melonvpn-original-cs/MelonVPNCore/DaemonSocketServer.cs

using System;
using System.Diagnostics;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;

namespace MelonVPNCore
{
    public static class DaemonSocketServer
    {
        private static Process currentVpnProcess = null;
        private static bool shouldBeRunning = false;

        public static void StartServer()
        {
            IPHostEntry host = Dns.GetHostEntry("localhost");
            IPAddress ipAddress = host.AddressList[0];
            IPEndPoint localEndPoint = new IPEndPoint(ipAddress, 22035);

            try
            {
                Socket listener = new Socket(ipAddress.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
                listener.Bind(localEndPoint);
                listener.Listen(10);

                string lastClientUpdate = Messages.EOF;

                while (true)
                {
                    Socket handler = listener.Accept();

                    string data = null;
                    byte[] bytes = null;

                    while (true)
                    {
                        bytes = new byte[1024];
                        int bytesRec = handler.Receive(bytes);
                        data += Encoding.ASCII.GetString(bytes, 0, bytesRec);
                        if (data.IndexOf(Messages.EOF, StringComparison.CurrentCulture) > -1) break;
                    }

                    if (data.StartsWith(Messages.ClientListStartMsg, StringComparison.CurrentCulture))
                    {
                        lastClientUpdate = data;
                        Client.SendCustomMessage(lastClientUpdate, true);
                    }
                    else if (data == Messages.StatusMsg)
                    {
                        Console.WriteLine("Status requested");
                        if (isProcessOnline(currentVpnProcess) || shouldBeRunning)
                        {
                            Console.WriteLine("Sending response: online");
                            Client.SendDataMessage(DataMessage.Online, true);
                            Client.SendCustomMessage(lastClientUpdate, true);
                        }
                        else
                        {
                            Console.WriteLine("Sending response: offline");
                            Client.SendDataMessage(DataMessage.Offline, true);
                            Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
                        }
                    }
                    else if (data == Messages.StartMsg)
                    {
                        if (! isProcessOnline(currentVpnProcess))
                        {
                            shouldBeRunning = true;
                            Console.WriteLine("Starting VPN");
                            try
                            {
                                Console.WriteLine("Starting embedded process");
                                StartEProcess(true);
                                Console.WriteLine("Sending online reply");
                                Thread.Sleep(1500);
                                Client.SendDataMessage(DataMessage.Online, true);
                                currentVpnProcess.EnableRaisingEvents = true;
                            }
                            catch (Exception e)
                            {
                                shouldBeRunning = false;
                                Console.WriteLine("There was an error. But I fixed it.");
                                Console.WriteLine("It looked like this: " + e);
                                currentVpnProcess = null;
                                Client.SendDataMessage(DataMessage.Error, true);
                                Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
                            }
                        }
                        else
                        {
                            Console.WriteLine("VPN already started");
                        }
                    }
                    else if (data == Messages.StopMsg)
                    {
                        shouldBeRunning = false;
                        bool haderr = false;
                        if (isProcessOnline(currentVpnProcess))
                        {
                            Console.WriteLine("Stopping VPN");
                            try
                            {
                                currentVpnProcess.EnableRaisingEvents = false;
                                Console.WriteLine("Stopping embedded process");
                                currentVpnProcess.Kill();
                                currentVpnProcess = null;
                            }
                            catch (Exception e)
                            {
                                Console.WriteLine("There was an error. But I fixed it.");
                                Console.WriteLine("It looked like this: " + e);
                                currentVpnProcess = null;
                                Client.SendDataMessage(DataMessage.Error, true);
                                Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
                                haderr = true;
                            }
                        }
                        else
                        {
                            Console.WriteLine("VPN already stopped");
                        }
                        if (!haderr)
                        {
                            Console.WriteLine("Sending offline reply");
                            Client.SendDataMessage(DataMessage.Offline, true);
                            Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
                        }
                    }

                    Client.SendDataMessage(DataMessage.Blank, true);

                    handler.Shutdown(SocketShutdown.Both);
                    handler.Close();
                }
            }
            catch (Exception e)
            {
                Console.WriteLine(e);
            }
        }

        public static bool isProcessOnline(Process p)
        {
            if (p == null)
            {
                return false;
            }
            else
            {
                if (p.HasExited)
                {
                    return false;
                }
                else
                {
                    return true;
                }
            }
        }

        static void CurrentVpnProcess_Exited(object sender, EventArgs e)
        {
            Console.WriteLine("Restarting embedded process");
            currentVpnProcess = null;
            Thread.Sleep(1500);
            StartEProcess(false);
        }

        static void StartEProcess(bool reraisee)
        {
            if (shouldBeRunning)
            {
                try
                {
                    currentVpnProcess = new Process
                    {
                        StartInfo = new ProcessStartInfo("simple-vpn", "client /etc/melon-vpn/client.cfg")
                    };
                    currentVpnProcess.Start();
                    currentVpnProcess.EnableRaisingEvents = false;
                    currentVpnProcess.Exited += CurrentVpnProcess_Exited;
                    Thread.Sleep(1500);
                    currentVpnProcess.EnableRaisingEvents = true;
                }
                catch (Exception e)
                {
                    Console.WriteLine("There was an error. But I fixed it.");
                    Console.WriteLine("It looked like this: " + e);
                    if (reraisee) { throw e; }
                }
            }
        }
    }
}