windows下的一个socket多人聊天程序(附源代码)

首先声明什么是socket?一个ip加一个端口就是一个socket,一对socket之间可以互相通信。

socket基本通讯过程是,客户端连接到服务器,就可以给服务器发消息,在客户端关闭了连接之前,服务器可以通过建立的连接回复消息。另外,在一个客户端连接到服务器之前,服务器一直处在监听状态。

根据上述所说,可以很容易地二人通信,只要保持socket连接不断开就行(权利在客户端)。但是多人通信有点麻烦,一般的做法是采用多线程,并且把服务器的socket监听设为非阻塞(默认阻塞,连上一个客户端后,除非客户端断开连接,其他客户端的就连不上了)。但是无疑多线程对于新手来说比较麻烦。

下面是一个很简单的思路,既能多人聊天又能避免采用多线程:

假如有A,B,C,D,E 5个客户端,服务器一直打开,让5个客户端同时进行连接服务器。某个客户端A连接成功后,客户端要做的是选择“是否向另外的客户端发送信息”,服务器要做的是查看自己的空间里是否有针对A的消息。如果有,发送给客户端。注意,A实际上是向服务器发消息,服务器暂时保存这条消息。这样做完后,A断开连接,休眠一段时间,服务器被释放,继续监听。 此时如果有客户端处于活动状态,假设B又连接上了服务器,服务器就会查看是否有针对B的消息,然后发送。完成后,B进入休眠,此时可能C又连接上了。当客户端休眠一段时间后,再次进入向服务器请求连接的循环中。

// server.cpp

#include <iostream>
#include <cstdio>
#include <Winsock2.h>

using namespace std;
int main() {
//准备好5个id所需的5个数组。
	char reg[5][100];
	reg[0][0]='A';
	reg[1][0]='B';
	reg[2][0]='C';
	reg[3][0]='D';
	reg[4][0]='E';
// 加载socket动态链接库(dll)
	WORD wVersionRequested;
	WSADATA wsaData;    // 这结构是用于接收Wjndows Socket的结构信息的
	int err;

	wVersionRequested = MAKEWORD( 1, 1 );   // 请求1.1版本的WinSock库

	err = WSAStartup( wVersionRequested, &wsaData );
	if ( err != 0 ) {
		return -1;          // 返回值为零的时候是表示成功申请WSAStartup
	}

	if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) {
		// 检查这个低字节是不是1,高字节是不是1以确定是否我们所请求的1.1版本
		// 否则的话,调用WSACleanup()清除信息,结束函数
		WSACleanup( );
		return -1;
	}

// 创建socket操作,建立流式套接字,返回套接字号sockSrv
	// SOCKET socket(int af, int type, int protocol);
	// 第一个参数,指定地址簇(TCP/IP只能是AF_INET,也可写成PF_INET)
	// 第二个,选择套接字的类型(流式套接字),第三个,特定地址家族相关协议(0为自动)
	SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0);

// 套接字sockSrv与本地地址相连
	// int bind(SOCKET s, const struct sockaddr* name, int namelen);
	// 第一个参数,指定需要绑定的套接字;
	// 第二个参数,指定该套接字的本地地址信息,该地址结构会随所用的网络协议的不同而不同
	// 第三个参数,指定该网络协议地址的长度
	// PS: struct sockaddr{ u_short sa_family; char sa_data[14];};
	//                      sa_family指定该地址家族, sa_data起到占位占用一块内存分配区的作用
	//     在TCP/IP中,可使用sockaddr_in结构替换sockaddr,以方便填写地址信息
	//
	//     struct sockaddr_in{ short sin_family; unsigned short sin_port; struct in_addr sin_addr; char sin_zero[8];};
	//     sin_family表示地址族,对于IP地址,sin_family成员将一直是AF_INET。
	//     sin_port指定将要分配给套接字的端口。
	//     sin_addr给出套接字的主机IP地址。
	//     sin_zero[8]给出填充数,让sockaddr_in与sockaddr结构的长度一样。
	//     将IP地址指定为INADDR_ANY,允许套接字向任何分配给本地机器的IP地址发送或接收数据。
	//     如果想只让套接字使用多个IP中的一个地址,可指定实际地址,用inet_addr()函数。
	SOCKADDR_IN addrSrv;
	//addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY); // 将INADDR_ANY转换为网络字节序,调用 htonl(long型)或htons(整型)
	addrSrv.sin_addr.S_un.S_addr = inet_addr("115.154.18.135");
	addrSrv.sin_family = AF_INET;
	addrSrv.sin_port = htons(6000);



