forked from KolibriOS/kolibrios
187 lines
5.4 KiB
NASM
187 lines
5.4 KiB
NASM
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;; ;;
|
|
;; Copyright (C) KolibriOS team 2010-2017. All rights reserved. ;;
|
|
;; Distributed under terms of the GNU General Public License ;;
|
|
;; ;;
|
|
;; tcpserv.asm - TCP server demo program for KolibriOS ;;
|
|
;; ;;
|
|
;; Written by hidnplayr@kolibrios.org ;;
|
|
;; ;;
|
|
;; GNU GENERAL PUBLIC LICENSE ;;
|
|
;; Version 2, June 1991 ;;
|
|
;; ;;
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
format binary as ""
|
|
|
|
BUFFERSIZE = 1500
|
|
|
|
use32
|
|
; standard header
|
|
db 'MENUET01' ; signature
|
|
dd 1 ; header version
|
|
dd start ; entry point
|
|
dd i_end ; initialized size
|
|
dd mem ; required memory
|
|
dd mem ; stack pointer
|
|
dd 0 ; parameters
|
|
dd 0 ; path
|
|
|
|
|
|
include '../../macros.inc'
|
|
purge mov,add,sub
|
|
include '../../proc32.inc'
|
|
include '../../dll.inc'
|
|
|
|
include '../../network.inc'
|
|
|
|
; entry point
|
|
start:
|
|
; load libraries
|
|
stdcall dll.Load, @IMPORT
|
|
test eax, eax
|
|
jnz exit
|
|
|
|
; initialize console
|
|
invoke con_start, 1
|
|
invoke con_init, 80, 25, 80, 25, title
|
|
|
|
; Set event mask to socket events only
|
|
mcall 40, EVM_STACK
|
|
|
|
; Write message str1 to the console
|
|
invoke con_write_asciiz, str1
|
|
|
|
; Allocate a TCP socket
|
|
mcall socket, AF_INET4, SOCK_STREAM, 0
|
|
cmp eax, -1
|
|
je sock_err
|
|
; Socket allocation succeeded, store it's number in socketnum
|
|
mov [socketnum], eax
|
|
|
|
; This might be needed in the future,
|
|
; SO_REUSEADDR option is not implemented in kernel yet.
|
|
; mcall setsockopt, [socketnum], SOL_SOCKET, SO_REUSEADDR, &yes,
|
|
; cmp eax, -1
|
|
; je opt_err
|
|
|
|
; Bind the socket to port 23 (as defined in sockaddr1)
|
|
mcall bind, [socketnum], sockaddr1, sockaddr1.length
|
|
cmp eax, -1
|
|
je bind_err
|
|
|
|
; Start listening for incoming connections, with a backlog of max 1
|
|
mcall listen, [socketnum], 1
|
|
cmp eax, -1
|
|
je listen_err
|
|
|
|
; Write message str2 to the console
|
|
invoke con_write_asciiz, str2
|
|
|
|
; (Wait for and) accept incoming connection
|
|
mcall accept, [socketnum], sockaddr1, sockaddr1.length
|
|
cmp eax, -1
|
|
je acpt_err
|
|
; We have a new incoming connection, store the new socket number in socketnum2
|
|
mov [socketnum2], eax
|
|
|
|
; Send a message on the incoming connection
|
|
mcall send, [socketnum2], hello, hello.length
|
|
|
|
; Print the received data to the console, untill socket is closed by remote end
|
|
.loop:
|
|
mcall recv, [socketnum2], buffer, BUFFERSIZE, 0
|
|
cmp eax, -1
|
|
je .loop
|
|
|
|
mov byte[buffer+eax], 0 ; Zero-terminate the data, so we can print it
|
|
invoke con_write_asciiz, buffer
|
|
jmp .loop
|
|
|
|
; Print error message
|
|
acpt_err:
|
|
invoke con_write_asciiz, str8
|
|
jmp done
|
|
|
|
listen_err:
|
|
invoke con_write_asciiz, str3
|
|
jmp done
|
|
|
|
bind_err:
|
|
invoke con_write_asciiz, str4
|
|
jmp done
|
|
|
|
sock_err:
|
|
invoke con_write_asciiz, str6
|
|
jmp done
|
|
|
|
|
|
done:
|
|
; Wait for user input
|
|
invoke con_getch2
|
|
; Close console
|
|
invoke con_exit, 1
|
|
exit:
|
|
; Close listening socket, if it is open
|
|
cmp [socketnum], 0
|
|
je @f
|
|
mcall close, [socketnum]
|
|
@@:
|
|
|
|
; Close second socket, if it is open
|
|
cmp [socketnum2], 0
|
|
je @f
|
|
mcall close, [socketnum2]
|
|
@@:
|
|
; Close application
|
|
mcall -1
|
|
|
|
|
|
|
|
; data
|
|
title db 'TCP stream server demo',0
|
|
str1 db 'Opening socket',10, 0
|
|
str2 db 'Listening for incoming connections...',10,0
|
|
str3 db 'Listen error',10,10,0
|
|
str4 db 'Bind error',10,10,0
|
|
str5 db 'Setsockopt error',10,10,0
|
|
str6 db 'Could not open socket',10,10,0
|
|
str7 db 'Got data',10,10,0
|
|
str8 db 'Error accepting connection',10,10,0
|
|
|
|
hello db 'Hello world!',0
|
|
.length = $ - hello
|
|
|
|
sockaddr1:
|
|
dw AF_INET4 ; IPv4
|
|
.port dw 23 shl 8 ; port 23 - network byte order
|
|
.ip dd 0
|
|
rb 10
|
|
.length = $ - sockaddr1
|
|
|
|
; import
|
|
align 4
|
|
@IMPORT:
|
|
|
|
library console, 'console.obj'
|
|
|
|
import console, \
|
|
con_start, 'START', \
|
|
con_init, 'con_init', \
|
|
con_write_asciiz, 'con_write_asciiz', \
|
|
con_exit, 'con_exit', \
|
|
con_gets, 'con_gets',\
|
|
con_cls, 'con_cls',\
|
|
con_printf, 'con_printf',\
|
|
con_getch2, 'con_getch2',\
|
|
con_set_cursor_pos, 'con_set_cursor_pos'
|
|
i_end:
|
|
|
|
socketnum dd 0
|
|
socketnum2 dd 0
|
|
buffer rb BUFFERSIZE
|
|
|
|
align 4
|
|
rb 4096 ; stack
|
|
mem:
|