libs/smux/per_support.c

   1 /*
   2  * Copyright (c) 2005-2014 Lev Walkin <vlm@lionet.info>.
   3  * All rights reserved.
   4  * Redistribution and modifications are permitted subject to BSD license.
   5  */
   6 #include <asn_system.h>
   7 #include <asn_internal.h>
   8 #include <per_support.h>
   9
  10 char *
  11 per_data_string(asn_per_data_t *pd) {
  12         static char buf[2][32];
  13         static int n;
  14         n = (n+1) % 2;
  15         snprintf(buf[n], sizeof(buf[n]),
  16                 "{m=%ld span %+ld[%d..%d] (%d)}",
  17                 (long)pd->moved,
  18                 (((long)pd->buffer) & 0xf),
  19                 (int)pd->nboff, (int)pd->nbits,
  20                 (int)(pd->nbits - pd->nboff));
  21         return buf[n];
  22 }
  23
  24 void
  25 per_get_undo(asn_per_data_t *pd, int nbits) {
  26         if((ssize_t)pd->nboff < nbits) {
  27                 assert((ssize_t)pd->nboff < nbits);
  28         } else {
  29                 pd->nboff -= nbits;
  30                 pd->moved -= nbits;
  31         }
  32 }
  33
  34 /*
  35  * Extract a small number of bits (<= 31) from the specified PER data pointer.
  36  */
  37 int32_t
  38 per_get_few_bits(asn_per_data_t *pd, int nbits) {
  39         size_t off;     /* Next after last bit offset */
  40         ssize_t nleft;  /* Number of bits left in this stream */
  41         uint32_t accum;
  42         const uint8_t *buf;
  43
  44         if(nbits < 0)
  45                 return -1;
  46
  47         nleft = pd->nbits - pd->nboff;
  48         if(nbits > nleft) {
  49                 int32_t tailv, vhead;
  50                 if(!pd->refill || nbits > 31) return -1;
  51                 /* Accumulate unused bytes before refill */
  52                 ASN_DEBUG("Obtain the rest %d bits (want %d)",
  53                         (int)nleft, (int)nbits);
  54                 tailv = per_get_few_bits(pd, nleft);
  55                 if(tailv < 0) return -1;
  56                 /* Refill (replace pd contents with new data) */
  57                 if(pd->refill(pd))
  58                         return -1;
  59                 nbits -= nleft;
  60                 vhead = per_get_few_bits(pd, nbits);
  61                 /* Combine the rest of previous pd with the head of new one */
  62                 tailv = (tailv << nbits) | vhead;  /* Could == -1 */
  63                 return tailv;
  64         }
  65
  66         /*
  67          * Normalize position indicator.
  68          */
  69         if(pd->nboff >= 8) {
  70                 pd->buffer += (pd->nboff >> 3);
  71                 pd->nbits  -= (pd->nboff & ~0x07);
  72                 pd->nboff  &= 0x07;
  73         }
  74         pd->moved += nbits;
  75         pd->nboff += nbits;
  76         off = pd->nboff;
  77         buf = pd->buffer;
  78
  79         /*
  80          * Extract specified number of bits.
  81          */
  82         if(off <= 8)
  83                 accum = nbits ? (buf[0]) >> (8 - off) : 0;
  84         else if(off <= 16)
  85                 accum = ((buf[0] << 8) + buf[1]) >> (16 - off);
  86         else if(off <= 24)
  87                 accum = ((buf[0] << 16) + (buf[1] << 8) + buf[2]) >> (24 - off);
  88         else if(off <= 31)
  89                 accum = ((buf[0] << 24) + (buf[1] << 16)
  90                         + (buf[2] << 8) + (buf[3])) >> (32 - off);
  91         else if(nbits <= 31) {
  92                 asn_per_data_t tpd = *pd;
  93                 /* Here are we with our 31-bits limit plus 1..7 bits offset. */
  94                 per_get_undo(&tpd, nbits);
  95                 /* The number of available bits in the stream allow
  96                  * for the following operations to take place without
  97                  * invoking the ->refill() function */
  98                 accum  = per_get_few_bits(&tpd, nbits - 24) << 24;
  99                 accum |= per_get_few_bits(&tpd, 24);
 100         } else {
 101                 per_get_undo(pd, nbits);
 102                 return -1;
 103         }
 104
 105         accum &= (((uint32_t)1 << nbits) - 1);
 106
 107         ASN_DEBUG("  [PER got %2d<=%2d bits => span %d %+ld[%d..%d]:%02x (%d) => 0x%x]",
 108                 (int)nbits, (int)nleft,
 109                 (int)pd->moved,
 110                 (((long)pd->buffer) & 0xf),
 111                 (int)pd->nboff, (int)pd->nbits,
 112                 ((pd->buffer != NULL)?pd->buffer[0]:0),
 113                 (int)(pd->nbits - pd->nboff),
 114                 (int)accum);
 115
 116         return accum;
 117 }
 118
 119 /*
 120  * Extract a large number of bits from the specified PER data pointer.
 121  */
 122 int
 123 per_get_many_bits(asn_per_data_t *pd, uint8_t *dst, int alright, int nbits) {
 124         int32_t value;
 125
 126         if(alright && (nbits & 7)) {
 127                 /* Perform right alignment of a first few bits */
 128                 value = per_get_few_bits(pd, nbits & 0x07);
 129                 if(value < 0) return -1;
 130                 *dst++ = value; /* value is already right-aligned */
 131                 nbits &= ~7;
 132         }
 133
 134         while(nbits) {
 135                 if(nbits >= 24) {
 136                         value = per_get_few_bits(pd, 24);
 137                         if(value < 0) return -1;
 138                         *(dst++) = value >> 16;
 139                         *(dst++) = value >> 8;
 140                         *(dst++) = value;
 141                         nbits -= 24;
 142                 } else {
 143                         value = per_get_few_bits(pd, nbits);
 144                         if(value < 0) return -1;
 145                         if(nbits & 7) { /* implies left alignment */
 146                                 value <<= 8 - (nbits & 7),
 147                                 nbits += 8 - (nbits & 7);
 148                                 if(nbits > 24)
 149                                         *dst++ = value >> 24;
 150                         }
 151                         if(nbits > 16)
 152                                 *dst++ = value >> 16;
 153                         if(nbits > 8)
 154                                 *dst++ = value >> 8;
 155                         *dst++ = value;
 156                         break;
 157                 }
 158         }
 159
 160         return 0;
 161 }
 162
 163 /*
 164  * Get the length "n" from the stream.
 165  */
 166 ssize_t
 167 uper_get_length(asn_per_data_t *pd, int ebits, int *repeat) {
 168         ssize_t value;
 169
 170         *repeat = 0;
 171
 172         if(ebits >= 0) return per_get_few_bits(pd, ebits);
 173
 174         value = per_get_few_bits(pd, 8);
 175         if(value < 0) return -1;
 176         if((value & 128) == 0)  /* #10.9.3.6 */
 177                 return (value & 0x7F);
 178         if((value & 64) == 0) { /* #10.9.3.7 */
 179                 value = ((value & 63) << 8) | per_get_few_bits(pd, 8);
 180                 if(value < 0) return -1;
 181                 return value;
 182         }
 183         value &= 63;    /* this is "m" from X.691, #10.9.3.8 */
 184         if(value < 1 || value > 4)
 185                 return -1;
 186         *repeat = 1;
 187         return (16384 * value);
 188 }
 189
 190 /*
 191  * Get the normally small length "n".
 192  * This procedure used to decode length of extensions bit-maps
 193  * for SET and SEQUENCE types.
 194  */
 195 ssize_t
 196 uper_get_nslength(asn_per_data_t *pd) {
 197         ssize_t length;
 198
 199         ASN_DEBUG("Getting normally small length");
 200
 201         if(per_get_few_bits(pd, 1) == 0) {
 202                 length = per_get_few_bits(pd, 6) + 1;
 203                 if(length <= 0) return -1;
 204                 ASN_DEBUG("l=%d", (int)length);
 205                 return length;
 206         } else {
 207                 int repeat;
 208                 length = uper_get_length(pd, -1, &repeat);
 209                 if(length >= 0 && !repeat) return length;
 210                 return -1; /* Error, or do not support >16K extensions */
 211         }
 212 }
 213
 214 /*
 215  * Get the normally small non-negative whole number.
 216  * X.691, #10.6
 217  */
 218 ssize_t
 219 uper_get_nsnnwn(asn_per_data_t *pd) {
 220         ssize_t value;
 221
 222         value = per_get_few_bits(pd, 7);
 223         if(value & 64) {        /* implicit (value < 0) */
 224                 value &= 63;
 225                 value <<= 2;
 226                 value |= per_get_few_bits(pd, 2);
 227                 if(value & 128) /* implicit (value < 0) */
 228                         return -1;
 229                 if(value == 0)
 230                         return 0;
 231                 if(value >= 3)
 232                         return -1;
 233                 value = per_get_few_bits(pd, 8 * value);
 234                 return value;
 235         }
 236
 237         return value;
 238 }
 239
 240 /*
 241  * X.691-11/2008, #11.6
 242  * Encoding of a normally small non-negative whole number
 243  */
 244 int
 245 uper_put_nsnnwn(asn_per_outp_t *po, int n) {
 246         int bytes;
 247
 248         if(n <= 63) {
 249                 if(n < 0) return -1;
 250                 return per_put_few_bits(po, n, 7);
 251         }
 252         if(n < 256)
 253                 bytes = 1;
 254         else if(n < 65536)
 255                 bytes = 2;
 256         else if(n < 256 * 65536)
 257                 bytes = 3;
 258         else
 259                 return -1;      /* This is not a "normally small" value */
 260         if(per_put_few_bits(po, bytes, 8))
 261                 return -1;
 262
 263         return per_put_few_bits(po, n, 8 * bytes);
 264 }
 265
 266
 267 /* X.691-2008/11, #11.5.6 -> #11.3 */
 268 int uper_get_constrained_whole_number(asn_per_data_t *pd, unsigned long *out_value, int nbits) {
 269         unsigned long lhalf;    /* Lower half of the number*/
 270         long half;
 271
 272         if(nbits <= 31) {
 273                 half = per_get_few_bits(pd, nbits);
 274                 if(half < 0) return -1;
 275                 *out_value = half;
 276                 return 0;
 277         }
 278
 279         if((size_t)nbits > 8 * sizeof(*out_value))
 280                 return -1;  /* RANGE */
 281
 282         half = per_get_few_bits(pd, 31);
 283         if(half < 0) return -1;
 284
 285         if(uper_get_constrained_whole_number(pd, &lhalf, nbits - 31))
 286                 return -1;
 287
 288         *out_value = ((unsigned long)half << (nbits - 31)) | lhalf;
 289         return 0;
 290 }
 291
 292
 293 /* X.691-2008/11, #11.5.6 -> #11.3 */
 294 int uper_put_constrained_whole_number_s(asn_per_outp_t *po, long v, int nbits) {
 295         /*
 296          * Assume signed number can be safely coerced into
 297          * unsigned of the same range.
 298          * The following testing code will likely be optimized out
 299          * by compiler if it is true.
 300          */
 301         unsigned long uvalue1 = ULONG_MAX;
 302                  long svalue  = uvalue1;
 303         unsigned long uvalue2 = svalue;
 304         assert(uvalue1 == uvalue2);
 305         return uper_put_constrained_whole_number_u(po, v, nbits);
 306 }
 307
 308 int uper_put_constrained_whole_number_u(asn_per_outp_t *po, unsigned long v, int nbits) {
 309         if(nbits <= 31) {
 310                 return per_put_few_bits(po, v, nbits);
 311         } else {
 312                 /* Put higher portion first, followed by lower 31-bit */
 313                 if(uper_put_constrained_whole_number_u(po, v >> 31, nbits - 31))
 314                         return -1;
 315                 return per_put_few_bits(po, v, 31);
 316         }
 317 }
 318
 319 /*
 320  * Put a small number of bits (<= 31).
 321  */
 322 int
 323 per_put_few_bits(asn_per_outp_t *po, uint32_t bits, int obits) {
 324         size_t off;     /* Next after last bit offset */
 325         size_t omsk;    /* Existing last byte meaningful bits mask */
 326         uint8_t *buf;
 327
 328         if(obits <= 0 || obits >= 32) return obits ? -1 : 0;
 329
 330         ASN_DEBUG("[PER put %d bits %x to %p+%d bits]",
 331                         obits, (int)bits, po->buffer, (int)po->nboff);
 332
 333         /*
 334          * Normalize position indicator.
 335          */
 336         if(po->nboff >= 8) {
 337                 po->buffer += (po->nboff >> 3);
 338                 po->nbits  -= (po->nboff & ~0x07);
 339                 po->nboff  &= 0x07;
 340         }
 341
 342         /*
 343          * Flush whole-bytes output, if necessary.
 344          */
 345         if(po->nboff + obits > po->nbits) {
 346                 int complete_bytes = (po->buffer - po->tmpspace);
 347                 ASN_DEBUG("[PER output %ld complete + %ld]",
 348                         (long)complete_bytes, (long)po->flushed_bytes);
 349                 if(po->outper(po->tmpspace, complete_bytes, po->op_key) < 0)
 350                         return -1;
 351                 if(po->nboff)
 352                         po->tmpspace[0] = po->buffer[0];
 353                 po->buffer = po->tmpspace;
 354                 po->nbits = 8 * sizeof(po->tmpspace);
 355                 po->flushed_bytes += complete_bytes;
 356         }
 357
 358         /*
 359          * Now, due to sizeof(tmpspace), we are guaranteed large enough space.
 360          */
 361         buf = po->buffer;
 362         omsk = ~((1 << (8 - po->nboff)) - 1);
 363         off = (po->nboff + obits);
 364
 365         /* Clear data of debris before meaningful bits */
 366         bits &= (((uint32_t)1 << obits) - 1);
 367
 368         ASN_DEBUG("[PER out %d %u/%x (t=%d,o=%d) %x&%x=%x]", obits,
 369                 (int)bits, (int)bits,
 370                 (int)po->nboff, (int)off,
 371                 buf[0], (int)(omsk&0xff),
 372                 (int)(buf[0] & omsk));
 373
 374         if(off <= 8)    /* Completely within 1 byte */
 375                 po->nboff = off,
 376                 bits <<= (8 - off),
 377                 buf[0] = (buf[0] & omsk) | bits;
 378         else if(off <= 16)
 379                 po->nboff = off,
 380                 bits <<= (16 - off),
 381                 buf[0] = (buf[0] & omsk) | (bits >> 8),
 382                 buf[1] = bits;
 383         else if(off <= 24)
 384                 po->nboff = off,
 385                 bits <<= (24 - off),
 386                 buf[0] = (buf[0] & omsk) | (bits >> 16),
 387                 buf[1] = bits >> 8,
 388                 buf[2] = bits;
 389         else if(off <= 31)
 390                 po->nboff = off,
 391                 bits <<= (32 - off),
 392                 buf[0] = (buf[0] & omsk) | (bits >> 24),
 393                 buf[1] = bits >> 16,
 394                 buf[2] = bits >> 8,
 395                 buf[3] = bits;
 396         else {
 397                 per_put_few_bits(po, bits >> (obits - 24), 24);
 398                 per_put_few_bits(po, bits, obits - 24);
 399         }
 400
 401         ASN_DEBUG("[PER out %u/%x => %02x buf+%ld]",
 402                 (int)bits, (int)bits, buf[0],
 403                 (long)(po->buffer - po->tmpspace));
 404
 405         return 0;
 406 }
 407
 408
 409 /*
 410  * Output a large number of bits.
 411  */
 412 int
 413 per_put_many_bits(asn_per_outp_t *po, const uint8_t *src, int nbits) {
 414
 415         while(nbits) {
 416                 uint32_t value;
 417
 418                 if(nbits >= 24) {
 419                         value = (src[0] << 16) | (src[1] << 8) | src[2];
 420                         src += 3;
 421                         nbits -= 24;
 422                         if(per_put_few_bits(po, value, 24))
 423                                 return -1;
 424                 } else {
 425                         value = src[0];
 426                         if(nbits > 8)
 427                                 value = (value << 8) | src[1];
 428                         if(nbits > 16)
 429                                 value = (value << 8) | src[2];
 430                         if(nbits & 0x07)
 431                                 value >>= (8 - (nbits & 0x07));
 432                         if(per_put_few_bits(po, value, nbits))
 433                                 return -1;
 434                         break;
 435                 }
 436         }
 437
 438         return 0;
 439 }
 440
 441 /*
 442  * Put the length "n" (or part of it) into the stream.
 443  */
 444 ssize_t
 445 uper_put_length(asn_per_outp_t *po, size_t length) {
 446
 447         if(length <= 127)       /* #10.9.3.6 */
 448                 return per_put_few_bits(po, length, 8)
 449                         ? -1 : (ssize_t)length;
 450         else if(length < 16384) /* #10.9.3.7 */
 451                 return per_put_few_bits(po, length|0x8000, 16)
 452                         ? -1 : (ssize_t)length;
 453
 454         length >>= 14;
 455         if(length > 4) length = 4;
 456
 457         return per_put_few_bits(po, 0xC0 | length, 8)
 458                         ? -1 : (ssize_t)(length << 14);
 459 }
 460
 461
 462 /*
 463  * Put the normally small length "n" into the stream.
 464  * This procedure used to encode length of extensions bit-maps
 465  * for SET and SEQUENCE types.
 466  */
 467 int
 468 uper_put_nslength(asn_per_outp_t *po, size_t length) {
 469
 470         if(length <= 64) {
 471                 /* #10.9.3.4 */
 472                 if(length == 0) return -1;
 473                 return per_put_few_bits(po, length-1, 7) ? -1 : 0;
 474         } else {
 475                 if(uper_put_length(po, length) != (ssize_t)length) {
 476                         /* This might happen in case of >16K extensions */
 477                         return -1;
 478                 }
 479         }
 480
 481         return 0;
 482 }
 483