gen_tcp async accept大致流程

erlang 调用 gen_tcp:accept时是会阻塞的，包括后续的gen_tcp:recv也是，但是这个阻塞实际是在erlang这边 receive等待driver返回消息，并不是阻塞在driver上，driver是不能阻塞的，这个mryufeng老大也很早就跟我说过,当时没明 白，现在终于理解了

看下erlang 这边是如何做的

Java代码  收藏代码
  1. accept0(L, Time) when is_port(L), is_integer(Time) ->  
  2.     case async_accept(L, Time) of  
  3.     {ok, Ref} ->  
  4.         receive   
  5.         {inet_async, L, Ref, {ok,S}} ->  
  6.             accept_opts(L, S);  
  7.         {inet_async, L, Ref, Error} ->  
  8.             Error  
  9.         end;  
  10.     Error -> Error  
  11.     end.  
  12.   
  13. async_accept(L, Time) ->  
  14.     case ctl_cmd(L,?TCP_REQ_ACCEPT, [enc_time(Time)]) of  
  15.     {ok, [R1,R0]} -> {ok, ?u16(R1,R0)};  
  16.     Error -> Error  
  17.     end.  


可以看到prim_inet调用 ctl_cmd实际是通过port_control调用driver,这个是会立即返回的，但只是返回这个调用的成功/失败状态，而实际的accept后的结果是通过receive得到的，格式：
{inet_async, L, Ref, {ok,S}}   S就是conn fd对应的port

那么driver里是怎么处理的呢，其实想让accept不阻塞肯定是使用的nonblocking + event notify 机制，也就是select/epoll之类的，看下

tcp_inet_ctl函数里的TCP_REQ_ACCEPT分支

C代码  收藏代码
  1. if (desc->inet.state == TCP_STATE_ACCEPTING) {  
  2.     //这个还没弄明白  
  3. } else if (desc->inet.state == TCP_STATE_MULTI_ACCEPTING) {  
  4.     ...  
  5. }else{  
  6.     ...  
  7. }  


进入这个分支首先是对当前连接状态的一个判断，accept之前的状态是LISTEN，所以肯定执行的是else部分代码

C代码  收藏代码
  1. s = sock_accept(desc->inet.s, (struct sockaddr*) &remote, &n);  
  2.     if (s == INVALID_SOCKET) {  
  3.     if (sock_errno() == ERRNO_BLOCK) {  
  4.         ErlDrvMonitor monitor;  
  5.         if (driver_monitor_process(desc->inet.port, driver_caller(desc->inet.port),  
  6.                        &monitor) != 0) {   
  7.         return ctl_xerror("noproc", rbuf, rsize);  
  8.         }  
  9.         enq_async_w_tmo(INETP(desc), tbuf, TCP_REQ_ACCEPT, timeout, &monitor);  
  10.         desc->inet.state = TCP_STATE_ACCEPTING;  
  11.         sock_select(INETP(desc),FD_ACCEPT,1);  
  12.         if (timeout != INET_INFINITY) {  
  13.         driver_set_timer(desc->inet.port, timeout);  
  14.         }  
  15.     } else {  
  16.         return ctl_error(sock_errno(), rbuf, rsize);  
  17.     }  
  18.     } else {  
  19.            }  


这段代码首先调用了sock_accept，这个封装的宏也是为了跨平台的通用，其实linux上就是accept,由于之前创建socket时 已经设置描述字为非阻塞了，所以这个会立即返回-1，就是INVALID_SOCKET,并且errno全局变量的值是 11,EWOULDBLOCK,sock_errno也是为了通用
后面的driver_monitor_process是监控调用者erlang进程，然后enq_async_w_tmo实际是保存当前异步请求 到一个队列中，核心的是调用sock_select这个函数，并且根据timeout设置了定时器，看下sock_select实现,实际最终调用的是 erl_driver.h中定义的

C代码  收藏代码
  1. int driver_select(ErlDrvPort port, ErlDrvEvent event, int mode, int on)  


实际就是注册当前关心的事件 socket/pipe 读写等事件，由emulator通过 epoll/select 进行event notify,回调的是driver里定义的ready_input/ready_output函数，这里对应的就是

C代码  收藏代码
  1. static int tcp_inet_input(tcp_descriptor* desc, HANDLE event)  


这个函数无非就是调用accept，然后返回连接描述字，最终通过tcp_inet_copy里的

C代码  收藏代码
  1. /* The new port will be linked and connected to the original caller */  
  2.    port = driver_create_port(port, owner, "tcp_inet", (ErlDrvData) copy_desc);  


创建了一个新的port返回给erlang端调用，处理后续连接描述字的读写事件

看完这些能对gen_tcp driver异步工作的机制有个大概了解，细节实现还有非常多的地方需要研究，erl_driver.h的实现回头也需要一起看一下

原文：http://www.iteye.com/topic/482507

posted on 2012-12-05 11:14  应无所住而生其心  阅读(678)  评论(0)    收藏  举报

导航