354 lines
8.6 KiB
C++
354 lines
8.6 KiB
C++
/*
|
|
|
|
Copyright (c) 2003, 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/entry.hpp"
|
|
#include "libtorrent/bencode.hpp"
|
|
#include "libtorrent/torrent_info.hpp"
|
|
#include "libtorrent/lazy_entry.hpp"
|
|
#include "libtorrent/magnet_uri.hpp"
|
|
|
|
int load_file(std::string const& filename, std::vector<char>& v, libtorrent::error_code& ec, int limit = 8000000)
|
|
{
|
|
ec.clear();
|
|
FILE* f = fopen(filename.c_str(), "rb");
|
|
if (f == NULL)
|
|
{
|
|
ec.assign(errno, boost::system::get_generic_category());
|
|
return -1;
|
|
}
|
|
|
|
int r = fseek(f, 0, SEEK_END);
|
|
if (r != 0)
|
|
{
|
|
ec.assign(errno, boost::system::get_generic_category());
|
|
fclose(f);
|
|
return -1;
|
|
}
|
|
long s = ftell(f);
|
|
if (s < 0)
|
|
{
|
|
ec.assign(errno, boost::system::get_generic_category());
|
|
fclose(f);
|
|
return -1;
|
|
}
|
|
|
|
if (s > limit)
|
|
{
|
|
fclose(f);
|
|
return -2;
|
|
}
|
|
|
|
r = fseek(f, 0, SEEK_SET);
|
|
if (r != 0)
|
|
{
|
|
ec.assign(errno, boost::system::get_generic_category());
|
|
fclose(f);
|
|
return -1;
|
|
}
|
|
|
|
v.resize(s);
|
|
if (s == 0)
|
|
{
|
|
fclose(f);
|
|
return 0;
|
|
}
|
|
|
|
r = fread(&v[0], 1, v.size(), f);
|
|
if (r < 0)
|
|
{
|
|
ec.assign(errno, boost::system::get_generic_category());
|
|
fclose(f);
|
|
return -1;
|
|
}
|
|
|
|
fclose(f);
|
|
|
|
if (r != s) return -3;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int line_longer_than(libtorrent::lazy_entry const& e, int limit)
|
|
{
|
|
using namespace libtorrent;
|
|
|
|
int line_len = 0;
|
|
switch (e.type())
|
|
{
|
|
case lazy_entry::list_t:
|
|
line_len += 4;
|
|
if (line_len > limit) return -1;
|
|
for (int i = 0; i < e.list_size(); ++i)
|
|
{
|
|
int ret = line_longer_than(*e.list_at(i), limit - line_len);
|
|
if (ret == -1) return -1;
|
|
line_len += ret + 2;
|
|
}
|
|
break;
|
|
case lazy_entry::dict_t:
|
|
line_len += 4;
|
|
if (line_len > limit) return -1;
|
|
for (int i = 0; i < e.dict_size(); ++i)
|
|
{
|
|
line_len += 4 + e.dict_at(i).first.size();
|
|
if (line_len > limit) return -1;
|
|
int ret = line_longer_than(*e.dict_at(i).second, limit - line_len);
|
|
if (ret == -1) return -1;
|
|
line_len += ret + 1;
|
|
}
|
|
break;
|
|
case lazy_entry::string_t:
|
|
line_len += 3 + e.string_length();
|
|
break;
|
|
case lazy_entry::int_t:
|
|
{
|
|
size_type val = e.int_value();
|
|
while (val > 0)
|
|
{
|
|
++line_len;
|
|
val /= 10;
|
|
}
|
|
line_len += 2;
|
|
}
|
|
break;
|
|
case lazy_entry::none_t:
|
|
line_len += 4;
|
|
break;
|
|
}
|
|
|
|
if (line_len > limit) return -1;
|
|
return line_len;
|
|
}
|
|
|
|
std::string print_entry(libtorrent::lazy_entry const& e, bool single_line = false, int indent = 0)
|
|
{
|
|
using namespace libtorrent;
|
|
|
|
char indent_str[200];
|
|
memset(indent_str, ' ', 200);
|
|
indent_str[0] = ',';
|
|
indent_str[1] = '\n';
|
|
indent_str[199] = 0;
|
|
if (indent < 197 && indent >= 0) indent_str[indent+2] = 0;
|
|
std::string ret;
|
|
switch (e.type())
|
|
{
|
|
case lazy_entry::none_t: return "none";
|
|
case lazy_entry::int_t:
|
|
{
|
|
char str[100];
|
|
snprintf(str, sizeof(str), "%"PRId64, e.int_value());
|
|
return str;
|
|
}
|
|
case lazy_entry::string_t:
|
|
{
|
|
bool printable = true;
|
|
char const* str = e.string_ptr();
|
|
for (int i = 0; i < e.string_length(); ++i)
|
|
{
|
|
using namespace std;
|
|
if (is_print((unsigned char)str[i])) continue;
|
|
printable = false;
|
|
break;
|
|
}
|
|
ret += "'";
|
|
if (printable)
|
|
{
|
|
ret += e.string_value();
|
|
ret += "'";
|
|
return ret;
|
|
}
|
|
for (int i = 0; i < e.string_length(); ++i)
|
|
{
|
|
char tmp[5];
|
|
snprintf(tmp, sizeof(tmp), "%02x", (unsigned char)str[i]);
|
|
ret += tmp;
|
|
}
|
|
ret += "'";
|
|
return ret;
|
|
}
|
|
case lazy_entry::list_t:
|
|
{
|
|
ret += '[';
|
|
bool one_liner = line_longer_than(e, 200) != -1 || single_line;
|
|
|
|
if (!one_liner) ret += indent_str + 1;
|
|
for (int i = 0; i < e.list_size(); ++i)
|
|
{
|
|
if (i == 0 && one_liner) ret += " ";
|
|
ret += ::print_entry(*e.list_at(i), single_line, indent + 2);
|
|
if (i < e.list_size() - 1) ret += (one_liner?", ":indent_str);
|
|
else ret += (one_liner?" ":indent_str+1);
|
|
}
|
|
ret += "]";
|
|
return ret;
|
|
}
|
|
case lazy_entry::dict_t:
|
|
{
|
|
ret += "{";
|
|
bool one_liner = line_longer_than(e, 200) != -1 || single_line;
|
|
|
|
if (!one_liner) ret += indent_str+1;
|
|
for (int i = 0; i < e.dict_size(); ++i)
|
|
{
|
|
if (i == 0 && one_liner) ret += " ";
|
|
std::pair<std::string, lazy_entry const*> ent = e.dict_at(i);
|
|
ret += "'";
|
|
ret += ent.first;
|
|
ret += "': ";
|
|
ret += ::print_entry(*ent.second, single_line, indent + 2);
|
|
if (i < e.dict_size() - 1) ret += (one_liner?", ":indent_str);
|
|
else ret += (one_liner?" ":indent_str+1);
|
|
}
|
|
ret += "}";
|
|
return ret;
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
using namespace libtorrent;
|
|
|
|
if (argc < 2 || argc > 4)
|
|
{
|
|
fputs("usage: dump_torrent torrent-file [total-items-limit] [recursion-limit]\n", stderr);
|
|
return 1;
|
|
}
|
|
|
|
int item_limit = 1000000;
|
|
int depth_limit = 1000;
|
|
|
|
if (argc > 2) item_limit = atoi(argv[2]);
|
|
if (argc > 3) depth_limit = atoi(argv[3]);
|
|
|
|
int size = file_size(argv[1]);
|
|
if (size > 40 * 1000000)
|
|
{
|
|
fprintf(stderr, "file too big (%d), aborting\n", size);
|
|
return 1;
|
|
}
|
|
std::vector<char> buf(size);
|
|
error_code ec;
|
|
int ret = load_file(argv[1], buf, ec, 40 * 1000000);
|
|
if (ret != 0)
|
|
{
|
|
fprintf(stderr, "failed to load file: %s\n", ec.message().c_str());
|
|
return 1;
|
|
}
|
|
lazy_entry e;
|
|
int pos;
|
|
printf("decoding. recursion limit: %d total item count limit: %d\n"
|
|
, depth_limit, item_limit);
|
|
ret = lazy_bdecode(&buf[0], &buf[0] + buf.size(), e, ec, &pos
|
|
, depth_limit, item_limit);
|
|
|
|
printf("\n\n----- raw info -----\n\n%s\n", ::print_entry(e).c_str());
|
|
|
|
if (ret != 0)
|
|
{
|
|
fprintf(stderr, "failed to decode: '%s' at character: %d\n", ec.message().c_str(), pos);
|
|
return 1;
|
|
}
|
|
|
|
torrent_info t(e, ec);
|
|
if (ec)
|
|
{
|
|
fprintf(stderr, "%s\n", ec.message().c_str());
|
|
return 1;
|
|
}
|
|
e.clear();
|
|
std::vector<char>().swap(buf);
|
|
|
|
// print info about torrent
|
|
printf("\n\n----- torrent file info -----\n\n"
|
|
"nodes:\n");
|
|
|
|
typedef std::vector<std::pair<std::string, int> > node_vec;
|
|
node_vec const& nodes = t.nodes();
|
|
for (node_vec::const_iterator i = nodes.begin(), end(nodes.end());
|
|
i != end; ++i)
|
|
{
|
|
printf("%s: %d\n", i->first.c_str(), i->second);
|
|
}
|
|
puts("trackers:\n");
|
|
for (std::vector<announce_entry>::const_iterator i = t.trackers().begin();
|
|
i != t.trackers().end(); ++i)
|
|
{
|
|
printf("%2d: %s\n", i->tier, i->url.c_str());
|
|
}
|
|
|
|
char ih[41];
|
|
to_hex((char const*)&t.info_hash()[0], 20, ih);
|
|
printf("number of pieces: %d\n"
|
|
"piece length: %d\n"
|
|
"info hash: %s\n"
|
|
"comment: %s\n"
|
|
"created by: %s\n"
|
|
"magnet link: %s\n"
|
|
"name: %s\n"
|
|
"number of files: %d\n"
|
|
"files:\n"
|
|
, t.num_pieces()
|
|
, t.piece_length()
|
|
, ih
|
|
, t.comment().c_str()
|
|
, t.creator().c_str()
|
|
, make_magnet_uri(t).c_str()
|
|
, t.name().c_str()
|
|
, t.num_files());
|
|
int index = 0;
|
|
for (torrent_info::file_iterator i = t.begin_files();
|
|
i != t.end_files(); ++i, ++index)
|
|
{
|
|
int first = t.map_file(index, 0, 0).piece;
|
|
int last = t.map_file(index, (std::max)(size_type(i->size)-1, size_type(0)), 0).piece;
|
|
printf(" %8"PRIx64" %11"PRId64" %c%c%c%c [ %5d, %5d ] %7u %s %s %s%s\n"
|
|
, i->offset
|
|
, i->size
|
|
, (i->pad_file?'p':'-')
|
|
, (i->executable_attribute?'x':'-')
|
|
, (i->hidden_attribute?'h':'-')
|
|
, (i->symlink_attribute?'l':'-')
|
|
, first, last
|
|
, boost::uint32_t(t.files().mtime(*i))
|
|
, t.files().hash(*i) != sha1_hash(0) ? to_hex(t.files().hash(*i).to_string()).c_str() : ""
|
|
, t.files().file_path(*i).c_str()
|
|
, i->symlink_attribute ? "-> ": ""
|
|
, i->symlink_attribute && i->symlink_index != -1 ? t.files().symlink(*i).c_str() : "");
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|