main.cc 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291
  1. /* <!-- copyright */
  2. /*
  3. * aria2 - The high speed download utility
  4. *
  5. * Copyright (C) 2006 Tatsuhiro Tsujikawa
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. *
  21. * In addition, as a special exception, the copyright holders give
  22. * permission to link the code of portions of this program with the
  23. * OpenSSL library under certain conditions as described in each
  24. * individual source file, and distribute linked combinations
  25. * including the two.
  26. * You must obey the GNU General Public License in all respects
  27. * for all of the code used other than OpenSSL. If you modify
  28. * file(s) with this exception, you may extend this exception to your
  29. * version of the file(s), but you are not obligated to do so. If you
  30. * do not wish to do so, delete this exception statement from your
  31. * version. If you delete this exception statement from all source
  32. * files in the program, then also delete it here.
  33. */
  34. /* copyright --> */
  35. #include "common.h"
  36. #include <signal.h>
  37. #include <unistd.h>
  38. #include <getopt.h>
  39. #include <fstream>
  40. #include <iostream>
  41. #include <numeric>
  42. #include <vector>
  43. #include "SharedHandle.h"
  44. #include "LogFactory.h"
  45. #include "Logger.h"
  46. #include "util.h"
  47. #include "FeatureConfig.h"
  48. #include "MultiUrlRequestInfo.h"
  49. #include "SimpleRandomizer.h"
  50. #include "File.h"
  51. #include "message.h"
  52. #include "prefs.h"
  53. #include "Option.h"
  54. #include "a2algo.h"
  55. #include "a2io.h"
  56. #include "a2time.h"
  57. #include "Platform.h"
  58. #include "FileEntry.h"
  59. #include "RequestGroup.h"
  60. #include "ConsoleStatCalc.h"
  61. #include "NullStatCalc.h"
  62. #include "download_helper.h"
  63. #include "Exception.h"
  64. #include "ProtocolDetector.h"
  65. #include "RecoverableException.h"
  66. #include "SocketCore.h"
  67. #include "DownloadContext.h"
  68. #include "fmt.h"
  69. #ifdef ENABLE_BITTORRENT
  70. # include "bittorrent_helper.h"
  71. #endif // ENABLE_BITTORRENT
  72. #ifdef ENABLE_METALINK
  73. # include "metalink_helper.h"
  74. # include "MetalinkEntry.h"
  75. #endif // ENABLE_METALINK
  76. #ifdef ENABLE_MESSAGE_DIGEST
  77. # include "message_digest_helper.h"
  78. #endif // ENABLE_MESSAGE_DIGEST
  79. extern char* optarg;
  80. extern int optind, opterr, optopt;
  81. namespace aria2 {
  82. // output stream to /dev/null
  83. std::ofstream nullout(DEV_NULL);
  84. SharedHandle<StatCalc> getStatCalc(const SharedHandle<Option>& op)
  85. {
  86. SharedHandle<StatCalc> statCalc;
  87. if(op->getAsBool(PREF_QUIET)) {
  88. statCalc.reset(new NullStatCalc());
  89. } else {
  90. SharedHandle<ConsoleStatCalc> impl
  91. (new ConsoleStatCalc(op->getAsInt(PREF_SUMMARY_INTERVAL),
  92. op->getAsBool(PREF_HUMAN_READABLE)));
  93. impl->setReadoutVisibility(op->getAsBool(PREF_SHOW_CONSOLE_READOUT));
  94. statCalc = impl;
  95. }
  96. return statCalc;
  97. }
  98. std::ostream& getSummaryOut(const SharedHandle<Option>& op)
  99. {
  100. if(op->getAsBool(PREF_QUIET)) {
  101. return nullout;
  102. } else {
  103. return std::cout;
  104. }
  105. }
  106. #ifdef ENABLE_BITTORRENT
  107. namespace {
  108. void showTorrentFile(const std::string& uri)
  109. {
  110. SharedHandle<Option> op(new Option());
  111. SharedHandle<DownloadContext> dctx(new DownloadContext());
  112. bittorrent::load(uri, dctx, op);
  113. bittorrent::print(std::cout, dctx);
  114. }
  115. } // namespace
  116. #endif // ENABLE_BITTORRENT
  117. #ifdef ENABLE_METALINK
  118. namespace {
  119. void showMetalinkFile
  120. (const std::string& uri, const SharedHandle<Option>& op)
  121. {
  122. std::vector<SharedHandle<MetalinkEntry> > metalinkEntries;
  123. metalink::parseAndQuery(metalinkEntries, uri, op.get(),
  124. op->get(PREF_METALINK_BASE_URI));
  125. std::vector<SharedHandle<FileEntry> > fileEntries;
  126. MetalinkEntry::toFileEntry(fileEntries, metalinkEntries);
  127. util::toStream(fileEntries.begin(), fileEntries.end(), std::cout);
  128. std::cout << std::endl;
  129. }
  130. } // namespace
  131. #endif // ENABLE_METALINK
  132. #if defined ENABLE_BITTORRENT || defined ENABLE_METALINK
  133. namespace {
  134. void showFiles
  135. (const std::vector<std::string>& uris, const SharedHandle<Option>& op)
  136. {
  137. ProtocolDetector dt;
  138. for(std::vector<std::string>::const_iterator i = uris.begin(),
  139. eoi = uris.end(); i != eoi; ++i) {
  140. printf(">>> ");
  141. printf(MSG_SHOW_FILES, (*i).c_str());
  142. printf("\n");
  143. try {
  144. #ifdef ENABLE_BITTORRENT
  145. if(dt.guessTorrentFile(*i)) {
  146. showTorrentFile(*i);
  147. } else
  148. #endif // ENABLE_BITTORRENT
  149. #ifdef ENABLE_METALINK
  150. if(dt.guessMetalinkFile(*i)) {
  151. showMetalinkFile(*i, op);
  152. } else
  153. #endif // ENABLE_METALINK
  154. {
  155. printf(MSG_NOT_TORRENT_METALINK);
  156. printf("\n\n");
  157. }
  158. } catch(RecoverableException& e) {
  159. std::cout << e.stackTrace() << std::endl;
  160. }
  161. }
  162. }
  163. } // namespace
  164. #endif // ENABLE_BITTORRENT || ENABLE_METALINK
  165. extern void option_processing(Option& option, std::vector<std::string>& uris,
  166. int argc, char* argv[]);
  167. error_code::Value main(int argc, char* argv[])
  168. {
  169. std::vector<std::string> args;
  170. SharedHandle<Option> op(new Option());
  171. option_processing(*op.get(), args, argc, argv);
  172. SimpleRandomizer::init();
  173. #ifdef ENABLE_BITTORRENT
  174. bittorrent::generateStaticPeerId(op->get(PREF_PEER_ID_PREFIX));
  175. #endif // ENABLE_BITTORRENT
  176. LogFactory::setLogFile(op->get(PREF_LOG));
  177. LogFactory::setLogLevel(op->get(PREF_LOG_LEVEL));
  178. if(op->getAsBool(PREF_QUIET)) {
  179. LogFactory::setConsoleOutput(false);
  180. }
  181. error_code::Value exitStatus = error_code::FINISHED;
  182. A2_LOG_INFO("<<--- --- --- ---");
  183. A2_LOG_INFO(" --- --- --- ---");
  184. A2_LOG_INFO(" --- --- --- --->>");
  185. A2_LOG_INFO(fmt("%s %s %s", PACKAGE, PACKAGE_VERSION, TARGET));
  186. A2_LOG_INFO(MSG_LOGGING_STARTED);
  187. #ifdef ENABLE_MESSAGE_DIGEST
  188. message_digest::staticSHA1DigestInit();
  189. #endif // ENABLE_MESSAGE_DIGEST
  190. if(op->getAsBool(PREF_DISABLE_IPV6)) {
  191. SocketCore::setProtocolFamily(AF_INET);
  192. // Get rid of AI_ADDRCONFIG. It causes name resolution error
  193. // when none of network interface has IPv4 address.
  194. setDefaultAIFlags(0);
  195. }
  196. // Bind interface
  197. if(!op->get(PREF_INTERFACE).empty()) {
  198. std::string iface = op->get(PREF_INTERFACE);
  199. SocketCore::bindAddress(iface);
  200. }
  201. #ifdef SIGPIPE
  202. util::setGlobalSignalHandler(SIGPIPE, SIG_IGN, 0);
  203. #endif
  204. #ifdef SIGCHLD
  205. // Avoid to create zombie process when forked child processes are
  206. // died.
  207. util::setGlobalSignalHandler(SIGCHLD, SIG_IGN, 0);
  208. #endif // SIGCHILD
  209. std::vector<SharedHandle<RequestGroup> > requestGroups;
  210. #ifdef ENABLE_BITTORRENT
  211. if(!op->blank(PREF_TORRENT_FILE)) {
  212. if(op->get(PREF_SHOW_FILES) == A2_V_TRUE) {
  213. showTorrentFile(op->get(PREF_TORRENT_FILE));
  214. return exitStatus;
  215. } else {
  216. createRequestGroupForBitTorrent(requestGroups, op, args);
  217. }
  218. }
  219. else
  220. #endif // ENABLE_BITTORRENT
  221. #ifdef ENABLE_METALINK
  222. if(!op->blank(PREF_METALINK_FILE)) {
  223. if(op->get(PREF_SHOW_FILES) == A2_V_TRUE) {
  224. showMetalinkFile(op->get(PREF_METALINK_FILE), op);
  225. return exitStatus;
  226. } else {
  227. createRequestGroupForMetalink(requestGroups, op);
  228. }
  229. }
  230. else
  231. #endif // ENABLE_METALINK
  232. if(!op->blank(PREF_INPUT_FILE)) {
  233. createRequestGroupForUriList(requestGroups, op);
  234. #if defined ENABLE_BITTORRENT || defined ENABLE_METALINK
  235. } else if(op->get(PREF_SHOW_FILES) == A2_V_TRUE) {
  236. showFiles(args, op);
  237. return exitStatus;
  238. #endif // ENABLE_METALINK || ENABLE_METALINK
  239. } else {
  240. createRequestGroupForUri(requestGroups, op, args, false, false, true);
  241. }
  242. // Remove option values which is only valid for URIs specified in
  243. // command-line. If they are left, because op is used as a template
  244. // for new RequestGroup(such as created in RPC command), they causes
  245. // unintentional effect.
  246. op->remove(PREF_OUT);
  247. op->remove(PREF_FORCE_SEQUENTIAL);
  248. op->remove(PREF_INPUT_FILE);
  249. op->remove(PREF_INDEX_OUT);
  250. op->remove(PREF_SELECT_FILE);
  251. if(!op->getAsBool(PREF_ENABLE_RPC) && requestGroups.empty()) {
  252. std::cout << MSG_NO_FILES_TO_DOWNLOAD << std::endl;
  253. } else {
  254. exitStatus = MultiUrlRequestInfo(requestGroups, op, getStatCalc(op),
  255. getSummaryOut(op)).execute();
  256. }
  257. return exitStatus;
  258. }
  259. } // namespace aria2
  260. int main(int argc, char* argv[])
  261. {
  262. aria2::error_code::Value r;
  263. try {
  264. aria2::Platform platform;
  265. r = aria2::main(argc, argv);
  266. } catch(aria2::Exception& ex) {
  267. std::cerr << EX_EXCEPTION_CAUGHT << "\n" << ex.stackTrace() << std::endl;
  268. r = ex.getErrorCode();
  269. }
  270. return r;
  271. }