socket编程，简单多线程服务端测试程序

　　前些天重温了MSDN关于socket编程的WSAStartup、WSACleanup、socket、closesocket、bind、listen、accept、recv、send等函数的介绍，今天写了一个CUI界面的测试程序（依赖MFC）作为补充。程序功能简介如下：

1：一个线程做监听用。

2：监听线程收到客户端连接后，创建新线程接收客户端数据。所有对客户端线程将加入容器，以便管理。

3：服务端打印所有客户端发来的信息。

4：服务端CUI界面输入数字0，将关闭所有连接线程，释放socket，并退出程序。

程序实现依赖类:

1：MFC

2：CSingleton模板，我关于singleton实现的文章中有源码。http://www.cnblogs.com/hgwang/p/6085922.html

3：CThreadLockCs，CRITICAL_SECTION封装类，我关于singleton实现的文章中有源码。http://www.cnblogs.com/hgwang/p/6085922.html。

4：windows socket库，我关于windows socket的文章中有介绍及调用方法，http://www.cnblogs.com/hgwang/p/6074038.html。

下面是该测试程序服务端封装类的源码：

Listen.h头文件

 1 #pragma once
 2 
 3 #ifndef WHG_LISTEN
 4 #define WHG_LISTEN
 5 
 6 #include "Singleton.h"
 7 
 8 class CListen
 9 {
10 public:
11     CListen(void);
12     ~CListen(void);
13     //类入口点，会创建监听线程
14     void SetAddress(unsigned short port,std::string ip="");
15     //用于线程访问关闭socket，并为CListen记录关闭信息
16     void DeleteLink(SOCKET s);
17 private:
18     //线程和CListen类共享的任务信息
19     struct  ThreadSocketInfo
20     {
21         CWinThread* pWt;
22         SOCKET*        pS;
23         CListen*    pListen;
24         ThreadSocketInfo()
25         {
26             pListen = NULL;
27             pWt = NULL;
28             pS = NULL;
29         }
30         ~ThreadSocketInfo()
31         {
32             pListen = NULL;
33             pWt = NULL;
34             pS = NULL;
35         }
36     };
37     //监听socket
38     SOCKET            listensocket;
39     //监听线程指针
40     CWinThread*        m_pListenThread;
41     //任务信息列表
42     std::vector<ThreadSocketInfo>    vec_WorkThreads;
43     //当前监听ip和端口
44     unsigned short    m_port;
45     std::string        m_ip; 
46     //线程互斥访问锁
47     CThreadLockCs    m_tlcs;
48  
49 private:
50     //监听线程工作对象
51     static UINT AFX_CDECL ListenThread(LPVOID);
52     //对客户端线程工作对象
53     static UINT AFX_CDECL WorkThread(LPVOID);
54     //bind、listen、accept实现
55     void Address(unsigned short port,std::string ip="");
56 };
57 
58 //申明singleton的监听线程访问对象，全局唯一实例
59 typedef CSingleton<CListen>    LISTEN;
60 
61 #endif

