aboutsummaryrefslogtreecommitdiff
path: root/client/parse.cpp
blob: 516445be5c1da672851a6f27ca010e6dd93a9c50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "parse.h"

static int parse_string(const char * str, char * data, size_t * offset) {
	char escape = false;
	int i = 0;
	size_t len = strlen(str);

	switch (str[i]) {
		case '\'':
			escape = false;
			break;
		case '\"':
			escape = true;
			break;
		default:
			return -i;
	}
	char closing = str[i];

	for (i = 1; i < len && str[i] != '\0'; i++, *offset += 1) {
		char c = str[i];

		if (c == closing) return i + 1; // +1 for closing quote

		if (escape && c == '\\') {
			char x = str[i + 1];
			if (x == '0') c = '\0';
			else if (x == 't') c = '\t';
			else if (x == 'n') c = '\n';
			else if (x == 'r') c = '\r';
			else if (x == '\\') c = '\\';
			else if (x == '\"') c = '\"';
			else if (x == '\'') c = '\'';
			else break;
			i++;
		}

		if (data != NULL) data[*offset] = c;
	}

	return -i;
}

static int parse_hexstr(const char * str, char * data, size_t * offset) {
	size_t len = strcspn(str, IFS);
	int i = 0;

	// check if token contains at least one colon
	const char * colon = strchr(str, ':');
	if (colon == NULL) return -i;
	if (colon >= str + len) return -i;

	// check if token only contains allowed characters [0-9a-fA-F:]
	size_t len_ok = strspn(str + i, SET_HEX_STR) + i;
	if (len != len_ok) return -len_ok;

	size_t c = 0;
	while (c < len) { // count bytes in bytestring
		if (strspn(str + c, SET_HEX) != 2) return -i - c;

		if (data != NULL) data[*offset] = strtol(str + c, NULL, 16) & 0xff;

		c += 2;
		*offset += 1;

		if (str[c] == ':') {
			c += 1;
			continue;
		}
		break;
	}

	i += len;
	return i;
}

static int parse_number(const char * str, char * data, size_t * offset) {
	size_t len = strcspn(str, IFS);
	int i = 0;
	int base = 10;
	bool bytestring = false;

	if (len > 2 && strncmp(str, "0x", 2) == 0) { // hexadecimal prefix
		base = 16;
		i += 2;
	} /* else if (len > 1 && strncmp(str, "0", 1) == 0) { // octal prefix
		base = 8;
		i += 1;
	}*/

	const char * set;
	// if (base == 8) set = SET_OCT;
	if (base == 10) set = SET_DEC;
	if (base == 16) set = SET_HEX;

	size_t len_ok = strspn(str + i, set) + i;
	if (len != len_ok) return -len_ok;

	size_t size = 1; // default integer size in bytes
	if (base == 16) {
		size_t prefixless = len - i;
		switch (prefixless) {
			case 2: //  8-bit (2 hex characters)
			case 4: // 16-bit
			case 8: // 32-bit
			case 16: // 64-bit
				break;
			default:
				return -i;
		}
		size = prefixless / 2;
	}

	if (data != NULL) {
		unsigned long number = strtol(str + i, NULL, base);
		long long mask = (1 << 8 * size) - 1;
		number &= mask;
		// NOTE: the hton? functions are used to convert host endianness to network
		// endianness (big), and are required
		switch (size) {
			case 1:
				data[*offset] = number & 0xff;
				break;
			case 2:
				number = htons(number);
				// TODO: check if the endianness is OK, or reverse these *offset indices*
				data[*offset + 1] = (number) & 0xff;
				data[*offset + 0] = (number >>= 8) & 0xff;
				break;
			case 4:
				number = htonl(number);
				data[*offset + 3] = (number) & 0xff;
				data[*offset + 2] = (number >>= 8) & 0xff;
				data[*offset + 1] = (number >>= 8) & 0xff;
				data[*offset + 0] = (number >>= 8) & 0xff;
				break;
		}
	}

	*offset += size;
	return len;
}

static int _strtodata_main(const char * str, char * data, size_t * offset) {
	size_t len = strlen(str);

	int i, run;
	for (i = 0; i < len; i += run) {
		i += strspn(&str[i], IFS); // skip whitespace
		if (str[i] == '\0') break; // end of string

		if ((run = parse_string(str + i, data, offset)) > 0) continue;
		if ((run = parse_hexstr(str + i, data, offset)) > 0) continue;
		if ((run = parse_number(str + i, data, offset)) > 0) continue;

		// no format detected
		return -i + run;
	}

	return i;
}

int strtodata(const char * str, char ** data, size_t * size) {
	*size = 0;

	// 1st pass: check data format
	int ret = _strtodata_main(str, NULL, size);
	if (ret <= 0) return ret; // on error

	// 2nd pass: convert string literals into binary data
	*data = (char *) malloc(*size);
	size_t written = 0;
	return _strtodata_main(str, *data, &written);
}