mirror of
https://github.com/wiz0u/WTelegramClient.git
synced 2025-12-06 06:52:01 +01:00
104 lines
3.4 KiB
C#
104 lines
3.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Security.Cryptography;
|
|
using System.Text.Json;
|
|
using System.Threading;
|
|
|
|
namespace WTelegram
|
|
{
|
|
internal class Session
|
|
{
|
|
public TL.User User;
|
|
public int MainDC;
|
|
public Dictionary<int, DCSession> DCSessions = new();
|
|
public TL.DcOption[] DcOptions;
|
|
|
|
public class DCSession
|
|
{
|
|
public long Id;
|
|
public long AuthKeyID;
|
|
public byte[] AuthKey; // 2048-bit = 256 bytes
|
|
public long UserId;
|
|
public long Salt;
|
|
public int Seqno;
|
|
public long ServerTicksOffset;
|
|
public long LastSentMsgId;
|
|
public TL.DcOption DataCenter;
|
|
|
|
internal Client Client;
|
|
internal int DcID => DataCenter?.id ?? 0;
|
|
internal IPEndPoint EndPoint => DataCenter == null ? null : new(IPAddress.Parse(DataCenter.ip_address), DataCenter.port);
|
|
}
|
|
|
|
public DateTime SessionStart => _sessionStart;
|
|
private readonly DateTime _sessionStart = DateTime.UtcNow;
|
|
private readonly SHA256 _sha256 = SHA256.Create();
|
|
private string _pathname;
|
|
private byte[] _apiHash; // used as AES key for encryption of session file
|
|
|
|
private static readonly JsonSerializerOptions JsonOptions = new(Helpers.JsonOptions)
|
|
{
|
|
Converters = {
|
|
new Helpers.PolymorphicConverter<TL.UserProfilePhoto>(),
|
|
new Helpers.PolymorphicConverter<TL.UserStatus>()
|
|
}
|
|
};
|
|
|
|
internal static Session LoadOrCreate(string pathname, byte[] apiHash)
|
|
{
|
|
if (File.Exists(pathname))
|
|
{
|
|
try
|
|
{
|
|
var session = Load(pathname, apiHash);
|
|
session._pathname = pathname;
|
|
session._apiHash = apiHash;
|
|
Helpers.Log(2, "Loaded previous session");
|
|
return session;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
throw new ApplicationException($"Exception while reading session file: {ex.Message}\nDelete the file to start a new session", ex);
|
|
}
|
|
}
|
|
return new Session { _pathname = pathname, _apiHash = apiHash };
|
|
}
|
|
|
|
internal static Session Load(string pathname, byte[] apiHash)
|
|
{
|
|
var input = File.ReadAllBytes(pathname);
|
|
using var sha256 = SHA256.Create();
|
|
using var aes = Aes.Create();
|
|
using var decryptor = aes.CreateDecryptor(apiHash, input[0..16]);
|
|
var utf8Json = decryptor.TransformFinalBlock(input, 16, input.Length - 16);
|
|
if (!sha256.ComputeHash(utf8Json, 32, utf8Json.Length - 32).SequenceEqual(utf8Json[0..32]))
|
|
throw new ApplicationException("Integrity check failed in session loading");
|
|
return JsonSerializer.Deserialize<Session>(utf8Json.AsSpan(32), JsonOptions);
|
|
}
|
|
|
|
internal void Save()
|
|
{
|
|
var utf8Json = JsonSerializer.SerializeToUtf8Bytes(this, JsonOptions);
|
|
var finalBlock = new byte[16];
|
|
var output = new byte[(16 + 32 + utf8Json.Length + 16) & ~15];
|
|
Encryption.RNG.GetBytes(output, 0, 16);
|
|
using var aes = Aes.Create();
|
|
using var encryptor = aes.CreateEncryptor(_apiHash, output[0..16]);
|
|
encryptor.TransformBlock(_sha256.ComputeHash(utf8Json), 0, 32, output, 16);
|
|
encryptor.TransformBlock(utf8Json, 0, utf8Json.Length & ~15, output, 48);
|
|
utf8Json.AsSpan(utf8Json.Length & ~15).CopyTo(finalBlock);
|
|
encryptor.TransformFinalBlock(finalBlock, 0, utf8Json.Length & 15).CopyTo(output.AsMemory(48 + utf8Json.Length & ~15));
|
|
if (!File.Exists(_pathname))
|
|
File.WriteAllBytes(_pathname, output);
|
|
else lock (this)
|
|
{
|
|
string tempPathname = _pathname + ".tmp";
|
|
File.WriteAllBytes(tempPathname, output);
|
|
File.Replace(tempPathname, _pathname, null);
|
|
}
|
|
}
|
|
}
|
|
} |