Linux Network- Muti_thread and Multi_process TCP server

最新推荐文章于 2024-05-17 16:41:20 发布

SatoshiNakamoto007

最新推荐文章于 2024-05-17 16:41:20 发布

阅读量46

点赞数

文章标签： linux c语言网络 tcp/ip

本文链接：https://blog.csdn.net/YamamotoIsoroku/article/details/129050436

版权

1.多进程

01_multi_process_tcp_server.c

#include <stdio.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <wait.h>
#include <stdlib.h>

#define IP "192.168.8.249"
#define PORT 8888

int deal_with_client(int newsockfd,struct sockaddr_in cin);

#define ERR_MSG(msg) do\
{\
   fprintf(stderr,"line:__%d__",__LINE__);\
   perror(msg);\
} while (0);

//回收僵尸进程
void handler(int sig)
{
    while (waitpid(-1,NULL,WNOHANG)>0);
}

int main(int argc, char const *argv[])
{
    //捕获子进程退出信号
    __sighandler_t s = signal(SIGCHLD,handler);
    if (SIG_ERR==s)
    {
        ERR_MSG("signal");
        return -1;
    }
    



    //1.socket
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if (sockfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    
    //2.bind
    //bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen);
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;;
    sin.sin_port = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);
    if (bind(sockfd,(struct sockaddr *)&sin,sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success\n");

    //3.listen
    if (listen(sockfd,5) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success\n");

    //4.accept
    // int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);
    pid_t cpid = 0;
    while (1)
    {
        //父进程只用来接收请求
        int newsockfd = accept(sockfd,(struct sockaddr *)&cin,&addrlen);
        if ( newsockfd < 0)
        {
            ERR_MSG("accept");
            return -1;
        }
        printf("accept success,the incoming ip==>%s,port==>%d\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port));

        cpid = fork();
        if (cpid > 0)
        {
            //关闭newsockfd
            close(newsockfd);
            
        } else if(0 == cpid)
        {
            //sockfd 关闭
            close(sockfd);
            deal_with_client(newsockfd,cin);
            //子进程必须退出，否则继续循环
            exit(0);
        } else
        {
            ERR_MSG("fork");
            break;
        }
        
    }
    close(sockfd);
    return 0;
}



int deal_with_client(int newsockfd,struct sockaddr_in cin)
{
    socklen_t addrlen = sizeof(cin);
    char buf[128] = {0};
    ssize_t ret = 0;

    while (1)
    {
        //接收之前先清0
        bzero(buf,sizeof(buf));
        //5.recv 
        //ssize_t recv(int sockfd, void *buf, size_t len, int flags);
        // recv(newsockfd, buf, sizeof(buf), 0) ==> read(newsockfd, buf, sizeof(buf))
        // recv(newsockfd, buf, sizeof(buf), 0) ==> recvfrom(newsockfd, buf, sizeof(buf), 0,NULL,NULL)
        ret = recv(newsockfd, buf, sizeof(buf), 0);
        if (ret < 0)
        {
            ERR_MSG("recv");
            break;
        } else if (0 == ret)
        {
            fprintf(stdout,"The remost host has shut down,ip==>%s,port==>%d\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port));
            break;
        }
        printf("message received from %s:%d is %s\n",inet_ntoa(cin.sin_addr),\
        htons(cin.sin_port),buf);
        
        //6.send
        strcat(buf,"_Satoshi_Nakamoto\n");
        //ssize_t send(int sockfd, const void *buf, size_t len, int flags);
        //send(newsockfd,buf,sizeof(buf),0) = write(newsockfd,buf,sizeof(buf),0)
        //send(newsockfd,buf,sizeof(buf),0) =  sendto(newsockfd,buf,sizeof(buf),0, NULL, 0);
        ret = send(newsockfd,buf,sizeof(buf),0);
        if (ret < 0)
        {
            ERR_MSG("write");
            break;
        }
        printf("%ld bytes sent successfully\n",ret);
        
    }
    close(newsockfd);
    return 0;
}

2.多线程

01_multi_thread_tcp_server.c

#include <stdio.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <wait.h>
#include <stdlib.h>
#include <pthread.h>

typedef struct _info
{
    int newsockfd;
    struct sockaddr_in cin;

} INFO;

#define IP "192.168.8.249"
#define PORT 8888

void * callback (void *arg);

int deal_with_client(int newsockfd,struct sockaddr_in cin);

#define ERR_MSG(msg) do\
{\
   fprintf(stderr,"line:__%d__",__LINE__);\
   perror(msg);\
} while (0);


int main(int argc, char const *argv[])
{
    //1.socket
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if (sockfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    
    //2.bind
    //bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen);
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;;
    sin.sin_port = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);
    if (bind(sockfd,(struct sockaddr *)&sin,sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success\n");

    //3.listen
    if (listen(sockfd,5) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success\n");

    //4.accept
    // int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);
    pid_t cpid = 0;

    pthread_t tid;

    while (1)
    {
        //主进程只用来接收请求
        int newsockfd = accept(sockfd,(struct sockaddr *)&cin,&addrlen);
        if ( newsockfd < 0)
        {
            ERR_MSG("accept");
            return -1;
        }
        printf("accept success,the incoming ip==>%s,port==>%d\n", inet_ntoa(cin.sin_addr),htons(cin.sin_port));
        INFO info;
        info.cin = cin;
        info.newsockfd = newsockfd;
        //主线程接收到请求后创建子线程进行处理
        if (pthread_create(&tid,NULL,callback,(void *)&info)!=0)
        {
            ERR_MSG("pthread_create");
            break;
        }
    }
    pthread_detach(tid);
    close(sockfd);
    return 0;
}



void * callback (void *arg)
{
    INFO *info = (INFO *)arg;
    int newsockfd = info->newsockfd;
    struct sockaddr_in cin = info->cin;
    socklen_t addrlen = sizeof(cin);
    char buf[128] = {0};
    ssize_t ret = 0;

    while (1)
    {
        //接收之前先清0
        bzero(buf,sizeof(buf));
        //5.recv 
        //ssize_t recv(int sockfd, void *buf, size_t len, int flags);
        // recv(newsockfd, buf, sizeof(buf), 0) ==> read(newsockfd, buf, sizeof(buf))
        // recv(newsockfd, buf, sizeof(buf), 0) ==> recvfrom(newsockfd, buf, sizeof(buf), 0,NULL,NULL)
        ret = recv(newsockfd, buf, sizeof(buf), 0);
        if (ret < 0)
        {
            ERR_MSG("recv");
            break;
        } else if (0 == ret)
        {
            fprintf(stdout,"The remost host has shut down,ip==>%s,port==>%d\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port));
            break;
        }
        printf("message received from %s:%d is %s\n",inet_ntoa(cin.sin_addr),\
        htons(cin.sin_port),buf);
        
        //6.send
        strcat(buf,"_Satoshi_Nakamoto\n");
        //ssize_t send(int sockfd, const void *buf, size_t len, int flags);
        //send(newsockfd,buf,sizeof(buf),0) = write(newsockfd,buf,sizeof(buf),0)
        //send(newsockfd,buf,sizeof(buf),0) =  sendto(newsockfd,buf,sizeof(buf),0, NULL, 0);
        ret = send(newsockfd,buf,sizeof(buf),0);
        if (ret < 0)
        {
            ERR_MSG("write");
            break;
        }
        printf("%ld bytes sent successfully\n",ret);
        
    }
    close(newsockfd);
    pthread_exit(NULL);
}