【无标题】

使用select实现TCP并发服务

#include <head.h>
#define SER_PORT 8812
#define SER_IP "192.168.125.128"

int main(int argc, const char *argv[])
{
    int sfd = socket(AF_INET,SOCK_STREAM,0);
    if(sfd == -1){
        perror("socket error");
        return -1;
    }
    printf("socket success sfd = %d\n",sfd);

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

    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(SER_PORT);
    sin.sin_addr.s_addr = inet_addr(SER_IP);
    if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
            perror("bind error");
            return -1;
    }
    printf("bind success");

    if(listen(sfd,128) == -1){
        perror("listen error");
        return -1;
    }
    printf("listen success\n");

    struct sockaddr_in cin;
    socklen_t addrlen = sizeof(cin);

    fd_set readfds,tempfds;
    FD_ZERO(&readfds);
    FD_SET(sfd,&readfds);
    FD_SET(0,&readfds);
    int newfd = 0;
    int maxfd = sfd;
    struct sockaddr_in cin_arr[1024];
    while(1){
        tempfds = readfds;
        int res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
        if(res==-1)
        {
            perror("select error");
            return -1;
        }
        else if(res == 0)
        {
            printf("time out\n");
            return -1;
        }
        for(int i = 0;i<=maxfd;i++)
        {
            if(!FD_ISSET(i,&tempfds))
            {
                continue;
            }
            if(i==sfd)
            {
                newfd = accept(i,(struct sockaddr*)&cin,&addrlen);
                if(newfd == -1)
                {
                    perror("accept error");
                    return -1;
                }
                printf("[%s:%d]:enter require\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
                FD_SET(newfd,&readfds);
                if(newfd>=maxfd)
                {
                    maxfd = newfd;
                }
                cin_arr[newfd] = cin;
            }else if(i==0)
            {
                char buf[128]="";
                fgets(buf,sizeof(buf),stdin);
                buf[strlen(buf)-1] = 0;
                printf("enter over:%s\n",buf);
                for(int k=4;k<=maxfd;k++)
                {
                    send(k,buf,strlen(buf),0);
                }
            }else
            {
                char rbuf[128] = "";
                bzero(rbuf,sizeof(rbuf));
                int res = recv(i,rbuf,sizeof(rbuf),0);
                if(res==0)
                {
                    printf("guse over\n");
                    close(i);
                    FD_CLR(i,&readfds);
                    for(int j=maxfd;j>=sfd;j--)
                    {
                        if(FD_ISSET(j,&readfds))
                        {
                            maxfd = j;
                            break;
                        }
                    }
                    continue;
                }
                printf("[%s:%d];%s\n",inet_ntoa(cin_arr[i].sin_addr),ntohs(cin_arr[i].sin_port),rbuf);
                for(int k=4;k<=maxfd;k++)
                {
                    if(k!=i)
                    {
                        send(k,rbuf,strlen(rbuf),0);
                    }
                }
            }
        }
    }
    close(sfd);
    return 0;
}

使用poll完成TCP客户端中发送数据和接收数据的并发执行

#include <head.h>
#define SER_PORT 8812
#define SER_IP "192.168.125.128"
#define CLI_PORT 8823
#define CLI_IP "192.168.125.128"

int main(int argc, const char *argv[])
{
    int cfd = socket(AF_INET,SOCK_STREAM,0);
    if(cfd == -1){
        perror("socket error");
        return -1;
    }
    printf("cfd = %d\n",cfd);

    struct sockaddr_in cin;
    cin.sin_family = AF_INET;
    cin.sin_port = htons(CLI_PORT);
    cin.sin_addr.s_addr = inet_addr(CLI_IP);
    if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin)) == -1){
        perror("bind error");
        return -1;
    }
    printf("bind success\n");
    
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(SER_PORT);
    sin.sin_addr.s_addr = inet_addr(SER_IP);
    if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
        perror("connect error");
        return -1;
    }
    printf("connect success\n");
    struct pollfd pfd[2];
    pfd[0].fd = 0;
    pfd[0].events = POLLIN;

    pfd[1].fd = cfd;
    pfd[1].events = POLLIN;

    char wbuf[128] ="";
    char rbuf[128] ="";
    while(1){
        int res = poll(pfd,2,-1);
        if(res == -1)
        {
            perror("poll error");
            return -1;
        }
        else if(res == 0)
        {
            printf("time out\n");
            return -1;
        }
        if(pfd[0].revents == POLLIN)
        {
            fgets(wbuf,sizeof(wbuf),stdin);
            wbuf[strlen(wbuf)-1] = '\0';
            send(cfd,wbuf,strlen(wbuf),0);
            printf("send succes\n");
        }
        if(pfd[1].revents == POLLIN)
        {
            bzero(rbuf,sizeof(rbuf));
            recv(cfd,rbuf,sizeof(rbuf),0);
            printf("message is:%s\n",rbuf);
        }
    }
    close(cfd);
    return 0;
}