diff options
Diffstat (limited to 'coreutils/split.c')
-rw-r--r-- | coreutils/split.c | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/coreutils/split.c b/coreutils/split.c new file mode 100644 index 0000000..312906f --- /dev/null +++ b/coreutils/split.c @@ -0,0 +1,118 @@ +/* vi: set sw=4 ts=4: */ +/* + * split - split a file into pieces + * Copyright (c) 2007 Bernhard Fischer + * + * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. + */ +/* BB_AUDIT: not yet SUSV3 compliant; FIXME: add -bN{k,m} + * SUSv3 requirements: + * http://www.opengroup.org/onlinepubs/009695399/utilities/split.html + */ +#include "busybox.h" +static unsigned suffix_len = 2; + +/* Increment the suffix part of the filename. + * Returns 0 on success and 1 on error (if we are out of files) + */ +static bool next_file(char **old) +{ + size_t end = strlen(*old); + unsigned i = 1; + char *curr; + + do { +// if (**(old + end - i) < 'z') { + curr = *old + end - i; + if (*curr < 'z') { + *(*old + end - i) += 1; + break; + } + *(*old +end - i) = 'a'; + i++; + } while (i <= suffix_len); + if ((*curr == 'z') && (i == suffix_len)) + return 1; + return 0; +} +#define SPLIT_OPT_l (1<<0) +#define SPLIT_OPT_b (1<<1) + +int split_main(int argc, char **argv); +int split_main(int argc, char **argv) +{ + char *pfx; + char *count_p = NULL; + char *sfx_len = NULL; + unsigned cnt = 1000; + char *input_file; + +//XXX: FIXME opt_complementary = "+2"; /* at most 2 non-option arguments */ + getopt32(argc, argv, "l:b:a:", &count_p, &count_p, &sfx_len); + argv += optind; + + if (count_p) + cnt = xatoi(count_p); + if (sfx_len) + suffix_len = xatoul(sfx_len); + + if (!*argv) + *--argv = (char*) "-"; + input_file = *argv; + if (NAME_MAX < strlen(*argv) + suffix_len) + bb_error_msg_and_die("Suffix too long"); + + { + char *char_p = xzalloc(suffix_len); + memset(char_p, 'a', suffix_len); + pfx = xasprintf("%s%s", (argc > optind + 1) ? *++argv : "x", char_p); + if (ENABLE_FEATURE_CLEAN_UP) + free(char_p); + } +//XXX:FIXME: unify those two file-handling schemata below (FILE vs fd) ! + if (option_mask32 & SPLIT_OPT_b) { + char *buf; + ssize_t i; + ssize_t bytes = 0; + int inp = xopen(input_file, O_RDONLY); + int flags = O_WRONLY | O_CREAT | O_TRUNC; + do { + int out = xopen(pfx, flags); + buf = xzalloc(cnt); + lseek(inp, bytes, SEEK_SET); + bytes += i = full_read(inp, buf, cnt); + xwrite(out, buf, i); + close(out); + free(buf); + if (next_file(&pfx)) + flags = O_WRONLY | O_APPEND; + } while(i > 0); + } else { /* -l */ + FILE *fp = fopen_or_warn_stdin(input_file); + char *buf; + do { + unsigned i = cnt; + int flags = O_WRONLY | O_CREAT | O_TRUNC; + int out = xopen(pfx, flags); + buf = NULL; + while (i--) { + buf = xmalloc_fgets(fp); + if (buf == NULL) + break; + xwrite(out, buf, buf ? strlen(buf) : 0); + free(buf); + }; + close(out); + + if (next_file(&pfx)) + flags = O_WRONLY | O_APPEND; + } while (buf); + if (ENABLE_FEATURE_CLEAN_UP) + fclose_if_not_stdin(fp); + } + + if (ENABLE_FEATURE_CLEAN_UP) { + free(pfx); + } + return EXIT_SUCCESS; +} |