lib/util/asn1.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    simple ASN1 routines
   4    Copyright (C) Andrew Tridgell 2001
   5
   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 3 of the License, or
   9    (at your option) any later version.
  10
  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.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "replace.h"
  21 #include "system/locale.h"
  22 #include "lib/util/asn1.h"
  23 #include "lib/util/debug.h"
  24 #include "lib/util/samba_util.h"
  25
  26 struct nesting {
  27         off_t start;
  28         size_t taglen; /* for parsing */
  29         struct nesting *next;
  30 };
  31
  32
  33 struct asn1_data {
  34         uint8_t *data;
  35         size_t length;
  36         off_t ofs;
  37         struct nesting *nesting;
  38         bool has_error;
  39 };
  40
  41 /* allocate an asn1 structure */
  42 struct asn1_data *asn1_init(TALLOC_CTX *mem_ctx)
  43 {
  44         struct asn1_data *ret = talloc_zero(mem_ctx, struct asn1_data);
  45         if (ret == NULL) {
  46                 DEBUG(0,("asn1_init failed! out of memory\n"));
  47         }
  48         return ret;
  49 }
  50
  51 /* free an asn1 structure */
  52 void asn1_free(struct asn1_data *data)
  53 {
  54         talloc_free(data);
  55 }
  56
  57 bool asn1_has_error(const struct asn1_data *data)
  58 {
  59         return data->has_error;
  60 }
  61
  62 void asn1_set_error(struct asn1_data *data)
  63 {
  64         data->has_error = true;
  65 }
  66
  67 bool asn1_has_nesting(const struct asn1_data *data)
  68 {
  69         return data->nesting != NULL;
  70 }
  71
  72 off_t asn1_current_ofs(const struct asn1_data *data)
  73 {
  74         return data->ofs;
  75 }
  76
  77 /* write to the ASN1 buffer, advancing the buffer pointer */
  78 bool asn1_write(struct asn1_data *data, const void *p, int len)
  79 {
  80         if (data->has_error) return false;
  81
  82         if ((len < 0) || (data->ofs + (size_t)len < data->ofs)) {
  83                 data->has_error = true;
  84                 return false;
  85         }
  86
  87         if (data->length < data->ofs+len) {
  88                 uint8_t *newp;
  89                 newp = talloc_realloc(data, data->data, uint8_t, data->ofs+len);
  90                 if (!newp) {
  91                         data->has_error = true;
  92                         return false;
  93                 }
  94                 data->data = newp;
  95                 data->length = data->ofs+len;
  96         }
  97         memcpy(data->data + data->ofs, p, len);
  98         data->ofs += len;
  99         return true;
 100 }
 101
 102 /* useful fn for writing a uint8_t */
 103 bool asn1_write_uint8(struct asn1_data *data, uint8_t v)
 104 {
 105         return asn1_write(data, &v, 1);
 106 }
 107
 108 /* push a tag onto the asn1 data buffer. Used for nested structures */
 109 bool asn1_push_tag(struct asn1_data *data, uint8_t tag)
 110 {
 111         struct nesting *nesting;
 112
 113         if (!asn1_write_uint8(data, tag)) {
 114                 return false;
 115         }
 116         nesting = talloc(data, struct nesting);
 117         if (!nesting) {
 118                 data->has_error = true;
 119                 return false;
 120         }
 121
 122         nesting->start = data->ofs;
 123         nesting->next = data->nesting;
 124         data->nesting = nesting;
 125         return asn1_write_uint8(data, 0xff);
 126 }
 127
 128 /* pop a tag */
 129 bool asn1_pop_tag(struct asn1_data *data)
 130 {
 131         struct nesting *nesting;
 132         size_t len;
 133
 134         if (data->has_error) {
 135                 return false;
 136         }
 137
 138         nesting = data->nesting;
 139
 140         if (!nesting) {
 141                 data->has_error = true;
 142                 return false;
 143         }
 144         len = data->ofs - (nesting->start+1);
 145         /* yes, this is ugly. We don't know in advance how many bytes the length
 146            of a tag will take, so we assumed 1 byte. If we were wrong then we
 147            need to correct our mistake */
 148         if (len > 0xFFFFFF) {
 149                 data->data[nesting->start] = 0x84;
 150                 if (!asn1_write_uint8(data, 0)) return false;
 151                 if (!asn1_write_uint8(data, 0)) return false;
 152                 if (!asn1_write_uint8(data, 0)) return false;
 153                 if (!asn1_write_uint8(data, 0)) return false;
 154                 memmove(data->data+nesting->start+5, data->data+nesting->start+1, len);
 155                 data->data[nesting->start+1] = (len>>24) & 0xFF;
 156                 data->data[nesting->start+2] = (len>>16) & 0xFF;
 157                 data->data[nesting->start+3] = (len>>8) & 0xFF;
 158                 data->data[nesting->start+4] = len&0xff;
 159         } else if (len > 0xFFFF) {
 160                 data->data[nesting->start] = 0x83;
 161                 if (!asn1_write_uint8(data, 0)) return false;
 162                 if (!asn1_write_uint8(data, 0)) return false;
 163                 if (!asn1_write_uint8(data, 0)) return false;
 164                 memmove(data->data+nesting->start+4, data->data+nesting->start+1, len);
 165                 data->data[nesting->start+1] = (len>>16) & 0xFF;
 166                 data->data[nesting->start+2] = (len>>8) & 0xFF;
 167                 data->data[nesting->start+3] = len&0xff;
 168         } else if (len > 255) {
 169                 data->data[nesting->start] = 0x82;
 170                 if (!asn1_write_uint8(data, 0)) return false;
 171                 if (!asn1_write_uint8(data, 0)) return false;
 172                 memmove(data->data+nesting->start+3, data->data+nesting->start+1, len);
 173                 data->data[nesting->start+1] = len>>8;
 174                 data->data[nesting->start+2] = len&0xff;
 175         } else if (len > 127) {
 176                 data->data[nesting->start] = 0x81;
 177                 if (!asn1_write_uint8(data, 0)) return false;
 178                 memmove(data->data+nesting->start+2, data->data+nesting->start+1, len);
 179                 data->data[nesting->start+1] = len;
 180         } else {
 181                 data->data[nesting->start] = len;
 182         }
 183
 184         data->nesting = nesting->next;
 185         talloc_free(nesting);
 186         return true;
 187 }
 188
 189 /* "i" is the one's complement representation, as is the normal result of an
 190  * implicit signed->unsigned conversion */
 191
 192 static bool push_int_bigendian(struct asn1_data *data, unsigned int i, bool negative)
 193 {
 194         uint8_t lowest = i & 0xFF;
 195
 196         i = i >> 8;
 197         if (i != 0)
 198                 if (!push_int_bigendian(data, i, negative))
 199                         return false;
 200
 201         if (data->nesting->start+1 == data->ofs) {
 202
 203                 /* We did not write anything yet, looking at the highest
 204                  * valued byte */
 205
 206                 if (negative) {
 207                         /* Don't write leading 0xff's */
 208                         if (lowest == 0xFF)
 209                                 return true;
 210
 211                         if ((lowest & 0x80) == 0) {
 212                                 /* The only exception for a leading 0xff is if
 213                                  * the highest bit is 0, which would indicate
 214                                  * a positive value */
 215                                 if (!asn1_write_uint8(data, 0xff))
 216                                         return false;
 217                         }
 218                 } else {
 219                         if (lowest & 0x80) {
 220                                 /* The highest bit of a positive integer is 1,
 221                                  * this would indicate a negative number. Push
 222                                  * a 0 to indicate a positive one */
 223                                 if (!asn1_write_uint8(data, 0))
 224                                         return false;
 225                         }
 226                 }
 227         }
 228
 229         return asn1_write_uint8(data, lowest);
 230 }
 231
 232 /* write an Integer without the tag framing. Needed for example for the LDAP
 233  * Abandon Operation */
 234
 235 bool asn1_write_implicit_Integer(struct asn1_data *data, int i)
 236 {
 237         if (data->has_error) {
 238                 return false;
 239         }
 240
 241         if (i == -1) {
 242                 /* -1 is special as it consists of all-0xff bytes. In
 243                     push_int_bigendian this is the only case that is not
 244                     properly handled, as all 0xff bytes would be handled as
 245                     leading ones to be ignored. */
 246                 return asn1_write_uint8(data, 0xff);
 247         } else {
 248                 return push_int_bigendian(data, i, i<0);
 249         }
 250 }
 251
 252
 253 /* write an integer */
 254 bool asn1_write_Integer(struct asn1_data *data, int i)
 255 {
 256         if (!asn1_push_tag(data, ASN1_INTEGER)) return false;
 257         if (!asn1_write_implicit_Integer(data, i)) return false;
 258         return asn1_pop_tag(data);
 259 }
 260
 261 /* write a BIT STRING */
 262 bool asn1_write_BitString(struct asn1_data *data, const void *p, size_t length, uint8_t padding)
 263 {
 264         if (!asn1_push_tag(data, ASN1_BIT_STRING)) return false;
 265         if (!asn1_write_uint8(data, padding)) return false;
 266         if (!asn1_write(data, p, length)) return false;
 267         return asn1_pop_tag(data);
 268 }
 269
 270 bool ber_write_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *OID)
 271 {
 272         unsigned int v, v2;
 273         const char *p = (const char *)OID;
 274         char *newp;
 275         int i;
 276
 277         if (!isdigit(*p)) return false;
 278         v = strtoul(p, &newp, 10);
 279         if (newp[0] != '.') return false;
 280         p = newp + 1;
 281
 282         if (!isdigit(*p)) return false;
 283         v2 = strtoul(p, &newp, 10);
 284         if (newp[0] != '.') return false;
 285         p = newp + 1;
 286
 287         /*the ber representation can't use more space than the string one */
 288         *blob = data_blob_talloc(mem_ctx, NULL, strlen(OID));
 289         if (!blob->data) return false;
 290
 291         blob->data[0] = 40*v + v2;
 292
 293         i = 1;
 294         while (*p) {
 295                 if (!isdigit(*p)) return false;
 296                 v = strtoul(p, &newp, 10);
 297                 if (newp[0] == '.') {
 298                         p = newp + 1;
 299                         /* check for empty last component */
 300                         if (!*p) return false;
 301                 } else if (newp[0] == '\0') {
 302                         p = newp;
 303                 } else {
 304                         data_blob_free(blob);
 305                         return false;
 306                 }
 307                 if (v >= (1<<28)) blob->data[i++] = (0x80 | ((v>>28)&0x7f));
 308                 if (v >= (1<<21)) blob->data[i++] = (0x80 | ((v>>21)&0x7f));
 309                 if (v >= (1<<14)) blob->data[i++] = (0x80 | ((v>>14)&0x7f));
 310                 if (v >= (1<<7)) blob->data[i++] = (0x80 | ((v>>7)&0x7f));
 311                 blob->data[i++] = (v&0x7f);
 312         }
 313
 314         blob->length = i;
 315
 316         return true;
 317 }
 318
 319 /**
 320  * Serialize partial OID string.
 321  * Partial OIDs are in the form:
 322  *   1:2.5.6:0x81
 323  *   1:2.5.6:0x8182
 324  */
 325 bool ber_write_partial_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, const char *partial_oid)
 326 {
 327         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 328         char *oid = talloc_strdup(tmp_ctx, partial_oid);
 329         char *p;
 330
 331         /* truncate partial part so ber_write_OID_String() works */
 332         p = strchr(oid, ':');
 333         if (p) {
 334                 *p = '\0';
 335                 p++;
 336         }
 337
 338         if (!ber_write_OID_String(mem_ctx, blob, oid)) {
 339                 talloc_free(tmp_ctx);
 340                 return false;
 341         }
 342
 343         /* Add partially encoded sub-identifier */
 344         if (p) {
 345                 DATA_BLOB tmp_blob = strhex_to_data_blob(tmp_ctx, p);
 346                 if (!data_blob_append(mem_ctx, blob, tmp_blob.data,
 347                                       tmp_blob.length)) {
 348                         talloc_free(tmp_ctx);
 349                         return false;
 350                 }
 351         }
 352
 353         talloc_free(tmp_ctx);
 354
 355         return true;
 356 }
 357
 358 /* write an object ID to a ASN1 buffer */
 359 bool asn1_write_OID(struct asn1_data *data, const char *OID)
 360 {
 361         DATA_BLOB blob;
 362
 363         if (!asn1_push_tag(data, ASN1_OID)) return false;
 364
 365         if (!ber_write_OID_String(NULL, &blob, OID)) {
 366                 data->has_error = true;
 367                 return false;
 368         }
 369
 370         if (!asn1_write(data, blob.data, blob.length)) {
 371                 data_blob_free(&blob);
 372                 data->has_error = true;
 373                 return false;
 374         }
 375         data_blob_free(&blob);
 376         return asn1_pop_tag(data);
 377 }
 378
 379 /* write an octet string */
 380 bool asn1_write_OctetString(struct asn1_data *data, const void *p, size_t length)
 381 {
 382         if (!asn1_push_tag(data, ASN1_OCTET_STRING)) return false;
 383         if (!asn1_write(data, p, length)) return false;
 384         return asn1_pop_tag(data);
 385 }
 386
 387 /* write a LDAP string */
 388 bool asn1_write_LDAPString(struct asn1_data *data, const char *s)
 389 {
 390         return asn1_write(data, s, strlen(s));
 391 }
 392
 393 /* write a LDAP string from a DATA_BLOB */
 394 bool asn1_write_DATA_BLOB_LDAPString(struct asn1_data *data, const DATA_BLOB *s)
 395 {
 396         return asn1_write(data, s->data, s->length);
 397 }
 398
 399 /* write a general string */
 400 bool asn1_write_GeneralString(struct asn1_data *data, const char *s)
 401 {
 402         if (!asn1_push_tag(data, ASN1_GENERAL_STRING)) return false;
 403         if (!asn1_write_LDAPString(data, s)) return false;
 404         return asn1_pop_tag(data);
 405 }
 406
 407 bool asn1_write_ContextSimple(struct asn1_data *data, uint8_t num, DATA_BLOB *blob)
 408 {
 409         if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(num))) return false;
 410         if (!asn1_write(data, blob->data, blob->length)) return false;
 411         return asn1_pop_tag(data);
 412 }
 413
 414 /* write a BOOLEAN */
 415 bool asn1_write_BOOLEAN(struct asn1_data *data, bool v)
 416 {
 417         if (!asn1_push_tag(data, ASN1_BOOLEAN)) return false;
 418         if (!asn1_write_uint8(data, v ? 0xFF : 0)) return false;
 419         return asn1_pop_tag(data);
 420 }
 421
 422 bool asn1_read_BOOLEAN(struct asn1_data *data, bool *v)
 423 {
 424         uint8_t tmp = 0;
 425         if (!asn1_start_tag(data, ASN1_BOOLEAN)) return false;
 426         *v = false;
 427         if (!asn1_read_uint8(data, &tmp)) return false;
 428         if (tmp == 0xFF) {
 429                 *v = true;
 430         }
 431         return asn1_end_tag(data);
 432 }
 433
 434 /* write a BOOLEAN in a simple context */
 435 bool asn1_write_BOOLEAN_context(struct asn1_data *data, bool v, int context)
 436 {
 437         if (!asn1_push_tag(data, ASN1_CONTEXT_SIMPLE(context))) return false;
 438         if (!asn1_write_uint8(data, v ? 0xFF : 0)) return false;
 439         return asn1_pop_tag(data);
 440 }
 441
 442 bool asn1_read_BOOLEAN_context(struct asn1_data *data, bool *v, int context)
 443 {
 444         uint8_t tmp = 0;
 445         if (!asn1_start_tag(data, ASN1_CONTEXT_SIMPLE(context))) return false;
 446         *v = false;
 447         if (!asn1_read_uint8(data, &tmp)) return false;
 448         if (tmp == 0xFF) {
 449                 *v = true;
 450         }
 451         return asn1_end_tag(data);
 452 }
 453
 454 /* check a BOOLEAN */
 455 bool asn1_check_BOOLEAN(struct asn1_data *data, bool v)
 456 {
 457         uint8_t b = 0;
 458
 459         if (!asn1_read_uint8(data, &b)) return false;
 460         if (b != ASN1_BOOLEAN) {
 461                 data->has_error = true;
 462                 return false;
 463         }
 464         if (!asn1_read_uint8(data, &b)) return false;
 465         if (b != v) {
 466                 data->has_error = true;
 467                 return false;
 468         }
 469         return !data->has_error;
 470 }
 471
 472
 473 /* load a struct asn1_data structure with a lump of data, ready to be parsed */
 474 bool asn1_load(struct asn1_data *data, DATA_BLOB blob)
 475 {
 476         ZERO_STRUCTP(data);
 477         data->data = (uint8_t *)talloc_memdup(data, blob.data, blob.length);
 478         if (!data->data) {
 479                 data->has_error = true;
 480                 return false;
 481         }
 482         data->length = blob.length;
 483         return true;
 484 }
 485
 486 /* Peek into an ASN1 buffer, not advancing the pointer */
 487 bool asn1_peek(struct asn1_data *data, void *p, int len)
 488 {
 489         if (data->has_error)
 490                 return false;
 491
 492         if (len < 0 || data->ofs + len < data->ofs || data->ofs + len < len)
 493                 return false;
 494
 495         if (data->ofs + len > data->length) {
 496                 /* we need to mark the buffer as consumed, so the caller knows
 497                    this was an out of data error, and not a decode error */
 498                 data->ofs = data->length;
 499                 return false;
 500         }
 501
 502         memcpy(p, data->data + data->ofs, len);
 503         return true;
 504 }
 505
 506 /* read from a ASN1 buffer, advancing the buffer pointer */
 507 bool asn1_read(struct asn1_data *data, void *p, int len)
 508 {
 509         if (!asn1_peek(data, p, len)) {
 510                 data->has_error = true;
 511                 return false;
 512         }
 513
 514         data->ofs += len;
 515         return true;
 516 }
 517
 518 /* read a uint8_t from a ASN1 buffer */
 519 bool asn1_read_uint8(struct asn1_data *data, uint8_t *v)
 520 {
 521         return asn1_read(data, v, 1);
 522 }
 523
 524 bool asn1_peek_uint8(struct asn1_data *data, uint8_t *v)
 525 {
 526         return asn1_peek(data, v, 1);
 527 }
 528
 529 bool asn1_peek_tag(struct asn1_data *data, uint8_t tag)
 530 {
 531         uint8_t b;
 532
 533         if (asn1_tag_remaining(data) <= 0) {
 534                 return false;
 535         }
 536
 537         if (!asn1_peek_uint8(data, &b))
 538                 return false;
 539
 540         return (b == tag);
 541 }
 542
 543 /*
 544  * just get the needed size the tag would consume
 545  */
 546 bool asn1_peek_tag_needed_size(struct asn1_data *data, uint8_t tag, size_t *size)
 547 {
 548         off_t start_ofs = data->ofs;
 549         uint8_t b;
 550         size_t taglen = 0;
 551
 552         if (data->has_error) {
 553                 return false;
 554         }
 555
 556         if (!asn1_read_uint8(data, &b)) {
 557                 data->ofs = start_ofs;
 558                 data->has_error = false;
 559                 return false;
 560         }
 561
 562         if (b != tag) {
 563                 data->ofs = start_ofs;
 564                 data->has_error = false;
 565                 return false;
 566         }
 567
 568         if (!asn1_read_uint8(data, &b)) {
 569                 data->ofs = start_ofs;
 570                 data->has_error = false;
 571                 return false;
 572         }
 573
 574         if (b & 0x80) {
 575                 int n = b & 0x7f;
 576                 if (!asn1_read_uint8(data, &b)) {
 577                         data->ofs = start_ofs;
 578                         data->has_error = false;
 579                         return false;
 580                 }
 581                 if (n > 4) {
 582                         /*
 583                          * We should not allow more than 4 bytes
 584                          * for the encoding of the tag length.
 585                          *
 586                          * Otherwise we'd overflow the taglen
 587                          * variable on 32 bit systems.
 588                          */
 589                         data->ofs = start_ofs;
 590                         data->has_error = false;
 591                         return false;
 592                 }
 593                 taglen = b;
 594                 while (n > 1) {
 595                         if (!asn1_read_uint8(data, &b)) {
 596                                 data->ofs = start_ofs;
 597                                 data->has_error = false;
 598                                 return false;
 599                         }
 600                         taglen = (taglen << 8) | b;
 601                         n--;
 602                 }
 603         } else {
 604                 taglen = b;
 605         }
 606
 607         *size = (data->ofs - start_ofs) + taglen;
 608
 609         data->ofs = start_ofs;
 610         data->has_error = false;
 611         return true;
 612 }
 613
 614 /* start reading a nested asn1 structure */
 615 bool asn1_start_tag(struct asn1_data *data, uint8_t tag)
 616 {
 617         uint8_t b;
 618         struct nesting *nesting;
 619
 620         if (!asn1_read_uint8(data, &b))
 621                 return false;
 622
 623         if (b != tag) {
 624                 data->has_error = true;
 625                 return false;
 626         }
 627         nesting = talloc(data, struct nesting);
 628         if (!nesting) {
 629                 data->has_error = true;
 630                 return false;
 631         }
 632
 633         if (!asn1_read_uint8(data, &b)) {
 634                 return false;
 635         }
 636
 637         if (b & 0x80) {
 638                 int n = b & 0x7f;
 639                 if (!asn1_read_uint8(data, &b))
 640                         return false;
 641                 nesting->taglen = b;
 642                 while (n > 1) {
 643                         if (!asn1_read_uint8(data, &b))
 644                                 return false;
 645                         nesting->taglen = (nesting->taglen << 8) | b;
 646                         n--;
 647                 }
 648         } else {
 649                 nesting->taglen = b;
 650         }
 651         nesting->start = data->ofs;
 652         nesting->next = data->nesting;
 653         data->nesting = nesting;
 654         if (asn1_tag_remaining(data) == -1) {
 655                 return false;
 656         }
 657         return !data->has_error;
 658 }
 659
 660 /* stop reading a tag */
 661 bool asn1_end_tag(struct asn1_data *data)
 662 {
 663         struct nesting *nesting;
 664
 665         /* make sure we read it all */
 666         if (asn1_tag_remaining(data) != 0) {
 667                 data->has_error = true;
 668                 return false;
 669         }
 670
 671         nesting = data->nesting;
 672
 673         if (!nesting) {
 674                 data->has_error = true;
 675                 return false;
 676         }
 677
 678         data->nesting = nesting->next;
 679         talloc_free(nesting);
 680         return true;
 681 }
 682
 683 /* work out how many bytes are left in this nested tag */
 684 int asn1_tag_remaining(struct asn1_data *data)
 685 {
 686         int remaining;
 687         if (data->has_error) {
 688                 return -1;
 689         }
 690
 691         if (!data->nesting) {
 692                 data->has_error = true;
 693                 return -1;
 694         }
 695         remaining = data->nesting->taglen - (data->ofs - data->nesting->start);
 696         if (remaining > (data->length - data->ofs)) {
 697                 data->has_error = true;
 698                 return -1;
 699         }
 700         return remaining;
 701 }
 702
 703 /**
 704  * Internal implementation for reading binary OIDs
 705  * Reading is done as far in the buffer as valid OID
 706  * till buffer ends or not valid sub-identifier is found.
 707  */
 708 static bool _ber_read_OID_String_impl(TALLOC_CTX *mem_ctx, DATA_BLOB blob,
 709                                       char **OID, size_t *bytes_eaten)
 710 {
 711         int i;
 712         uint8_t *b;
 713         unsigned int v;
 714         char *tmp_oid = NULL;
 715
 716         if (blob.length < 2) return false;
 717
 718         b = blob.data;
 719
 720         tmp_oid = talloc_asprintf(mem_ctx, "%u",  b[0]/40);
 721         if (!tmp_oid) goto nomem;
 722         tmp_oid = talloc_asprintf_append_buffer(tmp_oid, ".%u",  b[0]%40);
 723         if (!tmp_oid) goto nomem;
 724
 725         if (bytes_eaten != NULL) {
 726                 *bytes_eaten = 0;
 727         }
 728
 729         for(i = 1, v = 0; i < blob.length; i++) {
 730                 v = (v<<7) | (b[i]&0x7f);
 731                 if ( ! (b[i] & 0x80)) {
 732                         tmp_oid = talloc_asprintf_append_buffer(tmp_oid, ".%u",  v);
 733                         v = 0;
 734                         if (bytes_eaten)
 735                                 *bytes_eaten = i+1;
 736                 }
 737                 if (!tmp_oid) goto nomem;
 738         }
 739
 740         *OID = tmp_oid;
 741         return true;
 742
 743 nomem:
 744         return false;
 745 }
 746
 747 /* read an object ID from a data blob */
 748 bool ber_read_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB blob, char **OID)
 749 {
 750         size_t bytes_eaten;
 751
 752         if (!_ber_read_OID_String_impl(mem_ctx, blob, OID, &bytes_eaten))
 753                 return false;
 754
 755         return (bytes_eaten == blob.length);
 756 }
 757
 758 /**
 759  * Deserialize partial OID string.
 760  * Partial OIDs are in the form:
 761  *   1:2.5.6:0x81
 762  *   1:2.5.6:0x8182
 763  */
 764 bool ber_read_partial_OID_String(TALLOC_CTX *mem_ctx, DATA_BLOB blob,
 765                                  char **partial_oid)
 766 {
 767         size_t bytes_left;
 768         size_t bytes_eaten;
 769         char *identifier = NULL;
 770         char *tmp_oid = NULL;
 771
 772         if (!_ber_read_OID_String_impl(mem_ctx, blob, &tmp_oid, &bytes_eaten))
 773                 return false;
 774
 775         if (bytes_eaten < blob.length) {
 776                 bytes_left = blob.length - bytes_eaten;
 777                 identifier = hex_encode_talloc(mem_ctx, &blob.data[bytes_eaten], bytes_left);
 778                 if (!identifier)        goto nomem;
 779
 780                 *partial_oid = talloc_asprintf_append_buffer(tmp_oid, ":0x%s", identifier);
 781                 if (!*partial_oid)      goto nomem;
 782                 TALLOC_FREE(identifier);
 783         } else {
 784                 *partial_oid = tmp_oid;
 785         }
 786
 787         return true;
 788
 789 nomem:
 790         TALLOC_FREE(identifier);
 791         TALLOC_FREE(tmp_oid);
 792         return false;
 793 }
 794
 795 /* read an object ID from a ASN1 buffer */
 796 bool asn1_read_OID(struct asn1_data *data, TALLOC_CTX *mem_ctx, char **OID)
 797 {
 798         DATA_BLOB blob;
 799         int len;
 800
 801         if (!asn1_start_tag(data, ASN1_OID)) return false;
 802
 803         len = asn1_tag_remaining(data);
 804         if (len < 0) {
 805                 data->has_error = true;
 806                 return false;
 807         }
 808
 809         blob = data_blob(NULL, len);
 810         if (!blob.data) {
 811                 data->has_error = true;
 812                 return false;
 813         }
 814
 815         if (!asn1_read(data, blob.data, len)) return false;
 816         if (!asn1_end_tag(data)) {
 817                 data_blob_free(&blob);
 818                 return false;
 819         }
 820
 821         if (!ber_read_OID_String(mem_ctx, blob, OID)) {
 822                 data->has_error = true;
 823                 data_blob_free(&blob);
 824                 return false;
 825         }
 826
 827         data_blob_free(&blob);
 828         return true;
 829 }
 830
 831 /* check that the next object ID is correct */
 832 bool asn1_check_OID(struct asn1_data *data, const char *OID)
 833 {
 834         char *id;
 835
 836         if (!asn1_read_OID(data, data, &id)) return false;
 837
 838         if (strcmp(id, OID) != 0) {
 839                 talloc_free(id);
 840                 data->has_error = true;
 841                 return false;
 842         }
 843         talloc_free(id);
 844         return true;
 845 }
 846
 847 /* read a LDAPString from a ASN1 buffer */
 848 bool asn1_read_LDAPString(struct asn1_data *data, TALLOC_CTX *mem_ctx, char **s)
 849 {
 850         int len;
 851         len = asn1_tag_remaining(data);
 852         if (len < 0) {
 853                 data->has_error = true;
 854                 return false;
 855         }
 856         *s = talloc_array(mem_ctx, char, len+1);
 857         if (! *s) {
 858                 data->has_error = true;
 859                 return false;
 860         }
 861         (*s)[len] = 0;
 862         return asn1_read(data, *s, len);
 863 }
 864
 865
 866 /* read a GeneralString from a ASN1 buffer */
 867 bool asn1_read_GeneralString(struct asn1_data *data, TALLOC_CTX *mem_ctx, char **s)
 868 {
 869         if (!asn1_start_tag(data, ASN1_GENERAL_STRING)) return false;
 870         if (!asn1_read_LDAPString(data, mem_ctx, s)) return false;
 871         return asn1_end_tag(data);
 872 }
 873
 874
 875 /* read a octet string blob */
 876 bool asn1_read_OctetString(struct asn1_data *data, TALLOC_CTX *mem_ctx, DATA_BLOB *blob)
 877 {
 878         int len;
 879         ZERO_STRUCTP(blob);
 880         if (!asn1_start_tag(data, ASN1_OCTET_STRING)) return false;
 881         len = asn1_tag_remaining(data);
 882         if (len < 0) {
 883                 data->has_error = true;
 884                 return false;
 885         }
 886         *blob = data_blob_talloc(mem_ctx, NULL, len+1);
 887         if (!blob->data || blob->length < len) {
 888                 data->has_error = true;
 889                 return false;
 890         }
 891         if (!asn1_read(data, blob->data, len)) goto err;
 892         if (!asn1_end_tag(data)) goto err;
 893         blob->length--;
 894         blob->data[len] = 0;
 895         return true;
 896
 897   err:
 898
 899         data_blob_free(blob);
 900         *blob = data_blob_null;
 901         return false;
 902 }
 903
 904 bool asn1_read_ContextSimple(struct asn1_data *data, uint8_t num, DATA_BLOB *blob)
 905 {
 906         int len;
 907         ZERO_STRUCTP(blob);
 908         if (!asn1_start_tag(data, ASN1_CONTEXT_SIMPLE(num))) return false;
 909         len = asn1_tag_remaining(data);
 910         if (len < 0) {
 911                 data->has_error = true;
 912                 return false;
 913         }
 914         *blob = data_blob(NULL, len);
 915         if ((len != 0) && (!blob->data)) {
 916                 data->has_error = true;
 917                 return false;
 918         }
 919         if (!asn1_read(data, blob->data, len)) return false;
 920         return asn1_end_tag(data);
 921 }
 922
 923 /* read an integer without tag*/
 924 bool asn1_read_implicit_Integer(struct asn1_data *data, int *i)
 925 {
 926         uint8_t b;
 927         bool first_byte = true;
 928         *i = 0;
 929
 930         while (!data->has_error && asn1_tag_remaining(data)>0) {
 931                 if (!asn1_read_uint8(data, &b)) return false;
 932                 if (first_byte) {
 933                         if (b & 0x80) {
 934                                 /* Number is negative.
 935                                    Set i to -1 for sign extend. */
 936                                 *i = -1;
 937                         }
 938                         first_byte = false;
 939                 }
 940                 *i = (*i << 8) + b;
 941         }
 942         return !data->has_error;
 943
 944 }
 945
 946 /* read an integer */
 947 bool asn1_read_Integer(struct asn1_data *data, int *i)
 948 {
 949         *i = 0;
 950
 951         if (!asn1_start_tag(data, ASN1_INTEGER)) return false;
 952         if (!asn1_read_implicit_Integer(data, i)) return false;
 953         return asn1_end_tag(data);
 954 }
 955
 956 /* read a BIT STRING */
 957 bool asn1_read_BitString(struct asn1_data *data, TALLOC_CTX *mem_ctx, DATA_BLOB *blob, uint8_t *padding)
 958 {
 959         int len;
 960         ZERO_STRUCTP(blob);
 961         if (!asn1_start_tag(data, ASN1_BIT_STRING)) return false;
 962         len = asn1_tag_remaining(data);
 963         if (len < 0) {
 964                 data->has_error = true;
 965                 return false;
 966         }
 967         if (!asn1_read_uint8(data, padding)) return false;
 968
 969         *blob = data_blob_talloc(mem_ctx, NULL, len+1);
 970         if (!blob->data || blob->length < len) {
 971                 data->has_error = true;
 972                 return false;
 973         }
 974         if (asn1_read(data, blob->data, len - 1)) {
 975                 blob->length--;
 976                 blob->data[len] = 0;
 977                 asn1_end_tag(data);
 978         }
 979
 980         if (data->has_error) {
 981                 data_blob_free(blob);
 982                 *blob = data_blob_null;
 983                 *padding = 0;
 984                 return false;
 985         }
 986         return true;
 987 }
 988
 989 /* read an integer */
 990 bool asn1_read_enumerated(struct asn1_data *data, int *v)
 991 {
 992         *v = 0;
 993
 994         if (!asn1_start_tag(data, ASN1_ENUMERATED)) return false;
 995         while (!data->has_error && asn1_tag_remaining(data)>0) {
 996                 uint8_t b;
 997                 if (!asn1_read_uint8(data, &b)) {
 998                         return false;
 999                 }
1000                 *v = (*v << 8) + b;
1001         }
1002         return asn1_end_tag(data);
1003 }
1004
1005 /* check a enumerated value is correct */
1006 bool asn1_check_enumerated(struct asn1_data *data, int v)
1007 {
1008         uint8_t b;
1009         if (!asn1_start_tag(data, ASN1_ENUMERATED)) return false;
1010         if (!asn1_read_uint8(data, &b)) return false;
1011         if (!asn1_end_tag(data)) return false;
1012
1013         if (v != b)
1014                 data->has_error = false;
1015
1016         return !data->has_error;
1017 }
1018
1019 /* write an enumerated value to the stream */
1020 bool asn1_write_enumerated(struct asn1_data *data, uint8_t v)
1021 {
1022         if (!asn1_push_tag(data, ASN1_ENUMERATED)) return false;
1023         if (!asn1_write_uint8(data, v)) return false;
1024         return asn1_pop_tag(data);
1025 }
1026
1027 /*
1028   Get us the data just written without copying
1029 */
1030 bool asn1_blob(const struct asn1_data *asn1, DATA_BLOB *blob)
1031 {
1032         if (asn1->has_error) {
1033                 return false;
1034         }
1035         if (asn1->nesting != NULL) {
1036                 return false;
1037         }
1038         blob->data = asn1->data;
1039         blob->length = asn1->length;
1040         return true;
1041 }
1042
1043 bool asn1_extract_blob(struct asn1_data *asn1, TALLOC_CTX *mem_ctx,
1044                        DATA_BLOB *pblob)
1045 {
1046         DATA_BLOB blob;
1047
1048         if (!asn1_blob(asn1, &blob)) {
1049                 return false;
1050         }
1051
1052         *pblob = (DATA_BLOB) { .length = blob.length };
1053         pblob->data = talloc_move(mem_ctx, &blob.data);
1054
1055         /*
1056          * Stop access from here on
1057          */
1058         asn1->has_error = true;
1059
1060         return true;
1061 }
1062
1063 /*
1064   Fill in an asn1 struct without making a copy
1065 */
1066 void asn1_load_nocopy(struct asn1_data *data, uint8_t *buf, size_t len)
1067 {
1068         ZERO_STRUCTP(data);
1069         data->data = buf;
1070         data->length = len;
1071 }
1072
1073 int asn1_peek_full_tag(DATA_BLOB blob, uint8_t tag, size_t *packet_size)
1074 {
1075         struct asn1_data asn1;
1076         size_t size;
1077         bool ok;
1078
1079         ZERO_STRUCT(asn1);
1080         asn1.data = blob.data;
1081         asn1.length = blob.length;
1082
1083         ok = asn1_peek_tag_needed_size(&asn1, tag, &size);
1084         if (!ok) {
1085                 return EMSGSIZE;
1086         }
1087
1088         if (size > blob.length) {
1089                 *packet_size = size;
1090                 return EAGAIN;
1091         }
1092
1093         *packet_size = size;
1094         return 0;
1095 }