OpenCodeBox/fishbait
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update Program.cs

Browse source
This commit is contained in:
cxxpxr 2021-04-05 13:57:57 -04:00 committed by GitHub
parent aff75b9707
commit d6f0d9ca63
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
ServerProject-DONT-IMPORT-INTO-UNITY/LRM/Program.cs
View file

@ -1,4 +1,4 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.IO; using System.IO;
using System.Linq; using System.Linq;
@ -32,6 +32,7 @@ namespace LightReflectiveMirror
private BiDictionary<int, string> _pendingNATPunches = new BiDictionary<int, string>(); private BiDictionary<int, string> _pendingNATPunches = new BiDictionary<int, string>();
private int _currentHeartbeatTimer = 0; private int _currentHeartbeatTimer = 0;
private string _externalIp;
private byte[] _NATRequest = new byte[500]; private byte[] _NATRequest = new byte[500];
private int _NATRequestPosition = 0; private int _NATRequestPosition = 0;
@ -62,6 +63,7 @@ namespace LightReflectiveMirror
else else
{ {
conf = JsonConvert.DeserializeObject<Config>(File.ReadAllText(CONFIG_PATH)); conf = JsonConvert.DeserializeObject<Config>(File.ReadAllText(CONFIG_PATH));
_externalIp = await GetExternalIp();
WriteLogMessage("Loading Assembly... ", ConsoleColor.White, true); WriteLogMessage("Loading Assembly... ", ConsoleColor.White, true);
try try
@ -218,22 +220,23 @@ namespace LightReflectiveMirror
async Task<bool> RegisterSelfToLoadBalancer() async Task<bool> RegisterSelfToLoadBalancer()
{ {
// replace hard coded value for config value later
var url = "http://www.localhost:7070/api/auth";
string externalip = "127.0.0.1";//new WebClient().DownloadString("https://ipv4.icanhazip.com/");
string port = conf.EndpointPort.ToString();
HttpWebRequest myRequest = (HttpWebRequest)WebRequest.Create(url);
myRequest.Headers.Add("Auth", "AuthKey");
myRequest.Headers.Add("Address", $"{externalip}:" + port + "/api/stats");
try try
{ {
// replace hard coded value for config value later
var uri = new Uri("http://localhost:7070/api/auth");
string externalip = _externalIp.Normalize().Trim();
string port = conf.EndpointPort.ToString();
HttpWebRequest myRequest = (HttpWebRequest)WebRequest.Create(uri);
myRequest.Headers.Add("Auth", "AuthKey");
myRequest.Headers.Add("Address", externalip + ":" + port);
WebResponse myResponse = await myRequest.GetResponseAsync(); WebResponse myResponse = await myRequest.GetResponseAsync();
return true; return true;
} }
catch catch (Exception e)
{ {
// error adding or load balancer unavailable // error adding or load balancer unavailable
WriteLogMessage("Error registering", ConsoleColor.Red); WriteLogMessage("Error registering", ConsoleColor.Red);
@ -242,10 +245,21 @@ namespace LightReflectiveMirror
} }
async Task<string> GetExternalIp()
{
HttpWebRequest myRequest = (HttpWebRequest)WebRequest.Create("https://ipv4.icanhazip.com/");
WebResponse myResponse = await myRequest.GetResponseAsync();
Stream stream = myResponse.GetResponseStream();
var ip = new StreamReader(stream).ReadToEnd();
return ip;
}
void RunNATPunchLoop() void RunNATPunchLoop()
{ {
WriteLogMessage("OK\n", ConsoleColor.Green); WriteLogMessage("OK\n", ConsoleColor.Green);
IPEndPoint remoteEndpoint = new IPEndPoint(IPAddress.Any, conf.NATPunchtroughPort); IPEndPoint remoteEndpoint = new(IPAddress.Any, conf.NATPunchtroughPort);
// Stock Data server sends to everyone: // Stock Data server sends to everyone:
var serverResponse = new byte[1] { 1 }; var serverResponse = new byte[1] { 1 };