src/event/ngx_event_connect.c - nginx-1.7.10

Functions defined

Source code


  1. /*
  2. * Copyright (C) Igor Sysoev
  3. * Copyright (C) Nginx, Inc.
  4. */


  5. #include <ngx_config.h>
  6. #include <ngx_core.h>
  7. #include <ngx_event.h>
  8. #include <ngx_event_connect.h>


  9. ngx_int_t
  10. ngx_event_connect_peer(ngx_peer_connection_t *pc)
  11. {
  12.     int                rc;
  13.     ngx_int_t          event;
  14.     ngx_err_t          err;
  15.     ngx_uint_t         level;
  16.     ngx_socket_t       s;
  17.     ngx_event_t       *rev, *wev;
  18.     ngx_connection_t  *c;

  19.     rc = pc->get(pc, pc->data);
  20.     if (rc != NGX_OK) {
  21.         return rc;
  22.     }

  23.     s = ngx_socket(pc->sockaddr->sa_family, SOCK_STREAM, 0);

  24.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, pc->log, 0, "socket %d", s);

  25.     if (s == (ngx_socket_t) -1) {
  26.         ngx_log_error(NGX_LOG_ALERT, pc->log, ngx_socket_errno,
  27.                       ngx_socket_n " failed");
  28.         return NGX_ERROR;
  29.     }


  30.     c = ngx_get_connection(s, pc->log);

  31.     if (c == NULL) {
  32.         if (ngx_close_socket(s) == -1) {
  33.             ngx_log_error(NGX_LOG_ALERT, pc->log, ngx_socket_errno,
  34.                           ngx_close_socket_n "failed");
  35.         }

  36.         return NGX_ERROR;
  37.     }

  38.     if (pc->rcvbuf) {
  39.         if (setsockopt(s, SOL_SOCKET, SO_RCVBUF,
  40.                        (const void *) &pc->rcvbuf, sizeof(int)) == -1)
  41.         {
  42.             ngx_log_error(NGX_LOG_ALERT, pc->log, ngx_socket_errno,
  43.                           "setsockopt(SO_RCVBUF) failed");
  44.             goto failed;
  45.         }
  46.     }

  47.     if (ngx_nonblocking(s) == -1) {
  48.         ngx_log_error(NGX_LOG_ALERT, pc->log, ngx_socket_errno,
  49.                       ngx_nonblocking_n " failed");

  50.         goto failed;
  51.     }

  52.     if (pc->local) {
  53.         if (bind(s, pc->local->sockaddr, pc->local->socklen) == -1) {
  54.             ngx_log_error(NGX_LOG_CRIT, pc->log, ngx_socket_errno,
  55.                           "bind(%V) failed", &pc->local->name);

  56.             goto failed;
  57.         }
  58.     }

  59.     c->recv = ngx_recv;
  60.     c->send = ngx_send;
  61.     c->recv_chain = ngx_recv_chain;
  62.     c->send_chain = ngx_send_chain;

  63.     c->sendfile = 1;

  64.     c->log_error = pc->log_error;

  65.     if (pc->sockaddr->sa_family == AF_UNIX) {
  66.         c->tcp_nopush = NGX_TCP_NOPUSH_DISABLED;
  67.         c->tcp_nodelay = NGX_TCP_NODELAY_DISABLED;

  68. #if (NGX_SOLARIS)
  69.         /* Solaris's sendfilev() supports AF_NCA, AF_INET, and AF_INET6 */
  70.         c->sendfile = 0;
  71. #endif
  72.     }

  73.     rev = c->read;
  74.     wev = c->write;

  75.     rev->log = pc->log;
  76.     wev->log = pc->log;

  77.     pc->connection = c;

  78.     c->number = ngx_atomic_fetch_add(ngx_connection_counter, 1);

  79.     if (ngx_add_conn) {
  80.         if (ngx_add_conn(c) == NGX_ERROR) {
  81.             goto failed;
  82.         }
  83.     }

  84.     ngx_log_debug3(NGX_LOG_DEBUG_EVENT, pc->log, 0,
  85.                    "connect to %V, fd:%d #%uA", pc->name, s, c->number);

  86.     rc = connect(s, pc->sockaddr, pc->socklen);

  87.     if (rc == -1) {
  88.         err = ngx_socket_errno;


  89.         if (err != NGX_EINPROGRESS
  90. #if (NGX_WIN32)
  91.             /* Winsock returns WSAEWOULDBLOCK (NGX_EAGAIN) */
  92.             && err != NGX_EAGAIN
  93. #endif
  94.             )
  95.         {
  96.             if (err == NGX_ECONNREFUSED
  97. #if (NGX_LINUX)
  98.                 /*
  99.                  * Linux returns EAGAIN instead of ECONNREFUSED
  100.                  * for unix sockets if listen queue is full
  101.                  */
  102.                 || err == NGX_EAGAIN
  103. #endif
  104.                 || err == NGX_ECONNRESET
  105.                 || err == NGX_ENETDOWN
  106.                 || err == NGX_ENETUNREACH
  107.                 || err == NGX_EHOSTDOWN
  108.                 || err == NGX_EHOSTUNREACH)
  109.             {
  110.                 level = NGX_LOG_ERR;

  111.             } else {
  112.                 level = NGX_LOG_CRIT;
  113.             }

  114.             ngx_log_error(level, c->log, err, "connect() to %V failed",
  115.                           pc->name);

  116.             ngx_close_connection(c);
  117.             pc->connection = NULL;

  118.             return NGX_DECLINED;
  119.         }
  120.     }

  121.     if (ngx_add_conn) {
  122.         if (rc == -1) {

  123.             /* NGX_EINPROGRESS */

  124.             return NGX_AGAIN;
  125.         }

  126.         ngx_log_debug0(NGX_LOG_DEBUG_EVENT, pc->log, 0, "connected");

  127.         wev->ready = 1;

  128.         return NGX_OK;
  129.     }

  130.     if (ngx_event_flags & NGX_USE_AIO_EVENT) {

  131.         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, pc->log, ngx_socket_errno,
  132.                        "connect(): %d", rc);

  133.         /* aio, iocp */

  134.         if (ngx_blocking(s) == -1) {
  135.             ngx_log_error(NGX_LOG_ALERT, pc->log, ngx_socket_errno,
  136.                           ngx_blocking_n " failed");
  137.             goto failed;
  138.         }

  139.         /*
  140.          * FreeBSD's aio allows to post an operation on non-connected socket.
  141.          * NT does not support it.
  142.          *
  143.          * TODO: check in Win32, etc. As workaround we can use NGX_ONESHOT_EVENT
  144.          */

  145.         rev->ready = 1;
  146.         wev->ready = 1;

  147.         return NGX_OK;
  148.     }

  149.     if (ngx_event_flags & NGX_USE_CLEAR_EVENT) {

  150.         /* kqueue */

  151.         event = NGX_CLEAR_EVENT;

  152.     } else {

  153.         /* select, poll, /dev/poll */

  154.         event = NGX_LEVEL_EVENT;
  155.     }

  156.     if (ngx_add_event(rev, NGX_READ_EVENT, event) != NGX_OK) {
  157.         goto failed;
  158.     }

  159.     if (rc == -1) {

  160.         /* NGX_EINPROGRESS */

  161.         if (ngx_add_event(wev, NGX_WRITE_EVENT, event) != NGX_OK) {
  162.             goto failed;
  163.         }

  164.         return NGX_AGAIN;
  165.     }

  166.     ngx_log_debug0(NGX_LOG_DEBUG_EVENT, pc->log, 0, "connected");

  167.     wev->ready = 1;

  168.     return NGX_OK;

  169. failed:

  170.     ngx_close_connection(c);
  171.     pc->connection = NULL;

  172.     return NGX_ERROR;
  173. }


  174. ngx_int_t
  175. ngx_event_get_peer(ngx_peer_connection_t *pc, void *data)
  176. {
  177.     return NGX_OK;
  178. }