mirror of
https://github.com/Doczom/simple-httpd.git
synced 2025-09-21 02:50:09 +02:00
Doczom
00abdb31f0
Added new features: - begin_send_response - finish_send_response The close_server function has been updated and support for many flags for creating a response has been added Added a "chunked" type transmission New test modules have been added: - a module for server management - a module for test chunked transmission
148 lines
3.3 KiB
NASM
148 lines
3.3 KiB
NASM
; test api 0.1.0 - get cmd path and get context unit
|
|
format MS COFF ;<- this is lib format
|
|
public @EXPORT as 'EXPORTS'
|
|
|
|
NO_DEBUG_INPUT = 0
|
|
include "macros.inc"
|
|
purge mov,add,sub
|
|
include "proc32.inc"
|
|
|
|
include '../module_api.inc'
|
|
|
|
section '.flat' code readable align 16
|
|
|
|
unit_init:
|
|
xor eax, eax
|
|
push esi edi
|
|
mov esi, [esp + 4*2 + 4]
|
|
|
|
cmp dword[esi + IMPORT_DATA.version], API_VERSION
|
|
jne .exit
|
|
|
|
mov edi, IMPORT
|
|
mov ecx, [esi + IMPORT_DATA.sizeof]
|
|
shr ecx, 2 ; div 4
|
|
rep movsd
|
|
|
|
; create unit context
|
|
invoke IMPORT.Alloc, 4096 ; for cmd path
|
|
test eax, eax
|
|
jz .exit
|
|
|
|
mov esi, [esp + 4*2 + 8]
|
|
mov edi, eax
|
|
test esi, esi
|
|
jz .exit
|
|
pushfd
|
|
cld
|
|
@@:
|
|
movsb
|
|
cmp byte[esi - 1], 0
|
|
jnz @b
|
|
|
|
mov esi, [esp + 4*3 + 8]
|
|
mov edi, text_board.token
|
|
movsd
|
|
movsw
|
|
|
|
popfd
|
|
;unit init successful
|
|
.exit:
|
|
pop edi esi
|
|
ret 8
|
|
|
|
|
|
server_entry:
|
|
push esi edi
|
|
mov esi, [esp + 4*2 + 4] ; request context
|
|
|
|
mov edi, [esp + 4*2 + 8] ; unit context
|
|
cmp dword[edi], 0
|
|
je .err_403
|
|
|
|
; check arg "token"
|
|
|
|
invoke IMPORT.find_uri_arg, esi, key_token
|
|
test eax, eax
|
|
jz .err_403
|
|
mov ecx, esi
|
|
mov esi, eax
|
|
|
|
cmpsd
|
|
jne .err_403_1
|
|
cmpsw
|
|
jne .err_403_1
|
|
cmpsb
|
|
jne .err_403_1
|
|
cmp byte[edi - 1], 0
|
|
jne .err_403_1
|
|
mov esi, ecx
|
|
; check command
|
|
|
|
invoke IMPORT.find_uri_arg, esi, key_cmd
|
|
test eax, eax
|
|
jz .no_shutdown
|
|
|
|
cmp word[eax], 'S'
|
|
jnz .no_shutdown
|
|
|
|
invoke IMPORT.close_server
|
|
.no_shutdown:
|
|
|
|
invoke IMPORT.create_resp, esi, 0
|
|
test eax, eax
|
|
jz .exit
|
|
|
|
push eax
|
|
invoke IMPORT.send_resp, eax, text_board, text_board.size
|
|
invoke IMPORT.destruct_resp
|
|
.exit:
|
|
pop edi esi
|
|
ret 8
|
|
.err_403_1:
|
|
mov esi, ecx
|
|
.err_403:
|
|
invoke IMPORT.create_resp, esi, FLAG_NO_CACHE_CONTROL\
|
|
+ FLAG_NO_CONTENT_ENCODING
|
|
test eax, eax
|
|
jz .exit
|
|
|
|
mov edi, eax
|
|
invoke IMPORT.set_http_status, edi, dword '403'
|
|
invoke IMPORT.send_resp, edi, text403, text403.length
|
|
invoke IMPORT.destruct_resp, edi
|
|
jmp .exit
|
|
|
|
|
|
server_close:
|
|
mov ecx, [esp + 4]
|
|
invoke IMPORT.Free, ecx
|
|
ret 4
|
|
|
|
|
|
section '.data' data readable writable align 16
|
|
|
|
key_token:
|
|
db 'token', 0
|
|
key_cmd:
|
|
db 'cmd',0
|
|
|
|
text403:
|
|
db 'Access to server management is prohibited.',\
|
|
' An unknown token has been entered'
|
|
.length = $ - text403
|
|
|
|
text_board:
|
|
db '<body> <h1> Control panel of simple-httpd</h1> <a href="?cmd=S&token='
|
|
.token:
|
|
db ' ">Stop Server</a></body>'
|
|
.size = $ - text_board
|
|
|
|
@EXPORT:
|
|
export \
|
|
unit_init, 'httpd_init', \
|
|
server_entry, 'httpd_serv',\
|
|
server_close, 'httpd_close'
|
|
|
|
|
|
IMPORT IMPORT_DATA |