Fixed a bug
This commit is contained in:
parent
35e42ec851
commit
f457f2ab9e
|
@ -7,18 +7,22 @@
|
|||
#include <string.h>
|
||||
|
||||
|
||||
/* usage */
|
||||
static int usage(void)
|
||||
{
|
||||
fprintf(stderr, "%s", "Usage: basename string [suffix]\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
||||
/* main */
|
||||
int main(int argc, char * argv[])
|
||||
{
|
||||
char * str;
|
||||
|
||||
/* parse command line arguments */
|
||||
/* check for errors */
|
||||
if(argc != 2 && argc != 3)
|
||||
{
|
||||
fprintf(stderr, "%s", "Usage: basename string [suffix]\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
return usage();
|
||||
/* basename */
|
||||
str = basename(argv[1]);
|
||||
if(argc == 3)
|
||||
|
@ -26,12 +30,11 @@ int main(int argc, char * argv[])
|
|||
int slen;
|
||||
int alen;
|
||||
|
||||
slen = strlen(argv[1]);
|
||||
slen = strlen(str);
|
||||
alen = strlen(argv[2]);
|
||||
if(alen < slen && strcmp(argv[2], &str[slen - alen]) == 0)
|
||||
str[slen - alen] = '\0';
|
||||
}
|
||||
printf("%s\n", str);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user