C++编写Socket网络程序

keehoo

//server.cpp 服务器端

1. #include   <winsock2.h>
   
2. #include   <io.h>
   
3. #include   <windows.h>
   
4. #include   <iostream>
   
5. #include   <string>
   
6. #pragma   comment(lib, "WS2_32.lib ")
   
7. using   namespace   std;
   
8. 
   
9. void   OutErr(string   msg)
   
10. {
    
11. cout << "错误！发生在 " <<msg <<endl;
    
12. }
    
13. 
    
14. fd_set  fdread;
    
15. int  ret;
    
16. 
    
17. DWORD  WINAPI  ThreadProc(LPVOID   lpParameter)
    
18. {
    
19. SOCKET c=(SOCKET)lpParameter;
    
20. char   data[80];
    
21. UINT   ReciveBytes=0;
    
22. FD_ZERO(&fdread);
    
23. FD_SET(c,&fdread);
    
24. do
    
25. {
    
26.   if((ret=select(0,&fdread,NULL,NULL,NULL))==SOCKET_ERROR)
    
27.   {
    
28.    OutErr(string( "select "));
    
29.    break;
    
30.   }
    
31.   
    
32.   if(ret> 0)
    
33.   {
    
34.    if(FD_ISSET(c,&fdread))
    
35.    {
    
36.     ReciveBytes=recv(c,data,10,0);
    
37.     if(ReciveBytes==SOCKET_ERROR)
    
38.     {
    
39.      break;
    
40.     }
    
41.     cout << "收到字节数： "
    
42.      <<ReciveBytes
    
43.      <<endl;
    
44.     data[ReciveBytes]= '\0 ';
    
45.     cout <<data <<endl;
    
46.    }
    
47.   }
    
48. }while(ReciveBytes> 0);
    
49. FD_CLR(c,&fdread);
    
50. closesocket(c);
    
51. return   0;
    
52. }
    
53. 
    
54. int   main()
    
55. {
    
56. SOCKET   server;
    
57. SOCKET   client;
    
58. SOCKADDR_IN   ser_addr,cli_addr;
    
59. WSADATA   wdata;
    
60. int   cliaddrLen=0;
    
61. UINT   port=4000;
    
62. HANDLE       Hnd;
    
63. DWORD         dwThreadID;
    
64. if(WSAStartup(MAKEWORD(2,0),&wdata)!=0)
    
65. {
    
66.   OutErr(string( "WSAStartup "));
    
67.   return   1;
    
68. }
    
69. 
    
70. ser_addr.sin_addr.S_un.S_addr=inet_addr( "0.0.0.0 ");
    
71. ser_addr.sin_family=AF_INET;
    
72. ser_addr.sin_port=htons(port);
    
73. 
    
74. //if((server=WSASocket(AF_INET,SOCK_STREAM,IPPROTO_IP,NULL,0,1))==INVALID_SOCKET)
    
75. if((server=socket(AF_INET,SOCK_STREAM,IPPROTO_IP))==INVALID_SOCKET)
    
76. {
    
77.   OutErr(string( "WSASocket "));
    
78.   return   1;
    
79. }
    
80. 
    
81. if(bind(server,(SOCKADDR   *)&ser_addr,sizeof(SOCKADDR))!=0)
    
82. {
    
83.   OutErr(string( "bind "));
    
84.   return   1;
    
85. }
    
86. 
    
87. if(listen(server,5)!=0)
    
88. {
    
89.   OutErr(string( "listen "));
    
90.   return   1;
    
91. }
    
92. 
    
93. while(1)
    
94. {
    
95.   cliaddrLen=sizeof(SOCKADDR);
    
96.   client=accept(server,(SOCKADDR   *)&cli_addr,&cliaddrLen);
    
97.   cout << "客户IP： " <<inet_ntoa(cli_addr.sin_addr)
    
98.    << "，端口: " <<ntohs(cli_addr.sin_port) <<endl;
    
99.   
    
100.   ULONG   ul=1;
     
101.   if(ioctlsocket(client,FIONBIO,&ul)==SOCKET_ERROR)
     
102.   {
     
103.    OutErr(string( "ioctlsocket "));
     
104.    return   1;
     
105.   }
     
106.   FD_ZERO(&fdread);//清空集合
     
107.   FD_SET(client,&fdread);//将C加入到读取集合中
     
108.   
     
109.   //Hnd=CreateThread(NULL,0,ThreadProc,(LPVOID)client,0,&dwThreadID);
     
110.         //接收数据
     
111.   while(TRUE)
     
112.   {
     
113.    char   data[11];
     
114.    int     ReciveBytes;
     
115.    //这里阻断程序，知道缓冲区内有可读数据才返回
     
116.    if((ret=select(0,&fdread,NULL,NULL,NULL))==SOCKET_ERROR)
     
117.    {
     
118.                 //套接字错误
     
119.                 break;
     
120.    }
     
121.    
     
122.    if(ret> 0)
     
123.    {
     
124.                 if(FD_ISSET(client,&fdread))//c中是否属于fdread集合                        
     
125.                 {      
     
126.      do//循环读取，每次接收10个字符，直到把缓冲区中全部取完
     
127.      {
     
128.       ReciveBytes=recv(client,data,10,0);
     
129.       if(ReciveBytes==SOCKET_ERROR)
     
130.       {
     
131.        //SOCKET错误
     
132.        break;  
     
133.       }
     
134.       data[ReciveBytes]=0;//加上结束符
     
135.       cout << "收到 " <<ReciveBytes << "个字节:\n " <<data <<endl;
     
136.      }while(ReciveBytes> 0);
     
137.                 }
     
138.    }
     
139.   }//while接收数据
     
140.   
     
141.   if(Hnd==NULL)
     
142.   {
     
143.    CloseHandle(Hnd);
     
144.    closesocket(client);
     
145.   }
     
146. }
     
147. closesocket(server);
     
148. WSACleanup();
     
149. return   0;
     
150. }
     
