]> git.stg.codes - stg.git/blob - stglibs/smux.lib/per_support.c
Merge branch 'master' into full-month-stats
[stg.git] / stglibs / smux.lib / per_support.c
1 /*
2  * Copyright (c) 2005, 2006 Lev Walkin <vlm@lionet.info>. All rights reserved.
3  * Redistribution and modifications are permitted subject to BSD license.
4  */
5 #include <asn_system.h>
6 #include <asn_internal.h>
7 #include <per_support.h>
8
9 /*
10  * Extract a small number of bits (<= 31) from the specified PER data pointer.
11  */
12 int32_t
13 per_get_few_bits(asn_per_data_t *pd, int nbits) {
14         size_t off;     /* Next after last bit offset */
15         uint32_t accum;
16         const uint8_t *buf;
17
18         if(nbits < 0 || pd->nboff + nbits > pd->nbits)
19                 return -1;
20
21         ASN_DEBUG("[PER get %d bits from %p+%d bits]",
22                 nbits, pd->buffer, pd->nboff);
23
24         /*
25          * Normalize position indicator.
26          */
27         if(pd->nboff >= 8) {
28                 pd->buffer += (pd->nboff >> 3);
29                 pd->nbits  -= (pd->nboff & ~0x07);
30                 pd->nboff  &= 0x07;
31         }
32         off = (pd->nboff += nbits);
33         buf = pd->buffer;
34
35         /*
36          * Extract specified number of bits.
37          */
38         if(off <= 8)
39                 accum = nbits ? (buf[0]) >> (8 - off) : 0;
40         else if(off <= 16)
41                 accum = ((buf[0] << 8) + buf[1]) >> (16 - off);
42         else if(off <= 24)
43                 accum = ((buf[0] << 16) + (buf[1] << 8) + buf[2]) >> (24 - off);
44         else if(off <= 31)
45                 accum = ((buf[0] << 24) + (buf[1] << 16)
46                         + (buf[2] << 8) + (buf[3])) >> (32 - off);
47         else if(nbits <= 31) {
48                 asn_per_data_t tpd = *pd;
49                 /* Here are we with our 31-bits limit plus 1..7 bits offset. */
50                 tpd.nboff -= nbits;
51                 accum  = per_get_few_bits(&tpd, nbits - 24) << 24;
52                 accum |= per_get_few_bits(&tpd, 24);
53         } else {
54                 pd->nboff -= nbits;     /* Oops, revert back */
55                 return -1;
56         }
57
58         return (accum & (((uint32_t)1 << nbits) - 1));
59 }
60
61 /*
62  * Extract a large number of bits from the specified PER data pointer.
63  */
64 int
65 per_get_many_bits(asn_per_data_t *pd, uint8_t *dst, int alright, int nbits) {
66         int32_t value;
67
68         if(alright && (nbits & 7)) {
69                 /* Perform right alignment of a first few bits */
70                 value = per_get_few_bits(pd, nbits & 0x07);
71                 if(value < 0) return -1;
72                 *dst++ = value; /* value is already right-aligned */
73                 nbits &= ~7;
74         }
75
76         while(nbits) {
77                 if(nbits >= 24) {
78                         value = per_get_few_bits(pd, 24);
79                         if(value < 0) return -1;
80                         *(dst++) = value >> 16;
81                         *(dst++) = value >> 8;
82                         *(dst++) = value;
83                         nbits -= 24;
84                 } else {
85                         value = per_get_few_bits(pd, nbits);
86                         if(value < 0) return -1;
87                         if(nbits & 7) { /* implies left alignment */
88                                 value <<= 8 - (nbits & 7),
89                                 nbits += 8 - (nbits & 7);
90                                 if(nbits > 24)
91                                         *dst++ = value >> 24;
92                         }
93                         if(nbits > 16)
94                                 *dst++ = value >> 16;
95                         if(nbits > 8)
96                                 *dst++ = value >> 8;
97                         *dst++ = value;
98                         break;
99                 }
100         }
101
102         return 0;
103 }
104
105 /*
106  * Get the length "n" from the stream.
107  */
108 ssize_t
109 uper_get_length(asn_per_data_t *pd, int ebits, int *repeat) {
110         ssize_t value;
111
112         *repeat = 0;
113
114         if(ebits >= 0) return per_get_few_bits(pd, ebits);
115
116         value = per_get_few_bits(pd, 8);
117         if(value < 0) return -1;
118         if((value & 128) == 0)  /* #10.9.3.6 */
119                 return (value & 0x7F);
120         if((value & 64) == 0) { /* #10.9.3.7 */
121                 value = ((value & 63) << 8) | per_get_few_bits(pd, 8);
122                 if(value < 0) return -1;
123                 return value;
124         }
125         value &= 63;    /* this is "m" from X.691, #10.9.3.8 */
126         if(value < 1 || value > 4)
127                 return -1;
128         *repeat = 1;
129         return (16384 * value);
130 }
131
132 /*
133  * Get the normally small non-negative whole number.
134  * X.691, #10.6
135  */
136 ssize_t
137 uper_get_nsnnwn(asn_per_data_t *pd) {
138         ssize_t value;
139
140         value = per_get_few_bits(pd, 7);
141         if(value & 64) {        /* implicit (value < 0) */
142                 value &= 63;
143                 value <<= 2;
144                 value |= per_get_few_bits(pd, 2);
145                 if(value & 128) /* implicit (value < 0) */
146                         return -1;
147                 if(value == 0)
148                         return 0;
149                 if(value >= 3)
150                         return -1;
151                 value = per_get_few_bits(pd, 8 * value);
152                 return value;
153         }
154
155         return value;
156 }
157
158 /*
159  * Put the normally small non-negative whole number.
160  * X.691, #10.6
161  */
162 int
163 uper_put_nsnnwn(asn_per_outp_t *po, int n) {
164         int bytes;
165
166         if(n <= 63) {
167                 if(n < 0) return -1;
168                 return per_put_few_bits(po, n, 7);
169         }
170         if(n < 256)
171                 bytes = 1;
172         else if(n < 65536)
173                 bytes = 2;
174         else if(n < 256 * 65536)
175                 bytes = 3;
176         else
177                 return -1;      /* This is not a "normally small" value */
178         if(per_put_few_bits(po, bytes, 8))
179                 return -1;
180
181         return per_put_few_bits(po, n, 8 * bytes);
182 }
183
184
185 /*
186  * Put a small number of bits (<= 31).
187  */
188 int
189 per_put_few_bits(asn_per_outp_t *po, uint32_t bits, int obits) {
190         size_t off;     /* Next after last bit offset */
191         size_t omsk;    /* Existing last byte meaningful bits mask */
192         uint8_t *buf;
193
194         if(obits <= 0 || obits >= 32) return obits ? -1 : 0;
195
196         ASN_DEBUG("[PER put %d bits to %p+%d bits]",
197                         obits, po->buffer, po->nboff);
198
199         /*
200          * Normalize position indicator.
201          */
202         if(po->nboff >= 8) {
203                 po->buffer += (po->nboff >> 3);
204                 po->nbits  -= (po->nboff & ~0x07);
205                 po->nboff  &= 0x07;
206         }
207
208         /*
209          * Flush whole-bytes output, if necessary.
210          */
211         if(po->nboff + obits > po->nbits) {
212                 int complete_bytes = (po->buffer - po->tmpspace);
213                 if(po->outper(po->buffer, complete_bytes, po->op_key) < 0)
214                         return -1;
215                 if(po->nboff)
216                         po->tmpspace[0] = po->buffer[0];
217                 po->buffer = po->tmpspace;
218                 po->nbits = 8 * sizeof(po->tmpspace);
219                 po->flushed_bytes += complete_bytes;
220         }
221
222         /*
223          * Now, due to sizeof(tmpspace), we are guaranteed large enough space.
224          */
225         buf = po->buffer;
226         omsk = ~((1 << (8 - po->nboff)) - 1);
227         off = (po->nboff += obits);
228
229         /* Clear data of debris before meaningful bits */
230         bits &= (((uint32_t)1 << obits) - 1);
231
232         ASN_DEBUG("[PER out %d %u/%x (t=%d,o=%d) %x&%x=%x]", obits, bits, bits,
233                 po->nboff - obits, off, buf[0], omsk&0xff, buf[0] & omsk);
234
235         if(off <= 8)    /* Completely within 1 byte */
236                 bits <<= (8 - off),
237                 buf[0] = (buf[0] & omsk) | bits;
238         else if(off <= 16)
239                 bits <<= (16 - off),
240                 buf[0] = (buf[0] & omsk) | (bits >> 8),
241                 buf[1] = bits;
242         else if(off <= 24)
243                 bits <<= (24 - off),
244                 buf[0] = (buf[0] & omsk) | (bits >> 16),
245                 buf[1] = bits >> 8,
246                 buf[2] = bits;
247         else if(off <= 31)
248                 bits <<= (32 - off),
249                 buf[0] = (buf[0] & omsk) | (bits >> 24),
250                 buf[1] = bits >> 16,
251                 buf[2] = bits >> 8,
252                 buf[3] = bits;
253         else {
254                 ASN_DEBUG("->[PER out split %d]", obits);
255                 per_put_few_bits(po, bits >> 8, 24);
256                 per_put_few_bits(po, bits, obits - 24);
257                 ASN_DEBUG("<-[PER out split %d]", obits);
258         }
259
260         ASN_DEBUG("[PER out %u/%x => %02x buf+%d]",
261                 bits, bits, buf[0], po->buffer - po->tmpspace);
262
263         return 0;
264 }
265
266
267 /*
268  * Output a large number of bits.
269  */
270 int
271 per_put_many_bits(asn_per_outp_t *po, const uint8_t *src, int nbits) {
272
273         while(nbits) {
274                 uint32_t value;
275
276                 if(nbits >= 24) {
277                         value = (src[0] << 16) | (src[1] << 8) | src[2];
278                         src += 3;
279                         nbits -= 24;
280                         if(per_put_few_bits(po, value, 24))
281                                 return -1;
282                 } else {
283                         value = src[0];
284                         if(nbits > 8)
285                                 value = (value << 8) | src[1];
286                         if(nbits > 16)
287                                 value = (value << 8) | src[2];
288                         if(nbits & 0x07)
289                                 value >>= (8 - (nbits & 0x07));
290                         if(per_put_few_bits(po, value, nbits))
291                                 return -1;
292                         break;
293                 }
294         }
295
296         return 0;
297 }
298
299 /*
300  * Put the length "n" (or part of it) into the stream.
301  */
302 ssize_t
303 uper_put_length(asn_per_outp_t *po, size_t length) {
304
305         if(length <= 127)       /* #10.9.3.6 */
306                 return per_put_few_bits(po, length, 8)
307                         ? -1 : (ssize_t)length;
308         else if(length < 16384) /* #10.9.3.7 */
309                 return per_put_few_bits(po, length|0x8000, 16)
310                         ? -1 : (ssize_t)length;
311
312         length >>= 14;
313         if(length > 4) length = 4;
314
315         return per_put_few_bits(po, 0xC0 | length, 8)
316                         ? -1 : (ssize_t)(length << 14);
317 }
318