C++ TCP通信 客户端多线程程序

C++ TCP通信 客户端多线程程序

#define HAVE_STRUCT_TIMESPEC
#include <winsock.h>
#include <stdlib.h>
#include <stdio.h>
#include <pthread.h> 
#include <iostream>
#pragma comment (lib,"ws2_32.lib") //显示加载套接字库
#pragma comment(lib, "pthreadVC2.lib")

void* Write(void* temporary) //写线程，void*是线程创建的标准参数，用时修改
{
	SOCKET* sock_client = (SOCKET*)temporary; //类型转换
	printf("Write套接字指针的地址：%p\n", &sock_client);
	printf("Write套接字指针指向的地址：%p\n", sock_client);
	int result = 0;
	char buffer[128];
	memset(buffer, 0, sizeof(buffer)); //初始化缓冲区
	gets_s(buffer);
	while (1)
	{
		gets_s(buffer); //从键盘中读取数据
		if (strcmp(buffer, "quit") == 0)
		{
			pthread_exit(NULL); //退出线程
		}
		result = send(*sock_client, buffer, sizeof(buffer), 0); //将数据写进客户端套接字

		if (result < 0)
		{
			perror("发送失败。\n");
			exit(1);
		}
		else if (result == 0)
		{
			printf("没有数据发送。\n");
			continue;
		}
		else
		{
			printf("发送客户端数据：%s\n", buffer);
		}
		memset(buffer, 0, sizeof(buffer)); //初始化缓冲区
	}
}
void* Read(void* temporary) //读线程，当有数据过来时可以无阻塞读取
{
	SOCKET* sock_client = (SOCKET*)temporary;
	printf("Read套接字指针的地址：%p\n", &sock_client);
	printf("Read套接字指针指向的地址：%p\n", sock_client);
	int result = 0;
	char buffer[128];
	memset(buffer, 0, sizeof(buffer)); //初始化缓冲区
	while (1)
	{
		result = recv(*sock_client, buffer, sizeof(buffer), 0); //从套接字中读取数据
		if (result < 0)
		{
			perror("接收失败。\n");
			exit(1);
		}
		else if (result == 0)
		{
			printf("服务器已关闭.\n");
			exit(0);
		}
		else
		{
			printf("接收客户端数据: %s\n", buffer);
		}
		memset(buffer, 0, sizeof(buffer)); //初始化缓冲区
	}
}

int main()
{
	char server_addr[20];
	memset(server_addr, 0, sizeof(server_addr)); //初始化缓冲区
	printf("请输入服务器IP地址：");
	scanf_s("%s", &server_addr, 20);
	int server_port_number = 0;
	printf("请输入服务器端口号：");
	scanf_s("%d", &server_port_number);

	//初始化套接字库
	WSADATA wsadata; //定义一个WSADATA类型的结构体，存储被WSAStartup函数调用后返回的Windows Sockets数据
	WORD sock_version = MAKEWORD(2, 0); //设置版本号
	if (WSAStartup(sock_version, &wsadata)) //初始化套接字，启动构建，将“ws2_32.lib”加载到内存中
	{
		printf("WSAStartup failed \n");
		return 0;
	}

	//1、创建套接字socket（socket）
	//家族：ip地址IPV4 IPV6//协议类型//用户不确定什么方式 传0
	SOCKET sock_client = socket(AF_INET, SOCK_STREAM, 0); //创建一个对应IPv4的TCP套接字
	if (sock_client == INVALID_SOCKET)
	{
		printf("socket failed \n");
		return 0;
	}

	//地址处理以及端口处理
	sockaddr_in addr_client;
	addr_client.sin_family = AF_INET; //家族：ip地址IPV4
	addr_client.sin_port = htons(server_port_number); //htons 转换为u_short
	addr_client.sin_addr.S_un.S_addr = inet_addr(server_addr); //设置连接地址
	printf("客户端已启动。\n");

	//2、向服务器发出连接请求（connect）
	int result = connect(sock_client, (sockaddr *)&addr_client, sizeof(addr_client));
	if (result == 0)
	{
		printf("服务器已连接。\n");

		//3、请求连接后与服务器进行通信操作（send/recv）
		pthread_t pthread_write_id;
		pthread_t pthread_read_id;
		result = pthread_create(&pthread_write_id, NULL, Write, &sock_client); //写线程
		result = pthread_create(&pthread_read_id, NULL, Read, &sock_client); //读线程
		pthread_join(pthread_write_id, NULL);//等待写线程结束
	}
	//4、 释放套接字资源（closesocket）
	closesocket(sock_client);
	WSACleanup();
	system("puase");
	return 0;
}