運用線程實現服務器與客戶端相互通信

代碼

通過創建一個線程，使服務器的讀寫分離，主線程用於讀客戶端發送的數據，其他線程用來向客戶端寫數據，互不影響。

#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <unistd.h>
#include <pthread.h>
#include <string.h>
#include <arpa/inet.h>

#define PORT 6666

//讀
void Read(int client_socket)
{
	char buf[1024] = {"aaa"};
	while (1)
	{
		int ret = read(client_socket,buf,sizeof(buf)-1);
		if (-1 == ret)
		{
			perror ("read error");
		}
		if (0 == ret)
		{
			printf ("客戶端退出\n");
			break;
		}
		buf[ret] = '\0';
		printf ("收到客戶端數據：%s\n", buf);
	}
}

//寫
void *Write(void *v)
{
	int client_socket = *(int *)v;
	char buf[1024];
	while(1)
	{
		fgets(buf,1024,stdin);
		ssize_t fd = write(client_socket,buf,strlen(buf));
		if(0 == fd)
		{
			perror("寫入失敗");
			break;
		}
	}
}


// 監聽套接字
int init()
{
	int listen_socket = socket(AF_INET, SOCK_STREAM, 0);
	if(-1 == listen_socket)
	{
		perror("創建套接字失敗");
		return -1;
	}
	
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(struct sockaddr_in));
	addr.sin_family      = AF_INET;             /* Internet地址族 */
	addr.sin_port        = htons(PORT);         /* 端口號 */
	addr.sin_addr.s_addr = htonl(INADDR_ANY);   /* IP地址, 綁定本地的所有ip地址*/

	int ret = bind(listen_socket, (const struct sockaddr *)&addr, sizeof(addr));
	if(-1 == ret)
	{
		perror("綁定失敗");
		return -1;
	}
	
	// 3、監聽套接字
	ret = listen(listen_socket, 5);
	if(-1 == ret)
	{
		perror("監聽失敗");
		return -1;
	}
	
	return listen_socket;
}

// 通信套接字
int myAccept(int listen_socket)
{
	struct sockaddr_in client_addr;
	socklen_t  len = sizeof(client_addr);
	int client_socket = accept(listen_socket, (struct sockaddr *)&client_addr, &len);
	if (-1 == client_socket)
	{
		perror("accept  失敗");
		return -1;
	}
	
	printf ("客戶端的 ip = %s, 端口 = %d\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
	
	
	return client_socket;
}

int main()
{
	int listen_socket = init();
	if(-1 == listen_socket)
	{
		perror("創建套接字失敗");
		return -1;
	}
	
	int client_socket = myAccept(listen_socket);
	
	pthread_t thread;
	pthread_create(&thread,NULL,Write,(void *)&client_socket);
	
	Read(client_socket);
	
	close(listen_socket);
	
	return 0;
}