socket线程应用

布满荆棘的人生 2022-07-14 09:14 165阅读 0赞

# 声明：本博文用于学习总结及工作心得 #

**server端：**

#include <stdio.h> #include <stdlib.h> #include <sys/types.h> #include <sys/socket.h> #include <string.h> #include <errno.h> #include <unistd.h> #include <netinet/in.h> #include <arpa/inet.h> #include <pthread.h> //改写inet_ntoa void sockaddr_toa(struct sockaddr_in *addr, char *IPAddr) { unsigned char *p = (unsigned char *)&(addr->sin_addr.s_addr); sprintf(IPAddr, "%u.%u.%u.%u", p[0], p[1], p[2], p[3]); } pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; int status = 0; struct ps { int socket; pthread_t *t; }; //接收客户端的socket数据的线程 void *recvsocket(void *arg) { struct ps *p = (struct ps *)arg; int st = p->socket; pthread_t t = *(p->t); char s[1024]; char *ip = malloc(sizeof(char) *64); //得到远端sockaddr memset(ip, 0, sizeof(ip)); struct sockaddr_in client_addr; socklen_t client_len = sizeof(client_addr); memset(&client_addr, 0, client_len); getpeername(st, (struct sockaddr *)&client_addr, &client_len); sockaddr_toa(&client_addr, ip); printf("远端sockaddr:%s\n", ip); //得到自身的sockaddr memset(ip, 0, sizeof(ip)); struct sockaddr_in server_addr; socklen_t server_len = sizeof(server_addr); memset(&server_addr, 0, server_len); getsockname(st, (struct sockaddr *)&server_addr, &server_len); sockaddr_toa(&server_addr, ip); printf("自身sockaddr:%s\n", ip); while(1) { memset(s, 0, sizeof(s)); int rc = recv(st, s, sizeof(s), 0); if(rc <= 0)//如果recv小于等于0 代表socket已经关闭或者出错了 { break; } else { printf("client:%s\n", s); } } pthread_mutex_lock(&mutex); status--; pthread_mutex_unlock(&mutex); pthread_cancel(t);//被cancel的线程内部没有使用锁 return NULL; } void *sendsocket(void *arg)//向client端socket发送数据 { int st = *((int *)arg); char s[1024]; while(1) { memset(s, 0, sizeof(s)); read(STDIN_FILENO, s, sizeof(s));//读取用户输入信息 send(st, s, strlen(s), 0); } return NULL; } //第二套 int main(int arg, char *args[]) { if (arg < 2) { return -1; } int port = atoi(args[1]); int st = socket(AF_INET, SOCK_STREAM, 0); int on =1; //设置socket属性，第三个参数SO_REUSEADDR代表地址可重用 if(setsockopt(st, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) { printf("setsockopt failed %s\n", strerror(errno)); return EXIT_FAILURE; } struct sockaddr_in addr; memset(&addr, 0, sizeof(addr)); addr.sin_family = AF_INET;//设置结构地址类型为TCP/IP地址 addr.sin_port = htons(port);//指定一个端口号 8900 htons: 将short类型从host字节类型到net字节类型的转化 addr.sin_addr.s_addr = htonl(INADDR_ANY);//INADDR_ANY代表这个server上所有的地址 //将IP与server程序绑定 if(bind(st, (struct sockaddr *)&addr, sizeof(addr)) == -1) { printf("bind failed %s\n", strerror(errno)); return EXIT_FAILURE; } //server开始listen if(listen(st, 20) == -1) { printf("listen failed %s\n", strerror(errno)); return EXIT_FAILURE; } int client_st = 0; //client端socket struct sockaddr_in client_addr;//表示client段的IP地址 pthread_t thrd1, thrd2; struct ps p1, p2; while(1) { memset(&client_addr, 0, sizeof(client_addr)); socklen_t len = sizeof(client_addr); //accept会阻塞，直到有客户端连接过来，accept返回client的socket描述符 // client_st = accept(st, p, &len); client_st = accept(st, (struct sockaddr *)&client_addr, &len); pthread_mutex_lock(&mutex);//为全局变量加一个互斥锁，放在与线程函数同时读写变量的冲突 status ++; pthread_mutex_unlock(&mutex);//解锁 if(status > 1)//当client端连接数大于1时，断开以后的连接 { close(client_st); continue; } if(client_st == -1) { printf("accept failed %s\n", strerror(errno)); return EXIT_FAILURE; } printf("accept by %s\n", inet_ntoa(client_addr.sin_addr)); p1.socket = client_st; p1.t = &thrd1; p2.socket = client_st; p2.t = &thrd2; pthread_create(&thrd1, NULL, recvsocket, (void *)&p1); pthread_detach(thrd1);//设置线程可分离 pthread_create(&thrd2, NULL, sendsocket, (void *)&p2); pthread_detach(thrd2);//设置线程可分离 } close(st);//关闭server端listen的socket return EXIT_SUCCESS; }

