gh0s7/mckernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

atobytes(): restore postfix before return

Browse Source
This commit is contained in:
Balazs Gerofi
2017-10-14 09:17:09 +09:00
parent 60ac94cbb9
commit 4b652c9353
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
executer/user/mcexec.c
View File

@ -1653,6 +1653,8 @@ static void destroy_local_environ(char **local_env)
unsigned long atobytes(char *string)
{
unsigned long mult = 1;
unsigned long ret;
char orig_postfix = 0;
char *postfix;
errno = ERANGE;
@ -1664,19 +1666,26 @@ unsigned long atobytes(char *string)
if (*postfix == 'k' || *postfix == 'K') {
mult = 1024;
orig_postfix = *postfix;
*postfix = 0;
}
else if (*postfix == 'm' || *postfix == 'M') {
mult = 1024 * 1024;
orig_postfix = *postfix;
*postfix = 0;
}
else if (*postfix == 'g' || *postfix == 'G') {
mult = 1024 * 1024 * 1024;
orig_postfix = *postfix;
*postfix = 0;
}
ret = atol(string) * mult;
if (orig_postfix)
*postfix = orig_postfix;
errno = 0;
return atol(string) * mult;
return ret;
}
static struct option mcexec_options[] = {