使用示例（使用了默认用户root，和默认端口号22）：
./mooon_ssh --h=192.168.4.1,192.168.4.2 -P=password -c='cat /etc/hosts'
./mooon_ssh --h=192.168.4.1,192.168.4.2 -P=password -c='cat /etc/hosts'
- #include "mooon/net/libssh2.h" // 提供远程执行命令接口
- #include "mooon/sys/error.h"
- #include "mooon/sys/stop_watch.h"
- #include "mooon/utils/args_parser.h"
- #include "mooon/utils/print_color.h"
- #include "mooon/utils/string_utils.h"
- #include "mooon/utils/tokener.h"
- #include iostream>
-
- // 被执行的命令，可为一条或多条命令，如：ls /&&whoami
- STRING_ARG_DEFINE(c, "", "command to execute remotely");
- // 逗号分隔的远程主机列表
- STRING_ARG_DEFINE(h, "", "remote hosts");
- // 远程主机的sshd端口号
- INTEGER_ARG_DEFINE(uint16_t, p, 22, 10, 65535, "remote hosts port");
- // 用户名
- STRING_ARG_DEFINE(u, "root", "remote host user");
- // 密码
- STRING_ARG_DEFINE(P, "", "remote host password");
-
- // 连接超时，单位为秒
- INTEGER_ARG_DEFINE(uint16_t, t, 10, 1, 65535, "timeout seconds to remote host");
-
- // 结果信息
- struct ResultInfo
- {
- bool success; // 为true表示执行成功
- std::string ip; // 远程host的IP地址
- uint32_t seconds; // 运行花费的时长，精确到秒
-
- ResultInfo()
- : success(false), seconds(0)
- {
- }
-
- std::string str() const
- {
- std::string tag = success? "SUCCESS": "FAILURE";
- return mooon::utils::CStringUtils::format_string("[%s %s]: %u seconds", ip.c_str(), tag.c_str(), seconds);
- }
- };
-
- inline std::ostream& operator (std::ostream& out, const struct ResultInfo& result)
- {
- std::string tag = result.success? "SUCCESS": "FAILURE";
- out "["PRINT_COLOR_YELLOW result.ip PRINT_COLOR_NONE" " tag "] " result.seconds " seconds";
- return out;
- }
-
- // 使用示例：
- // mooon_ssh -u=root -P=test -p=2016 -h="127.0.0.1,192.168.0.1" -c='ls /tmp&&ps aux|grep -c test'
- int main(int argc, char* argv[])
- {
- // 解析命令行参数
- std::string errmsg;
- if (!mooon::utils::parse_arguments(argc, argv, &errmsg))
- {
- fprintf(stderr, "parameter error: %s\n", errmsg.c_str());
- exit(1);
- }
-
- uint16_t port = mooon::argument::p->value();
- std::string commands = mooon::argument::c->value();
- std::string hosts = mooon::argument::h->value();
- std::string user = mooon::argument::u->value();
- std::string password = mooon::argument::P->value();
- mooon::utils::CStringUtils::trim(commands);
- mooon::utils::CStringUtils::trim(hosts);
- mooon::utils::CStringUtils::trim(user);
- mooon::utils::CStringUtils::trim(password);
-
- // 检查参数（-c）
- if (commands.empty())
- {
- fprintf(stderr, "parameter[-c]'s value not set\n");
- exit(1);
- }
-
- // 检查参数（-h）
- if (hosts.empty())
- {
- // 尝试从环境变量取值
- const char* hosts_ = getenv("HOSTS");
- if (NULL == hosts_)
- {
- fprintf(stderr, "parameter[-h]'s value not set\n");
- exit(1);
- }
-
- hosts= hosts_;
- mooon::utils::CStringUtils::trim(hosts);
- if (hosts.empty())
- {
- fprintf(stderr, "parameter[-h]'s value not set\n");
- exit(1);
- }
- }
-
- // 检查参数（-u）
- if (user.empty())
- {
- fprintf(stderr, "parameter[-u]'s value not set\n");
- exit(1);
- }
-
- // 检查参数（-P）
- if (password.empty())
- {
- fprintf(stderr, "parameter[-P]'s value not set\n");
- exit(1);
- }
-
- std::vectorstd::string> hosts_ip;
- const std::string& remote_hosts_ip = hosts;
- int num_remote_hosts_ip = mooon::utils::CTokener::split(&hosts_ip, remote_hosts_ip, ",", true);
- if (0 == num_remote_hosts_ip)
- {
- fprintf(stderr, "parameter[-h] error\n");
- exit(1);
- }
-
- std::vectorstruct ResultInfo> results(num_remote_hosts_ip);
- for (int i=0; inum_remote_hosts_ip; ++i)
- {
- bool color = true;
- int num_bytes = 0;
- int exitcode = 0;
- std::string exitsignal;
- std::string errmsg;
- const std::string& remote_host_ip = hosts_ip[i];
- results[i].ip = remote_host_ip;
-
- fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"]\n", remote_host_ip.c_str());
- fprintf(stdout, PRINT_COLOR_GREEN);
-
- mooon::sys::CStopWatch stop_watch;
- try
- {
- mooon::net::CLibssh2 libssh2(remote_host_ip, port, user, password, mooon::argument::t->value());
-
- libssh2.remotely_execute(commands, std::cout, &exitcode, &exitsignal, &errmsg, &num_bytes);
- fprintf(stdout, PRINT_COLOR_NONE);
- color = false; // color = true;
-
- if ((0 == exitcode) && exitsignal.empty())
- {
- results[i].success = true;
- fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"] SUCCESS\n", remote_host_ip.c_str());
- }
- else
- {
- results[i].success = false;
-
- if (exitcode != 0)
- {
- fprintf(stderr, "%d: %s\n", exitcode, mooon::sys::Error::to_string(exitcode).c_str());
- }
- else if (!exitsignal.empty())
- {
- fprintf(stderr, "%s: %s\n", exitsignal.c_str(), errmsg.c_str());
- }
- }
- }
- catch (mooon::sys::CSyscallException& ex)
- {
- if (color)
- fprintf(stdout, PRINT_COLOR_NONE); // color = true;
-
- fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
- }
- catch (mooon::utils::CException& ex)
- {
- if (color)
- fprintf(stdout, PRINT_COLOR_NONE); // color = true;
-
- fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
- }
-
- results[i].seconds = stop_watch.get_elapsed_microseconds() / 1000000;
- std::cout std::endl;
- } // for
-
- // 输出总结
- std::cout std::endl;
- std::cout "================================" std::endl;
- int num_success = 0; // 成功的个数
- int num_failure = 0; // 失败的个数
- for (std::vectorstruct ResultInfo>::size_type i=0; iresults.size(); ++i)
- {
- const struct ResultInfo& result_info = results[i];
- std::cout result_info std::endl;
-
- if (result_info.success)
- ++num_success;
- else
- ++num_failure;
- }
- std::cout "SUCCESS: " num_success ", FAILURE: " num_failure std::endl;
-
- return 0;
- }
