busybox/coreutils/basename.c
<<
>>
Prefs
   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
  10/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
  11 *
  12 * Changes:
  13 * 1) Now checks for too many args.  Need at least one and at most two.
  14 * 2) Don't check for options, as per SUSv3.
  15 * 3) Save some space by using strcmp().  Calling strncmp() here was silly.
  16 */
  17
  18/* BB_AUDIT SUSv3 compliant */
  19/* http://www.opengroup.org/onlinepubs/007904975/utilities/basename.html */
  20
  21//kbuild:lib-$(CONFIG_BASENAME) += basename.o
  22
  23//config:config BASENAME
  24//config:       bool "basename"
  25//config:       default y
  26//config:       help
  27//config:         basename is used to strip the directory and suffix from filenames,
  28//config:         leaving just the filename itself. Enable this option if you wish
  29//config:         to enable the 'basename' utility.
  30
  31#include "libbb.h"
  32
  33/* This is a NOFORK applet. Be very careful! */
  34
  35int basename_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  36int basename_main(int argc, char **argv)
  37{
  38        size_t m, n;
  39        char *s;
  40
  41        if ((unsigned)(argc-2) >= 2) {
  42                bb_show_usage();
  43        }
  44
  45        /* It should strip slash: /abc/def/ -> def */
  46        s = bb_get_last_path_component_strip(*++argv);
  47
  48        m = strlen(s);
  49        if (*++argv) {
  50                n = strlen(*argv);
  51                if ((m > n) && (strcmp(s+m-n, *argv) == 0)) {
  52                        m -= n;
  53                        /*s[m] = '\0'; - redundant */
  54                }
  55        }
  56
  57        /* puts(s) will do, but we can do without stdio this way: */
  58        s[m++] = '\n';
  59        /* NB: != is correct here: */
  60        return full_write(STDOUT_FILENO, s, m) != (ssize_t)m;
  61}
  62