busybox/miscutils/dc.c

306 lines
5.1 KiB
C
Raw Normal View History

/* vi: set sw=4 ts=4: */
/*
* Licensed under GPLv2 or later, see file LICENSE in this source tree.
*/
/* config/applet/usage bits are in bc.c */
//#include "libbb.h"
//#include "common_bufsiz.h"
#include <math.h>
#if 0
typedef unsigned data_t;
#define DATA_FMT ""
#elif 0
typedef unsigned long data_t;
#define DATA_FMT "l"
#else
typedef unsigned long long data_t;
#define DATA_FMT "ll"
#endif
1999-10-05 21:54:54 +05:30
struct globals {
unsigned pointer;
unsigned base;
double stack[1];
} FIX_ALIASING;
enum { STACK_SIZE = (COMMON_BUFSIZE - offsetof(struct globals, stack)) / sizeof(double) };
#define G (*(struct globals*)bb_common_bufsiz1)
#define pointer (G.pointer )
#define base (G.base )
#define stack (G.stack )
#define INIT_G() do { \
setup_common_bufsiz(); \
base = 10; \
} while (0)
static unsigned check_under(void)
{
unsigned p = pointer;
if (p == 0)
bb_error_msg_and_die("stack underflow");
return p - 1;
}
static void push(double a)
1999-10-05 21:54:54 +05:30
{
if (pointer >= STACK_SIZE)
2003-03-19 14:43:01 +05:30
bb_error_msg_and_die("stack overflow");
stack[pointer++] = a;
1999-10-05 21:54:54 +05:30
}
2001-12-06 08:59:37 +05:30
static double pop(void)
1999-10-05 21:54:54 +05:30
{
unsigned p = check_under();
pointer = p;
return stack[p];
1999-10-05 21:54:54 +05:30
}
2001-12-06 08:59:37 +05:30
static void add(void)
1999-10-05 21:54:54 +05:30
{
push(pop() + pop());
}
2001-12-06 08:59:37 +05:30
static void sub(void)
1999-10-05 21:54:54 +05:30
{
double subtrahend = pop();
1999-10-05 21:54:54 +05:30
push(pop() - subtrahend);
}
2001-12-06 08:59:37 +05:30
static void mul(void)
1999-10-05 21:54:54 +05:30
{
push(pop() * pop());
}
#if ENABLE_FEATURE_DC_LIBM
static void power(void)
{
double topower = pop();
push(pow(pop(), topower));
}
#endif
2001-12-06 08:59:37 +05:30
static void divide(void)
1999-10-05 21:54:54 +05:30
{
double divisor = pop();
1999-10-05 21:54:54 +05:30
push(pop() / divisor);
}
static void mod(void)
{
data_t d = pop();
/* compat with dc (GNU bc 1.07.1) 1.4.1:
* $ dc -e '4 0 % p'
* dc: remainder by zero
* 0
*/
if (d == 0) {
bb_error_msg("remainder by zero");
pop();
push(0);
return;
}
/* ^^^^ without this, we simply get SIGFPE and die */
push((data_t) pop() % d);
}
2001-12-06 08:59:37 +05:30
static void and(void)
1999-10-05 21:54:54 +05:30
{
push((data_t) pop() & (data_t) pop());
1999-10-05 21:54:54 +05:30
}
2001-12-06 08:59:37 +05:30
static void or(void)
1999-10-05 21:54:54 +05:30
{
push((data_t) pop() | (data_t) pop());
1999-10-05 21:54:54 +05:30
}
2001-12-06 08:59:37 +05:30
static void eor(void)
1999-10-05 21:54:54 +05:30
{
push((data_t) pop() ^ (data_t) pop());
1999-10-05 21:54:54 +05:30
}
2001-12-06 08:59:37 +05:30
static void not(void)
1999-10-05 21:54:54 +05:30
{
push(~(data_t) pop());
1999-10-05 21:54:54 +05:30
}
static void set_output_base(void)
{
static const char bases[] ALIGN1 = { 2, 8, 10, 16, 0 };
unsigned b = (unsigned)pop();
base = *strchrnul(bases, b);
if (base == 0) {
bb_error_msg("error, base %u is not supported", b);
base = 10;
}
}
static void print_base(double print)
{
data_t x, i;
x = (data_t) print;
if (base == 10) {
if (x == print) /* exactly representable as unsigned integer */
printf("%"DATA_FMT"u\n", x);
else
printf("%g\n", print);
return;
}
switch (base) {
case 16:
printf("%"DATA_FMT"x\n", x);
break;
case 8:
printf("%"DATA_FMT"o\n", x);
break;
default: /* base 2 */
i = MAXINT(data_t) - (MAXINT(data_t) >> 1);
/* i is 100000...00000 */
do {
if (x & i)
break;
i >>= 1;
} while (i > 1);
do {
bb_putchar('1' - !(x & i));
i >>= 1;
} while (i);
bb_putchar('\n');
}
}
static void print_stack_no_pop(void)
{
unsigned i = pointer;
while (i)
print_base(stack[--i]);
}
static void print_no_pop(void)
{
print_base(stack[check_under()]);
}
1999-10-05 21:54:54 +05:30
struct op {
const char name[4];
2001-12-06 08:59:37 +05:30
void (*function) (void);
1999-10-05 21:54:54 +05:30
};
static const struct op operators[] = {
#if ENABLE_FEATURE_DC_LIBM
{"^", power},
// {"exp", power},
// {"pow", power},
#endif
{"%", mod},
// {"mod", mod},
// logic ops are not standard, remove?
{"and", and},
{"or", or},
{"not", not},
{"xor", eor},
{"+", add},
// {"add", add},
{"-", sub},
// {"sub", sub},
{"*", mul},
// {"mul", mul},
{"/", divide},
// {"div", divide},
{"p", print_no_pop},
{"f", print_stack_no_pop},
{"o", set_output_base},
1999-10-05 21:54:54 +05:30
};
/* Feed the stack machine */
static void stack_machine(const char *argument)
1999-10-05 21:54:54 +05:30
{
char *end;
double number;
const struct op *o;
1999-10-05 21:54:54 +05:30
next:
number = strtod(argument, &end);
if (end != argument) {
argument = end;
push(number);
goto next;
1999-10-05 21:54:54 +05:30
}
/* We might have matched a digit, eventually advance the argument */
argument = skip_whitespace(argument);
if (*argument == '\0')
return;
o = operators;
do {
libbb: introduce and use is_prefixed_with() function old new delta is_prefixed_with - 18 +18 complete_username 78 77 -1 man_main 737 735 -2 fsck_device 429 427 -2 unpack_ar_archive 80 76 -4 strip_unsafe_prefix 105 101 -4 singlemount 1054 1050 -4 rtc_adjtime_is_utc 90 86 -4 resolve_mount_spec 88 84 -4 parse_one_line 1029 1025 -4 parse_conf 1460 1456 -4 may_wakeup 83 79 -4 loadkmap_main 219 215 -4 get_irqs_from_stat 103 99 -4 get_header_cpio 913 909 -4 findfs_main 79 75 -4 fbsplash_main 1230 1226 -4 load_crontab 776 771 -5 expand_vars_to_list 1151 1146 -5 date_main 881 876 -5 skip_dev_pfx 30 24 -6 make_device 2199 2193 -6 complete_cmd_dir_file 773 767 -6 run_applet_and_exit 715 708 -7 uudecode_main 321 313 -8 pwdx_main 197 189 -8 execute 568 560 -8 i2cdetect_main 1186 1176 -10 procps_scan 1242 1230 -12 procps_read_smaps 1017 1005 -12 process_module 746 734 -12 patch_main 1903 1891 -12 nfsmount 3572 3560 -12 stack_machine 126 112 -14 process_timer_stats 449 435 -14 match_fstype 111 97 -14 do_ipaddr 1344 1330 -14 open_list_and_close 359 343 -16 get_header_tar 1795 1779 -16 prepend_new_eth_table 340 323 -17 fsck_main 1811 1794 -17 find_iface_state 56 38 -18 dnsd_main 1321 1303 -18 base_device 179 158 -21 find_keyword 104 82 -22 handle_incoming_and_exit 2785 2762 -23 parse_and_put_prompt 774 746 -28 modinfo 347 317 -30 find_action 204 171 -33 update_passwd 1470 1436 -34 ------------------------------------------------------------------------------ (add/remove: 1/0 grow/shrink: 0/49 up/down: 18/-540) Total: -522 bytes Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
2015-03-12 22:18:34 +05:30
char *after_name = is_prefixed_with(argument, o->name);
if (after_name) {
argument = after_name;
o->function();
goto next;
1999-10-05 21:54:54 +05:30
}
o++;
} while (o != operators + ARRAY_SIZE(operators));
bb_error_msg_and_die("syntax error at '%s'", argument);
1999-10-05 21:54:54 +05:30
}
static void process_file(FILE *fp)
{
char *line;
while ((line = xmalloc_fgetline(fp)) != NULL) {
stack_machine(line);
free(line);
}
}
int dc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
2008-07-05 14:48:54 +05:30
int dc_main(int argc UNUSED_PARAM, char **argv)
{
bool script = 0;
INIT_G();
/* Run -e'SCRIPT' and -fFILE in order of appearance, then handle FILEs */
for (;;) {
int n = getopt(argc, argv, "e:f:");
if (n <= 0)
break;
switch (n) {
case 'e':
script = 1;
stack_machine(optarg);
break;
case 'f':
script = 1;
process_file(xfopen_for_read(optarg));
break;
default:
bb_show_usage();
}
1999-10-05 21:54:54 +05:30
}
argv += optind;
if (*argv) {
do
process_file(xfopen_for_read(*argv++));
while (*argv);
} else if (!script) {
/* Take stuff from stdin if no args are given */
process_file(stdin);
}
return EXIT_SUCCESS;
1999-10-05 21:54:54 +05:30
}