socket聊天室

---

一、套接字相关知识

1.概念

套接字可以理解为通信连接的一组数据，其中主要包括协议、主机IP地址、端口号。
描述两个应用进程之间的端到端通信关联需要，协议、信源机IP地址、信源应用进程端口、信宿机IP地址、信宿应用进程端口。

2.TCP/IP的三种套接字

流套接字（SOCK_STREAM）（有连接）、数据报套接字（SOCK_DGRAM）（无连接）、原始套接字（SOCK_ROW）。

3.面向连接的CS模型

（1）服务器：socket()建立流套接字，bind()套接字与本地地址绑定，listen()监听，accept()接收连接，等待客户进程连接请求，recv()接收数据，send()发送数据，close()关闭套接字。

（2）客户端：socket()建立流套接字，connect()将套接字与远程主机相连，send()发送数据，recv()接收数据，close()关闭套接字。

4.无连接的CS模型

（1）服务器：socket()建立套接字，bind()套接字与本地地址绑定，recvfrom()接收数据，sendto()发送数据，close()关闭套接字。

（2）客户端：socket()建立套接字，bind()套接字与本地地址绑定，sendto()发送数据，recvfrom()接收数据，close()关闭套接字。

二、服务器端代码

#include<stdio.h>
#include<WS2tcpip.h>
#include<WinSock2.h>
#pragma comment(lib,"ws2_32.lib")
#include<Windows.h>

SOCKADDR_IN cAddr = { 0 };	//获取客户端协议地址簇
int len = sizeof cAddr;
SOCKET clientSocket[1024];	//最多1024个客户端
int count = 0;	//第几个客户端

void func(int index);	//收发数据

int main()
{
	//1.确定协议版本
	WSADATA wsaData;
	WSAStartup(MAKEWORD(2, 2), &wsaData);	//用于初始化WinSock，即检查系统中是否有Windows Sockets的实现库
	if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
	{
		printf("确定协议版本失败！\n");
		return -1;
	}
	printf("确定协议版本成功！\n");

	//2.创建socket
	SOCKET serverSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);	//创建套接字。参数1：协议类型（协议簇），参数2：通信载体，参数3：ip保护方式
	if (serverSocket == SOCKET_ERROR)	//SOCKET_ERROR=-1
	{
		printf("创建socket失败：%d\n", GetLastError());
		//9.清理协议版本信息
		WSACleanup();
		return -1;
	}
	printf("创建socket成功！\n");

	//3.创建服务器协议地址簇
	SOCKADDR_IN addr = { 0 };
	addr.sin_family = AF_INET;	//协议地址簇
	addr.sin_addr.S_un.S_addr = inet_addr("192.168.0.104");	//inet_addr将点分十进制表示的IP地址转换为网络字节顺序的IP地址
	//inet_pton(AF_INET, "192.168.0.104", &addr);
	addr.sin_port = htons(9527);	//10000左右，小端转大端

	//4.绑定
	int r = bind(serverSocket, (struct sockaddr*)&addr, sizeof addr);	//套接字与主机本地IP地址与端口号的绑定
	if (r == -1)
	{
		printf("绑定失败：%d\n", GetLastError());
		//8.关闭socket
		closesocket(serverSocket);
		//9.清理协议版本信息
		WSACleanup();
		return -1;
	}
	printf("绑定成功！\n");

	//5.监听
	r = listen(serverSocket, 10);	//监听，准备接收客户进程发送的连接请求，第二个参数：连接队列可达到的最大长度
	if (r == -1)
	{
		printf("监听失败：%d\n", GetLastError());
		//8.关闭socket
		closesocket(serverSocket);
		//9.清理协议版本信息
		WSACleanup();
		return -1;
	}
	printf("监听成功！\n");

	//6.等待客户端连接
	for (int i = 0; i < 1024; i++)
	{
		clientSocket[i] = accept(serverSocket, (sockaddr*)&cAddr, &len);	//接收客户端连接请求
		if (clientSocket[i] == SOCKET_ERROR)
		{
			printf("客户端连接失败：%d\n", GetLastError());
			//8.关闭socket
			closesocket(serverSocket);
			//9.清理协议版本信息
			WSACleanup();
			return -1;
		}
		printf("客户端连接成功！\n");
		count++;
		CreateThread(NULL,NULL,(LPTHREAD_START_ROUTINE)func,(LPVOID)i,NULL,NULL);	//创建线程,i是func的参数
	}

	//8.关闭socket
	closesocket(serverSocket);

	//9.清理协议版本信息
	WSACleanup();

	system("pause");
	return 0;
}

