org.telegram.api.engine.TelegramApi Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of telegramapi Show documentation
Show all versions of telegramapi Show documentation
Java library to create Telegram Clients
package org.telegram.api.engine;
import org.telegram.api.TLApiContext;
import org.telegram.api.TLConfig;
import org.telegram.api.auth.TLExportedAuthorization;
import org.telegram.api.engine.file.Downloader;
import org.telegram.api.engine.file.Uploader;
import org.telegram.api.engine.storage.AbsApiState;
import org.telegram.api.functions.TLRequestInitConnection;
import org.telegram.api.functions.TLRequestInvokeWithLayer;
import org.telegram.api.functions.auth.TLRequestAuthExportAuthorization;
import org.telegram.api.functions.auth.TLRequestAuthImportAuthorization;
import org.telegram.api.functions.help.TLRequestHelpGetConfig;
import org.telegram.api.functions.upload.TLRequestUploadGetCdnFile;
import org.telegram.api.functions.upload.TLRequestUploadGetFile;
import org.telegram.api.functions.upload.TLRequestUploadSaveBigFilePart;
import org.telegram.api.functions.upload.TLRequestUploadSaveFilePart;
import org.telegram.api.input.filelocation.TLAbsInputFileLocation;
import org.telegram.api.updates.TLAbsUpdates;
import org.telegram.api.upload.cdn.TLAbsCdnFile;
import org.telegram.api.upload.file.TLAbsFile;
import org.telegram.mtproto.CallWrapper;
import org.telegram.mtproto.MTProto;
import org.telegram.mtproto.MTProtoCallback;
import org.telegram.mtproto.pq.Authorizer;
import org.telegram.mtproto.pq.PqAuth;
import org.telegram.mtproto.state.ConnectionInfo;
import org.telegram.mtproto.util.BytesCache;
import org.telegram.tl.*;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.TreeMap;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Created with IntelliJ IDEA.
* User: Ruben Bermudez
* Date: 04.11.13
* Time: 21:54
*/
public class TelegramApi {
private static final AtomicInteger rpcCallIndex = new AtomicInteger(0);
private static final AtomicInteger instanceIndex = new AtomicInteger(1000);
private static final int CHANNELS_MAIN = 1;
private static final int CHANNELS_FS = 2;
private static final int DEFAULT_TIMEOUT_CHECK = 15000;
private static final int DEFAULT_TIMEOUT = 15000;
private static final int FILE_TIMEOUT = 45000;
private final String TAG;
private final int INSTANCE_INDEX;
private final HashMap dcProtos = new HashMap<>();
private final HashMap dcSync = new HashMap<>();
private final HashMap callbacks = new HashMap<>();
private final HashMap sentRequests = new HashMap<>();
private final TreeMap timeoutTimes = new TreeMap<>();
private final TreeMap dcRequired = new TreeMap<>();
private static final int DEFAULTCOMPETABLETIMEOUTMILLIS = 30000;
private boolean isClosed;
private int primaryDc;
private MTProto mainProto;
private ProtoCallback callback;
private SenderThread senderThread;
private TLApiContext apiContext;
private TimeoutThread timeoutThread;
private ConnectionThread dcThread;
private HashSet registeredInApi = new HashSet();
private AbsApiState state;
private AppInfo appInfo;
private ApiCallback apiCallback;
private Downloader downloader;
private Uploader uploader;
/**
* Instantiates a new Telegram api.
*
* @param state the state
* @param _appInfo the _ app info
* @param _apiCallback the _ api callback
*/
public TelegramApi(AbsApiState state, AppInfo _appInfo, ApiCallback _apiCallback) {
this.INSTANCE_INDEX = instanceIndex.incrementAndGet();
this.TAG = "TelegramApi#" + this.INSTANCE_INDEX;
long start = System.currentTimeMillis();
this.apiCallback = _apiCallback;
this.appInfo = _appInfo;
this.state = state;
this.primaryDc = state.getPrimaryDc();
this.isClosed = false;
this.callback = new ProtoCallback();
Logger.d(this.TAG, "Phase 0 in " + (System.currentTimeMillis() - start) + " ms");
start = System.currentTimeMillis();
this.apiContext = new TLApiContext() {
private AtomicInteger integer = new AtomicInteger(0);
@Override
public TLObject deserializeMessage(int clazzId, InputStream stream) throws IOException {
if ((this.integer.incrementAndGet() % 10) == 9) {
Thread.yield();
}
return super.deserializeMessage(clazzId, stream);
}
@Override
public TLBytes allocateBytes(int size) {
return new TLBytes(BytesCache.getInstance().allocate(size), 0, size);
}
@Override
public void releaseBytes(TLBytes unused) {
BytesCache.getInstance().put(unused.getData());
}
};
Logger.d(this.TAG, "Phase 1 in " + (System.currentTimeMillis() - start) + " ms");
start = System.currentTimeMillis();
this.timeoutThread = new TimeoutThread();
this.timeoutThread.start();
this.dcThread = new ConnectionThread();
this.dcThread.start();
this.senderThread = new SenderThread();
this.senderThread.start();
Logger.d(this.TAG, "Phase 2 in " + (System.currentTimeMillis() - start) + " ms");
start = System.currentTimeMillis();
this.downloader = new Downloader(this);
this.uploader = new Uploader(this);
Logger.d(this.TAG, "Phase 3 in " + (System.currentTimeMillis() - start) + " ms");
}
/**
* Gets downloader.
*
* @return the downloader
*/
public Downloader getDownloader() {
return this.downloader;
}
/**
* Gets uploader.
*
* @return the uploader
*/
public Uploader getUploader() {
return this.uploader;
}
/**
* Switch to dc.
*
* @param dcId the dc id
*/
public void switchToDc(int dcId) {
if (this.mainProto != null) {
this.mainProto.close();
}
this.mainProto = null;
this.primaryDc = dcId;
this.state.setPrimaryDc(dcId);
synchronized (this.dcRequired) {
this.dcRequired.notifyAll();
}
}
@Override
public String toString() {
return "api#" + this.INSTANCE_INDEX;
}
private TLMethod wrapForDc(int dcId, TLMethod method) {
if (this.registeredInApi.contains(dcId)) {
return new TLRequestInvokeWithLayer(method);
}
return new TLRequestInvokeWithLayer(new TLRequestInitConnection(
this.appInfo.getApiId(), this.appInfo.getDeviceModel(), this.appInfo.getSystemVersion(), this.appInfo.getAppVersion(), this.appInfo.getLangCode(), method));
}
/**
* Gets state.
*
* @return the state
*/
public AbsApiState getState() {
return this.state;
}
/**
* Gets api context.
*
* @return the api context
*/
public TLApiContext getApiContext() {
return this.apiContext;
}
/**
* On message arrived.
*
* @param object the object
*/
protected void onMessageArrived(TLObject object) {
if (object instanceof TLAbsUpdates) {
Logger.d(this.TAG, "<< update " + object.toString());
this.apiCallback.onUpdate((TLAbsUpdates) object);
} else {
Logger.d(this.TAG, "<< unknown object " + object.toString());
}
}
/**
* Is closed.
*
* @return the boolean
*/
public boolean isClosed() {
return this.isClosed;
}
/**
* Close void.
*/
public void close() {
if (!this.isClosed) {
this.apiCallback.onAuthCancelled(this);
this.isClosed = true;
if (this.timeoutThread != null) {
this.timeoutThread.interrupt();
this.timeoutThread = null;
}
this.mainProto.close();
}
}
/**
* Reset network backoff.
*/
public void resetNetworkBackoff() {
if (this.mainProto != null) {
this.mainProto.resetNetworkBackoff();
}
for (MTProto mtProto : this.dcProtos.values()) {
mtProto.resetNetworkBackoff();
}
}
/**
* Reset connection info.
*/
public void resetConnectionInfo() {
this.mainProto.reloadConnectionInformation();
synchronized (this.dcProtos) {
for (MTProto proto : this.dcProtos.values()) {
proto.reloadConnectionInformation();
}
}
}
// Basic sync and async methods
private void doRpcCall(TLMethod method, int timeout, RpcCallback callback, int destDc) {
doRpcCall(method, timeout, callback, destDc, true);
}
private void doRpcCall(TLMethod method, int timeout, RpcCallback callback, int destDc,
boolean authRequired) {
if (this.isClosed) {
if (callback != null) {
callback.onError(0, null);
}
return;
}
int localRpcId = rpcCallIndex.getAndIncrement();
synchronized (this.callbacks) {
RpcCallbackWrapper wrapper = new RpcCallbackWrapper(localRpcId, method, callback);
wrapper.dcId = destDc;
wrapper.timeout = timeout;
wrapper.isAuthRequred = authRequired;
this.callbacks.put(localRpcId, wrapper);
if (callback != null) {
long timeoutTime = System.nanoTime() + timeout * 2 * 1000 * 1000L;
synchronized (this.timeoutTimes) {
while (this.timeoutTimes.containsKey(timeoutTime)) {
timeoutTime++;
}
this.timeoutTimes.put(timeoutTime, localRpcId);
this.timeoutTimes.notifyAll();
}
wrapper.timeoutTime = timeoutTime;
}
if (authRequired) {
checkDcAuth(destDc);
} else {
checkDc(destDc);
}
this.callbacks.notifyAll();
}
Logger.d(this.TAG, ">> #" + +localRpcId + ": " + method.toString());
}
private T doRpcCall(TLMethod method, int timeout, int destDc) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(method, timeout, destDc, true);
}
private T doRpcCall(TLMethod method, int timeout, int destDc, boolean authRequired) throws RpcException, java.util.concurrent.TimeoutException {
if (this.isClosed) {
throw new RpcException(0, "Connection is closed");
}
T resultObject = null;
final CompletableFuture completableFuture = new CompletableFuture<>();
doRpcCall(method, timeout, new RpcCallback() {
@Override
public void onResult(T result) {
completableFuture.complete(result);
}
@Override
public void onError(int errorCode, String message) {
completableFuture.completeExceptionally(new RpcException(errorCode, message));
}
}, destDc, authRequired);
try {
resultObject = completableFuture.get(timeout, TimeUnit.MILLISECONDS);
} catch (InterruptedException e) {
Logger.w(TAG, method.toString());
Logger.e(TAG, e);
} catch (java.util.concurrent.TimeoutException e) {
Logger.w(TAG, method.toString());
Logger.e(TAG, e);
throw e;
} catch (ExecutionException e) {
Logger.w(TAG, method.toString());
Logger.e(TAG, e);
if (e.getCause() instanceof RpcException) {
throw (RpcException) e.getCause();
}
}
return resultObject;
}
/**
* Do rpc call weak.
*
* @param the type parameter
* @param method the method
*/
// Public async methods
public void doRpcCallWeak(TLMethod method) {
doRpcCallWeak(method, DEFAULT_TIMEOUT);
}
/**
* Do rpc call weak.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
*/
public void doRpcCallWeak(TLMethod method, int timeout) {
doRpcCall(method, timeout, null);
}
/**
* Do rpc call.
*
* @param the type parameter
* @param method the method
* @param callback the callback
*/
public void doRpcCall(TLMethod method, RpcCallback callback) {
doRpcCall(method, DEFAULT_TIMEOUT, callback);
}
/**
* Do rpc call.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @param callback the callback
*/
public void doRpcCall(TLMethod method, int timeout, RpcCallback callback) {
doRpcCall(method, timeout, callback, 0);
}
// Public sync methods
/**
* Do rpc call.
*
* @param the type parameter
* @param method the method
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCall(TLMethod method) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(method, DEFAULTCOMPETABLETIMEOUTMILLIS);
}
/**
* Do rpc call.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCall(TLMethod method, int timeout) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(method, timeout, 0);
}
/**
* Do rpc call side.
*
* @param the type parameter
* @param method the method
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallSide(TLMethod method) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(method, DEFAULTCOMPETABLETIMEOUTMILLIS, this.primaryDc, true);
}
/**
* Do rpc call side.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallSide(TLMethod method, int timeout) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(method, timeout, this.primaryDc, true);
}
/**
* Do rpc call side gzip.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallSideGzip(TLMethod method, int timeout) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(new GzipRequest(method), timeout, this.primaryDc, true);
}
/**
* Do rpc call gzip.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallGzip(TLMethod method, int timeout) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCall(new GzipRequest(method), timeout, 0);
}
/**
* Do rpc call non auth.
*
* @param the type parameter
* @param method the method
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallNonAuth(TLMethod method) throws RpcException, java.util.concurrent.TimeoutException {
return doRpcCallNonAuth(method, DEFAULTCOMPETABLETIMEOUTMILLIS, this.primaryDc);
}
/**
* Do rpc call non auth.
*
* @param the type parameter
* @param method the method
* @param dcId the dc id
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallNonAuth(TLMethod method, int dcId) throws IOException, java.util.concurrent.TimeoutException {
return doRpcCallNonAuth(method, DEFAULTCOMPETABLETIMEOUTMILLIS, dcId);
}
/**
* Do rpc call non auth.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @param dcId the dc id
* @return the t
* @throws IOException the iO exception
*/
public T doRpcCallNonAuth(TLMethod method, int timeout, int dcId) throws RpcException, java.util.concurrent.TimeoutException {
return doRpcCall(method, timeout, dcId, false);
}
/**
* Do rpc call non auth.
*
* @param the type parameter
* @param method the method
* @param timeout the timeout
* @param callback the callback
*/
public void doRpcCallNonAuth(TLMethod method, int timeout, RpcCallback callback) {
doRpcCall(method, timeout, callback, 0, false);
}
/**
* Do save file part.
*
* @param _fileId the _ file id
* @param _filePart the _ file part
* @param _bytes the _ bytes
* @return the boolean
* @throws IOException the iO exception
*/
public boolean doSaveFilePart(long _fileId, int _filePart, byte[] _bytes) throws IOException, java.util.concurrent.TimeoutException {
final TLRequestUploadSaveFilePart tlRequestUploadSaveFilePart = new TLRequestUploadSaveFilePart();
tlRequestUploadSaveFilePart.setFileId(_fileId);
tlRequestUploadSaveFilePart.setFilePart(_filePart);
tlRequestUploadSaveFilePart.setBytes(new TLBytes(_bytes));
final TLBool res = doRpcCall(tlRequestUploadSaveFilePart, FILE_TIMEOUT, this.primaryDc, true);
return res instanceof TLBoolTrue;
}
/**
* Do save big file part.
*
* @param _fileId the _ file id
* @param _filePart the _ file part
* @param _totalParts the _ total parts
* @param _bytes the _ bytes
* @return the boolean
* @throws IOException the iO exception
*/
public boolean doSaveBigFilePart(long _fileId, int _filePart, int _totalParts, byte[] _bytes) throws IOException, java.util.concurrent.TimeoutException {
final TLRequestUploadSaveBigFilePart tlRequestUploadSaveBigFilePart = new TLRequestUploadSaveBigFilePart();
tlRequestUploadSaveBigFilePart.setFileId(_fileId);
tlRequestUploadSaveBigFilePart.setFilePart(_filePart);
tlRequestUploadSaveBigFilePart.setFileTotalParts(_totalParts);
tlRequestUploadSaveBigFilePart.setBytes(new TLBytes(_bytes));
final TLBool res = doRpcCall(tlRequestUploadSaveBigFilePart, FILE_TIMEOUT, this.primaryDc);
return res instanceof TLBoolTrue;
}
/**
* Do get file.
*
* @param dcId the dc id
* @param _location the _ location
* @param _offset the _ offset
* @param _limit the _ limit
* @return the tL file
* @throws IOException the iO exception
*/
public TLAbsFile doGetFile(int dcId, TLAbsInputFileLocation _location, int _offset, int _limit) throws IOException, java.util.concurrent.TimeoutException {
final TLRequestUploadGetFile tlRequestUploadGetFile = new TLRequestUploadGetFile();
tlRequestUploadGetFile.setLocation(_location);
tlRequestUploadGetFile.setOffset(_offset);
tlRequestUploadGetFile.setLimit(_limit);
return doRpcCall(tlRequestUploadGetFile, FILE_TIMEOUT, dcId);
}
public TLAbsCdnFile doGetCdnFile(int dcId, TLBytes fileToken, int offset, int limit) throws IOException, java.util.concurrent.TimeoutException {
final TLRequestUploadGetCdnFile tlRequestUploadGetCdnFile = new TLRequestUploadGetCdnFile();
tlRequestUploadGetCdnFile.setFileToken(fileToken);
tlRequestUploadGetCdnFile.setOffset(offset);
tlRequestUploadGetCdnFile.setLimit(limit);
return doRpcCall(tlRequestUploadGetCdnFile, FILE_TIMEOUT, dcId);
}
private void checkDcAuth(int dcId) {
if (dcId != 0) {
synchronized (this.dcProtos) {
if (!this.dcProtos.containsKey(dcId)) {
synchronized (this.dcRequired) {
this.dcRequired.put(dcId, true);
this.dcRequired.notifyAll();
}
} else if (!this.state.isAuthenticated(dcId)) {
synchronized (this.dcRequired) {
this.dcRequired.put(dcId, true);
this.dcRequired.notifyAll();
}
}
}
}
}
private void checkDc(int dcId) {
if (dcId != 0) {
synchronized (this.dcProtos) {
if (!this.dcProtos.containsKey(dcId)) {
synchronized (this.dcRequired) {
if (!this.dcRequired.containsKey(dcId)) {
this.dcRequired.put(dcId, false);
}
this.dcRequired.notifyAll();
}
}
}
} else if (this.mainProto == null) {
synchronized (this.dcRequired) {
this.dcRequired.notifyAll();
}
}
}
public void notifyCallbacks() {
synchronized (this.callbacks) {
this.callbacks.notifyAll();
}
}
private class ProtoCallback implements MTProtoCallback {
@Override
public void onSessionCreated(MTProto proto) {
if (TelegramApi.this.isClosed) {
return;
}
Logger.w(TelegramApi.this.TAG, proto + ": onSessionCreated");
if (proto == TelegramApi.this.mainProto) {
TelegramApi.this.registeredInApi.add(TelegramApi.this.primaryDc);
} else {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
TelegramApi.this.registeredInApi.add(p.getKey());
break;
}
}
}
TelegramApi.this.apiCallback.onUpdatesInvalidated(TelegramApi.this);
}
@Override
public void onAuthInvalidated(MTProto proto) {
if (TelegramApi.this.isClosed) {
return;
}
if (proto == TelegramApi.this.mainProto) {
synchronized (TelegramApi.this.dcRequired) {
TelegramApi.this.mainProto.close();
TelegramApi.this.mainProto = null;
TelegramApi.this.state.setAuthenticated(TelegramApi.this.primaryDc, false);
TelegramApi.this.dcRequired.notifyAll();
}
synchronized (TelegramApi.this.dcProtos) {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
p.getValue().close();
TelegramApi.this.state.setAuthenticated(p.getKey(), false);
}
}
TelegramApi.this.apiCallback.onAuthCancelled(TelegramApi.this);
} else {
synchronized (TelegramApi.this.dcProtos) {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
TelegramApi.this.state.setAuthenticated(p.getKey(), false);
TelegramApi.this.dcProtos.remove(p.getKey());
break;
}
}
}
synchronized (TelegramApi.this.dcRequired) {
TelegramApi.this.dcRequired.notifyAll();
}
}
}
@Override
public void onApiMessage(byte[] message, MTProto proto) {
if (TelegramApi.this.isClosed) {
return;
}
if (proto == TelegramApi.this.mainProto) {
TelegramApi.this.registeredInApi.add(TelegramApi.this.primaryDc);
} else {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
TelegramApi.this.registeredInApi.add(p.getKey());
break;
}
}
}
try {
TLObject object = TelegramApi.this.apiContext.deserializeMessage(message);
onMessageArrived(object);
} catch (Throwable t) {
Logger.e(TelegramApi.this.TAG, t);
}
}
@Override
public void onRpcResult(int callId, byte[] response, MTProto proto) {
if (TelegramApi.this.isClosed) {
return;
}
if (proto == TelegramApi.this.mainProto) {
TelegramApi.this.registeredInApi.add(TelegramApi.this.primaryDc);
} else {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
TelegramApi.this.registeredInApi.add(p.getKey());
break;
}
}
}
try {
RpcCallbackWrapper currentCallback = null;
synchronized (TelegramApi.this.callbacks) {
if (TelegramApi.this.sentRequests.containsKey(callId)) {
currentCallback = TelegramApi.this.callbacks.remove(TelegramApi.this.sentRequests.remove(callId));
}
}
if (currentCallback != null && currentCallback.method != null) {
long start = System.currentTimeMillis();
TLObject object = currentCallback.method.deserializeResponse(response, TelegramApi.this.apiContext);
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " deserialized " + object + " in " + (System.currentTimeMillis() - start) + " ms");
synchronized (currentCallback) {
if (currentCallback.isCompleted) {
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " ignored " + object + " in " + currentCallback.elapsed() + " ms");
return;
} else {
currentCallback.isCompleted = true;
}
}
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " " + object + " in " + currentCallback.elapsed() + " ms");
synchronized (TelegramApi.this.timeoutTimes) {
TelegramApi.this.timeoutTimes.remove(currentCallback.timeoutTime);
}
if (currentCallback.callback != null) {
currentCallback.callback.onResult(object);
}
}
} catch (Throwable t) {
Logger.e(TelegramApi.this.TAG, t);
}
}
@Override
public void onRpcError(int callId, int errorCode, String message, MTProto proto) {
if (TelegramApi.this.isClosed) {
return;
}
if (errorCode == 400 && message != null &&
(message.startsWith("CONNECTION_NOT_INITED") || message.startsWith("CONNECTION_LAYER_INVALID"))) {
Logger.w(TelegramApi.this.TAG, proto + ": (!)Error #400 " + message);
int dc = -1;
if (proto == TelegramApi.this.mainProto) {
dc = TelegramApi.this.primaryDc;
} else {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
dc = p.getKey();
break;
}
}
}
if (dc < 0) {
return;
}
TelegramApi.this.registeredInApi.remove(dc);
RpcCallbackWrapper currentCallback;
synchronized (TelegramApi.this.callbacks) {
currentCallback = TelegramApi.this.callbacks.remove(TelegramApi.this.sentRequests.remove(callId));
if (currentCallback != null) {
currentCallback.isSent = false;
TelegramApi.this.callbacks.notifyAll();
}
}
return;
} else {
if (proto == TelegramApi.this.mainProto) {
TelegramApi.this.registeredInApi.add(TelegramApi.this.primaryDc);
} else {
for (Map.Entry p : TelegramApi.this.dcProtos.entrySet()) {
if (p.getValue() == proto) {
TelegramApi.this.registeredInApi.add(p.getKey());
break;
}
}
}
}
try {
RpcCallbackWrapper currentCallback = null;
synchronized (TelegramApi.this.callbacks) {
if (TelegramApi.this.sentRequests.containsKey(callId)) {
currentCallback = TelegramApi.this.callbacks.remove(TelegramApi.this.sentRequests.remove(callId));
}
}
if (currentCallback != null) {
synchronized (currentCallback) {
if (currentCallback.isCompleted) {
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " ignored error #" + errorCode + " " + message + " in " + currentCallback.elapsed() + " ms");
return;
} else {
currentCallback.isCompleted = true;
}
}
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " error #" + errorCode + " " + message + " in " + currentCallback.elapsed() + " ms");
synchronized (TelegramApi.this.timeoutTimes) {
TelegramApi.this.timeoutTimes.remove(currentCallback.timeoutTime);
}
if (currentCallback.callback != null) {
currentCallback.callback.onError(errorCode, message);
}
}
} catch (Throwable t) {
Logger.e(TelegramApi.this.TAG, t);
}
}
@Override
public void onConfirmed(int callId) {
RpcCallbackWrapper currentCallback = null;
synchronized (TelegramApi.this.callbacks) {
if (TelegramApi.this.sentRequests.containsKey(callId)) {
currentCallback = TelegramApi.this.callbacks.get(TelegramApi.this.sentRequests.get(callId));
}
}
if (currentCallback != null) {
Logger.d(TelegramApi.this.TAG, "<< #" + +currentCallback.id + " confirmed in " + currentCallback.elapsed() + " ms");
synchronized (currentCallback) {
if (currentCallback.isCompleted || currentCallback.isConfirmed) {
return;
} else {
currentCallback.isConfirmed = true;
}
}
if (currentCallback.callback instanceof RpcCallbackEx) {
((RpcCallbackEx) currentCallback.callback).onConfirmed();
}
}
}
}
private class SenderThread extends Thread {
/**
* Instantiates a new Sender thread.
*/
public SenderThread() {
setName("Sender#" + hashCode());
}
@Override
public void run() {
setPriority(Thread.MIN_PRIORITY);
while (!TelegramApi.this.isClosed) {
Logger.d(TelegramApi.this.TAG, "Sender iteration");
RpcCallbackWrapper wrapper = null;
synchronized (TelegramApi.this.callbacks) {
for (RpcCallbackWrapper w : TelegramApi.this.callbacks.values()) {
if (!w.isSent) {
if (w.dcId == 0 && TelegramApi.this.mainProto != null) {
if (TelegramApi.this.state.isAuthenticated(TelegramApi.this.primaryDc) || !w.isAuthRequred) {
wrapper = w;
break;
}
}
if (w.dcId != 0 && TelegramApi.this.dcProtos.containsKey(w.dcId)) {
if (TelegramApi.this.state.isAuthenticated(w.dcId) || !w.isAuthRequred) {
wrapper = w;
break;
}
}
}
}
if (wrapper == null) {
try {
TelegramApi.this.callbacks.wait();
} catch (InterruptedException e) {
Logger.e(TelegramApi.this.TAG, e);
return;
}
continue;
}
}
if (TelegramApi.this.mainProto == null) {
continue;
}
if (wrapper.dcId == 0) {
if (!TelegramApi.this.state.isAuthenticated(TelegramApi.this.primaryDc) && wrapper.isAuthRequred) {
continue;
}
synchronized (TelegramApi.this.callbacks) {
boolean isHighPriority = wrapper.callback != null && wrapper.callback instanceof RpcCallbackEx;
int rpcId = TelegramApi.this.mainProto.sendRpcMessage(wrapper.method, wrapper.timeout, isHighPriority);
TelegramApi.this.sentRequests.put(rpcId, wrapper.id);
wrapper.isSent = true;
Logger.d(TelegramApi.this.TAG, "#> #" + wrapper.id + " sent to MTProto #" + TelegramApi.this.mainProto.getInstanceIndex() + " with id #" + rpcId);
}
} else {
if (!TelegramApi.this.dcProtos.containsKey(wrapper.dcId) || (!TelegramApi.this.state.isAuthenticated(wrapper.dcId) && wrapper.isAuthRequred)) {
continue;
}
MTProto proto = TelegramApi.this.dcProtos.get(wrapper.dcId);
synchronized (TelegramApi.this.callbacks) {
boolean isHighPriority = wrapper.callback != null && wrapper.callback instanceof RpcCallbackEx;
int rpcId = proto.sendRpcMessage(wrapper.method, wrapper.timeout, isHighPriority);
TelegramApi.this.sentRequests.put(rpcId, wrapper.id);
wrapper.isSent = true;
Logger.d(TelegramApi.this.TAG, "#> #" + wrapper.id + " sent to MTProto #" + proto.getInstanceIndex() + " with id #" + rpcId);
}
}
}
}
}
private class ConnectionThread extends Thread {
/**
* Instantiates a new Connection thread.
*/
public ConnectionThread() {
setName("Connection#" + hashCode());
}
private MTProto waitForDc(final int dcId) throws IOException, java.util.concurrent.TimeoutException {
Logger.d(TelegramApi.this.TAG, "#" + dcId + ": waitForDc");
if (TelegramApi.this.isClosed) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Api is closed");
throw new TimeoutException();
}
// if (!state.isAuthenticated(primaryDc)) {
// Logger.w(TAG, "#" + dcId + ": Dc is not authenticated");
// throw new TimeoutException();
// }
Object syncObj;
synchronized (TelegramApi.this.dcSync) {
syncObj = TelegramApi.this.dcSync.get(dcId);
if (syncObj == null) {
syncObj = new Object();
TelegramApi.this.dcSync.put(dcId, syncObj);
}
}
synchronized (syncObj) {
MTProto proto;
synchronized (TelegramApi.this.dcProtos) {
proto = TelegramApi.this.dcProtos.get(dcId);
if (proto != null) {
if (proto.isClosed()) {
Logger.d(TelegramApi.this.TAG, "#" + dcId + "proto removed because of death");
TelegramApi.this.dcProtos.remove(dcId);
proto = null;
}
}
}
if (proto == null) {
Logger.d(TelegramApi.this.TAG, "#" + dcId + ": Creating proto for dc");
ConnectionInfo[] connectionInfo = TelegramApi.this.state.getAvailableConnections(dcId);
if (connectionInfo.length == 0) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Unable to find proper dc config");
TLConfig config = doRpcCall(new TLRequestHelpGetConfig());
TelegramApi.this.state.updateSettings(config);
resetConnectionInfo();
connectionInfo = TelegramApi.this.state.getAvailableConnections(dcId);
}
if (connectionInfo.length == 0) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Still unable to find proper dc config");
throw new TimeoutException();
}
if (TelegramApi.this.state.getAuthKey(dcId) != null) {
byte[] authKey = TelegramApi.this.state.getAuthKey(dcId);
if (authKey == null) {
throw new TimeoutException();
}
proto = new MTProto(TelegramApi.this.state.getMtProtoState(dcId), TelegramApi.this.callback,
new CallWrapper() {
@Override
public TLObject wrapObject(TLMethod srcRequest) {
return wrapForDc(dcId, srcRequest);
}
}, CHANNELS_FS);
TelegramApi.this.dcProtos.put(dcId, proto);
return proto;
} else {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Creating key");
Authorizer authorizer = new Authorizer();
PqAuth auth = authorizer.doAuth(connectionInfo);
if (auth == null) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Timed out");
throw new TimeoutException();
}
TelegramApi.this.state.putAuthKey(dcId, auth.getAuthKey());
TelegramApi.this.state.setAuthenticated(dcId, false);
TelegramApi.this.state.getMtProtoState(dcId).initialServerSalt(auth.getServerSalt());
byte[] authKey = TelegramApi.this.state.getAuthKey(dcId);
if (authKey == null) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": auth key == null");
throw new TimeoutException();
}
proto = new MTProto(TelegramApi.this.state.getMtProtoState(dcId), TelegramApi.this.callback,
new CallWrapper() {
@Override
public TLObject wrapObject(TLMethod srcRequest) {
return wrapForDc(dcId, srcRequest);
}
}, CHANNELS_FS);
TelegramApi.this.dcProtos.put(dcId, proto);
return proto;
}
} else {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": returning proper proto");
return proto;
}
}
}
private MTProto waitForAuthDc(final int dcId) throws IOException, java.util.concurrent.TimeoutException {
Logger.d(TelegramApi.this.TAG, "#" + dcId + ": waitForAuthDc");
if (TelegramApi.this.isClosed) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": Api is closed");
throw new TimeoutException();
}
MTProto proto = waitForDc(dcId);
if (!TelegramApi.this.state.isAuthenticated(dcId)) {
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": exporting auth");
TLRequestAuthExportAuthorization exportAuthorization = new TLRequestAuthExportAuthorization();
exportAuthorization.setDcId(dcId);
TLExportedAuthorization exAuth = doRpcCall(exportAuthorization);
Logger.w(TelegramApi.this.TAG, "#" + dcId + ": importing auth");
TLRequestAuthImportAuthorization tlRequestAuthImportAuthorization = new TLRequestAuthImportAuthorization();
tlRequestAuthImportAuthorization.setId(exAuth.getId());
tlRequestAuthImportAuthorization.setBytes(exAuth.getBytes());
doRpcCallNonAuth(tlRequestAuthImportAuthorization, DEFAULT_TIMEOUT, dcId);
TelegramApi.this.state.setAuthenticated(dcId, true);
}
return proto;
}
@Override
public void run() {
setPriority(Thread.MIN_PRIORITY);
while (!TelegramApi.this.isClosed) {
Logger.d(TelegramApi.this.TAG, "Connection iteration");
if (TelegramApi.this.mainProto == null) {
if (TelegramApi.this.state.getAuthKey(TelegramApi.this.primaryDc) == null) {
try {
long start = System.currentTimeMillis();
waitForDc(TelegramApi.this.primaryDc);
TelegramApi.this.mainProto = new MTProto(TelegramApi.this.state.getMtProtoState(TelegramApi.this.primaryDc), TelegramApi.this.callback,
new CallWrapper() {
@Override
public TLObject wrapObject(TLMethod srcRequest) {
return wrapForDc(TelegramApi.this.primaryDc, srcRequest);
}
}, CHANNELS_MAIN);
Logger.d(TelegramApi.this.TAG, "#MTProto #" + TelegramApi.this.mainProto.getInstanceIndex() + " created in " + (System.currentTimeMillis() - start) + " ms");
} catch (IOException | java.util.concurrent.TimeoutException e) {
Logger.e(TelegramApi.this.TAG, e);
try {
Thread.sleep(1000);
continue;
} catch (InterruptedException e1) {
Logger.e(TelegramApi.this.TAG, e1);
return;
}
}
} else {
long start = System.currentTimeMillis();
TelegramApi.this.mainProto = new MTProto(TelegramApi.this.state.getMtProtoState(TelegramApi.this.primaryDc), TelegramApi.this.callback,
new CallWrapper() {
@Override
public TLObject wrapObject(TLMethod srcRequest) {
return wrapForDc(TelegramApi.this.primaryDc, srcRequest);
}
}, CHANNELS_MAIN);
Logger.d(TelegramApi.this.TAG, "#MTProto #" + TelegramApi.this.mainProto.getInstanceIndex() + " created in " + (System.currentTimeMillis() - start) + " ms");
}
synchronized (TelegramApi.this.callbacks) {
TelegramApi.this.callbacks.notifyAll();
}
continue;
}
Integer dcId = null;
Boolean authRequired = null;
synchronized (TelegramApi.this.dcRequired) {
if (TelegramApi.this.dcRequired.isEmpty()) {
dcId = null;
authRequired = null;
} else {
try {
dcId = TelegramApi.this.dcRequired.firstKey();
} catch (Exception e) {
Logger.e(TelegramApi.this.TAG, e);
}
}
if (dcId == null) {
try {
TelegramApi.this.dcRequired.wait();
} catch (InterruptedException e) {
// e.printStackTrace();
}
continue;
}
authRequired = TelegramApi.this.dcRequired.remove(dcId);
}
if (TelegramApi.this.dcProtos.containsKey(dcId)) {
if (authRequired && !TelegramApi.this.state.isAuthenticated(dcId) && TelegramApi.this.state.isAuthenticated(TelegramApi.this.primaryDc)) {
try {
waitForAuthDc(dcId);
synchronized (TelegramApi.this.callbacks) {
TelegramApi.this.callbacks.notifyAll();
}
} catch (IOException | java.util.concurrent.TimeoutException e) {
try {
Thread.sleep(1000);
continue;
} catch (InterruptedException e1) {
Logger.e(TelegramApi.this.TAG, e1);
return;
}
}
}
} else {
try {
if (authRequired && !TelegramApi.this.state.isAuthenticated(dcId) && TelegramApi.this.state.isAuthenticated(TelegramApi.this.primaryDc)) {
waitForAuthDc(dcId);
} else {
waitForDc(dcId);
}
synchronized (TelegramApi.this.callbacks) {
TelegramApi.this.callbacks.notifyAll();
}
} catch (IOException | java.util.concurrent.TimeoutException e) {
Logger.e(TelegramApi.this.TAG, e);
}
}
}
}
}
private class TimeoutThread extends Thread {
/**
* Instantiates a new Timeout thread.
*/
public TimeoutThread() {
setName("Timeout#" + hashCode());
}
@Override
public void run() {
while (!TelegramApi.this.isClosed) {
Logger.d(TelegramApi.this.TAG, "Timeout Iteration");
Long key = null;
Integer id = null;
synchronized (TelegramApi.this.timeoutTimes) {
if (TelegramApi.this.timeoutTimes.isEmpty()) {
key = null;
} else {
try {
key = TelegramApi.this.timeoutTimes.firstKey();
} catch (Exception e) {
Logger.e(TelegramApi.this.TAG, e);
}
}
if (key == null) {
try {
TelegramApi.this.timeoutTimes.wait();
} catch (InterruptedException e) {
// e.printStackTrace();
}
continue;
}
long delta = (key - System.nanoTime()) / (1000 * 1000);
if (delta > 0) {
try {
TelegramApi.this.timeoutTimes.wait(delta);
} catch (InterruptedException e) {
// e.printStackTrace();
}
continue;
}
id = TelegramApi.this.timeoutTimes.remove(key);
if (id == null) {
continue;
}
}
RpcCallbackWrapper currentCallback;
synchronized (TelegramApi.this.callbacks) {
currentCallback = TelegramApi.this.callbacks.remove(id);
}
if (currentCallback != null) {
synchronized (currentCallback) {
if (currentCallback.isCompleted) {
Logger.d(TelegramApi.this.TAG, "RPC #" + id + ": Timeout ignored");
return;
} else {
currentCallback.isCompleted = true;
}
}
Logger.d(TelegramApi.this.TAG, "RPC #" + id + ": Timeout (" + currentCallback.elapsed() + " ms)");
currentCallback.callback.onError(0, null);
} else {
Logger.d(TelegramApi.this.TAG, "RPC #" + id + ": Timeout ignored2");
}
}
synchronized (TelegramApi.this.timeoutTimes) {
for (Map.Entry entry : TelegramApi.this.timeoutTimes.entrySet()) {
RpcCallbackWrapper currentCallback;
synchronized (TelegramApi.this.callbacks) {
currentCallback = TelegramApi.this.callbacks.remove(entry.getValue());
}
if (currentCallback != null) {
synchronized (currentCallback) {
if (currentCallback.isCompleted) {
return;
} else {
currentCallback.isCompleted = true;
}
}
Logger.d(TelegramApi.this.TAG, "RPC #" + entry.getValue() + ": Timeout (" + currentCallback.elapsed() + " ms)");
currentCallback.callback.onError(0, null);
}
}
}
}
}
private class RpcCallbackWrapper {
/**
* The Id.
*/
public int id;
/**
* The Request time.
*/
public long requestTime = System.currentTimeMillis();
/**
* The Is sent.
*/
public boolean isSent;
/**
* The Is completed.
*/
public boolean isCompleted;
/**
* The Is confirmed.
*/
public boolean isConfirmed;
/**
* The Callback.
*/
public RpcCallback callback;
/**
* The Timeout time.
*/
public long timeoutTime;
/**
* The Timeout.
*/
public long timeout;
/**
* The Method.
*/
public TLMethod method;
/**
* The Is auth requred.
*/
public boolean isAuthRequred;
/**
* The Dc id.
*/
public int dcId;
private RpcCallbackWrapper(int id, TLMethod method, RpcCallback callback) {
this.id = id;
this.method = method;
this.callback = callback;
}
/**
* Elapsed long.
*
* @return the long
*/
public long elapsed() {
return System.currentTimeMillis() - this.requestTime;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy