上傳單個文件到多台機器工具

上傳單個文件到多台機器工具

使用示例：
./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/目錄

1. #include "mooon/net/libssh2.h"
   
2. #include "mooon/sys/stop_watch.h"
   
3. #include "mooon/utils/args_parser.h"
   
4. #include "mooon/utils/print_color.h"
   
5. #include "mooon/utils/string_utils.h"
   
6. #include "mooon/utils/tokener.h"
   
7. #include <fstream>
   
8. #include <iostream>
   
9. 
   
10. // 逗號分隔的遠程主機列表
    
11. STRING_ARG_DEFINE(h, "", "remote hosts");
    
12. // 遠程主機的sshd端口號
    
13. INTEGER_ARG_DEFINE(uint16_t, p, 22, 10, 65535, "remote hosts port");
    
14. // 用戶名
    
15. STRING_ARG_DEFINE(u, "root", "remote host user");
    
16. // 密碼
    
17. STRING_ARG_DEFINE(P, "", "remote host password");
    
18. 
    
19. // 被上傳的文件路徑
    
20. STRING_ARG_DEFINE(s, "", "the source file uploaded");
    
21. // 文件上傳後存放的目錄路徑
    
22. STRING_ARG_DEFINE(d, "", "the directory to store");
    
23. 
    
24. // 連接超時，單位為秒
    
25. INTEGER_ARG_DEFINE(uint16_t, t, 10, 1, 65535, "timeout seconds to remote host");
    
26. 
    
27. // 結果信息
    
28. struct ResultInfo
    
29. {
    
30. bool success; // 為true表示執行成功
    
31. std::string ip; // 遠程host的IP地址
    
32. uint32_t seconds; // 運行花費的時長，精確到秒
    
33. 
    
34. ResultInfo()
    
35. : success(false), seconds(0)
    
36. {
    
37. }
    
38. 
    
39. std::string str() const
    
40. {
    
41. std::string tag = success? "SUCCESS": "FAILURE";
    
42. return mooon::utils::CStringUtils::format_string("[%s %s]: %u seconds", ip.c_str(), tag.c_str(), seconds);
    
43. }
    
44. };
    
45. 
    
46. inline std::ostream& operator <<(std::ostream& out, const struct ResultInfo& result)
    
47. {
    
48. std::string tag = result.success? "SUCCESS": "FAILURE";
    
49. out << "["PRINT_COLOR_YELLOW << result.ip << PRINT_COLOR_NONE" " << tag << "] " << result.seconds << " seconds";
    
50. return out;
    
51. }
    
52. 
    
53. int main(int argc, char* argv[])
    
