public ClientSession keyExchangeTcpServer(Socket s)
{
try
{
byte[] recvBuf = new byte[259];
byte[] command = new byte[4];
recvBuf = Networking.my_recv(259, s);
if (recvBuf != null)
{
byte[] modulus = new byte[256];
byte[] exponent = new byte[3];
MemoryStream ms = new MemoryStream(recvBuf);
ms.Read(modulus, 0, 256);
ms.Read(exponent, 0, 3);
RSACryptoServiceProvider rsa = Security.getPublicKey(modulus, exponent);
AesCryptoServiceProvider aes = Security.generateAESKey();
byte[] dataToEncrypt = new byte[48];
ms = new MemoryStream(dataToEncrypt);
ms.Write(aes.Key, 0, aes.Key.Length);
ms.Write(aes.IV, 0, aes.IV.Length);
byte[] encrypted = Security.RSAEncrypt(rsa, dataToEncrypt, false);
if (encrypted != null)
{
s.Send(encrypted);
command = Networking.my_recv(4, s);
if (command != null && (
((Networking.CONNECTION_CODES)BitConverter.ToUInt32(command, 0) ==
Networking.CONNECTION_CODES.OK)))
{
ClientSession clientSession = new ClientSession();
clientSession.AESKey = aes;
clientSession.Socket = s;
return clientSession;
}
}
}
return null;
}
catch (SocketException) { return null; }
}
private bool updateFile(SQLiteConnection conn, ClientSession clientSession, Socket s)
{
AesCryptoServiceProvider aes = clientSession.AESKey;
try
{
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
byte[] encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String filename = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String path = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime lastModTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
Int64 fileLen = BitConverter.ToInt64(recvBuf, 0);
byte[] file = Networking.recvEncryptedFile(fileLen, s, aes);
if (file == null)
return false;
return DBmanager.updateFile(conn, path, filename, clientSession.User.UserId, file, clientSession.CurrentStatus.CreationTime, lastModTime);
}
catch (SocketException) { return false;}
}
public Int64 authenticationTcpServer(ClientSession clientSession)
{
try
{
byte[] command = new byte[4];
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(16, s);
if (recvBuf != null)
{
byte[] decryptedData = Security.AESDecrypt(aes, recvBuf);
String userId = Encoding.UTF8.GetString(decryptedData);
String pwd = DBmanager.find_user(userId);
if (pwd != null)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
Random r = new Random();
byte[] challenge = new byte[8];
r.NextBytes(challenge);
byte[] encryptedData = Security.AESEncrypt(aes, challenge);
s.Send(encryptedData);
SHA1 sha = new SHA1CryptoServiceProvider();
byte[] p = Encoding.UTF8.GetBytes(pwd);
byte[] hash = sha.ComputeHash(Security.XOR(p, challenge));
byte[] hashClient = Networking.my_recv(20, s);
if (hashClient != null)
{
if (hash.SequenceEqual(hashClient))
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
Int64 sessionID = incrementAndGetIdCounter();
encryptedData = Security.AESEncrypt(aes, BitConverter.GetBytes(sessionID));
s.Send(encryptedData);
command = Networking.my_recv(4, s);
if (command != null && (
((Networking.CONNECTION_CODES)BitConverter.ToUInt32(command, 0) ==
Networking.CONNECTION_CODES.OK)))
{
clientSession.SessionId = sessionID;
clientSession.User = new User(userId, pwd);
clientSession.LastActivationTime = DateTime.Now;
id2client.GetOrAdd(sessionID, clientSession);
return sessionID;
}
else
{
return -1; //error on last ack
}
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.AUTH_FAILURE);
s.Send(command);
return -2;
}
}
}
else
{
//auth failed: user not existent
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.AUTH_FAILURE);
s.Send(command);
return -3;
}
}
return -1;
}
catch (SocketException) { return -1; }
}
private bool deleteFile(SQLiteConnection conn, ClientSession clientSession, Socket s)
{
AesCryptoServiceProvider aes = clientSession.AESKey;
try
{
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
byte[] encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String filename = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String path = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime lastModTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
if ((Networking.CONNECTION_CODES)BitConverter.ToUInt32(recvBuf, 0) == Networking.CONNECTION_CODES.DIR)
{
//directory
String fullname = Path.Combine(path, filename);
return DBmanager.deleteDirectory(conn, fullname, clientSession.User.UserId, clientSession.CurrentStatus.CreationTime);
}
if ((Networking.CONNECTION_CODES)BitConverter.ToUInt32(recvBuf, 0) == Networking.CONNECTION_CODES.FILE)
{
//file
return DBmanager.deleteFile(conn, path, filename, clientSession.User.UserId, clientSession.CurrentStatus.CreationTime);
}
return false;
}
catch (SocketException) { return false; }
}
private bool addFile(SQLiteConnection conn, ClientSession clientSession, DirectoryStatus newStatus)
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
try
{
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
byte[] encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String filename = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String path = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime lastModTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
if ((Networking.CONNECTION_CODES)BitConverter.ToUInt32(recvBuf, 0) == Networking.CONNECTION_CODES.DIR)
{
//directory
return DBmanager.insertDirectory(conn, newStatus, filename, path, lastModTime);
}
if ((Networking.CONNECTION_CODES)BitConverter.ToUInt32(recvBuf, 0) == Networking.CONNECTION_CODES.FILE)
{
//file
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
Int64 fileLen = BitConverter.ToInt64(recvBuf, 0);
byte[] file = Networking.recvEncryptedFile(fileLen, s, aes);
if (file == null)
return false;
return DBmanager.insertFile(conn, newStatus, filename, path, file, lastModTime);
}
return false;
}
catch (SocketException) { return false; }
}
private bool renameDirectory(SQLiteConnection conn, ClientSession clientSession, Socket s)
{
AesCryptoServiceProvider aes = clientSession.AESKey;
try
{
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
byte[] encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String oldPath = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
encryptedData = Networking.my_recv(BitConverter.ToInt32(recvBuf, 0), s);
if (encryptedData == null)
return false;
String newPath = Encoding.UTF8.GetString(Security.AESDecrypt(aes, encryptedData));
return DBmanager.renameDirectory(conn, clientSession.User.UserId, oldPath, newPath, clientSession.CurrentStatus.CreationTime);
}
catch (SocketException) { return false; }
}
public void synchronizationSession(ClientSession clientSession, Socket s)
{
SQLiteConnection con = null;
SQLiteTransaction transaction = null;
bool success = false;
bool exit = false;
try
{
con = new SQLiteConnection(DBmanager.connectionString);
con.Open();
transaction = con.BeginTransaction();
while (!exit)
{
byte[] command = Utility.Networking.my_recv(4, s);
if (command != null)
{
Networking.CONNECTION_CODES code = (Networking.CONNECTION_CODES)BitConverter.ToUInt32(command, 0);
switch (code)
{
case Networking.CONNECTION_CODES.ADD:
if (!this.addFile(con, clientSession, s))
exit = true;
break;
case Networking.CONNECTION_CODES.UPD:
if (!this.updateFile(con, clientSession, s))
exit = true;
break;
case Networking.CONNECTION_CODES.DEL:
if (!this.deleteFile(con, clientSession, s))
exit = true;
break;
case Networking.CONNECTION_CODES.RENAME_DIR:
if (!this.renameDirectory(con, clientSession, s))
exit = true;
break;
case Networking.CONNECTION_CODES.RENAME_FILE:
if (!this.renameFile(con, clientSession, s))
exit = true;
break;
case Networking.CONNECTION_CODES.END_SYNCH:
//success
success = true;
lock (this.id2client[clientSession.SessionId])
{
this.id2client[clientSession.SessionId].LastActivationTime = DateTime.Now;
}
exit = true;
break;
default:
exit = true;
break;
}
}
else
{
exit=true;
}
}
if (success)
{
transaction.Commit();
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
}
else
{
transaction.Rollback();
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.ERR);
s.Send(command);
}
}
catch (SocketException) {
if (transaction != null)
transaction.Rollback();
return;
}
catch (SQLiteException) {
if (transaction != null)
transaction.Rollback();
return;
}
finally
{
if (transaction != null)
{
transaction.Dispose();
}
if (con != null && con.State == System.Data.ConnectionState.Open)
con.Dispose();
if (s != null)
s.Close();
}
}
public bool getPreviousVersions(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
int pathLen = BitConverter.ToInt32(recvBuf, 0);
recvBuf = Networking.my_recv(pathLen, s);
if (recvBuf == null)
return false;
String path = Encoding.UTF8.GetString(Security.AESDecrypt(aes, recvBuf));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime date = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
DirectoryStatus requestedFiles = DBmanager.getPreviousVersion(path, clientSession.User.UserId, date);
int count = (requestedFiles == null) ? 0 : requestedFiles.Files.Count;
byte[] buf = BitConverter.GetBytes(count);
s.Send(buf);
if (requestedFiles != null)
{
foreach (var item in requestedFiles.Files)
{
DirectoryFile file = item.Value;
buf = BitConverter.GetBytes(file.LastModificationTime.ToBinary());
s.Send(buf);
buf = Encoding.UTF8.GetBytes(file.Filename);
buf = BitConverter.GetBytes(file.Id);
s.Send(buf);
buf = Encoding.UTF8.GetBytes(file.Checksum);
byte[] encryptedData = Security.AESEncrypt(aes, buf);
s.Send(encryptedData);
}
}
return true;
}
catch (SocketException) { return false; }
}
public bool synchronizationSession(ClientSession clientSession)
{
DirectoryStatus newStatus = new DirectoryStatus();
newStatus.FolderPath = clientSession.CurrentStatus.FolderPath;
newStatus.Username = clientSession.CurrentStatus.Username;
SQLiteConnection con = null;
SQLiteTransaction transaction = null;
bool success = false;
try
{
bool exit = false;
con = new SQLiteConnection(DBmanager.connectionString);
con.Open();
transaction = con.BeginTransaction();
while (!exit)
{
byte[] command = Utility.Networking.my_recv(4, clientSession.Socket);
if (command != null)
{
Networking.CONNECTION_CODES code = (Networking.CONNECTION_CODES)BitConverter.ToUInt32(command, 0);
switch (code)
{
case Networking.CONNECTION_CODES.ADD:
if (!this.addFile(con, clientSession, newStatus))
exit = true;
break;
case Networking.CONNECTION_CODES.UPD:
if (!this.updateFile(con, clientSession, newStatus))
exit = true;
break;
case Networking.CONNECTION_CODES.DEL:
if (!this.deleteFile(con, clientSession, newStatus))
exit = true;
break;
case Networking.CONNECTION_CODES.HELLO:
this.Hello(clientSession.Socket);
break;
case Networking.CONNECTION_CODES.SESSION:
if (!this.resumeSession(ref clientSession, clientSession.Socket))
exit = true;
break;
case Networking.CONNECTION_CODES.END_SYNCH:
if (DBmanager.insertUnchanged(con, clientSession.CurrentStatus, newStatus))
{
//success
success = true;
lock (this.id2client[clientSession.SessionId])
{
this.id2client[clientSession.SessionId].CurrentStatus = null;
this.id2client[clientSession.SessionId].CurrentStatus = newStatus;
this.id2client[clientSession.SessionId].LastActivationTime = DateTime.Now;
}
}
exit = true;
break;
default:
exit = true;
break;
}
}
else
{
exit = true;
}
}
if (success)
{
transaction.Commit();
transaction.Dispose();
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
clientSession.Socket.Send(command);
}
else
{
transaction.Rollback();
}
}
catch (SocketException)
{
newStatus = null;
if (transaction != null)
{
transaction.Rollback();
}
return false;
}
catch (SQLiteException)
{
newStatus = null;
if (transaction != null)
{
transaction.Rollback();
}
return false;
}
finally
{
if (transaction != null)
{
transaction.Dispose();
}
if (con != null && con.State == System.Data.ConnectionState.Open)
con.Dispose();
}
return success;
}
public bool resumeSession(ref ClientSession clientSession, Socket s, bool fromWatcher)
{
try
{
byte[] command = new byte[4];
Random r = new Random();
byte[] rand = new byte[8];
r.NextBytes(rand);
s.Send(rand);
byte[] hashClient = Networking.my_recv(20, s);
if (hashClient != null)
{
clientSession = getClientSessionFromHash(hashClient, rand);
if (clientSession != null)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
lock (clientSession)
{
clientSession.LastActivationTime = DateTime.Now;
}
return true;
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.AUTH);
s.Send(command);
return false;
}
}
return false;
}
catch (SocketException) { return false;}
}
public bool resumeSession(ref ClientSession clientSession, Socket s)
{
try
{
byte[] command = new byte[4];
Random r = new Random();
byte[] rand = new byte[8];
r.NextBytes(rand);
s.Send(rand);
byte[] hashClient = Networking.my_recv(20, s);
if (hashClient != null)
{
if (clientSession != null)
{
SHA1CryptoServiceProvider sha = new SHA1CryptoServiceProvider();
byte[] hash = null;
hash = sha.ComputeHash(Security.XOR(rand, BitConverter.GetBytes(clientSession.SessionId)));
if (hash.SequenceEqual(hashClient))
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
lock (clientSession)
{
clientSession.LastActivationTime = DateTime.Now;
clientSession.Socket = s;
}
return true;
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.AUTH);
s.Send(command);
return false;
}
}
else
{
clientSession = getClientSessionFromHash(hashClient, rand);
if (clientSession != null)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
lock (clientSession)
{
clientSession.LastActivationTime = DateTime.Now;
clientSession.Socket = s;
}
return true;
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.AUTH);
s.Send(command);
return false;
}
}
}
return false;
}
catch (SocketException) { return false; }
}
public void registrationTcpServer(ClientSession clientSession)
{
try
{
byte[] command = new byte[4];
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] encryptedData = Networking.my_recv(16, s);
if (encryptedData != null)
{
byte[] decryptedData = Security.AESDecrypt(aes, encryptedData);
String userid = Encoding.UTF8.GetString(decryptedData);
String pwd = DBmanager.find_user(userid);
if (pwd != null)
{
//user yet present
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.REG_FAILURE);
s.Send(command);
return;
}
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
byte[] pwdSize = Networking.my_recv(4, s);
if (pwdSize != null)
{
int size = BitConverter.ToInt32(pwdSize, 0);
encryptedData = Networking.my_recv(size, s);
decryptedData = Security.AESDecrypt(aes, encryptedData);
if (DBmanager.register(userid, Encoding.UTF8.GetString(decryptedData)))
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
return;
}
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.REG_FAILURE);
s.Send(command);
}
}
}
catch (SocketException) { return; }
}
public bool restoreFile(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
Int64 id = BitConverter.ToInt32(recvBuf, 0);
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
String path = "", name = "";
byte[] file = null;
DateTime lastModTime = DateTime.MinValue;
DBmanager.getFileFromID(id, ref path, ref name, ref file);
if (file != null && !path.Equals("") && !name.Equals(""))
{
long left = file.LongLength;
s.Send(BitConverter.GetBytes(left));
using (MemoryStream ms = new MemoryStream(file))
{
while (left > 0)
{
int dim = (left > 4096) ? 4096 : (int)left;
byte[] buf = new byte[dim];
ms.Read(buf, 0, dim);
byte[] encryptedData = Security.AESEncrypt(aes, buf);
s.Send(encryptedData);
left -= dim;
}
}
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
lastModTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
String checksum = Security.CalculateMD5Hash(file);
if (DBmanager.restoreFile(id, path, name, clientSession.User.UserId, checksum, lastModTime))
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
return true;
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.ERR);
s.Send(command);
return false;
}
}
return false;
}
catch (SocketException) { return false; }
}
public bool DownloadDirectory(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
int pathLen = BitConverter.ToInt32(recvBuf, 0);
recvBuf = Networking.my_recv(pathLen, s);
if (recvBuf == null)
return false;
String dir = Encoding.UTF8.GetString(Security.AESDecrypt(aes, recvBuf));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime creationTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
List<Int64> ids = DBmanager.getFilesIdToDownload(dir, clientSession.User.UserId, creationTime);
byte[] buf = BitConverter.GetBytes(ids.Count);
s.Send(buf);
bool result = true;
for (int i = 0; i < ids.Count; i++)
{
String path = "", name = "";
byte[] file = null;
DateTime lastModTime = DateTime.MinValue;
DBmanager.getFileFromID(ids[i], ref path, ref name, ref file);
if (file != null && !path.Equals("") && !name.Equals(""))
{
byte[] encryptedData = Security.AESEncrypt(aes, Encoding.UTF8.GetBytes(name));
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
encryptedData = Security.AESEncrypt(aes, Encoding.UTF8.GetBytes(path));
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
long left = file.LongLength;
s.Send(BitConverter.GetBytes(left));
using (MemoryStream ms = new MemoryStream(file))
{
while (left > 0)
{
int dim = (left > 4096) ? 4096 : (int)left;
buf = new byte[dim];
ms.Read(buf, 0, dim);
encryptedData = Security.AESEncrypt(aes, buf);
s.Send(encryptedData);
left -= dim;
}
}
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
lastModTime = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
if (DBmanager.restoreFile(ids[i], path, name, clientSession.User.UserId, lastModTime, creationTime))
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.ERR);
s.Send(command);
result = false;
break;
}
}
}
return result;
}
catch (SocketException) { return false; }
catch (Exception) { return false; }
}
public bool beginSynchronization(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
int pathLen = BitConverter.ToInt32(recvBuf, 0);
recvBuf = Networking.my_recv(pathLen, s);
if (recvBuf == null)
return false;
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
String dir = Encoding.UTF8.GetString(Security.AESDecrypt(aes, recvBuf));
DirectoryStatus status = DBmanager.getLastSnapshot(dir, clientSession.User.UserId);
clientSession.CurrentStatus = status;
if (status != null)
{
Dictionary<String, DirectoryFile> diff = status.Files.Where(x => !status.Files[x.Key].Deleted).ToDictionary(x => x.Key, x => x.Value);
byte[] buf = BitConverter.GetBytes(diff.Count);
s.Send(buf);
foreach (var item in diff)
{
DirectoryFile file = item.Value;
buf = Encoding.UTF8.GetBytes(file.Path);
byte[] encryptedData = Security.AESEncrypt(aes, buf);
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
buf = Encoding.UTF8.GetBytes(file.Filename);
encryptedData = Security.AESEncrypt(aes, buf);
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
buf = BitConverter.GetBytes(file.LastModificationTime.ToBinary());
s.Send(buf);
if (file.Directory)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.DIR);
s.Send(command);
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.FILE);
s.Send(command);
buf = BitConverter.GetBytes(file.Id);
s.Send(buf);
buf = Encoding.UTF8.GetBytes(file.Checksum);
encryptedData = Security.AESEncrypt(aes, buf);
s.Send(encryptedData);
}
}
}
else
{
byte[] buf = BitConverter.GetBytes(0);
s.Send(buf);
return false;
}
return true;
}
catch (SocketException) { return false; }
}
public bool getDirectoryName(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf != null)
{
int pathLen = BitConverter.ToInt32(recvBuf, 0);
recvBuf = Networking.my_recv(pathLen, s);
if (recvBuf != null)
{
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
return true;
}
}
return false;
}
catch (SocketException) { return false; }
}
public bool getSnapshotInfo(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
byte[] recvBuf = Networking.my_recv(4, s);
if (recvBuf == null)
return false;
int pathLen = BitConverter.ToInt32(recvBuf, 0);
recvBuf = Networking.my_recv(pathLen, s);
if (recvBuf == null)
return false;
String dir = Encoding.UTF8.GetString(Security.AESDecrypt(aes, recvBuf));
recvBuf = Networking.my_recv(8, s);
if (recvBuf == null)
return false;
DateTime creation_time = DateTime.FromBinary(BitConverter.ToInt64(recvBuf, 0));
byte[] command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
DirectoryStatus requestedDirectory = DBmanager.getSnapshotFilesOfDirectory(dir, clientSession.User.UserId,creation_time);
int count = (requestedDirectory == null) ? 0 : requestedDirectory.Files.Count;
byte[] buf = BitConverter.GetBytes(count);
s.Send(buf);
if (requestedDirectory != null)
{
foreach (var item in requestedDirectory.Files)
{
DirectoryFile file = item.Value;
buf = Encoding.UTF8.GetBytes(file.Path);
byte[] encryptedData = Security.AESEncrypt(aes, buf);
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
buf = Encoding.UTF8.GetBytes(file.Filename);
encryptedData = Security.AESEncrypt(aes, buf);
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
buf = BitConverter.GetBytes(file.LastModificationTime.ToBinary());
s.Send(buf);
if (file.Deleted)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.DEL);
s.Send(command);
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.OK);
s.Send(command);
}
if (file.Directory)
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.DIR);
s.Send(command);
}
else
{
command = BitConverter.GetBytes((UInt32)Networking.CONNECTION_CODES.FILE);
s.Send(command);
buf = BitConverter.GetBytes(file.Id);
s.Send(buf);
}
}
}
return true;
}
catch (SocketException) { return false; }
}
public bool getAllSnapshots(ClientSession clientSession)
{
try
{
Socket s = clientSession.Socket;
AesCryptoServiceProvider aes = clientSession.AESKey;
DirectoryStatus snapshots = DBmanager.getAllSnapshots(clientSession.User.UserId);
int count = (snapshots == null) ? 0 : snapshots.Files.Count;
byte[] buf = BitConverter.GetBytes(count);
s.Send(buf);
if (snapshots != null)
{
foreach (var item in snapshots.Files)
{
DirectoryFile file = item.Value;
buf = Encoding.UTF8.GetBytes(file.Path);
byte[] encryptedData = Security.AESEncrypt(aes, buf);
s.Send(BitConverter.GetBytes(encryptedData.Length));
s.Send(encryptedData);
buf = BitConverter.GetBytes(file.LastModificationTime.ToBinary());
s.Send(buf);
}
}
return true;
}
catch (SocketException) { return false; }
}