Listen.cpp：

  1 #include "StdAfx.h"
  2 #include "Listen.h"
  3 
  4 CListen::CListen(void)
  5 :m_pListenThread(NULL)
  6 {
  7     listensocket = 0;
  8     WSAData wsa;
  9     if (WSAStartup(MAKEWORD(1,1),&wsa) != 0)
 10     {
 11         cout<<"WSAStartup "<<endl;
 12         WSACleanup();
 13     }
 14     else
 15     {
 16         listensocket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
 17         if (listensocket == INVALID_SOCKET)
 18         {
 19             cout<<"socket() error,error code "<<WSAGetLastError()<<endl;
 20         }
 21     }
 22 }
 23 
 24 CListen::~CListen(void)
 25 {
 26     //要先关闭对客户端连接，再关闭监听socket
 27     if (vec_WorkThreads.size() > 0)
 28     {
 29         std::vector<ThreadSocketInfo>::iterator iter = vec_WorkThreads.begin();
 30         for (;iter!=vec_WorkThreads.end();iter++)
 31         {
 32             SOCKET* s = (*iter).pS;
 33             if (s)
 34             {
 35                 closesocket(*s);
 36                 delete s;
 37                 (*iter).pS = NULL;
 38             }
 39             CWinThread* pwt = (*iter).pWt;
 40             if (pwt)
 41             {
 42                 WaitForSingleObject(pwt->m_hThread,INFINITE);
 43                 delete pwt;
 44                 (*iter).pWt = NULL;
 45             }
 46         }
 47     }
 48     if (listensocket)
 49     {
 50         closesocket(listensocket);
 51     }
 52     WSACleanup();
 53     cout<<"WSACleanup "<<endl;
 54 
 55     if (m_pListenThread)
 56     {
 57         WaitForSingleObject(m_pListenThread->m_hThread,INFINITE);
 58         delete m_pListenThread;
 59     }
 60     vec_WorkThreads.clear();
 61 }
 62 
 63 //监听线程工作对象
 64 UINT CListen::ListenThread(LPVOID param)
 65 {
 66     CListen* p = (CListen*)param;
 67     if (p)
 68     {
 69         p->Address(p->m_port,p->m_ip);
 70     }
 71     return 0;
 72 }
 73 
 74 //对客户端线程工作对象
 75 UINT CListen::WorkThread(LPVOID param)
 76 {
 77     ThreadSocketInfo* tsi = (ThreadSocketInfo*)param;
 78     SOCKET acp = *(tsi->pS);
 79     CListen* pListen = tsi->pListen;
 80     delete tsi;
 81 
 82     do 
 83     {
 84         char buf[1024];
 85         int len = recv(acp,buf,1024,0);
 86         if (len == 0)
 87         {
 88             cout<<"connection has been closed "<<endl;
 89             break;
 90         }
 91         else if (len == SOCKET_ERROR)
 92         {
 93             cout<<"recv error,error code "<<WSAGetLastError()<<endl;
 94             break;
 95         }
 96         else
 97         {
 98             char* outbuf = new char[len+1];
 99             memcpy(outbuf,buf,len);
100             outbuf[len] = 0;
101             cout<<"recv data,"<<outbuf<<endl;
102             delete outbuf;
103         }
104     } while (1);
105     //删除当前连接记录
106     pListen->DeleteLink(acp);
107     return 0;
108 }
109 
110 //用于线程访问关闭socket，并为CListen记录关闭信息
111 void CListen::DeleteLink(SOCKET s)
112 {
113     m_tlcs.lock();
114     std::vector<ThreadSocketInfo>::iterator iter = vec_WorkThreads.begin();
115     for (;iter!=vec_WorkThreads.end();iter++)
116     {
117         SOCKET* ss = (*iter).pS;
118         if (ss && *ss == s)
119         {
120             closesocket(s);
121             delete ss;
122             iter->pS = NULL;
123         }
124     }
125     m_tlcs.unlock();
126 }
127 
128 //类入口点，会创建监听线程
129 void CListen::SetAddress(unsigned short port,std::string ip)
130 {
131     //监听线程只允许启动一次
132     if (m_pListenThread == NULL)
133     {
134         m_ip    = ip;
135         m_port    = port;
136         m_pListenThread = AfxBeginThread(ListenThread,this,THREAD_PRIORITY_NORMAL,0,CREATE_SUSPENDED,NULL);
137         if (m_pListenThread)
138         {
139             m_pListenThread->m_bAutoDelete = FALSE ;
140             m_pListenThread->ResumeThread();
141         }
142     }
143 }
144 
145 //bind、listen、accept实现
146 void CListen::Address(unsigned short port,std::string ip)
147 {
148     sockaddr_in service;
149     service.sin_family = AF_INET;
150     service.sin_port = htons(port);
151     if (ip.empty())
152     {
153         service.sin_addr.s_addr = INADDR_ANY;
154     }
155     else
156     {
157         service.sin_addr.s_addr = inet_addr(ip.c_str());
158     }
159 
160     if (bind(listensocket,(sockaddr*)&service,sizeof(service)) == SOCKET_ERROR)
161     {
162         cout<<"bind() error,error code "<<WSAGetLastError()<<endl;
163         return;
164     }
165     cout<<"bind "<<endl;
166 
167     if (listen(listensocket,SOMAXCONN) == SOCKET_ERROR)
168     {
169         cout<<"listen() error,error code "<<WSAGetLastError()<<endl;
170         return;
171     }
172     cout<<"listen "<<endl;
173 
174     while (1)
175     {
176         sockaddr_in recvLinkAddr;
177         int recvAddr = sizeof(recvLinkAddr);
178         SOCKET acp     = accept(listensocket,(sockaddr*)&recvLinkAddr,&recvAddr);
179         if (acp == INVALID_SOCKET)
180         {
181             cout<<"accept error,error code "<<WSAGetLastError()<<endl;
182             return;
183         }
184         cout<<"获取新的连接请求，ip:"<<inet_ntoa(recvLinkAddr.sin_addr)<<",port:"<<recvLinkAddr.sin_port<<endl;
185 
186         SOCKET* s = new SOCKET(acp);
187         ThreadSocketInfo* tsi = new ThreadSocketInfo;
188         tsi->pListen = this;
189         tsi->pS = s;
190         CWinThread* workthread = AfxBeginThread(WorkThread,tsi,THREAD_PRIORITY_NORMAL,0,CREATE_SUSPENDED,NULL);
191         if (workthread)
192         {
193             workthread->m_bAutoDelete = FALSE;
194             workthread->ResumeThread();
195             tsi->pWt = workthread;
196             ThreadSocketInfo t = *tsi;
197             m_tlcs.lock();
198             vec_WorkThreads.push_back(t);
199             m_tlcs.unlock();
200         }
201     }
202 }

