Mercurial > hg > nginx-quic
view src/core/ngx_inet.c @ 1900:55a9d867e3c8
avoid endless loop if epoll is used
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Tue, 19 Feb 2008 16:30:54 +0000 |
parents | 04b04e6d60f4 |
children | 2a92804f4109 |
line wrap: on
line source
/* * Copyright (C) Igor Sysoev */ #include <ngx_config.h> #include <ngx_core.h> static size_t ngx_sprint_uchar(u_char *text, u_char c, size_t len); /* AF_INET only */ in_addr_t ngx_inet_addr(u_char *text, size_t len) { u_char *p, c; in_addr_t addr; ngx_uint_t octet, n; addr = 0; octet = 0; n = 0; for (p = text; p < text + len; p++) { c = *p; if (c >= '0' && c <= '9') { octet = octet * 10 + (c - '0'); continue; } if (c == '.' && octet < 256) { addr = (addr << 8) + octet; octet = 0; n++; continue; } return INADDR_NONE; } if (n != 3) { return INADDR_NONE; } if (octet < 256) { addr = (addr << 8) + octet; return htonl(addr); } return INADDR_NONE; } /* * ngx_sock_ntop() and ngx_inet_ntop() may be implemented as * "ngx_sprintf(text, "%ud.%ud.%ud.%ud", p[0], p[1], p[2], p[3])", however, * they had been implemented long before the ngx_sprintf() had appeared * and they are faster by 1.5-2.5 times, so it is worth to keep them. * * By the way, the implementation using ngx_sprintf() is faster by 2.5-3 times * than using FreeBSD libc's snprintf(). */ /* AF_INET only */ size_t ngx_sock_ntop(int family, struct sockaddr *sa, u_char *text, size_t len) { u_char *p; size_t n; ngx_uint_t i; struct sockaddr_in *sin; if (len == 0) { return 0; } if (family != AF_INET) { return 0; } sin = (struct sockaddr_in *) sa; p = (u_char *) &sin->sin_addr; if (len > INET_ADDRSTRLEN) { len = INET_ADDRSTRLEN; } n = ngx_sprint_uchar(text, p[0], len); i = 1; do { if (len == n) { text[n - 1] = '\0'; return n; } text[n++] = '.'; if (len == n) { text[n - 1] = '\0'; return n; } n += ngx_sprint_uchar(&text[n], p[i++], len - n); } while (i < 4); if (len == n) { text[n] = '\0'; return n; } text[n] = '\0'; return n; } size_t ngx_inet_ntop(int family, void *addr, u_char *text, size_t len) { u_char *p; size_t n; ngx_uint_t i; if (len == 0) { return 0; } if (family != AF_INET) { return 0; } p = (u_char *) addr; if (len > INET_ADDRSTRLEN) { len = INET_ADDRSTRLEN; } n = ngx_sprint_uchar(text, p[0], len); i = 1; do { if (len == n) { text[n - 1] = '\0'; return n; } text[n++] = '.'; if (len == n) { text[n - 1] = '\0'; return n; } n += ngx_sprint_uchar(&text[n], p[i++], len - n); } while (i < 4); if (len == n) { text[n] = '\0'; return n; } text[n] = '\0'; return n; } static size_t ngx_sprint_uchar(u_char *text, u_char c, size_t len) { size_t n; ngx_uint_t c1, c2; n = 0; if (len == n) { return n; } c1 = c / 100; if (c1) { *text++ = (u_char) (c1 + '0'); n++; if (len == n) { return n; } } c2 = (c % 100) / 10; if (c1 || c2) { *text++ = (u_char) (c2 + '0'); n++; if (len == n) { return n; } } c2 = c % 10; *text = (u_char) (c2 + '0'); n++; return n; } /* AF_INET only */ ngx_int_t ngx_ptocidr(ngx_str_t *text, void *cidr) { ngx_int_t m; ngx_uint_t i; ngx_inet_cidr_t *in_cidr; in_cidr = cidr; for (i = 0; i < text->len; i++) { if (text->data[i] == '/') { break; } } if (i == text->len) { return NGX_ERROR; } text->data[i] = '\0'; in_cidr->addr = inet_addr((char *) text->data); text->data[i] = '/'; if (in_cidr->addr == INADDR_NONE) { return NGX_ERROR; } m = ngx_atoi(&text->data[i + 1], text->len - (i + 1)); if (m == NGX_ERROR) { return NGX_ERROR; } if (m == 0) { /* the x86 compilers use the shl instruction that shifts by modulo 32 */ in_cidr->mask = 0; return NGX_OK; } in_cidr->mask = htonl((ngx_uint_t) (0 - (1 << (32 - m)))); if (in_cidr->addr == (in_cidr->addr & in_cidr->mask)) { return NGX_OK; } in_cidr->addr &= in_cidr->mask; return NGX_DONE; } ngx_int_t ngx_parse_url(ngx_pool_t *pool, ngx_url_t *u) { u_char *p, *host, *port_start; size_t len, port_len; ngx_int_t port; ngx_uint_t i; struct hostent *h; #if (NGX_HAVE_UNIX_DOMAIN) struct sockaddr_un *saun; #endif len = u->url.len; p = u->url.data; if (ngx_strncasecmp(p, (u_char *) "unix:", 5) == 0) { #if (NGX_HAVE_UNIX_DOMAIN) p += 5; len -= 5; u->uri.len = len; u->uri.data = p; if (u->uri_part) { for (i = 0; i < len; i++) { if (p[i] == ':') { len = i; u->uri.len -= len + 1; u->uri.data += len + 1; break; } } } if (len == 0) { u->err = "no path in the unix domain socket"; return NGX_ERROR; } if (len + 1 > sizeof(saun->sun_path)) { u->err = "too long path in the unix domain socket"; return NGX_ERROR; } u->addrs = ngx_pcalloc(pool, sizeof(ngx_peer_addr_t)); if (u->addrs == NULL) { return NGX_ERROR; } saun = ngx_pcalloc(pool, sizeof(struct sockaddr_un)); if (saun == NULL) { return NGX_ERROR; } u->naddrs = 1; saun->sun_family = AF_UNIX; (void) ngx_cpystrn((u_char *) saun->sun_path, p, len + 1); u->addrs[0].sockaddr = (struct sockaddr *) saun; u->addrs[0].socklen = sizeof(struct sockaddr_un); u->addrs[0].name.len = len + 5; u->addrs[0].name.data = u->url.data; u->host.len = len; u->host.data = p; u->unix_socket = 1; return NGX_OK; #else u->err = "the unix domain sockets are not supported on this platform"; return NGX_ERROR; #endif } if ((p[0] == ':' || p[0] == '/') && !u->listen) { u->err = "invalid host"; return NGX_ERROR; } u->host.data = p; port_start = NULL; port_len = 0; for (i = 0; i < len; i++) { if (p[i] == ':') { port_start = &p[i + 1]; u->host.len = i; if (!u->uri_part) { port_len = len - (i + 1); break; } } if (p[i] == '/') { u->uri.len = len - i; u->uri.data = &p[i]; if (u->host.len == 0) { u->host.len = i; } if (port_start == NULL) { u->no_port = 1; goto no_port; } port_len = &p[i] - port_start; if (port_len == 0) { u->err = "invalid port"; return NGX_ERROR; } break; } } if (port_start) { if (port_len == 0) { port_len = &p[i] - port_start; if (port_len == 0) { u->err = "invalid port"; return NGX_ERROR; } } port = ngx_atoi(port_start, port_len); if (port == NGX_ERROR || port < 1 || port > 65536) { u->err = "invalid port"; return NGX_ERROR; } u->port_text.len = port_len; u->port_text.data = port_start; } else { port = ngx_atoi(p, len); if (port == NGX_ERROR) { u->host.len = len; u->no_port = 1; goto no_port; } u->wildcard = 1; } u->port = (in_port_t) port; no_port: if (u->listen) { if (u->port == 0) { if (u->default_port == 0) { u->err = "no port"; return NGX_ERROR; } u->port = u->default_port; } if (u->host.len == 1 && u->host.data[0] == '*') { u->host.len = 0; } /* AF_INET only */ if (u->host.len) { host = ngx_alloc(u->host.len + 1, pool->log); if (host == NULL) { return NGX_ERROR; } (void) ngx_cpystrn(host, u->host.data, u->host.len + 1); u->addr.in_addr = inet_addr((const char *) host); if (u->addr.in_addr == INADDR_NONE) { h = gethostbyname((const char *) host); if (h == NULL || h->h_addr_list[0] == NULL) { ngx_free(host); u->err = "host not found"; return NGX_ERROR; } u->addr.in_addr = *(in_addr_t *) (h->h_addr_list[0]); } ngx_free(host); } else { u->addr.in_addr = INADDR_ANY; } return NGX_OK; } if (u->host.len == 0) { u->err = "no host"; return NGX_ERROR; } if (u->no_resolve) { return NGX_OK; } if (u->no_port) { u->port = u->default_port; } if (u->port == 0) { u->err = "no port"; return NGX_ERROR; } if (ngx_inet_resolve_host(pool, u) != NGX_OK) { return NGX_ERROR; } return NGX_OK; } ngx_int_t ngx_inet_resolve_host(ngx_pool_t *pool, ngx_url_t *u) { u_char *p, *host; size_t len; in_addr_t in_addr; ngx_uint_t i; struct hostent *h; struct sockaddr_in *sin; host = ngx_alloc(u->host.len + 1, pool->log); if (host == NULL) { return NGX_ERROR; } (void) ngx_cpystrn(host, u->host.data, u->host.len + 1); /* AF_INET only */ in_addr = inet_addr((char *) host); if (in_addr == INADDR_NONE) { h = gethostbyname((char *) host); ngx_free(host); if (h == NULL || h->h_addr_list[0] == NULL) { u->err = "host not found"; return NGX_ERROR; } if (u->one_addr == 0) { for (i = 0; h->h_addr_list[i] != NULL; i++) { /* void */ } } else { i = 1; } /* MP: ngx_shared_palloc() */ u->addrs = ngx_pcalloc(pool, i * sizeof(ngx_peer_addr_t)); if (u->addrs == NULL) { return NGX_ERROR; } u->naddrs = i; for (i = 0; h->h_addr_list[i] != NULL; i++) { sin = ngx_pcalloc(pool, sizeof(struct sockaddr_in)); if (sin == NULL) { return NGX_ERROR; } sin->sin_family = AF_INET; sin->sin_port = htons(u->port); sin->sin_addr.s_addr = *(in_addr_t *) (h->h_addr_list[i]); u->addrs[i].sockaddr = (struct sockaddr *) sin; u->addrs[i].socklen = sizeof(struct sockaddr_in); len = INET_ADDRSTRLEN - 1 + 1 + sizeof(":65536") - 1; p = ngx_palloc(pool, len); if (p == NULL) { return NGX_ERROR; } len = ngx_sock_ntop(AF_INET, (struct sockaddr *) sin, p, len); u->addrs[i].name.len = ngx_sprintf(&p[len], ":%d", u->port) - p; u->addrs[i].name.data = p; } } else { ngx_free(host); /* MP: ngx_shared_palloc() */ u->addrs = ngx_pcalloc(pool, sizeof(ngx_peer_addr_t)); if (u->addrs == NULL) { return NGX_ERROR; } sin = ngx_pcalloc(pool, sizeof(struct sockaddr_in)); if (sin == NULL) { return NGX_ERROR; } u->naddrs = 1; sin->sin_family = AF_INET; sin->sin_port = htons(u->port); sin->sin_addr.s_addr = in_addr; u->addrs[0].sockaddr = (struct sockaddr *) sin; u->addrs[0].socklen = sizeof(struct sockaddr_in); p = ngx_palloc(pool, u->host.len + sizeof(":65536") - 1); if (p == NULL) { return NGX_ERROR; } u->addrs[0].name.len = ngx_sprintf(p, "%V:%d", &u->host, u->port) - p; u->addrs[0].name.data = p; } return NGX_OK; }