2022-8-15 TCP 和 UDP的域套接字模型

只能做本地通讯的套接字文件描述符，地址族： AF_UNIX ；

首先是TCP的多进程域套接字模型

#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 <unistd.h>
#include <signal.h>
#include <sys/wait.h>
#include <stdlib.h>
#include<sys/un.h>


//打印错误新的宏函数
#define ERR_MSG(msg)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 6666 				//1024~49151
#define IP 	"192.168.31.63" 	//本机IP,用ifconfig查看


typedef void (*sighandler_t)(int);

int rcv_cli_msg(int newfd,struct sockaddr_un cun);

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)
	{
		ERR_MSG("signal");
		return -1;
	}


	//创建流式套接字
	int sfd = socket(AF_UNIX, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success\n");

	//判断要绑定的文件是否存在
	//如果存在，则删除该文件
	if(access("./unix",F_OK) == 0)
	{
		//删除文件
		if(unlink("./unix") < 0 )
		{
			ERR_MSG("unlink");
			return -1;
		}
	
	}
	
	/*	//允许端口快速重用
	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_un sun;
	sun.sun_family 		= AF_UNIX;
	strcpy(sun.sun_path,"./unix");
	//	sin.sin_port 		= htons(PORT); 	//网络字节序的端口号
	//	sin.sin_addr.s_addr = inet_addr(IP); 	//网络字节序的IP地址


	//将地址信息结构体绑定到套接字上
	if(bind(sfd, (struct sockaddr*)&sun, sizeof(sun)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
//	printf(" %d   ,%s",sun.sun_family,sun.sun_path);

	//将套接字设置为被动监听状态，让内核去监听是否有客户端连接;
	if(listen(sfd, 10) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	struct sockaddr_un cun;
	socklen_t addrlen = sizeof(cun);
//	printf(" %d   ,%s\n",cun.sun_family,cun.sun_path);
	//从已完成连接的队列头中，取出一个客户端的信息，创建生成一个新的套接字文件描述符，
	//该文件描述符才是与客户端通信的文件描述符!!!
	int newfd =0;
	pid_t pid =0;
	//	int newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
	while(1)
	{
		//父进程负责连接
		newfd =accept(sfd,(struct sockaddr*)&cun,&addrlen);
		if(newfd < 0)
		{
			perror("accept");
			return -1;
		}

		//网络字节序的IP-->点分十进制  网络字节序的port--->本机字节序
		printf("newfd = %d\n",newfd);

		printf(" %d   ,%s\n",cun.sun_family,cun.sun_path);
		//一旦连接成功，则需要创建一个子进程用于与客户端进行交互
		pid=fork();
		if(pid >0)
		{
			//父进程进行
			close(newfd);
		}
		else if(0 == pid)
		{
			close(sfd);
			//子进程运行，子进程负责与客户端交互

			rcv_cli_msg(newfd,cun);
			close(newfd);
			//子进程只负责交互，所以不能让子进程继续往后运行，需要直接退出
			exit(0);
		
		}
		else
		{
			ERR_MSG("fork");
			return -1;
		}

	}
		close(sfd);
//		close(newfd);
		return 0;
	

}

下面的是UDP的域套接字模型

#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<sys/un.h>

//打印错误新的宏函数
#define ERR_MSG(msg)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg);\
}while(0)


int main(int argc, const char *argv[])
{
	/*	if(argc < 3)
		{
		fprintf(stderr, "请输入IP port\n");
		return -1;
		}*/
	//将获取到的端口号字符串，转换成整形
	/*	int port = atoi(argv[2]);
		if(port < 1024 || port > 49151)
		{
		fprintf(stderr, "port %d input error!! 1024~49151\n", port);
		return -1;
		}*/

	//创建报式套接字
	int sfd = socket(AF_UNIX, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//判断要绑定的文件是否存在                 
	//如果存在，则删除                         
	if(access("./udp_unix",F_OK) == 0) //文件存在   
	{                                          
		//删除文件                             
		if(unlink("./udp_unix") < 0 )               
		{                                      
			ERR_MSG("unlink");                 
			return -1;                         
		}                                      
	}                                          

	//填充服务器的IP地址以及端口号
	struct sockaddr_un sun;
	sun.sun_family 		= AF_UNIX;
	strcpy(sun.sun_path,"udp_unix");
	//	sin.sin_port 		= htons(port);
	//	sin.sin_addr.s_addr = inet_addr(argv[1]);


	//绑定服务器的地址信息结构体
	if(bind(sfd, (struct sockaddr*)&sun, sizeof(sun)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	struct sockaddr_un cun; 	//存储接收到的数据包来自哪里
	socklen_t addrlen = sizeof(cun);

	char buf[128] = "";
	while(1)
	{
		bzero(buf, sizeof(buf));
		//接收
		if(recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&cun, &addrlen) < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		printf("[%s:%s\n",cun.sun_path, buf);

		//发送
		strcat(buf,"*_*");
		//谁发给我，我发给谁
		if(sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cun,sizeof(cun)) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf("sendto success\n");
	}
	//关闭套接字
	close(sfd);
	return 0;
}