// 将套接字设置为监听模式(连接请求), listen()通知TCP服务器准备好接收连接
	// int listen(SOCKET s,  int backlog);
	// 第一个参数指定需要设置的套接字,第二个参数为(等待连接队列的最大长度)


// accept(),接收连接,等待客户端连接
	// SOCKET accept(  SOCKET s,  struct sockaddr* addr,  int* addrlen);
	// 第一个参数,接收一个处于监听状态下的套接字
	// 第二个参数,sockaddr用于保存客户端地址的信息
	// 第三个参数,用于指定这个地址的长度
	// 返回的是向与这个监听状态下的套接字通信的套接字

// 客户端与用户端进行通信

	// send(), 在套接字上发送数据
	// int send( SOCKET s,  const char* buf,  int len,  int flags);
	// 第一个参数,需要发送信息的套接字,
	// 第二个参数,包含了需要被传送的数据,
	// 第三个参数是buffer的数据长度,
	// 第四个参数,一些传送参数的设置

	// recv(), 在套接字上接收数据
	// int recv(  SOCKET s,  char* buf,  int len,  int flags);
	// 第一个参数,建立连接后的套接字,
	// 第二个参数,接收数据
	// 第三个参数,接收数据的长度,
	// 第四个参数,一些传送参数的设置

	SOCKADDR_IN  addrClient;
	int len = sizeof(SOCKADDR);
	char sendBuf[100];
	char recvBuf[100];
	char id,tmp;

	while(true) {   // 不断等待客户端请求的到来 ,注意接收一次发送一次后应该断开以使得其他客户端也能连接;
		bind(sockSrv, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)); // 第二参数要强制类型转换
		listen(sockSrv, 10);
		//cout<<"cycle begin"<<endl;
		SOCKET sockConn = accept(sockSrv, (SOCKADDR*)&addrClient, &len);  //客户端socket的备份。
		recv(sockConn, recvBuf, 100, 0);//对于接收信息后,应该先简单判断是请求消息还是发送消息;发送消息的话不急着处理recvBuf;
		id=recvBuf[0];
		//cout<<id<<endl<<endl;

		for(tmp=0; tmp<5; tmp++) {
			if(id==reg[tmp][1]) {
				//memcpy(sendBuf,reg[tmp],100);
				//如上,这样用会导致sendBuf超界,send失败。
				for(id=0; id<100; id++) { //注意,这样做是合法的,id已经失去了作用;
					sendBuf[id]=reg[tmp][id];
				}
				//cout<<"below-"<<sendBuf<<endl;
				//getchar();
				send(sockConn, sendBuf, strlen(sendBuf)+1, 0);
				reg[tmp][1]=' ';
				reg[tmp][2]='\0';
				break;//查到未读信息就结束for循环;
			}
		}
		if(tmp==5)send(sockConn, " ", strlen(" ")+1, 0);
		closesocket(sockConn);
		//WSACleanup();
		//将recv的信息录入;
		if(recvBuf[1]!='\0') { //如果没有来消息就不保存信息了;
			tmp=recvBuf[0]-65;
			for(id=0; id<100; id++) { //注意,这样做是合法的,id已经失去了作用;
				reg[tmp][id]=recvBuf[id];
			}

		}
		//else就比较尴尬了,本来应该给每个客户端设置多条数组空间来存储多条消息(可能是发送给不同人的)
		//下面这样设置,如果还是发给同一个人,后缀上就可以了,不是发给同一个人,也直接覆盖掉。

		//memcpy(reg[tmp],recvBuf,strlen(recvBuf));
		//cout<<"a cycle done"<<endl;
	}

	printf("\n");
	system("pause");
	return 0;
}
这是服务器的源码。

// client.cpp


#include <iostream>
#include <cstdio>
#include <Winsock2.h>
#include <windows.h>
#include<conio.h>
#include <cstring>
#define ID 'C'
using namespace std;


