博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
将单个文件上传到多机器工具
阅读量:6412 次
发布时间:2019-06-23

本文共 4903 字,大约阅读时间需要 16 分钟。

// 使用示例:

// ./mooon_upload -h=192.168.10.11,192.168.10.12 -p=6000 -u=root -P='root123' -s=./abc -d=/tmp/
// 表示将本地的文件./abc上传到两台机器192.168.10.11和192.168.10.12的/tmp/目录

#include "mooon/net/libssh2.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 
#include
// 逗号分隔的远程主机列表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");// 被上传的文件路径STRING_ARG_DEFINE(s, "", "the source file uploaded");// 文件上传后存放的目录路径STRING_ARG_DEFINE(d, "", "the directory to store");// 连接超时,单位为秒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;}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 source = mooon::argument::s->value(); std::string directory = mooon::argument::d->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(source); mooon::utils::CStringUtils::trim(directory); mooon::utils::CStringUtils::trim(hosts); mooon::utils::CStringUtils::trim(user); mooon::utils::CStringUtils::trim(password); // 检查参数(-s) if (source.empty()) { fprintf(stderr, "parameter[-s]'s value not set\n"); exit(1); } // 检查参数(-d) if (directory.empty()) { fprintf(stderr, "parameter[-d]'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::vector
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::string remote_filepath = directory + std::string("/") + mooon::utils::CStringUtils::extract_filename(source); std::vector
results(num_remote_hosts_ip); for (int i=0; i
value()); libssh2.upload(source, remote_filepath, &file_size); fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"] SUCCESS: %d bytes\n", remote_host_ip.c_str(), file_size); results[i].success = true; } 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; } // 输出总结 std::cout << std::endl; std::cout << "================================" << std::endl; int num_success = 0; // 成功的个数 int num_failure = 0; // 失败的个数 for (std::vector
::size_type i=0; i

转载地址:http://kikra.baihongyu.com/

你可能感兴趣的文章
Ubuntu构建LVS+Keepalived高可用负载均衡集群【生产环境部署】
查看>>
lvm实现快速备份文件及数据库,lvm快照原理
查看>>
设计模式之Factory Method(工厂方法)
查看>>
10K入职linux运维岗位小伙伴感谢信及面试经历分享
查看>>
zookeeper入门之Curator的使用之几种监听器的使用
查看>>
[转]Reporting Service部署之访问权限
查看>>
innerxml and outerxml
查看>>
validform校验框架不显示错误提示
查看>>
flink 获取上传的Jar源码
查看>>
Spring Data JPA Batch Insertion
查看>>
UEditor自动调节宽度
查看>>
JAVA做验证码图片(转自CSDN)
查看>>
Delphi TServerSocket,TClientSocket实现传送文件代码
查看>>
JS无聊之作
查看>>
Mac上搭建ELK
查看>>
443 Chapter7.Planning for High Availability in the Enterprise
查看>>
框架和语言的作用
查看>>
unidac连接ORACLE免装客户端驱动
查看>>
Cygwin + OpenSSH FOR Windows的安装配置
查看>>
咏南中间件支持手机客户端
查看>>