blob: 2aef2d95511f9df59369ee9170ca4eb76eec4e8c [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
John Beppuc0ca4731999-12-21 20:00:35 +00002/*
John Beppu568cb7b1999-12-22 23:02:12 +00003 * Mini sort implementation for busybox
John Beppuc0ca4731999-12-21 20:00:35 +00004 *
5 *
Matt Kraai5e8c0ff2000-12-20 20:49:56 +00006 * Copyright (C) 2000 by Matt Kraai <kraai@alumni.carnegiemellon.edu>
John Beppuc0ca4731999-12-21 20:00:35 +00007 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 *
22 */
23
Eric Andersen3570a342000-09-25 21:45:58 +000024#include "busybox.h"
Eric Andersened3ef502001-01-27 08:24:39 +000025#include <getopt.h>
26#include <stdlib.h>
John Beppuc0ca4731999-12-21 20:00:35 +000027
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000028int compare_ascii(const void *x, const void *y)
John Beppuc0ca4731999-12-21 20:00:35 +000029{
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000030 return strcmp(*(char **)x, *(char **)y);
John Beppu38efa791999-12-22 00:30:29 +000031}
32
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000033int compare_numeric(const void *x, const void *y)
John Beppu38efa791999-12-22 00:30:29 +000034{
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000035 return atoi(*(char **)x) - atoi(*(char **)y);
John Beppuc0ca4731999-12-21 20:00:35 +000036}
37
Erik Andersene49d5ec2000-02-08 19:58:47 +000038int sort_main(int argc, char **argv)
John Beppuc0ca4731999-12-21 20:00:35 +000039{
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000040 FILE *fp;
41 char *line, **lines = NULL;
42 int i, opt, nlines = 0;
43 int (*compare)(const void *, const void *) = compare_ascii;
44#ifdef BB_FEATURE_SORT_REVERSE
45 int reverse = FALSE;
46#endif
John Beppuc0ca4731999-12-21 20:00:35 +000047
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000048 while ((opt = getopt(argc, argv, "nr")) != -1) {
49 switch (opt) {
Erik Andersene49d5ec2000-02-08 19:58:47 +000050 case 'n':
Erik Andersene49d5ec2000-02-08 19:58:47 +000051 compare = compare_numeric;
52 break;
Erik Andersen029011b2000-03-04 21:19:32 +000053#ifdef BB_FEATURE_SORT_REVERSE
Erik Andersene49d5ec2000-02-08 19:58:47 +000054 case 'r':
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000055 reverse = TRUE;
Erik Andersene49d5ec2000-02-08 19:58:47 +000056 break;
Erik Andersen029011b2000-03-04 21:19:32 +000057#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +000058 default:
Erik Andersene49d5ec2000-02-08 19:58:47 +000059 usage(sort_usage);
Erik Andersene49d5ec2000-02-08 19:58:47 +000060 }
61 }
62
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000063 /* read the input */
64 for (i = optind; i == optind || i < argc; i++) {
65 if (argv[i] == NULL)
66 fp = stdin;
67 else
68 fp = xfopen(argv[i], "r");
John Beppu8d369e92000-09-28 17:49:59 +000069
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000070 while ((line = get_line_from_file(fp)) != NULL) {
71 lines = xrealloc(lines, sizeof(char *) * (nlines + 1));
72 lines[nlines++] = line;
Erik Andersene49d5ec2000-02-08 19:58:47 +000073 }
John Beppuc0ca4731999-12-21 20:00:35 +000074 }
John Beppuc0ca4731999-12-21 20:00:35 +000075
Matt Kraai5e8c0ff2000-12-20 20:49:56 +000076 /* sort it */
77 qsort(lines, nlines, sizeof(char *), compare);
78
79 /* print it */
80#ifdef BB_FEATURE_SORT_REVERSE
81 if (reverse)
82 for (i = nlines - 1; 0 <= i; i--)
83 fputs(lines[i], stdout);
84 else
85#endif
86 for (i = 0; i < nlines; i++)
87 fputs(lines[i], stdout);
88 return EXIT_SUCCESS;
John Beppuc0ca4731999-12-21 20:00:35 +000089}