今天學習了些關于VC網絡編程中關于Winsock的知識,并做了相應的整理。在這里根據了一個聊天室例子幫助我了解這些知識的用法。
(1)Server端調用函數順序:
WSAStartup()初始化Winsock
socket()創建一個監聽Socket
bind()為監聽Socket指定通信對象
listen()設置等待連接狀態
accept()接收連接并生成會話Socket
send和recv進行對話
closesocket()關閉socket
(2)Client端調用函數順序
WSAStartup()初始化Winsock
socket()創建一個監聽Socket
connect()與Server端連接
send和recv進行對話
closesocket()關閉socket
1、程序頭部分,包括引用的頭文件、宏定義、全局變量和函數聲明
#include "stdafx.h"
#include "resource.h"
#include <VECTOR> //采用STL
#include <algorithm>
#include <WINSOCK2.H> //調用winsock
#include <ASSERT.H>

#define BUFFER_SIZE 4096 //緩沖區的大小
#define SERVER_MESSAGE WM_USER+100 //服務器SOCKET異步消息ID
#define CLIENT_MESSAGE WM_USER+101 //客戶機SOCKET異步消息ID

// Global Variables:
SOCKET g_ListenSocket; //Listen socket (for server)
std::vector<SOCKET> g_DataSockets; //All data sockets with clients (for server)
SOCKET g_ClientSocket; //Client data socket (for client)
BOOL g_bActive; //A tag of active socket (for both server and client)
std::string g_ChatWords;
BOOL g_bClient; //TRUE as a client, FALSE as a server

BOOL ServerInit(HWND hWnd,UINT port);
BOOL ClientInit(HWND hWnd,UINT port,const char* serverIP);
int OnClientMessage(HWND hWnd, WPARAM wParam, LPARAM lParam);
int OnServerMessage(HWND hWnd, WPARAM wParam, LPARAM lParam);
void SendMessageToPeer(HWND hWnd);
void Send(SOCKET sock, const char* buffer, int length);
void RefreshScreen(HWND hWnd);
void ExitChat(HWND hWnd);
void GetErrorReason();
2、Win32程序函數入口點WinMain:
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
int APIENTRY WinMain(HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPSTR lpCmdLine,
int nCmdShow)


{
// Perform application initialization:
g_bActive=FALSE;
DialogBox(hInstance, MAKEINTRESOURCE(IDD_CHAT_DIALOG), NULL, (DLGPROC)WndProc);
return 0;
}
3、對話框的消息循環函數WndProc:
在WndProc消息循環中,只處理4中消息:對話框初始化、對話框控件、服務器SOCKET異步和客戶端SOCKET異步,在對話框初始化中,程序啟動了SOCKET棧。在對話框控件中,程序分別響應了各種按鈕消息以及文字輸入框消息EN_UPDATE。在服務器SOCKET異步消息中,調用OnServerMessage處理。在客戶端SOCKET異步消息中,調用了OnclientMessage處理。
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)


{
HWND subWnd;
char buffer[BUFFER_SIZE];
UINT port;
char serverIP[16];
switch (message)

{
case WM_INITDIALOG:

{
WSADATA wsaData;
if(WSAStartup(MAKEWORD(2,2),&wsaData)!=0)

{
MessageBox(hWnd,"Socket Stack Error!","Error",IDOK);
return -1;
}
}
return TRUE;
case WM_COMMAND:

{
switch(LOWORD(wParam))

{
case IDOK:
case IDCANCEL:
EndDialog(hWnd,LOWORD(wParam));
break;
case IDC_CONNECT:

{
subWnd=GetDlgItem(hWnd,IDC_LISTEN_PORT_C);
GetWindowText(subWnd,buffer, BUFFER_SIZE);
port= UINT(atoi(buffer));
subWnd=GetDlgItem(hWnd,IDC_SERVERIP);
GetWindowText(subWnd,buffer,BUFFER_SIZE);
strncpy(serverIP,buffer,16);
ClientInit(hWnd,port,serverIP);
//g_ChatWords="";
}
break;
case IDC_LISTEN:

{
subWnd=GetDlgItem(hWnd,IDC_LISTEN_PORT_S);
GetWindowText(subWnd,buffer, BUFFER_SIZE);
port= UINT(atoi(buffer));
ServerInit(hWnd,port);
//g_ChatWords="";
}
break;
case IDC_INPUTTEXT:

{
if(HIWORD(wParam)==EN_UPDATE)

{
SendMessageToPeer(hWnd);
}
}
default:
break;
}
}
break;
case SERVER_MESSAGE:

{
OnServerMessage(hWnd,wParam, lParam);
}
break;
case CLIENT_MESSAGE:

{
OnClientMessage(hWnd, wParam, lParam);
}
break;
}
return 0;
}

