linux下socket编程改进版

通过在bbs提问，增加一个标志位，解决了线程ID不能重复，灰常感谢。

功能还是一样的，服务器接收消息然后打印，加了些服务器发送的确认消息。

注意点：开始客户端的recv函数阻塞住导致我后面无法发送，后来学习到要用非阻塞模式。

下面是代码，写得有点乱。

服务器端：

//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>

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


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];

int main(){
	int sockfd,confd,len,err,i=0;
    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("Receive 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;
                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;
    struct tm *t;
	struct trans_data *ptr;
	ptr = (struct trans_data *)p;
    send((*ptr).fd,"服务器线程创建成功...",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("send");
            exit(EXIT_FAIL_NUM);
        }
        if(len ==0){
            for(i=0;i<MAX_NUM;i++){
                if(myThread[i].tid == pthread_self()){
                    myThread[i].flag = 0;
                    break;
                }
            }
            pthread_exit(NULL);
        }
        time_t timer;
        timer = time(NULL);
        t = localtime(&timer);
        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);
        puts((*ptr).buff);
	}
}

客户端：

//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>


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

int main(){
	int sockfd,confd,len;
	char buff[MAX_LEN];
	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);//"127.0.0.1");
	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){
        char message[100];
        memset(message,0,sizeof(message));
        int io_block_var = fcntl(sockfd, F_GETFL, 0);
        fcntl(sockfd, F_SETFL, io_block_var|O_NONBLOCK);
        if((len = recv(sockfd,message,sizeof(message),0))>0){
            puts(message);
            if(0 == strcmp(message,"服务器没有空余线程，连接即将关闭......")){
                close(sockfd);
                sleep(3);
                exit(EXIT_FAIL);		
            }
            printf("连接成功！！！\n");
        }
        printf("等待发送......\n");
        gets(buff);
        len = send(sockfd,buff,strlen(buff),0);
        if(len == -1){
            perror("send");
            exit(EXIT_FAIL);
        }
        memset(buff,0,sizeof(buff));
	}
    close(sockfd);
	return 0;
}

运行结果图：


微信扫一扫，订阅我的博客动态^_^