mirror of
https://github.com/Ylianst/MeshCentralRouter
synced 2025-12-06 00:13:33 +00:00
First working remote desktop viewer.
This commit is contained in:
Ylian Saint-Hilaire
549
KVMViewer.cs
549
KVMViewer.cs
@@ -34,6 +34,8 @@ namespace MeshCentralRouter
|
||||
private MeshCentralServer server = null;
|
||||
private NodeClass node = null;
|
||||
private int state = 0;
|
||||
private RandomNumberGenerator rand = RandomNumberGenerator.Create();
|
||||
private string randomIdHex = null;
|
||||
public webSocketClient wc = null;
|
||||
|
||||
public KVMViewer(MeshCentralServer server, NodeClass node)
|
||||
@@ -83,22 +85,51 @@ namespace MeshCentralRouter
|
||||
public void MenuItemConnect_Click(object sender, EventArgs e)
|
||||
{
|
||||
if (wc != null) return;
|
||||
byte[] randomid = new byte[10];
|
||||
rand.GetBytes(randomid);
|
||||
randomIdHex = BitConverter.ToString(randomid).Replace("-", string.Empty);
|
||||
|
||||
state = 1;
|
||||
string randomId = "abc"; // TODO
|
||||
Uri u = new Uri(server.wsurl.ToString().Replace("/control.ashx", "/") + "meshrelay.ashx?browser=1&p=2&nodeid=" + node.nodeid + "&id=" + randomId + "&auth=" + server.authCookie);
|
||||
Uri u = new Uri(server.wsurl.ToString().Replace("/control.ashx", "/") + "meshrelay.ashx?browser=1&p=2&nodeid=" + node.nodeid + "&id=" + randomIdHex + "&auth=" + server.authCookie);
|
||||
wc = new webSocketClient();
|
||||
wc.Start(this, u, server.wshash);
|
||||
wc.onStateChanged += Wc_onStateChanged;
|
||||
wc.onBinaryData += Wc_onBinaryData;
|
||||
wc.onStringData += Wc_onStringData;
|
||||
wc.Start(u, server.wshash);
|
||||
}
|
||||
|
||||
public void onWebSocketConnected()
|
||||
private void Wc_onStateChanged(webSocketClient.ConnectionStates wsstate)
|
||||
{
|
||||
string randomId = "abc"; // TODO
|
||||
string u = "*/meshrelay.ashx?p=2&nodeid=" + node.nodeid + "&id=" + randomId + "&rauth=" + server.rauthCookie;
|
||||
server.sendCommand("{ \"action\": \"msg\", \"type\": \"tunnel\", \"nodeid\": \"" + node.nodeid + "\", \"value\": \"" + u.ToString() + "\", \"usage\": 2 }");
|
||||
switch (wsstate)
|
||||
{
|
||||
case webSocketClient.ConnectionStates.Disconnected:
|
||||
{
|
||||
// Disconnect
|
||||
state = 0;
|
||||
wc.Dispose();
|
||||
wc = null;
|
||||
break;
|
||||
}
|
||||
case webSocketClient.ConnectionStates.Connecting:
|
||||
{
|
||||
state = 1;
|
||||
break;
|
||||
}
|
||||
case webSocketClient.ConnectionStates.Connected:
|
||||
{
|
||||
state = 2;
|
||||
string u = "*/meshrelay.ashx?p=2&nodeid=" + node.nodeid + "&id=" + randomIdHex + "&rauth=" + server.rauthCookie;
|
||||
server.sendCommand("{ \"action\": \"msg\", \"type\": \"tunnel\", \"nodeid\": \"" + node.nodeid + "\", \"value\": \"" + u.ToString() + "\", \"usage\": 2 }");
|
||||
break;
|
||||
}
|
||||
}
|
||||
UpdateStatus();
|
||||
}
|
||||
|
||||
public void processServerData(string data) {
|
||||
if ((state == 1) && ((data == "c") || (data == "cr"))) {
|
||||
private void Wc_onStringData(string data)
|
||||
{
|
||||
if ((state == 2) && ((data == "c") || (data == "cr")))
|
||||
{
|
||||
state = 3;
|
||||
kvmControl.Send("2");
|
||||
kvmControl.SendCompressionLevel();
|
||||
@@ -110,10 +141,10 @@ namespace MeshCentralRouter
|
||||
if (state != 3) return;
|
||||
}
|
||||
|
||||
public void processServerBinaryData(byte[] data, int offset, int len)
|
||||
private void Wc_onBinaryData(byte[] data, int offset, int length)
|
||||
{
|
||||
if (state != 3) return;
|
||||
kvmControl.ProcessData(data, offset, len);
|
||||
kvmControl.ProcessData(data, offset, length);
|
||||
}
|
||||
|
||||
private void MenuItemDisconnect_Click(object sender, EventArgs e)
|
||||
@@ -125,7 +156,9 @@ namespace MeshCentralRouter
|
||||
wc.Dispose();
|
||||
wc = null;
|
||||
UpdateStatus();
|
||||
} else {
|
||||
}
|
||||
else
|
||||
{
|
||||
// Connect
|
||||
MenuItemConnect_Click(null, null);
|
||||
}
|
||||
@@ -136,7 +169,7 @@ namespace MeshCentralRouter
|
||||
|
||||
private void UpdateStatus()
|
||||
{
|
||||
if (this.InvokeRequired) { this.Invoke(new UpdateStatusHandler(UpdateStatus)); return; }
|
||||
if (this.InvokeRequired) { this.Invoke(new UpdateStatusHandler(UpdateStatus)); return; }
|
||||
|
||||
//if (kvmControl == null) return;
|
||||
switch (state)
|
||||
@@ -178,10 +211,15 @@ namespace MeshCentralRouter
|
||||
|
||||
private void Main_FormClosing(object sender, FormClosingEventArgs e)
|
||||
{
|
||||
if (wc != null)
|
||||
{
|
||||
// Disconnect
|
||||
state = 0;
|
||||
wc.Dispose();
|
||||
wc = null;
|
||||
UpdateStatus();
|
||||
}
|
||||
node.desktopViewer = null;
|
||||
//if (kvmControl != null) kvmControl.Disconnect();
|
||||
//controller.OnNodeStateUpdated -= new MeshControl.NodeUpdateHandler(controller_NodeStateUpdated);
|
||||
//controller.OnPolicyUpdated -= new MeshControl.PolicyUpdateHandler(controller_OnPolicyUpdated);
|
||||
}
|
||||
|
||||
private void statusToolStripMenuItem_CheckedChanged(object sender, EventArgs e)
|
||||
@@ -221,7 +259,8 @@ namespace MeshCentralRouter
|
||||
form.Compression = kvmControl.CompressionLevel;
|
||||
form.Scaling = kvmControl.ScalingLevel;
|
||||
form.FrameRate = kvmControl.FrameRate;
|
||||
if (form.ShowDialog(this) == System.Windows.Forms.DialogResult.OK) {
|
||||
if (form.ShowDialog(this) == System.Windows.Forms.DialogResult.OK)
|
||||
{
|
||||
kvmControl.SetCompressionParams(form.Compression, form.Scaling, form.FrameRate);
|
||||
}
|
||||
}
|
||||
@@ -333,481 +372,5 @@ namespace MeshCentralRouter
|
||||
try { return Registry.GetValue(@"HKEY_CURRENT_USER\SOFTWARE\OpenSource\MeshRouter", name, "").ToString(); } catch (Exception) { return ""; }
|
||||
}
|
||||
|
||||
public static string GetProxyForUrlUsingPac(string DestinationUrl, string PacUri)
|
||||
{
|
||||
IntPtr WinHttpSession = Win32Api.WinHttpOpen("User", Win32Api.WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, IntPtr.Zero, IntPtr.Zero, 0);
|
||||
|
||||
Win32Api.WINHTTP_AUTOPROXY_OPTIONS ProxyOptions = new Win32Api.WINHTTP_AUTOPROXY_OPTIONS();
|
||||
Win32Api.WINHTTP_PROXY_INFO ProxyInfo = new Win32Api.WINHTTP_PROXY_INFO();
|
||||
|
||||
ProxyOptions.dwFlags = Win32Api.WINHTTP_AUTOPROXY_CONFIG_URL;
|
||||
ProxyOptions.dwAutoDetectFlags = (Win32Api.WINHTTP_AUTO_DETECT_TYPE_DHCP | Win32Api.WINHTTP_AUTO_DETECT_TYPE_DNS_A);
|
||||
ProxyOptions.lpszAutoConfigUrl = PacUri;
|
||||
|
||||
// Get Proxy
|
||||
bool IsSuccess = Win32Api.WinHttpGetProxyForUrl(WinHttpSession, DestinationUrl, ref ProxyOptions, ref ProxyInfo);
|
||||
Win32Api.WinHttpCloseHandle(WinHttpSession);
|
||||
|
||||
if (IsSuccess)
|
||||
{
|
||||
return ProxyInfo.lpszProxy;
|
||||
}
|
||||
else
|
||||
{
|
||||
Console.WriteLine("Error: {0}", Win32Api.GetLastError());
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public class webSocketClient : IDisposable
|
||||
{
|
||||
private KVMViewer parent = null;
|
||||
private TcpClient wsclient = null;
|
||||
private SslStream wsstream = null;
|
||||
private NetworkStream wsrawstream = null;
|
||||
private int state = 0;
|
||||
private Uri url = null;
|
||||
private byte[] readBuffer = new Byte[500];
|
||||
private int readBufferLen = 0;
|
||||
private int accopcodes = 0;
|
||||
private bool accmask = false;
|
||||
private int acclen = 0;
|
||||
private bool proxyInUse = false;
|
||||
private string user = null;
|
||||
private string pass = null;
|
||||
private string token = null;
|
||||
public bool xdebug = false;
|
||||
public bool xignoreCert = false;
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
try { wsstream.Close(); } catch (Exception) { }
|
||||
try { wsstream.Dispose(); } catch (Exception) { }
|
||||
wsstream = null;
|
||||
wsclient = null;
|
||||
state = -1;
|
||||
}
|
||||
|
||||
public void Debug(string msg) { if (xdebug) { try { File.AppendAllText("debug.log", "Debug-" + msg + "\r\n"); } catch (Exception) { } } }
|
||||
|
||||
public bool Start(KVMViewer parent, Uri url, string fingerprint)
|
||||
{
|
||||
if (state != 0) return false;
|
||||
state = 1;
|
||||
this.url = url;
|
||||
this.parent = parent;
|
||||
Uri proxyUri = null;
|
||||
|
||||
// Check if we need to use a HTTP proxy (Auto-proxy way)
|
||||
try
|
||||
{
|
||||
RegistryKey registryKey = Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings", true);
|
||||
Object x = registryKey.GetValue("AutoConfigURL", null);
|
||||
if ((x != null) && (x.GetType() == typeof(string)))
|
||||
{
|
||||
string proxyStr = GetProxyForUrlUsingPac("http" + ((url.Port == 80) ? "" : "s") + "://" + url.Host + ":" + url.Port, x.ToString());
|
||||
if (proxyStr != null) { proxyUri = new Uri("http://" + proxyStr); }
|
||||
}
|
||||
}
|
||||
catch (Exception) { proxyUri = null; }
|
||||
|
||||
// Check if we need to use a HTTP proxy (Normal way)
|
||||
if (proxyUri == null)
|
||||
{
|
||||
var proxy = System.Net.HttpWebRequest.GetSystemWebProxy();
|
||||
proxyUri = proxy.GetProxy(url);
|
||||
if ((url.Host.ToLower() == proxyUri.Host.ToLower()) && (url.Port == proxyUri.Port)) { proxyUri = null; }
|
||||
}
|
||||
|
||||
if (proxyUri != null)
|
||||
{
|
||||
// Proxy in use
|
||||
proxyInUse = true;
|
||||
wsclient = new TcpClient();
|
||||
wsclient.BeginConnect(proxyUri.Host, proxyUri.Port, new AsyncCallback(OnConnectSink), this);
|
||||
}
|
||||
else
|
||||
{
|
||||
// No proxy in use
|
||||
proxyInUse = false;
|
||||
wsclient = new TcpClient();
|
||||
wsclient.BeginConnect(url.Host, url.Port, new AsyncCallback(OnConnectSink), this);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
private void OnConnectSink(IAsyncResult ar)
|
||||
{
|
||||
if (wsclient == null) return;
|
||||
|
||||
// Accept the connection
|
||||
try
|
||||
{
|
||||
wsclient.EndConnect(ar);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Debug("Websocket TCP failed to connect: " + ex.ToString());
|
||||
Dispose();
|
||||
return;
|
||||
}
|
||||
|
||||
if (proxyInUse == true)
|
||||
{
|
||||
// Send proxy connection request
|
||||
wsrawstream = wsclient.GetStream();
|
||||
byte[] proxyRequestBuf = UTF8Encoding.UTF8.GetBytes("CONNECT " + url.Host + ":" + url.Port + " HTTP/1.1\r\nHost: " + url.Host + ":" + url.Port + "\r\n\r\n");
|
||||
wsrawstream.Write(proxyRequestBuf, 0, proxyRequestBuf.Length);
|
||||
wsrawstream.BeginRead(readBuffer, readBufferLen, readBuffer.Length - readBufferLen, new AsyncCallback(OnProxyResponseSink), this);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Start TLS connection
|
||||
Debug("Websocket TCP connected, doing TLS...");
|
||||
wsstream = new SslStream(wsclient.GetStream(), false, VerifyServerCertificate, null);
|
||||
wsstream.BeginAuthenticateAsClient(url.Host, null, System.Security.Authentication.SslProtocols.Tls12, false, new AsyncCallback(OnTlsSetupSink), this);
|
||||
}
|
||||
}
|
||||
|
||||
private void OnProxyResponseSink(IAsyncResult ar)
|
||||
{
|
||||
if (wsrawstream == null) return;
|
||||
|
||||
int len = 0;
|
||||
try { len = wsrawstream.EndRead(ar); } catch (Exception) { }
|
||||
if (len == 0)
|
||||
{
|
||||
// Disconnect
|
||||
Debug("Websocket proxy disconnected, length = 0.");
|
||||
Dispose();
|
||||
return;
|
||||
}
|
||||
|
||||
readBufferLen += len;
|
||||
string proxyResponse = UTF8Encoding.UTF8.GetString(readBuffer, 0, readBufferLen);
|
||||
if (proxyResponse.IndexOf("\r\n\r\n") >= 0)
|
||||
{
|
||||
// We get a full proxy response, we should get something like "HTTP/1.1 200 Connection established\r\n\r\n"
|
||||
if (proxyResponse.StartsWith("HTTP/1.1 200 "))
|
||||
{
|
||||
// All good, start TLS setup.
|
||||
readBufferLen = 0;
|
||||
Debug("Websocket TCP connected, doing TLS...");
|
||||
wsstream = new SslStream(wsrawstream, false, VerifyServerCertificate, null);
|
||||
wsstream.BeginAuthenticateAsClient(url.Host, null, System.Security.Authentication.SslProtocols.Tls12, false, new AsyncCallback(OnTlsSetupSink), this);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Invalid response
|
||||
Debug("Proxy connection failed: " + proxyResponse);
|
||||
Dispose();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (readBufferLen == readBuffer.Length)
|
||||
{
|
||||
// Buffer overflow
|
||||
Debug("Proxy connection failed");
|
||||
Dispose();
|
||||
}
|
||||
else
|
||||
{
|
||||
// Read more proxy data
|
||||
wsrawstream.BeginRead(readBuffer, readBufferLen, readBuffer.Length - readBufferLen, new AsyncCallback(OnProxyResponseSink), this);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public string Base64Encode(string plainText)
|
||||
{
|
||||
var plainTextBytes = System.Text.Encoding.UTF8.GetBytes(plainText);
|
||||
return System.Convert.ToBase64String(plainTextBytes);
|
||||
}
|
||||
|
||||
public string Base64Decode(string base64EncodedData)
|
||||
{
|
||||
var base64EncodedBytes = System.Convert.FromBase64String(base64EncodedData);
|
||||
return System.Text.Encoding.UTF8.GetString(base64EncodedBytes);
|
||||
}
|
||||
|
||||
private void OnTlsSetupSink(IAsyncResult ar)
|
||||
{
|
||||
if (wsstream == null) return;
|
||||
|
||||
// Accept the connection
|
||||
try
|
||||
{
|
||||
wsstream.EndAuthenticateAsClient(ar);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
// Disconnect
|
||||
MessageBox.Show(ex.Message, "MeshRouter");
|
||||
Debug("Websocket TLS failed: " + ex.ToString());
|
||||
Dispose();
|
||||
return;
|
||||
}
|
||||
|
||||
// Fetch remote certificate
|
||||
//parent.wshash = wsstream.RemoteCertificate.GetCertHashString();
|
||||
|
||||
// Setup extra headers if needed
|
||||
string extraHeaders = "";
|
||||
if (user != null && pass != null && token != null) { extraHeaders = "x-meshauth: " + Base64Encode(user) + "," + Base64Encode(pass) + "," + Base64Encode(token) + "\r\n"; }
|
||||
else if (user != null && pass != null) { extraHeaders = "x-meshauth: " + Base64Encode(user) + "," + Base64Encode(pass) + "\r\n"; }
|
||||
|
||||
// Send the HTTP headers
|
||||
Debug("Websocket TLS setup, sending HTTP header...");
|
||||
string header = "GET " + url.PathAndQuery + " HTTP/1.1\r\nHost: " + url.Host + "\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\nSec-WebSocket-Version: 13\r\n" + extraHeaders + "\r\n";
|
||||
wsstream.Write(UTF8Encoding.UTF8.GetBytes(header));
|
||||
|
||||
// Start receiving data
|
||||
wsstream.BeginRead(readBuffer, readBufferLen, readBuffer.Length - readBufferLen, new AsyncCallback(OnTlsDataSink), this);
|
||||
}
|
||||
|
||||
private void OnTlsDataSink(IAsyncResult ar)
|
||||
{
|
||||
if (wsstream == null) return;
|
||||
|
||||
int len = 0;
|
||||
try { len = wsstream.EndRead(ar); } catch (Exception) { }
|
||||
if (len == 0)
|
||||
{
|
||||
// Disconnect
|
||||
Debug("Websocket disconnected, length = 0.");
|
||||
Dispose();
|
||||
return;
|
||||
}
|
||||
//parent.Debug("#" + counter + ": Websocket got new data: " + len);
|
||||
readBufferLen += len;
|
||||
|
||||
// Consume all of the data
|
||||
int consumed = 0;
|
||||
int ptr = 0;
|
||||
do
|
||||
{
|
||||
consumed = ProcessBuffer(readBuffer, ptr, readBufferLen - ptr);
|
||||
if (consumed < 0) { Dispose(); return; } // Error, close the connection
|
||||
ptr += consumed;
|
||||
} while ((consumed > 0) && ((readBufferLen - consumed) > 0));
|
||||
|
||||
// Move the data forward
|
||||
if ((ptr > 0) && (readBufferLen - ptr) > 0)
|
||||
{
|
||||
//Console.Write("MOVE FORWARD\r\n");
|
||||
Array.Copy(readBuffer, ptr, readBuffer, 0, (readBufferLen - ptr));
|
||||
}
|
||||
readBufferLen = (readBufferLen - ptr);
|
||||
|
||||
// If the buffer is too small, double the size here.
|
||||
if (readBuffer.Length - readBufferLen == 0)
|
||||
{
|
||||
Debug("Increasing the read buffer size from " + readBuffer.Length + " to " + (readBuffer.Length * 2) + ".");
|
||||
byte[] readBuffer2 = new byte[readBuffer.Length * 2];
|
||||
Array.Copy(readBuffer, 0, readBuffer2, 0, readBuffer.Length);
|
||||
readBuffer = readBuffer2;
|
||||
}
|
||||
|
||||
// Receive more data
|
||||
try { wsstream.BeginRead(readBuffer, readBufferLen, readBuffer.Length - readBufferLen, new AsyncCallback(OnTlsDataSink), this); } catch (Exception) { }
|
||||
}
|
||||
|
||||
private int ProcessBuffer(byte[] buffer, int offset, int len)
|
||||
{
|
||||
string ss = UTF8Encoding.UTF8.GetString(buffer, offset, len);
|
||||
|
||||
if (state == 1)
|
||||
{
|
||||
// Look for the end of the http header
|
||||
string header = UTF8Encoding.UTF8.GetString(buffer, offset, len);
|
||||
int i = header.IndexOf("\r\n\r\n");
|
||||
if (i == -1) return 0;
|
||||
Dictionary<string, string> parsedHeader = ParseHttpHeader(header.Substring(0, i));
|
||||
if ((parsedHeader == null) || (parsedHeader["_Path"] != "101")) { Debug("Websocket bad header."); return -1; } // Bad header, close the connection
|
||||
Debug("Websocket got setup upgrade header.");
|
||||
state = 2;
|
||||
this.parent.onWebSocketConnected();
|
||||
return len; // TODO: Technically we need to return the header length before UTF8 convert.
|
||||
}
|
||||
else if (state == 2)
|
||||
{
|
||||
// Parse a websocket fragment header
|
||||
if (len < 2) return 0;
|
||||
int headsize = 2;
|
||||
accopcodes = buffer[offset];
|
||||
accmask = ((buffer[offset + 1] & 0x80) != 0);
|
||||
acclen = (buffer[offset + 1] & 0x7F);
|
||||
|
||||
if ((accopcodes & 0x0F) == 8)
|
||||
{
|
||||
// Close the websocket
|
||||
Debug("Websocket got closed fragment.");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (acclen == 126)
|
||||
{
|
||||
if (len < 4) return 0;
|
||||
headsize = 4;
|
||||
acclen = (buffer[offset + 2] << 8) + (buffer[offset + 3]);
|
||||
}
|
||||
else if (acclen == 127)
|
||||
{
|
||||
if (len < 10) return 0;
|
||||
headsize = 10;
|
||||
acclen = (buffer[offset + 6] << 24) + (buffer[offset + 7] << 16) + (buffer[offset + 8] << 8) + (buffer[offset + 9]);
|
||||
Debug("Websocket receive large fragment: " + acclen);
|
||||
}
|
||||
if (accmask == true)
|
||||
{
|
||||
// TODO: Do unmasking here.
|
||||
headsize += 4;
|
||||
}
|
||||
//parent.Debug("#" + counter + ": Websocket frag header - FIN: " + ((accopcodes & 0x80) != 0) + ", OP: " + (accopcodes & 0x0F) + ", LEN: " + acclen + ", MASK: " + accmask);
|
||||
state = 3;
|
||||
return headsize;
|
||||
}
|
||||
else if (state == 3)
|
||||
{
|
||||
// Parse a websocket fragment data
|
||||
if (len < acclen) return 0;
|
||||
//Console.Write("WSREAD: " + acclen + "\r\n");
|
||||
ProcessWsBuffer(buffer, offset, acclen, accopcodes);
|
||||
state = 2;
|
||||
return acclen;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
private void ProcessWsBuffer(byte[] data, int offset, int len, int op)
|
||||
{
|
||||
bool binary = ((op & 1) == 0);
|
||||
|
||||
Debug("Websocket got data.");
|
||||
//try { parent.processServerData(UTF8Encoding.UTF8.GetString(data, offset, len)); } catch (Exception ex) { }
|
||||
|
||||
if (binary == false) {
|
||||
parent.processServerData(UTF8Encoding.UTF8.GetString(data, offset, len));
|
||||
} else {
|
||||
parent.processServerBinaryData(data, offset, len);
|
||||
}
|
||||
}
|
||||
|
||||
private Dictionary<string, string> ParseHttpHeader(string header)
|
||||
{
|
||||
string[] lines = header.Replace("\r\n", "\r").Split('\r');
|
||||
if (lines.Length < 2) { return null; }
|
||||
string[] directive = lines[0].Split(' ');
|
||||
Dictionary<string, string> values = new Dictionary<string, string>();
|
||||
values["_Action"] = directive[0];
|
||||
values["_Path"] = directive[1];
|
||||
values["_Protocol"] = directive[2];
|
||||
for (int i = 1; i < lines.Length; i++)
|
||||
{
|
||||
var j = lines[i].IndexOf(":");
|
||||
values[lines[i].Substring(0, j).ToLower()] = lines[i].Substring(j + 1).Trim();
|
||||
}
|
||||
return values;
|
||||
}
|
||||
|
||||
// Return a modified base64 SHA384 hash string of the certificate public key
|
||||
public static string GetMeshKeyHash(X509Certificate cert)
|
||||
{
|
||||
return ByteArrayToHexString(new SHA384Managed().ComputeHash(cert.GetPublicKey()));
|
||||
}
|
||||
|
||||
// Return a modified base64 SHA384 hash string of the certificate
|
||||
public static string GetMeshCertHash(X509Certificate cert)
|
||||
{
|
||||
return ByteArrayToHexString(new SHA384Managed().ComputeHash(cert.GetRawCertData()));
|
||||
}
|
||||
|
||||
public static string ByteArrayToHexString(byte[] Bytes)
|
||||
{
|
||||
StringBuilder Result = new StringBuilder(Bytes.Length * 2);
|
||||
string HexAlphabet = "0123456789ABCDEF";
|
||||
foreach (byte B in Bytes) { Result.Append(HexAlphabet[(int)(B >> 4)]); Result.Append(HexAlphabet[(int)(B & 0xF)]); }
|
||||
return Result.ToString();
|
||||
}
|
||||
|
||||
private bool VerifyServerCertificate(object sender, X509Certificate certificate, X509Chain chain, SslPolicyErrors sslPolicyErrors)
|
||||
{
|
||||
return (parent.server.certHash == GetMeshKeyHash(certificate));
|
||||
}
|
||||
|
||||
public void WriteStringWebSocket(string data)
|
||||
{
|
||||
if (state < 2) return;
|
||||
|
||||
// Convert the string into a buffer with 4 byte of header space.
|
||||
int len = UTF8Encoding.UTF8.GetByteCount(data);
|
||||
byte[] buf = new byte[4 + len];
|
||||
UTF8Encoding.UTF8.GetBytes(data, 0, data.Length, buf, 4);
|
||||
len = buf.Length - 4;
|
||||
|
||||
// Check that everything is ok
|
||||
if ((state < 2) || (len < 1) || (len > 65535)) { Dispose(); return; }
|
||||
|
||||
//Console.Write("Length: " + len + "\r\n");
|
||||
//System.Threading.Thread.Sleep(0);
|
||||
|
||||
if (len < 126)
|
||||
{
|
||||
// Small fragment
|
||||
buf[2] = 129; // Fragment op code (129 = text, 130 = binary)
|
||||
buf[3] = (byte)(len & 0x7F);
|
||||
//try { wsstream.BeginWrite(buf, 2, len + 2, new AsyncCallback(WriteWebSocketAsyncDone), args); } catch (Exception) { Dispose(); return; }
|
||||
wsstream.Write(buf, 2, len + 2);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Large fragment
|
||||
buf[0] = 129; // Fragment op code (129 = text, 130 = binary)
|
||||
buf[1] = 126;
|
||||
buf[2] = (byte)((len >> 8) & 0xFF);
|
||||
buf[3] = (byte)(len & 0xFF);
|
||||
//try { wsstream.BeginWrite(buf, 0, len + 4, new AsyncCallback(WriteWebSocketAsyncDone), args); } catch (Exception) { Dispose(); return; }
|
||||
wsstream.Write(buf, 0, len + 4);
|
||||
}
|
||||
}
|
||||
|
||||
public void WriteBinaryWebSocket(byte[] data, int offset, int len)
|
||||
{
|
||||
if (state < 2) return;
|
||||
|
||||
// Convert the string into a buffer with 4 byte of header space.
|
||||
byte[] buf = new byte[4 + len];
|
||||
Array.Copy(data, offset, buf, 4, len);
|
||||
len = buf.Length - 4;
|
||||
|
||||
// Check that everything is ok
|
||||
if ((state < 2) || (len < 1) || (len > 65535)) { Dispose(); return; }
|
||||
|
||||
//Console.Write("Length: " + len + "\r\n");
|
||||
//System.Threading.Thread.Sleep(0);
|
||||
|
||||
if (len < 126)
|
||||
{
|
||||
// Small fragment
|
||||
buf[2] = 130; // Fragment op code (129 = text, 130 = binary)
|
||||
buf[3] = (byte)(len & 0x7F);
|
||||
//try { wsstream.BeginWrite(buf, 2, len + 2, new AsyncCallback(WriteWebSocketAsyncDone), args); } catch (Exception) { Dispose(); return; }
|
||||
wsstream.Write(buf, 2, len + 2);
|
||||
}
|
||||
else
|
||||
{
|
||||
// Large fragment
|
||||
buf[0] = 130; // Fragment op code (129 = text, 130 = binary)
|
||||
buf[1] = 126;
|
||||
buf[2] = (byte)((len >> 8) & 0xFF);
|
||||
buf[3] = (byte)(len & 0xFF);
|
||||
//try { wsstream.BeginWrite(buf, 0, len + 4, new AsyncCallback(WriteWebSocketAsyncDone), args); } catch (Exception) { Dispose(); return; }
|
||||
wsstream.Write(buf, 0, len + 4);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user