download_helper.h 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  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. #ifndef D_DOWNLOAD_HELPER_H
  36. #define D_DOWNLOAD_HELPER_H
  37. #include "common.h"
  38. #include <string>
  39. #include <vector>
  40. #include <set>
  41. #include <memory>
  42. namespace aria2 {
  43. class RequestGroup;
  44. class Option;
  45. class MetadataInfo;
  46. class DownloadContext;
  47. class UriListParser;
  48. class ValueBase;
  49. class GroupId;
  50. #ifdef ENABLE_BITTORRENT
  51. // Create RequestGroup object using torrent file specified by
  52. // metaInfoUri, which is treated as local file path. If non-empty
  53. // torrentData is specified, then it is used as a content of torrent
  54. // file instead. If adjustAnnounceUri is true, announce URIs are
  55. // adjusted using bittorrent::adjustAnnounceUri(). In this function,
  56. // force-sequential is ignored.
  57. void createRequestGroupForBitTorrent(
  58. std::vector<std::shared_ptr<RequestGroup>>& result,
  59. const std::shared_ptr<Option>& option, const std::vector<std::string>& uris,
  60. const std::string& metaInfoUri, const std::string& torrentData = "",
  61. bool adjustAnnounceUri = true);
  62. // Create RequestGroup object using already decoded torrent metainfo
  63. // structure. If adjustAnnounceUri is true, announce URIs are
  64. // adjusted using bittorrent::adjustAnnounceUri(). In this function,
  65. // force-sequential is ignored.
  66. void createRequestGroupForBitTorrent(
  67. std::vector<std::shared_ptr<RequestGroup>>& result,
  68. const std::shared_ptr<Option>& option, const std::vector<std::string>& uris,
  69. const std::string& metaInfoUri, const ValueBase* torrent,
  70. bool adjustAnnounceUri = true);
  71. #endif // ENABLE_BITTORRENT
  72. #ifdef ENABLE_METALINK
  73. // Create RequestGroup objects using Metalink file specified by
  74. // metalink-file option. If non-empty metalinkData is specified, it is
  75. // used as a content of metalink file instead.
  76. void createRequestGroupForMetalink(
  77. std::vector<std::shared_ptr<RequestGroup>>& result,
  78. const std::shared_ptr<Option>& option,
  79. const std::string& metalinkData = "");
  80. #endif // ENABLE_METALINK
  81. // Reads one entry from uriListParser and creates RequestGroups from
  82. // it and store them in result. If the bad entry is found, this
  83. // function just skips it and reads next entry. If at least one
  84. // RequestGroup is created successfully, this function returns true
  85. // and created RequestGroups are stored in result. If no RequestGroup
  86. // is created and uriListParser reads all input, this function returns
  87. // false. The option is used as a option template.
  88. bool createRequestGroupFromUriListParser(
  89. std::vector<std::shared_ptr<RequestGroup>>& result, const Option* option,
  90. UriListParser* uriListParser);
  91. // Creates UriListParser using given filename. If filename is "-",
  92. // then UriListParser is configured to read from standard input.
  93. // Otherwise, this function first checks file denoted by filename
  94. // exists. If it does not exist, this function throws exception.
  95. // This function returns std::shared_ptr<UriListParser> object if it
  96. // succeeds.
  97. std::shared_ptr<UriListParser> openUriListParser(const std::string& filename);
  98. // Create RequestGroup objects from reading file specified by input-file option.
  99. // If the value of input-file option is "-", stdin is used as a input source.
  100. // Each line is treated as if it is provided in command-line argument.
  101. // The additional out and dir options can be specified after each line of URIs.
  102. // This optional line must start with white space(s).
  103. void createRequestGroupForUriList(
  104. std::vector<std::shared_ptr<RequestGroup>>& result,
  105. const std::shared_ptr<Option>& option);
  106. // Create RequestGroup object using provided uris. If ignoreLocalPath
  107. // is true, a path to torrent file and metalink file are ignored. If
  108. // throwOnError is true, exception will be thrown when Metalink
  109. // Document or .torrent file cannot be parsed or unrecognized URI is
  110. // given. If throwOnError is false, these errors are just logged as
  111. // error.
  112. void createRequestGroupForUri(
  113. std::vector<std::shared_ptr<RequestGroup>>& result,
  114. const std::shared_ptr<Option>& option, const std::vector<std::string>& uris,
  115. bool ignoreForceSequential = false, bool ignoreLocalPath = false,
  116. bool throwOnError = false);
  117. template <typename InputIterator>
  118. void setMetadataInfo(InputIterator first, InputIterator last,
  119. const std::shared_ptr<MetadataInfo>& mi)
  120. {
  121. for (; first != last; ++first) {
  122. (*first)->setMetadataInfo(mi);
  123. }
  124. }
  125. std::shared_ptr<MetadataInfo> createMetadataInfoFromFirstFileEntry(
  126. const std::shared_ptr<GroupId>& gid,
  127. const std::shared_ptr<DownloadContext>& dctx);
  128. // Removes option value which is only effective at the first
  129. // construction time.
  130. void removeOneshotOption(const std::shared_ptr<Option>& option);
  131. } // namespace aria2
  132. #endif // D_DOWNLOAD_HELPER_H