util/sbase/basename.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

/* See LICENSE file for copyright and license details. */
#include <libgen.h>
#include <stdio.h>
#include <string.h>

#include "util.h"

static void
usage(void)
{
	eprintf("usage: %s path [suffix]\n", argv0);
}

int
main(int argc, char *argv[])
{
	ssize_t off;
	char *p;

	ARGBEGIN {
	default:
		usage();
	} ARGEND

	if (argc != 1 && argc != 2)
		usage();

	p = basename(argv[0]);
	if (argc == 2) {
		off = strlen(p) - strlen(argv[1]);
		if (off > 0 && !strcmp(p + off, argv[1]))
			p[off] = '\0';
	}
	puts(p);

	return fshut(stdout, "<stdout>");
}