coreutils/basename.c

   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 (3.7 kb)"
  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;
  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 }