01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen#include "lib.h"

e130bb802c8bfb6c6cc44e5c8bc098b4fa5af789Timo Sirainen#include "array.h"

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen#include "str.h"

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen#include "net.h"

5f4e547bb810403e8cfb19a49d8fe34713507ffdTimo Sirainen#include "uri-util.h"

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen#include <ctype.h>

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

5f4e547bb810403e8cfb19a49d8fe34713507ffdTimo Sirainen

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define URI_MAX_SCHEME_NAME_LEN 64

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_UNRESERVED (1<<0)

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_SUB_DELIMS (1<<1)

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_PCHAR ((1<<0)|(1<<1)|(1<<3))

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_PFCHAR ((1<<0)|(1<<1)|(1<<3)|(1<<5))

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_UCHAR ((1<<0)|(1<<1)|(1<<4))

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen#define CHAR_MASK_QCHAR ((1<<0)|(1<<1)|(1<<3)|(1<<5)|(1<<6))

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainenstatic unsigned const char _uri_char_lookup[256] = {

d2cf6522779802d0edeab7dcf960ffea2f2e1828Timo Sirainen 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 00

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 10

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 0, 2, 0, 4, 2, 0, 2, 2, 2, 2, 2, 2, 2, 1, 1, 36, // 20

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28, 2, 0, 2, 0, 68, // 30

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 12, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, // 40

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 4, 0, 4, 0, 1, // 50

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, // 60

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 0, // 70

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen};

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

