mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
Replace malloc with mrb_malloc in mrbc
This commit is contained in:
+5
-5
@@ -47,12 +47,12 @@ usage(const char *name)
|
||||
}
|
||||
|
||||
static char *
|
||||
get_outfilename(char *infile, char *ext)
|
||||
get_outfilename(mrb_state *mrb, char *infile, char *ext)
|
||||
{
|
||||
char *outfile;
|
||||
char *p;
|
||||
|
||||
outfile = (char*)malloc(strlen(infile) + strlen(ext) + 1);
|
||||
outfile = (char*)mrb_malloc(mrb, strlen(infile) + strlen(ext) + 1);
|
||||
strcpy(outfile, infile);
|
||||
if (*ext) {
|
||||
if ((p = strrchr(outfile, '.')) == NULL)
|
||||
@@ -91,7 +91,7 @@ parse_args(mrb_state *mrb, int argc, char **argv, struct _args *args)
|
||||
result = EXIT_FAILURE;
|
||||
goto exit;
|
||||
}
|
||||
outfile = get_outfilename((*argv) + 2, "");
|
||||
outfile = get_outfilename(mrb, (*argv) + 2, "");
|
||||
break;
|
||||
case 'B':
|
||||
args->ext = C_EXT;
|
||||
@@ -150,7 +150,7 @@ parse_args(mrb_state *mrb, int argc, char **argv, struct _args *args)
|
||||
outfile = infile;
|
||||
}
|
||||
else {
|
||||
outfile = get_outfilename(infile, args->ext);
|
||||
outfile = get_outfilename(mrb, infile, args->ext);
|
||||
}
|
||||
}
|
||||
if (strcmp("-", outfile) == 0) {
|
||||
@@ -163,7 +163,7 @@ parse_args(mrb_state *mrb, int argc, char **argv, struct _args *args)
|
||||
}
|
||||
}
|
||||
exit:
|
||||
if (outfile && infile != outfile) free(outfile);
|
||||
if (outfile && infile != outfile) mrb_free(mrb, outfile);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user