src/event/modules/ngx_eventport_module.c - nginx source code

Global variables defined

Data types defined

Functions defined

Macros 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. #if (NGX_TEST_BUILD_EVENTPORT)

  9. #define ushort_t  u_short
  10. #define uint_t    u_int

  11. #ifndef CLOCK_REALTIME
  12. #define CLOCK_REALTIME          0
  13. typedef int     clockid_t;
  14. typedef voidtimer_t;
  15. #elif (NGX_DARWIN)
  16. typedef voidtimer_t;
  17. #endif

  18. /* Solaris declarations */

  19. #define PORT_SOURCE_AIO         1
  20. #define PORT_SOURCE_TIMER       2
  21. #define PORT_SOURCE_USER        3
  22. #define PORT_SOURCE_FD          4
  23. #define PORT_SOURCE_ALERT       5
  24. #define PORT_SOURCE_MQ          6

  25. #ifndef ETIME
  26. #define ETIME                   64
  27. #endif

  28. #define SIGEV_PORT              4

  29. typedef struct {
  30.     int         portev_events;  /* event data is source specific */
  31.     ushort_t    portev_source;  /* event source */
  32.     ushort_t    portev_pad;     /* port internal use */
  33.     uintptr_t   portev_object;  /* source specific object */
  34.     void       *portev_user;    /* user cookie */
  35. } port_event_t;

  36. typedef struct  port_notify {
  37.     int         portnfy_port;   /* bind request(s) to port */
  38.     void       *portnfy_user;   /* user defined */
  39. } port_notify_t;

  40. #if (__FreeBSD__ && __FreeBSD_version < 700005) || (NGX_DARWIN)

  41. typedef struct itimerspec {     /* definition per POSIX.4 */
  42.     struct timespec it_interval;/* timer period */
  43.     struct timespec it_value;   /* timer expiration */
  44. } itimerspec_t;

  45. #endif

  46. int port_create(void);

  47. int port_create(void)
  48. {
  49.     return -1;
  50. }


  51. int port_associate(int port, int source, uintptr_t object, int events,
  52.     void *user);

  53. int port_associate(int port, int source, uintptr_t object, int events,
  54.     void *user)
  55. {
  56.     return -1;
  57. }


  58. int port_dissociate(int port, int source, uintptr_t object);

  59. int port_dissociate(int port, int source, uintptr_t object)
  60. {
  61.     return -1;
  62. }


  63. int port_getn(int port, port_event_t list[], uint_t max, uint_t *nget,
  64.     struct timespec *timeout);

  65. int port_getn(int port, port_event_t list[], uint_t max, uint_t *nget,
  66.     struct timespec *timeout)
  67. {
  68.     return -1;
  69. }

  70. int port_send(int port, int events, void *user);

  71. int port_send(int port, int events, void *user)
  72. {
  73.     return -1;
  74. }


  75. int timer_create(clockid_t clock_id, struct sigevent *evp, timer_t *timerid);

  76. int timer_create(clockid_t clock_id, struct sigevent *evp, timer_t *timerid)
  77. {
  78.     return -1;
  79. }


  80. int timer_settime(timer_t timerid, int flags, const struct itimerspec *value,
  81.     struct itimerspec *ovalue);

  82. int timer_settime(timer_t timerid, int flags, const struct itimerspec *value,
  83.     struct itimerspec *ovalue)
  84. {
  85.     return -1;
  86. }


  87. int timer_delete(timer_t timerid);

  88. int timer_delete(timer_t timerid)
  89. {
  90.     return -1;
  91. }

  92. #endif


  93. typedef struct {
  94.     ngx_uint_t  events;
  95. } ngx_eventport_conf_t;


  96. static ngx_int_t ngx_eventport_init(ngx_cycle_t *cycle, ngx_msec_t timer);
  97. static void ngx_eventport_done(ngx_cycle_t *cycle);
  98. static ngx_int_t ngx_eventport_add_event(ngx_event_t *ev, ngx_int_t event,
  99.     ngx_uint_t flags);
  100. static ngx_int_t ngx_eventport_del_event(ngx_event_t *ev, ngx_int_t event,
  101.     ngx_uint_t flags);
  102. static ngx_int_t ngx_eventport_notify(ngx_event_handler_pt handler);
  103. static ngx_int_t ngx_eventport_process_events(ngx_cycle_t *cycle,
  104.     ngx_msec_t timer, ngx_uint_t flags);

  105. static void *ngx_eventport_create_conf(ngx_cycle_t *cycle);
  106. static char *ngx_eventport_init_conf(ngx_cycle_t *cycle, void *conf);

  107. static int            ep = -1;
  108. static port_event_t  *event_list;
  109. static ngx_uint_t     nevents;
  110. static timer_t        event_timer = (timer_t) -1;
  111. static ngx_event_t    notify_event;

  112. static ngx_str_t      eventport_name = ngx_string("eventport");


  113. static ngx_command_t  ngx_eventport_commands[] = {

  114.     { ngx_string("eventport_events"),
  115.       NGX_EVENT_CONF|NGX_CONF_TAKE1,
  116.       ngx_conf_set_num_slot,
  117.       0,
  118.       offsetof(ngx_eventport_conf_t, events),
  119.       NULL },

  120.       ngx_null_command
  121. };


  122. static ngx_event_module_t  ngx_eventport_module_ctx = {
  123.     &eventport_name,
  124.     ngx_eventport_create_conf,             /* create configuration */
  125.     ngx_eventport_init_conf,               /* init configuration */

  126.     {
  127.         ngx_eventport_add_event,           /* add an event */
  128.         ngx_eventport_del_event,           /* delete an event */
  129.         ngx_eventport_add_event,           /* enable an event */
  130.         ngx_eventport_del_event,           /* disable an event */
  131.         NULL,                              /* add an connection */
  132.         NULL,                              /* delete an connection */
  133.         ngx_eventport_notify,              /* trigger a notify */
  134.         ngx_eventport_process_events,      /* process the events */
  135.         ngx_eventport_init,                /* init the events */
  136.         ngx_eventport_done,                /* done the events */
  137.     }

  138. };

  139. ngx_module_t  ngx_eventport_module = {
  140.     NGX_MODULE_V1,
  141.     &ngx_eventport_module_ctx,             /* module context */
  142.     ngx_eventport_commands,                /* module directives */
  143.     NGX_EVENT_MODULE,                      /* module type */
  144.     NULL,                                  /* init master */
  145.     NULL,                                  /* init module */
  146.     NULL,                                  /* init process */
  147.     NULL,                                  /* init thread */
  148.     NULL,                                  /* exit thread */
  149.     NULL,                                  /* exit process */
  150.     NULL,                                  /* exit master */
  151.     NGX_MODULE_V1_PADDING
  152. };


  153. static ngx_int_t
  154. ngx_eventport_init(ngx_cycle_t *cycle, ngx_msec_t timer)
  155. {
  156.     port_notify_t          pn;
  157.     struct itimerspec      its;
  158.     struct sigevent        sev;
  159.     ngx_eventport_conf_t  *epcf;

  160.     epcf = ngx_event_get_conf(cycle->conf_ctx, ngx_eventport_module);

  161.     if (ep == -1) {
  162.         ep = port_create();

  163.         if (ep == -1) {
  164.             ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
  165.                           "port_create() failed");
  166.             return NGX_ERROR;
  167.         }

  168.         notify_event.active = 1;
  169.         notify_event.log = cycle->log;
  170.     }

  171.     if (nevents < epcf->events) {
  172.         if (event_list) {
  173.             ngx_free(event_list);
  174.         }

  175.         event_list = ngx_alloc(sizeof(port_event_t) * epcf->events,
  176.                                cycle->log);
  177.         if (event_list == NULL) {
  178.             return NGX_ERROR;
  179.         }
  180.     }

  181.     ngx_event_flags = NGX_USE_EVENTPORT_EVENT;

  182.     if (timer) {
  183.         ngx_memzero(&pn, sizeof(port_notify_t));
  184.         pn.portnfy_port = ep;

  185.         ngx_memzero(&sev, sizeof(struct sigevent));
  186.         sev.sigev_notify = SIGEV_PORT;
  187.         sev.sigev_value.sival_ptr = &pn;

  188.         if (timer_create(CLOCK_REALTIME, &sev, &event_timer) == -1) {
  189.             ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
  190.                           "timer_create() failed");
  191.             return NGX_ERROR;
  192.         }

  193.         its.it_interval.tv_sec = timer / 1000;
  194.         its.it_interval.tv_nsec = (timer % 1000) * 1000000;
  195.         its.it_value.tv_sec = timer / 1000;
  196.         its.it_value.tv_nsec = (timer % 1000) * 1000000;

  197.         if (timer_settime(event_timer, 0, &its, NULL) == -1) {
  198.             ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
  199.                           "timer_settime() failed");
  200.             return NGX_ERROR;
  201.         }

  202.         ngx_event_flags |= NGX_USE_TIMER_EVENT;
  203.     }

  204.     nevents = epcf->events;

  205.     ngx_io = ngx_os_io;

  206.     ngx_event_actions = ngx_eventport_module_ctx.actions;

  207.     return NGX_OK;
  208. }


  209. static void
  210. ngx_eventport_done(ngx_cycle_t *cycle)
  211. {
  212.     if (event_timer != (timer_t) -1) {
  213.         if (timer_delete(event_timer) == -1) {
  214.             ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno,
  215.                           "timer_delete() failed");
  216.         }

  217.         event_timer = (timer_t) -1;
  218.     }

  219.     if (close(ep) == -1) {
  220.         ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno,
  221.                       "close() event port failed");
  222.     }

  223.     ep = -1;

  224.     ngx_free(event_list);

  225.     event_list = NULL;
  226.     nevents = 0;
  227. }


  228. static ngx_int_t
  229. ngx_eventport_add_event(ngx_event_t *ev, ngx_int_t event, ngx_uint_t flags)
  230. {
  231.     ngx_int_t          events, prev;
  232.     ngx_event_t       *e;
  233.     ngx_connection_t  *c;

  234.     c = ev->data;

  235.     events = event;

  236.     if (event == NGX_READ_EVENT) {
  237.         e = c->write;
  238.         prev = POLLOUT;
  239. #if (NGX_READ_EVENT != POLLIN)
  240.         events = POLLIN;
  241. #endif

  242.     } else {
  243.         e = c->read;
  244.         prev = POLLIN;
  245. #if (NGX_WRITE_EVENT != POLLOUT)
  246.         events = POLLOUT;
  247. #endif
  248.     }

  249.     if (e->oneshot) {
  250.         events |= prev;
  251.     }

  252.     ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0,
  253.                    "eventport add event: fd:%d ev:%04Xi", c->fd, events);

  254.     if (port_associate(ep, PORT_SOURCE_FD, c->fd, events,
  255.                        (void *) ((uintptr_t) ev | ev->instance))
  256.         == -1)
  257.     {
  258.         ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno,
  259.                       "port_associate() failed");
  260.         return NGX_ERROR;
  261.     }

  262.     ev->active = 1;
  263.     ev->oneshot = 1;

  264.     return NGX_OK;
  265. }


  266. static ngx_int_t
  267. ngx_eventport_del_event(ngx_event_t *ev, ngx_int_t event, ngx_uint_t flags)
  268. {
  269.     ngx_event_t       *e;
  270.     ngx_connection_t  *c;

  271.     /*
  272.      * when the file descriptor is closed, the event port automatically
  273.      * dissociates it from the port, so we do not need to dissociate explicitly
  274.      * the event before the closing the file descriptor
  275.      */

  276.     if (flags & NGX_CLOSE_EVENT) {
  277.         ev->active = 0;
  278.         ev->oneshot = 0;
  279.         return NGX_OK;
  280.     }

  281.     c = ev->data;

  282.     if (event == NGX_READ_EVENT) {
  283.         e = c->write;
  284.         event = POLLOUT;

  285.     } else {
  286.         e = c->read;
  287.         event = POLLIN;
  288.     }

  289.     if (e->oneshot) {
  290.         ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0,
  291.                        "eventport change event: fd:%d ev:%04Xi", c->fd, event);

  292.         if (port_associate(ep, PORT_SOURCE_FD, c->fd, event,
  293.                            (void *) ((uintptr_t) ev | ev->instance))
  294.             == -1)
  295.         {
  296.             ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno,
  297.                           "port_associate() failed");
  298.             return NGX_ERROR;
  299.         }

  300.     } else if (ev->active) {
  301.         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, ev->log, 0,
  302.                        "eventport del event: fd:%d", c->fd);

  303.         if (port_dissociate(ep, PORT_SOURCE_FD, c->fd) == -1) {
  304.             ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno,
  305.                           "port_dissociate() failed");
  306.             return NGX_ERROR;
  307.         }
  308.     }

  309.     ev->active = 0;
  310.     ev->oneshot = 0;

  311.     return NGX_OK;
  312. }


  313. static ngx_int_t
  314. ngx_eventport_notify(ngx_event_handler_pt handler)
  315. {
  316.     notify_event.handler = handler;

  317.     if (port_send(ep, 0, &notify_event) != 0) {
  318.         ngx_log_error(NGX_LOG_ALERT, notify_event.log, ngx_errno,
  319.                       "port_send() failed");
  320.         return NGX_ERROR;
  321.     }

  322.     return NGX_OK;
  323. }


  324. static ngx_int_t
  325. ngx_eventport_process_events(ngx_cycle_t *cycle, ngx_msec_t timer,
  326.     ngx_uint_t flags)
  327. {
  328.     int                 n, revents;
  329.     u_int               events;
  330.     ngx_err_t           err;
  331.     ngx_int_t           instance;
  332.     ngx_uint_t          i, level;
  333.     ngx_event_t        *ev, *rev, *wev;
  334.     ngx_queue_t        *queue;
  335.     ngx_connection_t   *c;
  336.     struct timespec     ts, *tp;

  337.     if (timer == NGX_TIMER_INFINITE) {
  338.         tp = NULL;

  339.     } else {
  340.         ts.tv_sec = timer / 1000;
  341.         ts.tv_nsec = (timer % 1000) * 1000000;
  342.         tp = &ts;
  343.     }

  344.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, cycle->log, 0,
  345.                    "eventport timer: %M", timer);

  346.     events = 1;

  347.     n = port_getn(ep, event_list, (u_int) nevents, &events, tp);

  348.     err = ngx_errno;

  349.     if (flags & NGX_UPDATE_TIME) {
  350.         ngx_time_update();
  351.     }

  352.     if (n == -1) {
  353.         if (err == ETIME) {
  354.             if (timer != NGX_TIMER_INFINITE) {
  355.                 return NGX_OK;
  356.             }

  357.             ngx_log_error(NGX_LOG_ALERT, cycle->log, 0,
  358.                           "port_getn() returned no events without timeout");
  359.             return NGX_ERROR;
  360.         }

  361.         level = (err == NGX_EINTR) ? NGX_LOG_INFO : NGX_LOG_ALERT;
  362.         ngx_log_error(level, cycle->log, err, "port_getn() failed");
  363.         return NGX_ERROR;
  364.     }

  365.     if (events == 0) {
  366.         if (timer != NGX_TIMER_INFINITE) {
  367.             return NGX_OK;
  368.         }

  369.         ngx_log_error(NGX_LOG_ALERT, cycle->log, 0,
  370.                       "port_getn() returned no events without timeout");
  371.         return NGX_ERROR;
  372.     }

  373.     for (i = 0; i < events; i++) {

  374.         if (event_list[i].portev_source == PORT_SOURCE_TIMER) {
  375.             ngx_time_update();
  376.             continue;
  377.         }

  378.         ev = event_list[i].portev_user;

  379.         switch (event_list[i].portev_source) {

  380.         case PORT_SOURCE_FD:

  381.             instance = (uintptr_t) ev & 1;
  382.             ev = (ngx_event_t *) ((uintptr_t) ev & (uintptr_t) ~1);

  383.             if (ev->closed || ev->instance != instance) {

  384.                 /*
  385.                  * the stale event from a file descriptor
  386.                  * that was just closed in this iteration
  387.                  */

  388.                 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, cycle->log, 0,
  389.                                "eventport: stale event %p", ev);
  390.                 continue;
  391.             }

  392.             revents = event_list[i].portev_events;

  393.             ngx_log_debug2(NGX_LOG_DEBUG_EVENT, cycle->log, 0,
  394.                            "eventport: fd:%d, ev:%04Xd",
  395.                            (int) event_list[i].portev_object, revents);

  396.             if (revents & (POLLERR|POLLHUP|POLLNVAL)) {
  397.                 ngx_log_debug2(NGX_LOG_DEBUG_EVENT, cycle->log, 0,
  398.                                "port_getn() error fd:%d ev:%04Xd",
  399.                                (int) event_list[i].portev_object, revents);
  400.             }

  401.             if (revents & ~(POLLIN|POLLOUT|POLLERR|POLLHUP|POLLNVAL)) {
  402.                 ngx_log_error(NGX_LOG_ALERT, cycle->log, 0,
  403.                               "strange port_getn() events fd:%d ev:%04Xd",
  404.                               (int) event_list[i].portev_object, revents);
  405.             }

  406.             if (revents & (POLLERR|POLLHUP|POLLNVAL)) {

  407.                 /*
  408.                  * if the error events were returned, add POLLIN and POLLOUT
  409.                  * to handle the events at least in one active handler
  410.                  */

  411.                 revents |= POLLIN|POLLOUT;
  412.             }

  413.             c = ev->data;
  414.             rev = c->read;
  415.             wev = c->write;

  416.             rev->active = 0;
  417.             wev->active = 0;

  418.             if (revents & POLLIN) {
  419.                 rev->ready = 1;
  420.                 rev->available = -1;

  421.                 if (flags & NGX_POST_EVENTS) {
  422.                     queue = rev->accept ? &ngx_posted_accept_events
  423.                                         : &ngx_posted_events;

  424.                     ngx_post_event(rev, queue);

  425.                 } else {
  426.                     rev->handler(rev);

  427.                     if (ev->closed || ev->instance != instance) {
  428.                         continue;
  429.                     }
  430.                 }

  431.                 if (rev->accept) {
  432.                     if (ngx_use_accept_mutex) {
  433.                         ngx_accept_events = 1;
  434.                         continue;
  435.                     }

  436.                     if (port_associate(ep, PORT_SOURCE_FD, c->fd, POLLIN,
  437.                                        (void *) ((uintptr_t) ev | ev->instance))
  438.                         == -1)
  439.                     {
  440.                         ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno,
  441.                                       "port_associate() failed");
  442.                         return NGX_ERROR;
  443.                     }
  444.                 }
  445.             }

  446.             if (revents & POLLOUT) {
  447.                 wev->ready = 1;

  448.                 if (flags & NGX_POST_EVENTS) {
  449.                     ngx_post_event(wev, &ngx_posted_events);

  450.                 } else {
  451.                     wev->handler(wev);
  452.                 }
  453.             }

  454.             continue;

  455.         case PORT_SOURCE_USER:

  456.             ev->handler(ev);

  457.             continue;

  458.         default:
  459.             ngx_log_error(NGX_LOG_ALERT, cycle->log, 0,
  460.                           "unexpected eventport object %d",
  461.                           (int) event_list[i].portev_object);
  462.             continue;
  463.         }
  464.     }

  465.     return NGX_OK;
  466. }


  467. static void *
  468. ngx_eventport_create_conf(ngx_cycle_t *cycle)
  469. {
  470.     ngx_eventport_conf_t  *epcf;

  471.     epcf = ngx_palloc(cycle->pool, sizeof(ngx_eventport_conf_t));
  472.     if (epcf == NULL) {
  473.         return NULL;
  474.     }

  475.     epcf->events = NGX_CONF_UNSET;

  476.     return epcf;
  477. }


  478. static char *
  479. ngx_eventport_init_conf(ngx_cycle_t *cycle, void *conf)
  480. {
  481.     ngx_eventport_conf_t *epcf = conf;

  482.     ngx_conf_init_uint_value(epcf->events, 32);

  483.     return NGX_CONF_OK;
  484. }