1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini basename implementation for busybox
4  *
5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6  *
7  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
8  */
9 /* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
10  *
11  * Changes:
12  * 1) Now checks for too many args.  Need at least one and at most two.
13  * 2) Don't check for options, as per SUSv3.
14  * 3) Save some space by using strcmp().  Calling strncmp() here was silly.
15  */
16 //config:config BASENAME
17 //config:	bool "basename (438 bytes)"
18 //config:	default y
19 //config:	help
20 //config:	basename is used to strip the directory and suffix from filenames,
21 //config:	leaving just the filename itself. Enable this option if you wish
22 //config:	to enable the 'basename' utility.
23 
24 //applet:IF_BASENAME(APPLET_NOFORK(basename, basename, BB_DIR_USR_BIN, BB_SUID_DROP, basename))
25 
26 //kbuild:lib-$(CONFIG_BASENAME) += basename.o
27 
28 /* BB_AUDIT SUSv3 compliant */
29 /* http://www.opengroup.org/onlinepubs/007904975/utilities/basename.html */
30 
31 //usage:#define basename_trivial_usage
32 //usage:       "FILE [SUFFIX] | -a FILE... | -s SUFFIX FILE..."
33 //usage:#define basename_full_usage "\n\n"
34 //usage:       "Strip directory path and SUFFIX from FILE\n"
35 //usage:     "\n	-a		All arguments are FILEs"
36 //usage:     "\n	-s SUFFIX	Remove SUFFIX (implies -a)"
37 //usage:
38 //usage:#define basename_example_usage
39 //usage:       "$ basename /usr/local/bin/foo\n"
40 //usage:       "foo\n"
41 //usage:       "$ basename /usr/local/bin/\n"
42 //usage:       "bin\n"
43 //usage:       "$ basename /foo/bar.txt .txt\n"
44 //usage:       "bar"
45 
46 #include "libbb.h"
47 
48 /* This is a NOFORK applet. Be very careful! */
49 
50 int basename_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
basename_main(int argc UNUSED_PARAM,char ** argv)51 int basename_main(int argc UNUSED_PARAM, char **argv)
52 {
53 	unsigned opts;
54 	const char *suffix = NULL;
55 
56 	/* '+': stop at first non-option */
57 	opts = getopt32(argv, "^+" "as:"
58 		"\0" "-1" /* At least one argument */
59 		, &suffix
60 	);
61 	argv += optind;
62 
63 	do {
64 		char *s;
65 		size_t m;
66 
67 		/* It should strip slash: /abc/def/ -> def */
68 		s = bb_get_last_path_component_strip(*argv++);
69 		m = strlen(s);
70 		if (!opts) {
71 			if (*argv) {
72 				suffix = *argv;
73 				if (argv[1])
74 					bb_show_usage();
75 			}
76 		}
77 		if (suffix) {
78 			size_t n = strlen(suffix);
79 			if ((m > n) && (strcmp(s + m - n, suffix) == 0)) {
80 				m -= n;
81 				/*s[m] = '\0'; - redundant */
82 			}
83 		}
84 		/* puts(s) will do, but we can do without stdio this way: */
85 		s[m++] = '\n';
86 		/* NB: != is correct here: */
87 		if (full_write(STDOUT_FILENO, s, m) != (ssize_t)m)
88 			return EXIT_FAILURE;
89 	} while (opts && *argv);
90 
91 	return EXIT_SUCCESS;
92 }
93