1 /* vi: set sw=4 ts=4: */
3 * shuf: Write a random permutation of the input lines to standard output.
5 * Copyright (C) 2014 by Bartosz Golaszewski <bartekgola@gmail.com>
7 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
14 //config: Generate random permutations
16 //kbuild:lib-$(CONFIG_SHUF) += shuf.o
17 //applet:IF_SHUF(APPLET_NOEXEC(shuf, shuf, BB_DIR_USR_BIN, BB_SUID_DROP, shuf))
19 //usage:#define shuf_trivial_usage
20 //usage: "[-e|-i L-H] [-n NUM] [-o FILE] [-z] [FILE|ARG...]"
21 //usage:#define shuf_full_usage "\n\n"
22 //usage: "Randomly permute lines\n"
23 //usage: "\n -e Treat ARGs as lines"
24 //usage: "\n -i L-H Treat numbers L-H as lines"
25 //usage: "\n -n NUM Output at most NUM lines"
26 //usage: "\n -o FILE Write to FILE, not standard output"
27 //usage: "\n -z End lines with zero byte, not newline"
31 /* This is a NOEXEC applet. Be very careful! */
33 #define OPT_e (1 << 0)
34 #define OPT_i (1 << 1)
35 #define OPT_n (1 << 2)
36 #define OPT_o (1 << 3)
37 #define OPT_z (1 << 4)
38 #define OPT_STR "ei:n:o:z"
41 * Use the Fisher-Yates shuffle algorithm on an array of lines.
43 static void shuffle_lines(char **lines
, unsigned numlines
)
49 srand(monotonic_us());
51 for (i
= numlines
-1; i
> 0; i
--) {
53 /* RAND_MAX can be as small as 32767 */
63 int shuf_main(int argc
, char **argv
) MAIN_EXTERNALLY_VISIBLE
;
64 int shuf_main(int argc
, char **argv
)
67 char *opt_i_str
, *opt_n_str
, *opt_o_str
;
73 opt_complementary
= "e--i:i--e"; /* mutually exclusive */
74 opts
= getopt32(argv
, OPT_STR
, &opt_i_str
, &opt_n_str
, &opt_o_str
);
79 /* Prepare lines for shuffling - either: */
81 /* make lines from command-line arguments */
86 /* create a range of numbers */
90 dash
= strchr(opt_i_str
, '-');
92 bb_error_msg_and_die("bad range '%s'", opt_i_str
);
95 lo
= xatou(opt_i_str
);
99 bb_error_msg_and_die("bad range '%s'", opt_i_str
);
102 numlines
= (hi
+1) - lo
;
103 lines
= xmalloc(numlines
* sizeof(lines
[0]));
104 for (i
= 0; i
< numlines
; i
++) {
105 lines
[i
] = (char*)(uintptr_t)lo
;
109 /* default - read lines from stdin or the input file */
115 fp
= xfopen_stdin(argv
[0] ? argv
[0] : "-");
119 char *line
= xmalloc_fgetline(fp
);
122 lines
= xrealloc_vector(lines
, 6, numlines
);
123 lines
[numlines
++] = line
;
125 fclose_if_not_stdin(fp
);
129 shuffle_lines(lines
, numlines
);
132 xmove_fd(xopen(opt_o_str
, O_WRONLY
|O_CREAT
|O_TRUNC
), STDOUT_FILENO
);
136 maxlines
= xatou(opt_n_str
);
137 if (numlines
> maxlines
)
145 for (i
= 0; i
< numlines
; i
++) {
147 printf("%u%c", (unsigned)(uintptr_t)lines
[i
], eol
);
149 printf("%s%c", lines
[i
], eol
);
152 fflush_stdout_and_exit(EXIT_SUCCESS
);