Draft totally work in progress 0.6 ctrl msg receiving

Author: ChillerDragon

Diff for: src/data/teeworlds.asm

@@ -8,6 +8,12 @@ MSG_CTRL_ACCEPT equ 2
 MSG_CTRL_CLOSE equ 4
 MSG_CTRL_TOKEN equ 5
 
+MSG6_CTRL_KEEPALIVE equ 0
+MSG6_CTRL_CONNECT equ 1
+MSG6_CTRL_CONNECTACCEPT equ 2
+; MSG6_CTRL_ACCEPT equ 3 ; UNUSED!
+MSG6_CTRL_CLOSE equ 4
+
 MSG_SYSTEM_NULL equ 0
 MSG_SYSTEM_INFO equ 1
 MSG_SYSTEM_MAP_CHANGE equ 2

Diff for: src/on_packet.asm

@@ -154,7 +154,15 @@ on_packet_end:
     ret
 
 on_packet6:
-    puts "0.6 not implemented yet"
+
+    push rax
+    mov rax, udp_recv_buf
+    call unpack_packet_header6
+    pop rax
+
+    is_packet_in_flag PACKETFLAG6_CONTROL
+    je on_ctrl6_message
+
     jmp on_packet_end
 
 on_packet7:

Diff for: src/packet_header.asm

@@ -35,3 +35,31 @@ unpack_packet_header:
     pop_registers
     ret
 
+unpack_packet_header6:
+    ; unpack_packet_header6 [rax]
+    ;  rax = buffer to unpack
+    ; warning this is not getting the ddnet security token!
+    ; because it is at the end of the payload
+    ; so it is parsed at a later point in time
+    push_registers
+
+    ; dereference pointer
+    mov rcx, [rax]
+    ; move 1 byte at offset 0
+    mov byte [in_packet_header_flags], cl
+
+    ; dereference pointer
+    mov rcx, [rax + 2]
+    ; move 1 byte at offset 2
+    mov byte [in_packet_header_num_chunks], cl
+
+    ; explicitly empty the token field to avoid confusion
+    ; we do not know the token yet!
+    mov byte [in_packet_header_token+0], 0xff
+    mov byte [in_packet_header_token+1], 0xff
+    mov byte [in_packet_header_token+2], 0xff
+    mov byte [in_packet_header_token+3], 0xff
+
+    pop_registers
+    ret
+

Diff for: src/receive_control6.asm

@@ -0,0 +1,54 @@
+on_ctrl6_msg_connect_accept:
+    ; ignore cursed token only ddnet servers send
+    ; https://github.com/ddnet/ddnet/issues/8805
+
+    ; mov rax, [udp_recv_buf + 8]
+    ; mov [peer_token], rax
+
+    print_label s_got_accept
+    ; call send_msg_info
+
+    jmp on_ctrl6_message_end
+
+on_ctrl6_msg_close:
+    push rax
+
+    mov rax, [packet_payload + 1]
+    cmp rax, 0
+    je .on_ctrl6_msg_close_no_reason
+
+.on_ctrl6_msg_close_reason:
+    print_label s_got_disconnect_with_reason
+    lea rax, [packet_payload + 1]
+    print_c_str rax
+    call print_newline
+    jmp .on_ctrl6_msg_close_end
+
+.on_ctrl6_msg_close_no_reason:
+    print_label s_got_disconnect
+
+.on_ctrl6_msg_close_end:
+    exit 0
+    pop rax
+    jmp on_ctrl6_message_end
+
+on_ctrl6_message:
+    push_registers ; popped in on_ctrl6_message_end
+
+    print_label s_got_ctrl_msg
+
+    mov al, [udp_recv_buf + PACKET_HEADER_LEN]
+    call println_uint32
+
+    cmp al, MSG6_CTRL_CONNECTACCEPT
+    je on_ctrl6_msg_connect_accept
+    cmp al, MSG6_CTRL_CLOSE
+    je on_ctrl6_msg_close
+
+    print_label s_unknown_ctrl_msg
+    call println_uint32
+
+on_ctrl6_message_end:
+    pop_registers ; pushed in on_ctrl6_message
+    jmp on_packet_end
+

Diff for: src/teeworlds_asmr.asm

@@ -140,6 +140,7 @@ section .text
 %include "src/send_system.asm"
 %include "src/send_game.asm"
 %include "src/receive_control.asm"
+%include "src/receive_control6.asm"
 %include "src/system.asm"
 %include "src/string.asm"
 %include "src/packet_header.asm"

Diff for: tests/assert.asm

@@ -73,6 +73,7 @@ section .text
 %include "src/send_system.asm"
 %include "src/send_game.asm"
 %include "src/receive_control.asm"
+%include "src/receive_control6.asm"
 %include "src/system.asm"
 %include "src/string.asm"
 %include "src/packet_header.asm"