TCP实现点对点聊天(多线程)-C语言

TCP实现点对点聊天(多线程)-服务器端

//**********操作步骤********************//
//1、编译：gcc TCPM.c -lws2_32 -o TCPM.exe
//2、运行：TCPM
#include <stdio.h>
#include <winsock2.h>
#include <pthread.h>

int nclients = 0;
void *hanld_client(void *arg);
int main(int argc, char *argv[])
{
	//初始化动态链接库ws2_32.dll
	WORD sockVersion = MAKEWORD(2, 2);
	WSADATA wsaData;
	if(WSAStartup(sockVersion, &wsaData) != 0)
	{
		return 0;
	}
	//创建socket套接字，指定使用TCP协议
	SOCKET listensocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(listensocket == INVALID_SOCKET)
	{
		printf("创建socket失败！！！");
		return 0;
	}
	//填充服务器相关地址信息，绑定IP地址、协议及端口
	struct sockaddr_in server;
	server.sin_family = AF_INET;
	server.sin_port = htons(8888);
	server.sin_addr.S_un.S_addr = INADDR_ANY;

	if(bind(listensocket, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR)
	{
        printf("绑定地址失败！");
        return 0;
	}

	//开始监听
	if(listen(listensocket, 6) == SOCKET_ERROR)
	{
        printf("监听失败！");
        return 0;
	}

	SOCKET clientsocket;
	struct sockaddr_in clientaddr;
	int nclientaddrlen = sizeof(clientaddr);
	char recvData[256];
	
	//循环接收数据
	while(1)
	{
		//等待连接
		printf("等待连接！！");
		//accept会阻塞
		clientsocket = accept(listensocket, (struct sockaddr *)&clientaddr, &nclientaddrlen);
		if(clientsocket == INVALID_SOCKET)
		{
			printf("接收客户端连接错误！");
			return 0;
		}
		printf("客户端%s连接成功！", inet_ntoa(clientaddr.sin_addr));
		//接收数据
		pthread_t id;
		//(SOCKET *)&clientsocket是hanld_client的参数
		pthread_create(&id, NULL, hanld_client, (SOCKET *)&clientsocket);  //创建一个线程，来处理客户端。
		pthread_detach(id);   //把线程分离出去。
	}
	//	关闭套接字
	closesocket(listensocket);
	WSACleanup();
	return 0;
}

//新建线程需要执行的函数
void *hanld_client(void *arg)
{
	SOCKET clientsocket = *(SOCKET *)(arg);
	char recvData[256];
	while(1)
	{
		int ret = recv(clientsocket, recvData, 256, 0);
		if(ret > 0)
		{
			recvData[ret] = '\0';
			printf(recvData);
		}
		if(strncmp(recvData, "end", 3) == 0)
		{
			break;
		}
  		memset(recvData, 0, sizeof(recvData));
  		nclients++;
  		char sendData[256];
  		sprintf(sendData, "您好，第%03d号客户\n", nclients);

  		send(clientsocket, sendData, strlen(sendData), 0);
  		memset(sendData, 0, sizeof(sendData));
	}
	closesocket(clientsocket);
}

TCP实现点对点聊天-客户端

//**********操作步骤********************//
//1、编译：gcc TCPC.c -lws2_32 -o TCPC.exe
//2、运行：TCPC
#include <stdio.h>
#include <winsock2.h>
int main(int argc, char *argv[])
{
	//设定socket版本为2.2
	WORD sockVersion = MAKEWORD(2, 2);
	//初始化动态链接库ws2_32.dll
	WSADATA wsaData;
	if(WSAStartup(sockVersion, &wsaData))
	{
		return 0;
	}
	//创建socket套接字，指定使用TCP协议
	SOCKET clientsocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(clientsocket == INVALID_SOCKET)
	{
		printf("socket创建错误!");
		return 0;
	}
	//填充相关地址信息，绑定ip地址、协议及端口
	struct sockaddr_in serveraddr;
	//使用TCP/IP协议栈
	serveraddr.sin_family = AF_INET;
	//将端口8888转换为网络字节序
	serveraddr.sin_port = htons(8888);
	//绑定本机的IP地址，可以使用本机的任何IP地址
	serveraddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
	int len = sizeof(serveraddr);
	if(connect(clientsocket, (struct sockaddr*)&serveraddr, len) == SOCKET_ERROR)
	{
		printf("连接服务器失败！");
		closesocket(clientsocket);
		return 0;
	}
    while(1)
    {
    	char sendData[20];
		gets(sendData);
		sendData[strlen(sendData)] = '\n';
		send(clientsocket, sendData, strlen(sendData), 0);
		memset(sendData, 0, sizeof(sendData));
	    char recvData[256];
		//接收UDP的数据包  接收存到recvData 长度返回ret 客户端相关信息给clientaddr
		int ret = recv(clientsocket, recvData, 256, 0);
		if(ret > 0)
		{
			recvData[ret] = '\0';   //结束字符串
			printf("服务器回应：%s", recvData);
		}
		memset(recvData, 0, sizeof(recvData));
	}
	clockscket(clientsocket);
	//释放WinSock动态链接库相关的资源
	WSACleanup();
	return 0;
}