zoukankan      html  css  js  c++  java
  • 批量远程执行shell命令工具

    使用示例(使用了默认用户root,和默认端口号22):
    ./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
    ", 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
    ");
            exit(1);
        }
    
        // 检查参数(-h)
        if (hosts.empty())
        {
            // 尝试从环境变量取值
            const char* hosts_ = getenv("HOSTS");
            if (NULL == hosts_)
            {
                fprintf(stderr, "parameter[-h]'s value not set
    ");
                exit(1);
            }
    
            hosts= hosts_;
            mooon::utils::CStringUtils::trim(hosts);
            if (hosts.empty())
            {
                fprintf(stderr, "parameter[-h]'s value not set
    ");
                exit(1);
            }
        }
    
        // 检查参数(-u)
        if (user.empty())
        {
            fprintf(stderr, "parameter[-u]'s value not set
    ");
            exit(1);
        }
    
        // 检查参数(-P)
        if (password.empty())
        {
            fprintf(stderr, "parameter[-P]'s value not set
    ");
            exit(1);
        }
    
        std::vector<std::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
    ");
            exit(1);
        }
    
        std::vector<struct ResultInfo> results(num_remote_hosts_ip);
        for (int i=0; i<num_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"]
    ", 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
    ", remote_host_ip.c_str());
                }
                else
                {
                    results[i].success = false;
    
                    if (exitcode != 0)
                    {
                        fprintf(stderr, "%d: %s
    ", exitcode, mooon::sys::Error::to_string(exitcode).c_str());
                    }
                    else if (!exitsignal.empty())
                    {
                        fprintf(stderr, "%s: %s
    ", 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
    ", 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
    ", 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::vector<struct ResultInfo>::size_type i=0; i<results.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;
    }


  • 相关阅读:
    Revit二次开发 C#程序员的佳好选择
    查询性能调优和索引优化那些事
    步步为营 .NET 设计模式学习笔记 十七、Flyweight(享元模式)
    初窥Ruby Metaprogramming
    线程间操作无效: 从不是创建控件“”的线程访问它
    全文搜索的,Lucene.net
    认识Lucene
    一些ObjectiveC学习资源
    步步为营 .NET 设计模式学习笔记 十五、Composite(组合模式)
    步步为营 .NET 设计模式学习笔记 十六、Facade(外观模式)
  • 原文地址:https://www.cnblogs.com/aquester/p/9891537.html
Copyright © 2011-2022 走看看