One Level Up Top Level

src/core/ngx_proxy_protocol.c - nginx-1.7.10

Functions defined

Source code


  2. /*
  3. * Copyright (C) Roman Arutyunyan
  4. * Copyright (C) Nginx, Inc.
  5. */


  8. #include <ngx_config.h>
  9. #include <ngx_core.h>


  12. u_char *
  13. ‌ngx_proxy_protocol_parse(ngx_connection_t *c, u_char *buf, u_char *last)
  14. {
  15.     size_t  len;
  16.     u_char  ch, *p, *addr;

  18.     p = buf;
  19.     len = last - buf;

  21.     if (len < 8 || ngx_strncmp(p, "PROXY ", 6) != 0) {
  22.         goto invalid;
  23.     }

  25.     p += 6;
  26.     len -= 6;

  28.     if (len >= 7 && ngx_strncmp(p, "UNKNOWN", 7) == 0) {
  29.         ngx_log_debug0(NGX_LOG_DEBUG_CORE, c->log, 0,
  30.                        "PROXY protocol unknown protocol");
  31.         p += 7;
  32.         goto skip;
  33.     }

  35.     if (len < 5 || ngx_strncmp(p, "TCP", 3) != 0
  36.         || (p[3] != '4' && p[3] != '6') || p[4] != ' ')
  37.     {
  38.         goto invalid;
  39.     }

  41.     p += 5;
  42.     addr = p;

  44.     for ( ;; ) {
  45.         if (p == last) {
  46.             goto invalid;
  47.         }

  49.         ch = *p++;

  51.         if (ch == ' ') {
  52.             break;
  53.         }

  55.         if (ch != ':' && ch != '.'
  56.             && (ch < 'a' || ch > 'f')
  57.             && (ch < 'A' || ch > 'F')
  58.             && (ch < '0' || ch > '9'))
  59.         {
  60.             goto invalid;
  61.         }
  62.     }

  64.     len = p - addr - 1;
  65.     c->proxy_protocol_addr.data = ngx_pnalloc(c->pool, len);

  67.     if (c->proxy_protocol_addr.data == NULL) {
  68.         return NULL;
  69.     }

  71.     ngx_memcpy(c->proxy_protocol_addr.data, addr, len);
  72.     c->proxy_protocol_addr.len = len;

  74.     ngx_log_debug1(NGX_LOG_DEBUG_CORE, c->log, 0,
  75.                    "PROXY protocol address: \"%V\"", &c->proxy_protocol_addr);

  77. skip:

  79.     for ( /* void */ ; p < last - 1; p++) {
  80.         if (p[0] == CR && p[1] == LF) {
  81.             return p + 2;
  82.         }
  83.     }

  85. invalid:

  87.     ngx_log_error(NGX_LOG_ERR, c->log, 0,
  88.                   "broken header: \"%*s\"", (size_t) (last - buf), buf);

  90.     return NULL;
  91. }

One Level Up Top Level