客户端代码：

 1 // WinsockClient.cpp : Defines the entry point for the console application.
 2 //
 3 
 4 #include "stdafx.h"
 5 int _tmain(int argc, _TCHAR* argv[])
 6 {
 7     cout<<"input id:";
 8     std::string str;   
 9     cin>>str;
10 
11     WSAData wsa;
12     if (WSAStartup(MAKEWORD(1,1),&wsa) != 0)
13     {
14         WSACleanup();
15         return 0;
16     }
17     SOCKET cnetsocket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
18     do 
19     {
20         if (cnetsocket == INVALID_SOCKET)
21             break;
22         sockaddr_in server;
23         server.sin_family = AF_INET;
24         server.sin_port = htons(8828);
25         server.sin_addr.s_addr = inet_addr("127.0.0.1");
26         if (connect(cnetsocket,(sockaddr*)&server,sizeof(server)) == SOCKET_ERROR)
27         {
28             break;
29         }
30         str += " : windows socket test!";
31         while (1)
32         {
33             int len = send(cnetsocket,str.c_str(),str.length(),0);
34             cout<<"send data:"<<str.c_str()<<" ,length = "<<str.length()<<endl;
35             if (len < str.length())
36             {
37                 cout<<"data send uncompleted"<<endl;
38                 str = str.substr(len+1,str.length());
39                 len = send(cnetsocket,str.c_str(),str.length(),0);
40                 cout<<"send data uncomplete,send remaining data :"<<str.c_str()<<" ,length = "<<str.length()<<endl;
41             }
42             else if (len == SOCKET_ERROR)
43             {
44                 break;
45             }
46             Sleep(5000);
47         }
48     } while (0);
49     closesocket(cnetsocket);
50     WSACleanup();
51 
52     return 1;
53 }

main函数：

 1 // MultithreadServer.cpp : Defines the entry point for the console application.
 2 //
 3 
 4 #include "stdafx.h"
 5 #include "MultithreadServer.h"
 6 
 7 #include "Listen.h"
 8 #include "Singleton.h"
 9 #ifdef _DEBUG
10 #define new DEBUG_NEW
11 #endif
12 // The one and only application object
13 CWinApp theApp;
14 using namespace std;
15 
16 int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
17 {
18     int nRetCode = 0;
19 
20     // initialize MFC and print and error on failure
21     if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
22     {
23         // TODO: change error code to suit your needs
24         _tprintf(_T("Fatal Error: MFC initialization failed\n"));
25         nRetCode = 1;
26     }
27     LISTEN::Instance()->SetAddress(8828,"127.0.0.1");
28 
29     int outId;
30     cin>>outId;
31     if (outId == 0)
32     {
33         LISTEN::Close();
34     }
35     return nRetCode;
36 }