aboutsummaryrefslogtreecommitdiff
path: root/contrib/epee
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/epee')
-rw-r--r--contrib/epee/include/console_handler.h6
-rw-r--r--contrib/epee/include/net/abstract_tcp_server2.inl6
-rw-r--r--contrib/epee/include/readline_buffer.h13
-rw-r--r--contrib/epee/src/mlog.cpp6
-rw-r--r--contrib/epee/src/readline_buffer.cpp72
5 files changed, 81 insertions, 22 deletions
diff --git a/contrib/epee/include/console_handler.h b/contrib/epee/include/console_handler.h
index 6832f2ea1..e780ad4de 100644
--- a/contrib/epee/include/console_handler.h
+++ b/contrib/epee/include/console_handler.h
@@ -374,6 +374,9 @@ namespace epee
}
else
{
+#ifdef HAVE_READLINE
+ rdln::suspend_readline pause_readline;
+#endif
std::cout << "unknown command: " << command << std::endl;
std::cout << usage;
}
@@ -477,6 +480,9 @@ namespace epee
lookup::mapped_type & vt = m_command_handlers[cmd];
vt.first = hndlr;
vt.second = usage;
+#ifdef HAVE_READLINE
+ rdln::readline_buffer::add_completion(cmd);
+#endif
}
bool process_command_vec(const std::vector<std::string>& cmd)
diff --git a/contrib/epee/include/net/abstract_tcp_server2.inl b/contrib/epee/include/net/abstract_tcp_server2.inl
index 0fbd9ed28..61276e761 100644
--- a/contrib/epee/include/net/abstract_tcp_server2.inl
+++ b/contrib/epee/include/net/abstract_tcp_server2.inl
@@ -209,14 +209,14 @@ PRAGMA_WARNING_DISABLE_VS(4355)
bool connection<t_protocol_handler>::add_ref()
{
TRY_ENTRY();
- //_dbg3("[sock " << socket_.native_handle() << "] add_ref, m_peer_number=" << mI->m_peer_number);
- CRITICAL_REGION_LOCAL(m_self_refs_lock);
- //_dbg3("[sock " << socket_.native_handle() << "] add_ref 2, m_peer_number=" << mI->m_peer_number);
// Use safe_shared_from_this, because of this is public method and it can be called on the object being deleted
auto self = safe_shared_from_this();
if(!self)
return false;
+ //_dbg3("[sock " << socket_.native_handle() << "] add_ref, m_peer_number=" << mI->m_peer_number);
+ CRITICAL_REGION_LOCAL(self->m_self_refs_lock);
+ //_dbg3("[sock " << socket_.native_handle() << "] add_ref 2, m_peer_number=" << mI->m_peer_number);
if(m_was_shutdown)
return false;
m_self_refs.push_back(self);
diff --git a/contrib/epee/include/readline_buffer.h b/contrib/epee/include/readline_buffer.h
index 916d14f01..8dd082a70 100644
--- a/contrib/epee/include/readline_buffer.h
+++ b/contrib/epee/include/readline_buffer.h
@@ -3,6 +3,8 @@
#include <streambuf>
#include <sstream>
#include <iostream>
+#include <vector>
+#include <algorithm>
namespace rdln
{
@@ -19,12 +21,23 @@ namespace rdln
}
void get_line(std::string& line) const;
void set_prompt(const std::string& prompt);
+ static void add_completion(const std::string& command)
+ {
+ if(std::find(completion_commands.begin(), completion_commands.end(), command) != completion_commands.end())
+ return;
+ completion_commands.push_back(command);
+ }
+ static const std::vector<std::string>& get_completions()
+ {
+ return completion_commands;
+ }
protected:
virtual int sync();
private:
std::streambuf* m_cout_buf;
+ static std::vector<std::string> completion_commands;
};
class suspend_readline
diff --git a/contrib/epee/src/mlog.cpp b/contrib/epee/src/mlog.cpp
index 3be992374..a51654d36 100644
--- a/contrib/epee/src/mlog.cpp
+++ b/contrib/epee/src/mlog.cpp
@@ -91,10 +91,10 @@ static const char *get_default_categories(int level)
switch (level)
{
case 0:
- categories = "*:WARNING,net:FATAL,net.p2p:FATAL,net.cn:FATAL,global:INFO,verify:FATAL,stacktrace:INFO,logging:INFO";
+ categories = "*:WARNING,net:FATAL,net.p2p:FATAL,net.cn:FATAL,global:INFO,verify:FATAL,stacktrace:INFO,logging:INFO,msgwriter:INFO";
break;
case 1:
- categories = "*:WARNING,global:INFO,stacktrace:INFO,logging:INFO";
+ categories = "*:WARNING,global:INFO,stacktrace:INFO,logging:INFO,msgwriter:INFO";
break;
case 2:
categories = "*:DEBUG";
@@ -139,7 +139,7 @@ void mlog_configure(const std::string &filename_base, bool console)
{
monero_log = get_default_categories(0);
}
- mlog_set_categories(monero_log);
+ mlog_set_log(monero_log);
}
void mlog_set_categories(const char *categories)
diff --git a/contrib/epee/src/readline_buffer.cpp b/contrib/epee/src/readline_buffer.cpp
index c846641bd..549f49d45 100644
--- a/contrib/epee/src/readline_buffer.cpp
+++ b/contrib/epee/src/readline_buffer.cpp
@@ -6,6 +6,7 @@
#include <mutex>
#include <condition_variable>
#include <boost/thread.hpp>
+#include <boost/algorithm/string.hpp>
static int process_input();
static void install_line_handler();
@@ -16,6 +17,8 @@ static std::string last_prompt;
std::mutex line_mutex, sync_mutex, process_mutex;
std::condition_variable have_line;
+std::vector<std::string> rdln::readline_buffer::completion_commands = {"exit"};
+
namespace
{
rdln::readline_buffer* current = NULL;
@@ -59,6 +62,7 @@ void rdln::readline_buffer::start()
void rdln::readline_buffer::stop()
{
std::unique_lock<std::mutex> lock(process_mutex);
+ have_line.notify_all();
if(m_cout_buf == NULL)
return;
std::cout.rdbuf(m_cout_buf);
@@ -78,6 +82,7 @@ void rdln::readline_buffer::set_prompt(const std::string& prompt)
last_prompt = prompt;
if(m_cout_buf == NULL)
return;
+ std::lock_guard<std::mutex> lock(sync_mutex);
rl_set_prompt(last_prompt.c_str());
rl_redisplay();
}
@@ -112,8 +117,7 @@ int rdln::readline_buffer::sync()
do
{
- char x = this->sgetc();
- m_cout_buf->sputc(x);
+ m_cout_buf->sputc( this->sgetc() );
}
while ( this->snextc() != EOF );
@@ -148,43 +152,44 @@ static int process_input()
static void handle_line(char* line)
{
- if (line != NULL)
+ if(last_line == "exit" || last_line == "q")
{
- std::lock_guard<std::mutex> lock(sync_mutex);
- rl_set_prompt(last_prompt.c_str());
- rl_already_prompted = 1;
return;
}
- rl_set_prompt("");
- rl_replace_line("", 0);
- rl_redisplay();
+ std::lock_guard<std::mutex> lock(sync_mutex);
rl_set_prompt(last_prompt.c_str());
+ rl_already_prompted = 1;
}
static int handle_enter(int x, int y)
{
std::lock_guard<std::mutex> lock(sync_mutex);
char* line = NULL;
-
+
line = rl_copy_text(0, rl_end);
+ rl_crlf();
+ rl_on_new_line();
rl_set_prompt("");
rl_replace_line("", 1);
rl_redisplay();
- if (strcmp(line, "") != 0)
+ std::string test_line = line;
+ boost::trim_right(test_line);
+ if (test_line.length() > 0)
{
- last_line = line;
- add_history(line);
+ last_line = test_line;
+ add_history(test_line.c_str());
have_line.notify_one();
}
free(line);
-
- if(last_line != "exit")
+
+ if(last_line != "exit" && last_line != "q")
{
rl_set_prompt(last_prompt.c_str());
+ rl_on_new_line_with_prompt();
rl_redisplay();
}
-
+
rl_done = 1;
return 0;
}
@@ -196,15 +201,50 @@ static int startup_hook()
return 0;
}
+static char* completion_matches(const char* text, int state)
+{
+ static size_t list_index;
+ static size_t len;
+
+ if(state == 0)
+ {
+ list_index = 0;
+ len = strlen(text);
+ }
+
+ const std::vector<std::string>& completions = rdln::readline_buffer::get_completions();
+ for(; list_index<completions.size(); )
+ {
+ const std::string& cmd = completions[list_index++];
+ if(cmd.compare(0, len, text) == 0)
+ {
+ return strdup(cmd.c_str());
+ }
+ }
+
+ return NULL;
+}
+
+static char** attempted_completion(const char* text, int start, int end)
+{
+ rl_attempted_completion_over = 1;
+ return rl_completion_matches(text, completion_matches);
+}
+
static void install_line_handler()
{
rl_startup_hook = startup_hook;
+ rl_attempted_completion_function = attempted_completion;
rl_callback_handler_install("", handle_line);
}
static void remove_line_handler()
{
+ rl_replace_line("", 0);
+ rl_set_prompt("");
+ rl_redisplay();
rl_unbind_key(RETURN);
+ rl_unbind_key(NEWLINE);
rl_callback_handler_remove();
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-12-28 21:10:58 +0000