protected void StatisticsInit(AuthDataAes128 authData)
{
#if PROTOCOL_STATISTICS
if (authData.tree == null)
{
authData.tree = new Model.MinSearchTree(Server.tcp_mss - tree_offset);
authData.tree.Init();
}
tree = authData.tree.Clone();
#endif
}
public override void Dispose()
{
#if PROTOCOL_STATISTICS
if (Server != null && Server.data != null && packet_cnt != null)
{
AuthDataAes128 authData = Server.data as AuthDataAes128;
if (authData != null && authData.tree != null)
{
lock (authData)
{
authData.tree.Update(packet_cnt);
}
}
}
#endif
}
protected void Sync()
{
#if PROTOCOL_STATISTICS
if (Server.data != null)
{
AuthDataAes128 authData = Server.data as AuthDataAes128;
lock (authData)
{
if (authData.tree != null && packet_cnt != null)
{
authData.tree.Update(packet_cnt);
tree = authData.tree.Clone();
}
}
}
#endif
}
public void PackAuthData(byte[] data, int datalength, byte[] outdata, out int outlength)
{
const int authhead_len = 7 + 4 + 16 + 4;
const int overhead = authhead_len + 4;
byte[] encrypt = new byte[24];
AuthDataAes128 authData = this.Server.data as AuthDataAes128;
lock (authData)
{
if (authData.connectionID > 0xFF000000)
{
authData.clientID = null;
}
if (authData.clientID == null)
{
authData.clientID = new byte[4];
g_random.GetBytes(authData.clientID);
authData.connectionID = (UInt32)BitConverter.ToInt32(authData.clientID, 0) % 0xFFFFFD;
}
authData.connectionID += 1;
Array.Copy(authData.clientID, 0, encrypt, 4, 4);
Array.Copy(BitConverter.GetBytes(authData.connectionID), 0, encrypt, 8, 4);
StatisticsInit(authData);
}
#if !PROTOCOL_STATISTICS
int rand_len = TrapezoidRandomInt(Server.tcp_mss - datalength - overhead + 1, -0.3); //(datalength > 400 ? LinearRandomInt(512) : LinearRandomInt(1024));
#else
int rand_len = GenRandLenFull(datalength + overhead, datalength, false) - datalength - overhead;
#endif
int data_offset = rand_len + authhead_len;
outlength = data_offset + datalength + 4;
byte[] encrypt_data = new byte[32];
byte[] key = new byte[Server.iv.Length + Server.key.Length];
Server.iv.CopyTo(key, 0);
Server.key.CopyTo(key, Server.iv.Length);
{
byte[] rnd_data = new byte[rand_len];
random.NextBytes(rnd_data);
rnd_data.CopyTo(outdata, data_offset - rand_len);
}
UInt64 utc_time_second = (UInt64)Math.Floor(DateTime.UtcNow.Subtract(new DateTime(1970, 1, 1, 0, 0, 0)).TotalSeconds);
UInt32 utc_time = (UInt32)(utc_time_second);
Array.Copy(BitConverter.GetBytes(utc_time), 0, encrypt, 0, 4);
encrypt[12] = (byte)(outlength);
encrypt[13] = (byte)(outlength >> 8);
encrypt[14] = (byte)(rand_len);
encrypt[15] = (byte)(rand_len >> 8);
{
byte[] uid = new byte[4];
int index_of_split = Server.param.IndexOf(':');
if (index_of_split > 0)
{
try
{
uint user = uint.Parse(Server.param.Substring(0, index_of_split));
user_key = hash(System.Text.Encoding.UTF8.GetBytes(Server.param.Substring(index_of_split + 1)));
BitConverter.GetBytes(user).CopyTo(uid, 0);
}
catch (Exception ex)
{
Logging.Log(LogLevel.Warn, $"Faild to parse auth param, fallback to basic mode. {ex}");
}
}
if (user_key == null)
{
random.NextBytes(uid);
user_key = Server.key;
}
byte[] encrypt_key = user_key;
Encryption.IEncryptor encryptor = Encryption.EncryptorFactory.GetEncryptor("aes-128-cbc", System.Convert.ToBase64String(encrypt_key) + SALT);
int enc_outlen;
encryptor.SetIV(new byte[] { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 });
encryptor.Encrypt(encrypt, 16, encrypt_data, out enc_outlen);
encryptor.Dispose();
Array.Copy(encrypt_data, 16, encrypt, 4, 16);
uid.CopyTo(encrypt, 0);
}
{
MbedTLS.HMAC sha1 = CreateHMAC(key);
byte[] sha1data = sha1.ComputeHash(encrypt, 0, 20);
Array.Copy(sha1data, 0, encrypt, 20, 4);
}
{
byte[] rnd = new byte[1];
random.NextBytes(rnd);
rnd.CopyTo(outdata, 0);
MbedTLS.HMAC sha1 = CreateHMAC(key);
byte[] sha1data = sha1.ComputeHash(rnd, 0, rnd.Length);
Array.Copy(sha1data, 0, outdata, rnd.Length, 7 - rnd.Length);
}
encrypt.CopyTo(outdata, 7);
Array.Copy(data, 0, outdata, data_offset, datalength);
{
MbedTLS.HMAC sha1 = CreateHMAC(user_key);
byte[] sha1data = sha1.ComputeHash(outdata, 0, outlength - 4);
Array.Copy(sha1data, 0, outdata, outlength - 4, 4);
}
}
