3.15 作业

head.h

#ifndef __HEAD_H__
#define __HEAD_H__
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <pthread.h>

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

#define ERR_MSG_1(msg) ERR_MSG(msg);return -1

#define PORT 2222
#define IP "192.168.1.33"
#endif

多进程并发服务器

#include "head.h"

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[])
{
    __sighandler_t s=signal(SIGCHLD,zombie_callback);
    if (SIG_ERR==s)
    {
        ERR_MSG_1("signal");
    }

    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if (sfd<0)
    {
        ERR_MSG_1("socket");
    }

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

    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_1("bind");
    }

    if (listen(sfd,128)<0)
    {
        ERR_MSG_1("listen");
    }

    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_1("accept");
        }

        printf("[%s:%d]客户端连接成功\n",\
                inet_ntoa(cin.sin_addr),\
                ntohs(cin.sin_port));

        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_1("fork");
        }
    }

    close(sfd);
    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_1("recv");
        }
        else if(0==res)
        {
            fprintf(stderr,"[%s:%d]客户端下线\n",\
                    inet_ntoa(cin.sin_addr),\
                    ntohs(cin.sin_port));
            break;
        }
        printf("[%s:%d]:%s\n",
                inet_ntoa(cin.sin_addr),\
                ntohs(cin.sin_port),\
                buf);
        strcat(buf,"*_*");
        if(send(newfd,buf,sizeof(buf),0)<0)
        {
            ERR_MSG_1("send");
        }
    }
}

多线程并发服务器

#include "head.h"

struct cli_msg
{
    int newfd;
    struct sockaddr_in cin;
};

void* deal_cli_msg(void *arg);

int main(int argc, const char *argv[])
{
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if (sfd<0)
    {
        ERR_MSG_1("socket");
    }

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

    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_1("bind");
    }

    if (listen(sfd,128)<0)
    {
        ERR_MSG_1("listen");
    }

    struct sockaddr_in cin;
    socklen_t addrlen=sizeof(cin);
    int newfd=-1;
    pthread_t pid;

    while (1)
    {
        newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
        if (newfd<0)
        {
            ERR_MSG_1("accept");
        }

        printf("[%s:%d]客户端连接成功\n",\
                inet_ntoa(cin.sin_addr),\
                ntohs(cin.sin_port));

        struct cli_msg info;
        info.newfd=newfd;
        info.cin=cin;

        if (pthread_create(&pid,NULL,deal_cli_msg,&info)!=0)
        {
            fprintf(stderr,"pthread_create error");
            return -1;
        }

        pthread_detach(pid);

    }

    close(sfd);
    return 0;
}


void* deal_cli_msg(void *arg)
{
    struct sockaddr_in cin=((struct cli_msg*)arg)->cin;
    int newfd=((struct cli_msg*)arg)->newfd;
    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");
            break;
        }
        else if(0==res)
        {
            fprintf(stderr,"[%s:%d]客户端下线\n",\
                    inet_ntoa(cin.sin_addr),\
                    ntohs(cin.sin_port));
            break;
        }
        printf("[%s:%d]:%s\n",
                inet_ntoa(cin.sin_addr),\
                ntohs(cin.sin_port),\
                buf);
        strcat(buf,"*_*");
        if(send(newfd,buf,sizeof(buf),0)<0)
        {
            ERR_MSG("send");
            break;
        }
    }
    close(newfd);
    pthread_exit(NULL);
}