1 /*- 2 * Copyright (c) 1991, 1993, 1994 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of the University nor the names of its contributors 14 * may be used to endorse or promote products derived from this software 15 * without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 */ 29 30 #ifndef lint 31 static const char copyright[] = 32 "@(#) Copyright (c) 1991, 1993, 1994\n\ 33 The Regents of the University of California. All rights reserved.\n"; 34 #endif 35 36 #if 0 37 #ifndef lint 38 static char sccsid[] = "@(#)basename.c 8.4 (Berkeley) 5/4/95"; 39 #endif /* not lint */ 40 #endif 41 42 #include <sys/cdefs.h> 43 __FBSDID("$FreeBSD$"); 44 45 #include <capsicum_helpers.h> 46 #include <err.h> 47 #include <libgen.h> 48 #include <limits.h> 49 #include <locale.h> 50 #include <stdio.h> 51 #include <stdlib.h> 52 #include <string.h> 53 #include <unistd.h> 54 #include <wchar.h> 55 56 void stripsuffix(char *, const char *, size_t); 57 void usage(void); 58 59 int 60 main(int argc, char **argv) 61 { 62 char *p, *suffix; 63 size_t suffixlen; 64 int aflag, ch; 65 66 setlocale(LC_ALL, ""); 67 68 if (caph_limit_stdio() < 0 || (cap_enter() < 0 && errno != ENOSYS)) 69 err(1, "capsicum"); 70 71 aflag = 0; 72 suffix = NULL; 73 suffixlen = 0; 74 75 while ((ch = getopt(argc, argv, "as:")) != -1) 76 switch(ch) { 77 case 'a': 78 aflag = 1; 79 break; 80 case 's': 81 suffix = optarg; 82 break; 83 case '?': 84 default: 85 usage(); 86 } 87 argc -= optind; 88 argv += optind; 89 90 if (argc < 1) 91 usage(); 92 93 if (!*argv[0]) { 94 printf("\n"); 95 exit(0); 96 } 97 if ((p = basename(argv[0])) == NULL) 98 err(1, "%s", argv[0]); 99 if ((suffix == NULL && !aflag) && argc == 2) { 100 suffix = argv[1]; 101 argc--; 102 } 103 if (suffix != NULL) 104 suffixlen = strlen(suffix); 105 while (argc--) { 106 if ((p = basename(*argv)) == NULL) 107 err(1, "%s", argv[0]); 108 stripsuffix(p, suffix, suffixlen); 109 argv++; 110 (void)printf("%s\n", p); 111 } 112 exit(0); 113 } 114 115 void 116 stripsuffix(char *p, const char *suffix, size_t suffixlen) 117 { 118 char *q, *r; 119 mbstate_t mbs; 120 size_t n; 121 122 if (suffixlen && (q = strchr(p, '\0') - suffixlen) > p && 123 strcmp(suffix, q) == 0) { 124 /* Ensure that the match occurred on a character boundary. */ 125 memset(&mbs, 0, sizeof(mbs)); 126 for (r = p; r < q; r += n) { 127 n = mbrlen(r, MB_LEN_MAX, &mbs); 128 if (n == (size_t)-1 || n == (size_t)-2) { 129 memset(&mbs, 0, sizeof(mbs)); 130 n = 1; 131 } 132 } 133 /* Chop off the suffix. */ 134 if (q == r) 135 *q = '\0'; 136 } 137 } 138 139 void 140 usage(void) 141 { 142 143 (void)fprintf(stderr, 144 "usage: basename string [suffix]\n" 145 " basename [-a] [-s suffix] string [...]\n"); 146 exit(1); 147 } 148