2008-07-01 20:59:13 +02:00
|
|
|
/*
|
|
|
|
|
|
|
|
Copyright (c) 2008, 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.
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
2005-08-11 13:06:52 +02:00
|
|
|
#include <iostream>
|
2008-11-05 04:17:41 +01:00
|
|
|
#include <boost/config.hpp>
|
2010-12-24 02:20:39 +01:00
|
|
|
#include <fcntl.h>
|
2011-03-01 22:57:03 +01:00
|
|
|
#include <stdio.h>
|
2013-01-06 19:38:33 +01:00
|
|
|
#include <stdlib.h> // for exit()
|
2014-07-06 21:18:00 +02:00
|
|
|
#include "libtorrent/address.hpp"
|
|
|
|
#include "libtorrent/socket.hpp"
|
2015-05-30 06:31:23 +02:00
|
|
|
#include "setup_transfer.hpp" // for _g_test_failures
|
2014-07-06 21:18:00 +02:00
|
|
|
#include "test.hpp"
|
2013-08-04 01:30:37 +02:00
|
|
|
#include "dht_server.hpp" // for stop_dht
|
|
|
|
#include "peer_server.hpp" // for stop_peer
|
2014-02-23 02:32:55 +01:00
|
|
|
#include "udp_tracker.hpp" // for stop_udp_tracker
|
2005-08-11 13:06:52 +02:00
|
|
|
|
2012-04-21 07:54:49 +02:00
|
|
|
#include "libtorrent/assert.hpp"
|
2013-06-17 02:45:17 +02:00
|
|
|
#include "libtorrent/file.hpp"
|
2012-04-21 07:54:49 +02:00
|
|
|
#include <signal.h>
|
|
|
|
|
2013-08-04 05:04:59 +02:00
|
|
|
#ifdef WIN32
|
|
|
|
#include <windows.h> // fot SetErrorMode
|
2015-05-30 23:46:59 +02:00
|
|
|
#include <io.h> // for _dup and _dup2
|
|
|
|
|
|
|
|
#define dup _dup
|
|
|
|
#define dup2 _dup2
|
|
|
|
|
2013-08-04 05:04:59 +02:00
|
|
|
#endif
|
|
|
|
|
2014-07-06 21:18:00 +02:00
|
|
|
using namespace libtorrent;
|
|
|
|
|
2015-05-31 18:14:46 +02:00
|
|
|
// these are global so we can restore them on abnormal exits and print stuff
|
|
|
|
// out, such as the log
|
|
|
|
int old_stdout = -1;
|
|
|
|
int old_stderr = -1;
|
2015-06-14 00:52:17 +02:00
|
|
|
bool redirect_output = true;
|
2015-05-31 18:14:46 +02:00
|
|
|
|
|
|
|
// the current tests file descriptor
|
|
|
|
unit_test_t* current_test = NULL;
|
|
|
|
|
|
|
|
void output_test_log_to_terminal()
|
|
|
|
{
|
2015-06-14 00:52:17 +02:00
|
|
|
if (current_test == NULL || old_stdout == -1 || old_stderr == -1
|
|
|
|
|| !redirect_output)
|
2015-05-31 18:14:46 +02:00
|
|
|
return;
|
|
|
|
|
|
|
|
fflush(stdout);
|
|
|
|
fflush(stderr);
|
|
|
|
dup2(old_stdout, fileno(stdout));
|
|
|
|
dup2(old_stderr, fileno(stderr));
|
|
|
|
|
|
|
|
fseek(current_test->output, 0, SEEK_SET);
|
|
|
|
fprintf(stderr, "\x1b[1m[%s]\x1b[0m\n\n", current_test->name);
|
|
|
|
char buf[4096];
|
|
|
|
int size = 0;
|
|
|
|
do {
|
|
|
|
size = fread(buf, 1, sizeof(buf), current_test->output);
|
|
|
|
if (size > 0) fwrite(buf, 1, size, stderr);
|
|
|
|
} while (size > 0);
|
|
|
|
}
|
|
|
|
|
2012-04-21 07:54:49 +02:00
|
|
|
void sig_handler(int sig)
|
|
|
|
{
|
|
|
|
char stack_text[10000];
|
2012-06-05 07:55:40 +02:00
|
|
|
|
2015-05-17 23:00:47 +02:00
|
|
|
#if (defined TORRENT_DEBUG && TORRENT_USE_ASSERTS) \
|
|
|
|
|| defined TORRENT_ASIO_DEBUGGING \
|
|
|
|
|| defined TORRENT_PROFILE_CALLS \
|
|
|
|
|| defined TORRENT_RELEASE_ASSERTS \
|
|
|
|
|| defined TORRENT_DEBUG_BUFFERS
|
2012-04-21 07:54:49 +02:00
|
|
|
print_backtrace(stack_text, sizeof(stack_text), 30);
|
2012-06-05 07:55:40 +02:00
|
|
|
#elif defined __FUNCTION__
|
|
|
|
strcat(stack_text, __FUNCTION__);
|
|
|
|
#else
|
|
|
|
stack_text[0] = 0;
|
|
|
|
#endif
|
2012-04-21 07:54:49 +02:00
|
|
|
char const* sig_name = 0;
|
|
|
|
switch (sig)
|
|
|
|
{
|
|
|
|
#define SIG(x) case x: sig_name = #x; break
|
|
|
|
SIG(SIGSEGV);
|
2012-04-28 20:49:53 +02:00
|
|
|
#ifdef SIGBUS
|
2012-04-21 07:54:49 +02:00
|
|
|
SIG(SIGBUS);
|
2012-04-28 20:49:53 +02:00
|
|
|
#endif
|
2015-06-14 00:52:17 +02:00
|
|
|
SIG(SIGINT);
|
|
|
|
SIG(SIGTERM);
|
2012-04-21 07:54:49 +02:00
|
|
|
SIG(SIGILL);
|
|
|
|
SIG(SIGABRT);
|
|
|
|
SIG(SIGFPE);
|
2012-04-28 20:49:53 +02:00
|
|
|
#ifdef SIGSYS
|
2012-04-21 07:54:49 +02:00
|
|
|
SIG(SIGSYS);
|
2012-04-28 20:49:53 +02:00
|
|
|
#endif
|
2012-04-21 07:54:49 +02:00
|
|
|
#undef SIG
|
|
|
|
};
|
|
|
|
fprintf(stderr, "signal: %s caught:\n%s\n", sig_name, stack_text);
|
2015-05-31 18:14:46 +02:00
|
|
|
|
|
|
|
output_test_log_to_terminal();
|
|
|
|
|
2012-04-21 07:54:49 +02:00
|
|
|
exit(138);
|
|
|
|
}
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
void print_usage(char const* executable)
|
2005-08-11 13:06:52 +02:00
|
|
|
{
|
2015-05-30 06:31:23 +02:00
|
|
|
printf("%s [options] [tests...]\n"
|
|
|
|
"\n"
|
|
|
|
"OPTIONS:\n"
|
|
|
|
"-h,--help show this help\n"
|
|
|
|
"-l,--list list the tests available to run\n"
|
2015-06-14 00:52:17 +02:00
|
|
|
"-n,--no-redirect don't redirect test output to\n"
|
|
|
|
" temporary file, but let it go straight\n"
|
|
|
|
" to stdout\n"
|
2015-05-30 06:31:23 +02:00
|
|
|
"\n"
|
|
|
|
"for tests, specify one or more test names as printed\n"
|
2015-06-14 00:52:17 +02:00
|
|
|
"by -l. If no test is specified, all tests are run\n", executable);
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
|
|
|
|
2015-06-20 18:08:49 +02:00
|
|
|
EXPORT int main(int argc, char const* argv[])
|
2015-05-30 06:31:23 +02:00
|
|
|
{
|
2015-06-14 00:52:17 +02:00
|
|
|
char const* executable = argv[0];
|
|
|
|
// skip executable name
|
|
|
|
++argv;
|
|
|
|
--argc;
|
2015-05-30 06:31:23 +02:00
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
// pick up options
|
|
|
|
while (argc > 0 && argv[0][0] == '-')
|
2015-05-30 06:31:23 +02:00
|
|
|
{
|
2015-06-14 00:52:17 +02:00
|
|
|
if (strcmp(argv[0], "-h") == 0 || strcmp(argv[0], "--help") == 0)
|
2015-05-30 06:31:23 +02:00
|
|
|
{
|
2015-06-14 00:52:17 +02:00
|
|
|
print_usage(executable);
|
|
|
|
return 0;
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
2015-06-14 00:52:17 +02:00
|
|
|
|
|
|
|
if (strcmp(argv[0], "-l") == 0 || strcmp(argv[0], "--list") == 0)
|
|
|
|
{
|
|
|
|
printf("TESTS:\n");
|
|
|
|
for (int i = 0; i < _g_num_unit_tests; ++i)
|
|
|
|
{
|
|
|
|
printf(" - %s\n", _g_unit_tests[i].name);
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (strcmp(argv[0], "-n") == 0 || strcmp(argv[0], "--no-redirect") == 0)
|
|
|
|
{
|
|
|
|
redirect_output = false;
|
|
|
|
}
|
|
|
|
++argv;
|
|
|
|
--argc;
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
std::set<std::string> tests_to_run;
|
|
|
|
bool filter = false;
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
for (int i = 0; i < argc; ++i)
|
2015-05-30 06:31:23 +02:00
|
|
|
{
|
|
|
|
tests_to_run.insert(argv[i]);
|
|
|
|
filter = true;
|
|
|
|
}
|
|
|
|
|
2013-08-04 05:04:59 +02:00
|
|
|
#ifdef WIN32
|
|
|
|
// try to suppress hanging the process by windows displaying
|
|
|
|
// modal dialogs.
|
|
|
|
SetErrorMode(SEM_NOALIGNMENTFAULTEXCEPT | SEM_NOALIGNMENTFAULTEXCEPT
|
|
|
|
| SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX);
|
|
|
|
#endif
|
|
|
|
|
2015-03-22 06:31:55 +01:00
|
|
|
srand(total_microseconds(clock_type::now().time_since_epoch()) & 0x7fffffff);
|
2010-12-24 01:37:01 +01:00
|
|
|
#ifdef O_NONBLOCK
|
|
|
|
// on darwin, stdout is set to non-blocking mode by default
|
|
|
|
// which sometimes causes tests to fail with EAGAIN just
|
|
|
|
// by printing logs
|
2010-12-24 02:20:39 +01:00
|
|
|
int flags = fcntl(fileno(stdout), F_GETFL, 0);
|
|
|
|
fcntl(fileno(stdout), F_SETFL, flags & ~O_NONBLOCK);
|
|
|
|
flags = fcntl(fileno(stderr), F_GETFL, 0);
|
|
|
|
fcntl(fileno(stderr), F_SETFL, flags & ~O_NONBLOCK);
|
2010-12-24 01:37:01 +01:00
|
|
|
#endif
|
|
|
|
|
2012-04-21 07:54:49 +02:00
|
|
|
signal(SIGSEGV, &sig_handler);
|
2012-04-28 20:49:53 +02:00
|
|
|
#ifdef SIGBUS
|
2012-04-21 07:54:49 +02:00
|
|
|
signal(SIGBUS, &sig_handler);
|
2012-04-28 20:49:53 +02:00
|
|
|
#endif
|
2012-04-21 07:54:49 +02:00
|
|
|
signal(SIGILL, &sig_handler);
|
2015-05-31 18:14:46 +02:00
|
|
|
signal(SIGINT, &sig_handler);
|
2012-04-21 07:54:49 +02:00
|
|
|
signal(SIGABRT, &sig_handler);
|
|
|
|
signal(SIGFPE, &sig_handler);
|
2012-04-28 20:49:53 +02:00
|
|
|
#ifdef SIGSYS
|
2012-04-21 07:54:49 +02:00
|
|
|
signal(SIGSYS, &sig_handler);
|
2012-04-28 20:49:53 +02:00
|
|
|
#endif
|
2012-04-21 07:54:49 +02:00
|
|
|
|
2013-06-17 02:45:17 +02:00
|
|
|
char dir[40];
|
|
|
|
snprintf(dir, sizeof(dir), "test_tmp_%u", rand());
|
|
|
|
std::string test_dir = complete(dir);
|
|
|
|
error_code ec;
|
|
|
|
create_directory(test_dir, ec);
|
|
|
|
if (ec)
|
|
|
|
{
|
|
|
|
fprintf(stderr, "Failed to create test directory: %s\n", ec.message().c_str());
|
|
|
|
return 1;
|
|
|
|
}
|
2013-06-17 06:12:18 +02:00
|
|
|
#ifdef TORRENT_WINDOWS
|
2013-06-17 07:50:53 +02:00
|
|
|
SetCurrentDirectoryA(dir);
|
2013-06-17 06:12:18 +02:00
|
|
|
#else
|
2013-06-17 02:45:17 +02:00
|
|
|
chdir(dir);
|
2013-06-17 06:12:18 +02:00
|
|
|
#endif
|
2014-01-24 10:31:47 +01:00
|
|
|
fprintf(stderr, "cwd = \"%s\"\n", test_dir.c_str());
|
2013-06-17 02:45:17 +02:00
|
|
|
|
2015-05-30 06:31:23 +02:00
|
|
|
int total_failures = 0;
|
|
|
|
|
|
|
|
if (_g_num_unit_tests == 0)
|
|
|
|
{
|
|
|
|
fprintf(stderr, "\x1b[31mERROR: no unit tests registered\x1b[0m\n");
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
if (redirect_output)
|
|
|
|
{
|
|
|
|
old_stdout = dup(fileno(stdout));
|
|
|
|
old_stderr = dup(fileno(stderr));
|
|
|
|
}
|
2015-05-30 06:31:23 +02:00
|
|
|
|
|
|
|
int num_run = 0;
|
|
|
|
for (int i = 0; i < _g_num_unit_tests; ++i)
|
2005-08-11 13:06:52 +02:00
|
|
|
{
|
2015-05-30 06:31:23 +02:00
|
|
|
if (filter && tests_to_run.count(_g_unit_tests[i].name) == 0)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
unit_test_t& t = _g_unit_tests[i];
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
if (redirect_output)
|
2015-05-30 06:31:23 +02:00
|
|
|
{
|
2015-06-14 00:52:17 +02:00
|
|
|
// redirect test output to a temporary file
|
|
|
|
fflush(stdout);
|
|
|
|
fflush(stderr);
|
|
|
|
|
|
|
|
t.output = tmpfile();
|
|
|
|
int ret1 = dup2(fileno(t.output), fileno(stdout));
|
|
|
|
int ret2 = dup2(fileno(t.output), fileno(stderr));
|
|
|
|
if (ret1 < 0 /*|| ret2 < 0*/)
|
|
|
|
{
|
|
|
|
fprintf(stderr, "failed to redirect output: (%d) %s\n"
|
|
|
|
, errno, strerror(errno));
|
|
|
|
continue;
|
|
|
|
}
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
|
|
|
|
2015-05-31 18:14:46 +02:00
|
|
|
current_test = &t;
|
|
|
|
|
2015-05-30 06:31:23 +02:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
|
|
try
|
|
|
|
{
|
2008-11-03 05:23:16 +01:00
|
|
|
#endif
|
2015-05-30 06:31:23 +02:00
|
|
|
|
|
|
|
_g_test_failures = 0;
|
|
|
|
(*t.fun)();
|
2008-11-03 05:23:16 +01:00
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
|
|
|
catch (std::exception const& e)
|
|
|
|
{
|
|
|
|
char buf[200];
|
|
|
|
snprintf(buf, sizeof(buf), "Terminated with exception: \"%s\"", e.what());
|
|
|
|
report_failure(buf, __FILE__, __LINE__);
|
|
|
|
}
|
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
report_failure("Terminated with unknown exception", __FILE__, __LINE__);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (!tests_to_run.empty()) tests_to_run.erase(t.name);
|
|
|
|
|
|
|
|
if (_g_test_failures > 0)
|
|
|
|
{
|
2015-05-31 18:14:46 +02:00
|
|
|
output_test_log_to_terminal();
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
t.num_failures = _g_test_failures;
|
|
|
|
t.run = true;
|
|
|
|
total_failures += _g_test_failures;
|
|
|
|
++num_run;
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
if (redirect_output)
|
|
|
|
{
|
|
|
|
fclose(t.output);
|
|
|
|
}
|
2005-08-11 13:06:52 +02:00
|
|
|
}
|
2015-05-30 06:31:23 +02:00
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
if (redirect_output)
|
|
|
|
{
|
|
|
|
dup2(old_stdout, fileno(stdout));
|
|
|
|
dup2(old_stderr, fileno(stderr));
|
|
|
|
}
|
2015-05-30 06:31:23 +02:00
|
|
|
|
|
|
|
if (!tests_to_run.empty())
|
2005-08-11 13:06:52 +02:00
|
|
|
{
|
2015-05-30 23:10:34 +02:00
|
|
|
fprintf(stderr, "\x1b[1mUNKONWN tests:\x1b[0m\n");
|
2015-05-30 06:31:23 +02:00
|
|
|
for (std::set<std::string>::iterator i = tests_to_run.begin()
|
|
|
|
, end(tests_to_run.end()); i != end; ++i)
|
|
|
|
{
|
2015-05-30 23:10:34 +02:00
|
|
|
fprintf(stderr, " %s\n", i->c_str());
|
2015-05-30 06:31:23 +02:00
|
|
|
}
|
2005-08-11 13:06:52 +02:00
|
|
|
}
|
2015-05-30 06:31:23 +02:00
|
|
|
|
|
|
|
if (num_run == 0)
|
2005-08-11 13:06:52 +02:00
|
|
|
{
|
2015-05-30 06:31:23 +02:00
|
|
|
fprintf(stderr, "\x1b[31mERROR: no unit tests run\x1b[0m\n");
|
|
|
|
return 1;
|
2005-08-11 13:06:52 +02:00
|
|
|
}
|
2013-08-04 01:30:37 +02:00
|
|
|
|
|
|
|
// just in case of premature exits
|
|
|
|
// make sure we try to clean up some
|
2014-02-23 02:32:55 +01:00
|
|
|
stop_udp_tracker();
|
2013-08-04 01:30:37 +02:00
|
|
|
stop_all_proxies();
|
|
|
|
stop_web_server();
|
|
|
|
stop_peer();
|
|
|
|
stop_dht();
|
|
|
|
|
2015-06-14 00:52:17 +02:00
|
|
|
if (redirect_output)
|
|
|
|
{
|
|
|
|
fflush(stdout);
|
|
|
|
fflush(stderr);
|
|
|
|
}
|
2013-08-27 09:02:46 +02:00
|
|
|
|
2013-08-31 22:19:03 +02:00
|
|
|
int ret = print_failures();
|
2015-01-08 23:54:11 +01:00
|
|
|
#if !defined TORRENT_LOGGING
|
2013-08-31 22:19:03 +02:00
|
|
|
if (ret == 0)
|
|
|
|
{
|
|
|
|
remove_all(test_dir, ec);
|
|
|
|
if (ec)
|
|
|
|
fprintf(stderr, "failed to remove test dir: %s\n", ec.message().c_str());
|
|
|
|
}
|
2013-10-14 03:04:02 +02:00
|
|
|
#endif
|
2013-08-27 09:02:46 +02:00
|
|
|
|
2015-05-30 06:31:23 +02:00
|
|
|
return total_failures ? 333 : 0;
|
2005-08-11 13:06:52 +02:00
|
|
|
}
|
|
|
|
|