6.21作业

最新推荐文章于 2024-09-13 16:27:15 发布

m0_68526641

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量248

点赞数 1

文章标签：算法

本文链接：https://blog.csdn.net/m0_68526641/article/details/139907782

版权

服务器

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


#define PORT 6666
#define IP   "192.168.1.4"
#define errormsg(msg)   do{\
                        fprintf(stderr,"line%d",__LINE__);\
                        perror(msg);\
                        }while(0)
                                                                                                                                                                                                                            
int deal_keyboard_msg(fd_set readfds);
int deal_cliRecSend(int i ,struct sockaddr_in savecin[], fd_set* preadfds,int *pmaxfd);
int deal_cliconnect(int sfd, struct sockaddr_in psavecin[],fd_set *preadfds,int *pmaxfd);
int main(int argc, const char *argv[])
{
    //创建流失套接字
    int sfd=socket(AF_INET,SOCK_STREAM,0);
    if(sfd<0)
    {
        errormsg("socket");
        return -1;
    }
    printf("流式套接字创建成功 sfd=%d\n",sfd);

    //允许端口快速被复用
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        errormsg("setsockopt");
        return -1;
    }
    printf("允许端口快速重用\n");
    //填充服务器地址信息结构体，给bing函数用
    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)
    {
        errormsg("bind");
        return -1;
    }
    printf("绑定服务器地址信息成功\n");


    //套接字设置为监听状态
    if(listen(sfd,128)<0)
    {
        errormsg("listen");
        return -1;
    }
    printf("监听状态设置成功\n");

    fd_set readfds ,tempfds;
    FD_ZERO(&readfds);

    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);

    int maxfd = sfd;

    //获取客户端信息

    int s_res;
    int newfd = -1;

    struct sockaddr_in savecin[1024];

    while(1)
    {
        tempfds=readfds;

        s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);

        if(s_res<0)
        {
            errormsg("select");
            return -1;
        }
        if(s_res==0)
        {
            printf("time out\n");
            break;
        }

        
        for(int i=0;i<=maxfd;i++)
        {
            if(FD_ISSET(i, &tempfds)==0)
            {
                continue;
            }
            if(0==i)
            {
                printf("键盘输入\n");
                deal_keyboard_msg(readfds);
            }
            else if(sfd==i)
            {
                deal_cliconnect(sfd,savecin,&readfds,&maxfd);
            }
            else
            {

                deal_cliRecSend(i,savecin,&readfds,&maxfd);
            }
        }
    }
    if(close(sfd)<0)
    {
        errormsg("close");
        return -1;
    }


    return 0;
}

//-------------------------------------处理客户端交互事件----------------
int deal_cliRecSend(int i ,struct sockaddr_in savecin[], fd_set* preadfds,int *pmaxfd)
{
    char buf[128]="";
    ssize_t res ;

    bzero(buf,sizeof(buf));
    res=recv(i,buf,sizeof(buf),0);
    if(res<0)
    {
        errormsg("recv");
        return -1;
    }
    else if(0==res)
    {
        printf("[%s:%d] newfd=%d客户端关闭\n",\
                inet_ntoa(savecin[i].sin_addr),ntohs(savecin[i].sin_port),i);
        close(i);
        FD_CLR(i,preadfds);
        return 0;
    }
    printf("[%s:%d] newfd=%d\t%s\n",\
            inet_ntoa(savecin[i].sin_addr),ntohs(savecin[i].sin_port),i,buf);
    //发送
    strcat(buf,"*_*");
    if(send(i,buf,sizeof(buf),0)<0)
    {
        errormsg("send");
        return -1;
    }
    printf("发送成功\n");
    return 0;
}

