1 /* 2 * $OpenBSD: dc.c,v 1.6 2004/10/18 07:49:00 otto Exp $ 3 * $DragonFly: src/usr.bin/dc/dc.c,v 1.2 2005/04/21 18:50:50 swildner Exp $ 4 */ 5 6 /* 7 * Copyright (c) 2003, Otto Moerbeek <otto@drijf.net> 8 * 9 * Permission to use, copy, modify, and distribute this software for any 10 * purpose with or without fee is hereby granted, provided that the above 11 * copyright notice and this permission notice appear in all copies. 12 * 13 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 14 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 15 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 16 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 17 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 18 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 19 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 20 */ 21 22 #include <err.h> 23 #include <stdlib.h> 24 #include <string.h> 25 #include <unistd.h> 26 27 #include "extern.h" 28 29 static __dead2 void usage(void); 30 31 extern char *__progname; 32 33 static __dead2 void 34 usage(void) 35 { 36 fprintf(stderr, "usage: %s [-x] [-e expr] [file]\n", __progname); 37 exit(1); 38 } 39 40 int 41 main(int argc, char *argv[]) 42 { 43 int ch; 44 bool extended_regs = false; 45 FILE *file; 46 struct source src; 47 char *buf, *p; 48 49 if ((buf = strdup("")) == NULL) 50 err(1, NULL); 51 /* accept and ignore a single dash to be 4.4BSD dc(1) compatible */ 52 while ((ch = getopt(argc, argv, "e:x-")) != -1) { 53 switch (ch) { 54 case 'e': 55 p = buf; 56 if (asprintf(&buf, "%s %s", buf, optarg) == -1) 57 err(1, NULL); 58 free(p); 59 break; 60 case 'x': 61 extended_regs = true; 62 break; 63 case '-': 64 break; 65 default: 66 usage(); 67 } 68 } 69 argc -= optind; 70 argv += optind; 71 72 init_bmachine(extended_regs); 73 setlinebuf(stdout); 74 setlinebuf(stderr); 75 76 if (argc > 1) 77 usage(); 78 if (buf[0] != '\0') { 79 src_setstring(&src, buf); 80 reset_bmachine(&src); 81 eval(); 82 free(buf); 83 if (argc == 0) 84 return (0); 85 } 86 if (argc == 1) { 87 file = fopen(argv[0], "r"); 88 if (file == NULL) 89 err(1, "cannot open file %s", argv[0]); 90 src_setstream(&src, file); 91 reset_bmachine(&src); 92 eval(); 93 fclose(file); 94 /* 95 * BSD and Solaris dc(1) continue with stdin after processing 96 * the file given as the argument. We follow GNU dc(1). 97 */ 98 return (0); 99 } 100 src_setstream(&src, stdin); 101 reset_bmachine(&src); 102 eval(); 103 104 return 0; 105 } 106