kolibrios-fun/programs/network/tcpserv/tcpserv.asm

187 lines
5.4 KiB
NASM
Raw Normal View History

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; ;;
;; 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: