使用C/C++实现Socket聊天程序

原文链接:http://blog.csdn.net/yexianyi/article/details/3486766

非常感谢原作者

使用C/C++实现Socket聊天程序

Initsock.h文件

1. // initsock.h文件
2. #include <winsock2.h>
3. #include <stdlib.h> 
4. #include <conio.h> 
5. #include <stdio.h> 
6. #pragma comment(lib, "WS2_32")  // 链接到WS2_32.lib
7. class CInitSock     
8. {
9. public:
10.     CInitSock(BYTE minorVer = 2, BYTE majorVer = 2)
11.     {
12.         // 初始化WS2_32.dll
13.         WSADATA wsaData;
14.         WORD sockVersion = MAKEWORD(minorVer, majorVer);
15.         if(::WSAStartup(sockVersion, &wsaData) != 0)
16.         {
17.             exit(0);
18.         }
19.     }
20.     ~CInitSock()
21.     {   
22.         ::WSACleanup(); 
23.     }
24. };

TCP版

TCPClient.cpp文件

1. //
2. // TCPClient.cpp文件
3. /*
4.                 《使用说明》
5. 0.运行程序前请查看是否将initsock.h
6.   文件引入到项目中。
7. 1.首先修改聊天对方的IP地址
8. 2.请首先运行服务端（TCPServer）程序，再运行客户端（TCPClient）程序：
9.   如配置正确服务端会收到相关连接信息。
10. 3.连接成功后，需要由服务器端首先发起会话（输入消息并确认发送），
11.   客户端收到消息后才能输入消息并确认发送到服务器端。
12.   并且双方每次只能发送一条消息。如想发送第二条消息，需要等待该方成功
13.   接受到另一方的消息后才能继续输入消息。
14. */
15. #include "InitSock.h"
16. #include <stdio.h>
17. #include <iostream.h>
18. CInitSock initSock;     // 初始化Winsock库
19. int main()
20. {
21.     // 创建套节字
22.     SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
23.     if(s == INVALID_SOCKET)
24.     {
25.         printf(" Failed socket() /n");
26.         return 0;
27.     }
28.     
29.     // 也可以在这里调用bind函数绑定一个本地地址
30.     // 否则系统将会自动安排
31.     
32.     // 填写远程地址信息
33.     sockaddr_in servAddr; 
34.     servAddr.sin_family = AF_INET;
35.     servAddr.sin_port = htons(4567);
36.     // 注意，这里要填写服务器程序（TCPServer程序）所在机器的IP地址
37.     // 如果你的计算机没有联网，直接使用127.0.0.1即可
38.     servAddr.sin_addr.S_un.S_addr = inet_addr("192.168.1.129");
39.     
40.     if(::connect(s, (sockaddr*)&servAddr, sizeof(servAddr)) == -1)
41.     {
42.         printf(" Failed connect() /n");
43.         return 0;
44.     }
45.     
46.     char buff[256];
47.     char szText[256] ;
48.     
49.     while(TRUE)
50.     {
51.         //从服务器端接收数据
52.         int nRecv = ::recv(s, buff, 256, 0);
53.         if(nRecv > 0)
54.         {
55.             buff[nRecv] = '/0';
56.             printf("接收到数据：%s/n", buff);
57.         }
58.         // 向服务器端发送数据
59.         cin>>szText ;
60.         szText[255] = '/0';
61.         ::send(s, szText, strlen(szText), 0) ;
62.         
63.     }
64.     
65.     // 关闭套节字
66.     ::closesocket(s);
67.     return 0;
68. }

TCPServer.cpp文件

1. // TCPServer.cpp文件
2. /*
3.                 《使用说明》
4. 0.运行程序前请查看是否将initsock.h
5.   文件引入到项目中。
6. 1.首先修改聊天对方的IP地址
7. 2.请首先运行服务端（TCPServer）程序，再运行客户端（TCPClient）程序：
8.   如配置正确服务端会收到相关连接信息。
9. 3.连接成功后，需要由服务器端首先发起会话（输入消息并确认发送），
10.   客户端收到消息后才能输入消息并确认发送到服务器端。
11.   并且双方每次只能发送一条消息。如想发送第二条消息，需要等待该方成功
12.   接受到另一方的消息后才能继续输入消息。
13. */
14. #include "InitSock.h"
15. #include <stdio.h>
16. #include <iostream.h>
17. CInitSock initSock;     // 初始化Winsock库
18. int main()
19. {
20.     // 创建套节字
21.     SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
22.     if(sListen == INVALID_SOCKET)
23.     {
24.         printf("Failed socket() /n");
25.         return 0;
26.     }
27.     
28.     // 填充sockaddr_in结构
29.     sockaddr_in sin;
30.     sin.sin_family = AF_INET;
31.     sin.sin_port = htons(4567);
32.     sin.sin_addr.S_un.S_addr = INADDR_ANY;
33.     
34.     // 绑定这个套节字到一个本地地址
35.     if(::bind(sListen, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)
36.     {
37.         printf("Failed bind() /n");
38.         return 0;
39.     }
40.     
41.     // 进入监听模式
42.     if(::listen(sListen, 2) == SOCKET_ERROR)
43.     {
44.         printf("Failed listen() /n");
45.         return 0;
46.     }
47.     
48.     // 循环接受客户的连接请求
49.     sockaddr_in remoteAddr; 
50.     int nAddrLen = sizeof(remoteAddr);
51.     SOCKET sClient = 0;
52.     char szText[] = " TCP Server Demo! /r/n";
53.     while(sClient==0)
54.     {
55.         // 接受一个新连接
56.         sClient = ::accept(sListen, (SOCKADDR*)&remoteAddr, &nAddrLen);
57.         if(sClient == INVALID_SOCKET)
58.         {
59.             printf("Failed accept()");
60.         }
61.         
62.         
63.         printf("接受到一个连接：%s /r/n", inet_ntoa(remoteAddr.sin_addr));
64.         continue ;
65.     }
66.     while(TRUE)
67.     {
68.         // 向客户端发送数据
69.         gets(szText) ;
70.         ::send(sClient, szText, strlen(szText), 0);
71.         
72.         // 从客户端接收数据
73.         char buff[256] ;
74.         int nRecv = ::recv(sClient, buff, 256, 0);
75.         if(nRecv > 0)
76.         {
77.             buff[nRecv] = '/0';
78.             printf(" 接收到数据：%s/n", buff);
79.         }
80.     
81.     }
82.     // 关闭同客户端的连接
83.     ::closesocket(sClient);
84.         
85.     // 关闭监听套节字
86.     ::closesocket(sListen);
87.     return 0;
88. }

UDP版

1. // Chat.cpp : Defines the entry point for the console application.
2. //
3. /*
4.                 《使用说明》
5. 0.运行程序前请查看是否将initsock.h
6.   文件引入到项目中。
7. 1.首先修改聊天对方的IP地址
8. 2.运行程序：如配置正确另一方会收到相关连接信息。
9. 3.输入消息：在每次输入完欲发送的消息后，需要连续敲击两次回车。
10. 4.本程序有诸多缺陷：对用户输入的消息不能即时回显到控制台，
11.   需要在敲击两次回车后回显到屏幕。
12. */
13. #include "stdafx.h"
14. #include <iostream>
15. #include <stdlib.h>
16. #include <STDIO.H>
17. #include <windows.h>
18. #include "InitSock.h"
19. using namespace std;
20. CInitSock initSock;     // 初始化Winsock库
21. DWORD receiverMark ; //接收消息者线程标识符                      
22. DWORD senderMark ;//发送者线程标识符
23. /**
24. *定义信号量
25. */
26. DWORD WINAPI Receiver(LPVOID) ;
27. DWORD WINAPI Sender(LPVOID) ;
28. // 接收数据
29. char buff[1024];
30. sockaddr_in addr;
31. int nLen = sizeof(addr);
32. SOCKET s ;
33. int main(int argc, char *argv[])
34. {
35.     // 创建套节字
36.     s = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
37.     //u_long iMode = 1;
38.     //ioctlsocket(s, FIONBIO, &iMode);
39.     if(s == INVALID_SOCKET)
40.     {
41.         printf("Failed socket() /n");
42.         return 0;
43.     }
44.     
45.     // 填充sockaddr_in结构
46.     sockaddr_in sin;
47.     sin.sin_family = AF_INET;
48.     sin.sin_port = htons(4567);
49.     sin.sin_addr.S_un.S_addr = INADDR_ANY;
50.     addr.sin_family = AF_INET;
51.     addr.sin_port = htons(4567);
52.     // 注意，这里要填写服务器程序所在机器的IP地址
53.     // 如果你的计算机没有联网，直接使用127.0.0.1即可
54.     addr.sin_addr.S_un.S_addr = inet_addr("192.168.1.129");
55.     // 绑定这个套节字到一个本地地址
56.     if(::bind(s, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)
57.     {
58.         printf("Failed bind() /n");
59.         return 0;
60.     }
61.     // 发送数据
62.     char szText[] = "PC请求连接... /r/n";
63.     ::sendto(s, szText, strlen(szText), 0, (sockaddr*)&addr, sizeof(addr));
64.     CreateThread(NULL,0,Receiver,NULL,0,&receiverMark);
65.     CreateThread(NULL,0,Sender,NULL,0,&senderMark);
66.        
67.     bool isContinue = true ;
68.     while(isContinue)
69.     { 
70.           if(getche()==96){  //按~后终止程序运行
71.                isContinue = false; 
72.            }
73.     }
74.     system("PAUSE");
75.     return 0;
76. }
77. /**
78. *接收者
79. */
80. DWORD WINAPI Receiver(LPVOID lpParam )
81. {
82.     while(1)
83.     {
84.         int nRecv = ::recvfrom(s, buff, 1024, 0, (sockaddr*)&addr, &nLen); 
85.         if(nRecv > 0)
86.         {
87.             buff[nRecv] = '/0';
88.             printf(" Received data(%s)：%s/n", ::inet_ntoa(addr.sin_addr), buff);
89.             
90.         }        
91.     }
92.     return 0;
93. }
94. /**
95. *发送者
96. */
97. DWORD WINAPI Sender(LPVOID lpPara)
98. {
99.     while(1)
100.     {
101.         cout<<"Input your message: " ;
102.     
103.         // 发送数据
104.         char text[256] ;
105.         cin>>text ;
106.         text[255] = '/0' ;
107.         cout<<text<<endl ;
108.         
109.         ::sendto(s, text, strlen(text), 0, (sockaddr*)&addr, sizeof(addr));
110.         
111.     }
112.     return 0;
113. }

鉴于很多同学说拷贝的代码无法运行，

特此将源代码和实验报告上传于此：下载

解压密码：yexianyi20081216

 

北航软件学院《一级实践》实验报告

学号：GS0821594　　   姓名：叶现一    　　　　　　　　　　　　　　第　13　周

内容 训练	使用C/C++实现Socket聊天程序
	解决该问题的思路	解决该问题的整体方案不外有二： （1）基于TCP的Socket连接；（2）基于UDP的Socket连接； 但是，针对每种方案又各有很多具体的实现方法。 在本次实验中，我先后开发了基于TCP连接和UDP连接的Socket聊天程序。具体实现思路如下： （一）  基于TCP连接Socket聊天程序 基于该连接的聊天程序需要至少具备一个服务器端（Server）和一个客户端（Client）。在本程序中，一个用户作为Server端，另一个用户作为Client端。也就是说，作为Server端的用户，需要首先启动程序，等待Client端的连接请求。当TCP连接握手以后，双方方可进行交互。（注：在本程序中Server端并不是单独存在。它也可以向他的Client端发送消息。）但是本程序实现的交互功能十分简单，具有很多限制。当Client端与Server端握手以后，Server端需要首先发起会话；Client端在收到消息后再回复一条消息给Server端；同样，Server端在收到消息后再回复一条消息给Client端……以此类推。并且，无论是Server端还是Client端每次发送消息只能发送一条。 造成交互操作具有诸多限制的主要原因是，我在Server端和Client端使用了一个While循环，它们的伪代码分别如下： Client端 Server端     while(TRUE)     {         从Server端接收消息         {           … }        向Server端发送消息         {           … } } while(TRUE)     {        向Client端发送消息         {           … }         从Client端接收消息         {           … }      } （二）  基于UDP连接Socket聊天程序 基于该连接的聊天程序不需要具备服务器端（Server），每个客户端（Client）既是服务器端也是客户端。也就是说每个Client端自身既可以自行接收其它用户发来的消息，也可以向其它Client端发送消息，不需要事先与其他用户进行握手连接。 由于在默认情况下WinSock接口的recvfrom()和sendto()都会在被调用时阻塞当前线程，也就是说如果程序正在接受其他用户发来的数据，那么它就不能够执行发送数据的任务，反之相同。所以为解决该问题一般有以下几种解决方案：采用Select模型、WSAAsyncSelect模型、WSAEventSelect模型、重叠（Overlapped）模型和完成接口（Completion port）模型。在本程序中，由于我没能在短时间内学会上述方案中的任一种，因此采用了多线程技术去实现消息接收和发送的同步问题。也就是说，在程序中创建两个线程，一个线程负责发送消息，另一个消息负责接受消息。两个线程交替运行，从而达到同时收发消息的目的。当然采用多线程方式解决消息收发同步问题可以移除上个程序中每个用户一次只能发送一条消息的限制。
本周开发源代码	代码的功能简述	使用C/C++实现Socket聊天程序： TCP版：服务器端用户和客户端用户在成功连接后，其中一方通过控制台输入消息，依次轮流向另一方发送数据。要求，服务器端首先发起会话，并且双方每次只能发送一条消息。 UDP版：任一端用户通过指定IP地址将消息发送到另一端的用户。交互双方通过控制台输入消息向另一方发送数据。没有任何发送限制。
	开发的收获	理解了TCP连接和UDP连接的基本原理和工作过程。 复习了关于Windows多线程及进程同步的相关知识。
	开发中碰到的主要困难	对于TCP版：     在考虑如何解决消息的收发同步问题上遇到了困难。最终使用了不佳方案：通过在服务器端和客户端分别运行while循环并依次进行数据收发工作的方式解决数据收发同步问题。 对于UDP版：     同样在考虑如何解决消息的收发同步问题上遇到了困难。但这次使用了多线程解决该问题。
	开发中未能解决的问题	对于UDP版： 为何第一次消息输入完毕敲击一次回车后，只有消息的第一个字符没能发送出去，而其它字符却可以被成功发送出去？而且当第二次输入消息敲回车后消息就能被全部发送出去？ 为何消息输入完毕后需要按两次回车键才能将消息发送到另一端？ 为什么输入的消息不能即时回显到发送者屏幕上？只有当敲击二次回车后用户输入的欲发送消息才会显现出来？ 如何才能避免用户在输入消息的同时也能正常接收消息？也就是不至于打断用户已输入的消息的前提下，显示接收到的消息。
针对本周训练内容自己设计的案例	案例的主要功能	同代码的功能简述
	用到的基本知识	相关Winsock编程接口；TCP连接和UDP连接基本工作原理；Windows多线程； 进程同步
	程序注意了哪些规范	代码格式、常量的定义、函数的声明
	你进行了哪些测试	略
	程序进行了哪些有效性检查	略
	你是如何保证程序的高效率的	略

注意：实验报告和案例源代码须在本次小组讨论会前提交