1)int WSAStartup(WORD wVersionRequested,LPWSADATA lpWSAData);
wVersionRequested:表示欲使用的Windows Sockets API版本,使用宏MAKEWORD(X,Y)設置參數。表示版本為X.Y(本例中為2.2)
4、客戶端初始化函數ClientInit:
當單擊“連接服務器“按鈕時,程序分別從子窗口IDC_LISTEN_POST_C和IDC_SERVERIP中提取服務器的IP和監聽端口,然后調用客戶端函數ClientInit。
BOOL ClientInit(HWND hWnd,UINT port,const char* serverIP)


{
sockaddr_in addr;
ExitChat(hWnd);
g_bClient=TRUE;
if(inet_addr(serverIP)==INADDR_NONE)

{
MessageBox(hWnd,"Invalid IP Address!","Warnning",IDOK);
return FALSE;
}

//create tcp/stream data socket
g_ClientSocket = socket(AF_INET, SOCK_STREAM,0);
assert(g_ClientSocket != INVALID_SOCKET);

//set the socket as async selection tag, the related message is CLIENT_MESSAGE
//the async events contain: read, write, close, connect
WSAAsyncSelect(g_ClientSocket,hWnd,CLIENT_MESSAGE,FD_READ|FD_WRITE|FD_CLOSE|FD_CONNECT);

//set the server address and port
addr.sin_family = AF_INET;
addr.sin_addr.S_un.S_addr = inet_addr(serverIP);
addr.sin_port = htons(port); //short from host to network format
//connect to server
connect(g_ClientSocket, (sockaddr*)&addr, sizeof(addr));

g_ChatWords+="Connecting
\r\n";
RefreshScreen(hWnd);
return TRUE;
}

1)SOCKET socket(int af,int type,int protocol);
af:指協議的地址族,如果想建立一個UDP或TCP的套接字,則該值設置成AF_INET,表示在網絡層采用網際協議(IP)
type:協議的套接字類型,可以是SOCK_STREAM、SOCK_DGRAM 、SOCK_RAW、SOCK_RDM、SOCK_SEQPACKET。 當采用流連接方式時用SOCK_STREAM,采用數據報文方式用SOCK_DGRAM
proctocol:協議字段,當af和type都指定之后,該值的取值范圍就被去定了,默認值為0。
2)int PASCAL FAR WSAAsyncSelect ( SOCKET s, HWND hWnd, unsigned int wMsg, long lEvent );
S:標識一個需要事件通知的套接口的描述符.
hWnd:標識一個在網絡事件發生時需要接收消息的窗口句柄.
wMsg :在網絡事件發生時要接收的消息.
lEvent:位屏蔽碼,用于指明應用程序感興趣的網絡事件集合.
3)int connect(SOCKET s,const struct sockaddr FAR * name,int namelen);
S是建立好的套接字,name指向描述通信對象地址信息的結構體的指針,namelen是該結構體的長度。
4)struct sockaddr_in {short sin_family;u_short sin_port;struct in_addr sin_addr;char sin_zero[8];};
5)unsigned long PASCAL FAR inet_addr( const struct FAR* cp);
cp:一個以Internet標準“.”間隔的字符串。
本函數解釋cp參數中的字符串,這個字符串用Internet的“.”間隔格式表示一個數字的Internet地址。
6)u_short PASCAL FAR htons( u_short hostshort);
hostshort:主機字節順序表達的16位數。
本函數將一個16位數從主機字節順序轉換成網絡字節順序。
其中sin_family是指地址族;sin_port是制定的端口號;sin_addr是制定的IP地址;sin_zero冗余字段,使sockaddr_in結構和SOCKADDR結構大小相同
5、服務器初始化函數ServerInit:
單擊“建立服務器“按鈕時,程序從子窗口IDC_LISTEN_PORT_S中提取監聽端口,然后調用服務器初始化函數ServerInit。
BOOL ServerInit(HWND hWnd,UINT port)


