Vladimir Kartashev 373440 · @ColdDirol · 52bbc1b4 · 8295ca45 · fd4727a1 · 7226c080
--- a/lib.inc

+ 185

- 14

View file @ adcefd01
+++ b/lib.inc

+ 185

- 14

View file @ adcefd01
 
 
 ; Принимает код возврата и завершает текущий процесс
-exit: 
-    xor rax, rax,
-    ret 
+exit:
+    xor rdi, rdi  ; Очищаем rdi с использованием xor
+    mov rax, 60   ; Устанавливаем системный вызов для выхода
+    syscall       ; Вызываем системный вызов
+

 ; Принимает указатель на нуль-терминированную строку, возвращает её длину
 string_length:
    xor rax, rax
+.loop:
+    cmp byte [rdi + rax], 0
+    je .end
+    inc rax
+    jmp .loop
+.end:
    ret

+
+
 ; Принимает указатель на нуль-терминированную строку, выводит её в stdout
 print_string:
-    xor rax, rax
+    push rdi
+    call string_length
+    pop rsi
+    mov rdx, rax
+    mov rax, 1
+    mov rdi, 1
+    syscall
    ret

+
+
 ; Принимает код символа и выводит его в stdout
 print_char:
-    xor rax, rax
+    sub rsp, 16         ; Выделение места на стеке, выравненного по 16 байтам
+    mov [rsp], dil     ; Сохранение значения в стеке
+    mov rax, 1
+    mov rdi, 1
+    lea rsi, [rsp]     ; Использование lea для эффективного вычисления адреса
+    mov rdx, 1
+    syscall
+    add rsp, 16         ; Восстановление стека
    ret

+
+
 ; Переводит строку (выводит символ с кодом 0xA)
 print_newline:
-    xor rax, rax
+    mov rdi, 0xA
+    call print_char
    ret

+
+
 ; Выводит беззнаковое 8-байтовое число в десятичном формате 
 ; Совет: выделите место в стеке и храните там результаты деления
 ; Не забудьте перевести цифры в их ASCII коды.
 print_uint:
    xor rax, rax
+    xor rdx, rdx
+    xor rsi, rsi
+    push rbx
+    sub rsp, 16            ; Выделение места на стеке, выравненного по 16 байтам
+    mov rbx, rsp
+    mov rax, rdi
+    mov rsi, 10
+    mov [rsp], byte 0
+.loop:
+    div rsi
+    lea rdx, ['0' + rdx]
+    dec rsp
+    mov [rsp], dl
+    xor rdx, rdx
+    test rax, rax
+    jnz .loop
+    mov rdi, rsp
+    call print_string
+    mov rsp, rbx
+    add rsp, 16            ; Восстановление стека
+    pop rbx
    ret

-; Выводит знаковое 8-байтовое число в десятичном формате 
+
+
+; Выводит знаковое 8-байтовое число в десятичном формате
 print_int:
-    xor rax, rax
+    mov rax, rdi
+    test rax, rax
+    jns .print_uint
+    push rdi
+    mov rdi, '-'
+    call print_char
+    pop rdi
+    neg rdi
+.print_uint:
+    call print_uint
    ret

+
+
 ; Принимает два указателя на нуль-терминированные строки, возвращает 1 если они равны, 0 иначе
 string_equals:
    xor rax, rax
+.loop:
+    mov al, [rdi]
+    cmp al, [rsi]
+    jne .not_equal
+    test al, al
+    jz .done
+    inc rdi
+    inc rsi
+    jmp .loop
+.not_equal:
+    xor rax, rax
+    ret
+.done:
+    mov rax, 1
    ret

+
+
+
 ; Читает один символ из stdin и возвращает его. Возвращает 0 если достигнут конец потока
 read_char:
-    xor rax, rax
-    ret 
+    sub rsp, 16            ; Выделение места на стеке, выравненного по 16 байтам
+    xor rax, rax          ; Замена mov rax, 0 на xor rax, rax
+    mov rdi, 0
+    mov rsi, rsp
+    mov rdx, 1
+    syscall
+    test rax, rax         ; Замена cmp rax, 0 на test rax, rax
+    jle .end
+    movzx rax, byte [rsp]
+.end:
+    add rsp, 16            ; Восстановление стека
+    ret
+
+

 ; Принимает: адрес начала буфера, размер буфера
 ; Читает в буфер слово из stdin, пропуская пробельные символы в начале, .
 ; При успехе возвращает адрес буфера в rax, длину слова в rdx.
 ; При неудаче возвращает 0 в rax
 ; Эта функция должна дописывать к слову нуль-терминатор
-
 read_word:
+    push r12
+    push r13
+    push r14
+    mov r12, rdi
+    mov r13, rsi
+    xor r14, r14
+.loop:
+    call read_char
+    test rax, rax
+    jz .done
+    cmp r14, r13
+    jnl .error
+    cmp rax, 0x20
+    jz .skip
+    cmp rax, 0x9
+    jz .skip
+    cmp rax, 0xA
+    jz .skip
+    mov [r12 + r14], al
+    inc r14
+    jmp .loop
+.skip:
+    test r14, r14
+    jz .loop
+.done:
+    mov byte [r12 + r14], 0
+    mov rax, r12
+    mov rdx, r14
+    jmp .exit
+.error:
+    xor rax, rax
+.exit:
+    pop r14
+    pop r13
+    pop r12
    ret
 

+
 ; Принимает указатель на строку, пытается
 ; прочитать из её начала беззнаковое число.
 ; Возвращает в rax: число, rdx : его длину в символах
 ; rdx = 0 если число прочитать не удалось
 parse_uint:
    xor rax, rax
+    xor rdx, rdx
+.parse_loop:
+    movzx rcx, byte [rdi + rdx]
+    cmp rcx, '0'
+    jb .end
+    cmp rcx, '9'
+    ja .end
+    sub rcx, '0'
+    imul rax, rax, 10
+    add rax, rcx
+    inc rdx
+    jmp .parse_loop
+.end:
    ret



-
 ; Принимает указатель на строку, пытается
 ; прочитать из её начала знаковое число.
 ; Если есть знак, пробелы между ним и числом не разрешены.
 ; Возвращает в rax: число, rdx : его длину в символах (включая знак, если он был) 
 ; rdx = 0 если число прочитать не удалось
 parse_int:
-    xor rax, rax
-    ret 
+    xor rsi, rsi
+    cmp byte [rdi], '-'
+    je .negative
+    cmp byte [rdi], '+'
+    je .positive
+.unsigned:
+    call parse_uint
+    ret
+.negative:
+    inc rdi
+    call parse_uint
+    inc rdx
+    neg rax
+    ret
+.positive:
+    inc rdi
+    call parse_uint
+    inc rdx
+    ret
+
+

 ; Принимает указатель на строку, указатель на буфер и длину буфера
 ; Копирует строку в буфер
 ; Возвращает длину строки если она умещается в буфер, иначе 0
 string_copy:
    xor rax, rax
+.loop:
+    cmp rax, rdx
+    jz .return_zero
+    mov r9b, [rdi + rax]
+    mov [rsi + rax], r9b
+    inc rax
+    test r9b, r9b
+    jz .return
+    jmp .loop
+.return_zero:
+    xor rax, rax
+.return:
    ret