fcb5f4cd72b413a5356a8db55e679403c6a1adb5Timo Sirainenstatic inline int _decode_hex_digit(const unsigned char digit)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen switch (digit) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen case '0': case '1': case '2': case '3': case '4':

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen case '5': case '6': case '7': case '8': case '9':

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return digit - '0';

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen case 'a': case 'b': case 'c': case 'd': case 'e': case 'f':

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return digit - 'a' + 0x0a;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen case 'A': case 'B': case 'C': case 'D': case 'E': case 'F':

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return digit - 'A' + 0x0A;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen}

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenstatic int ATTR_NULL(3)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenuri_parse_pct_encoded(struct uri_parser *parser, const unsigned char **p,

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen const unsigned char *pend, unsigned char *ch_r)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen int value;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (**p == 0 || *(*p+1) == 0 || (pend != NULL && *p+1 >= pend)) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->error = "Unexpected URI boundary after '%'";

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if ((value = _decode_hex_digit(**p)) < 0) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->error = t_strdup_printf(

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen "Expecting hex digit after '%%', but found '%c'", **p);

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *ch_r = (value & 0x0f) << 4;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *p += 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if ((value = _decode_hex_digit(**p)) < 0) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->error = t_strdup_printf(

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen "Expecting hex digit after '%%%c', but found '%c'", *((*p)-1), **p);

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *ch_r |= (value & 0x0f);

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *p += 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (*ch_r == '\0') {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->error =

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen "Percent encoding is not allowed to encode NUL character";

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen}

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenstatic int

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainenuri_parse_unreserved_char(struct uri_parser *parser, unsigned char *ch_r)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen{

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if (*parser->cur == '%') {

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen parser->cur++;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen if (uri_parse_pct_encoded(parser, &parser->cur,

84a5175b9768da401404635c9b606264585739bdTimo Sirainen parser->end, ch_r) <= 0)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return -1;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return 1;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen }

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if ((*parser->cur & 0x80) != 0)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return 0;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen

f2686912e0156c04296d6dc306f39d61089a1363Timo Sirainen if ((_uri_char_lookup[*parser->cur] & CHAR_MASK_UNRESERVED) != 0) {

f2686912e0156c04296d6dc306f39d61089a1363Timo Sirainen *ch_r = *parser->cur;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen parser->cur++;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return 1;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen }

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return 0;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen}

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainenint uri_parse_unreserved(struct uri_parser *parser, string_t *part)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen{

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen int len = 0;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen while (parser->cur < parser->end) {

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen int ret;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen unsigned char ch = 0;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if ((ret = uri_parse_unreserved_char(parser, &ch)) < 0)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return -1;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if (ret == 0)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen break;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if (part != NULL)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen str_append_c(part, ch);

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen len++;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen }

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen return len > 0 ? 1 : 0;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen}

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainenbool uri_data_decode(struct uri_parser *parser, const char *data,

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen const char *until, const char **decoded_r)

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen{

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen const unsigned char *p = (const unsigned char *)data;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen const unsigned char *pend = (const unsigned char *)until;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen string_t *decoded;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (pend == NULL) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen /* NULL means unlimited; solely rely on '\0' */

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen pend = (const unsigned char *)(size_t)-1;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (p >= pend || *p == '\0') {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (decoded_r != NULL)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *decoded_r = "";

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return TRUE;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen }

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen decoded = uri_parser_get_tmpbuf(parser, 256);

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen while (p < pend && *p != '\0') {

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen unsigned char ch;

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if (*p == '%') {

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen p++;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen if (uri_parse_pct_encoded(parser, &p, NULL, &ch) <= 0)

84a5175b9768da401404635c9b606264585739bdTimo Sirainen return FALSE;

84a5175b9768da401404635c9b606264585739bdTimo Sirainen

84a5175b9768da401404635c9b606264585739bdTimo Sirainen str_append_c(decoded, ch);

84a5175b9768da401404635c9b606264585739bdTimo Sirainen } else {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen str_append_c(decoded, *p);

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen p++;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (decoded_r != NULL)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *decoded_r = t_strdup(str_c(decoded));

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return TRUE;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen}

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenint uri_cut_scheme(const char **uri_p, const char **scheme_r)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen const char *p = *uri_p;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen size_t len = 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

0d6ae58916bee3452c91d9d81be72227761ec33dTimo Sirainen /* RFC 3968:

0d6ae58916bee3452c91d9d81be72227761ec33dTimo Sirainen

9f0a996c22ebe39dcfe5cb84c8fd2f22ef5ce9d8Timo Sirainen if (!i_isalpha(*p))

0d6ae58916bee3452c91d9d81be72227761ec33dTimo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen p++;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen while (len < URI_MAX_SCHEME_NAME_LEN && *p != '\0') {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (!i_isalnum(*p) && *p != '+' && *p != '-' && *p != '.')

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen break;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen p++;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen len++;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen if (*p != ':')

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen return -1;

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen *scheme_r = t_strdup_until(*uri_p, p);

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen *uri_p = p + 1;

b13d76faf0c82162c29050382cd7f4a808294622Timo Sirainen return 0;

2026684bfc269f4ab3196dea3594f9fe17b1238dTimo Sirainen}

0d6ae58916bee3452c91d9d81be72227761ec33dTimo Sirainen

0d6ae58916bee3452c91d9d81be72227761ec33dTimo Sirainenint uri_parse_scheme(struct uri_parser *parser, const char **scheme_r)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

60d1fdf2c17fd0c7020234590dbd73da81c3ce8fTimo Sirainen const char *p;

60d1fdf2c17fd0c7020234590dbd73da81c3ce8fTimo Sirainen

60d1fdf2c17fd0c7020234590dbd73da81c3ce8fTimo Sirainen if (parser->cur >= parser->end)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen p = (const char *)parser->cur;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (uri_cut_scheme(&p, scheme_r) < 0)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->cur = (const unsigned char *)p;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen}

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenstatic int

e130bb802c8bfb6c6cc44e5c8bc098b4fa5af789Timo Sirainenuri_parse_dec_octet(struct uri_parser *parser, string_t *literal,

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen uint8_t *octet_r)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen uint8_t octet = 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen int count = 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen /* RFC 3986:

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen while (parser->cur < parser->end && i_isdigit(*parser->cur)) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen uint8_t prev = octet;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen octet = octet * 10 + (parser->cur[0] - '0');

afc77c5375cdb8f2bf0ab6280d9229ac27c933c6Timo Sirainen if (octet < prev)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return -1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (literal != NULL)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen str_append_c(literal, *parser->cur);

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen parser->cur++;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen count++;

f46885a5b78b15a8d2419f6e5d13b643bd85e41fTimo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen if (count > 0) {

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen *octet_r = octet;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 1;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen }

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen return 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen}

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenstatic int

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainenuri_parse_ipv4address(struct uri_parser *parser, string_t *literal,

306b3f41b05da642d87e7ca7a1496efce9f5902fTimo Sirainen struct in_addr *ip4_r)

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen{

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen uint8_t octet;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen uint32_t ip = 0;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen int ret;

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen int i;

f46885a5b78b15a8d2419f6e5d13b643bd85e41fTimo Sirainen

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen /* RFC 3986:

01435c38e7d671d5a892c4b802cfb204881cd454Timo Sirainen

if ((ret = uri_parse_dec_octet(parser, literal, &octet)) <= 0)

return ret;

ip = octet;

for (i = 0; i < 3 && parser->cur < parser->end; i++) {

if (*parser->cur != '.')

return -1;

if (literal != NULL)

str_append_c(literal, '.');

parser->cur++;

if ((ret = uri_parse_dec_octet(parser, literal, &octet)) <= 0)

return -1;

ip = (ip << 8) + octet;

}

if (ip4_r != NULL)

ip4_r->s_addr = htonl(ip);

return 1;

}

static int uri_parse_reg_name(struct uri_parser *parser, string_t *reg_name)

{

int len = 0;

/* RFC 3986:

while (parser->cur < parser->end) {

int ret;

unsigned char c;

/* unreserved / pct-encoded */

if ((ret = uri_parse_unreserved_char(parser, &c)) < 0)

return -1;

if (ret > 0) {

if (reg_name != NULL)

str_append_c(reg_name, c);

len++;

continue;

}

/* sub-delims */

c = *parser->cur;

if ((c & 0x80) == 0 && (_uri_char_lookup[c] & CHAR_MASK_SUB_DELIMS) != 0) {

if (reg_name != NULL)

str_append_c(reg_name, *parser->cur);

parser->cur++;

len++;

continue;

}

break;

}

return len > 0 ? 1 : 0;

}

