2019独角兽企业重金招聘Python工程师标准>>>

所谓的回射是指:客户端A向服务端B发送数据,服务端B接收到数据之后,再将接收到的数据发送回客户端B。所谓的迭代服务器,是指服务器端只用一个进程处理或线程处理所有客户端的请求。与之对应的是并发服务器,并发服务器是指对于每一一个客户端的请求,服务端都分配一个进程或是线程独立来处理客户端的处理。下面介绍使用select函数实现TCP回射迭代服务。直接上代码:

服务端程序:

/*=============================================================================
#     FileName: tcpservselect.c
#         Desc: receive client data and then send they back.
#       Author: Licaibiao
#   LastChange: 2017-02-12 
=============================================================================*/
#include<stdio.h>  
#include<sys/types.h>  
#include<sys/socket.h>  
#include<unistd.h>  
#include<stdlib.h>  
#include<errno.h>  
#include<arpa/inet.h>  
#include<netinet/in.h>  
#include<string.h>  
#include<signal.h>
#define MAXLINE      1024
#define LISTENLEN 10
#define SERV_PORT 6666
 
int main(int argc, char **argv)
{
    int                    i, maxi, maxfd, listenfd, connfd, sockfd;
    int                    nready, client[FD_SETSIZE];
    ssize_t                n;
    fd_set                rset, allset;
    char                buf[MAXLINE];
    socklen_t            clilen;
    struct sockaddr_in    cliaddr, servaddr;
 
    listenfd = socket(AF_INET, SOCK_STREAM, 0);
 
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family      = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port        = htons(SERV_PORT);
 
    bind(listenfd, (struct sockaddr*) &servaddr, sizeof(servaddr));
 
    listen(listenfd, LISTENLEN);
 
    maxfd = listenfd;            /* initialize */
    maxi = -1;                    /* index into client[] array */
    for (i = 0; i < FD_SETSIZE; i++)
        client[i] = -1;            /* -1 indicates available entry */
    FD_ZERO(&allset);
    FD_SET(listenfd, &allset);
 
    for ( ; ; ) 
    {
        rset = allset;        /* structure assignment */
        nready = select(maxfd+1, &rset, NULL, NULL, NULL);
 
        if (FD_ISSET(listenfd, &rset)) /* new client connection */
        {    
            clilen = sizeof(cliaddr);
            connfd = accept(listenfd, (struct sockaddr*) &cliaddr, &clilen);
#ifdef    NOTDEF
            printf("new client: %s, port %d\n",
                    inet_ntop(AF_INET, &cliaddr.sin_addr, 4, NULL),
                    ntohs(cliaddr.sin_port));
#endif
 
            for (i = 0; i < FD_SETSIZE; i++)
                if (client[i] < 0) {
                    client[i] = connfd;    /* save descriptor */
                    break;
                }
            if (i == FD_SETSIZE)
            {
                printf("too many clients");
                exit(0);
            }
 
            FD_SET(connfd, &allset);    /* add new descriptor to set */
            if (connfd > maxfd)
                maxfd = connfd;            /* for select */
            if (i > maxi)
                maxi = i;                /* max index in client[] array */
 
            if (--nready <= 0)
                continue;                /* no more readable descriptors */
        }
 
        for (i = 0; i <= maxi; i++)     /* check all clients for data */
        {    
            if ( (sockfd = client[i]) < 0)
                continue;
            if (FD_ISSET(sockfd, &rset)) 
            {
                if ( (n = read(sockfd, buf, MAXLINE)) == 0)/* connection closed by client */ 
                {
                    close(sockfd);
                    FD_CLR(sockfd, &allset);
                    client[i] = -1;
                } else
                    write(sockfd, buf, n);
 
                if (--nready <= 0)
                    break;                /* no more readable descriptors */
            }
        }
    }
}
 
在服务端的程序中,我们使用select 来处理任意个客户的单进程程序,而不是派生一个子程序。
客户端程序:

