| @@ -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; |
| + | } |