nasm_tdjson_example

; TDLib Minimal Linux Client in x86-64 Assembly (NASM) ; ; nasm -f elf64 -o nasm_tdjson_example.o nasm_tdjson_example.asm ; gcc -o nasm_tdjson_example nasm_tdjson_example.o -ldl -no-pie ; ./nasm_tdjson_example

section .data ; Error messages lib_error db "Error: libtdjson.so not found", 10, 0 client_error db "Error: Failed to create client", 10, 0 input_error db "Error reading input", 10, 0 receive_error db "Error in receive function", 10, 0

; Function names
create_name db "td_json_client_create", 0
send_name db "td_json_client_send", 0
receive_name db "td_json_client_receive", 0
destroy_name db "td_json_client_destroy", 0
execute_name db "td_json_client_execute", 0

; Library name
lib_name db "libtdjson.so", 0

; Prompts
phone_prompt db "Enter phone number (international format): ", 0
code_prompt db "Enter authentication code: ", 0
password_prompt db "Enter password: ", 0
auth_success db "Authorization successful!", 10, 0
auth_wait db "Waiting for authorization...", 10, 0

; JSON messages
params_msg db '{"@type":"setTdlibParameters","database_directory":"td_data","use_message_database":true,"api_id":94575,"api_hash":"a3406de8d171bb422bb6ddf3bbd800e2","system_language_code":"en","device_model":"AsmClient","application_version":"1.0"}', 0
verbosity_msg db '{"@type":"setLogVerbosityLevel","new_verbosity_level":2}', 0
close_msg db '{"@type":"close"}', 0
newline db 10, 0

; Authorization states
auth_wait_phone db "authorizationStateWaitPhoneNumber", 0
auth_wait_code db "authorizationStateWaitCode", 0
auth_wait_password db "authorizationStateWaitPassword", 0
auth_ready db "authorizationStateReady", 0

; Format strings
fmt_phone db '{"@type":"setAuthenticationPhoneNumber","phone_number":"%s"}', 0
fmt_code db '{"@type":"checkAuthenticationCode","code":"%s"}', 0
fmt_password db '{"@type":"checkAuthenticationPassword","password":"%s"}', 0

section .bss ; Handle pointers lib_handle resq 1 client_handle resq 1

; Function pointers
create_func resq 1
send_func resq 1
receive_func resq 1
destroy_func resq 1
execute_func resq 1

; Buffers
input_buffer resb 256
phone_buffer resb 32
code_buffer resb 16
password_buffer resb 64
json_buffer resb 1024
auth_state resq 1

section .text global main extern dlopen, dlsym, printf, sleep, exit, fgets, stdin, strstr, sprintf, strlen, putchar, usleep, fflush

main: push rbp mov rbp, rsp sub rsp, 32

; Load TDLib library
mov rdi, lib_name
mov rsi, 1
call dlopen
test rax, rax
jz .dl_error
mov [lib_handle], rax

; Get function addresses
mov rdi, [lib_handle]
mov rsi, create_name
call dlsym
mov [create_func], rax

mov rdi, [lib_handle]
mov rsi, send_name
call dlsym
mov [send_func], rax

mov rdi, [lib_handle]
mov rsi, receive_name
call dlsym
mov [receive_func], rax

mov rdi, [lib_handle]
mov rsi, destroy_name
call dlsym
mov [destroy_func], rax

mov rdi, [lib_handle]
mov rsi, execute_name
call dlsym
mov [execute_func], rax

; Create TDLib client
call [create_func]
test rax, rax
jz .client_error
mov [client_handle], rax

; Set log verbosity
mov rdi, [client_handle]
mov rsi, verbosity_msg
call [execute_func]

; Set TDLib parameters
mov rdi, [client_handle]
mov rsi, params_msg
call [send_func]

; Authorization loop
xor r12, r12
mov qword [auth_state], 0

mov edi, 50000
call usleep

.auth_loop: mov rdi, [client_handle] xor rsi, rsi call [receive_func] test rax, rax jz .no_message

; Process message
mov rdi, rax
call process_message
cmp r12, 1
je .auth_complete

