112 lines
3.6 KiB
NASM
112 lines
3.6 KiB
NASM
global _start
|
||
|
||
%define STDIN 0
|
||
%define STDOUT 1
|
||
%define STDERR 2
|
||
|
||
section .data
|
||
src db 1, 2, 3, 4, 5, 6, 7, 8, 9, 0
|
||
src_size equ $-src
|
||
|
||
; резервируем 1 килобайт для буффера ввода и вывода
|
||
; также в отдельной переменной сохраняем размер этого буфера
|
||
print_buf: times 1024 db 0
|
||
buf_size equ $-print_buf
|
||
input_buf: times 1024 db 0 ; буфер, в который будут читаться символы со стандартного ввода
|
||
input_size equ $-src
|
||
|
||
; Для poll
|
||
%define POLLIN 0x001 ; Есть ли что почитать с буфера ввода. Понадобится для продолжения ввода
|
||
input_pollfd: dd STDIN
|
||
dw POLLIN
|
||
revents: dw 0 ; возвращаемые события
|
||
|
||
section .text
|
||
|
||
%macro DIGIT_TO_ASCII 1 ; макрос, принимающий один аргумент (регистр или память)
|
||
add %1, '0'
|
||
%endmacro
|
||
|
||
%macro ASCII_TO_DIGIT 1 ; макрос, принимающий один аргумент (регистр или память)
|
||
sub %1, '0'
|
||
%endmacro
|
||
|
||
%macro PUSH_M 1-* ; push many; пушит в порядке следования
|
||
%rep %0
|
||
push %1
|
||
%rotate 1
|
||
%endrep
|
||
%endmacro
|
||
|
||
%macro POP_M 1-* ; pop many. читает в порядке следования
|
||
%rep %0
|
||
pop %1
|
||
%rotate 1
|
||
%endrep
|
||
%endmacro
|
||
|
||
%macro RPOP_M 1-* ; pop many. читает в обратном порядке
|
||
%rotate -1
|
||
%rep %0
|
||
pop %1
|
||
%rotate -1
|
||
%endrep
|
||
%endmacro
|
||
|
||
; Передачу аргументов будем делать при помощи ABI - стандартная практика для linux
|
||
; Аргументы передаются в следующем порядке: rdi, rsi, rdx, rcx, r8, r9. Все, что не влезло, пушится в стек
|
||
; У передачи через стек тоже есть особенности, но их мы пока касаться не будем
|
||
|
||
clean_print_buf: ; none -> void
|
||
PUSH_M rax, rcx, rdi
|
||
mov rcx, buf_size
|
||
mov rdi, print_buf
|
||
xor rax, rax ; будем заносить нули во всю память
|
||
rep stosb
|
||
RPOP_M rax, rcx, rdi
|
||
ret
|
||
|
||
print_from_buf: ; word -> void
|
||
PUSH_M rax, rsi, rdx, rdi ; сохраним регистры, которые точно попортим
|
||
mov rdx, rdi ; сколько выводить, в rdi содержится единственный аргумент
|
||
mov rsi, print_buf ; откуда выводить. Адрес буфера
|
||
mov rdi, STDOUT; куда выводить. Дескриптор файла. В нашем случае стандартного вывода
|
||
mov rax, 1
|
||
syscall
|
||
RPOP_M rax, rsi, rdx, rdi ; вернем значения регистров
|
||
ret
|
||
|
||
read_to_buf: ; none -> void. Пытается заполнить буфер из стандартного ввода
|
||
PUSH_M rax, rdi, rsi
|
||
mov rdi, STDIN ; откуда читать (дескриптор файла)
|
||
mov rsi, input_buf ; куда читать
|
||
mov rdx, input_size ; Сколько пытаемся читать
|
||
mov rax, 0 ; системный вызов чтения
|
||
RPOP_M rax, rdi, rsi
|
||
ret
|
||
|
||
poll_stdin:
|
||
PUSH_M rdi, rsi, rdx
|
||
mov rsi, 1 ; следим только за одним потоком
|
||
mov rax, 7 ; poll syscall
|
||
mov rdi, input_pollfd
|
||
mov rsi, 1 ; одна структура данных
|
||
mov rdx, 0 ; не ждать
|
||
RPOP_M rdi, rsi, rdx
|
||
ret
|
||
|
||
_start:
|
||
.read_loop:
|
||
call read_to_buf
|
||
; process
|
||
|
||
call poll_stdin
|
||
test BYTE [revents], POLLIN
|
||
jnz .read_loop
|
||
|
||
exit:
|
||
mov rax, 60
|
||
mov rdi, 0
|
||
syscall
|
||
|