Commit 9d597f4d authored by cremno's avatar cremno

allocate `history_path` dynamically

`PATH_MAX` doesn't have to be defined and if it is, then its value is
usually incorrect (possible buffer overflow).
Use `malloc` and `snprintf` instead to prevent this.

https://www.gnu.org/software/hurd/community/gsoc/project_ideas/maxpath.html
parent ec45dbe1
......@@ -29,18 +29,48 @@
#define MIRB_USING_HISTORY()
#endif
#ifdef ENABLE_READLINE
#include <limits.h>
static const char *history_file_name = ".mirb_history";
char history_path[PATH_MAX];
#endif
#include "mruby.h"
#include "mruby/array.h"
#include "mruby/proc.h"
#include "mruby/compile.h"
#include "mruby/string.h"
#ifdef ENABLE_READLINE
static const char history_file_name[] = ".mirb_history";
static char *
get_history_path(void)
{
char *path = NULL;
const char *home = getenv("HOME");
#ifdef _WIN32
if (home != NULL) {
home = getenv("USERPROFILE");
}
#endif
if (home != NULL) {
int len = snprintf(NULL, 0, "%s/%s", home, history_file_name);
if (len >= 0) {
size_t size = len + 1;
path = (char *)malloc(size);
if (path != NULL) {
int n = snprintf(path, size, "%s/%s", home, history_file_name);
if (n != len) {
free(path);
path = NULL;
}
}
}
}
return path;
}
#endif
static void
p(mrb_state *mrb, mrb_value obj, int prompt)
{
......@@ -283,7 +313,7 @@ main(int argc, char **argv)
int last_char;
int char_index;
#else
char *home = NULL;
char *history_path;
#endif
mrbc_context *cxt;
struct mrb_parser_state *parser;
......@@ -295,6 +325,17 @@ main(int argc, char **argv)
int ai;
unsigned int stack_keep = 0;
#ifdef ENABLE_READLINE
history_path = get_history_path();
if (history_path == NULL) {
fputs("failed to get history path\n", stderr);
return EXIT_FAILURE;
}
MIRB_USING_HISTORY();
MIRB_READ_HISTORY(history_path);
#endif
/* new interpreter instance */
mrb = mrb_open();
if (mrb == NULL) {
......@@ -320,22 +361,6 @@ main(int argc, char **argv)
ai = mrb_gc_arena_save(mrb);
#ifdef ENABLE_READLINE
MIRB_USING_HISTORY();
home = getenv("HOME");
#ifdef _WIN32
if (!home)
home = getenv("USERPROFILE");
#endif
if (home) {
strcpy(history_path, home);
strcat(history_path, "/");
strcat(history_path, history_file_name);
MIRB_READ_HISTORY(history_path);
}
#endif
while (TRUE) {
#ifndef ENABLE_READLINE
print_cmdline(code_block_open);
......@@ -441,6 +466,7 @@ main(int argc, char **argv)
#ifdef ENABLE_READLINE
MIRB_WRITE_HISTORY(history_path);
free(history_path);
#endif
return 0;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment