linux下socket编程再改进版

最新推荐文章于 2024-08-08 16:07:11 发布

安菲尔德的门柱

最新推荐文章于 2024-08-08 16:07:11 发布

阅读量568

点赞数

分类专栏： C/C++学习

本文链接：https://blog.csdn.net/gerrard0616/article/details/13837159

版权

C/C++学习专栏收录该内容

8 篇文章 0 订阅

订阅专栏

基于上次的改进版，实现了服务器的转发功能，基本实现了局域网聊天室的功能。

此版改进之处：

1、给client端开了一个线程，recv()函数在线程中单独处理，这样又可不必使用非阻塞的recv()了。

2、给服务器端成功accept()产生的confd加入一个链表中，当收到消息的时候，将转发标志flag置为1，转发的时候遍历链表，和接收到消息的confd不同的就转发。

3、每当一个客户端退出时，收回线程id的同时，将此次连接的confd从链表中删除。

下面是代码：

//linklist.c
#include<stdio.h>
#include<stdlib.h>

typedef int elemType;

typedef struct listnode{
	elemType info;
	struct listnode * next;
}listnode;

listnode * linkListInit(listnode * head){
	head = NULL;
	return head;
}

listnode * addNode(listnode *head,elemType value){
	listnode *ptr = (listnode *)malloc(sizeof(listnode));
	ptr->info = value;
	if(head == NULL){
		ptr->next = NULL;
	}
	else{
		ptr->next = head;
	}
	head = ptr;
	return head;
}

listnode * findNode(listnode * head,elemType value){
	listnode *ptr = head;
	if(NULL == head)//0 elem
		return NULL;
	else if(NULL == ptr->next){ //1 elem
		if(value == ptr->info)
			return ptr;
		else 
			return NULL;
	}
	while(ptr != NULL){
		if(value == ptr->info)
			return ptr;
		ptr = ptr->next;

	}
	return NULL;
}

listnode * delNode(listnode *head,elemType value){
	listnode *elemPtr = findNode(head,value);
	listnode *ptr = head;
	if(NULL == elemPtr)
		printf("Cannot find value: %d in list.\n",value);
	else{
		if(elemPtr == head ){
			if(head->next ==NULL)
				head = NULL;
			else{
				head = head->next;
			}
		}
		else if(elemPtr->next == NULL){
			while(elemPtr != ptr->next){
				ptr = ptr->next;
			}
			ptr->next = NULL;
		}
		else{
			while(ptr->next != elemPtr){
				ptr = ptr->next;
			}
			ptr->next = elemPtr->next;
		}
		free(elemPtr);
	}		
	return head;
}

void printList(listnode *head){
	if(head == NULL)
		return ;
	listnode *ptr = head;
	while(ptr != NULL){
		printf("%d -> ",ptr->info);
		ptr = ptr->next;
	}
	printf("\n");
}

//server.c
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/socket.h>
#include<sys/types.h>
#include<error.h>
#include<errno.h>
#include<arpa/inet.h>
#include<pthread.h>
#include<time.h>
#include"linklist.c"

#define MAX_LEN 100
#define PORT_NUM 4321
#define EXIT_FAIL_NUM 1
#define MAX_NUM 3

void trans(void *p);

struct trans_data{
	int fd;
	char buff[MAX_LEN];
	struct sockaddr_in info;
}p[MAX_NUM];

struct myTherad{
    pthread_t tid;
    int flag;
}myThread[MAX_NUM];

listnode *head;

int main(){
	int sockfd,confd,len,err,i=0;
	head=linkListInit(head);
    for(i=0;i<MAX_NUM;i++){
        memset(&myThread[i].flag,0,sizeof(int));
        printf("thread[%d] flag %d\n",i,myThread[i].flag);
    }
	struct sockaddr_in ser,cli;
	if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0){
		perror("socket");
		exit(EXIT_FAIL_NUM);
	}
	printf("scoket_fd = %d\n",sockfd);
	bzero(&ser,sizeof(ser));
	ser.sin_family = AF_INET;
	ser.sin_port = htons(PORT_NUM);
	ser.sin_addr.s_addr = htonl(INADDR_ANY);
	if(-1 == bind(sockfd,(struct sockaddr *)&ser,sizeof(ser))){
		perror("bind");
		exit(EXIT_FAIL_NUM);
	}
	if(0 != listen(sockfd,10)){
		perror("listen");
		exit(EXIT_FAIL_NUM);
	}
	printf("Receiving message......\n");
    while(1){ 
	    len = sizeof(cli);
	    confd = accept(sockfd,(struct sockaddr *)&cli,&len);
	    if(-1 == confd){
		    perror("accept");
		    exit(EXIT_FAIL_NUM);
	    }
        for(i=0;i<MAX_NUM;i++){ 
            if(myThread[i].flag == 0){
		        p[i].fd = confd;
		        memcpy(&p[i].info,&cli,sizeof(p[i].info));
		        err = pthread_create(&myThread[i].tid,NULL,(void *)&trans,(void *)&p[i]);
		        if(0 != err){
			        printf("Can not create thread!\n");
			        exit(EXIT_FAIL_NUM);
		        }
                myThread[i].flag = 1;
		printf("confd:%d\n",confd);
		head = addNode(head,confd);
//		printList(head);
                break;   
            }      
	    }
        if(MAX_NUM  <= i){
            char message[100];
            memset(message,0,sizeof(message));
            strcpy(message,"服务器没有空余线程，连接即将关闭......");
            if( -1 ==send(confd,message,sizeof(message),0)){
                perror("send");
                exit(EXIT_FAIL_NUM);
            }
        }
    }
	return 0;
}


