xref: /dragonfly/usr.bin/basename/basename.c (revision 0a488908)
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. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * $FreeBSD: src/usr.bin/basename/basename.c,v 1.15 2004/07/15 06:15:10 tjr Exp $
34  * $DragonFly: src/usr.bin/basename/basename.c,v 1.5 2005/09/18 10:12:55 asmodai Exp $
35  *
36  * @(#) Copyright (c) 1991, 1993, 1994 The Regents of the University of California.  All rights reserved.
37  * @(#)basename.c	8.4 (Berkeley) 5/4/95
38  */
39 
40 #include <err.h>
41 #include <libgen.h>
42 #include <limits.h>
43 #include <locale.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48 #include <wchar.h>
49 
50 void stripsuffix(char *, const char *, size_t);
51 void usage(void);
52 
53 int
54 main(int argc, char **argv)
55 {
56 	char *p, *suffix;
57 	size_t suffixlen;
58 	int ch;
59 
60 	setlocale(LC_ALL, "");
61 
62 	suffix = 0;
63 	suffixlen = NULL;
64 
65 	while ((ch = getopt(argc, argv, "")) != -1)
66 		switch(ch) {
67 		case '?':
68 		default:
69 			usage();
70 		}
71 	argc -= optind;
72 	argv += optind;
73 
74 	if (argc != 1 && argc != 2)
75 		usage();
76 
77 	if (!*argv[0]) {
78 		printf("\n");
79 		exit(0);
80 	}
81 	if ((p = basename(argv[0])) == NULL)
82 		err(1, "%s", argv[0]);
83 	stripsuffix(p, suffix, suffixlen);
84 	(void)printf("%s\n", p);
85 	exit(0);
86 }
87 
88 void
89 stripsuffix(char *p, const char *suffix, size_t suffixlen)
90 {
91 	char *q, *r;
92 	mbstate_t mbs;
93 	size_t n;
94 
95 	if (suffixlen && (q = strchr(p, '\0') - suffixlen) > p &&
96 	    strcmp(suffix, q) == 0) {
97 		/* Ensure that the match occurred on a character boundary. */
98 		memset(&mbs, 0, sizeof(mbs));
99 		for (r = p; r < q; r += n) {
100 			n = mbrlen(r, MB_LEN_MAX, &mbs);
101 			if (n == (size_t)-1 || n == (size_t)-2) {
102 				memset(&mbs, 0, sizeof(mbs));
103 				n = 1;
104 			}
105 		}
106 		/* Chop off the suffix. */
107 		if (q == r)
108 			*q = '\0';
109 	}
110 }
111 
112 void
113 usage(void)
114 {
115 
116 	(void)fprintf(stderr, "usage: basename string [suffix]\n");
117 	exit(1);
118 }
119