一个TCP非阻塞client端简单的例子
如果我们要产生一个非阻塞的socket,在C语言中如下代码所示:
// 创建socket
intsock_fd = socket(AF_INET,SOCK_STREAM,0); ...
// 更改socket为nonblock
fcntl(sock_fd,F_SETFL,fdflags | O_NONBLOCK); // connect .... while(1) {
intrecvlen = recv(sock_fd,recvbuf,RECV_BUF_SIZE); ...... } ...
由于网络协议非常复杂,内核里面用到了大量的面向对象的技巧,所以我们从创建连接开始,一步一步追述到最后代码的调用点。 socket的创建
很明显,内核的第一步应该是通过AF_INET、SOCK_STREAM以及最后一个参数0定位到需要创建一个TCP的socket,如下图绿线所示:
我们跟踪源码调用
socket(AF_INET,SOCK_STREAM,0) |->sys_socket进入系统调用 |->sock_create |->__sock_create
进一步分析__sock_create的代码判断: conststructnet_proto_family *pf;
// RCU(Read-Copy Update)是linux的一种内核同步方法,在此不阐述 // family=INET
pf = rcu_dereference(net_families[family]); err = pf->create(net,sock,protocol);
const struct net_proto_family *pf; // RCU(Read-Copy Update)是linux的一种内核同步方法,在此不阐述 // family=INET pf = rcu_dereference(net_families[family]); err = pf->create(net, sock, protocol); 则
通
过
源
码
可
知
,
由
于
是
AF_INET(PF_INET),
所
以
net_families[PF_INET].create=inet_create(以后我们都用PF_INET表示),即pf->create = inet_create; 进一步追溯调用:
inet_create(structnet *net,structsocket *sock,intprotocol){ Sock* sock; ......
// 此处是寻找对应协议处理器的过程 lookup_protocol:
// 迭代寻找protocol==answer->protocol的情况 list_for_each_rcu(p,
/* Check the non-wild match. */ if(protocol == answer->protocol){ if(protocol != IPPROTO_IP) break; } ......
// 这边answer指的是SOCK_STREAM sock->ops = answer->ops;
answer_no_check = answer->no_check; // 这边sk->prot就是answer_prot=>tcp_prot
sk = sk_alloc(net,PF_INET,GFP_KERNEL,answer_prot); sock_init_data(sock,sk); ...... }
上面的代码就是在INET中寻找SOCK_STREAM的过程了 我们再看一下inetsw[SOCK_STREAM]的具体配置: staticstructinet_protosw inetsw_array[] = { {
.type = SOCK_STREAM, .protocol = IPPROTO_TCP,
.prot = \.no_check = 0,
.flags = INET_PROTOSW_PERMANENT | INET_PROTOSW_ICSK, }, ...... }
这边也用了重载,AF_INET有TCP、UDP以及Raw三种:
从上述代码,我们可以清楚的发现sock->ops= conststructproto_ops inet_stream_ops = { .family = PF_INET, .owner = THIS_MODULE, ......
.sendmsg = tcp_sendmsg,
.recvmsg = sock_common_recvmsg, ...... }
即sock->ops->recvmsg = sock_common_recvmsg; 同时sock->sk->sk_prot = tcp_prot;
我们再看下tcp_prot中的各个函数重载的定义: structproto tcp_prot = { .name = \.close = tcp_close,