146 lines
4.1 KiB
C++
146 lines
4.1 KiB
C++
/*
|
|
|
|
Copyright (c) 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 "test.hpp"
|
|
#include "libtorrent/file_storage.hpp"
|
|
#include "libtorrent/torrent_info.hpp"
|
|
#include "libtorrent/create_torrent.hpp"
|
|
#include <boost/make_shared.hpp>
|
|
|
|
using namespace libtorrent;
|
|
|
|
void test_storage()
|
|
{
|
|
file_storage fs;
|
|
|
|
fs.add_file("test/temporary.txt", 0x4000);
|
|
fs.add_file("test/A/tmp", 0x4000);
|
|
fs.add_file("test/Temporary.txt", 0x4000);
|
|
fs.add_file("test/TeMPorArY.txT", 0x4000);
|
|
fs.add_file("test/a", 0x4000);
|
|
fs.add_file("test/b.exe", 0x4000);
|
|
fs.add_file("test/B.ExE", 0x4000);
|
|
fs.add_file("test/B.exe", 0x4000);
|
|
fs.add_file("test/test/TEMPORARY.TXT", 0x4000);
|
|
fs.add_file("test/A", 0x4000);
|
|
|
|
libtorrent::create_torrent t(fs, 0x4000);
|
|
|
|
// calculate the hash for all pieces
|
|
int num = t.num_pieces();
|
|
sha1_hash ph;
|
|
for (int i = 0; i < num; ++i)
|
|
t.set_hash(i, ph);
|
|
|
|
std::vector<char> tmp;
|
|
std::back_insert_iterator<std::vector<char> > out(tmp);
|
|
|
|
entry tor = t.generate();
|
|
bencode(out, tor);
|
|
|
|
torrent_info ti(&tmp[0], tmp.size());
|
|
|
|
char const* filenames[10] =
|
|
{
|
|
"test/temporary.txt",
|
|
"test/A/tmp",
|
|
"test/Temporary.1.txt", // duplicate of temporary.txt
|
|
"test/TeMPorArY.2.txT", // duplicate of temporary.txt
|
|
"test/a.1", // a file may not have the same name as a directory
|
|
"test/b.exe",
|
|
"test/B.1.ExE", // duplicate of b.exe
|
|
"test/B.2.exe", // duplicate of b.exe
|
|
"test/test/TEMPORARY.TXT", // a file with the same name in a seprate directory is fine
|
|
"test/A.2", // duplicate of directory a
|
|
};
|
|
|
|
for (int i = 0; i < ti.num_files(); ++i)
|
|
{
|
|
std::string p = ti.file_at(i).path;
|
|
convert_path_to_posix(p);
|
|
fprintf(stderr, "%s == %s\n", p.c_str(), filenames[i]);
|
|
TEST_CHECK(p == filenames[i]);
|
|
}
|
|
}
|
|
|
|
void test_copy()
|
|
{
|
|
using namespace libtorrent;
|
|
|
|
boost::shared_ptr<torrent_info> a(boost::make_shared<torrent_info>(
|
|
combine_path(parent_path(current_working_directory())
|
|
, combine_path("test_torrents", "sample.torrent"))));
|
|
|
|
boost::shared_ptr<torrent_info> b(boost::make_shared<torrent_info>(*a));
|
|
|
|
// clear out the buffer for a, just to make sure b doesn't have any
|
|
// references into it by mistake
|
|
int s = a->metadata_size();
|
|
memset(a->metadata().get(), 0, s);
|
|
|
|
a.reset();
|
|
|
|
TEST_EQUAL(b->num_files(), 3);
|
|
|
|
char const* expected_files[] =
|
|
{
|
|
"sample/text_file2.txt",
|
|
"sample/.____padding_file/0",
|
|
"sample/text_file.txt",
|
|
};
|
|
|
|
sha1_hash file_hashes[] =
|
|
{
|
|
sha1_hash(0),
|
|
sha1_hash(0),
|
|
sha1_hash("abababababababababab")
|
|
};
|
|
|
|
for (int i = 0; i < b->num_files(); ++i)
|
|
{
|
|
std::string p = b->file_at(i).path;
|
|
convert_path_to_posix(p);
|
|
TEST_EQUAL(p, expected_files[i]);
|
|
fprintf(stderr, "%s\n", p.c_str());
|
|
|
|
TEST_EQUAL(b->files().hash(i), file_hashes[i]);
|
|
}
|
|
|
|
}
|
|
|
|
int test_main()
|
|
{
|
|
test_storage();
|
|
test_copy();
|
|
return 0;
|
|
}
|