From 8e1fcfd0dd7dc8fa9f5450253d12e89dfafdd855 Mon Sep 17 00:00:00 2001 From: Trumeet Date: Mon, 11 Oct 2021 00:31:10 -0700 Subject: First Commit --- easy-tg.c | 300 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 300 insertions(+) create mode 100644 easy-tg.c (limited to 'easy-tg.c') diff --git a/easy-tg.c b/easy-tg.c new file mode 100644 index 0000000..1d89a2b --- /dev/null +++ b/easy-tg.c @@ -0,0 +1,300 @@ +#include "easy-tg.h" + +#include +#include + +#include +#include +#include + +void *td = NULL; +const char *tg_update_type = NULL; +json_object *tg_update = NULL; +int tg_errno = 0; +const void *tg_reg1 = NULL; +const void *tg_reg2 = NULL; +const void *tg_reg3 = NULL; +const void *tg_reg4 = NULL; +const void *tg_reg5 = NULL; +const void *tg_reg6 = NULL; + +#ifdef TG_ENABLE_PTHREAD +#include +static pthread_mutex_t mutex_close = PTHREAD_MUTEX_INITIALIZER; +#endif + +json_object *tg_mkreq(const char *extra, const char *type, const char *args_key, json_object *args) +{ + json_object *root = json_object_new_object(); + json_object_object_add(root, "@extra", json_object_new_string(extra)); + json_object_object_add(root, "@type", json_object_new_string(type)); + if(args_key != NULL) json_object_object_add(root, args_key, args); + return root; +} + +int tg_send_raw(bool execute, const char *req) +{ +#ifdef TG_DEBUG + printf("> %s\n", req); +#endif + if(execute) + td_json_client_execute(td, req); + else + td_json_client_send(td, req); + return 0; +} + +int tg_send(bool execute, json_object *root, const bool dispose) +{ + int r = 0; + const char *req = json_object_to_json_string(root); + r = tg_send_raw(execute, req); + if(dispose) + { + json_object_put(root); + } + return r; +} + +int tg_login_phone( + const char *phone_number + ) +{ + int r = 0; + json_object *raw_req = json_object_new_string(phone_number); + tg_send(false, tg_mkreq(TG_REQ_LOGIN_PHONE, "setAuthenticationPhoneNumber", "phone_number", raw_req), true); + goto cleanup; +cleanup: + return r; +} + +int tg_login_code( + const char *code + ) +{ + int r = 0; + json_object *raw_req = json_object_new_string(code); + tg_send(false, tg_mkreq(TG_REQ_LOGIN_CODE, "checkAuthenticationCode", "code", raw_req), true); + goto cleanup; +cleanup: + return r; +} + +int tg_login_pass( + const char *pass + ) +{ + int r = 0; + json_object *raw_req = json_object_new_string(pass); + tg_send(false, tg_mkreq(TG_REQ_LOGIN_PASS, "checkAuthenticationPassword", "password", raw_req), true); + goto cleanup; +cleanup: + return r; +} + +int tg_login_enc_key( + const char *encryption_key + ) +{ + int r = 0; + json_object *raw_req = json_object_new_string(encryption_key); + tg_send(false, tg_mkreq("-1", "checkDatabaseEncryptionKey", "encryption_key", raw_req), true); + goto cleanup; +cleanup: + return r; +} + +int tg_set_params( + const bool use_test_dc, + const char *database_directory, + const char *files_directory, + const bool use_file_database, + const bool use_chat_info_database, + const bool use_message_database, + const bool use_secret_chats, + const int32_t api_id, + const char *api_hash, + const char *system_language_code, + const char *device_model, + const char *system_version, + const char *application_version, + const bool enable_storage_optimizer, + const bool ignore_file_names + ) +{ + int r = 0; + json_object *raw_req = json_object_new_object(); + json_object_object_add(raw_req, "use_test_dc", json_object_new_boolean(use_test_dc)); + json_object_object_add(raw_req, "database_directory", json_object_new_string(database_directory)); + if(files_directory != NULL) + json_object_object_add(raw_req, "files_directory", json_object_new_string(files_directory)); + json_object_object_add(raw_req, "use_file_database", json_object_new_boolean(use_file_database)); + json_object_object_add(raw_req, "use_message_database", json_object_new_boolean(use_message_database)); + json_object_object_add(raw_req, "use_secret_chats", json_object_new_boolean(use_secret_chats)); + json_object_object_add(raw_req, "api_id", json_object_new_int(api_id)); + json_object_object_add(raw_req, "api_hash", json_object_new_string(api_hash)); + json_object_object_add(raw_req, "system_language_code", json_object_new_string(system_language_code)); + json_object_object_add(raw_req, "device_model", json_object_new_string(device_model)); + json_object_object_add(raw_req, "system_version", json_object_new_string(system_version)); + json_object_object_add(raw_req, "application_version", json_object_new_string(application_version)); + json_object_object_add(raw_req, "use_secret_chats", json_object_new_boolean(use_secret_chats)); + json_object_object_add(raw_req, "enable_storage_optimizer", json_object_new_boolean(enable_storage_optimizer)); + json_object_object_add(raw_req, "ignore_file_names", json_object_new_boolean(ignore_file_names)); + tg_send(false, tg_mkreq(TG_REQ_SET_PARAMS, "setTdlibParameters", "parameters", raw_req), true); + goto cleanup; +cleanup: + return r; +} + +static int handle_update_error(const json_object *json) +{ + int r = TG_IGNORE; + int code = -1; + const char *msg = NULL; + const char *extra = NULL; + json_object *obj = NULL; + if(!json_object_object_get_ex(json, "code", &obj)) + goto cleanup; + code = json_object_get_int(obj); + if(!json_object_object_get_ex(json, "message", &obj)) + goto cleanup; + msg = json_object_get_string(obj); + if(json_object_object_get_ex(json, "@extra", &obj)) + extra = json_object_get_string(obj); + tg_errno = code; + tg_reg1 = extra; + tg_reg2 = msg; + r = TG_ERROR; + goto cleanup; +cleanup: + return r; +} + +static int handle_update_authorization_state(const json_object *json) +{ + int r = 0; + json_object *obj = NULL; + if(!json_object_object_get_ex(json, "authorization_state", &obj)) + goto cleanup; + json_object *type_obj = NULL; + if(!json_object_object_get_ex(obj, "@type", &type_obj)) + goto cleanup; + const char *type = json_object_get_string(type_obj); +#ifdef TG_DEBUG + printf("AUTH: %s\n", type); +#endif + if(!strcmp(type, "authorizationStateClosed")) + r = TG_CLOSED; + else if(!strcmp(type, "authorizationStateClosing")) + r = TG_CLOSING; + else if(!strcmp(type, "authorizationStateLoggingOut")) + r = TG_LOGGING_OUT; + else if(!strcmp(type, "authorizationStateWaitTdlibParameters")) + r = TG_PARAMS; + else if(!strcmp(type, "authorizationStateWaitEncryptionKey")) + r = TG_LOGIN_ENC; + else if(!strcmp(type, "authorizationStateWaitPhoneNumber")) + r = TG_LOGIN_PHONE; + else if(!strcmp(type, "authorizationStateWaitCode")) + r = TG_LOGIN_CODE; + else if(!strcmp(type, "authorizationStateWaitPassword")) + r = TG_LOGIN_PW; + else if(!strcmp(type, "authorizationStateWaitRegistration")) + r = TG_LOGIN_REG; + else if(!strcmp(type, "authorizationStateWaitOtherDeviceConfirmation")) + { + r = TG_LOGIN_QR; + if(json_object_object_get_ex(obj, "link", &obj)) + tg_reg1 = json_object_get_string(obj); + } + else if(!strcmp(type, "authorizationStateReady")) + r = TG_LOGGED_IN; + else + r = TG_RUN; + goto cleanup; +cleanup: + return r; +} + +int tg_init() +{ + int r = 0; + td = td_json_client_create(); + if(td == NULL) + { + r = TG_SYSERR; + goto cleanup; + } + tg_send_raw(true, "{ \"@extra\": \"log\", \"@type\": \"setLogVerbosityLevel\", \"@args\": { \"new_verbosity_level\": 0 } }"); + goto cleanup; +cleanup: + return r; +} + +int tg_destroy() +{ +#ifdef TG_ENABLE_PTHREAD + pthread_mutex_lock(&mutex_close); +#endif + if(td == NULL) + { + return 0; + } + td_json_client_destroy(td); + td = NULL; +#ifdef TG_ENABLE_PTHREAD + pthread_mutex_unlock(&mutex_close); +#endif + return 0; +} + +int tg_loop() +{ + if(tg_update != NULL) + { + json_object_put(tg_update); + tg_update = NULL; + tg_update_type = NULL; + tg_errno = 0; + tg_reg1 = NULL; + tg_reg2 = NULL; + tg_reg3 = NULL; + tg_reg4 = NULL; + tg_reg5 = NULL; + tg_reg6 = NULL; + } + const char *event = td_json_client_receive(td, 3); + int r = TG_IGNORE; + if(event == NULL) goto cleanup; +#ifdef TG_DEBUG + printf("< %s\n", event); +#endif + tg_update = json_tokener_parse(event); + json_object *obj = NULL; + if(!json_object_object_get_ex(tg_update, "@type", &obj)) + goto cleanup; + tg_update_type = json_object_get_string(obj); + if(!strcmp(tg_update_type, "updateAuthorizationState")) + r = handle_update_authorization_state(tg_update); + else if(!strcmp(tg_update_type, "error")) + r = handle_update_error(tg_update); + else + r = TG_RUN; + goto cleanup; +cleanup: + return r; +} + +int tg_close() +{ +#ifdef TG_ENABLE_PTHREAD + pthread_mutex_lock(&mutex_close); +#endif + if(td == NULL) return 0; + /* Using json object seems to be leaking */ + tg_send_raw(false, "{ \"@type\": \"close\" }"); +#ifdef TG_ENABLE_PTHREAD + pthread_mutex_unlock(&mutex_close); +#endif + return 0; +} -- cgit v1.2.3