Server/Client Socket
Всем известно, то, что первое появилась это глобальная сеть, а не локальная с приходом протокола TCP/IP для военных целей. Но так как этот способ связи был ненадёжным и опасным для передачи секретных данных, - его начали использовать во всём мире.
Каждый когда-нибудь хотел написать свой клиент сервер, чтобы пообщаться с другом через Интернет или в локальной сети. Сейчас я покажу и подробно прокомментирую, как это делается средствами API – Windows на MASM(ассемблере).
Итак, сервер: главной задачей, которого, является ожидание клиента, обработка пришедшего сообщения, добавление его в LISTBOX – controle и отсылка, что мы написали в EDIT – controle;
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
Компилируется: .386
.model flat, stdcall
option casemap :none
include \masm32\include\windows.inc
include \masm32\include\user32.inc
include \masm32\include\kernel32.inc
include \masm32\include\wsock32.inc
includelib \masm32\lib\wsock32.lib
includelib \masm32\lib\user32.lib
includelib \masm32\lib\kernel32.lib
_T macro p1, p2 ;макрос обработки строк
local l
ifb <p2>; Строка без имени
.data
l db p1, 0
.code
exitm <addr l>
else ; Строка с именем
.data
p1 db p2, 0
.code
exitm <addr p1>
endif
endm
DlgProc proto :HWND, :UINT, :WPARAM, :LPARAM
EditProc proto :HWND, :UINT, :WPARAM, :LPARAM
MESSG proto :HWND, :DWORD
ERROR proto :HWND, :DWORD, :BYTE
Port equ 80
WM_SOCKET equ WM_USER+100
VER_SOCKET equ 101h ; требуемая версия WinSock
BUF_LEN equ 1024 ; размер буфера для приема/передачи
; контролы из ресурсов
IDC_OUT equ 101 ; строка отсылки (контрол EDIT)
IDC_IN equ 102 ; приемное "окно" (контрол ListBox)
.data?
hInstance HINSTANCE ?
hDlg HWND ? ; хэндл окна диалога
lpfnEditProc HWND ? ; перехваченная оконная функция EDIT
hSocket dd ? ; хэндл слушающего сокета
hClient dd ? ; хэндл сокета клиента
IsSend dd ? ; при значении >0 разрешена передача данных клиенту
wsa WSADATA <> ; информация о версии WinSock
sin sockaddr_in <> ;структура узла
; буфер для операций ввода/вывода
IOBuff db BUF_LEN+1 dup(?)
.code
start:
invoke GetModuleHandle, NULL
mov hInstance, eax
; инициализируем Windows Sockets DLL
invoke WSAStartup, VER_SOCKET, addr wsa
.if (eax == NULL) && (wsa.wVersion == VER_SOCKET)
; создаем диалог
invoke DialogBoxParam, hInstance, _T("MAINDIALOG"), 0, addr DlgProc, 0
.else
invoke WSACleanup
invoke MessageBox, 0, _T("Windows socket not found!"), _T("ERROR"), \
MB_OK+MB_ICONERROR
mov eax, 90h ; заканчиваем программу с кодом ошибки
.endif
; уходим в хаОС
invoke ExitProcess, eax
; оконная функция диалога
DlgProc proc hWnd: HWND, uMsg: UINT, wParam: WPARAM, lParam: LPARAM
; определяем пришедшее сообщение
.if (uMsg == WM_INITDIALOG)
; меняем заголовок окна диалога
invoke SendMessage, hWnd, WM_SETTEXT, NULL, _T("Server")
mov IsSend, FALSE ; запрещаем отправку данных клиенту
mov eax, hWnd
mov hDlg, eax ; хэндл окна нам еще понадобится :)
; сабклассируем контрол EDIT
invoke GetDlgItem, eax, IDC_OUT
invoke SetWindowLong, eax, GWL_WNDPROC, addr EditProc
mov lpfnEditProc, eax ; сохраняем адрес предыдущей функции
; создаем сокет для прослушки порта на предмет входящих соединений
invoke socket, AF_INET, SOCK_STREAM, 0
mov hSocket, eax
.if (eax != SOCKET_ERROR)
; задаем окно для приема сообщений от сокета
; нам понадобится только FD_ACCEPT и FD_CLOSE(только для SOCK_STREAM)
; автоматически для сокета устанавливается non-block mode.
invoke WSAAsyncSelect, hSocket, hWnd, WM_SOCKET, FD_ACCEPT+FD_CLOSE
.if (eax != SOCKET_ERROR)
; заполням структуру sockaddr_in
; для начала обнуляем, поскольку заполняем не все поля
mov edi, offset sin ;структуру сокета
mov ecx, sizeof sin ;размер структуры
xor eax, eax
rep stosb
; преобразовываем номер порта в сетевой порядок байт
invoke htons, Port
mov sin.sin_port, ax ; и сохраняем его
mov sin.sin_family, AF_INET ;семейство протоколов
mov sin.sin_addr, INADDR_ANY; принимать с любого адреса
; ассоциируем локальный адрес (Port) с открытым сокетом
invoke bind, hSocket, addr sin, sizeof sin
.if (eax != SOCKET_ERROR)
; ставим сокет на прослушку порта
invoke listen, hSocket, 5 ; 5 - размер очереди на входящие запросы
.if (eax == SOCKET_ERROR)
invoke ERROR, hWnd, _T("Can't listen socket!"), TRUE
.endif
.else
invoke ERROR, hWnd, _T("Can't binding socket!"), TRUE
.endif
.else
invoke ERROR, hWnd, _T("Can't select event for socket!"), TRUE
.endif
.else
invoke ERROR, hWnd, _T("Can't create socket!"), TRUE
.endif
.elseif (uMsg == WM_CLOSE)
; подчищаем за собой и выходим
invoke closesocket, hClient ; закрываем сокеты
invoke closesocket, hSocket ; если уже закрыты, то ничего страшного :)
invoke WSACleanup ; выгружаем WinSock DLL
invoke EndDialog, hWnd, NULL
.elseif (uMsg == WM_SOCKET) ;события сокета
mov eax, lParam
mov edx, eax
shr edx, 16 ; dx - код ошибки или ноль
.if (edx != NULL) ; если dx <> NULL то получено извещение о ошибке
invoke ERROR, hWnd, _T("soket error!"), FALSE
; удаляем сокет вызвавший ошибку
invoke closesocket, wParam
.else
.if (ax == FD_ACCEPT)
; извлекаем из очереди ожидающих подключений первое и
; создаем новый сокет для связи с клиентом
invoke accept, wParam, 0, 0
mov hClient, eax
; от него мы будем получать уведомления на чтение/запись/закрытие
invoke WSAAsyncSelect, eax, hWnd, WM_SOCKET, FD_READ+FD_WRITE+FD_CLOSE
invoke MESSG, hWnd, _T("added client socket!")
.elseif (ax == FD_CLOSE)
; закрываем сокет (в wParam его хэндл)
invoke MESSG, hWnd, _T("close socket!")
invoke closesocket, wParam
.elseif (ax == FD_READ)
; получаем данные с сокета (в wParam его хэндл)
invoke recv, wParam, addr IOBuff, BUF_LEN, 0
invoke MESSG, hWnd, addr IOBuff; отправляем их в ListBox
.elseif (ax == FD_WRITE)
; извещение о готовности сокета(в wParam его хэндл) к отправке данных
mov IsSend, TRUE ; устанавливам флаг
invoke MESSG, hWnd, _T("socket is ready for writing!")
.endif
.endif ; WM_SOCKET
.else ; если сообщение не обрабатывали, то вернем NULL
xor eax, eax
ret
.endif
; сообщение обработано, возвращаем ненулевой результат
mov eax, TRUE
ret
DlgProc endp
; оконная функция контрола EDIT
; просто перехватывает нажатие Enter и отправляет данные клиенту
EditProc proc hCtl: HWND, uMsg: UINT, wParam: WPARAM, lParam: LPARAM
mov eax, wParam
.if (uMsg == WM_KEYDOWN) && (al == VK_RETURN) && (IsSend != FALSE)
; нажат Enter, отсылаем строку клиенту
invoke GetDlgItemText, hDlg, IDC_OUT, addr IOBuff, BUF_LEN
invoke send, hClient, addr IOBuff, BUF_LEN, 0
.if (eax==SOCKET_ERROR)
invoke ERROR, hDlg, _T("send failed!"), FALSE
.endif
; очищаем строку ввода
invoke SetDlgItemText, hDlg, IDC_OUT, NULL
ret
.endif
; нажата любая другая клавиша, вызываем старую функцию контрола
invoke CallWindowProc, lpfnEditProc, hCtl, uMsg, wParam, lParam
ret
EditProc endp
; просто добавляет строку в ListBox
MESSG proc hWnd: HWND, message: DWORD
invoke GetDlgItem, hWnd, IDC_IN
push eax
invoke SendMessage, eax, LB_ADDSTRING, 0, message
pop ecx
invoke SendMessage, ecx, LB_SETTOPINDEX, eax, 0
ret
MESSG endp
; информирует о последней ошибке, получая ее с помощью WSAGetLastError
; при Fatal <> FALSE сообщение выводится в MessageBox, и программа закрывается.
; при Fatal == FALSE сообщение просто выводится в ListBox
ERROR proc hWnd: HWND, message: DWORD, Fatal: BYTE
local szStr[256]: byte ; временный буфер для создания строки
; получаем код последней ошибки
invoke WSAGetLastError
; формируем строку с сообщением об ошибке
invoke wsprintf, addr szStr, _T("%s Error code: %u"), message, eax
.if (Fatal != FALSE)
invoke MessageBox, hWnd, addr szStr, _T("FATAL ERROR"), MB_OK+MB_ICONERROR
; закрываем диалог, все равно смысла нет в дальнейшей работе :)
invoke SendMessage, hWnd, WM_CLOSE, 0, 0
.else
; информируем пользователя, засылая строку в ListBox
invoke MESSG, hWnd, addr szStr
.endif
ret
ERROR endp
end start
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
ml /c /coff Server.ASM
Link /SUBSYSTEM:WINDOWS Server.obj Server.RES
pause
Клиент: главной задачей, которого, является подключение к серверу обработка пришедшего сообщения и отсылка, того, что мы написали в EDIT – controle;
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
Компилируется: .386
.model flat, stdcall
option casemap :none
include \masm32\include\windows.inc
include \masm32\include\user32.inc
include \masm32\include\kernel32.inc
include \masm32\include\wsock32.inc
includelib \masm32\lib\wsock32.lib
includelib \masm32\lib\user32.lib
includelib \masm32\lib\kernel32.lib
_T macro p1, p2 ;макрос обработки строк
local l
ifb <p2>; Строка без имени
.data
l db p1, 0
.code
exitm <addr l>
else ; Строка с именем
.data
p1 db p2, 0
.code
exitm <addr p1>
endif
endm
DlgProc proto :HWND, :UINT, :WPARAM, :LPARAM
EditProc proto :HWND, :UINT, :WPARAM, :LPARAM
MESSG proto :HWND, :DWORD
ERROR proto :HWND, :DWORD, :BYTE
Port equ 80
WM_SOCKET equ WM_USER+100
VER_SOCKET equ 101h ; требуемая версия WinSock
BUF_LEN equ 1024 ; размер буфера для приема/передачи
; контролы из ресурсов
IDC_OUT equ 101 ; строка отсылки (контрол EDIT)
IDC_IN equ 102 ; приемное "окно" (контрол ListBox)
.data?
hInstance HINSTANCE ?
hDlg HWND ? ; дескриптор окна диалога
lpfnEditProc HWND ? ; перехваченная оконная функция EDIT
hSocket dd ? ; дескриптор сокета
IsSend dd ? ; при значении >0 разрешена передача данных серверу
wsa WSADATA <> ; информация о версии WinSock
sin sockaddr_in <>
; буфер для операций ввода/вывода
IOBuff db BUF_LEN+1 dup(?)
.code
start:
invoke GetModuleHandle, NULL ;;получает дескриптор для определенного модуля если
mov hInstance, eax ;файл отображен в пространстве адреса вызова процесса и
;возвращает дескриптор окна в регистр eax
; инициализируем Windows Sockets DLL
invoke WSAStartup, VER_SOCKET, addr wsa ;версия и структура сокета
.if (eax == NULL) && (wsa.wVersion == VER_SOCKET);если всё ок
; создаем диалог
invoke DialogBoxParam, hInstance, _T("MAINDIALOG"), 0, addr DlgProc, 0
.else
invoke WSACleanup ;в обратном случае выгружаем библу
invoke MessageBox, 0, _T("Windows socket not found!"), _T("ERROR"),\ ;окно ошибки
MB_OK+MB_ICONERROR
mov eax, 90h ; заканчиваем программу с кодом ошибки
.endif
; уходим в хаОС
invoke ExitProcess, eax ;выход из процесса
; оконная функция диалога
DlgProc proc hWnd: HWND, uMsg: UINT, wParam: WPARAM, lParam: LPARAM
; определяем пришедшее сообщение окну!
.if (uMsg == WM_INITDIALOG)
; меняем заголовок окна диалога
invoke SendMessage, hWnd, WM_SETTEXT, NULL, _T("Client")
mov IsSend, FALSE ; запрещаем отправку данных клиенту
mov eax, hWnd ;дескриптор окна в регистр еах
mov hDlg, eax ; дескриптор окна нам еще понадобится :)
; сабклассируем контрол EDIT
invoke GetDlgItem, eax, IDC_OUT ;извлекает дескриптор для управления
invoke SetWindowLong, eax, GWL_WNDPROC, addr EditProc ;изменяет атрибут определенного окна.
; Функция также устанавливает 32-
; бит величины в определенной компенсации
;в дополнительную память окна
mov lpfnEditProc, eax ; сохраняем адрес предыдущей функции
; создаем сокет
invoke socket, AF_INET, SOCK_STREAM, 0
mov hSocket, eax ;запоминаем дескриптор сокета
.if (eax != SOCKET_ERROR) ; если нет ошибки, то продолжаем
; задаем окно для приема сообщений от сокета
invoke WSAAsyncSelect, hSocket, hWnd, WM_SOCKET, \ ;дескриптор сокета, окна; события сокета, "какие события"
FD_READ+FD_WRITE+FD_CONNECT+FD_CLOSE
.if (eax != SOCKET_ERROR);если всё ок
; заполням структуру sockaddr_in
; для начала обнуляем, поскольку заполняем не все поля
mov edi, offset sin ;структуру сокета
mov ecx, sizeof sin ;размер структуры
xor eax, eax
rep stosb
; преобразовываем номер порта в сетевой порядок байт
invoke htons, Port
mov sin.sin_port, ax
mov sin.sin_family, AF_INET ;семейство используемых протоколов для интернет "AF_INET"
; конвертируем строковый формат в IP-адрес
invoke inet_addr, _T("127.0.0.1")
mov sin.sin_addr, eax ;загружаем IP в параметр структуры
; подключаем созданный сокет к указанному в sin IP- адресу
invoke connect, hSocket, addr sin, sizeof sin ;дескриптор сокета, адрес и порт удалённого узла и размер структуры
.if (eax == SOCKET_ERROR) ;ошибка небеда :)
; уточняем ошибку :)
invoke WSAGetLastError
; если ошибка не равна нижеследующим, то можно продолжать :)
.if (eax != WSAEWOULDBLOCK) && (eax != WSAEINPROGRESS);ресурс временно недоступен, действие в процессе развития
invoke ERROR, hWnd, _T("Can't connect!"), TRUE ;ошибки :)
.endif
.endif
.else
invoke ERROR, hWnd, _T("Can't select event for socket!"), TRUE
.endif
.else
invoke ERROR, hWnd, _T("Can't create socket!"), TRUE
.endif
.elseif (uMsg == WM_CLOSE) ;событие закрытия ;)
; подчищаем за собой и выходим
invoke closesocket, hSocket; закрываем сокет, если он еще не закрыт
invoke WSACleanup ; выгружаем WinSock DLL
invoke EndDialog, hWnd, NULL;закрываем диалог
.elseif (uMsg == WM_SOCKET) ;а вот и узнаём какие события сокета
mov eax, lParam ; ax - код события
mov edx, eax
shr edx, 16 ; dx - код ошибки или ноль
.if (edx != NULL) ; если dx <> NULL то получено извещение о ошибке
invoke ERROR, hWnd, _T("soket error!"), FALSE
; закрываем открытый сокет
invoke closesocket, wParam
.else
.if (ax == FD_READ) ;событие чтения
; получаем данные с сокета (в wParam его дескриптор)
invoke recv, wParam, addr IOBuff, BUF_LEN,0 ;буфер, размер буфера,флаг
invoke MESSG, hWnd, addr IOBuff ; отправляем их в ListBox, hWnd - дескриптор окна, IOBuff - буфер
.elseif (ax == FD_WRITE) ;событие отправки
; извещение о готовности сокета(в wParam его дескриптор) к отправке данных
mov IsSend, TRUE ; выставляем флаг
invoke MESSG, hWnd, _T("socket is ready for writing!")
.elseif (ax == FD_CONNECT) ;событие присоединения
; извещение о успешном соединении с сервером (в wParam дескриптор сокета)
invoke MESSG, hWnd, _T("connected to server!")
.elseif (ax == FD_CLOSE) ;событие закрытия
; закрываем сокет (в wParam его дескриптор)
invoke MESSG, hWnd, _T("close server!")
invoke closesocket, wParam ;закрываем сокет
.endif
.endif ; WM_SOCKET
.else ; если сообщение не обрабатывали, то вернем NULL
xor eax, eax
ret
.endif
; сообщение обработано, возвращаем ненулевой результат
mov eax, TRUE
ret
DlgProc endp
; оконная функция контрола EDIT
; просто перехватывает нажатие Enter и отправляет данные серверу
EditProc proc hCtl: HWND, uMsg: UINT, wParam: WPARAM, lParam: LPARAM
mov eax, wParam
.if (uMsg == WM_KEYDOWN) && (al == VK_RETURN) && (IsSend != FALSE) ;если клавиша энтер и "чистый" флаг
; отсылаем строку серверу
invoke GetDlgItemText, hDlg, IDC_OUT, addr IOBuff, BUF_LEN ;дескриптор контрола EDIT его идентификатор, буфер и его размер
invoke send, hSocket, addr IOBuff, BUF_LEN, 0 ; дескриптор сокета, буфер его размер и флаг
.if (eax==SOCKET_ERROR) ;эх ошибка ;)
invoke ERROR, hDlg, _T("send failed!"), FALSE
.endif
; очищаем строку ввода
invoke SetDlgItemText, hDlg, IDC_OUT, NULL
ret
.endif
; нажата любая другая клавиша, вызываем старую функцию контрола
invoke CallWindowProc, lpfnEditProc, hCtl, uMsg, wParam, lParam
ret
EditProc endp
; просто добавляет строку в ListBox
MESSG proc hWnd: HWND, message: DWORD
invoke GetDlgItem, hWnd, IDC_IN
push eax
invoke SendMessage, eax, LB_ADDSTRING, 0, message
pop ecx
invoke SendMessage, ecx, LB_SETTOPINDEX, eax, 0
ret
MESSG endp
; информирует о последней ошибке, получая ее с помощью WSAGetLastError
; при Fatal <> FALSE сообщение выводится в MessageBox, и программа закрывается.
; при Fatal == FALSE сообщение просто выводится в ListBox
ERROR proc hWnd: HWND, message: DWORD, Fatal: BYTE
local szStr[256]: byte ; временный буфер для создания строки
; получаем код последней ошибки
invoke WSAGetLastError
; формируем строку с сообщением об ошибке
invoke wsprintf, addr szStr, _T("%s Error code: %u"), message, eax
.if (Fatal != FALSE)
invoke MessageBox, hWnd, addr szStr, _T("FATAL ERROR"), MB_OK+MB_ICONERROR
; закрываем диалог, все равно смысла нет в дальнейшей работе :)
invoke SendMessage, hWnd, WM_CLOSE, 0, 0
.else
; информируем пользователя, засылая строку в ListBox
invoke MESSG, hWnd, addr szStr
.endif
ret
ERROR endp
end start
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
ml /c /coff Client.ASM
Link /SUBSYSTEM:WINDOWS Client.obj Client.RES
pause
Ресурс диалога для «общения» с пользователем:
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
Компилируется борландским компилятором ресурсов brcc32 MAINDIALOG DIALOG 17, 30, 160, 82
STYLE DS_MODALFRAME | WS_POPUP | WS_CAPTION | WS_SYSMENU
CAPTION "MAINDIALOG"
BEGIN
CONTROL "", 101, "EDIT", ES_LEFT | ES_MULTILINE | WS_CHILD | WS_VISIBLE | \
WS_BORDER | WS_TABSTOP, 8, 64, 144, 12
LISTBOX 102, 7, 7, 145, 49, LBS_NOTIFY | WS_CHILD | WS_VISIBLE | WS_BORDER |\
WS_VSCROLL
END
<{CODE_COLLAPSE_OFF}><{CODE_WRAP_OFF}>
brcc32.exe -32 Client/Server.txt
pause
В некоторых местах не была вставлена функция обработки ошибок, только в самых необходимых. Это только пример, и в реальных прогах такое не приветствуется
Насчет recv и send. Данные функции не гарантируют передачу/прием блока данных за один раз. В данном примере это несущественно. Этот пример относится к WSAAsyncSelect Model. Есть также Select model, WSAEventSelect Model, и Overlapped Model(пожалуй самая интересная ).