linux学习笔记12 多线程服务器编程

#include <sys/types.h>
#include <sys/wait.h>
#include <errno.h>
#include <pthread.h>
#include <fcntl.h>

using namespace std;

struct s_info
{
	struct sockaddr_in addr_client;
	int cfd;
};//定义这个结构体是因为线程调用函数只有一个入口参数，但需要传入多个参数，所以使用结构体传参


void *thr(void *arg)//线程的调用函数
{
	struct s_info* ts=(s_info*)arg;
	char ip[64];
	char buf[256];//线程调用函数中定义的变量只有这个线程单独使用，但是在主线程中定义的变量子线程都可以使用。
	while (1)
	{
		cout<<"the client's ip is"<<inet_ntop(AF_INET, &ts->addr_client.sin_addr.s_addr, ip, sizeof(ip))<<"\n";
		cout<<"the client's port is"<<ntohs(ts->addr_client.sin_port)<<"\n";
		cout<<"the server's thread is"<<pthread_self()<<endl;
		int n=read(ts->cfd, buf, sizeof(buf));
		if (n==-1)
		{
			cout<<"read is error!";
			exit(1);
		}else if (n==0)//read是阻塞的，如果返回结果是0， 说明是客户端发送了文件结束符，表明要结束了，不是客户端不发消息就是0
		{
			cout<<"client is over!";
			break;
		}else
		{
			for (int i=0; i<n; i++)
				buf[i]=toupper(buf[i]);
			write(ts->cfd, buf, n);
		}
	}
	close(ts->cfd);
	return NULL;
}

int main(int argc, const char *argv[])
{
	if (argc<2)
	{
		cout<<"input :./a.out port";
		exit(1);	
	}

	int sfd, cfd;
	int port=atoi(argv[1]);

	struct s_info ts[256];
	int n,i=0;
	pthread_t tid;

	struct sockaddr_in addr_server;
	addr_server.sin_port=htons(port);
	addr_server.sin_addr.s_addr=htonl(INADDR_ANY);
	addr_server.sin_family=AF_INET;

	struct sockaddr_in addr_client;
	socklen_t addrlen=sizeof(addr_client);

	sfd=socket(AF_INET, SOCK_STREAM, 0);
	bind(sfd, (struct sockaddr*) &addr_server, sizeof(addr_server));
	listen(sfd, 20);
		
	while (1)
	{
		cfd=accept(sfd, (struct sockaddr*) &addr_client, &addrlen);
		while (cfd==-1&&errno==EINTR)
			cfd=accept(sfd, (struct sockaddr*) &addr_client, &addrlen);
		if (cfd==-1&&errno!=EINTR)
		{
			cout<<"accept is error!";
			exit(1);	
		}
		ts[i].addr_client=addr_client;
		ts[i].cfd=cfd;
		pthread_create(&tid, NULL, thr, &ts[i]);//创建一个线程
		pthread_detach(tid);//当线程结束时，可以自动回收线程
		i++;
		if (i==256) break;
	}
	close(sfd);
	return 0;
}

运行结果