8.15 网络编程

TCP多进程并发服务器

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/wait.h>
#include<netinet/in.h>
#include<signal.h>
#include<arpa/inet.h>
#include<stdio.h>
#include<stdio.h>

#define PORT 8888
#define IP "192.168.31.190"
#define MSG(msg) do{\
	printf("__%d__\n",__LINE__);\
	perror(msg);\
}while(0)

typedef void (*sighandler_t)(int);

void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
}

int main(int argc, const char *argv[])
{
	sighandler_t s = signal(17,handler);
	if(SIG_ERR == s)
	{
		MSG("signal");
		return -1;
	}

	int sfd = socket(AF_INET,SOCK_STREAM,0);

	//允许端口快速重用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		MSG("setsockopt");
		return -1;
	}
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		MSG("bind");
		return -1;
	}

	if(listen(sfd,10)<0)
	{
		MSG("listen");
		return -1;
	}

	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	int newfd;
	pid_t pid;

	while(1)
	{
		newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd<0)
		{
			MSG("accept");
			return -1;
		}

		pid = fork();
		if(pid>0)
		{
			close(newfd);
		}
		else if(0 == pid)
		{
			close(sfd);
			//子进程运行，负责与客户端进行交互
			char buf[128];
			ssize_t res = 0;
			while(1)
			{
				bzero(buf,sizeof(buf));
				//接收
				if((res = recv(newfd,buf,sizeof(buf),0))<0)
				{
					MSG("recv");
					return -1;
				}
				else if(res==0)
				{
					printf("%s : %d newfd = %d client off_line\n",\
							inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
					break;
				}
				printf("%s : %d newfd = %d:%s \n",inet_ntoa(cin.sin_addr),\
						ntohs(cin.sin_port),newfd,buf);
				//发送数据
				strcat(buf,"<<<<<haihaihai");
				if(send(newfd,buf,sizeof(buf),0)<0)
				{
					MSG("send");
					return -1;
				}
			}
			close(newfd);
			exit(0);
		}
	}
	close(sfd);
	return 0;
}

多线程并发服务器

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>

#define ERR_MSG(msg)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 6666
#define IP 	"192.168.31.135"

//需要传入到线程中的成员
struct msg
{
	int newfd;
	struct sockaddr_in cin;
};

void* rcv_cli_msg(void* arg);	//void* arg = (void*)&cliInfo
int main(int argc, const char *argv[])
{
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success\n");

	//允许端口快速重用
	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	if(listen(sfd, 10) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}

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

	int newfd = 0;
	pthread_t tid;
	struct msg cliInfo;

	while(1)
	{
		//主线程主要负责连接
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
			perror("accept");
			return -1;
		}

		//网络字节序的IP-->点分十进制  网络字节序的port--->本机字节序
		printf("[%s : %d] newfd = %d\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd);

		cliInfo.newfd = newfd;
		cliInfo.cin = cin;
		//一旦连接成功后，创建一个分支线程用户与客户端交互;
		if(pthread_create(&tid, NULL, rcv_cli_msg, (void*)&cliInfo) != 0)
		{
			ERR_MSG("pthread_create");
			return -1;
		}
	}

	close(sfd);
	return 0;
}

//线程处理函数
void* rcv_cli_msg(void* arg) 	//void* arg = (void*)&cliInfo
{
	//分离分支线程
	pthread_detach(pthread_self());

	int newfd = (*(struct msg*)arg).newfd;
	struct sockaddr_in cin = (*(struct msg*)arg).cin;

	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		bzero(buf, sizeof(buf));
		//循环接收 
		res = recv(newfd, buf, sizeof(buf), 0);
		if(res < 0)
		{
			ERR_MSG("recv");
			break;
		}
		else if(0 == res)
		{
			printf("[%s : %d] newfd = %d client off-line\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd);
			break;
		}
		printf("[%s : %d] newfd = %d : %s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd, buf);

		//发送数据
		strcat(buf, "congnidequansijieluguo");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			break;
		}
		printf("send message success\n");
	}
	close(newfd);
	pthread_exit(NULL);
}

TCP循环服务器模型

sfd = socket();
bind();
listen();
while(1)
{
    newfd = accept();
    while(1)
    {
        recv();
        send();
    }
    close(newfd);
}
close(sfd);