{
sockaddr_in addr;
char buffer[BUFFER_SIZE];
ExitChat(hWnd);
g_bClient=FALSE;
//create a tcp/stream based socket
g_ListenSocket = socket(AF_INET, SOCK_STREAM,0);
assert(g_ListenSocket != INVALID_SOCKET);

//set the socket as async selection tag, the related message is SERVER_MESSAGE
//the async events contain: accept, read, write, close
WSAAsyncSelect(g_ListenSocket, hWnd, SERVER_MESSAGE, FD_ACCEPT|FD_READ|FD_WRITE|FD_CLOSE);

addr.sin_family = AF_INET; //IP layer takes IP router protocol
addr.sin_addr.S_un.S_addr = INADDR_ANY;
addr.sin_port = htons(port);//port from short of host to short of network

//binding listening socket
if(bind(g_ListenSocket, (sockaddr*)&addr, sizeof(sockaddr)) == SOCKET_ERROR)

{
sprintf(buffer,"Port %d has been taken. Change another port.\r\n",port);
g_ChatWords+=buffer;
RefreshScreen(hWnd);
MessageBox(hWnd,"Binding Error","Warnning",IDOK);
return FALSE;
}
//listening
if(listen(g_ListenSocket, 5) == SOCKET_ERROR)

{
strcpy(buffer,"Listen error.\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
MessageBox(hWnd,"Listen Error.","Warnning",IDOK);
return FALSE;
}

g_ChatWords+="Server is listening
\r\n";
RefreshScreen(hWnd);
g_bActive=TRUE;
return TRUE;
}

1) int bind(SOCKET s,const struct sockaddr FAR * name,int namelen);
S是建立好的套接字,name指向描述通信對象地址信息的結構體的指針,namelen是該結構體的長度。
2)int listen(SOCKET s,int backlog);
S是建立好的套接字,backlog是并發連接等待隊列的長度。
6、SendMessageToPeer函數:
當用戶在文本框中輸入文字時,程序會接收到該文本框的EN_UPDATE消息,函數SendMessageToPeer負責處理該消息。在SendMessageToPeer中,一旦檢測到用戶按下了回車鍵,程序就將文本內容刷新到上面的顯示框中,如果當前在線,文本內容還將發送到會話的另一端。
void SendMessageToPeer(HWND hWnd)


{
HWND subwnd;
char buffer[BUFFER_SIZE];
int i;
static int oldNumOfChars=0;

subwnd=GetDlgItem(hWnd,IDC_INPUTTEXT);
GetWindowText(subwnd,buffer,BUFFER_SIZE-2);
if(oldNumOfChars!=strlen(buffer))

{
oldNumOfChars=strlen(buffer);
return;
}
//CString s;
//s("%d還是%d",oldNumOfChars,strlen(buffer));
// g_ChatWords+=s;
//empty content of input edit box
SetWindowText(subwnd,"");
oldNumOfChars=0;
if(!g_bActive)

{
g_ChatWords+=buffer;
g_ChatWords.erase(g_ChatWords.size(),1);
g_ChatWords+="(Hint: you are isolated now.)\r\n";
RefreshScreen(hWnd);
return;
}
strcat(buffer,"\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
if(g_bClient)

{
Send(g_ClientSocket, buffer, strlen(buffer));
}
else

{
for(i=0;i<g_DataSockets.size();i++)
Send(g_DataSockets[i],buffer,strlen(buffer));
}
}
void Send(SOCKET sock, const char* buffer, int length)


{
int ret,i;
i=0;
while(length>0)

{
ret=send(sock,&(buffer[i]),length,0);
if(ret==0)
break;
else if(ret == SOCKET_ERROR)

{
g_ChatWords+="Error sending.\r\n";
break;
}
length-=ret;
i+=ret;
}
}
1)int send(SOCKET s,const char FAR * buf,int len,int flags);
參數s是建立連接的套接字。buf和len是發送的數據包及其長度。flags一般取0,還有MSG_DONTROUTE要求套接字傳輸的數據不要路由,如果傳輸協議不支持該選項,則這個要求被忽略。MSG_OOB表示套接字此時傳輸的數據室外帶數據,需要緊急處理。
7、OnClientMessage函數:
在主循環的消息循環中,OnClientMessage函數用來處理客戶端SOCKET的一部消息CLIENT_MESSAGE。需要指出的是:當某個網絡消息到達窗口時,可能暗含著套接字上發生了某種網絡錯誤。此時,應該先判斷套接字上是否發生了錯誤,如果有錯誤發生就直接關閉套接字然后返回,只有在沒有錯誤發生的情況下,才需要處理具體的網絡消息。在實現上,需要利用消息參數lParam的高字節,宏WSAGETSELECTERROR可以檢測錯誤。而具體的消息類型是存儲在lParam的低字節,宏WSAGETSELECTEVENT用來獲取該低字節值。
int OnClientMessage(HWND hWnd, WPARAM wParam, LPARAM lParam)


