Multi-thread_Asm_Study/threads.asm at main · miojo-dev/Multi-thread_Asm_Study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
section .data
    msg_thread1 db 'Hello from thread 1 to ', 0
    msg_thread2 db 'Hello from thread 2 to ', 0
    msg db 'Type your name: ', 0
    len equ $ - msg

section .bss
    stack2 resb 4096 ; 4KB para stack da thread 2
    name resb 100 ; name buffer
    temp_msg1 resb 100  ; temporary buffer for thread 1
    temp_msg2 resb 100  ; temporary buffer for thread 2

section .text
    global _start

_start:
    mov eax, 4 ; Write method(sys_write)
    mov ebx, 1 ; Output method(std_out)
    mov ecx, msg ; Message
    mov edx, len ; Message size
    int 0x80 ; Kernel call

    mov eax, 3 ; Read method(sys_read)
    mov ebx, 0 ; Input method(std_in)
    mov ecx, name ; Var who stores the input
    mov edx, 99 ; Input max size
    int 0x80 ; Kernel call

    ; Create thread 1
    mov eax, 56 ; syscall clone
    mov edi, 0x00000100 ; CLONE_VM - share memory
    mov esi, 0 ; stack (NULL = use current stack)
    mov edx, thread1_func ; function of the thread
    mov ecx, 0 ; arguments
    int 0x80

    ; Create thread 2
    mov eax, 56 ; syscall clone
    mov edi, 0x00000100 ; CLONE_VM
    mov esi, stack2 + 4096 ; top of the stack (grows down)
    mov edx, thread2_func ; function of the thread
    mov ecx, 0 ; arguments
    int 0x80

    ; wait the threads to finish
    mov eax, 114     ; syscall wait4
    mov ebx, -1      ; wait any child
    mov ecx, 0       ; status
    mov edx, 0       ; options
    int 0x80

    ; exit
    mov eax, 60
    xor edi, edi
    int 0x80

thread1_func:
    ; thead 1 code
    mov esi, msg_thread1 ; 1st string
    mov edi, name ; 2nd string
    mov eax, temp_msg1 ; buffer destiny
    call concat_strings ; call func to concat strings

    ; write concatenated msg
    mov ebx, 1 ; stdout
    mov ecx, temp_msg1 ; concatenated msg
    mov edx, eax ; returned size
    mov eax, 4 ; write
    int 0x80

    ; exit thread
    mov eax, 60
    xor edi, edi
    int 0x80

thread2_func:
    ; thread 2 code
    mov esi, msg_thread2 ; 1st string
    mov edi, name ; 2nd string
    mov eax, temp_msg2 ; buffer destiny
    call concat_strings ; same func

    ; write concatenated msg
    mov ebx, 1 ; stdout
    mov ecx, temp_msg2 ; concatenated msg
    mov edx, eax ; returned size
    mov eax, 4 ; write
    int 0x80

    ; exit thread
    mov eax, 60
    xor edi, edi
    int 0x80

concat_strings:
    push ebx
    push ecx
    push edx

    ; copy 1st string
    mov ebx, esi ; ebx = src1
    mov ecx, eax ; ecx = dest

.copy_loop1:
    mov dl, [ebx] ; read character
    mov [ecx], dl ; write on destiny
    inc ebx
    inc ecx
    cmp dl, 0 ; end of the string?
    jne .copy_loop1

    ; copy 2nd string
    mov ebx, edi ; ebx = src2

.copy_loop2:
    mov dl, [ebx] ; read character
    mov [ecx], dl ; write on destiny
    inc ebx
    inc ecx
    cmp dl, 0 ; end of the string?
    jne .copy_loop2

    ; calculate total size
    sub ecx, eax ; ecx - eax = size
    mov eax, ecx ; return size

    pop edx
    pop ecx
    pop ebx
    ret