Novetus_src/Novetus/NovetusCore/StorageAndFunctions/NetFuncs.cs

132 lines
5.0 KiB
C#

#if !BASICLAUNCHER
#region Usings
using Mono.Nat;
using System;
using System.Collections.Generic;
using System.Web;
using Titanium.Web.Proxy.Models;
#endregion
namespace Novetus.Core
{
#region NetFuncs
public static class NetFuncs
{
public static void InitUPnP()
{
if (GlobalVars.UserConfiguration.ReadSettingBool("UPnP"))
{
try
{
NatUtility.DeviceFound += DeviceFound;
NatUtility.StartDiscovery();
Util.ConsolePrint("UPnP: Service initialized", 3);
}
catch (Exception ex)
{
Util.LogExceptions(ex);
Util.ConsolePrint("UPnP: Unable to initialize UPnP. Reason - " + ex.Message, 2);
}
}
}
public static void DeviceFound(object sender, DeviceEventArgs args)
{
try
{
INatDevice device = args.Device;
string IP = !string.IsNullOrWhiteSpace(GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP")) ? GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP") : device.GetExternalIP().ToString();
Util.ConsolePrint("UPnP: Device '" + IP + "' registered.", 3);
StartUPnP(device, Protocol.Udp, GlobalVars.UserConfiguration.ReadSettingInt("RobloxPort"));
StartUPnP(device, Protocol.Tcp, GlobalVars.UserConfiguration.ReadSettingInt("RobloxPort"));
}
catch (Exception ex)
{
Util.LogExceptions(ex);
Util.ConsolePrint("UPnP: Unable to register device. Reason - " + ex.Message, 2);
}
}
public static void StartUPnP(INatDevice device, Protocol protocol, int port)
{
if (GlobalVars.UserConfiguration.ReadSettingBool("UPnP"))
{
try
{
Mapping checker = device.GetSpecificMapping(protocol, port);
int mapPublic = checker.PublicPort;
int mapPrivate = checker.PrivatePort;
if (mapPublic == -1 && mapPrivate == -1)
{
Mapping portmap = new Mapping(protocol, port, port);
device.CreatePortMap(portmap);
}
string IP = !string.IsNullOrWhiteSpace(GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP")) ? GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP") : device.GetExternalIP().ToString();
Util.ConsolePrint("UPnP: Port " + port + " opened on '" + IP + "' (" + protocol.ToString() + ")", 3);
}
catch (Exception ex)
{
Util.LogExceptions(ex);
Util.ConsolePrint("UPnP: Unable to open port mapping. Reason - " + ex.Message, 2);
}
}
}
public static void StopUPnP(INatDevice device, Protocol protocol, int port)
{
if (GlobalVars.UserConfiguration.ReadSettingBool("UPnP"))
{
try
{
Mapping checker = device.GetSpecificMapping(protocol, port);
int mapPublic = checker.PublicPort;
int mapPrivate = checker.PrivatePort;
if (mapPublic != -1 && mapPrivate != -1)
{
Mapping portmap = new Mapping(protocol, port, port);
device.DeletePortMap(portmap);
}
string IP = !string.IsNullOrWhiteSpace(GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP")) ? GlobalVars.UserConfiguration.ReadSetting("AlternateServerIP") : device.GetExternalIP().ToString();
Util.ConsolePrint("UPnP: Port " + port + " closed on '" + IP + "' (" + protocol.ToString() + ")", 3);
}
catch (Exception ex)
{
Util.LogExceptions(ex);
Util.ConsolePrint("UPnP: Unable to close port mapping. Reason - " + ex.Message, 2);
}
}
}
public static string FindQueryString(Uri uri, string searchQuery)
{
return FindQueryString(uri.Query, searchQuery);
}
public static string FindQueryString(string query, string searchQuery)
{
return HttpUtility.ParseQueryString(query)[searchQuery];
}
public static IEnumerable<HttpHeader> GenerateHeaders(string content, string contenttype = "")
{
List<HttpHeader> HeaderList = new List<HttpHeader>();
if (!string.IsNullOrWhiteSpace(contenttype))
{
HeaderList.Add(new HttpHeader("Content-Type", contenttype));
}
HeaderList.Add(new HttpHeader("Content-Length", content));
HeaderList.Add(new HttpHeader("Cache-Control", "no-cache"));
return HeaderList;
}
}
#endregion
}
#endif