melonvpn-original-cs/MelonVPNCore/DaemonSocketServer.cs

196 lines
7.6 KiB
C#
Raw Normal View History

2020-09-22 23:10:16 +01:00
using System;
using System.Diagnostics;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
namespace MelonVPNCore
{
public static class DaemonSocketServer
{
2021-03-08 16:55:26 +00:00
private static Process currentVpnProcess = null;
private static bool shouldBeRunning = false;
2020-09-22 23:10:16 +01:00
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");
2021-03-08 16:55:26 +00:00
if (isProcessOnline(currentVpnProcess) || shouldBeRunning)
2020-09-22 23:10:16 +01:00
{
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);
2020-09-23 20:24:30 +01:00
Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
2020-09-22 23:10:16 +01:00
}
}
else if (data == Messages.StartMsg)
{
2021-03-08 16:55:26 +00:00
if (! isProcessOnline(currentVpnProcess))
2020-09-22 23:10:16 +01:00
{
2021-03-08 16:55:26 +00:00
shouldBeRunning = true;
2020-09-22 23:10:16 +01:00
Console.WriteLine("Starting VPN");
try
{
Console.WriteLine("Starting embedded process");
2021-03-08 16:55:26 +00:00
StartEProcess(true);
2020-09-22 23:10:16 +01:00
Console.WriteLine("Sending online reply");
Thread.Sleep(1500);
Client.SendDataMessage(DataMessage.Online, true);
2021-03-08 16:55:26 +00:00
currentVpnProcess.EnableRaisingEvents = true;
2020-09-22 23:10:16 +01:00
}
catch (Exception e)
{
2021-03-08 16:55:26 +00:00
shouldBeRunning = false;
2020-09-22 23:10:16 +01:00
Console.WriteLine("There was an error. But I fixed it.");
Console.WriteLine("It looked like this: " + e);
currentVpnProcess = null;
Client.SendDataMessage(DataMessage.Error, true);
2020-09-23 20:24:30 +01:00
Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
2020-09-22 23:10:16 +01:00
}
}
else
{
Console.WriteLine("VPN already started");
}
}
else if (data == Messages.StopMsg)
{
2021-03-08 16:55:26 +00:00
shouldBeRunning = false;
bool haderr = false;
if (isProcessOnline(currentVpnProcess))
2020-09-22 23:10:16 +01:00
{
Console.WriteLine("Stopping VPN");
try
{
2021-03-08 16:55:26 +00:00
currentVpnProcess.EnableRaisingEvents = false;
2020-09-22 23:10:16 +01:00
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);
2020-09-23 20:24:30 +01:00
Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
2021-03-08 16:55:26 +00:00
haderr = true;
2020-09-22 23:10:16 +01:00
}
}
else
{
Console.WriteLine("VPN already stopped");
}
2021-03-08 16:55:26 +00:00
if (!haderr)
{
Console.WriteLine("Sending offline reply");
Client.SendDataMessage(DataMessage.Offline, true);
Client.SendCustomMessage(Messages.ClientListEmptyMsg, true);
}
2020-09-22 23:10:16 +01:00
}
Client.SendDataMessage(DataMessage.Blank, true);
handler.Shutdown(SocketShutdown.Both);
handler.Close();
}
}
catch (Exception e)
{
Console.WriteLine(e);
}
}
2021-03-08 16:55:26 +00:00
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; }
}
}
}
2020-09-22 23:10:16 +01:00
}
}