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