[NASM] Bot irc basique

[ark]

Okay, donc j'ai un peu avancer, on a le droit a un bot qui se co/deco instant! \o/ Magnifique. x) le seul probleme etant que j'ai hardcoded pas mal de trucs.

Voici le code pour l'instant, je compte virer mes trucs trop hardcoded et les remplacer par des trucs plus generiques. Et une fois que ca sera fait, je rajouterais la reponse au ping.

Code ASM :

BITS 64

    EXTERN  puts

    %DEFINE    BUFF_SIZE    1024
    %DEFINE INT_SIZE    4
    %DEFINE AF_INET        2
    %DEFINE    SOCK_STREAM    1

    %DEFINE SYS_SOCKET    41
    %DEFINE SYS_CONNECT    42
    %DEFINE SYS_CLOSE    3
    %DEFINE SYS_WRITE    1

    SECTION .rodata
            sock_error      db      "[-] Error while creating socket", 0
            sock_ok         db      "[+] The socket have been successfuly created", 0
            connect_error   db      "[-] Error: connexion to server failed", 0
            connect_ok      db      "[+] Connecting to irc.n-pn.fr", 0
            nick_cmd        db      "NICK asm64_bot", 0xd, 0xa, 0
            user_cmd        db      "USER BotDe7Lieu 0 0 :BotDe7Lieu", 0xd, 0xa, 0
            input_print     db      "\ninput: %s", 0
            channel         db      "JOIN #bot", 0xd, 0xa, 0
        server        db    "localhost", 0
    SECTION .bss
            buffer          resb    1024
        sock_fd        resb    4

    SECTION .text

        GLOBAL  main
main:
        push rbp
        mov rbp, rsp
;; make_socket
            mov     rdi, AF_INET
            mov     rsi, SOCK_STREAM
        mov    rdx, 0
            mov     rax, SYS_SOCKET
            syscall            ; Syscall socket()
            mov      [sock_fd], eax     ; Saving socket fd
            cmp     rax, 0
            jl      sock_fail
        mov     rdi, sock_ok
            call    puts
;; connect
        xor     rax, rax
        push       rax                ; padding
        push     DWORD 0xc589bf58     ; Hardcoded server's IP
        push     WORD 0x0b1a
        push     WORD AF_INET
        mov    rsi, rsp
        xor     rdi, rdi
        mov     edi, DWORD [sock_fd]
        mov    rdx, 0x10
        mov     rax, SYS_CONNECT
        syscall
        cmp    rax, 0
        jl    connect_fail
        mov    rdi, connect_ok
        call    puts

;; send data identification
        xor    rdi, rdi
        mov    edi, [sock_fd]
        mov    rsi, nick_cmd
        mov    rdx, 16
        mov     rax, 1
        syscall
        xor    rdi, rdi
        mov    edi, [sock_fd]
        mov    rsi, user_cmd
        mov    rdx, 33
        mov     rax, 1
        syscall

;; receive datas
        xor     rax, rax
        inc    rax            ; set rax to 1 for the cmp
        jmp     recv_data_loop1
recv_data_start1:
        xor     rdi, rdi
        mov    edi, [sock_fd]
        mov    rsi, buffer
        mov    rdx, 1024
        mov     rax, 0
        syscall                ; read what's on the socket 'till there's nothing left

        push     rax            ; save rax's value  which is read's return value
        mov    rdi, 1
        mov    rsi, buffer
        mov    rdx, rax
        mov     rax, 1
        syscall                ; print the buffer to stdout
        pop    rax

recv_data_loop1:
        cmp rax, 287            ; Hardcore cheat of the death <img src="https://dev.n-pn.fr/forum/images/smilies/confused.png" alt="Confused" title="Confused" class="smilie smilie_13" />hame:
        jne recv_data_start1

;; join chan
        xor    rdi, rdi
        mov    edi, [sock_fd]
        mov    rsi, channel
        mov    rdx, 11
        mov     rax, 1
        syscall

;; close connexion and exit
        call     close_socket
        mov     rax, 0
        leave
        ret

sock_fail:
        mov     rdi, sock_error
        call     puts
        mov     rax, 1
            leave
            ret

connect_fail:
        mov    rdi, connect_error
        call    puts
        call     close_socket
        mov     rax, 1
        leave
        ret

close_socket:                     ; This just closes the socket
        push     rbp
        mov    rbp, rsp
        xor    rdi, rdi
        mov     edi, DWORD [sock_fd]
        mov     rax, SYS_CLOSE
        syscall
        leave
        ret