format PE GUI 4.0
entry main

include 'win32a.inc'

section '.data' data readable writeable
        hello   db 'hello', 0
        error   db 'Error: %d', 0
        ip      db '127.0.0.1', 0
        wdata   WSADATA
        laddr   sockaddr_in
        caddr   sockaddr_in
        sock    dd ?
        csock   dd ?
        temp    rb 15
        bufer   rb 1024

section '.code' code readable executable
  main:
        invoke  WSAStartup, 0101h, wdata
        invoke  socket, AF_INET, SOCK_STREAM, 6
        mov     [sock], eax
        mov     [laddr.sin_family], AF_INET
        invoke  htons, 6112
        mov     [laddr.sin_port], ax
        invoke  inet_addr, ip
        mov     [laddr.sin_addr], eax
        invoke  bind, [sock], laddr, sizeof.sockaddr_in
        cmp     eax, 0
        jne     .error
        invoke  listen, [sock], 0

  .while:
        invoke  accept, [sock], caddr, sizeof.sockaddr_in
        mov     [csock], eax
        invoke  recv, [csock], bufer, 1024, 0
        invoke  lstrcmp, bufer, hello
        cmp     eax, 0
        jne     .while
        invoke  MessageBox, 0, hello, hello, 0
        jmp     .while

  .error:
        invoke  GetLastError
        invoke  wsprintf, temp, error, eax
        invoke  MessageBox, 0, temp, hello, MB_OK
        invoke  ExitProcess, 0

section '.idata' import data readable
        library kernel32, 'kernel32.dll',\
                user32, 'user32.dll',\
                wsock32, 'wsock32.dll'

        include 'api\kernel32.inc'
        include 'api\user32.inc'
        include 'api\wsock32.inc'

; TCP/IP функции для правильной работы клиента

;отправить все
;ecx - len
;ebx - &buf
proc sendall
local total dd 0

.w:
	
	cmp [total],ecx
	jnl .exit
		add ebx,[total]
		sub ecx,[total]

 		invoke send,[sock],ebx,ecx,0
 		inc eax
 		jz .exit
 		dec eax
		add [total],eax
		jmp .w
.exit:
ret
endp

;принять все
;ecx - len
;ebx - &buf
proc recvall
local total dd 0

.w:
	
	cmp [total],ecx
	jnl .exit
		add ebx,[total]
		sub ecx,[total]

 		invoke recv,[sock],ebx,ecx,0
 		inc eax
 		jz .exit
 		dec eax
		add [total],eax
		jmp .w
.exit:
ret
endp