151. 
     

复制代码

//client.cpp 客户端

1. #include   <iostream>
   
2. #include   <winsock2.h>
   
3. #include   <string>
   
4. #pragma   comment(lib, "ws2_32.lib ")
   
5. using   namespace   std;
   
6. 
   
7. void   OutErr(string   msg)
   
8. {
   
9. cout << "错误发生： " <<msg <<endl;
   
10. }
    
11. 
    
12. int   main()
    
13. {
    
14. SOCKET   c;
    
15. SOCKADDR_IN   server_addr;
    
16. WSADATA       wsd;
    
17. string         msg;
    
18. string         remote_ip;
    
19. 
    
20. cout << "输入对方IP地址: ";
    
21. cin>> remote_ip;
    
22. if(WSAStartup(MAKEWORD(2,2),&wsd)!=0)
    
23. {
    
24.   OutErr(string( "WSAStartup "));
    
25.   return   1;
    
26. }
    
27. 
    
28. if((c=WSASocket(AF_INET,SOCK_STREAM,IPPROTO_IP,NULL,0,1))==INVALID_SOCKET)
    
29. {
    
30.   OutErr(string( "WSASocket "));
    
31.   return   1;
    
32. }
    
33. server_addr.sin_addr.S_un.S_addr=inet_addr(remote_ip.c_str());
    
34. server_addr.sin_family=AF_INET;
    
35. server_addr.sin_port=htons(4000);
    
36. 
    
37. if(connect(c,(SOCKADDR   *)&server_addr,sizeof(SOCKADDR_IN))!=0)
    
38. {
    
39.   OutErr(string( "connect "));
    
40.   return   1;
    
41. }
    
42. 
    
43. do
    
44. {
    
45.   int   SentBytes;
    
46.   cin>> msg;
    
47.   if(msg== "exit ")break;
    
48.   if((SentBytes=send(c,msg.c_str(),msg.size(),0))==SOCKET_ERROR)
    
49.   {
    
50.    OutErr(string( "send "));
    
51.    break;
    
52.   }
    
53. }while(1);
    
54. 
    
55. closesocket(c);
    
56. WSACleanup();
    
57. return   0;
    
58. }
    

复制代码

转载自：http://blog.sina.com.cn/s/blog_68b606350100yi5l.html

hniu

用Boost不是更方便吗？

nketc

这和c有啥子区别？