void func(int index)
{
	//7.通信
	char buff[1024];
	int r;
	while (1)
	{
		r = recv(clientSocket[index], buff, 1023, NULL);	//接收数据
		if (r > 0)
		{
			buff[r] = 0;
			printf("%s发来的数据：%s\n", inet_ntoa(cAddr.sin_addr), buff);	//将网络字节顺序表示的IP地址转换为点分十进制
			//printf("%s发来的数据：%s\n", inet_ntop(AF_INET, &addr, buff, 1024), buff);

			for (int i = 0; i < count; i++)	//广播发送给所有客户端
				send(clientSocket[i], buff, r, NULL);
		}
	}
}

三、服务器端代码

#include<stdio.h>
#include<WS2tcpip.h>
#include<WinSock2.h>
#pragma comment(lib,"ws2_32.lib")
#include<Windows.h>
#include<iostream>

using namespace std;

void recvServerAndShow();

SOCKET clientSocket;

int main()
{
	//1.确定协议版本
	WSADATA wsaData;
	WSAStartup(MAKEWORD(2, 2), &wsaData);
	if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
	{
		printf("确定协议版本失败！\n");
		return -1;
	}
	printf("确定协议版本成功！\n");

	//2.创建socket
	clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);	//参数1：协议类型（协议簇），参数2：通信载体，参数3：ip保护方式
	if (clientSocket == SOCKET_ERROR)	//SOCKET_ERROR=-1
	{
		printf("创建socket失败：%d\n", GetLastError());
		//8.关闭socket
		closesocket(clientSocket);
		//9.清理协议版本信息
		WSACleanup();
		return -1;
	}
	printf("创建socket成功！\n");

	//3.获取服务器协议地址簇
	SOCKADDR_IN addr = { 0 };
	addr.sin_family = AF_INET;	//协议地址簇
	addr.sin_addr.S_un.S_addr = inet_addr("192.168.0.104");
	//inet_pton(AF_INET, "192.168.0.104", &addr);
	addr.sin_port = htons(9527);	//10000左右，小端转大端

	//4.连接服务器
	int r = connect(clientSocket, (sockaddr*)&addr, sizeof addr);	//提出一条与服务器建立连接的请求
	if (r == -1)
	{
		printf("连接服务器失败：%d\n", GetLastError());
		//6.关闭socket
		closesocket(clientSocket);
		//7.清理协议版本信息
		WSACleanup();
		return -1;
	}
	printf("连接服务器成功！\n");

	CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)recvServerAndShow, NULL, NULL, NULL);	//创建线程

	//5.通信
	char buff[1024];
	while (1)
	{
		memset(buff, 0, 1024);//清空
		cin>>buff;//接收用户输入
		send(clientSocket, buff, strlen(buff), NULL);//发送给服务器
	}

	//6.关闭socket
	closesocket(clientSocket);

	//9.清理协议版本信息
	WSACleanup();

	return 0;
}

void recvServerAndShow()
{
	char buff[1024];
	int r;
	while (1)
	{
		r = recv(clientSocket, buff, 1023, NULL);
		if (r > 0)
		{
			printf("消息：%s\n", buff);
		}
	}
}