asn1c_defs/all-defs/asn_bit_data.c

   1 /*\r
   2  * Copyright (c) 2005-2017 Lev Walkin <vlm@lionet.info>.\r
   3  * All rights reserved.\r
   4  * Redistribution and modifications are permitted subject to BSD license.\r
   5  */\r
   6 #include <asn_system.h>\r
   7 #include <asn_internal.h>\r
   8 #include <asn_bit_data.h>\r
   9 \r
  10 /*\r
  11  * Create a contiguous non-refillable bit data structure.\r
  12  * Can be freed by FREEMEM().\r
  13  */\r
  14 asn_bit_data_t *\r
  15 asn_bit_data_new_contiguous(const void *data, size_t size_bits) {\r
  16     size_t size_bytes = (size_bits + 7) / 8;\r
  17     asn_bit_data_t *pd;\r
  18     uint8_t *bytes;\r
  19 \r
  20     /* Get the extensions map */\r
  21     pd = CALLOC(1, sizeof(*pd) + size_bytes + 1);\r
  22     if(!pd) {\r
  23         return NULL;\r
  24     }\r
  25     bytes = (void *)(((char *)pd) + sizeof(*pd));\r
  26     memcpy(bytes, data, size_bytes);\r
  27     bytes[size_bytes] = 0;\r
  28     pd->buffer = bytes;\r
  29     pd->nboff = 0;\r
  30     pd->nbits = size_bits;\r
  31 \r
  32     return pd;\r
  33 }\r
  34 \r
  35 \r
  36 char *\r
  37 asn_bit_data_string(asn_bit_data_t *pd) {\r
  38         static char buf[2][32];\r
  39         static int n;\r
  40         n = (n+1) % 2;\r
  41     snprintf(buf[n], sizeof(buf[n]),\r
  42              "{m=%" ASN_PRI_SIZE " span %" ASN_PRI_SIZE "[%" ASN_PRI_SIZE\r
  43              "..%" ASN_PRI_SIZE "] (%" ASN_PRI_SIZE ")}",\r
  44              pd->moved, ((uintptr_t)(pd->buffer) & 0xf), pd->nboff, pd->nbits,\r
  45              pd->nbits - pd->nboff);\r
  46     return buf[n];\r
  47 }\r
  48 \r
  49 void\r
  50 asn_get_undo(asn_bit_data_t *pd, int nbits) {\r
  51         if((ssize_t)pd->nboff < nbits) {\r
  52                 assert((ssize_t)pd->nboff < nbits);\r
  53         } else {\r
  54                 pd->nboff -= nbits;\r
  55                 pd->moved -= nbits;\r
  56         }\r
  57 }\r
  58 \r
  59 /*\r
  60  * Extract a small number of bits (<= 31) from the specified PER data pointer.\r
  61  */\r
  62 int32_t\r
  63 asn_get_few_bits(asn_bit_data_t *pd, int nbits) {\r
  64         size_t off;     /* Next after last bit offset */\r
  65         ssize_t nleft;  /* Number of bits left in this stream */\r
  66         uint32_t accum;\r
  67         const uint8_t *buf;\r
  68 \r
  69         if(nbits < 0)\r
  70                 return -1;\r
  71 \r
  72         nleft = pd->nbits - pd->nboff;\r
  73         if(nbits > nleft) {\r
  74                 int32_t tailv, vhead;\r
  75                 if(!pd->refill || nbits > 31) return -1;\r
  76                 /* Accumulate unused bytes before refill */\r
  77                 ASN_DEBUG("Obtain the rest %d bits (want %d)",\r
  78                         (int)nleft, (int)nbits);\r
  79                 tailv = asn_get_few_bits(pd, nleft);\r
  80                 if(tailv < 0) return -1;\r
  81                 /* Refill (replace pd contents with new data) */\r
  82                 if(pd->refill(pd))\r
  83                         return -1;\r
  84                 nbits -= nleft;\r
  85                 vhead = asn_get_few_bits(pd, nbits);\r
  86                 /* Combine the rest of previous pd with the head of new one */\r
  87                 tailv = (tailv << nbits) | vhead;  /* Could == -1 */\r
  88                 return tailv;\r
  89         }\r
  90 \r
  91         /*\r
  92          * Normalize position indicator.\r
  93          */\r
  94         if(pd->nboff >= 8) {\r
  95                 pd->buffer += (pd->nboff >> 3);\r
  96                 pd->nbits  -= (pd->nboff & ~0x07);\r
  97                 pd->nboff  &= 0x07;\r
  98         }\r
  99         pd->moved += nbits;\r
 100         pd->nboff += nbits;\r
 101         off = pd->nboff;\r
 102         buf = pd->buffer;\r
 103 \r
 104         /*\r
 105          * Extract specified number of bits.\r
 106          */\r
 107         if(off <= 8)\r
 108                 accum = nbits ? (buf[0]) >> (8 - off) : 0;\r
 109         else if(off <= 16)\r
 110                 accum = ((buf[0] << 8) + buf[1]) >> (16 - off);\r
 111         else if(off <= 24)\r
 112                 accum = ((buf[0] << 16) + (buf[1] << 8) + buf[2]) >> (24 - off);\r
 113         else if(off <= 31)\r
 114                 accum = (((uint32_t)buf[0] << 24) + (buf[1] << 16)\r
 115                         + (buf[2] << 8) + (buf[3])) >> (32 - off);\r
 116         else if(nbits <= 31) {\r
 117                 asn_bit_data_t tpd = *pd;\r
 118                 /* Here are we with our 31-bits limit plus 1..7 bits offset. */\r
 119                 asn_get_undo(&tpd, nbits);\r
 120                 /* The number of available bits in the stream allow\r
 121                  * for the following operations to take place without\r
 122                  * invoking the ->refill() function */\r
 123                 accum  = asn_get_few_bits(&tpd, nbits - 24) << 24;\r
 124                 accum |= asn_get_few_bits(&tpd, 24);\r
 125         } else {\r
 126                 asn_get_undo(pd, nbits);\r
 127                 return -1;\r
 128         }\r
 129 \r
 130         accum &= (((uint32_t)1 << nbits) - 1);\r
 131 \r
 132         ASN_DEBUG("  [PER got %2d<=%2d bits => span %d %+ld[%d..%d]:%02x (%d) => 0x%x]",\r
 133                 (int)nbits, (int)nleft,\r
 134                 (int)pd->moved,\r
 135                 (((long)pd->buffer) & 0xf),\r
 136                 (int)pd->nboff, (int)pd->nbits,\r
 137                 ((pd->buffer != NULL)?pd->buffer[0]:0),\r
 138                 (int)(pd->nbits - pd->nboff),\r
 139                 (int)accum);\r
 140 \r
 141         return accum;\r
 142 }\r
 143 \r
 144 /*\r
 145  * Extract a large number of bits from the specified PER data pointer.\r
 146  */\r
 147 int\r
 148 asn_get_many_bits(asn_bit_data_t *pd, uint8_t *dst, int alright, int nbits) {\r
 149         int32_t value;\r
 150 \r
 151         if(alright && (nbits & 7)) {\r
 152                 /* Perform right alignment of a first few bits */\r
 153                 value = asn_get_few_bits(pd, nbits & 0x07);\r
 154                 if(value < 0) return -1;\r
 155                 *dst++ = value; /* value is already right-aligned */\r
 156                 nbits &= ~7;\r
 157         }\r
 158 \r
 159         while(nbits) {\r
 160                 if(nbits >= 24) {\r
 161                         value = asn_get_few_bits(pd, 24);\r
 162                         if(value < 0) return -1;\r
 163                         *(dst++) = value >> 16;\r
 164                         *(dst++) = value >> 8;\r
 165                         *(dst++) = value;\r
 166                         nbits -= 24;\r
 167                 } else {\r
 168                         value = asn_get_few_bits(pd, nbits);\r
 169                         if(value < 0) return -1;\r
 170                         if(nbits & 7) { /* implies left alignment */\r
 171                                 value <<= 8 - (nbits & 7),\r
 172                                 nbits += 8 - (nbits & 7);\r
 173                                 if(nbits > 24)\r
 174                                         *dst++ = value >> 24;\r
 175                         }\r
 176                         if(nbits > 16)\r
 177                                 *dst++ = value >> 16;\r
 178                         if(nbits > 8)\r
 179                                 *dst++ = value >> 8;\r
 180                         *dst++ = value;\r
 181                         break;\r
 182                 }\r
 183         }\r
 184 \r
 185         return 0;\r
 186 }\r
 187 \r
 188 /*\r
 189  * Put a small number of bits (<= 31).\r
 190  */\r
 191 int\r
 192 asn_put_few_bits(asn_bit_outp_t *po, uint32_t bits, int obits) {\r
 193         size_t off;     /* Next after last bit offset */\r
 194         size_t omsk;    /* Existing last byte meaningful bits mask */\r
 195         uint8_t *buf;\r
 196 \r
 197         if(obits <= 0 || obits >= 32) return obits ? -1 : 0;\r
 198 \r
 199         ASN_DEBUG("[PER put %d bits %x to %p+%d bits]",\r
 200                         obits, (int)bits, (void *)po->buffer, (int)po->nboff);\r
 201 \r
 202         /*\r
 203          * Normalize position indicator.\r
 204          */\r
 205         if(po->nboff >= 8) {\r
 206                 po->buffer += (po->nboff >> 3);\r
 207                 po->nbits  -= (po->nboff & ~0x07);\r
 208                 po->nboff  &= 0x07;\r
 209         }\r
 210 \r
 211         /*\r
 212          * Flush whole-bytes output, if necessary.\r
 213          */\r
 214         if(po->nboff + obits > po->nbits) {\r
 215                 size_t complete_bytes;\r
 216                 if(!po->buffer) po->buffer = po->tmpspace;\r
 217                 complete_bytes = (po->buffer - po->tmpspace);\r
 218                 ASN_DEBUG("[PER output %ld complete + %ld]",\r
 219                         (long)complete_bytes, (long)po->flushed_bytes);\r
 220                 if(po->output(po->tmpspace, complete_bytes, po->op_key) < 0)\r
 221                         return -1;\r
 222                 if(po->nboff)\r
 223                         po->tmpspace[0] = po->buffer[0];\r
 224                 po->buffer = po->tmpspace;\r
 225                 po->nbits = 8 * sizeof(po->tmpspace);\r
 226                 po->flushed_bytes += complete_bytes;\r
 227         }\r
 228 \r
 229         /*\r
 230          * Now, due to sizeof(tmpspace), we are guaranteed large enough space.\r
 231          */\r
 232         buf = po->buffer;\r
 233         omsk = ~((1 << (8 - po->nboff)) - 1);\r
 234         off = (po->nboff + obits);\r
 235 \r
 236         /* Clear data of debris before meaningful bits */\r
 237         bits &= (((uint32_t)1 << obits) - 1);\r
 238 \r
 239         ASN_DEBUG("[PER out %d %u/%x (t=%d,o=%d) %x&%x=%x]", obits,\r
 240                 (int)bits, (int)bits,\r
 241                 (int)po->nboff, (int)off,\r
 242                 buf[0], (int)(omsk&0xff),\r
 243                 (int)(buf[0] & omsk));\r
 244 \r
 245         if(off <= 8)    /* Completely within 1 byte */\r
 246                 po->nboff = off,\r
 247                 bits <<= (8 - off),\r
 248                 buf[0] = (buf[0] & omsk) | bits;\r
 249         else if(off <= 16)\r
 250                 po->nboff = off,\r
 251                 bits <<= (16 - off),\r
 252                 buf[0] = (buf[0] & omsk) | (bits >> 8),\r
 253                 buf[1] = bits;\r
 254         else if(off <= 24)\r
 255                 po->nboff = off,\r
 256                 bits <<= (24 - off),\r
 257                 buf[0] = (buf[0] & omsk) | (bits >> 16),\r
 258                 buf[1] = bits >> 8,\r
 259                 buf[2] = bits;\r
 260         else if(off <= 31)\r
 261                 po->nboff = off,\r
 262                 bits <<= (32 - off),\r
 263                 buf[0] = (buf[0] & omsk) | (bits >> 24),\r
 264                 buf[1] = bits >> 16,\r
 265                 buf[2] = bits >> 8,\r
 266                 buf[3] = bits;\r
 267         else {\r
 268                 if(asn_put_few_bits(po, bits >> (obits - 24), 24)) return -1;\r
 269                 if(asn_put_few_bits(po, bits, obits - 24)) return -1;\r
 270         }\r
 271 \r
 272         ASN_DEBUG("[PER out %u/%x => %02x buf+%ld]",\r
 273                 (int)bits, (int)bits, buf[0],\r
 274                 (long)(po->buffer - po->tmpspace));\r
 275 \r
 276         return 0;\r
 277 }\r
 278 \r
 279 \r
 280 /*\r
 281  * Output a large number of bits.\r
 282  */\r
 283 int\r
 284 asn_put_many_bits(asn_bit_outp_t *po, const uint8_t *src, int nbits) {\r
 285 \r
 286         while(nbits) {\r
 287                 uint32_t value;\r
 288 \r
 289                 if(nbits >= 24) {\r
 290                         value = (src[0] << 16) | (src[1] << 8) | src[2];\r
 291                         src += 3;\r
 292                         nbits -= 24;\r
 293                         if(asn_put_few_bits(po, value, 24))\r
 294                                 return -1;\r
 295                 } else {\r
 296                         value = src[0];\r
 297                         if(nbits > 8)\r
 298                                 value = (value << 8) | src[1];\r
 299                         if(nbits > 16)\r
 300                                 value = (value << 8) | src[2];\r
 301                         if(nbits & 0x07)\r
 302                                 value >>= (8 - (nbits & 0x07));\r
 303                         if(asn_put_few_bits(po, value, nbits))\r
 304                                 return -1;\r
 305                         break;\r
 306                 }\r
 307         }\r
 308 \r
 309         return 0;\r
 310 }\r
 311 \r
 312 \r
 313 int\r
 314 asn_put_aligned_flush(asn_bit_outp_t *po) {\r
 315     uint32_t unused_bits = (0x7 & (8 - (po->nboff & 0x07)));\r
 316     size_t complete_bytes =\r
 317         (po->buffer ? po->buffer - po->tmpspace : 0) + ((po->nboff + 7) >> 3);\r
 318 \r
 319     if(unused_bits) {\r
 320         po->buffer[po->nboff >> 3] &= ~0u << unused_bits;\r
 321     }\r
 322 \r
 323     if(po->output(po->tmpspace, complete_bytes, po->op_key) < 0) {\r
 324         return -1;\r
 325     } else {\r
 326         po->buffer = po->tmpspace;\r
 327         po->nboff = 0;\r
 328         po->nbits = 8 * sizeof(po->tmpspace);\r
 329         po->flushed_bytes += complete_bytes;\r
 330         return 0;\r
 331     }\r
 332 }\r
 333 \r