int main() {
// 加载socket动态链接库(dll)
	char recvBuf[100];
	char sendBuf[100];
	char scanBuf[100];
	sendBuf[0]=ID;//self id;
	WORD wVersionRequested;
	WSADATA wsaData;    // 这结构是用于接收Wjndows Socket的结构信息的
	int err;


	wVersionRequested = MAKEWORD( 1, 1 );   // 请求1.1版本的WinSock库


	err = WSAStartup( wVersionRequested, &wsaData );
	if ( err != 0 ) {
		return -1;          // 返回值为零的时候是表示成功申请WSAStartup
	}


	if ( LOBYTE( wsaData.wVersion ) != 1 || HIBYTE( wsaData.wVersion ) != 1 ) {
		// 检查这个低字节是不是1,高字节是不是1以确定是否我们所请求的1.1版本
		// 否则的话,调用WSACleanup()清除信息,结束函数
		WSACleanup( );
		return -1;
	}


// 创建socket操作,建立流式套接字,返回套接字号sockClient
	// SOCKET socket(int af, int type, int protocol);
	// 第一个参数,指定地址簇(TCP/IP只能是AF_INET,也可写成PF_INET)
	// 第二个,选择套接字的类型(流式套接字),第三个,特定地址家族相关协议(0为自动)


	char tmp;
	for(;;) {
		//cout<<"cycle begin"<<endl;
		SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);


// 将套接字sockClient与远程主机相连
		// int connect( SOCKET s,  const struct sockaddr* name,  int namelen);
		// 第一个参数:需要进行连接操作的套接字
		// 第二个参数:设定所需要连接的地址信息
		// 第三个参数:地址的长度
		SOCKADDR_IN addrSrv;
		addrSrv.sin_addr.S_un.S_addr = inet_addr("115.154.18.135");      // 本地回路地址是127.0.0.1;
		addrSrv.sin_family = AF_INET;
		addrSrv.sin_port = htons(6000);
		while((connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)))!=0);  //0代表成功,-1代表失败,加循环
		//cout<<"linked the server"<<endl;
		if(_kbhit())
			tmp=getch();
		if(tmp=='s') {//此处要保证tmp被重置为不是s;
			closesocket(sockClient);
			cout<<"Enter the destination id:";
			Sleep(1000);//这些是为了防止键盘抖动。
			tmp=getch();
			sendBuf[1]=tmp;
			cout<<endl<<"Enter the message:"<<endl;
			Sleep(1000);
			gets(scanBuf);
			//strcat(sendBuf,scanBuf);
			//如上,不能这样用,导致发送出问题;
			for(tmp=0;tmp<strlen(scanBuf);tmp++){
				sendBuf[tmp+2]=scanBuf[tmp];
			}
			sendBuf[tmp+3]='\0';
			tmp='\0';
			Sleep(600);
			continue;
		}//捕捉到发消息请求后立刻关闭socket,设置好sendBuf重连socket;
		//cout<<"check key ok"<<endl;
//应该启动就进入发消息/发空消息模式,为了简便,服务器端寄存器定义好头部,收到连接后直接分析数据然后回传验证消息(0或者一个id+消息);
		send(sockClient, sendBuf, strlen(sendBuf)+1, 0);//该客户端的id为A;另外此处需要等待服务器确认消息
		//最后发现那一群烫烫烫烫错在了上面,具体杂错的不管了。
		//cout<<"send message ok"<<endl;
		recvBuf[0]='\0';
		recv(sockClient, recvBuf, 100, 0);
		//cout<<"rec message ok"<<endl;
		closesocket(sockClient);
		//WSACleanup();
		//发完接收完,处理sendBuf和recvBuf,休眠后再次连接;
		sendBuf[0]=ID;
		sendBuf[1]='\0';
		if(recvBuf[0]!=' ') { //服务器设置为:没有要发给客户端A的消息,则发送一个空格;如果B发给A信息了(校验方式是regB的第二位是A,就把regB发送过来。
			cout<<recvBuf[0]<<" send you a message! below:"<<endl;
			recvBuf[0]=' ';
			recvBuf[1]=' ';
			cout<<recvBuf<<endl;
		}
		Sleep(600);//休眠600毫秒再次打开;
		//cout<<"sleep done;"<<endl;
	}


	printf("End linking...\n");
	//closesocket(sockClient);
	   // 终止对套接字库的使用


	printf("\n");
	system("pause");
	return 0;
}

如上,客户端的代号被宏定义为C,可替换。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值