Audit basename(1)
1) be stricter which number of arguments is accepted (1 or 2) 2) basename already returns a pointer to "." is argv[0] is "" 3) No need to check for *p != '/', because basename() only returns a string beginning with '/' which has length 1, so if strlen(p) == 1, the only way for suffix to be "evaluated" is for off to be > 0, being equal to suffix being "", but "" != "/". 4) don't calculate strlen twice for each string. Store it in a ssize_t and check if it's > 0.
This commit is contained in:
parent
44bd931834
commit
e7c33c4af3
2
README
2
README
|
@ -9,7 +9,7 @@ The following tools are implemented ('*' == finished, '#' == UTF-8 support,
|
||||||
|
|
||||||
UTILITY POSIX 2008 COMPLIANT MISSING OPTIONS
|
UTILITY POSIX 2008 COMPLIANT MISSING OPTIONS
|
||||||
------- -------------------- ---------------
|
------- -------------------- ---------------
|
||||||
=* basename yes none
|
=*| basename yes none
|
||||||
=* cal yes none
|
=* cal yes none
|
||||||
=* cat yes none
|
=* cat yes none
|
||||||
=* chgrp yes none
|
=* chgrp yes none
|
||||||
|
|
15
basename.c
15
basename.c
|
@ -5,8 +5,6 @@
|
||||||
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
static void usage(void);
|
|
||||||
|
|
||||||
void
|
void
|
||||||
usage(void)
|
usage(void)
|
||||||
{
|
{
|
||||||
|
@ -16,25 +14,24 @@ usage(void)
|
||||||
int
|
int
|
||||||
main(int argc, char *argv[])
|
main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
ssize_t off;
|
||||||
char *p;
|
char *p;
|
||||||
size_t off;
|
|
||||||
|
|
||||||
ARGBEGIN {
|
ARGBEGIN {
|
||||||
default:
|
default:
|
||||||
usage();
|
usage();
|
||||||
} ARGEND;
|
} ARGEND;
|
||||||
|
|
||||||
if (argc < 1)
|
if (argc != 1 && argc != 2)
|
||||||
usage();
|
usage();
|
||||||
|
|
||||||
p = strlen(argv[0]) ? basename(argv[0]) : ".";
|
p = basename(argv[0]);
|
||||||
if (argc == 2 && *p != '/') {
|
if (argc == 2) {
|
||||||
if (strlen(argv[1]) < strlen(p)) {
|
|
||||||
off = strlen(p) - strlen(argv[1]);
|
off = strlen(p) - strlen(argv[1]);
|
||||||
if (strcmp(&p[off], argv[1]) == 0)
|
if (off > 0 && !strcmp(p + off, argv[1]))
|
||||||
p[off] = '\0';
|
p[off] = '\0';
|
||||||
}
|
}
|
||||||
}
|
|
||||||
puts(p);
|
puts(p);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user