Linux C——基于线程的tco的服务器与客户端

1、服务器程序

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<pthread.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<stdlib.h>
#include<string.h>
#define PORT 2003


void *read_msg(void *arg)
{
    int fd = *((int *)arg);
    int nread = 0;
    char buffer[1024];


    while((nread = read(fd,buffer,sizeof(buffer))) > 0)
    {
        buffer[nread] = '\0';
printf("Message from client:%s\n",buffer);
memset(buffer,0,sizeof(buffer));
    }
}


void *write_msg(void *arg)
{
    char buf2[1024];
    int fd = *((int *)arg);


    while(1)
    {
        fgets(buf2,1024,stdin);
write(fd,buf2,100);
sleep(2);
    }
}


int main()
{
    int sockfd,new_fd;
    struct sockaddr_in server_addr;
    struct sockaddr_in client_addr;
    int r_byte;
    int size;
    char buffer[1024];
    pthread_t id;


    if((sockfd = socket(AF_INET,SOCK_STREAM,0)) < 0)
    {
        printf("socket error!\n");
    }
    
    bzero(&server_addr,sizeof(struct sockaddr_in));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(PORT);
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);


    if((bind(sockfd,(struct sockaddr*)&server_addr,sizeof(struct sockaddr))) < 0)
    {
        printf("bind error!\n");
    }


    if(listen(sockfd,5) < 0)
    {
        printf("listen error!\n");
    }


    while(1)
    {
        size = sizeof(struct sockaddr_in);
printf("accepting...\n");
        if((new_fd = accept(sockfd,(struct sockaddr*)&client_addr,&size)) < 0)
{
   printf("accept error!\n");
}
fprintf(stderr,"Server get connection from %s\n",inet_ntoa(client_addr.sin_addr));


pthread_create(&id,NULL,(void *)read_msg,(void *)&new_fd);
pthread_create(&id,NULL,(void *)write_msg,(void *)&new_fd);
    }
    close(sockfd);
    exit(0);
}

2、客户端程序

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<string.h>
#include<pthread.h>
#define PORT 2003


void *read_msg(void *arg)
{
    int fd = *((int *)arg);
    int nread = 0;
    char buffer[1024];


    while((nread = read(fd,buffer,sizeof(buffer))) > 0)
    {
        buffer[nread] = '\0';
printf("Meaasage from server is:%s\n",buffer);
memset(buffer,0,sizeof(buffer));
    }
}


void *write_msg(void *arg)
{
    char buf2[1024];
    int fd = *((int *)arg);
    while(1)
    {
        fgets(buf2,1024,stdin);
write(fd,buf2,5);
sleep(2);
    }


}
int main()
{ 
    int sockfd;
    struct sockaddr_in server_addr;
    char buffer[1024];


    if((sockfd = socket(AF_INET,SOCK_STREAM,0)) < 0)
    {
        printf("socket error!\n");
    }
    bzero(&server_addr,sizeof(struct sockaddr_in));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(PORT);
    server_addr.sin_addr.s_addr = inet_addr("192.168.1.145");


    if(connect(sockfd,(struct sockaddr*)&server_addr,sizeof(server_addr)) < 0)
    {
        printf("connect error!\n");
    }


    pthread_t id;


    pthread_create(&id,NULL,(void *)read_msg,(int *)&sockfd);


    printf("please input char :");
    while(1)
    {
        //printf("please input char :");
scanf("%s",buffer);
write(sockfd,buffer,strlen(buffer));
    }
    close(sockfd);
    exit(0);
}