void luadebug_interactive_user(struct luadebug_user *user) { mutex_lock(¤t_user_mutex); luadebug_user_release(¤t_user); if (user) { current_user = user; luadebug_user_addref(user); } mutex_unlock(¤t_user_mutex); }
static int run_console(int fd, int argc, char *argv[]) { struct lua_state *state; struct luadebug_user *user; state = lua_state_init(); if (!state) { messagef(HAKA_LOG_FATAL, "hakactl", clear_error()); return COMMAND_FAILED; } console_fd = fd; lua_pushcfunction(state->L, luaopen_hakactl); lua_call(state->L, 0, 1); lua_setglobal(state->L, "hakactl"); if (!initialize_console(state)) { messagef(HAKA_LOG_FATAL, "hakactl", clear_error()); lua_state_close(state); return COMMAND_FAILED; } user = luadebug_user_readline(); if (!user) { messagef(HAKA_LOG_FATAL, "hakactl", clear_error()); lua_state_close(state); return COMMAND_FAILED; } /* Load Lua console utilities */ luadebug_interactive_enter(state->L, "> ", ">> ", NULL, -1, user); luadebug_user_release(&user); lua_state_close(state); return COMMAND_SUCCESS; }
void luadebug_user_remote_server(int fd, struct luadebug_user *user) { char command; while ((read(fd, &command, 1) == 1)) { switch (command) { case 's': { char *line = read_string(fd); if (!line) { messagef(HAKA_LOG_ERROR, MODULE, L"remote communication error: %s", errno_error(errno)); return; } user->start(user, line); free(line); if (!luadebug_user_remote_server_session(fd, user)) { message(HAKA_LOG_ERROR, MODULE, L"remote communication error"); return; } break; } case 'P': { break; } default: message(HAKA_LOG_ERROR, MODULE, L"remote communication error"); return; } } luadebug_user_release(&user); }
void luadebug_interactive_enter(struct lua_State *L, const char *single, const char *multi, const char *msg) { char *line, *full_line = NULL; bool multiline = false; bool stop = false; struct luadebug_interactive session; { mutex_lock(¤t_user_mutex); session.user = current_user; if (!session.user) { message(HAKA_LOG_ERROR, L"interactive", L"no input/output handler"); mutex_unlock(¤t_user_mutex); return; } luadebug_user_addref(session.user); mutex_unlock(¤t_user_mutex); } mutex_lock(&active_session_mutex); if (!single) single = "> "; if (!multi) multi = ">> "; session.user->completion = completion; if (!session.user->start(session.user, "haka")) { on_user_error(); luadebug_user_release(&session.user); mutex_unlock(&active_session_mutex); return; } current_session = &session; LUA_STACK_MARK(L); session.user->print(session.user, GREEN "entering interactive session" CLEAR ": %s\n", msg); session.L = L; session.env_index = capture_env(L, 1); assert(session.env_index != -1); session.complete.stack_env = session.env_index; while (!stop && (line = session.user->readline(session.user, multiline ? multi : single))) { int status = -1; bool is_return = false; char *current_line; if (multiline) { const size_t full_line_size = strlen(full_line); assert(full_line); full_line = realloc(full_line, full_line_size + 1 + strlen(line) + 1); full_line[full_line_size] = ' '; strcpy(full_line + full_line_size + 1, line); current_line = full_line; } else { session.user->addhistory(session.user, line); current_line = line; } while (*current_line == ' ' || *current_line == '\t') { ++current_line; } is_return = (strncmp(current_line, "return ", 7) == 0) || (strcmp(current_line, "return") == 0); if (!multiline && !is_return) { char *return_line = malloc(7 + strlen(current_line) + 1); strcpy(return_line, "return "); strcpy(return_line+7, current_line); status = luaL_loadbuffer(L, return_line, strlen(return_line), "stdin"); if (status) { lua_pop(L, 1); status = -1; } free(return_line); } if (status == -1) { status = luaL_loadbuffer(L, current_line, strlen(current_line), "stdin"); } multiline = false; /* Change the chunk environment */ lua_pushvalue(L, session.env_index); lua_setfenv(L, -2); if (status == LUA_ERRSYNTAX) { const char *message; const int k = sizeof(EOF_MARKER) / sizeof(char) - 1; size_t n; message = lua_tolstring(L, -1, &n); if (n > k && !strncmp(message + n - k, EOF_MARKER, k) && strlen(line) > 0) { multiline = true; } else { session.user->print(session.user, RED "%s" CLEAR "\n", lua_tostring(L, -1)); } lua_pop(L, 1); } else if (status == LUA_ERRMEM) { session.user->print(session.user, RED "%s" CLEAR "\n", lua_tostring(L, -1)); lua_pop(L, 1); } else { execute_print(L, session.user); lua_pop(L, 1); if (is_return) { stop = true; } } if (!multiline) { free(full_line); full_line = NULL; } else if (!full_line) { full_line = strdup(line); } free(line); } free(full_line); session.user->print(session.user, "\n"); session.user->print(session.user, GREEN "continue" CLEAR "\n"); lua_pop(L, 1); LUA_STACK_CHECK(L, 0); if (!session.user->stop(session.user)) { on_user_error(); } luadebug_user_release(&session.user); current_session = NULL; mutex_unlock(&active_session_mutex); }