{
char buffer[BUFFER_SIZE+1];
int retCode;
sockaddr_in name;
int namelen=sizeof(sockaddr_in);

if(WSAGETSELECTERROR(lParam))

{
closesocket(g_ClientSocket);
strcpy(buffer,"Server has no response.\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
g_bActive=FALSE;
return 0;
}

switch(WSAGETSELECTEVENT(lParam))

{
case FD_CONNECT:
getpeername(g_ClientSocket,(sockaddr*)&name,&namelen);
sprintf(buffer,"Successfully connected to %s:%d.\r\n",inet_ntoa(name.sin_addr),ntohs(name.sin_port));
g_ChatWords+=buffer;
RefreshScreen(hWnd);
g_bActive=TRUE;
break;
case FD_READ:
retCode=recv(g_ClientSocket,buffer,BUFFER_SIZE,0);
if(retCode!=SOCKET_ERROR)

{
buffer[retCode]=NULL;
g_ChatWords+=buffer;
}
else
GetErrorReason();
RefreshScreen(hWnd);
break;
case FD_WRITE:
break;
case FD_CLOSE:
closesocket(g_ClientSocket);
strcpy(buffer,"Server close session.Successfully log out.\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
g_bActive=FALSE;
break;
}
return 0;
}
1)char FAR* PASCAL FAR inet_ntoa( struct in_addr in);
in:一個表示Internet主機地址的結構。
本函數將一個用in參數所表示的Internet地址結構轉換成以“.” 間隔的諸如“a.b.c.d”的字符串形式。
2) int recv(SOCKET s,char FAR * buf,int len,int flags);
參數與send一致
3)int closesocket(SOCKET s);
關閉套接字
8、OnServerMessage函數:
在主程序的消息循環中,OnServerMessage函數用來處理客戶端SOCKET的一部消息SERVER_MESSAGE。在服務器初始化函數中的g_ListenSocket被設置成異步監聽socket,所有被g_ListenSocket接收的數據socket將同時被設置成異步socket,并共享同一個消息SERVER_MESSAGE。及時服務器關閉了監聽socket,會話socket的連接依然有效,除非服務器同時也將會話socket關閉。
int OnServerMessage(HWND hWnd, WPARAM wParam, LPARAM lParam)