void trans(void *p){
	int i;
	int flag=0;
    	struct tm *t;
	struct trans_data *ptr;
	ptr = (struct trans_data *)p;
	printf("线程创建成功，线程号:%lu\n",pthread_self());
    	send((*ptr).fd,"服务器线程创建成功...\n连接成功！！！",100,0);
	while(1){
		memset((*ptr).buff,0,sizeof((*ptr).buff));
		int len;
		if((len = recv((*ptr).fd,(*ptr).buff,sizeof((*ptr).buff),0))<0){
            		perror("recv");
            		exit(EXIT_FAIL_NUM);
        }
        if(len ==0){
            for(i=0;i<MAX_NUM;i++){
                if(myThread[i].tid == pthread_self()){
                    myThread[i].flag = 0;
		    printf("confd:%d\n",(*ptr).fd);
		    head = delNode(head,(*ptr).fd);
//		    printList(head);
		    printf("IP:%s\t端口号：%d 断开连接.\n",inet_ntoa((struct in_addr)(*ptr).info.sin_addr),(*ptr).info.sin_port);
		    printf("线程:%lu 被收回\n",myThread[i].tid);
                    break;
                }
            }
            pthread_exit(NULL);
        }
        time_t timer;
        timer = time(NULL);
        t = localtime(&timer);
	flag = 1;
		printf("消息来自:\nIP:%s\t",inet_ntoa((struct in_addr )(*ptr).info.sin_addr));
		printf("端口号:%d\t",(*ptr).info.sin_port);
        printf("(%d:%d:%d)\n",(*t).tm_hour,(*t).tm_min,(*t).tm_sec);
		printf("confd:%d\t线程号:%lu\n",(*ptr).fd,pthread_self());
		puts((*ptr).buff);
		char message[1024];
		strcat(message,"\n来自IP:");
		strcat(message,inet_ntoa((struct in_addr)(*ptr).info.sin_addr));
		strcat(message,"\tPort:");
		char m[10];
		sprintf(m,"%d",(*ptr).info.sin_port);
		strcat(message,m);
		strcat(message,"\n消息：");
		strcat(message,(*ptr).buff);
		listnode *s = head;
		int l;
//		printList(head);
		while(flag){
			while(s !=NULL){
				if(s->info !=(*ptr).fd){
					if((l=send(s->info,message,sizeof(message),0))<0){
						perror("send");
					}
				}
				s = s->next;
			}
			flag = 0;
		}
	}
}

//client.c
#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<sys/types.h>
#include<netinet/in.h>
#include<unistd.h>
#include<string.h>
#include<error.h>
#include<errno.h>
#include<fcntl.h>
#include<pthread.h>

#define MAX_LEN 100
#define EXIT_FAIL 1
#define SERV_IP "127.0.0.1"
#define PORT_NUM 4321


void doRecv(void *p){
	char message[1024];
	int len;
	int *s = p;
	while(1){
		memset(message,0,sizeof(message));	
//		int io_block_var = fcntl(*s, F_GETFL, 0);
//	        fcntl(*s, F_SETFL, io_block_var|O_NONBLOCK);
        	if((len = recv(*s,message,sizeof(message),0))>0){
			puts(message);
	        	if(0 == strcmp(message,"服务器没有空余线程，连接即将关闭......")){
	            		close(*s);
               			sleep(3);
		        	exit(EXIT_FAIL);		
	        }
        }
		
	}
}

int main(){
	int sockfd,confd,len;
	char buff[MAX_LEN];
	int *p;
	struct sockaddr_in cli,ser;
	if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1){
		perror("socket");
		exit(EXIT_FAIL);
	}
	printf("socket_fd = %d\n",sockfd);
	bzero(&ser,sizeof(struct sockaddr_in));
	ser.sin_family = AF_INET;
	ser.sin_addr.s_addr = inet_addr(SERV_IP);
	ser.sin_port = htons(PORT_NUM);
	if((confd = connect(sockfd,(struct sockaddr *)&ser,sizeof(struct sockaddr))) == -1){
		perror("connect");
		exit(EXIT_FAIL);
	}
	printf("正在连接......\n");
	sleep(2);
	while(1){
	pthread_t pid;
	int err;
	p = &sockfd;
	err = pthread_create(&pid,NULL,(void *)&doRecv,(void *)p);
        if(err != 0){
		printf("Can not create thread!!!\n");
		exit(EXIT_FAIL);
	}
	gets(buff);
//	fgets(buff,sizeof(buff),stdin);
	fflush(stdin);
		len = send(sockfd,buff,strlen(buff),0);
		if(len == -1){
			perror("send");
			exit(EXIT_FAIL);
		}
		memset(buff,0,sizeof(buff));
	}
    close(sockfd);
	return 0;
}

运行截图