diff options
| -rw-r--r-- | test-instr.c | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/test-instr.c b/test-instr.c index 71838462..3750b1b5 100644 --- a/test-instr.c +++ b/test-instr.c @@ -23,15 +23,19 @@ int main(int argc, char** argv) { char buff[8]; char* buf = buff; - if (argc > 1) + // we support command line parameter and stdin + if (argc > 1) { + buf = argv[1]; - else if (read(0, buf, sizeof(buf)) < 1) { + + } else if (read(0, buf, sizeof(buf)) < 1) { printf("Hum?\n"); - exit(1); + return 1; } + // we support three input cases (plus a 4th if stdin is used but there is no input) if (buf[0] == '0') printf("Looks like a zero to me!\n"); else if (buf[0] == '1') @@ -39,7 +43,6 @@ int main(int argc, char** argv) { else printf("Neither one or zero? How quaint!\n"); - exit(0); + return 0; } - |