From 4cb6e07eff50cdac3f7749db23fbc6e2027d314f Mon Sep 17 00:00:00 2001 From: des Date: Wed, 6 Sep 2000 07:28:02 +0000 Subject: Reintroduce functionality and error checks that were boneheadedly removed in the previous commit. Pass me the pointy hat. PR: bin/21061 --- usr.bin/basename/basename.c | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'usr.bin/basename') diff --git a/usr.bin/basename/basename.c b/usr.bin/basename/basename.c index d807ed9..e22e5a6 100644 --- a/usr.bin/basename/basename.c +++ b/usr.bin/basename/basename.c @@ -43,8 +43,10 @@ static const char copyright[] = static const char sccsid[] = "@(#)basename.c 8.4 (Berkeley) 5/4/95"; #endif /* not lint */ +#include #include #include +#include #include void usage __P((void)); @@ -54,6 +56,7 @@ main(argc, argv) int argc; char **argv; { + char *p, *q; int ch; while ((ch = getopt(argc, argv, "")) != -1) @@ -68,7 +71,11 @@ main(argc, argv) if (argc != 1 && argc != 2) usage(); - (void)printf("%s\n", basename(*argv)); + if ((p = basename(argv[0])) == NULL) + err(1, "%s", argv[0]); + if (*++argv && (q = strstr(p, *argv)) && strcmp(q, *argv) == 0) + *q = '\0'; + (void)printf("%s\n", p); exit(0); } -- cgit v1.1