/*=============================================================================
#     FileName: tcpcliselect.c
#         Desc: send data to server and receive data from server
#       Author: Licaibiao
#   LastChange: 2017-02-12 
=============================================================================*/
#include<stdio.h>  
#include<sys/types.h>  
#include<sys/socket.h>  
#include<unistd.h>  
#include<stdlib.h>  
#include<errno.h>  
#include<arpa/inet.h>  
#include<netinet/in.h>  
#include<string.h>  
#include<signal.h>
#define MAXLINE      1024
#define LISTENLEN 10
#define SERV_PORT 6666
 
int max(int a, int b)
{
    return a>b ? a : b;
}
 
void str_cli(FILE *fp, int sockfd)
{
    int            maxfdp1, stdineof;
    fd_set        rset;
    char        buf[MAXLINE];
    int        n;
 
    stdineof = 0;
    FD_ZERO(&rset);
    for ( ; ; ) 
    {
        if (stdineof == 0)
            FD_SET(fileno(fp), &rset);
        FD_SET(sockfd, &rset);
        maxfdp1 = max(fileno(fp), sockfd) + 1;
        select(maxfdp1, &rset, NULL, NULL, NULL);
 
        if (FD_ISSET(sockfd, &rset)) 
        {    
            if ( (n = read(sockfd, buf, MAXLINE)) == 0) /* socket is readable */
            {
                if (stdineof == 1)
                    return;        /* normal termination */
                else
                    printf("str_cli: server terminated prematurely");
            }
            write(fileno(stdout), buf, n);
        }
 
        if (FD_ISSET(fileno(fp), &rset))  /* input is readable */
        {  
            if ( (n = read(fileno(fp), buf, MAXLINE)) == 0) 
            {
                stdineof = 1;
                shutdown(sockfd, SHUT_WR);    /* send FIN */
                FD_CLR(fileno(fp), &rset);
                continue;
            }
 
            write(sockfd, buf, n);
        }
    }
}
 
int main(int argc, char **argv)
{
    int    sockfd;
    struct sockaddr_in    servaddr;
 
    if (argc != 2)
    {
        printf("usage: tcpcli <IPaddress>");
        exit(0);
    }
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
 
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
 
    connect(sockfd, (struct sockaddr*) &servaddr, sizeof(servaddr));
 
    str_cli(stdin, sockfd);        /* do it all */
 
    exit(0);
}
 
 
    在第60行我们使用了shutdown函数。我们知道,TCP是全双工工作,在我们做批量输入批量输出的时候,我们客户端已经把数据发送完毕,这个时候并不能直接关闭描述符,因为可能还有数据在从服务端发送回来的路上。close函数是直接终止读和写两个方向的数据传送。但是使用shutdown可以单方向关闭数据传输。
    在客户端,我们使用Ctrl + d 来结束客户端程序。Ctrl + d 会发送一个exit 。在TCP传输中,如果对端TCP发送一个FIN(finish 对端进程终止),那么该套接字变为可读,并且read返回0(EOF)

运行结果:

运行服务端程序

root@ubuntu:/home/share/test# ./strserselect
另外一个终端运行客户端程序:
root@ubuntu:/home/share/test# ./strcliselect 127.0.0.1
china                        /*发送*/
china                        /*接收*/

注意:上面的程序存在一个问题,如果有一个恶意客户端只发送一个字节数据(不是换行符)后进入睡眠,服务器调用read读入一个字节,后面就阻塞在read函数以等待其他的数据,这样一来服务端就阻塞在一个客户端,不能再处理其他客户端的请求(拒绝服务型攻击)
解决上面问题有下面的几种方法:

(a)使用非阻塞式IO

(b)对IO操作设置一个超时

(c)让每个客户由单独的控制线程提供服务

--------------------- 
作者:li_wen01 
来源:CSDN 
原文:https://blog.csdn.net/li_wen01/article/details/55004918 
版权声明:本文为博主原创文章,转载请附上博文链接!

转载于:https://my.oschina.net/u/4000302/blog/3044645

