实现socket非阻塞设置

刚开始学习，难免有些疏漏之处，请各位多多指教。

程序亲测可以使用。废话不多说。

网络IO模型分为阻塞IO、非阻塞IO、异步IO、IO复用、信号驱动IO。

阻塞IO： 有数据时，直接读；没有数据时，等待读。

非阻塞IO: 无论内核有没有数据都读。

以下是程序：

1. serevr.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <sys/types.h>    
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/stat.h>
#include <fcntl.h>



#define PORT 4455
#define MAX_LEN 5

//1. 初始化socket  -  tcp
int init_socket()
{
    //1-1. socket
    int sfd = socket(AF_INET,SOCK_STREAM,0);
    if(sfd == -1)
    {
        perror("socket");
        exit(1);
    }
    //1-2 REUSEADDR
    int val = 1;
    int ret = setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,(void *)&val,sizeof(val));
    if(ret == -1)
    {
        perror("setsockopt");
        exit(1);
    }
    // 1-3 bind
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;
    saddr.sin_port = htons(PORT); 
    ret = bind(sfd,(struct sockaddr *)&saddr,sizeof(saddr));
    if(ret == -1)
    {
        perror("bind");
        exit(1);
    }
    // 1-4 listen
    ret = listen(sfd,MAX_LEN);
    if( ret == -1)
    {
        perror("listen");
        exit(1);
    }
    
    printf("Srever is ready ...\n");
    return sfd;
}
//2. 处理接收，设置为非阻塞型的
int deal_accept(int sfd)
{
    // 2-1 accept
    int cfd = accept(sfd,NULL,NULL);
    if(cfd == -1)
    {
        perror("accept");
        exit(1);
    }
    // 2-2 获得cfd属性
    int flags = fcntl(cfd,F_GETFL);
    if(flags == -1)
    {
        perror("fcntl F_GETFL");
        exit(1);
    }
    // 2-2 判断flags是否具有O_NONBLOCK属性
    if(!(flags&O_NONBLOCK))
    {
        flags |= O_NONBLOCK;
        int ret = fcntl(cfd,F_SETFL,flags);
        if(ret == -1)
        {
            perror("fcntl F_SETFL");
            exit(1);
        }
    }
    return cfd;
}

void deal_communication(int cfd)
{
    char buf[64] = {'\0'};
    while(1)
    {
        memset(buf,'\0',sizeof(buf));
        recv(cfd,buf,sizeof(buf),0);
        printf("client say : %s\n",buf);
        if(strncmp(buf,"quit",4) == 0)  break;
        sleep(3);
    }
    return ;
}

int main()
{
    //1. 初始化socket  -  tcp
    int sfd = init_socket();
    //2. 处理接收，设置为非阻塞型的
    int cfd = deal_accept(sfd);
    //3. 通信
    deal_communication(cfd);
    return 0;
}

2. client.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <sys/types.h>    
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/stat.h>
#include <fcntl.h>


#define IP "127.0.0.1"
#define PORT 4455
//1. 初始化socket
int init_socket()
{
    // 1-1 socket
    int sfd = socket(AF_INET,SOCK_STREAM,0);
    if(-1 == sfd )
    {
        perror("socket");
        exit(1);
    }
    // 1-2 connect
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = inet_addr(IP);
    saddr.sin_port = htons(PORT);
    int  ret = connect(sfd,(struct sockaddr *)&saddr,sizeof(saddr));
    if(-1 == ret )
    {
        perror("connect");
        exit(1);
    }
    
    return sfd;
}

void deal_communication(int sfd)
{
    char buf[64] = {'\0'};
    while(1)
    {
        memset(buf,'\0',sizeof(buf));
        printf("请输入:");
        scanf("%s",buf);
        send(sfd,buf,strlen(buf),0);
        printf("client say :%s\n",buf);
        if(strncmp("quit",buf,4) == 0)
        break;        
    }
    return ;
}

int main()
{
    //1. 初始化socket
    int sfd = init_socket();
    //2. 通信
    deal_communication(sfd);
    return 0;
}