3.15作业

多进程并发服务器

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

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

#define PORT 6666               
#define IP  "192.168.31.123"    

int deal_cli_msg(int newfd, struct sockaddr_in cin);

void zombie_callBack(int sig)
{
    while(waitpid(-1, NULL, WNOHANG) > 0);
}

int main(int argc, const char *argv[])
{
    //捕获17号信号
    __sighandler_t s=signal(SIGCHLD,zombie_callBack);
    if(SIG_ERR==s)
    {
        ERR_MSG("signal");
        return -1;
    }
    //创建流式套接字
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if(sfd<0)
    {
        ERR_MSG("socket");
        return -1;
    }

    int reuse=1;
    if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }

    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;         
    sin.sin_port        = htons(PORT);  
    sin.sin_addr.s_addr = inet_addr(IP);  

    //绑定
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);
    
    //将套接字设置为被动监听状态
    if(listen(sfd,128)<0)
    {
        ERR_MSG("listen");
        return -1;
    }

    struct sockaddr_in cin;
    socklen_t addrlen=sizeof(cin);
    int newfd=-1;
    pid_t cpid=0;

    while(1)
    {
        newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
        if(newfd<0)
        {
            ERR_MSG("accept");
            return -1;
        }
        printf("[%s : %d] newfd=%d 客户端连接成功\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);
        cpid=fork();
         if(0 == cpid)       //子进程运行
        {
            close(sfd);        

            deal_cli_msg(newfd, cin);

            close(newfd);
            exit(0);            //退出子进程
        }
        else if(cpid > 0)
        {
            close(newfd);      
        }
        else
        {
            ERR_MSG("fork");
            return -1;
        }
    }
    close(sfd);
    close(newfd);
    return 0;
}
int deal_cli_msg(int newfd, struct sockaddr_in cin)
{
    char buf[128] = "";
    ssize_t res = -1;
    while(1)
    {
        bzero(buf, sizeof(buf));
        //接收
        res = recv(newfd, buf, sizeof(buf), 0);
        if(res < 0)
        {
            ERR_MSG("recv");
            return -1;
        }
        else if(0 == res)
        {
            fprintf(stderr, "[%s : %d] newfd=%d 客户端下线\n", \
                    inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);
            break;
        }
        printf("[%s : %d] newfd=%d : %s\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, buf);

        //发送
        strcat(buf, "*_*");
        if(send(newfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            return -1;
        }
        printf("send success\n");
    }
    return 0;
}

多线程并发服务器

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

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

#define PORT 6666             
#define IP  "127.0.0.1"  

int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0); 
    if(sfd < 0)
    {   
        ERR_MSG("socket");
        return -1; 
    }   
    printf("socket create success  sfd = %d\n", sfd);

    //设置允许端口快速被重用
    int resue = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &resue, sizeof(resue)) < 0)
    {   
        ERR_MSG("setsockopt");
        return -1; 
    }   

    //填充服务器的地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;    
    sin.sin_port        = htons(PORT);  
    sin.sin_addr.s_addr = inet_addr(IP);   

    //绑定
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {   
        ERR_MSG("bind");
        return -1; 
    }   
    printf("bind success __%d__\n", __LINE__);

    //将套接字设置为被动监听状态
    if(listen(sfd, 128) < 0)
    {   
        ERR_MSG("listen");
        return -1; 
    }   
    printf("listen success __%d__\n", __LINE__);
    struct sockaddr_in cin;    
    socklen_t addrlen = sizeof(cin);
    int newfd = -1; 
    
    pthread_t tid;
    struct cli_msg info;

    while(1)
    {
        newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
        if(newfd<0)
        {
            ERR_MSG("accept");
            retuen -1;
        }
        printf("[%s : %d] newfd=%d 客户端连接成功\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);
        info.newfd=newfd;
        info.cin=cin;

        if(pthread_create(&tid,NULL,deal_cil_msg,&info)!=0)
        {
            fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
            return -1;
        }
        pthread_detach(tid);
    }
    close(sfd);
    return 0;
}
void* deal_cli_msg(void* arg)   //void* arg = (void*)&info
{
    int newfd = ((struct cli_msg*)arg)->newfd;
    struct sockaddr_in cin = ((struct cli_msg*)arg)->cin;

    char buf[128] = "";
    ssize_t res = -1;
    
    while(1)
    {
        bzero(buf,sizeof(buf));
        res=recv(newfd,buf,sizeof(buf),0);
        if(res<0)
        {
            ERR_MSG("recv");
            return -1;
        }
        else if(0==res)
        {
            fprintf(stderr,"[%s : %d] newfd=%d 客户端下线\n",\
                    inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
            break;
        }
        printf("[%s : %d] newfd=%d : %s\n", \
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, buf);
        strcat(buf, "*_*");
        if(send(newfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            break;
        }
        printf("send success\n");
    }   
    close(newfd);
    
    pthread_exit(NULL);
    }
}