diff options
Diffstat (limited to 'toys')
| -rw-r--r-- | toys/other/hello.c | 56 | ||||
| -rw-r--r-- | toys/other/vconfig.c | 14 | ||||
| -rw-r--r-- | toys/pending/compress.c | 189 | ||||
| -rw-r--r-- | toys/pending/init.c | 44 | ||||
| -rw-r--r-- | toys/posix/date.c | 160 | ||||
| -rw-r--r-- | toys/posix/du.c | 14 | ||||
| -rw-r--r-- | toys/posix/grep.c | 98 |
7 files changed, 154 insertions, 421 deletions
diff --git a/toys/other/hello.c b/toys/other/hello.c index b909ed53..eb80972e 100644 --- a/toys/other/hello.c +++ b/toys/other/hello.c @@ -8,28 +8,17 @@ // Accept many different kinds of command line argument: USE_HELLO(NEWTOY(hello, "(walrus)(blubber):;(also):e@d*c#b:a", TOYFLAG_USR|TOYFLAG_BIN)) -USE_HELLO_ALIAS(NEWTOY(hello_alias, "b:dq", TOYFLAG_USR|TOYFLAG_BIN)) config HELLO bool "hello" default n help - usage: hello [-a] [-b STRING] [-c NUMBER] [-d LIST] [-e COUNT] [...] + usage: hello [-a] [-b string] [-c number] [-d list] [-e count] [...] A hello world program. You don't need this. Mostly used as an example/skeleton file for adding new commands, occasionally nice to test kernel booting via "init=/bin/hello". - -config HELLO_ALIAS - bool "hello_alias" - default n - depends on HELLO - help - usage: hello_alias [-dq] [-b NUMBER] - - Example of a second command with different arguments in the same source - file as the first. Allows shared infrastructure not added to lib. */ #define FOR_hello @@ -38,19 +27,12 @@ config HELLO_ALIAS // Hello doesn't use these globals, they're here for example/skeleton purposes. GLOBALS( - union { - struct { - char *b_string; - long c_number; - struct arg_list *d_list; - long e_count; - char *also_string; - char *blubber_string; - } h; - struct { - long b_number; - } a; - }; + char *b_string; + long c_number; + struct arg_list *d_list; + long e_count; + char *also_string; + char *blubber_string; int more_globals; ) @@ -65,25 +47,15 @@ void hello_main(void) if (toys.optflags) printf("flags=%x\n", toys.optflags); if (toys.optflags & FLAG_a) printf("Saw a\n"); - if (toys.optflags & FLAG_b) printf("b=%s\n", TT.h.b_string); - if (toys.optflags & FLAG_c) printf("c=%ld\n", TT.h.c_number); - while (TT.h.d_list) { - printf("d=%s\n", TT.h.d_list->arg); - TT.h.d_list = TT.h.d_list->next; + if (toys.optflags & FLAG_b) printf("b=%s\n", TT.b_string); + if (toys.optflags & FLAG_c) printf("c=%ld\n", TT.c_number); + while (TT.d_list) { + printf("d=%s\n", TT.d_list->arg); + TT.d_list = TT.d_list->next; } - if (TT.h.e_count) printf("e was seen %ld times\n", TT.h.e_count); + if (TT.e_count) printf("e was seen %ld times\n", TT.e_count); for (optargs = toys.optargs; *optargs; optargs++) printf("optarg=%s\n", *optargs); if (toys.optflags & FLAG_walrus) printf("Saw --walrus\n"); - if (TT.h.blubber_string) printf("--blubber=%s\n", TT.h.blubber_string); -} - -#define CLEANUP_hello -#define FOR_hello_alias -#include "generated/flags.h" - -void hello_alias_main(void) -{ - printf("hello world %x\n", toys.optflags); - if (toys.optflags & FLAG_b) printf("b=%ld", TT.a.b_number); + if (TT.blubber_string) printf("--blubber=%s\n", TT.blubber_string); } diff --git a/toys/other/vconfig.c b/toys/other/vconfig.c index 5a737c78..d2422529 100644 --- a/toys/other/vconfig.c +++ b/toys/other/vconfig.c @@ -47,18 +47,6 @@ void vconfig_main(void) memset(&request, 0, sizeof(struct vlan_ioctl_args)); cmd = toys.optargs[0]; -//add ADD_VLAN_CMD 4094 0 // ADD_VLAN_CMD -//rem DEL_VLAN_CMD 0 0 // DEL_VLAN_CMD -//set_ingress_map INT_MAX 0 // SET_VLAN_INGRESS_PRIORITY_CMD -//set_egress_map // SET_VLAN_EGRESS_PRIORITY_CMD -//GET_VLAN_INGRESS_PRIORITY_CMD, -//GET_VLAN_EGRESS_PRIORITY_CMD, -//set_name_type // SET_VLAN_NAME_TYPE_CMD -//set_flag // SET_VLAN_FLAG_CMD, -//GET_VLAN_REALDEV_NAME_CMD, -//GET_VLAN_VID_CMD - - if (!strcmp(cmd, "set_name_type")) { char *types[] = {"VLAN_PLUS_VID", "DEV_PLUS_VID", "VLAN_PLUS_VID_NO_PAD", "DEV_PLUS_VID_NO_PAD"}; @@ -77,7 +65,7 @@ void vconfig_main(void) } // Store interface name - xstrncpy(request.device1, toys.optargs[1], 23); + xstrncpy(request.device1, toys.optargs[1], 16); if (!strcmp(cmd, "add")) { request.cmd = ADD_VLAN_CMD; diff --git a/toys/pending/compress.c b/toys/pending/compress.c index 48cec494..bb4af46c 100644 --- a/toys/pending/compress.c +++ b/toys/pending/compress.c @@ -5,8 +5,7 @@ * The inflate/deflate code lives here, so the various things that use it * either live here or call these commands to pipe data through them. * - * Divergence from posix: replace obsolete/patented "compress" with mutiplexer. - * (gzip already replaces "uncompress".) + * Divergence from posix: replace obsolete "compress" with mutiplexer. * * See RFCs 1950 (zlib), 1951 (deflate), and 1952 (gzip) * LSB 4.1 has gzip, gunzip, and zcat @@ -15,10 +14,8 @@ // Accept many different kinds of command line argument. // Leave Lrg at end so flag values line up. -USE_COMPRESS(NEWTOY(compress, "zcd9lrg[-cd][!zgLr]", TOYFLAG_USR|TOYFLAG_BIN)) -USE_COMPRESS(NEWTOY(gzip, USE_GZIP_D("d")"19dcflqStvgLRz[!gLRz]", TOYFLAG_USR|TOYFLAG_BIN)) -USE_DECOMPRESS(NEWTOY(zcat, 0, TOYFLAG_USR|TOYFLAG_BIN)) -USE_DECOMPRESS(NEWTOY(gunzip, "cflqStv", TOYFLAG_USR|TOYFLAG_BIN)) +USE_COMPRESS(NEWTOY(compress, "zcd9Lrg[-cd][!zgLr]", TOYFLAG_USR|TOYFLAG_BIN)) +USE_COMPRESS(NEWTOY(zcat, "aLrg[!aLrg]", TOYFLAG_USR|TOYFLAG_BIN)) //zip unzip gzip gunzip zcat @@ -26,114 +23,38 @@ config COMPRESS bool "compress" default n help - usage: compress [-zgLR19] [FILE] - - Compress or decompress file (or stdin) using "deflate" algorithm. - - -1 min compression - -9 max compression (default) - -g gzip (default) - -L zlib - -R raw - -z zip - -config GZIP - bool "gzip" - default y - depends on COMPRESS - help - usage: gzip [-19cfqStvzgLR] [FILE...] - - Compess (deflate) file(s). With no files, compress stdin to stdout. - - On successful decompression, compressed files are replaced with the - uncompressed version. The input file is removed and replaced with - a new file without the .gz extension (with same ownership/permissions). - - -1 Minimal compression (fastest) - -9 Max compression (default) - -c cat to stdout (act as zcat) - -f force (if output file exists, input is tty, unrecognized extension) - -q quiet (no warnings) - -S specify exension (default .*) - -t test compressed file(s) - -v verbose (like -l, but compress files) - - Compression type: - -g gzip (default) -L zlib -R raw -z zip - -config GZIP_D - bool - default y - depends on GZIP && DECOMPRESS - help - usage: gzip [-d] - - -d decompress (act as gunzip) - -config DECOMPRESS - bool "decompress" - default n - help usage: compress [-zglrcd9] [FILE] Compress or decompress file (or stdin) using "deflate" algorithm. - -c compress with -g gzip (default) -l zlib -r raw -z zip + -c compress with -g gzip (default) -L zlib -r raw -z zip -d decompress (autodetects type) - config ZCAT bool "zcat" - default y - depends on DECOMPRESS + default n + depends on COMPRESS help usage: zcat [FILE...] Decompress deflated file(s) to stdout - -config GUNZIP - bool "gunzip" - default y - depends on DECOMPRESS - help - usage: gunzip [-cflqStv] [FILE...] - - Decompess (deflate) file(s). With no files, compress stdin to stdout. - - On successful decompression, compressed files are replaced with the - uncompressed version. The input file is removed and replaced with - a new file without the .gz extension (with same ownership/permissions). - - -c cat to stdout (act as zcat) - -f force (output file exists, input is tty, unrecognized extension) - -l list compressed/uncompressed/ratio/name for each input file. - -q quiet (no warnings) - -S specify exension (default .*) - -t test compressed file(s) - -v verbose (like -l, but decompress files) */ #define FOR_compress #include "toys.h" GLOBALS( - // Huffman codes: base offset and extra bits tables (length and distance) + // base offset and extra bits tables (length and distance) char lenbits[29], distbits[30]; unsigned short lenbase[29], distbase[30]; void *fixdisthuff, *fixlithuff; - // CRC void (*crcfunc)(char *data, int len); - unsigned crc; - - // Compressed data buffer - char *data; - unsigned pos, len; - int fd; + unsigned crc, len; - // Tables only used for deflation - unsigned short *head, *chain; + char *outbuf; + unsigned outlen; + int outfd; ) // little endian bit buffer @@ -204,12 +125,12 @@ unsigned bitbuf_get(struct bitbuf *bb, int bits) return result; } -static void data_crc(char sym) +static void outbuf_crc(char sym) { - TT.data[TT.pos++ & 32767] = sym; + TT.outbuf[TT.outlen++ & 32767] = sym; - if (!(TT.pos & 32767)) { - xwrite(TT.fd, TT.data, 32768); + if (!(TT.outlen & 32767)) { + xwrite(TT.outfd, TT.outbuf, 32768); if (TT.crcfunc) TT.crcfunc(0, 32768); } } @@ -225,7 +146,7 @@ struct huff { // Create simple huffman tree from array of bit lengths. -// The symbols in the huffman trees are sorted (first by bit length +// The symbols in deflate's huffman trees are sorted (first by bit length // of the code to reach them, then by symbol number). This means that given // the bit length of each symbol, we can construct a unique tree. static void len2huff(struct huff *huff, char bitlen[], int len) @@ -264,33 +185,6 @@ static unsigned huff_and_puff(struct bitbuf *bb, struct huff *huff) return huff->symbol[start + offset]; } -// For deflate, TT.len = input read, TT.pos = input consumed -static void deflate(struct bitbuf *bb) -{ - char *data = TT.data; - int len, end = 0; - - TT.crc = ~0; - - // Read next half-window of data if we haven't hit EOF yet. -read_more: - len = readall(TT.fd, data + (TT.len & 32768), 32768); -fprintf(stderr, "read %d@%d\n", len, TT.pos); - if (len < 0) perror_exit("read"); // todo: add filename - if (len != 32768) end++; - TT.len += len; - - // repeat until spanked - while (TT.pos != TT.len) { - unsigned pos = TT.pos & 65535; - - if (!(pos & 32767) && !end) goto read_more; - - TT.pos++; - } -fprintf(stderr, "numberwang %d\n", TT.pos); -} - // Decompress deflated data from bitbuf to filehandle. static void inflate(struct bitbuf *bb) { @@ -322,7 +216,7 @@ static void inflate(struct bitbuf *bb) // dump bytes until done or end of current bitbuf contents if (bblen > len) bblen = len; pos = bblen; - while (pos--) data_crc(*(p++)); + while (pos--) outbuf_crc(*(p++)); bitbuf_skip(bb, bblen << 3); len -= bblen; } @@ -382,7 +276,7 @@ static void inflate(struct bitbuf *bb) int sym = huff_and_puff(bb, lithuff); // Literal? - if (sym < 256) data_crc(sym); + if (sym < 256) outbuf_crc(sym); // Copy range? else if (sym > 256) { @@ -392,9 +286,9 @@ static void inflate(struct bitbuf *bb) len = TT.lenbase[sym] + bitbuf_get(bb, TT.lenbits[sym]); sym = huff_and_puff(bb, disthuff); dist = TT.distbase[sym] + bitbuf_get(bb, TT.distbits[sym]); - sym = TT.pos & 32767; + sym = TT.outlen & 32767; - while (len--) data_crc(TT.data[(TT.pos-dist) & 32767]); + while (len--) outbuf_crc(TT.outbuf[(TT.outlen-dist) & 32767]); // End of block } else break; @@ -405,25 +299,18 @@ static void inflate(struct bitbuf *bb) if (final) break; } - if (TT.pos & 32767) { - xwrite(TT.fd, TT.data, TT.pos & 32767); - if (TT.crcfunc) TT.crcfunc(0, TT.pos & 32767); + if (TT.outlen & 32767) { + xwrite(TT.outfd, TT.outbuf, TT.outlen & 32767); + if (TT.crcfunc) TT.crcfunc(0, TT.outlen & 32767); } } -static void init_deflate(int compress) +static void init_deflate(void) { - int i, n = 1, size = 32768; + int i, n = 1; // Ye olde deflate window - TT.data = xmalloc(32768*(compress+1)); - if (compress) { - TT.head = (unsigned short *)(TT.data+65536); - TT.chain = TT.head + - -HASH_SIZE = 1 << 15; - - } + TT.outbuf = xmalloc(32768); // Calculate lenbits, lenbase, distbits, distbase *TT.lenbase = 3; @@ -478,22 +365,17 @@ void gzip_crc(char *data, int len) unsigned crc, *crc_table = (unsigned *)(toybuf+sizeof(toybuf)-1024); crc = TT.crc; - for (i=0; i<len; i++) crc = crc_table[(crc^TT.data[i])&0xff] ^ (crc>>8); + for (i=0; i<len; i++) crc = crc_table[(crc^TT.outbuf[i])&0xff] ^ (crc>>8); TT.crc = crc; TT.len += len; } -static void do_compress(int fd, char *name) -{ - xwrite(1, "\x1f\x8b\x08\0\0\0\0\0\x02\xff", 10); -} - static void do_zcat(int fd, char *name) { struct bitbuf *bb = bitbuf_init(fd, sizeof(toybuf)); if (!is_gzip(bb)) error_exit("not gzip"); - TT.fd = 1; + TT.outfd = 1; // Use last 1k of toybuf for little endian crc table crc_init((unsigned *)(toybuf+sizeof(toybuf)-1024), 1); @@ -516,20 +398,13 @@ void compress_main(void) zcat_main(); } -#define CLEANUP_compress -#define FOR_zcat -#include "generated/flags.h" +//#define CLEANUP_compress +//#define FOR_zcat +//#include "generated/flags.h" void zcat_main(void) { - init_deflate(0); + init_deflate(); loopfiles(toys.optargs, do_zcat); } - -//void gunzip_main(void) -//{ -// init_deflate(0); -// -// loopfiles(toys.optargs, do_zcat); -//} diff --git a/toys/pending/init.c b/toys/pending/init.c index ca3c6cc4..8199e748 100644 --- a/toys/pending/init.c +++ b/toys/pending/init.c @@ -13,10 +13,7 @@ config INIT help usage: init - System V style init. - - First program to run (as PID 1) when the system comes up, reading - /etc/inittab to determine actions. + init the system. */ #include "toys.h" @@ -44,9 +41,8 @@ int caught_signal; static void initialize_console(void) { int fd; - char *p = getenv("CONSOLE"); + char *p = (p = getenv("CONSOLE")) ? p : getenv("console"); - if (!p) p = getenv("console"); if (!p) { fd = open("/dev/null", O_RDWR); if (fd >= 0) { @@ -66,11 +62,11 @@ static void initialize_console(void) if (!getenv("TERM")) putenv("TERM=linux"); } -static void reset_term(int fd) +static void set_sane_term(void) { struct termios terminal; - tcgetattr(fd, &terminal); + tcgetattr(0, &terminal); terminal.c_cc[VINTR] = 3; //ctrl-c terminal.c_cc[VQUIT] = 28; /*ctrl-\*/ terminal.c_cc[VERASE] = 127; //ctrl-? @@ -81,16 +77,12 @@ static void reset_term(int fd) terminal.c_cc[VSUSP] = 26; //ctrl-z terminal.c_line = 0; - terminal.c_cflag &= CRTSCTS|PARODD|PARENB|CSTOPB|CSIZE|CBAUDEX|CBAUD; - terminal.c_cflag |= CLOCAL|HUPCL|CREAD; - - //enable start/stop input and output control + map CR to NL on input - terminal.c_iflag = IXON|IXOFF|ICRNL; - - //Map NL to CR-NL on output - terminal.c_oflag = ONLCR|OPOST; + terminal.c_cflag = terminal.c_cflag&(CRTSCTS|PARODD|PARENB|CSTOPB|CSIZE|CBAUDEX|CBAUD); + terminal.c_cflag = terminal.c_cflag|(CLOCAL|HUPCL|CREAD); + terminal.c_iflag = IXON|IXOFF|ICRNL;//enable start/stop input and output control + map CR to NL on input + terminal.c_oflag = ONLCR|OPOST;//Map NL to CR-NL on output terminal.c_lflag = IEXTEN|ECHOKE|ECHOCTL|ECHOK|ECHOE|ECHO|ICANON|ISIG; - tcsetattr(fd, TCSANOW, &terminal); + tcsetattr(0, TCSANOW, &terminal); } static void add_new_action(uint8_t action,char *command,char *term) @@ -258,7 +250,7 @@ static pid_t final_run(struct action_list_seed *x) dup2(0, 2); } } - reset_term(0); + set_sane_term(); run_command(x->command); _exit(-1); } @@ -289,7 +281,6 @@ static void waitforpid(pid_t pid) if (kill(y, 0)) break; } } - static void run_action_from_list(int action) { pid_t pid; @@ -311,7 +302,14 @@ static void set_default(void) { sigset_t signal_set_c; - sigatexit(SIG_DFL); + signal(SIGUSR1,SIG_DFL); + signal(SIGUSR2,SIG_DFL); + signal(SIGTERM,SIG_DFL); + signal(SIGQUIT,SIG_DFL); + signal(SIGINT,SIG_DFL); + signal(SIGHUP,SIG_DFL); + signal(SIGTSTP,SIG_DFL); + signal(SIGSTOP,SIG_DFL); sigfillset(&signal_set_c); sigprocmask(SIG_UNBLOCK,&signal_set_c, NULL); @@ -359,7 +357,6 @@ static void halt_poweroff_reboot_handler(int sig_no) while(1) sleep(1); } - static void restart_init_handler(int sig_no) { struct action_list_seed *x; @@ -389,7 +386,7 @@ static void restart_init_handler(int sig_no) } else { dup2(0, 1); dup2(0, 2); - reset_term(0); + set_sane_term(); run_command(x->command); } } @@ -443,7 +440,7 @@ void init_main(void) if (getpid() != 1) error_exit("Already running"); printf("Started init\n"); initialize_console(); - reset_term(0); + set_sane_term(); if (chdir("/")) perror_exit("Can't cd to /"); setsid(); @@ -452,7 +449,6 @@ void init_main(void) putenv("PATH=/sbin:/usr/sbin:/bin:/usr/bin"); putenv("SHELL=/bin/sh"); putenv("USER=root"); - inittab_parsing(); signal(SIGUSR1, halt_poweroff_reboot_handler);//halt signal(SIGUSR2, halt_poweroff_reboot_handler);//poweroff diff --git a/toys/posix/date.c b/toys/posix/date.c index a4e5e27c..e1ba636d 100644 --- a/toys/posix/date.c +++ b/toys/posix/date.c @@ -7,39 +7,21 @@ * Note: setting a 2 year date is 50 years back/forward from today, * not posix's hardwired magic dates. -USE_DATE(NEWTOY(date, "d:s:r:u", TOYFLAG_BIN)) +USE_DATE(NEWTOY(date, "r:u", TOYFLAG_BIN)) config DATE bool "date" default y help - usage: date [-u] [-r FILE] [-d DATE] [+DISPLAY_FORMAT] [-s SET_FORMAT] [SET] + usage: date [-u] [-r FILE] [+FORMAT] | mmddhhmm[[cc]yy[.ss]] - Set/get the current date/time. With no SET shows the current date. + Set/get the current date/time. - Default SET format is "MMDDhhmm[[CC]YY][.ss]", that's (2 digits each) - month, day, hour (0-23), and minute. Optionally century, year, and second. + Setting the date requires month, day, hour (0-23), and minute, each + two digits. It can optionally include year, century, and .seconds. - -d Show DATE instead of current time (convert date format) - -r Use modification time of FILE instead of current date - -s +FORMAT for SET or -d (instead of MMDDhhmm[[CC]YY][.ss]) - -u Use UTC instead of current timezone - - +FORMAT specifies display format string using these escapes: - - %% literal % %n newline %t tab - %S seconds (00-60) %M minute (00-59) %m month (01-12) - %H hour (0-23) %I hour (01-12) %p AM/PM - %y short year (00-99) %Y year %C century - %a short weekday name %A weekday name %u day of week (1-7, 1=mon) - %b short month name %B month name %Z timezone name - %j day of year (001-366) %d day of month (01-31) %e day of month ( 1-31) - - %U Week of year (0-53 start sunday) %W Week of year (0-53 start monday) - %V Week of year (1-53 start monday, week < 4 days not part of this year) - - %D = "%m/%d/%y" %r = "%I : %M : %S %p" %T = "%H:%M:%S" %h = "%b" - %x locale date %X locale time %c locale date/time + -u Use UTC timezone instead of current + -r Use date from FILE instead of current date */ #define FOR_date @@ -47,95 +29,72 @@ config DATE GLOBALS( char *file; - char *setfmt; - char *showdate; ) -// Handle default posix date format: mmddhhmm[[cc]yy] -// returns 0 success, nonzero for error -int parse_posixdate(char *str, struct tm *tm) -{ - struct timeval tv; - int len; - - len = 0; - sscanf(str, "%2u%2u%2u%2u%n", &tm.tm_mon, &tm.tm_mday, &tm.tm_hour, - &tm.tm_min, &len); - if (len != 8) goto bad_date; - str += len; - tm.tm_mon--; - - // If year specified, overwrite one we fetched earlier - if (*str && *str != '.') { - unsigned year, r1 = tm.tm_year % 100, r2 = (tm.tm_year + 50) % 100, - century = tm.tm_year - r1; - - len = 0; - sscanf(str, "%u%n", &year, &len); - if (len == 4) year -= 1900; - else if (len != 2) goto bad_date; - str += len; - - // 2 digit years, next 50 years are "future", last 50 years are "past". - // A "future" date in past is a century ahead. - // A non-future date in the future is a century behind. - if ((r1 < r2) ? (r1 < year && year < r2) : (year < r1 || year > r2)) { - if (year < r1) year += 100; - } else if (year > r1) year -= 100; - tm.tm_year = year + century; - } - if (*str == '.') { - len = 0; - sscanf(str, ".%u%n", &tm.tm_sec, &len); - str += len; - } - - return *str; -} - - void date_main(void) { - char *setdate = *toys.optargs, *format_string = "%a %b %e %H:%M:%S %Z %Y", - *tz; + const char *format_string = "%a %b %e %H:%M:%S %Z %Y"; time_t now = time(NULL); struct tm tm; - // We can't just pass a timezone to mktime because posix. - if (toys.optflags & FLAG_u) { - tz = CFG_TOYBOX_FREE ? getenv("TZ") : 0; - setenv("TZ", "UTC", 1); - tzset(); - } - if (TT.file) { struct stat st; xstat(TT.file, &st); now = st.st_mtim.tv_sec; - } else if (TT.showdate) { - if (TT.setfmt) { - char *s = strptime(TT.showdate, TT.setfmt, &tm); - - if (!s || !*s) goto bad_date; - } else if (parse_posixdate(TT.showdate, &tm)) goto bad_date; - } else localtime_r(&now, &tm); + } + ((toys.optflags & FLAG_u) ? gmtime_r : localtime_r)(&now, &tm); - // Fall through if no arguments - if (!setdate); // Display the date? - else if (*setdate == '+') { - format_string = toys.optargs[0]+1; - setdate = toys.optargs[1]; + if (!toys.optargs[0] || toys.optargs[0][0] == '+') { + if (toys.optargs[0]) format_string = toys.optargs[0]+1; + if (!strftime(toybuf, sizeof(toybuf), format_string, &tm)) goto bad_format; + + puts(toybuf); // Set the date - } else if (setdate) { - if (parse_posixdate(setdate, tm)) goto bad_date; + } else { + struct timeval tv; + char *s = *toys.optargs; + int len; + + // Date format: mmddhhmm[[cc]yy] + len = 0; + sscanf(s, "%2u%2u%2u%2u%n", &tm.tm_mon, &tm.tm_mday, &tm.tm_hour, + &tm.tm_min, &len); + if (len != 8) goto bad_date; + s += len; + tm.tm_mon--; + + // If year specified, overwrite one we fetched earlier + if (*s && *s != '.') { + unsigned year, r1 = tm.tm_year % 100, r2 = (tm.tm_year + 50) % 100, + century = tm.tm_year - r1; + + len = 0; + sscanf(s, "%u%n", &year, &len); + if (len == 4) year -= 1900; + else if (len != 2) goto bad_date; + s += len; + + // 2 digit years, next 50 years are "future", last 50 years are "past". + // A "future" date in past is a century ahead. + // A non-future date in the future is a century behind. + if ((r1 < r2) ? (r1 < year && year < r2) : (year < r1 || year > r2)) { + if (year < r1) year += 100; + } else if (year > r1) year -= 100; + tm.tm_year = year + century; + } + if (*s == '.') { + len = 0; + sscanf(s, ".%u%n", &tm.tm_sec, &len); + s += len; + } + if (*s) goto bad_date; if (toys.optflags & FLAG_u) { - // We can't just pass a timezone to mktime because posix. + // Get the UTC version of a struct tm char *tz = CFG_TOYBOX_FREE ? getenv("TZ") : 0; - setenv("TZ", "UTC", 1); tzset(); tv.tv_sec = mktime(&tm); @@ -148,18 +107,11 @@ void date_main(void) if (tv.tv_sec == (time_t)-1) goto bad_date; tv.tv_usec = 0; + if (!strftime(toybuf, sizeof(toybuf), format_string, &tm)) goto bad_format; + puts(toybuf); if (settimeofday(&tv, NULL) < 0) perror_msg("cannot set date"); } - if (toys.optflags & FLAG_u) { - if (tz) setenv("TZ", tz, 1); - else unsetenv("TZ"); - tzset(); - } - - if (!strftime(toybuf, sizeof(toybuf), format_string, &tm)) goto bad_format; - puts(toybuf); - return; bad_date: diff --git a/toys/posix/du.c b/toys/posix/du.c index 64ba0517..96922bca 100644 --- a/toys/posix/du.c +++ b/toys/posix/du.c @@ -54,8 +54,18 @@ static void print(long long size, struct dirtree *node) if (TT.maxdepth && TT.depth > TT.maxdepth) return; - if (toys.optflags & FLAG_h) printf("%s", human_readable(size)); - else { + if (toys.optflags & FLAG_h) { + char buf[32]; + int index, sz; + + for (index = 0; 1024 < size>>(10*index); index++); + sz = size>>(10*index); + if (sz < 10) { + sprintf(buf, "%llu", size>>(10*(index-1))); + printf("%c.%c", buf[0], buf[1]); + } else printf("%d", sz); + if (index) printf("%c", " KMGTPE"[index]); + } else { int bits = 10; if (toys.optflags & FLAG_K) bits = 9; diff --git a/toys/posix/grep.c b/toys/posix/grep.c index 2837cc24..aba70878 100644 --- a/toys/posix/grep.c +++ b/toys/posix/grep.c @@ -4,7 +4,7 @@ * * See http://pubs.opengroup.org/onlinepubs/9699919799/utilities/grep.html -USE_GREP(NEWTOY(grep, "A#B#C#ZzEFHabhinorsvwclqe*f*m#x[!wx][!EFw]", TOYFLAG_BIN)) +USE_GREP(NEWTOY(grep, "ZzEFHabhinorsvwclqe*f*m#x[!wx][!EFw]", TOYFLAG_BIN)) USE_GREP(OLDTOY(egrep, grep, OPTSTR_grep, TOYFLAG_BIN)) USE_GREP(OLDTOY(fgrep, grep, OPTSTR_grep, TOYFLAG_BIN)) @@ -46,67 +46,40 @@ GLOBALS( long m; struct arg_list *f; struct arg_list *e; - long C; - long B; - long A; struct arg_list *regex; - struct double_list *blist; ) -struct dlist_off { - char *next, *prev; - long offset; - char *data; -}; - static void do_grep(int fd, char *name) { FILE *file = fdopen(fd, "r"); long offset = 0; - int lcount = 0, mcount = 0, which = toys.optflags & FLAG_w ? 2 : 0, - blines = 0, alines = 0, dash = 0; + int lcount = 0, mcount = 0, which = toys.optflags & FLAG_w ? 2 : 0; char indelim = '\n' * !(toys.optflags&FLAG_z), outdelim = '\n' * !(toys.optflags&FLAG_Z); if (!fd) name = "(standard input)"; -fprintf(stderr, "boo\n"); if (!file) { perror_msg("%s", name); return; } - // Loop through lines of input for (;;) { - char *oline = 0, *line = 0, *start; + char *line = 0, *start; regmatch_t matches[3]; size_t unused; long len; int mmatch = 0; - // Read next line of input lcount++; if (0 > (len = getdelim(&line, &unused, indelim, file))) break; if (line[len-1] == indelim) line[len-1] = 0; -fprintf(stderr, "line=%s\n", line); - // Unconditionally add line to blist so output can always just dump blist. - dlist_add(&TT.blist, line); -fprintf(stderr, "added=%s\n", TT.blist->data); -fprintf(stderr, "prev=%s\n", TT.blist->prev->data); - if (blines <= TT.B) blines++; - else { - struct double_list *temp = dlist_pop(&TT.blist); -fprintf(stderr, "bird=%s\n", temp->data); - free(temp->data); - free(temp); - } start = line; - // Loop to match multiple times within the same line (if necessary) - for (;;) { -fprintf(stderr, "match?\n"); + for (;;) + { int rc = 0, skip = 0; if (toys.optflags & FLAG_F) { @@ -159,51 +132,30 @@ fprintf(stderr, "match?\n"); } matches[which].rm_so = 0; } else if (rc) break; -fprintf(stderr, "got match %s\n", line); - // We got a match, figure out how to display it + mmatch++; toys.exitval = 0; if (toys.optflags & FLAG_q) xexit(); if (toys.optflags & FLAG_l) { printf("%s%c", name, outdelim); - goto finish; + free(line); + fclose(file); + return; } - - line = 0; -fprintf(stderr, "here=%s\n", TT.blist->prev->data); - // Yes, -o sometimes counts things as a match (-c) but doesn't display it if (toys.optflags & FLAG_o) if (matches[which].rm_eo == matches[which].rm_so) break; -// List of lines that DIDN'T match, print backlog? -// Except this includes the one we just matched...? - while (TT.blist) { - struct double_list *dlist = dlist_pop(&TT.blist); - char *ll = dlist->data; -fprintf(stderr, "popped %s\n", ll); - if (dash) printf("--%c", outdelim); - dash = 0; - - if (!(toys.optflags & FLAG_c)) { - if (toys.optflags & FLAG_H) printf("%s:", name); - if (toys.optflags & FLAG_n) printf("%d:", lcount); - if (toys.optflags & FLAG_b) - printf("%ld:", offset + (start - dlist->data) + - ((toys.optflags & FLAG_o) ? matches[which].rm_so : 0)); - if (!(toys.optflags & FLAG_o)) xprintf("%s%c", dlist->data, outdelim); - else if (!TT.blist) { - -// TODO: FLAG_o prints multiple times, can't free it yet? - xprintf("%.*s%c", matches[which].rm_eo - matches[which].rm_so, - start + matches[which].rm_so, outdelim); - line = dlist->data; - } - } - if (oline && !TT.blist) TT.blist = dlist; + if (!(toys.optflags & FLAG_c)) { + if (toys.optflags & FLAG_H) printf("%s:", name); + if (toys.optflags & FLAG_n) printf("%d:", lcount); + if (toys.optflags & FLAG_b) + printf("%ld:", offset + (start-line) + + ((toys.optflags & FLAG_o) ? matches[which].rm_so : 0)); + if (!(toys.optflags & FLAG_o)) xprintf("%s%c", line, outdelim); else { - free(dlist->data); - free(dlist); + xprintf("%.*s%c", matches[which].rm_eo - matches[which].rm_so, + start + matches[which].rm_so, outdelim); } } @@ -211,7 +163,7 @@ fprintf(stderr, "popped %s\n", ll); if (!(toys.optflags & FLAG_o) || !*start) break; } offset += len; -fprintf(stderr, "Spacious skies\n"); + free(line); if (mmatch) mcount++; @@ -223,14 +175,6 @@ fprintf(stderr, "Spacious skies\n"); xprintf("%d%c", mcount, outdelim); } -finish: - while (CFG_TOYBOX_FREE && TT.blist) { - struct double_list *dlist = dlist_pop(&TT.blist); - - free(dlist->data); - free(dlist); - } - // loopfiles will also close the fd, but this frees an (opaque) struct. fclose(file); } @@ -329,10 +273,6 @@ void grep_main(void) toys.optc--; } - if (!TT.A) TT.A = TT.C; - if (!TT.B) TT.B = TT.C; - if (toys.optflags & (FLAG_l|FLAG_o)) TT.B = 0; // avoid memory leak - parse_regex(); if (!(toys.optflags & FLAG_h) && toys.optc>1) toys.optflags |= FLAG_H; |