;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; ;; ;; Copyright (C) KolibriOS team 2004-2013. All rights reserved. ;; ;; Distributed under terms of the GNU General Public License ;; ;; ;; ;; Part of the TCP/IP network stack for KolibriOS ;; ;; ;; ;; Written by hidnplayr@kolibrios.org, ;; ;; and Clevermouse. ;; ;; ;; ;; Based on code by mike.dld ;; ;; ;; ;; GNU GENERAL PUBLIC LICENSE ;; ;; Version 2, June 1991 ;; ;; ;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; $Revision: 3514 $ struct SOCKET NextPtr dd ? ; pointer to next socket in list PrevPtr dd ? ; pointer to previous socket in list Number dd ? ; socket number mutex MUTEX PID dd ? ; process ID TID dd ? ; thread ID Domain dd ? ; INET/LOCAL/.. Type dd ? ; RAW/STREAM/DGRAP Protocol dd ? ; ICMP/IPv4/ARP/TCP/UDP errorcode dd ? device dd ? ; driver pointer, socket pointer if it's an LOCAL socket options dd ? state dd ? backlog dw ? ; how many incoming connections that can be queued snd_proc dd ? rcv_proc dd ? connect_proc dd ? ends struct IP_SOCKET SOCKET LocalIP rd 4 ; network byte order RemoteIP rd 4 ; network byte order ends struct TCP_SOCKET IP_SOCKET LocalPort dw ? ; network byte order RemotePort dw ? ; network byte order t_state dd ? ; TCB state t_rxtshift db ? rb 3 ; align t_rxtcur dd ? t_dupacks dd ? t_maxseg dd ? t_force dd ? t_flags dd ? ;--------------- ; RFC783 page 21 ; send sequence SND_UNA dd ? ; sequence number of unack'ed sent Packets SND_NXT dd ? ; next send sequence number to use SND_UP dd ? ; urgent pointer SND_WL1 dd ? ; window minus one SND_WL2 dd ? ; ISS dd ? ; initial send sequence number SND_WND dd ? ; send window ; receive sequence RCV_WND dd ? ; receive window RCV_NXT dd ? ; next receive sequence number to use RCV_UP dd ? ; urgent pointer IRS dd ? ; initial receive sequence number ;--------------------- ; Additional variables ; receive variables RCV_ADV dd ? ; retransmit variables SND_MAX dd ? ; congestion control SND_CWND dd ? SND_SSTHRESH dd ? ;---------------------- ; Transmit timing stuff t_idle dd ? t_rtt dd ? t_rtseq dd ? t_srtt dd ? t_rttvar dd ? t_rttmin dd ? max_sndwnd dd ? ;----------------- ; Out-of-band data t_oobflags dd ? t_iobc dd ? t_softerror dd ? ;--------- ; RFC 1323 ; the order of next 4 elements may not change SND_SCALE db ? RCV_SCALE db ? requested_s_scale db ? request_r_scale db ? ts_recent dd ? ; a copy of the most-recent valid timestamp from the other end ts_recent_age dd ? last_ack_sent dd ? ;------- ; Timers timer_flags dd ? timer_retransmission dd ? ; rexmt timer_persist dd ? timer_keepalive dd ? ; keepalive/syn timeout timer_timed_wait dd ? ; also used as 2msl timer timer_connect dd ? ; extra ts_ecr dd ? ; timestamp echo reply ts_val dd ? seg_next dd ? ; re-assembly queue temp_bits db ? rb 3 ; align ends struct UDP_SOCKET IP_SOCKET LocalPort dw ? ; network byte order RemotePort dw ? ; network byte order ends struct ICMP_SOCKET IP_SOCKET Identifier dw ? ends struct RING_BUFFER mutex MUTEX start_ptr dd ? ; Pointer to start of buffer end_ptr dd ? ; pointer to end of buffer read_ptr dd ? ; Read pointer write_ptr dd ? ; Write pointer size dd ? ; Number of bytes buffered ends struct STREAM_SOCKET TCP_SOCKET rcv RING_BUFFER snd RING_BUFFER ends struct socket_queue_entry data_ptr dd ? buf_ptr dd ? data_size dd ? ends SOCKETBUFFSIZE = 4096 ; in bytes SOCKET_QUEUE_SIZE = 10 ; maximum number of incoming packets queued for 1 socket ; the incoming packet queue for sockets is placed in the socket struct itself, at this location from start SOCKET_QUEUE_LOCATION = (SOCKETBUFFSIZE - SOCKET_QUEUE_SIZE*sizeof.socket_queue_entry - sizeof.queue) uglobal align 4 net_sockets rd 4 last_socket_num dd ? last_UDP_port dw ? ; These values give the number of the last used ephemeral port last_TCP_port dw ? ; socket_mutex MUTEX endg ;----------------------------------------------------------------- ; ; SOCKET_init ; ;----------------------------------------------------------------- macro SOCKET_init { xor eax, eax mov edi, net_sockets mov ecx, 5 rep stosd @@: pseudo_random eax cmp ax, EPHEMERAL_PORT_MIN jb @r cmp ax, EPHEMERAL_PORT_MAX ja @r xchg al, ah mov [last_UDP_port], ax @@: pseudo_random eax cmp ax, EPHEMERAL_PORT_MIN jb @r cmp ax, EPHEMERAL_PORT_MAX ja @r xchg al, ah mov [last_TCP_port], ax mov ecx, socket_mutex call mutex_init } ;----------------------------------------------------------------- ; ; Socket API (function 74) ; ;----------------------------------------------------------------- align 4 sys_socket: mov dword[esp+20], 0 ; Set error code to 0 cmp ebx, 255 jz SOCKET_debug cmp ebx, .number ja .error jmp dword [.table + 4*ebx] .table: dd SOCKET_open ; 0 dd SOCKET_close ; 1 dd SOCKET_bind ; 2 dd SOCKET_listen ; 3 dd SOCKET_connect ; 4 dd SOCKET_accept ; 5 dd SOCKET_send ; 6 dd SOCKET_receive ; 7 dd SOCKET_set_opt ; 8 dd SOCKET_get_opt ; 9 dd SOCKET_pair ; 10 .number = ($ - .table) / 4 - 1 .error: mov dword[esp+32], -1 mov dword[esp+20], EINVAL ret ;----------------------------------------------------------------- ; ; SOCKET_open ; ; IN: domain in ecx ; type in edx ; protocol in esi ; OUT: eax is socket num, -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_open: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_open: domain=%u type=%u protocol=%x ", ecx, edx, esi push ecx edx esi call SOCKET_alloc pop esi edx ecx jz .nobuffs mov [esp+32], edi ; return socketnumber DEBUGF DEBUG_NETWORK_VERBOSE, "socknum=%u\n", edi test edx, SO_NONBLOCK jz @f or [eax + SOCKET.options], SO_NONBLOCK and edx, not SO_NONBLOCK @@: mov [eax + SOCKET.Domain], ecx mov [eax + SOCKET.Type], edx mov [eax + SOCKET.Protocol], esi mov [eax + SOCKET.connect_proc], connect_notsupp cmp ecx, AF_INET4 jne .no_inet4 cmp edx, SOCK_DGRAM je .udp cmp edx, SOCK_STREAM je .tcp cmp edx, SOCK_RAW je .raw .no_inet4: cmp ecx, AF_PPP jne .no_ppp cmp esi, PPP_PROTO_ETHERNET je .pppoe .no_ppp: .unsupported: push eax call SOCKET_free pop eax mov dword[esp+20], EOPNOTSUPP mov dword[esp+32], -1 ret .nobuffs: mov dword[esp+20], ENOBUFS mov dword[esp+32], -1 ret .raw: test esi, esi ; IP_PROTO_IP jz .raw_ip cmp esi, IP_PROTO_ICMP je .raw_icmp jmp .unsupported align 4 .udp: mov [eax + SOCKET.Protocol], IP_PROTO_UDP mov [eax + SOCKET.snd_proc], SOCKET_send_udp mov [eax + SOCKET.rcv_proc], SOCKET_receive_dgram mov [eax + SOCKET.connect_proc], UDP_connect ret align 4 .tcp: mov [eax + SOCKET.Protocol], IP_PROTO_TCP mov [eax + SOCKET.snd_proc], SOCKET_send_tcp mov [eax + SOCKET.rcv_proc], SOCKET_receive_stream mov [eax + SOCKET.connect_proc], TCP_connect TCP_init_socket eax ret align 4 .raw_ip: mov [eax + SOCKET.snd_proc], SOCKET_send_ip mov [eax + SOCKET.rcv_proc], SOCKET_receive_dgram mov [eax + SOCKET.connect_proc], IPv4_connect ret align 4 .raw_icmp: mov [eax + SOCKET.snd_proc], SOCKET_send_icmp mov [eax + SOCKET.rcv_proc], SOCKET_receive_dgram mov [eax + SOCKET.connect_proc], IPv4_connect ret align 4 .pppoe: push eax init_queue (eax + SOCKET_QUEUE_LOCATION) ; Set up data receiving queue pop eax mov [eax + SOCKET.snd_proc], SOCKET_send_pppoe mov [eax + SOCKET.rcv_proc], SOCKET_receive_dgram ret ;----------------------------------------------------------------- ; ; SOCKET_bind ; ; IN: socket number in ecx ; pointer to sockaddr struct in edx ; length of that struct in esi ; OUT: 0 on success ; ;----------------------------------------------------------------- align 4 SOCKET_bind: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_bind: socknum=%u sockaddr=%x length=%u\n", ecx, edx, esi call SOCKET_num_to_ptr jz .invalid cmp esi, 2 jb .invalid cmp [eax + UDP_SOCKET.LocalPort], 0 ; Socket can only be bound once jnz .invalid cmp word [edx], AF_INET4 je .af_inet4 cmp word [edx], AF_LOCAL je .af_local .notsupp: mov dword[esp+20], EOPNOTSUPP mov dword[esp+32], -1 ret .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret .af_local: ; TODO: write code here mov dword[esp+32], 0 ret .af_inet4: cmp esi, 6 jb .invalid cmp [eax + SOCKET.Protocol], IP_PROTO_UDP je .udp cmp [eax + SOCKET.Protocol], IP_PROTO_TCP je .tcp jmp .notsupp .tcp: .udp: pushd [edx + 4] ; First, fill in the IP popd [eax + IP_SOCKET.LocalIP] mov bx, [edx + 2] ; Did caller specify a local port? test bx, bx jnz .just_check call SOCKET_find_port ; Nope, find an ephemeral one jmp .done .just_check: call SOCKET_check_port ; Yes, check if it's still available jz .addrinuse ; ZF is set by socket_check_port on error .done: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_bind: local ip=%u.%u.%u.%u\n",\ [eax + IP_SOCKET.LocalIP + 0]:1,[eax + IP_SOCKET.LocalIP + 1]:1,\ [eax + IP_SOCKET.LocalIP + 2]:1,[eax + IP_SOCKET.LocalIP + 3]:1 mov dword[esp+32], 0 ret .addrinuse: mov dword[esp+32], -1 mov dword[esp+20], EADDRINUSE ret ;----------------------------------------------------------------- ; ; SOCKET_connect ; ; IN: socket number in ecx ; pointer to sockaddr struct in edx ; length of that struct in esi ; OUT: 0 on success ; ;----------------------------------------------------------------- align 4 SOCKET_connect: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_connect: socknum=%u sockaddr=%x length=%u\n", ecx, edx, esi call SOCKET_num_to_ptr jz .invalid cmp esi, 8 jb .invalid cmp [eax + SOCKET.state], SS_ISCONNECTING je .already test [eax + SOCKET.options], SO_ACCEPTCON jnz .notsupp call [eax + SOCKET.connect_proc] mov dword[esp+20], ebx mov dword[esp+32], eax ret .notsupp: mov dword[esp+20], EOPNOTSUPP mov dword[esp+32], -1 ret .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret .already: mov dword[esp+20], EALREADY mov dword[esp+32], -1 ret connect_notsupp: xor eax, eax dec eax mov ebx, EOPNOTSUPP ret ;----------------------------------------------------------------- ; ; SOCKET_listen ; ; IN: socket number in ecx ; backlog in edx ; OUT: eax is socket num, -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_listen: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_listen: socknum=%u backlog=%u\n", ecx, edx call SOCKET_num_to_ptr jz .invalid cmp [eax + SOCKET.Domain], AF_INET4 jne .notsupp cmp [eax + SOCKET.Protocol], IP_PROTO_TCP jne .invalid cmp [eax + TCP_SOCKET.LocalPort], 0 je .already cmp [eax + IP_SOCKET.LocalIP], 0 jne @f push [IP_LIST + 4] ;;; fixme!!!! pop [eax + IP_SOCKET.LocalIP] @@: cmp edx, MAX_backlog jbe @f mov edx, MAX_backlog @@: mov [eax + SOCKET.backlog], dx or [eax + SOCKET.options], SO_ACCEPTCON mov [eax + TCP_SOCKET.t_state], TCPS_LISTEN mov [eax + TCP_SOCKET.timer_keepalive], 0 ; disable keepalive timer push eax init_queue (eax + SOCKET_QUEUE_LOCATION) ; Set up sockets queue pop eax mov dword[esp+32], 0 ret .notsupp: mov dword[esp+20], EOPNOTSUPP mov dword[esp+32], -1 ret .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret .already: mov dword[esp+20], EALREADY mov dword[esp+32], -1 ret ;----------------------------------------------------------------- ; ; SOCKET_accept ; ; IN: socket number in ecx ; addr in edx ; addrlen in esi ; OUT: eax is socket num, -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_accept: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_accept: socknum=%u sockaddr=%x length=%u\n", ecx, edx, esi call SOCKET_num_to_ptr jz .invalid test [eax + SOCKET.options], SO_ACCEPTCON jz .invalid cmp [eax + SOCKET.Domain], AF_INET4 jne .notsupp cmp [eax + SOCKET.Protocol], IP_PROTO_TCP jne .invalid .loop: get_from_queue (eax + SOCKET_QUEUE_LOCATION), MAX_backlog, 4, .block ; Ok, we got a socket ptr mov eax, [esi] ; Change thread ID to that of the current thread mov ebx, [TASK_BASE] mov ebx, [ebx + TASKDATA.pid] mov [eax + SOCKET.TID], ebx ; Convert it to a socket number call SOCKET_ptr_to_num jz .invalid ; FIXME ? ; and return it to caller mov [esp+32], eax ret .block: test [eax + SOCKET.options], SO_NONBLOCK jnz .wouldblock call SOCKET_block jmp .loop .wouldblock: mov dword[esp+20], EWOULDBLOCK mov dword[esp+32], -1 ret .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret .notsupp: mov dword[esp+20], EOPNOTSUPP mov dword[esp+32], -1 ret ;----------------------------------------------------------------- ; ; SOCKET_close ; ; IN: socket number in ecx ; OUT: eax is socket num, -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_close: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_close: socknum=%u\n", ecx call SOCKET_num_to_ptr jz .invalid mov dword[esp+32], 0 ; The socket exists, so we will succeed in closing it. or [eax + SOCKET.options], SO_NONBLOCK ; Mark the socket as non blocking, we dont want it to block any longer! test [eax + SOCKET.state], SS_BLOCKED ; Is the socket still in blocked state? jz @f call SOCKET_notify.unblock ; Unblock it. @@: cmp [eax + SOCKET.Domain], AF_INET4 jne .free cmp [eax + SOCKET.Protocol], IP_PROTO_TCP je .tcp .free: call SOCKET_free ret .tcp: cmp [eax + TCP_SOCKET.t_state], TCPS_SYN_RECEIVED ; state must be LISTEN, SYN_SENT or CLOSED jb .free call TCP_usrclosed call TCP_output ;;;; Fixme: is this nescessary?? call SOCKET_free ret .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret ;----------------------------------------------------------------- ; ; SOCKET_receive ; ; IN: socket number in ecx ; addr to buffer in edx ; length of buffer in esi ; flags in edi ; OUT: eax is number of bytes copied, -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_receive: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_receive: socknum=%u bufaddr=%x buflength=%u flags=%x\n", ecx, edx, esi, edi call SOCKET_num_to_ptr jz .invalid .loop: push edi call [eax + SOCKET.rcv_proc] pop edi test [eax + SOCKET.state], SS_CANTRCVMORE jnz .return cmp ebx, EWOULDBLOCK jne .return test edi, MSG_DONTWAIT jnz .return_err ; test [eax + SOCKET.options], SO_NONBLOCK ; jnz .return_err call SOCKET_block jmp .loop .invalid: push EINVAL pop ebx .return_err: mov ecx, -1 .return: mov [esp+20], ebx mov [esp+32], ecx ret align 4 SOCKET_receive_dgram: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_receive: DGRAM\n" mov ebx, esi ; bufferlength get_from_queue (eax + SOCKET_QUEUE_LOCATION), SOCKET_QUEUE_SIZE, sizeof.socket_queue_entry, .wouldblock ; sets esi only on success. mov ecx, [esi + socket_queue_entry.data_size] DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_receive: %u bytes data\n", ecx cmp ecx, ebx ; If data segment does not fit in applications buffer, abort ja .too_small push eax ecx push [esi + socket_queue_entry.buf_ptr] ; save the buffer addr so we can clear it later mov esi, [esi + socket_queue_entry.data_ptr] DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_receive: Source buffer=%x real addr=%x\n", [esp], esi ; copy the data from kernel buffer to application buffer mov edi, edx ; bufferaddr shr ecx, 1 jnc .nb movsb .nb: shr ecx, 1 jnc .nw movsw .nw: test ecx, ecx jz .nd rep movsd .nd: call NET_packet_free pop ecx eax ; return number of bytes copied to application xor ebx, ebx ret .too_small: mov ecx, -1 push EMSGSIZE pop ebx ret .wouldblock: push EWOULDBLOCK pop ebx ret align 4 SOCKET_receive_local: ; does this socket have a PID yet? cmp [eax + SOCKET.PID], 0 jne @f ; Change PID to that of current process mov ebx, [TASK_BASE] mov ebx, [ebx + TASKDATA.pid] mov [eax + SOCKET.PID], ebx mov [eax + SOCKET.TID], ebx ; currently TID = PID in kolibrios :( @@: mov [eax + SOCKET.rcv_proc], SOCKET_receive_stream ; ... continue to SOCKET_receive_stream align 4 SOCKET_receive_stream: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_receive: STREAM\n" cmp [eax + STREAM_SOCKET.rcv + RING_BUFFER.size], 0 je .wouldblock test edi, MSG_PEEK jnz .peek mov ecx, esi mov edi, edx xor edx, edx push eax add eax, STREAM_SOCKET.rcv call SOCKET_ring_read ; copy data from kernel buffer to application buffer call SOCKET_ring_free ; free read memory pop eax xor ebx, ebx ; errorcode = 0 (no error) ret .wouldblock: push EWOULDBLOCK pop ebx xor ecx, ecx ret .peek: mov ecx, [eax + STREAM_SOCKET.rcv + RING_BUFFER.size] xor ebx, ebx ret ;----------------------------------------------------------------- ; ; SOCKET_send ; ; ; IN: socket number in ecx ; pointer to data in edx ; datalength in esi ; flags in edi ; OUT: -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_send: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: socknum=%u data ptr=%x length=%u flags=%x\n", ecx, edx, esi, edi call SOCKET_num_to_ptr jz .invalid mov ecx, esi mov esi, edx jmp [eax + SOCKET.snd_proc] .invalid: mov dword[esp+20], EINVAL mov dword[esp+32], -1 ret align 4 SOCKET_send_udp: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: UDP\n" mov [esp+32], ecx call UDP_output cmp eax, -1 je .error ret .error: mov dword[esp+32], -1 mov dword[esp+20], EMSGSIZE ; FIXME: UDP_output should return error codes! ret align 4 SOCKET_send_tcp: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: TCP\n" push eax add eax, STREAM_SOCKET.snd call SOCKET_ring_write pop eax mov [esp+32], ecx mov [eax + SOCKET.errorcode], 0 push eax call TCP_output ; FIXME: this doesnt look pretty, does it? pop eax mov eax, [eax + SOCKET.errorcode] mov [esp+20], eax ret align 4 SOCKET_send_ip: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: IPv4\n" mov [esp+32], ecx call IPv4_output_raw ; FIXME: IPv4_output_raw should return error codes! cmp eax, -1 je .error ret .error: mov dword[esp+32], -1 mov dword[esp+20], EMSGSIZE ret align 4 SOCKET_send_icmp: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: ICMP\n" mov [esp+32], ecx call ICMP_output_raw ; FIXME: errorcodes cmp eax, -1 je .error ret .error: mov dword[esp+32], -1 mov dword[esp+20], EMSGSIZE ret align 4 SOCKET_send_pppoe: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: PPPoE\n" mov [esp+32], ecx mov ebx, [eax + SOCKET.device] call PPPoE_discovery_output ; FIXME: errorcodes cmp eax, -1 je .error ret .error: mov dword[esp+32], -1 mov dword[esp+20], EMSGSIZE ret align 4 SOCKET_send_local: ; does this socket have a PID yet? cmp [eax + SOCKET.PID], 0 jne @f ; Change PID to that of current process mov ebx, [TASK_BASE] mov ebx, [ebx + TASKDATA.pid] mov [eax + SOCKET.PID], ebx mov [eax + SOCKET.TID], ebx ; currently TID = PID in kolibrios :( @@: mov [eax + SOCKET.snd_proc], SOCKET_send_local_ align 4 SOCKET_send_local_: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_send: LOCAL\n" ; get the other side's socket and check if it still exists mov eax, [eax + SOCKET.device] call SOCKET_check jz .invalid ; allright, shove in the data! push eax add eax, STREAM_SOCKET.rcv call SOCKET_ring_write pop eax ; return the number of written bytes (or errorcode) to application mov [esp+32], ecx ; and notify the other end call SOCKET_notify ret .invalid: mov dword[esp+32], -1 mov dword[esp+20], EINVAL ret ;----------------------------------------------------------------- ; ; SOCKET_get_options ; ; IN: ecx = socket number ; edx = pointer to the options: ; dd level, optname, optval, optlen ; OUT: -1 on error ; ; At moment, uses only pseudo-optname -2 for get last_ack_number for TCP. ; TODO: find best way to notify that send()'ed data were acknowledged ; Also pseudo-optname -3 is valid and returns socket state, one of TCPS_*. ; ;----------------------------------------------------------------- align 4 SOCKET_get_opt: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_get_opt\n" call SOCKET_num_to_ptr jz .invalid cmp dword [edx], IP_PROTO_TCP jne .invalid cmp dword [edx+4], -2 je @f cmp dword [edx+4], -3 jne .invalid @@: ; mov eax, [edx+12] ; test eax, eax ; jz .fail ; cmp dword [eax], 4 ; mov dword [eax], 4 ; jb .fail ; stdcall net_socket_num_to_addr, ecx ; test eax, eax ; jz .fail ; ; todo: check that eax is really TCP socket ; mov ecx, [eax + TCP_SOCKET.last_ack_number] ; cmp dword [edx+4], -2 ; jz @f ; mov ecx, [eax + TCP_SOCKET.state] @@: mov eax, [edx+8] test eax, eax jz @f mov [eax], ecx @@: mov dword [esp+32], 0 ret .invalid: mov dword[esp+32], -1 mov dword[esp+20], EINVAL ret ;----------------------------------------------------------------- ; ; SOCKET_set_options ; ; IN: ecx = socket number ; edx = pointer to the options: ; dd level, optname, optlen, optval ; OUT: -1 on error ; ;----------------------------------------------------------------- align 4 SOCKET_set_opt: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_set_opt\n" call SOCKET_num_to_ptr jz .invalid cmp dword [edx], SOL_SOCKET jne .invalid cmp dword [edx+4], SO_BINDTODEVICE je .bind .invalid: mov dword[esp+32], -1 mov dword[esp+20], EINVAL ret .bind: cmp dword[edx+8], 0 je .unbind movzx edx, byte[edx + 9] cmp edx, NET_DEVICES_MAX ja .invalid mov edx, [NET_DRV_LIST + 4*edx] test edx, edx jz .already mov [eax + SOCKET.device], edx DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_set_opt: Bound socket %x to device %x\n",eax, edx mov dword[esp+32], 0 ; success! ret .unbind: mov [eax + SOCKET.device], 0 mov dword[esp+32], 0 ; success! ret .already: mov dword[esp+20], EALREADY mov dword[esp+32], -1 ret ;----------------------------------------------------------------- ; ; SOCKET_pair ; ; Allocates a pair of linked LOCAL domain sockets ; ; IN: / ; OUT: eax is socket1 num, -1 on error ; ebx is socket2 num ; ;----------------------------------------------------------------- align 4 SOCKET_pair: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_pair\n" call SOCKET_alloc jz .nomem1 mov [esp+32], edi ; application's eax mov [eax + SOCKET.Domain], AF_LOCAL mov [eax + SOCKET.Type], SOCK_STREAM mov [eax + SOCKET.Protocol], 0 ;;; CHECKME mov [eax + SOCKET.snd_proc], SOCKET_send_local mov [eax + SOCKET.rcv_proc], SOCKET_receive_local mov [eax + SOCKET.PID], 0 mov ebx, eax call SOCKET_alloc jz .nomem2 mov [esp+20], edi ; application's ebx mov [eax + SOCKET.Domain], AF_LOCAL mov [eax + SOCKET.Type], SOCK_STREAM mov [eax + SOCKET.Protocol], 0 ;;; CHECKME mov [eax + SOCKET.snd_proc], SOCKET_send_local mov [eax + SOCKET.rcv_proc], SOCKET_receive_local mov [eax + SOCKET.PID], 0 ; Link the two sockets to eachother mov [eax + SOCKET.device], ebx mov [ebx + SOCKET.device], eax lea eax, [eax + STREAM_SOCKET.rcv] call SOCKET_ring_create lea eax, [ebx + STREAM_SOCKET.rcv] call SOCKET_ring_create pop eax ret .nomem2: mov eax, ebx call SOCKET_free .nomem1: mov dword[esp+32], -1 mov dword[esp+28], ENOMEM ret ;----------------------------------------------------------------- ; ; SOCKET_debug ; ; Copies socket variables to application buffer ; ; IN: ecx = socket number ; edx = pointer to buffer ; ; OUT: -1 on error ;----------------------------------------------------------------- align 4 SOCKET_debug: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_debug\n" mov edi, edx test ecx, ecx jz .returnall call SOCKET_num_to_ptr jz .invalid mov esi, eax mov ecx, SOCKETBUFFSIZE/4 rep movsd mov dword[esp+32], 0 ret .returnall: mov ebx, net_sockets .next_socket: mov ebx, [ebx + SOCKET.NextPtr] test ebx, ebx jz .done mov eax, [ebx + SOCKET.Number] stosd jmp .next_socket .done: xor eax, eax stosd mov dword[esp+32], eax ret .invalid: mov dword[esp+32], -1 mov dword[esp+28], EINVAL ret ;----------------------------------------------------------------- ; ; SOCKET_find_port ; ; Fills in the local port number for TCP and UDP sockets ; This procedure always works because the number of sockets is ; limited to a smaller number then the number of possible ports ; ; IN: eax = socket pointer ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_find_port: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_find_port\n" push ebx esi ecx cmp [eax + SOCKET.Protocol], IP_PROTO_UDP je .udp cmp [eax + SOCKET.Protocol], IP_PROTO_TCP je .tcp pop ecx esi ebx ret .udp: mov bx, [last_UDP_port] call .findit mov [last_UDP_port], bx pop ecx esi ebx ret .tcp: mov bx, [last_TCP_port] call .findit mov [last_TCP_port], bx pop ecx esi ebx ret .restart: mov bx, MIN_EPHEMERAL_PORT_N .findit: cmp bx, MAX_EPHEMERAL_PORT_N je .restart add bh, 1 adc bl, 0 call SOCKET_check_port jz .findit ret ;----------------------------------------------------------------- ; ; SOCKET_check_port (to be used with AF_INET only!) ; ; Checks if a local port number is unused ; If the proposed port number is unused, it is filled in in the socket structure ; ; IN: eax = socket ptr (to find out if its a TCP/UDP socket) ; bx = proposed socket number (network byte order) ; ; OUT: ZF = set on error ; ;----------------------------------------------------------------- align 4 SOCKET_check_port: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_check_port: " pusha mov ecx, socket_mutex call mutex_lock popa mov ecx, [eax + SOCKET.Protocol] mov edx, [eax + IP_SOCKET.LocalIP] mov esi, net_sockets .next_socket: mov esi, [esi + SOCKET.NextPtr] or esi, esi jz .port_ok cmp [esi + SOCKET.Protocol], ecx jne .next_socket cmp [esi + IP_SOCKET.LocalIP], edx jne .next_socket cmp [esi + UDP_SOCKET.LocalPort], bx jne .next_socket pusha mov ecx, socket_mutex call mutex_unlock popa DEBUGF DEBUG_NETWORK_VERBOSE, "local port %x already in use\n", bx ; FIXME: find a way to print big endian values with debugf ret .port_ok: pusha mov ecx, socket_mutex call mutex_unlock popa DEBUGF DEBUG_NETWORK_VERBOSE, "local port %x is free\n", bx ; FIXME: find a way to print big endian values with debugf mov [eax + UDP_SOCKET.LocalPort], bx or bx, bx ; clear the zero-flag ret ;----------------------------------------------------------------- ; ; SOCKET_input ; ; Updates a (stateless) socket with received data ; ; Note: the mutex should already be set ! ; ; IN: eax = socket ptr ; ecx = data size ; esi = ptr to data ; [esp] = ptr to buf ; [esp + 4] = buf size ; ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_input: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_input: socket=%x, data=%x size=%u\n", eax, esi, ecx mov [esp+4], ecx push esi mov esi, esp add_to_queue (eax + SOCKET_QUEUE_LOCATION), SOCKET_QUEUE_SIZE, sizeof.socket_queue_entry, SOCKET_input.full DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_input: success\n" add esp, sizeof.socket_queue_entry pusha lea ecx, [eax + SOCKET.mutex] call mutex_unlock popa jmp SOCKET_notify .full: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_input: socket %x is full!\n", eax pusha lea ecx, [eax + SOCKET.mutex] call mutex_unlock popa call NET_packet_free add esp, 8 ret ;-------------------------- ; ; eax = ptr to ring struct (just a buffer of the right size) ; align 4 SOCKET_ring_create: push esi mov esi, eax push edx stdcall create_ring_buffer, SOCKET_MAXDATA, PG_SW pop edx DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_created: %x\n", eax pusha lea ecx, [esi + RING_BUFFER.mutex] call mutex_init popa mov [esi + RING_BUFFER.start_ptr], eax mov [esi + RING_BUFFER.write_ptr], eax mov [esi + RING_BUFFER.read_ptr], eax mov [esi + RING_BUFFER.size], 0 add eax, SOCKET_MAXDATA mov [esi + RING_BUFFER.end_ptr], eax mov eax, esi pop esi ret ;----------------------------------------------------------------- ; ; SOCKET_ring_write ; ; Adds data to a stream socket, and updates write pointer and size ; ; IN: eax = ptr to ring struct ; ecx = data size ; esi = ptr to data ; ; OUT: ecx = number of bytes stored ; ;----------------------------------------------------------------- align 4 SOCKET_ring_write: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_write: ringbuff=%x ptr=%x size=%u\n", eax, esi, ecx ; lock mutex pusha lea ecx, [eax + RING_BUFFER.mutex] call mutex_lock ; TODO: check what registers this function actually destroys popa ; calculate available size mov edi, SOCKET_MAXDATA sub edi, [eax + RING_BUFFER.size] ; available buffer size in edi cmp ecx, edi jbe .copy mov ecx, edi .copy: mov edi, [eax + RING_BUFFER.write_ptr] DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_write: %u bytes from %x to %x\n", ecx, esi, edi ; update write ptr push edi add edi, ecx cmp edi, [eax + RING_BUFFER.end_ptr] jb @f sub edi, SOCKET_MAXDATA ; WRAP @@: mov [eax + RING_BUFFER.write_ptr], edi pop edi ; update size add [eax + RING_BUFFER.size], ecx ; copy the data push ecx shr ecx, 1 jnc .nb movsb .nb: shr ecx, 1 jnc .nw movsw .nw: test ecx, ecx jz .nd rep movsd .nd: pop ecx ; unlock mutex push eax ecx lea ecx, [eax + RING_BUFFER.mutex] call mutex_unlock ; TODO: check what registers this function actually destroys pop ecx eax ret ;----------------------------------------------------------------- ; ; SOCKET_ring_read ; ; IN: eax = ring struct ptr ; ecx = bytes to read ; edx = offset ; edi = ptr to buffer start ; ; OUT: eax = unchanged ; ecx = number of bytes read (0 on error) ; edx = destroyed ; esi = destroyed ; edi = ptr to buffer end ; ;----------------------------------------------------------------- align 4 SOCKET_ring_read: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_read: ringbuff=%x ptr=%x size=%u offset=%x\n", eax, edi, ecx, edx pusha lea ecx, [eax + RING_BUFFER.mutex] call mutex_lock ; TODO: check what registers this function actually destroys popa mov esi, [eax + RING_BUFFER.read_ptr] add esi, edx ; esi = start_ptr + offset neg edx add edx, [eax + RING_BUFFER.size] ; edx = snd.size - offset jle .no_data_at_all pusha lea ecx, [eax + RING_BUFFER.mutex] call mutex_unlock ; TODO: check what registers this function actually destroys popa cmp ecx, edx ja .less_data .copy: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_read: %u bytes from %x to %x\n", ecx, esi, edi push ecx shr ecx, 1 jnc .nb movsb .nb: shr ecx, 1 jnc .nw movsw .nw: test ecx, ecx jz .nd rep movsd .nd: pop ecx ret .no_data_at_all: pusha lea ecx, [eax + RING_BUFFER.mutex] call mutex_unlock ; TODO: check what registers this function actually destroys popa DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_read: no data at all!\n" xor ecx, ecx ret .less_data: mov ecx, edx jmp .copy ;----------------------------------------------------------------- ; ; SOCKET_ring_free ; ; Free's some bytes from the ringbuffer ; ; IN: eax = ptr to ring struct ; ecx = data size ; ; OUT: ecx = number of bytes free-ed ; ;----------------------------------------------------------------- align 4 SOCKET_ring_free: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_free: %u bytes from ring %x\n", ecx, eax push eax ecx lea ecx, [eax + RING_BUFFER.mutex] call mutex_lock ; TODO: check what registers this function actually destroys pop ecx eax sub [eax + RING_BUFFER.size], ecx jb .error add [eax + RING_BUFFER.read_ptr], ecx mov edx, [eax + RING_BUFFER.end_ptr] cmp [eax + RING_BUFFER.read_ptr], edx jb @f sub [eax + RING_BUFFER.read_ptr], SOCKET_MAXDATA @@: push eax ecx lea ecx, [eax + RING_BUFFER.mutex] ; TODO: check what registers this function actually destroys call mutex_unlock pop ecx eax ret .error: ; we could free all available bytes, but that would be stupid, i guess.. DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ring_free: buffer=%x error!\n", eax add [eax + RING_BUFFER.size], ecx push eax lea ecx, [eax + RING_BUFFER.mutex] call mutex_unlock ; TODO: check what registers this function actually destroys pop eax xor ecx, ecx ret ;----------------------------------------------------------------- ; ; SOCKET_block ; ; Suspends the thread attached to a socket ; ; IN: eax = socket ptr ; OUT: eax = unchanged ; ;----------------------------------------------------------------- align 4 SOCKET_block: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_block: %x\n", eax pushf push eax cli ; Set the 'socket is blocked' flag or [eax + SOCKET.state], SS_BLOCKED ; Suspend the thread push edx mov edx, [TASK_BASE] mov [edx + TASKDATA.state], 1 ; Suspended ; Remember the thread ID so we can wake it up again mov edx, [edx + TASKDATA.pid] DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_block: suspending thread: %u\n", edx mov [eax + SOCKET.TID], edx pop edx call change_task pop eax popf DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_block: continueing\n" ret ;----------------------------------------------------------------- ; ; SOCKET_notify ; ; notify's the owner of a socket that something happened ; ; IN: eax = socket ptr ; OUT: eax = unchanged ; ;----------------------------------------------------------------- align 4 SOCKET_notify: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_notify: %x\n", eax call SOCKET_check jz .error test [eax + SOCKET.state], SS_BLOCKED jnz .unblock ; test [eax + SOCKET.options], SO_NONBLOCK ; jz .error push eax ecx esi ; socket exists and is of non blocking type. ; We'll try to flag an event to the thread mov eax, [eax + SOCKET.TID] test eax, eax jz .done mov ecx, 1 mov esi, TASK_DATA + TASKDATA.pid .next_pid: cmp [esi], eax je .found_pid inc ecx add esi, 0x20 cmp ecx, [TASK_COUNT] jbe .next_pid ; PID not found, TODO: close socket! jmp .done .found_pid: shl ecx, 8 or [ecx + SLOT_BASE + APPDATA.event_mask], EVENT_NETWORK DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_notify: poking thread %u!\n", eax jmp .done .unblock: push eax ecx esi ; Clear the 'socket is blocked' flag and [eax + SOCKET.state], not SS_BLOCKED ; Find the thread's TASK_DATA mov eax, [eax + SOCKET.TID] test eax, eax jz .error xor ecx, ecx inc ecx mov esi, TASK_DATA .next: cmp [esi + TASKDATA.pid], eax je .found inc ecx add esi, 0x20 cmp ecx, [TASK_COUNT] jbe .next jmp .error .found: ; Run the thread mov [esi + TASKDATA.state], 0 ; Running DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_notify: Unblocked socket!\n" .done: pop esi ecx eax .error: ret ;-------------------------------------------------------------------- ; ; SOCKET_alloc ; ; Allocate memory for socket data and put new socket into the list ; Newly created socket is initialized with calling PID and number and ; put into beginning of list (which is a fastest way). ; ; IN: / ; OUT: eax = 0 on error, socket ptr otherwise ; edi = socket number ; ZF = cleared on error ; ;-------------------------------------------------------------------- align 4 SOCKET_alloc: push ebx stdcall kernel_alloc, SOCKETBUFFSIZE DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_alloc: ptr=%x\n", eax or eax, eax jz .exit ; zero-initialize allocated memory push eax mov edi, eax mov ecx, SOCKETBUFFSIZE / 4 xor eax, eax rep stosd pop eax ; set send-and receive procedures to return -1 mov [eax + SOCKET.snd_proc], .not_yet mov [eax + SOCKET.rcv_proc], .not_yet pusha mov ecx, socket_mutex call mutex_lock popa ; find first free socket number and use it mov edi, [last_socket_num] .next_socket_number: inc edi jz .next_socket_number ; avoid socket nr 0 cmp edi, -1 je .next_socket_number ; avoid socket nr -1 mov ebx, net_sockets .next_socket: mov ebx, [ebx + SOCKET.NextPtr] test ebx, ebx jz .last_socket cmp [ebx + SOCKET.Number], edi jne .next_socket jmp .next_socket_number .last_socket: mov [last_socket_num], edi mov [eax + SOCKET.Number], edi DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_alloc: number=%u\n", edi ; Fill in PID mov ebx, [TASK_BASE] mov ebx, [ebx + TASKDATA.pid] mov [eax + SOCKET.PID], ebx mov [eax + SOCKET.TID], ebx ; currently TID = PID in kolibrios :( ; init mutex pusha lea ecx, [eax + SOCKET.mutex] call mutex_init popa ; add socket to the list by re-arranging some pointers mov ebx, [net_sockets + SOCKET.NextPtr] mov [eax + SOCKET.PrevPtr], net_sockets mov [eax + SOCKET.NextPtr], ebx test ebx, ebx jz @f pusha lea ecx, [ebx + SOCKET.mutex] call mutex_lock popa mov [ebx + SOCKET.PrevPtr], eax pusha lea ecx, [ebx + SOCKET.mutex] call mutex_unlock popa @@: mov [net_sockets + SOCKET.NextPtr], eax or eax, eax ; used to clear zero flag pusha mov ecx, socket_mutex call mutex_unlock popa .exit: pop ebx ret .not_yet: mov dword[esp+20], ENOTCONN mov dword[esp+32], -1 ret ;---------------------------------------------------- ; ; SOCKET_free ; ; Free socket data memory and remove socket from the list ; Caller should lock and unlock socket_mutex ; ; IN: eax = socket ptr ; OUT: / ; ;---------------------------------------------------- align 4 SOCKET_free: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_free: %x\n", eax call SOCKET_check jz .error push ebx pusha lea ecx, [eax + SOCKET.mutex] call mutex_lock popa cmp [eax + SOCKET.Domain], AF_INET4 jnz .no_tcp cmp [eax + SOCKET.Protocol], IP_PROTO_TCP jnz .no_tcp mov ebx, eax stdcall kernel_free, [ebx + STREAM_SOCKET.rcv.start_ptr] stdcall kernel_free, [ebx + STREAM_SOCKET.snd.start_ptr] mov eax, ebx .no_tcp: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_free: freeing socket %x\n", eax push eax ; this will be passed to kernel_free mov ebx, [eax + SOCKET.NextPtr] mov eax, [eax + SOCKET.PrevPtr] DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_free: linking socket %x to socket %x\n", eax, ebx test eax, eax jz @f mov [eax + SOCKET.NextPtr], ebx @@: test ebx, ebx jz @f mov [ebx + SOCKET.PrevPtr], eax @@: call kernel_free pop ebx DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_free: success!\n" .error: ret ;------------------------------------ ; ; SOCKET_fork ; ; Create a child socket ; ; IN: socket nr in ebx ; OUT: child socket nr in eax ; ;----------------------------------- align 4 SOCKET_fork: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_fork: %x\n", ebx ; Exit if backlog queue is full mov eax, [ebx + SOCKET_QUEUE_LOCATION + queue.size] cmp ax, [ebx + SOCKET.backlog] jae .fail ; Allocate new socket push ebx call SOCKET_alloc pop ebx jz .fail push eax mov esi, esp add_to_queue (ebx + SOCKET_QUEUE_LOCATION), MAX_backlog, 4, .fail2 pop eax ; Copy structure from current socket to new ; We start at PID to preserve the socket num, 2 pointers and mutex ; TID will be filled in later lea esi, [ebx + SOCKET.PID] lea edi, [eax + SOCKET.PID] mov ecx, (SOCKET_QUEUE_LOCATION - SOCKET.PID + 3)/4 rep movsd and [eax + SOCKET.options], not SO_ACCEPTCON ; Notify owner of parent socket push eax mov eax, ebx call SOCKET_notify pop eax ret .fail2: add esp, 4+4+4 .fail: DEBUGF DEBUG_NETWORK_ERROR, "SOCKET_fork: failed\n" xor eax, eax ret ;--------------------------------------------------- ; ; SOCKET_num_to_ptr ; ; Get socket structure address by its number ; ; IN: ecx = socket number ; OUT: eax = 0 on error, socket ptr otherwise ; ZF = set on error ; ;--------------------------------------------------- align 4 SOCKET_num_to_ptr: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_num_to_ptr: num=%u ", ecx pusha mov ecx, socket_mutex call mutex_lock popa mov eax, net_sockets .next_socket: mov eax, [eax + SOCKET.NextPtr] or eax, eax jz .error diff16 "tetten", 0, $ cmp [eax + SOCKET.Number], ecx jne .next_socket test eax, eax pusha mov ecx, socket_mutex call mutex_unlock popa DEBUGF DEBUG_NETWORK_VERBOSE, "ptr=%x\n", eax ret .error: pusha mov ecx, socket_mutex call mutex_unlock popa DEBUGF DEBUG_NETWORK_ERROR, "SOCKET_num_to_ptr: not found\n", eax ret ;--------------------------------------------------- ; ; SOCKET_ptr_to_num ; ; Get socket number by its address ; ; IN: eax = socket ptr ; OUT: eax = 0 on error, socket num otherwise ; ZF = set on error ; ;--------------------------------------------------- align 4 SOCKET_ptr_to_num: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_ptr_to_num: ptr=%x ", eax call SOCKET_check jz .error mov eax, [eax + SOCKET.Number] DEBUGF DEBUG_NETWORK_VERBOSE, "num=%u\n", eax ret .error: DEBUGF DEBUG_NETWORK_ERROR, "SOCKET_ptr_to_num: not found\n", eax ret ;--------------------------------------------------- ; ; SOCKET_check ; ; checks if the given value is really a socket ptr ; ; IN: eax = socket ptr ; OUT: eax = 0 on error, unchanged otherwise ; ZF = set on error ; ;--------------------------------------------------- align 4 SOCKET_check: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_check: %x\n", eax push ebx mov ebx, net_sockets .next_socket: mov ebx, [ebx + SOCKET.NextPtr] or ebx, ebx jz .done cmp ebx, eax jnz .next_socket .done: mov eax, ebx test eax, eax pop ebx ret ;--------------------------------------------------- ; ; SOCKET_check_owner ; ; checks if the caller application owns the socket ; ; IN: eax = socket ptr ; OUT: ZF = true/false ; ;--------------------------------------------------- align 4 SOCKET_check_owner: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_check_owner: %x\n", eax push ebx mov ebx, [TASK_BASE] mov ebx, [ebx + TASKDATA.pid] cmp [eax + SOCKET.PID], ebx pop ebx ret ;------------------------------------------------------ ; ; SOCKET_process_end ; ; Kernel calls this function when a certain process ends ; This function will check if the process had any open sockets ; And update them accordingly ; ; IN: edx = pid ; OUT: / ; ;------------------------------------------------------ align 4 SOCKET_process_end: cmp [net_sockets + SOCKET.NextPtr], 0 ; Are there any active sockets at all? je .quickret ; nope, exit immediately ; TODO: run the following code in another thread, to avoid deadlock DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_process_end: %x\n", edx pusha mov ecx, socket_mutex call mutex_lock popa push ebx mov ebx, net_sockets .next_socket: mov ebx, [ebx + SOCKET.NextPtr] .next_socket_test: test ebx, ebx jz .done cmp [ebx + SOCKET.PID], edx jne .next_socket DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_process_end: killing socket %x\n", ebx mov [ebx + SOCKET.PID], 0 mov eax, ebx mov ebx, [ebx + SOCKET.NextPtr] pusha cmp [eax + SOCKET.Domain], AF_INET4 jne .free cmp [eax + SOCKET.Protocol], IP_PROTO_TCP jne .free call TCP_close jmp .closed .free: call SOCKET_free .closed: popa jmp .next_socket_test .done: pop ebx pusha mov ecx, socket_mutex call mutex_unlock popa .quickret: ret ;----------------------------------------------------------------- ; ; SOCKET_is_connecting ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_is_connecting: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_is_connecting: %x\n", eax and [eax + SOCKET.state], not (SS_ISCONNECTED + SS_ISDISCONNECTING + SS_ISCONFIRMING) or [eax + SOCKET.state], SS_ISCONNECTING ret ;----------------------------------------------------------------- ; ; SOCKET_is_connected ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_is_connected: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_is_connected: %x\n", eax and [eax + SOCKET.state], not (SS_ISCONNECTING + SS_ISDISCONNECTING + SS_ISCONFIRMING) or [eax + SOCKET.state], SS_ISCONNECTED jmp SOCKET_notify ;----------------------------------------------------------------- ; ; SOCKET_is_disconnecting ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_is_disconnecting: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_is_disconnecting: %x\n", eax and [eax + SOCKET.state], not (SS_ISCONNECTING) or [eax + SOCKET.state], SS_ISDISCONNECTING + SS_CANTRCVMORE + SS_CANTSENDMORE jmp SOCKET_notify ;----------------------------------------------------------------- ; ; SOCKET_is_disconnected ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_is_disconnected: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_is_disconnected: %x\n", eax and [eax + SOCKET.state], not (SS_ISCONNECTING + SS_ISCONNECTED + SS_ISDISCONNECTING) or [eax + SOCKET.state], SS_CANTRCVMORE + SS_CANTSENDMORE cmp [eax + SOCKET.Protocol], IP_PROTO_TCP je .tcp cmp [eax + SOCKET.Protocol], IP_PROTO_UDP je .udp jmp SOCKET_notify .tcp: .udp: mov [eax + UDP_SOCKET.LocalPort], 0 ; UDP and TCP structs store localport at the same offset mov [eax + UDP_SOCKET.RemotePort], 0 jmp SOCKET_notify ;----------------------------------------------------------------- ; ; SOCKET_cant_recv_more ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_cant_recv_more: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_cant_recv_more: %x\n", eax or [eax + SOCKET.state], SS_CANTRCVMORE call SOCKET_notify ret ;----------------------------------------------------------------- ; ; SOCKET_cant_send_more ; ; IN: eax = socket ptr ; OUT: / ; ;----------------------------------------------------------------- align 4 SOCKET_cant_send_more: DEBUGF DEBUG_NETWORK_VERBOSE, "SOCKET_cant_send_more: %x\n", eax or [eax + SOCKET.state], SS_CANTSENDMORE mov [eax + SOCKET.snd_proc], .notconn call SOCKET_notify ret .notconn: mov dword[esp+20], ENOTCONN mov dword[esp+32], -1 ret