Add a test suite for seq.

[oweals/busybox.git] / miscutils / dc.c

diff --git a/miscutils/dc.c b/miscutils/dc.c
index 48aa830d6117ffb676c00be3910ae79f05afd5ca..112f6df3ff866ccc3d1834b1810c0cb957f2dd00 100644 (file)
--- a/miscutils/dc.c
+++ b/miscutils/dc.c
@@ -1,86 +1,122 @@
 /* vi: set sw=4 ts=4: */
-#include "busybox.h"
 #include <ctype.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 #include <unistd.h>
 #include <math.h>
+#include "busybox.h"
 
 /* Tiny RPN calculator, because "expr" didn't give me bitwise operations. */
 
 static double stack[100];
 static unsigned int pointer;
+static unsigned char base;
 
 static void push(double a)
 {
-       if (pointer >= (sizeof(stack) / sizeof(*stack))) {
-               errorMsg("stack overflow\n");
-               exit(-1);
-       } else
-               stack[pointer++] = a;
+       if (pointer >= (sizeof(stack) / sizeof(*stack)))
+               bb_error_msg_and_die("stack overflow");
+       stack[pointer++] = a;
 }
 
-static double pop()
+static double pop(void)
 {
-       if (pointer == 0) {
-               errorMsg("stack underflow\n");
-               exit(-1);
-       }
+       if (pointer == 0)
+               bb_error_msg_and_die("stack underflow");
        return stack[--pointer];
 }
 
-static void add()
+static void add(void)
 {
        push(pop() + pop());
 }
 
-static void sub()
+static void sub(void)
 {
        double subtrahend = pop();
 
        push(pop() - subtrahend);
 }
 
-static void mul()
+static void mul(void)
 {
        push(pop() * pop());
 }
 
-static void divide()
+static void power(void)
+{
+       double topower = pop();
+
+       push(pow(pop(), topower));
+}
+
+static void divide(void)
 {
        double divisor = pop();
 
        push(pop() / divisor);
 }
 
-static void and()
+static void mod(void)
+{
+       unsigned int d = pop();
+
+       push((unsigned int) pop() % d);
+}
+
+static void and(void)
 {
        push((unsigned int) pop() & (unsigned int) pop());
 }
 
-static void or()
+static void or(void)
 {
        push((unsigned int) pop() | (unsigned int) pop());
 }
 
-static void eor()
+static void eor(void)
 {
        push((unsigned int) pop() ^ (unsigned int) pop());
 }
 
-static void not()
+static void not(void)
 {
        push(~(unsigned int) pop());
 }
 
-static void print()
+static void set_output_base(void)
 {
-       printf("%g\n", pop());
+       base=(unsigned char)pop();
+       if ((base != 10) && (base != 16)) {
+               fprintf(stderr, "Error: base = %d is not supported.\n", base);
+               base=10;
+       }
+}
+
+static void print_base(double print)
+{
+       if (base == 16)
+               printf("%x\n", (unsigned int)print);
+       else
+       printf("%g\n", print);
+}
+
+static void print_stack_no_pop(void)
+{
+       unsigned int i=pointer;
+       while (i)
+               print_base(stack[--i]);
+}
+
+static void print_no_pop(void)
+{
+       print_base(stack[pointer-1]);
 }
 
 struct op {
        const char *name;
-       void (*function) ();
+       void (*function) (void);
 };
 
 static const struct op operators[] = {
@@ -92,10 +128,19 @@ static const struct op operators[] = {
        {"mul", mul},
        {"/",   divide},
        {"div", divide},
+       {"**",  power},
+       {"exp", power},
+       {"pow", power},
+       {"%",   mod},
+       {"mod", mod},
        {"and", and},
        {"or",  or},
        {"not", not},
        {"eor", eor},
+       {"xor", eor},
+       {"p", print_no_pop},
+       {"f", print_stack_no_pop},
+       {"o", set_output_base},
        {0,     0}
 };
 
@@ -105,10 +150,8 @@ static void stack_machine(const char *argument)
        double d;
        const struct op *o = operators;
 
-       if (argument == 0) {
-               print();
+       if (argument == 0)
                return;
-       }
 
        d = strtod(argument, &endPointer);
 
@@ -124,12 +167,11 @@ static void stack_machine(const char *argument)
                }
                o++;
        }
-       errorMsg("%s: syntax error.\n", argument);
-       exit(-1);
+       bb_error_msg_and_die("%s: syntax error.", argument);
 }
 
 /* return pointer to next token in buffer and set *buffer to one char
- * past the end of the above mentioned token 
+ * past the end of the above mentioned token
  */
 static char *get_token(char **buffer)
 {
@@ -139,7 +181,7 @@ static char *get_token(char **buffer)
        while (isspace(*current)) { current++; }
        if (*current != 0) {
                start = current;
-               while (!isspace(*current) && current != 0) { current++; }
+               while (!isspace(*current) && *current != 0) { current++; }
                *buffer = current;
        }
        return start;
@@ -162,7 +204,7 @@ int dc_main(int argc, char **argv)
                char *line   = NULL;
                char *cursor = NULL;
                char *token  = NULL;
-               while ((line = get_line_from_file(stdin))) {
+               while ((line = bb_get_chomped_line_from_file(stdin))) {
                        cursor = line;
                        len = number_of_tokens(line);
                        for (i = 0; i < len; i++) {
@@ -174,7 +216,7 @@ int dc_main(int argc, char **argv)
                }
        } else {
                if (*argv[1]=='-')
-                       usage(dc_usage);
+                       bb_show_usage();
                while (argc >= 2) {
                        stack_machine(argv[1]);
                        argv++;
@@ -182,5 +224,5 @@ int dc_main(int argc, char **argv)
                }
        }
        stack_machine(0);
-       return( TRUE);
+       return EXIT_SUCCESS;
 }