Linux网络编程:使用select函数实现socket 收发数据相关推荐

  1. linux 网络编程:使用两线程实现socket同时收发数据

    http://blog.csdn.net/li_wen01/article/details/52665505 工作中最近有使用到socket 向客户端同时发送和接收数据,因为是嵌入式linux设备,且 ...

  2. Linux 网络编程详解二(socket创建流程、多进程版)

    netstat -na | grep "8080" --查看TCP/IP协议连接状态 //socket编程提高版--服务器 #include <stdio.h> #in ...

  3. linux网络编程:splice函数和tee( )函数高效的零拷贝

    splice( )函数 在两个文件描述符之间移动数据,同sendfile( )函数一样,也是零拷贝.  函数原型: #include <fcntl.h> ssize_t splice(in ...

  4. linux网络编程中listen函数 backlog的含义

    结论: backlog 是用来指定在TCP连接时,同时进行 3次握手建立连接的客户端数量 listen函数在一般在调用bind之后-调用accept之前调用, 它的函数原型是: #include< ...

  5. linux网络编程系列-select和epoll的区别

    select和epoll属于I/O多路复用模型,用于持续监听多个socket,获取其IO事件. select(轮询) 该模型轮询各socket,不管socket是否活跃,随着socket数的增加,性能 ...

  6. linux网络编程--数据结构与函数原型

    套接字有三种类型:流式套接字(SOCK_STREAM),数据报套接字(SOCK_DGRAM)及原始套接字. socket()    |  bind()    |                     ...

  7. Linux网络编程中出现 listen fail : Socket operation on non-socket错误

    错误代码: int main (int argc,char *argv[]) {int lfd = 0,cfd = 0;/* 定义服务器地址结构 和 客户端地址结构*/struct sockaddr_ ...

  8. 【Linux网络编程】并发服务器之select模型

    00. 目录 文章目录 00. 目录 01. 概述 02. I/O复用技术概述 03. select模型服务器实现思路 04. select模型服务器实现 05. 附录 01. 概述 服务器设计技术有 ...

  9. Linux网络编程---I/O复用模型之select

    https://blog.csdn.net/men_wen/article/details/53456435 Linux网络编程-I/O复用模型之select 1. IO复用模型 IO复用能够预先告知 ...

  10. linux网络编程(三)select、poll和epoll

    linux网络编程(三)select.poll和epoll 一.为什么会有多路I/O转接服务器? 二.select 三.poll 三.epoll 一.为什么会有多路I/O转接服务器? 为什么会有多路I ...

最新文章

  1. uwsgi: invalid option -- 'x'
  2. CentOS重启启动Apache,VNC
  3. miniconda安装,及channels配置,安装其他软件
  4. python打印异常
  5. Android Activity动画属性简介
  6. 谈谈HashMap线程不安全的体现
  7. php redis并发读写,PHP使用Redis实现防止大并发下二次写入的方法
  8. Python导入全局、局部模块以及如何让避免循环导入
  9. 以云战“疫”,这次阿里云又让人们惊了……
  10. spring学习--AOP术语
  11. java类的生命周期
  12. 面向对象-java控制台计算器简单实现[50行]
  13. numpy array和python list_Python list与NumPy array 区分详解
  14. 【机器学习Machine Learning】资料大全
  15. Python实现多线程并发下载大文件(断点续传支持)
  16. STC12C5A60S2-定时器+数码管
  17. mysql 导入 没有数据库文件怎么打开_没有SQL Server数据库时如何打开.MDF文件
  18. hdf heg 批量拼接_MODIS处理工具MRT已被HEG代替
  19. 关系抽取调研——工业界
  20. 【高级篇】详解Zookeeper客户端Curator

热门文章

  1. 242.有效的字母异位词
  2. python学习——和python的第一次亲密接触
  3. 2篇word文档比较重复率_【软件】PDF转word黑科技 快来get!
  4. POST 和GET传输的最大容量分别是多少?
  5. 【NLP】揭秘马尔可夫模型神秘面纱系列文章(五)
  6. HBASE自带小工具,统计表的行数
  7. 【机器学习】主成分分析 (PCA)、无监督特征提取
  8. 常系数齐次线性微分方程的解法
  9. java添加录音,大家说说java程序里怎么插入视频和音频啊
  10. 使用ppmap检测和利用XSS漏洞