//-------------------------------------客户端连接-------------------------
int deal_cliconnect(int sfd, struct sockaddr_in psavecin[],fd_set *preadfds,int *pmaxfd)
{
    struct sockaddr_in cin;
    socklen_t addrlen=sizeof(cin);
    printf("有客户端连接\n");
    int newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
    if(newfd<0)
    {
        errormsg("accept");
        return -1;
    }
    printf("[%s:%d] newfd=%d客户端链接成功\n",\
            inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
    psavecin[newfd] = cin;

    FD_SET(newfd,preadfds);

    *pmaxfd = *pmaxfd>newfd?*pmaxfd:newfd;
    return 0;
}



//-------------------------------------键盘输入---------------------------
int deal_keyboard_msg(fd_set readfds)
{
    int sndfd;
    char buf[128]="";
    int res=scanf("%d %s",&sndfd, buf);
    while(getchar()!='\n');
    if(res!=2)
    {
        printf("请输入正确格式  fd  string\n");
        return -1;
    }
    if(sndfd<3 || FD_ISSET(sndfd,&readfds)==0)
    {
        printf("请输入正确范围\n");
        return -1;
    }
    if(send(sndfd,buf,sizeof(buf),0)<0)
    {
        errormsg("send");
        return -1;
    }
    printf("发送成功\n");
    return 0;
}

客户端

 #include <stdio.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <unistd.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
 #include <string.h>
 #include <stdlib.h>
  #include <sys/select.h>
 #include <sys/time.h>
 
 
 #define errormsg(msg)   do{\
                         fprintf(stderr,"line%d",__LINE__);\
                         perror(msg);\
                         }while(0)
 
 int main(int argc, const char *argv[])
 {
 
     //创建流失套接字
     int cfd=socket(AF_INET,SOCK_STREAM,0);
     if(cfd<0)
     {
         errormsg("socket");
         return -1;
     }
     printf("流式套接字创建成功 cfd=%d\n",cfd);
 
     //允许端口快速被复用
     int reuse = 1;
     if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
     {
         errormsg("setsockopt");
         return -1;
     }
     printf("允许端口快速重用\n");
     //填充服务器地址信息结构
     int PORT=atoi(argv[1]);
     struct sockaddr_in sin ;
     sin.sin_family          =AF_INET;
     sin.sin_port            =htons(PORT);
     sin.sin_addr.s_addr     =inet_addr(argv[2]);
     //链接服务器
     if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
     {
         errormsg("connect");
         return -1;
     }
     printf("连接到服务器[%s:%d]\n",argv[2],PORT);
 
     fd_set readfds ,tempfds;
     FD_ZERO(&readfds);
 
     FD_SET(0,&readfds);
     FD_SET(cfd,&readfds);
     int s_res;
                                                                                                                               
 
     char buf[128]="";
     ssize_t res;
     while(1)
     {
         tempfds=readfds;
 
         s_res=select(cfd+1,&tempfds,NULL,NULL,NULL);
 
         if(s_res<0)
         {
             errormsg("select");
             return -1;
         }
         if(s_res==0)
         {
             printf("time out\n");
             break;
         }
 
         if(FD_ISSET(0, &tempfds))
         {
 
 
         bzero(buf,sizeof(buf));
         printf("请输入>>>\n");
         fgets(buf,sizeof(buf),stdin);
         buf[strlen(buf)-1]='\0';
         if(send(cfd,buf,sizeof(buf),0)<0)
         {
             errormsg("send");
             return -1;
         }
         printf("发送成功\n");
         }
         //接收
         if(FD_ISSET(cfd, &tempfds))
         {
         bzero(buf,sizeof(buf));
         res=recv(cfd,buf,sizeof(buf),0);
         if(res<0)
         {
             errormsg("recv");
             return -1;
         }
         else if(0==res)
         {
             printf("[%s:%d] cfd=%d客户端关闭\n",\
                     inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),cfd);
             break;
         }
         printf("[%s:%d] cfd=%d\t%s\n",\
                 inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),cfd,buf);
         }
 
     }
     if(close(cfd)<0)
     {
         errormsg("close");
         return -1;
     }
 
     return 0;
 }