#ifdef HAVE_IPV6

static int

uri_parse_ip_literal(struct uri_parser *parser, string_t *literal,

struct in6_addr *ip6_r)

{

const unsigned char *p;

const char *address;

int ret;

/* IP-literal = "[" ( IPv6address / IPvFuture ) "]"

/* "[" already verified */

/* Scan for end of address */

for (p = parser->cur+1; p < parser->end; p++) {

if (*p == ']')

break;

}

if (p >= parser->end || *p != ']') {

parser->error = "Expecting ']' at end of IP-literal";

return -1;

}

if (literal != NULL)

str_append_n(literal, parser->cur, p-parser->cur+1);

address = t_strdup_until(parser->cur+1, p);

parser->cur = p + 1;

if (*address == '\0') {

parser->error = "Empty IPv6 host address";

return -1;

}

if (*address == 'v') {

parser->error = t_strdup_printf(

"Future IP host address '%s' not supported", address);

return -1;

}

if ((ret = inet_pton(AF_INET6, address, ip6_r)) <= 0) {

parser->error = t_strdup_printf(

"Invalid IPv6 host address '%s'", address);

return -1;

}

return 1;

}

#endif

static int uri_parse_host(struct uri_parser *parser, struct uri_authority *auth)

{

const unsigned char *preserve;

struct in_addr ip4;

struct in6_addr ip6;

string_t *literal = NULL;

int ret;

/* RFC 3986:

literal = uri_parser_get_tmpbuf(parser, 256);

/* IP-literal / */

if (parser->cur < parser->end && *parser->cur == '[') {

#ifdef HAVE_IPV6

if ((ret=uri_parse_ip_literal(parser, literal, &ip6)) <= 0)

return -1;

if (auth != NULL) {

auth->host_literal = t_strdup(str_c(literal));

auth->host_ip.family = AF_INET6;

auth->host_ip.u.ip6 = ip6;

auth->have_host_ip = TRUE;

}

return 1;

#else

parser->error = "IPv6 host address is not supported";

return -1;

#endif

}

/* IPv4address /

preserve = parser->cur;

if ((ret = uri_parse_ipv4address(parser, literal, &ip4)) > 0) {

if (auth != NULL) {

auth->host_literal = t_strdup(str_c(literal));

auth->host_ip.family = AF_INET;

auth->host_ip.u.ip4 = ip4;

auth->have_host_ip = TRUE;

}

return ret;

}

parser->cur = preserve;

str_truncate(literal, 0);

/* reg-name */

if ((ret = uri_parse_reg_name(parser, literal)) != 0) {

if (ret > 0 && auth != NULL) {

auth->host_literal = t_strdup(str_c(literal));

auth->have_host_ip = FALSE;

}

return ret;

}

return 0;

}

static int uri_parse_port(struct uri_parser *parser, struct uri_authority *auth)