54. {
    
55. // 解析命令行參數
    
56. std::string errmsg;
    
57. if (!mooon::utils::parse_arguments(argc, argv, &errmsg))
    
58. {
    
59. fprintf(stderr, "parameter error: %s\n", errmsg.c_str());
    
60. exit(1);
    
61. }
    
62. 
    
63. uint16_t port = mooon::argument::p->value();
    
64. std::string source = mooon::argument::s->value();
    
65. std::string directory = mooon::argument::d->value();
    
66. std::string hosts = mooon::argument::h->value();
    
67. std::string user = mooon::argument::u->value();
    
68. std::string password = mooon::argument::P->value();
    
69. mooon::utils::CStringUtils::trim(source);
    
70. mooon::utils::CStringUtils::trim(directory);
    
71. mooon::utils::CStringUtils::trim(hosts);
    
72. mooon::utils::CStringUtils::trim(user);
    
73. mooon::utils::CStringUtils::trim(password);
    
74. 
    
75. // 檢查參數（-s）
    
76. if (source.empty())
    
77. {
    
78. fprintf(stderr, "parameter[-s]'s value not set\n");
    
79. exit(1);
    
80. }
    
81. 
    
82. // 檢查參數（-d）
    
83. if (directory.empty())
    
84. {
    
85. fprintf(stderr, "parameter[-d]'s value not set\n");
    
86. exit(1);
    
87. }
    
88. 
    
89. // 檢查參數（-h）
    
90. if (hosts.empty())
    
91. {
    
92. // 嘗試從環境變量取值
    
93. const char* hosts_ = getenv("HOSTS");
    
94. if (NULL == hosts_)
    
95. {
    
96. fprintf(stderr, "parameter[-h]'s value not set\n");
    
97. exit(1);
    
98. }
    
99. 
    
100. hosts= hosts_;
     
101. mooon::utils::CStringUtils::trim(hosts);
     
102. if (hosts.empty())
     
103. {
     
104. fprintf(stderr, "parameter[-h]'s value not set\n");
     
105. exit(1);
     
106. }
     
107. }
     
108. 
     
109. // 檢查參數（-u）
     
110. if (user.empty())
     
111. {
     
112. fprintf(stderr, "parameter[-u]'s value not set\n");
     
113. exit(1);
     
114. }
     
115. 
     
116. // 檢查參數（-P）
     
117. if (password.empty())
     
118. {
     
119. fprintf(stderr, "parameter[-P]'s value not set\n");
     
120. exit(1);
     
121. }
     
122. 
     
123. std::vector<std::string> hosts_ip;
     
124. const std::string& remote_hosts_ip = hosts;
     
125. int num_remote_hosts_ip = mooon::utils::CTokener::split(&hosts_ip, remote_hosts_ip, ",", true);
     
126. if (0 == num_remote_hosts_ip)
     
127. {
     
128. fprintf(stderr, "parameter[-h] error\n");
     
129. exit(1);
     
130. }
     
131. 
     
132. std::string remote_filepath = directory + std::string("/") + mooon::utils::CStringUtils::extract_filename(source);
     
133. std::vector<struct ResultInfo> results(num_remote_hosts_ip);
     
134. for (int i=0; i<num_remote_hosts_ip; ++i)
     
135. {
     
136. bool color = true;
     
137. const std::string& remote_host_ip = hosts_ip[i];
     
138. results[i].ip = remote_host_ip;
     
139. results[i].success = false;
     
140. 
     
141. fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"]\n", remote_host_ip.c_str());
     
142. fprintf(stdout, PRINT_COLOR_GREEN);
     
143. 
     
144. mooon::sys::CStopWatch stop_watch;
     
145. try
     
146. {
     
147. int file_size = 0;
     
148. mooon::net::CLibssh2 libssh2(remote_host_ip, port, user, password, mooon::argument::t->value());
     
149. libssh2.upload(source, remote_filepath, &file_size);
     
150. 
     
151. fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"] SUCCESS: %d bytes\n", remote_host_ip.c_str(), file_size);
     
152. results[i].success = true;
     
153. }
     
154. catch (mooon::sys::CSyscallException& ex)
     
155. {
     
156. if (color)
     
157. fprintf(stdout, PRINT_COLOR_NONE); // color = true;
     
158. 
     
159. fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
     
160. }
     
161. catch (mooon::utils::CException& ex)
     
162. {
     
163. if (color)
     
164. fprintf(stdout, PRINT_COLOR_NONE); // color = true;
     
165. 
     
166. fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
     
167. }
     
168. 
     
169. results[i].seconds = stop_watch.get_elapsed_microseconds() / 1000000;
     
170. std::cout << std::endl;
     
171. }
     
172. 
     
173. // 輸出總結
     
174. std::cout << std::endl;
     
175. std::cout << "================================" << std::endl;
     
176. int num_success = 0; // 成功的個數
     
177. int num_failure = 0; // 失敗的個數
     
178. for (std::vector<struct ResultInfo>::size_type i=0; i<results.size(); ++i)
     
179. {
     
180. const struct ResultInfo& result_info = results[i];
     
181. std::cout << result_info << std::endl;
     
182. 
     
183. if (result_info.success)
     
184. ++num_success;
     
185. else
     
186. ++num_failure;
     
187. }
     
188. std::cout << "SUCCESS: " << num_success << ", FAILURE: " << num_failure << std::endl;
     
189. 
     
190. return 0;
     
191. }