Telegram Web K with changes to work inside I2P https://web.telegram.i2p/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

603 lines
21 KiB

/*
* https://github.com/morethanwords/tweb
* Copyright (C) 2019-2021 Eduard Kuzmenko
* https://github.com/morethanwords/tweb/blob/master/LICENSE
*
* Originally from:
* https://github.com/zhukov/webogram
* Copyright (C) 2014 Igor Zhukov <igor.beatle@gmail.com>
* https://github.com/zhukov/webogram/blob/master/LICENSE
*/
import { TLSerialization, TLDeserialization } from "./tl_utils";
import dcConfigurator from "./dcConfigurator";
import rsaKeysManager from "./rsaKeysManager";
import timeManager from "./timeManager";
import CryptoWorker from "../crypto/cryptoworker";
import { logger, LogTypes } from "../logger";
import { bytesCmp, bytesToHex, bytesFromHex, bytesXor } from "../../helpers/bytes";
import DEBUG from "../../config/debug";
import { cmp, int2bigInt, one, pow, str2bigInt, sub } from "../../vendor/leemon";
/* let fNewNonce: any = bytesFromHex('8761970c24cb2329b5b2459752c502f3057cb7e8dbab200e526e8767fdc73b3c').reverse();
let fNonce: any = bytesFromHex('b597720d11faa5914ef485c529cde414').reverse();
let fResult: any = new Uint8Array(bytesFromHex('000000000000000001b473a0661b285e480000006324160514e4cd29c585f44e91a5fa110d7297b5c0c4134c84893db5715ecd56af5ed618082182053cc5de91cd00000015c4b51c02000000a5b7f709355fc30b216be86c022bb4c3'));
fNewNonce = false;
fNonce = false;
fResult = false; */
type AuthOptions = {
dcId: number,
nonce: Uint8Array,
serverNonce?: Uint8Array,
pq?: Uint8Array,
fingerprints?: string[],
publicKey?: {
modulus: string,
exponent: string,
fingerprint: string
},
// good
p?: number[],
q?: number[],
newNonce?: Uint8Array,
retry?: number,
b?: number[],
g?: number,
gA?: Uint8Array,
dhPrime?: Uint8Array,
tmpAesKey?: Uint8Array,
tmpAesIv?: Uint8Array,
authKeyId?: Uint8Array,
authKey?: number[],
serverSalt?: number[],
localTime?: number,
serverTime?: any
};
export class Authorizer {
private cached: {
[dcId: number]: Promise<AuthOptions>
} = {};
private log: ReturnType<typeof logger>;
constructor() {
this.log = logger(`AUTHORIZER`, LogTypes.Error | LogTypes.Log);
}
public mtpSendPlainRequest(dcId: number, requestArray: Uint8Array) {
const requestLength = requestArray.byteLength;
//requestArray = new /* Int32Array */Uint8Array(requestBuffer);
const header = new TLSerialization();
header.storeLongP(0, 0, 'auth_key_id'); // Auth key
header.storeLong(timeManager.generateId(), 'msg_id'); // Msg_id
header.storeInt(requestLength, 'request_length');
const headerArray = header.getBytes(true) as Uint8Array;
const resultArray = new Uint8Array(headerArray.byteLength + requestLength);
resultArray.set(headerArray);
resultArray.set(requestArray, headerArray.length);
/* var headerBuffer = header.getBuffer(),
headerArray = new Int32Array(headerBuffer);
var headerLength = headerBuffer.byteLength;
var resultBuffer = new ArrayBuffer(headerLength + requestLength),
resultArray = new Int32Array(resultBuffer);
resultArray.set(headerArray);
resultArray.set(requestArray, headerArray.length);
const requestData = xhrSendBuffer ? resultBuffer : resultArray; */
const transport = dcConfigurator.chooseServer(dcId);
const baseError = {
code: 406,
type: 'NETWORK_BAD_RESPONSE',
transport: transport
};
if(DEBUG) {
this.log('mtpSendPlainRequest: creating requestPromise');
}
const promise = transport.send(resultArray) as any as Promise<Uint8Array>;
return promise.then(result => {
if(DEBUG) {
this.log('mtpSendPlainRequest: in good sector', result);
}
if(!result || !result.byteLength) {
return Promise.reject(baseError);
}
try {
/* result = fResult ? fResult : result;
fResult = new Uint8Array(0); */
const deserializer = new TLDeserialization(result, {mtproto: true});
const auth_key_id = deserializer.fetchLong('auth_key_id');
if(auth_key_id !== '0') this.log.error('auth_key_id !== 0', auth_key_id);
const msg_id = deserializer.fetchLong('msg_id');
if(msg_id === '0') this.log.error('msg_id === 0', msg_id);
const msg_len = deserializer.fetchInt('msg_len');
if(!msg_len) this.log.error('no msg_len', msg_len);
return deserializer;
} catch(e) {
this.log.error('mtpSendPlainRequest: deserialization went bad', e);
const error = Object.assign(baseError, {originalError: e});
throw error;
}
}, error => {
if(!error.message && !error.type) {
error = Object.assign(baseError, {
originalError: error
});
}
return Promise.reject(error);
});
}
public async mtpSendReqPQ(auth: AuthOptions) {
var request = new TLSerialization({mtproto: true});
request.storeMethod('req_pq_multi', {nonce: auth.nonce});
// need
rsaKeysManager.prepare().then(() => {});
if(DEBUG) {
this.log('Send req_pq', auth.nonce.hex);
}
try {
var deserializer = await this.mtpSendPlainRequest(auth.dcId, request.getBytes(true));
} catch(error) {
this.log.error('req_pq error', error.message);
throw error;
}
var response = deserializer.fetchObject('ResPQ');
if(response._ !== 'resPQ') {
throw new Error('[MT] resPQ response invalid: ' + response._);
}
if(!bytesCmp(auth.nonce, response.nonce)) {
this.log.error(auth.nonce, response.nonce);
throw new Error('[MT] resPQ nonce mismatch');
}
//auth.serverNonce = response.server_nonce;
auth.serverNonce = new Uint8Array(response.server_nonce); // need
auth.pq = response.pq;
auth.fingerprints = response.server_public_key_fingerprints;
if(DEBUG) {
this.log('Got ResPQ', bytesToHex(auth.serverNonce), bytesToHex(auth.pq), auth.fingerprints);
}
let publicKey = await rsaKeysManager.select(auth.fingerprints);
if(!publicKey) {
throw new Error('[MT] No public key found');
}
auth.publicKey = publicKey;
if(DEBUG) {
this.log('PQ factorization start', auth.pq);
}
try {
var pAndQ = await CryptoWorker.factorize(auth.pq);
} catch(error) {
this.log.error('worker error factorize', error);
throw error;
}
auth.p = pAndQ[0];
auth.q = pAndQ[1];
if(DEBUG) {
this.log('PQ factorization done', pAndQ);
}
/* let p = new Uint32Array(new Uint8Array(auth.p).buffer)[0];
let q = new Uint32Array(new Uint8Array(auth.q).buffer)[0];
console.log(dT(), 'PQ factorization done', pAndQ, p.toString(16), q.toString(16)); */
return this.mtpSendReqDhParams(auth);
}
public async mtpSendReqDhParams(auth: AuthOptions) {
auth.newNonce = new Uint8Array(32).randomize();
/* auth.newNonce = new Array(32); // need array, not uint8array!
MTProto.secureRandom.nextBytes(auth.newNonce); */
//console.log("TCL: Authorizer -> mtpSendReqDhParams -> auth.newNonce", auth.newNonce)
// remove
// auth.newNonce = fNewNonce ? fNewNonce : auth.newNonce;
// console.log("TCL: Authorizer -> mtpSendReqDhParams -> auth.newNonce", auth.newNonce);
let p_q_inner_data = {
_: 'p_q_inner_data',
pq: auth.pq,
p: auth.p,
q: auth.q,
nonce: auth.nonce,
server_nonce: auth.serverNonce,
new_nonce: auth.newNonce
};
let data = new TLSerialization({mtproto: true});
data.storeObject(p_q_inner_data, 'P_Q_inner_data', 'DECRYPTED_DATA');
/* console.log('p_q_inner_data', p_q_inner_data,
bytesToHex(bytesFromArrayBuffer(data.getBuffer())),
sha1BytesSync(data.getBuffer()),
bytesFromArrayBuffer(await CryptoWorker.sha1Hash(data.getBuffer()))); */
let uint8Data = data.getBytes(true);
let sha1Hashed = await CryptoWorker.sha1Hash(uint8Data);
//var dataWithHash = sha1BytesSync(data.getBuffer()).concat(data.getBytes() as number[]);
let dataWithHash = sha1Hashed.concat(uint8Data);
//dataWithHash = addPadding(dataWithHash, 255);
//dataWithHash = dataWithHash.concat(bytesFromHex('96228ea7790e71caaabc2ab67f4412e9aa224c664d232cc08617a32ce1796aa052da4a737083211689858f461e4473fd6394afd3aa0c8014840dc13f47beaf4fc3b9229aea9cfa83f9f6e676e50ee7676542fb75606879ee7e65cf3a2295b4ba0934ceec1011560c62395a6e9593bfb117cd0da75ba56723672d100ac17ec4d805aa59f7852e3a25a79ee4'));
//console.log('sha1Hashed', bytesToHex(sha1Hashed), 'dataWithHash', bytesToHex(dataWithHash), dataWithHash.length);
let rsaEncrypted = await CryptoWorker.rsaEncrypt(auth.publicKey, dataWithHash);
//let rsaEncrypted = await CryptoWorker.rsaEncrypt(auth.publicKey, dataWithHash);
//console.log('rsaEncrypted', rsaEncrypted, new Uint8Array(rsaEncrypted).hex);
let req_DH_params = {
nonce: auth.nonce,
server_nonce: auth.serverNonce,
p: auth.p,
q: auth.q,
public_key_fingerprint: auth.publicKey.fingerprint,
encrypted_data: rsaEncrypted
};
var request = new TLSerialization({mtproto: true});
request.storeMethod('req_DH_params', req_DH_params);
let requestBytes = request.getBytes(true);
if(DEBUG) {
this.log('Send req_DH_params', req_DH_params/* , requestBytes.hex */);
}
try {
var deserializer = await this.mtpSendPlainRequest(auth.dcId, requestBytes);
} catch(error) {
this.log.error('Send req_DH_params FAIL!', error);
throw error;
}
var response = deserializer.fetchObject('Server_DH_Params', 'RESPONSE');
if(DEBUG) {
this.log('Sent req_DH_params, response:', response);
}
if(response._ !== 'server_DH_params_fail' && response._ !== 'server_DH_params_ok') {
throw new Error('[MT] Server_DH_Params response invalid: ' + response._);
}
if(!bytesCmp(auth.nonce, response.nonce)) {
throw new Error('[MT] Server_DH_Params nonce mismatch');
}
if(!bytesCmp(auth.serverNonce, response.server_nonce)) {
throw new Error('[MT] Server_DH_Params server_nonce mismatch');
}
if(response._ === 'server_DH_params_fail') {
//var newNonceHash = sha1BytesSync(auth.newNonce).slice(-16);
var newNonceHash = (await CryptoWorker.sha1Hash(auth.newNonce)).slice(-16);
if(!bytesCmp(newNonceHash, response.new_nonce_hash)) {
throw new Error('[MT] server_DH_params_fail new_nonce_hash mismatch');
}
throw new Error('[MT] server_DH_params_fail');
}
// fill auth object
try {
await this.mtpDecryptServerDhDataAnswer(auth, response.encrypted_answer);
} catch(e) {
this.log.error('mtpDecryptServerDhDataAnswer FAILED!', e);
throw e;
}
//console.log(dT(), 'mtpSendReqDhParams: executing mtpSendSetClientDhParams...');
return this.mtpSendSetClientDhParams(auth as any); // костыль
}
public async mtpDecryptServerDhDataAnswer(auth: AuthOptions, encryptedAnswer: any) {
auth.localTime = Date.now();
// can't concat Array with Uint8Array!
//auth.tmpAesKey = sha1BytesSync(auth.newNonce.concat(auth.serverNonce)).concat(sha1BytesSync(auth.serverNonce.concat(auth.newNonce)).slice(0, 12));
//auth.tmpAesIv = sha1BytesSync(auth.serverNonce.concat(auth.newNonce)).slice(12).concat(sha1BytesSync([].concat(auth.newNonce, auth.newNonce)), auth.newNonce.slice(0, 4));
auth.tmpAesKey = (await CryptoWorker.sha1Hash(auth.newNonce.concat(auth.serverNonce)))
.concat((await CryptoWorker.sha1Hash(auth.serverNonce.concat(auth.newNonce))).slice(0, 12));
auth.tmpAesIv = (await CryptoWorker.sha1Hash(auth.serverNonce.concat(auth.newNonce))).slice(12)
.concat(await CryptoWorker.sha1Hash(auth.newNonce.concat(auth.newNonce)), auth.newNonce.slice(0, 4));
/* console.log(auth.serverNonce.concat(auth.newNonce));
console.log(auth.newNonce.concat(auth.serverNonce));
console.log(auth.newNonce.concat(auth.newNonce)); */
//var answerWithHash = aesDecryptSync(encryptedAnswer, auth.tmpAesKey, auth.tmpAesIv);
var answerWithHash = new Uint8Array(await CryptoWorker.aesDecrypt(encryptedAnswer, auth.tmpAesKey, auth.tmpAesIv));
var hash = answerWithHash.slice(0, 20);
var answerWithPadding = answerWithHash.slice(20);
// console.log('hash', hash);
var deserializer = new TLDeserialization(answerWithPadding, {mtproto: true});
var response = deserializer.fetchObject('Server_DH_inner_data');
if(response._ !== 'server_DH_inner_data') {
throw new Error('[MT] server_DH_inner_data response invalid: ' + response);
}
if(!bytesCmp(auth.nonce, response.nonce)) {
throw new Error('[MT] server_DH_inner_data nonce mismatch');
}
if(!bytesCmp(auth.serverNonce, response.server_nonce)) {
throw new Error('[MT] server_DH_inner_data serverNonce mismatch');
}
if(DEBUG) {
this.log('Done decrypting answer');
}
auth.g = response.g;
auth.dhPrime = response.dh_prime;
auth.gA = response.g_a;
auth.serverTime = response.server_time;
auth.retry = 0;
this.mtpVerifyDhParams(auth.g, auth.dhPrime, auth.gA);
var offset = deserializer.getOffset();
//if(!bytesCmp(hash, sha1BytesSync(answerWithPadding.slice(0, offset)))) {
if(!bytesCmp(hash, await CryptoWorker.sha1Hash(answerWithPadding.slice(0, offset)))) {
throw new Error('[MT] server_DH_inner_data SHA1-hash mismatch');
}
timeManager.applyServerTime(auth.serverTime, auth.localTime);
}
public mtpVerifyDhParams(g: number, dhPrime: Uint8Array, gA: Uint8Array) {
if(DEBUG) {
this.log('Verifying DH params', g, dhPrime, gA);
}
var dhPrimeHex = bytesToHex(dhPrime);
if(g !== 3 || dhPrimeHex !== 'c71caeb9c6b1c9048e6c522f70f13f73980d40238e3e21c14934d037563d930f48198a0aa7c14058229493d22530f4dbfa336f6e0ac925139543aed44cce7c3720fd51f69458705ac68cd4fe6b6b13abdc9746512969328454f18faf8c595f642477fe96bb2a941d5bcd1d4ac8cc49880708fa9b378e3c4f3a9060bee67cf9a4a4a695811051907e162753b56b0f6b410dba74d8a84b2a14b3144e0ef1284754fd17ed950d5965b4b9dd46582db1178d169c6bc465b0d6ff9ca3928fef5b9ae4e418fc15e83ebea0f87fa9ff5eed70050ded2849f47bf959d956850ce929851f0d8115f635b105ee2e4e15d04b2454bf6f4fadf034b10403119cd8e3b92fcc5b') {
// The verified value is from https://core.telegram.org/mtproto/security_guidelines
throw new Error('[MT] DH params are not verified: unknown dhPrime');
}
if(DEBUG) {
this.log('dhPrime cmp OK');
}
//var gABigInt = new BigInteger(bytesToHex(gA), 16);
const _gABigInt = str2bigInt(bytesToHex(gA), 16);
//var dhPrimeBigInt = new BigInteger(dhPrimeHex, 16);
const _dhPrimeBigInt = str2bigInt(dhPrimeHex, 16);
//this.log('gABigInt.compareTo(BigInteger.ONE) <= 0', gABigInt.compareTo(BigInteger.ONE), BigInteger.ONE.compareTo(BigInteger.ONE), greater(_gABigInt, one));
//if(gABigInt.compareTo(BigInteger.ONE) <= 0) {
if(cmp(_gABigInt, one) <= 0) {
throw new Error('[MT] DH params are not verified: gA <= 1');
}
/* this.log('gABigInt.compareTo(dhPrimeBigInt.subtract(BigInteger.ONE)) >= 0', gABigInt.compareTo(dhPrimeBigInt.subtract(BigInteger.ONE)),
greater(gABigInt, sub(_dhPrimeBigInt, one))); */
//if(gABigInt.compareTo(dhPrimeBigInt.subtract(BigInteger.ONE)) >= 0) {
if(cmp(_gABigInt, sub(_dhPrimeBigInt, one)) >= 0) {
throw new Error('[MT] DH params are not verified: gA >= dhPrime - 1');
}
if(DEBUG) {
this.log('1 < gA < dhPrime-1 OK');
}
//var two = new BigInteger(/* null */'');
//two.fromInt(2);
const _two = int2bigInt(2, 32, 0);
//this.log('_two:', bigInt2str(_two, 16), two.toString(16));
let perf = performance.now();
//var twoPow = two.pow(2048 - 64);
//console.log('jsbn pow', performance.now() - perf);
perf = performance.now();
const _twoPow = pow(_two, 2048 - 64);
//console.log('leemon pow', performance.now() - perf);
//this.log('twoPow:', twoPow.toString(16), bigInt2str(_twoPow, 16));
// this.log('gABigInt.compareTo(twoPow) < 0');
//if(gABigInt.compareTo(twoPow) < 0) {
if(cmp(_gABigInt, _twoPow) < 0) {
throw new Error('[MT] DH params are not verified: gA < 2^{2048-64}');
}
//if(gABigInt.compareTo(dhPrimeBigInt.subtract(twoPow)) >= 0) {
if(cmp(_gABigInt, sub(_dhPrimeBigInt, _twoPow)) >= 0) {
throw new Error('[MT] DH params are not verified: gA > dhPrime - 2^{2048-64}');
}
if(DEBUG) {
this.log('2^{2048-64} < gA < dhPrime-2^{2048-64} OK');
}
return true;
}
public async mtpSendSetClientDhParams(auth: AuthOptions): Promise<AuthOptions> {
var gBytes = bytesFromHex(auth.g.toString(16));
auth.b = new Array(256);
auth.b = [...new Uint8Array(auth.b.length).randomize()];
//MTProto.secureRandom.nextBytes(auth.b);
try {
var gB = await CryptoWorker.modPow(gBytes, auth.b, auth.dhPrime);
} catch(error) {
throw error;
}
var data = new TLSerialization({mtproto: true});
data.storeObject({
_: 'client_DH_inner_data',
nonce: auth.nonce,
server_nonce: auth.serverNonce,
retry_id: [0, auth.retry++],
g_b: gB
}, 'Client_DH_Inner_Data');
//var dataWithHash = sha1BytesSync(data.getBuffer()).concat(data.getBytes());
var dataWithHash = (await CryptoWorker.sha1Hash(data.getBuffer())).concat(data.getBytes());
//var encryptedData = aesEncryptSync(dataWithHash, auth.tmpAesKey, auth.tmpAesIv);
var encryptedData = await CryptoWorker.aesEncrypt(dataWithHash, auth.tmpAesKey, auth.tmpAesIv);
var request = new TLSerialization({mtproto: true});
request.storeMethod('set_client_DH_params', {
nonce: auth.nonce,
server_nonce: auth.serverNonce,
encrypted_data: encryptedData
});
if(DEBUG) {
this.log('Send set_client_DH_params');
}
try {
var deserializer = await this.mtpSendPlainRequest(auth.dcId, request.getBytes(true));
} catch(err) {
throw err;
}
let response = deserializer.fetchObject('Set_client_DH_params_answer');
if(response._ !== 'dh_gen_ok' && response._ !== 'dh_gen_retry' && response._ !== 'dh_gen_fail') {
throw new Error('[MT] Set_client_DH_params_answer response invalid: ' + response._);
}
if(!bytesCmp(auth.nonce, response.nonce)) {
throw new Error('[MT] Set_client_DH_params_answer nonce mismatch');
}
if(!bytesCmp(auth.serverNonce, response.server_nonce)) {
throw new Error('[MT] Set_client_DH_params_answer server_nonce mismatch');
}
try {
var authKey = await CryptoWorker.modPow(auth.gA, auth.b, auth.dhPrime);
} catch(err) {
throw authKey;
}
//var authKeyHash = sha1BytesSync(authKey),
let authKeyHash = await CryptoWorker.sha1Hash(new Uint8Array(authKey)),
authKeyAux = authKeyHash.slice(0, 8),
authKeyId = authKeyHash.slice(-8);
if(DEBUG) {
this.log('Got Set_client_DH_params_answer', response._, authKey);
}
switch(response._) {
case 'dh_gen_ok':
var newNonceHash1 = (await CryptoWorker.sha1Hash(auth.newNonce.concat([1], authKeyAux))).slice(-16);
//var newNonceHash1 = sha1BytesSync(auth.newNonce.concat([1], authKeyAux)).slice(-16);
if(!bytesCmp(newNonceHash1, response.new_nonce_hash1)) {
throw new Error('[MT] Set_client_DH_params_answer new_nonce_hash1 mismatch');
}
var serverSalt = bytesXor(auth.newNonce.slice(0, 8), auth.serverNonce.slice(0, 8));
if(DEBUG) {
this.log('Auth successfull!', authKeyId, authKey, serverSalt);
}
auth.authKeyId = authKeyId;
auth.authKey = authKey;
auth.serverSalt = serverSalt;
return auth;
break;
case 'dh_gen_retry':
//var newNonceHash2 = sha1BytesSync(auth.newNonce.concat([2], authKeyAux)).slice(-16);
var newNonceHash2 = (await CryptoWorker.sha1Hash(auth.newNonce.concat([2], authKeyAux))).slice(-16);
if(!bytesCmp(newNonceHash2, response.new_nonce_hash2)) {
throw new Error('[MT] Set_client_DH_params_answer new_nonce_hash2 mismatch');
}
return this.mtpSendSetClientDhParams(auth);
case 'dh_gen_fail':
//var newNonceHash3 = sha1BytesSync(auth.newNonce.concat([3], authKeyAux)).slice(-16);
var newNonceHash3 = (await CryptoWorker.sha1Hash(auth.newNonce.concat([3], authKeyAux))).slice(-16);
if(!bytesCmp(newNonceHash3, response.new_nonce_hash3)) {
throw new Error('[MT] Set_client_DH_params_answer new_nonce_hash3 mismatch');
}
throw new Error('[MT] Set_client_DH_params_answer fail');
}
}
// mtpAuth
public async auth(dcId: number): Promise<AuthOptions> {
if(dcId in this.cached) {
return this.cached[dcId];
}
let nonce = /* fNonce ? fNonce : */new Uint8Array(16).randomize();
/* var nonce = new Array(16);
MTProto.secureRandom.nextBytes(nonce); */
if(!dcConfigurator.chooseServer(dcId)) {
return Promise.reject(new Error('[MT] No server found for dc ' + dcId));
}
try {
let promise = this.mtpSendReqPQ({dcId, nonce});
this.cached[dcId] = promise;
return await promise;
} catch(err) {
delete this.cached[dcId];
throw err;
}
}
}
export default new Authorizer();