2 Unix SMB/CIFS implementation.
4 Copyright (C) Andrew Tridgell 2001
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 /* free an asn1 structure */
24 void asn1_free(ASN1_DATA *data)
26 SAFE_FREE(data->data);
29 /* write to the ASN1 buffer, advancing the buffer pointer */
30 BOOL asn1_write(ASN1_DATA *data, const void *p, int len)
32 if (data->has_error) return False;
33 if (data->length < data->ofs+len) {
34 data->data = SMB_REALLOC(data->data, data->ofs+len);
36 data->has_error = True;
39 data->length = data->ofs+len;
41 memcpy(data->data + data->ofs, p, len);
46 /* useful fn for writing a uint8 */
47 BOOL asn1_write_uint8(ASN1_DATA *data, uint8 v)
49 return asn1_write(data, &v, 1);
52 /* push a tag onto the asn1 data buffer. Used for nested structures */
53 BOOL asn1_push_tag(ASN1_DATA *data, uint8 tag)
55 struct nesting *nesting;
57 asn1_write_uint8(data, tag);
58 nesting = SMB_MALLOC_P(struct nesting);
60 data->has_error = True;
64 nesting->start = data->ofs;
65 nesting->next = data->nesting;
66 data->nesting = nesting;
67 return asn1_write_uint8(data, 0xff);
71 BOOL asn1_pop_tag(ASN1_DATA *data)
73 struct nesting *nesting;
76 nesting = data->nesting;
79 data->has_error = True;
82 len = data->ofs - (nesting->start+1);
83 /* yes, this is ugly. We don't know in advance how many bytes the length
84 of a tag will take, so we assumed 1 byte. If we were wrong then we
85 need to correct our mistake */
87 data->data[nesting->start] = 0x83;
88 if (!asn1_write_uint8(data, 0)) return False;
89 if (!asn1_write_uint8(data, 0)) return False;
90 if (!asn1_write_uint8(data, 0)) return False;
91 memmove(data->data+nesting->start+4, data->data+nesting->start+1, len);
92 data->data[nesting->start+1] = (len>>16) & 0xFF;
93 data->data[nesting->start+2] = (len>>8) & 0xFF;
94 data->data[nesting->start+3] = len&0xff;
95 } else if (len > 255) {
96 data->data[nesting->start] = 0x82;
97 if (!asn1_write_uint8(data, 0)) return False;
98 if (!asn1_write_uint8(data, 0)) return False;
99 memmove(data->data+nesting->start+3, data->data+nesting->start+1, len);
100 data->data[nesting->start+1] = len>>8;
101 data->data[nesting->start+2] = len&0xff;
102 } else if (len > 127) {
103 data->data[nesting->start] = 0x81;
104 if (!asn1_write_uint8(data, 0)) return False;
105 memmove(data->data+nesting->start+2, data->data+nesting->start+1, len);
106 data->data[nesting->start+1] = len;
108 data->data[nesting->start] = len;
111 data->nesting = nesting->next;
117 /* write an integer */
118 BOOL asn1_write_Integer(ASN1_DATA *data, int i)
120 if (!asn1_push_tag(data, ASN1_INTEGER)) return False;
122 asn1_write_uint8(data, i);
125 return asn1_pop_tag(data);
128 /* write an object ID to a ASN1 buffer */
129 BOOL asn1_write_OID(ASN1_DATA *data, const char *OID)
132 const char *p = (const char *)OID;
135 if (!asn1_push_tag(data, ASN1_OID))
137 v = strtol(p, &newp, 10);
139 v2 = strtol(p, &newp, 10);
141 if (!asn1_write_uint8(data, 40*v + v2))
145 v = strtol(p, &newp, 10);
147 if (v >= (1<<28)) asn1_write_uint8(data, 0x80 | ((v>>28)&0xff));
148 if (v >= (1<<21)) asn1_write_uint8(data, 0x80 | ((v>>21)&0xff));
149 if (v >= (1<<14)) asn1_write_uint8(data, 0x80 | ((v>>14)&0xff));
150 if (v >= (1<<7)) asn1_write_uint8(data, 0x80 | ((v>>7)&0xff));
151 if (!asn1_write_uint8(data, v&0x7f))
154 return asn1_pop_tag(data);
157 /* write an octet string */
158 BOOL asn1_write_OctetString(ASN1_DATA *data, const void *p, size_t length)
160 asn1_push_tag(data, ASN1_OCTET_STRING);
161 asn1_write(data, p, length);
163 return !data->has_error;
166 /* write a general string */
167 BOOL asn1_write_GeneralString(ASN1_DATA *data, const char *s)
169 asn1_push_tag(data, ASN1_GENERAL_STRING);
170 asn1_write(data, s, strlen(s));
172 return !data->has_error;
175 /* write a BOOLEAN */
176 BOOL asn1_write_BOOLEAN(ASN1_DATA *data, BOOL v)
178 asn1_write_uint8(data, ASN1_BOOLEAN);
179 asn1_write_uint8(data, v);
180 return !data->has_error;
183 /* write a BOOLEAN - hmm, I suspect this one is the correct one, and the
184 above boolean is bogus. Need to check */
185 BOOL asn1_write_BOOLEAN2(ASN1_DATA *data, BOOL v)
187 asn1_push_tag(data, ASN1_BOOLEAN);
188 asn1_write_uint8(data, v);
190 return !data->has_error;
193 /* check a BOOLEAN */
194 BOOL asn1_check_BOOLEAN(ASN1_DATA *data, BOOL v)
198 asn1_read_uint8(data, &b);
199 if (b != ASN1_BOOLEAN) {
200 data->has_error = True;
203 asn1_read_uint8(data, &b);
205 data->has_error = True;
208 return !data->has_error;
212 /* load a ASN1_DATA structure with a lump of data, ready to be parsed */
213 BOOL asn1_load(ASN1_DATA *data, DATA_BLOB blob)
216 data->data = memdup(blob.data, blob.length);
218 data->has_error = True;
221 data->length = blob.length;
225 /* read from a ASN1 buffer, advancing the buffer pointer */
226 BOOL asn1_read(ASN1_DATA *data, void *p, int len)
231 if (len < 0 || data->ofs + len < data->ofs || data->ofs + len < len) {
232 data->has_error = True;
236 if (data->ofs + len > data->length) {
237 data->has_error = True;
240 memcpy(p, data->data + data->ofs, len);
245 /* read a uint8 from a ASN1 buffer */
246 BOOL asn1_read_uint8(ASN1_DATA *data, uint8 *v)
248 return asn1_read(data, v, 1);
251 /* start reading a nested asn1 structure */
252 BOOL asn1_start_tag(ASN1_DATA *data, uint8 tag)
255 struct nesting *nesting;
257 if (!asn1_read_uint8(data, &b))
261 data->has_error = True;
264 nesting = SMB_MALLOC_P(struct nesting);
266 data->has_error = True;
270 if (!asn1_read_uint8(data, &b)) {
276 if (!asn1_read_uint8(data, &b))
280 if (!asn1_read_uint8(data, &b))
282 nesting->taglen = (nesting->taglen << 8) | b;
288 nesting->start = data->ofs;
289 nesting->next = data->nesting;
290 data->nesting = nesting;
291 return !data->has_error;
295 /* stop reading a tag */
296 BOOL asn1_end_tag(ASN1_DATA *data)
298 struct nesting *nesting;
300 /* make sure we read it all */
301 if (asn1_tag_remaining(data) != 0) {
302 data->has_error = True;
306 nesting = data->nesting;
309 data->has_error = True;
313 data->nesting = nesting->next;
318 /* work out how many bytes are left in this nested tag */
319 int asn1_tag_remaining(ASN1_DATA *data)
324 if (!data->nesting) {
325 data->has_error = True;
328 return data->nesting->taglen - (data->ofs - data->nesting->start);
331 /* read an object ID from a ASN1 buffer */
332 BOOL asn1_read_OID(ASN1_DATA *data, char **OID)
338 if (!asn1_start_tag(data, ASN1_OID)) return False;
339 asn1_read_uint8(data, &b);
342 fstr_sprintf(el, "%u", b/40);
343 pstrcat(oid_str, el);
344 fstr_sprintf(el, " %u", b%40);
345 pstrcat(oid_str, el);
347 while (asn1_tag_remaining(data) > 0) {
350 asn1_read_uint8(data, &b);
351 v = (v<<7) | (b&0x7f);
352 } while (!data->has_error && b & 0x80);
353 fstr_sprintf(el, " %u", v);
354 pstrcat(oid_str, el);
359 *OID = SMB_STRDUP(oid_str);
361 return !data->has_error;
364 /* check that the next object ID is correct */
365 BOOL asn1_check_OID(ASN1_DATA *data, const char *OID)
369 if (!asn1_read_OID(data, &id)) return False;
371 if (strcmp(id, OID) != 0) {
372 data->has_error = True;
379 /* read a GeneralString from a ASN1 buffer */
380 BOOL asn1_read_GeneralString(ASN1_DATA *data, char **s)
383 if (!asn1_start_tag(data, ASN1_GENERAL_STRING)) return False;
384 len = asn1_tag_remaining(data);
386 data->has_error = True;
389 *s = SMB_MALLOC(len+1);
391 data->has_error = True;
394 asn1_read(data, *s, len);
397 return !data->has_error;
400 /* read a octet string blob */
401 BOOL asn1_read_OctetString(ASN1_DATA *data, DATA_BLOB *blob)
405 if (!asn1_start_tag(data, ASN1_OCTET_STRING)) return False;
406 len = asn1_tag_remaining(data);
408 data->has_error = True;
411 *blob = data_blob(NULL, len);
412 asn1_read(data, blob->data, len);
414 return !data->has_error;
417 /* read an interger */
418 BOOL asn1_read_Integer(ASN1_DATA *data, int *i)
423 if (!asn1_start_tag(data, ASN1_INTEGER)) return False;
424 while (asn1_tag_remaining(data)>0) {
425 asn1_read_uint8(data, &b);
428 return asn1_end_tag(data);
432 /* check a enumarted value is correct */
433 BOOL asn1_check_enumerated(ASN1_DATA *data, int v)
436 if (!asn1_start_tag(data, ASN1_ENUMERATED)) return False;
437 asn1_read_uint8(data, &b);
441 data->has_error = False;
443 return !data->has_error;
446 /* write an enumarted value to the stream */
447 BOOL asn1_write_enumerated(ASN1_DATA *data, uint8 v)
449 if (!asn1_push_tag(data, ASN1_ENUMERATED)) return False;
450 asn1_write_uint8(data, v);
452 return !data->has_error;