full adaptation to c99
[brainfuck.git] / bf.c
CommitLineData
048e6e55
LM
1/* depend: */
2/* cflags: */
3/* linker: debug.o */
bd9ffcf2
LM
4/* winlnk: debug.o */
5/* doslnk: debug.o */
048e6e55 6
a213b0ae 7#include <errno.h>
048e6e55
LM
8#include <getopt.h>
9#include <malloc.h>
10#include <stdio.h>
11#include <stdlib.h>
12#include <string.h>
13
14#include "debug.h"
15
16/* macros */
17
18#define BUFSIZE 256
3d8520e3
LM
19#define MAXJUMP 4
20#define MEMSIZE 10
048e6e55 21
72c52012
LM
22#define CHECKALLOC(ptr) \
23 do { \
24 if ((ptr) == NULL) { \
25 VERBOSE (ERROR, fprintf (stderr, "can't get enough memory for '%s'\n", #ptr)); \
26 exit (1); \
27 } \
28 } while (0)
29
048e6e55
LM
30/* type definition */
31
32/* gobal variables */
33
34char *progname = NULL;
35int p = 0;
6ab28783 36signed char mem[MEMSIZE] = {0};
3d8520e3
LM
37int j = 0;
38int jump[MAXJUMP] = {0};
048e6e55
LM
39
40/* help function */
41
42int usage (int ret)
43{
44 FILE *fd = ret ? stderr : stdout;
e07f5d2a 45 fprintf (fd, "usage: %s [-i file] [-h] [-m memory] [-o file] [-v]\n", progname);
048e6e55
LM
46 fprintf (fd, " -i : input file\n");
47 fprintf (fd, " -h : help message\n");
e07f5d2a 48 fprintf (fd, " -m : memory [0..0]\n");
048e6e55
LM
49 fprintf (fd, " -o : output file\n");
50 fprintf (fd, " -v : verbose level (%d)\n", verbose);
51
52 return ret;
53}
54
07cc2c73
LM
55/* main process */
56
57int process (char *buffer, int nb, FILE *out) {
07cc2c73 58
04c960d9 59 for (int i = 0; i < nb; i++) {
3d8520e3
LM
60 if (buffer[i] == 0) {
61 break;
62 }
63
04c960d9 64 VERBOSE (DEBUG, fprintf (stderr, "%s: p=%d, buffer='%s', memory=[ ", progname, p, buffer + i); for (int _i = 0; _i < MEMSIZE; _i++) fprintf (stderr," %d", mem[_i]); fprintf (stderr," ]\n"));
07cc2c73 65
07cc2c73
LM
66 switch (buffer[i]) {
67 case '>': /* increase pointer */
68 p++;
69 break;
70 case '<': /* decrease pointer */
71 p--;
72 break;
73 case '+': /* increase pointer value */
74 if ((p >= 0) && (p < MEMSIZE)) {
75 mem[p]++;
76 } else {
77 VERBOSE (ERROR, fprintf (stderr, "%s: invalid address (%d)\n", progname, p));
78 return 1;
79 }
80 break;
81 case '-': /* decrease pointer value */
82 if ((p >= 0) && (p < MEMSIZE)) {
83 mem[p]--;
84 } else {
85 VERBOSE (ERROR, fprintf (stderr, "%s: invalid address (%d)\n", progname, p));
86 return 1;
87 }
88 break;
89 case '.': /* output pointer value */
90 if ((p >= 0) && (p < MEMSIZE)) {
91 fprintf (out, "%c", mem[p]);
92 } else {
93 VERBOSE (ERROR, fprintf (stderr, "%s: invalid address (%d)\n", progname, p));
94 return 1;
95 }
96 break;
97 case ',': /* read a byte and store it in memory */
98 if ((p >= 0) && (p < MEMSIZE)) {
096fb1e3
LM
99 int c = getchar ();
100 mem[p] = (c > 0) ? c : 0;
07cc2c73
LM
101 } else {
102 VERBOSE (ERROR, fprintf (stderr, "%s: invalid address (%d)\n", progname, p));
103 return 1;
104 }
105 break;
106 case '[': /* jump to right bracket if pointer is set to 0 */
3d8520e3
LM
107 if ((p < 0) || (p >= MEMSIZE)) {
108 VERBOSE (ERROR, fprintf (stderr, "%s: invalid address (%d)\n", progname, p));
109 return 1;
110 }
07cc2c73
LM
111 if (mem[p] == 0) {
112 int bracket = 1;
113 while ((++i < nb) && (bracket > 0)) {
114 bracket += (buffer[i] == '[') ? +1 : (buffer[i] == ']') ? -1 :0;
115 }
116 if (bracket) {
117 VERBOSE (ERROR, fprintf (stderr, "%s: brace not closed\n", progname));
118 return 1;
119 }
3d8520e3 120 i--;
07cc2c73 121 } else {
3d8520e3
LM
122 if (j >= MAXJUMP) {
123 VERBOSE (ERROR, fprintf (stderr, "%s: too many jump\n", progname));
124 return 1;
125 }
126 jump[j++] = i;
07cc2c73
LM
127 }
128 break;
129 case ']': /* jump to left bracket if pointer is different to 0 */
3d8520e3
LM
130 if (j <= 0) {
131 VERBOSE (ERROR, fprintf (stderr, "%s: can't jump back\n", progname));
132 return 1;
07cc2c73 133 }
3d8520e3 134 i = jump[--j] - 1;
07cc2c73 135 break;
096fb1e3
LM
136 case ' ':
137 case '\t':
07cc2c73
LM
138 case '\n':
139 case '\r':
140 break;
141 default:
142 VERBOSE (WARNING, fprintf (stderr, "%s: can't understand '%c'\n", progname, buffer[i]));
143 }
144
04c960d9 145 //VERBOSE (DEBUG, fprintf (stderr, "%s: p: %d mem:", progname, p); for (int _i = 0; _i < MEMSIZE; _i++) fprintf (stderr, " %d", mem[_i]); fprintf (stderr, "\n"));
07cc2c73
LM
146
147 }
148
149 return 0;
150}
151
048e6e55
LM
152/* main function */
153
154int main (int argc, char *argv[])
155{
156 char *input = NULL;
157 char *output = NULL;
158 char *buffer = NULL;
159 int n = 0;
160 int size = 0;
161 FILE *fid = NULL;
162
163 /* get basename */
164 char *pt = progname = argv[0];
165 while (*pt) {
166 if ((*pt == '/') || (*pt == '\\')) {
167 progname = pt + 1;
168 }
169 pt++;
170 }
171
172 int c;
096fb1e3 173 while ((c = getopt(argc, argv, "e:i:m:o:hv:")) != EOF) {
048e6e55 174 switch (c) {
096fb1e3 175 case 'e':
04c960d9 176 free (buffer);
096fb1e3 177 buffer = strdup (optarg);
72c52012 178 CHECKALLOC (buffer);
096fb1e3
LM
179 size = strlen (buffer) + 1;
180 break;
048e6e55
LM
181 case 'i':
182 input = optarg;
183 break;
e07f5d2a 184 case 'm':
04c960d9 185 for (int i = 0; i < MEMSIZE; i++) {
e07f5d2a
LM
186 if (optarg) {
187 mem[i] = strtol (optarg, &optarg, 10);
188 VERBOSE (DEBUG, fprintf (stderr, "%s: mem[%d] = %d\n", progname, i, mem[i]));
189 }
190 }
191 if (*optarg != 0) {
32237385 192 VERBOSE (WARNING, fprintf (stderr, "%s: too many memory values or incorrect value\n", progname));
04c960d9 193 free (buffer);
e07f5d2a
LM
194 return 1;
195 }
196 break;
048e6e55
LM
197 case 'o':
198 output = optarg;
199 break;
200 case 'v':
201 verbose = atoi (optarg);
202 break;
203 case 'h':
204 default:
04c960d9 205 free (buffer);
048e6e55
LM
206 return usage (c != 'h');
207 }
208 }
209 if (argc - optind != 0) {
210 VERBOSE (ERROR, fprintf (stderr, "%s: invalid option -- %s\n", progname, argv[optind]));
04c960d9 211 free (buffer);
048e6e55
LM
212 return usage (1);
213 }
214
215 /* check input file */
216 if (input) {
217 fid = fopen (input, "r");
218 if (fid == NULL) {
219 VERBOSE (ERROR, fprintf (stderr, "%s: can't open file '%s' for reading\n", progname, input));
04c960d9 220 free (buffer);
048e6e55
LM
221 return 1;
222 }
096fb1e3 223 } else if (buffer == NULL) {
048e6e55
LM
224 fid = stdin;
225 }
226
227 /* read input file */
096fb1e3
LM
228 if (fid) {
229 while (!feof (fid)) {
230 buffer = (char *) realloc (buffer, size + BUFSIZE);
72c52012 231 CHECKALLOC (buffer);
096fb1e3
LM
232 memset (buffer + size, 0, BUFSIZE);
233 n = fread (buffer + size, 1, BUFSIZE, fid);
04c960d9 234 if (ferror (fid)) {
096fb1e3 235 VERBOSE (ERROR, fprintf (stderr, "%s: can't read data from file '%s'\n", progname, input));
04c960d9
LM
236 if (fid != stdin) {
237 fclose (fid);
238 }
239 free (buffer);
096fb1e3
LM
240 return 1;
241 }
242 size += BUFSIZE;
048e6e55 243 }
048e6e55 244
3d8520e3
LM
245 if (fid != stdin) {
246 fclose (fid);
247 }
096fb1e3
LM
248 VERBOSE (DEBUG, fprintf (stderr, "%s: read %d bytes\n", progname, size + n - BUFSIZE));
249 } else {
250 VERBOSE (DEBUG, fprintf (stderr, "%s: prog %d bytes\n", progname, size -1));
251 }
048e6e55
LM
252
253 /* check output file */
254 if (output) {
255 fid = fopen (output, "w");
256 if (fid == NULL) {
257 VERBOSE (ERROR, fprintf (stderr, "%s: can't open file '%s' for writing\n", progname, output));
04c960d9 258 free (buffer);
048e6e55
LM
259 return 1;
260 }
261 } else {
262 fid = stdout;
263 }
264
265 /* main process */
266 int rc = process (buffer, size, fid);
267
268 /* close output file */
3d8520e3
LM
269 if (fid != stdout) {
270 fclose (fid);
271 }
272
a213b0ae 273
04c960d9 274 VERBOSE (INFO, fprintf (stdout, "\nmemory:"); for (int _i = 0; _i < MEMSIZE; _i++) fprintf (stdout," %d", (int)mem[_i]); fprintf (stdout,"\n"));
048e6e55 275
04c960d9 276 free (buffer);
048e6e55
LM
277 return rc;
278}
279
048e6e55
LM
280// test: bf.exe -h
281// test: bf.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
32237385 282
048e6e55
LM
283// test: bf.exe -_ 2> /dev/null | awk 'END { if (NR == 0) { exit(0) } else exit (1) }'
284// test: bf.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
32237385
LM
285// test: bf.exe error 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
286
096fb1e3
LM
287// test: bf.exe -i error.b 2>&1 | grep "can't open" | grep -q "reading"
288// test: echo ">>." | bf.exe -o error/error.b 2>&1 | grep "can't open" | grep -q "writing"
32237385
LM
289// test: echo "error" | bf.exe -v1 2>&1 | grep -q "can't understand"
290// test: echo "<+" | bf.exe -v1 2>&1 | grep -q "invalid address"
291// test: echo "<-" | bf.exe -v1 2>&1 | grep -q "invalid address"
292// test: echo "<." | bf.exe -v1 2>&1 | grep -q "invalid address"
293// test: echo "<," | bf.exe -v1 2>&1 | grep -q "invalid address"
294// test: echo "<[" | bf.exe -v1 2>&1 | grep -q "invalid address"
295// test: echo "[" | bf.exe -v1 2>&1 | grep -q "brace not closed"
296// test: echo "+[[[[[]]]]]" | bf.exe -v1 2>&1 | grep -q "too many jump"
297// test: bf.exe -m "0 1 2 3 4 5 6 7 8 9 10" -e '.' -v1 2>&1 | grep -q "too many memory values"
298// test: bf.exe -m "0 error" -e '.' -v1 2>&1 | grep -q "incorrect value"
299// test: echo "]" | bf.exe -v1 2>&1 | grep -q "can't jump back"
300
096fb1e3 301// test: echo '+++>++>>-<--' | bf.exe -v2 | grep -q "memory: 3 2 -2 -1 0"
32237385 302// test: bf.exe -e '+++' -e '+++>++>>-<--' -v2 | grep -q "memory: 3 2 -2 -1 0"
096fb1e3 303// test: bf.exe -e '+++>++>>-<--' -v2 | grep -q "memory: 3 2 -2 -1 0"
3d8520e3 304// test: bf.exe -m "51 50 49" -e '.>.>.' |grep -q "321"
20c60362 305// test: bf.exe -m "51 50 49" -e '.>.>.' -o test.log && grep -q "321" test.log; rc=$?; rm test.log; test $rc -eq 0
ee55e48c 306// test: echo '+++>++>>-<--' > test.b; bf.exe -i test.b | grep -q "memory: 3 2 -2 -1 0"; rc=$?; rm test.b; test $rc -eq 0
32237385 307
e07f5d2a
LM
308// test: echo '++++++++++[>+++++++>++++++++++>+++>+<<<<-]>++.>+.+++++++..+++.>++.<<+++++++++++++++.>.+++.------.--------.>+.>.' | bf.exe -v1 | grep -q "Hello World!"
309// test: echo '++++++++++[>+>+++>+++++++>++++++++++<<<<-]>>>++++++++++++++.>+++++++++++++++++.<<++.>+++++++++++++.>--.<<.>+++.+.--.>----.++++++.+.<++.>----.++.<<.>>+.-------.<<.>>++.<.>+++++.<<.>-.+.<.>---.>---.<-.++++++++.>----.<---.>+++++++.<---.++++++++.' | bf.exe -v1 | grep -q "Tu as decouvert un peu de brainfuck"
32237385 310
096fb1e3 311// test: echo -e "123\0" | bf.exe -e ',[>,]' -v2 | grep -q "memory: 49 50 51 0"
3d8520e3
LM
312// test: echo -e "4+3\0" | bf.exe -e ',>++++++[<-------->-],,[<+>-]<.' -v1 | grep -q 7
313// test: echo -e "1+7\0" | bf.exe -e ',>++++++[<-------->-],,[<+>-]<.' -v1 | grep -q 8
314// test: echo -e "3*2\0" | bf.exe -e ',>,,>++++++++[<------<------>>-]<<[>[>+>+<<-]>>[<<+>>-]<<<-]>>>++++++[<++++++++>-]<.' -v1 | grep -q 6
315// test: echo -e "1*7\0" | bf.exe -e ',>,,>++++++++[<------<------>>-]<<[>[>+>+<<-]>>[<<+>>-]<<<-]>>>++++++[<++++++++>-]<.' -v1 | grep -q 7
048e6e55
LM
316
317/* vim: set ts=4 sw=4 et: */