5.18作业

最新推荐文章于 2024-08-27 19:58:48 发布

沈✘✘

最新推荐文章于 2024-08-27 19:58:48 发布

阅读量45

点赞数

文章标签： c语言开发语言

本文链接：https://blog.csdn.net/qq_59539828/article/details/130755269

版权

这篇文章展示了使用C语言实现的基于套接字(socket)的多线程和多进程TCP服务器。代码中包含了错误处理函数，用于在socket操作失败时打印错误信息。在多线程版本中，主线程接收客户端连接并创建新线程处理通信；在多进程版本中，主线程接收连接后通过fork创建子进程来处理通信。两者都实现了接收客户端数据并在数据末尾添加特定标记后回传的功能。

摘要由CSDN通过智能技术生成

多线程

#include<string.h>	
#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<pthread.h>
#define ERR_MSG(msg) do{\
						printf("line: %d\n",__LINE__);\
						perror(msg);\
						}while(0)
#define IP "192.168.10.111"

void* callback(void* arg);    // arg=(struct climsg)&info
struct climsg
{
	int newswd;
	struct sockaddr_in cin;
};

int main(int argc, const char *argv[])
{
	int swd=socket(AF_INET,SOCK_STREAM,0);
	if(swd<0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//绑定
	struct sockaddr_in sin;
	sin.sin_family= AF_INET;
	sin.sin_port = htons(8888);
	sin.sin_addr.s_addr=inet_addr(IP);

	//允许端口快速被重用
	int reuse =1;
	if(setsockopt(swd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}

	if(bind(swd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//监听
	if(listen(swd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success__%d__\n",__LINE__);

	int newswd=0;
	struct sockaddr_in cin;
	socklen_t len=sizeof(cin);
	struct climsg info; 

	pthread_t bit;
	while(1)
	{	
		//获取新创建的套节字
		newswd=accept(swd,(struct sockaddr*)&cin,&len);
		if(newswd<0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d]客户端连接成功 newswd=%d __%d__\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newswd,__LINE__);
				
			info.newswd=newswd;
			info.cin=cin;

		//创建字线程
		if(pthread_create(&bit,NULL,callback,&info)!=0)//子线程负责接发
		{
			fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
			return -1;
		}
		pthread_detach(bit);
		
	}
	close(swd);
	return 0;
}

void* callback(void* arg)    // arg=(struct climsg)&info
{
	 int newswd=((struct climsg*)arg)->newswd;
	 struct sockaddr_in cin =((struct climsg*)arg)->cin;

	 char buf[120]={0};

	 ssize_t ret=0;
	 while(1)
	 {
		 bzero(buf,sizeof(buf));
		 ret=recv(newswd,buf,sizeof(buf),0);
		 if(ret<0)
		 {
			 ERR_MSG("recv");
			 break;
		 }	else if(0==ret)
		 {
		 	printf("[%s:%d]客户端下线 newfd=%d __%d__\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newswd,__LINE__);
			break;
		 }
		 	printf("[%s:%d newfd=%d  %s  __%d__\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newswd,buf,__LINE__);

			strcat(buf,"*__*");

			if(send(newswd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				break;
			}
			printf("send success\n");
		
	 }
	 close(newswd);
	 pthread_exit(NULL);


}

多进程

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


#define ERR_MSG(msg) do{\
    printf("line: %d\n", __LINE__); \
    perror(msg);\
}while(0)

#define PORT 8888   //端口号的网络字节序 1024~49151
#define IP "192.168.10.111" //ifconfig查看本机IP

int  recvsent(int newfd,struct sockaddr_in cin);
void zombie_handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);

}
int main(int argc, const char *argv[])
{                                                                                     
	if(signal(SIGCHLD,zombie_handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket create success sfd=%d __%d__\n", sfd, __LINE__);

    //功能：允许端口快速被重用，快速被复用
    int reuse =1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }

    //填充服务器自身的地址信息结构体，
    //真实的地址信息结构体根据地址族制定 AF_INET: man 7 ip
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;      //必须填充AF_INET;
    sin.sin_port        = htons(PORT);  //端口号的网络字节序 1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);    //ifconfig查看本机IP

    //将IP地址和端口号绑定到指定的套接字文件描述符上
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);

    //功能：将通讯套接字设置为被动监听状态。用于监听是否有客户端连接;
    //listen函数会让内核维护两个队列：未完成连接的队列， 已完成连接的队列
    if(listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success __%d__\n", __LINE__);

	pid_t fd=0;

	struct sockaddr_in cin;     //存储客户端信息
	socklen_t len = sizeof(cin);
	int newfd=0;
	while(1)

	{

		//从已完成连接的队列里面，获取一个客户端信息，生成一个新的文件描述符，
		//该文件描述符才是与客户端通信的文件描述符。

		newfd = accept(sfd, (struct sockaddr*)&cin, &len);
		if(newfd < 0)
		{
			perror("accept");
			return -1;
		}
		printf("[%s:%d] 客户端连接成功 newfd=%d __%d__\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);

		fd=fork();
		if(fd>0)
		{
			if(close(newfd) < 0)
			{
				ERR_MSG("close");
				return -1;
			}
		
		}

		else if(fd==0)
		{
			if(close(sfd) < 0)
			{
				ERR_MSG("close");
				return -1;
			}
			recvsent(newfd, cin);
			exit(0);
		}

	}

	return 0;
}

int  recvsent(int newfd,struct sockaddr_in cin)
{
	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		bzero(buf, sizeof(buf));
		//接收数据
		// res = recv(newfd, buf, sizeof(buf), 0);
		res = recvfrom(newfd, buf, sizeof(buf), 0,NULL,NULL);
		if(res < 0)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(0 == res)
		{
			printf("[%s:%d] 客户端下线 newfd=%d __%d__\n",\
					inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);
			break;
		}

		printf("[%s:%d] newfd=%d : %s, __%d__\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, buf, __LINE__);

		//发送数据
		strcat(buf, "*_*");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("发送成功\n");
	}
	return 0;
}