socket服务器端代码实例

 

下面就是源代码，其中部分很基本的代码我就省略掉了，编译平台为
Win2000 Server with SP2 + VC6.0 with SP5

#include <windows.h>
#include <winsock2.h>

#define PORT 5150
#define DATA_BUFSIZE 8192

typedef struct _SOCKET_INFORMATION {
　BOOL RecvPosted;
　CHAR Buffer[DATA_BUFSIZE];
　WSABUF DataBuf;
　SOCKET Socket;
　DWORD BytesSEND;
　DWORD BytesRECV;
　_SOCKET_INFORMATION *Next;
} SOCKET_INFORMATION, * LPSOCKET_INFORMATION;

#define WM_SOCKET (WM_USER + 1)

void CreateSocketInformation(SOCKET s, HWND);
LPSOCKET_INFORMATION GetSocketInformation(SOCKET s);
void FreeSocketInformation(SOCKET s);

LPSOCKET_INFORMATION SocketInfoList;

LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM) ;

int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
　　　　LPSTR lpCmdLine, int nCmdShow)
{
　DWORD Ret;
　SOCKET Listen;
　SOCKADDR_IN InternetAddr;
　WSADATA wsaData;
　static TCHAR szAppName[] = TEXT ("HelloWin") ;
　HWND hwnd ;
　MSG msg ;
　WNDCLASS wndclass ;

　// Prepare echo server
　wndclass.style = CS_HREDRAW | CS_VREDRAW ;
　...
　...
　RegisterClass (&wndclass);
　hwnd = CreateWindow (...) ; // creation parameters
　ShowWindow (hwnd, nCmdShow) ;
　UpdateWindow (hwnd) ;

　if ((Ret = WSAStartup(0x0202, &wsaData)) != 0)
　{
　　MessageBox(hwnd, TEXT("Start socket failed"), TEXT("error"), MB_OK);
　　ExitProcess(1);
　}
　if ((Listen = socket (PF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET)
　{
　　MessageBox(hwnd, TEXT("socket() failed"), TEXT("error"), MB_OK);
　　ExitProcess(1);
　}
　WSAAsyncSelect(Listen, hwnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE);
　InternetAddr.sin_family = AF_INET;
　InternetAddr.sin_addr.s_addr = htonl(INADDR_ANY);
　InternetAddr.sin_port = htons(PORT);

　if (bind(Listen, (PSOCKADDR) &InternetAddr, sizeof(InternetAddr))
　　　　== SOCKET_ERROR)
　{
　　MessageBox(hwnd, TEXT("bind() failed"), TEXT("error"), MB_OK);
　　ExitProcess(1);
　}
　if (listen(Listen, 5))
　{
　　MessageBox(hwnd, TEXT("listen() failed"), TEXT("error"), MB_OK);
　　ExitProcess(1);
　}
　// Translate and dispatch window messages for the application thread
　while (GetMessage (&msg, NULL, 0, 0))
　{
　　TranslateMessage (&msg) ;
　　DispatchMessage (&msg) ;
　}
　return msg.wParam ;
}

LRESULT CALLBACK WndProc (HWND hwnd, UINT message,
　　　　　　　　WPARAM wParam, LPARAM lParam)
{
　HDC hdc ;
　PAINTSTRUCT ps ;
　RECT rect ;
　SOCKET Accept;
　LPSOCKET_INFORMATION SocketInfo;
　DWORD RecvBytes, SendBytes;
　DWORD Flags;

　switch (message)
　{
　　case WM_CREATE:
　　　return 0 ;

　　case WM_PAINT:
　　　hdc = BeginPaint (hwnd, &ps) ;
　　　GetClientRect (hwnd, &rect) ;
　　　DrawText (hdc, TEXT ("Server Started!"), -1, &rect,
　　　DT_SINGLELINE | DT_CENTER | DT_VCENTER) ;
　　　EndPaint (hwnd, &ps) ;
　　　return 0 ;

　　case WM_DESTROY:
　　　PostQuitMessage (0) ;
　　　return 0 ;

　　case WM_SOCKET:
　　　if (WSAGETSELECTERROR(lParam))
　　　{
　　　　MessageBox(...);
　　　　FreeSocketInformation(wParam);
　　　}
　　　else
　　　{
　　　　switch(WSAGETSELECTEVENT(lParam))
　　　　{
　　　　　case FD_ACCEPT:
　　　　　　if ((Accept = accept(wParam, NULL, NULL)) == INVALID_SOCKET)
　　　　　　{
　　　　　　　MessageBox(...);
　　　　　　　break;
　　　　　　}
　　　　　　// Create a socket information structure to associate with the
　　　　　　// socket for processing I/O.
　　　　　　CreateSocketInformation(Accept, hwnd);
　　　　　　WSAAsyncSelect(Accept, hwnd, WM_SOCKET,
　　　　　　　　　FD_READ|FD_WRITE|FD_CLOSE);
　　　　　　break;

　　　　　case FD_READ:
　　　　　　SocketInfo = GetSocketInformation(wParam);
　　　　　　// Read data only if the receive buffer is empty.
　　　　　　if (SocketInfo->BytesRECV != 0)
　　　　　　{
　　　　　　　SocketInfo->RecvPosted = TRUE;
　　　　　　　return 0;
　　　　　　}
　　　　　　else
　　　　　　{
　　　　　　　SocketInfo->DataBuf.buf = SocketInfo->Buffer;
　　　　　　　SocketInfo->DataBuf.len = DATA_BUFSIZE;
　　　　　　　Flags = 0;
　　　　　　　if (WSARecv(SocketInfo->Socket, &(SocketInfo->DataBuf),
　　　　　　　　　　　1, &RecvBytes, &Flags, NULL, NULL) == SOCKET_ERROR)
　　　　　　　{
　　　　　　　　if (WSAGetLastError() != WSAEWOULDBLOCK)
　　　　　　　　{
　　　　　　　　　MessageBox(...);
　　　　　　　　　FreeSocketInformation(wParam);
　　　　　　　　　return 0;
　　　　　　　　}
　　　　　　　}
　　　　　　　else // No error so update the byte count
　　　　　　　　SocketInfo->BytesRECV = RecvBytes;
　　　　　　}
　　　　　　// DO NOT BREAK HERE SINCE WE GOT A SUCCESSFUL RECV.
　　　　　　// Go ahead and begin writing data to the client.

　　　　　　case FD_WRITE:
　　　　　　　SocketInfo = GetSocketInformation(wParam);
　　　　　　　if (SocketInfo->BytesRECV > SocketInfo->BytesSEND)
　　　　　　　{
　　　　　　　　SocketInfo->DataBuf.buf =
　　　　　　　　　SocketInfo->Buffer + SocketInfo->BytesSEND;
　　　　　　　　SocketInfo->DataBuf.len =
　　　　　　　　　SocketInfo->BytesRECV - SocketInfo->BytesSEND;
　　　　　　　　if (WSASend(SocketInfo->Socket, &(SocketInfo->DataBuf),
　　　　　　　　　　　　1, &SendBytes, 0,NULL, NULL) == SOCKET_ERROR)
　　　　　　　　{
　　　　　　　　　if (WSAGetLastError() != WSAEWOULDBLOCK)
　　　　　　　　　{
　　　　　　　　　　MessageBox(...);
　　　　　　　　　　FreeSocketInformation(wParam);
　　　　　　　　　　return 0;
　　　　　　　　　}
　　　　　　　　}
　　　　　　　　else // No error so update the byte count
　　　　　　　　　SocketInfo->BytesSEND += SendBytes;
　　　　　　　}

　　　　　　　if (SocketInfo->BytesSEND == SocketInfo->BytesRECV)
　　　　　　　{
　　　　　　　　SocketInfo->BytesSEND = 0;
　　　　　　　　SocketInfo->BytesRECV = 0;
　　　　　　　　// If a RECV occurred during our SENDs then we need to post
　　　　　　　　// an FD_READ notification on the socket.

　　　　　　　　if (SocketInfo->RecvPosted == TRUE)
　　　　　　　　{
　　　　　　　　　SocketInfo->RecvPosted = FALSE;
　　　　　　　　　PostMessage(hwnd, WM_SOCKET, wParam, FD_READ);
　　　　　　　　}
　　　　　　　}
　　　　　　　if(SocketInfo->DataBuf.buf != NULL)
　　　　　　　　MessageBox(hwnd, SocketInfo->DataBuf.buf,
　　　　　　　　　　 TEXT("Received"), MB_OK);
　　　　　　　break;

　　　　　　case FD_CLOSE:
　　　　　　　FreeSocketInformation(wParam);
　　　　　　　break;
　　　　}
　　　}
　　　return 0;
　}
　return DefWindowProc(hwnd, message, wParam, lParam);
}

void CreateSocketInformation(SOCKET s, HWND hwnd)
{
　LPSOCKET_INFORMATION SI;
　if ((SI = (LPSOCKET_INFORMATION) GlobalAlloc(GPTR,
　sizeof(SOCKET_INFORMATION))) == NULL)
　{
　　MessageBox(...);
　　return;
　}
　// Prepare SocketInfo structure for use.
　SI->Socket = s;
　SI->RecvPosted = FALSE;
　SI->BytesSEND = 0;
　SI->BytesRECV = 0;
　SI->Next = SocketInfoList;
　SocketInfoList = SI;
}

LPSOCKET_INFORMATION GetSocketInformation(SOCKET s)
{
　SOCKET_INFORMATION *SI = SocketInfoList;
　while(SI)
　{
　　if (SI->Socket == s)
　　return SI;
　　SI = SI->Next;
　}
　return NULL;
}

void FreeSocketInformation(SOCKET s)
{
　SOCKET_INFORMATION *SI = SocketInfoList;
　SOCKET_INFORMATION *PrevSI = NULL;
　while(SI)
　{
　　if (SI->Socket == s)
　　{
　　　if (PrevSI)
　　　　PrevSI->Next = SI->Next;
　　　else
　　　　SocketInfoList = SI->Next;
　　　closesocket(SI->Socket);
　　　GlobalFree(SI);
　　　return;
　　}
　　PrevSI = SI;
　　SI = SI->Next;
　}
}

服务器就这样建好了，只需要一个客户机就可以通信了，具体的代码我就不再贴了，各位可
以自己设计客户机，或者去下载区下载源程序。最后向大家推荐《Windows网络编程技术》,
真的不错。

本文中部分内容翻译自MSDN，客户机程序使用的是《Windows网络编程技术》中的例子