{

in_port_t port = 0;

int count = 0;

/* RFC 3986:

while (parser->cur < parser->end && i_isdigit(*parser->cur)) {

in_port_t prev = port;

port = port * 10 + (in_port_t)(parser->cur[0] - '0');

if (port < prev) {

parser->error = "Port number is too high";

return -1;

}

parser->cur++;

count++;

}

if (count > 0) {

if (auth != NULL) {

auth->port = port;

auth->have_port = TRUE;

}

return 1;

}

return 0;

}

int uri_parse_authority(struct uri_parser *parser, struct uri_authority *auth)

{

const unsigned char *p;

int ret;

/* hier-part = "//" authority {...}

/* Parse "//" as part of authority */

if ((parser->end - parser->cur) <= 2 || parser->cur[0] != '/' ||

parser->cur[1] != '/')

return 0;

parser->cur += 2;

if (auth != NULL)

memset(auth, 0, sizeof(*auth));

/* Scan ahead to check whether there is a [userinfo "@"] uri component */

for (p = parser->cur; p < parser->end; p++){

/* refuse 8bit characters */

if ((*p & 0x80) != 0)

break;

/* break at first delimiter */

if (*p != '%' && (_uri_char_lookup[*p] & CHAR_MASK_UCHAR) == 0)

break;

}

/* Extract userinfo */

if (p < parser->end && *p == '@') {

if (auth != NULL)

auth->enc_userinfo = t_strdup_until(parser->cur, p);

parser->cur = p+1;

}

/* host */

if ((ret = uri_parse_host(parser, auth)) <= 0) {

if (ret == 0) {

parser->error = "Missing 'host' component";

return -1;

}

return ret;

}

/* [":" ... */

if (parser->cur >= parser->end || *parser->cur != ':')

return 1;

parser->cur++;

/* ... port] */

if ((ret = uri_parse_port(parser, auth)) < 0)

return ret;

return 1;

}

int uri_parse_path_segment(struct uri_parser *parser, const char **segment_r)

{

const unsigned char *p = parser->cur;

while (p < parser->end) {

if (*p == '%') {

p++;

continue;

}

if ((*p & 0x80) != 0 || (_uri_char_lookup[*p] & CHAR_MASK_PCHAR) == 0)

break;

p++;

}

if (p == parser->cur)

return 0;

if (segment_r != NULL)

*segment_r = t_strdup_until(parser->cur, p);

parser->cur = p;

return 1;

}

int uri_parse_path(struct uri_parser *parser,

int *relative_r, const char *const **path_r)

{

const unsigned char *pbegin = parser->cur;

ARRAY_TYPE(const_string) segments;

const char *segment = NULL;

unsigned int count;

int relative = 1;

int ret;

t_array_init(&segments, 16);

/* check for a leading '/' and indicate absolute path

if (parser->cur < parser->end && *parser->cur == '/') {

parser->cur++;

relative = 0;

}

/* parse first segment */

if ((ret = uri_parse_path_segment(parser, &segment)) < 0)

return -1;

for (;;) {

if (ret > 0) {

/* strip dot segments */

if (segment[0] == '.') {

if (segment[1] == '.') {

if (segment[2] == '\0') {

/* '..' -> skip and... */

segment = NULL;

/* ... pop last segment (if any) */

count = array_count(&segments);

if (count > 0) {

array_delete(&segments, count-1, 1);

} else if ( relative > 0 ) {

relative++;

}

}

} else if (segment[1] == '\0') {

/* '.' -> skip */

segment = NULL;

}

}

} else {

segment = "";

}

if (segment != NULL)

array_append(&segments, &segment, 1);

if (parser->cur >= parser->end || *parser->cur != '/')

break;

parser->cur++;

/* parse next path segment */

if ((ret = uri_parse_path_segment(parser, &segment)) < 0)

return -1;

}

if (parser->cur == pbegin) {

/* path part of URI is missing */

return 0;

}

/* special treatment for a trailing '..' or '.' */

if (segment == NULL) {

segment = "";

array_append(&segments, &segment, 1);

}

array_append_zero(&segments);

*path_r = array_get(&segments, &count);

*relative_r = relative;

return 1;

}

int uri_parse_query(struct uri_parser *parser, const char **query_r)

