| 
 | 1 | +#include "git-compat-util.h"  | 
 | 2 | +#include "hex.h"  | 
 | 3 | +#include "json-parser.h"  | 
 | 4 | + | 
 | 5 | +int reset_iterator(struct json_iterator *it)  | 
 | 6 | +{  | 
 | 7 | +	it->p = it->begin = it->json;  | 
 | 8 | +	strbuf_release(&it->key);  | 
 | 9 | +	strbuf_release(&it->string_value);  | 
 | 10 | +	it->type = JSON_NULL;  | 
 | 11 | +	return -1;  | 
 | 12 | +}  | 
 | 13 | + | 
 | 14 | +static int parse_json_string(struct json_iterator *it, struct strbuf *out)  | 
 | 15 | +{  | 
 | 16 | +	const char *begin = it->p;  | 
 | 17 | + | 
 | 18 | +	if (*(it->p)++ != '"')  | 
 | 19 | +		return error("expected double quote: '%.*s'", 5, begin),  | 
 | 20 | +			reset_iterator(it);  | 
 | 21 | + | 
 | 22 | +	strbuf_reset(&it->string_value);  | 
 | 23 | +#define APPEND(c) strbuf_addch(out, c)  | 
 | 24 | +	while (*it->p != '"') {  | 
 | 25 | +		switch (*it->p) {  | 
 | 26 | +		case '\0':  | 
 | 27 | +			return error("incomplete string: '%s'", begin),  | 
 | 28 | +				reset_iterator(it);  | 
 | 29 | +		case '\\':  | 
 | 30 | +			it->p++;  | 
 | 31 | +			if (*it->p == '\\' || *it->p == '"')  | 
 | 32 | +				APPEND(*it->p);  | 
 | 33 | +			else if (*it->p == 'b')  | 
 | 34 | +				APPEND(8);  | 
 | 35 | +			else if (*it->p == 't')  | 
 | 36 | +				APPEND(9);  | 
 | 37 | +			else if (*it->p == 'n')  | 
 | 38 | +				APPEND(10);  | 
 | 39 | +			else if (*it->p == 'f')  | 
 | 40 | +				APPEND(12);  | 
 | 41 | +			else if (*it->p == 'r')  | 
 | 42 | +				APPEND(13);  | 
 | 43 | +			else if (*it->p == 'u') {  | 
 | 44 | +				unsigned char binary[2];  | 
 | 45 | +				int i;  | 
 | 46 | + | 
 | 47 | +				if (hex_to_bytes(binary, it->p + 1, 2) < 0)  | 
 | 48 | +					return error("invalid: '%.*s'",  | 
 | 49 | +						     6, it->p - 1),  | 
 | 50 | +						reset_iterator(it);  | 
 | 51 | +				it->p += 4;  | 
 | 52 | + | 
 | 53 | +				i = (binary[0] << 8) | binary[1];  | 
 | 54 | +				if (i < 0x80)  | 
 | 55 | +					APPEND(i);  | 
 | 56 | +				else if (i < 0x0800) {  | 
 | 57 | +					APPEND(0xc0 | ((i >> 6) & 0x1f));  | 
 | 58 | +					APPEND(0x80 | (i & 0x3f));  | 
 | 59 | +				} else if (i < 0x10000) {  | 
 | 60 | +					APPEND(0xe0 | ((i >> 12) & 0x0f));  | 
 | 61 | +					APPEND(0x80 | ((i >> 6) & 0x3f));  | 
 | 62 | +					APPEND(0x80 | (i & 0x3f));  | 
 | 63 | +				} else {  | 
 | 64 | +					APPEND(0xf0 | ((i >> 18) & 0x07));  | 
 | 65 | +					APPEND(0x80 | ((i >> 12) & 0x3f));  | 
 | 66 | +					APPEND(0x80 | ((i >> 6) & 0x3f));  | 
 | 67 | +					APPEND(0x80 | (i & 0x3f));  | 
 | 68 | +				}  | 
 | 69 | +			}  | 
 | 70 | +			break;  | 
 | 71 | +		default:  | 
 | 72 | +			APPEND(*it->p);  | 
 | 73 | +		}  | 
 | 74 | +		it->p++;  | 
 | 75 | +	}  | 
 | 76 | + | 
 | 77 | +	it->end = it->p++;  | 
 | 78 | +	return 0;  | 
 | 79 | +}  | 
 | 80 | + | 
 | 81 | +static void skip_whitespace(struct json_iterator *it)  | 
 | 82 | +{  | 
 | 83 | +	while (isspace(*it->p))  | 
 | 84 | +		it->p++;  | 
 | 85 | +}  | 
 | 86 | + | 
 | 87 | +int iterate_json(struct json_iterator *it)  | 
 | 88 | +{  | 
 | 89 | +	skip_whitespace(it);  | 
 | 90 | +	it->begin = it->p;  | 
 | 91 | + | 
 | 92 | +	switch (*it->p) {  | 
 | 93 | +	case '\0':  | 
 | 94 | +		return reset_iterator(it), 0;  | 
 | 95 | +	case 'n':  | 
 | 96 | +		if (!starts_with(it->p, "null"))  | 
 | 97 | +			return error("unexpected value: %.*s", 4, it->p),  | 
 | 98 | +				reset_iterator(it);  | 
 | 99 | +		it->type = JSON_NULL;  | 
 | 100 | +		it->end = it->p = it->begin + 4;  | 
 | 101 | +		break;  | 
 | 102 | +	case 't':  | 
 | 103 | +		if (!starts_with(it->p, "true"))  | 
 | 104 | +			return error("unexpected value: %.*s", 4, it->p),  | 
 | 105 | +				reset_iterator(it);  | 
 | 106 | +		it->type = JSON_TRUE;  | 
 | 107 | +		it->end = it->p = it->begin + 4;  | 
 | 108 | +		break;  | 
 | 109 | +	case 'f':  | 
 | 110 | +		if (!starts_with(it->p, "false"))  | 
 | 111 | +			return error("unexpected value: %.*s", 5, it->p),  | 
 | 112 | +				reset_iterator(it);  | 
 | 113 | +		it->type = JSON_FALSE;  | 
 | 114 | +		it->end = it->p = it->begin + 5;  | 
 | 115 | +		break;  | 
 | 116 | +	case '-': case '.':  | 
 | 117 | +	case '0': case '1': case '2': case '3': case '4':  | 
 | 118 | +	case '5': case '6': case '7': case '8': case '9':  | 
 | 119 | +		it->type = JSON_NUMBER;  | 
 | 120 | +		it->end = it->p = it->begin + strspn(it->p, "-.0123456789");  | 
 | 121 | +		break;  | 
 | 122 | +	case '"':  | 
 | 123 | +		it->type = JSON_STRING;  | 
 | 124 | +		if (parse_json_string(it, &it->string_value) < 0)  | 
 | 125 | +			return -1;  | 
 | 126 | +		break;  | 
 | 127 | +	case '[': {  | 
 | 128 | +		const char *save = it->begin;  | 
 | 129 | +		size_t key_offset = it->key.len;  | 
 | 130 | +		int i = 0, res;  | 
 | 131 | + | 
 | 132 | +		for (it->p++, skip_whitespace(it); *it->p != ']'; i++) {  | 
 | 133 | +			strbuf_addf(&it->key, "[%d]", i);  | 
 | 134 | + | 
 | 135 | +			if ((res = iterate_json(it)))  | 
 | 136 | +				return reset_iterator(it), res;  | 
 | 137 | +			strbuf_setlen(&it->key, key_offset);  | 
 | 138 | + | 
 | 139 | +			skip_whitespace(it);  | 
 | 140 | +			if (*it->p == ',')  | 
 | 141 | +				it->p++;  | 
 | 142 | +		}  | 
 | 143 | + | 
 | 144 | +		it->type = JSON_ARRAY;  | 
 | 145 | +		it->begin = save;  | 
 | 146 | +		it->end = it->p;  | 
 | 147 | +		it->p++;  | 
 | 148 | +		break;  | 
 | 149 | +	}  | 
 | 150 | +	case '{': {  | 
 | 151 | +		const char *save = it->begin;  | 
 | 152 | +		size_t key_offset = it->key.len;  | 
 | 153 | +		int res;  | 
 | 154 | + | 
 | 155 | +		strbuf_addch(&it->key, '.');  | 
 | 156 | +		for (it->p++, skip_whitespace(it); *it->p != '}'; ) {  | 
 | 157 | +			strbuf_setlen(&it->key, key_offset + 1);  | 
 | 158 | +			if (parse_json_string(it, &it->key) < 0)  | 
 | 159 | +				return -1;  | 
 | 160 | +			skip_whitespace(it);  | 
 | 161 | +			if (*(it->p)++ != ':')  | 
 | 162 | +				return error("expected colon: %.*s", 5, it->p),  | 
 | 163 | +					reset_iterator(it);  | 
 | 164 | + | 
 | 165 | +			if ((res = iterate_json(it)))  | 
 | 166 | +				return res;  | 
 | 167 | + | 
 | 168 | +			skip_whitespace(it);  | 
 | 169 | +			if (*it->p == ',')  | 
 | 170 | +				it->p++;  | 
 | 171 | +		}  | 
 | 172 | +		strbuf_setlen(&it->key, key_offset);  | 
 | 173 | + | 
 | 174 | +		it->type = JSON_OBJECT;  | 
 | 175 | +		it->begin = save;  | 
 | 176 | +		it->end = it->p;  | 
 | 177 | +		it->p++;  | 
 | 178 | +		break;  | 
 | 179 | +	}  | 
 | 180 | +	}  | 
 | 181 | + | 
 | 182 | +	return it->fn(it);  | 
 | 183 | +}  | 
0 commit comments