sbase/basename.c
Brandon Mulcahy 757cf0651a Fix basename suffix treatment
Explicitly use "." instead of the result of basename(3) when argv[0] is
an empty string in order to avoid a segfault.

Skip suffix treatment if the result of basename(3) is "/", per POSIX.

Fix the suffix check, which was previously checking for a match at any
location in the string.
2014-12-04 08:00:03 +00:00

43 lines
605 B
C

/* See LICENSE file for copyright and license details. */
#include <libgen.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "util.h"
static void usage(void);
void
usage(void)
{
eprintf("usage: %s name [suffix]\n", argv0);
}
int
main(int argc, char *argv[])
{
char *s, *p;
size_t d;
ARGBEGIN {
default:
usage();
} ARGEND;
if (argc < 1)
usage();
s = strlen(argv[0]) ? basename(argv[0]) : ".";
if (argc == 2 && *s != '/') {
d = strlen(s) - strlen(argv[1]);
if (d >= 0) {
p = s + d;
if (strcmp(p, argv[1]) == 0)
*p = '\0';
}
}
puts(s);
return 0;
}