{

const unsigned char *p = parser->cur;

/* RFC 3986:

if (p >= parser->end || *p != '?')

return 0;

p++;

while (p < parser->end) {

if (*p == '%') {

p++;

continue;

}

if ((*p & 0x80) != 0 || (_uri_char_lookup[*p] & CHAR_MASK_QCHAR) == 0)

break;

p++;

}

if (query_r != NULL)

*query_r = t_strdup_until(parser->cur+1, p);

parser->cur = p;

return 1;

}

int uri_parse_fragment(struct uri_parser *parser, const char **fragment_r)

{

const unsigned char *p = parser->cur;

/* RFC 3986:

if (p >= parser->end || *p != '#')

return 0;

p++;

while (p < parser->end) {

if (*p == '%') {

p++;

continue;

}

if ((*p & 0x80) != 0 || (_uri_char_lookup[*p] & CHAR_MASK_QCHAR) == 0)

break;

p++;

}

if (fragment_r != NULL)

*fragment_r = t_strdup_until(parser->cur+1, p);

parser->cur = p;

return 1;

}

void uri_parser_init(struct uri_parser *parser, pool_t pool, const char *data)

{

parser->pool = pool;

parser->begin = parser->cur = (unsigned char *)data;

parser->end = (unsigned char *)data + strlen(data);

parser->error = NULL;

parser->tmpbuf = NULL;

}

string_t *uri_parser_get_tmpbuf(struct uri_parser *parser, size_t size)

{

if (parser->tmpbuf == NULL)

parser->tmpbuf = t_str_new(size);

else

str_truncate(parser->tmpbuf, 0);

return parser->tmpbuf;

}

static void

uri_data_encode(string_t *out, const unsigned char esc_table[256],

unsigned char esc_mask, const char *esc_extra, const char *data)

{

const unsigned char *p = (const unsigned char *)data;

while (*p != '\0') {

if ((*p & 0x80) != 0 || (esc_table[*p] & esc_mask) == 0 ||

strchr(esc_extra, (char)*p) != NULL) {

str_printfa(out, "%%%2x", *p);

} else {

str_append_c(out, *p);

}

p++;

}

}

void uri_append_scheme(string_t *out, const char *scheme)

{

str_append(out, scheme);

str_append_c(out, ':');

}

void uri_append_user_data(string_t *out, const char *esc,

const char *data)

{

uri_data_encode(out, _uri_char_lookup, CHAR_MASK_UCHAR, esc, data);

}

void uri_append_userinfo(string_t *out, const char *userinfo)

{

uri_append_user_data(out, "", userinfo);

str_append_c(out, '@');

}

void uri_append_host_name(string_t *out, const char *name)

{

uri_data_encode(out, _uri_char_lookup,

CHAR_MASK_UNRESERVED | CHAR_MASK_SUB_DELIMS, "", name);

}

void uri_append_host_ip(string_t *out, const struct ip_addr *host_ip)

{

const char *addr = net_ip2addr(host_ip);

i_assert(addr != NULL);

if (host_ip->family == AF_INET) {

str_append(out, addr);

return;

}

i_assert(host_ip->family == AF_INET6);

str_append_c(out, '[');

str_append(out, addr);

str_append_c(out, ']');

}

void uri_append_port(string_t *out, in_port_t port)

{

str_printfa(out, ":%u", port);

}

void uri_append_path_segment_data(string_t *out, const char *esc,

const char *data)

{

uri_data_encode(out, _uri_char_lookup, CHAR_MASK_PCHAR, esc, data);

}

void uri_append_path_segment(string_t *out, const char *segment)

{

str_append_c(out, '/');

if (*segment != '\0')

uri_append_path_data(out, "", segment);

}

void uri_append_path_data(string_t *out, const char *esc,

const char *data)

{

uri_data_encode(out, _uri_char_lookup, CHAR_MASK_PFCHAR, esc, data);

}

void uri_append_path(string_t *out, const char *path)

{

str_append_c(out, '/');

if (*path != '\0')

uri_append_path_data(out, "", path);

}

void uri_append_query_data(string_t *out, const char *esc,

const char *data)

{

uri_data_encode(out, _uri_char_lookup, CHAR_MASK_QCHAR, esc, data);

}

void uri_append_query(string_t *out, const char *query)

{

str_append_c(out, '?');

if (*query != '\0')

uri_append_query_data(out, "", query);

}

void uri_append_fragment_data(string_t *out, const char *esc,

const char *data)

{

uri_data_encode(out, _uri_char_lookup, CHAR_MASK_QCHAR, esc, data);

}

void uri_append_fragment(string_t *out, const char *fragment)

{

str_append_c(out, '#');

if (*fragment != '\0')

uri_append_fragment_data(out, "", fragment);

}