{
SOCKET socket;
int i,retCode,namelen;
char buffer[BUFFER_SIZE+1];
std::vector<SOCKET>::iterator ite;
sockaddr_in name;
namelen=sizeof(name);

if(WSAGETSELECTERROR(lParam))

{
getpeername((SOCKET)wParam,(sockaddr*)&name,&namelen);
closesocket((SOCKET)wParam);
//erase the client socket from client socks list
ite=std::find(g_DataSockets.begin(), g_DataSockets.end(),(SOCKET)wParam);
assert(ite!=g_DataSockets.end());
g_DataSockets.erase(ite);
//refresh screen
sprintf(buffer, "Client %s:%d lost contact with us.\r\n", inet_ntoa(name.sin_addr),ntohs(name.sin_port));
for(i=0; i<g_DataSockets.size(); i++)
Send(g_DataSockets[i],buffer,strlen(buffer));
g_ChatWords+=buffer;
RefreshScreen(hWnd);
return 0;
}

switch(WSAGETSELECTEVENT(lParam))

{
case FD_ACCEPT:
//accept the client request
socket= accept(g_ListenSocket,NULL,NULL);
//in fact we can get the peer when calling accept, here we use getpeername instead
getpeername(socket,(sockaddr*)&name,&namelen);
//send a message of logining to other clients
sprintf(buffer,"A guest joins us.(%s:%d)\r\n",inet_ntoa(name.sin_addr),ntohs(name.sin_port));
g_ChatWords+=buffer;
for(i=0;i<g_DataSockets.size();i++)
Send(g_DataSockets[i],buffer,strlen(buffer));
//send a welcome message to current client
sprintf(buffer, "Welcome !(You ID is: %s:%d.)\r\n",inet_ntoa(name.sin_addr),ntohs(name.sin_port));
//send(socket,buffer,strlen(buffer),0);
Send(socket,buffer,strlen(buffer));
//refresh srceen text
RefreshScreen(hWnd);
//push the client socket down to the client sockets list
g_DataSockets.push_back(socket);
break;
case FD_READ:

{
//get the client message, the client socket ID is wParam
retCode=recv((SOCKET)wParam,buffer,BUFFER_SIZE,0);
buffer[retCode]=NULL;
//send the received message to other clients
for(i=0;i<g_DataSockets.size();i++)

{
if(wParam!=g_DataSockets[i])
Send(g_DataSockets[i],buffer,strlen(buffer));
}
g_ChatWords+=buffer;
//refresh screen
RefreshScreen(hWnd);
}
break;
case FD_WRITE:
return 0;
case FD_CLOSE: //Client gracefully close the socket
//close the client socket who has left, the client socket ID is wParam
getpeername((SOCKET)wParam,(sockaddr*)&name,&namelen);
closesocket((SOCKET)wParam);
//erase the client socket from client socks list
ite=std::find(g_DataSockets.begin(), g_DataSockets.end(),(SOCKET)wParam);
assert(ite!=g_DataSockets.end());
g_DataSockets.erase(ite);
//refresh screen
sprintf(buffer, "Client %s:%d left.\r\n", inet_ntoa(name.sin_addr),ntohs(name.sin_port));
for(i=0; i<g_DataSockets.size(); i++)
Send(g_DataSockets[i],buffer,strlen(buffer));
g_ChatWords+=buffer;
RefreshScreen(hWnd);
break;
}
return 0;
}
1)int PASCAL FAR getpeername( SOCKET s, struct sockaddr FAR* name,int FAR* namelen);
s:標識一已連接套接口的描述字。
name:接收端地址的名字結構。
namelen:一個指向名字結構的指針。
2)u_short PASCAL FAR ntohs( u_short netshort);
netshort:一個以網絡字節順序表達的16位數。
本函數將一個16位數由網絡字節順序轉換為主機字節順序。
3) SOCKET accept(SOCKET s,struct sockaddr FAR * addr,int FAR * addrlen);
如果是服務器并不關心對方的地址,則可將accept函數中后兩個參數設置為NULL。
9、最后,介紹3個輔助函數:ExitChat、GetErrorReason和RefreshScreen
當新建服務器或鏈接服務器的時候,首先需要將上一次會話關閉,ExitChat函數完成此功能。
RefreshScreen函數負責刷新顯示內容并滾動顯示文本框。
GetErrorReason函數負責獲得當前操作中的具體錯誤原因。
void ExitChat(HWND hWnd)


{
int i;
char buffer[BUFFER_SIZE];
if(g_bActive)

{
if(g_bClient)

{
closesocket(g_ClientSocket);
strcpy(buffer,"Successfully log out.\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
}
else

{
strcpy(buffer,"I will leave. Pls clients log out ASAP.\r\n");
for(i=0; i<g_DataSockets.size(); i++)

{
Send(g_DataSockets[i],buffer,strlen(buffer));
closesocket(g_DataSockets[i]);
}
g_DataSockets.clear();
closesocket(g_ListenSocket);
strcpy(buffer,"Successfully close server.\r\n");
g_ChatWords+=buffer;
RefreshScreen(hWnd);
}
}
g_bActive=FALSE;
}

void GetErrorReason()


{
LPVOID lpMsgBuf;
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,WSAGetLastError(),MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &lpMsgBuf,0,NULL);
g_ChatWords+=(char *)lpMsgBuf;
LocalFree( lpMsgBuf );
}
void RefreshScreen(HWND hWnd)


{
HWND subWnd;
std::string::size_type pos;
int n;
subWnd=GetDlgItem(hWnd,IDC_SHOWTEXT);
SetWindowText(subWnd, g_ChatWords.c_str());
n=0;
pos=0;
while((pos=g_ChatWords.find('\n',pos))!=std::string::npos)

{
pos++;
n++;
}
SendMessage(subWnd, EM_LINESCROLL, 0, n);
}

posted on 2009-06-21 00:23
The_Moment 閱讀(1925)
評論(0) 編輯 收藏 引用 所屬分類:
VC理論