WebSVN – Kolibri OS – Diff – /kernel/branches/net/network/socket.inc

 ;;          GNU GENERAL PUBLIC LICENSE                             ;;
 ;;             Version 2, June 1991                                ;;
 ;;                                                                 ;;
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-$Revision: 1019 $
+$Revision: 1206 $
 align 4
 struct	 SOCKET
 	 .Domain		dd ? ; INET/UNIX/..
 	 .Type			dd ? ; RAW/UDP/TCP/...
 	 .Protocol		dd ? ; ICMP/IPv4/ARP/
 	 .LocalIP		dd ? ; local IP address
 	 .RemoteIP		dd ? ; remote IP address
-	 .LocalPort		dw ? ; local port
+	 .LocalPort		dw ? ; local port       (In INET byte order)
-	 .RemotePort		dw ? ; remote port
+	 .RemotePort		dw ? ; remote port      (IN INET byte order
 	 .OrigRemoteIP		dd ? ; original remote IP address (used to reset to LISTEN state)
 	 .OrigRemotePort	dw ? ; original remote port (used to reset to LISTEN state)
 	 .rxDataCount		dd ? ; rx data count
 	 .TCBState		dd ? ; TCB state
 	 .TCBTimer		dd ? ; TCB timer (seconds)
 ;       type in edx
 ;       protocol in esi
 ;  OUT: eax is socket num, -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_open:
 	DEBUGF	1,"socket_open: domain: %u, type: %u",ecx, edx
 ;       pointer to sockaddr struct in edx
 ;       length of that struct in esi
 ;  OUT: 0 on success
+;
 ;-----------------------------------------------
+align 4
 socket_bind:
 	DEBUGF	1,"Socket_bind: socknum: %u sockaddr: %x, length: %u, ",ecx,edx,esi
 	cmp	esi, 2
 	jl	s_error
-	cmp	word [edx], AF_INET4
-	je	.af_inet4
 	cmp	word [edx], AF_INET4
-	jmp	s_error
+	jne	s_error
+  .af_inet4:
   .af_inet4:
 	cmp	esi, 6
 	jl	s_error
-	mov	bx, word [edx + 2]
-	rol	bx,8			;;;
-	DEBUGF	1,"local port: %u ",bx
-	test	bx, bx
-	jnz	.check_only
-	mov	bx , [last_UDP_port]
-  .find_port_loop:
-	inc	bx
-	inc	[last_UDP_port]
-  .check_only:
-	mov	esi, net_sockets
-  .next_udp_socket:
-	mov	esi, [esi + SOCKET.NextPtr]
-	or	esi, esi
-	jz	.udp_port_ok
+	mov	ecx, [eax + SOCKET.Type]
-	cmp	[esi + SOCKET.Type], IP_PROTO_UDP
+	mov	bx, word [edx + 2]
-	jne	.next_udp_socket
+	DEBUGF	1,"local port: %x ",bx
+	test	bx, bx
-	cmp	[esi + SOCKET.LocalPort], bx
+	jz	.find_free
-	jne	.next_udp_socket
+	call	socket_check_port
+	test	bx, bx
-	cmp	word [edx + 2], 0
+	je	s_error
+	jmp	.got_port
 	jne	s_error
   .find_free:
 	cmp	bx, MAX_EPHEMERAL_PORT
 ;       length of that struct in esi
 ;  OUT: 0 on success
+;
 ;-----------------------------------------------
 align 4
 socket_connect:
 	DEBUGF	1,"Socket_connect: socknum: %u sockaddr: %x, length: %u,",ecx,edx,esi
 	je	.udp
 	cmp	[eax + SOCKET.Type], IP_PROTO_ICMP
 	je	.icmp
-;        cmp     [eax + SOCKET.Type], IP_PROTO_TCP
+	cmp	[eax + SOCKET.Type], IP_PROTO_TCP
-;        je      .tcp
+	je	.tcp
-	jmp	s_error
 	jmp	s_error
-  .udp:
-	mov	bx , word [edx + 2]
+  .udp:
 	rol	bx, 8
-	mov	word [eax + SOCKET.RemotePort], bx
 	mov	bx , word [edx + 2]
-	DEBUGF	1,"remote port: %u ",bx
+	mov	word [eax + SOCKET.RemotePort], bx
 	DEBUGF	1,"remote port: %x ",bx
 ;  IN:  socket number in ecx
 ;       backlog in edx
 ;  OUT: eax is socket num, -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_listen:
 	DEBUGF	1,"Socket_listen: socknum: %u backlog: %u\n",ecx,edx
 ;       addr in edx
 ;       addrlen in esi
 ;  OUT: eax is socket num, -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_accept:
 	DEBUGF	1,"Socket_accept: socknum: %u sockaddr: %x, length: %u\n",ecx,edx,esi
+;
 ;  IN:  socket number in ecx
 ;  OUT: eax is socket num, -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_close:
 	DEBUGF	1,"Socket_close: socknum: %u\n",ecx
 	je	.udp
 	cmp	[eax + SOCKET.Type], IP_PROTO_ICMP
 	je	.icmp
-;        cmp     [eax + SOCKET.Type], IP_PROTO_TCP
+	cmp	[eax + SOCKET.Type], IP_PROTO_TCP
-;        je      .tcp
+	je	.tcp
 ;       addrlen in esi
 ;       flags in edi
 ;  OUT: eax is number of bytes copied, -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_recv:
 	DEBUGF	1,"Socket_receive: socknum: %u sockaddr: %x, length: %u, flags: %x\n",ecx,edx,esi,edi
+;
 ; SOCKET_send
+;
+;
 ;  IN:  socket number in ecx
-;       addr in edx
+;       pointer to data in edx
-;       addrlen in esi
+;       datalength in esi
 ;       flags in edi
 ;  OUT: -1 on error
+;
 ;-----------------------------------------------
+align 4
 socket_send:
 	DEBUGF	1,"Socket_send: socknum: %u sockaddr: %x, length: %u, flags: %x, ",ecx,edx,esi,edi
 	stdcall net_socket_num_to_addr, ecx		   ; get real socket address
+	or	eax, eax
+	jz	s_error
+	cmp	word [eax + SOCKET.Domain], AF_INET4
+	je	.af_inet4
+	jmp	s_error
 	or	eax, eax
-	jz	s_error
+  .af_inet4:
 	DEBUGF	1,"Socket type:%u\n", [eax + SOCKET.Type]:4
 	cmp	[eax + SOCKET.Type], IP_PROTO_UDP
 	je	.udp
 	cmp	[eax + SOCKET.Type], IP_PROTO_ICMP
 	je	.icmp
+	cmp	[eax + SOCKET.Type], IP_PROTO_TCP
+	je	.tcp
+	jmp	s_error
+  .udp:
+	DEBUGF	1,"type: UDP, "
+	cmp	[eax + SOCKET.LocalPort],0
 	jne	.port_ok
 ;        cmp     [eax + SOCKET.Type], IP_PROTO_TCP
-;        je      .tcp
+	mov	ecx, [eax + SOCKET.Type]
+	call	socket_find_port
-	jmp	s_error
-  .udp:
+	test	bx, bx
 	je	s_error
-	DEBUGF	1,"type: UDP\n"
+	mov	[eax + SOCKET.LocalPort], bx
 	ret
+;-----------------------------------------------
+;
+; SOCKET_find_free_port (local port)
+;
+; works with INET byte order
+;
+;  IN:  type in ecx (TCP/UDP)
+;  OUT: bx = 0 on error, portnumber otherwise
+;
+;-----------------------------------------------
+align 4
+socket_find_port:
+	DEBUGF	1,"Socket_find_free_port, type: %u ",eax
+	cmp	ecx, IP_PROTO_UDP
+	je	.udp
+	cmp	ecx, IP_PROTO_TCP
+	je	.tcp
+  .udp:
+	mov	bx, [last_UDP_port]
+	je	.continue
+  .tcp:
+	mov	bx, [last_TCP_port]
+  .continue:
+	inc	bx
+  .check_only:
+	mov	esi, net_sockets
+  .next_socket:
+	mov	esi, [esi + SOCKET.NextPtr]
+	or	esi, esi
+	jz	.port_ok
+	cmp	[esi + SOCKET.Type], ecx
+	jne	.next_socket
+	rol	bx, 8
+	cmp	[esi + SOCKET.LocalPort], bx
+	rol	bx, 8				; this doesnt change the zero flag, does it ?
+	jne	.next_socket
+	cmp	bx, MAX_EPHEMERAL_PORT
+	jle	.continue
+	; todo: WRAP!
+;        mov     [last_UDP_port], MIN_EPHEMERAL_PORT
+  .exit:
+	xor	ebx, ebx
+  .port_ok:
+	rol	bx, 8
+	ret
+;-----------------------------------------------
+;
+; SOCKET_check_port (local port)
+;
+; works with INET byte order
+;
+;  IN:  type in ecx (TCP/UDP)
+;       port to check in bx
+;  OUT: bx = 0 on error, unchanged otherwise
+;
+;-----------------------------------------------
+align 4
+socket_check_port:
+	mov	esi, net_sockets
+  .next_socket:
+	mov	esi, [esi + SOCKET.NextPtr]
+	or	esi, esi
+	jz	.port_ok
+	cmp	[esi + SOCKET.Type], ecx
+	jne	.next_socket
+	cmp	[esi + SOCKET.LocalPort], bx
+	jne	.next_socket
+	xor	ebx, ebx
+  .port_ok:
+	ret
+;-----------------------------------------------
+;
+; SOCKET_internal_receiver
+;
+; Checks if any socket wants the received data
+; If so, update the socket
+;
+;  IN:  eax = socket number
+;       ecx = number of bytes
+;       esi = pointer to beginning of data
+;       dx = Remote port (in INET byte order)
+;       edi = IP address of sender
+;
+;  OUT: xxx
+;
+;-----------------------------------------------
+align 4
+socket_internal_receiver:
+	DEBUGF 1,"internal socket receiver\n"
+	lea	ebx, [eax + SOCKET.lock]
+	call	wait_mutex
+	mov	[eax + SOCKET.RemotePort], dx		; update remote port number
+	mov	[eax + SOCKET.RemoteIP], edi
+	mov	edx, [eax + SOCKET.rxDataCount] 	; get # of bytes already in buffer
+	DEBUGF 1,"bytes already in socket: %u ", edx
+	lea	edi, [ecx + edx]			; check for buffer overflow
+	cmp	edi, SOCKETBUFFSIZE - SOCKETHEADERSIZE	;
+	jg	.dump					;
+	lea	edi, [eax + SOCKET.rxData + edx]
+	add	[eax + SOCKET.rxDataCount], ecx 	; increment the count of bytes in buffer
+	DEBUGF 1,"adding %u bytes\n", ecx
+	; copy the data across
+	push	cx
+	shr	ecx, 2
+	rep	movsd
+	pop	cx
+	and	cx, 3
+	rep	movsb
+	DEBUGF 1,"socket updated\n"
+	mov	[eax + SOCKET.lock], 0
+	; flag an event to the application
+	mov	edx, [eax + SOCKET.PID] 		; get socket owner PID
+	mov	ecx, 1
+	mov	esi, TASK_DATA + TASKDATA.pid
+       .next_pid:
+	cmp	[esi], edx
+	je	.found_pid
+	inc	ecx
+	add	esi, 0x20
+	cmp	ecx, [TASK_COUNT]
+	jbe	.next_pid
+	ret
+       .found_pid:
+	shl	ecx, 8
+	or	[ecx + SLOT_BASE + APPDATA.event_mask], EVENT_NETWORK ; stack event
+	mov	[check_idle_semaphore], 200
+	ret
        .dump:

Subversion Repositories Kolibri OS

(root)/kernel/branches/net/network/socket.inc – Rev 1196 → 1206