forked from premiere/premiere-libtorrent
217 lines
6.3 KiB
C++
217 lines
6.3 KiB
C++
/*
|
|
|
|
Copyright (c) 2007-2012, Arvid Norberg
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in
|
|
the documentation and/or other materials provided with the distribution.
|
|
* Neither the name of the author nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#include "libtorrent/magnet_uri.hpp"
|
|
#include "libtorrent/session.hpp"
|
|
#include "libtorrent/torrent_handle.hpp"
|
|
#include "libtorrent/escape_string.hpp"
|
|
#include "libtorrent/error_code.hpp"
|
|
|
|
#include <string>
|
|
|
|
namespace libtorrent
|
|
{
|
|
std::string make_magnet_uri(torrent_handle const& handle)
|
|
{
|
|
if (!handle.is_valid()) return "";
|
|
|
|
std::string ret;
|
|
sha1_hash const& ih = handle.info_hash();
|
|
ret += "magnet:?xt=urn:btih:";
|
|
ret += to_hex(ih.to_string());
|
|
|
|
torrent_status st = handle.status(torrent_handle::query_name);
|
|
|
|
if (!st.name.empty())
|
|
{
|
|
ret += "&dn=";
|
|
ret += escape_string(st.name.c_str(), st.name.length());
|
|
}
|
|
|
|
std::vector<announce_entry> const& tr = handle.trackers();
|
|
|
|
for (std::vector<announce_entry>::const_iterator i = tr.begin(), end(tr.end()); i != end; ++i)
|
|
{
|
|
ret += "&tr=";
|
|
ret += escape_string(i->url.c_str(), i->url.length());
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
std::string make_magnet_uri(torrent_info const& info)
|
|
{
|
|
std::string ret;
|
|
sha1_hash const& ih = info.info_hash();
|
|
ret += "magnet:?xt=urn:btih:";
|
|
ret += to_hex(ih.to_string());
|
|
|
|
std::string const& name = info.name();
|
|
|
|
if (!name.empty())
|
|
{
|
|
ret += "&dn=";
|
|
ret += escape_string(name.c_str(), name.length());
|
|
}
|
|
|
|
std::vector<announce_entry> const& tr = info.trackers();
|
|
|
|
for (std::vector<announce_entry>::const_iterator i = tr.begin(), end(tr.end()); i != end; ++i)
|
|
{
|
|
ret += "&tr=";
|
|
ret += escape_string(i->url.c_str(), i->url.length());
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
#ifndef TORRENT_NO_DEPRECATE
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
torrent_handle add_magnet_uri(session& ses, std::string const& uri
|
|
, std::string const& save_path
|
|
, storage_mode_t storage_mode
|
|
, bool paused
|
|
, storage_constructor_type sc
|
|
, void* userdata)
|
|
{
|
|
std::string name;
|
|
std::string tracker;
|
|
|
|
error_code ec;
|
|
std::string display_name = url_has_argument(uri, "dn");
|
|
if (!display_name.empty()) name = unescape_string(display_name.c_str(), ec);
|
|
std::string tracker_string = url_has_argument(uri, "tr");
|
|
if (!tracker_string.empty()) tracker = unescape_string(tracker_string.c_str(), ec);
|
|
|
|
std::string btih = url_has_argument(uri, "xt");
|
|
if (btih.empty()) return torrent_handle();
|
|
|
|
if (btih.compare(0, 9, "urn:btih:") != 0) return torrent_handle();
|
|
|
|
sha1_hash info_hash;
|
|
if (btih.size() == 40 + 9) from_hex(&btih[9], 40, (char*)&info_hash[0]);
|
|
else info_hash.assign(base32decode(btih.substr(9)));
|
|
|
|
return ses.add_torrent(tracker.empty() ? 0 : tracker.c_str(), info_hash
|
|
, name.empty() ? 0 : name.c_str(), save_path, entry()
|
|
, storage_mode, paused, sc, userdata);
|
|
}
|
|
|
|
torrent_handle add_magnet_uri(session& ses, std::string const& uri
|
|
, add_torrent_params p)
|
|
{
|
|
error_code ec;
|
|
torrent_handle ret = add_magnet_uri(ses, uri, p, ec);
|
|
if (ec) throw libtorrent_exception(ec);
|
|
return ret;
|
|
}
|
|
#endif // BOOST_NO_EXCEPTIONS
|
|
|
|
torrent_handle add_magnet_uri(session& ses, std::string const& uri
|
|
, add_torrent_params p, error_code& ec)
|
|
{
|
|
parse_magnet_uri(uri, p, ec);
|
|
if (ec) return torrent_handle();
|
|
return ses.add_torrent(p, ec);
|
|
}
|
|
|
|
#endif // TORRENT_NO_DEPRECATE
|
|
|
|
void parse_magnet_uri(std::string const& uri, add_torrent_params& p, error_code& ec)
|
|
{
|
|
ec.clear();
|
|
std::string name;
|
|
std::string tracker;
|
|
|
|
error_code e;
|
|
std::string display_name = url_has_argument(uri, "dn");
|
|
if (!display_name.empty()) name = unescape_string(display_name.c_str(), e);
|
|
|
|
// parse trackers out of the magnet link
|
|
std::string::size_type pos = std::string::npos;
|
|
std::string url = url_has_argument(uri, "tr", &pos);
|
|
while (pos != std::string::npos)
|
|
{
|
|
error_code e;
|
|
url = unescape_string(url, e);
|
|
if (e) continue;
|
|
p.trackers.push_back(url);
|
|
pos = uri.find("&tr=", pos);
|
|
if (pos == std::string::npos) break;
|
|
pos += 4;
|
|
url = uri.substr(pos, uri.find('&', pos) - pos);
|
|
}
|
|
|
|
std::string btih = url_has_argument(uri, "xt");
|
|
if (btih.empty())
|
|
{
|
|
ec = errors::missing_info_hash_in_uri;
|
|
return;
|
|
}
|
|
|
|
if (btih.compare(0, 9, "urn:btih:") != 0)
|
|
{
|
|
ec = errors::missing_info_hash_in_uri;
|
|
return;
|
|
}
|
|
|
|
#ifndef TORRENT_DISABLE_DHT
|
|
std::string::size_type node_pos = std::string::npos;
|
|
std::string node = url_has_argument(uri, "dht", &node_pos);
|
|
while (!node.empty())
|
|
{
|
|
std::string::size_type divider = node.find_last_of(':');
|
|
if (divider != std::string::npos)
|
|
{
|
|
int port = atoi(node.c_str()+divider+1);
|
|
if (port != 0)
|
|
p.dht_nodes.push_back(std::make_pair(node.substr(0, divider), port));
|
|
}
|
|
|
|
node_pos = uri.find("&dht=", node_pos);
|
|
if (node_pos == std::string::npos) break;
|
|
node_pos += 5;
|
|
node = uri.substr(node_pos, uri.find('&', node_pos) - node_pos);
|
|
}
|
|
#endif
|
|
|
|
sha1_hash info_hash;
|
|
if (btih.size() == 40 + 9) from_hex(&btih[9], 40, (char*)&info_hash[0]);
|
|
else info_hash.assign(base32decode(btih.substr(9)));
|
|
|
|
p.info_hash = info_hash;
|
|
if (!name.empty()) p.name = name;
|
|
}
|
|
}
|
|
|
|
|