1 /* $OpenBSD: dc.c,v 1.21 2024/04/23 13:34:50 jsg Exp $ */
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 <sys/stat.h>
20 #include <err.h>
21 #include <errno.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25
26 #include "extern.h"
27
28 static __dead void usage(void);
29
30 extern char *__progname;
31
32 static __dead void
usage(void)33 usage(void)
34 {
35 (void)fprintf(stderr, "usage: %s [-x] [-e expression] [file]\n",
36 __progname);
37 exit(1);
38 }
39
40 int
dc_main(int argc,char * argv[])41 dc_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 struct stat st;
49
50 buf = bstrdup("");
51 /* accept and ignore a single dash to be 4.4BSD dc(1) compatible */
52 optind = 1;
53 optreset = 1;
54 while ((ch = getopt(argc, argv, "e:x-")) != -1) {
55 switch (ch) {
56 case 'e':
57 p = buf;
58 if (asprintf(&buf, "%s %s", buf, optarg) == -1)
59 err(1, NULL);
60 free(p);
61 break;
62 case 'x':
63 extended_regs = true;
64 break;
65 case '-':
66 break;
67 default:
68 usage();
69 }
70 }
71 argc -= optind;
72 argv += optind;
73
74 if (argc == 0) {
75 if (pledge("stdio", NULL) == -1)
76 err(1, "pledge");
77 }
78
79 init_bmachine(extended_regs);
80 (void)setvbuf(stdout, NULL, _IOLBF, 0);
81 (void)setvbuf(stderr, NULL, _IOLBF, 0);
82
83 if (argc > 1)
84 usage();
85 if (buf[0] != '\0') {
86 src_setstring(&src, buf);
87 reset_bmachine(&src);
88 eval();
89 free(buf);
90 if (argc == 0)
91 return (0);
92 }
93 if (argc == 1) {
94 file = fopen(argv[0], "r");
95 if (file == NULL)
96 err(1, "cannot open file %s", argv[0]);
97
98 if (pledge("stdio", NULL) == -1)
99 err(1, "pledge");
100
101 if (fstat(fileno(file), &st) == -1)
102 err(1, "%s", argv[0]);
103 if (S_ISDIR(st.st_mode))
104 errc(1, EISDIR, "%s", argv[0]);
105 src_setstream(&src, file);
106 reset_bmachine(&src);
107 eval();
108 (void)fclose(file);
109 /*
110 * BSD and Solaris dc(1) continue with stdin after processing
111 * the file given as the argument. We follow GNU dc(1).
112 */
113 return (0);
114 }
115
116 src_setstream(&src, stdin);
117 reset_bmachine(&src);
118 eval();
119
120 return (0);
121 }
122