Prevent sonic-db-cli generate core dump

sonic-db-cli/main.cpp

@@ -1,5 +1,6 @@
 #include "sonic-db-cli.h"
 #include "common/dbconnector.h"
+#include <iostream>
 
 using namespace swss;
 using namespace std;
@@ -16,7 +17,7 @@ int main(int argc, char** argv)
         SonicDBConfig::initialize(SonicDBConfig::DEFAULT_SONIC_DB_CONFIG_FILE);
     };
 
-    return sonic_db_cli(
+    return cli_exception_wrapper(
                     argc,
                     argv,
                     initializeGlobalConfig,

sonic-db-cli/sonic-db-cli.cpp

@@ -343,6 +343,38 @@ int sonic_db_cli(
     return 0;
 }
 
+
+int cli_exception_wrapper(
+    int argc,
+    char** argv,
+    function<void()> initializeGlobalConfig,
+    function<void()> initializeConfig)
+{
+    try
+    {
+        return sonic_db_cli(
+                        argc,
+                        argv,
+                        initializeGlobalConfig,
+                        initializeConfig);
+    }
+    catch (const exception& e)
+    {
+        // sonic-db-cli is porting from python version.
+        // in python version, when any exception happen sonic-db-cli will crash but will not generate core dump file.
+        // catch all exception here to avoid a core dump file generated.
+        cerr << "An exception of type " << e.what() << " occurred. Arguments:" << endl;
+        for (int idx = 0; idx < argc; idx++)
+        {
+            cerr << argv[idx]  << " ";
+        }
+        cerr << endl;
+
+        // when python version crash, exit code is 1.
+        return 1;
+    }
+}
+
 string getCommandName(vector<string>& commands)
 {
     if (commands.size() == 0)

sonic-db-cli/sonic-db-cli.h

@@ -54,4 +54,10 @@ int sonic_db_cli(
     std::function<void()> initializeGlobalConfig,
     std::function<void()> initializeConfig);
 
+int cli_exception_wrapper(
+    int argc,
+    char** argv,
+    std::function<void()> initializeGlobalConfig,
+    std::function<void()> initializeConfig);
+
 std::string getCommandName(std::vector<std::string>& command);

tests/cli_ut.cpp

@@ -508,3 +508,35 @@ TEST(sonic_db_cli, test_parallel_cmd) {
     auto parallen_time = float( clock () - begin_time );
     EXPECT_TRUE(parallen_time < sequential_time);
 }
+
+TEST(sonic_db_cli, test_cli_not_throw_exception)
+{
+    char *args[5];
+    args[0] = "sonic-db-cli";
+    args[1] = "TEST_DB";
+
+    // set key to test DB
+    args[2] = "SET";
+    args[3] = "testkey";
+    args[4] = "testvalue";
+
+    // data base file does not exist, will throw exception
+    auto initializeGlobalConfig = []()
+    {
+        throw std::system_error();
+    };
+
+    auto initializeConfig = []()
+    {
+        throw std::system_error();
+    };
+
+    optind = 0;
+    int exit_code = cli_exception_wrapper(
+                        5,
+                        args,
+                        initializeGlobalConfig,
+                        initializeConfig);
+
+    EXPECT_EQ(1, exit_code);
+}