d/freebsd-skq
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure that suffix matches occur on character boundaries.

Browse Source
This commit is contained in:
tjr 2004-07-15 06:15:10 +00:00
parent 51777f0d81
commit fdd18f5e83
1 changed files with 32 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
usr.bin/basename/basename.c
View File

@ -48,20 +48,26 @@ __FBSDID("$FreeBSD$");
#include <err.h>
#include <libgen.h>
#include <limits.h>
#include <locale.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <wchar.h>
void stripsuffix(char *, const char *, size_t);
void usage(void);
int
main(int argc, char **argv)
{
char *p, *q, *suffix;
char *p, *suffix;
size_t suffixlen;
int aflag, ch;
setlocale(LC_ALL, "");
aflag = 0;
suffix = NULL;
suffixlen = 0;
@ -99,15 +105,37 @@ main(int argc, char **argv)
while (argc--) {
if ((p = basename(*argv)) == NULL)
err(1, "%s", argv[0]);
if (suffixlen && (q = strchr(p, '\0') - suffixlen) > p &&
strcmp(suffix, q) == 0)
*q = '\0';
stripsuffix(p, suffix, suffixlen);
argv++;
(void)printf("%s\n", p);
}
exit(0);
}
void
stripsuffix(char *p, const char *suffix, size_t suffixlen)
{
char *q, *r;
mbstate_t mbs;
size_t n;
if (suffixlen && (q = strchr(p, '\0') - suffixlen) > p &&
strcmp(suffix, q) == 0) {
/* Ensure that the match occurred on a character boundary. */
memset(&mbs, 0, sizeof(mbs));
for (r = p; r < q; r += n) {
n = mbrlen(r, MB_LEN_MAX, &mbs);
if (n == (size_t)-1 || n == (size_t)-2) {
memset(&mbs, 0, sizeof(mbs));
n = 1;
}
}
/* Chop off the suffix. */
if (q == r)
*q = '\0';
}
}
void
usage(void)
{