1 /* vi: set sw=4 ts=4: */
3 * split - split a file into pieces
4 * Copyright (c) 2007 Bernhard Reutner-Fischer
6 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
8 /* BB_AUDIT: SUSv3 compliant
10 * http://www.opengroup.org/onlinepubs/009695399/utilities/split.html
13 //usage:#define split_trivial_usage
14 //usage: "[OPTIONS] [INPUT [PREFIX]]"
15 //usage:#define split_full_usage "\n\n"
16 //usage: " -b N[k|m] Split by N (kilo|mega)bytes"
17 //usage: "\n -l N Split by N lines"
18 //usage: "\n -a N Use N letters as suffix"
20 //usage:#define split_example_usage
21 //usage: "$ split TODO foo\n"
22 //usage: "$ cat TODO | split -a 2 -l 2 TODO_\n"
26 static const struct suffix_mult split_suffices
[] = {
27 #if ENABLE_FEATURE_SPLIT_FANCY
32 #if ENABLE_FEATURE_SPLIT_FANCY
33 { "g", 1024*1024*1024 },
38 /* Increment the suffix part of the filename.
39 * Returns NULL if we are out of filenames.
41 static char *next_file(char *old
, unsigned suffix_len
)
43 size_t end
= strlen(old
);
63 #define read_buffer bb_common_bufsiz1
64 enum { READ_BUFFER_SIZE
= COMMON_BUFSIZE
- 1 };
66 #define SPLIT_OPT_l (1<<0)
67 #define SPLIT_OPT_b (1<<1)
68 #define SPLIT_OPT_a (1<<2)
70 int split_main(int argc
, char **argv
) MAIN_EXTERNALLY_VISIBLE
;
71 int split_main(int argc UNUSED_PARAM
, char **argv
)
73 unsigned suffix_len
= 2;
80 ssize_t bytes_read
, to_write
;
83 opt_complementary
= "?2:a+"; /* max 2 args; -a N */
84 opt
= getopt32(argv
, "l:b:a:", &count_p
, &count_p
, &suffix_len
);
86 if (opt
& SPLIT_OPT_l
)
87 cnt
= XATOOFF(count_p
);
88 if (opt
& SPLIT_OPT_b
) // FIXME: also needs XATOOFF
89 cnt
= xatoull_sfx(count_p
, split_suffices
);
97 fd
= xopen_stdin(argv
[0]);
98 xmove_fd(fd
, STDIN_FILENO
);
100 argv
[0] = (char *) bb_msg_standard_input
;
103 if (NAME_MAX
< strlen(sfx
) + suffix_len
)
104 bb_error_msg_and_die("suffix too long");
107 char *char_p
= xzalloc(suffix_len
+ 1);
108 memset(char_p
, 'a', suffix_len
);
109 pfx
= xasprintf("%s%s", sfx
, char_p
);
110 if (ENABLE_FEATURE_CLEAN_UP
)
115 bytes_read
= safe_read(STDIN_FILENO
, read_buffer
, READ_BUFFER_SIZE
);
119 bb_simple_perror_msg_and_die(argv
[0]);
124 bb_error_msg_and_die("suffixes exhausted");
125 xmove_fd(xopen(pfx
, O_WRONLY
| O_CREAT
| O_TRUNC
), 1);
126 pfx
= next_file(pfx
, suffix_len
);
130 if (opt
& SPLIT_OPT_b
) {
132 to_write
= (bytes_read
< remaining
) ? bytes_read
: remaining
;
133 remaining
-= to_write
;
136 /* can be sped up by using _memrchr_
137 * and writing many lines at once... */
138 char *end
= memchr(src
, '\n', bytes_read
);
141 to_write
= end
- src
+ 1;
143 to_write
= bytes_read
;
147 xwrite(STDOUT_FILENO
, src
, to_write
);
148 bytes_read
-= to_write
;
150 } while (bytes_read
);