Capture Ctrl+C in cli_wallet when not in daemon mode #1193

bitshares · Oct 19, 2018 · 7dc0de7 · 7dc0de7
1 parent fc7ed0c
commit 7dc0de7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/programs/cli_wallet/main.cpp b/programs/cli_wallet/main.cpp
@@ -279,6 +279,17 @@ int main( int argc, char** argv )
       {
          wallet_cli->register_api( wapi );
          wallet_cli->start();
+
+         fc::set_signal_handler([](int signal) {
+            ilog( "Captured SIGINT not in daemon mode" );
+            fclose(stdin);
+         }, SIGINT);
+
+         fc::set_signal_handler([](int signal) {
+            ilog( "Captured SIGTERM not in daemon mode" );
+            fclose(stdin);
+         }, SIGTERM);
+
          wallet_cli->wait();
       }
       else
@@ -288,6 +299,10 @@ int main( int argc, char** argv )
            exit_promise->set_value(signal);
         }, SIGINT);
 
+        fc::set_signal_handler([&exit_promise](int signal) {
+           exit_promise->set_value(signal);
+        }, SIGTERM);
+
         ilog( "Entering Daemon Mode, ^C to exit" );
         exit_promise->wait();
       }