**client端:**

#include <stdio.h>
    #include <stdlib.h>
    #include <sys/types.h>
    #include <sys/socket.h>
    #include <string.h>
    #include <errno.h>
    #include <unistd.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <pthread.h>
    
    //接收客户端的socket数据的线程
    void *recvsocket(void *arg)
    {
    	int st = *((int *)arg);
    	char s[1024];
    
    	while(1)
    	{
    		memset(s, 0, sizeof(s));
    		int rc = recv(st, s, sizeof(s), 0);
    		if(rc <= 0)//如果recv小于等于0 代表socket已经关闭或者出错了
    		{
    			break;
    		}
    		else
    		{
    			printf("client:%s\n", s);
    		}
    	}
    	return NULL;
    }
    
    
    void *sendsocket(void *arg)//向client端socket发送数据
    {
    	int st = *((int *)arg);
    	char s[1024];
    	while(1)
    	{
    		memset(s, 0, sizeof(s));
    		read(STDIN_FILENO, s, sizeof(s));//读取用户输入信息
    		send(st, s, strlen(s), 0);
    	}
    	return NULL;
    }
    //第二套
    int main(int arg, char *args[])
    {
        if (arg < 3) {
            return -1;
        }
        
        char *ipv4 = args[1];
        int port = atoi(args[2]);
    	int st = socket(AF_INET, SOCK_STREAM, 0);//初始化socket
    	if(st == -1)
    	{
    		printf("create socket error:%s\n", strerror(errno));
    		return -1;
    	}
    	struct sockaddr_in addr;//定义一个IP地址的结构
    	memset(&addr, 0 , sizeof(addr));
    	addr.sin_family = AF_INET;//设置结构地址类型为TCP/IP地址
    	addr.sin_port = htons(port);//指定一个端口号 8900 htons: 将short类型从host字节类型到net字节类型的转化
    	addr.sin_addr.s_addr = inet_addr(ipv4);//将字符串类型的IP地址转化为int，赋给addr结构成员
    
    	//调用connect连接到结构addr指定的IP地址和端口号
    	if(connect(st, (struct sockaddr *)&addr, sizeof(addr)) == -1)
    	{
    		printf("connect failed %s\n", strerror(errno));
    		return -1;
    	}
    	pthread_t thrd1, thrd2;
    	pthread_create(&thrd1, NULL, recvsocket, (void *)&st);
    	pthread_create(&thrd2, NULL, sendsocket, (void *)&st);
    	pthread_join(thrd1, NULL);
    	//pthread_join(thrd2, NULL);
    
    	close(st);//关闭socket
    
    
    	return EXIT_SUCCESS;
    }

makefile:

.SUFFIXES:.c .o
    
    CC=gcc
    SRCS1=socket.c
    SRCS2=server.c
    OBJS1=$(SRCS1:.c=.o)
    OBJS2=$(SRCS2:.c=.o)
    EXEC1=socket
    EXEC2=server
    start: $(OBJS1) $(OBJS2)
    	$(CC) -o $(EXEC1) $(OBJS1) -lpthread
    	$(CC) -o $(EXEC2) $(OBJS2) -lpthread
    	@echo '-----------OK-----------'
    .c.o:
    	$(CC) -Wall -g -o $@ -c $<
    clean:
    	rm -rf $(OBJS1) $(ONJS2)