Merge pull request #6020 from Icinga/fix/console-home-env

Fix crash when running 'icinga2 console' without HOME environment variable
This commit is contained in:
Noah Hilverling 2018-01-29 16:37:24 +01:00 committed by GitHub
commit 9cb7c9d7f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 10 deletions

View File

@ -287,17 +287,22 @@ int ConsoleCommand::RunScriptConsole(ScriptFrame& scriptFrame, const String& add
int next_line = 1;
#ifdef HAVE_EDITLINE
String homeEnv = getenv("HOME");
String historyPath = homeEnv + "/.icinga2_history";
char *homeEnv = getenv("HOME");
String historyPath;
std::fstream historyfp;
historyfp.open(historyPath.CStr(), std::fstream::in);
String line;
while (std::getline(historyfp, line.GetData()))
add_history(line.CStr());
if (homeEnv) {
historyPath = String(homeEnv) + "/.icinga2_history";
historyfp.close();
historyfp.open(historyPath.CStr(), std::fstream::in);
String line;
while (std::getline(historyfp, line.GetData()))
add_history(line.CStr());
historyfp.close();
}
#endif /* HAVE_EDITLINE */
l_ScriptFrame = &scriptFrame;
@ -370,9 +375,11 @@ incomplete:
if (commandOnce.IsEmpty() && cline[0] != '\0') {
add_history(cline);
historyfp.open(historyPath.CStr(), std::fstream::out | std::fstream::app);
historyfp << cline << "\n";
historyfp.close();
if (!historyPath.IsEmpty()) {
historyfp.open(historyPath.CStr(), std::fstream::out | std::fstream::app);
historyfp << cline << "\n";
historyfp.close();
}
}
line = cline;