Annotation of src/usr.bin/dc/inout.c, Revision 1.18
1.18 ! deraadt 1: /* $OpenBSD: inout.c,v 1.18 2014/12/01 13:11:21 deraadt Exp $ */
1.1 otto 2:
3: /*
4: * Copyright (c) 2003, Otto Moerbeek <otto@drijf.net>
5: *
6: * Permission to use, copy, modify, and distribute this software for any
7: * purpose with or without fee is hereby granted, provided that the above
8: * copyright notice and this permission notice appear in all copies.
9: *
10: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17: */
18:
19: #include <ssl/ssl.h>
20: #include <ctype.h>
21: #include <err.h>
22: #include <string.h>
23:
24: #include "extern.h"
25:
1.11 otto 26: #define MAX_CHARS_PER_LINE 68
1.1 otto 27:
1.12 otto 28: static int lastchar;
29: static int charcount;
1.1 otto 30:
31: static int src_getcharstream(struct source *);
1.14 otto 32: static void src_ungetcharstream(struct source *);
1.1 otto 33: static char *src_getlinestream(struct source *);
34: static void src_freestream(struct source *);
35: static int src_getcharstring(struct source *);
1.14 otto 36: static void src_ungetcharstring(struct source *);
1.1 otto 37: static char *src_getlinestring(struct source *);
38: static void src_freestring(struct source *);
1.12 otto 39: static void flushwrap(FILE *);
1.1 otto 40: static void putcharwrap(FILE *, int);
41: static void printwrap(FILE *, const char *);
42: static char *get_digit(u_long, int, u_int);
43:
44: static struct vtable stream_vtable = {
45: src_getcharstream,
46: src_ungetcharstream,
47: src_getlinestream,
48: src_freestream
49: };
50:
51: static struct vtable string_vtable = {
52: src_getcharstring,
53: src_ungetcharstring,
54: src_getlinestring,
55: src_freestring
56: };
57:
58: void
59: src_setstream(struct source *src, FILE *stream)
60: {
61: src->u.stream = stream;
62: src->vtable = &stream_vtable;
63: }
64:
65: void
66: src_setstring(struct source *src, char *p)
67: {
68: src->u.string.buf = (u_char *)p;
69: src->u.string.pos = 0;
70: src->vtable = &string_vtable;
71: }
72:
73: static int
74: src_getcharstream(struct source *src)
75: {
76: return src->lastchar = getc(src->u.stream);
77: }
78:
1.14 otto 79: static void
1.1 otto 80: src_ungetcharstream(struct source *src)
81: {
1.14 otto 82: (void)ungetc(src->lastchar, src->u.stream);
1.1 otto 83: }
84:
1.14 otto 85: /* ARGSUSED */
1.1 otto 86: static void
87: src_freestream(struct source *src)
88: {
89: }
90:
91: static char *
92: src_getlinestream(struct source *src)
93: {
94: char buf[BUFSIZ];
95:
96: if (fgets(buf, BUFSIZ, src->u.stream) == NULL)
97: return bstrdup("");
98: return bstrdup(buf);
99: }
100:
101: static int
102: src_getcharstring(struct source *src)
103: {
104: src->lastchar = src->u.string.buf[src->u.string.pos];
105: if (src->lastchar == '\0')
106: return EOF;
107: else {
108: src->u.string.pos++;
109: return src->lastchar;
110: }
111: }
112:
1.14 otto 113: static void
1.1 otto 114: src_ungetcharstring(struct source *src)
115: {
1.7 otto 116: if (src->u.string.pos > 0) {
117: if (src->lastchar != '\0')
118: --src->u.string.pos;
1.14 otto 119: }
1.1 otto 120: }
121:
122: static char *
123: src_getlinestring(struct source *src)
124: {
125: char buf[BUFSIZ];
126: int ch, i;
127:
128: i = 0;
129: while (i < BUFSIZ-1) {
130: ch = src_getcharstring(src);
131: if (ch == EOF)
132: break;
133: buf[i++] = ch;
134: if (ch == '\n')
135: break;
136: }
137: buf[i] = '\0';
138: return bstrdup(buf);
139: }
140:
141: static void
142: src_freestring(struct source *src)
143: {
144: free(src->u.string.buf);
145: }
146:
147: static void
1.12 otto 148: flushwrap(FILE *f)
149: {
150: if (lastchar != -1)
1.14 otto 151: (void)putc(lastchar, f);
1.12 otto 152: }
153:
154: static void
1.1 otto 155: putcharwrap(FILE *f, int ch)
156: {
1.12 otto 157: if (charcount >= MAX_CHARS_PER_LINE) {
158: charcount = 0;
1.14 otto 159: (void)fputs("\\\n", f);
1.1 otto 160: }
1.12 otto 161: if (lastchar != -1) {
162: charcount++;
1.14 otto 163: (void)putc(lastchar, f);
1.12 otto 164: }
165: lastchar = ch;
1.1 otto 166: }
167:
168: static void
169: printwrap(FILE *f, const char *p)
170: {
171: char buf[12];
172: char *q = buf;
173:
1.14 otto 174: (void)strlcpy(buf, p, sizeof(buf));
1.1 otto 175: while (*q)
176: putcharwrap(f, *q++);
177: }
178:
179: struct number *
180: readnumber(struct source *src, u_int base)
181: {
182: struct number *n;
183: int ch;
184: bool sign = false;
185: bool dot = false;
186: BN_ULONG v;
1.13 otto 187: u_int i;
1.1 otto 188:
189: n = new_number();
190: bn_check(BN_zero(n->number));
191:
192: while ((ch = (*src->vtable->readchar)(src)) != EOF) {
193:
194: if ('0' <= ch && ch <= '9')
195: v = ch - '0';
196: else if ('A' <= ch && ch <= 'F')
197: v = ch - 'A' + 10;
198: else if (ch == '_') {
199: sign = true;
200: continue;
201: } else if (ch == '.') {
202: if (dot)
203: break;
204: dot = true;
205: continue;
206: } else {
207: (*src->vtable->unreadchar)(src);
208: break;
209: }
210: if (dot)
211: n->scale++;
212:
213: bn_check(BN_mul_word(n->number, base));
214:
1.8 otto 215: #if 0
1.1 otto 216: /* work around a bug in BN_add_word: 0 += 0 is buggy.... */
217: if (v > 0)
1.8 otto 218: #endif
1.1 otto 219: bn_check(BN_add_word(n->number, v));
1.13 otto 220: }
221: if (base != 10) {
222: scale_number(n->number, n->scale);
223: for (i = 0; i < n->scale; i++)
1.14 otto 224: (void)BN_div_word(n->number, base);
1.1 otto 225: }
226: if (sign)
227: negate(n);
228: return n;
229: }
230:
231: char *
232: read_string(struct source *src)
233: {
1.4 otto 234: int count, i, sz, new_sz, ch;
1.1 otto 235: char *p;
1.5 otto 236: bool escape;
1.1 otto 237:
1.5 otto 238: escape = false;
1.1 otto 239: count = 1;
240: i = 0;
241: sz = 15;
242: p = bmalloc(sz + 1);
243:
244: while ((ch = (*src->vtable->readchar)(src)) != EOF) {
1.5 otto 245: if (!escape) {
246: if (ch == '[')
247: count++;
248: else if (ch == ']')
249: count--;
250: if (count == 0)
251: break;
252: }
253: if (ch == '\\' && !escape)
254: escape = true;
255: else {
256: escape = false;
257: if (i == sz) {
258: new_sz = sz * 2;
1.18 ! deraadt 259: p = breallocarray(p, 1, new_sz + 1);
1.5 otto 260: sz = new_sz;
261: }
262: p[i++] = ch;
1.1 otto 263: }
264: }
265: p[i] = '\0';
266: return p;
267: }
268:
269: static char *
270: get_digit(u_long num, int digits, u_int base)
271: {
272: char *p;
1.3 deraadt 273:
1.1 otto 274: if (base <= 16) {
275: p = bmalloc(2);
1.2 deraadt 276: p[0] = num >= 10 ? num + 'A' - 10 : num + '0';
1.1 otto 277: p[1] = '\0';
278: } else {
279: if (asprintf(&p, "%0*lu", digits, num) == -1)
1.6 otto 280: err(1, NULL);
1.1 otto 281: }
282: return p;
283: }
284:
285: void
286: printnumber(FILE *f, const struct number *b, u_int base)
287: {
288: struct number *int_part, *fract_part;
289: int digits;
290: char buf[11];
291: size_t sz;
292: int i;
293: struct stack stack;
294: char *p;
295:
1.12 otto 296: charcount = 0;
297: lastchar = -1;
1.1 otto 298: if (BN_is_zero(b->number))
299: putcharwrap(f, '0');
300:
301: int_part = new_number();
302: fract_part = new_number();
303: fract_part->scale = b->scale;
304:
305: if (base <= 16)
306: digits = 1;
307: else {
308: digits = snprintf(buf, sizeof(buf), "%u", base-1);
309: }
310: split_number(b, int_part->number, fract_part->number);
311:
312: i = 0;
313: stack_init(&stack);
314: while (!BN_is_zero(int_part->number)) {
315: BN_ULONG rem = BN_div_word(int_part->number, base);
316: stack_pushstring(&stack, get_digit(rem, digits, base));
317: i++;
318: }
319: sz = i;
1.16 otto 320: if (BN_is_negative(b->number))
1.1 otto 321: putcharwrap(f, '-');
322: for (i = 0; i < sz; i++) {
323: p = stack_popstring(&stack);
324: if (base > 16)
325: putcharwrap(f, ' ');
326: printwrap(f, p);
327: free(p);
328: }
329: stack_clear(&stack);
330: if (b->scale > 0) {
331: struct number *num_base;
332: BIGNUM mult, stop;
333:
334: putcharwrap(f, '.');
335: num_base = new_number();
1.14 otto 336: bn_check(BN_set_word(num_base->number, base));
1.1 otto 337: BN_init(&mult);
1.14 otto 338: bn_check(BN_one(&mult));
1.1 otto 339: BN_init(&stop);
1.14 otto 340: bn_check(BN_one(&stop));
1.1 otto 341: scale_number(&stop, b->scale);
342:
343: i = 0;
344: while (BN_cmp(&mult, &stop) < 0) {
345: u_long rem;
346:
347: if (i && base > 16)
348: putcharwrap(f, ' ');
349: i = 1;
350:
1.17 otto 351: bmul_number(fract_part, fract_part, num_base,
352: bmachine_scale());
1.1 otto 353: split_number(fract_part, int_part->number, NULL);
354: rem = BN_get_word(int_part->number);
355: p = get_digit(rem, digits, base);
356: int_part->scale = 0;
357: normalize(int_part, fract_part->scale);
1.14 otto 358: bn_check(BN_sub(fract_part->number, fract_part->number,
359: int_part->number));
1.1 otto 360: printwrap(f, p);
361: free(p);
1.14 otto 362: bn_check(BN_mul_word(&mult, base));
1.1 otto 363: }
364: free_number(num_base);
365: BN_free(&mult);
366: BN_free(&stop);
367: }
1.12 otto 368: flushwrap(f);
1.1 otto 369: free_number(int_part);
370: free_number(fract_part);
371: }
372:
373: void
374: print_value(FILE *f, const struct value *value, const char *prefix, u_int base)
375: {
1.14 otto 376: (void)fputs(prefix, f);
1.1 otto 377: switch (value->type) {
378: case BCODE_NONE:
379: if (value->array != NULL)
1.14 otto 380: (void)fputs("<array>", f);
1.1 otto 381: break;
382: case BCODE_NUMBER:
383: printnumber(f, value->u.num, base);
384: break;
385: case BCODE_STRING:
1.14 otto 386: (void)fputs(value->u.string, f);
1.1 otto 387: break;
388: }
389: }
390:
391: void
392: print_ascii(FILE *f, const struct number *n)
393: {
394: BIGNUM *v;
395: int numbits, i, ch;
396:
397: v = BN_dup(n->number);
398: bn_checkp(v);
399:
1.16 otto 400: if (BN_is_negative(v))
401: BN_set_negative(v, 0);
1.1 otto 402:
403: numbits = BN_num_bytes(v) * 8;
404: while (numbits > 0) {
405: ch = 0;
406: for (i = 0; i < 8; i++)
407: ch |= BN_is_bit_set(v, numbits-i-1) << (7 - i);
1.14 otto 408: (void)putc(ch, f);
1.1 otto 409: numbits -= 8;
410: }
411: BN_free(v);
412: }