.no_message: ; Check if we need to request input cmp qword [auth_state], 1 ; wait_phone je .request_phone cmp qword [auth_state], 2 ; wait_code je .request_code cmp qword [auth_state], 3 ; wait_password je .request_password

mov edi, 10000
call usleep
jmp .auth_loop

.request_phone: mov rdi, phone_prompt call printf xor rdi, rdi call fflush

; Read phone number
mov rdi, phone_buffer
mov rsi, 32
mov rdx, [stdin]
call fgets
test rax, rax
jz .input_error

; Remove newline
mov rdi, phone_buffer
call remove_newline

; Create JSON for phone number
mov rdi, json_buffer
mov rsi, fmt_phone
mov rdx, phone_buffer
call sprintf

; Send to TDLib
mov rdi, [client_handle]
mov rsi, json_buffer
call [send_func]

mov qword [auth_state], 0

mov edi, 10000
call usleep
jmp .auth_loop

.request_code: mov rdi, code_prompt call printf xor rdi, rdi call fflush

; Read code
mov rdi, code_buffer
mov rsi, 16
mov rdx, [stdin]
call fgets
test rax, rax
jz .input_error

; Remove newline
mov rdi, code_buffer
call remove_newline

; Create JSON for code
mov rdi, json_buffer
mov rsi, fmt_code
mov rdx, code_buffer
call sprintf

; Send to TDLib
mov rdi, [client_handle]
mov rsi, json_buffer
call [send_func]

mov qword [auth_state], 0

mov edi, 10000
call usleep
jmp .auth_loop

.request_password: mov rdi, password_prompt call printf xor rdi, rdi call fflush

; Read password
mov rdi, password_buffer
mov rsi, 64
mov rdx, [stdin]
call fgets
test rax, rax
jz .input_error

; Remove newline
mov rdi, password_buffer
call remove_newline

; Create JSON for password
mov rdi, json_buffer
mov rsi, fmt_password
mov rdx, password_buffer
call sprintf

; Send to TDLib
mov rdi, [client_handle]
mov rsi, json_buffer
call [send_func]

mov qword [auth_state], 0

mov edi, 10000
call usleep
jmp .auth_loop

.auth_complete: mov rdi, auth_success call printf

; Main loop
mov r13, 100

.main_loop: ; Receive messages mov rdi, [client_handle] xor rsi, rsi call [receive_func] test rax, rax jz .no_msg_main

; Print message
mov rdi, rax
call printf
mov rdi, newline
call printf

.no_msg_main: mov edi, 50000 call usleep dec r13 jnz .main_loop

.cleanup: ; Send close command mov rdi, [client_handle] mov rsi, close_msg call [send_func]

mov edi, 50000
call usleep

; Destroy client
mov rdi, [client_handle]
call [destroy_func]

; Exit
xor rdi, rdi
call exit

.dl_error: mov rdi, lib_error call printf mov rdi, 1 call exit

.client_error: mov rdi, client_error call printf jmp .cleanup

.input_error: mov rdi, input_error call printf jmp .cleanup

process_message: push rbp mov rbp, rsp push r12 push r13 mov r12, rdi

mov rdi, r12
mov rsi, auth_wait_phone
call strstr
test rax, rax
jnz .wait_phone

mov rdi, r12
mov rsi, auth_wait_code
call strstr
test rax, rax
jnz .wait_code

mov rdi, r12
mov rsi, auth_wait_password
call strstr
test rax, rax
jnz .wait_password

mov rdi, r12
mov rsi, auth_ready
call strstr
test rax, rax
jnz .auth_ready

jmp .done

.wait_phone: mov qword [auth_state], 1 jmp .done

.wait_code: mov qword [auth_state], 2 jmp .done

.wait_password: mov qword [auth_state], 3 jmp .done

.auth_ready: mov r12, 1 mov qword [auth_state], 0

.done: pop r13 pop r12 pop rbp ret

remove_newline: push rbp mov rbp, rsp

mov rcx, rdi

.find_end: cmp byte [rcx], 0 je .check_newline inc rcx jmp .find_end

.check_newline: dec rcx cmp byte [rcx], 10 jne .end mov byte [rcx], 0

.end: pop rbp ret