mirror of
https://git.wownero.com/wownero/wownero.git
synced 2024-12-22 22:18:53 +00:00
Merge remote-tracking branch 'monero-official/master' into network-1.6-work1
This commit is contained in:
commit
3cbdf198f1
@ -238,12 +238,18 @@ if (${BOOST_IGNORE_SYSTEM_PATHS} STREQUAL "ON")
|
|||||||
set(Boost_NO_SYSTEM_PATHS TRUE)
|
set(Boost_NO_SYSTEM_PATHS TRUE)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
set(OLD_LIB_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES})
|
||||||
if(STATIC)
|
if(STATIC)
|
||||||
|
if(MINGW)
|
||||||
|
set(CMAKE_FIND_LIBRARY_SUFFIXES .a)
|
||||||
|
endif()
|
||||||
|
|
||||||
set(Boost_USE_STATIC_LIBS ON)
|
set(Boost_USE_STATIC_LIBS ON)
|
||||||
set(Boost_USE_STATIC_RUNTIME ON)
|
set(Boost_USE_STATIC_RUNTIME ON)
|
||||||
endif()
|
endif()
|
||||||
find_package(Boost 1.53 QUIET REQUIRED COMPONENTS system filesystem thread date_time chrono regex serialization program_options)
|
find_package(Boost 1.53 QUIET REQUIRED COMPONENTS system filesystem thread date_time chrono regex serialization program_options)
|
||||||
|
|
||||||
|
set(CMAKE_FIND_LIBRARY_SUFFIXES ${OLD_LIB_SUFFIXES})
|
||||||
if(NOT Boost_FOUND)
|
if(NOT Boost_FOUND)
|
||||||
die("Could not find Boost libraries, please make sure you have installed Boost or libboost-all-dev (1.53 or 1.55+) or the equivalent")
|
die("Could not find Boost libraries, please make sure you have installed Boost or libboost-all-dev (1.53 or 1.55+) or the equivalent")
|
||||||
endif()
|
endif()
|
||||||
|
@ -4,8 +4,9 @@ Copyright (c) 2014-2015, The Monero Project
|
|||||||
|
|
||||||
## Development Resources
|
## Development Resources
|
||||||
|
|
||||||
Web: [monero.cc](http://monero.cc)
|
Web: [getmonero.org](https://getmonero.org)
|
||||||
Mail: [dev@monero.cc](mailto:dev@monero.cc)
|
Forum: [forum.getmonero.org](https://forum.getmonero.org)
|
||||||
|
Mail: [dev@getmonero.org](mailto:dev@getmonero.org)
|
||||||
Github (staging): [https://github.com/monero-project/bitmonero](https://github.com/monero-project/bitmonero)
|
Github (staging): [https://github.com/monero-project/bitmonero](https://github.com/monero-project/bitmonero)
|
||||||
Github (development): [http://github.com/monero-project/bitmonero/tree/development](http://github.com/monero-project/bitmonero/tree/development)
|
Github (development): [http://github.com/monero-project/bitmonero/tree/development](http://github.com/monero-project/bitmonero/tree/development)
|
||||||
IRC: [#monero-dev on Freenode](irc://chat.freenode.net/#monero-dev)
|
IRC: [#monero-dev on Freenode](irc://chat.freenode.net/#monero-dev)
|
||||||
@ -32,7 +33,7 @@ Anyone is able to contribute to Monero. If you have a fix or code change, feel f
|
|||||||
|
|
||||||
Monero development can be supported directly through donations.
|
Monero development can be supported directly through donations.
|
||||||
|
|
||||||
Both Monero and Bitcoin donations can be made to donate.monero.cc if using a client that supports the [OpenAlias](https://openalias.org) standard
|
Both Monero and Bitcoin donations can be made to donate.getmonero.org if using a client that supports the [OpenAlias](https://openalias.org) standard
|
||||||
|
|
||||||
The Monero donation address is: 46BeWrHpwXmHDpDEUmZBWZfoQpdc6HaERCNmx1pEYL2rAcuwufPN9rXHHtyUA4QVy66qeFQkn6sfK8aHYjA3jk3o1Bv16em (viewkey: e422831985c9205238ef84daf6805526c14d96fd7b059fe68c7ab98e495e5703)
|
The Monero donation address is: 46BeWrHpwXmHDpDEUmZBWZfoQpdc6HaERCNmx1pEYL2rAcuwufPN9rXHHtyUA4QVy66qeFQkn6sfK8aHYjA3jk3o1Bv16em (viewkey: e422831985c9205238ef84daf6805526c14d96fd7b059fe68c7ab98e495e5703)
|
||||||
|
|
||||||
|
@ -26,11 +26,13 @@
|
|||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#include "misc_log_ex.h"
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
#include <condition_variable>
|
#include <condition_variable>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
namespace epee
|
namespace epee
|
||||||
{
|
{
|
||||||
@ -267,17 +269,19 @@ namespace epee
|
|||||||
string_tools::trim(command);
|
string_tools::trim(command);
|
||||||
|
|
||||||
LOG_PRINT_L2("Read command: " << command);
|
LOG_PRINT_L2("Read command: " << command);
|
||||||
if(0 == command.compare("exit") || 0 == command.compare("q"))
|
if (command.empty())
|
||||||
{
|
|
||||||
continue_handle = false;
|
|
||||||
}else if (command.empty())
|
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
else if(cmd_handler(command))
|
else if(cmd_handler(command))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
} else
|
}
|
||||||
|
else if(0 == command.compare("exit") || 0 == command.compare("q"))
|
||||||
|
{
|
||||||
|
continue_handle = false;
|
||||||
|
}
|
||||||
|
else
|
||||||
{
|
{
|
||||||
std::cout << "unknown command: " << command << std::endl;
|
std::cout << "unknown command: " << command << std::endl;
|
||||||
std::cout << usage;
|
std::cout << usage;
|
||||||
@ -290,7 +294,7 @@ namespace epee
|
|||||||
|
|
||||||
private:
|
private:
|
||||||
async_stdin_reader m_stdin_reader;
|
async_stdin_reader m_stdin_reader;
|
||||||
bool m_running = true;
|
std::atomic<bool> m_running = {true};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -350,17 +354,11 @@ namespace epee
|
|||||||
return true;
|
return true;
|
||||||
}*/
|
}*/
|
||||||
|
|
||||||
/************************************************************************/
|
class command_handler {
|
||||||
/* */
|
|
||||||
/************************************************************************/
|
|
||||||
class console_handlers_binder
|
|
||||||
{
|
|
||||||
typedef boost::function<bool (const std::vector<std::string> &)> console_command_handler;
|
|
||||||
typedef std::map<std::string, std::pair<console_command_handler, std::string> > command_handlers_map;
|
|
||||||
std::unique_ptr<boost::thread> m_console_thread;
|
|
||||||
command_handlers_map m_command_handlers;
|
|
||||||
async_console_handler m_console_handler;
|
|
||||||
public:
|
public:
|
||||||
|
typedef boost::function<bool (const std::vector<std::string> &)> callback;
|
||||||
|
typedef std::map<std::string, std::pair<callback, std::string> > lookup;
|
||||||
|
|
||||||
std::string get_usage()
|
std::string get_usage()
|
||||||
{
|
{
|
||||||
std::stringstream ss;
|
std::stringstream ss;
|
||||||
@ -376,12 +374,14 @@ namespace epee
|
|||||||
}
|
}
|
||||||
return ss.str();
|
return ss.str();
|
||||||
}
|
}
|
||||||
void set_handler(const std::string& cmd, const console_command_handler& hndlr, const std::string& usage = "")
|
|
||||||
|
void set_handler(const std::string& cmd, const callback& hndlr, const std::string& usage = "")
|
||||||
{
|
{
|
||||||
command_handlers_map::mapped_type & vt = m_command_handlers[cmd];
|
lookup::mapped_type & vt = m_command_handlers[cmd];
|
||||||
vt.first = hndlr;
|
vt.first = hndlr;
|
||||||
vt.second = usage;
|
vt.second = usage;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool process_command_vec(const std::vector<std::string>& cmd)
|
bool process_command_vec(const std::vector<std::string>& cmd)
|
||||||
{
|
{
|
||||||
if(!cmd.size())
|
if(!cmd.size())
|
||||||
@ -399,14 +399,20 @@ namespace epee
|
|||||||
boost::split(cmd_v,cmd,boost::is_any_of(" "), boost::token_compress_on);
|
boost::split(cmd_v,cmd,boost::is_any_of(" "), boost::token_compress_on);
|
||||||
return process_command_vec(cmd_v);
|
return process_command_vec(cmd_v);
|
||||||
}
|
}
|
||||||
|
private:
|
||||||
|
lookup m_command_handlers;
|
||||||
|
};
|
||||||
|
|
||||||
/*template<class t_srv>
|
/************************************************************************/
|
||||||
bool start_handling(t_srv& srv, const std::string& usage_string = "")
|
/* */
|
||||||
|
/************************************************************************/
|
||||||
|
class console_handlers_binder : public command_handler
|
||||||
{
|
{
|
||||||
start_default_console_handler_no_srv_param(&srv, boost::bind(&console_handlers_binder::process_command_str, this, _1));
|
typedef command_handler::callback console_command_handler;
|
||||||
return true;
|
typedef command_handler::lookup command_handlers_map;
|
||||||
}*/
|
std::unique_ptr<boost::thread> m_console_thread;
|
||||||
|
async_console_handler m_console_handler;
|
||||||
|
public:
|
||||||
bool start_handling(const std::string& prompt, const std::string& usage_string = "")
|
bool start_handling(const std::string& prompt, const std::string& usage_string = "")
|
||||||
{
|
{
|
||||||
m_console_thread.reset(new boost::thread(boost::bind(&console_handlers_binder::run_handling, this, prompt, usage_string)));
|
m_console_thread.reset(new boost::thread(boost::bind(&console_handlers_binder::run_handling, this, prompt, usage_string)));
|
||||||
@ -423,40 +429,33 @@ namespace epee
|
|||||||
{
|
{
|
||||||
return m_console_handler.run(boost::bind(&console_handlers_binder::process_command_str, this, _1), prompt, usage_string);
|
return m_console_handler.run(boost::bind(&console_handlers_binder::process_command_str, this, _1), prompt, usage_string);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*template<class t_srv>
|
|
||||||
bool run_handling(t_srv& srv, const std::string& usage_string)
|
|
||||||
{
|
|
||||||
return run_default_console_handler_no_srv_param(&srv, boost::bind<bool>(&console_handlers_binder::process_command_str, this, _1), usage_string);
|
|
||||||
}*/
|
|
||||||
};
|
};
|
||||||
|
|
||||||
/* work around because of broken boost bind */
|
///* work around because of broken boost bind */
|
||||||
template<class t_server>
|
//template<class t_server>
|
||||||
class srv_console_handlers_binder: public console_handlers_binder
|
//class srv_console_handlers_binder: public command_handler
|
||||||
{
|
//{
|
||||||
bool process_command_str(t_server* /*psrv*/, const std::string& cmd)
|
// async_console_handler m_console_handler;
|
||||||
{
|
//public:
|
||||||
return console_handlers_binder::process_command_str(cmd);
|
// bool start_handling(t_server* psrv, const std::string& prompt, const std::string& usage_string = "")
|
||||||
}
|
// {
|
||||||
public:
|
// boost::thread(boost::bind(&srv_console_handlers_binder<t_server>::run_handling, this, psrv, prompt, usage_string)).detach();
|
||||||
bool start_handling(t_server* psrv, const std::string& prompt, const std::string& usage_string = "")
|
// return true;
|
||||||
{
|
// }
|
||||||
boost::thread(boost::bind(&srv_console_handlers_binder<t_server>::run_handling, this, psrv, prompt, usage_string)).detach();
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool run_handling(t_server* psrv, const std::string& prompt, const std::string& usage_string)
|
// bool run_handling(t_server* psrv, const std::string& prompt, const std::string& usage_string)
|
||||||
{
|
// {
|
||||||
return m_console_handler.run(psrv, boost::bind(&srv_console_handlers_binder<t_server>::process_command_str, this, _1, _2), prompt, usage_string);
|
// return m_console_handler.run(psrv, boost::bind(&srv_console_handlers_binder<t_server>::process_command_str, this, _1, _2), prompt, usage_string);
|
||||||
}
|
// }
|
||||||
|
|
||||||
void stop_handling()
|
// void stop_handling()
|
||||||
{
|
// {
|
||||||
m_console_handler.stop();
|
// m_console_handler.stop();
|
||||||
}
|
// }
|
||||||
|
//private:
|
||||||
private:
|
// bool process_command_str(t_server* /*psrv*/, const std::string& cmd)
|
||||||
async_console_handler m_console_handler;
|
// {
|
||||||
};
|
// return console_handlers_binder::process_command_str(cmd);
|
||||||
|
// }
|
||||||
|
//};
|
||||||
}
|
}
|
||||||
|
@ -65,7 +65,7 @@
|
|||||||
CHECK_AND_ASSERT_MES(parse_res, false, "Failed to parse json: \r\n" << query_info.m_body); \
|
CHECK_AND_ASSERT_MES(parse_res, false, "Failed to parse json: \r\n" << query_info.m_body); \
|
||||||
uint64_t ticks1 = epee::misc_utils::get_tick_count(); \
|
uint64_t ticks1 = epee::misc_utils::get_tick_count(); \
|
||||||
boost::value_initialized<command_type::response> resp;\
|
boost::value_initialized<command_type::response> resp;\
|
||||||
if(!callback_f(static_cast<command_type::request&>(req), static_cast<command_type::response&>(resp), m_conn_context)) \
|
if(!callback_f(static_cast<command_type::request&>(req), static_cast<command_type::response&>(resp))) \
|
||||||
{ \
|
{ \
|
||||||
LOG_ERROR("Failed to " << #callback_f << "()"); \
|
LOG_ERROR("Failed to " << #callback_f << "()"); \
|
||||||
response_info.m_response_code = 500; \
|
response_info.m_response_code = 500; \
|
||||||
@ -90,7 +90,7 @@
|
|||||||
CHECK_AND_ASSERT_MES(parse_res, false, "Failed to parse bin body data, body size=" << query_info.m_body.size()); \
|
CHECK_AND_ASSERT_MES(parse_res, false, "Failed to parse bin body data, body size=" << query_info.m_body.size()); \
|
||||||
uint64_t ticks1 = misc_utils::get_tick_count(); \
|
uint64_t ticks1 = misc_utils::get_tick_count(); \
|
||||||
boost::value_initialized<command_type::response> resp;\
|
boost::value_initialized<command_type::response> resp;\
|
||||||
if(!callback_f(static_cast<command_type::request&>(req), static_cast<command_type::response&>(resp), m_conn_context)) \
|
if(!callback_f(static_cast<command_type::request&>(req), static_cast<command_type::response&>(resp))) \
|
||||||
{ \
|
{ \
|
||||||
LOG_ERROR("Failed to " << #callback_f << "()"); \
|
LOG_ERROR("Failed to " << #callback_f << "()"); \
|
||||||
response_info.m_response_code = 500; \
|
response_info.m_response_code = 500; \
|
||||||
@ -173,7 +173,7 @@
|
|||||||
epee::json_rpc::error_response fail_resp = AUTO_VAL_INIT(fail_resp); \
|
epee::json_rpc::error_response fail_resp = AUTO_VAL_INIT(fail_resp); \
|
||||||
fail_resp.jsonrpc = "2.0"; \
|
fail_resp.jsonrpc = "2.0"; \
|
||||||
fail_resp.id = req.id; \
|
fail_resp.id = req.id; \
|
||||||
if(!callback_f(req.params, resp.result, fail_resp.error, m_conn_context)) \
|
if(!callback_f(req.params, resp.result, fail_resp.error)) \
|
||||||
{ \
|
{ \
|
||||||
epee::serialization::store_t_to_json(static_cast<epee::json_rpc::error_response&>(fail_resp), response_info.m_body); \
|
epee::serialization::store_t_to_json(static_cast<epee::json_rpc::error_response&>(fail_resp), response_info.m_body); \
|
||||||
return true; \
|
return true; \
|
||||||
@ -202,7 +202,7 @@
|
|||||||
else if(callback_name == method_name) \
|
else if(callback_name == method_name) \
|
||||||
{ \
|
{ \
|
||||||
PREPARE_OBJECTS_FROM_JSON(command_type) \
|
PREPARE_OBJECTS_FROM_JSON(command_type) \
|
||||||
if(!callback_f(req.params, resp.result, m_conn_context)) \
|
if(!callback_f(req.params, resp.result)) \
|
||||||
{ \
|
{ \
|
||||||
epee::json_rpc::error_response fail_resp = AUTO_VAL_INIT(fail_resp); \
|
epee::json_rpc::error_response fail_resp = AUTO_VAL_INIT(fail_resp); \
|
||||||
fail_resp.jsonrpc = "2.0"; \
|
fail_resp.jsonrpc = "2.0"; \
|
||||||
|
2
external/CMakeLists.txt
vendored
2
external/CMakeLists.txt
vendored
@ -35,7 +35,7 @@
|
|||||||
# ...except for FreeBSD, because FreeBSD is a special case that doesn't play well with
|
# ...except for FreeBSD, because FreeBSD is a special case that doesn't play well with
|
||||||
# others.
|
# others.
|
||||||
|
|
||||||
find_package(MiniUpnpc QUIET)
|
find_package(Miniupnpc QUIET)
|
||||||
|
|
||||||
# FreeBSD doesn't play well with the local copy, so default to using shared
|
# FreeBSD doesn't play well with the local copy, so default to using shared
|
||||||
set(USE_SHARED_MINIUPNPC false)
|
set(USE_SHARED_MINIUPNPC false)
|
||||||
|
@ -98,4 +98,5 @@ add_subdirectory(cryptonote_protocol)
|
|||||||
add_subdirectory(connectivity_tool)
|
add_subdirectory(connectivity_tool)
|
||||||
add_subdirectory(miner)
|
add_subdirectory(miner)
|
||||||
add_subdirectory(simplewallet)
|
add_subdirectory(simplewallet)
|
||||||
|
add_subdirectory(daemonizer)
|
||||||
add_subdirectory(daemon)
|
add_subdirectory(daemon)
|
||||||
|
@ -39,8 +39,11 @@ set(common_private_headers
|
|||||||
boost_serialization_helper.h
|
boost_serialization_helper.h
|
||||||
command_line.h
|
command_line.h
|
||||||
dns_utils.h
|
dns_utils.h
|
||||||
|
http_connection.h
|
||||||
int-util.h
|
int-util.h
|
||||||
pod-class.h
|
pod-class.h
|
||||||
|
rpc_client.h
|
||||||
|
scoped_message_writer.h
|
||||||
unordered_containers_boost_serialization.h
|
unordered_containers_boost_serialization.h
|
||||||
util.h
|
util.h
|
||||||
varint.h)
|
varint.h)
|
||||||
|
@ -34,7 +34,65 @@
|
|||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include "include_base_utils.h"
|
#include "include_base_utils.h"
|
||||||
|
#include <boost/filesystem/fstream.hpp>
|
||||||
using namespace epee;
|
using namespace epee;
|
||||||
|
namespace bf = boost::filesystem;
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
|
||||||
|
/*
|
||||||
|
* The following two functions were taken from unbound-anchor.c, from
|
||||||
|
* the unbound library packaged with this source. The license and source
|
||||||
|
* can be found in $PROJECT_ROOT/external/unbound
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* Cert builtin commented out until it's used, as the compiler complains
|
||||||
|
|
||||||
|
// return the built in root update certificate
|
||||||
|
static const char*
|
||||||
|
get_builtin_cert(void)
|
||||||
|
{
|
||||||
|
return
|
||||||
|
// The ICANN CA fetched at 24 Sep 2010. Valid to 2028
|
||||||
|
"-----BEGIN CERTIFICATE-----\n"
|
||||||
|
"MIIDdzCCAl+gAwIBAgIBATANBgkqhkiG9w0BAQsFADBdMQ4wDAYDVQQKEwVJQ0FO\n"
|
||||||
|
"TjEmMCQGA1UECxMdSUNBTk4gQ2VydGlmaWNhdGlvbiBBdXRob3JpdHkxFjAUBgNV\n"
|
||||||
|
"BAMTDUlDQU5OIFJvb3QgQ0ExCzAJBgNVBAYTAlVTMB4XDTA5MTIyMzA0MTkxMloX\n"
|
||||||
|
"DTI5MTIxODA0MTkxMlowXTEOMAwGA1UEChMFSUNBTk4xJjAkBgNVBAsTHUlDQU5O\n"
|
||||||
|
"IENlcnRpZmljYXRpb24gQXV0aG9yaXR5MRYwFAYDVQQDEw1JQ0FOTiBSb290IENB\n"
|
||||||
|
"MQswCQYDVQQGEwJVUzCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAKDb\n"
|
||||||
|
"cLhPNNqc1NB+u+oVvOnJESofYS9qub0/PXagmgr37pNublVThIzyLPGCJ8gPms9S\n"
|
||||||
|
"G1TaKNIsMI7d+5IgMy3WyPEOECGIcfqEIktdR1YWfJufXcMReZwU4v/AdKzdOdfg\n"
|
||||||
|
"ONiwc6r70duEr1IiqPbVm5T05l1e6D+HkAvHGnf1LtOPGs4CHQdpIUcy2kauAEy2\n"
|
||||||
|
"paKcOcHASvbTHK7TbbvHGPB+7faAztABLoneErruEcumetcNfPMIjXKdv1V1E3C7\n"
|
||||||
|
"MSJKy+jAqqQJqjZoQGB0necZgUMiUv7JK1IPQRM2CXJllcyJrm9WFxY0c1KjBO29\n"
|
||||||
|
"iIKK69fcglKcBuFShUECAwEAAaNCMEAwDwYDVR0TAQH/BAUwAwEB/zAOBgNVHQ8B\n"
|
||||||
|
"Af8EBAMCAf4wHQYDVR0OBBYEFLpS6UmDJIZSL8eZzfyNa2kITcBQMA0GCSqGSIb3\n"
|
||||||
|
"DQEBCwUAA4IBAQAP8emCogqHny2UYFqywEuhLys7R9UKmYY4suzGO4nkbgfPFMfH\n"
|
||||||
|
"6M+Zj6owwxlwueZt1j/IaCayoKU3QsrYYoDRolpILh+FPwx7wseUEV8ZKpWsoDoD\n"
|
||||||
|
"2JFbLg2cfB8u/OlE4RYmcxxFSmXBg0yQ8/IoQt/bxOcEEhhiQ168H2yE5rxJMt9h\n"
|
||||||
|
"15nu5JBSewrCkYqYYmaxyOC3WrVGfHZxVI7MpIFcGdvSb2a1uyuua8l0BKgk3ujF\n"
|
||||||
|
"0/wsHNeP22qNyVO+XVBzrM8fk8BSUFuiT/6tZTYXRtEt5aKQZgXbKU5dUF3jT9qg\n"
|
||||||
|
"j/Br5BZw3X/zd325TvnswzMC1+ljLzHnQGGk\n"
|
||||||
|
"-----END CERTIFICATE-----\n"
|
||||||
|
;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
/** return the built in root DS trust anchor */
|
||||||
|
static const char*
|
||||||
|
get_builtin_ds(void)
|
||||||
|
{
|
||||||
|
return
|
||||||
|
". IN DS 19036 8 2 49AAC11D7B6F6446702E54A1607371607A1A41855200FD2CE1CDDE32F24E8FB5\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
/************************************************************
|
||||||
|
************************************************************
|
||||||
|
***********************************************************/
|
||||||
|
|
||||||
|
} // anonymous namespace
|
||||||
|
|
||||||
namespace tools
|
namespace tools
|
||||||
{
|
{
|
||||||
@ -109,6 +167,8 @@ DNSResolver::DNSResolver() : m_data(new DNSResolverData())
|
|||||||
// look for "/etc/resolv.conf" and "/etc/hosts" or platform equivalent
|
// look for "/etc/resolv.conf" and "/etc/hosts" or platform equivalent
|
||||||
ub_ctx_resolvconf(m_data->m_ub_context, &empty_string);
|
ub_ctx_resolvconf(m_data->m_ub_context, &empty_string);
|
||||||
ub_ctx_hosts(m_data->m_ub_context, &empty_string);
|
ub_ctx_hosts(m_data->m_ub_context, &empty_string);
|
||||||
|
|
||||||
|
ub_ctx_add_ta(m_data->m_ub_context, ::get_builtin_ds());
|
||||||
}
|
}
|
||||||
|
|
||||||
DNSResolver::~DNSResolver()
|
DNSResolver::~DNSResolver()
|
||||||
@ -143,6 +203,8 @@ std::vector<std::string> DNSResolver::get_ipv4(const std::string& url, bool& dns
|
|||||||
// call DNS resolver, blocking. if return value not zero, something went wrong
|
// call DNS resolver, blocking. if return value not zero, something went wrong
|
||||||
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_A, DNS_CLASS_IN, &(result.ptr)))
|
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_A, DNS_CLASS_IN, &(result.ptr)))
|
||||||
{
|
{
|
||||||
|
dnssec_available = (result.ptr->secure || (!result.ptr->secure && result.ptr->bogus));
|
||||||
|
dnssec_valid = !result.ptr->bogus;
|
||||||
if (result.ptr->havedata)
|
if (result.ptr->havedata)
|
||||||
{
|
{
|
||||||
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
||||||
@ -175,6 +237,8 @@ std::vector<std::string> DNSResolver::get_ipv6(const std::string& url, bool& dns
|
|||||||
// call DNS resolver, blocking. if return value not zero, something went wrong
|
// call DNS resolver, blocking. if return value not zero, something went wrong
|
||||||
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_AAAA, DNS_CLASS_IN, &(result.ptr)))
|
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_AAAA, DNS_CLASS_IN, &(result.ptr)))
|
||||||
{
|
{
|
||||||
|
dnssec_available = (result.ptr->secure || (!result.ptr->secure && result.ptr->bogus));
|
||||||
|
dnssec_valid = !result.ptr->bogus;
|
||||||
if (result.ptr->havedata)
|
if (result.ptr->havedata)
|
||||||
{
|
{
|
||||||
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
||||||
@ -207,6 +271,8 @@ std::vector<std::string> DNSResolver::get_txt_record(const std::string& url, boo
|
|||||||
// call DNS resolver, blocking. if return value not zero, something went wrong
|
// call DNS resolver, blocking. if return value not zero, something went wrong
|
||||||
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_TXT, DNS_CLASS_IN, &(result.ptr)))
|
if (!ub_resolve(m_data->m_ub_context, urlC, DNS_TYPE_TXT, DNS_CLASS_IN, &(result.ptr)))
|
||||||
{
|
{
|
||||||
|
dnssec_available = (result.ptr->secure || (!result.ptr->secure && result.ptr->bogus));
|
||||||
|
dnssec_valid = !result.ptr->bogus;
|
||||||
if (result.ptr->havedata)
|
if (result.ptr->havedata)
|
||||||
{
|
{
|
||||||
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
for (size_t i=0; result.ptr->data[i] != NULL; i++)
|
||||||
|
42
src/common/http_connection.h
Normal file
42
src/common/http_connection.h
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "string_tools.h"
|
||||||
|
#include "net/http_client.h"
|
||||||
|
|
||||||
|
namespace tools {
|
||||||
|
|
||||||
|
class t_http_connection {
|
||||||
|
private:
|
||||||
|
epee::net_utils::http::http_simple_client * mp_http_client;
|
||||||
|
bool m_ok;
|
||||||
|
public:
|
||||||
|
static unsigned int const TIMEOUT = 200000;
|
||||||
|
|
||||||
|
t_http_connection(
|
||||||
|
epee::net_utils::http::http_simple_client * p_http_client
|
||||||
|
, uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
)
|
||||||
|
: mp_http_client(p_http_client)
|
||||||
|
{
|
||||||
|
// TODO fix http client so that it accepts properly typed arguments
|
||||||
|
std::string ip_str = epee::string_tools::get_ip_string_from_int32(ip);
|
||||||
|
std::string port_str = boost::lexical_cast<std::string>(port);
|
||||||
|
m_ok = mp_http_client->connect(ip_str, port_str, TIMEOUT);
|
||||||
|
}
|
||||||
|
|
||||||
|
~t_http_connection()
|
||||||
|
{
|
||||||
|
if (m_ok)
|
||||||
|
{
|
||||||
|
mp_http_client->disconnect();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool is_open()
|
||||||
|
{
|
||||||
|
return m_ok;
|
||||||
|
}
|
||||||
|
}; // class t_http_connection
|
||||||
|
|
||||||
|
} // namespace tools
|
132
src/common/rpc_client.h
Normal file
132
src/common/rpc_client.h
Normal file
@ -0,0 +1,132 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "common/http_connection.h"
|
||||||
|
#include "common/scoped_message_writer.h"
|
||||||
|
#include "rpc/core_rpc_server_commands_defs.h"
|
||||||
|
#include "storages/http_abstract_invoke.h"
|
||||||
|
#include "net/http_client.h"
|
||||||
|
#include "string_tools.h"
|
||||||
|
#include <boost/lexical_cast.hpp>
|
||||||
|
|
||||||
|
namespace tools
|
||||||
|
{
|
||||||
|
class t_rpc_client final
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
epee::net_utils::http::http_simple_client m_http_client;
|
||||||
|
uint32_t m_ip;
|
||||||
|
uint16_t m_port;
|
||||||
|
public:
|
||||||
|
t_rpc_client(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
)
|
||||||
|
: m_http_client{}
|
||||||
|
, m_ip{ip}
|
||||||
|
, m_port{port}
|
||||||
|
{}
|
||||||
|
|
||||||
|
std::string build_url(std::string const & relative_url)
|
||||||
|
{
|
||||||
|
std::string result =
|
||||||
|
"http://"
|
||||||
|
+ epee::string_tools::get_ip_string_from_int32(m_ip)
|
||||||
|
+ ":"
|
||||||
|
+ boost::lexical_cast<std::string>(m_port)
|
||||||
|
+ relative_url;
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_req, typename T_res>
|
||||||
|
bool basic_json_rpc_request(
|
||||||
|
T_req & req
|
||||||
|
, T_res & res
|
||||||
|
, std::string const & method_name
|
||||||
|
)
|
||||||
|
{
|
||||||
|
std::string rpc_url = build_url("/json_rpc");
|
||||||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||||||
|
|
||||||
|
bool ok = connection.is_open();
|
||||||
|
if (!ok)
|
||||||
|
{
|
||||||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
ok = ok && epee::net_utils::invoke_http_json_rpc(rpc_url, method_name, req, res, m_http_client);
|
||||||
|
if (!ok)
|
||||||
|
{
|
||||||
|
fail_msg_writer() << "Daemon request failed";
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_req, typename T_res>
|
||||||
|
bool json_rpc_request(
|
||||||
|
T_req & req
|
||||||
|
, T_res & res
|
||||||
|
, std::string const & method_name
|
||||||
|
, std::string const & fail_msg
|
||||||
|
)
|
||||||
|
{
|
||||||
|
std::string rpc_url = build_url("/json_rpc");
|
||||||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||||||
|
|
||||||
|
bool ok = connection.is_open();
|
||||||
|
ok = ok && epee::net_utils::invoke_http_json_rpc(rpc_url, method_name, req, res, m_http_client);
|
||||||
|
if (!ok)
|
||||||
|
{
|
||||||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else if (res.status != CORE_RPC_STATUS_OK) // TODO - handle CORE_RPC_STATUS_BUSY ?
|
||||||
|
{
|
||||||
|
fail_msg_writer() << fail_msg << " -- " << res.status;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_req, typename T_res>
|
||||||
|
bool rpc_request(
|
||||||
|
T_req & req
|
||||||
|
, T_res & res
|
||||||
|
, std::string const & relative_url
|
||||||
|
, std::string const & fail_msg
|
||||||
|
)
|
||||||
|
{
|
||||||
|
std::string rpc_url = build_url(relative_url);
|
||||||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||||||
|
|
||||||
|
bool ok = connection.is_open();
|
||||||
|
ok = ok && epee::net_utils::invoke_http_json_remote_command2(rpc_url, req, res, m_http_client);
|
||||||
|
if (!ok)
|
||||||
|
{
|
||||||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else if (res.status != CORE_RPC_STATUS_OK) // TODO - handle CORE_RPC_STATUS_BUSY ?
|
||||||
|
{
|
||||||
|
fail_msg_writer() << fail_msg << " -- " << res.status;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool check_connection()
|
||||||
|
{
|
||||||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||||||
|
return connection.is_open();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
95
src/common/scoped_message_writer.h
Normal file
95
src/common/scoped_message_writer.h
Normal file
@ -0,0 +1,95 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
namespace tools
|
||||||
|
{
|
||||||
|
|
||||||
|
class scoped_message_writer
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
bool m_flush;
|
||||||
|
std::stringstream m_oss;
|
||||||
|
epee::log_space::console_colors m_color;
|
||||||
|
bool m_bright;
|
||||||
|
int m_log_level;
|
||||||
|
public:
|
||||||
|
scoped_message_writer(
|
||||||
|
epee::log_space::console_colors color = epee::log_space::console_color_default
|
||||||
|
, bool bright = false
|
||||||
|
, std::string&& prefix = std::string()
|
||||||
|
, int log_level = LOG_LEVEL_2
|
||||||
|
)
|
||||||
|
: m_flush(true)
|
||||||
|
, m_color(color)
|
||||||
|
, m_bright(bright)
|
||||||
|
, m_log_level(log_level)
|
||||||
|
{
|
||||||
|
m_oss << prefix;
|
||||||
|
}
|
||||||
|
|
||||||
|
scoped_message_writer(scoped_message_writer&& rhs)
|
||||||
|
: m_flush(std::move(rhs.m_flush))
|
||||||
|
#if defined(_MSC_VER)
|
||||||
|
, m_oss(std::move(rhs.m_oss))
|
||||||
|
#else
|
||||||
|
// GCC bug: http://gcc.gnu.org/bugzilla/show_bug.cgi?id=54316
|
||||||
|
, m_oss(rhs.m_oss.str(), std::ios_base::out | std::ios_base::ate)
|
||||||
|
#endif
|
||||||
|
, m_color(std::move(rhs.m_color))
|
||||||
|
, m_log_level(std::move(rhs.m_log_level))
|
||||||
|
{
|
||||||
|
rhs.m_flush = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
scoped_message_writer(scoped_message_writer& rhs) = delete;
|
||||||
|
scoped_message_writer& operator=(scoped_message_writer& rhs) = delete;
|
||||||
|
scoped_message_writer& operator=(scoped_message_writer&& rhs) = delete;
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
std::ostream& operator<<(const T& val)
|
||||||
|
{
|
||||||
|
m_oss << val;
|
||||||
|
return m_oss;
|
||||||
|
}
|
||||||
|
|
||||||
|
~scoped_message_writer()
|
||||||
|
{
|
||||||
|
if (m_flush)
|
||||||
|
{
|
||||||
|
m_flush = false;
|
||||||
|
|
||||||
|
LOG_PRINT(m_oss.str(), m_log_level)
|
||||||
|
|
||||||
|
if (epee::log_space::console_color_default == m_color)
|
||||||
|
{
|
||||||
|
std::cout << m_oss.str();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
epee::log_space::set_console_color(m_color, m_bright);
|
||||||
|
std::cout << m_oss.str();
|
||||||
|
epee::log_space::reset_console_color();
|
||||||
|
}
|
||||||
|
std::cout << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
inline scoped_message_writer success_msg_writer()
|
||||||
|
{
|
||||||
|
return scoped_message_writer(epee::log_space::console_color_green, false, std::string(), LOG_LEVEL_2);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline scoped_message_writer msg_writer(epee::log_space::console_colors color = epee::log_space::console_color_default)
|
||||||
|
{
|
||||||
|
return scoped_message_writer(color, false, std::string(), LOG_LEVEL_2);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline scoped_message_writer fail_msg_writer()
|
||||||
|
{
|
||||||
|
return scoped_message_writer(epee::log_space::console_color_red, true, "Error: ", LOG_LEVEL_0);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace tools
|
@ -326,7 +326,7 @@ std::string get_nix_version_display_string()
|
|||||||
std::string config_folder;
|
std::string config_folder;
|
||||||
|
|
||||||
#ifdef WIN32
|
#ifdef WIN32
|
||||||
config_folder = get_special_folder_path(CSIDL_APPDATA, true) + "\\" + CRYPTONOTE_NAME;
|
config_folder = get_special_folder_path(CSIDL_COMMON_APPDATA, true) + "\\" + CRYPTONOTE_NAME;
|
||||||
#else
|
#else
|
||||||
std::string pathRet;
|
std::string pathRet;
|
||||||
char* pszHome = getenv("HOME");
|
char* pszHome = getenv("HOME");
|
||||||
|
@ -58,6 +58,18 @@ namespace tools
|
|||||||
*/
|
*/
|
||||||
std::string get_default_data_dir();
|
std::string get_default_data_dir();
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
/**
|
||||||
|
* @brief
|
||||||
|
*
|
||||||
|
* @param nfolder
|
||||||
|
* @param iscreate
|
||||||
|
*
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
std::string get_special_folder_path(int nfolder, bool iscreate);
|
||||||
|
#endif
|
||||||
|
|
||||||
/*! \brief Returns the OS version string
|
/*! \brief Returns the OS version string
|
||||||
*
|
*
|
||||||
* \details This is a wrapper around the primitives
|
* \details This is a wrapper around the primitives
|
||||||
|
@ -51,6 +51,7 @@
|
|||||||
// MONEY_SUPPLY - total number coins to be generated
|
// MONEY_SUPPLY - total number coins to be generated
|
||||||
#define MONEY_SUPPLY ((uint64_t)(-1))
|
#define MONEY_SUPPLY ((uint64_t)(-1))
|
||||||
#define EMISSION_SPEED_FACTOR (20)
|
#define EMISSION_SPEED_FACTOR (20)
|
||||||
|
#define FINAL_SUBSIDY_PER_MINUTE ((uint64_t)300000000000) // 3 * pow(10, 11)
|
||||||
|
|
||||||
#define CRYPTONOTE_REWARD_BLOCKS_WINDOW 100
|
#define CRYPTONOTE_REWARD_BLOCKS_WINDOW 100
|
||||||
#define CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE 20000 //size of block (bytes) after which reward for block calculated using block size
|
#define CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE 20000 //size of block (bytes) after which reward for block calculated using block size
|
||||||
@ -61,9 +62,6 @@
|
|||||||
|
|
||||||
#define FEE_PER_KB ((uint64_t)10000000000) // pow(10, 10)
|
#define FEE_PER_KB ((uint64_t)10000000000) // pow(10, 10)
|
||||||
|
|
||||||
// temporarily to allow backward compatibility during the switch to per-kb
|
|
||||||
//#define MINING_ALLOWED_LEGACY_FEE ((uint64_t)100000000000) // pow(10, 11)
|
|
||||||
|
|
||||||
#define ORPHANED_BLOCKS_MAX_COUNT 100
|
#define ORPHANED_BLOCKS_MAX_COUNT 100
|
||||||
|
|
||||||
|
|
||||||
|
@ -89,6 +89,7 @@ bool blockchain_storage::init(const std::string& config_folder, bool testnet)
|
|||||||
{
|
{
|
||||||
CRITICAL_REGION_LOCAL(m_blockchain_lock);
|
CRITICAL_REGION_LOCAL(m_blockchain_lock);
|
||||||
m_config_folder = config_folder;
|
m_config_folder = config_folder;
|
||||||
|
m_testnet = testnet;
|
||||||
LOG_PRINT_L0("Loading blockchain...");
|
LOG_PRINT_L0("Loading blockchain...");
|
||||||
const std::string filename = m_config_folder + "/" CRYPTONOTE_BLOCKCHAINDATA_FILENAME;
|
const std::string filename = m_config_folder + "/" CRYPTONOTE_BLOCKCHAINDATA_FILENAME;
|
||||||
if(tools::unserialize_obj_from_file(*this, filename))
|
if(tools::unserialize_obj_from_file(*this, filename))
|
||||||
@ -1365,7 +1366,7 @@ bool blockchain_storage::pop_transaction_from_global_index(const transaction& tx
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------
|
//------------------------------------------------------------------
|
||||||
bool blockchain_storage::add_transaction_from_block(const transaction& tx, const crypto::hash& tx_id, const crypto::hash& bl_id, uint64_t bl_height)
|
bool blockchain_storage::add_transaction_from_block(const transaction& tx, const crypto::hash& tx_id, const crypto::hash& bl_id, uint64_t bl_height, size_t blob_size)
|
||||||
{
|
{
|
||||||
CRITICAL_REGION_LOCAL(m_blockchain_lock);
|
CRITICAL_REGION_LOCAL(m_blockchain_lock);
|
||||||
struct add_transaction_input_visitor: public boost::static_visitor<bool>
|
struct add_transaction_input_visitor: public boost::static_visitor<bool>
|
||||||
@ -1404,6 +1405,7 @@ bool blockchain_storage::add_transaction_from_block(const transaction& tx, const
|
|||||||
}
|
}
|
||||||
transaction_chain_entry ch_e;
|
transaction_chain_entry ch_e;
|
||||||
ch_e.m_keeper_block_height = bl_height;
|
ch_e.m_keeper_block_height = bl_height;
|
||||||
|
ch_e.m_blob_size = blob_size;
|
||||||
ch_e.tx = tx;
|
ch_e.tx = tx;
|
||||||
auto i_r = m_transactions.insert(std::pair<crypto::hash, transaction_chain_entry>(tx_id, ch_e));
|
auto i_r = m_transactions.insert(std::pair<crypto::hash, transaction_chain_entry>(tx_id, ch_e));
|
||||||
if(!i_r.second)
|
if(!i_r.second)
|
||||||
@ -1669,10 +1671,12 @@ bool blockchain_storage::handle_block_to_main_chain(const block& bl, const crypt
|
|||||||
bvc.m_verifivation_failed = true;
|
bvc.m_verifivation_failed = true;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
size_t coinbase_blob_size = get_object_blobsize(bl.miner_tx);
|
crypto::hash coinbase_hash = null_hash;
|
||||||
|
size_t coinbase_blob_size = 0;
|
||||||
|
get_transaction_hash(bl.miner_tx, coinbase_hash, coinbase_blob_size);
|
||||||
size_t cumulative_block_size = coinbase_blob_size;
|
size_t cumulative_block_size = coinbase_blob_size;
|
||||||
//process transactions
|
//process transactions
|
||||||
if(!add_transaction_from_block(bl.miner_tx, get_transaction_hash(bl.miner_tx), id, get_current_blockchain_height()))
|
if(!add_transaction_from_block(bl.miner_tx, coinbase_hash, id, get_current_blockchain_height(), coinbase_blob_size))
|
||||||
{
|
{
|
||||||
LOG_PRINT_L1("Block with id: " << id << " failed to add transaction to blockchain storage");
|
LOG_PRINT_L1("Block with id: " << id << " failed to add transaction to blockchain storage");
|
||||||
bvc.m_verifivation_failed = true;
|
bvc.m_verifivation_failed = true;
|
||||||
@ -1706,7 +1710,7 @@ bool blockchain_storage::handle_block_to_main_chain(const block& bl, const crypt
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if(!add_transaction_from_block(tx, tx_id, id, get_current_blockchain_height()))
|
if(!add_transaction_from_block(tx, tx_id, id, get_current_blockchain_height(), blob_size))
|
||||||
{
|
{
|
||||||
LOG_PRINT_L1("Block with id: " << id << " failed to add transaction to blockchain storage");
|
LOG_PRINT_L1("Block with id: " << id << " failed to add transaction to blockchain storage");
|
||||||
cryptonote::tx_verification_context tvc = AUTO_VAL_INIT(tvc);
|
cryptonote::tx_verification_context tvc = AUTO_VAL_INIT(tvc);
|
||||||
@ -1736,7 +1740,14 @@ bool blockchain_storage::handle_block_to_main_chain(const block& bl, const crypt
|
|||||||
bei.bl = bl;
|
bei.bl = bl;
|
||||||
bei.block_cumulative_size = cumulative_block_size;
|
bei.block_cumulative_size = cumulative_block_size;
|
||||||
bei.cumulative_difficulty = current_diffic;
|
bei.cumulative_difficulty = current_diffic;
|
||||||
bei.already_generated_coins = already_generated_coins + base_reward;
|
|
||||||
|
// In the "tail" state when the minimum subsidy (implemented in get_block_reward) is in effect, the number of
|
||||||
|
// coins will eventually exceed MONEY_SUPPLY and overflow a uint64. To prevent overflow, cap already_generated_coins
|
||||||
|
// at MONEY_SUPPLY. already_generated_coins is only used to compute the block subsidy and MONEY_SUPPLY yields a
|
||||||
|
// subsidy of 0 under the base formula and therefore the minimum subsidy >0 in the tail state.
|
||||||
|
|
||||||
|
bei.already_generated_coins = base_reward < (MONEY_SUPPLY-already_generated_coins) ? already_generated_coins + base_reward : MONEY_SUPPLY;
|
||||||
|
|
||||||
if(m_blocks.size())
|
if(m_blocks.size())
|
||||||
bei.cumulative_difficulty += m_blocks.back().cumulative_difficulty;
|
bei.cumulative_difficulty += m_blocks.back().cumulative_difficulty;
|
||||||
|
|
||||||
@ -1862,7 +1873,7 @@ bool blockchain_storage::update_checkpoints(const std::string& file_path, bool c
|
|||||||
else if (check_dns)
|
else if (check_dns)
|
||||||
{
|
{
|
||||||
checkpoints dns_points;
|
checkpoints dns_points;
|
||||||
cryptonote::load_checkpoints_from_dns(dns_points);
|
cryptonote::load_checkpoints_from_dns(dns_points, m_testnet);
|
||||||
if (m_checkpoints.check_for_conflicts(dns_points))
|
if (m_checkpoints.check_for_conflicts(dns_points))
|
||||||
{
|
{
|
||||||
check_against_checkpoints(dns_points, false);
|
check_against_checkpoints(dns_points, false);
|
||||||
|
@ -219,6 +219,7 @@ namespace cryptonote
|
|||||||
std::atomic<bool> m_is_blockchain_storing;
|
std::atomic<bool> m_is_blockchain_storing;
|
||||||
|
|
||||||
bool m_enforce_dns_checkpoints;
|
bool m_enforce_dns_checkpoints;
|
||||||
|
bool m_testnet;
|
||||||
|
|
||||||
bool switch_to_alternative_blockchain(std::list<blocks_ext_by_hash::iterator>& alt_chain, bool discard_disconnected_chain);
|
bool switch_to_alternative_blockchain(std::list<blocks_ext_by_hash::iterator>& alt_chain, bool discard_disconnected_chain);
|
||||||
bool pop_block_from_blockchain();
|
bool pop_block_from_blockchain();
|
||||||
@ -234,7 +235,7 @@ namespace cryptonote
|
|||||||
bool validate_miner_transaction(const block& b, size_t cumulative_block_size, uint64_t fee, uint64_t& base_reward, uint64_t already_generated_coins);
|
bool validate_miner_transaction(const block& b, size_t cumulative_block_size, uint64_t fee, uint64_t& base_reward, uint64_t already_generated_coins);
|
||||||
bool validate_transaction(const block& b, uint64_t height, const transaction& tx);
|
bool validate_transaction(const block& b, uint64_t height, const transaction& tx);
|
||||||
bool rollback_blockchain_switching(std::list<block>& original_chain, size_t rollback_height);
|
bool rollback_blockchain_switching(std::list<block>& original_chain, size_t rollback_height);
|
||||||
bool add_transaction_from_block(const transaction& tx, const crypto::hash& tx_id, const crypto::hash& bl_id, uint64_t bl_height);
|
bool add_transaction_from_block(const transaction& tx, const crypto::hash& tx_id, const crypto::hash& bl_id, uint64_t bl_height, size_t blob_size);
|
||||||
bool push_transaction_to_global_outs_index(const transaction& tx, const crypto::hash& tx_id, std::vector<uint64_t>& global_indexes);
|
bool push_transaction_to_global_outs_index(const transaction& tx, const crypto::hash& tx_id, std::vector<uint64_t>& global_indexes);
|
||||||
bool pop_transaction_from_global_index(const transaction& tx, const crypto::hash& tx_id);
|
bool pop_transaction_from_global_index(const transaction& tx, const crypto::hash& tx_id);
|
||||||
bool get_last_n_blocks_sizes(std::vector<size_t>& sz, size_t count);
|
bool get_last_n_blocks_sizes(std::vector<size_t>& sz, size_t count);
|
||||||
|
@ -76,6 +76,7 @@ bool create_checkpoints(cryptonote::checkpoints& checkpoints)
|
|||||||
ADD_CHECKPOINT(231350, "b5add137199b820e1ea26640e5c3e121fd85faa86a1e39cf7e6cc097bdeb1131");
|
ADD_CHECKPOINT(231350, "b5add137199b820e1ea26640e5c3e121fd85faa86a1e39cf7e6cc097bdeb1131");
|
||||||
ADD_CHECKPOINT(232150, "955de8e6b6508af2c24f7334f97beeea651d78e9ade3ab18fec3763be3201aa8");
|
ADD_CHECKPOINT(232150, "955de8e6b6508af2c24f7334f97beeea651d78e9ade3ab18fec3763be3201aa8");
|
||||||
ADD_CHECKPOINT(249380, "654fb0a81ce3e5caf7e3264a70f447d4bd07586c08fa50f6638cc54da0a52b2d");
|
ADD_CHECKPOINT(249380, "654fb0a81ce3e5caf7e3264a70f447d4bd07586c08fa50f6638cc54da0a52b2d");
|
||||||
|
ADD_CHECKPOINT(460000, "75037a7aed3e765db96c75bcf908f59d690a5f3390baebb9edeafd336a1c4831");
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -112,7 +113,7 @@ bool load_checkpoints_from_json(cryptonote::checkpoints& checkpoints, std::strin
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints)
|
bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints, bool testnet)
|
||||||
{
|
{
|
||||||
// All four MoneroPulse domains have DNSSEC on and valid
|
// All four MoneroPulse domains have DNSSEC on and valid
|
||||||
static const std::vector<std::string> dns_urls = { "checkpoints.moneropulse.se"
|
static const std::vector<std::string> dns_urls = { "checkpoints.moneropulse.se"
|
||||||
@ -120,6 +121,12 @@ bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints)
|
|||||||
, "checkpoints.moneropulse.net"
|
, "checkpoints.moneropulse.net"
|
||||||
, "checkpoints.moneropulse.co"
|
, "checkpoints.moneropulse.co"
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static const std::vector<std::string> testnet_dns_urls = { "testpoints.moneropulse.se"
|
||||||
|
, "testpoints.moneropulse.org"
|
||||||
|
, "testpoints.moneropulse.net"
|
||||||
|
, "testpoints.moneropulse.co"
|
||||||
|
};
|
||||||
bool avail, valid;
|
bool avail, valid;
|
||||||
std::vector<std::string> records;
|
std::vector<std::string> records;
|
||||||
|
|
||||||
@ -131,14 +138,34 @@ bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints)
|
|||||||
size_t cur_index = first_index;
|
size_t cur_index = first_index;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
records = tools::DNSResolver::instance().get_txt_record(dns_urls[cur_index], avail, valid);
|
std::string url;
|
||||||
if (records.size() == 0 || (avail && !valid))
|
if (testnet)
|
||||||
|
{
|
||||||
|
url = testnet_dns_urls[cur_index];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
url = dns_urls[cur_index];
|
||||||
|
}
|
||||||
|
|
||||||
|
records = tools::DNSResolver::instance().get_txt_record(url, avail, valid);
|
||||||
|
if (!avail)
|
||||||
|
{
|
||||||
|
LOG_PRINT_L2("DNSSEC not available for checkpoint update at URL: " << url << ", skipping.");
|
||||||
|
}
|
||||||
|
if (!valid)
|
||||||
|
{
|
||||||
|
LOG_PRINT_L2("DNSSEC validation failed for checkpoint update at URL: " << url << ", skipping.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (records.size() == 0 || !avail || !valid)
|
||||||
{
|
{
|
||||||
cur_index++;
|
cur_index++;
|
||||||
if (cur_index == dns_urls.size())
|
if (cur_index == dns_urls.size())
|
||||||
{
|
{
|
||||||
cur_index = 0;
|
cur_index = 0;
|
||||||
}
|
}
|
||||||
|
records.clear();
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
@ -146,13 +173,7 @@ bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints)
|
|||||||
|
|
||||||
if (records.size() == 0)
|
if (records.size() == 0)
|
||||||
{
|
{
|
||||||
LOG_PRINT_L1("Fetching MoneroPulse checkpoints failed, no TXT records available.");
|
LOG_PRINT_L0("WARNING: All MoneroPulse checkpoint URLs failed DNSSEC validation and/or returned no records");
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (avail && !valid)
|
|
||||||
{
|
|
||||||
LOG_PRINT_L0("WARNING: MoneroPulse failed DNSSEC validation and/or returned no records");
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -42,7 +42,7 @@ namespace cryptonote
|
|||||||
bool create_checkpoints(cryptonote::checkpoints& checkpoints);
|
bool create_checkpoints(cryptonote::checkpoints& checkpoints);
|
||||||
|
|
||||||
bool load_checkpoints_from_json(cryptonote::checkpoints& checkpoints, std::string json_hashfile_fullpath);
|
bool load_checkpoints_from_json(cryptonote::checkpoints& checkpoints, std::string json_hashfile_fullpath);
|
||||||
bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints);
|
bool load_checkpoints_from_dns(cryptonote::checkpoints& checkpoints, bool testnet = false);
|
||||||
bool load_new_checkpoints(cryptonote::checkpoints& checkpoints, std::string json_hashfile_fullpath);
|
bool load_new_checkpoints(cryptonote::checkpoints& checkpoints, std::string json_hashfile_fullpath);
|
||||||
|
|
||||||
} // namespace cryptonote
|
} // namespace cryptonote
|
||||||
|
@ -60,6 +60,10 @@ namespace cryptonote {
|
|||||||
//-----------------------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------------------
|
||||||
bool get_block_reward(size_t median_size, size_t current_block_size, uint64_t already_generated_coins, uint64_t &reward) {
|
bool get_block_reward(size_t median_size, size_t current_block_size, uint64_t already_generated_coins, uint64_t &reward) {
|
||||||
uint64_t base_reward = (MONEY_SUPPLY - already_generated_coins) >> EMISSION_SPEED_FACTOR;
|
uint64_t base_reward = (MONEY_SUPPLY - already_generated_coins) >> EMISSION_SPEED_FACTOR;
|
||||||
|
if (base_reward < FINAL_SUBSIDY_PER_MINUTE)
|
||||||
|
{
|
||||||
|
base_reward = FINAL_SUBSIDY_PER_MINUTE;
|
||||||
|
}
|
||||||
|
|
||||||
//make it soft
|
//make it soft
|
||||||
if (median_size < CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE) {
|
if (median_size < CRYPTONOTE_BLOCK_GRANTED_FULL_REWARD_ZONE) {
|
||||||
|
@ -42,6 +42,8 @@ using namespace epee;
|
|||||||
#include "cryptonote_format_utils.h"
|
#include "cryptonote_format_utils.h"
|
||||||
#include "misc_language.h"
|
#include "misc_language.h"
|
||||||
#include <csignal>
|
#include <csignal>
|
||||||
|
#include "daemon/command_line_args.h"
|
||||||
|
#include "cryptonote_core/checkpoints_create.h"
|
||||||
|
|
||||||
DISABLE_VS_WARNINGS(4355)
|
DISABLE_VS_WARNINGS(4355)
|
||||||
|
|
||||||
@ -108,14 +110,41 @@ namespace cryptonote
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
//-----------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------
|
||||||
|
void core::stop()
|
||||||
|
{
|
||||||
|
graceful_exit();
|
||||||
|
}
|
||||||
|
//-----------------------------------------------------------------------------------
|
||||||
void core::init_options(boost::program_options::options_description& /*desc*/)
|
void core::init_options(boost::program_options::options_description& /*desc*/)
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
//-----------------------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------------------
|
||||||
bool core::handle_command_line(const boost::program_options::variables_map& vm, bool testnet)
|
bool core::handle_command_line(const boost::program_options::variables_map& vm)
|
||||||
{
|
{
|
||||||
auto data_dir_arg = testnet ? command_line::arg_testnet_data_dir : command_line::arg_data_dir;
|
m_testnet = command_line::get_arg(vm, daemon_args::arg_testnet_on);
|
||||||
|
|
||||||
|
auto data_dir_arg = m_testnet ? command_line::arg_testnet_data_dir : command_line::arg_data_dir;
|
||||||
m_config_folder = command_line::get_arg(vm, data_dir_arg);
|
m_config_folder = command_line::get_arg(vm, data_dir_arg);
|
||||||
|
|
||||||
|
auto data_dir = boost::filesystem::path(m_config_folder);
|
||||||
|
|
||||||
|
if (!m_testnet)
|
||||||
|
{
|
||||||
|
cryptonote::checkpoints checkpoints;
|
||||||
|
if (!cryptonote::create_checkpoints(checkpoints))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to initialize checkpoints");
|
||||||
|
}
|
||||||
|
set_checkpoints(std::move(checkpoints));
|
||||||
|
|
||||||
|
boost::filesystem::path json(JSON_HASH_FILE_NAME);
|
||||||
|
boost::filesystem::path checkpoint_json_hashfile_fullpath = data_dir / json;
|
||||||
|
|
||||||
|
set_checkpoints_file_path(checkpoint_json_hashfile_fullpath.string());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
set_enforce_dns_checkpoints(command_line::get_arg(vm, daemon_args::arg_dns_checkpoints));
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//-----------------------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------------------
|
||||||
@ -154,21 +183,21 @@ namespace cryptonote
|
|||||||
return m_blockchain_storage.get_alternative_blocks_count();
|
return m_blockchain_storage.get_alternative_blocks_count();
|
||||||
}
|
}
|
||||||
//-----------------------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------------------
|
||||||
bool core::init(const boost::program_options::variables_map& vm, bool testnet)
|
bool core::init(const boost::program_options::variables_map& vm)
|
||||||
{
|
{
|
||||||
bool r = handle_command_line(vm, testnet);
|
bool r = handle_command_line(vm);
|
||||||
|
|
||||||
r = m_mempool.init(m_config_folder);
|
r = m_mempool.init(m_config_folder);
|
||||||
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize memory pool");
|
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize memory pool");
|
||||||
|
|
||||||
r = m_blockchain_storage.init(m_config_folder, testnet);
|
r = m_blockchain_storage.init(m_config_folder, m_testnet);
|
||||||
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize blockchain storage");
|
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize blockchain storage");
|
||||||
|
|
||||||
// load json & DNS checkpoints, and verify them
|
// load json & DNS checkpoints, and verify them
|
||||||
// with respect to what blocks we already have
|
// with respect to what blocks we already have
|
||||||
CHECK_AND_ASSERT_MES(update_checkpoints(), false, "One or more checkpoints loaded from json or dns conflicted with existing checkpoints.");
|
CHECK_AND_ASSERT_MES(update_checkpoints(), false, "One or more checkpoints loaded from json or dns conflicted with existing checkpoints.");
|
||||||
|
|
||||||
r = m_miner.init(vm, testnet);
|
r = m_miner.init(vm, m_testnet);
|
||||||
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize blockchain storage");
|
CHECK_AND_ASSERT_MES(r, false, "Failed to initialize blockchain storage");
|
||||||
|
|
||||||
return load_state_data();
|
return load_state_data();
|
||||||
|
@ -72,7 +72,7 @@ namespace cryptonote
|
|||||||
|
|
||||||
miner& get_miner(){return m_miner;}
|
miner& get_miner(){return m_miner;}
|
||||||
static void init_options(boost::program_options::options_description& desc);
|
static void init_options(boost::program_options::options_description& desc);
|
||||||
bool init(const boost::program_options::variables_map& vm, bool testnet);
|
bool init(const boost::program_options::variables_map& vm);
|
||||||
bool set_genesis_block(const block& b);
|
bool set_genesis_block(const block& b);
|
||||||
bool deinit();
|
bool deinit();
|
||||||
static void set_fast_exit();
|
static void set_fast_exit();
|
||||||
@ -131,6 +131,8 @@ namespace cryptonote
|
|||||||
|
|
||||||
bool update_checkpoints();
|
bool update_checkpoints();
|
||||||
|
|
||||||
|
void stop();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
bool add_new_tx(const transaction& tx, const crypto::hash& tx_hash, const crypto::hash& tx_prefix_hash, size_t blob_size, tx_verification_context& tvc, bool keeped_by_block);
|
bool add_new_tx(const transaction& tx, const crypto::hash& tx_hash, const crypto::hash& tx_prefix_hash, size_t blob_size, tx_verification_context& tvc, bool keeped_by_block);
|
||||||
bool add_new_tx(const transaction& tx, tx_verification_context& tvc, bool keeped_by_block);
|
bool add_new_tx(const transaction& tx, tx_verification_context& tvc, bool keeped_by_block);
|
||||||
@ -148,7 +150,7 @@ namespace cryptonote
|
|||||||
bool check_tx_ring_signature(const txin_to_key& tx, const crypto::hash& tx_prefix_hash, const std::vector<crypto::signature>& sig);
|
bool check_tx_ring_signature(const txin_to_key& tx, const crypto::hash& tx_prefix_hash, const std::vector<crypto::signature>& sig);
|
||||||
bool is_tx_spendtime_unlocked(uint64_t unlock_time);
|
bool is_tx_spendtime_unlocked(uint64_t unlock_time);
|
||||||
bool update_miner_block_template();
|
bool update_miner_block_template();
|
||||||
bool handle_command_line(const boost::program_options::variables_map& vm, bool testnet);
|
bool handle_command_line(const boost::program_options::variables_map& vm);
|
||||||
bool on_update_blocktemplate_interval();
|
bool on_update_blocktemplate_interval();
|
||||||
bool check_tx_inputs_keyimages_diff(const transaction& tx);
|
bool check_tx_inputs_keyimages_diff(const transaction& tx);
|
||||||
void graceful_exit();
|
void graceful_exit();
|
||||||
@ -171,6 +173,7 @@ namespace cryptonote
|
|||||||
|
|
||||||
uint64_t m_target_blockchain_height;
|
uint64_t m_target_blockchain_height;
|
||||||
|
|
||||||
|
bool m_testnet;
|
||||||
std::string m_checkpoints_path;
|
std::string m_checkpoints_path;
|
||||||
time_t m_last_dns_checkpoints_update;
|
time_t m_last_dns_checkpoints_update;
|
||||||
time_t m_last_json_checkpoints_update;
|
time_t m_last_json_checkpoints_update;
|
||||||
|
@ -27,12 +27,43 @@
|
|||||||
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
set(daemon_sources
|
set(daemon_sources
|
||||||
daemon.cpp)
|
command_parser_executor.cpp
|
||||||
|
command_server.cpp
|
||||||
|
daemon.cpp
|
||||||
|
executor.cpp
|
||||||
|
main.cpp
|
||||||
|
rpc_command_executor.cpp
|
||||||
|
)
|
||||||
|
|
||||||
set(daemon_headers)
|
set(daemon_headers)
|
||||||
|
|
||||||
set(daemon_private_headers
|
set(daemon_private_headers
|
||||||
daemon_commands_handler.h)
|
command_parser_executor.h
|
||||||
|
command_server.h
|
||||||
|
core.h
|
||||||
|
daemon.h
|
||||||
|
daemon_commands_handler.h
|
||||||
|
executor.h
|
||||||
|
p2p.h
|
||||||
|
protocol.h
|
||||||
|
rpc.h
|
||||||
|
rpc_command_executor.h
|
||||||
|
|
||||||
|
# cryptonote_protocol
|
||||||
|
../cryptonote_protocol/blobdatatype.h
|
||||||
|
../cryptonote_protocol/cryptonote_protocol_defs.h
|
||||||
|
../cryptonote_protocol/cryptonote_protocol_handler.h
|
||||||
|
../cryptonote_protocol/cryptonote_protocol_handler.inl
|
||||||
|
../cryptonote_protocol/cryptonote_protocol_handler_common.h
|
||||||
|
|
||||||
|
# p2p
|
||||||
|
../p2p/net_node.h
|
||||||
|
../p2p/net_node.inl
|
||||||
|
../p2p/net_node_common.h
|
||||||
|
../p2p/net_peerlist.h
|
||||||
|
../p2p/net_peerlist_boost_serialization.h
|
||||||
|
../p2p/p2p_protocol_defs.h
|
||||||
|
../p2p/stdafx.h)
|
||||||
|
|
||||||
bitmonero_private_headers(daemon
|
bitmonero_private_headers(daemon
|
||||||
${daemon_private_headers})
|
${daemon_private_headers})
|
||||||
@ -46,9 +77,7 @@ target_link_libraries(daemon
|
|||||||
cryptonote_core
|
cryptonote_core
|
||||||
crypto
|
crypto
|
||||||
common
|
common
|
||||||
otshell_utils
|
daemonizer
|
||||||
p2p
|
|
||||||
cryptonote_protocol
|
|
||||||
${Boost_CHRONO_LIBRARY}
|
${Boost_CHRONO_LIBRARY}
|
||||||
${Boost_FILESYSTEM_LIBRARY}
|
${Boost_FILESYSTEM_LIBRARY}
|
||||||
${Boost_PROGRAM_OPTIONS_LIBRARY}
|
${Boost_PROGRAM_OPTIONS_LIBRARY}
|
||||||
|
76
src/daemon/command_line_args.h
Normal file
76
src/daemon/command_line_args.h
Normal file
@ -0,0 +1,76 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#ifndef DAEMON_COMMAND_LINE_ARGS_H
|
||||||
|
#define DAEMON_COMMAND_LINE_ARGS_H
|
||||||
|
|
||||||
|
#include "common/command_line.h"
|
||||||
|
#include "cryptonote_config.h"
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
|
||||||
|
namespace daemon_args
|
||||||
|
{
|
||||||
|
std::string const WINDOWS_SERVICE_NAME = "Monero Daemon";
|
||||||
|
|
||||||
|
const command_line::arg_descriptor<std::string> arg_config_file = {
|
||||||
|
"config-file"
|
||||||
|
, "Specify configuration file"
|
||||||
|
, std::string(CRYPTONOTE_NAME ".conf")
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<std::string> arg_log_file = {
|
||||||
|
"log-file"
|
||||||
|
, "Specify log file"
|
||||||
|
, ""
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<int> arg_log_level = {
|
||||||
|
"log-level"
|
||||||
|
, ""
|
||||||
|
, LOG_LEVEL_0
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<std::vector<std::string>> arg_command = {
|
||||||
|
"daemon_command"
|
||||||
|
, "Hidden"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_os_version = {
|
||||||
|
"os-version"
|
||||||
|
, "OS for which this executable was compiled"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_testnet_on = {
|
||||||
|
"testnet"
|
||||||
|
, "Run on testnet. The wallet must be launched with --testnet flag."
|
||||||
|
, false
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_dns_checkpoints = {
|
||||||
|
"enforce-dns-checkpointing"
|
||||||
|
, "checkpoints from DNS server will be enforced"
|
||||||
|
, false
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace daemon_args
|
||||||
|
|
||||||
|
#endif // DAEMON_COMMAND_LINE_ARGS_H
|
299
src/daemon/command_parser_executor.cpp
Normal file
299
src/daemon/command_parser_executor.cpp
Normal file
@ -0,0 +1,299 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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 "cryptonote_core/cryptonote_basic_impl.h"
|
||||||
|
#include "daemon/command_parser_executor.h"
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
t_command_parser_executor::t_command_parser_executor(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc
|
||||||
|
, cryptonote::core_rpc_server* rpc_server
|
||||||
|
)
|
||||||
|
: m_executor(ip, port, is_rpc, rpc_server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_peer_list(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_peer_list();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::save_blockchain(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.save_blockchain();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::show_hash_rate(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.show_hash_rate();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::hide_hash_rate(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.hide_hash_rate();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::show_difficulty(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.show_difficulty();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_connections(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_connections();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_blockchain_info(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(!args.size())
|
||||||
|
{
|
||||||
|
std::cout << "need block index parameter" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
uint64_t start_index = 0;
|
||||||
|
uint64_t end_index = 0;
|
||||||
|
if(!epee::string_tools::get_xtype_from_string(start_index, args[0]))
|
||||||
|
{
|
||||||
|
std::cout << "wrong starter block index parameter" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if(args.size() >1 && !epee::string_tools::get_xtype_from_string(end_index, args[1]))
|
||||||
|
{
|
||||||
|
std::cout << "wrong end block index parameter" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return m_executor.print_blockchain_info(start_index, end_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::set_log_level(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size() != 1)
|
||||||
|
{
|
||||||
|
std::cout << "use: set_log <log_level_number_0-4>" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint16_t l = 0;
|
||||||
|
if(!epee::string_tools::get_xtype_from_string(l, args[0]))
|
||||||
|
{
|
||||||
|
std::cout << "wrong number format, use: set_log <log_level_number_0-4>" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(LOG_LEVEL_4 < l)
|
||||||
|
{
|
||||||
|
std::cout << "wrong number range, use: set_log <log_level_number_0-4>" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return m_executor.set_log_level(l);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_height(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_height();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_block(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (args.empty())
|
||||||
|
{
|
||||||
|
std::cout << "expected: print_block (<block_hash> | <block_height>)" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string& arg = args.front();
|
||||||
|
try
|
||||||
|
{
|
||||||
|
uint64_t height = boost::lexical_cast<uint64_t>(arg);
|
||||||
|
return m_executor.print_block_by_height(height);
|
||||||
|
}
|
||||||
|
catch (boost::bad_lexical_cast&)
|
||||||
|
{
|
||||||
|
crypto::hash block_hash;
|
||||||
|
if (parse_hash256(arg, block_hash))
|
||||||
|
{
|
||||||
|
return m_executor.print_block_by_hash(block_hash);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_transaction(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (args.empty())
|
||||||
|
{
|
||||||
|
std::cout << "expected: print_tx <transaction hash>" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string& str_hash = args.front();
|
||||||
|
crypto::hash tx_hash;
|
||||||
|
if (parse_hash256(str_hash, tx_hash))
|
||||||
|
{
|
||||||
|
m_executor.print_transaction(tx_hash);
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_transaction_pool_long(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_transaction_pool_long();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_transaction_pool_short(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_transaction_pool_short();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::start_mining(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(!args.size())
|
||||||
|
{
|
||||||
|
std::cout << "Please specify a wallet address to mine for: start_mining <addr> [threads=1]" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
cryptonote::account_public_address adr;
|
||||||
|
if(!cryptonote::get_account_address_from_str(adr, false, args.front()))
|
||||||
|
{
|
||||||
|
if(!cryptonote::get_account_address_from_str(adr, true, args.front()))
|
||||||
|
{
|
||||||
|
std::cout << "target account address has wrong format" << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
std::cout << "Mining to a testnet address, make sure this is intentional!" << std::endl;
|
||||||
|
}
|
||||||
|
uint64_t threads_count = 1;
|
||||||
|
if(args.size() > 2)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else if(args.size() == 2)
|
||||||
|
{
|
||||||
|
bool ok = epee::string_tools::get_xtype_from_string(threads_count, args[1]);
|
||||||
|
threads_count = (ok && 0 < threads_count) ? threads_count : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
m_executor.start_mining(adr, threads_count);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::stop_mining(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.stop_mining();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::stop_daemon(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.stop_daemon();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::print_status(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if (!args.empty()) return false;
|
||||||
|
|
||||||
|
return m_executor.print_status();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::set_limit(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size()!=1) return false;
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (limit==-1) limit=128;
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
return m_executor.set_limit(limit);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::set_limit_up(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size()!=1) return false;
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (limit==-1) limit=128;
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
return m_executor.set_limit_up(limit);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_parser_executor::set_limit_down(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size()!=1) return false;
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (limit==-1) limit=128;
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
return m_executor.set_limit_down(limit);
|
||||||
|
}
|
||||||
|
} // namespace daemonize
|
98
src/daemon/command_parser_executor.h
Normal file
98
src/daemon/command_parser_executor.h
Normal file
@ -0,0 +1,98 @@
|
|||||||
|
/**
|
||||||
|
@file
|
||||||
|
@details
|
||||||
|
|
||||||
|
@image html images/other/runtime-commands.png
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "daemon/rpc_command_executor.h"
|
||||||
|
#include "rpc/core_rpc_server.h"
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
class t_command_parser_executor final
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
t_rpc_command_executor m_executor;
|
||||||
|
public:
|
||||||
|
t_command_parser_executor(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc
|
||||||
|
, cryptonote::core_rpc_server* rpc_server = NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
bool print_peer_list(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool save_blockchain(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool show_hash_rate(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool hide_hash_rate(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool show_difficulty(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_connections(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_blockchain_info(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool set_log_level(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_height(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_block(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_transaction(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_transaction_pool_long(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_transaction_pool_short(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool start_mining(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool stop_mining(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool stop_daemon(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool print_status(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool set_limit(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool set_limit_up(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
bool set_limit_down(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace daemonize
|
213
src/daemon/command_server.cpp
Normal file
213
src/daemon/command_server.cpp
Normal file
@ -0,0 +1,213 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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 "cryptonote_config.h"
|
||||||
|
#include "version.h"
|
||||||
|
#include "daemon/command_server.h"
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
namespace p = std::placeholders;
|
||||||
|
|
||||||
|
t_command_server::t_command_server(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc
|
||||||
|
, cryptonote::core_rpc_server* rpc_server
|
||||||
|
)
|
||||||
|
: m_parser(ip, port, is_rpc, rpc_server)
|
||||||
|
, m_command_lookup()
|
||||||
|
, m_is_rpc(is_rpc)
|
||||||
|
{
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"q"
|
||||||
|
, [] (const std::vector<std::string>& args) {return true;}
|
||||||
|
, "ignored"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"help"
|
||||||
|
, std::bind(&t_command_server::help, this, p::_1)
|
||||||
|
, "Show this help"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_height"
|
||||||
|
, std::bind(&t_command_parser_executor::print_height, &m_parser, p::_1)
|
||||||
|
, "Print local blockchain height"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_pl"
|
||||||
|
, std::bind(&t_command_parser_executor::print_peer_list, &m_parser, p::_1)
|
||||||
|
, "Print peer list"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_cn"
|
||||||
|
, std::bind(&t_command_parser_executor::print_connections, &m_parser, p::_1)
|
||||||
|
, "Print connections"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_bc"
|
||||||
|
, std::bind(&t_command_parser_executor::print_blockchain_info, &m_parser, p::_1)
|
||||||
|
, "Print blockchain info in a given blocks range, print_bc <begin_height> [<end_height>]"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_block"
|
||||||
|
, std::bind(&t_command_parser_executor::print_block, &m_parser, p::_1)
|
||||||
|
, "Print block, print_block <block_hash> | <block_height>"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_tx"
|
||||||
|
, std::bind(&t_command_parser_executor::print_transaction, &m_parser, p::_1)
|
||||||
|
, "Print transaction, print_tx <transaction_hash>"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"start_mining"
|
||||||
|
, std::bind(&t_command_parser_executor::start_mining, &m_parser, p::_1)
|
||||||
|
, "Start mining for specified address, start_mining <addr> [threads=1]"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"stop_mining"
|
||||||
|
, std::bind(&t_command_parser_executor::stop_mining, &m_parser, p::_1)
|
||||||
|
, "Stop mining"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_pool"
|
||||||
|
, std::bind(&t_command_parser_executor::print_transaction_pool_long, &m_parser, p::_1)
|
||||||
|
, "Print transaction pool (long format)"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_pool_sh"
|
||||||
|
, std::bind(&t_command_parser_executor::print_transaction_pool_short, &m_parser, p::_1)
|
||||||
|
, "Print transaction pool (short format)"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"show_hr"
|
||||||
|
, std::bind(&t_command_parser_executor::show_hash_rate, &m_parser, p::_1)
|
||||||
|
, "Start showing hash rate"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"hide_hr"
|
||||||
|
, std::bind(&t_command_parser_executor::hide_hash_rate, &m_parser, p::_1)
|
||||||
|
, "Stop showing hash rate"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"save"
|
||||||
|
, std::bind(&t_command_parser_executor::save_blockchain, &m_parser, p::_1)
|
||||||
|
, "Save blockchain"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"set_log"
|
||||||
|
, std::bind(&t_command_parser_executor::set_log_level, &m_parser, p::_1)
|
||||||
|
, "set_log <level> - Change current log detalization level, <level> is a number 0-4"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"diff"
|
||||||
|
, std::bind(&t_command_parser_executor::show_difficulty, &m_parser, p::_1)
|
||||||
|
, "Show difficulty"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"stop_daemon"
|
||||||
|
, std::bind(&t_command_parser_executor::stop_daemon, &m_parser, p::_1)
|
||||||
|
, "Stop the daemon"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"exit"
|
||||||
|
, std::bind(&t_command_parser_executor::stop_daemon, &m_parser, p::_1)
|
||||||
|
, "Stop the daemon"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"print_status"
|
||||||
|
, std::bind(&t_command_parser_executor::print_status, &m_parser, p::_1)
|
||||||
|
, "Prints daemon status"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"limit"
|
||||||
|
, std::bind(&t_command_parser_executor::set_limit, &m_parser, p::_1)
|
||||||
|
, "limit <kB/s> - Set download and upload limit"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"limit-up"
|
||||||
|
, std::bind(&t_command_parser_executor::set_limit_up, &m_parser, p::_1)
|
||||||
|
, "limit <kB/s> - Set upload limit"
|
||||||
|
);
|
||||||
|
m_command_lookup.set_handler(
|
||||||
|
"limit-down"
|
||||||
|
, std::bind(&t_command_parser_executor::set_limit_down, &m_parser, p::_1)
|
||||||
|
, "limit <kB/s> - Set download limit"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_server::process_command_str(const std::string& cmd)
|
||||||
|
{
|
||||||
|
return m_command_lookup.process_command_str(cmd);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_server::process_command_vec(const std::vector<std::string>& cmd)
|
||||||
|
{
|
||||||
|
bool result = m_command_lookup.process_command_vec(cmd);
|
||||||
|
if (!result)
|
||||||
|
{
|
||||||
|
help(std::vector<std::string>());
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_server::start_handling()
|
||||||
|
{
|
||||||
|
if (m_is_rpc) return false;
|
||||||
|
|
||||||
|
m_command_lookup.start_handling("", get_commands_str());
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void t_command_server::stop_handling()
|
||||||
|
{
|
||||||
|
if (m_is_rpc) return;
|
||||||
|
|
||||||
|
m_command_lookup.stop_handling();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_command_server::help(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
std::cout << get_commands_str() << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string t_command_server::get_commands_str()
|
||||||
|
{
|
||||||
|
std::stringstream ss;
|
||||||
|
ss << CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL << std::endl;
|
||||||
|
ss << "Commands: " << std::endl;
|
||||||
|
std::string usage = m_command_lookup.get_usage();
|
||||||
|
boost::replace_all(usage, "\n", "\n ");
|
||||||
|
usage.insert(0, " ");
|
||||||
|
ss << usage << std::endl;
|
||||||
|
return ss.str();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace daemonize
|
75
src/daemon/command_server.h
Normal file
75
src/daemon/command_server.h
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
/**
|
||||||
|
@file
|
||||||
|
@details
|
||||||
|
|
||||||
|
|
||||||
|
Passing RPC commands:
|
||||||
|
|
||||||
|
@image html images/other/runtime-commands.png
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "console_handler.h"
|
||||||
|
#include "daemon/command_parser_executor.h"
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
class t_command_server {
|
||||||
|
private:
|
||||||
|
t_command_parser_executor m_parser;
|
||||||
|
epee::console_handlers_binder m_command_lookup;
|
||||||
|
bool m_is_rpc;
|
||||||
|
|
||||||
|
public:
|
||||||
|
t_command_server(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc = true
|
||||||
|
, cryptonote::core_rpc_server* rpc_server = NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
bool process_command_str(const std::string& cmd);
|
||||||
|
|
||||||
|
bool process_command_vec(const std::vector<std::string>& cmd);
|
||||||
|
|
||||||
|
bool start_handling();
|
||||||
|
|
||||||
|
void stop_handling();
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool help(const std::vector<std::string>& args);
|
||||||
|
|
||||||
|
std::string get_commands_str();
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace daemonize
|
99
src/daemon/core.h
Normal file
99
src/daemon/core.h
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "cryptonote_core/checkpoints_create.h"
|
||||||
|
#include "cryptonote_core/cryptonote_core.h"
|
||||||
|
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
#include "daemon/command_line_args.h"
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
|
||||||
|
class t_core final
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
static void init_options(boost::program_options::options_description & option_spec)
|
||||||
|
{
|
||||||
|
cryptonote::core::init_options(option_spec);
|
||||||
|
cryptonote::miner::init_options(option_spec);
|
||||||
|
}
|
||||||
|
private:
|
||||||
|
typedef cryptonote::t_cryptonote_protocol_handler<cryptonote::core> t_protocol_raw;
|
||||||
|
cryptonote::core m_core;
|
||||||
|
// TEMPORARY HACK - Yes, this creates a copy, but otherwise the original
|
||||||
|
// variable map could go out of scope before the run method is called
|
||||||
|
boost::program_options::variables_map const m_vm_HACK;
|
||||||
|
public:
|
||||||
|
t_core(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
: m_core{nullptr}
|
||||||
|
, m_vm_HACK{vm}
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO - get rid of circular dependencies in internals
|
||||||
|
void set_protocol(t_protocol_raw & protocol)
|
||||||
|
{
|
||||||
|
m_core.set_cryptonote_protocol(&protocol);
|
||||||
|
}
|
||||||
|
|
||||||
|
void run()
|
||||||
|
{
|
||||||
|
//initialize core here
|
||||||
|
LOG_PRINT_L0("Initializing core...");
|
||||||
|
if (!m_core.init(m_vm_HACK))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to initialize core");
|
||||||
|
}
|
||||||
|
LOG_PRINT_L0("Core initialized OK");
|
||||||
|
}
|
||||||
|
|
||||||
|
cryptonote::core & get()
|
||||||
|
{
|
||||||
|
return m_core;
|
||||||
|
}
|
||||||
|
|
||||||
|
~t_core()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Deinitializing core...");
|
||||||
|
try {
|
||||||
|
m_core.deinit();
|
||||||
|
m_core.set_cryptonote_protocol(nullptr);
|
||||||
|
} catch (...) {
|
||||||
|
LOG_PRINT_L0("Failed to deinitialize core...");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
@ -1,4 +1,4 @@
|
|||||||
// Copyright (c) 2014-2015, The Monero Project
|
// Copyright (c) 2014, The Monero Project
|
||||||
//
|
//
|
||||||
// All rights reserved.
|
// All rights reserved.
|
||||||
//
|
//
|
||||||
@ -28,302 +28,133 @@
|
|||||||
//
|
//
|
||||||
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
// node.cpp : Defines the entry point for the console application.
|
#include "daemon/daemon.h"
|
||||||
// Does this file exist?
|
|
||||||
|
|
||||||
|
#include "common/util.h"
|
||||||
#include "include_base_utils.h"
|
#include "daemon/core.h"
|
||||||
|
#include "daemon/p2p.h"
|
||||||
|
#include "daemon/protocol.h"
|
||||||
|
#include "daemon/rpc.h"
|
||||||
|
#include "daemon/command_server.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
#include "version.h"
|
#include "version.h"
|
||||||
#include "../../contrib/epee/include/syncobj.h"
|
|
||||||
|
|
||||||
using namespace epee;
|
|
||||||
|
|
||||||
#include <boost/program_options.hpp>
|
#include <boost/program_options.hpp>
|
||||||
|
#include <functional>
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
#include "crypto/hash.h"
|
namespace daemonize {
|
||||||
#include "console_handler.h"
|
|
||||||
#include "p2p/net_node.h"
|
|
||||||
#include "cryptonote_config.h"
|
|
||||||
#include "cryptonote_core/checkpoints_create.h"
|
|
||||||
#include "cryptonote_core/checkpoints.h"
|
|
||||||
#include "cryptonote_core/cryptonote_core.h"
|
|
||||||
#include "rpc/core_rpc_server.h"
|
|
||||||
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
|
||||||
#include "daemon_commands_handler.h"
|
|
||||||
#include "version.h"
|
|
||||||
|
|
||||||
#if defined(WIN32)
|
struct t_internals {
|
||||||
#include <crtdbg.h>
|
private:
|
||||||
#endif
|
t_protocol protocol;
|
||||||
|
public:
|
||||||
|
t_core core;
|
||||||
|
t_p2p p2p;
|
||||||
|
t_rpc rpc;
|
||||||
|
|
||||||
namespace po = boost::program_options;
|
t_internals(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
: core{vm}
|
||||||
|
, protocol{vm, core}
|
||||||
|
, p2p{vm, protocol}
|
||||||
|
, rpc{vm, core, p2p}
|
||||||
|
{
|
||||||
|
// Handle circular dependencies
|
||||||
|
protocol.set_p2p_endpoint(p2p.get());
|
||||||
|
core.set_protocol(protocol.get());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
unsigned int epee::g_test_dbg_lock_sleep = 0;
|
void t_daemon::init_options(boost::program_options::options_description & option_spec)
|
||||||
|
|
||||||
namespace
|
|
||||||
{
|
{
|
||||||
const command_line::arg_descriptor<std::string> arg_config_file = {"config-file", "Specify configuration file", std::string(CRYPTONOTE_NAME ".conf")};
|
t_core::init_options(option_spec);
|
||||||
const command_line::arg_descriptor<bool> arg_os_version = {"os-version", ""};
|
t_p2p::init_options(option_spec);
|
||||||
const command_line::arg_descriptor<std::string> arg_log_file = {"log-file", "", ""};
|
t_rpc::init_options(option_spec);
|
||||||
const command_line::arg_descriptor<int> arg_log_level = {"log-level", "", LOG_LEVEL_0};
|
|
||||||
const command_line::arg_descriptor<bool> arg_console = {"no-console", "Disable daemon console commands"};
|
|
||||||
const command_line::arg_descriptor<bool> arg_testnet_on = {
|
|
||||||
"testnet"
|
|
||||||
, "Run on testnet. The wallet must be launched with --testnet flag."
|
|
||||||
, false
|
|
||||||
};
|
|
||||||
const command_line::arg_descriptor<bool> arg_dns_checkpoints = {"enforce-dns-checkpointing", "checkpoints from DNS server will be enforced", false};
|
|
||||||
const command_line::arg_descriptor<bool> arg_test_drop_download = {"test-drop-download", "For net tests: in download, discard ALL blocks instead checking/saving them (very fast)"};
|
|
||||||
const command_line::arg_descriptor<uint64_t> arg_test_drop_download_height = {"test-drop-download-height", "Like test-drop-download but disards only after around certain height", 0};
|
|
||||||
const command_line::arg_descriptor<bool> arg_save_graph = {"save-graph", "Save data for dr monero", false};
|
|
||||||
const command_line::arg_descriptor<int> test_dbg_lock_sleep = {"test-dbg-lock-sleep", "Sleep time in ms, defaults to 0 (off), used to debug before/after locking mutex. Values 100 to 1000 are good for tests.", 0};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool command_line_preprocessor(const boost::program_options::variables_map& vm)
|
t_daemon::t_daemon(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
: mp_internals{new t_internals{vm}}
|
||||||
|
{}
|
||||||
|
|
||||||
|
t_daemon::~t_daemon() = default;
|
||||||
|
|
||||||
|
// MSVC is brain-dead and can't default this...
|
||||||
|
t_daemon::t_daemon(t_daemon && other)
|
||||||
{
|
{
|
||||||
bool exit = false;
|
if (this != &other)
|
||||||
if (command_line::get_arg(vm, command_line::arg_version))
|
|
||||||
{
|
{
|
||||||
std::cout << CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL << ENDL;
|
mp_internals = std::move(other.mp_internals);
|
||||||
exit = true;
|
other.mp_internals.reset(nullptr);
|
||||||
}
|
}
|
||||||
if (command_line::get_arg(vm, arg_os_version))
|
|
||||||
{
|
|
||||||
std::cout << "OS: " << tools::get_os_version_string() << ENDL;
|
|
||||||
exit = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (exit)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
int new_log_level = command_line::get_arg(vm, arg_log_level);
|
|
||||||
if(new_log_level < LOG_LEVEL_MIN || new_log_level > LOG_LEVEL_MAX)
|
|
||||||
{
|
|
||||||
LOG_PRINT_L0("Wrong log level value: ");
|
|
||||||
}
|
|
||||||
else if (log_space::get_set_log_detalisation_level(false) != new_log_level)
|
|
||||||
{
|
|
||||||
log_space::get_set_log_detalisation_level(true, new_log_level);
|
|
||||||
int otshell_utils_log_level = 100 - (new_log_level * 25);
|
|
||||||
gCurrentLogger.setDebugLevel(otshell_utils_log_level);
|
|
||||||
LOG_PRINT_L0("LOG_LEVEL set to " << new_log_level);
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(int argc, char* argv[])
|
// or this
|
||||||
|
t_daemon & t_daemon::operator=(t_daemon && other)
|
||||||
{
|
{
|
||||||
string_tools::set_module_name_and_folder(argv[0]);
|
if (this != &other)
|
||||||
#ifdef WIN32
|
|
||||||
_CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
|
|
||||||
#endif
|
|
||||||
log_space::get_set_log_detalisation_level(true, LOG_LEVEL_0);
|
|
||||||
log_space::log_singletone::add_logger(LOGGER_CONSOLE, NULL, NULL);
|
|
||||||
LOG_PRINT_L0("Starting...");
|
|
||||||
|
|
||||||
nOT::nUtils::cFilesystemUtils::CreateDirTree("log/dr-monero/net/");
|
|
||||||
_warn_c("test","Starting program (a test message)");
|
|
||||||
_warn_c("main/program","Starting program");
|
|
||||||
|
|
||||||
TRY_ENTRY();
|
|
||||||
|
|
||||||
boost::filesystem::path default_data_path {tools::get_default_data_dir()};
|
|
||||||
boost::filesystem::path default_testnet_data_path {default_data_path / "testnet"};
|
|
||||||
|
|
||||||
po::options_description desc_cmd_only("Command line options");
|
|
||||||
po::options_description desc_cmd_sett("Command line options and settings options");
|
|
||||||
|
|
||||||
command_line::add_arg(desc_cmd_only, command_line::arg_help);
|
|
||||||
command_line::add_arg(desc_cmd_only, command_line::arg_version);
|
|
||||||
command_line::add_arg(desc_cmd_only, arg_os_version);
|
|
||||||
// tools::get_default_data_dir() can't be called during static initialization
|
|
||||||
command_line::add_arg(desc_cmd_only, command_line::arg_data_dir, default_data_path.string());
|
|
||||||
command_line::add_arg(desc_cmd_only, command_line::arg_testnet_data_dir, default_testnet_data_path.string());
|
|
||||||
command_line::add_arg(desc_cmd_only, arg_config_file);
|
|
||||||
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_log_file);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_log_level);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_console);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_testnet_on);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_dns_checkpoints);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_test_drop_download);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_test_drop_download_height);
|
|
||||||
command_line::add_arg(desc_cmd_sett, arg_save_graph);
|
|
||||||
command_line::add_arg(desc_cmd_sett, test_dbg_lock_sleep);
|
|
||||||
|
|
||||||
cryptonote::core::init_options(desc_cmd_sett);
|
|
||||||
cryptonote::core_rpc_server::init_options(desc_cmd_sett);
|
|
||||||
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >::init_options(desc_cmd_sett);
|
|
||||||
cryptonote::miner::init_options(desc_cmd_sett);
|
|
||||||
|
|
||||||
po::options_description desc_options("Allowed options");
|
|
||||||
desc_options.add(desc_cmd_only).add(desc_cmd_sett);
|
|
||||||
|
|
||||||
po::variables_map vm;
|
|
||||||
bool r = command_line::handle_error_helper(desc_options, [&]()
|
|
||||||
{
|
{
|
||||||
po::store(po::parse_command_line(argc, argv, desc_options), vm);
|
mp_internals = std::move(other.mp_internals);
|
||||||
po::notify(vm);
|
other.mp_internals.reset(nullptr);
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
return true;
|
bool t_daemon::run(bool interactive)
|
||||||
});
|
{
|
||||||
if (!r)
|
if (nullptr == mp_internals)
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (command_line::get_arg(vm, command_line::arg_help))
|
|
||||||
{
|
{
|
||||||
std::cout << CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL << ENDL << ENDL;
|
throw std::runtime_error{"Can't run stopped daemon"};
|
||||||
std::cout << desc_options << std::endl;
|
}
|
||||||
return false;
|
tools::signal_handler::install(std::bind(&daemonize::t_daemon::stop, this));
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
mp_internals->core.run();
|
||||||
|
mp_internals->rpc.run();
|
||||||
|
|
||||||
|
daemonize::t_command_server* rpc_commands;
|
||||||
|
|
||||||
|
if (interactive)
|
||||||
|
{
|
||||||
|
rpc_commands = new daemonize::t_command_server(0, 0, false, mp_internals->rpc.get_server());
|
||||||
|
rpc_commands->start_handling();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool testnet_mode = command_line::get_arg(vm, arg_testnet_on);
|
mp_internals->p2p.run(); // blocks until p2p goes down
|
||||||
|
|
||||||
auto data_dir_arg = testnet_mode ? command_line::arg_testnet_data_dir : command_line::arg_data_dir;
|
if (interactive)
|
||||||
|
|
||||||
std::string data_dir = command_line::get_arg(vm, data_dir_arg);
|
|
||||||
tools::create_directories_if_necessary(data_dir);
|
|
||||||
std::string config = command_line::get_arg(vm, arg_config_file);
|
|
||||||
|
|
||||||
boost::filesystem::path data_dir_path(data_dir);
|
|
||||||
boost::filesystem::path config_path(config);
|
|
||||||
if (!config_path.has_parent_path())
|
|
||||||
{
|
{
|
||||||
config_path = data_dir_path / config_path;
|
rpc_commands->stop_handling();
|
||||||
}
|
}
|
||||||
|
|
||||||
boost::system::error_code ec;
|
mp_internals->rpc.stop();
|
||||||
if (boost::filesystem::exists(config_path, ec))
|
|
||||||
{
|
|
||||||
po::store(po::parse_config_file<char>(config_path.string<std::string>().c_str(), desc_cmd_sett), vm);
|
|
||||||
}
|
|
||||||
|
|
||||||
//set up logging options
|
|
||||||
boost::filesystem::path log_file_path(command_line::get_arg(vm, arg_log_file));
|
|
||||||
if (log_file_path.empty())
|
|
||||||
log_file_path = log_space::log_singletone::get_default_log_file();
|
|
||||||
std::string log_dir;
|
|
||||||
log_dir = log_file_path.has_parent_path() ? log_file_path.parent_path().string() : log_space::log_singletone::get_default_log_folder();
|
|
||||||
|
|
||||||
log_space::log_singletone::add_logger(LOGGER_FILE, log_file_path.filename().string().c_str(), log_dir.c_str());
|
|
||||||
LOG_PRINT_L0(CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL);
|
|
||||||
|
|
||||||
if (command_line_preprocessor(vm))
|
|
||||||
{
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
LOG_PRINT("Module folder: " << argv[0], LOG_LEVEL_0);
|
|
||||||
|
|
||||||
bool res = true;
|
|
||||||
cryptonote::checkpoints checkpoints;
|
|
||||||
res = cryptonote::create_checkpoints(checkpoints);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize checkpoints");
|
|
||||||
boost::filesystem::path json(JSON_HASH_FILE_NAME);
|
|
||||||
boost::filesystem::path checkpoint_json_hashfile_fullpath = data_dir / json;
|
|
||||||
|
|
||||||
//create objects and link them
|
|
||||||
cryptonote::core ccore(NULL);
|
|
||||||
|
|
||||||
// tell core if we're enforcing dns checkpoints
|
|
||||||
bool enforce_dns = command_line::get_arg(vm, arg_dns_checkpoints);
|
|
||||||
ccore.set_enforce_dns_checkpoints(enforce_dns);
|
|
||||||
|
|
||||||
if (testnet_mode) {
|
|
||||||
LOG_PRINT_L0("Starting in testnet mode!");
|
|
||||||
} else {
|
|
||||||
ccore.set_checkpoints(std::move(checkpoints));
|
|
||||||
ccore.set_checkpoints_file_path(checkpoint_json_hashfile_fullpath.string());
|
|
||||||
}
|
|
||||||
|
|
||||||
cryptonote::t_cryptonote_protocol_handler<cryptonote::core> cprotocol(ccore, NULL);
|
|
||||||
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> > p2psrv {
|
|
||||||
cprotocol
|
|
||||||
, testnet_mode ? std::move(config::testnet::NETWORK_ID) : std::move(config::NETWORK_ID)
|
|
||||||
};
|
|
||||||
cryptonote::core_rpc_server rpc_server {ccore, p2psrv, testnet_mode};
|
|
||||||
cprotocol.set_p2p_endpoint(&p2psrv);
|
|
||||||
ccore.set_cryptonote_protocol(&cprotocol);
|
|
||||||
std::shared_ptr<daemon_cmmands_handler> dch(new daemon_cmmands_handler(p2psrv, testnet_mode));
|
|
||||||
if(command_line::has_arg(vm, arg_save_graph))
|
|
||||||
p2psrv.set_save_graph(true);
|
|
||||||
|
|
||||||
epee::g_test_dbg_lock_sleep = command_line::get_arg(vm, test_dbg_lock_sleep);
|
|
||||||
|
|
||||||
//initialize core here
|
|
||||||
LOG_PRINT_L0("Initializing core...");
|
|
||||||
res = ccore.init(vm, testnet_mode);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize core");
|
|
||||||
if (command_line::get_arg(vm, arg_test_drop_download))
|
|
||||||
ccore.test_drop_download();
|
|
||||||
|
|
||||||
ccore.test_drop_download_height(command_line::get_arg(vm, arg_test_drop_download_height));
|
|
||||||
LOG_PRINT_L0("Core initialized OK");
|
|
||||||
|
|
||||||
//initialize objects
|
|
||||||
LOG_PRINT_L0("Initializing P2P server...");
|
|
||||||
res = p2psrv.init(vm, testnet_mode);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize P2P server.");
|
|
||||||
LOG_PRINT_L0("P2P server initialized OK");
|
|
||||||
|
|
||||||
LOG_PRINT_L0("Initializing protocol...");
|
|
||||||
res = cprotocol.init(vm);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize protocol.");
|
|
||||||
LOG_PRINT_L0("Protocol initialized OK");
|
|
||||||
|
|
||||||
LOG_PRINT_L0("Initializing core RPC server...");
|
|
||||||
res = rpc_server.init(vm);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize core RPC server.");
|
|
||||||
LOG_PRINT_GREEN("Core RPC server initialized OK on port: " << rpc_server.get_binded_port(), LOG_LEVEL_0);
|
|
||||||
|
|
||||||
// start components
|
|
||||||
if(!command_line::has_arg(vm, arg_console))
|
|
||||||
{
|
|
||||||
dch->start_handling();
|
|
||||||
}
|
|
||||||
|
|
||||||
LOG_PRINT_L0("Starting core RPC server...");
|
|
||||||
res = rpc_server.run(2, false);
|
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize core RPC server.");
|
|
||||||
LOG_PRINT_L0("Core RPC server started ok");
|
|
||||||
|
|
||||||
tools::signal_handler::install([&dch, &p2psrv] {
|
|
||||||
dch->stop_handling();
|
|
||||||
p2psrv.send_stop_signal();
|
|
||||||
});
|
|
||||||
|
|
||||||
LOG_PRINT_L0("Starting P2P net loop...");
|
|
||||||
p2psrv.run();
|
|
||||||
LOG_PRINT_L0("P2P net loop stopped");
|
|
||||||
|
|
||||||
//stop components
|
|
||||||
dch->stop_handling();
|
|
||||||
dch.reset();
|
|
||||||
LOG_PRINT_L0("Stopping core rpc server...");
|
|
||||||
rpc_server.send_stop_signal();
|
|
||||||
rpc_server.timed_wait_server_stop(5000);
|
|
||||||
|
|
||||||
//deinitialize components
|
|
||||||
LOG_PRINT_L0("Deinitializing core...");
|
|
||||||
ccore.deinit();
|
|
||||||
LOG_PRINT_L0("Deinitializing RPC server ...");
|
|
||||||
rpc_server.deinit();
|
|
||||||
LOG_PRINT_L0("Deinitializing protocol...");
|
|
||||||
cprotocol.deinit();
|
|
||||||
LOG_PRINT_L0("Deinitializing P2P...");
|
|
||||||
p2psrv.deinit();
|
|
||||||
|
|
||||||
|
|
||||||
ccore.set_cryptonote_protocol(NULL);
|
|
||||||
cprotocol.set_p2p_endpoint(NULL);
|
|
||||||
|
|
||||||
epee::net_utils::data_logger::kill_instance();
|
|
||||||
LOG_PRINT("Node stopped.", LOG_LEVEL_0);
|
LOG_PRINT("Node stopped.", LOG_LEVEL_0);
|
||||||
return 0;
|
return true;
|
||||||
|
}
|
||||||
CATCH_ENTRY_L0("main", 1);
|
catch (std::exception const & ex)
|
||||||
|
{
|
||||||
|
LOG_ERROR("Uncaught exception! " << ex.what());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
catch (...)
|
||||||
|
{
|
||||||
|
LOG_ERROR("Uncaught exception!");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void t_daemon::stop()
|
||||||
|
{
|
||||||
|
if (nullptr == mp_internals)
|
||||||
|
{
|
||||||
|
throw std::runtime_error{"Can't stop stopped daemon"};
|
||||||
|
}
|
||||||
|
mp_internals->p2p.stop();
|
||||||
|
mp_internals->rpc.stop();
|
||||||
|
mp_internals.reset(nullptr); // Ensure resources are cleaned up before we return
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace daemonize
|
||||||
|
54
src/daemon/daemon.h
Normal file
54
src/daemon/daemon.h
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
class t_internals;
|
||||||
|
|
||||||
|
class t_daemon final {
|
||||||
|
public:
|
||||||
|
static void init_options(boost::program_options::options_description & option_spec);
|
||||||
|
private:
|
||||||
|
std::unique_ptr<t_internals> mp_internals;
|
||||||
|
public:
|
||||||
|
t_daemon(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
);
|
||||||
|
t_daemon(t_daemon && other);
|
||||||
|
t_daemon & operator=(t_daemon && other);
|
||||||
|
~t_daemon();
|
||||||
|
|
||||||
|
bool run(bool interactive = false);
|
||||||
|
void stop();
|
||||||
|
};
|
||||||
|
}
|
@ -1,35 +1,7 @@
|
|||||||
// Copyright (c) 2014-2015, The Monero Project
|
// Copyright (c) 2012-2013 The Cryptonote developers
|
||||||
//
|
// Distributed under the MIT/X11 software license, see the accompanying
|
||||||
// All rights reserved.
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||||
//
|
|
||||||
// Redistribution and use in source and binary forms, with or without modification, are
|
|
||||||
// permitted provided that the following conditions are met:
|
|
||||||
//
|
|
||||||
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
|
||||||
// conditions and the following disclaimer.
|
|
||||||
//
|
|
||||||
// 2. 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.
|
|
||||||
//
|
|
||||||
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
|
||||||
//
|
|
||||||
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
|
||||||
|
|
||||||
|
|
||||||
/* This isn't a header file, may want to refactor this... */
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <boost/lexical_cast.hpp>
|
#include <boost/lexical_cast.hpp>
|
||||||
@ -42,21 +14,18 @@
|
|||||||
#include "version.h"
|
#include "version.h"
|
||||||
#include "../../contrib/otshell_utils/utils.hpp"
|
#include "../../contrib/otshell_utils/utils.hpp"
|
||||||
|
|
||||||
/*!
|
//#include "net/net_helper.h"
|
||||||
* \brief I don't really know right now
|
//#include "../p2p/p2p_protocol_defs.h"
|
||||||
*
|
//#include "../p2p/net_peerlist_boost_serialization.h"
|
||||||
*
|
//#include "net/local_ip.h"
|
||||||
*/
|
//#include "crypto/crypto.h"
|
||||||
|
//#include "storages/levin_abstract_invoke2.h"
|
||||||
|
|
||||||
class daemon_cmmands_handler
|
class daemon_cmmands_handler
|
||||||
{
|
{
|
||||||
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& m_srv;
|
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& m_srv;
|
||||||
public:
|
public:
|
||||||
daemon_cmmands_handler(
|
daemon_cmmands_handler(nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& srv):m_srv(srv)
|
||||||
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& srv
|
|
||||||
, bool testnet
|
|
||||||
)
|
|
||||||
: m_srv(srv)
|
|
||||||
, m_testnet(testnet)
|
|
||||||
{
|
{
|
||||||
m_cmd_binder.set_handler("help", boost::bind(&daemon_cmmands_handler::help, this, _1), "Show this help");
|
m_cmd_binder.set_handler("help", boost::bind(&daemon_cmmands_handler::help, this, _1), "Show this help");
|
||||||
m_cmd_binder.set_handler("print_pl", boost::bind(&daemon_cmmands_handler::print_pl, this, _1), "Print peer list");
|
m_cmd_binder.set_handler("print_pl", boost::bind(&daemon_cmmands_handler::print_pl, this, _1), "Print peer list");
|
||||||
@ -75,14 +44,10 @@ public:
|
|||||||
m_cmd_binder.set_handler("save", boost::bind(&daemon_cmmands_handler::save, this, _1), "Save blockchain");
|
m_cmd_binder.set_handler("save", boost::bind(&daemon_cmmands_handler::save, this, _1), "Save blockchain");
|
||||||
m_cmd_binder.set_handler("set_log", boost::bind(&daemon_cmmands_handler::set_log, this, _1), "set_log <level> - Change current log detalization level, <level> is a number 0-4");
|
m_cmd_binder.set_handler("set_log", boost::bind(&daemon_cmmands_handler::set_log, this, _1), "set_log <level> - Change current log detalization level, <level> is a number 0-4");
|
||||||
m_cmd_binder.set_handler("diff", boost::bind(&daemon_cmmands_handler::diff, this, _1), "Show difficulty");
|
m_cmd_binder.set_handler("diff", boost::bind(&daemon_cmmands_handler::diff, this, _1), "Show difficulty");
|
||||||
|
m_cmd_binder.set_handler("limit-up", boost::bind(&daemon_cmmands_handler::limit_up, this, _1), "Set upload limit");
|
||||||
|
m_cmd_binder.set_handler("limit-down", boost::bind(&daemon_cmmands_handler::limit_down, this, _1), "Set download limit");
|
||||||
|
m_cmd_binder.set_handler("limit", boost::bind(&daemon_cmmands_handler::limit, this, _1), "Set download and upload limit");
|
||||||
m_cmd_binder.set_handler("out_peers", boost::bind(&daemon_cmmands_handler::out_peers_limit, this, _1), "Set max limit of out peers");
|
m_cmd_binder.set_handler("out_peers", boost::bind(&daemon_cmmands_handler::out_peers_limit, this, _1), "Set max limit of out peers");
|
||||||
m_cmd_binder.set_handler("limit_up", boost::bind(&daemon_cmmands_handler::limit_up, this, _1), "Set upload limit [kB/s]");
|
|
||||||
m_cmd_binder.set_handler("limit_down", boost::bind(&daemon_cmmands_handler::limit_down, this, _1), "Set download limit [kB/s]");
|
|
||||||
m_cmd_binder.set_handler("limit", boost::bind(&daemon_cmmands_handler::limit, this, _1), "Set download and upload limit [kB/s]");
|
|
||||||
m_cmd_binder.set_handler("fast_exit", boost::bind(&daemon_cmmands_handler::fast_exit, this, _1), "Exit");
|
|
||||||
m_cmd_binder.set_handler("test_drop_download", boost::bind(&daemon_cmmands_handler::test_drop_download, this, _1), "For network testing, drop downloaded blocks instead checking/adding them to blockchain. Can fake-download blocks very fast.");
|
|
||||||
m_cmd_binder.set_handler("start_save_graph", boost::bind(&daemon_cmmands_handler::start_save_graph, this, _1), "");
|
|
||||||
m_cmd_binder.set_handler("stop_save_graph", boost::bind(&daemon_cmmands_handler::stop_save_graph, this, _1), "");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool start_handling()
|
bool start_handling()
|
||||||
@ -98,7 +63,7 @@ public:
|
|||||||
|
|
||||||
private:
|
private:
|
||||||
epee::srv_console_handlers_binder<nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> > > m_cmd_binder;
|
epee::srv_console_handlers_binder<nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> > > m_cmd_binder;
|
||||||
bool m_testnet;
|
|
||||||
|
|
||||||
//--------------------------------------------------------------------------------
|
//--------------------------------------------------------------------------------
|
||||||
std::string get_commands_str()
|
std::string get_commands_str()
|
||||||
@ -130,6 +95,122 @@ private:
|
|||||||
m_srv.get_payload_object().get_core().get_blockchain_storage().store_blockchain();
|
m_srv.get_payload_object().get_core().get_blockchain_storage().store_blockchain();
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
//--------------------------------------------------------------------------------
|
||||||
|
bool limit_up(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size()!=1) {
|
||||||
|
std::cout << "Usage: limit_up <speed>" << ENDL;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (limit==-1) {
|
||||||
|
limit=128;
|
||||||
|
//this->islimitup=false;
|
||||||
|
}
|
||||||
|
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
|
||||||
|
//nodetool::epee::net_utils::connection<epee::levin::async_protocol_handler<nodetool::p2p_connection_context> >::set_rate_up_limit( limit );
|
||||||
|
epee::net_utils::connection_basic::set_rate_up_limit( limit );
|
||||||
|
std::cout << "Set limit-up to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
//--------------------------------------------------------------------------------
|
||||||
|
bool limit_down(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
|
||||||
|
if(args.size()!=1) {
|
||||||
|
std::cout << "Usage: limit_down <speed>" << ENDL;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (limit==-1) {
|
||||||
|
limit=128;
|
||||||
|
//this->islimitup=false;
|
||||||
|
}
|
||||||
|
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
|
||||||
|
//nodetool::epee::net_utils::connection<epee::levin::async_protocol_handler<nodetool::p2p_connection_context> >::set_rate_up_limit( limit );
|
||||||
|
epee::net_utils::connection_basic::set_rate_down_limit( limit );
|
||||||
|
std::cout << "Set limit-down to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
//--------------------------------------------------------------------------------
|
||||||
|
bool limit(const std::vector<std::string>& args)
|
||||||
|
{
|
||||||
|
if(args.size()!=1) {
|
||||||
|
std::cout << "Usage: limit_down <speed>" << ENDL;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (limit==-1) {
|
||||||
|
limit=128;
|
||||||
|
//this->islimitup=false;
|
||||||
|
}
|
||||||
|
|
||||||
|
limit *= 1024;
|
||||||
|
|
||||||
|
|
||||||
|
//nodetool::epee::net_utils::connection<epee::levin::async_protocol_handler<nodetool::p2p_connection_context> >::set_rate_up_limit( limit );
|
||||||
|
epee::net_utils::connection_basic::set_rate_down_limit( limit );
|
||||||
|
epee::net_utils::connection_basic::set_rate_up_limit( limit );
|
||||||
|
std::cout << "Set limit-down to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
std::cout << "Set limit-up to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//--------------------------------------------------------------------------------
|
||||||
|
bool out_peers_limit(const std::vector<std::string>& args) {
|
||||||
|
if(args.size()!=1) {
|
||||||
|
std::cout << "Usage: limit_down <speed>" << ENDL;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
int limit;
|
||||||
|
try {
|
||||||
|
limit = std::stoi(args[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
catch(std::invalid_argument& ex) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_PRINT_RED_L0("connections_count: " << limit);
|
||||||
|
m_srv.m_config.m_net_config.connections_count = limit;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
//--------------------------------------------------------------------------------
|
//--------------------------------------------------------------------------------
|
||||||
bool show_hr(const std::vector<std::string>& args)
|
bool show_hr(const std::vector<std::string>& args)
|
||||||
{
|
{
|
||||||
@ -243,9 +324,11 @@ private:
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO what the hell causes compilation warning in following code line
|
||||||
|
PUSH_WARNINGS
|
||||||
|
DISABLE_GCC_WARNING(maybe-uninitialized)
|
||||||
log_space::log_singletone::get_set_log_detalisation_level(true, l);
|
log_space::log_singletone::get_set_log_detalisation_level(true, l);
|
||||||
int otshell_utils_log_level = 100 - (l * 25);
|
POP_WARNINGS
|
||||||
gCurrentLogger.setDebugLevel(otshell_utils_log_level);
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -385,7 +468,7 @@ private:
|
|||||||
}
|
}
|
||||||
|
|
||||||
cryptonote::account_public_address adr;
|
cryptonote::account_public_address adr;
|
||||||
if(!cryptonote::get_account_address_from_str(adr, m_testnet, args.front()))
|
if(!cryptonote::get_account_address_from_str(adr, args.front()))
|
||||||
{
|
{
|
||||||
std::cout << "target account address has wrong format" << std::endl;
|
std::cout << "target account address has wrong format" << std::endl;
|
||||||
return true;
|
return true;
|
||||||
|
71
src/daemon/executor.cpp
Normal file
71
src/daemon/executor.cpp
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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 "daemon/executor.h"
|
||||||
|
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
|
||||||
|
#include "common/command_line.h"
|
||||||
|
#include "cryptonote_config.h"
|
||||||
|
#include "version.h"
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
std::string const t_executor::NAME = "Monero Daemon";
|
||||||
|
|
||||||
|
void t_executor::init_options(
|
||||||
|
boost::program_options::options_description & configurable_options
|
||||||
|
)
|
||||||
|
{
|
||||||
|
t_daemon::init_options(configurable_options);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string const & t_executor::name()
|
||||||
|
{
|
||||||
|
return NAME;
|
||||||
|
}
|
||||||
|
|
||||||
|
t_daemon t_executor::create_daemon(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0(CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL);
|
||||||
|
return t_daemon{vm};
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_executor::run_interactive(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
epee::log_space::log_singletone::add_logger(LOGGER_CONSOLE, NULL, NULL);
|
||||||
|
return t_daemon{vm}.run(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
60
src/daemon/executor.h
Normal file
60
src/daemon/executor.h
Normal file
@ -0,0 +1,60 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "daemon/daemon.h"
|
||||||
|
#include <boost/program_options/options_description.hpp>
|
||||||
|
#include <boost/program_options/variables_map.hpp>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
class t_executor final
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
typedef ::daemonize::t_daemon t_daemon;
|
||||||
|
|
||||||
|
static std::string const NAME;
|
||||||
|
|
||||||
|
static void init_options(
|
||||||
|
boost::program_options::options_description & configurable_options
|
||||||
|
);
|
||||||
|
|
||||||
|
std::string const & name();
|
||||||
|
|
||||||
|
t_daemon create_daemon(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
);
|
||||||
|
|
||||||
|
bool run_interactive(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
);
|
||||||
|
};
|
||||||
|
}
|
238
src/daemon/main.cpp
Normal file
238
src/daemon/main.cpp
Normal file
@ -0,0 +1,238 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#include "common/command_line.h"
|
||||||
|
#include "common/scoped_message_writer.h"
|
||||||
|
#include "common/util.h"
|
||||||
|
#include "cryptonote_core/cryptonote_core.h"
|
||||||
|
#include "cryptonote_core/miner.h"
|
||||||
|
#include "daemon/command_server.h"
|
||||||
|
#include "daemon/daemon.h"
|
||||||
|
#include "daemon/executor.h"
|
||||||
|
#include "daemonizer/daemonizer.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include "p2p/net_node.h"
|
||||||
|
#include "rpc/core_rpc_server.h"
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
#include "daemon/command_line_args.h"
|
||||||
|
|
||||||
|
namespace po = boost::program_options;
|
||||||
|
namespace bf = boost::filesystem;
|
||||||
|
|
||||||
|
int main(int argc, char const * argv[])
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
|
||||||
|
epee::string_tools::set_module_name_and_folder(argv[0]);
|
||||||
|
|
||||||
|
// Build argument description
|
||||||
|
po::options_description all_options("All");
|
||||||
|
po::options_description hidden_options("Hidden");
|
||||||
|
po::options_description visible_options("Options");
|
||||||
|
po::options_description core_settings("Settings");
|
||||||
|
po::positional_options_description positional_options;
|
||||||
|
{
|
||||||
|
bf::path default_data_dir = daemonizer::get_default_data_dir();
|
||||||
|
bf::path default_testnet_data_dir = {default_data_dir / "testnet"};
|
||||||
|
|
||||||
|
// Misc Options
|
||||||
|
|
||||||
|
command_line::add_arg(visible_options, command_line::arg_help);
|
||||||
|
command_line::add_arg(visible_options, command_line::arg_version);
|
||||||
|
command_line::add_arg(visible_options, daemon_args::arg_os_version);
|
||||||
|
command_line::add_arg(visible_options, command_line::arg_data_dir, default_data_dir.string());
|
||||||
|
command_line::add_arg(visible_options, command_line::arg_testnet_data_dir, default_testnet_data_dir.string());
|
||||||
|
bf::path default_conf = default_data_dir / std::string(CRYPTONOTE_NAME ".conf");
|
||||||
|
command_line::add_arg(visible_options, daemon_args::arg_config_file, default_conf.string());
|
||||||
|
|
||||||
|
// Settings
|
||||||
|
bf::path default_log = default_data_dir / std::string(CRYPTONOTE_NAME ".log");
|
||||||
|
command_line::add_arg(core_settings, daemon_args::arg_log_file, default_log.string());
|
||||||
|
command_line::add_arg(core_settings, daemon_args::arg_log_level);
|
||||||
|
command_line::add_arg(core_settings, daemon_args::arg_testnet_on);
|
||||||
|
command_line::add_arg(core_settings, daemon_args::arg_dns_checkpoints);
|
||||||
|
daemonizer::init_options(hidden_options, visible_options);
|
||||||
|
daemonize::t_executor::init_options(core_settings);
|
||||||
|
|
||||||
|
// Hidden options
|
||||||
|
command_line::add_arg(hidden_options, daemon_args::arg_command);
|
||||||
|
|
||||||
|
visible_options.add(core_settings);
|
||||||
|
all_options.add(visible_options);
|
||||||
|
all_options.add(hidden_options);
|
||||||
|
|
||||||
|
// Positional
|
||||||
|
positional_options.add(daemon_args::arg_command.name, -1); // -1 for unlimited arguments
|
||||||
|
}
|
||||||
|
|
||||||
|
// Do command line parsing
|
||||||
|
po::variables_map vm;
|
||||||
|
bool ok = command_line::handle_error_helper(visible_options, [&]()
|
||||||
|
{
|
||||||
|
boost::program_options::store(
|
||||||
|
boost::program_options::command_line_parser(argc, argv)
|
||||||
|
.options(all_options).positional(positional_options).run()
|
||||||
|
, vm
|
||||||
|
);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
if (!ok) return 1;
|
||||||
|
|
||||||
|
if (command_line::get_arg(vm, command_line::arg_help))
|
||||||
|
{
|
||||||
|
std::cout << CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL << ENDL << ENDL;
|
||||||
|
std::cout << "Usage: " + std::string{argv[0]} + " [options|settings] [daemon_command...]" << std::endl << std::endl;
|
||||||
|
std::cout << visible_options << std::endl;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Monero Version
|
||||||
|
if (command_line::get_arg(vm, command_line::arg_version))
|
||||||
|
{
|
||||||
|
std::cout << CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL << ENDL;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// OS
|
||||||
|
if (command_line::get_arg(vm, daemon_args::arg_os_version))
|
||||||
|
{
|
||||||
|
std::cout << "OS: " << tools::get_os_version_string() << ENDL;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool testnet_mode = command_line::get_arg(vm, daemon_args::arg_testnet_on);
|
||||||
|
|
||||||
|
auto data_dir_arg = testnet_mode ? command_line::arg_testnet_data_dir : command_line::arg_data_dir;
|
||||||
|
|
||||||
|
// Create data dir if it doesn't exist
|
||||||
|
boost::filesystem::path data_dir = boost::filesystem::absolute(
|
||||||
|
command_line::get_arg(vm, data_dir_arg));
|
||||||
|
tools::create_directories_if_necessary(data_dir.string());
|
||||||
|
|
||||||
|
// FIXME: not sure on windows implementation default, needs further review
|
||||||
|
//bf::path relative_path_base = daemonizer::get_relative_path_base(vm);
|
||||||
|
bf::path relative_path_base = data_dir;
|
||||||
|
|
||||||
|
std::string config = command_line::get_arg(vm, daemon_args::arg_config_file);
|
||||||
|
|
||||||
|
boost::filesystem::path data_dir_path(data_dir);
|
||||||
|
boost::filesystem::path config_path(config);
|
||||||
|
if (!config_path.has_parent_path())
|
||||||
|
{
|
||||||
|
config_path = data_dir / config_path;
|
||||||
|
}
|
||||||
|
|
||||||
|
boost::system::error_code ec;
|
||||||
|
if (bf::exists(config_path, ec))
|
||||||
|
{
|
||||||
|
po::store(po::parse_config_file<char>(config_path.string<std::string>().c_str(), core_settings), vm);
|
||||||
|
}
|
||||||
|
po::notify(vm);
|
||||||
|
|
||||||
|
// If there are positional options, we're running a daemon command
|
||||||
|
{
|
||||||
|
auto command = command_line::get_arg(vm, daemon_args::arg_command);
|
||||||
|
|
||||||
|
if (command.size())
|
||||||
|
{
|
||||||
|
auto rpc_ip_str = command_line::get_arg(vm, cryptonote::core_rpc_server::arg_rpc_bind_ip);
|
||||||
|
auto rpc_port_str = command_line::get_arg(vm, cryptonote::core_rpc_server::arg_rpc_bind_port);
|
||||||
|
if (testnet_mode)
|
||||||
|
{
|
||||||
|
rpc_port_str = command_line::get_arg(vm, cryptonote::core_rpc_server::arg_testnet_rpc_bind_port);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t rpc_ip;
|
||||||
|
uint16_t rpc_port;
|
||||||
|
if (!epee::string_tools::get_ip_int32_from_string(rpc_ip, rpc_ip_str))
|
||||||
|
{
|
||||||
|
std::cerr << "Invalid IP: " << rpc_ip_str << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
if (!epee::string_tools::get_xtype_from_string(rpc_port, rpc_port_str))
|
||||||
|
{
|
||||||
|
std::cerr << "Invalid port: " << rpc_port_str << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
daemonize::t_command_server rpc_commands{rpc_ip, rpc_port};
|
||||||
|
if (rpc_commands.process_command_vec(command))
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
std::cerr << "Unknown command" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start with log level 0
|
||||||
|
epee::log_space::get_set_log_detalisation_level(true, LOG_LEVEL_0);
|
||||||
|
|
||||||
|
// Set log level
|
||||||
|
{
|
||||||
|
int new_log_level = command_line::get_arg(vm, daemon_args::arg_log_level);
|
||||||
|
if(new_log_level < LOG_LEVEL_MIN || new_log_level > LOG_LEVEL_MAX)
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Wrong log level value: " << new_log_level);
|
||||||
|
}
|
||||||
|
else if (epee::log_space::get_set_log_detalisation_level(false) != new_log_level)
|
||||||
|
{
|
||||||
|
epee::log_space::get_set_log_detalisation_level(true, new_log_level);
|
||||||
|
LOG_PRINT_L0("LOG_LEVEL set to " << new_log_level);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set log file
|
||||||
|
{
|
||||||
|
bf::path log_file_path{bf::absolute(command_line::get_arg(vm, daemon_args::arg_log_file), relative_path_base)};
|
||||||
|
|
||||||
|
epee::log_space::log_singletone::add_logger(
|
||||||
|
LOGGER_FILE
|
||||||
|
, log_file_path.filename().string().c_str()
|
||||||
|
, log_file_path.parent_path().string().c_str()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return daemonizer::daemonize(argc, argv, daemonize::t_executor{}, vm);
|
||||||
|
}
|
||||||
|
catch (std::exception const & ex)
|
||||||
|
{
|
||||||
|
LOG_ERROR("Exception in main! " << ex.what());
|
||||||
|
}
|
||||||
|
catch (...)
|
||||||
|
{
|
||||||
|
LOG_ERROR("Exception in main!");
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
99
src/daemon/p2p.h
Normal file
99
src/daemon/p2p.h
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
||||||
|
#include "daemon/protocol.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include "p2p/net_node.h"
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
|
||||||
|
class t_p2p final
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
typedef cryptonote::t_cryptonote_protocol_handler<cryptonote::core> t_protocol_raw;
|
||||||
|
typedef nodetool::node_server<t_protocol_raw> t_node_server;
|
||||||
|
public:
|
||||||
|
static void init_options(boost::program_options::options_description & option_spec)
|
||||||
|
{
|
||||||
|
t_node_server::init_options(option_spec);
|
||||||
|
}
|
||||||
|
private:
|
||||||
|
t_node_server m_server;
|
||||||
|
public:
|
||||||
|
t_p2p(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
, t_protocol & protocol
|
||||||
|
)
|
||||||
|
: m_server{protocol.get()}
|
||||||
|
{
|
||||||
|
//initialize objects
|
||||||
|
LOG_PRINT_L0("Initializing p2p server...");
|
||||||
|
if (!m_server.init(vm))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to initialize p2p server.");
|
||||||
|
}
|
||||||
|
LOG_PRINT_L0("P2p server initialized OK");
|
||||||
|
}
|
||||||
|
|
||||||
|
t_node_server & get()
|
||||||
|
{
|
||||||
|
return m_server;
|
||||||
|
}
|
||||||
|
|
||||||
|
void run()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Starting p2p net loop...");
|
||||||
|
m_server.run();
|
||||||
|
LOG_PRINT_L0("p2p net loop stopped");
|
||||||
|
}
|
||||||
|
|
||||||
|
void stop()
|
||||||
|
{
|
||||||
|
m_server.send_stop_signal();
|
||||||
|
}
|
||||||
|
|
||||||
|
~t_p2p()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Deinitializing p2p...");
|
||||||
|
try {
|
||||||
|
m_server.deinit();
|
||||||
|
} catch (...) {
|
||||||
|
LOG_PRINT_L0("Failed to deinitialize p2p...");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
88
src/daemon/protocol.h
Normal file
88
src/daemon/protocol.h
Normal file
@ -0,0 +1,88 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include "p2p/net_node.h"
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
|
||||||
|
class t_protocol final
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
typedef cryptonote::t_cryptonote_protocol_handler<cryptonote::core> t_protocol_raw;
|
||||||
|
typedef nodetool::node_server<t_protocol_raw> t_node_server;
|
||||||
|
|
||||||
|
t_protocol_raw m_protocol;
|
||||||
|
public:
|
||||||
|
t_protocol(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
, t_core & core
|
||||||
|
)
|
||||||
|
: m_protocol{core.get(), nullptr}
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Initializing cryptonote protocol...");
|
||||||
|
if (!m_protocol.init(vm))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to initialize cryptonote protocol.");
|
||||||
|
}
|
||||||
|
LOG_PRINT_L0("Cryptonote protocol initialized OK");
|
||||||
|
}
|
||||||
|
|
||||||
|
t_protocol_raw & get()
|
||||||
|
{
|
||||||
|
return m_protocol;
|
||||||
|
}
|
||||||
|
|
||||||
|
void set_p2p_endpoint(
|
||||||
|
t_node_server & server
|
||||||
|
)
|
||||||
|
{
|
||||||
|
m_protocol.set_p2p_endpoint(&server);
|
||||||
|
}
|
||||||
|
|
||||||
|
~t_protocol()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Deinitializing cryptonote_protocol...");
|
||||||
|
try {
|
||||||
|
m_protocol.deinit();
|
||||||
|
m_protocol.set_p2p_endpoint(nullptr);
|
||||||
|
} catch (...) {
|
||||||
|
LOG_PRINT_L0("Failed to deinitialize protocol...");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
101
src/daemon/rpc.h
Normal file
101
src/daemon/rpc.h
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "daemon/core.h"
|
||||||
|
#include "daemon/p2p.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include "rpc/core_rpc_server.h"
|
||||||
|
#include <boost/program_options.hpp>
|
||||||
|
#include <stdexcept>
|
||||||
|
|
||||||
|
namespace daemonize
|
||||||
|
{
|
||||||
|
|
||||||
|
class t_rpc final
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
static void init_options(boost::program_options::options_description & option_spec)
|
||||||
|
{
|
||||||
|
cryptonote::core_rpc_server::init_options(option_spec);
|
||||||
|
}
|
||||||
|
private:
|
||||||
|
cryptonote::core_rpc_server m_server;
|
||||||
|
public:
|
||||||
|
t_rpc(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
, t_core & core
|
||||||
|
, t_p2p & p2p
|
||||||
|
)
|
||||||
|
: m_server{core.get(), p2p.get()}
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Initializing core rpc server...");
|
||||||
|
if (!m_server.init(vm))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to initialize core rpc server.");
|
||||||
|
}
|
||||||
|
LOG_PRINT_GREEN("Core rpc server initialized OK on port: " << m_server.get_binded_port(), LOG_LEVEL_0);
|
||||||
|
}
|
||||||
|
|
||||||
|
void run()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Starting core rpc server...");
|
||||||
|
if (!m_server.run(2, false))
|
||||||
|
{
|
||||||
|
throw std::runtime_error("Failed to start core rpc server.");
|
||||||
|
}
|
||||||
|
LOG_PRINT_L0("Core rpc server started ok");
|
||||||
|
}
|
||||||
|
|
||||||
|
void stop()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Stopping core rpc server...");
|
||||||
|
m_server.send_stop_signal();
|
||||||
|
m_server.timed_wait_server_stop(5000);
|
||||||
|
}
|
||||||
|
|
||||||
|
cryptonote::core_rpc_server* get_server()
|
||||||
|
{
|
||||||
|
return &m_server;
|
||||||
|
}
|
||||||
|
|
||||||
|
~t_rpc()
|
||||||
|
{
|
||||||
|
LOG_PRINT_L0("Deinitializing rpc server...");
|
||||||
|
try {
|
||||||
|
m_server.deinit();
|
||||||
|
} catch (...) {
|
||||||
|
LOG_PRINT_L0("Failed to deinitialize rpc server...");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
692
src/daemon/rpc_command_executor.cpp
Normal file
692
src/daemon/rpc_command_executor.cpp
Normal file
@ -0,0 +1,692 @@
|
|||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#include "string_tools.h"
|
||||||
|
#include "common/scoped_message_writer.h"
|
||||||
|
#include "daemon/rpc_command_executor.h"
|
||||||
|
#include "rpc/core_rpc_server_commands_defs.h"
|
||||||
|
#include <boost/format.hpp>
|
||||||
|
#include <ctime>
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
void print_peer(std::string const & prefix, cryptonote::peer const & peer)
|
||||||
|
{
|
||||||
|
time_t now;
|
||||||
|
time(&now);
|
||||||
|
time_t last_seen = static_cast<time_t>(peer.last_seen);
|
||||||
|
|
||||||
|
std::string id_str;
|
||||||
|
std::string port_str;
|
||||||
|
std::string elapsed = epee::misc_utils::get_time_interval_string(now - last_seen);
|
||||||
|
std::string ip_str = epee::string_tools::get_ip_string_from_int32(peer.ip);
|
||||||
|
epee::string_tools::xtype_to_string(peer.id, id_str);
|
||||||
|
epee::string_tools::xtype_to_string(peer.port, port_str);
|
||||||
|
std::string addr_str = ip_str + ":" + port_str;
|
||||||
|
tools::msg_writer() << boost::format("%-10s %-25s %-25s %s") % prefix % id_str % addr_str % elapsed;
|
||||||
|
}
|
||||||
|
|
||||||
|
void print_block_header(cryptonote::block_header_responce const & header)
|
||||||
|
{
|
||||||
|
tools::success_msg_writer()
|
||||||
|
<< "timestamp: " << boost::lexical_cast<std::string>(header.timestamp) << std::endl
|
||||||
|
<< "previous hash: " << header.prev_hash << std::endl
|
||||||
|
<< "nonce: " << boost::lexical_cast<std::string>(header.nonce) << std::endl
|
||||||
|
<< "is orphan: " << header.orphan_status << std::endl
|
||||||
|
<< "height: " << boost::lexical_cast<std::string>(header.height) << std::endl
|
||||||
|
<< "depth: " << boost::lexical_cast<std::string>(header.depth) << std::endl
|
||||||
|
<< "hash: " << header.hash
|
||||||
|
<< "difficulty: " << boost::lexical_cast<std::string>(header.difficulty) << std::endl
|
||||||
|
<< "reward: " << boost::lexical_cast<std::string>(header.reward);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
t_rpc_command_executor::t_rpc_command_executor(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc
|
||||||
|
, cryptonote::core_rpc_server* rpc_server
|
||||||
|
)
|
||||||
|
: m_rpc_client(NULL), m_rpc_server(rpc_server)
|
||||||
|
{
|
||||||
|
if (is_rpc)
|
||||||
|
{
|
||||||
|
m_rpc_client = new tools::t_rpc_client(ip, port);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (rpc_server == NULL)
|
||||||
|
{
|
||||||
|
throw std::runtime_error("If not calling commands via RPC, rpc_server pointer must be non-null");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
m_is_rpc = is_rpc;
|
||||||
|
}
|
||||||
|
|
||||||
|
t_rpc_command_executor::~t_rpc_command_executor()
|
||||||
|
{
|
||||||
|
if (m_rpc_client != NULL)
|
||||||
|
{
|
||||||
|
delete m_rpc_client;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_peer_list() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_PEER_LIST::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_PEER_LIST::response res;
|
||||||
|
|
||||||
|
std::string failure_message = "Couldn't retrieve peer list";
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/get_peer_list", failure_message.c_str()))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_peer_list(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << failure_message;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto & peer : res.white_list)
|
||||||
|
{
|
||||||
|
print_peer("white", peer);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto & peer : res.gray_list)
|
||||||
|
{
|
||||||
|
print_peer("gray", peer);
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::save_blockchain() {
|
||||||
|
cryptonote::COMMAND_RPC_SAVE_BC::request req;
|
||||||
|
cryptonote::COMMAND_RPC_SAVE_BC::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Couldn't save blockchain";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/save_bc", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_save_bc(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Blockchain saved";
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::show_hash_rate() {
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_HASH_RATE::request req;
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_HASH_RATE::response res;
|
||||||
|
req.visible = true;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/set_log_hash_rate", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_set_log_hash_rate(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Hash rate logging is on";
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::hide_hash_rate() {
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_HASH_RATE::request req;
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_HASH_RATE::response res;
|
||||||
|
req.visible = false;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/set_log_hash_rate", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_set_log_hash_rate(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Hash rate logging is off";
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::show_difficulty() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_INFO::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_INFO::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Problem fetching info";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/getinfo", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_info(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "BH: " << res.height
|
||||||
|
<< ", DIFF: " << res.difficulty
|
||||||
|
<< ", HR: " << (int) res.difficulty / 60L << " H/s";
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_connections() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_CONNECTIONS::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_CONNECTIONS::response res;
|
||||||
|
epee::json_rpc::error error_resp;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->json_rpc_request(req, res, "/get_connections", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_connections(req, res, error_resp))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto & info : res.connections)
|
||||||
|
{
|
||||||
|
std::string address = info.ip + ":" + info.port;
|
||||||
|
std::string in_out = info.incoming ? "INC" : "OUT";
|
||||||
|
tools::msg_writer() << boost::format("%-25s peer_id: %-25s %s") % address % info.peer_id % in_out;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_blockchain_info(uint64_t start_block_index, uint64_t end_block_index) {
|
||||||
|
|
||||||
|
// this function appears to not exist in the json rpc api, and so is commented
|
||||||
|
// until such a time as it does.
|
||||||
|
|
||||||
|
/*
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADERS_RANGE::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADERS_RANGE::response res;
|
||||||
|
epee::json_rpc::error error_resp;
|
||||||
|
|
||||||
|
req.start_height = start_block_index;
|
||||||
|
req.end_height = end_block_index;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->json_rpc_request(req, res, "getblockheadersrange", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_getblockheadersrange(req, res, error_resp))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto & header : res.headers)
|
||||||
|
{
|
||||||
|
std::cout
|
||||||
|
<< "major version: " << header.major_version << std::endl
|
||||||
|
<< "minor version: " << header.minor_version << std::endl
|
||||||
|
<< "height: " << header.height << ", timestamp: " << header.timestamp << ", difficulty: " << header.difficulty << std::endl
|
||||||
|
<< "block id: " << header.hash << std::endl
|
||||||
|
<< "previous block id: " << header.prev_hash << std::endl
|
||||||
|
<< "difficulty: " << header.difficulty << ", nonce " << header.nonce << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
*/
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::set_log_level(int8_t level) {
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_LEVEL::request req;
|
||||||
|
cryptonote::COMMAND_RPC_SET_LOG_LEVEL::response res;
|
||||||
|
req.level = level;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/set_log_level", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_set_log_level(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Log level is now " << boost::lexical_cast<std::string>(level);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_height() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_HEIGHT::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_HEIGHT::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/getheight", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_height(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << boost::lexical_cast<std::string>(res.height);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_block_by_hash(crypto::hash block_hash) {
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::response res;
|
||||||
|
epee::json_rpc::error error_resp;
|
||||||
|
|
||||||
|
req.hash = epee::string_tools::pod_to_hex(block_hash);
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->json_rpc_request(req, res, "getblockheaderbyhash", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_block_header_by_hash(req, res, error_resp))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
print_block_header(res.block_header);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_block_by_height(uint64_t height) {
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::response res;
|
||||||
|
epee::json_rpc::error error_resp;
|
||||||
|
|
||||||
|
req.height = height;
|
||||||
|
|
||||||
|
std::string fail_message = "Unsuccessful";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->json_rpc_request(req, res, "getblockheaderbyheight", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_block_header_by_height(req, res, error_resp))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
print_block_header(res.block_header);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_transaction(crypto::hash transaction_hash) {
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTIONS::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTIONS::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Problem fetching transaction";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/gettransactions", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_transactions(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (1 == res.txs_as_hex.size())
|
||||||
|
{
|
||||||
|
tools::success_msg_writer() << res.txs_as_hex.front();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "transaction wasn't found: <" << transaction_hash << '>' << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_transaction_pool_long() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTION_POOL::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTION_POOL::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Problem fetching transaction pool";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/get_transaction_pool", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_transaction_pool(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res.transactions.empty())
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "Pool is empty" << std::endl;
|
||||||
|
}
|
||||||
|
for (auto & tx_info : res.transactions)
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "id: " << tx_info.id_hash << std::endl
|
||||||
|
<< "blob_size: " << tx_info.blob_size << std::endl
|
||||||
|
<< "fee: " << tx_info.fee << std::endl
|
||||||
|
<< "kept_by_block: " << tx_info.kept_by_block << std::endl
|
||||||
|
<< "max_used_block_height: " << tx_info.max_used_block_height << std::endl
|
||||||
|
<< "max_used_block_id: " << tx_info.max_used_block_id_hash << std::endl
|
||||||
|
<< "last_failed_height: " << tx_info.last_failed_height << std::endl
|
||||||
|
<< "last_failed_id: " << tx_info.last_failed_id_hash << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_transaction_pool_short() {
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTION_POOL::request req;
|
||||||
|
cryptonote::COMMAND_RPC_GET_TRANSACTION_POOL::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Problem fetching transaction pool";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/get_transaction_pool", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_get_transaction_pool(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res.transactions.empty())
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "Pool is empty" << std::endl;
|
||||||
|
}
|
||||||
|
for (auto & tx_info : res.transactions)
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "id: " << tx_info.id_hash << std::endl
|
||||||
|
<< tx_info.tx_json << std::endl
|
||||||
|
<< "blob_size: " << tx_info.blob_size << std::endl
|
||||||
|
<< "fee: " << tx_info.fee << std::endl
|
||||||
|
<< "kept_by_block: " << tx_info.kept_by_block << std::endl
|
||||||
|
<< "max_used_block_height: " << tx_info.max_used_block_height << std::endl
|
||||||
|
<< "max_used_block_id: " << tx_info.max_used_block_id_hash << std::endl
|
||||||
|
<< "last_failed_height: " << tx_info.last_failed_height << std::endl
|
||||||
|
<< "last_failed_id: " << tx_info.last_failed_id_hash << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: update this for testnet
|
||||||
|
bool t_rpc_command_executor::start_mining(cryptonote::account_public_address address, uint64_t num_threads) {
|
||||||
|
cryptonote::COMMAND_RPC_START_MINING::request req;
|
||||||
|
cryptonote::COMMAND_RPC_START_MINING::response res;
|
||||||
|
req.miner_address = cryptonote::get_account_address_as_str(false, address);
|
||||||
|
req.threads_count = num_threads;
|
||||||
|
|
||||||
|
if (m_rpc_client->rpc_request(req, res, "/start_mining", "Mining did not start"))
|
||||||
|
{
|
||||||
|
tools::success_msg_writer() << "Mining started";
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::stop_mining() {
|
||||||
|
cryptonote::COMMAND_RPC_STOP_MINING::request req;
|
||||||
|
cryptonote::COMMAND_RPC_STOP_MINING::response res;
|
||||||
|
|
||||||
|
std::string fail_message = "Mining did not stop";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if (!m_rpc_client->rpc_request(req, res, "/stop_mining", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_stop_mining(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Mining stopped";
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::stop_daemon()
|
||||||
|
{
|
||||||
|
cryptonote::COMMAND_RPC_STOP_DAEMON::request req;
|
||||||
|
cryptonote::COMMAND_RPC_STOP_DAEMON::response res;
|
||||||
|
|
||||||
|
//# ifdef WIN32
|
||||||
|
// // Stop via service API
|
||||||
|
// // TODO - this is only temporary! Get rid of hard-coded constants!
|
||||||
|
// bool ok = windows::stop_service("BitMonero Daemon");
|
||||||
|
// ok = windows::uninstall_service("BitMonero Daemon");
|
||||||
|
// //bool ok = windows::stop_service(SERVICE_NAME);
|
||||||
|
// //ok = windows::uninstall_service(SERVICE_NAME);
|
||||||
|
// if (ok)
|
||||||
|
// {
|
||||||
|
// return true;
|
||||||
|
// }
|
||||||
|
//# endif
|
||||||
|
|
||||||
|
// Stop via RPC
|
||||||
|
std::string fail_message = "Daemon did not stop";
|
||||||
|
|
||||||
|
if (m_is_rpc)
|
||||||
|
{
|
||||||
|
if(!m_rpc_client->rpc_request(req, res, "/stop_daemon", fail_message.c_str()))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!m_rpc_server->on_stop_daemon(req, res))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << fail_message.c_str();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Stop signal sent";
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::print_status()
|
||||||
|
{
|
||||||
|
if (!m_is_rpc)
|
||||||
|
{
|
||||||
|
tools::success_msg_writer() << "print_status makes no sense in interactive mode";
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool daemon_is_alive = m_rpc_client->check_connection();
|
||||||
|
|
||||||
|
if(daemon_is_alive) {
|
||||||
|
tools::success_msg_writer() << "bitmonerod is running";
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
tools::fail_msg_writer() << "bitmonerod is NOT running";
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::set_limit(int limit)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
epee::net_utils::connection_basic::set_rate_down_limit( limit );
|
||||||
|
epee::net_utils::connection_basic::set_rate_up_limit( limit );
|
||||||
|
std::cout << "Set limit-down to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
std::cout << "Set limit-up to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
*/
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::set_limit_up(int limit)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
epee::net_utils::connection_basic::set_rate_up_limit( limit );
|
||||||
|
std::cout << "Set limit-up to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
*/
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool t_rpc_command_executor::set_limit_down(int limit)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
epee::net_utils::connection_basic::set_rate_down_limit( limit );
|
||||||
|
std::cout << "Set limit-down to " << limit/1024 << " kB/s" << std::endl;
|
||||||
|
*/
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
}// namespace daemonize
|
111
src/daemon/rpc_command_executor.h
Normal file
111
src/daemon/rpc_command_executor.h
Normal file
@ -0,0 +1,111 @@
|
|||||||
|
/**
|
||||||
|
@file
|
||||||
|
@details
|
||||||
|
|
||||||
|
@image html images/other/runtime-commands.png
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Copyright (c) 2014, The Monero Project
|
||||||
|
//
|
||||||
|
// All rights reserved.
|
||||||
|
//
|
||||||
|
// Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
// permitted provided that the following conditions are met:
|
||||||
|
//
|
||||||
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
// conditions and the following disclaimer.
|
||||||
|
//
|
||||||
|
// 2. 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.
|
||||||
|
//
|
||||||
|
// 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
//
|
||||||
|
// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "common/rpc_client.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
#include "cryptonote_core/cryptonote_core.h"
|
||||||
|
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
||||||
|
#include "p2p/net_node.h"
|
||||||
|
#include "rpc/core_rpc_server.h"
|
||||||
|
|
||||||
|
namespace daemonize {
|
||||||
|
|
||||||
|
class t_rpc_command_executor final {
|
||||||
|
private:
|
||||||
|
tools::t_rpc_client* m_rpc_client;
|
||||||
|
cryptonote::core_rpc_server* m_rpc_server;
|
||||||
|
bool m_is_rpc;
|
||||||
|
|
||||||
|
public:
|
||||||
|
t_rpc_command_executor(
|
||||||
|
uint32_t ip
|
||||||
|
, uint16_t port
|
||||||
|
, bool is_rpc = true
|
||||||
|
, cryptonote::core_rpc_server* rpc_server = NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
~t_rpc_command_executor();
|
||||||
|
|
||||||
|
bool print_peer_list();
|
||||||
|
|
||||||
|
bool save_blockchain();
|
||||||
|
|
||||||
|
bool show_hash_rate();
|
||||||
|
|
||||||
|
bool hide_hash_rate();
|
||||||
|
|
||||||
|
bool show_difficulty();
|
||||||
|
|
||||||
|
bool print_connections();
|
||||||
|
|
||||||
|
bool print_blockchain_info(uint64_t start_block_index, uint64_t end_block_index);
|
||||||
|
|
||||||
|
bool set_log_level(int8_t level);
|
||||||
|
|
||||||
|
bool print_height();
|
||||||
|
|
||||||
|
bool print_block_by_hash(crypto::hash block_hash);
|
||||||
|
|
||||||
|
bool print_block_by_height(uint64_t height);
|
||||||
|
|
||||||
|
bool print_transaction(crypto::hash transaction_hash);
|
||||||
|
|
||||||
|
bool print_transaction_pool_long();
|
||||||
|
|
||||||
|
bool print_transaction_pool_short();
|
||||||
|
|
||||||
|
bool start_mining(cryptonote::account_public_address address, uint64_t num_threads);
|
||||||
|
|
||||||
|
bool stop_mining();
|
||||||
|
|
||||||
|
bool stop_daemon();
|
||||||
|
|
||||||
|
bool print_status();
|
||||||
|
|
||||||
|
bool set_limit(int limit);
|
||||||
|
|
||||||
|
bool set_limit_up(int limit);
|
||||||
|
|
||||||
|
bool set_limit_down(int limit);
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace daemonize
|
74
src/daemonizer/CMakeLists.txt
Normal file
74
src/daemonizer/CMakeLists.txt
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
# Copyright (c) 2014-2015, The Monero Project
|
||||||
|
#
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
# Redistribution and use in source and binary forms, with or without modification, are
|
||||||
|
# permitted provided that the following conditions are met:
|
||||||
|
#
|
||||||
|
# 1. Redistributions of source code must retain the above copyright notice, this list of
|
||||||
|
# conditions and the following disclaimer.
|
||||||
|
#
|
||||||
|
# 2. 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.
|
||||||
|
#
|
||||||
|
# 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||||
|
|
||||||
|
if(MSVC OR MINGW)
|
||||||
|
set(daemonizer_sources
|
||||||
|
windows_service.cpp
|
||||||
|
windows_daemonizer.inl
|
||||||
|
)
|
||||||
|
else()
|
||||||
|
set(daemonizer_sources
|
||||||
|
posix_fork.cpp
|
||||||
|
posix_daemonizer.inl
|
||||||
|
)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
set(daemonizer_headers
|
||||||
|
)
|
||||||
|
|
||||||
|
if(MSVC OR MINGW)
|
||||||
|
set(daemonizer_private_headers
|
||||||
|
daemonizer.h
|
||||||
|
windows_service.h
|
||||||
|
windows_service_runner.h
|
||||||
|
)
|
||||||
|
else()
|
||||||
|
set(daemonizer_private_headers
|
||||||
|
daemonizer.h
|
||||||
|
posix_fork.h
|
||||||
|
)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
bitmonero_private_headers(daemonizer
|
||||||
|
${daemonizer_private_headers})
|
||||||
|
bitmonero_add_library(daemonizer
|
||||||
|
${daemonizer_sources}
|
||||||
|
${daemonizer_headers}
|
||||||
|
${daemonizer_private_headers})
|
||||||
|
target_link_libraries(daemonizer
|
||||||
|
LINK_PRIVATE
|
||||||
|
common
|
||||||
|
${Boost_CHRONO_LIBRARY}
|
||||||
|
${Boost_FILESYSTEM_LIBRARY}
|
||||||
|
${Boost_PROGRAM_OPTIONS_LIBRARY}
|
||||||
|
${Boost_REGEX_LIBRARY}
|
||||||
|
${Boost_SYSTEM_LIBRARY}
|
||||||
|
${Boost_THREAD_LIBRARY}
|
||||||
|
${CMAKE_THREAD_LIBS_INIT}
|
||||||
|
${UPNP_LIBRARIES}
|
||||||
|
${EXTRA_LIBRARIES})
|
38
src/daemonizer/daemonizer.h
Normal file
38
src/daemonizer/daemonizer.h
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <boost/filesystem/path.hpp>
|
||||||
|
#include <boost/program_options/options_description.hpp>
|
||||||
|
#include <boost/program_options/variables_map.hpp>
|
||||||
|
|
||||||
|
namespace daemonizer
|
||||||
|
{
|
||||||
|
void init_options(
|
||||||
|
boost::program_options::options_description & hidden_options
|
||||||
|
, boost::program_options::options_description & normal_options
|
||||||
|
);
|
||||||
|
|
||||||
|
boost::filesystem::path get_default_data_dir();
|
||||||
|
|
||||||
|
boost::filesystem::path get_relative_path_base(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @arg create_before_detach - this indicates that the daemon should be
|
||||||
|
* created before the fork, giving it a chance to report initialization
|
||||||
|
* errors. At the time of this writing, this is not possible in the primary
|
||||||
|
* daemon (likely due to the size of the blockchain in memory).
|
||||||
|
*/
|
||||||
|
template <typename T_executor>
|
||||||
|
bool daemonize(
|
||||||
|
int argc, char const * argv[]
|
||||||
|
, T_executor && executor // universal ref
|
||||||
|
, boost::program_options::variables_map const & vm
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
# include "daemonizer/windows_daemonizer.inl"
|
||||||
|
#else
|
||||||
|
# include "daemonizer/posix_daemonizer.inl"
|
||||||
|
#endif
|
60
src/daemonizer/posix_daemonizer.inl
Normal file
60
src/daemonizer/posix_daemonizer.inl
Normal file
@ -0,0 +1,60 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "common/scoped_message_writer.h"
|
||||||
|
#include "common/util.h"
|
||||||
|
#include "daemonizer/posix_fork.h"
|
||||||
|
|
||||||
|
#include <boost/filesystem/operations.hpp>
|
||||||
|
#include <boost/filesystem/path.hpp>
|
||||||
|
|
||||||
|
namespace daemonizer
|
||||||
|
{
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
const command_line::arg_descriptor<bool> arg_detach = {
|
||||||
|
"detach"
|
||||||
|
, "Run as daemon"
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void init_options(
|
||||||
|
boost::program_options::options_description & hidden_options
|
||||||
|
, boost::program_options::options_description & normal_options
|
||||||
|
)
|
||||||
|
{
|
||||||
|
command_line::add_arg(normal_options, arg_detach);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline boost::filesystem::path get_default_data_dir()
|
||||||
|
{
|
||||||
|
return boost::filesystem::absolute(tools::get_default_data_dir());
|
||||||
|
}
|
||||||
|
|
||||||
|
inline boost::filesystem::path get_relative_path_base(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
return boost::filesystem::current_path();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_executor>
|
||||||
|
inline bool daemonize(
|
||||||
|
int argc, char const * argv[]
|
||||||
|
, T_executor && executor // universal ref
|
||||||
|
, boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
if (command_line::has_arg(vm, arg_detach))
|
||||||
|
{
|
||||||
|
auto daemon = executor.create_daemon(vm);
|
||||||
|
tools::success_msg_writer() << "Forking to background...";
|
||||||
|
posix::fork();
|
||||||
|
return daemon.run();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//LOG_PRINT_L0(CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL);
|
||||||
|
return executor.run_interactive(vm);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
108
src/daemonizer/posix_fork.cpp
Normal file
108
src/daemonizer/posix_fork.cpp
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
// Copyright (c) 2003-2011 Christopher M. Kohlhoff (chris at kohlhoff dot com)
|
||||||
|
//
|
||||||
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||||||
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
||||||
|
//
|
||||||
|
|
||||||
|
#include "daemonizer/posix_fork.h"
|
||||||
|
#include "misc_log_ex.h"
|
||||||
|
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <string>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
|
||||||
|
namespace posix {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
void quit(std::string const & message)
|
||||||
|
{
|
||||||
|
LOG_ERROR(message);
|
||||||
|
throw std::runtime_error(message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void fork()
|
||||||
|
{
|
||||||
|
// Fork the process and have the parent exit. If the process was started
|
||||||
|
// from a shell, this returns control to the user. Forking a new process is
|
||||||
|
// also a prerequisite for the subsequent call to setsid().
|
||||||
|
if (pid_t pid = ::fork())
|
||||||
|
{
|
||||||
|
if (pid > 0)
|
||||||
|
{
|
||||||
|
// We're in the parent process and need to exit.
|
||||||
|
//
|
||||||
|
// When the exit() function is used, the program terminates without
|
||||||
|
// invoking local variables' destructors. Only global variables are
|
||||||
|
// destroyed.
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
quit("First fork failed");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make the process a new session leader. This detaches it from the
|
||||||
|
// terminal.
|
||||||
|
setsid();
|
||||||
|
|
||||||
|
// A process inherits its working directory from its parent. This could be
|
||||||
|
// on a mounted filesystem, which means that the running daemon would
|
||||||
|
// prevent this filesystem from being unmounted. Changing to the root
|
||||||
|
// directory avoids this problem.
|
||||||
|
if (chdir("/") < 0)
|
||||||
|
{
|
||||||
|
quit("Unable to change working directory to root");
|
||||||
|
}
|
||||||
|
|
||||||
|
// The file mode creation mask is also inherited from the parent process.
|
||||||
|
// We don't want to restrict the permissions on files created by the
|
||||||
|
// daemon, so the mask is cleared.
|
||||||
|
umask(0);
|
||||||
|
|
||||||
|
// A second fork ensures the process cannot acquire a controlling terminal.
|
||||||
|
if (pid_t pid = ::fork())
|
||||||
|
{
|
||||||
|
if (pid > 0)
|
||||||
|
{
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
quit("Second fork failed");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close the standard streams. This decouples the daemon from the terminal
|
||||||
|
// that started it.
|
||||||
|
close(0);
|
||||||
|
close(1);
|
||||||
|
close(2);
|
||||||
|
|
||||||
|
// We don't want the daemon to have any standard input.
|
||||||
|
if (open("/dev/null", O_RDONLY) < 0)
|
||||||
|
{
|
||||||
|
quit("Unable to open /dev/null");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send standard output to a log file.
|
||||||
|
const char* output = "/tmp/bitmonero.daemon.stdout.stderr";
|
||||||
|
const int flags = O_WRONLY | O_CREAT | O_APPEND;
|
||||||
|
const mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
|
||||||
|
if (open(output, flags, mode) < 0)
|
||||||
|
{
|
||||||
|
quit("Unable to open output file: " + std::string(output));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also send standard error to the same log file.
|
||||||
|
if (dup(1) < 0)
|
||||||
|
{
|
||||||
|
quit("Unable to dup output descriptor");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace posix
|
11
src/daemonizer/posix_fork.h
Normal file
11
src/daemonizer/posix_fork.h
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#ifndef WIN32
|
||||||
|
|
||||||
|
namespace posix {
|
||||||
|
|
||||||
|
void fork();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
156
src/daemonizer/windows_daemonizer.inl
Normal file
156
src/daemonizer/windows_daemonizer.inl
Normal file
@ -0,0 +1,156 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "common/util.h"
|
||||||
|
#include "daemonizer/windows_service.h"
|
||||||
|
#include "daemonizer/windows_service_runner.h"
|
||||||
|
|
||||||
|
#include <shlobj.h>
|
||||||
|
#include <boost/filesystem/operations.hpp>
|
||||||
|
#include <boost/filesystem/path.hpp>
|
||||||
|
|
||||||
|
namespace daemonizer
|
||||||
|
{
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
const command_line::arg_descriptor<bool> arg_install_service = {
|
||||||
|
"install-service"
|
||||||
|
, "Install Windows service"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_uninstall_service = {
|
||||||
|
"uninstall-service"
|
||||||
|
, "Uninstall Windows service"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_start_service = {
|
||||||
|
"start-service"
|
||||||
|
, "Start Windows service"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_stop_service = {
|
||||||
|
"stop-service"
|
||||||
|
, "Stop Windows service"
|
||||||
|
};
|
||||||
|
const command_line::arg_descriptor<bool> arg_is_service = {
|
||||||
|
"run-as-service"
|
||||||
|
, "Hidden -- true if running as windows service"
|
||||||
|
};
|
||||||
|
|
||||||
|
std::string get_argument_string(int argc, char const * argv[])
|
||||||
|
{
|
||||||
|
std::string result = "";
|
||||||
|
for (int i = 1; i < argc; ++i)
|
||||||
|
{
|
||||||
|
result += " " + std::string{argv[i]};
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void init_options(
|
||||||
|
boost::program_options::options_description & hidden_options
|
||||||
|
, boost::program_options::options_description & normal_options
|
||||||
|
)
|
||||||
|
{
|
||||||
|
command_line::add_arg(normal_options, arg_install_service);
|
||||||
|
command_line::add_arg(normal_options, arg_uninstall_service);
|
||||||
|
command_line::add_arg(normal_options, arg_start_service);
|
||||||
|
command_line::add_arg(normal_options, arg_stop_service);
|
||||||
|
command_line::add_arg(hidden_options, arg_is_service);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline boost::filesystem::path get_default_data_dir()
|
||||||
|
{
|
||||||
|
bool admin;
|
||||||
|
if (!windows::check_admin(admin))
|
||||||
|
{
|
||||||
|
admin = false;
|
||||||
|
}
|
||||||
|
if (admin)
|
||||||
|
{
|
||||||
|
return boost::filesystem::absolute(
|
||||||
|
tools::get_special_folder_path(CSIDL_COMMON_APPDATA, true) + "\\" + CRYPTONOTE_NAME
|
||||||
|
);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return boost::filesystem::absolute(
|
||||||
|
tools::get_special_folder_path(CSIDL_APPDATA, true) + "\\" + CRYPTONOTE_NAME
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
inline boost::filesystem::path get_relative_path_base(
|
||||||
|
boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
if (command_line::has_arg(vm, arg_is_service))
|
||||||
|
{
|
||||||
|
if (command_line::has_arg(vm, command_line::arg_data_dir))
|
||||||
|
{
|
||||||
|
return command_line::get_arg(vm, command_line::arg_data_dir);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return tools::get_default_data_dir();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return boost::filesystem::current_path();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_executor>
|
||||||
|
inline bool daemonize(
|
||||||
|
int argc, char const * argv[]
|
||||||
|
, T_executor && executor // universal ref
|
||||||
|
, boost::program_options::variables_map const & vm
|
||||||
|
)
|
||||||
|
{
|
||||||
|
std::string arguments = get_argument_string(argc, argv);
|
||||||
|
|
||||||
|
if (command_line::has_arg(vm, arg_is_service))
|
||||||
|
{
|
||||||
|
// TODO - Set the service status here for return codes
|
||||||
|
windows::t_service_runner<typename T_executor::t_daemon>::run(
|
||||||
|
executor.name()
|
||||||
|
, executor.create_daemon(vm)
|
||||||
|
);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else if (command_line::has_arg(vm, arg_install_service))
|
||||||
|
{
|
||||||
|
if (windows::ensure_admin(arguments))
|
||||||
|
{
|
||||||
|
arguments += " --run-as-service";
|
||||||
|
return windows::install_service(executor.name(), arguments);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (command_line::has_arg(vm, arg_uninstall_service))
|
||||||
|
{
|
||||||
|
if (windows::ensure_admin(arguments))
|
||||||
|
{
|
||||||
|
return windows::uninstall_service(executor.name());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (command_line::has_arg(vm, arg_start_service))
|
||||||
|
{
|
||||||
|
if (windows::ensure_admin(arguments))
|
||||||
|
{
|
||||||
|
return windows::start_service(executor.name());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (command_line::has_arg(vm, arg_stop_service))
|
||||||
|
{
|
||||||
|
if (windows::ensure_admin(arguments))
|
||||||
|
{
|
||||||
|
return windows::stop_service(executor.name());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else // interactive
|
||||||
|
{
|
||||||
|
//LOG_PRINT_L0(CRYPTONOTE_NAME << " v" << MONERO_VERSION_FULL);
|
||||||
|
return executor.run_interactive(vm);
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
341
src/daemonizer/windows_service.cpp
Normal file
341
src/daemonizer/windows_service.cpp
Normal file
@ -0,0 +1,341 @@
|
|||||||
|
#undef UNICODE
|
||||||
|
#undef _UNICODE
|
||||||
|
|
||||||
|
#include "common/scoped_message_writer.h"
|
||||||
|
#include "daemonizer/windows_service.h"
|
||||||
|
#include "string_tools.h"
|
||||||
|
#include <chrono>
|
||||||
|
#include <iostream>
|
||||||
|
#include <utility>
|
||||||
|
#include <memory>
|
||||||
|
#include <shellapi.h>
|
||||||
|
#include <thread>
|
||||||
|
#include <windows.h>
|
||||||
|
|
||||||
|
namespace windows {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
typedef std::unique_ptr<std::remove_pointer<SC_HANDLE>::type, decltype(&::CloseServiceHandle)> service_handle;
|
||||||
|
|
||||||
|
std::string get_last_error()
|
||||||
|
{
|
||||||
|
LPSTR p_error_text = nullptr;
|
||||||
|
|
||||||
|
FormatMessage(
|
||||||
|
FORMAT_MESSAGE_FROM_SYSTEM
|
||||||
|
| FORMAT_MESSAGE_ALLOCATE_BUFFER
|
||||||
|
| FORMAT_MESSAGE_IGNORE_INSERTS
|
||||||
|
, nullptr
|
||||||
|
, GetLastError()
|
||||||
|
, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT)
|
||||||
|
, reinterpret_cast<LPSTR>(&p_error_text)
|
||||||
|
, 0
|
||||||
|
, nullptr
|
||||||
|
);
|
||||||
|
|
||||||
|
if (nullptr == p_error_text)
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return std::string{p_error_text};
|
||||||
|
LocalFree(p_error_text);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool relaunch_as_admin(
|
||||||
|
std::string const & command
|
||||||
|
, std::string const & arguments
|
||||||
|
)
|
||||||
|
{
|
||||||
|
SHELLEXECUTEINFO info{};
|
||||||
|
info.cbSize = sizeof(info);
|
||||||
|
info.lpVerb = "runas";
|
||||||
|
info.lpFile = command.c_str();
|
||||||
|
info.lpParameters = arguments.c_str();
|
||||||
|
info.hwnd = nullptr;
|
||||||
|
info.nShow = SW_SHOWNORMAL;
|
||||||
|
if (!ShellExecuteEx(&info))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Admin relaunch failed: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// When we relaunch as admin, Windows opens a new window. This just pauses
|
||||||
|
// to allow the user to read any output.
|
||||||
|
void pause_to_display_admin_window_messages()
|
||||||
|
{
|
||||||
|
std::chrono::milliseconds how_long{1500};
|
||||||
|
std::this_thread::sleep_for(how_long);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool check_admin(bool & result)
|
||||||
|
{
|
||||||
|
BOOL is_admin = FALSE;
|
||||||
|
PSID p_administrators_group = nullptr;
|
||||||
|
|
||||||
|
SID_IDENTIFIER_AUTHORITY nt_authority = SECURITY_NT_AUTHORITY;
|
||||||
|
|
||||||
|
if (!AllocateAndInitializeSid(
|
||||||
|
&nt_authority
|
||||||
|
, 2
|
||||||
|
, SECURITY_BUILTIN_DOMAIN_RID
|
||||||
|
, DOMAIN_ALIAS_RID_ADMINS
|
||||||
|
, 0, 0, 0, 0, 0, 0
|
||||||
|
, &p_administrators_group
|
||||||
|
))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Security Identifier creation failed: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!CheckTokenMembership(
|
||||||
|
nullptr
|
||||||
|
, p_administrators_group
|
||||||
|
, &is_admin
|
||||||
|
))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Permissions check failed: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
result = is_admin ? true : false;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ensure_admin(
|
||||||
|
std::string const & arguments
|
||||||
|
)
|
||||||
|
{
|
||||||
|
bool admin;
|
||||||
|
|
||||||
|
if (!check_admin(admin))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (admin)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
std::string command = epee::string_tools::get_current_module_path();
|
||||||
|
relaunch_as_admin(command, arguments);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool install_service(
|
||||||
|
std::string const & service_name
|
||||||
|
, std::string const & arguments
|
||||||
|
)
|
||||||
|
{
|
||||||
|
std::string command = epee::string_tools::get_current_module_path();
|
||||||
|
std::string full_command = command + arguments;
|
||||||
|
|
||||||
|
service_handle p_manager{
|
||||||
|
OpenSCManager(
|
||||||
|
nullptr
|
||||||
|
, nullptr
|
||||||
|
, SC_MANAGER_CONNECT | SC_MANAGER_CREATE_SERVICE
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_manager == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
service_handle p_service{
|
||||||
|
CreateService(
|
||||||
|
p_manager.get()
|
||||||
|
, service_name.c_str()
|
||||||
|
, service_name.c_str()
|
||||||
|
, 0
|
||||||
|
//, GENERIC_EXECUTE | GENERIC_READ
|
||||||
|
, SERVICE_WIN32_OWN_PROCESS
|
||||||
|
, SERVICE_DEMAND_START
|
||||||
|
, SERVICE_ERROR_NORMAL
|
||||||
|
, full_command.c_str()
|
||||||
|
, nullptr
|
||||||
|
, nullptr
|
||||||
|
, ""
|
||||||
|
//, "NT AUTHORITY\\LocalService"
|
||||||
|
, nullptr // Implies LocalSystem account
|
||||||
|
, nullptr
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_service == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't create service: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Service installed";
|
||||||
|
|
||||||
|
pause_to_display_admin_window_messages();
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool start_service(
|
||||||
|
std::string const & service_name
|
||||||
|
)
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "Starting service";
|
||||||
|
|
||||||
|
SERVICE_STATUS_PROCESS service_status = {};
|
||||||
|
DWORD unused = 0;
|
||||||
|
|
||||||
|
service_handle p_manager{
|
||||||
|
OpenSCManager(
|
||||||
|
nullptr
|
||||||
|
, nullptr
|
||||||
|
, SC_MANAGER_CONNECT
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_manager == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
service_handle p_service{
|
||||||
|
OpenService(
|
||||||
|
p_manager.get()
|
||||||
|
, service_name.c_str()
|
||||||
|
//, SERVICE_START | SERVICE_QUERY_STATUS
|
||||||
|
, SERVICE_START
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_service == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!StartService(
|
||||||
|
p_service.get()
|
||||||
|
, 0
|
||||||
|
, nullptr
|
||||||
|
))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Service start request failed: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Service started";
|
||||||
|
|
||||||
|
pause_to_display_admin_window_messages();
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool stop_service(
|
||||||
|
std::string const & service_name
|
||||||
|
)
|
||||||
|
{
|
||||||
|
tools::msg_writer() << "Stopping service";
|
||||||
|
|
||||||
|
service_handle p_manager{
|
||||||
|
OpenSCManager(
|
||||||
|
nullptr
|
||||||
|
, nullptr
|
||||||
|
, SC_MANAGER_CONNECT
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_manager == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
service_handle p_service{
|
||||||
|
OpenService(
|
||||||
|
p_manager.get()
|
||||||
|
, service_name.c_str()
|
||||||
|
, SERVICE_STOP | SERVICE_QUERY_STATUS
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_service == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
SERVICE_STATUS status = {};
|
||||||
|
if (!ControlService(p_service.get(), SERVICE_CONTROL_STOP, &status))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't request service stop: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Service stopped";
|
||||||
|
|
||||||
|
pause_to_display_admin_window_messages();
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool uninstall_service(
|
||||||
|
std::string const & service_name
|
||||||
|
)
|
||||||
|
{
|
||||||
|
service_handle p_manager{
|
||||||
|
OpenSCManager(
|
||||||
|
nullptr
|
||||||
|
, nullptr
|
||||||
|
, SC_MANAGER_CONNECT
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_manager == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
service_handle p_service{
|
||||||
|
OpenService(
|
||||||
|
p_manager.get()
|
||||||
|
, service_name.c_str()
|
||||||
|
, SERVICE_QUERY_STATUS | DELETE
|
||||||
|
)
|
||||||
|
, &::CloseServiceHandle
|
||||||
|
};
|
||||||
|
if (p_service == nullptr)
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
SERVICE_STATUS status = {};
|
||||||
|
if (!DeleteService(p_service.get()))
|
||||||
|
{
|
||||||
|
tools::fail_msg_writer() << "Couldn't uninstall service: " << get_last_error();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
tools::success_msg_writer() << "Service uninstalled";
|
||||||
|
|
||||||
|
pause_to_display_admin_window_messages();
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace windows
|
36
src/daemonizer/windows_service.h
Normal file
36
src/daemonizer/windows_service.h
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
|
||||||
|
#undef UNICODE
|
||||||
|
#undef _UNICODE
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <windows.h>
|
||||||
|
|
||||||
|
namespace windows
|
||||||
|
{
|
||||||
|
bool check_admin(bool & result);
|
||||||
|
|
||||||
|
bool ensure_admin(
|
||||||
|
std::string const & arguments
|
||||||
|
);
|
||||||
|
|
||||||
|
bool install_service(
|
||||||
|
std::string const & service_name
|
||||||
|
, std::string const & arguments
|
||||||
|
);
|
||||||
|
|
||||||
|
bool uninstall_service(
|
||||||
|
std::string const & service_name
|
||||||
|
);
|
||||||
|
|
||||||
|
bool start_service(
|
||||||
|
std::string const & service_name
|
||||||
|
);
|
||||||
|
|
||||||
|
bool stop_service(
|
||||||
|
std::string const & service_name
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#endif
|
157
src/daemonizer/windows_service_runner.h
Normal file
157
src/daemonizer/windows_service_runner.h
Normal file
@ -0,0 +1,157 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#ifdef WIN32
|
||||||
|
|
||||||
|
#undef UNICODE
|
||||||
|
#undef _UNICODE
|
||||||
|
|
||||||
|
#include "daemonizer/windows_service.h"
|
||||||
|
#include <memory>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <windows.h>
|
||||||
|
|
||||||
|
namespace windows {
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
std::vector<char> vecstring(std::string const & str)
|
||||||
|
{
|
||||||
|
std::vector<char> result{str.begin(), str.end()};
|
||||||
|
result.push_back('\0');
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T_handler>
|
||||||
|
class t_service_runner final
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
SERVICE_STATUS_HANDLE m_status_handle{nullptr};
|
||||||
|
SERVICE_STATUS m_status{};
|
||||||
|
std::mutex m_lock{};
|
||||||
|
std::string m_name;
|
||||||
|
T_handler m_handler;
|
||||||
|
|
||||||
|
static std::unique_ptr<t_service_runner<T_handler>> sp_instance;
|
||||||
|
public:
|
||||||
|
t_service_runner(
|
||||||
|
std::string name
|
||||||
|
, T_handler handler
|
||||||
|
)
|
||||||
|
: m_name{std::move(name)}
|
||||||
|
, m_handler{std::move(handler)}
|
||||||
|
{
|
||||||
|
m_status.dwServiceType = SERVICE_WIN32;
|
||||||
|
m_status.dwCurrentState = SERVICE_STOPPED;
|
||||||
|
m_status.dwControlsAccepted = 0;
|
||||||
|
m_status.dwWin32ExitCode = NO_ERROR;
|
||||||
|
m_status.dwServiceSpecificExitCode = NO_ERROR;
|
||||||
|
m_status.dwCheckPoint = 0;
|
||||||
|
m_status.dwWaitHint = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
t_service_runner & operator=(t_service_runner && other)
|
||||||
|
{
|
||||||
|
if (this != &other)
|
||||||
|
{
|
||||||
|
m_status_handle = std::move(other.m_status_handle);
|
||||||
|
m_status = std::move(other.m_status);
|
||||||
|
m_name = std::move(other.m_name);
|
||||||
|
m_handler = std::move(other.m_handler);
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void run(
|
||||||
|
std::string name
|
||||||
|
, T_handler handler
|
||||||
|
)
|
||||||
|
{
|
||||||
|
sp_instance.reset(new t_service_runner<T_handler>{
|
||||||
|
std::move(name)
|
||||||
|
, std::move(handler)
|
||||||
|
});
|
||||||
|
|
||||||
|
sp_instance->run_();
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
void run_()
|
||||||
|
{
|
||||||
|
SERVICE_TABLE_ENTRY table[] =
|
||||||
|
{
|
||||||
|
{ vecstring(m_name).data(), &service_main }
|
||||||
|
, { 0, 0 }
|
||||||
|
};
|
||||||
|
|
||||||
|
StartServiceCtrlDispatcher(table);
|
||||||
|
}
|
||||||
|
|
||||||
|
void report_status(DWORD status)
|
||||||
|
{
|
||||||
|
m_status.dwCurrentState = status;
|
||||||
|
if (status == SERVICE_RUNNING)
|
||||||
|
{
|
||||||
|
m_status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
|
||||||
|
}
|
||||||
|
else if(status == SERVICE_STOP_PENDING)
|
||||||
|
{
|
||||||
|
m_status.dwControlsAccepted = 0;
|
||||||
|
}
|
||||||
|
SetServiceStatus(m_status_handle, &m_status);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void WINAPI service_main(DWORD argc, LPSTR * argv)
|
||||||
|
{
|
||||||
|
sp_instance->service_main_(argc, argv);
|
||||||
|
}
|
||||||
|
|
||||||
|
void service_main_(DWORD argc, LPSTR * argv)
|
||||||
|
{
|
||||||
|
m_status_handle = RegisterServiceCtrlHandler(m_name.c_str(), &on_state_change_request);
|
||||||
|
if (m_status_handle == nullptr) return;
|
||||||
|
|
||||||
|
report_status(SERVICE_START_PENDING);
|
||||||
|
|
||||||
|
report_status(SERVICE_RUNNING);
|
||||||
|
|
||||||
|
m_handler.run();
|
||||||
|
|
||||||
|
on_state_change_request_(SERVICE_CONTROL_STOP);
|
||||||
|
|
||||||
|
// Ensure that the service is uninstalled
|
||||||
|
uninstall_service(m_name);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void WINAPI on_state_change_request(DWORD control_code)
|
||||||
|
{
|
||||||
|
sp_instance->on_state_change_request_(control_code);
|
||||||
|
}
|
||||||
|
|
||||||
|
void on_state_change_request_(DWORD control_code)
|
||||||
|
{
|
||||||
|
switch (control_code)
|
||||||
|
{
|
||||||
|
case SERVICE_CONTROL_INTERROGATE:
|
||||||
|
break;
|
||||||
|
case SERVICE_CONTROL_SHUTDOWN:
|
||||||
|
case SERVICE_CONTROL_STOP:
|
||||||
|
report_status(SERVICE_STOP_PENDING);
|
||||||
|
m_handler.stop();
|
||||||
|
report_status(SERVICE_STOPPED);
|
||||||
|
break;
|
||||||
|
case SERVICE_CONTROL_PAUSE:
|
||||||
|
break;
|
||||||
|
case SERVICE_CONTROL_CONTINUE:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T_handler>
|
||||||
|
std::unique_ptr<t_service_runner<T_handler>> t_service_runner<T_handler>::sp_instance;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
@ -728,7 +728,7 @@ namespace Language
|
|||||||
"initiate",
|
"initiate",
|
||||||
"injury",
|
"injury",
|
||||||
"inkling",
|
"inkling",
|
||||||
"incline",
|
"inline",
|
||||||
"inmate",
|
"inmate",
|
||||||
"innocent",
|
"innocent",
|
||||||
"inorganic",
|
"inorganic",
|
||||||
@ -1073,7 +1073,7 @@ namespace Language
|
|||||||
"ouch",
|
"ouch",
|
||||||
"ought",
|
"ought",
|
||||||
"ounce",
|
"ounce",
|
||||||
"launchpad",
|
"ourselves",
|
||||||
"oust",
|
"oust",
|
||||||
"outbreak",
|
"outbreak",
|
||||||
"oval",
|
"oval",
|
||||||
|
@ -80,22 +80,18 @@ namespace nodetool
|
|||||||
public:
|
public:
|
||||||
typedef t_payload_net_handler payload_net_handler;
|
typedef t_payload_net_handler payload_net_handler;
|
||||||
|
|
||||||
node_server(t_payload_net_handler& payload_handler, boost::uuids::uuid network_id)
|
node_server(
|
||||||
:m_payload_handler(payload_handler),
|
t_payload_net_handler& payload_handler
|
||||||
m_allow_local_ip(false),
|
)
|
||||||
m_no_igd(false),
|
: m_payload_handler(payload_handler)
|
||||||
m_hide_my_port(false),
|
, m_allow_local_ip(false)
|
||||||
m_network_id(std::move(network_id))
|
, m_hide_my_port(false)
|
||||||
{
|
{}
|
||||||
m_current_number_of_out_peers = 0;
|
|
||||||
m_save_graph = false;
|
|
||||||
is_closing = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void init_options(boost::program_options::options_description& desc);
|
static void init_options(boost::program_options::options_description& desc);
|
||||||
|
|
||||||
bool run();
|
bool run();
|
||||||
bool init(const boost::program_options::variables_map& vm, bool testnet);
|
bool init(const boost::program_options::variables_map& vm);
|
||||||
bool deinit();
|
bool deinit();
|
||||||
bool send_stop_signal();
|
bool send_stop_signal();
|
||||||
uint32_t get_this_peer_port(){return m_listenning_port;}
|
uint32_t get_this_peer_port(){return m_listenning_port;}
|
||||||
|
@ -46,7 +46,7 @@
|
|||||||
#include "net/local_ip.h"
|
#include "net/local_ip.h"
|
||||||
#include "crypto/crypto.h"
|
#include "crypto/crypto.h"
|
||||||
#include "storages/levin_abstract_invoke2.h"
|
#include "storages/levin_abstract_invoke2.h"
|
||||||
#include "data_logger.hpp"
|
#include "daemon/command_line_args.h"
|
||||||
|
|
||||||
// We have to look for miniupnpc headers in different places, dependent on if its compiled or external
|
// We have to look for miniupnpc headers in different places, dependent on if its compiled or external
|
||||||
#ifdef UPNP_STATIC
|
#ifdef UPNP_STATIC
|
||||||
@ -273,16 +273,20 @@ namespace nodetool
|
|||||||
|
|
||||||
//-----------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------
|
||||||
template<class t_payload_net_handler>
|
template<class t_payload_net_handler>
|
||||||
bool node_server<t_payload_net_handler>::init(const boost::program_options::variables_map& vm, bool testnet)
|
bool node_server<t_payload_net_handler>::init(const boost::program_options::variables_map& vm)
|
||||||
{
|
{
|
||||||
|
bool testnet = command_line::get_arg(vm, daemon_args::arg_testnet_on);
|
||||||
|
|
||||||
if (testnet)
|
if (testnet)
|
||||||
{
|
{
|
||||||
|
memcpy(&m_network_id, &::config::testnet::NETWORK_ID, 16);
|
||||||
append_net_address(m_seed_nodes, "107.152.187.202:28080");
|
append_net_address(m_seed_nodes, "107.152.187.202:28080");
|
||||||
append_net_address(m_seed_nodes, "197.242.158.240:28080");
|
append_net_address(m_seed_nodes, "197.242.158.240:28080");
|
||||||
append_net_address(m_seed_nodes, "107.152.130.98:28080");
|
append_net_address(m_seed_nodes, "107.152.130.98:28080");
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
|
memcpy(&m_network_id, &::config::NETWORK_ID, 16);
|
||||||
// for each hostname in the seed nodes list, attempt to DNS resolve and
|
// for each hostname in the seed nodes list, attempt to DNS resolve and
|
||||||
// add the result addresses as seed nodes
|
// add the result addresses as seed nodes
|
||||||
// TODO: at some point add IPv6 support, but that won't be relevant
|
// TODO: at some point add IPv6 support, but that won't be relevant
|
||||||
|
@ -40,29 +40,10 @@ using namespace epee;
|
|||||||
#include "misc_language.h"
|
#include "misc_language.h"
|
||||||
#include "crypto/hash.h"
|
#include "crypto/hash.h"
|
||||||
#include "core_rpc_server_error_codes.h"
|
#include "core_rpc_server_error_codes.h"
|
||||||
|
#include "daemon/command_line_args.h"
|
||||||
|
|
||||||
namespace cryptonote
|
namespace cryptonote
|
||||||
{
|
{
|
||||||
namespace
|
|
||||||
{
|
|
||||||
const command_line::arg_descriptor<std::string> arg_rpc_bind_ip = {
|
|
||||||
"rpc-bind-ip"
|
|
||||||
, "IP for RPC server"
|
|
||||||
, "127.0.0.1"
|
|
||||||
};
|
|
||||||
|
|
||||||
const command_line::arg_descriptor<std::string> arg_rpc_bind_port = {
|
|
||||||
"rpc-bind-port"
|
|
||||||
, "Port for RPC server"
|
|
||||||
, std::to_string(config::RPC_DEFAULT_PORT)
|
|
||||||
};
|
|
||||||
|
|
||||||
const command_line::arg_descriptor<std::string> arg_testnet_rpc_bind_port = {
|
|
||||||
"testnet-rpc-bind-port"
|
|
||||||
, "Port for testnet RPC server"
|
|
||||||
, std::to_string(config::testnet::RPC_DEFAULT_PORT)
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------------
|
||||||
void core_rpc_server::init_options(boost::program_options::options_description& desc)
|
void core_rpc_server::init_options(boost::program_options::options_description& desc)
|
||||||
@ -75,11 +56,9 @@ namespace cryptonote
|
|||||||
core_rpc_server::core_rpc_server(
|
core_rpc_server::core_rpc_server(
|
||||||
core& cr
|
core& cr
|
||||||
, nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& p2p
|
, nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& p2p
|
||||||
, bool testnet
|
|
||||||
)
|
)
|
||||||
: m_core(cr)
|
: m_core(cr)
|
||||||
, m_p2p(p2p)
|
, m_p2p(p2p)
|
||||||
, m_testnet(testnet)
|
|
||||||
{}
|
{}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::handle_command_line(
|
bool core_rpc_server::handle_command_line(
|
||||||
@ -97,6 +76,8 @@ namespace cryptonote
|
|||||||
const boost::program_options::variables_map& vm
|
const boost::program_options::variables_map& vm
|
||||||
)
|
)
|
||||||
{
|
{
|
||||||
|
m_testnet = command_line::get_arg(vm, daemon_args::arg_testnet_on);
|
||||||
|
|
||||||
m_net_server.set_threads_prefix("RPC");
|
m_net_server.set_threads_prefix("RPC");
|
||||||
bool r = handle_command_line(vm);
|
bool r = handle_command_line(vm);
|
||||||
CHECK_AND_ASSERT_MES(r, false, "Failed to process command line in core_rpc_server");
|
CHECK_AND_ASSERT_MES(r, false, "Failed to process command line in core_rpc_server");
|
||||||
@ -124,7 +105,7 @@ namespace cryptonote
|
|||||||
#define CHECK_CORE_READY() do { if(!check_core_ready()){res.status = CORE_RPC_STATUS_BUSY;return true;} } while(0)
|
#define CHECK_CORE_READY() do { if(!check_core_ready()){res.status = CORE_RPC_STATUS_BUSY;return true;} } while(0)
|
||||||
|
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_height(const COMMAND_RPC_GET_HEIGHT::request& req, COMMAND_RPC_GET_HEIGHT::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_height(const COMMAND_RPC_GET_HEIGHT::request& req, COMMAND_RPC_GET_HEIGHT::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
res.height = m_core.get_current_blockchain_height();
|
res.height = m_core.get_current_blockchain_height();
|
||||||
@ -132,7 +113,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_info(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_info(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
res.height = m_core.get_current_blockchain_height();
|
res.height = m_core.get_current_blockchain_height();
|
||||||
@ -150,7 +131,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_blocks(const COMMAND_RPC_GET_BLOCKS_FAST::request& req, COMMAND_RPC_GET_BLOCKS_FAST::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_blocks(const COMMAND_RPC_GET_BLOCKS_FAST::request& req, COMMAND_RPC_GET_BLOCKS_FAST::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
std::list<std::pair<block, std::list<transaction> > > bs;
|
std::list<std::pair<block, std::list<transaction> > > bs;
|
||||||
@ -175,7 +156,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_random_outs(const COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::request& req, COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_random_outs(const COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::request& req, COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
res.status = "Failed";
|
res.status = "Failed";
|
||||||
@ -204,7 +185,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_indexes(const COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::request& req, COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_indexes(const COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::request& req, COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
bool r = m_core.get_tx_outputs_gindexs(req.txid, res.o_indexes);
|
bool r = m_core.get_tx_outputs_gindexs(req.txid, res.o_indexes);
|
||||||
@ -218,7 +199,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_transactions(const COMMAND_RPC_GET_TRANSACTIONS::request& req, COMMAND_RPC_GET_TRANSACTIONS::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_transactions(const COMMAND_RPC_GET_TRANSACTIONS::request& req, COMMAND_RPC_GET_TRANSACTIONS::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
std::vector<crypto::hash> vh;
|
std::vector<crypto::hash> vh;
|
||||||
@ -260,7 +241,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_send_raw_tx(const COMMAND_RPC_SEND_RAW_TX::request& req, COMMAND_RPC_SEND_RAW_TX::response& res, connection_context& cntx)
|
bool core_rpc_server::on_send_raw_tx(const COMMAND_RPC_SEND_RAW_TX::request& req, COMMAND_RPC_SEND_RAW_TX::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_READY();
|
CHECK_CORE_READY();
|
||||||
|
|
||||||
@ -303,7 +284,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_start_mining(const COMMAND_RPC_START_MINING::request& req, COMMAND_RPC_START_MINING::response& res, connection_context& cntx)
|
bool core_rpc_server::on_start_mining(const COMMAND_RPC_START_MINING::request& req, COMMAND_RPC_START_MINING::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_READY();
|
CHECK_CORE_READY();
|
||||||
account_public_address adr;
|
account_public_address adr;
|
||||||
@ -325,7 +306,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_stop_mining(const COMMAND_RPC_STOP_MINING::request& req, COMMAND_RPC_STOP_MINING::response& res, connection_context& cntx)
|
bool core_rpc_server::on_stop_mining(const COMMAND_RPC_STOP_MINING::request& req, COMMAND_RPC_STOP_MINING::response& res)
|
||||||
{
|
{
|
||||||
if(!m_core.get_miner().stop())
|
if(!m_core.get_miner().stop())
|
||||||
{
|
{
|
||||||
@ -336,7 +317,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_mining_status(const COMMAND_RPC_MINING_STATUS::request& req, COMMAND_RPC_MINING_STATUS::response& res, connection_context& cntx)
|
bool core_rpc_server::on_mining_status(const COMMAND_RPC_MINING_STATUS::request& req, COMMAND_RPC_MINING_STATUS::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_READY();
|
CHECK_CORE_READY();
|
||||||
|
|
||||||
@ -354,7 +335,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_save_bc(const COMMAND_RPC_SAVE_BC::request& req, COMMAND_RPC_SAVE_BC::response& res, connection_context& cntx)
|
bool core_rpc_server::on_save_bc(const COMMAND_RPC_SAVE_BC::request& req, COMMAND_RPC_SAVE_BC::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
if( !m_core.get_blockchain_storage().store_blockchain() )
|
if( !m_core.get_blockchain_storage().store_blockchain() )
|
||||||
@ -366,7 +347,76 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_getblockcount(const COMMAND_RPC_GETBLOCKCOUNT::request& req, COMMAND_RPC_GETBLOCKCOUNT::response& res, connection_context& cntx)
|
bool core_rpc_server::on_get_peer_list(const COMMAND_RPC_GET_PEER_LIST::request& req, COMMAND_RPC_GET_PEER_LIST::response& res)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
std::list<nodetool::peerlist_entry> white_list;
|
||||||
|
std::list<nodetool::peerlist_entry> gray_list;
|
||||||
|
m_p2p.get_peerlist(white_list, gray_list);
|
||||||
|
|
||||||
|
for (auto & entry : white_list)
|
||||||
|
{
|
||||||
|
res.white_list.emplace_back(entry.id, entry.adr.ip, entry.adr.port, entry.last_seen);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto & entry : gray_list)
|
||||||
|
{
|
||||||
|
res.gray_list.emplace_back(entry.id, entry.adr.ip, entry.adr.port, entry.last_seen);
|
||||||
|
}
|
||||||
|
|
||||||
|
*/
|
||||||
|
res.status = CORE_RPC_STATUS_OK;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
|
bool core_rpc_server::on_set_log_hash_rate(const COMMAND_RPC_SET_LOG_HASH_RATE::request& req, COMMAND_RPC_SET_LOG_HASH_RATE::response& res)
|
||||||
|
{
|
||||||
|
if(m_core.get_miner().is_mining())
|
||||||
|
{
|
||||||
|
m_core.get_miner().do_print_hashrate(req.visible);
|
||||||
|
res.status = CORE_RPC_STATUS_OK;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
res.status = CORE_RPC_STATUS_NOT_MINING;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
|
bool core_rpc_server::on_set_log_level(const COMMAND_RPC_SET_LOG_LEVEL::request& req, COMMAND_RPC_SET_LOG_LEVEL::response& res)
|
||||||
|
{
|
||||||
|
if (req.level < LOG_LEVEL_MIN || req.level > LOG_LEVEL_MAX)
|
||||||
|
{
|
||||||
|
res.status = "Error: log level not valid";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
epee::log_space::log_singletone::get_set_log_detalisation_level(true, req.level);
|
||||||
|
res.status = CORE_RPC_STATUS_OK;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
|
bool core_rpc_server::on_get_transaction_pool(const COMMAND_RPC_GET_TRANSACTION_POOL::request& req, COMMAND_RPC_GET_TRANSACTION_POOL::response& res)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
CHECK_CORE_BUSY();
|
||||||
|
res.transactions = m_core.transaction_pool_info();
|
||||||
|
*/
|
||||||
|
res.status = CORE_RPC_STATUS_OK;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
|
bool core_rpc_server::on_stop_daemon(const COMMAND_RPC_STOP_DAEMON::request& req, COMMAND_RPC_STOP_DAEMON::response& res)
|
||||||
|
{
|
||||||
|
// FIXME: replace back to original m_p2p.send_stop_signal() after
|
||||||
|
// investigating why that isn't working quite right.
|
||||||
|
m_p2p.send_stop_signal();
|
||||||
|
res.status = CORE_RPC_STATUS_OK;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
|
bool core_rpc_server::on_getblockcount(const COMMAND_RPC_GETBLOCKCOUNT::request& req, COMMAND_RPC_GETBLOCKCOUNT::response& res)
|
||||||
{
|
{
|
||||||
CHECK_CORE_BUSY();
|
CHECK_CORE_BUSY();
|
||||||
res.count = m_core.get_current_blockchain_height();
|
res.count = m_core.get_current_blockchain_height();
|
||||||
@ -374,7 +424,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_getblockhash(const COMMAND_RPC_GETBLOCKHASH::request& req, COMMAND_RPC_GETBLOCKHASH::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_getblockhash(const COMMAND_RPC_GETBLOCKHASH::request& req, COMMAND_RPC_GETBLOCKHASH::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
@ -415,7 +465,7 @@ namespace cryptonote
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_getblocktemplate(const COMMAND_RPC_GETBLOCKTEMPLATE::request& req, COMMAND_RPC_GETBLOCKTEMPLATE::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_getblocktemplate(const COMMAND_RPC_GETBLOCKTEMPLATE::request& req, COMMAND_RPC_GETBLOCKTEMPLATE::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
if(!check_core_ready())
|
if(!check_core_ready())
|
||||||
{
|
{
|
||||||
@ -481,7 +531,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_submitblock(const COMMAND_RPC_SUBMITBLOCK::request& req, COMMAND_RPC_SUBMITBLOCK::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_submitblock(const COMMAND_RPC_SUBMITBLOCK::request& req, COMMAND_RPC_SUBMITBLOCK::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
CHECK_CORE_READY();
|
CHECK_CORE_READY();
|
||||||
if(req.size()!=1)
|
if(req.size()!=1)
|
||||||
@ -553,7 +603,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_last_block_header(const COMMAND_RPC_GET_LAST_BLOCK_HEADER::request& req, COMMAND_RPC_GET_LAST_BLOCK_HEADER::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_get_last_block_header(const COMMAND_RPC_GET_LAST_BLOCK_HEADER::request& req, COMMAND_RPC_GET_LAST_BLOCK_HEADER::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
@ -589,7 +639,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_block_header_by_hash(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::response& res, epee::json_rpc::error& error_resp, connection_context& cntx){
|
bool core_rpc_server::on_get_block_header_by_hash(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::response& res, epee::json_rpc::error& error_resp){
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
error_resp.code = CORE_RPC_ERROR_CODE_CORE_BUSY;
|
error_resp.code = CORE_RPC_ERROR_CODE_CORE_BUSY;
|
||||||
@ -630,7 +680,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_block_header_by_height(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::response& res, epee::json_rpc::error& error_resp, connection_context& cntx){
|
bool core_rpc_server::on_get_block_header_by_height(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::response& res, epee::json_rpc::error& error_resp){
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
error_resp.code = CORE_RPC_ERROR_CODE_CORE_BUSY;
|
error_resp.code = CORE_RPC_ERROR_CODE_CORE_BUSY;
|
||||||
@ -663,7 +713,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_connections(const COMMAND_RPC_GET_CONNECTIONS::request& req, COMMAND_RPC_GET_CONNECTIONS::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_get_connections(const COMMAND_RPC_GET_CONNECTIONS::request& req, COMMAND_RPC_GET_CONNECTIONS::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
@ -679,7 +729,7 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool core_rpc_server::on_get_info_json(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, epee::json_rpc::error& error_resp, connection_context& cntx)
|
bool core_rpc_server::on_get_info_json(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, epee::json_rpc::error& error_resp)
|
||||||
{
|
{
|
||||||
if(!check_core_busy())
|
if(!check_core_busy())
|
||||||
{
|
{
|
||||||
@ -703,4 +753,23 @@ namespace cryptonote
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
}
|
|
||||||
|
const command_line::arg_descriptor<std::string> core_rpc_server::arg_rpc_bind_ip = {
|
||||||
|
"rpc-bind-ip"
|
||||||
|
, "IP for RPC server"
|
||||||
|
, "127.0.0.1"
|
||||||
|
};
|
||||||
|
|
||||||
|
const command_line::arg_descriptor<std::string> core_rpc_server::arg_rpc_bind_port = {
|
||||||
|
"rpc-bind-port"
|
||||||
|
, "Port for RPC server"
|
||||||
|
, std::to_string(config::RPC_DEFAULT_PORT)
|
||||||
|
};
|
||||||
|
|
||||||
|
const command_line::arg_descriptor<std::string> core_rpc_server::arg_testnet_rpc_bind_port = {
|
||||||
|
"testnet-rpc-bind-port"
|
||||||
|
, "Port for testnet RPC server"
|
||||||
|
, std::to_string(config::testnet::RPC_DEFAULT_PORT)
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace cryptonote
|
||||||
|
@ -39,6 +39,10 @@
|
|||||||
#include "p2p/net_node.h"
|
#include "p2p/net_node.h"
|
||||||
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
#include "cryptonote_protocol/cryptonote_protocol_handler.h"
|
||||||
|
|
||||||
|
// yes, epee doesn't properly use its full namespace when calling its
|
||||||
|
// functions from macros. *sigh*
|
||||||
|
using namespace epee;
|
||||||
|
|
||||||
namespace cryptonote
|
namespace cryptonote
|
||||||
{
|
{
|
||||||
/************************************************************************/
|
/************************************************************************/
|
||||||
@ -47,19 +51,22 @@ namespace cryptonote
|
|||||||
class core_rpc_server: public epee::http_server_impl_base<core_rpc_server>
|
class core_rpc_server: public epee::http_server_impl_base<core_rpc_server>
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
|
||||||
|
static const command_line::arg_descriptor<std::string> arg_rpc_bind_ip;
|
||||||
|
static const command_line::arg_descriptor<std::string> arg_rpc_bind_port;
|
||||||
|
static const command_line::arg_descriptor<std::string> arg_testnet_rpc_bind_port;
|
||||||
|
|
||||||
typedef epee::net_utils::connection_context_base connection_context;
|
typedef epee::net_utils::connection_context_base connection_context;
|
||||||
|
|
||||||
core_rpc_server(
|
core_rpc_server(
|
||||||
core& cr
|
core& cr
|
||||||
, nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& p2p
|
, nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<cryptonote::core> >& p2p
|
||||||
, bool testnet
|
|
||||||
);
|
);
|
||||||
|
|
||||||
static void init_options(boost::program_options::options_description& desc);
|
static void init_options(boost::program_options::options_description& desc);
|
||||||
bool init(
|
bool init(
|
||||||
const boost::program_options::variables_map& vm
|
const boost::program_options::variables_map& vm
|
||||||
);
|
);
|
||||||
private:
|
|
||||||
|
|
||||||
CHAIN_HTTP_TO_MAP2(connection_context); //forward http requests to uri map
|
CHAIN_HTTP_TO_MAP2(connection_context); //forward http requests to uri map
|
||||||
|
|
||||||
@ -74,6 +81,11 @@ namespace cryptonote
|
|||||||
MAP_URI_AUTO_JON2("/stop_mining", on_stop_mining, COMMAND_RPC_STOP_MINING)
|
MAP_URI_AUTO_JON2("/stop_mining", on_stop_mining, COMMAND_RPC_STOP_MINING)
|
||||||
MAP_URI_AUTO_JON2("/mining_status", on_mining_status, COMMAND_RPC_MINING_STATUS)
|
MAP_URI_AUTO_JON2("/mining_status", on_mining_status, COMMAND_RPC_MINING_STATUS)
|
||||||
MAP_URI_AUTO_JON2("/save_bc", on_save_bc, COMMAND_RPC_SAVE_BC)
|
MAP_URI_AUTO_JON2("/save_bc", on_save_bc, COMMAND_RPC_SAVE_BC)
|
||||||
|
MAP_URI_AUTO_JON2("/get_peer_list", on_get_peer_list, COMMAND_RPC_GET_PEER_LIST)
|
||||||
|
MAP_URI_AUTO_JON2("/set_log_hash_rate", on_set_log_hash_rate, COMMAND_RPC_SET_LOG_HASH_RATE)
|
||||||
|
MAP_URI_AUTO_JON2("/set_log_level", on_set_log_level, COMMAND_RPC_SET_LOG_LEVEL)
|
||||||
|
MAP_URI_AUTO_JON2("/get_transaction_pool", on_get_transaction_pool, COMMAND_RPC_GET_TRANSACTION_POOL)
|
||||||
|
MAP_URI_AUTO_JON2("/stop_daemon", on_stop_daemon, COMMAND_RPC_STOP_DAEMON)
|
||||||
MAP_URI_AUTO_JON2("/getinfo", on_get_info, COMMAND_RPC_GET_INFO)
|
MAP_URI_AUTO_JON2("/getinfo", on_get_info, COMMAND_RPC_GET_INFO)
|
||||||
BEGIN_JSON_RPC_MAP("/json_rpc")
|
BEGIN_JSON_RPC_MAP("/json_rpc")
|
||||||
MAP_JON_RPC("getblockcount", on_getblockcount, COMMAND_RPC_GETBLOCKCOUNT)
|
MAP_JON_RPC("getblockcount", on_getblockcount, COMMAND_RPC_GETBLOCKCOUNT)
|
||||||
@ -88,29 +100,37 @@ namespace cryptonote
|
|||||||
END_JSON_RPC_MAP()
|
END_JSON_RPC_MAP()
|
||||||
END_URI_MAP2()
|
END_URI_MAP2()
|
||||||
|
|
||||||
bool on_get_height(const COMMAND_RPC_GET_HEIGHT::request& req, COMMAND_RPC_GET_HEIGHT::response& res, connection_context& cntx);
|
bool on_get_height(const COMMAND_RPC_GET_HEIGHT::request& req, COMMAND_RPC_GET_HEIGHT::response& res);
|
||||||
bool on_get_blocks(const COMMAND_RPC_GET_BLOCKS_FAST::request& req, COMMAND_RPC_GET_BLOCKS_FAST::response& res, connection_context& cntx);
|
bool on_get_blocks(const COMMAND_RPC_GET_BLOCKS_FAST::request& req, COMMAND_RPC_GET_BLOCKS_FAST::response& res);
|
||||||
bool on_get_transactions(const COMMAND_RPC_GET_TRANSACTIONS::request& req, COMMAND_RPC_GET_TRANSACTIONS::response& res, connection_context& cntx);
|
bool on_get_transactions(const COMMAND_RPC_GET_TRANSACTIONS::request& req, COMMAND_RPC_GET_TRANSACTIONS::response& res);
|
||||||
bool on_get_indexes(const COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::request& req, COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::response& res, connection_context& cntx);
|
bool on_get_indexes(const COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::request& req, COMMAND_RPC_GET_TX_GLOBAL_OUTPUTS_INDEXES::response& res);
|
||||||
bool on_send_raw_tx(const COMMAND_RPC_SEND_RAW_TX::request& req, COMMAND_RPC_SEND_RAW_TX::response& res, connection_context& cntx);
|
bool on_send_raw_tx(const COMMAND_RPC_SEND_RAW_TX::request& req, COMMAND_RPC_SEND_RAW_TX::response& res);
|
||||||
bool on_start_mining(const COMMAND_RPC_START_MINING::request& req, COMMAND_RPC_START_MINING::response& res, connection_context& cntx);
|
bool on_start_mining(const COMMAND_RPC_START_MINING::request& req, COMMAND_RPC_START_MINING::response& res);
|
||||||
bool on_stop_mining(const COMMAND_RPC_STOP_MINING::request& req, COMMAND_RPC_STOP_MINING::response& res, connection_context& cntx);
|
bool on_stop_mining(const COMMAND_RPC_STOP_MINING::request& req, COMMAND_RPC_STOP_MINING::response& res);
|
||||||
bool on_mining_status(const COMMAND_RPC_MINING_STATUS::request& req, COMMAND_RPC_MINING_STATUS::response& res, connection_context& cntx);
|
bool on_mining_status(const COMMAND_RPC_MINING_STATUS::request& req, COMMAND_RPC_MINING_STATUS::response& res);
|
||||||
bool on_get_random_outs(const COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::request& req, COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::response& res, connection_context& cntx);
|
bool on_get_random_outs(const COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::request& req, COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::response& res);
|
||||||
bool on_get_info(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, connection_context& cntx);
|
bool on_get_info(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res);
|
||||||
bool on_save_bc(const COMMAND_RPC_SAVE_BC::request& req, COMMAND_RPC_SAVE_BC::response& res, connection_context& cntx);
|
bool on_save_bc(const COMMAND_RPC_SAVE_BC::request& req, COMMAND_RPC_SAVE_BC::response& res);
|
||||||
|
bool on_get_peer_list(const COMMAND_RPC_GET_PEER_LIST::request& req, COMMAND_RPC_GET_PEER_LIST::response& res);
|
||||||
|
bool on_set_log_hash_rate(const COMMAND_RPC_SET_LOG_HASH_RATE::request& req, COMMAND_RPC_SET_LOG_HASH_RATE::response& res);
|
||||||
|
bool on_set_log_level(const COMMAND_RPC_SET_LOG_LEVEL::request& req, COMMAND_RPC_SET_LOG_LEVEL::response& res);
|
||||||
|
bool on_get_transaction_pool(const COMMAND_RPC_GET_TRANSACTION_POOL::request& req, COMMAND_RPC_GET_TRANSACTION_POOL::response& res);
|
||||||
|
bool on_stop_daemon(const COMMAND_RPC_STOP_DAEMON::request& req, COMMAND_RPC_STOP_DAEMON::response& res);
|
||||||
|
|
||||||
//json_rpc
|
//json_rpc
|
||||||
bool on_getblockcount(const COMMAND_RPC_GETBLOCKCOUNT::request& req, COMMAND_RPC_GETBLOCKCOUNT::response& res, connection_context& cntx);
|
bool on_getblockcount(const COMMAND_RPC_GETBLOCKCOUNT::request& req, COMMAND_RPC_GETBLOCKCOUNT::response& res);
|
||||||
bool on_getblockhash(const COMMAND_RPC_GETBLOCKHASH::request& req, COMMAND_RPC_GETBLOCKHASH::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_getblockhash(const COMMAND_RPC_GETBLOCKHASH::request& req, COMMAND_RPC_GETBLOCKHASH::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_getblocktemplate(const COMMAND_RPC_GETBLOCKTEMPLATE::request& req, COMMAND_RPC_GETBLOCKTEMPLATE::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_getblocktemplate(const COMMAND_RPC_GETBLOCKTEMPLATE::request& req, COMMAND_RPC_GETBLOCKTEMPLATE::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_submitblock(const COMMAND_RPC_SUBMITBLOCK::request& req, COMMAND_RPC_SUBMITBLOCK::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_submitblock(const COMMAND_RPC_SUBMITBLOCK::request& req, COMMAND_RPC_SUBMITBLOCK::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_get_last_block_header(const COMMAND_RPC_GET_LAST_BLOCK_HEADER::request& req, COMMAND_RPC_GET_LAST_BLOCK_HEADER::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_get_last_block_header(const COMMAND_RPC_GET_LAST_BLOCK_HEADER::request& req, COMMAND_RPC_GET_LAST_BLOCK_HEADER::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_get_block_header_by_hash(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_get_block_header_by_hash(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HASH::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_get_block_header_by_height(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_get_block_header_by_height(const COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::request& req, COMMAND_RPC_GET_BLOCK_HEADER_BY_HEIGHT::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_get_connections(const COMMAND_RPC_GET_CONNECTIONS::request& req, COMMAND_RPC_GET_CONNECTIONS::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_get_connections(const COMMAND_RPC_GET_CONNECTIONS::request& req, COMMAND_RPC_GET_CONNECTIONS::response& res, epee::json_rpc::error& error_resp);
|
||||||
bool on_get_info_json(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, epee::json_rpc::error& error_resp, connection_context& cntx);
|
bool on_get_info_json(const COMMAND_RPC_GET_INFO::request& req, COMMAND_RPC_GET_INFO::response& res, epee::json_rpc::error& error_resp);
|
||||||
//-----------------------
|
//-----------------------
|
||||||
|
|
||||||
|
private:
|
||||||
|
|
||||||
bool handle_command_line(
|
bool handle_command_line(
|
||||||
const boost::program_options::variables_map& vm
|
const boost::program_options::variables_map& vm
|
||||||
);
|
);
|
||||||
|
@ -39,6 +39,7 @@ namespace cryptonote
|
|||||||
//-----------------------------------------------
|
//-----------------------------------------------
|
||||||
#define CORE_RPC_STATUS_OK "OK"
|
#define CORE_RPC_STATUS_OK "OK"
|
||||||
#define CORE_RPC_STATUS_BUSY "BUSY"
|
#define CORE_RPC_STATUS_BUSY "BUSY"
|
||||||
|
#define CORE_RPC_STATUS_NOT_MINING "NOT MINING"
|
||||||
|
|
||||||
struct COMMAND_RPC_GET_HEIGHT
|
struct COMMAND_RPC_GET_HEIGHT
|
||||||
{
|
{
|
||||||
@ -510,6 +511,135 @@ namespace cryptonote
|
|||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
struct peer {
|
||||||
|
uint64_t id;
|
||||||
|
uint32_t ip;
|
||||||
|
uint16_t port;
|
||||||
|
uint64_t last_seen;
|
||||||
|
|
||||||
|
peer() = default;
|
||||||
|
|
||||||
|
peer(uint64_t id, uint32_t ip, uint16_t port, uint64_t last_seen)
|
||||||
|
: id(id), ip(ip), port(port), last_seen(last_seen)
|
||||||
|
{}
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(id)
|
||||||
|
KV_SERIALIZE(ip)
|
||||||
|
KV_SERIALIZE(port)
|
||||||
|
KV_SERIALIZE(last_seen)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_GET_PEER_LIST
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
std::vector<peer> white_list;
|
||||||
|
std::vector<peer> gray_list;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
KV_SERIALIZE(white_list)
|
||||||
|
KV_SERIALIZE(gray_list)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_SET_LOG_HASH_RATE
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
bool visible;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(visible)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_SET_LOG_LEVEL
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
int8_t level;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(level)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
struct tx_info
|
||||||
|
{
|
||||||
|
std::string id_hash;
|
||||||
|
std::string tx_json; // TODO - expose this data directly
|
||||||
|
uint64_t blob_size;
|
||||||
|
uint64_t fee;
|
||||||
|
std::string max_used_block_id_hash;
|
||||||
|
uint64_t max_used_block_height;
|
||||||
|
bool kept_by_block;
|
||||||
|
uint64_t last_failed_height;
|
||||||
|
std::string last_failed_id_hash;
|
||||||
|
uint64_t receive_time;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(id_hash)
|
||||||
|
KV_SERIALIZE(tx_json)
|
||||||
|
KV_SERIALIZE(blob_size)
|
||||||
|
KV_SERIALIZE(fee)
|
||||||
|
KV_SERIALIZE(max_used_block_id_hash)
|
||||||
|
KV_SERIALIZE(max_used_block_height)
|
||||||
|
KV_SERIALIZE(kept_by_block)
|
||||||
|
KV_SERIALIZE(last_failed_height)
|
||||||
|
KV_SERIALIZE(last_failed_id_hash)
|
||||||
|
KV_SERIALIZE(receive_time)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_GET_TRANSACTION_POOL
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
std::vector<tx_info> transactions;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
KV_SERIALIZE(transactions)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
struct COMMAND_RPC_GET_CONNECTIONS
|
struct COMMAND_RPC_GET_CONNECTIONS
|
||||||
{
|
{
|
||||||
struct request
|
struct request
|
||||||
@ -530,5 +660,48 @@ namespace cryptonote
|
|||||||
};
|
};
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_GET_BLOCK_HEADERS_RANGE
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
uint64_t start_height;
|
||||||
|
uint64_t end_height;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(start_height)
|
||||||
|
KV_SERIALIZE(end_height)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
std::vector<block_header_responce> headers;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
KV_SERIALIZE(headers)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
struct COMMAND_RPC_STOP_DAEMON
|
||||||
|
{
|
||||||
|
struct request
|
||||||
|
{
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
|
||||||
|
struct response
|
||||||
|
{
|
||||||
|
std::string status;
|
||||||
|
|
||||||
|
BEGIN_KV_SERIALIZE_MAP()
|
||||||
|
KV_SERIALIZE(status)
|
||||||
|
END_KV_SERIALIZE_MAP()
|
||||||
|
};
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -67,7 +67,7 @@ namespace po = boost::program_options;
|
|||||||
|
|
||||||
#define EXTENDED_LOGS_FILE "wallet_details.log"
|
#define EXTENDED_LOGS_FILE "wallet_details.log"
|
||||||
|
|
||||||
unsigned int epee::g_test_dbg_lock_sleep = 0;
|
#define DEFAULT_MIX 3
|
||||||
|
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
@ -281,7 +281,7 @@ simple_wallet::simple_wallet()
|
|||||||
m_cmd_binder.set_handler("incoming_transfers", boost::bind(&simple_wallet::show_incoming_transfers, this, _1), "incoming_transfers [available|unavailable] - Show incoming transfers - all of them or filter them by availability");
|
m_cmd_binder.set_handler("incoming_transfers", boost::bind(&simple_wallet::show_incoming_transfers, this, _1), "incoming_transfers [available|unavailable] - Show incoming transfers - all of them or filter them by availability");
|
||||||
m_cmd_binder.set_handler("payments", boost::bind(&simple_wallet::show_payments, this, _1), "payments <payment_id_1> [<payment_id_2> ... <payment_id_N>] - Show payments <payment_id_1>, ... <payment_id_N>");
|
m_cmd_binder.set_handler("payments", boost::bind(&simple_wallet::show_payments, this, _1), "payments <payment_id_1> [<payment_id_2> ... <payment_id_N>] - Show payments <payment_id_1>, ... <payment_id_N>");
|
||||||
m_cmd_binder.set_handler("bc_height", boost::bind(&simple_wallet::show_blockchain_height, this, _1), "Show blockchain height");
|
m_cmd_binder.set_handler("bc_height", boost::bind(&simple_wallet::show_blockchain_height, this, _1), "Show blockchain height");
|
||||||
m_cmd_binder.set_handler("transfer", boost::bind(&simple_wallet::transfer, this, _1), "transfer <mixin_count> <addr_1> <amount_1> [<addr_2> <amount_2> ... <addr_N> <amount_N>] [payment_id] - Transfer <amount_1>,... <amount_N> to <address_1>,... <address_N>, respectively. <mixin_count> is the number of transactions yours is indistinguishable from (from 0 to maximum available)");
|
m_cmd_binder.set_handler("transfer", boost::bind(&simple_wallet::transfer, this, _1), "transfer [<mixin_count>] <addr_1> <amount_1> [<addr_2> <amount_2> ... <addr_N> <amount_N>] [payment_id] - Transfer <amount_1>,... <amount_N> to <address_1>,... <address_N>, respectively. <mixin_count> is the number of transactions yours is indistinguishable from (from 0 to maximum available)");
|
||||||
m_cmd_binder.set_handler("set_log", boost::bind(&simple_wallet::set_log, this, _1), "set_log <level> - Change current log detalization level, <level> is a number 0-4");
|
m_cmd_binder.set_handler("set_log", boost::bind(&simple_wallet::set_log, this, _1), "set_log <level> - Change current log detalization level, <level> is a number 0-4");
|
||||||
m_cmd_binder.set_handler("address", boost::bind(&simple_wallet::print_address, this, _1), "Show current wallet public address");
|
m_cmd_binder.set_handler("address", boost::bind(&simple_wallet::print_address, this, _1), "Show current wallet public address");
|
||||||
m_cmd_binder.set_handler("save", boost::bind(&simple_wallet::save, this, _1), "Save wallet synchronized data");
|
m_cmd_binder.set_handler("save", boost::bind(&simple_wallet::save, this, _1), "Save wallet synchronized data");
|
||||||
@ -1068,19 +1068,24 @@ bool simple_wallet::transfer(const std::vector<std::string> &args_)
|
|||||||
return true;
|
return true;
|
||||||
|
|
||||||
std::vector<std::string> local_args = args_;
|
std::vector<std::string> local_args = args_;
|
||||||
if(local_args.size() < 3)
|
|
||||||
{
|
|
||||||
fail_msg_writer() << "wrong number of arguments, expected at least 3, got " << local_args.size();
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
size_t fake_outs_count;
|
size_t fake_outs_count;
|
||||||
|
if(local_args.size() > 0) {
|
||||||
if(!epee::string_tools::get_xtype_from_string(fake_outs_count, local_args[0]))
|
if(!epee::string_tools::get_xtype_from_string(fake_outs_count, local_args[0]))
|
||||||
{
|
{
|
||||||
fail_msg_writer() << "mixin_count should be non-negative integer, got " << local_args[0];
|
fake_outs_count = DEFAULT_MIX;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
local_args.erase(local_args.begin());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(local_args.size() < 2)
|
||||||
|
{
|
||||||
|
fail_msg_writer() << "wrong number of arguments";
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
local_args.erase(local_args.begin());
|
|
||||||
|
|
||||||
std::vector<uint8_t> extra;
|
std::vector<uint8_t> extra;
|
||||||
if (1 == local_args.size() % 2)
|
if (1 == local_args.size() % 2)
|
||||||
|
@ -85,7 +85,7 @@ namespace tools
|
|||||||
return epee::http_server_impl_base<wallet_rpc_server, connection_context>::init(m_port, m_bind_ip);
|
return epee::http_server_impl_base<wallet_rpc_server, connection_context>::init(m_port, m_bind_ip);
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_getbalance(const wallet_rpc::COMMAND_RPC_GET_BALANCE::request& req, wallet_rpc::COMMAND_RPC_GET_BALANCE::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_getbalance(const wallet_rpc::COMMAND_RPC_GET_BALANCE::request& req, wallet_rpc::COMMAND_RPC_GET_BALANCE::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
@ -101,7 +101,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_getaddress(const wallet_rpc::COMMAND_RPC_GET_ADDRESS::request& req, wallet_rpc::COMMAND_RPC_GET_ADDRESS::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_getaddress(const wallet_rpc::COMMAND_RPC_GET_ADDRESS::request& req, wallet_rpc::COMMAND_RPC_GET_ADDRESS::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
@ -161,7 +161,7 @@ namespace tools
|
|||||||
}
|
}
|
||||||
|
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_transfer(const wallet_rpc::COMMAND_RPC_TRANSFER::request& req, wallet_rpc::COMMAND_RPC_TRANSFER::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_transfer(const wallet_rpc::COMMAND_RPC_TRANSFER::request& req, wallet_rpc::COMMAND_RPC_TRANSFER::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
|
|
||||||
std::vector<cryptonote::tx_destination_entry> dsts;
|
std::vector<cryptonote::tx_destination_entry> dsts;
|
||||||
@ -219,7 +219,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_transfer_split(const wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::request& req, wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_transfer_split(const wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::request& req, wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
|
|
||||||
std::vector<cryptonote::tx_destination_entry> dsts;
|
std::vector<cryptonote::tx_destination_entry> dsts;
|
||||||
@ -273,7 +273,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_store(const wallet_rpc::COMMAND_RPC_STORE::request& req, wallet_rpc::COMMAND_RPC_STORE::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_store(const wallet_rpc::COMMAND_RPC_STORE::request& req, wallet_rpc::COMMAND_RPC_STORE::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
if (m_wallet.restricted())
|
if (m_wallet.restricted())
|
||||||
{
|
{
|
||||||
@ -295,7 +295,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_get_payments(const wallet_rpc::COMMAND_RPC_GET_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_PAYMENTS::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_get_payments(const wallet_rpc::COMMAND_RPC_GET_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_PAYMENTS::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
crypto::hash payment_id;
|
crypto::hash payment_id;
|
||||||
cryptonote::blobdata payment_id_blob;
|
cryptonote::blobdata payment_id_blob;
|
||||||
@ -332,7 +332,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_get_bulk_payments(const wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_get_bulk_payments(const wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
res.payments.clear();
|
res.payments.clear();
|
||||||
|
|
||||||
@ -397,7 +397,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_incoming_transfers(const wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::request& req, wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_incoming_transfers(const wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::request& req, wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
if(req.transfer_type.compare("all") != 0 && req.transfer_type.compare("available") != 0 && req.transfer_type.compare("unavailable") != 0)
|
if(req.transfer_type.compare("all") != 0 && req.transfer_type.compare("available") != 0 && req.transfer_type.compare("unavailable") != 0)
|
||||||
{
|
{
|
||||||
@ -450,7 +450,7 @@ namespace tools
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
//------------------------------------------------------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------------------------------------------------------
|
||||||
bool wallet_rpc_server::on_query_key(const wallet_rpc::COMMAND_RPC_QUERY_KEY::request& req, wallet_rpc::COMMAND_RPC_QUERY_KEY::response& res, epee::json_rpc::error& er, connection_context& cntx)
|
bool wallet_rpc_server::on_query_key(const wallet_rpc::COMMAND_RPC_QUERY_KEY::request& req, wallet_rpc::COMMAND_RPC_QUERY_KEY::response& res, epee::json_rpc::error& er)
|
||||||
{
|
{
|
||||||
if (m_wallet.restricted())
|
if (m_wallet.restricted())
|
||||||
{
|
{
|
||||||
|
@ -74,20 +74,20 @@ namespace tools
|
|||||||
END_URI_MAP2()
|
END_URI_MAP2()
|
||||||
|
|
||||||
//json_rpc
|
//json_rpc
|
||||||
bool on_getbalance(const wallet_rpc::COMMAND_RPC_GET_BALANCE::request& req, wallet_rpc::COMMAND_RPC_GET_BALANCE::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_getbalance(const wallet_rpc::COMMAND_RPC_GET_BALANCE::request& req, wallet_rpc::COMMAND_RPC_GET_BALANCE::response& res, epee::json_rpc::error& er);
|
||||||
bool on_getaddress(const wallet_rpc::COMMAND_RPC_GET_ADDRESS::request& req, wallet_rpc::COMMAND_RPC_GET_ADDRESS::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_getaddress(const wallet_rpc::COMMAND_RPC_GET_ADDRESS::request& req, wallet_rpc::COMMAND_RPC_GET_ADDRESS::response& res, epee::json_rpc::error& er);
|
||||||
bool validate_transfer(const std::list<wallet_rpc::transfer_destination> destinations, const std::string payment_id, std::vector<cryptonote::tx_destination_entry>& dsts, std::vector<uint8_t>& extra, epee::json_rpc::error& er);
|
bool validate_transfer(const std::list<wallet_rpc::transfer_destination> destinations, const std::string payment_id, std::vector<cryptonote::tx_destination_entry>& dsts, std::vector<uint8_t>& extra, epee::json_rpc::error& er);
|
||||||
bool on_transfer(const wallet_rpc::COMMAND_RPC_TRANSFER::request& req, wallet_rpc::COMMAND_RPC_TRANSFER::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_transfer(const wallet_rpc::COMMAND_RPC_TRANSFER::request& req, wallet_rpc::COMMAND_RPC_TRANSFER::response& res, epee::json_rpc::error& er);
|
||||||
bool on_transfer_split(const wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::request& req, wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_transfer_split(const wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::request& req, wallet_rpc::COMMAND_RPC_TRANSFER_SPLIT::response& res, epee::json_rpc::error& er);
|
||||||
bool on_store(const wallet_rpc::COMMAND_RPC_STORE::request& req, wallet_rpc::COMMAND_RPC_STORE::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_store(const wallet_rpc::COMMAND_RPC_STORE::request& req, wallet_rpc::COMMAND_RPC_STORE::response& res, epee::json_rpc::error& er);
|
||||||
bool on_get_payments(const wallet_rpc::COMMAND_RPC_GET_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_PAYMENTS::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_get_payments(const wallet_rpc::COMMAND_RPC_GET_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_PAYMENTS::response& res, epee::json_rpc::error& er);
|
||||||
bool on_get_bulk_payments(const wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_get_bulk_payments(const wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::request& req, wallet_rpc::COMMAND_RPC_GET_BULK_PAYMENTS::response& res, epee::json_rpc::error& er);
|
||||||
bool on_incoming_transfers(const wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::request& req, wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_incoming_transfers(const wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::request& req, wallet_rpc::COMMAND_RPC_INCOMING_TRANSFERS::response& res, epee::json_rpc::error& er);
|
||||||
|
|
||||||
bool handle_command_line(const boost::program_options::variables_map& vm);
|
bool handle_command_line(const boost::program_options::variables_map& vm);
|
||||||
|
|
||||||
//json rpc v2
|
//json rpc v2
|
||||||
bool on_query_key(const wallet_rpc::COMMAND_RPC_QUERY_KEY::request& req, wallet_rpc::COMMAND_RPC_QUERY_KEY::response& res, epee::json_rpc::error& er, connection_context& cntx);
|
bool on_query_key(const wallet_rpc::COMMAND_RPC_QUERY_KEY::request& req, wallet_rpc::COMMAND_RPC_QUERY_KEY::response& res, epee::json_rpc::error& er);
|
||||||
|
|
||||||
wallet2& m_wallet;
|
wallet2& m_wallet;
|
||||||
std::string m_port;
|
std::string m_port;
|
||||||
|
@ -108,7 +108,6 @@ int main(int argc, char* argv[])
|
|||||||
cryptonote::t_cryptonote_protocol_handler<tests::proxy_core> cprotocol(pr_core, NULL);
|
cryptonote::t_cryptonote_protocol_handler<tests::proxy_core> cprotocol(pr_core, NULL);
|
||||||
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<tests::proxy_core> > p2psrv {
|
nodetool::node_server<cryptonote::t_cryptonote_protocol_handler<tests::proxy_core> > p2psrv {
|
||||||
cprotocol
|
cprotocol
|
||||||
, std::move(config::NETWORK_ID)
|
|
||||||
};
|
};
|
||||||
cprotocol.set_p2p_endpoint(&p2psrv);
|
cprotocol.set_p2p_endpoint(&p2psrv);
|
||||||
//pr_core.set_cryptonote_protocol(&cprotocol);
|
//pr_core.set_cryptonote_protocol(&cprotocol);
|
||||||
@ -117,7 +116,7 @@ int main(int argc, char* argv[])
|
|||||||
//initialize objects
|
//initialize objects
|
||||||
|
|
||||||
LOG_PRINT_L0("Initializing p2p server...");
|
LOG_PRINT_L0("Initializing p2p server...");
|
||||||
bool res = p2psrv.init(vm, false);
|
bool res = p2psrv.init(vm);
|
||||||
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize p2p server.");
|
CHECK_AND_ASSERT_MES(res, 1, "Failed to initialize p2p server.");
|
||||||
LOG_PRINT_L0("P2p server initialized OK");
|
LOG_PRINT_L0("P2p server initialized OK");
|
||||||
|
|
||||||
|
@ -487,7 +487,9 @@ inline bool do_replay_events(std::vector<test_event_entry>& events)
|
|||||||
|
|
||||||
cryptonote::cryptonote_protocol_stub pr; //TODO: stub only for this kind of test, make real validation of relayed objects
|
cryptonote::cryptonote_protocol_stub pr; //TODO: stub only for this kind of test, make real validation of relayed objects
|
||||||
cryptonote::core c(&pr);
|
cryptonote::core c(&pr);
|
||||||
if (!c.init(vm, false))
|
// FIXME: make sure that vm has arg_testnet_on set to true or false if
|
||||||
|
// this test needs for it to be so.
|
||||||
|
if (!c.init(vm))
|
||||||
{
|
{
|
||||||
std::cout << concolor::magenta << "Failed to init core" << concolor::normal << std::endl;
|
std::cout << concolor::magenta << "Failed to init core" << concolor::normal << std::endl;
|
||||||
return false;
|
return false;
|
||||||
|
157
tests/gtest/CHANGES
Normal file
157
tests/gtest/CHANGES
Normal file
@ -0,0 +1,157 @@
|
|||||||
|
Changes for 1.7.0:
|
||||||
|
|
||||||
|
* New feature: death tests are supported on OpenBSD and in iOS
|
||||||
|
simulator now.
|
||||||
|
* New feature: Google Test now implements a protocol to allow
|
||||||
|
a test runner to detect that a test program has exited
|
||||||
|
prematurely and report it as a failure (before it would be
|
||||||
|
falsely reported as a success if the exit code is 0).
|
||||||
|
* New feature: Test::RecordProperty() can now be used outside of the
|
||||||
|
lifespan of a test method, in which case it will be attributed to
|
||||||
|
the current test case or the test program in the XML report.
|
||||||
|
* New feature (potentially breaking): --gtest_list_tests now prints
|
||||||
|
the type parameters and value parameters for each test.
|
||||||
|
* Improvement: char pointers and char arrays are now escaped properly
|
||||||
|
in failure messages.
|
||||||
|
* Improvement: failure summary in XML reports now includes file and
|
||||||
|
line information.
|
||||||
|
* Improvement: the <testsuites> XML element now has a timestamp attribute.
|
||||||
|
* Improvement: When --gtest_filter is specified, XML report now doesn't
|
||||||
|
contain information about tests that are filtered out.
|
||||||
|
* Fixed the bug where long --gtest_filter flag values are truncated in
|
||||||
|
death tests.
|
||||||
|
* Potentially breaking change: RUN_ALL_TESTS() is now implemented as a
|
||||||
|
function instead of a macro in order to work better with Clang.
|
||||||
|
* Compatibility fixes with C++ 11 and various platforms.
|
||||||
|
* Bug/warning fixes.
|
||||||
|
|
||||||
|
Changes for 1.6.0:
|
||||||
|
|
||||||
|
* New feature: ADD_FAILURE_AT() for reporting a test failure at the
|
||||||
|
given source location -- useful for writing testing utilities.
|
||||||
|
* New feature: the universal value printer is moved from Google Mock
|
||||||
|
to Google Test.
|
||||||
|
* New feature: type parameters and value parameters are reported in
|
||||||
|
the XML report now.
|
||||||
|
* A gtest_disable_pthreads CMake option.
|
||||||
|
* Colored output works in GNU Screen sessions now.
|
||||||
|
* Parameters of value-parameterized tests are now printed in the
|
||||||
|
textual output.
|
||||||
|
* Failures from ad hoc test assertions run before RUN_ALL_TESTS() are
|
||||||
|
now correctly reported.
|
||||||
|
* Arguments of ASSERT_XY and EXPECT_XY no longer need to support << to
|
||||||
|
ostream.
|
||||||
|
* More complete handling of exceptions.
|
||||||
|
* GTEST_ASSERT_XY can be used instead of ASSERT_XY in case the latter
|
||||||
|
name is already used by another library.
|
||||||
|
* --gtest_catch_exceptions is now true by default, allowing a test
|
||||||
|
program to continue after an exception is thrown.
|
||||||
|
* Value-parameterized test fixtures can now derive from Test and
|
||||||
|
WithParamInterface<T> separately, easing conversion of legacy tests.
|
||||||
|
* Death test messages are clearly marked to make them more
|
||||||
|
distinguishable from other messages.
|
||||||
|
* Compatibility fixes for Android, Google Native Client, MinGW, HP UX,
|
||||||
|
PowerPC, Lucid autotools, libCStd, Sun C++, Borland C++ Builder (Code Gear),
|
||||||
|
IBM XL C++ (Visual Age C++), and C++0x.
|
||||||
|
* Bug fixes and implementation clean-ups.
|
||||||
|
* Potentially incompatible changes: disables the harmful 'make install'
|
||||||
|
command in autotools.
|
||||||
|
|
||||||
|
Changes for 1.5.0:
|
||||||
|
|
||||||
|
* New feature: assertions can be safely called in multiple threads
|
||||||
|
where the pthreads library is available.
|
||||||
|
* New feature: predicates used inside EXPECT_TRUE() and friends
|
||||||
|
can now generate custom failure messages.
|
||||||
|
* New feature: Google Test can now be compiled as a DLL.
|
||||||
|
* New feature: fused source files are included.
|
||||||
|
* New feature: prints help when encountering unrecognized Google Test flags.
|
||||||
|
* Experimental feature: CMake build script (requires CMake 2.6.4+).
|
||||||
|
* Experimental feature: the Pump script for meta programming.
|
||||||
|
* double values streamed to an assertion are printed with enough precision
|
||||||
|
to differentiate any two different values.
|
||||||
|
* Google Test now works on Solaris and AIX.
|
||||||
|
* Build and test script improvements.
|
||||||
|
* Bug fixes and implementation clean-ups.
|
||||||
|
|
||||||
|
Potentially breaking changes:
|
||||||
|
|
||||||
|
* Stopped supporting VC++ 7.1 with exceptions disabled.
|
||||||
|
* Dropped support for 'make install'.
|
||||||
|
|
||||||
|
Changes for 1.4.0:
|
||||||
|
|
||||||
|
* New feature: the event listener API
|
||||||
|
* New feature: test shuffling
|
||||||
|
* New feature: the XML report format is closer to junitreport and can
|
||||||
|
be parsed by Hudson now.
|
||||||
|
* New feature: when a test runs under Visual Studio, its failures are
|
||||||
|
integrated in the IDE.
|
||||||
|
* New feature: /MD(d) versions of VC++ projects.
|
||||||
|
* New feature: elapsed time for the tests is printed by default.
|
||||||
|
* New feature: comes with a TR1 tuple implementation such that Boost
|
||||||
|
is no longer needed for Combine().
|
||||||
|
* New feature: EXPECT_DEATH_IF_SUPPORTED macro and friends.
|
||||||
|
* New feature: the Xcode project can now produce static gtest
|
||||||
|
libraries in addition to a framework.
|
||||||
|
* Compatibility fixes for Solaris, Cygwin, minGW, Windows Mobile,
|
||||||
|
Symbian, gcc, and C++Builder.
|
||||||
|
* Bug fixes and implementation clean-ups.
|
||||||
|
|
||||||
|
Changes for 1.3.0:
|
||||||
|
|
||||||
|
* New feature: death tests on Windows, Cygwin, and Mac.
|
||||||
|
* New feature: ability to use Google Test assertions in other testing
|
||||||
|
frameworks.
|
||||||
|
* New feature: ability to run disabled test via
|
||||||
|
--gtest_also_run_disabled_tests.
|
||||||
|
* New feature: the --help flag for printing the usage.
|
||||||
|
* New feature: access to Google Test flag values in user code.
|
||||||
|
* New feature: a script that packs Google Test into one .h and one
|
||||||
|
.cc file for easy deployment.
|
||||||
|
* New feature: support for distributing test functions to multiple
|
||||||
|
machines (requires support from the test runner).
|
||||||
|
* Bug fixes and implementation clean-ups.
|
||||||
|
|
||||||
|
Changes for 1.2.1:
|
||||||
|
|
||||||
|
* Compatibility fixes for Linux IA-64 and IBM z/OS.
|
||||||
|
* Added support for using Boost and other TR1 implementations.
|
||||||
|
* Changes to the build scripts to support upcoming release of Google C++
|
||||||
|
Mocking Framework.
|
||||||
|
* Added Makefile to the distribution package.
|
||||||
|
* Improved build instructions in README.
|
||||||
|
|
||||||
|
Changes for 1.2.0:
|
||||||
|
|
||||||
|
* New feature: value-parameterized tests.
|
||||||
|
* New feature: the ASSERT/EXPECT_(NON)FATAL_FAILURE(_ON_ALL_THREADS)
|
||||||
|
macros.
|
||||||
|
* Changed the XML report format to match JUnit/Ant's.
|
||||||
|
* Added tests to the Xcode project.
|
||||||
|
* Added scons/SConscript for building with SCons.
|
||||||
|
* Added src/gtest-all.cc for building Google Test from a single file.
|
||||||
|
* Fixed compatibility with Solaris and z/OS.
|
||||||
|
* Enabled running Python tests on systems with python 2.3 installed,
|
||||||
|
e.g. Mac OS X 10.4.
|
||||||
|
* Bug fixes.
|
||||||
|
|
||||||
|
Changes for 1.1.0:
|
||||||
|
|
||||||
|
* New feature: type-parameterized tests.
|
||||||
|
* New feature: exception assertions.
|
||||||
|
* New feature: printing elapsed time of tests.
|
||||||
|
* Improved the robustness of death tests.
|
||||||
|
* Added an Xcode project and samples.
|
||||||
|
* Adjusted the output format on Windows to be understandable by Visual Studio.
|
||||||
|
* Minor bug fixes.
|
||||||
|
|
||||||
|
Changes for 1.0.1:
|
||||||
|
|
||||||
|
* Added project files for Visual Studio 7.1.
|
||||||
|
* Fixed issues with compiling on Mac OS X.
|
||||||
|
* Fixed issues with compiling on Cygwin.
|
||||||
|
|
||||||
|
Changes for 1.0.0:
|
||||||
|
|
||||||
|
* Initial Open Source release of Google Test
|
@ -59,6 +59,16 @@ include_directories(
|
|||||||
# Where Google Test's libraries can be found.
|
# Where Google Test's libraries can be found.
|
||||||
link_directories(${gtest_BINARY_DIR}/src)
|
link_directories(${gtest_BINARY_DIR}/src)
|
||||||
|
|
||||||
|
# Summary of tuple support for Microsoft Visual Studio:
|
||||||
|
# Compiler version(MS) version(cmake) Support
|
||||||
|
# ---------- ----------- -------------- -----------------------------
|
||||||
|
# <= VS 2010 <= 10 <= 1600 Use Google Tests's own tuple.
|
||||||
|
# VS 2012 11 1700 std::tr1::tuple + _VARIADIC_MAX=10
|
||||||
|
# VS 2013 12 1800 std::tr1::tuple
|
||||||
|
if (MSVC AND MSVC_VERSION EQUAL 1700)
|
||||||
|
add_definitions(/D _VARIADIC_MAX=10)
|
||||||
|
endif()
|
||||||
|
|
||||||
########################################################################
|
########################################################################
|
||||||
#
|
#
|
||||||
# Defines the gtest & gtest_main libraries. User tests should link
|
# Defines the gtest & gtest_main libraries. User tests should link
|
||||||
@ -77,7 +87,7 @@ target_link_libraries(gtest_main gtest)
|
|||||||
#
|
#
|
||||||
# They are not built by default. To build them, set the
|
# They are not built by default. To build them, set the
|
||||||
# gtest_build_samples option to ON. You can do it by running ccmake
|
# gtest_build_samples option to ON. You can do it by running ccmake
|
||||||
# or specifying the -Dbuild_gtest_samples=ON flag when running cmake.
|
# or specifying the -Dgtest_build_samples=ON flag when running cmake.
|
||||||
|
|
||||||
if (gtest_build_samples)
|
if (gtest_build_samples)
|
||||||
cxx_executable(sample1_unittest samples gtest_main samples/sample1.cc)
|
cxx_executable(sample1_unittest samples gtest_main samples/sample1.cc)
|
||||||
@ -124,6 +134,8 @@ if (gtest_build_tests)
|
|||||||
test/gtest-param-test2_test.cc)
|
test/gtest-param-test2_test.cc)
|
||||||
cxx_test(gtest-port_test gtest_main)
|
cxx_test(gtest-port_test gtest_main)
|
||||||
cxx_test(gtest_pred_impl_unittest gtest_main)
|
cxx_test(gtest_pred_impl_unittest gtest_main)
|
||||||
|
cxx_test(gtest_premature_exit_test gtest
|
||||||
|
test/gtest_premature_exit_test.cc)
|
||||||
cxx_test(gtest-printers_test gtest_main)
|
cxx_test(gtest-printers_test gtest_main)
|
||||||
cxx_test(gtest_prod_test gtest_main
|
cxx_test(gtest_prod_test gtest_main
|
||||||
test/production.cc)
|
test/production.cc)
|
||||||
@ -140,10 +152,13 @@ if (gtest_build_tests)
|
|||||||
############################################################
|
############################################################
|
||||||
# C++ tests built with non-standard compiler flags.
|
# C++ tests built with non-standard compiler flags.
|
||||||
|
|
||||||
|
# MSVC 7.1 does not support STL with exceptions disabled.
|
||||||
|
if (NOT MSVC OR MSVC_VERSION GREATER 1310)
|
||||||
cxx_library(gtest_no_exception "${cxx_no_exception}"
|
cxx_library(gtest_no_exception "${cxx_no_exception}"
|
||||||
src/gtest-all.cc)
|
src/gtest-all.cc)
|
||||||
cxx_library(gtest_main_no_exception "${cxx_no_exception}"
|
cxx_library(gtest_main_no_exception "${cxx_no_exception}"
|
||||||
src/gtest-all.cc src/gtest_main.cc)
|
src/gtest-all.cc src/gtest_main.cc)
|
||||||
|
endif()
|
||||||
cxx_library(gtest_main_no_rtti "${cxx_no_rtti}"
|
cxx_library(gtest_main_no_rtti "${cxx_no_rtti}"
|
||||||
src/gtest-all.cc src/gtest_main.cc)
|
src/gtest-all.cc src/gtest_main.cc)
|
||||||
|
|
||||||
@ -166,12 +181,10 @@ if (gtest_build_tests)
|
|||||||
PROPERTIES
|
PROPERTIES
|
||||||
COMPILE_DEFINITIONS "GTEST_LINKED_AS_SHARED_LIBRARY=1")
|
COMPILE_DEFINITIONS "GTEST_LINKED_AS_SHARED_LIBRARY=1")
|
||||||
|
|
||||||
if (NOT MSVC OR NOT MSVC_VERSION EQUAL 1600)
|
if (NOT MSVC OR MSVC_VERSION LESS 1600) # 1600 is Visual Studio 2010.
|
||||||
# The C++ Standard specifies tuple_element<int, class>.
|
# Visual Studio 2010, 2012, and 2013 define symbols in std::tr1 that
|
||||||
# Yet MSVC 10's <utility> declares tuple_element<size_t, class>.
|
# conflict with our own definitions. Therefore using our own tuple does not
|
||||||
# That declaration conflicts with our own standard-conforming
|
# work on those compilers.
|
||||||
# tuple implementation. Therefore using our own tuple with
|
|
||||||
# MSVC 10 doesn't compile.
|
|
||||||
cxx_library(gtest_main_use_own_tuple "${cxx_use_own_tuple}"
|
cxx_library(gtest_main_use_own_tuple "${cxx_use_own_tuple}"
|
||||||
src/gtest-all.cc src/gtest_main.cc)
|
src/gtest-all.cc src/gtest_main.cc)
|
||||||
|
|
||||||
@ -189,11 +202,15 @@ if (gtest_build_tests)
|
|||||||
cxx_executable(gtest_break_on_failure_unittest_ test gtest)
|
cxx_executable(gtest_break_on_failure_unittest_ test gtest)
|
||||||
py_test(gtest_break_on_failure_unittest)
|
py_test(gtest_break_on_failure_unittest)
|
||||||
|
|
||||||
|
# Visual Studio .NET 2003 does not support STL with exceptions disabled.
|
||||||
|
if (NOT MSVC OR MSVC_VERSION GREATER 1310) # 1310 is Visual Studio .NET 2003
|
||||||
cxx_executable_with_flags(
|
cxx_executable_with_flags(
|
||||||
gtest_catch_exceptions_no_ex_test_
|
gtest_catch_exceptions_no_ex_test_
|
||||||
"${cxx_no_exception}"
|
"${cxx_no_exception}"
|
||||||
gtest_main_no_exception
|
gtest_main_no_exception
|
||||||
test/gtest_catch_exceptions_test_.cc)
|
test/gtest_catch_exceptions_test_.cc)
|
||||||
|
endif()
|
||||||
|
|
||||||
cxx_executable_with_flags(
|
cxx_executable_with_flags(
|
||||||
gtest_catch_exceptions_ex_test_
|
gtest_catch_exceptions_ex_test_
|
||||||
"${cxx_exception}"
|
"${cxx_exception}"
|
||||||
@ -222,11 +239,14 @@ if (gtest_build_tests)
|
|||||||
cxx_executable(gtest_shuffle_test_ test gtest)
|
cxx_executable(gtest_shuffle_test_ test gtest)
|
||||||
py_test(gtest_shuffle_test)
|
py_test(gtest_shuffle_test)
|
||||||
|
|
||||||
|
# MSVC 7.1 does not support STL with exceptions disabled.
|
||||||
|
if (NOT MSVC OR MSVC_VERSION GREATER 1310)
|
||||||
cxx_executable(gtest_throw_on_failure_test_ test gtest_no_exception)
|
cxx_executable(gtest_throw_on_failure_test_ test gtest_no_exception)
|
||||||
set_target_properties(gtest_throw_on_failure_test_
|
set_target_properties(gtest_throw_on_failure_test_
|
||||||
PROPERTIES
|
PROPERTIES
|
||||||
COMPILE_FLAGS "${cxx_no_exception}")
|
COMPILE_FLAGS "${cxx_no_exception}")
|
||||||
py_test(gtest_throw_on_failure_test)
|
py_test(gtest_throw_on_failure_test)
|
||||||
|
endif()
|
||||||
|
|
||||||
cxx_executable(gtest_uninitialized_test_ test gtest)
|
cxx_executable(gtest_uninitialized_test_ test gtest)
|
||||||
py_test(gtest_uninitialized_test)
|
py_test(gtest_uninitialized_test)
|
||||||
|
37
tests/gtest/CONTRIBUTORS
Normal file
37
tests/gtest/CONTRIBUTORS
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
# This file contains a list of people who've made non-trivial
|
||||||
|
# contribution to the Google C++ Testing Framework project. People
|
||||||
|
# who commit code to the project are encouraged to add their names
|
||||||
|
# here. Please keep the list sorted by first names.
|
||||||
|
|
||||||
|
Ajay Joshi <jaj@google.com>
|
||||||
|
Balázs Dán <balazs.dan@gmail.com>
|
||||||
|
Bharat Mediratta <bharat@menalto.com>
|
||||||
|
Chandler Carruth <chandlerc@google.com>
|
||||||
|
Chris Prince <cprince@google.com>
|
||||||
|
Chris Taylor <taylorc@google.com>
|
||||||
|
Dan Egnor <egnor@google.com>
|
||||||
|
Eric Roman <eroman@chromium.org>
|
||||||
|
Hady Zalek <hady.zalek@gmail.com>
|
||||||
|
Jeffrey Yasskin <jyasskin@google.com>
|
||||||
|
Jói Sigurðsson <joi@google.com>
|
||||||
|
Keir Mierle <mierle@gmail.com>
|
||||||
|
Keith Ray <keith.ray@gmail.com>
|
||||||
|
Kenton Varda <kenton@google.com>
|
||||||
|
Manuel Klimek <klimek@google.com>
|
||||||
|
Markus Heule <markus.heule@gmail.com>
|
||||||
|
Mika Raento <mikie@iki.fi>
|
||||||
|
Miklós Fazekas <mfazekas@szemafor.com>
|
||||||
|
Pasi Valminen <pasi.valminen@gmail.com>
|
||||||
|
Patrick Hanna <phanna@google.com>
|
||||||
|
Patrick Riley <pfr@google.com>
|
||||||
|
Peter Kaminski <piotrk@google.com>
|
||||||
|
Preston Jackson <preston.a.jackson@gmail.com>
|
||||||
|
Rainer Klaffenboeck <rainer.klaffenboeck@dynatrace.com>
|
||||||
|
Russ Cox <rsc@google.com>
|
||||||
|
Russ Rufer <russ@pentad.com>
|
||||||
|
Sean Mcafee <eefacm@gmail.com>
|
||||||
|
Sigurður Ásgeirsson <siggi@google.com>
|
||||||
|
Tracy Bialik <tracy@pentad.com>
|
||||||
|
Vadim Berman <vadimb@google.com>
|
||||||
|
Vlad Losev <vladl@google.com>
|
||||||
|
Zhanyong Wan <wan@google.com>
|
28
tests/gtest/LICENSE
Normal file
28
tests/gtest/LICENSE
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
Copyright 2008, Google Inc.
|
||||||
|
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 Google Inc. 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.
|
435
tests/gtest/README
Normal file
435
tests/gtest/README
Normal file
@ -0,0 +1,435 @@
|
|||||||
|
Google C++ Testing Framework
|
||||||
|
============================
|
||||||
|
|
||||||
|
http://code.google.com/p/googletest/
|
||||||
|
|
||||||
|
Overview
|
||||||
|
--------
|
||||||
|
|
||||||
|
Google's framework for writing C++ tests on a variety of platforms
|
||||||
|
(Linux, Mac OS X, Windows, Windows CE, Symbian, etc). Based on the
|
||||||
|
xUnit architecture. Supports automatic test discovery, a rich set of
|
||||||
|
assertions, user-defined assertions, death tests, fatal and non-fatal
|
||||||
|
failures, various options for running the tests, and XML test report
|
||||||
|
generation.
|
||||||
|
|
||||||
|
Please see the project page above for more information as well as the
|
||||||
|
mailing list for questions, discussions, and development. There is
|
||||||
|
also an IRC channel on OFTC (irc.oftc.net) #gtest available. Please
|
||||||
|
join us!
|
||||||
|
|
||||||
|
Requirements for End Users
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
Google Test is designed to have fairly minimal requirements to build
|
||||||
|
and use with your projects, but there are some. Currently, we support
|
||||||
|
Linux, Windows, Mac OS X, and Cygwin. We will also make our best
|
||||||
|
effort to support other platforms (e.g. Solaris, AIX, and z/OS).
|
||||||
|
However, since core members of the Google Test project have no access
|
||||||
|
to these platforms, Google Test may have outstanding issues there. If
|
||||||
|
you notice any problems on your platform, please notify
|
||||||
|
googletestframework@googlegroups.com. Patches for fixing them are
|
||||||
|
even more welcome!
|
||||||
|
|
||||||
|
### Linux Requirements ###
|
||||||
|
|
||||||
|
These are the base requirements to build and use Google Test from a source
|
||||||
|
package (as described below):
|
||||||
|
* GNU-compatible Make or gmake
|
||||||
|
* POSIX-standard shell
|
||||||
|
* POSIX(-2) Regular Expressions (regex.h)
|
||||||
|
* A C++98-standard-compliant compiler
|
||||||
|
|
||||||
|
### Windows Requirements ###
|
||||||
|
|
||||||
|
* Microsoft Visual C++ 7.1 or newer
|
||||||
|
|
||||||
|
### Cygwin Requirements ###
|
||||||
|
|
||||||
|
* Cygwin 1.5.25-14 or newer
|
||||||
|
|
||||||
|
### Mac OS X Requirements ###
|
||||||
|
|
||||||
|
* Mac OS X 10.4 Tiger or newer
|
||||||
|
* Developer Tools Installed
|
||||||
|
|
||||||
|
Also, you'll need CMake 2.6.4 or higher if you want to build the
|
||||||
|
samples using the provided CMake script, regardless of the platform.
|
||||||
|
|
||||||
|
Requirements for Contributors
|
||||||
|
-----------------------------
|
||||||
|
|
||||||
|
We welcome patches. If you plan to contribute a patch, you need to
|
||||||
|
build Google Test and its own tests from an SVN checkout (described
|
||||||
|
below), which has further requirements:
|
||||||
|
|
||||||
|
* Python version 2.3 or newer (for running some of the tests and
|
||||||
|
re-generating certain source files from templates)
|
||||||
|
* CMake 2.6.4 or newer
|
||||||
|
|
||||||
|
Getting the Source
|
||||||
|
------------------
|
||||||
|
|
||||||
|
There are two primary ways of getting Google Test's source code: you
|
||||||
|
can download a stable source release in your preferred archive format,
|
||||||
|
or directly check out the source from our Subversion (SVN) repository.
|
||||||
|
The SVN checkout requires a few extra steps and some extra software
|
||||||
|
packages on your system, but lets you track the latest development and
|
||||||
|
make patches much more easily, so we highly encourage it.
|
||||||
|
|
||||||
|
### Source Package ###
|
||||||
|
|
||||||
|
Google Test is released in versioned source packages which can be
|
||||||
|
downloaded from the download page [1]. Several different archive
|
||||||
|
formats are provided, but the only difference is the tools used to
|
||||||
|
manipulate them, and the size of the resulting file. Download
|
||||||
|
whichever you are most comfortable with.
|
||||||
|
|
||||||
|
[1] http://code.google.com/p/googletest/downloads/list
|
||||||
|
|
||||||
|
Once the package is downloaded, expand it using whichever tools you
|
||||||
|
prefer for that type. This will result in a new directory with the
|
||||||
|
name "gtest-X.Y.Z" which contains all of the source code. Here are
|
||||||
|
some examples on Linux:
|
||||||
|
|
||||||
|
tar -xvzf gtest-X.Y.Z.tar.gz
|
||||||
|
tar -xvjf gtest-X.Y.Z.tar.bz2
|
||||||
|
unzip gtest-X.Y.Z.zip
|
||||||
|
|
||||||
|
### SVN Checkout ###
|
||||||
|
|
||||||
|
To check out the main branch (also known as the "trunk") of Google
|
||||||
|
Test, run the following Subversion command:
|
||||||
|
|
||||||
|
svn checkout http://googletest.googlecode.com/svn/trunk/ gtest-svn
|
||||||
|
|
||||||
|
Setting up the Build
|
||||||
|
--------------------
|
||||||
|
|
||||||
|
To build Google Test and your tests that use it, you need to tell your
|
||||||
|
build system where to find its headers and source files. The exact
|
||||||
|
way to do it depends on which build system you use, and is usually
|
||||||
|
straightforward.
|
||||||
|
|
||||||
|
### Generic Build Instructions ###
|
||||||
|
|
||||||
|
Suppose you put Google Test in directory ${GTEST_DIR}. To build it,
|
||||||
|
create a library build target (or a project as called by Visual Studio
|
||||||
|
and Xcode) to compile
|
||||||
|
|
||||||
|
${GTEST_DIR}/src/gtest-all.cc
|
||||||
|
|
||||||
|
with ${GTEST_DIR}/include in the system header search path and ${GTEST_DIR}
|
||||||
|
in the normal header search path. Assuming a Linux-like system and gcc,
|
||||||
|
something like the following will do:
|
||||||
|
|
||||||
|
g++ -isystem ${GTEST_DIR}/include -I${GTEST_DIR} \
|
||||||
|
-pthread -c ${GTEST_DIR}/src/gtest-all.cc
|
||||||
|
ar -rv libgtest.a gtest-all.o
|
||||||
|
|
||||||
|
(We need -pthread as Google Test uses threads.)
|
||||||
|
|
||||||
|
Next, you should compile your test source file with
|
||||||
|
${GTEST_DIR}/include in the system header search path, and link it
|
||||||
|
with gtest and any other necessary libraries:
|
||||||
|
|
||||||
|
g++ -isystem ${GTEST_DIR}/include -pthread path/to/your_test.cc libgtest.a \
|
||||||
|
-o your_test
|
||||||
|
|
||||||
|
As an example, the make/ directory contains a Makefile that you can
|
||||||
|
use to build Google Test on systems where GNU make is available
|
||||||
|
(e.g. Linux, Mac OS X, and Cygwin). It doesn't try to build Google
|
||||||
|
Test's own tests. Instead, it just builds the Google Test library and
|
||||||
|
a sample test. You can use it as a starting point for your own build
|
||||||
|
script.
|
||||||
|
|
||||||
|
If the default settings are correct for your environment, the
|
||||||
|
following commands should succeed:
|
||||||
|
|
||||||
|
cd ${GTEST_DIR}/make
|
||||||
|
make
|
||||||
|
./sample1_unittest
|
||||||
|
|
||||||
|
If you see errors, try to tweak the contents of make/Makefile to make
|
||||||
|
them go away. There are instructions in make/Makefile on how to do
|
||||||
|
it.
|
||||||
|
|
||||||
|
### Using CMake ###
|
||||||
|
|
||||||
|
Google Test comes with a CMake build script (CMakeLists.txt) that can
|
||||||
|
be used on a wide range of platforms ("C" stands for cross-platform.).
|
||||||
|
If you don't have CMake installed already, you can download it for
|
||||||
|
free from http://www.cmake.org/.
|
||||||
|
|
||||||
|
CMake works by generating native makefiles or build projects that can
|
||||||
|
be used in the compiler environment of your choice. The typical
|
||||||
|
workflow starts with:
|
||||||
|
|
||||||
|
mkdir mybuild # Create a directory to hold the build output.
|
||||||
|
cd mybuild
|
||||||
|
cmake ${GTEST_DIR} # Generate native build scripts.
|
||||||
|
|
||||||
|
If you want to build Google Test's samples, you should replace the
|
||||||
|
last command with
|
||||||
|
|
||||||
|
cmake -Dgtest_build_samples=ON ${GTEST_DIR}
|
||||||
|
|
||||||
|
If you are on a *nix system, you should now see a Makefile in the
|
||||||
|
current directory. Just type 'make' to build gtest.
|
||||||
|
|
||||||
|
If you use Windows and have Visual Studio installed, a gtest.sln file
|
||||||
|
and several .vcproj files will be created. You can then build them
|
||||||
|
using Visual Studio.
|
||||||
|
|
||||||
|
On Mac OS X with Xcode installed, a .xcodeproj file will be generated.
|
||||||
|
|
||||||
|
### Legacy Build Scripts ###
|
||||||
|
|
||||||
|
Before settling on CMake, we have been providing hand-maintained build
|
||||||
|
projects/scripts for Visual Studio, Xcode, and Autotools. While we
|
||||||
|
continue to provide them for convenience, they are not actively
|
||||||
|
maintained any more. We highly recommend that you follow the
|
||||||
|
instructions in the previous two sections to integrate Google Test
|
||||||
|
with your existing build system.
|
||||||
|
|
||||||
|
If you still need to use the legacy build scripts, here's how:
|
||||||
|
|
||||||
|
The msvc\ folder contains two solutions with Visual C++ projects.
|
||||||
|
Open the gtest.sln or gtest-md.sln file using Visual Studio, and you
|
||||||
|
are ready to build Google Test the same way you build any Visual
|
||||||
|
Studio project. Files that have names ending with -md use DLL
|
||||||
|
versions of Microsoft runtime libraries (the /MD or the /MDd compiler
|
||||||
|
option). Files without that suffix use static versions of the runtime
|
||||||
|
libraries (the /MT or the /MTd option). Please note that one must use
|
||||||
|
the same option to compile both gtest and the test code. If you use
|
||||||
|
Visual Studio 2005 or above, we recommend the -md version as /MD is
|
||||||
|
the default for new projects in these versions of Visual Studio.
|
||||||
|
|
||||||
|
On Mac OS X, open the gtest.xcodeproj in the xcode/ folder using
|
||||||
|
Xcode. Build the "gtest" target. The universal binary framework will
|
||||||
|
end up in your selected build directory (selected in the Xcode
|
||||||
|
"Preferences..." -> "Building" pane and defaults to xcode/build).
|
||||||
|
Alternatively, at the command line, enter:
|
||||||
|
|
||||||
|
xcodebuild
|
||||||
|
|
||||||
|
This will build the "Release" configuration of gtest.framework in your
|
||||||
|
default build location. See the "xcodebuild" man page for more
|
||||||
|
information about building different configurations and building in
|
||||||
|
different locations.
|
||||||
|
|
||||||
|
If you wish to use the Google Test Xcode project with Xcode 4.x and
|
||||||
|
above, you need to either:
|
||||||
|
* update the SDK configuration options in xcode/Config/General.xconfig.
|
||||||
|
Comment options SDKROOT, MACOS_DEPLOYMENT_TARGET, and GCC_VERSION. If
|
||||||
|
you choose this route you lose the ability to target earlier versions
|
||||||
|
of MacOS X.
|
||||||
|
* Install an SDK for an earlier version. This doesn't appear to be
|
||||||
|
supported by Apple, but has been reported to work
|
||||||
|
(http://stackoverflow.com/questions/5378518).
|
||||||
|
|
||||||
|
Tweaking Google Test
|
||||||
|
--------------------
|
||||||
|
|
||||||
|
Google Test can be used in diverse environments. The default
|
||||||
|
configuration may not work (or may not work well) out of the box in
|
||||||
|
some environments. However, you can easily tweak Google Test by
|
||||||
|
defining control macros on the compiler command line. Generally,
|
||||||
|
these macros are named like GTEST_XYZ and you define them to either 1
|
||||||
|
or 0 to enable or disable a certain feature.
|
||||||
|
|
||||||
|
We list the most frequently used macros below. For a complete list,
|
||||||
|
see file include/gtest/internal/gtest-port.h.
|
||||||
|
|
||||||
|
### Choosing a TR1 Tuple Library ###
|
||||||
|
|
||||||
|
Some Google Test features require the C++ Technical Report 1 (TR1)
|
||||||
|
tuple library, which is not yet available with all compilers. The
|
||||||
|
good news is that Google Test implements a subset of TR1 tuple that's
|
||||||
|
enough for its own need, and will automatically use this when the
|
||||||
|
compiler doesn't provide TR1 tuple.
|
||||||
|
|
||||||
|
Usually you don't need to care about which tuple library Google Test
|
||||||
|
uses. However, if your project already uses TR1 tuple, you need to
|
||||||
|
tell Google Test to use the same TR1 tuple library the rest of your
|
||||||
|
project uses, or the two tuple implementations will clash. To do
|
||||||
|
that, add
|
||||||
|
|
||||||
|
-DGTEST_USE_OWN_TR1_TUPLE=0
|
||||||
|
|
||||||
|
to the compiler flags while compiling Google Test and your tests. If
|
||||||
|
you want to force Google Test to use its own tuple library, just add
|
||||||
|
|
||||||
|
-DGTEST_USE_OWN_TR1_TUPLE=1
|
||||||
|
|
||||||
|
to the compiler flags instead.
|
||||||
|
|
||||||
|
If you don't want Google Test to use tuple at all, add
|
||||||
|
|
||||||
|
-DGTEST_HAS_TR1_TUPLE=0
|
||||||
|
|
||||||
|
and all features using tuple will be disabled.
|
||||||
|
|
||||||
|
### Multi-threaded Tests ###
|
||||||
|
|
||||||
|
Google Test is thread-safe where the pthread library is available.
|
||||||
|
After #include "gtest/gtest.h", you can check the GTEST_IS_THREADSAFE
|
||||||
|
macro to see whether this is the case (yes if the macro is #defined to
|
||||||
|
1, no if it's undefined.).
|
||||||
|
|
||||||
|
If Google Test doesn't correctly detect whether pthread is available
|
||||||
|
in your environment, you can force it with
|
||||||
|
|
||||||
|
-DGTEST_HAS_PTHREAD=1
|
||||||
|
|
||||||
|
or
|
||||||
|
|
||||||
|
-DGTEST_HAS_PTHREAD=0
|
||||||
|
|
||||||
|
When Google Test uses pthread, you may need to add flags to your
|
||||||
|
compiler and/or linker to select the pthread library, or you'll get
|
||||||
|
link errors. If you use the CMake script or the deprecated Autotools
|
||||||
|
script, this is taken care of for you. If you use your own build
|
||||||
|
script, you'll need to read your compiler and linker's manual to
|
||||||
|
figure out what flags to add.
|
||||||
|
|
||||||
|
### As a Shared Library (DLL) ###
|
||||||
|
|
||||||
|
Google Test is compact, so most users can build and link it as a
|
||||||
|
static library for the simplicity. You can choose to use Google Test
|
||||||
|
as a shared library (known as a DLL on Windows) if you prefer.
|
||||||
|
|
||||||
|
To compile *gtest* as a shared library, add
|
||||||
|
|
||||||
|
-DGTEST_CREATE_SHARED_LIBRARY=1
|
||||||
|
|
||||||
|
to the compiler flags. You'll also need to tell the linker to produce
|
||||||
|
a shared library instead - consult your linker's manual for how to do
|
||||||
|
it.
|
||||||
|
|
||||||
|
To compile your *tests* that use the gtest shared library, add
|
||||||
|
|
||||||
|
-DGTEST_LINKED_AS_SHARED_LIBRARY=1
|
||||||
|
|
||||||
|
to the compiler flags.
|
||||||
|
|
||||||
|
Note: while the above steps aren't technically necessary today when
|
||||||
|
using some compilers (e.g. GCC), they may become necessary in the
|
||||||
|
future, if we decide to improve the speed of loading the library (see
|
||||||
|
http://gcc.gnu.org/wiki/Visibility for details). Therefore you are
|
||||||
|
recommended to always add the above flags when using Google Test as a
|
||||||
|
shared library. Otherwise a future release of Google Test may break
|
||||||
|
your build script.
|
||||||
|
|
||||||
|
### Avoiding Macro Name Clashes ###
|
||||||
|
|
||||||
|
In C++, macros don't obey namespaces. Therefore two libraries that
|
||||||
|
both define a macro of the same name will clash if you #include both
|
||||||
|
definitions. In case a Google Test macro clashes with another
|
||||||
|
library, you can force Google Test to rename its macro to avoid the
|
||||||
|
conflict.
|
||||||
|
|
||||||
|
Specifically, if both Google Test and some other code define macro
|
||||||
|
FOO, you can add
|
||||||
|
|
||||||
|
-DGTEST_DONT_DEFINE_FOO=1
|
||||||
|
|
||||||
|
to the compiler flags to tell Google Test to change the macro's name
|
||||||
|
from FOO to GTEST_FOO. Currently FOO can be FAIL, SUCCEED, or TEST.
|
||||||
|
For example, with -DGTEST_DONT_DEFINE_TEST=1, you'll need to write
|
||||||
|
|
||||||
|
GTEST_TEST(SomeTest, DoesThis) { ... }
|
||||||
|
|
||||||
|
instead of
|
||||||
|
|
||||||
|
TEST(SomeTest, DoesThis) { ... }
|
||||||
|
|
||||||
|
in order to define a test.
|
||||||
|
|
||||||
|
Upgrating from an Earlier Version
|
||||||
|
---------------------------------
|
||||||
|
|
||||||
|
We strive to keep Google Test releases backward compatible.
|
||||||
|
Sometimes, though, we have to make some breaking changes for the
|
||||||
|
users' long-term benefits. This section describes what you'll need to
|
||||||
|
do if you are upgrading from an earlier version of Google Test.
|
||||||
|
|
||||||
|
### Upgrading from 1.3.0 or Earlier ###
|
||||||
|
|
||||||
|
You may need to explicitly enable or disable Google Test's own TR1
|
||||||
|
tuple library. See the instructions in section "Choosing a TR1 Tuple
|
||||||
|
Library".
|
||||||
|
|
||||||
|
### Upgrading from 1.4.0 or Earlier ###
|
||||||
|
|
||||||
|
The Autotools build script (configure + make) is no longer officially
|
||||||
|
supportted. You are encouraged to migrate to your own build system or
|
||||||
|
use CMake. If you still need to use Autotools, you can find
|
||||||
|
instructions in the README file from Google Test 1.4.0.
|
||||||
|
|
||||||
|
On platforms where the pthread library is available, Google Test uses
|
||||||
|
it in order to be thread-safe. See the "Multi-threaded Tests" section
|
||||||
|
for what this means to your build script.
|
||||||
|
|
||||||
|
If you use Microsoft Visual C++ 7.1 with exceptions disabled, Google
|
||||||
|
Test will no longer compile. This should affect very few people, as a
|
||||||
|
large portion of STL (including <string>) doesn't compile in this mode
|
||||||
|
anyway. We decided to stop supporting it in order to greatly simplify
|
||||||
|
Google Test's implementation.
|
||||||
|
|
||||||
|
Developing Google Test
|
||||||
|
----------------------
|
||||||
|
|
||||||
|
This section discusses how to make your own changes to Google Test.
|
||||||
|
|
||||||
|
### Testing Google Test Itself ###
|
||||||
|
|
||||||
|
To make sure your changes work as intended and don't break existing
|
||||||
|
functionality, you'll want to compile and run Google Test's own tests.
|
||||||
|
For that you can use CMake:
|
||||||
|
|
||||||
|
mkdir mybuild
|
||||||
|
cd mybuild
|
||||||
|
cmake -Dgtest_build_tests=ON ${GTEST_DIR}
|
||||||
|
|
||||||
|
Make sure you have Python installed, as some of Google Test's tests
|
||||||
|
are written in Python. If the cmake command complains about not being
|
||||||
|
able to find Python ("Could NOT find PythonInterp (missing:
|
||||||
|
PYTHON_EXECUTABLE)"), try telling it explicitly where your Python
|
||||||
|
executable can be found:
|
||||||
|
|
||||||
|
cmake -DPYTHON_EXECUTABLE=path/to/python -Dgtest_build_tests=ON ${GTEST_DIR}
|
||||||
|
|
||||||
|
Next, you can build Google Test and all of its own tests. On *nix,
|
||||||
|
this is usually done by 'make'. To run the tests, do
|
||||||
|
|
||||||
|
make test
|
||||||
|
|
||||||
|
All tests should pass.
|
||||||
|
|
||||||
|
### Regenerating Source Files ###
|
||||||
|
|
||||||
|
Some of Google Test's source files are generated from templates (not
|
||||||
|
in the C++ sense) using a script. A template file is named FOO.pump,
|
||||||
|
where FOO is the name of the file it will generate. For example, the
|
||||||
|
file include/gtest/internal/gtest-type-util.h.pump is used to generate
|
||||||
|
gtest-type-util.h in the same directory.
|
||||||
|
|
||||||
|
Normally you don't need to worry about regenerating the source files,
|
||||||
|
unless you need to modify them. In that case, you should modify the
|
||||||
|
corresponding .pump files instead and run the pump.py Python script to
|
||||||
|
regenerate them. You can find pump.py in the scripts/ directory.
|
||||||
|
Read the Pump manual [2] for how to use it.
|
||||||
|
|
||||||
|
[2] http://code.google.com/p/googletest/wiki/PumpManual
|
||||||
|
|
||||||
|
### Contributing a Patch ###
|
||||||
|
|
||||||
|
We welcome patches. Please read the Google Test developer's guide [3]
|
||||||
|
for how you can contribute. In particular, make sure you have signed
|
||||||
|
the Contributor License Agreement, or we won't be able to accept the
|
||||||
|
patch.
|
||||||
|
|
||||||
|
[3] http://code.google.com/p/googletest/wiki/GoogleTestDevGuide
|
||||||
|
|
||||||
|
Happy testing!
|
@ -37,7 +37,7 @@ macro(fix_default_compiler_settings_)
|
|||||||
|
|
||||||
# We prefer more strict warning checking for building Google Test.
|
# We prefer more strict warning checking for building Google Test.
|
||||||
# Replaces /W3 with /W4 in defaults.
|
# Replaces /W3 with /W4 in defaults.
|
||||||
string(REPLACE "/W3" "-W4" ${flag_var} "${${flag_var}}")
|
string(REPLACE "/W3" "/W4" ${flag_var} "${${flag_var}}")
|
||||||
endforeach()
|
endforeach()
|
||||||
endif()
|
endif()
|
||||||
endmacro()
|
endmacro()
|
||||||
@ -55,7 +55,32 @@ macro(config_compiler_and_linker)
|
|||||||
if (MSVC)
|
if (MSVC)
|
||||||
# Newlines inside flags variables break CMake's NMake generator.
|
# Newlines inside flags variables break CMake's NMake generator.
|
||||||
# TODO(vladl@google.com): Add -RTCs and -RTCu to debug builds.
|
# TODO(vladl@google.com): Add -RTCs and -RTCu to debug builds.
|
||||||
set(cxx_base_flags "-GS -W4 -WX -wd4127 -wd4251 -wd4275 -nologo -J -Zi")
|
set(cxx_base_flags "-GS -W4 -WX -wd4251 -wd4275 -nologo -J -Zi")
|
||||||
|
if (MSVC_VERSION LESS 1400) # 1400 is Visual Studio 2005
|
||||||
|
# Suppress spurious warnings MSVC 7.1 sometimes issues.
|
||||||
|
# Forcing value to bool.
|
||||||
|
set(cxx_base_flags "${cxx_base_flags} -wd4800")
|
||||||
|
# Copy constructor and assignment operator could not be generated.
|
||||||
|
set(cxx_base_flags "${cxx_base_flags} -wd4511 -wd4512")
|
||||||
|
# Compatibility warnings not applicable to Google Test.
|
||||||
|
# Resolved overload was found by argument-dependent lookup.
|
||||||
|
set(cxx_base_flags "${cxx_base_flags} -wd4675")
|
||||||
|
endif()
|
||||||
|
if (MSVC_VERSION LESS 1500) # 1500 is Visual Studio 2008
|
||||||
|
# Conditional expression is constant.
|
||||||
|
# When compiling with /W4, we get several instances of C4127
|
||||||
|
# (Conditional expression is constant). In our code, we disable that
|
||||||
|
# warning on a case-by-case basis. However, on Visual Studio 2005,
|
||||||
|
# the warning fires on std::list. Therefore on that compiler and earlier,
|
||||||
|
# we disable the warning project-wide.
|
||||||
|
set(cxx_base_flags "${cxx_base_flags} -wd4127")
|
||||||
|
endif()
|
||||||
|
if (NOT (MSVC_VERSION LESS 1700)) # 1700 is Visual Studio 2012.
|
||||||
|
# Suppress "unreachable code" warning on VS 2012 and later.
|
||||||
|
# http://stackoverflow.com/questions/3232669 explains the issue.
|
||||||
|
set(cxx_base_flags "${cxx_base_flags} -wd4702")
|
||||||
|
endif()
|
||||||
|
|
||||||
set(cxx_base_flags "${cxx_base_flags} -D_UNICODE -DUNICODE -DWIN32 -D_WIN32")
|
set(cxx_base_flags "${cxx_base_flags} -D_UNICODE -DUNICODE -DWIN32 -D_WIN32")
|
||||||
set(cxx_base_flags "${cxx_base_flags} -DSTRICT -DWIN32_LEAN_AND_MEAN")
|
set(cxx_base_flags "${cxx_base_flags} -DSTRICT -DWIN32_LEAN_AND_MEAN")
|
||||||
set(cxx_exception_flags "-EHsc -D_HAS_EXCEPTIONS=1")
|
set(cxx_exception_flags "-EHsc -D_HAS_EXCEPTIONS=1")
|
||||||
@ -69,7 +94,8 @@ macro(config_compiler_and_linker)
|
|||||||
# whether RTTI is enabled. Therefore we define GTEST_HAS_RTTI
|
# whether RTTI is enabled. Therefore we define GTEST_HAS_RTTI
|
||||||
# explicitly.
|
# explicitly.
|
||||||
set(cxx_no_rtti_flags "-fno-rtti -DGTEST_HAS_RTTI=0")
|
set(cxx_no_rtti_flags "-fno-rtti -DGTEST_HAS_RTTI=0")
|
||||||
set(cxx_strict_flags "-Wextra")
|
set(cxx_strict_flags
|
||||||
|
"-Wextra -Wno-unused-parameter -Wno-missing-field-initializers")
|
||||||
elseif (CMAKE_CXX_COMPILER_ID STREQUAL "SunPro")
|
elseif (CMAKE_CXX_COMPILER_ID STREQUAL "SunPro")
|
||||||
set(cxx_exception_flags "-features=except")
|
set(cxx_exception_flags "-features=except")
|
||||||
# Sun Pro doesn't provide macros to indicate whether exceptions and
|
# Sun Pro doesn't provide macros to indicate whether exceptions and
|
||||||
|
@ -51,6 +51,17 @@ GTEST_DECLARE_string_(death_test_style);
|
|||||||
|
|
||||||
#if GTEST_HAS_DEATH_TEST
|
#if GTEST_HAS_DEATH_TEST
|
||||||
|
|
||||||
|
namespace internal {
|
||||||
|
|
||||||
|
// Returns a Boolean value indicating whether the caller is currently
|
||||||
|
// executing in the context of the death test child process. Tools such as
|
||||||
|
// Valgrind heap checkers may need this to modify their behavior in death
|
||||||
|
// tests. IMPORTANT: This is an internal utility. Using it may break the
|
||||||
|
// implementation of death tests. User code MUST NOT use it.
|
||||||
|
GTEST_API_ bool InDeathTestChild();
|
||||||
|
|
||||||
|
} // namespace internal
|
||||||
|
|
||||||
// The following macros are useful for writing death tests.
|
// The following macros are useful for writing death tests.
|
||||||
|
|
||||||
// Here's what happens when an ASSERT_DEATH* or EXPECT_DEATH* is
|
// Here's what happens when an ASSERT_DEATH* or EXPECT_DEATH* is
|
||||||
@ -75,7 +86,7 @@ GTEST_DECLARE_string_(death_test_style);
|
|||||||
// for (int i = 0; i < 5; i++) {
|
// for (int i = 0; i < 5; i++) {
|
||||||
// EXPECT_DEATH(server.ProcessRequest(i),
|
// EXPECT_DEATH(server.ProcessRequest(i),
|
||||||
// "Invalid request .* in ProcessRequest()")
|
// "Invalid request .* in ProcessRequest()")
|
||||||
// << "Failed to die on request " << i);
|
// << "Failed to die on request " << i;
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
// ASSERT_EXIT(server.ExitNow(), ::testing::ExitedWithCode(0), "Exiting");
|
// ASSERT_EXIT(server.ExitNow(), ::testing::ExitedWithCode(0), "Exiting");
|
||||||
@ -245,10 +256,10 @@ class GTEST_API_ KilledBySignal {
|
|||||||
# ifdef NDEBUG
|
# ifdef NDEBUG
|
||||||
|
|
||||||
# define EXPECT_DEBUG_DEATH(statement, regex) \
|
# define EXPECT_DEBUG_DEATH(statement, regex) \
|
||||||
do { statement; } while (::testing::internal::AlwaysFalse())
|
GTEST_EXECUTE_STATEMENT_(statement, regex)
|
||||||
|
|
||||||
# define ASSERT_DEBUG_DEATH(statement, regex) \
|
# define ASSERT_DEBUG_DEATH(statement, regex) \
|
||||||
do { statement; } while (::testing::internal::AlwaysFalse())
|
GTEST_EXECUTE_STATEMENT_(statement, regex)
|
||||||
|
|
||||||
# else
|
# else
|
||||||
|
|
||||||
|
@ -48,8 +48,11 @@
|
|||||||
|
|
||||||
#include <limits>
|
#include <limits>
|
||||||
|
|
||||||
#include "gtest/internal/gtest-string.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
#include "gtest/internal/gtest-internal.h"
|
|
||||||
|
// Ensures that there is at least one operator<< in the global namespace.
|
||||||
|
// See Message& operator<<(...) below for why.
|
||||||
|
void operator<<(const testing::internal::Secret&, int);
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
|
|
||||||
@ -87,15 +90,7 @@ class GTEST_API_ Message {
|
|||||||
|
|
||||||
public:
|
public:
|
||||||
// Constructs an empty Message.
|
// Constructs an empty Message.
|
||||||
// We allocate the stringstream separately because otherwise each use of
|
Message();
|
||||||
// ASSERT/EXPECT in a procedure adds over 200 bytes to the procedure's
|
|
||||||
// stack frame leading to huge stack frames in some cases; gcc does not reuse
|
|
||||||
// the stack space.
|
|
||||||
Message() : ss_(new ::std::stringstream) {
|
|
||||||
// By default, we want there to be enough precision when printing
|
|
||||||
// a double to a Message.
|
|
||||||
*ss_ << std::setprecision(std::numeric_limits<double>::digits10 + 2);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Copy constructor.
|
// Copy constructor.
|
||||||
Message(const Message& msg) : ss_(new ::std::stringstream) { // NOLINT
|
Message(const Message& msg) : ss_(new ::std::stringstream) { // NOLINT
|
||||||
@ -118,7 +113,22 @@ class GTEST_API_ Message {
|
|||||||
// Streams a non-pointer value to this object.
|
// Streams a non-pointer value to this object.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
inline Message& operator <<(const T& val) {
|
inline Message& operator <<(const T& val) {
|
||||||
::GTestStreamToHelper(ss_.get(), val);
|
// Some libraries overload << for STL containers. These
|
||||||
|
// overloads are defined in the global namespace instead of ::std.
|
||||||
|
//
|
||||||
|
// C++'s symbol lookup rule (i.e. Koenig lookup) says that these
|
||||||
|
// overloads are visible in either the std namespace or the global
|
||||||
|
// namespace, but not other namespaces, including the testing
|
||||||
|
// namespace which Google Test's Message class is in.
|
||||||
|
//
|
||||||
|
// To allow STL containers (and other types that has a << operator
|
||||||
|
// defined in the global namespace) to be used in Google Test
|
||||||
|
// assertions, testing::Message must access the custom << operator
|
||||||
|
// from the global namespace. With this using declaration,
|
||||||
|
// overloads of << defined in the global namespace and those
|
||||||
|
// visible via Koenig lookup are both exposed in this function.
|
||||||
|
using ::operator <<;
|
||||||
|
*ss_ << val;
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -140,7 +150,7 @@ class GTEST_API_ Message {
|
|||||||
if (pointer == NULL) {
|
if (pointer == NULL) {
|
||||||
*ss_ << "(null)";
|
*ss_ << "(null)";
|
||||||
} else {
|
} else {
|
||||||
::GTestStreamToHelper(ss_.get(), pointer);
|
*ss_ << pointer;
|
||||||
}
|
}
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
@ -164,12 +174,8 @@ class GTEST_API_ Message {
|
|||||||
|
|
||||||
// These two overloads allow streaming a wide C string to a Message
|
// These two overloads allow streaming a wide C string to a Message
|
||||||
// using the UTF-8 encoding.
|
// using the UTF-8 encoding.
|
||||||
Message& operator <<(const wchar_t* wide_c_str) {
|
Message& operator <<(const wchar_t* wide_c_str);
|
||||||
return *this << internal::String::ShowWideCString(wide_c_str);
|
Message& operator <<(wchar_t* wide_c_str);
|
||||||
}
|
|
||||||
Message& operator <<(wchar_t* wide_c_str) {
|
|
||||||
return *this << internal::String::ShowWideCString(wide_c_str);
|
|
||||||
}
|
|
||||||
|
|
||||||
#if GTEST_HAS_STD_WSTRING
|
#if GTEST_HAS_STD_WSTRING
|
||||||
// Converts the given wide string to a narrow string using the UTF-8
|
// Converts the given wide string to a narrow string using the UTF-8
|
||||||
@ -183,13 +189,11 @@ class GTEST_API_ Message {
|
|||||||
Message& operator <<(const ::wstring& wstr);
|
Message& operator <<(const ::wstring& wstr);
|
||||||
#endif // GTEST_HAS_GLOBAL_WSTRING
|
#endif // GTEST_HAS_GLOBAL_WSTRING
|
||||||
|
|
||||||
// Gets the text streamed to this object so far as a String.
|
// Gets the text streamed to this object so far as an std::string.
|
||||||
// Each '\0' character in the buffer is replaced with "\\0".
|
// Each '\0' character in the buffer is replaced with "\\0".
|
||||||
//
|
//
|
||||||
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
||||||
internal::String GetString() const {
|
std::string GetString() const;
|
||||||
return internal::StringStreamToString(ss_.get());
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
||||||
@ -199,16 +203,20 @@ class GTEST_API_ Message {
|
|||||||
// decide between class template specializations for T and T*, so a
|
// decide between class template specializations for T and T*, so a
|
||||||
// tr1::type_traits-like is_pointer works, and we can overload on that.
|
// tr1::type_traits-like is_pointer works, and we can overload on that.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
inline void StreamHelper(internal::true_type /*dummy*/, T* pointer) {
|
inline void StreamHelper(internal::true_type /*is_pointer*/, T* pointer) {
|
||||||
if (pointer == NULL) {
|
if (pointer == NULL) {
|
||||||
*ss_ << "(null)";
|
*ss_ << "(null)";
|
||||||
} else {
|
} else {
|
||||||
::GTestStreamToHelper(ss_.get(), pointer);
|
*ss_ << pointer;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
template <typename T>
|
template <typename T>
|
||||||
inline void StreamHelper(internal::false_type /*dummy*/, const T& value) {
|
inline void StreamHelper(internal::false_type /*is_pointer*/,
|
||||||
::GTestStreamToHelper(ss_.get(), value);
|
const T& value) {
|
||||||
|
// See the comments in Message& operator <<(const T&) above for why
|
||||||
|
// we need this using statement.
|
||||||
|
using ::operator <<;
|
||||||
|
*ss_ << value;
|
||||||
}
|
}
|
||||||
#endif // GTEST_OS_SYMBIAN
|
#endif // GTEST_OS_SYMBIAN
|
||||||
|
|
||||||
@ -225,6 +233,18 @@ inline std::ostream& operator <<(std::ostream& os, const Message& sb) {
|
|||||||
return os << sb.GetString();
|
return os << sb.GetString();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
namespace internal {
|
||||||
|
|
||||||
|
// Converts a streamable value to an std::string. A NULL pointer is
|
||||||
|
// converted to "(null)". When the input value is a ::string,
|
||||||
|
// ::std::string, ::wstring, or ::std::wstring object, each NUL
|
||||||
|
// character in it is replaced with "\\0".
|
||||||
|
template <typename T>
|
||||||
|
std::string StreamableToString(const T& streamable) {
|
||||||
|
return (Message() << streamable).GetString();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace internal
|
||||||
} // namespace testing
|
} // namespace testing
|
||||||
|
|
||||||
#endif // GTEST_INCLUDE_GTEST_GTEST_MESSAGE_H_
|
#endif // GTEST_INCLUDE_GTEST_GTEST_MESSAGE_H_
|
||||||
|
@ -1257,7 +1257,7 @@ inline internal::ParamGenerator<bool> Bool() {
|
|||||||
// Boolean flags:
|
// Boolean flags:
|
||||||
//
|
//
|
||||||
// class FlagDependentTest
|
// class FlagDependentTest
|
||||||
// : public testing::TestWithParam<tuple(bool, bool)> > {
|
// : public testing::TestWithParam<tuple<bool, bool> > {
|
||||||
// virtual void SetUp() {
|
// virtual void SetUp() {
|
||||||
// // Assigns external_flag_1 and external_flag_2 values from the tuple.
|
// // Assigns external_flag_1 and external_flag_2 values from the tuple.
|
||||||
// tie(external_flag_1, external_flag_2) = GetParam();
|
// tie(external_flag_1, external_flag_2) = GetParam();
|
||||||
|
@ -414,7 +414,7 @@ inline internal::ParamGenerator<bool> Bool() {
|
|||||||
// Boolean flags:
|
// Boolean flags:
|
||||||
//
|
//
|
||||||
// class FlagDependentTest
|
// class FlagDependentTest
|
||||||
// : public testing::TestWithParam<tuple(bool, bool)> > {
|
// : public testing::TestWithParam<tuple<bool, bool> > {
|
||||||
// virtual void SetUp() {
|
// virtual void SetUp() {
|
||||||
// // Assigns external_flag_1 and external_flag_2 values from the tuple.
|
// // Assigns external_flag_1 and external_flag_2 values from the tuple.
|
||||||
// tie(external_flag_1, external_flag_2) = GetParam();
|
// tie(external_flag_1, external_flag_2) = GetParam();
|
||||||
|
@ -103,6 +103,10 @@
|
|||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
#include "gtest/internal/gtest-internal.h"
|
#include "gtest/internal/gtest-internal.h"
|
||||||
|
|
||||||
|
#if GTEST_HAS_STD_TUPLE_
|
||||||
|
# include <tuple>
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
|
|
||||||
// Definitions in the 'internal' and 'internal2' name spaces are
|
// Definitions in the 'internal' and 'internal2' name spaces are
|
||||||
@ -480,14 +484,16 @@ inline void PrintTo(const ::std::wstring& s, ::std::ostream* os) {
|
|||||||
}
|
}
|
||||||
#endif // GTEST_HAS_STD_WSTRING
|
#endif // GTEST_HAS_STD_WSTRING
|
||||||
|
|
||||||
#if GTEST_HAS_TR1_TUPLE
|
#if GTEST_HAS_TR1_TUPLE || GTEST_HAS_STD_TUPLE_
|
||||||
// Overload for ::std::tr1::tuple. Needed for printing function arguments,
|
|
||||||
// which are packed as tuples.
|
|
||||||
|
|
||||||
// Helper function for printing a tuple. T must be instantiated with
|
// Helper function for printing a tuple. T must be instantiated with
|
||||||
// a tuple type.
|
// a tuple type.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void PrintTupleTo(const T& t, ::std::ostream* os);
|
void PrintTupleTo(const T& t, ::std::ostream* os);
|
||||||
|
#endif // GTEST_HAS_TR1_TUPLE || GTEST_HAS_STD_TUPLE_
|
||||||
|
|
||||||
|
#if GTEST_HAS_TR1_TUPLE
|
||||||
|
// Overload for ::std::tr1::tuple. Needed for printing function arguments,
|
||||||
|
// which are packed as tuples.
|
||||||
|
|
||||||
// Overloaded PrintTo() for tuples of various arities. We support
|
// Overloaded PrintTo() for tuples of various arities. We support
|
||||||
// tuples of up-to 10 fields. The following implementation works
|
// tuples of up-to 10 fields. The following implementation works
|
||||||
@ -561,6 +567,13 @@ void PrintTo(
|
|||||||
}
|
}
|
||||||
#endif // GTEST_HAS_TR1_TUPLE
|
#endif // GTEST_HAS_TR1_TUPLE
|
||||||
|
|
||||||
|
#if GTEST_HAS_STD_TUPLE_
|
||||||
|
template <typename... Types>
|
||||||
|
void PrintTo(const ::std::tuple<Types...>& t, ::std::ostream* os) {
|
||||||
|
PrintTupleTo(t, os);
|
||||||
|
}
|
||||||
|
#endif // GTEST_HAS_STD_TUPLE_
|
||||||
|
|
||||||
// Overload for std::pair.
|
// Overload for std::pair.
|
||||||
template <typename T1, typename T2>
|
template <typename T1, typename T2>
|
||||||
void PrintTo(const ::std::pair<T1, T2>& value, ::std::ostream* os) {
|
void PrintTo(const ::std::pair<T1, T2>& value, ::std::ostream* os) {
|
||||||
@ -580,10 +593,7 @@ class UniversalPrinter {
|
|||||||
public:
|
public:
|
||||||
// MSVC warns about adding const to a function type, so we want to
|
// MSVC warns about adding const to a function type, so we want to
|
||||||
// disable the warning.
|
// disable the warning.
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4180)
|
||||||
# pragma warning(push) // Saves the current warning state.
|
|
||||||
# pragma warning(disable:4180) // Temporarily disables warning 4180.
|
|
||||||
#endif // _MSC_VER
|
|
||||||
|
|
||||||
// Note: we deliberately don't call this PrintTo(), as that name
|
// Note: we deliberately don't call this PrintTo(), as that name
|
||||||
// conflicts with ::testing::internal::PrintTo in the body of the
|
// conflicts with ::testing::internal::PrintTo in the body of the
|
||||||
@ -600,9 +610,7 @@ class UniversalPrinter {
|
|||||||
PrintTo(value, os);
|
PrintTo(value, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
# pragma warning(pop) // Restores the warning state.
|
|
||||||
#endif // _MSC_VER
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// UniversalPrintArray(begin, len, os) prints an array of 'len'
|
// UniversalPrintArray(begin, len, os) prints an array of 'len'
|
||||||
@ -630,9 +638,12 @@ void UniversalPrintArray(const T* begin, size_t len, ::std::ostream* os) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
// This overload prints a (const) char array compactly.
|
// This overload prints a (const) char array compactly.
|
||||||
GTEST_API_ void UniversalPrintArray(const char* begin,
|
GTEST_API_ void UniversalPrintArray(
|
||||||
size_t len,
|
const char* begin, size_t len, ::std::ostream* os);
|
||||||
::std::ostream* os);
|
|
||||||
|
// This overload prints a (const) wchar_t array compactly.
|
||||||
|
GTEST_API_ void UniversalPrintArray(
|
||||||
|
const wchar_t* begin, size_t len, ::std::ostream* os);
|
||||||
|
|
||||||
// Implements printing an array type T[N].
|
// Implements printing an array type T[N].
|
||||||
template <typename T, size_t N>
|
template <typename T, size_t N>
|
||||||
@ -651,10 +662,7 @@ class UniversalPrinter<T&> {
|
|||||||
public:
|
public:
|
||||||
// MSVC warns about adding const to a function type, so we want to
|
// MSVC warns about adding const to a function type, so we want to
|
||||||
// disable the warning.
|
// disable the warning.
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4180)
|
||||||
# pragma warning(push) // Saves the current warning state.
|
|
||||||
# pragma warning(disable:4180) // Temporarily disables warning 4180.
|
|
||||||
#endif // _MSC_VER
|
|
||||||
|
|
||||||
static void Print(const T& value, ::std::ostream* os) {
|
static void Print(const T& value, ::std::ostream* os) {
|
||||||
// Prints the address of the value. We use reinterpret_cast here
|
// Prints the address of the value. We use reinterpret_cast here
|
||||||
@ -665,27 +673,78 @@ class UniversalPrinter<T&> {
|
|||||||
UniversalPrint(value, os);
|
UniversalPrint(value, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
# pragma warning(pop) // Restores the warning state.
|
|
||||||
#endif // _MSC_VER
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// Prints a value tersely: for a reference type, the referenced value
|
// Prints a value tersely: for a reference type, the referenced value
|
||||||
// (but not the address) is printed; for a (const) char pointer, the
|
// (but not the address) is printed; for a (const) char pointer, the
|
||||||
// NUL-terminated string (but not the pointer) is printed.
|
// NUL-terminated string (but not the pointer) is printed.
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void UniversalTersePrint(const T& value, ::std::ostream* os) {
|
class UniversalTersePrinter {
|
||||||
|
public:
|
||||||
|
static void Print(const T& value, ::std::ostream* os) {
|
||||||
UniversalPrint(value, os);
|
UniversalPrint(value, os);
|
||||||
}
|
}
|
||||||
inline void UniversalTersePrint(const char* str, ::std::ostream* os) {
|
};
|
||||||
|
template <typename T>
|
||||||
|
class UniversalTersePrinter<T&> {
|
||||||
|
public:
|
||||||
|
static void Print(const T& value, ::std::ostream* os) {
|
||||||
|
UniversalPrint(value, os);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
template <typename T, size_t N>
|
||||||
|
class UniversalTersePrinter<T[N]> {
|
||||||
|
public:
|
||||||
|
static void Print(const T (&value)[N], ::std::ostream* os) {
|
||||||
|
UniversalPrinter<T[N]>::Print(value, os);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
template <>
|
||||||
|
class UniversalTersePrinter<const char*> {
|
||||||
|
public:
|
||||||
|
static void Print(const char* str, ::std::ostream* os) {
|
||||||
if (str == NULL) {
|
if (str == NULL) {
|
||||||
*os << "NULL";
|
*os << "NULL";
|
||||||
} else {
|
} else {
|
||||||
UniversalPrint(string(str), os);
|
UniversalPrint(string(str), os);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
inline void UniversalTersePrint(char* str, ::std::ostream* os) {
|
};
|
||||||
UniversalTersePrint(static_cast<const char*>(str), os);
|
template <>
|
||||||
|
class UniversalTersePrinter<char*> {
|
||||||
|
public:
|
||||||
|
static void Print(char* str, ::std::ostream* os) {
|
||||||
|
UniversalTersePrinter<const char*>::Print(str, os);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#if GTEST_HAS_STD_WSTRING
|
||||||
|
template <>
|
||||||
|
class UniversalTersePrinter<const wchar_t*> {
|
||||||
|
public:
|
||||||
|
static void Print(const wchar_t* str, ::std::ostream* os) {
|
||||||
|
if (str == NULL) {
|
||||||
|
*os << "NULL";
|
||||||
|
} else {
|
||||||
|
UniversalPrint(::std::wstring(str), os);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
|
template <>
|
||||||
|
class UniversalTersePrinter<wchar_t*> {
|
||||||
|
public:
|
||||||
|
static void Print(wchar_t* str, ::std::ostream* os) {
|
||||||
|
UniversalTersePrinter<const wchar_t*>::Print(str, os);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
void UniversalTersePrint(const T& value, ::std::ostream* os) {
|
||||||
|
UniversalTersePrinter<T>::Print(value, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints a value using the type inferred by the compiler. The
|
// Prints a value using the type inferred by the compiler. The
|
||||||
@ -694,19 +753,71 @@ inline void UniversalTersePrint(char* str, ::std::ostream* os) {
|
|||||||
// NUL-terminated string.
|
// NUL-terminated string.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void UniversalPrint(const T& value, ::std::ostream* os) {
|
void UniversalPrint(const T& value, ::std::ostream* os) {
|
||||||
UniversalPrinter<T>::Print(value, os);
|
// A workarond for the bug in VC++ 7.1 that prevents us from instantiating
|
||||||
|
// UniversalPrinter with T directly.
|
||||||
|
typedef T T1;
|
||||||
|
UniversalPrinter<T1>::Print(value, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
#if GTEST_HAS_TR1_TUPLE
|
|
||||||
typedef ::std::vector<string> Strings;
|
typedef ::std::vector<string> Strings;
|
||||||
|
|
||||||
|
// TuplePolicy<TupleT> must provide:
|
||||||
|
// - tuple_size
|
||||||
|
// size of tuple TupleT.
|
||||||
|
// - get<size_t I>(const TupleT& t)
|
||||||
|
// static function extracting element I of tuple TupleT.
|
||||||
|
// - tuple_element<size_t I>::type
|
||||||
|
// type of element I of tuple TupleT.
|
||||||
|
template <typename TupleT>
|
||||||
|
struct TuplePolicy;
|
||||||
|
|
||||||
|
#if GTEST_HAS_TR1_TUPLE
|
||||||
|
template <typename TupleT>
|
||||||
|
struct TuplePolicy {
|
||||||
|
typedef TupleT Tuple;
|
||||||
|
static const size_t tuple_size = ::std::tr1::tuple_size<Tuple>::value;
|
||||||
|
|
||||||
|
template <size_t I>
|
||||||
|
struct tuple_element : ::std::tr1::tuple_element<I, Tuple> {};
|
||||||
|
|
||||||
|
template <size_t I>
|
||||||
|
static typename AddReference<
|
||||||
|
const typename ::std::tr1::tuple_element<I, Tuple>::type>::type get(
|
||||||
|
const Tuple& tuple) {
|
||||||
|
return ::std::tr1::get<I>(tuple);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
template <typename TupleT>
|
||||||
|
const size_t TuplePolicy<TupleT>::tuple_size;
|
||||||
|
#endif // GTEST_HAS_TR1_TUPLE
|
||||||
|
|
||||||
|
#if GTEST_HAS_STD_TUPLE_
|
||||||
|
template <typename... Types>
|
||||||
|
struct TuplePolicy< ::std::tuple<Types...> > {
|
||||||
|
typedef ::std::tuple<Types...> Tuple;
|
||||||
|
static const size_t tuple_size = ::std::tuple_size<Tuple>::value;
|
||||||
|
|
||||||
|
template <size_t I>
|
||||||
|
struct tuple_element : ::std::tuple_element<I, Tuple> {};
|
||||||
|
|
||||||
|
template <size_t I>
|
||||||
|
static const typename ::std::tuple_element<I, Tuple>::type& get(
|
||||||
|
const Tuple& tuple) {
|
||||||
|
return ::std::get<I>(tuple);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
template <typename... Types>
|
||||||
|
const size_t TuplePolicy< ::std::tuple<Types...> >::tuple_size;
|
||||||
|
#endif // GTEST_HAS_STD_TUPLE_
|
||||||
|
|
||||||
|
#if GTEST_HAS_TR1_TUPLE || GTEST_HAS_STD_TUPLE_
|
||||||
// This helper template allows PrintTo() for tuples and
|
// This helper template allows PrintTo() for tuples and
|
||||||
// UniversalTersePrintTupleFieldsToStrings() to be defined by
|
// UniversalTersePrintTupleFieldsToStrings() to be defined by
|
||||||
// induction on the number of tuple fields. The idea is that
|
// induction on the number of tuple fields. The idea is that
|
||||||
// TuplePrefixPrinter<N>::PrintPrefixTo(t, os) prints the first N
|
// TuplePrefixPrinter<N>::PrintPrefixTo(t, os) prints the first N
|
||||||
// fields in tuple t, and can be defined in terms of
|
// fields in tuple t, and can be defined in terms of
|
||||||
// TuplePrefixPrinter<N - 1>.
|
// TuplePrefixPrinter<N - 1>.
|
||||||
|
//
|
||||||
// The inductive case.
|
// The inductive case.
|
||||||
template <size_t N>
|
template <size_t N>
|
||||||
struct TuplePrefixPrinter {
|
struct TuplePrefixPrinter {
|
||||||
@ -714,9 +825,14 @@ struct TuplePrefixPrinter {
|
|||||||
template <typename Tuple>
|
template <typename Tuple>
|
||||||
static void PrintPrefixTo(const Tuple& t, ::std::ostream* os) {
|
static void PrintPrefixTo(const Tuple& t, ::std::ostream* os) {
|
||||||
TuplePrefixPrinter<N - 1>::PrintPrefixTo(t, os);
|
TuplePrefixPrinter<N - 1>::PrintPrefixTo(t, os);
|
||||||
|
GTEST_INTENTIONAL_CONST_COND_PUSH_()
|
||||||
|
if (N > 1) {
|
||||||
|
GTEST_INTENTIONAL_CONST_COND_POP_()
|
||||||
*os << ", ";
|
*os << ", ";
|
||||||
UniversalPrinter<typename ::std::tr1::tuple_element<N - 1, Tuple>::type>
|
}
|
||||||
::Print(::std::tr1::get<N - 1>(t), os);
|
UniversalPrinter<
|
||||||
|
typename TuplePolicy<Tuple>::template tuple_element<N - 1>::type>
|
||||||
|
::Print(TuplePolicy<Tuple>::template get<N - 1>(t), os);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tersely prints the first N fields of a tuple to a string vector,
|
// Tersely prints the first N fields of a tuple to a string vector,
|
||||||
@ -725,12 +841,12 @@ struct TuplePrefixPrinter {
|
|||||||
static void TersePrintPrefixToStrings(const Tuple& t, Strings* strings) {
|
static void TersePrintPrefixToStrings(const Tuple& t, Strings* strings) {
|
||||||
TuplePrefixPrinter<N - 1>::TersePrintPrefixToStrings(t, strings);
|
TuplePrefixPrinter<N - 1>::TersePrintPrefixToStrings(t, strings);
|
||||||
::std::stringstream ss;
|
::std::stringstream ss;
|
||||||
UniversalTersePrint(::std::tr1::get<N - 1>(t), &ss);
|
UniversalTersePrint(TuplePolicy<Tuple>::template get<N - 1>(t), &ss);
|
||||||
strings->push_back(ss.str());
|
strings->push_back(ss.str());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Base cases.
|
// Base case.
|
||||||
template <>
|
template <>
|
||||||
struct TuplePrefixPrinter<0> {
|
struct TuplePrefixPrinter<0> {
|
||||||
template <typename Tuple>
|
template <typename Tuple>
|
||||||
@ -739,34 +855,13 @@ struct TuplePrefixPrinter<0> {
|
|||||||
template <typename Tuple>
|
template <typename Tuple>
|
||||||
static void TersePrintPrefixToStrings(const Tuple&, Strings*) {}
|
static void TersePrintPrefixToStrings(const Tuple&, Strings*) {}
|
||||||
};
|
};
|
||||||
// We have to specialize the entire TuplePrefixPrinter<> class
|
|
||||||
// template here, even though the definition of
|
|
||||||
// TersePrintPrefixToStrings() is the same as the generic version, as
|
|
||||||
// Embarcadero (formerly CodeGear, formerly Borland) C++ doesn't
|
|
||||||
// support specializing a method template of a class template.
|
|
||||||
template <>
|
|
||||||
struct TuplePrefixPrinter<1> {
|
|
||||||
template <typename Tuple>
|
|
||||||
static void PrintPrefixTo(const Tuple& t, ::std::ostream* os) {
|
|
||||||
UniversalPrinter<typename ::std::tr1::tuple_element<0, Tuple>::type>::
|
|
||||||
Print(::std::tr1::get<0>(t), os);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename Tuple>
|
// Helper function for printing a tuple.
|
||||||
static void TersePrintPrefixToStrings(const Tuple& t, Strings* strings) {
|
// Tuple must be either std::tr1::tuple or std::tuple type.
|
||||||
::std::stringstream ss;
|
template <typename Tuple>
|
||||||
UniversalTersePrint(::std::tr1::get<0>(t), &ss);
|
void PrintTupleTo(const Tuple& t, ::std::ostream* os) {
|
||||||
strings->push_back(ss.str());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Helper function for printing a tuple. T must be instantiated with
|
|
||||||
// a tuple type.
|
|
||||||
template <typename T>
|
|
||||||
void PrintTupleTo(const T& t, ::std::ostream* os) {
|
|
||||||
*os << "(";
|
*os << "(";
|
||||||
TuplePrefixPrinter< ::std::tr1::tuple_size<T>::value>::
|
TuplePrefixPrinter<TuplePolicy<Tuple>::tuple_size>::PrintPrefixTo(t, os);
|
||||||
PrintPrefixTo(t, os);
|
|
||||||
*os << ")";
|
*os << ")";
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -776,18 +871,18 @@ void PrintTupleTo(const T& t, ::std::ostream* os) {
|
|||||||
template <typename Tuple>
|
template <typename Tuple>
|
||||||
Strings UniversalTersePrintTupleFieldsToStrings(const Tuple& value) {
|
Strings UniversalTersePrintTupleFieldsToStrings(const Tuple& value) {
|
||||||
Strings result;
|
Strings result;
|
||||||
TuplePrefixPrinter< ::std::tr1::tuple_size<Tuple>::value>::
|
TuplePrefixPrinter<TuplePolicy<Tuple>::tuple_size>::
|
||||||
TersePrintPrefixToStrings(value, &result);
|
TersePrintPrefixToStrings(value, &result);
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
#endif // GTEST_HAS_TR1_TUPLE
|
#endif // GTEST_HAS_TR1_TUPLE || GTEST_HAS_STD_TUPLE_
|
||||||
|
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
::std::string PrintToString(const T& value) {
|
::std::string PrintToString(const T& value) {
|
||||||
::std::stringstream ss;
|
::std::stringstream ss;
|
||||||
internal::UniversalTersePrint(value, &ss);
|
internal::UniversalTersePrinter<T>::Print(value, &ss);
|
||||||
return ss.str();
|
return ss.str();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -223,7 +223,7 @@ class GTEST_API_ SingleFailureChecker {
|
|||||||
(substr));\
|
(substr));\
|
||||||
{\
|
{\
|
||||||
::testing::ScopedFakeTestPartResultReporter gtest_reporter(\
|
::testing::ScopedFakeTestPartResultReporter gtest_reporter(\
|
||||||
::testing::ScopedFakeTestPartResultReporter::INTERCEPT_ALL_THREADS,\
|
::testing::ScopedFakeTestPartResultReporter::INTERCEPT_ALL_THREADS, \
|
||||||
>est_failures);\
|
>est_failures);\
|
||||||
if (::testing::internal::AlwaysTrue()) { statement; }\
|
if (::testing::internal::AlwaysTrue()) { statement; }\
|
||||||
}\
|
}\
|
||||||
|
@ -62,7 +62,7 @@ class GTEST_API_ TestPartResult {
|
|||||||
int a_line_number,
|
int a_line_number,
|
||||||
const char* a_message)
|
const char* a_message)
|
||||||
: type_(a_type),
|
: type_(a_type),
|
||||||
file_name_(a_file_name),
|
file_name_(a_file_name == NULL ? "" : a_file_name),
|
||||||
line_number_(a_line_number),
|
line_number_(a_line_number),
|
||||||
summary_(ExtractSummary(a_message)),
|
summary_(ExtractSummary(a_message)),
|
||||||
message_(a_message) {
|
message_(a_message) {
|
||||||
@ -73,7 +73,9 @@ class GTEST_API_ TestPartResult {
|
|||||||
|
|
||||||
// Gets the name of the source file where the test part took place, or
|
// Gets the name of the source file where the test part took place, or
|
||||||
// NULL if it's unknown.
|
// NULL if it's unknown.
|
||||||
const char* file_name() const { return file_name_.c_str(); }
|
const char* file_name() const {
|
||||||
|
return file_name_.empty() ? NULL : file_name_.c_str();
|
||||||
|
}
|
||||||
|
|
||||||
// Gets the line in the source file where the test part took place,
|
// Gets the line in the source file where the test part took place,
|
||||||
// or -1 if it's unknown.
|
// or -1 if it's unknown.
|
||||||
@ -96,21 +98,22 @@ class GTEST_API_ TestPartResult {
|
|||||||
|
|
||||||
// Returns true iff the test part fatally failed.
|
// Returns true iff the test part fatally failed.
|
||||||
bool fatally_failed() const { return type_ == kFatalFailure; }
|
bool fatally_failed() const { return type_ == kFatalFailure; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Type type_;
|
Type type_;
|
||||||
|
|
||||||
// Gets the summary of the failure message by omitting the stack
|
// Gets the summary of the failure message by omitting the stack
|
||||||
// trace in it.
|
// trace in it.
|
||||||
static internal::String ExtractSummary(const char* message);
|
static std::string ExtractSummary(const char* message);
|
||||||
|
|
||||||
// The name of the source file where the test part took place, or
|
// The name of the source file where the test part took place, or
|
||||||
// NULL if the source file is unknown.
|
// "" if the source file is unknown.
|
||||||
internal::String file_name_;
|
std::string file_name_;
|
||||||
// The line in the source file where the test part took place, or -1
|
// The line in the source file where the test part took place, or -1
|
||||||
// if the line number is unknown.
|
// if the line number is unknown.
|
||||||
int line_number_;
|
int line_number_;
|
||||||
internal::String summary_; // The test failure summary.
|
std::string summary_; // The test failure summary.
|
||||||
internal::String message_; // The test failure message.
|
std::string message_; // The test failure message.
|
||||||
};
|
};
|
||||||
|
|
||||||
// Prints a TestPartResult object.
|
// Prints a TestPartResult object.
|
||||||
|
@ -52,6 +52,7 @@
|
|||||||
#define GTEST_INCLUDE_GTEST_GTEST_H_
|
#define GTEST_INCLUDE_GTEST_GTEST_H_
|
||||||
|
|
||||||
#include <limits>
|
#include <limits>
|
||||||
|
#include <ostream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "gtest/internal/gtest-internal.h"
|
#include "gtest/internal/gtest-internal.h"
|
||||||
@ -69,14 +70,14 @@
|
|||||||
// class ::string, which has the same interface as ::std::string, but
|
// class ::string, which has the same interface as ::std::string, but
|
||||||
// has a different implementation.
|
// has a different implementation.
|
||||||
//
|
//
|
||||||
// The user can define GTEST_HAS_GLOBAL_STRING to 1 to indicate that
|
// You can define GTEST_HAS_GLOBAL_STRING to 1 to indicate that
|
||||||
// ::string is available AND is a distinct type to ::std::string, or
|
// ::string is available AND is a distinct type to ::std::string, or
|
||||||
// define it to 0 to indicate otherwise.
|
// define it to 0 to indicate otherwise.
|
||||||
//
|
//
|
||||||
// If the user's ::std::string and ::string are the same class due to
|
// If ::std::string and ::string are the same class on your platform
|
||||||
// aliasing, he should define GTEST_HAS_GLOBAL_STRING to 0.
|
// due to aliasing, you should define GTEST_HAS_GLOBAL_STRING to 0.
|
||||||
//
|
//
|
||||||
// If the user doesn't define GTEST_HAS_GLOBAL_STRING, it is defined
|
// If you do not define GTEST_HAS_GLOBAL_STRING, it is defined
|
||||||
// heuristically.
|
// heuristically.
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
@ -153,25 +154,15 @@ class ExecDeathTest;
|
|||||||
class NoExecDeathTest;
|
class NoExecDeathTest;
|
||||||
class FinalSuccessChecker;
|
class FinalSuccessChecker;
|
||||||
class GTestFlagSaver;
|
class GTestFlagSaver;
|
||||||
|
class StreamingListenerTest;
|
||||||
class TestResultAccessor;
|
class TestResultAccessor;
|
||||||
class TestEventListenersAccessor;
|
class TestEventListenersAccessor;
|
||||||
class TestEventRepeater;
|
class TestEventRepeater;
|
||||||
|
class UnitTestRecordPropertyTestHelper;
|
||||||
class WindowsDeathTest;
|
class WindowsDeathTest;
|
||||||
class UnitTestImpl* GetUnitTestImpl();
|
class UnitTestImpl* GetUnitTestImpl();
|
||||||
void ReportFailureInUnknownLocation(TestPartResult::Type result_type,
|
void ReportFailureInUnknownLocation(TestPartResult::Type result_type,
|
||||||
const String& message);
|
const std::string& message);
|
||||||
|
|
||||||
// Converts a streamable value to a String. A NULL pointer is
|
|
||||||
// converted to "(null)". When the input value is a ::string,
|
|
||||||
// ::std::string, ::wstring, or ::std::wstring object, each NUL
|
|
||||||
// character in it is replaced with "\\0".
|
|
||||||
// Declared in gtest-internal.h but defined here, so that it has access
|
|
||||||
// to the definition of the Message class, required by the ARM
|
|
||||||
// compiler.
|
|
||||||
template <typename T>
|
|
||||||
String StreamableToString(const T& streamable) {
|
|
||||||
return (Message() << streamable).GetString();
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
|
|
||||||
@ -267,8 +258,31 @@ class GTEST_API_ AssertionResult {
|
|||||||
// Copy constructor.
|
// Copy constructor.
|
||||||
// Used in EXPECT_TRUE/FALSE(assertion_result).
|
// Used in EXPECT_TRUE/FALSE(assertion_result).
|
||||||
AssertionResult(const AssertionResult& other);
|
AssertionResult(const AssertionResult& other);
|
||||||
|
|
||||||
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4800 /* forcing value to bool */)
|
||||||
|
|
||||||
// Used in the EXPECT_TRUE/FALSE(bool_expression).
|
// Used in the EXPECT_TRUE/FALSE(bool_expression).
|
||||||
explicit AssertionResult(bool success) : success_(success) {}
|
//
|
||||||
|
// T must be contextually convertible to bool.
|
||||||
|
//
|
||||||
|
// The second parameter prevents this overload from being considered if
|
||||||
|
// the argument is implicitly convertible to AssertionResult. In that case
|
||||||
|
// we want AssertionResult's copy constructor to be used.
|
||||||
|
template <typename T>
|
||||||
|
explicit AssertionResult(
|
||||||
|
const T& success,
|
||||||
|
typename internal::EnableIf<
|
||||||
|
!internal::ImplicitlyConvertible<T, AssertionResult>::value>::type*
|
||||||
|
/*enabler*/ = NULL)
|
||||||
|
: success_(success) {}
|
||||||
|
|
||||||
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
|
|
||||||
|
// Assignment operator.
|
||||||
|
AssertionResult& operator=(AssertionResult other) {
|
||||||
|
swap(other);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
// Returns true iff the assertion succeeded.
|
// Returns true iff the assertion succeeded.
|
||||||
operator bool() const { return success_; } // NOLINT
|
operator bool() const { return success_; } // NOLINT
|
||||||
@ -309,6 +323,9 @@ class GTEST_API_ AssertionResult {
|
|||||||
message_->append(a_message.GetString().c_str());
|
message_->append(a_message.GetString().c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Swap the contents of this AssertionResult with other.
|
||||||
|
void swap(AssertionResult& other);
|
||||||
|
|
||||||
// Stores result of the assertion predicate.
|
// Stores result of the assertion predicate.
|
||||||
bool success_;
|
bool success_;
|
||||||
// Stores the message describing the condition in case the expectation
|
// Stores the message describing the condition in case the expectation
|
||||||
@ -316,8 +333,6 @@ class GTEST_API_ AssertionResult {
|
|||||||
// Referenced via a pointer to avoid taking too much stack frame space
|
// Referenced via a pointer to avoid taking too much stack frame space
|
||||||
// with test assertions.
|
// with test assertions.
|
||||||
internal::scoped_ptr< ::std::string> message_;
|
internal::scoped_ptr< ::std::string> message_;
|
||||||
|
|
||||||
GTEST_DISALLOW_ASSIGN_(AssertionResult);
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// Makes a successful assertion result.
|
// Makes a successful assertion result.
|
||||||
@ -344,8 +359,8 @@ GTEST_API_ AssertionResult AssertionFailure(const Message& msg);
|
|||||||
//
|
//
|
||||||
// class FooTest : public testing::Test {
|
// class FooTest : public testing::Test {
|
||||||
// protected:
|
// protected:
|
||||||
// virtual void SetUp() { ... }
|
// void SetUp() override { ... }
|
||||||
// virtual void TearDown() { ... }
|
// void TearDown() override { ... }
|
||||||
// ...
|
// ...
|
||||||
// };
|
// };
|
||||||
//
|
//
|
||||||
@ -391,20 +406,21 @@ class GTEST_API_ Test {
|
|||||||
// non-fatal) failure.
|
// non-fatal) failure.
|
||||||
static bool HasFailure() { return HasFatalFailure() || HasNonfatalFailure(); }
|
static bool HasFailure() { return HasFatalFailure() || HasNonfatalFailure(); }
|
||||||
|
|
||||||
// Logs a property for the current test. Only the last value for a given
|
// Logs a property for the current test, test case, or for the entire
|
||||||
// key is remembered.
|
// invocation of the test program when used outside of the context of a
|
||||||
// These are public static so they can be called from utility functions
|
// test case. Only the last value for a given key is remembered. These
|
||||||
// that are not members of the test fixture.
|
// are public static so they can be called from utility functions that are
|
||||||
// The arguments are const char* instead strings, as Google Test is used
|
// not members of the test fixture. Calls to RecordProperty made during
|
||||||
// on platforms where string doesn't compile.
|
// lifespan of the test (from the moment its constructor starts to the
|
||||||
//
|
// moment its destructor finishes) will be output in XML as attributes of
|
||||||
// Note that a driving consideration for these RecordProperty methods
|
// the <testcase> element. Properties recorded from fixture's
|
||||||
// was to produce xml output suited to the Greenspan charting utility,
|
// SetUpTestCase or TearDownTestCase are logged as attributes of the
|
||||||
// which at present will only chart values that fit in a 32-bit int. It
|
// corresponding <testsuite> element. Calls to RecordProperty made in the
|
||||||
// is the user's responsibility to restrict their values to 32-bit ints
|
// global context (before or after invocation of RUN_ALL_TESTS and from
|
||||||
// if they intend them to be used with Greenspan.
|
// SetUp/TearDown method of Environment objects registered with Google
|
||||||
static void RecordProperty(const char* key, const char* value);
|
// Test) will be output as attributes of the <testsuites> element.
|
||||||
static void RecordProperty(const char* key, int value);
|
static void RecordProperty(const std::string& key, const std::string& value);
|
||||||
|
static void RecordProperty(const std::string& key, int value);
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
// Creates a Test object.
|
// Creates a Test object.
|
||||||
@ -439,17 +455,17 @@ class GTEST_API_ Test {
|
|||||||
// Uses a GTestFlagSaver to save and restore all Google Test flags.
|
// Uses a GTestFlagSaver to save and restore all Google Test flags.
|
||||||
const internal::GTestFlagSaver* const gtest_flag_saver_;
|
const internal::GTestFlagSaver* const gtest_flag_saver_;
|
||||||
|
|
||||||
// Often a user mis-spells SetUp() as Setup() and spends a long time
|
// Often a user misspells SetUp() as Setup() and spends a long time
|
||||||
// wondering why it is never called by Google Test. The declaration of
|
// wondering why it is never called by Google Test. The declaration of
|
||||||
// the following method is solely for catching such an error at
|
// the following method is solely for catching such an error at
|
||||||
// compile time:
|
// compile time:
|
||||||
//
|
//
|
||||||
// - The return type is deliberately chosen to be not void, so it
|
// - The return type is deliberately chosen to be not void, so it
|
||||||
// will be a conflict if a user declares void Setup() in his test
|
// will be a conflict if void Setup() is declared in the user's
|
||||||
// fixture.
|
// test fixture.
|
||||||
//
|
//
|
||||||
// - This method is private, so it will be another compiler error
|
// - This method is private, so it will be another compiler error
|
||||||
// if a user calls it from his test fixture.
|
// if the method is called from the user's test fixture.
|
||||||
//
|
//
|
||||||
// DO NOT OVERRIDE THIS FUNCTION.
|
// DO NOT OVERRIDE THIS FUNCTION.
|
||||||
//
|
//
|
||||||
@ -473,7 +489,7 @@ class TestProperty {
|
|||||||
// C'tor. TestProperty does NOT have a default constructor.
|
// C'tor. TestProperty does NOT have a default constructor.
|
||||||
// Always use this constructor (with parameters) to create a
|
// Always use this constructor (with parameters) to create a
|
||||||
// TestProperty object.
|
// TestProperty object.
|
||||||
TestProperty(const char* a_key, const char* a_value) :
|
TestProperty(const std::string& a_key, const std::string& a_value) :
|
||||||
key_(a_key), value_(a_value) {
|
key_(a_key), value_(a_value) {
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -488,15 +504,15 @@ class TestProperty {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Sets a new value, overriding the one supplied in the constructor.
|
// Sets a new value, overriding the one supplied in the constructor.
|
||||||
void SetValue(const char* new_value) {
|
void SetValue(const std::string& new_value) {
|
||||||
value_ = new_value;
|
value_ = new_value;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// The key supplied by the user.
|
// The key supplied by the user.
|
||||||
internal::String key_;
|
std::string key_;
|
||||||
// The value supplied by the user.
|
// The value supplied by the user.
|
||||||
internal::String value_;
|
std::string value_;
|
||||||
};
|
};
|
||||||
|
|
||||||
// The result of a single Test. This includes a list of
|
// The result of a single Test. This includes a list of
|
||||||
@ -547,6 +563,7 @@ class GTEST_API_ TestResult {
|
|||||||
|
|
||||||
private:
|
private:
|
||||||
friend class TestInfo;
|
friend class TestInfo;
|
||||||
|
friend class TestCase;
|
||||||
friend class UnitTest;
|
friend class UnitTest;
|
||||||
friend class internal::DefaultGlobalTestPartResultReporter;
|
friend class internal::DefaultGlobalTestPartResultReporter;
|
||||||
friend class internal::ExecDeathTest;
|
friend class internal::ExecDeathTest;
|
||||||
@ -571,13 +588,16 @@ class GTEST_API_ TestResult {
|
|||||||
// a non-fatal failure if invalid (e.g., if it conflicts with reserved
|
// a non-fatal failure if invalid (e.g., if it conflicts with reserved
|
||||||
// key names). If a property is already recorded for the same key, the
|
// key names). If a property is already recorded for the same key, the
|
||||||
// value will be updated, rather than storing multiple values for the same
|
// value will be updated, rather than storing multiple values for the same
|
||||||
// key.
|
// key. xml_element specifies the element for which the property is being
|
||||||
void RecordProperty(const TestProperty& test_property);
|
// recorded and is used for validation.
|
||||||
|
void RecordProperty(const std::string& xml_element,
|
||||||
|
const TestProperty& test_property);
|
||||||
|
|
||||||
// Adds a failure if the key is a reserved attribute of Google Test
|
// Adds a failure if the key is a reserved attribute of Google Test
|
||||||
// testcase tags. Returns true if the property is valid.
|
// testcase tags. Returns true if the property is valid.
|
||||||
// TODO(russr): Validate attribute names are legal and human readable.
|
// TODO(russr): Validate attribute names are legal and human readable.
|
||||||
static bool ValidateTestProperty(const TestProperty& test_property);
|
static bool ValidateTestProperty(const std::string& xml_element,
|
||||||
|
const TestProperty& test_property);
|
||||||
|
|
||||||
// Adds a test part result to the list.
|
// Adds a test part result to the list.
|
||||||
void AddTestPartResult(const TestPartResult& test_part_result);
|
void AddTestPartResult(const TestPartResult& test_part_result);
|
||||||
@ -650,9 +670,9 @@ class GTEST_API_ TestInfo {
|
|||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns true if this test should run, that is if the test is not disabled
|
// Returns true if this test should run, that is if the test is not
|
||||||
// (or it is disabled but the also_run_disabled_tests flag has been specified)
|
// disabled (or it is disabled but the also_run_disabled_tests flag has
|
||||||
// and its full name matches the user-specified filter.
|
// been specified) and its full name matches the user-specified filter.
|
||||||
//
|
//
|
||||||
// Google Test allows the user to filter the tests by their full names.
|
// Google Test allows the user to filter the tests by their full names.
|
||||||
// The full name of a test Bar in test case Foo is defined as
|
// The full name of a test Bar in test case Foo is defined as
|
||||||
@ -668,22 +688,28 @@ class GTEST_API_ TestInfo {
|
|||||||
// contains the character 'A' or starts with "Foo.".
|
// contains the character 'A' or starts with "Foo.".
|
||||||
bool should_run() const { return should_run_; }
|
bool should_run() const { return should_run_; }
|
||||||
|
|
||||||
// Returns true if the test was filtered out by --gtest_filter
|
// Returns true iff this test will appear in the XML report.
|
||||||
bool filtered_out() const { return !matches_filter_; }
|
bool is_reportable() const {
|
||||||
|
// For now, the XML report includes all tests matching the filter.
|
||||||
|
// In the future, we may trim tests that are excluded because of
|
||||||
|
// sharding.
|
||||||
|
return matches_filter_;
|
||||||
|
}
|
||||||
|
|
||||||
// Returns the result of the test.
|
// Returns the result of the test.
|
||||||
const TestResult* result() const { return &result_; }
|
const TestResult* result() const { return &result_; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
||||||
#if GTEST_HAS_DEATH_TEST
|
#if GTEST_HAS_DEATH_TEST
|
||||||
friend class internal::DefaultDeathTestFactory;
|
friend class internal::DefaultDeathTestFactory;
|
||||||
#endif // GTEST_HAS_DEATH_TEST
|
#endif // GTEST_HAS_DEATH_TEST
|
||||||
friend class Test;
|
friend class Test;
|
||||||
friend class TestCase;
|
friend class TestCase;
|
||||||
friend class internal::UnitTestImpl;
|
friend class internal::UnitTestImpl;
|
||||||
|
friend class internal::StreamingListenerTest;
|
||||||
friend TestInfo* internal::MakeAndRegisterTestInfo(
|
friend TestInfo* internal::MakeAndRegisterTestInfo(
|
||||||
const char* test_case_name, const char* name,
|
const char* test_case_name,
|
||||||
|
const char* name,
|
||||||
const char* type_param,
|
const char* type_param,
|
||||||
const char* value_param,
|
const char* value_param,
|
||||||
internal::TypeId fixture_class_id,
|
internal::TypeId fixture_class_id,
|
||||||
@ -693,9 +719,10 @@ class GTEST_API_ TestInfo {
|
|||||||
|
|
||||||
// Constructs a TestInfo object. The newly constructed instance assumes
|
// Constructs a TestInfo object. The newly constructed instance assumes
|
||||||
// ownership of the factory object.
|
// ownership of the factory object.
|
||||||
TestInfo(const char* test_case_name, const char* name,
|
TestInfo(const std::string& test_case_name,
|
||||||
const char* a_type_param,
|
const std::string& name,
|
||||||
const char* a_value_param,
|
const char* a_type_param, // NULL if not a type-parameterized test
|
||||||
|
const char* a_value_param, // NULL if not a value-parameterized test
|
||||||
internal::TypeId fixture_class_id,
|
internal::TypeId fixture_class_id,
|
||||||
internal::TestFactoryBase* factory);
|
internal::TestFactoryBase* factory);
|
||||||
|
|
||||||
@ -775,18 +802,21 @@ class GTEST_API_ TestCase {
|
|||||||
// Returns true if any test in this test case should run.
|
// Returns true if any test in this test case should run.
|
||||||
bool should_run() const { return should_run_; }
|
bool should_run() const { return should_run_; }
|
||||||
|
|
||||||
// Returns true if this test case should be skipped in the report.
|
|
||||||
bool should_skip_report() const { return should_skip_report_; }
|
|
||||||
|
|
||||||
// Gets the number of successful tests in this test case.
|
// Gets the number of successful tests in this test case.
|
||||||
int successful_test_count() const;
|
int successful_test_count() const;
|
||||||
|
|
||||||
// Gets the number of failed tests in this test case.
|
// Gets the number of failed tests in this test case.
|
||||||
int failed_test_count() const;
|
int failed_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of disabled tests that will be reported in the XML report.
|
||||||
|
int reportable_disabled_test_count() const;
|
||||||
|
|
||||||
// Gets the number of disabled tests in this test case.
|
// Gets the number of disabled tests in this test case.
|
||||||
int disabled_test_count() const;
|
int disabled_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of tests to be printed in the XML report.
|
||||||
|
int reportable_test_count() const;
|
||||||
|
|
||||||
// Get the number of tests in this test case that should run.
|
// Get the number of tests in this test case that should run.
|
||||||
int test_to_run_count() const;
|
int test_to_run_count() const;
|
||||||
|
|
||||||
@ -806,6 +836,10 @@ class GTEST_API_ TestCase {
|
|||||||
// total_test_count() - 1. If i is not in that range, returns NULL.
|
// total_test_count() - 1. If i is not in that range, returns NULL.
|
||||||
const TestInfo* GetTestInfo(int i) const;
|
const TestInfo* GetTestInfo(int i) const;
|
||||||
|
|
||||||
|
// Returns the TestResult that holds test properties recorded during
|
||||||
|
// execution of SetUpTestCase and TearDownTestCase.
|
||||||
|
const TestResult& ad_hoc_test_result() const { return ad_hoc_test_result_; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
friend class Test;
|
friend class Test;
|
||||||
friend class internal::UnitTestImpl;
|
friend class internal::UnitTestImpl;
|
||||||
@ -824,7 +858,6 @@ class GTEST_API_ TestCase {
|
|||||||
|
|
||||||
// Sets the should_run member.
|
// Sets the should_run member.
|
||||||
void set_should_run(bool should) { should_run_ = should; }
|
void set_should_run(bool should) { should_run_ = should; }
|
||||||
void set_should_skip_report(bool should) { should_skip_report_ = should; }
|
|
||||||
|
|
||||||
// Adds a TestInfo to this test case. Will delete the TestInfo upon
|
// Adds a TestInfo to this test case. Will delete the TestInfo upon
|
||||||
// destruction of the TestCase object.
|
// destruction of the TestCase object.
|
||||||
@ -859,11 +892,22 @@ class GTEST_API_ TestCase {
|
|||||||
return test_info->should_run() && test_info->result()->Failed();
|
return test_info->should_run() && test_info->result()->Failed();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns true iff the test is disabled and will be reported in the XML
|
||||||
|
// report.
|
||||||
|
static bool TestReportableDisabled(const TestInfo* test_info) {
|
||||||
|
return test_info->is_reportable() && test_info->is_disabled_;
|
||||||
|
}
|
||||||
|
|
||||||
// Returns true iff test is disabled.
|
// Returns true iff test is disabled.
|
||||||
static bool TestDisabled(const TestInfo* test_info) {
|
static bool TestDisabled(const TestInfo* test_info) {
|
||||||
return test_info->is_disabled_;
|
return test_info->is_disabled_;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns true iff this test will appear in the XML report.
|
||||||
|
static bool TestReportable(const TestInfo* test_info) {
|
||||||
|
return test_info->is_reportable();
|
||||||
|
}
|
||||||
|
|
||||||
// Returns true if the given test should run.
|
// Returns true if the given test should run.
|
||||||
static bool ShouldRunTest(const TestInfo* test_info) {
|
static bool ShouldRunTest(const TestInfo* test_info) {
|
||||||
return test_info->should_run();
|
return test_info->should_run();
|
||||||
@ -876,7 +920,7 @@ class GTEST_API_ TestCase {
|
|||||||
void UnshuffleTests();
|
void UnshuffleTests();
|
||||||
|
|
||||||
// Name of the test case.
|
// Name of the test case.
|
||||||
internal::String name_;
|
std::string name_;
|
||||||
// Name of the parameter type, or NULL if this is not a typed or a
|
// Name of the parameter type, or NULL if this is not a typed or a
|
||||||
// type-parameterized test.
|
// type-parameterized test.
|
||||||
const internal::scoped_ptr<const ::std::string> type_param_;
|
const internal::scoped_ptr<const ::std::string> type_param_;
|
||||||
@ -893,17 +937,18 @@ class GTEST_API_ TestCase {
|
|||||||
Test::TearDownTestCaseFunc tear_down_tc_;
|
Test::TearDownTestCaseFunc tear_down_tc_;
|
||||||
// True iff any test in this test case should run.
|
// True iff any test in this test case should run.
|
||||||
bool should_run_;
|
bool should_run_;
|
||||||
// True if this test case should not be reported
|
|
||||||
bool should_skip_report_;
|
|
||||||
// Elapsed time, in milliseconds.
|
// Elapsed time, in milliseconds.
|
||||||
TimeInMillis elapsed_time_;
|
TimeInMillis elapsed_time_;
|
||||||
|
// Holds test properties recorded during execution of SetUpTestCase and
|
||||||
|
// TearDownTestCase.
|
||||||
|
TestResult ad_hoc_test_result_;
|
||||||
|
|
||||||
// We disallow copying TestCases.
|
// We disallow copying TestCases.
|
||||||
GTEST_DISALLOW_COPY_AND_ASSIGN_(TestCase);
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(TestCase);
|
||||||
};
|
};
|
||||||
|
|
||||||
// An Environment object is capable of setting up and tearing down an
|
// An Environment object is capable of setting up and tearing down an
|
||||||
// environment. The user should subclass this to define his own
|
// environment. You should subclass this to define your own
|
||||||
// environment(s).
|
// environment(s).
|
||||||
//
|
//
|
||||||
// An Environment object does the set-up and tear-down in virtual
|
// An Environment object does the set-up and tear-down in virtual
|
||||||
@ -1116,11 +1161,13 @@ class GTEST_API_ UnitTest {
|
|||||||
|
|
||||||
// Returns the TestCase object for the test that's currently running,
|
// Returns the TestCase object for the test that's currently running,
|
||||||
// or NULL if no test is running.
|
// or NULL if no test is running.
|
||||||
const TestCase* current_test_case() const;
|
const TestCase* current_test_case() const
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// Returns the TestInfo object for the test that's currently running,
|
// Returns the TestInfo object for the test that's currently running,
|
||||||
// or NULL if no test is running.
|
// or NULL if no test is running.
|
||||||
const TestInfo* current_test_info() const;
|
const TestInfo* current_test_info() const
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// Returns the random seed used at the start of the current test run.
|
// Returns the random seed used at the start of the current test run.
|
||||||
int random_seed() const;
|
int random_seed() const;
|
||||||
@ -1130,7 +1177,8 @@ class GTEST_API_ UnitTest {
|
|||||||
// value-parameterized tests and instantiate and register them.
|
// value-parameterized tests and instantiate and register them.
|
||||||
//
|
//
|
||||||
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
||||||
internal::ParameterizedTestCaseRegistry& parameterized_test_registry();
|
internal::ParameterizedTestCaseRegistry& parameterized_test_registry()
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
#endif // GTEST_HAS_PARAM_TEST
|
#endif // GTEST_HAS_PARAM_TEST
|
||||||
|
|
||||||
// Gets the number of successful test cases.
|
// Gets the number of successful test cases.
|
||||||
@ -1152,15 +1200,25 @@ class GTEST_API_ UnitTest {
|
|||||||
// Gets the number of failed tests.
|
// Gets the number of failed tests.
|
||||||
int failed_test_count() const;
|
int failed_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of disabled tests that will be reported in the XML report.
|
||||||
|
int reportable_disabled_test_count() const;
|
||||||
|
|
||||||
// Gets the number of disabled tests.
|
// Gets the number of disabled tests.
|
||||||
int disabled_test_count() const;
|
int disabled_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of tests to be printed in the XML report.
|
||||||
|
int reportable_test_count() const;
|
||||||
|
|
||||||
// Gets the number of all tests.
|
// Gets the number of all tests.
|
||||||
int total_test_count() const;
|
int total_test_count() const;
|
||||||
|
|
||||||
// Gets the number of tests that should run.
|
// Gets the number of tests that should run.
|
||||||
int test_to_run_count() const;
|
int test_to_run_count() const;
|
||||||
|
|
||||||
|
// Gets the time of the test program start, in ms from the start of the
|
||||||
|
// UNIX epoch.
|
||||||
|
TimeInMillis start_timestamp() const;
|
||||||
|
|
||||||
// Gets the elapsed time, in milliseconds.
|
// Gets the elapsed time, in milliseconds.
|
||||||
TimeInMillis elapsed_time() const;
|
TimeInMillis elapsed_time() const;
|
||||||
|
|
||||||
@ -1175,6 +1233,10 @@ class GTEST_API_ UnitTest {
|
|||||||
// total_test_case_count() - 1. If i is not in that range, returns NULL.
|
// total_test_case_count() - 1. If i is not in that range, returns NULL.
|
||||||
const TestCase* GetTestCase(int i) const;
|
const TestCase* GetTestCase(int i) const;
|
||||||
|
|
||||||
|
// Returns the TestResult containing information on test failures and
|
||||||
|
// properties logged outside of individual test cases.
|
||||||
|
const TestResult& ad_hoc_test_result() const;
|
||||||
|
|
||||||
// Returns the list of event listeners that can be used to track events
|
// Returns the list of event listeners that can be used to track events
|
||||||
// inside Google Test.
|
// inside Google Test.
|
||||||
TestEventListeners& listeners();
|
TestEventListeners& listeners();
|
||||||
@ -1198,12 +1260,16 @@ class GTEST_API_ UnitTest {
|
|||||||
void AddTestPartResult(TestPartResult::Type result_type,
|
void AddTestPartResult(TestPartResult::Type result_type,
|
||||||
const char* file_name,
|
const char* file_name,
|
||||||
int line_number,
|
int line_number,
|
||||||
const internal::String& message,
|
const std::string& message,
|
||||||
const internal::String& os_stack_trace);
|
const std::string& os_stack_trace)
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// Adds a TestProperty to the current TestResult object. If the result already
|
// Adds a TestProperty to the current TestResult object when invoked from
|
||||||
// contains a property with the same key, the value will be updated.
|
// inside a test, to current TestCase's ad_hoc_test_result_ when invoked
|
||||||
void RecordPropertyForCurrentTest(const char* key, const char* value);
|
// from SetUpTestCase or TearDownTestCase, or to the global property set
|
||||||
|
// when invoked elsewhere. If the result already contains a property with
|
||||||
|
// the same key, the value will be updated.
|
||||||
|
void RecordProperty(const std::string& key, const std::string& value);
|
||||||
|
|
||||||
// Gets the i-th test case among all the test cases. i can range from 0 to
|
// Gets the i-th test case among all the test cases. i can range from 0 to
|
||||||
// total_test_case_count() - 1. If i is not in that range, returns NULL.
|
// total_test_case_count() - 1. If i is not in that range, returns NULL.
|
||||||
@ -1218,11 +1284,13 @@ class GTEST_API_ UnitTest {
|
|||||||
friend class Test;
|
friend class Test;
|
||||||
friend class internal::AssertHelper;
|
friend class internal::AssertHelper;
|
||||||
friend class internal::ScopedTrace;
|
friend class internal::ScopedTrace;
|
||||||
|
friend class internal::StreamingListenerTest;
|
||||||
|
friend class internal::UnitTestRecordPropertyTestHelper;
|
||||||
friend Environment* AddGlobalTestEnvironment(Environment* env);
|
friend Environment* AddGlobalTestEnvironment(Environment* env);
|
||||||
friend internal::UnitTestImpl* internal::GetUnitTestImpl();
|
friend internal::UnitTestImpl* internal::GetUnitTestImpl();
|
||||||
friend void internal::ReportFailureInUnknownLocation(
|
friend void internal::ReportFailureInUnknownLocation(
|
||||||
TestPartResult::Type result_type,
|
TestPartResult::Type result_type,
|
||||||
const internal::String& message);
|
const std::string& message);
|
||||||
|
|
||||||
// Creates an empty UnitTest.
|
// Creates an empty UnitTest.
|
||||||
UnitTest();
|
UnitTest();
|
||||||
@ -1232,10 +1300,12 @@ class GTEST_API_ UnitTest {
|
|||||||
|
|
||||||
// Pushes a trace defined by SCOPED_TRACE() on to the per-thread
|
// Pushes a trace defined by SCOPED_TRACE() on to the per-thread
|
||||||
// Google Test trace stack.
|
// Google Test trace stack.
|
||||||
void PushGTestTrace(const internal::TraceInfo& trace);
|
void PushGTestTrace(const internal::TraceInfo& trace)
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// Pops a trace from the per-thread Google Test trace stack.
|
// Pops a trace from the per-thread Google Test trace stack.
|
||||||
void PopGTestTrace();
|
void PopGTestTrace()
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// Protects mutable state in *impl_. This is mutable as some const
|
// Protects mutable state in *impl_. This is mutable as some const
|
||||||
// methods need to lock it too.
|
// methods need to lock it too.
|
||||||
@ -1290,24 +1360,115 @@ GTEST_API_ void InitGoogleTest(int* argc, wchar_t** argv);
|
|||||||
|
|
||||||
namespace internal {
|
namespace internal {
|
||||||
|
|
||||||
|
// FormatForComparison<ToPrint, OtherOperand>::Format(value) formats a
|
||||||
|
// value of type ToPrint that is an operand of a comparison assertion
|
||||||
|
// (e.g. ASSERT_EQ). OtherOperand is the type of the other operand in
|
||||||
|
// the comparison, and is used to help determine the best way to
|
||||||
|
// format the value. In particular, when the value is a C string
|
||||||
|
// (char pointer) and the other operand is an STL string object, we
|
||||||
|
// want to format the C string as a string, since we know it is
|
||||||
|
// compared by value with the string object. If the value is a char
|
||||||
|
// pointer but the other operand is not an STL string object, we don't
|
||||||
|
// know whether the pointer is supposed to point to a NUL-terminated
|
||||||
|
// string, and thus want to print it as a pointer to be safe.
|
||||||
|
//
|
||||||
|
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
||||||
|
|
||||||
|
// The default case.
|
||||||
|
template <typename ToPrint, typename OtherOperand>
|
||||||
|
class FormatForComparison {
|
||||||
|
public:
|
||||||
|
static ::std::string Format(const ToPrint& value) {
|
||||||
|
return ::testing::PrintToString(value);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Array.
|
||||||
|
template <typename ToPrint, size_t N, typename OtherOperand>
|
||||||
|
class FormatForComparison<ToPrint[N], OtherOperand> {
|
||||||
|
public:
|
||||||
|
static ::std::string Format(const ToPrint* value) {
|
||||||
|
return FormatForComparison<const ToPrint*, OtherOperand>::Format(value);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// By default, print C string as pointers to be safe, as we don't know
|
||||||
|
// whether they actually point to a NUL-terminated string.
|
||||||
|
|
||||||
|
#define GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_(CharType) \
|
||||||
|
template <typename OtherOperand> \
|
||||||
|
class FormatForComparison<CharType*, OtherOperand> { \
|
||||||
|
public: \
|
||||||
|
static ::std::string Format(CharType* value) { \
|
||||||
|
return ::testing::PrintToString(static_cast<const void*>(value)); \
|
||||||
|
} \
|
||||||
|
}
|
||||||
|
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_(char);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_(const char);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_(wchar_t);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_(const wchar_t);
|
||||||
|
|
||||||
|
#undef GTEST_IMPL_FORMAT_C_STRING_AS_POINTER_
|
||||||
|
|
||||||
|
// If a C string is compared with an STL string object, we know it's meant
|
||||||
|
// to point to a NUL-terminated string, and thus can print it as a string.
|
||||||
|
|
||||||
|
#define GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(CharType, OtherStringType) \
|
||||||
|
template <> \
|
||||||
|
class FormatForComparison<CharType*, OtherStringType> { \
|
||||||
|
public: \
|
||||||
|
static ::std::string Format(CharType* value) { \
|
||||||
|
return ::testing::PrintToString(value); \
|
||||||
|
} \
|
||||||
|
}
|
||||||
|
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(char, ::std::string);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(const char, ::std::string);
|
||||||
|
|
||||||
|
#if GTEST_HAS_GLOBAL_STRING
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(char, ::string);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(const char, ::string);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if GTEST_HAS_GLOBAL_WSTRING
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(wchar_t, ::wstring);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(const wchar_t, ::wstring);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if GTEST_HAS_STD_WSTRING
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(wchar_t, ::std::wstring);
|
||||||
|
GTEST_IMPL_FORMAT_C_STRING_AS_STRING_(const wchar_t, ::std::wstring);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#undef GTEST_IMPL_FORMAT_C_STRING_AS_STRING_
|
||||||
|
|
||||||
// Formats a comparison assertion (e.g. ASSERT_EQ, EXPECT_LT, and etc)
|
// Formats a comparison assertion (e.g. ASSERT_EQ, EXPECT_LT, and etc)
|
||||||
// operand to be used in a failure message. The type (but not value)
|
// operand to be used in a failure message. The type (but not value)
|
||||||
// of the other operand may affect the format. This allows us to
|
// of the other operand may affect the format. This allows us to
|
||||||
// print a char* as a raw pointer when it is compared against another
|
// print a char* as a raw pointer when it is compared against another
|
||||||
// char*, and print it as a C string when it is compared against an
|
// char* or void*, and print it as a C string when it is compared
|
||||||
// std::string object, for example.
|
// against an std::string object, for example.
|
||||||
//
|
|
||||||
// The default implementation ignores the type of the other operand.
|
|
||||||
// Some specialized versions are used to handle formatting wide or
|
|
||||||
// narrow C strings.
|
|
||||||
//
|
//
|
||||||
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
||||||
template <typename T1, typename T2>
|
template <typename T1, typename T2>
|
||||||
String FormatForComparisonFailureMessage(const T1& value,
|
std::string FormatForComparisonFailureMessage(
|
||||||
const T2& /* other_operand */) {
|
const T1& value, const T2& /* other_operand */) {
|
||||||
// C++Builder compiles this incorrectly if the namespace isn't explicitly
|
return FormatForComparison<T1, T2>::Format(value);
|
||||||
// given.
|
}
|
||||||
return ::testing::PrintToString(value);
|
|
||||||
|
// Separate the error generating code from the code path to reduce the stack
|
||||||
|
// frame size of CmpHelperEQ. This helps reduce the overhead of some sanitizers
|
||||||
|
// when calling EXPECT_* in a tight loop.
|
||||||
|
template <typename T1, typename T2>
|
||||||
|
AssertionResult CmpHelperEQFailure(const char* expected_expression,
|
||||||
|
const char* actual_expression,
|
||||||
|
const T1& expected, const T2& actual) {
|
||||||
|
return EqFailure(expected_expression,
|
||||||
|
actual_expression,
|
||||||
|
FormatForComparisonFailureMessage(expected, actual),
|
||||||
|
FormatForComparisonFailureMessage(actual, expected),
|
||||||
|
false);
|
||||||
}
|
}
|
||||||
|
|
||||||
// The helper function for {ASSERT|EXPECT}_EQ.
|
// The helper function for {ASSERT|EXPECT}_EQ.
|
||||||
@ -1316,25 +1477,14 @@ AssertionResult CmpHelperEQ(const char* expected_expression,
|
|||||||
const char* actual_expression,
|
const char* actual_expression,
|
||||||
const T1& expected,
|
const T1& expected,
|
||||||
const T2& actual) {
|
const T2& actual) {
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4389 /* signed/unsigned mismatch */)
|
||||||
# pragma warning(push) // Saves the current warning state.
|
|
||||||
# pragma warning(disable:4389) // Temporarily disables warning on
|
|
||||||
// signed/unsigned mismatch.
|
|
||||||
#endif
|
|
||||||
|
|
||||||
if (expected == actual) {
|
if (expected == actual) {
|
||||||
return AssertionSuccess();
|
return AssertionSuccess();
|
||||||
}
|
}
|
||||||
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
|
|
||||||
#ifdef _MSC_VER
|
return CmpHelperEQFailure(expected_expression, actual_expression, expected,
|
||||||
# pragma warning(pop) // Restores the warning state.
|
actual);
|
||||||
#endif
|
|
||||||
|
|
||||||
return EqFailure(expected_expression,
|
|
||||||
actual_expression,
|
|
||||||
FormatForComparisonFailureMessage(expected, actual),
|
|
||||||
FormatForComparisonFailureMessage(actual, expected),
|
|
||||||
false);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// With this overloaded version, we allow anonymous enums to be used
|
// With this overloaded version, we allow anonymous enums to be used
|
||||||
@ -1422,6 +1572,19 @@ class EqHelper<true> {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Separate the error generating code from the code path to reduce the stack
|
||||||
|
// frame size of CmpHelperOP. This helps reduce the overhead of some sanitizers
|
||||||
|
// when calling EXPECT_OP in a tight loop.
|
||||||
|
template <typename T1, typename T2>
|
||||||
|
AssertionResult CmpHelperOpFailure(const char* expr1, const char* expr2,
|
||||||
|
const T1& val1, const T2& val2,
|
||||||
|
const char* op) {
|
||||||
|
return AssertionFailure()
|
||||||
|
<< "Expected: (" << expr1 << ") " << op << " (" << expr2
|
||||||
|
<< "), actual: " << FormatForComparisonFailureMessage(val1, val2)
|
||||||
|
<< " vs " << FormatForComparisonFailureMessage(val2, val1);
|
||||||
|
}
|
||||||
|
|
||||||
// A macro for implementing the helper functions needed to implement
|
// A macro for implementing the helper functions needed to implement
|
||||||
// ASSERT_?? and EXPECT_??. It is here just to avoid copy-and-paste
|
// ASSERT_?? and EXPECT_??. It is here just to avoid copy-and-paste
|
||||||
// of similar code.
|
// of similar code.
|
||||||
@ -1432,6 +1595,7 @@ class EqHelper<true> {
|
|||||||
// with gcc 4.
|
// with gcc 4.
|
||||||
//
|
//
|
||||||
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
// INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM.
|
||||||
|
|
||||||
#define GTEST_IMPL_CMP_HELPER_(op_name, op)\
|
#define GTEST_IMPL_CMP_HELPER_(op_name, op)\
|
||||||
template <typename T1, typename T2>\
|
template <typename T1, typename T2>\
|
||||||
AssertionResult CmpHelper##op_name(const char* expr1, const char* expr2, \
|
AssertionResult CmpHelper##op_name(const char* expr1, const char* expr2, \
|
||||||
@ -1439,10 +1603,7 @@ AssertionResult CmpHelper##op_name(const char* expr1, const char* expr2, \
|
|||||||
if (val1 op val2) {\
|
if (val1 op val2) {\
|
||||||
return AssertionSuccess();\
|
return AssertionSuccess();\
|
||||||
} else {\
|
} else {\
|
||||||
return AssertionFailure() \
|
return CmpHelperOpFailure(expr1, expr2, val1, val2, #op);\
|
||||||
<< "Expected: (" << expr1 << ") " #op " (" << expr2\
|
|
||||||
<< "), actual: " << FormatForComparisonFailureMessage(val1, val2)\
|
|
||||||
<< " vs " << FormatForComparisonFailureMessage(val2, val1);\
|
|
||||||
}\
|
}\
|
||||||
}\
|
}\
|
||||||
GTEST_API_ AssertionResult CmpHelper##op_name(\
|
GTEST_API_ AssertionResult CmpHelper##op_name(\
|
||||||
@ -1455,11 +1616,11 @@ GTEST_IMPL_CMP_HELPER_(NE, !=);
|
|||||||
// Implements the helper function for {ASSERT|EXPECT}_LE
|
// Implements the helper function for {ASSERT|EXPECT}_LE
|
||||||
GTEST_IMPL_CMP_HELPER_(LE, <=);
|
GTEST_IMPL_CMP_HELPER_(LE, <=);
|
||||||
// Implements the helper function for {ASSERT|EXPECT}_LT
|
// Implements the helper function for {ASSERT|EXPECT}_LT
|
||||||
GTEST_IMPL_CMP_HELPER_(LT, < );
|
GTEST_IMPL_CMP_HELPER_(LT, <);
|
||||||
// Implements the helper function for {ASSERT|EXPECT}_GE
|
// Implements the helper function for {ASSERT|EXPECT}_GE
|
||||||
GTEST_IMPL_CMP_HELPER_(GE, >=);
|
GTEST_IMPL_CMP_HELPER_(GE, >=);
|
||||||
// Implements the helper function for {ASSERT|EXPECT}_GT
|
// Implements the helper function for {ASSERT|EXPECT}_GT
|
||||||
GTEST_IMPL_CMP_HELPER_(GT, > );
|
GTEST_IMPL_CMP_HELPER_(GT, >);
|
||||||
|
|
||||||
#undef GTEST_IMPL_CMP_HELPER_
|
#undef GTEST_IMPL_CMP_HELPER_
|
||||||
|
|
||||||
@ -1625,7 +1786,7 @@ class GTEST_API_ AssertHelper {
|
|||||||
TestPartResult::Type const type;
|
TestPartResult::Type const type;
|
||||||
const char* const file;
|
const char* const file;
|
||||||
int const line;
|
int const line;
|
||||||
String const message;
|
std::string const message;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
GTEST_DISALLOW_COPY_AND_ASSIGN_(AssertHelperData);
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(AssertHelperData);
|
||||||
@ -1684,7 +1845,12 @@ class WithParamInterface {
|
|||||||
// references static data, to reduce the opportunity for incorrect uses
|
// references static data, to reduce the opportunity for incorrect uses
|
||||||
// like writing 'WithParamInterface<bool>::GetParam()' for a test that
|
// like writing 'WithParamInterface<bool>::GetParam()' for a test that
|
||||||
// uses a fixture whose parameter type is int.
|
// uses a fixture whose parameter type is int.
|
||||||
const ParamType& GetParam() const { return *parameter_; }
|
const ParamType& GetParam() const {
|
||||||
|
GTEST_CHECK_(parameter_ != NULL)
|
||||||
|
<< "GetParam() can only be called inside a value-parameterized test "
|
||||||
|
<< "-- did you intend to write TEST_P instead of TEST_F?";
|
||||||
|
return *parameter_;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Sets parameter value. The caller is responsible for making sure the value
|
// Sets parameter value. The caller is responsible for making sure the value
|
||||||
@ -1730,12 +1896,6 @@ class TestWithParam : public Test, public WithParamInterface<T> {
|
|||||||
// usually want the fail-fast behavior of FAIL and ASSERT_*, but those
|
// usually want the fail-fast behavior of FAIL and ASSERT_*, but those
|
||||||
// writing data-driven tests often find themselves using ADD_FAILURE
|
// writing data-driven tests often find themselves using ADD_FAILURE
|
||||||
// and EXPECT_* more.
|
// and EXPECT_* more.
|
||||||
//
|
|
||||||
// Examples:
|
|
||||||
//
|
|
||||||
// EXPECT_TRUE(server.StatusIsOK());
|
|
||||||
// ASSERT_FALSE(server.HasPendingRequest(port))
|
|
||||||
// << "There are still pending requests " << "on port " << port;
|
|
||||||
|
|
||||||
// Generates a nonfatal failure with a generic message.
|
// Generates a nonfatal failure with a generic message.
|
||||||
#define ADD_FAILURE() GTEST_NONFATAL_FAILURE_("Failed")
|
#define ADD_FAILURE() GTEST_NONFATAL_FAILURE_("Failed")
|
||||||
@ -1909,7 +2069,7 @@ class TestWithParam : public Test, public WithParamInterface<T> {
|
|||||||
# define ASSERT_GT(val1, val2) GTEST_ASSERT_GT(val1, val2)
|
# define ASSERT_GT(val1, val2) GTEST_ASSERT_GT(val1, val2)
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// C String Comparisons. All tests treat NULL and any non-NULL string
|
// C-string Comparisons. All tests treat NULL and any non-NULL string
|
||||||
// as different. Two NULLs are equal.
|
// as different. Two NULLs are equal.
|
||||||
//
|
//
|
||||||
// * {ASSERT|EXPECT}_STREQ(s1, s2): Tests that s1 == s2
|
// * {ASSERT|EXPECT}_STREQ(s1, s2): Tests that s1 == s2
|
||||||
@ -2093,8 +2253,8 @@ bool StaticAssertTypeEq() {
|
|||||||
// The convention is to end the test case name with "Test". For
|
// The convention is to end the test case name with "Test". For
|
||||||
// example, a test case for the Foo class can be named FooTest.
|
// example, a test case for the Foo class can be named FooTest.
|
||||||
//
|
//
|
||||||
// The user should put his test code between braces after using this
|
// Test code should appear between braces after an invocation of
|
||||||
// macro. Example:
|
// this macro. Example:
|
||||||
//
|
//
|
||||||
// TEST(FooTest, InitializesCorrectly) {
|
// TEST(FooTest, InitializesCorrectly) {
|
||||||
// Foo foo;
|
// Foo foo;
|
||||||
@ -2150,15 +2310,20 @@ bool StaticAssertTypeEq() {
|
|||||||
GTEST_TEST_(test_fixture, test_name, test_fixture, \
|
GTEST_TEST_(test_fixture, test_name, test_fixture, \
|
||||||
::testing::internal::GetTypeId<test_fixture>())
|
::testing::internal::GetTypeId<test_fixture>())
|
||||||
|
|
||||||
// Use this macro in main() to run all tests. It returns 0 if all
|
} // namespace testing
|
||||||
|
|
||||||
|
// Use this function in main() to run all tests. It returns 0 if all
|
||||||
// tests are successful, or 1 otherwise.
|
// tests are successful, or 1 otherwise.
|
||||||
//
|
//
|
||||||
// RUN_ALL_TESTS() should be invoked after the command line has been
|
// RUN_ALL_TESTS() should be invoked after the command line has been
|
||||||
// parsed by InitGoogleTest().
|
// parsed by InitGoogleTest().
|
||||||
|
//
|
||||||
|
// This function was formerly a macro; thus, it is in the global
|
||||||
|
// namespace and has an all-caps name.
|
||||||
|
int RUN_ALL_TESTS() GTEST_MUST_USE_RESULT_;
|
||||||
|
|
||||||
#define RUN_ALL_TESTS()\
|
inline int RUN_ALL_TESTS() {
|
||||||
(::testing::UnitTest::GetInstance()->Run())
|
return ::testing::UnitTest::GetInstance()->Run();
|
||||||
|
}
|
||||||
} // namespace testing
|
|
||||||
|
|
||||||
#endif // GTEST_INCLUDE_GTEST_GTEST_H_
|
#endif // GTEST_INCLUDE_GTEST_GTEST_H_
|
||||||
|
@ -27,7 +27,7 @@
|
|||||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
// This file is AUTOMATICALLY GENERATED on 09/24/2010 by command
|
// This file is AUTOMATICALLY GENERATED on 10/31/2011 by command
|
||||||
// 'gen_gtest_pred_impl.py 5'. DO NOT EDIT BY HAND!
|
// 'gen_gtest_pred_impl.py 5'. DO NOT EDIT BY HAND!
|
||||||
//
|
//
|
||||||
// Implements a family of generic predicate assertion macros.
|
// Implements a family of generic predicate assertion macros.
|
||||||
@ -98,7 +98,7 @@ AssertionResult AssertPred1Helper(const char* pred_text,
|
|||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT1.
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT1.
|
||||||
// Don't use this in your code.
|
// Don't use this in your code.
|
||||||
#define GTEST_PRED_FORMAT1_(pred_format, v1, on_failure)\
|
#define GTEST_PRED_FORMAT1_(pred_format, v1, on_failure)\
|
||||||
GTEST_ASSERT_(pred_format(#v1, v1),\
|
GTEST_ASSERT_(pred_format(#v1, v1), \
|
||||||
on_failure)
|
on_failure)
|
||||||
|
|
||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED1. Don't use
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED1. Don't use
|
||||||
@ -144,7 +144,7 @@ AssertionResult AssertPred2Helper(const char* pred_text,
|
|||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT2.
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT2.
|
||||||
// Don't use this in your code.
|
// Don't use this in your code.
|
||||||
#define GTEST_PRED_FORMAT2_(pred_format, v1, v2, on_failure)\
|
#define GTEST_PRED_FORMAT2_(pred_format, v1, v2, on_failure)\
|
||||||
GTEST_ASSERT_(pred_format(#v1, #v2, v1, v2),\
|
GTEST_ASSERT_(pred_format(#v1, #v2, v1, v2), \
|
||||||
on_failure)
|
on_failure)
|
||||||
|
|
||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED2. Don't use
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED2. Don't use
|
||||||
@ -197,7 +197,7 @@ AssertionResult AssertPred3Helper(const char* pred_text,
|
|||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT3.
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT3.
|
||||||
// Don't use this in your code.
|
// Don't use this in your code.
|
||||||
#define GTEST_PRED_FORMAT3_(pred_format, v1, v2, v3, on_failure)\
|
#define GTEST_PRED_FORMAT3_(pred_format, v1, v2, v3, on_failure)\
|
||||||
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, v1, v2, v3),\
|
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, v1, v2, v3), \
|
||||||
on_failure)
|
on_failure)
|
||||||
|
|
||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED3. Don't use
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED3. Don't use
|
||||||
@ -257,7 +257,7 @@ AssertionResult AssertPred4Helper(const char* pred_text,
|
|||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT4.
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT4.
|
||||||
// Don't use this in your code.
|
// Don't use this in your code.
|
||||||
#define GTEST_PRED_FORMAT4_(pred_format, v1, v2, v3, v4, on_failure)\
|
#define GTEST_PRED_FORMAT4_(pred_format, v1, v2, v3, v4, on_failure)\
|
||||||
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, #v4, v1, v2, v3, v4),\
|
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, #v4, v1, v2, v3, v4), \
|
||||||
on_failure)
|
on_failure)
|
||||||
|
|
||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED4. Don't use
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED4. Don't use
|
||||||
@ -324,7 +324,7 @@ AssertionResult AssertPred5Helper(const char* pred_text,
|
|||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT5.
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT5.
|
||||||
// Don't use this in your code.
|
// Don't use this in your code.
|
||||||
#define GTEST_PRED_FORMAT5_(pred_format, v1, v2, v3, v4, v5, on_failure)\
|
#define GTEST_PRED_FORMAT5_(pred_format, v1, v2, v3, v4, v5, on_failure)\
|
||||||
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, #v4, #v5, v1, v2, v3, v4, v5),\
|
GTEST_ASSERT_(pred_format(#v1, #v2, #v3, #v4, #v5, v1, v2, v3, v4, v5), \
|
||||||
on_failure)
|
on_failure)
|
||||||
|
|
||||||
// Internal macro for implementing {EXPECT|ASSERT}_PRED5. Don't use
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED5. Don't use
|
||||||
|
@ -127,11 +127,11 @@ class GTEST_API_ DeathTest {
|
|||||||
// the last death test.
|
// the last death test.
|
||||||
static const char* LastMessage();
|
static const char* LastMessage();
|
||||||
|
|
||||||
static void set_last_death_test_message(const String& message);
|
static void set_last_death_test_message(const std::string& message);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// A string containing a description of the outcome of the last death test.
|
// A string containing a description of the outcome of the last death test.
|
||||||
static String last_death_test_message_;
|
static std::string last_death_test_message_;
|
||||||
|
|
||||||
GTEST_DISALLOW_COPY_AND_ASSIGN_(DeathTest);
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(DeathTest);
|
||||||
};
|
};
|
||||||
@ -217,12 +217,23 @@ GTEST_API_ bool ExitedUnsuccessfully(int exit_status);
|
|||||||
// The symbol "fail" here expands to something into which a message
|
// The symbol "fail" here expands to something into which a message
|
||||||
// can be streamed.
|
// can be streamed.
|
||||||
|
|
||||||
|
// This macro is for implementing ASSERT/EXPECT_DEBUG_DEATH when compiled in
|
||||||
|
// NDEBUG mode. In this case we need the statements to be executed, the regex is
|
||||||
|
// ignored, and the macro must accept a streamed message even though the message
|
||||||
|
// is never printed.
|
||||||
|
# define GTEST_EXECUTE_STATEMENT_(statement, regex) \
|
||||||
|
GTEST_AMBIGUOUS_ELSE_BLOCKER_ \
|
||||||
|
if (::testing::internal::AlwaysTrue()) { \
|
||||||
|
GTEST_SUPPRESS_UNREACHABLE_CODE_WARNING_BELOW_(statement); \
|
||||||
|
} else \
|
||||||
|
::testing::Message()
|
||||||
|
|
||||||
// A class representing the parsed contents of the
|
// A class representing the parsed contents of the
|
||||||
// --gtest_internal_run_death_test flag, as it existed when
|
// --gtest_internal_run_death_test flag, as it existed when
|
||||||
// RUN_ALL_TESTS was called.
|
// RUN_ALL_TESTS was called.
|
||||||
class InternalRunDeathTestFlag {
|
class InternalRunDeathTestFlag {
|
||||||
public:
|
public:
|
||||||
InternalRunDeathTestFlag(const String& a_file,
|
InternalRunDeathTestFlag(const std::string& a_file,
|
||||||
int a_line,
|
int a_line,
|
||||||
int an_index,
|
int an_index,
|
||||||
int a_write_fd)
|
int a_write_fd)
|
||||||
@ -234,13 +245,13 @@ class InternalRunDeathTestFlag {
|
|||||||
posix::Close(write_fd_);
|
posix::Close(write_fd_);
|
||||||
}
|
}
|
||||||
|
|
||||||
String file() const { return file_; }
|
const std::string& file() const { return file_; }
|
||||||
int line() const { return line_; }
|
int line() const { return line_; }
|
||||||
int index() const { return index_; }
|
int index() const { return index_; }
|
||||||
int write_fd() const { return write_fd_; }
|
int write_fd() const { return write_fd_; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
String file_;
|
std::string file_;
|
||||||
int line_;
|
int line_;
|
||||||
int index_;
|
int index_;
|
||||||
int write_fd_;
|
int write_fd_;
|
||||||
|
@ -61,11 +61,7 @@ class GTEST_API_ FilePath {
|
|||||||
FilePath() : pathname_("") { }
|
FilePath() : pathname_("") { }
|
||||||
FilePath(const FilePath& rhs) : pathname_(rhs.pathname_) { }
|
FilePath(const FilePath& rhs) : pathname_(rhs.pathname_) { }
|
||||||
|
|
||||||
explicit FilePath(const char* pathname) : pathname_(pathname) {
|
explicit FilePath(const std::string& pathname) : pathname_(pathname) {
|
||||||
Normalize();
|
|
||||||
}
|
|
||||||
|
|
||||||
explicit FilePath(const String& pathname) : pathname_(pathname) {
|
|
||||||
Normalize();
|
Normalize();
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -78,7 +74,7 @@ class GTEST_API_ FilePath {
|
|||||||
pathname_ = rhs.pathname_;
|
pathname_ = rhs.pathname_;
|
||||||
}
|
}
|
||||||
|
|
||||||
String ToString() const { return pathname_; }
|
const std::string& string() const { return pathname_; }
|
||||||
const char* c_str() const { return pathname_.c_str(); }
|
const char* c_str() const { return pathname_.c_str(); }
|
||||||
|
|
||||||
// Returns the current working directory, or "" if unsuccessful.
|
// Returns the current working directory, or "" if unsuccessful.
|
||||||
@ -111,8 +107,8 @@ class GTEST_API_ FilePath {
|
|||||||
const FilePath& base_name,
|
const FilePath& base_name,
|
||||||
const char* extension);
|
const char* extension);
|
||||||
|
|
||||||
// Returns true iff the path is NULL or "".
|
// Returns true iff the path is "".
|
||||||
bool IsEmpty() const { return c_str() == NULL || *c_str() == '\0'; }
|
bool IsEmpty() const { return pathname_.empty(); }
|
||||||
|
|
||||||
// If input name has a trailing separator character, removes it and returns
|
// If input name has a trailing separator character, removes it and returns
|
||||||
// the name, otherwise return the name string unmodified.
|
// the name, otherwise return the name string unmodified.
|
||||||
@ -201,7 +197,7 @@ class GTEST_API_ FilePath {
|
|||||||
// separators. Returns NULL if no path separator was found.
|
// separators. Returns NULL if no path separator was found.
|
||||||
const char* FindLastPathSeparator() const;
|
const char* FindLastPathSeparator() const;
|
||||||
|
|
||||||
String pathname_;
|
std::string pathname_;
|
||||||
}; // class FilePath
|
}; // class FilePath
|
||||||
|
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
|
@ -44,18 +44,22 @@
|
|||||||
# include <sys/types.h>
|
# include <sys/types.h>
|
||||||
# include <sys/wait.h>
|
# include <sys/wait.h>
|
||||||
# include <unistd.h>
|
# include <unistd.h>
|
||||||
# include <signal.h>
|
|
||||||
#endif // GTEST_OS_LINUX
|
#endif // GTEST_OS_LINUX
|
||||||
#if GTEST_CAN_STREAM_RESULTS_
|
|
||||||
# include <sys/socket.h>
|
#if GTEST_HAS_EXCEPTIONS
|
||||||
#endif // GTEST_CAN_STREAM_RESULTS_
|
# include <stdexcept>
|
||||||
|
#endif
|
||||||
|
|
||||||
#include <ctype.h>
|
#include <ctype.h>
|
||||||
|
#include <float.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <iomanip>
|
#include <iomanip>
|
||||||
#include <limits>
|
#include <limits>
|
||||||
#include <set>
|
#include <set>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "gtest/gtest-message.h"
|
||||||
#include "gtest/internal/gtest-string.h"
|
#include "gtest/internal/gtest-string.h"
|
||||||
#include "gtest/internal/gtest-filepath.h"
|
#include "gtest/internal/gtest-filepath.h"
|
||||||
#include "gtest/internal/gtest-type-util.h"
|
#include "gtest/internal/gtest-type-util.h"
|
||||||
@ -71,36 +75,6 @@
|
|||||||
#define GTEST_CONCAT_TOKEN_(foo, bar) GTEST_CONCAT_TOKEN_IMPL_(foo, bar)
|
#define GTEST_CONCAT_TOKEN_(foo, bar) GTEST_CONCAT_TOKEN_IMPL_(foo, bar)
|
||||||
#define GTEST_CONCAT_TOKEN_IMPL_(foo, bar) foo ## bar
|
#define GTEST_CONCAT_TOKEN_IMPL_(foo, bar) foo ## bar
|
||||||
|
|
||||||
// Google Test defines the testing::Message class to allow construction of
|
|
||||||
// test messages via the << operator. The idea is that anything
|
|
||||||
// streamable to std::ostream can be streamed to a testing::Message.
|
|
||||||
// This allows a user to use his own types in Google Test assertions by
|
|
||||||
// overloading the << operator.
|
|
||||||
//
|
|
||||||
// util/gtl/stl_logging-inl.h overloads << for STL containers. These
|
|
||||||
// overloads cannot be defined in the std namespace, as that will be
|
|
||||||
// undefined behavior. Therefore, they are defined in the global
|
|
||||||
// namespace instead.
|
|
||||||
//
|
|
||||||
// C++'s symbol lookup rule (i.e. Koenig lookup) says that these
|
|
||||||
// overloads are visible in either the std namespace or the global
|
|
||||||
// namespace, but not other namespaces, including the testing
|
|
||||||
// namespace which Google Test's Message class is in.
|
|
||||||
//
|
|
||||||
// To allow STL containers (and other types that has a << operator
|
|
||||||
// defined in the global namespace) to be used in Google Test assertions,
|
|
||||||
// testing::Message must access the custom << operator from the global
|
|
||||||
// namespace. Hence this helper function.
|
|
||||||
//
|
|
||||||
// Note: Jeffrey Yasskin suggested an alternative fix by "using
|
|
||||||
// ::operator<<;" in the definition of Message's operator<<. That fix
|
|
||||||
// doesn't require a helper function, but unfortunately doesn't
|
|
||||||
// compile with MSVC.
|
|
||||||
template <typename T>
|
|
||||||
inline void GTestStreamToHelper(std::ostream* os, const T& val) {
|
|
||||||
*os << val;
|
|
||||||
}
|
|
||||||
|
|
||||||
class ProtocolMessage;
|
class ProtocolMessage;
|
||||||
namespace proto2 { class Message; }
|
namespace proto2 { class Message; }
|
||||||
|
|
||||||
@ -126,17 +100,12 @@ class TestInfoImpl; // Opaque implementation of TestInfo
|
|||||||
class UnitTestImpl; // Opaque implementation of UnitTest
|
class UnitTestImpl; // Opaque implementation of UnitTest
|
||||||
|
|
||||||
// How many times InitGoogleTest() has been called.
|
// How many times InitGoogleTest() has been called.
|
||||||
extern int g_init_gtest_count;
|
GTEST_API_ extern int g_init_gtest_count;
|
||||||
|
|
||||||
// The text used in failure messages to indicate the start of the
|
// The text used in failure messages to indicate the start of the
|
||||||
// stack trace.
|
// stack trace.
|
||||||
GTEST_API_ extern const char kStackTraceMarker[];
|
GTEST_API_ extern const char kStackTraceMarker[];
|
||||||
|
|
||||||
// A secret type that Google Test users don't know about. It has no
|
|
||||||
// definition on purpose. Therefore it's impossible to create a
|
|
||||||
// Secret object, which is what we want.
|
|
||||||
class Secret;
|
|
||||||
|
|
||||||
// Two overloaded helpers for checking at compile time whether an
|
// Two overloaded helpers for checking at compile time whether an
|
||||||
// expression is a null pointer literal (i.e. NULL or any 0-valued
|
// expression is a null pointer literal (i.e. NULL or any 0-valued
|
||||||
// compile-time integral constant). Their return values have
|
// compile-time integral constant). Their return values have
|
||||||
@ -167,8 +136,23 @@ char (&IsNullLiteralHelper(...))[2]; // NOLINT
|
|||||||
#endif // GTEST_ELLIPSIS_NEEDS_POD_
|
#endif // GTEST_ELLIPSIS_NEEDS_POD_
|
||||||
|
|
||||||
// Appends the user-supplied message to the Google-Test-generated message.
|
// Appends the user-supplied message to the Google-Test-generated message.
|
||||||
GTEST_API_ String AppendUserMessage(const String& gtest_msg,
|
GTEST_API_ std::string AppendUserMessage(
|
||||||
const Message& user_msg);
|
const std::string& gtest_msg, const Message& user_msg);
|
||||||
|
|
||||||
|
#if GTEST_HAS_EXCEPTIONS
|
||||||
|
|
||||||
|
// This exception is thrown by (and only by) a failed Google Test
|
||||||
|
// assertion when GTEST_FLAG(throw_on_failure) is true (if exceptions
|
||||||
|
// are enabled). We derive it from std::runtime_error, which is for
|
||||||
|
// errors presumably detectable only at run time. Since
|
||||||
|
// std::runtime_error inherits from std::exception, many testing
|
||||||
|
// frameworks know how to extract and print the message inside it.
|
||||||
|
class GTEST_API_ GoogleTestFailureException : public ::std::runtime_error {
|
||||||
|
public:
|
||||||
|
explicit GoogleTestFailureException(const TestPartResult& failure);
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // GTEST_HAS_EXCEPTIONS
|
||||||
|
|
||||||
// A helper class for creating scoped traces in user programs.
|
// A helper class for creating scoped traces in user programs.
|
||||||
class GTEST_API_ ScopedTrace {
|
class GTEST_API_ ScopedTrace {
|
||||||
@ -189,76 +173,35 @@ class GTEST_API_ ScopedTrace {
|
|||||||
// c'tor and d'tor. Therefore it doesn't
|
// c'tor and d'tor. Therefore it doesn't
|
||||||
// need to be used otherwise.
|
// need to be used otherwise.
|
||||||
|
|
||||||
// Converts a streamable value to a String. A NULL pointer is
|
namespace edit_distance {
|
||||||
// converted to "(null)". When the input value is a ::string,
|
// Returns the optimal edits to go from 'left' to 'right'.
|
||||||
// ::std::string, ::wstring, or ::std::wstring object, each NUL
|
// All edits cost the same, with replace having lower priority than
|
||||||
// character in it is replaced with "\\0".
|
// add/remove.
|
||||||
// Declared here but defined in gtest.h, so that it has access
|
// Simple implementation of the Wagner–Fischer algorithm.
|
||||||
// to the definition of the Message class, required by the ARM
|
// See http://en.wikipedia.org/wiki/Wagner-Fischer_algorithm
|
||||||
// compiler.
|
enum EditType { kMatch, kAdd, kRemove, kReplace };
|
||||||
template <typename T>
|
GTEST_API_ std::vector<EditType> CalculateOptimalEdits(
|
||||||
String StreamableToString(const T& streamable);
|
const std::vector<size_t>& left, const std::vector<size_t>& right);
|
||||||
|
|
||||||
// The Symbian compiler has a bug that prevents it from selecting the
|
// Same as above, but the input is represented as strings.
|
||||||
// correct overload of FormatForComparisonFailureMessage (see below)
|
GTEST_API_ std::vector<EditType> CalculateOptimalEdits(
|
||||||
// unless we pass the first argument by reference. If we do that,
|
const std::vector<std::string>& left,
|
||||||
// however, Visual Age C++ 10.1 generates a compiler error. Therefore
|
const std::vector<std::string>& right);
|
||||||
// we only apply the work-around for Symbian.
|
|
||||||
#if defined(__SYMBIAN32__)
|
|
||||||
# define GTEST_CREF_WORKAROUND_ const&
|
|
||||||
#else
|
|
||||||
# define GTEST_CREF_WORKAROUND_
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// When this operand is a const char* or char*, if the other operand
|
// Create a diff of the input strings in Unified diff format.
|
||||||
// is a ::std::string or ::string, we print this operand as a C string
|
GTEST_API_ std::string CreateUnifiedDiff(const std::vector<std::string>& left,
|
||||||
// rather than a pointer (we do the same for wide strings); otherwise
|
const std::vector<std::string>& right,
|
||||||
// we print it as a pointer to be safe.
|
size_t context = 2);
|
||||||
|
|
||||||
// This internal macro is used to avoid duplicated code.
|
} // namespace edit_distance
|
||||||
#define GTEST_FORMAT_IMPL_(operand2_type, operand1_printer)\
|
|
||||||
inline String FormatForComparisonFailureMessage(\
|
|
||||||
operand2_type::value_type* GTEST_CREF_WORKAROUND_ str, \
|
|
||||||
const operand2_type& /*operand2*/) {\
|
|
||||||
return operand1_printer(str);\
|
|
||||||
}\
|
|
||||||
inline String FormatForComparisonFailureMessage(\
|
|
||||||
const operand2_type::value_type* GTEST_CREF_WORKAROUND_ str, \
|
|
||||||
const operand2_type& /*operand2*/) {\
|
|
||||||
return operand1_printer(str);\
|
|
||||||
}
|
|
||||||
|
|
||||||
GTEST_FORMAT_IMPL_(::std::string, String::ShowCStringQuoted)
|
// Calculate the diff between 'left' and 'right' and return it in unified diff
|
||||||
#if GTEST_HAS_STD_WSTRING
|
// format.
|
||||||
GTEST_FORMAT_IMPL_(::std::wstring, String::ShowWideCStringQuoted)
|
// If not null, stores in 'total_line_count' the total number of lines found
|
||||||
#endif // GTEST_HAS_STD_WSTRING
|
// in left + right.
|
||||||
|
GTEST_API_ std::string DiffStrings(const std::string& left,
|
||||||
#if GTEST_HAS_GLOBAL_STRING
|
const std::string& right,
|
||||||
GTEST_FORMAT_IMPL_(::string, String::ShowCStringQuoted)
|
size_t* total_line_count);
|
||||||
#endif // GTEST_HAS_GLOBAL_STRING
|
|
||||||
#if GTEST_HAS_GLOBAL_WSTRING
|
|
||||||
GTEST_FORMAT_IMPL_(::wstring, String::ShowWideCStringQuoted)
|
|
||||||
#endif // GTEST_HAS_GLOBAL_WSTRING
|
|
||||||
|
|
||||||
#undef GTEST_FORMAT_IMPL_
|
|
||||||
|
|
||||||
// The next four overloads handle the case where the operand being
|
|
||||||
// printed is a char/wchar_t pointer and the other operand is not a
|
|
||||||
// string/wstring object. In such cases, we just print the operand as
|
|
||||||
// a pointer to be safe.
|
|
||||||
#define GTEST_FORMAT_CHAR_PTR_IMPL_(CharType) \
|
|
||||||
template <typename T> \
|
|
||||||
String FormatForComparisonFailureMessage(CharType* GTEST_CREF_WORKAROUND_ p, \
|
|
||||||
const T&) { \
|
|
||||||
return PrintToString(static_cast<const void*>(p)); \
|
|
||||||
}
|
|
||||||
|
|
||||||
GTEST_FORMAT_CHAR_PTR_IMPL_(char)
|
|
||||||
GTEST_FORMAT_CHAR_PTR_IMPL_(const char)
|
|
||||||
GTEST_FORMAT_CHAR_PTR_IMPL_(wchar_t)
|
|
||||||
GTEST_FORMAT_CHAR_PTR_IMPL_(const wchar_t)
|
|
||||||
|
|
||||||
#undef GTEST_FORMAT_CHAR_PTR_IMPL_
|
|
||||||
|
|
||||||
// Constructs and returns the message for an equality assertion
|
// Constructs and returns the message for an equality assertion
|
||||||
// (e.g. ASSERT_EQ, EXPECT_STREQ, etc) failure.
|
// (e.g. ASSERT_EQ, EXPECT_STREQ, etc) failure.
|
||||||
@ -277,12 +220,12 @@ GTEST_FORMAT_CHAR_PTR_IMPL_(const wchar_t)
|
|||||||
// be inserted into the message.
|
// be inserted into the message.
|
||||||
GTEST_API_ AssertionResult EqFailure(const char* expected_expression,
|
GTEST_API_ AssertionResult EqFailure(const char* expected_expression,
|
||||||
const char* actual_expression,
|
const char* actual_expression,
|
||||||
const String& expected_value,
|
const std::string& expected_value,
|
||||||
const String& actual_value,
|
const std::string& actual_value,
|
||||||
bool ignoring_case);
|
bool ignoring_case);
|
||||||
|
|
||||||
// Constructs a failure message for Boolean assertions such as EXPECT_TRUE.
|
// Constructs a failure message for Boolean assertions such as EXPECT_TRUE.
|
||||||
GTEST_API_ String GetBoolAssertionFailureMessage(
|
GTEST_API_ std::string GetBoolAssertionFailureMessage(
|
||||||
const AssertionResult& assertion_result,
|
const AssertionResult& assertion_result,
|
||||||
const char* expression_text,
|
const char* expression_text,
|
||||||
const char* actual_predicate_value,
|
const char* actual_predicate_value,
|
||||||
@ -357,7 +300,7 @@ class FloatingPoint {
|
|||||||
// bits. Therefore, 4 should be enough for ordinary use.
|
// bits. Therefore, 4 should be enough for ordinary use.
|
||||||
//
|
//
|
||||||
// See the following article for more details on ULP:
|
// See the following article for more details on ULP:
|
||||||
// http://www.cygnus-software.com/papers/comparingfloats/comparingfloats.htm.
|
// http://randomascii.wordpress.com/2012/02/25/comparing-floating-point-numbers-2012-edition/
|
||||||
static const size_t kMaxUlps = 4;
|
static const size_t kMaxUlps = 4;
|
||||||
|
|
||||||
// Constructs a FloatingPoint from a raw floating-point number.
|
// Constructs a FloatingPoint from a raw floating-point number.
|
||||||
@ -384,6 +327,9 @@ class FloatingPoint {
|
|||||||
return ReinterpretBits(kExponentBitMask);
|
return ReinterpretBits(kExponentBitMask);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns the maximum representable finite floating-point number.
|
||||||
|
static RawType Max();
|
||||||
|
|
||||||
// Non-static methods
|
// Non-static methods
|
||||||
|
|
||||||
// Returns the bits that represents this number.
|
// Returns the bits that represents this number.
|
||||||
@ -464,6 +410,13 @@ class FloatingPoint {
|
|||||||
FloatingPointUnion u_;
|
FloatingPointUnion u_;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// We cannot use std::numeric_limits<T>::max() as it clashes with the max()
|
||||||
|
// macro defined by <windows.h>.
|
||||||
|
template <>
|
||||||
|
inline float FloatingPoint<float>::Max() { return FLT_MAX; }
|
||||||
|
template <>
|
||||||
|
inline double FloatingPoint<double>::Max() { return DBL_MAX; }
|
||||||
|
|
||||||
// Typedefs the instances of the FloatingPoint template class that we
|
// Typedefs the instances of the FloatingPoint template class that we
|
||||||
// care to use.
|
// care to use.
|
||||||
typedef FloatingPoint<float> Float;
|
typedef FloatingPoint<float> Float;
|
||||||
@ -568,7 +521,8 @@ typedef void (*TearDownTestCaseFunc)();
|
|||||||
// The newly created TestInfo instance will assume
|
// The newly created TestInfo instance will assume
|
||||||
// ownership of the factory object.
|
// ownership of the factory object.
|
||||||
GTEST_API_ TestInfo* MakeAndRegisterTestInfo(
|
GTEST_API_ TestInfo* MakeAndRegisterTestInfo(
|
||||||
const char* test_case_name, const char* name,
|
const char* test_case_name,
|
||||||
|
const char* name,
|
||||||
const char* type_param,
|
const char* type_param,
|
||||||
const char* value_param,
|
const char* value_param,
|
||||||
TypeId fixture_class_id,
|
TypeId fixture_class_id,
|
||||||
@ -628,9 +582,9 @@ inline const char* SkipComma(const char* str) {
|
|||||||
|
|
||||||
// Returns the prefix of 'str' before the first comma in it; returns
|
// Returns the prefix of 'str' before the first comma in it; returns
|
||||||
// the entire string if it contains no comma.
|
// the entire string if it contains no comma.
|
||||||
inline String GetPrefixUntilComma(const char* str) {
|
inline std::string GetPrefixUntilComma(const char* str) {
|
||||||
const char* comma = strchr(str, ',');
|
const char* comma = strchr(str, ',');
|
||||||
return comma == NULL ? String(str) : String(str, comma - str);
|
return comma == NULL ? str : std::string(str, comma);
|
||||||
}
|
}
|
||||||
|
|
||||||
// TypeParameterizedTest<Fixture, TestSel, Types>::Register()
|
// TypeParameterizedTest<Fixture, TestSel, Types>::Register()
|
||||||
@ -656,9 +610,9 @@ class TypeParameterizedTest {
|
|||||||
// First, registers the first type-parameterized test in the type
|
// First, registers the first type-parameterized test in the type
|
||||||
// list.
|
// list.
|
||||||
MakeAndRegisterTestInfo(
|
MakeAndRegisterTestInfo(
|
||||||
String::Format("%s%s%s/%d", prefix, prefix[0] == '\0' ? "" : "/",
|
(std::string(prefix) + (prefix[0] == '\0' ? "" : "/") + case_name + "/"
|
||||||
case_name, index).c_str(),
|
+ StreamableToString(index)).c_str(),
|
||||||
GetPrefixUntilComma(test_names).c_str(),
|
StripTrailingSpaces(GetPrefixUntilComma(test_names)).c_str(),
|
||||||
GetTypeName<Type>().c_str(),
|
GetTypeName<Type>().c_str(),
|
||||||
NULL, // No value parameter.
|
NULL, // No value parameter.
|
||||||
GetTypeId<FixtureClass>(),
|
GetTypeId<FixtureClass>(),
|
||||||
@ -715,7 +669,7 @@ class TypeParameterizedTestCase<Fixture, Templates0, Types> {
|
|||||||
|
|
||||||
#endif // GTEST_HAS_TYPED_TEST || GTEST_HAS_TYPED_TEST_P
|
#endif // GTEST_HAS_TYPED_TEST || GTEST_HAS_TYPED_TEST_P
|
||||||
|
|
||||||
// Returns the current OS stack trace as a String.
|
// Returns the current OS stack trace as an std::string.
|
||||||
//
|
//
|
||||||
// The maximum number of stack frames to be included is specified by
|
// The maximum number of stack frames to be included is specified by
|
||||||
// the gtest_stack_trace_depth flag. The skip_count parameter
|
// the gtest_stack_trace_depth flag. The skip_count parameter
|
||||||
@ -725,8 +679,8 @@ class TypeParameterizedTestCase<Fixture, Templates0, Types> {
|
|||||||
// For example, if Foo() calls Bar(), which in turn calls
|
// For example, if Foo() calls Bar(), which in turn calls
|
||||||
// GetCurrentOsStackTraceExceptTop(..., 1), Foo() will be included in
|
// GetCurrentOsStackTraceExceptTop(..., 1), Foo() will be included in
|
||||||
// the trace but Bar() and GetCurrentOsStackTraceExceptTop() won't.
|
// the trace but Bar() and GetCurrentOsStackTraceExceptTop() won't.
|
||||||
GTEST_API_ String GetCurrentOsStackTraceExceptTop(UnitTest* unit_test,
|
GTEST_API_ std::string GetCurrentOsStackTraceExceptTop(
|
||||||
int skip_count);
|
UnitTest* unit_test, int skip_count);
|
||||||
|
|
||||||
// Helpers for suppressing warnings on unreachable code or constant
|
// Helpers for suppressing warnings on unreachable code or constant
|
||||||
// condition.
|
// condition.
|
||||||
@ -801,13 +755,19 @@ struct RemoveConst<const T> { typedef T type; }; // NOLINT
|
|||||||
// MSVC 8.0, Sun C++, and IBM XL C++ have a bug which causes the above
|
// MSVC 8.0, Sun C++, and IBM XL C++ have a bug which causes the above
|
||||||
// definition to fail to remove the const in 'const int[3]' and 'const
|
// definition to fail to remove the const in 'const int[3]' and 'const
|
||||||
// char[3][4]'. The following specialization works around the bug.
|
// char[3][4]'. The following specialization works around the bug.
|
||||||
// However, it causes trouble with GCC and thus needs to be
|
|
||||||
// conditionally compiled.
|
|
||||||
#if defined(_MSC_VER) || defined(__SUNPRO_CC) || defined(__IBMCPP__)
|
|
||||||
template <typename T, size_t N>
|
template <typename T, size_t N>
|
||||||
struct RemoveConst<const T[N]> {
|
struct RemoveConst<const T[N]> {
|
||||||
typedef typename RemoveConst<T>::type type[N];
|
typedef typename RemoveConst<T>::type type[N];
|
||||||
};
|
};
|
||||||
|
|
||||||
|
#if defined(_MSC_VER) && _MSC_VER < 1400
|
||||||
|
// This is the only specialization that allows VC++ 7.1 to remove const in
|
||||||
|
// 'const int[3] and 'const int[3][4]'. However, it causes trouble with GCC
|
||||||
|
// and thus needs to be conditionally compiled.
|
||||||
|
template <typename T, size_t N>
|
||||||
|
struct RemoveConst<T[N]> {
|
||||||
|
typedef typename RemoveConst<T>::type type[N];
|
||||||
|
};
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// A handy wrapper around RemoveConst that works when the argument
|
// A handy wrapper around RemoveConst that works when the argument
|
||||||
@ -856,7 +816,7 @@ class ImplicitlyConvertible {
|
|||||||
// MakeFrom() is an expression whose type is From. We cannot simply
|
// MakeFrom() is an expression whose type is From. We cannot simply
|
||||||
// use From(), as the type From may not have a public default
|
// use From(), as the type From may not have a public default
|
||||||
// constructor.
|
// constructor.
|
||||||
static From MakeFrom();
|
static typename AddReference<From>::type MakeFrom();
|
||||||
|
|
||||||
// These two functions are overloaded. Given an expression
|
// These two functions are overloaded. Given an expression
|
||||||
// Helper(x), the compiler will pick the first version if x can be
|
// Helper(x), the compiler will pick the first version if x can be
|
||||||
@ -874,25 +834,20 @@ class ImplicitlyConvertible {
|
|||||||
// We have to put the 'public' section after the 'private' section,
|
// We have to put the 'public' section after the 'private' section,
|
||||||
// or MSVC refuses to compile the code.
|
// or MSVC refuses to compile the code.
|
||||||
public:
|
public:
|
||||||
// MSVC warns about implicitly converting from double to int for
|
#if defined(__BORLANDC__)
|
||||||
// possible loss of data, so we need to temporarily disable the
|
|
||||||
// warning.
|
|
||||||
#ifdef _MSC_VER
|
|
||||||
# pragma warning(push) // Saves the current warning state.
|
|
||||||
# pragma warning(disable:4244) // Temporarily disables warning 4244.
|
|
||||||
|
|
||||||
static const bool value =
|
|
||||||
sizeof(Helper(ImplicitlyConvertible::MakeFrom())) == 1;
|
|
||||||
# pragma warning(pop) // Restores the warning state.
|
|
||||||
#elif defined(__BORLANDC__)
|
|
||||||
// C++Builder cannot use member overload resolution during template
|
// C++Builder cannot use member overload resolution during template
|
||||||
// instantiation. The simplest workaround is to use its C++0x type traits
|
// instantiation. The simplest workaround is to use its C++0x type traits
|
||||||
// functions (C++Builder 2009 and above only).
|
// functions (C++Builder 2009 and above only).
|
||||||
static const bool value = __is_convertible(From, To);
|
static const bool value = __is_convertible(From, To);
|
||||||
#else
|
#else
|
||||||
|
// MSVC warns about implicitly converting from double to int for
|
||||||
|
// possible loss of data, so we need to temporarily disable the
|
||||||
|
// warning.
|
||||||
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4244)
|
||||||
static const bool value =
|
static const bool value =
|
||||||
sizeof(Helper(ImplicitlyConvertible::MakeFrom())) == 1;
|
sizeof(Helper(ImplicitlyConvertible::MakeFrom())) == 1;
|
||||||
#endif // _MSV_VER
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
|
#endif // __BORLANDC__
|
||||||
};
|
};
|
||||||
template <typename From, typename To>
|
template <typename From, typename To>
|
||||||
const bool ImplicitlyConvertible<From, To>::value;
|
const bool ImplicitlyConvertible<From, To>::value;
|
||||||
@ -1018,11 +973,10 @@ void CopyArray(const T* from, size_t size, U* to) {
|
|||||||
|
|
||||||
// The relation between an NativeArray object (see below) and the
|
// The relation between an NativeArray object (see below) and the
|
||||||
// native array it represents.
|
// native array it represents.
|
||||||
enum RelationToSource {
|
// We use 2 different structs to allow non-copyable types to be used, as long
|
||||||
kReference, // The NativeArray references the native array.
|
// as RelationToSourceReference() is passed.
|
||||||
kCopy // The NativeArray makes a copy of the native array and
|
struct RelationToSourceReference {};
|
||||||
// owns the copy.
|
struct RelationToSourceCopy {};
|
||||||
};
|
|
||||||
|
|
||||||
// Adapts a native array to a read-only STL-style container. Instead
|
// Adapts a native array to a read-only STL-style container. Instead
|
||||||
// of the complete STL container concept, this adaptor only implements
|
// of the complete STL container concept, this adaptor only implements
|
||||||
@ -1040,22 +994,23 @@ class NativeArray {
|
|||||||
typedef Element* iterator;
|
typedef Element* iterator;
|
||||||
typedef const Element* const_iterator;
|
typedef const Element* const_iterator;
|
||||||
|
|
||||||
// Constructs from a native array.
|
// Constructs from a native array. References the source.
|
||||||
NativeArray(const Element* array, size_t count, RelationToSource relation) {
|
NativeArray(const Element* array, size_t count, RelationToSourceReference) {
|
||||||
Init(array, count, relation);
|
InitRef(array, count);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Constructs from a native array. Copies the source.
|
||||||
|
NativeArray(const Element* array, size_t count, RelationToSourceCopy) {
|
||||||
|
InitCopy(array, count);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Copy constructor.
|
// Copy constructor.
|
||||||
NativeArray(const NativeArray& rhs) {
|
NativeArray(const NativeArray& rhs) {
|
||||||
Init(rhs.array_, rhs.size_, rhs.relation_to_source_);
|
(this->*rhs.clone_)(rhs.array_, rhs.size_);
|
||||||
}
|
}
|
||||||
|
|
||||||
~NativeArray() {
|
~NativeArray() {
|
||||||
// Ensures that the user doesn't instantiate NativeArray with a
|
if (clone_ != &NativeArray::InitRef)
|
||||||
// const or reference type.
|
|
||||||
static_cast<void>(StaticAssertTypeEqHelper<Element,
|
|
||||||
GTEST_REMOVE_REFERENCE_AND_CONST_(Element)>());
|
|
||||||
if (relation_to_source_ == kCopy)
|
|
||||||
delete[] array_;
|
delete[] array_;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1069,23 +1024,30 @@ class NativeArray {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Initializes this object; makes a copy of the input array if
|
enum {
|
||||||
// 'relation' is kCopy.
|
kCheckTypeIsNotConstOrAReference = StaticAssertTypeEqHelper<
|
||||||
void Init(const Element* array, size_t a_size, RelationToSource relation) {
|
Element, GTEST_REMOVE_REFERENCE_AND_CONST_(Element)>::value,
|
||||||
if (relation == kReference) {
|
};
|
||||||
array_ = array;
|
|
||||||
} else {
|
// Initializes this object with a copy of the input.
|
||||||
|
void InitCopy(const Element* array, size_t a_size) {
|
||||||
Element* const copy = new Element[a_size];
|
Element* const copy = new Element[a_size];
|
||||||
CopyArray(array, a_size, copy);
|
CopyArray(array, a_size, copy);
|
||||||
array_ = copy;
|
array_ = copy;
|
||||||
}
|
|
||||||
size_ = a_size;
|
size_ = a_size;
|
||||||
relation_to_source_ = relation;
|
clone_ = &NativeArray::InitCopy;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initializes this object with a reference of the input.
|
||||||
|
void InitRef(const Element* array, size_t a_size) {
|
||||||
|
array_ = array;
|
||||||
|
size_ = a_size;
|
||||||
|
clone_ = &NativeArray::InitRef;
|
||||||
}
|
}
|
||||||
|
|
||||||
const Element* array_;
|
const Element* array_;
|
||||||
size_t size_;
|
size_t size_;
|
||||||
RelationToSource relation_to_source_;
|
void (NativeArray::*clone_)(const Element*, size_t);
|
||||||
|
|
||||||
GTEST_DISALLOW_ASSIGN_(NativeArray);
|
GTEST_DISALLOW_ASSIGN_(NativeArray);
|
||||||
};
|
};
|
||||||
@ -1228,3 +1190,4 @@ class GTEST_TEST_CLASS_NAME_(test_case_name, test_name) : public parent_class {\
|
|||||||
void GTEST_TEST_CLASS_NAME_(test_case_name, test_name)::TestBody()
|
void GTEST_TEST_CLASS_NAME_(test_case_name, test_name)::TestBody()
|
||||||
|
|
||||||
#endif // GTEST_INCLUDE_GTEST_INTERNAL_GTEST_INTERNAL_H_
|
#endif // GTEST_INCLUDE_GTEST_INTERNAL_GTEST_INTERNAL_H_
|
||||||
|
|
||||||
|
@ -105,25 +105,35 @@ class linked_ptr_internal {
|
|||||||
// framework.
|
// framework.
|
||||||
|
|
||||||
// Join an existing circle.
|
// Join an existing circle.
|
||||||
// L < g_linked_ptr_mutex
|
void join(linked_ptr_internal const* ptr)
|
||||||
void join(linked_ptr_internal const* ptr) {
|
GTEST_LOCK_EXCLUDED_(g_linked_ptr_mutex) {
|
||||||
MutexLock lock(&g_linked_ptr_mutex);
|
MutexLock lock(&g_linked_ptr_mutex);
|
||||||
|
|
||||||
linked_ptr_internal const* p = ptr;
|
linked_ptr_internal const* p = ptr;
|
||||||
while (p->next_ != ptr) p = p->next_;
|
while (p->next_ != ptr) {
|
||||||
|
assert(p->next_ != this &&
|
||||||
|
"Trying to join() a linked ring we are already in. "
|
||||||
|
"Is GMock thread safety enabled?");
|
||||||
|
p = p->next_;
|
||||||
|
}
|
||||||
p->next_ = this;
|
p->next_ = this;
|
||||||
next_ = ptr;
|
next_ = ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Leave whatever circle we're part of. Returns true if we were the
|
// Leave whatever circle we're part of. Returns true if we were the
|
||||||
// last member of the circle. Once this is done, you can join() another.
|
// last member of the circle. Once this is done, you can join() another.
|
||||||
// L < g_linked_ptr_mutex
|
bool depart()
|
||||||
bool depart() {
|
GTEST_LOCK_EXCLUDED_(g_linked_ptr_mutex) {
|
||||||
MutexLock lock(&g_linked_ptr_mutex);
|
MutexLock lock(&g_linked_ptr_mutex);
|
||||||
|
|
||||||
if (next_ == this) return true;
|
if (next_ == this) return true;
|
||||||
linked_ptr_internal const* p = next_;
|
linked_ptr_internal const* p = next_;
|
||||||
while (p->next_ != this) p = p->next_;
|
while (p->next_ != this) {
|
||||||
|
assert(p->next_ != next_ &&
|
||||||
|
"Trying to depart() a linked ring we are not in. "
|
||||||
|
"Is GMock thread safety enabled?");
|
||||||
|
p = p->next_;
|
||||||
|
}
|
||||||
p->next_ = next_;
|
p->next_ = next_;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -39,7 +39,7 @@ $var maxtuple = 10 $$ Maximum number of Combine arguments we want to support.
|
|||||||
// and at most $maxtuple arguments in Combine. Please contact
|
// and at most $maxtuple arguments in Combine. Please contact
|
||||||
// googletestframework@googlegroups.com if you need more.
|
// googletestframework@googlegroups.com if you need more.
|
||||||
// Please note that the number of arguments to Combine is limited
|
// Please note that the number of arguments to Combine is limited
|
||||||
// by the maximum arity of the implementation of tr1::tuple which is
|
// by the maximum arity of the implementation of tuple which is
|
||||||
// currently set at $maxtuple.
|
// currently set at $maxtuple.
|
||||||
|
|
||||||
#ifndef GTEST_INCLUDE_GTEST_INTERNAL_GTEST_PARAM_UTIL_GENERATED_H_
|
#ifndef GTEST_INCLUDE_GTEST_INTERNAL_GTEST_PARAM_UTIL_GENERATED_H_
|
||||||
@ -98,7 +98,7 @@ class ValueArray$i {
|
|||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
operator ParamGenerator<T>() const {
|
operator ParamGenerator<T>() const {
|
||||||
const T array[] = {$for j, [[v$(j)_]]};
|
const T array[] = {$for j, [[static_cast<T>(v$(j)_)]]};
|
||||||
return ValuesIn(array);
|
return ValuesIn(array);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -128,9 +128,9 @@ $range k 2..i
|
|||||||
|
|
||||||
template <$for j, [[typename T$j]]>
|
template <$for j, [[typename T$j]]>
|
||||||
class CartesianProductGenerator$i
|
class CartesianProductGenerator$i
|
||||||
: public ParamGeneratorInterface< ::std::tr1::tuple<$for j, [[T$j]]> > {
|
: public ParamGeneratorInterface< ::testing::tuple<$for j, [[T$j]]> > {
|
||||||
public:
|
public:
|
||||||
typedef ::std::tr1::tuple<$for j, [[T$j]]> ParamType;
|
typedef ::testing::tuple<$for j, [[T$j]]> ParamType;
|
||||||
|
|
||||||
CartesianProductGenerator$i($for j, [[const ParamGenerator<T$j>& g$j]])
|
CartesianProductGenerator$i($for j, [[const ParamGenerator<T$j>& g$j]])
|
||||||
: $for j, [[g$(j)_(g$j)]] {}
|
: $for j, [[g$(j)_(g$j)]] {}
|
||||||
@ -269,8 +269,8 @@ class CartesianProductHolder$i {
|
|||||||
CartesianProductHolder$i($for j, [[const Generator$j& g$j]])
|
CartesianProductHolder$i($for j, [[const Generator$j& g$j]])
|
||||||
: $for j, [[g$(j)_(g$j)]] {}
|
: $for j, [[g$(j)_(g$j)]] {}
|
||||||
template <$for j, [[typename T$j]]>
|
template <$for j, [[typename T$j]]>
|
||||||
operator ParamGenerator< ::std::tr1::tuple<$for j, [[T$j]]> >() const {
|
operator ParamGenerator< ::testing::tuple<$for j, [[T$j]]> >() const {
|
||||||
return ParamGenerator< ::std::tr1::tuple<$for j, [[T$j]]> >(
|
return ParamGenerator< ::testing::tuple<$for j, [[T$j]]> >(
|
||||||
new CartesianProductGenerator$i<$for j, [[T$j]]>(
|
new CartesianProductGenerator$i<$for j, [[T$j]]>(
|
||||||
$for j,[[
|
$for j,[[
|
||||||
|
|
||||||
|
@ -494,10 +494,10 @@ class ParameterizedTestCaseInfo : public ParameterizedTestCaseInfoBase {
|
|||||||
const string& instantiation_name = gen_it->first;
|
const string& instantiation_name = gen_it->first;
|
||||||
ParamGenerator<ParamType> generator((*gen_it->second)());
|
ParamGenerator<ParamType> generator((*gen_it->second)());
|
||||||
|
|
||||||
Message test_case_name_stream;
|
string test_case_name;
|
||||||
if ( !instantiation_name.empty() )
|
if ( !instantiation_name.empty() )
|
||||||
test_case_name_stream << instantiation_name << "/";
|
test_case_name = instantiation_name + "/";
|
||||||
test_case_name_stream << test_info->test_case_base_name;
|
test_case_name += test_info->test_case_base_name;
|
||||||
|
|
||||||
int i = 0;
|
int i = 0;
|
||||||
for (typename ParamGenerator<ParamType>::iterator param_it =
|
for (typename ParamGenerator<ParamType>::iterator param_it =
|
||||||
@ -506,7 +506,7 @@ class ParameterizedTestCaseInfo : public ParameterizedTestCaseInfoBase {
|
|||||||
Message test_name_stream;
|
Message test_name_stream;
|
||||||
test_name_stream << test_info->test_base_name << "/" << i;
|
test_name_stream << test_info->test_base_name << "/" << i;
|
||||||
MakeAndRegisterTestInfo(
|
MakeAndRegisterTestInfo(
|
||||||
test_case_name_stream.GetString().c_str(),
|
test_case_name.c_str(),
|
||||||
test_name_stream.GetString().c_str(),
|
test_name_stream.GetString().c_str(),
|
||||||
NULL, // No type parameter.
|
NULL, // No type parameter.
|
||||||
PrintToString(*param_it).c_str(),
|
PrintToString(*param_it).c_str(),
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -47,50 +47,18 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include "gtest/internal/gtest-port.h"
|
|
||||||
|
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
|
#include "gtest/internal/gtest-port.h"
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
namespace internal {
|
namespace internal {
|
||||||
|
|
||||||
// String - a UTF-8 string class.
|
// String - an abstract class holding static string utilities.
|
||||||
//
|
|
||||||
// For historic reasons, we don't use std::string.
|
|
||||||
//
|
|
||||||
// TODO(wan@google.com): replace this class with std::string or
|
|
||||||
// implement it in terms of the latter.
|
|
||||||
//
|
|
||||||
// Note that String can represent both NULL and the empty string,
|
|
||||||
// while std::string cannot represent NULL.
|
|
||||||
//
|
|
||||||
// NULL and the empty string are considered different. NULL is less
|
|
||||||
// than anything (including the empty string) except itself.
|
|
||||||
//
|
|
||||||
// This class only provides minimum functionality necessary for
|
|
||||||
// implementing Google Test. We do not intend to implement a full-fledged
|
|
||||||
// string class here.
|
|
||||||
//
|
|
||||||
// Since the purpose of this class is to provide a substitute for
|
|
||||||
// std::string on platforms where it cannot be used, we define a copy
|
|
||||||
// constructor and assignment operators such that we don't need
|
|
||||||
// conditional compilation in a lot of places.
|
|
||||||
//
|
|
||||||
// In order to make the representation efficient, the d'tor of String
|
|
||||||
// is not virtual. Therefore DO NOT INHERIT FROM String.
|
|
||||||
class GTEST_API_ String {
|
class GTEST_API_ String {
|
||||||
public:
|
public:
|
||||||
// Static utility methods
|
// Static utility methods
|
||||||
|
|
||||||
// Returns the input enclosed in double quotes if it's not NULL;
|
|
||||||
// otherwise returns "(null)". For example, "\"Hello\"" is returned
|
|
||||||
// for input "Hello".
|
|
||||||
//
|
|
||||||
// This is useful for printing a C string in the syntax of a literal.
|
|
||||||
//
|
|
||||||
// Known issue: escape sequences are not handled yet.
|
|
||||||
static String ShowCStringQuoted(const char* c_str);
|
|
||||||
|
|
||||||
// Clones a 0-terminated C string, allocating memory using new. The
|
// Clones a 0-terminated C string, allocating memory using new. The
|
||||||
// caller is responsible for deleting the return value using
|
// caller is responsible for deleting the return value using
|
||||||
// delete[]. Returns the cloned string, or NULL if the input is
|
// delete[]. Returns the cloned string, or NULL if the input is
|
||||||
@ -137,11 +105,7 @@ class GTEST_API_ String {
|
|||||||
// NULL will be converted to "(null)". If an error occurred during
|
// NULL will be converted to "(null)". If an error occurred during
|
||||||
// the conversion, "(failed to convert from wide string)" is
|
// the conversion, "(failed to convert from wide string)" is
|
||||||
// returned.
|
// returned.
|
||||||
static String ShowWideCString(const wchar_t* wide_c_str);
|
static std::string ShowWideCString(const wchar_t* wide_c_str);
|
||||||
|
|
||||||
// Similar to ShowWideCString(), except that this function encloses
|
|
||||||
// the converted string in double quotes.
|
|
||||||
static String ShowWideCStringQuoted(const wchar_t* wide_c_str);
|
|
||||||
|
|
||||||
// Compares two wide C strings. Returns true iff they have the same
|
// Compares two wide C strings. Returns true iff they have the same
|
||||||
// content.
|
// content.
|
||||||
@ -175,174 +139,27 @@ class GTEST_API_ String {
|
|||||||
static bool CaseInsensitiveWideCStringEquals(const wchar_t* lhs,
|
static bool CaseInsensitiveWideCStringEquals(const wchar_t* lhs,
|
||||||
const wchar_t* rhs);
|
const wchar_t* rhs);
|
||||||
|
|
||||||
// Formats a list of arguments to a String, using the same format
|
// Returns true iff the given string ends with the given suffix, ignoring
|
||||||
// spec string as for printf.
|
// case. Any string is considered to end with an empty suffix.
|
||||||
//
|
static bool EndsWithCaseInsensitive(
|
||||||
// We do not use the StringPrintf class as it is not universally
|
const std::string& str, const std::string& suffix);
|
||||||
// available.
|
|
||||||
//
|
|
||||||
// The result is limited to 4096 characters (including the tailing
|
|
||||||
// 0). If 4096 characters are not enough to format the input,
|
|
||||||
// "<buffer exceeded>" is returned.
|
|
||||||
static String Format(const char* format, ...);
|
|
||||||
|
|
||||||
// C'tors
|
// Formats an int value as "%02d".
|
||||||
|
static std::string FormatIntWidth2(int value); // "%02d" for width == 2
|
||||||
|
|
||||||
// The default c'tor constructs a NULL string.
|
// Formats an int value as "%X".
|
||||||
String() : c_str_(NULL), length_(0) {}
|
static std::string FormatHexInt(int value);
|
||||||
|
|
||||||
// Constructs a String by cloning a 0-terminated C string.
|
// Formats a byte as "%02X".
|
||||||
String(const char* a_c_str) { // NOLINT
|
static std::string FormatByte(unsigned char value);
|
||||||
if (a_c_str == NULL) {
|
|
||||||
c_str_ = NULL;
|
|
||||||
length_ = 0;
|
|
||||||
} else {
|
|
||||||
ConstructNonNull(a_c_str, strlen(a_c_str));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Constructs a String by copying a given number of chars from a
|
|
||||||
// buffer. E.g. String("hello", 3) creates the string "hel",
|
|
||||||
// String("a\0bcd", 4) creates "a\0bc", String(NULL, 0) creates "",
|
|
||||||
// and String(NULL, 1) results in access violation.
|
|
||||||
String(const char* buffer, size_t a_length) {
|
|
||||||
ConstructNonNull(buffer, a_length);
|
|
||||||
}
|
|
||||||
|
|
||||||
// The copy c'tor creates a new copy of the string. The two
|
|
||||||
// String objects do not share content.
|
|
||||||
String(const String& str) : c_str_(NULL), length_(0) { *this = str; }
|
|
||||||
|
|
||||||
// D'tor. String is intended to be a final class, so the d'tor
|
|
||||||
// doesn't need to be virtual.
|
|
||||||
~String() { delete[] c_str_; }
|
|
||||||
|
|
||||||
// Allows a String to be implicitly converted to an ::std::string or
|
|
||||||
// ::string, and vice versa. Converting a String containing a NULL
|
|
||||||
// pointer to ::std::string or ::string is undefined behavior.
|
|
||||||
// Converting a ::std::string or ::string containing an embedded NUL
|
|
||||||
// character to a String will result in the prefix up to the first
|
|
||||||
// NUL character.
|
|
||||||
String(const ::std::string& str) {
|
|
||||||
ConstructNonNull(str.c_str(), str.length());
|
|
||||||
}
|
|
||||||
|
|
||||||
operator ::std::string() const { return ::std::string(c_str(), length()); }
|
|
||||||
|
|
||||||
#if GTEST_HAS_GLOBAL_STRING
|
|
||||||
String(const ::string& str) {
|
|
||||||
ConstructNonNull(str.c_str(), str.length());
|
|
||||||
}
|
|
||||||
|
|
||||||
operator ::string() const { return ::string(c_str(), length()); }
|
|
||||||
#endif // GTEST_HAS_GLOBAL_STRING
|
|
||||||
|
|
||||||
// Returns true iff this is an empty string (i.e. "").
|
|
||||||
bool empty() const { return (c_str() != NULL) && (length() == 0); }
|
|
||||||
|
|
||||||
// Compares this with another String.
|
|
||||||
// Returns < 0 if this is less than rhs, 0 if this is equal to rhs, or > 0
|
|
||||||
// if this is greater than rhs.
|
|
||||||
int Compare(const String& rhs) const;
|
|
||||||
|
|
||||||
// Returns true iff this String equals the given C string. A NULL
|
|
||||||
// string and a non-NULL string are considered not equal.
|
|
||||||
bool operator==(const char* a_c_str) const { return Compare(a_c_str) == 0; }
|
|
||||||
|
|
||||||
// Returns true iff this String is less than the given String. A
|
|
||||||
// NULL string is considered less than "".
|
|
||||||
bool operator<(const String& rhs) const { return Compare(rhs) < 0; }
|
|
||||||
|
|
||||||
// Returns true iff this String doesn't equal the given C string. A NULL
|
|
||||||
// string and a non-NULL string are considered not equal.
|
|
||||||
bool operator!=(const char* a_c_str) const { return !(*this == a_c_str); }
|
|
||||||
|
|
||||||
// Returns true iff this String ends with the given suffix. *Any*
|
|
||||||
// String is considered to end with a NULL or empty suffix.
|
|
||||||
bool EndsWith(const char* suffix) const;
|
|
||||||
|
|
||||||
// Returns true iff this String ends with the given suffix, not considering
|
|
||||||
// case. Any String is considered to end with a NULL or empty suffix.
|
|
||||||
bool EndsWithCaseInsensitive(const char* suffix) const;
|
|
||||||
|
|
||||||
// Returns the length of the encapsulated string, or 0 if the
|
|
||||||
// string is NULL.
|
|
||||||
size_t length() const { return length_; }
|
|
||||||
|
|
||||||
// Gets the 0-terminated C string this String object represents.
|
|
||||||
// The String object still owns the string. Therefore the caller
|
|
||||||
// should NOT delete the return value.
|
|
||||||
const char* c_str() const { return c_str_; }
|
|
||||||
|
|
||||||
// Assigns a C string to this object. Self-assignment works.
|
|
||||||
const String& operator=(const char* a_c_str) {
|
|
||||||
return *this = String(a_c_str);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assigns a String object to this object. Self-assignment works.
|
|
||||||
const String& operator=(const String& rhs) {
|
|
||||||
if (this != &rhs) {
|
|
||||||
delete[] c_str_;
|
|
||||||
if (rhs.c_str() == NULL) {
|
|
||||||
c_str_ = NULL;
|
|
||||||
length_ = 0;
|
|
||||||
} else {
|
|
||||||
ConstructNonNull(rhs.c_str(), rhs.length());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Constructs a non-NULL String from the given content. This
|
String(); // Not meant to be instantiated.
|
||||||
// function can only be called when c_str_ has not been allocated.
|
|
||||||
// ConstructNonNull(NULL, 0) results in an empty string ("").
|
|
||||||
// ConstructNonNull(NULL, non_zero) is undefined behavior.
|
|
||||||
void ConstructNonNull(const char* buffer, size_t a_length) {
|
|
||||||
char* const str = new char[a_length + 1];
|
|
||||||
memcpy(str, buffer, a_length);
|
|
||||||
str[a_length] = '\0';
|
|
||||||
c_str_ = str;
|
|
||||||
length_ = a_length;
|
|
||||||
}
|
|
||||||
|
|
||||||
const char* c_str_;
|
|
||||||
size_t length_;
|
|
||||||
}; // class String
|
}; // class String
|
||||||
|
|
||||||
// Streams a String to an ostream. Each '\0' character in the String
|
// Gets the content of the stringstream's buffer as an std::string. Each '\0'
|
||||||
// is replaced with "\\0".
|
|
||||||
inline ::std::ostream& operator<<(::std::ostream& os, const String& str) {
|
|
||||||
if (str.c_str() == NULL) {
|
|
||||||
os << "(null)";
|
|
||||||
} else {
|
|
||||||
const char* const c_str = str.c_str();
|
|
||||||
for (size_t i = 0; i != str.length(); i++) {
|
|
||||||
if (c_str[i] == '\0') {
|
|
||||||
os << "\\0";
|
|
||||||
} else {
|
|
||||||
os << c_str[i];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return os;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Gets the content of the stringstream's buffer as a String. Each '\0'
|
|
||||||
// character in the buffer is replaced with "\\0".
|
// character in the buffer is replaced with "\\0".
|
||||||
GTEST_API_ String StringStreamToString(::std::stringstream* stream);
|
GTEST_API_ std::string StringStreamToString(::std::stringstream* stream);
|
||||||
|
|
||||||
// Converts a streamable value to a String. A NULL pointer is
|
|
||||||
// converted to "(null)". When the input value is a ::string,
|
|
||||||
// ::std::string, ::wstring, or ::std::wstring object, each NUL
|
|
||||||
// character in it is replaced with "\\0".
|
|
||||||
|
|
||||||
// Declared here but defined in gtest.h, so that it has access
|
|
||||||
// to the definition of the Message class, required by the ARM
|
|
||||||
// compiler.
|
|
||||||
template <typename T>
|
|
||||||
String StreamableToString(const T& streamable);
|
|
||||||
|
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
} // namespace testing
|
} // namespace testing
|
||||||
|
@ -1,4 +1,6 @@
|
|||||||
// This file was GENERATED by a script. DO NOT EDIT BY HAND!!!
|
// This file was GENERATED by command:
|
||||||
|
// pump.py gtest-tuple.h.pump
|
||||||
|
// DO NOT EDIT BY HAND!!!
|
||||||
|
|
||||||
// Copyright 2009 Google Inc.
|
// Copyright 2009 Google Inc.
|
||||||
// All Rights Reserved.
|
// All Rights Reserved.
|
||||||
@ -51,6 +53,14 @@
|
|||||||
private:
|
private:
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Visual Studio 2010, 2012, and 2013 define symbols in std::tr1 that conflict
|
||||||
|
// with our own definitions. Therefore using our own tuple does not work on
|
||||||
|
// those compilers.
|
||||||
|
#if defined(_MSC_VER) && _MSC_VER >= 1600 /* 1600 is Visual Studio 2010 */
|
||||||
|
# error "gtest's tuple doesn't compile on Visual Studio 2010 or later. \
|
||||||
|
GTEST_USE_OWN_TR1_TUPLE must be set to 0 on those compilers."
|
||||||
|
#endif
|
||||||
|
|
||||||
// GTEST_n_TUPLE_(T) is the type of an n-tuple.
|
// GTEST_n_TUPLE_(T) is the type of an n-tuple.
|
||||||
#define GTEST_0_TUPLE_(T) tuple<>
|
#define GTEST_0_TUPLE_(T) tuple<>
|
||||||
#define GTEST_1_TUPLE_(T) tuple<T##0, void, void, void, void, void, void, \
|
#define GTEST_1_TUPLE_(T) tuple<T##0, void, void, void, void, void, void, \
|
||||||
@ -140,34 +150,54 @@ template <bool kIndexValid, int kIndex, class Tuple>
|
|||||||
struct TupleElement;
|
struct TupleElement;
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 0, GTEST_10_TUPLE_(T)> { typedef T0 type; };
|
struct TupleElement<true, 0, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T0 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 1, GTEST_10_TUPLE_(T)> { typedef T1 type; };
|
struct TupleElement<true, 1, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T1 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 2, GTEST_10_TUPLE_(T)> { typedef T2 type; };
|
struct TupleElement<true, 2, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T2 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 3, GTEST_10_TUPLE_(T)> { typedef T3 type; };
|
struct TupleElement<true, 3, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T3 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 4, GTEST_10_TUPLE_(T)> { typedef T4 type; };
|
struct TupleElement<true, 4, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T4 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 5, GTEST_10_TUPLE_(T)> { typedef T5 type; };
|
struct TupleElement<true, 5, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T5 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 6, GTEST_10_TUPLE_(T)> { typedef T6 type; };
|
struct TupleElement<true, 6, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T6 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 7, GTEST_10_TUPLE_(T)> { typedef T7 type; };
|
struct TupleElement<true, 7, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T7 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 8, GTEST_10_TUPLE_(T)> { typedef T8 type; };
|
struct TupleElement<true, 8, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T8 type;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, 9, GTEST_10_TUPLE_(T)> { typedef T9 type; };
|
struct TupleElement<true, 9, GTEST_10_TUPLE_(T) > {
|
||||||
|
typedef T9 type;
|
||||||
|
};
|
||||||
|
|
||||||
} // namespace gtest_internal
|
} // namespace gtest_internal
|
||||||
|
|
||||||
@ -708,37 +738,59 @@ inline GTEST_10_TUPLE_(T) make_tuple(const T0& f0, const T1& f1, const T2& f2,
|
|||||||
template <typename Tuple> struct tuple_size;
|
template <typename Tuple> struct tuple_size;
|
||||||
|
|
||||||
template <GTEST_0_TYPENAMES_(T)>
|
template <GTEST_0_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_0_TUPLE_(T)> { static const int value = 0; };
|
struct tuple_size<GTEST_0_TUPLE_(T) > {
|
||||||
|
static const int value = 0;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_1_TYPENAMES_(T)>
|
template <GTEST_1_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_1_TUPLE_(T)> { static const int value = 1; };
|
struct tuple_size<GTEST_1_TUPLE_(T) > {
|
||||||
|
static const int value = 1;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_2_TYPENAMES_(T)>
|
template <GTEST_2_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_2_TUPLE_(T)> { static const int value = 2; };
|
struct tuple_size<GTEST_2_TUPLE_(T) > {
|
||||||
|
static const int value = 2;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_3_TYPENAMES_(T)>
|
template <GTEST_3_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_3_TUPLE_(T)> { static const int value = 3; };
|
struct tuple_size<GTEST_3_TUPLE_(T) > {
|
||||||
|
static const int value = 3;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_4_TYPENAMES_(T)>
|
template <GTEST_4_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_4_TUPLE_(T)> { static const int value = 4; };
|
struct tuple_size<GTEST_4_TUPLE_(T) > {
|
||||||
|
static const int value = 4;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_5_TYPENAMES_(T)>
|
template <GTEST_5_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_5_TUPLE_(T)> { static const int value = 5; };
|
struct tuple_size<GTEST_5_TUPLE_(T) > {
|
||||||
|
static const int value = 5;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_6_TYPENAMES_(T)>
|
template <GTEST_6_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_6_TUPLE_(T)> { static const int value = 6; };
|
struct tuple_size<GTEST_6_TUPLE_(T) > {
|
||||||
|
static const int value = 6;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_7_TYPENAMES_(T)>
|
template <GTEST_7_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_7_TUPLE_(T)> { static const int value = 7; };
|
struct tuple_size<GTEST_7_TUPLE_(T) > {
|
||||||
|
static const int value = 7;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_8_TYPENAMES_(T)>
|
template <GTEST_8_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_8_TUPLE_(T)> { static const int value = 8; };
|
struct tuple_size<GTEST_8_TUPLE_(T) > {
|
||||||
|
static const int value = 8;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_9_TYPENAMES_(T)>
|
template <GTEST_9_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_9_TUPLE_(T)> { static const int value = 9; };
|
struct tuple_size<GTEST_9_TUPLE_(T) > {
|
||||||
|
static const int value = 9;
|
||||||
|
};
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T)>
|
template <GTEST_10_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_10_TUPLE_(T)> { static const int value = 10; };
|
struct tuple_size<GTEST_10_TUPLE_(T) > {
|
||||||
|
static const int value = 10;
|
||||||
|
};
|
||||||
|
|
||||||
template <int k, class Tuple>
|
template <int k, class Tuple>
|
||||||
struct tuple_element {
|
struct tuple_element {
|
||||||
@ -922,8 +974,8 @@ template <GTEST_10_TYPENAMES_(T), GTEST_10_TYPENAMES_(U)>
|
|||||||
inline bool operator==(const GTEST_10_TUPLE_(T)& t,
|
inline bool operator==(const GTEST_10_TUPLE_(T)& t,
|
||||||
const GTEST_10_TUPLE_(U)& u) {
|
const GTEST_10_TUPLE_(U)& u) {
|
||||||
return gtest_internal::SameSizeTuplePrefixComparator<
|
return gtest_internal::SameSizeTuplePrefixComparator<
|
||||||
tuple_size<GTEST_10_TUPLE_(T)>::value,
|
tuple_size<GTEST_10_TUPLE_(T) >::value,
|
||||||
tuple_size<GTEST_10_TUPLE_(U)>::value>::Eq(t, u);
|
tuple_size<GTEST_10_TUPLE_(U) >::value>::Eq(t, u);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <GTEST_10_TYPENAMES_(T), GTEST_10_TYPENAMES_(U)>
|
template <GTEST_10_TYPENAMES_(T), GTEST_10_TYPENAMES_(U)>
|
||||||
|
@ -52,6 +52,14 @@ $$ This meta comment fixes auto-indentation in Emacs. }}
|
|||||||
private:
|
private:
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Visual Studio 2010, 2012, and 2013 define symbols in std::tr1 that conflict
|
||||||
|
// with our own definitions. Therefore using our own tuple does not work on
|
||||||
|
// those compilers.
|
||||||
|
#if defined(_MSC_VER) && _MSC_VER >= 1600 /* 1600 is Visual Studio 2010 */
|
||||||
|
# error "gtest's tuple doesn't compile on Visual Studio 2010 or later. \
|
||||||
|
GTEST_USE_OWN_TR1_TUPLE must be set to 0 on those compilers."
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
$range i 0..n-1
|
$range i 0..n-1
|
||||||
$range j 0..n
|
$range j 0..n
|
||||||
@ -118,8 +126,9 @@ struct TupleElement;
|
|||||||
|
|
||||||
$for i [[
|
$for i [[
|
||||||
template <GTEST_$(n)_TYPENAMES_(T)>
|
template <GTEST_$(n)_TYPENAMES_(T)>
|
||||||
struct TupleElement<true, $i, GTEST_$(n)_TUPLE_(T)> [[]]
|
struct TupleElement<true, $i, GTEST_$(n)_TUPLE_(T) > {
|
||||||
{ typedef T$i type; };
|
typedef T$i type;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
]]
|
]]
|
||||||
@ -220,7 +229,9 @@ template <typename Tuple> struct tuple_size;
|
|||||||
|
|
||||||
$for j [[
|
$for j [[
|
||||||
template <GTEST_$(j)_TYPENAMES_(T)>
|
template <GTEST_$(j)_TYPENAMES_(T)>
|
||||||
struct tuple_size<GTEST_$(j)_TUPLE_(T)> { static const int value = $j; };
|
struct tuple_size<GTEST_$(j)_TUPLE_(T) > {
|
||||||
|
static const int value = $j;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
]]
|
]]
|
||||||
@ -302,8 +313,8 @@ template <GTEST_$(n)_TYPENAMES_(T), GTEST_$(n)_TYPENAMES_(U)>
|
|||||||
inline bool operator==(const GTEST_$(n)_TUPLE_(T)& t,
|
inline bool operator==(const GTEST_$(n)_TUPLE_(T)& t,
|
||||||
const GTEST_$(n)_TUPLE_(U)& u) {
|
const GTEST_$(n)_TUPLE_(U)& u) {
|
||||||
return gtest_internal::SameSizeTuplePrefixComparator<
|
return gtest_internal::SameSizeTuplePrefixComparator<
|
||||||
tuple_size<GTEST_$(n)_TUPLE_(T)>::value,
|
tuple_size<GTEST_$(n)_TUPLE_(T) >::value,
|
||||||
tuple_size<GTEST_$(n)_TUPLE_(U)>::value>::Eq(t, u);
|
tuple_size<GTEST_$(n)_TUPLE_(U) >::value>::Eq(t, u);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <GTEST_$(n)_TYPENAMES_(T), GTEST_$(n)_TYPENAMES_(U)>
|
template <GTEST_$(n)_TYPENAMES_(T), GTEST_$(n)_TYPENAMES_(U)>
|
||||||
|
@ -45,15 +45,14 @@
|
|||||||
#define GTEST_INCLUDE_GTEST_INTERNAL_GTEST_TYPE_UTIL_H_
|
#define GTEST_INCLUDE_GTEST_INTERNAL_GTEST_TYPE_UTIL_H_
|
||||||
|
|
||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
#include "gtest/internal/gtest-string.h"
|
|
||||||
|
|
||||||
// #ifdef __GNUC__ is too general here. It is possible to use gcc without using
|
// #ifdef __GNUC__ is too general here. It is possible to use gcc without using
|
||||||
// libstdc++ (which is where cxxabi.h comes from).
|
// libstdc++ (which is where cxxabi.h comes from).
|
||||||
# ifdef __GLIBCXX__
|
# if GTEST_HAS_CXXABI_H_
|
||||||
# include <cxxabi.h>
|
# include <cxxabi.h>
|
||||||
# elif defined(__HP_aCC)
|
# elif defined(__HP_aCC)
|
||||||
# include <acxx_demangle.h>
|
# include <acxx_demangle.h>
|
||||||
# endif // __GLIBCXX__
|
# endif // GTEST_HASH_CXXABI_H_
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
namespace internal {
|
namespace internal {
|
||||||
@ -62,24 +61,24 @@ namespace internal {
|
|||||||
// NB: This function is also used in Google Mock, so don't move it inside of
|
// NB: This function is also used in Google Mock, so don't move it inside of
|
||||||
// the typed-test-only section below.
|
// the typed-test-only section below.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
String GetTypeName() {
|
std::string GetTypeName() {
|
||||||
# if GTEST_HAS_RTTI
|
# if GTEST_HAS_RTTI
|
||||||
|
|
||||||
const char* const name = typeid(T).name();
|
const char* const name = typeid(T).name();
|
||||||
# if defined(__GLIBCXX__) || defined(__HP_aCC)
|
# if GTEST_HAS_CXXABI_H_ || defined(__HP_aCC)
|
||||||
int status = 0;
|
int status = 0;
|
||||||
// gcc's implementation of typeid(T).name() mangles the type name,
|
// gcc's implementation of typeid(T).name() mangles the type name,
|
||||||
// so we have to demangle it.
|
// so we have to demangle it.
|
||||||
# ifdef __GLIBCXX__
|
# if GTEST_HAS_CXXABI_H_
|
||||||
using abi::__cxa_demangle;
|
using abi::__cxa_demangle;
|
||||||
# endif // __GLIBCXX__
|
# endif // GTEST_HAS_CXXABI_H_
|
||||||
char* const readable_name = __cxa_demangle(name, 0, 0, &status);
|
char* const readable_name = __cxa_demangle(name, 0, 0, &status);
|
||||||
const String name_str(status == 0 ? readable_name : name);
|
const std::string name_str(status == 0 ? readable_name : name);
|
||||||
free(readable_name);
|
free(readable_name);
|
||||||
return name_str;
|
return name_str;
|
||||||
# else
|
# else
|
||||||
return name;
|
return name;
|
||||||
# endif // __GLIBCXX__ || __HP_aCC
|
# endif // GTEST_HAS_CXXABI_H_ || __HP_aCC
|
||||||
|
|
||||||
# else
|
# else
|
||||||
|
|
||||||
@ -3300,7 +3299,9 @@ struct Templates<T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14,
|
|||||||
// INSTANTIATE_TYPED_TEST_CASE_P().
|
// INSTANTIATE_TYPED_TEST_CASE_P().
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
struct TypeList { typedef Types1<T> type; };
|
struct TypeList {
|
||||||
|
typedef Types1<T> type;
|
||||||
|
};
|
||||||
|
|
||||||
template <typename T1, typename T2, typename T3, typename T4, typename T5,
|
template <typename T1, typename T2, typename T3, typename T4, typename T5,
|
||||||
typename T6, typename T7, typename T8, typename T9, typename T10,
|
typename T6, typename T7, typename T8, typename T9, typename T10,
|
||||||
|
@ -43,15 +43,14 @@ $var n = 50 $$ Maximum length of type lists we want to support.
|
|||||||
#define GTEST_INCLUDE_GTEST_INTERNAL_GTEST_TYPE_UTIL_H_
|
#define GTEST_INCLUDE_GTEST_INTERNAL_GTEST_TYPE_UTIL_H_
|
||||||
|
|
||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
#include "gtest/internal/gtest-string.h"
|
|
||||||
|
|
||||||
// #ifdef __GNUC__ is too general here. It is possible to use gcc without using
|
// #ifdef __GNUC__ is too general here. It is possible to use gcc without using
|
||||||
// libstdc++ (which is where cxxabi.h comes from).
|
// libstdc++ (which is where cxxabi.h comes from).
|
||||||
# ifdef __GLIBCXX__
|
# if GTEST_HAS_CXXABI_H_
|
||||||
# include <cxxabi.h>
|
# include <cxxabi.h>
|
||||||
# elif defined(__HP_aCC)
|
# elif defined(__HP_aCC)
|
||||||
# include <acxx_demangle.h>
|
# include <acxx_demangle.h>
|
||||||
# endif // __GLIBCXX__
|
# endif // GTEST_HASH_CXXABI_H_
|
||||||
|
|
||||||
namespace testing {
|
namespace testing {
|
||||||
namespace internal {
|
namespace internal {
|
||||||
@ -60,24 +59,24 @@ namespace internal {
|
|||||||
// NB: This function is also used in Google Mock, so don't move it inside of
|
// NB: This function is also used in Google Mock, so don't move it inside of
|
||||||
// the typed-test-only section below.
|
// the typed-test-only section below.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
String GetTypeName() {
|
std::string GetTypeName() {
|
||||||
# if GTEST_HAS_RTTI
|
# if GTEST_HAS_RTTI
|
||||||
|
|
||||||
const char* const name = typeid(T).name();
|
const char* const name = typeid(T).name();
|
||||||
# if defined(__GLIBCXX__) || defined(__HP_aCC)
|
# if GTEST_HAS_CXXABI_H_ || defined(__HP_aCC)
|
||||||
int status = 0;
|
int status = 0;
|
||||||
// gcc's implementation of typeid(T).name() mangles the type name,
|
// gcc's implementation of typeid(T).name() mangles the type name,
|
||||||
// so we have to demangle it.
|
// so we have to demangle it.
|
||||||
# ifdef __GLIBCXX__
|
# if GTEST_HAS_CXXABI_H_
|
||||||
using abi::__cxa_demangle;
|
using abi::__cxa_demangle;
|
||||||
# endif // __GLIBCXX__
|
# endif // GTEST_HAS_CXXABI_H_
|
||||||
char* const readable_name = __cxa_demangle(name, 0, 0, &status);
|
char* const readable_name = __cxa_demangle(name, 0, 0, &status);
|
||||||
const String name_str(status == 0 ? readable_name : name);
|
const std::string name_str(status == 0 ? readable_name : name);
|
||||||
free(readable_name);
|
free(readable_name);
|
||||||
return name_str;
|
return name_str;
|
||||||
# else
|
# else
|
||||||
return name;
|
return name;
|
||||||
# endif // __GLIBCXX__ || __HP_aCC
|
# endif // GTEST_HAS_CXXABI_H_ || __HP_aCC
|
||||||
|
|
||||||
# else
|
# else
|
||||||
|
|
||||||
@ -279,7 +278,9 @@ struct Templates<$for j, [[T$j]]$for k[[, NoneT]]> {
|
|||||||
// INSTANTIATE_TYPED_TEST_CASE_P().
|
// INSTANTIATE_TYPED_TEST_CASE_P().
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
struct TypeList { typedef Types1<T> type; };
|
struct TypeList {
|
||||||
|
typedef Types1<T> type;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
$range i 1..n
|
$range i 1..n
|
||||||
|
@ -43,6 +43,11 @@
|
|||||||
# include <errno.h>
|
# include <errno.h>
|
||||||
# include <fcntl.h>
|
# include <fcntl.h>
|
||||||
# include <limits.h>
|
# include <limits.h>
|
||||||
|
|
||||||
|
# if GTEST_OS_LINUX
|
||||||
|
# include <signal.h>
|
||||||
|
# endif // GTEST_OS_LINUX
|
||||||
|
|
||||||
# include <stdarg.h>
|
# include <stdarg.h>
|
||||||
|
|
||||||
# if GTEST_OS_WINDOWS
|
# if GTEST_OS_WINDOWS
|
||||||
@ -52,6 +57,10 @@
|
|||||||
# include <sys/wait.h>
|
# include <sys/wait.h>
|
||||||
# endif // GTEST_OS_WINDOWS
|
# endif // GTEST_OS_WINDOWS
|
||||||
|
|
||||||
|
# if GTEST_OS_QNX
|
||||||
|
# include <spawn.h>
|
||||||
|
# endif // GTEST_OS_QNX
|
||||||
|
|
||||||
#endif // GTEST_HAS_DEATH_TEST
|
#endif // GTEST_HAS_DEATH_TEST
|
||||||
|
|
||||||
#include "gtest/gtest-message.h"
|
#include "gtest/gtest-message.h"
|
||||||
@ -59,9 +68,9 @@
|
|||||||
|
|
||||||
// Indicates that this translation unit is part of Google Test's
|
// Indicates that this translation unit is part of Google Test's
|
||||||
// implementation. It must come before gtest-internal-inl.h is
|
// implementation. It must come before gtest-internal-inl.h is
|
||||||
// included, or there will be a compiler error. This trick is to
|
// included, or there will be a compiler error. This trick exists to
|
||||||
// prevent a user from accidentally including gtest-internal-inl.h in
|
// prevent the accidental inclusion of gtest-internal-inl.h in the
|
||||||
// his code.
|
// user's code.
|
||||||
#define GTEST_IMPLEMENTATION_ 1
|
#define GTEST_IMPLEMENTATION_ 1
|
||||||
#include "src/gtest-internal-inl.h"
|
#include "src/gtest-internal-inl.h"
|
||||||
#undef GTEST_IMPLEMENTATION_
|
#undef GTEST_IMPLEMENTATION_
|
||||||
@ -100,13 +109,42 @@ GTEST_DEFINE_string_(
|
|||||||
"Indicates the file, line number, temporal index of "
|
"Indicates the file, line number, temporal index of "
|
||||||
"the single death test to run, and a file descriptor to "
|
"the single death test to run, and a file descriptor to "
|
||||||
"which a success code may be sent, all separated by "
|
"which a success code may be sent, all separated by "
|
||||||
"colons. This flag is specified if and only if the current "
|
"the '|' characters. This flag is specified if and only if the current "
|
||||||
"process is a sub-process launched for running a thread-safe "
|
"process is a sub-process launched for running a thread-safe "
|
||||||
"death test. FOR INTERNAL USE ONLY.");
|
"death test. FOR INTERNAL USE ONLY.");
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
|
|
||||||
#if GTEST_HAS_DEATH_TEST
|
#if GTEST_HAS_DEATH_TEST
|
||||||
|
|
||||||
|
namespace internal {
|
||||||
|
|
||||||
|
// Valid only for fast death tests. Indicates the code is running in the
|
||||||
|
// child process of a fast style death test.
|
||||||
|
static bool g_in_fast_death_test_child = false;
|
||||||
|
|
||||||
|
// Returns a Boolean value indicating whether the caller is currently
|
||||||
|
// executing in the context of the death test child process. Tools such as
|
||||||
|
// Valgrind heap checkers may need this to modify their behavior in death
|
||||||
|
// tests. IMPORTANT: This is an internal utility. Using it may break the
|
||||||
|
// implementation of death tests. User code MUST NOT use it.
|
||||||
|
bool InDeathTestChild() {
|
||||||
|
# if GTEST_OS_WINDOWS
|
||||||
|
|
||||||
|
// On Windows, death tests are thread-safe regardless of the value of the
|
||||||
|
// death_test_style flag.
|
||||||
|
return !GTEST_FLAG(internal_run_death_test).empty();
|
||||||
|
|
||||||
|
# else
|
||||||
|
|
||||||
|
if (GTEST_FLAG(death_test_style) == "threadsafe")
|
||||||
|
return !GTEST_FLAG(internal_run_death_test).empty();
|
||||||
|
else
|
||||||
|
return g_in_fast_death_test_child;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace internal
|
||||||
|
|
||||||
// ExitedWithCode constructor.
|
// ExitedWithCode constructor.
|
||||||
ExitedWithCode::ExitedWithCode(int exit_code) : exit_code_(exit_code) {
|
ExitedWithCode::ExitedWithCode(int exit_code) : exit_code_(exit_code) {
|
||||||
}
|
}
|
||||||
@ -141,7 +179,7 @@ namespace internal {
|
|||||||
|
|
||||||
// Generates a textual description of a given exit code, in the format
|
// Generates a textual description of a given exit code, in the format
|
||||||
// specified by wait(2).
|
// specified by wait(2).
|
||||||
static String ExitSummary(int exit_code) {
|
static std::string ExitSummary(int exit_code) {
|
||||||
Message m;
|
Message m;
|
||||||
|
|
||||||
# if GTEST_OS_WINDOWS
|
# if GTEST_OS_WINDOWS
|
||||||
@ -176,7 +214,7 @@ bool ExitedUnsuccessfully(int exit_status) {
|
|||||||
// one thread running, or cannot determine the number of threads, prior
|
// one thread running, or cannot determine the number of threads, prior
|
||||||
// to executing the given statement. It is the responsibility of the
|
// to executing the given statement. It is the responsibility of the
|
||||||
// caller not to pass a thread_count of 1.
|
// caller not to pass a thread_count of 1.
|
||||||
static String DeathTestThreadWarning(size_t thread_count) {
|
static std::string DeathTestThreadWarning(size_t thread_count) {
|
||||||
Message msg;
|
Message msg;
|
||||||
msg << "Death tests use fork(), which is unsafe particularly"
|
msg << "Death tests use fork(), which is unsafe particularly"
|
||||||
<< " in a threaded context. For this test, " << GTEST_NAME_ << " ";
|
<< " in a threaded context. For this test, " << GTEST_NAME_ << " ";
|
||||||
@ -210,7 +248,7 @@ enum DeathTestOutcome { IN_PROGRESS, DIED, LIVED, RETURNED, THREW };
|
|||||||
// message is propagated back to the parent process. Otherwise, the
|
// message is propagated back to the parent process. Otherwise, the
|
||||||
// message is simply printed to stderr. In either case, the program
|
// message is simply printed to stderr. In either case, the program
|
||||||
// then exits with status 1.
|
// then exits with status 1.
|
||||||
void DeathTestAbort(const String& message) {
|
void DeathTestAbort(const std::string& message) {
|
||||||
// On a POSIX system, this function may be called from a threadsafe-style
|
// On a POSIX system, this function may be called from a threadsafe-style
|
||||||
// death test child process, which operates on a very small stack. Use
|
// death test child process, which operates on a very small stack. Use
|
||||||
// the heap for any additional non-minuscule memory requirements.
|
// the heap for any additional non-minuscule memory requirements.
|
||||||
@ -234,9 +272,10 @@ void DeathTestAbort(const String& message) {
|
|||||||
# define GTEST_DEATH_TEST_CHECK_(expression) \
|
# define GTEST_DEATH_TEST_CHECK_(expression) \
|
||||||
do { \
|
do { \
|
||||||
if (!::testing::internal::IsTrue(expression)) { \
|
if (!::testing::internal::IsTrue(expression)) { \
|
||||||
DeathTestAbort(::testing::internal::String::Format( \
|
DeathTestAbort( \
|
||||||
"CHECK failed: File %s, line %d: %s", \
|
::std::string("CHECK failed: File ") + __FILE__ + ", line " \
|
||||||
__FILE__, __LINE__, #expression)); \
|
+ ::testing::internal::StreamableToString(__LINE__) + ": " \
|
||||||
|
+ #expression); \
|
||||||
} \
|
} \
|
||||||
} while (::testing::internal::AlwaysFalse())
|
} while (::testing::internal::AlwaysFalse())
|
||||||
|
|
||||||
@ -254,15 +293,16 @@ void DeathTestAbort(const String& message) {
|
|||||||
gtest_retval = (expression); \
|
gtest_retval = (expression); \
|
||||||
} while (gtest_retval == -1 && errno == EINTR); \
|
} while (gtest_retval == -1 && errno == EINTR); \
|
||||||
if (gtest_retval == -1) { \
|
if (gtest_retval == -1) { \
|
||||||
DeathTestAbort(::testing::internal::String::Format( \
|
DeathTestAbort( \
|
||||||
"CHECK failed: File %s, line %d: %s != -1", \
|
::std::string("CHECK failed: File ") + __FILE__ + ", line " \
|
||||||
__FILE__, __LINE__, #expression)); \
|
+ ::testing::internal::StreamableToString(__LINE__) + ": " \
|
||||||
|
+ #expression + " != -1"); \
|
||||||
} \
|
} \
|
||||||
} while (::testing::internal::AlwaysFalse())
|
} while (::testing::internal::AlwaysFalse())
|
||||||
|
|
||||||
// Returns the message describing the last system error in errno.
|
// Returns the message describing the last system error in errno.
|
||||||
String GetLastErrnoDescription() {
|
std::string GetLastErrnoDescription() {
|
||||||
return String(errno == 0 ? "" : posix::StrError(errno));
|
return errno == 0 ? "" : posix::StrError(errno);
|
||||||
}
|
}
|
||||||
|
|
||||||
// This is called from a death test parent process to read a failure
|
// This is called from a death test parent process to read a failure
|
||||||
@ -312,11 +352,11 @@ const char* DeathTest::LastMessage() {
|
|||||||
return last_death_test_message_.c_str();
|
return last_death_test_message_.c_str();
|
||||||
}
|
}
|
||||||
|
|
||||||
void DeathTest::set_last_death_test_message(const String& message) {
|
void DeathTest::set_last_death_test_message(const std::string& message) {
|
||||||
last_death_test_message_ = message;
|
last_death_test_message_ = message;
|
||||||
}
|
}
|
||||||
|
|
||||||
String DeathTest::last_death_test_message_;
|
std::string DeathTest::last_death_test_message_;
|
||||||
|
|
||||||
// Provides cross platform implementation for some death functionality.
|
// Provides cross platform implementation for some death functionality.
|
||||||
class DeathTestImpl : public DeathTest {
|
class DeathTestImpl : public DeathTest {
|
||||||
@ -491,7 +531,7 @@ bool DeathTestImpl::Passed(bool status_ok) {
|
|||||||
if (!spawned())
|
if (!spawned())
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
const String error_message = GetCapturedStderr();
|
const std::string error_message = GetCapturedStderr();
|
||||||
|
|
||||||
bool success = false;
|
bool success = false;
|
||||||
Message buffer;
|
Message buffer;
|
||||||
@ -673,22 +713,19 @@ DeathTest::TestRole WindowsDeathTest::AssumeRole() {
|
|||||||
FALSE, // The initial state is non-signalled.
|
FALSE, // The initial state is non-signalled.
|
||||||
NULL)); // The even is unnamed.
|
NULL)); // The even is unnamed.
|
||||||
GTEST_DEATH_TEST_CHECK_(event_handle_.Get() != NULL);
|
GTEST_DEATH_TEST_CHECK_(event_handle_.Get() != NULL);
|
||||||
const String filter_flag = String::Format("--%s%s=%s.%s",
|
const std::string filter_flag =
|
||||||
GTEST_FLAG_PREFIX_, kFilterFlag,
|
std::string("--") + GTEST_FLAG_PREFIX_ + kFilterFlag + "=" +
|
||||||
info->test_case_name(),
|
info->test_case_name() + "." + info->name();
|
||||||
info->name());
|
const std::string internal_flag =
|
||||||
const String internal_flag = String::Format(
|
std::string("--") + GTEST_FLAG_PREFIX_ + kInternalRunDeathTestFlag +
|
||||||
"--%s%s=%s|%d|%d|%u|%Iu|%Iu",
|
"=" + file_ + "|" + StreamableToString(line_) + "|" +
|
||||||
GTEST_FLAG_PREFIX_,
|
StreamableToString(death_test_index) + "|" +
|
||||||
kInternalRunDeathTestFlag,
|
StreamableToString(static_cast<unsigned int>(::GetCurrentProcessId())) +
|
||||||
file_, line_,
|
// size_t has the same width as pointers on both 32-bit and 64-bit
|
||||||
death_test_index,
|
|
||||||
static_cast<unsigned int>(::GetCurrentProcessId()),
|
|
||||||
// size_t has the same with as pointers on both 32-bit and 64-bit
|
|
||||||
// Windows platforms.
|
// Windows platforms.
|
||||||
// See http://msdn.microsoft.com/en-us/library/tcxf1dw6.aspx.
|
// See http://msdn.microsoft.com/en-us/library/tcxf1dw6.aspx.
|
||||||
reinterpret_cast<size_t>(write_handle),
|
"|" + StreamableToString(reinterpret_cast<size_t>(write_handle)) +
|
||||||
reinterpret_cast<size_t>(event_handle_.Get()));
|
"|" + StreamableToString(reinterpret_cast<size_t>(event_handle_.Get()));
|
||||||
|
|
||||||
char executable_path[_MAX_PATH + 1]; // NOLINT
|
char executable_path[_MAX_PATH + 1]; // NOLINT
|
||||||
GTEST_DEATH_TEST_CHECK_(
|
GTEST_DEATH_TEST_CHECK_(
|
||||||
@ -696,10 +733,9 @@ DeathTest::TestRole WindowsDeathTest::AssumeRole() {
|
|||||||
executable_path,
|
executable_path,
|
||||||
_MAX_PATH));
|
_MAX_PATH));
|
||||||
|
|
||||||
String command_line = String::Format("%s %s \"%s\"",
|
std::string command_line =
|
||||||
::GetCommandLineA(),
|
std::string(::GetCommandLineA()) + " " + filter_flag + " \"" +
|
||||||
filter_flag.c_str(),
|
internal_flag + "\"";
|
||||||
internal_flag.c_str());
|
|
||||||
|
|
||||||
DeathTest::set_last_death_test_message("");
|
DeathTest::set_last_death_test_message("");
|
||||||
|
|
||||||
@ -816,6 +852,7 @@ DeathTest::TestRole NoExecDeathTest::AssumeRole() {
|
|||||||
// Event forwarding to the listeners of event listener API mush be shut
|
// Event forwarding to the listeners of event listener API mush be shut
|
||||||
// down in death test subprocesses.
|
// down in death test subprocesses.
|
||||||
GetUnitTestImpl()->listeners()->SuppressEventForwarding();
|
GetUnitTestImpl()->listeners()->SuppressEventForwarding();
|
||||||
|
g_in_fast_death_test_child = true;
|
||||||
return EXECUTE_TEST;
|
return EXECUTE_TEST;
|
||||||
} else {
|
} else {
|
||||||
GTEST_DEATH_TEST_CHECK_SYSCALL_(close(pipe_fd[1]));
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(close(pipe_fd[1]));
|
||||||
@ -835,6 +872,11 @@ class ExecDeathTest : public ForkingDeathTest {
|
|||||||
ForkingDeathTest(a_statement, a_regex), file_(file), line_(line) { }
|
ForkingDeathTest(a_statement, a_regex), file_(file), line_(line) { }
|
||||||
virtual TestRole AssumeRole();
|
virtual TestRole AssumeRole();
|
||||||
private:
|
private:
|
||||||
|
static ::std::vector<testing::internal::string>
|
||||||
|
GetArgvsForDeathTestChildProcess() {
|
||||||
|
::std::vector<testing::internal::string> args = GetInjectableArgvs();
|
||||||
|
return args;
|
||||||
|
}
|
||||||
// The name of the file in which the death test is located.
|
// The name of the file in which the death test is located.
|
||||||
const char* const file_;
|
const char* const file_;
|
||||||
// The line number on which the death test is located.
|
// The line number on which the death test is located.
|
||||||
@ -869,6 +911,7 @@ class Arguments {
|
|||||||
char* const* Argv() {
|
char* const* Argv() {
|
||||||
return &args_[0];
|
return &args_[0];
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::vector<char*> args_;
|
std::vector<char*> args_;
|
||||||
};
|
};
|
||||||
@ -894,6 +937,7 @@ extern "C" char** environ;
|
|||||||
inline char** GetEnviron() { return environ; }
|
inline char** GetEnviron() { return environ; }
|
||||||
# endif // GTEST_OS_MAC
|
# endif // GTEST_OS_MAC
|
||||||
|
|
||||||
|
# if !GTEST_OS_QNX
|
||||||
// The main function for a threadsafe-style death test child process.
|
// The main function for a threadsafe-style death test child process.
|
||||||
// This function is called in a clone()-ed process and thus must avoid
|
// This function is called in a clone()-ed process and thus must avoid
|
||||||
// any potentially unsafe operations like malloc or libc functions.
|
// any potentially unsafe operations like malloc or libc functions.
|
||||||
@ -908,9 +952,8 @@ static int ExecDeathTestChildMain(void* child_arg) {
|
|||||||
UnitTest::GetInstance()->original_working_dir();
|
UnitTest::GetInstance()->original_working_dir();
|
||||||
// We can safely call chdir() as it's a direct system call.
|
// We can safely call chdir() as it's a direct system call.
|
||||||
if (chdir(original_dir) != 0) {
|
if (chdir(original_dir) != 0) {
|
||||||
DeathTestAbort(String::Format("chdir(\"%s\") failed: %s",
|
DeathTestAbort(std::string("chdir(\"") + original_dir + "\") failed: " +
|
||||||
original_dir,
|
GetLastErrnoDescription());
|
||||||
GetLastErrnoDescription().c_str()));
|
|
||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -920,12 +963,12 @@ static int ExecDeathTestChildMain(void* child_arg) {
|
|||||||
// invoke the test program via a valid path that contains at least
|
// invoke the test program via a valid path that contains at least
|
||||||
// one path separator.
|
// one path separator.
|
||||||
execve(args->argv[0], args->argv, GetEnviron());
|
execve(args->argv[0], args->argv, GetEnviron());
|
||||||
DeathTestAbort(String::Format("execve(%s, ...) in %s failed: %s",
|
DeathTestAbort(std::string("execve(") + args->argv[0] + ", ...) in " +
|
||||||
args->argv[0],
|
original_dir + " failed: " +
|
||||||
original_dir,
|
GetLastErrnoDescription());
|
||||||
GetLastErrnoDescription().c_str()));
|
|
||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
# endif // !GTEST_OS_QNX
|
||||||
|
|
||||||
// Two utility routines that together determine the direction the stack
|
// Two utility routines that together determine the direction the stack
|
||||||
// grows.
|
// grows.
|
||||||
@ -936,24 +979,76 @@ static int ExecDeathTestChildMain(void* child_arg) {
|
|||||||
// GTEST_NO_INLINE_ is required to prevent GCC 4.6 from inlining
|
// GTEST_NO_INLINE_ is required to prevent GCC 4.6 from inlining
|
||||||
// StackLowerThanAddress into StackGrowsDown, which then doesn't give
|
// StackLowerThanAddress into StackGrowsDown, which then doesn't give
|
||||||
// correct answer.
|
// correct answer.
|
||||||
bool StackLowerThanAddress(const void* ptr) GTEST_NO_INLINE_;
|
void StackLowerThanAddress(const void* ptr, bool* result) GTEST_NO_INLINE_;
|
||||||
bool StackLowerThanAddress(const void* ptr) {
|
void StackLowerThanAddress(const void* ptr, bool* result) {
|
||||||
int dummy;
|
int dummy;
|
||||||
return &dummy < ptr;
|
*result = (&dummy < ptr);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Make sure AddressSanitizer does not tamper with the stack here.
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_ADDRESS_
|
||||||
bool StackGrowsDown() {
|
bool StackGrowsDown() {
|
||||||
int dummy;
|
int dummy;
|
||||||
return StackLowerThanAddress(&dummy);
|
bool result;
|
||||||
|
StackLowerThanAddress(&dummy, &result);
|
||||||
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
// A threadsafe implementation of fork(2) for threadsafe-style death tests
|
// Spawns a child process with the same executable as the current process in
|
||||||
// that uses clone(2). It dies with an error message if anything goes
|
// a thread-safe manner and instructs it to run the death test. The
|
||||||
// wrong.
|
// implementation uses fork(2) + exec. On systems where clone(2) is
|
||||||
static pid_t ExecDeathTestFork(char* const* argv, int close_fd) {
|
// available, it is used instead, being slightly more thread-safe. On QNX,
|
||||||
|
// fork supports only single-threaded environments, so this function uses
|
||||||
|
// spawn(2) there instead. The function dies with an error message if
|
||||||
|
// anything goes wrong.
|
||||||
|
static pid_t ExecDeathTestSpawnChild(char* const* argv, int close_fd) {
|
||||||
ExecDeathTestArgs args = { argv, close_fd };
|
ExecDeathTestArgs args = { argv, close_fd };
|
||||||
pid_t child_pid = -1;
|
pid_t child_pid = -1;
|
||||||
|
|
||||||
|
# if GTEST_OS_QNX
|
||||||
|
// Obtains the current directory and sets it to be closed in the child
|
||||||
|
// process.
|
||||||
|
const int cwd_fd = open(".", O_RDONLY);
|
||||||
|
GTEST_DEATH_TEST_CHECK_(cwd_fd != -1);
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(fcntl(cwd_fd, F_SETFD, FD_CLOEXEC));
|
||||||
|
// We need to execute the test program in the same environment where
|
||||||
|
// it was originally invoked. Therefore we change to the original
|
||||||
|
// working directory first.
|
||||||
|
const char* const original_dir =
|
||||||
|
UnitTest::GetInstance()->original_working_dir();
|
||||||
|
// We can safely call chdir() as it's a direct system call.
|
||||||
|
if (chdir(original_dir) != 0) {
|
||||||
|
DeathTestAbort(std::string("chdir(\"") + original_dir + "\") failed: " +
|
||||||
|
GetLastErrnoDescription());
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
|
int fd_flags;
|
||||||
|
// Set close_fd to be closed after spawn.
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(fd_flags = fcntl(close_fd, F_GETFD));
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(fcntl(close_fd, F_SETFD,
|
||||||
|
fd_flags | FD_CLOEXEC));
|
||||||
|
struct inheritance inherit = {0};
|
||||||
|
// spawn is a system call.
|
||||||
|
child_pid = spawn(args.argv[0], 0, NULL, &inherit, args.argv, GetEnviron());
|
||||||
|
// Restores the current working directory.
|
||||||
|
GTEST_DEATH_TEST_CHECK_(fchdir(cwd_fd) != -1);
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(close(cwd_fd));
|
||||||
|
|
||||||
|
# else // GTEST_OS_QNX
|
||||||
|
# if GTEST_OS_LINUX
|
||||||
|
// When a SIGPROF signal is received while fork() or clone() are executing,
|
||||||
|
// the process may hang. To avoid this, we ignore SIGPROF here and re-enable
|
||||||
|
// it after the call to fork()/clone() is complete.
|
||||||
|
struct sigaction saved_sigprof_action;
|
||||||
|
struct sigaction ignore_sigprof_action;
|
||||||
|
memset(&ignore_sigprof_action, 0, sizeof(ignore_sigprof_action));
|
||||||
|
sigemptyset(&ignore_sigprof_action.sa_mask);
|
||||||
|
ignore_sigprof_action.sa_handler = SIG_IGN;
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(sigaction(
|
||||||
|
SIGPROF, &ignore_sigprof_action, &saved_sigprof_action));
|
||||||
|
# endif // GTEST_OS_LINUX
|
||||||
|
|
||||||
# if GTEST_HAS_CLONE
|
# if GTEST_HAS_CLONE
|
||||||
const bool use_fork = GTEST_FLAG(death_test_use_fork);
|
const bool use_fork = GTEST_FLAG(death_test_use_fork);
|
||||||
|
|
||||||
@ -964,8 +1059,19 @@ static pid_t ExecDeathTestFork(char* const* argv, int close_fd) {
|
|||||||
void* const stack = mmap(NULL, stack_size, PROT_READ | PROT_WRITE,
|
void* const stack = mmap(NULL, stack_size, PROT_READ | PROT_WRITE,
|
||||||
MAP_ANON | MAP_PRIVATE, -1, 0);
|
MAP_ANON | MAP_PRIVATE, -1, 0);
|
||||||
GTEST_DEATH_TEST_CHECK_(stack != MAP_FAILED);
|
GTEST_DEATH_TEST_CHECK_(stack != MAP_FAILED);
|
||||||
|
|
||||||
|
// Maximum stack alignment in bytes: For a downward-growing stack, this
|
||||||
|
// amount is subtracted from size of the stack space to get an address
|
||||||
|
// that is within the stack space and is aligned on all systems we care
|
||||||
|
// about. As far as I know there is no ABI with stack alignment greater
|
||||||
|
// than 64. We assume stack and stack_size already have alignment of
|
||||||
|
// kMaxStackAlignment.
|
||||||
|
const size_t kMaxStackAlignment = 64;
|
||||||
void* const stack_top =
|
void* const stack_top =
|
||||||
static_cast<char*>(stack) + (stack_grows_down ? stack_size : 0);
|
static_cast<char*>(stack) +
|
||||||
|
(stack_grows_down ? stack_size - kMaxStackAlignment : 0);
|
||||||
|
GTEST_DEATH_TEST_CHECK_(stack_size > kMaxStackAlignment &&
|
||||||
|
reinterpret_cast<intptr_t>(stack_top) % kMaxStackAlignment == 0);
|
||||||
|
|
||||||
child_pid = clone(&ExecDeathTestChildMain, stack_top, SIGCHLD, &args);
|
child_pid = clone(&ExecDeathTestChildMain, stack_top, SIGCHLD, &args);
|
||||||
|
|
||||||
@ -979,6 +1085,11 @@ static pid_t ExecDeathTestFork(char* const* argv, int close_fd) {
|
|||||||
ExecDeathTestChildMain(&args);
|
ExecDeathTestChildMain(&args);
|
||||||
_exit(0);
|
_exit(0);
|
||||||
}
|
}
|
||||||
|
# endif // GTEST_OS_QNX
|
||||||
|
# if GTEST_OS_LINUX
|
||||||
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(
|
||||||
|
sigaction(SIGPROF, &saved_sigprof_action, NULL));
|
||||||
|
# endif // GTEST_OS_LINUX
|
||||||
|
|
||||||
GTEST_DEATH_TEST_CHECK_(child_pid != -1);
|
GTEST_DEATH_TEST_CHECK_(child_pid != -1);
|
||||||
return child_pid;
|
return child_pid;
|
||||||
@ -1006,16 +1117,16 @@ DeathTest::TestRole ExecDeathTest::AssumeRole() {
|
|||||||
// it be closed when the child process does an exec:
|
// it be closed when the child process does an exec:
|
||||||
GTEST_DEATH_TEST_CHECK_(fcntl(pipe_fd[1], F_SETFD, 0) != -1);
|
GTEST_DEATH_TEST_CHECK_(fcntl(pipe_fd[1], F_SETFD, 0) != -1);
|
||||||
|
|
||||||
const String filter_flag =
|
const std::string filter_flag =
|
||||||
String::Format("--%s%s=%s.%s",
|
std::string("--") + GTEST_FLAG_PREFIX_ + kFilterFlag + "="
|
||||||
GTEST_FLAG_PREFIX_, kFilterFlag,
|
+ info->test_case_name() + "." + info->name();
|
||||||
info->test_case_name(), info->name());
|
const std::string internal_flag =
|
||||||
const String internal_flag =
|
std::string("--") + GTEST_FLAG_PREFIX_ + kInternalRunDeathTestFlag + "="
|
||||||
String::Format("--%s%s=%s|%d|%d|%d",
|
+ file_ + "|" + StreamableToString(line_) + "|"
|
||||||
GTEST_FLAG_PREFIX_, kInternalRunDeathTestFlag,
|
+ StreamableToString(death_test_index) + "|"
|
||||||
file_, line_, death_test_index, pipe_fd[1]);
|
+ StreamableToString(pipe_fd[1]);
|
||||||
Arguments args;
|
Arguments args;
|
||||||
args.AddArguments(GetArgvs());
|
args.AddArguments(GetArgvsForDeathTestChildProcess());
|
||||||
args.AddArgument(filter_flag.c_str());
|
args.AddArgument(filter_flag.c_str());
|
||||||
args.AddArgument(internal_flag.c_str());
|
args.AddArgument(internal_flag.c_str());
|
||||||
|
|
||||||
@ -1026,7 +1137,7 @@ DeathTest::TestRole ExecDeathTest::AssumeRole() {
|
|||||||
// is necessary.
|
// is necessary.
|
||||||
FlushInfoLog();
|
FlushInfoLog();
|
||||||
|
|
||||||
const pid_t child_pid = ExecDeathTestFork(args.Argv(), pipe_fd[0]);
|
const pid_t child_pid = ExecDeathTestSpawnChild(args.Argv(), pipe_fd[0]);
|
||||||
GTEST_DEATH_TEST_CHECK_SYSCALL_(close(pipe_fd[1]));
|
GTEST_DEATH_TEST_CHECK_SYSCALL_(close(pipe_fd[1]));
|
||||||
set_child_pid(child_pid);
|
set_child_pid(child_pid);
|
||||||
set_read_fd(pipe_fd[0]);
|
set_read_fd(pipe_fd[0]);
|
||||||
@ -1052,9 +1163,10 @@ bool DefaultDeathTestFactory::Create(const char* statement, const RE* regex,
|
|||||||
|
|
||||||
if (flag != NULL) {
|
if (flag != NULL) {
|
||||||
if (death_test_index > flag->index()) {
|
if (death_test_index > flag->index()) {
|
||||||
DeathTest::set_last_death_test_message(String::Format(
|
DeathTest::set_last_death_test_message(
|
||||||
"Death test count (%d) somehow exceeded expected maximum (%d)",
|
"Death test count (" + StreamableToString(death_test_index)
|
||||||
death_test_index, flag->index()));
|
+ ") somehow exceeded expected maximum ("
|
||||||
|
+ StreamableToString(flag->index()) + ")");
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1083,9 +1195,9 @@ bool DefaultDeathTestFactory::Create(const char* statement, const RE* regex,
|
|||||||
# endif // GTEST_OS_WINDOWS
|
# endif // GTEST_OS_WINDOWS
|
||||||
|
|
||||||
else { // NOLINT - this is more readable than unbalanced brackets inside #if.
|
else { // NOLINT - this is more readable than unbalanced brackets inside #if.
|
||||||
DeathTest::set_last_death_test_message(String::Format(
|
DeathTest::set_last_death_test_message(
|
||||||
"Unknown death test style \"%s\" encountered",
|
"Unknown death test style \"" + GTEST_FLAG(death_test_style)
|
||||||
GTEST_FLAG(death_test_style).c_str()));
|
+ "\" encountered");
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1123,8 +1235,8 @@ int GetStatusFileDescriptor(unsigned int parent_process_id,
|
|||||||
FALSE, // Non-inheritable.
|
FALSE, // Non-inheritable.
|
||||||
parent_process_id));
|
parent_process_id));
|
||||||
if (parent_process_handle.Get() == INVALID_HANDLE_VALUE) {
|
if (parent_process_handle.Get() == INVALID_HANDLE_VALUE) {
|
||||||
DeathTestAbort(String::Format("Unable to open parent process %u",
|
DeathTestAbort("Unable to open parent process " +
|
||||||
parent_process_id));
|
StreamableToString(parent_process_id));
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO(vladl@google.com): Replace the following check with a
|
// TODO(vladl@google.com): Replace the following check with a
|
||||||
@ -1144,9 +1256,10 @@ int GetStatusFileDescriptor(unsigned int parent_process_id,
|
|||||||
// DUPLICATE_SAME_ACCESS is used.
|
// DUPLICATE_SAME_ACCESS is used.
|
||||||
FALSE, // Request non-inheritable handler.
|
FALSE, // Request non-inheritable handler.
|
||||||
DUPLICATE_SAME_ACCESS)) {
|
DUPLICATE_SAME_ACCESS)) {
|
||||||
DeathTestAbort(String::Format(
|
DeathTestAbort("Unable to duplicate the pipe handle " +
|
||||||
"Unable to duplicate the pipe handle %Iu from the parent process %u",
|
StreamableToString(write_handle_as_size_t) +
|
||||||
write_handle_as_size_t, parent_process_id));
|
" from the parent process " +
|
||||||
|
StreamableToString(parent_process_id));
|
||||||
}
|
}
|
||||||
|
|
||||||
const HANDLE event_handle = reinterpret_cast<HANDLE>(event_handle_as_size_t);
|
const HANDLE event_handle = reinterpret_cast<HANDLE>(event_handle_as_size_t);
|
||||||
@ -1157,17 +1270,18 @@ int GetStatusFileDescriptor(unsigned int parent_process_id,
|
|||||||
0x0,
|
0x0,
|
||||||
FALSE,
|
FALSE,
|
||||||
DUPLICATE_SAME_ACCESS)) {
|
DUPLICATE_SAME_ACCESS)) {
|
||||||
DeathTestAbort(String::Format(
|
DeathTestAbort("Unable to duplicate the event handle " +
|
||||||
"Unable to duplicate the event handle %Iu from the parent process %u",
|
StreamableToString(event_handle_as_size_t) +
|
||||||
event_handle_as_size_t, parent_process_id));
|
" from the parent process " +
|
||||||
|
StreamableToString(parent_process_id));
|
||||||
}
|
}
|
||||||
|
|
||||||
const int write_fd =
|
const int write_fd =
|
||||||
::_open_osfhandle(reinterpret_cast<intptr_t>(dup_write_handle), O_APPEND);
|
::_open_osfhandle(reinterpret_cast<intptr_t>(dup_write_handle), O_APPEND);
|
||||||
if (write_fd == -1) {
|
if (write_fd == -1) {
|
||||||
DeathTestAbort(String::Format(
|
DeathTestAbort("Unable to convert pipe handle " +
|
||||||
"Unable to convert pipe handle %Iu to a file descriptor",
|
StreamableToString(write_handle_as_size_t) +
|
||||||
write_handle_as_size_t));
|
" to a file descriptor");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Signals the parent that the write end of the pipe has been acquired
|
// Signals the parent that the write end of the pipe has been acquired
|
||||||
@ -1204,9 +1318,8 @@ InternalRunDeathTestFlag* ParseInternalRunDeathTestFlag() {
|
|||||||
|| !ParseNaturalNumber(fields[3], &parent_process_id)
|
|| !ParseNaturalNumber(fields[3], &parent_process_id)
|
||||||
|| !ParseNaturalNumber(fields[4], &write_handle_as_size_t)
|
|| !ParseNaturalNumber(fields[4], &write_handle_as_size_t)
|
||||||
|| !ParseNaturalNumber(fields[5], &event_handle_as_size_t)) {
|
|| !ParseNaturalNumber(fields[5], &event_handle_as_size_t)) {
|
||||||
DeathTestAbort(String::Format(
|
DeathTestAbort("Bad --gtest_internal_run_death_test flag: " +
|
||||||
"Bad --gtest_internal_run_death_test flag: %s",
|
GTEST_FLAG(internal_run_death_test));
|
||||||
GTEST_FLAG(internal_run_death_test).c_str()));
|
|
||||||
}
|
}
|
||||||
write_fd = GetStatusFileDescriptor(parent_process_id,
|
write_fd = GetStatusFileDescriptor(parent_process_id,
|
||||||
write_handle_as_size_t,
|
write_handle_as_size_t,
|
||||||
@ -1217,9 +1330,8 @@ InternalRunDeathTestFlag* ParseInternalRunDeathTestFlag() {
|
|||||||
|| !ParseNaturalNumber(fields[1], &line)
|
|| !ParseNaturalNumber(fields[1], &line)
|
||||||
|| !ParseNaturalNumber(fields[2], &index)
|
|| !ParseNaturalNumber(fields[2], &index)
|
||||||
|| !ParseNaturalNumber(fields[3], &write_fd)) {
|
|| !ParseNaturalNumber(fields[3], &write_fd)) {
|
||||||
DeathTestAbort(String::Format(
|
DeathTestAbort("Bad --gtest_internal_run_death_test flag: "
|
||||||
"Bad --gtest_internal_run_death_test flag: %s",
|
+ GTEST_FLAG(internal_run_death_test));
|
||||||
GTEST_FLAG(internal_run_death_test).c_str()));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# endif // GTEST_OS_WINDOWS
|
# endif // GTEST_OS_WINDOWS
|
||||||
|
@ -29,6 +29,7 @@
|
|||||||
//
|
//
|
||||||
// Authors: keith.ray@gmail.com (Keith Ray)
|
// Authors: keith.ray@gmail.com (Keith Ray)
|
||||||
|
|
||||||
|
#include "gtest/gtest-message.h"
|
||||||
#include "gtest/internal/gtest-filepath.h"
|
#include "gtest/internal/gtest-filepath.h"
|
||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
|
|
||||||
@ -39,8 +40,8 @@
|
|||||||
#elif GTEST_OS_WINDOWS
|
#elif GTEST_OS_WINDOWS
|
||||||
# include <direct.h>
|
# include <direct.h>
|
||||||
# include <io.h>
|
# include <io.h>
|
||||||
#elif GTEST_OS_SYMBIAN || GTEST_OS_NACL
|
#elif GTEST_OS_SYMBIAN
|
||||||
// Symbian OpenC and NaCl have PATH_MAX in sys/syslimits.h
|
// Symbian OpenC has PATH_MAX in sys/syslimits.h
|
||||||
# include <sys/syslimits.h>
|
# include <sys/syslimits.h>
|
||||||
#else
|
#else
|
||||||
# include <limits.h>
|
# include <limits.h>
|
||||||
@ -69,7 +70,6 @@ namespace internal {
|
|||||||
// of them.
|
// of them.
|
||||||
const char kPathSeparator = '\\';
|
const char kPathSeparator = '\\';
|
||||||
const char kAlternatePathSeparator = '/';
|
const char kAlternatePathSeparator = '/';
|
||||||
const char kPathSeparatorString[] = "\\";
|
|
||||||
const char kAlternatePathSeparatorString[] = "/";
|
const char kAlternatePathSeparatorString[] = "/";
|
||||||
# if GTEST_OS_WINDOWS_MOBILE
|
# if GTEST_OS_WINDOWS_MOBILE
|
||||||
// Windows CE doesn't have a current directory. You should not use
|
// Windows CE doesn't have a current directory. You should not use
|
||||||
@ -83,7 +83,6 @@ const char kCurrentDirectoryString[] = ".\\";
|
|||||||
# endif // GTEST_OS_WINDOWS_MOBILE
|
# endif // GTEST_OS_WINDOWS_MOBILE
|
||||||
#else
|
#else
|
||||||
const char kPathSeparator = '/';
|
const char kPathSeparator = '/';
|
||||||
const char kPathSeparatorString[] = "/";
|
|
||||||
const char kCurrentDirectoryString[] = "./";
|
const char kCurrentDirectoryString[] = "./";
|
||||||
#endif // GTEST_OS_WINDOWS
|
#endif // GTEST_OS_WINDOWS
|
||||||
|
|
||||||
@ -98,7 +97,7 @@ static bool IsPathSeparator(char c) {
|
|||||||
|
|
||||||
// Returns the current working directory, or "" if unsuccessful.
|
// Returns the current working directory, or "" if unsuccessful.
|
||||||
FilePath FilePath::GetCurrentDir() {
|
FilePath FilePath::GetCurrentDir() {
|
||||||
#if GTEST_OS_WINDOWS_MOBILE
|
#if GTEST_OS_WINDOWS_MOBILE || GTEST_OS_WINDOWS_PHONE || GTEST_OS_WINDOWS_RT
|
||||||
// Windows CE doesn't have a current directory, so we just return
|
// Windows CE doesn't have a current directory, so we just return
|
||||||
// something reasonable.
|
// something reasonable.
|
||||||
return FilePath(kCurrentDirectoryString);
|
return FilePath(kCurrentDirectoryString);
|
||||||
@ -107,7 +106,14 @@ FilePath FilePath::GetCurrentDir() {
|
|||||||
return FilePath(_getcwd(cwd, sizeof(cwd)) == NULL ? "" : cwd);
|
return FilePath(_getcwd(cwd, sizeof(cwd)) == NULL ? "" : cwd);
|
||||||
#else
|
#else
|
||||||
char cwd[GTEST_PATH_MAX_ + 1] = { '\0' };
|
char cwd[GTEST_PATH_MAX_ + 1] = { '\0' };
|
||||||
return FilePath(getcwd(cwd, sizeof(cwd)) == NULL ? "" : cwd);
|
char* result = getcwd(cwd, sizeof(cwd));
|
||||||
|
# if GTEST_OS_NACL
|
||||||
|
// getcwd will likely fail in NaCl due to the sandbox, so return something
|
||||||
|
// reasonable. The user may have provided a shim implementation for getcwd,
|
||||||
|
// however, so fallback only when failure is detected.
|
||||||
|
return FilePath(result == NULL ? kCurrentDirectoryString : cwd);
|
||||||
|
# endif // GTEST_OS_NACL
|
||||||
|
return FilePath(result == NULL ? "" : cwd);
|
||||||
#endif // GTEST_OS_WINDOWS_MOBILE
|
#endif // GTEST_OS_WINDOWS_MOBILE
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -116,9 +122,10 @@ FilePath FilePath::GetCurrentDir() {
|
|||||||
// FilePath("dir/file"). If a case-insensitive extension is not
|
// FilePath("dir/file"). If a case-insensitive extension is not
|
||||||
// found, returns a copy of the original FilePath.
|
// found, returns a copy of the original FilePath.
|
||||||
FilePath FilePath::RemoveExtension(const char* extension) const {
|
FilePath FilePath::RemoveExtension(const char* extension) const {
|
||||||
String dot_extension(String::Format(".%s", extension));
|
const std::string dot_extension = std::string(".") + extension;
|
||||||
if (pathname_.EndsWithCaseInsensitive(dot_extension.c_str())) {
|
if (String::EndsWithCaseInsensitive(pathname_, dot_extension)) {
|
||||||
return FilePath(String(pathname_.c_str(), pathname_.length() - 4));
|
return FilePath(pathname_.substr(
|
||||||
|
0, pathname_.length() - dot_extension.length()));
|
||||||
}
|
}
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
@ -147,7 +154,7 @@ const char* FilePath::FindLastPathSeparator() const {
|
|||||||
// On Windows platform, '\' is the path separator, otherwise it is '/'.
|
// On Windows platform, '\' is the path separator, otherwise it is '/'.
|
||||||
FilePath FilePath::RemoveDirectoryName() const {
|
FilePath FilePath::RemoveDirectoryName() const {
|
||||||
const char* const last_sep = FindLastPathSeparator();
|
const char* const last_sep = FindLastPathSeparator();
|
||||||
return last_sep ? FilePath(String(last_sep + 1)) : *this;
|
return last_sep ? FilePath(last_sep + 1) : *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
// RemoveFileName returns the directory path with the filename removed.
|
// RemoveFileName returns the directory path with the filename removed.
|
||||||
@ -158,9 +165,9 @@ FilePath FilePath::RemoveDirectoryName() const {
|
|||||||
// On Windows platform, '\' is the path separator, otherwise it is '/'.
|
// On Windows platform, '\' is the path separator, otherwise it is '/'.
|
||||||
FilePath FilePath::RemoveFileName() const {
|
FilePath FilePath::RemoveFileName() const {
|
||||||
const char* const last_sep = FindLastPathSeparator();
|
const char* const last_sep = FindLastPathSeparator();
|
||||||
String dir;
|
std::string dir;
|
||||||
if (last_sep) {
|
if (last_sep) {
|
||||||
dir = String(c_str(), last_sep + 1 - c_str());
|
dir = std::string(c_str(), last_sep + 1 - c_str());
|
||||||
} else {
|
} else {
|
||||||
dir = kCurrentDirectoryString;
|
dir = kCurrentDirectoryString;
|
||||||
}
|
}
|
||||||
@ -177,11 +184,12 @@ FilePath FilePath::MakeFileName(const FilePath& directory,
|
|||||||
const FilePath& base_name,
|
const FilePath& base_name,
|
||||||
int number,
|
int number,
|
||||||
const char* extension) {
|
const char* extension) {
|
||||||
String file;
|
std::string file;
|
||||||
if (number == 0) {
|
if (number == 0) {
|
||||||
file = String::Format("%s.%s", base_name.c_str(), extension);
|
file = base_name.string() + "." + extension;
|
||||||
} else {
|
} else {
|
||||||
file = String::Format("%s_%d.%s", base_name.c_str(), number, extension);
|
file = base_name.string() + "_" + StreamableToString(number)
|
||||||
|
+ "." + extension;
|
||||||
}
|
}
|
||||||
return ConcatPaths(directory, FilePath(file));
|
return ConcatPaths(directory, FilePath(file));
|
||||||
}
|
}
|
||||||
@ -193,8 +201,7 @@ FilePath FilePath::ConcatPaths(const FilePath& directory,
|
|||||||
if (directory.IsEmpty())
|
if (directory.IsEmpty())
|
||||||
return relative_path;
|
return relative_path;
|
||||||
const FilePath dir(directory.RemoveTrailingPathSeparator());
|
const FilePath dir(directory.RemoveTrailingPathSeparator());
|
||||||
return FilePath(String::Format("%s%c%s", dir.c_str(), kPathSeparator,
|
return FilePath(dir.string() + kPathSeparator + relative_path.string());
|
||||||
relative_path.c_str()));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns true if pathname describes something findable in the file-system,
|
// Returns true if pathname describes something findable in the file-system,
|
||||||
@ -338,7 +345,7 @@ bool FilePath::CreateFolder() const {
|
|||||||
// On Windows platform, uses \ as the separator, other platforms use /.
|
// On Windows platform, uses \ as the separator, other platforms use /.
|
||||||
FilePath FilePath::RemoveTrailingPathSeparator() const {
|
FilePath FilePath::RemoveTrailingPathSeparator() const {
|
||||||
return IsDirectory()
|
return IsDirectory()
|
||||||
? FilePath(String(pathname_.c_str(), pathname_.length() - 1))
|
? FilePath(pathname_.substr(0, pathname_.length() - 1))
|
||||||
: *this;
|
: *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -40,7 +40,7 @@
|
|||||||
// GTEST_IMPLEMENTATION_ is defined to 1 iff the current translation unit is
|
// GTEST_IMPLEMENTATION_ is defined to 1 iff the current translation unit is
|
||||||
// part of Google Test's implementation; otherwise it's undefined.
|
// part of Google Test's implementation; otherwise it's undefined.
|
||||||
#if !GTEST_IMPLEMENTATION_
|
#if !GTEST_IMPLEMENTATION_
|
||||||
// A user is trying to include this from his code - just say no.
|
// If this file is included from the user's code, just say no.
|
||||||
# error "gtest-internal-inl.h is part of Google Test's internal implementation."
|
# error "gtest-internal-inl.h is part of Google Test's internal implementation."
|
||||||
# error "It must not be included except by Google Test itself."
|
# error "It must not be included except by Google Test itself."
|
||||||
#endif // GTEST_IMPLEMENTATION_
|
#endif // GTEST_IMPLEMENTATION_
|
||||||
@ -58,6 +58,11 @@
|
|||||||
|
|
||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
|
|
||||||
|
#if GTEST_CAN_STREAM_RESULTS_
|
||||||
|
# include <arpa/inet.h> // NOLINT
|
||||||
|
# include <netdb.h> // NOLINT
|
||||||
|
#endif
|
||||||
|
|
||||||
#if GTEST_OS_WINDOWS
|
#if GTEST_OS_WINDOWS
|
||||||
# include <windows.h> // NOLINT
|
# include <windows.h> // NOLINT
|
||||||
#endif // GTEST_OS_WINDOWS
|
#endif // GTEST_OS_WINDOWS
|
||||||
@ -112,6 +117,12 @@ GTEST_API_ bool ShouldUseColor(bool stdout_is_tty);
|
|||||||
// Formats the given time in milliseconds as seconds.
|
// Formats the given time in milliseconds as seconds.
|
||||||
GTEST_API_ std::string FormatTimeInMillisAsSeconds(TimeInMillis ms);
|
GTEST_API_ std::string FormatTimeInMillisAsSeconds(TimeInMillis ms);
|
||||||
|
|
||||||
|
// Converts the given time in milliseconds to a date string in the ISO 8601
|
||||||
|
// format, without the timezone information. N.B.: due to the use the
|
||||||
|
// non-reentrant localtime() function, this function is not thread safe. Do
|
||||||
|
// not use it in any code that can be called from multiple threads.
|
||||||
|
GTEST_API_ std::string FormatEpochTimeInMillisAsIso8601(TimeInMillis ms);
|
||||||
|
|
||||||
// Parses a string for an Int32 flag, in the form of "--flag=value".
|
// Parses a string for an Int32 flag, in the form of "--flag=value".
|
||||||
//
|
//
|
||||||
// On success, stores the value of the flag in *value, and returns
|
// On success, stores the value of the flag in *value, and returns
|
||||||
@ -190,37 +201,35 @@ class GTestFlagSaver {
|
|||||||
GTEST_FLAG(stream_result_to) = stream_result_to_;
|
GTEST_FLAG(stream_result_to) = stream_result_to_;
|
||||||
GTEST_FLAG(throw_on_failure) = throw_on_failure_;
|
GTEST_FLAG(throw_on_failure) = throw_on_failure_;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Fields for saving the original values of flags.
|
// Fields for saving the original values of flags.
|
||||||
bool also_run_disabled_tests_;
|
bool also_run_disabled_tests_;
|
||||||
bool break_on_failure_;
|
bool break_on_failure_;
|
||||||
bool catch_exceptions_;
|
bool catch_exceptions_;
|
||||||
String color_;
|
std::string color_;
|
||||||
String death_test_style_;
|
std::string death_test_style_;
|
||||||
bool death_test_use_fork_;
|
bool death_test_use_fork_;
|
||||||
String filter_;
|
std::string filter_;
|
||||||
String internal_run_death_test_;
|
std::string internal_run_death_test_;
|
||||||
bool list_tests_;
|
bool list_tests_;
|
||||||
String output_;
|
std::string output_;
|
||||||
bool print_time_;
|
bool print_time_;
|
||||||
bool pretty_;
|
|
||||||
internal::Int32 random_seed_;
|
internal::Int32 random_seed_;
|
||||||
internal::Int32 repeat_;
|
internal::Int32 repeat_;
|
||||||
bool shuffle_;
|
bool shuffle_;
|
||||||
internal::Int32 stack_trace_depth_;
|
internal::Int32 stack_trace_depth_;
|
||||||
String stream_result_to_;
|
std::string stream_result_to_;
|
||||||
bool throw_on_failure_;
|
bool throw_on_failure_;
|
||||||
} GTEST_ATTRIBUTE_UNUSED_;
|
} GTEST_ATTRIBUTE_UNUSED_;
|
||||||
|
|
||||||
// Converts a Unicode code point to a narrow string in UTF-8 encoding.
|
// Converts a Unicode code point to a narrow string in UTF-8 encoding.
|
||||||
// code_point parameter is of type UInt32 because wchar_t may not be
|
// code_point parameter is of type UInt32 because wchar_t may not be
|
||||||
// wide enough to contain a code point.
|
// wide enough to contain a code point.
|
||||||
// The output buffer str must containt at least 32 characters.
|
|
||||||
// The function returns the address of the output buffer.
|
|
||||||
// If the code_point is not a valid Unicode code point
|
// If the code_point is not a valid Unicode code point
|
||||||
// (i.e. outside of Unicode range U+0 to U+10FFFF) it will be output
|
// (i.e. outside of Unicode range U+0 to U+10FFFF) it will be converted
|
||||||
// as '(Invalid Unicode 0xXXXXXXXX)'.
|
// to "(Invalid Unicode 0xXXXXXXXX)".
|
||||||
GTEST_API_ char* CodePointToUtf8(UInt32 code_point, char* str);
|
GTEST_API_ std::string CodePointToUtf8(UInt32 code_point);
|
||||||
|
|
||||||
// Converts a wide string to a narrow string in UTF-8 encoding.
|
// Converts a wide string to a narrow string in UTF-8 encoding.
|
||||||
// The wide string is assumed to have the following encoding:
|
// The wide string is assumed to have the following encoding:
|
||||||
@ -235,7 +244,7 @@ GTEST_API_ char* CodePointToUtf8(UInt32 code_point, char* str);
|
|||||||
// as '(Invalid Unicode 0xXXXXXXXX)'. If the string is in UTF16 encoding
|
// as '(Invalid Unicode 0xXXXXXXXX)'. If the string is in UTF16 encoding
|
||||||
// and contains invalid UTF-16 surrogate pairs, values in those pairs
|
// and contains invalid UTF-16 surrogate pairs, values in those pairs
|
||||||
// will be encoded as individual Unicode characters from Basic Normal Plane.
|
// will be encoded as individual Unicode characters from Basic Normal Plane.
|
||||||
GTEST_API_ String WideStringToUtf8(const wchar_t* str, int num_chars);
|
GTEST_API_ std::string WideStringToUtf8(const wchar_t* str, int num_chars);
|
||||||
|
|
||||||
// Reads the GTEST_SHARD_STATUS_FILE environment variable, and creates the file
|
// Reads the GTEST_SHARD_STATUS_FILE environment variable, and creates the file
|
||||||
// if the variable is present. If a file already exists at this location, this
|
// if the variable is present. If a file already exists at this location, this
|
||||||
@ -339,16 +348,15 @@ class TestPropertyKeyIs {
|
|||||||
// Constructor.
|
// Constructor.
|
||||||
//
|
//
|
||||||
// TestPropertyKeyIs has NO default constructor.
|
// TestPropertyKeyIs has NO default constructor.
|
||||||
explicit TestPropertyKeyIs(const char* key)
|
explicit TestPropertyKeyIs(const std::string& key) : key_(key) {}
|
||||||
: key_(key) {}
|
|
||||||
|
|
||||||
// Returns true iff the test name of test property matches on key_.
|
// Returns true iff the test name of test property matches on key_.
|
||||||
bool operator()(const TestProperty& test_property) const {
|
bool operator()(const TestProperty& test_property) const {
|
||||||
return String(test_property.key()).Compare(key_) == 0;
|
return test_property.key() == key_;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
String key_;
|
std::string key_;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Class UnitTestOptions.
|
// Class UnitTestOptions.
|
||||||
@ -366,12 +374,12 @@ class GTEST_API_ UnitTestOptions {
|
|||||||
// Functions for processing the gtest_output flag.
|
// Functions for processing the gtest_output flag.
|
||||||
|
|
||||||
// Returns the output format, or "" for normal printed output.
|
// Returns the output format, or "" for normal printed output.
|
||||||
static String GetOutputFormat();
|
static std::string GetOutputFormat();
|
||||||
|
|
||||||
// Returns the absolute path of the requested output file, or the
|
// Returns the absolute path of the requested output file, or the
|
||||||
// default (test_detail.xml in the original working directory) if
|
// default (test_detail.xml in the original working directory) if
|
||||||
// none was explicitly specified.
|
// none was explicitly specified.
|
||||||
static String GetAbsolutePathToOutputFile();
|
static std::string GetAbsolutePathToOutputFile();
|
||||||
|
|
||||||
// Functions for processing the gtest_filter flag.
|
// Functions for processing the gtest_filter flag.
|
||||||
|
|
||||||
@ -384,8 +392,8 @@ class GTEST_API_ UnitTestOptions {
|
|||||||
|
|
||||||
// Returns true iff the user-specified filter matches the test case
|
// Returns true iff the user-specified filter matches the test case
|
||||||
// name and the test name.
|
// name and the test name.
|
||||||
static bool FilterMatchesTest(const String &test_case_name,
|
static bool FilterMatchesTest(const std::string &test_case_name,
|
||||||
const String &test_name);
|
const std::string &test_name);
|
||||||
|
|
||||||
#if GTEST_OS_WINDOWS
|
#if GTEST_OS_WINDOWS
|
||||||
// Function for supporting the gtest_catch_exception flag.
|
// Function for supporting the gtest_catch_exception flag.
|
||||||
@ -398,7 +406,7 @@ class GTEST_API_ UnitTestOptions {
|
|||||||
|
|
||||||
// Returns true if "name" matches the ':' separated list of glob-style
|
// Returns true if "name" matches the ':' separated list of glob-style
|
||||||
// filters in "filter".
|
// filters in "filter".
|
||||||
static bool MatchesFilter(const String& name, const char* filter);
|
static bool MatchesFilter(const std::string& name, const char* filter);
|
||||||
};
|
};
|
||||||
|
|
||||||
// Returns the current application's name, removing directory path if that
|
// Returns the current application's name, removing directory path if that
|
||||||
@ -411,13 +419,13 @@ class OsStackTraceGetterInterface {
|
|||||||
OsStackTraceGetterInterface() {}
|
OsStackTraceGetterInterface() {}
|
||||||
virtual ~OsStackTraceGetterInterface() {}
|
virtual ~OsStackTraceGetterInterface() {}
|
||||||
|
|
||||||
// Returns the current OS stack trace as a String. Parameters:
|
// Returns the current OS stack trace as an std::string. Parameters:
|
||||||
//
|
//
|
||||||
// max_depth - the maximum number of stack frames to be included
|
// max_depth - the maximum number of stack frames to be included
|
||||||
// in the trace.
|
// in the trace.
|
||||||
// skip_count - the number of top frames to be skipped; doesn't count
|
// skip_count - the number of top frames to be skipped; doesn't count
|
||||||
// against max_depth.
|
// against max_depth.
|
||||||
virtual String CurrentStackTrace(int max_depth, int skip_count) = 0;
|
virtual string CurrentStackTrace(int max_depth, int skip_count) = 0;
|
||||||
|
|
||||||
// UponLeavingGTest() should be called immediately before Google Test calls
|
// UponLeavingGTest() should be called immediately before Google Test calls
|
||||||
// user code. It saves some information about the current stack that
|
// user code. It saves some information about the current stack that
|
||||||
@ -432,8 +440,11 @@ class OsStackTraceGetterInterface {
|
|||||||
class OsStackTraceGetter : public OsStackTraceGetterInterface {
|
class OsStackTraceGetter : public OsStackTraceGetterInterface {
|
||||||
public:
|
public:
|
||||||
OsStackTraceGetter() : caller_frame_(NULL) {}
|
OsStackTraceGetter() : caller_frame_(NULL) {}
|
||||||
virtual String CurrentStackTrace(int max_depth, int skip_count);
|
|
||||||
virtual void UponLeavingGTest();
|
virtual string CurrentStackTrace(int max_depth, int skip_count)
|
||||||
|
GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
|
virtual void UponLeavingGTest() GTEST_LOCK_EXCLUDED_(mutex_);
|
||||||
|
|
||||||
// This string is inserted in place of stack frames that are part of
|
// This string is inserted in place of stack frames that are part of
|
||||||
// Google Test's implementation.
|
// Google Test's implementation.
|
||||||
@ -455,7 +466,7 @@ class OsStackTraceGetter : public OsStackTraceGetterInterface {
|
|||||||
struct TraceInfo {
|
struct TraceInfo {
|
||||||
const char* file;
|
const char* file;
|
||||||
int line;
|
int line;
|
||||||
String message;
|
std::string message;
|
||||||
};
|
};
|
||||||
|
|
||||||
// This is the default global test part result reporter used in UnitTestImpl.
|
// This is the default global test part result reporter used in UnitTestImpl.
|
||||||
@ -539,15 +550,25 @@ class GTEST_API_ UnitTestImpl {
|
|||||||
// Gets the number of failed tests.
|
// Gets the number of failed tests.
|
||||||
int failed_test_count() const;
|
int failed_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of disabled tests that will be reported in the XML report.
|
||||||
|
int reportable_disabled_test_count() const;
|
||||||
|
|
||||||
// Gets the number of disabled tests.
|
// Gets the number of disabled tests.
|
||||||
int disabled_test_count() const;
|
int disabled_test_count() const;
|
||||||
|
|
||||||
|
// Gets the number of tests to be printed in the XML report.
|
||||||
|
int reportable_test_count() const;
|
||||||
|
|
||||||
// Gets the number of all tests.
|
// Gets the number of all tests.
|
||||||
int total_test_count() const;
|
int total_test_count() const;
|
||||||
|
|
||||||
// Gets the number of tests that should run.
|
// Gets the number of tests that should run.
|
||||||
int test_to_run_count() const;
|
int test_to_run_count() const;
|
||||||
|
|
||||||
|
// Gets the time of the test program start, in ms from the start of the
|
||||||
|
// UNIX epoch.
|
||||||
|
TimeInMillis start_timestamp() const { return start_timestamp_; }
|
||||||
|
|
||||||
// Gets the elapsed time, in milliseconds.
|
// Gets the elapsed time, in milliseconds.
|
||||||
TimeInMillis elapsed_time() const { return elapsed_time_; }
|
TimeInMillis elapsed_time() const { return elapsed_time_; }
|
||||||
|
|
||||||
@ -596,7 +617,7 @@ class GTEST_API_ UnitTestImpl {
|
|||||||
// getter, and returns it.
|
// getter, and returns it.
|
||||||
OsStackTraceGetterInterface* os_stack_trace_getter();
|
OsStackTraceGetterInterface* os_stack_trace_getter();
|
||||||
|
|
||||||
// Returns the current OS stack trace as a String.
|
// Returns the current OS stack trace as an std::string.
|
||||||
//
|
//
|
||||||
// The maximum number of stack frames to be included is specified by
|
// The maximum number of stack frames to be included is specified by
|
||||||
// the gtest_stack_trace_depth flag. The skip_count parameter
|
// the gtest_stack_trace_depth flag. The skip_count parameter
|
||||||
@ -606,7 +627,7 @@ class GTEST_API_ UnitTestImpl {
|
|||||||
// For example, if Foo() calls Bar(), which in turn calls
|
// For example, if Foo() calls Bar(), which in turn calls
|
||||||
// CurrentOsStackTraceExceptTop(1), Foo() will be included in the
|
// CurrentOsStackTraceExceptTop(1), Foo() will be included in the
|
||||||
// trace but Bar() and CurrentOsStackTraceExceptTop() won't.
|
// trace but Bar() and CurrentOsStackTraceExceptTop() won't.
|
||||||
String CurrentOsStackTraceExceptTop(int skip_count);
|
std::string CurrentOsStackTraceExceptTop(int skip_count) GTEST_NO_INLINE_;
|
||||||
|
|
||||||
// Finds and returns a TestCase with the given name. If one doesn't
|
// Finds and returns a TestCase with the given name. If one doesn't
|
||||||
// exist, creates one and returns it.
|
// exist, creates one and returns it.
|
||||||
@ -696,6 +717,12 @@ class GTEST_API_ UnitTestImpl {
|
|||||||
ad_hoc_test_result_.Clear();
|
ad_hoc_test_result_.Clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Adds a TestProperty to the current TestResult object when invoked in a
|
||||||
|
// context of a test or a test case, or to the global property set. If the
|
||||||
|
// result already contains a property with the same key, the value will be
|
||||||
|
// updated.
|
||||||
|
void RecordProperty(const TestProperty& test_property);
|
||||||
|
|
||||||
enum ReactionToSharding {
|
enum ReactionToSharding {
|
||||||
HONOR_SHARDING_PROTOCOL,
|
HONOR_SHARDING_PROTOCOL,
|
||||||
IGNORE_SHARDING_PROTOCOL
|
IGNORE_SHARDING_PROTOCOL
|
||||||
@ -880,6 +907,10 @@ class GTEST_API_ UnitTestImpl {
|
|||||||
// Our random number generator.
|
// Our random number generator.
|
||||||
internal::Random random_;
|
internal::Random random_;
|
||||||
|
|
||||||
|
// The time of the test program start, in ms from the start of the
|
||||||
|
// UNIX epoch.
|
||||||
|
TimeInMillis start_timestamp_;
|
||||||
|
|
||||||
// How long the test took to run, in milliseconds.
|
// How long the test took to run, in milliseconds.
|
||||||
TimeInMillis elapsed_time_;
|
TimeInMillis elapsed_time_;
|
||||||
|
|
||||||
@ -935,33 +966,7 @@ GTEST_API_ void ParseGoogleTestFlagsOnly(int* argc, wchar_t** argv);
|
|||||||
|
|
||||||
// Returns the message describing the last system error, regardless of the
|
// Returns the message describing the last system error, regardless of the
|
||||||
// platform.
|
// platform.
|
||||||
GTEST_API_ String GetLastErrnoDescription();
|
GTEST_API_ std::string GetLastErrnoDescription();
|
||||||
|
|
||||||
# if GTEST_OS_WINDOWS
|
|
||||||
// Provides leak-safe Windows kernel handle ownership.
|
|
||||||
class AutoHandle {
|
|
||||||
public:
|
|
||||||
AutoHandle() : handle_(INVALID_HANDLE_VALUE) {}
|
|
||||||
explicit AutoHandle(HANDLE handle) : handle_(handle) {}
|
|
||||||
|
|
||||||
~AutoHandle() { Reset(); }
|
|
||||||
|
|
||||||
HANDLE Get() const { return handle_; }
|
|
||||||
void Reset() { Reset(INVALID_HANDLE_VALUE); }
|
|
||||||
void Reset(HANDLE handle) {
|
|
||||||
if (handle != handle_) {
|
|
||||||
if (handle_ != INVALID_HANDLE_VALUE)
|
|
||||||
::CloseHandle(handle_);
|
|
||||||
handle_ = handle;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
HANDLE handle_;
|
|
||||||
|
|
||||||
GTEST_DISALLOW_COPY_AND_ASSIGN_(AutoHandle);
|
|
||||||
};
|
|
||||||
# endif // GTEST_OS_WINDOWS
|
|
||||||
|
|
||||||
// Attempts to parse a string into a positive integer pointed to by the
|
// Attempts to parse a string into a positive integer pointed to by the
|
||||||
// number parameter. Returns true if that is possible.
|
// number parameter. Returns true if that is possible.
|
||||||
@ -1018,8 +1023,9 @@ bool ParseNaturalNumber(const ::std::string& str, Integer* number) {
|
|||||||
class TestResultAccessor {
|
class TestResultAccessor {
|
||||||
public:
|
public:
|
||||||
static void RecordProperty(TestResult* test_result,
|
static void RecordProperty(TestResult* test_result,
|
||||||
|
const std::string& xml_element,
|
||||||
const TestProperty& property) {
|
const TestProperty& property) {
|
||||||
test_result->RecordProperty(property);
|
test_result->RecordProperty(xml_element, property);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ClearTestPartResults(TestResult* test_result) {
|
static void ClearTestPartResults(TestResult* test_result) {
|
||||||
@ -1032,6 +1038,154 @@ class TestResultAccessor {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
#if GTEST_CAN_STREAM_RESULTS_
|
||||||
|
|
||||||
|
// Streams test results to the given port on the given host machine.
|
||||||
|
class StreamingListener : public EmptyTestEventListener {
|
||||||
|
public:
|
||||||
|
// Abstract base class for writing strings to a socket.
|
||||||
|
class AbstractSocketWriter {
|
||||||
|
public:
|
||||||
|
virtual ~AbstractSocketWriter() {}
|
||||||
|
|
||||||
|
// Sends a string to the socket.
|
||||||
|
virtual void Send(const string& message) = 0;
|
||||||
|
|
||||||
|
// Closes the socket.
|
||||||
|
virtual void CloseConnection() {}
|
||||||
|
|
||||||
|
// Sends a string and a newline to the socket.
|
||||||
|
void SendLn(const string& message) {
|
||||||
|
Send(message + "\n");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Concrete class for actually writing strings to a socket.
|
||||||
|
class SocketWriter : public AbstractSocketWriter {
|
||||||
|
public:
|
||||||
|
SocketWriter(const string& host, const string& port)
|
||||||
|
: sockfd_(-1), host_name_(host), port_num_(port) {
|
||||||
|
MakeConnection();
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual ~SocketWriter() {
|
||||||
|
if (sockfd_ != -1)
|
||||||
|
CloseConnection();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sends a string to the socket.
|
||||||
|
virtual void Send(const string& message) {
|
||||||
|
GTEST_CHECK_(sockfd_ != -1)
|
||||||
|
<< "Send() can be called only when there is a connection.";
|
||||||
|
|
||||||
|
const int len = static_cast<int>(message.length());
|
||||||
|
if (write(sockfd_, message.c_str(), len) != len) {
|
||||||
|
GTEST_LOG_(WARNING)
|
||||||
|
<< "stream_result_to: failed to stream to "
|
||||||
|
<< host_name_ << ":" << port_num_;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
// Creates a client socket and connects to the server.
|
||||||
|
void MakeConnection();
|
||||||
|
|
||||||
|
// Closes the socket.
|
||||||
|
void CloseConnection() {
|
||||||
|
GTEST_CHECK_(sockfd_ != -1)
|
||||||
|
<< "CloseConnection() can be called only when there is a connection.";
|
||||||
|
|
||||||
|
close(sockfd_);
|
||||||
|
sockfd_ = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int sockfd_; // socket file descriptor
|
||||||
|
const string host_name_;
|
||||||
|
const string port_num_;
|
||||||
|
|
||||||
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(SocketWriter);
|
||||||
|
}; // class SocketWriter
|
||||||
|
|
||||||
|
// Escapes '=', '&', '%', and '\n' characters in str as "%xx".
|
||||||
|
static string UrlEncode(const char* str);
|
||||||
|
|
||||||
|
StreamingListener(const string& host, const string& port)
|
||||||
|
: socket_writer_(new SocketWriter(host, port)) { Start(); }
|
||||||
|
|
||||||
|
explicit StreamingListener(AbstractSocketWriter* socket_writer)
|
||||||
|
: socket_writer_(socket_writer) { Start(); }
|
||||||
|
|
||||||
|
void OnTestProgramStart(const UnitTest& /* unit_test */) {
|
||||||
|
SendLn("event=TestProgramStart");
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestProgramEnd(const UnitTest& unit_test) {
|
||||||
|
// Note that Google Test current only report elapsed time for each
|
||||||
|
// test iteration, not for the entire test program.
|
||||||
|
SendLn("event=TestProgramEnd&passed=" + FormatBool(unit_test.Passed()));
|
||||||
|
|
||||||
|
// Notify the streaming server to stop.
|
||||||
|
socket_writer_->CloseConnection();
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestIterationStart(const UnitTest& /* unit_test */, int iteration) {
|
||||||
|
SendLn("event=TestIterationStart&iteration=" +
|
||||||
|
StreamableToString(iteration));
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestIterationEnd(const UnitTest& unit_test, int /* iteration */) {
|
||||||
|
SendLn("event=TestIterationEnd&passed=" +
|
||||||
|
FormatBool(unit_test.Passed()) + "&elapsed_time=" +
|
||||||
|
StreamableToString(unit_test.elapsed_time()) + "ms");
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestCaseStart(const TestCase& test_case) {
|
||||||
|
SendLn(std::string("event=TestCaseStart&name=") + test_case.name());
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestCaseEnd(const TestCase& test_case) {
|
||||||
|
SendLn("event=TestCaseEnd&passed=" + FormatBool(test_case.Passed())
|
||||||
|
+ "&elapsed_time=" + StreamableToString(test_case.elapsed_time())
|
||||||
|
+ "ms");
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestStart(const TestInfo& test_info) {
|
||||||
|
SendLn(std::string("event=TestStart&name=") + test_info.name());
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestEnd(const TestInfo& test_info) {
|
||||||
|
SendLn("event=TestEnd&passed=" +
|
||||||
|
FormatBool((test_info.result())->Passed()) +
|
||||||
|
"&elapsed_time=" +
|
||||||
|
StreamableToString((test_info.result())->elapsed_time()) + "ms");
|
||||||
|
}
|
||||||
|
|
||||||
|
void OnTestPartResult(const TestPartResult& test_part_result) {
|
||||||
|
const char* file_name = test_part_result.file_name();
|
||||||
|
if (file_name == NULL)
|
||||||
|
file_name = "";
|
||||||
|
SendLn("event=TestPartResult&file=" + UrlEncode(file_name) +
|
||||||
|
"&line=" + StreamableToString(test_part_result.line_number()) +
|
||||||
|
"&message=" + UrlEncode(test_part_result.message()));
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
// Sends the given message and a newline to the socket.
|
||||||
|
void SendLn(const string& message) { socket_writer_->SendLn(message); }
|
||||||
|
|
||||||
|
// Called at the start of streaming to notify the receiver what
|
||||||
|
// protocol we are using.
|
||||||
|
void Start() { SendLn("gtest_streaming_protocol_version=1.0"); }
|
||||||
|
|
||||||
|
string FormatBool(bool value) { return value ? "1" : "0"; }
|
||||||
|
|
||||||
|
const scoped_ptr<AbstractSocketWriter> socket_writer_;
|
||||||
|
|
||||||
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(StreamingListener);
|
||||||
|
}; // class StreamingListener
|
||||||
|
|
||||||
|
#endif // GTEST_CAN_STREAM_RESULTS_
|
||||||
|
|
||||||
} // namespace internal
|
} // namespace internal
|
||||||
} // namespace testing
|
} // namespace testing
|
||||||
|
|
||||||
|
@ -36,14 +36,14 @@
|
|||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#if GTEST_OS_WINDOWS_MOBILE
|
#if GTEST_OS_WINDOWS
|
||||||
# include <windows.h> // For TerminateProcess()
|
# include <windows.h>
|
||||||
#elif GTEST_OS_WINDOWS
|
|
||||||
# include <io.h>
|
# include <io.h>
|
||||||
# include <sys/stat.h>
|
# include <sys/stat.h>
|
||||||
|
# include <map> // Used in ThreadLocal.
|
||||||
#else
|
#else
|
||||||
# include <unistd.h>
|
# include <unistd.h>
|
||||||
#endif // GTEST_OS_WINDOWS_MOBILE
|
#endif // GTEST_OS_WINDOWS
|
||||||
|
|
||||||
#if GTEST_OS_MAC
|
#if GTEST_OS_MAC
|
||||||
# include <mach/mach_init.h>
|
# include <mach/mach_init.h>
|
||||||
@ -51,6 +51,12 @@
|
|||||||
# include <mach/vm_map.h>
|
# include <mach/vm_map.h>
|
||||||
#endif // GTEST_OS_MAC
|
#endif // GTEST_OS_MAC
|
||||||
|
|
||||||
|
#if GTEST_OS_QNX
|
||||||
|
# include <devctl.h>
|
||||||
|
# include <fcntl.h>
|
||||||
|
# include <sys/procfs.h>
|
||||||
|
#endif // GTEST_OS_QNX
|
||||||
|
|
||||||
#include "gtest/gtest-spi.h"
|
#include "gtest/gtest-spi.h"
|
||||||
#include "gtest/gtest-message.h"
|
#include "gtest/gtest-message.h"
|
||||||
#include "gtest/internal/gtest-internal.h"
|
#include "gtest/internal/gtest-internal.h"
|
||||||
@ -58,9 +64,9 @@
|
|||||||
|
|
||||||
// Indicates that this translation unit is part of Google Test's
|
// Indicates that this translation unit is part of Google Test's
|
||||||
// implementation. It must come before gtest-internal-inl.h is
|
// implementation. It must come before gtest-internal-inl.h is
|
||||||
// included, or there will be a compiler error. This trick is to
|
// included, or there will be a compiler error. This trick exists to
|
||||||
// prevent a user from accidentally including gtest-internal-inl.h in
|
// prevent the accidental inclusion of gtest-internal-inl.h in the
|
||||||
// his code.
|
// user's code.
|
||||||
#define GTEST_IMPLEMENTATION_ 1
|
#define GTEST_IMPLEMENTATION_ 1
|
||||||
#include "src/gtest-internal-inl.h"
|
#include "src/gtest-internal-inl.h"
|
||||||
#undef GTEST_IMPLEMENTATION_
|
#undef GTEST_IMPLEMENTATION_
|
||||||
@ -98,6 +104,26 @@ size_t GetThreadCount() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#elif GTEST_OS_QNX
|
||||||
|
|
||||||
|
// Returns the number of threads running in the process, or 0 to indicate that
|
||||||
|
// we cannot detect it.
|
||||||
|
size_t GetThreadCount() {
|
||||||
|
const int fd = open("/proc/self/as", O_RDONLY);
|
||||||
|
if (fd < 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
procfs_info process_info;
|
||||||
|
const int status =
|
||||||
|
devctl(fd, DCMD_PROC_INFO, &process_info, sizeof(process_info), NULL);
|
||||||
|
close(fd);
|
||||||
|
if (status == EOK) {
|
||||||
|
return static_cast<size_t>(process_info.num_threads);
|
||||||
|
} else {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
size_t GetThreadCount() {
|
size_t GetThreadCount() {
|
||||||
@ -108,6 +134,389 @@ size_t GetThreadCount() {
|
|||||||
|
|
||||||
#endif // GTEST_OS_MAC
|
#endif // GTEST_OS_MAC
|
||||||
|
|
||||||
|
#if GTEST_IS_THREADSAFE && GTEST_OS_WINDOWS
|
||||||
|
|
||||||
|
void SleepMilliseconds(int n) {
|
||||||
|
::Sleep(n);
|
||||||
|
}
|
||||||
|
|
||||||
|
AutoHandle::AutoHandle()
|
||||||
|
: handle_(INVALID_HANDLE_VALUE) {}
|
||||||
|
|
||||||
|
AutoHandle::AutoHandle(Handle handle)
|
||||||
|
: handle_(handle) {}
|
||||||
|
|
||||||
|
AutoHandle::~AutoHandle() {
|
||||||
|
Reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
AutoHandle::Handle AutoHandle::Get() const {
|
||||||
|
return handle_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void AutoHandle::Reset() {
|
||||||
|
Reset(INVALID_HANDLE_VALUE);
|
||||||
|
}
|
||||||
|
|
||||||
|
void AutoHandle::Reset(HANDLE handle) {
|
||||||
|
// Resetting with the same handle we already own is invalid.
|
||||||
|
if (handle_ != handle) {
|
||||||
|
if (IsCloseable()) {
|
||||||
|
::CloseHandle(handle_);
|
||||||
|
}
|
||||||
|
handle_ = handle;
|
||||||
|
} else {
|
||||||
|
GTEST_CHECK_(!IsCloseable())
|
||||||
|
<< "Resetting a valid handle to itself is likely a programmer error "
|
||||||
|
"and thus not allowed.";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool AutoHandle::IsCloseable() const {
|
||||||
|
// Different Windows APIs may use either of these values to represent an
|
||||||
|
// invalid handle.
|
||||||
|
return handle_ != NULL && handle_ != INVALID_HANDLE_VALUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
Notification::Notification()
|
||||||
|
: event_(::CreateEvent(NULL, // Default security attributes.
|
||||||
|
TRUE, // Do not reset automatically.
|
||||||
|
FALSE, // Initially unset.
|
||||||
|
NULL)) { // Anonymous event.
|
||||||
|
GTEST_CHECK_(event_.Get() != NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Notification::Notify() {
|
||||||
|
GTEST_CHECK_(::SetEvent(event_.Get()) != FALSE);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Notification::WaitForNotification() {
|
||||||
|
GTEST_CHECK_(
|
||||||
|
::WaitForSingleObject(event_.Get(), INFINITE) == WAIT_OBJECT_0);
|
||||||
|
}
|
||||||
|
|
||||||
|
Mutex::Mutex()
|
||||||
|
: type_(kDynamic),
|
||||||
|
owner_thread_id_(0),
|
||||||
|
critical_section_init_phase_(0),
|
||||||
|
critical_section_(new CRITICAL_SECTION) {
|
||||||
|
::InitializeCriticalSection(critical_section_);
|
||||||
|
}
|
||||||
|
|
||||||
|
Mutex::~Mutex() {
|
||||||
|
// Static mutexes are leaked intentionally. It is not thread-safe to try
|
||||||
|
// to clean them up.
|
||||||
|
// TODO(yukawa): Switch to Slim Reader/Writer (SRW) Locks, which requires
|
||||||
|
// nothing to clean it up but is available only on Vista and later.
|
||||||
|
// http://msdn.microsoft.com/en-us/library/windows/desktop/aa904937.aspx
|
||||||
|
if (type_ == kDynamic) {
|
||||||
|
::DeleteCriticalSection(critical_section_);
|
||||||
|
delete critical_section_;
|
||||||
|
critical_section_ = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Mutex::Lock() {
|
||||||
|
ThreadSafeLazyInit();
|
||||||
|
::EnterCriticalSection(critical_section_);
|
||||||
|
owner_thread_id_ = ::GetCurrentThreadId();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Mutex::Unlock() {
|
||||||
|
ThreadSafeLazyInit();
|
||||||
|
// We don't protect writing to owner_thread_id_ here, as it's the
|
||||||
|
// caller's responsibility to ensure that the current thread holds the
|
||||||
|
// mutex when this is called.
|
||||||
|
owner_thread_id_ = 0;
|
||||||
|
::LeaveCriticalSection(critical_section_);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Does nothing if the current thread holds the mutex. Otherwise, crashes
|
||||||
|
// with high probability.
|
||||||
|
void Mutex::AssertHeld() {
|
||||||
|
ThreadSafeLazyInit();
|
||||||
|
GTEST_CHECK_(owner_thread_id_ == ::GetCurrentThreadId())
|
||||||
|
<< "The current thread is not holding the mutex @" << this;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initializes owner_thread_id_ and critical_section_ in static mutexes.
|
||||||
|
void Mutex::ThreadSafeLazyInit() {
|
||||||
|
// Dynamic mutexes are initialized in the constructor.
|
||||||
|
if (type_ == kStatic) {
|
||||||
|
switch (
|
||||||
|
::InterlockedCompareExchange(&critical_section_init_phase_, 1L, 0L)) {
|
||||||
|
case 0:
|
||||||
|
// If critical_section_init_phase_ was 0 before the exchange, we
|
||||||
|
// are the first to test it and need to perform the initialization.
|
||||||
|
owner_thread_id_ = 0;
|
||||||
|
critical_section_ = new CRITICAL_SECTION;
|
||||||
|
::InitializeCriticalSection(critical_section_);
|
||||||
|
// Updates the critical_section_init_phase_ to 2 to signal
|
||||||
|
// initialization complete.
|
||||||
|
GTEST_CHECK_(::InterlockedCompareExchange(
|
||||||
|
&critical_section_init_phase_, 2L, 1L) ==
|
||||||
|
1L);
|
||||||
|
break;
|
||||||
|
case 1:
|
||||||
|
// Somebody else is already initializing the mutex; spin until they
|
||||||
|
// are done.
|
||||||
|
while (::InterlockedCompareExchange(&critical_section_init_phase_,
|
||||||
|
2L,
|
||||||
|
2L) != 2L) {
|
||||||
|
// Possibly yields the rest of the thread's time slice to other
|
||||||
|
// threads.
|
||||||
|
::Sleep(0);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 2:
|
||||||
|
break; // The mutex is already initialized and ready for use.
|
||||||
|
|
||||||
|
default:
|
||||||
|
GTEST_CHECK_(false)
|
||||||
|
<< "Unexpected value of critical_section_init_phase_ "
|
||||||
|
<< "while initializing a static mutex.";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
class ThreadWithParamSupport : public ThreadWithParamBase {
|
||||||
|
public:
|
||||||
|
static HANDLE CreateThread(Runnable* runnable,
|
||||||
|
Notification* thread_can_start) {
|
||||||
|
ThreadMainParam* param = new ThreadMainParam(runnable, thread_can_start);
|
||||||
|
DWORD thread_id;
|
||||||
|
// TODO(yukawa): Consider to use _beginthreadex instead.
|
||||||
|
HANDLE thread_handle = ::CreateThread(
|
||||||
|
NULL, // Default security.
|
||||||
|
0, // Default stack size.
|
||||||
|
&ThreadWithParamSupport::ThreadMain,
|
||||||
|
param, // Parameter to ThreadMainStatic
|
||||||
|
0x0, // Default creation flags.
|
||||||
|
&thread_id); // Need a valid pointer for the call to work under Win98.
|
||||||
|
GTEST_CHECK_(thread_handle != NULL) << "CreateThread failed with error "
|
||||||
|
<< ::GetLastError() << ".";
|
||||||
|
if (thread_handle == NULL) {
|
||||||
|
delete param;
|
||||||
|
}
|
||||||
|
return thread_handle;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
struct ThreadMainParam {
|
||||||
|
ThreadMainParam(Runnable* runnable, Notification* thread_can_start)
|
||||||
|
: runnable_(runnable),
|
||||||
|
thread_can_start_(thread_can_start) {
|
||||||
|
}
|
||||||
|
scoped_ptr<Runnable> runnable_;
|
||||||
|
// Does not own.
|
||||||
|
Notification* thread_can_start_;
|
||||||
|
};
|
||||||
|
|
||||||
|
static DWORD WINAPI ThreadMain(void* ptr) {
|
||||||
|
// Transfers ownership.
|
||||||
|
scoped_ptr<ThreadMainParam> param(static_cast<ThreadMainParam*>(ptr));
|
||||||
|
if (param->thread_can_start_ != NULL)
|
||||||
|
param->thread_can_start_->WaitForNotification();
|
||||||
|
param->runnable_->Run();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Prohibit instantiation.
|
||||||
|
ThreadWithParamSupport();
|
||||||
|
|
||||||
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(ThreadWithParamSupport);
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
ThreadWithParamBase::ThreadWithParamBase(Runnable *runnable,
|
||||||
|
Notification* thread_can_start)
|
||||||
|
: thread_(ThreadWithParamSupport::CreateThread(runnable,
|
||||||
|
thread_can_start)) {
|
||||||
|
}
|
||||||
|
|
||||||
|
ThreadWithParamBase::~ThreadWithParamBase() {
|
||||||
|
Join();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ThreadWithParamBase::Join() {
|
||||||
|
GTEST_CHECK_(::WaitForSingleObject(thread_.Get(), INFINITE) == WAIT_OBJECT_0)
|
||||||
|
<< "Failed to join the thread with error " << ::GetLastError() << ".";
|
||||||
|
}
|
||||||
|
|
||||||
|
// Maps a thread to a set of ThreadIdToThreadLocals that have values
|
||||||
|
// instantiated on that thread and notifies them when the thread exits. A
|
||||||
|
// ThreadLocal instance is expected to persist until all threads it has
|
||||||
|
// values on have terminated.
|
||||||
|
class ThreadLocalRegistryImpl {
|
||||||
|
public:
|
||||||
|
// Registers thread_local_instance as having value on the current thread.
|
||||||
|
// Returns a value that can be used to identify the thread from other threads.
|
||||||
|
static ThreadLocalValueHolderBase* GetValueOnCurrentThread(
|
||||||
|
const ThreadLocalBase* thread_local_instance) {
|
||||||
|
DWORD current_thread = ::GetCurrentThreadId();
|
||||||
|
MutexLock lock(&mutex_);
|
||||||
|
ThreadIdToThreadLocals* const thread_to_thread_locals =
|
||||||
|
GetThreadLocalsMapLocked();
|
||||||
|
ThreadIdToThreadLocals::iterator thread_local_pos =
|
||||||
|
thread_to_thread_locals->find(current_thread);
|
||||||
|
if (thread_local_pos == thread_to_thread_locals->end()) {
|
||||||
|
thread_local_pos = thread_to_thread_locals->insert(
|
||||||
|
std::make_pair(current_thread, ThreadLocalValues())).first;
|
||||||
|
StartWatcherThreadFor(current_thread);
|
||||||
|
}
|
||||||
|
ThreadLocalValues& thread_local_values = thread_local_pos->second;
|
||||||
|
ThreadLocalValues::iterator value_pos =
|
||||||
|
thread_local_values.find(thread_local_instance);
|
||||||
|
if (value_pos == thread_local_values.end()) {
|
||||||
|
value_pos =
|
||||||
|
thread_local_values
|
||||||
|
.insert(std::make_pair(
|
||||||
|
thread_local_instance,
|
||||||
|
linked_ptr<ThreadLocalValueHolderBase>(
|
||||||
|
thread_local_instance->NewValueForCurrentThread())))
|
||||||
|
.first;
|
||||||
|
}
|
||||||
|
return value_pos->second.get();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void OnThreadLocalDestroyed(
|
||||||
|
const ThreadLocalBase* thread_local_instance) {
|
||||||
|
std::vector<linked_ptr<ThreadLocalValueHolderBase> > value_holders;
|
||||||
|
// Clean up the ThreadLocalValues data structure while holding the lock, but
|
||||||
|
// defer the destruction of the ThreadLocalValueHolderBases.
|
||||||
|
{
|
||||||
|
MutexLock lock(&mutex_);
|
||||||
|
ThreadIdToThreadLocals* const thread_to_thread_locals =
|
||||||
|
GetThreadLocalsMapLocked();
|
||||||
|
for (ThreadIdToThreadLocals::iterator it =
|
||||||
|
thread_to_thread_locals->begin();
|
||||||
|
it != thread_to_thread_locals->end();
|
||||||
|
++it) {
|
||||||
|
ThreadLocalValues& thread_local_values = it->second;
|
||||||
|
ThreadLocalValues::iterator value_pos =
|
||||||
|
thread_local_values.find(thread_local_instance);
|
||||||
|
if (value_pos != thread_local_values.end()) {
|
||||||
|
value_holders.push_back(value_pos->second);
|
||||||
|
thread_local_values.erase(value_pos);
|
||||||
|
// This 'if' can only be successful at most once, so theoretically we
|
||||||
|
// could break out of the loop here, but we don't bother doing so.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Outside the lock, let the destructor for 'value_holders' deallocate the
|
||||||
|
// ThreadLocalValueHolderBases.
|
||||||
|
}
|
||||||
|
|
||||||
|
static void OnThreadExit(DWORD thread_id) {
|
||||||
|
GTEST_CHECK_(thread_id != 0) << ::GetLastError();
|
||||||
|
std::vector<linked_ptr<ThreadLocalValueHolderBase> > value_holders;
|
||||||
|
// Clean up the ThreadIdToThreadLocals data structure while holding the
|
||||||
|
// lock, but defer the destruction of the ThreadLocalValueHolderBases.
|
||||||
|
{
|
||||||
|
MutexLock lock(&mutex_);
|
||||||
|
ThreadIdToThreadLocals* const thread_to_thread_locals =
|
||||||
|
GetThreadLocalsMapLocked();
|
||||||
|
ThreadIdToThreadLocals::iterator thread_local_pos =
|
||||||
|
thread_to_thread_locals->find(thread_id);
|
||||||
|
if (thread_local_pos != thread_to_thread_locals->end()) {
|
||||||
|
ThreadLocalValues& thread_local_values = thread_local_pos->second;
|
||||||
|
for (ThreadLocalValues::iterator value_pos =
|
||||||
|
thread_local_values.begin();
|
||||||
|
value_pos != thread_local_values.end();
|
||||||
|
++value_pos) {
|
||||||
|
value_holders.push_back(value_pos->second);
|
||||||
|
}
|
||||||
|
thread_to_thread_locals->erase(thread_local_pos);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Outside the lock, let the destructor for 'value_holders' deallocate the
|
||||||
|
// ThreadLocalValueHolderBases.
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
// In a particular thread, maps a ThreadLocal object to its value.
|
||||||
|
typedef std::map<const ThreadLocalBase*,
|
||||||
|
linked_ptr<ThreadLocalValueHolderBase> > ThreadLocalValues;
|
||||||
|
// Stores all ThreadIdToThreadLocals having values in a thread, indexed by
|
||||||
|
// thread's ID.
|
||||||
|
typedef std::map<DWORD, ThreadLocalValues> ThreadIdToThreadLocals;
|
||||||
|
|
||||||
|
// Holds the thread id and thread handle that we pass from
|
||||||
|
// StartWatcherThreadFor to WatcherThreadFunc.
|
||||||
|
typedef std::pair<DWORD, HANDLE> ThreadIdAndHandle;
|
||||||
|
|
||||||
|
static void StartWatcherThreadFor(DWORD thread_id) {
|
||||||
|
// The returned handle will be kept in thread_map and closed by
|
||||||
|
// watcher_thread in WatcherThreadFunc.
|
||||||
|
HANDLE thread = ::OpenThread(SYNCHRONIZE | THREAD_QUERY_INFORMATION,
|
||||||
|
FALSE,
|
||||||
|
thread_id);
|
||||||
|
GTEST_CHECK_(thread != NULL);
|
||||||
|
// We need to to pass a valid thread ID pointer into CreateThread for it
|
||||||
|
// to work correctly under Win98.
|
||||||
|
DWORD watcher_thread_id;
|
||||||
|
HANDLE watcher_thread = ::CreateThread(
|
||||||
|
NULL, // Default security.
|
||||||
|
0, // Default stack size
|
||||||
|
&ThreadLocalRegistryImpl::WatcherThreadFunc,
|
||||||
|
reinterpret_cast<LPVOID>(new ThreadIdAndHandle(thread_id, thread)),
|
||||||
|
CREATE_SUSPENDED,
|
||||||
|
&watcher_thread_id);
|
||||||
|
GTEST_CHECK_(watcher_thread != NULL);
|
||||||
|
// Give the watcher thread the same priority as ours to avoid being
|
||||||
|
// blocked by it.
|
||||||
|
::SetThreadPriority(watcher_thread,
|
||||||
|
::GetThreadPriority(::GetCurrentThread()));
|
||||||
|
::ResumeThread(watcher_thread);
|
||||||
|
::CloseHandle(watcher_thread);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Monitors exit from a given thread and notifies those
|
||||||
|
// ThreadIdToThreadLocals about thread termination.
|
||||||
|
static DWORD WINAPI WatcherThreadFunc(LPVOID param) {
|
||||||
|
const ThreadIdAndHandle* tah =
|
||||||
|
reinterpret_cast<const ThreadIdAndHandle*>(param);
|
||||||
|
GTEST_CHECK_(
|
||||||
|
::WaitForSingleObject(tah->second, INFINITE) == WAIT_OBJECT_0);
|
||||||
|
OnThreadExit(tah->first);
|
||||||
|
::CloseHandle(tah->second);
|
||||||
|
delete tah;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns map of thread local instances.
|
||||||
|
static ThreadIdToThreadLocals* GetThreadLocalsMapLocked() {
|
||||||
|
mutex_.AssertHeld();
|
||||||
|
static ThreadIdToThreadLocals* map = new ThreadIdToThreadLocals;
|
||||||
|
return map;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Protects access to GetThreadLocalsMapLocked() and its return value.
|
||||||
|
static Mutex mutex_;
|
||||||
|
// Protects access to GetThreadMapLocked() and its return value.
|
||||||
|
static Mutex thread_map_mutex_;
|
||||||
|
};
|
||||||
|
|
||||||
|
Mutex ThreadLocalRegistryImpl::mutex_(Mutex::kStaticMutex);
|
||||||
|
Mutex ThreadLocalRegistryImpl::thread_map_mutex_(Mutex::kStaticMutex);
|
||||||
|
|
||||||
|
ThreadLocalValueHolderBase* ThreadLocalRegistry::GetValueOnCurrentThread(
|
||||||
|
const ThreadLocalBase* thread_local_instance) {
|
||||||
|
return ThreadLocalRegistryImpl::GetValueOnCurrentThread(
|
||||||
|
thread_local_instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
void ThreadLocalRegistry::OnThreadLocalDestroyed(
|
||||||
|
const ThreadLocalBase* thread_local_instance) {
|
||||||
|
ThreadLocalRegistryImpl::OnThreadLocalDestroyed(thread_local_instance);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // GTEST_IS_THREADSAFE && GTEST_OS_WINDOWS
|
||||||
|
|
||||||
#if GTEST_USES_POSIX_RE
|
#if GTEST_USES_POSIX_RE
|
||||||
|
|
||||||
// Implements RE. Currently only needed for death tests.
|
// Implements RE. Currently only needed for death tests.
|
||||||
@ -222,7 +631,7 @@ bool AtomMatchesChar(bool escaped, char pattern_char, char ch) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Helper function used by ValidateRegex() to format error messages.
|
// Helper function used by ValidateRegex() to format error messages.
|
||||||
String FormatRegexSyntaxError(const char* regex, int index) {
|
std::string FormatRegexSyntaxError(const char* regex, int index) {
|
||||||
return (Message() << "Syntax error at index " << index
|
return (Message() << "Syntax error at index " << index
|
||||||
<< " in simple regular expression \"" << regex << "\": ").GetString();
|
<< " in simple regular expression \"" << regex << "\": ").GetString();
|
||||||
}
|
}
|
||||||
@ -429,15 +838,15 @@ const char kUnknownFile[] = "unknown file";
|
|||||||
// Formats a source file path and a line number as they would appear
|
// Formats a source file path and a line number as they would appear
|
||||||
// in an error message from the compiler used to compile this code.
|
// in an error message from the compiler used to compile this code.
|
||||||
GTEST_API_ ::std::string FormatFileLocation(const char* file, int line) {
|
GTEST_API_ ::std::string FormatFileLocation(const char* file, int line) {
|
||||||
const char* const file_name = file == NULL ? kUnknownFile : file;
|
const std::string file_name(file == NULL ? kUnknownFile : file);
|
||||||
|
|
||||||
if (line < 0) {
|
if (line < 0) {
|
||||||
return String::Format("%s:", file_name).c_str();
|
return file_name + ":";
|
||||||
}
|
}
|
||||||
#ifdef _MSC_VER
|
#ifdef _MSC_VER
|
||||||
return String::Format("%s(%d):", file_name, line).c_str();
|
return file_name + "(" + StreamableToString(line) + "):";
|
||||||
#else
|
#else
|
||||||
return String::Format("%s:%d:", file_name, line).c_str();
|
return file_name + ":" + StreamableToString(line) + ":";
|
||||||
#endif // _MSC_VER
|
#endif // _MSC_VER
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -448,12 +857,12 @@ GTEST_API_ ::std::string FormatFileLocation(const char* file, int line) {
|
|||||||
// to the file location it produces, unlike FormatFileLocation().
|
// to the file location it produces, unlike FormatFileLocation().
|
||||||
GTEST_API_ ::std::string FormatCompilerIndependentFileLocation(
|
GTEST_API_ ::std::string FormatCompilerIndependentFileLocation(
|
||||||
const char* file, int line) {
|
const char* file, int line) {
|
||||||
const char* const file_name = file == NULL ? kUnknownFile : file;
|
const std::string file_name(file == NULL ? kUnknownFile : file);
|
||||||
|
|
||||||
if (line < 0)
|
if (line < 0)
|
||||||
return file_name;
|
return file_name;
|
||||||
else
|
else
|
||||||
return String::Format("%s:%d", file_name, line).c_str();
|
return file_name + ":" + StreamableToString(line);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -477,10 +886,7 @@ GTestLog::~GTestLog() {
|
|||||||
}
|
}
|
||||||
// Disable Microsoft deprecation warnings for POSIX functions called from
|
// Disable Microsoft deprecation warnings for POSIX functions called from
|
||||||
// this class (creat, dup, dup2, and close)
|
// this class (creat, dup, dup2, and close)
|
||||||
#ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_PUSH_(4996)
|
||||||
# pragma warning(push)
|
|
||||||
# pragma warning(disable: 4996)
|
|
||||||
#endif // _MSC_VER
|
|
||||||
|
|
||||||
#if GTEST_HAS_STREAM_REDIRECTION
|
#if GTEST_HAS_STREAM_REDIRECTION
|
||||||
|
|
||||||
@ -488,8 +894,7 @@ GTestLog::~GTestLog() {
|
|||||||
class CapturedStream {
|
class CapturedStream {
|
||||||
public:
|
public:
|
||||||
// The ctor redirects the stream to a temporary file.
|
// The ctor redirects the stream to a temporary file.
|
||||||
CapturedStream(int fd) : fd_(fd), uncaptured_fd_(dup(fd)) {
|
explicit CapturedStream(int fd) : fd_(fd), uncaptured_fd_(dup(fd)) {
|
||||||
|
|
||||||
# if GTEST_OS_WINDOWS
|
# if GTEST_OS_WINDOWS
|
||||||
char temp_dir_path[MAX_PATH + 1] = { '\0' }; // NOLINT
|
char temp_dir_path[MAX_PATH + 1] = { '\0' }; // NOLINT
|
||||||
char temp_file_path[MAX_PATH + 1] = { '\0' }; // NOLINT
|
char temp_file_path[MAX_PATH + 1] = { '\0' }; // NOLINT
|
||||||
@ -506,10 +911,29 @@ class CapturedStream {
|
|||||||
<< temp_file_path;
|
<< temp_file_path;
|
||||||
filename_ = temp_file_path;
|
filename_ = temp_file_path;
|
||||||
# else
|
# else
|
||||||
// There's no guarantee that a test has write access to the
|
// There's no guarantee that a test has write access to the current
|
||||||
// current directory, so we create the temporary file in the /tmp
|
// directory, so we create the temporary file in the /tmp directory
|
||||||
// directory instead.
|
// instead. We use /tmp on most systems, and /sdcard on Android.
|
||||||
|
// That's because Android doesn't have /tmp.
|
||||||
|
# if GTEST_OS_LINUX_ANDROID
|
||||||
|
// Note: Android applications are expected to call the framework's
|
||||||
|
// Context.getExternalStorageDirectory() method through JNI to get
|
||||||
|
// the location of the world-writable SD Card directory. However,
|
||||||
|
// this requires a Context handle, which cannot be retrieved
|
||||||
|
// globally from native code. Doing so also precludes running the
|
||||||
|
// code as part of a regular standalone executable, which doesn't
|
||||||
|
// run in a Dalvik process (e.g. when running it through 'adb shell').
|
||||||
|
//
|
||||||
|
// The location /sdcard is directly accessible from native code
|
||||||
|
// and is the only location (unofficially) supported by the Android
|
||||||
|
// team. It's generally a symlink to the real SD Card mount point
|
||||||
|
// which can be /mnt/sdcard, /mnt/sdcard0, /system/media/sdcard, or
|
||||||
|
// other OEM-customized locations. Never rely on these, and always
|
||||||
|
// use /sdcard.
|
||||||
|
char name_template[] = "/sdcard/gtest_captured_stream.XXXXXX";
|
||||||
|
# else
|
||||||
char name_template[] = "/tmp/captured_stream.XXXXXX";
|
char name_template[] = "/tmp/captured_stream.XXXXXX";
|
||||||
|
# endif // GTEST_OS_LINUX_ANDROID
|
||||||
const int captured_fd = mkstemp(name_template);
|
const int captured_fd = mkstemp(name_template);
|
||||||
filename_ = name_template;
|
filename_ = name_template;
|
||||||
# endif // GTEST_OS_WINDOWS
|
# endif // GTEST_OS_WINDOWS
|
||||||
@ -522,7 +946,7 @@ class CapturedStream {
|
|||||||
remove(filename_.c_str());
|
remove(filename_.c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
String GetCapturedString() {
|
std::string GetCapturedString() {
|
||||||
if (uncaptured_fd_ != -1) {
|
if (uncaptured_fd_ != -1) {
|
||||||
// Restores the original stream.
|
// Restores the original stream.
|
||||||
fflush(NULL);
|
fflush(NULL);
|
||||||
@ -532,14 +956,14 @@ class CapturedStream {
|
|||||||
}
|
}
|
||||||
|
|
||||||
FILE* const file = posix::FOpen(filename_.c_str(), "r");
|
FILE* const file = posix::FOpen(filename_.c_str(), "r");
|
||||||
const String content = ReadEntireFile(file);
|
const std::string content = ReadEntireFile(file);
|
||||||
posix::FClose(file);
|
posix::FClose(file);
|
||||||
return content;
|
return content;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Reads the entire content of a file as a String.
|
// Reads the entire content of a file as an std::string.
|
||||||
static String ReadEntireFile(FILE* file);
|
static std::string ReadEntireFile(FILE* file);
|
||||||
|
|
||||||
// Returns the size (in bytes) of a file.
|
// Returns the size (in bytes) of a file.
|
||||||
static size_t GetFileSize(FILE* file);
|
static size_t GetFileSize(FILE* file);
|
||||||
@ -559,7 +983,7 @@ size_t CapturedStream::GetFileSize(FILE* file) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Reads the entire content of a file as a string.
|
// Reads the entire content of a file as a string.
|
||||||
String CapturedStream::ReadEntireFile(FILE* file) {
|
std::string CapturedStream::ReadEntireFile(FILE* file) {
|
||||||
const size_t file_size = GetFileSize(file);
|
const size_t file_size = GetFileSize(file);
|
||||||
char* const buffer = new char[file_size];
|
char* const buffer = new char[file_size];
|
||||||
|
|
||||||
@ -575,15 +999,13 @@ String CapturedStream::ReadEntireFile(FILE* file) {
|
|||||||
bytes_read += bytes_last_read;
|
bytes_read += bytes_last_read;
|
||||||
} while (bytes_last_read > 0 && bytes_read < file_size);
|
} while (bytes_last_read > 0 && bytes_read < file_size);
|
||||||
|
|
||||||
const String content(buffer, bytes_read);
|
const std::string content(buffer, bytes_read);
|
||||||
delete[] buffer;
|
delete[] buffer;
|
||||||
|
|
||||||
return content;
|
return content;
|
||||||
}
|
}
|
||||||
|
|
||||||
# ifdef _MSC_VER
|
GTEST_DISABLE_MSC_WARNINGS_POP_()
|
||||||
# pragma warning(pop)
|
|
||||||
# endif // _MSC_VER
|
|
||||||
|
|
||||||
static CapturedStream* g_captured_stderr = NULL;
|
static CapturedStream* g_captured_stderr = NULL;
|
||||||
static CapturedStream* g_captured_stdout = NULL;
|
static CapturedStream* g_captured_stdout = NULL;
|
||||||
@ -598,8 +1020,8 @@ void CaptureStream(int fd, const char* stream_name, CapturedStream** stream) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Stops capturing the output stream and returns the captured string.
|
// Stops capturing the output stream and returns the captured string.
|
||||||
String GetCapturedStream(CapturedStream** captured_stream) {
|
std::string GetCapturedStream(CapturedStream** captured_stream) {
|
||||||
const String content = (*captured_stream)->GetCapturedString();
|
const std::string content = (*captured_stream)->GetCapturedString();
|
||||||
|
|
||||||
delete *captured_stream;
|
delete *captured_stream;
|
||||||
*captured_stream = NULL;
|
*captured_stream = NULL;
|
||||||
@ -618,21 +1040,37 @@ void CaptureStderr() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Stops capturing stdout and returns the captured string.
|
// Stops capturing stdout and returns the captured string.
|
||||||
String GetCapturedStdout() { return GetCapturedStream(&g_captured_stdout); }
|
std::string GetCapturedStdout() {
|
||||||
|
return GetCapturedStream(&g_captured_stdout);
|
||||||
|
}
|
||||||
|
|
||||||
// Stops capturing stderr and returns the captured string.
|
// Stops capturing stderr and returns the captured string.
|
||||||
String GetCapturedStderr() { return GetCapturedStream(&g_captured_stderr); }
|
std::string GetCapturedStderr() {
|
||||||
|
return GetCapturedStream(&g_captured_stderr);
|
||||||
|
}
|
||||||
|
|
||||||
#endif // GTEST_HAS_STREAM_REDIRECTION
|
#endif // GTEST_HAS_STREAM_REDIRECTION
|
||||||
|
|
||||||
#if GTEST_HAS_DEATH_TEST
|
#if GTEST_HAS_DEATH_TEST
|
||||||
|
|
||||||
// A copy of all command line arguments. Set by InitGoogleTest().
|
// A copy of all command line arguments. Set by InitGoogleTest().
|
||||||
::std::vector<String> g_argvs;
|
::std::vector<testing::internal::string> g_argvs;
|
||||||
|
|
||||||
// Returns the command line as a vector of strings.
|
static const ::std::vector<testing::internal::string>* g_injected_test_argvs =
|
||||||
const ::std::vector<String>& GetArgvs() { return g_argvs; }
|
NULL; // Owned.
|
||||||
|
|
||||||
|
void SetInjectableArgvs(const ::std::vector<testing::internal::string>* argvs) {
|
||||||
|
if (g_injected_test_argvs != argvs)
|
||||||
|
delete g_injected_test_argvs;
|
||||||
|
g_injected_test_argvs = argvs;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ::std::vector<testing::internal::string>& GetInjectableArgvs() {
|
||||||
|
if (g_injected_test_argvs != NULL) {
|
||||||
|
return *g_injected_test_argvs;
|
||||||
|
}
|
||||||
|
return g_argvs;
|
||||||
|
}
|
||||||
#endif // GTEST_HAS_DEATH_TEST
|
#endif // GTEST_HAS_DEATH_TEST
|
||||||
|
|
||||||
#if GTEST_OS_WINDOWS_MOBILE
|
#if GTEST_OS_WINDOWS_MOBILE
|
||||||
@ -647,8 +1085,8 @@ void Abort() {
|
|||||||
// Returns the name of the environment variable corresponding to the
|
// Returns the name of the environment variable corresponding to the
|
||||||
// given flag. For example, FlagToEnvVar("foo") will return
|
// given flag. For example, FlagToEnvVar("foo") will return
|
||||||
// "GTEST_FOO" in the open-source version.
|
// "GTEST_FOO" in the open-source version.
|
||||||
static String FlagToEnvVar(const char* flag) {
|
static std::string FlagToEnvVar(const char* flag) {
|
||||||
const String full_flag =
|
const std::string full_flag =
|
||||||
(Message() << GTEST_FLAG_PREFIX_ << flag).GetString();
|
(Message() << GTEST_FLAG_PREFIX_ << flag).GetString();
|
||||||
|
|
||||||
Message env_var;
|
Message env_var;
|
||||||
@ -705,7 +1143,7 @@ bool ParseInt32(const Message& src_text, const char* str, Int32* value) {
|
|||||||
//
|
//
|
||||||
// The value is considered true iff it's not "0".
|
// The value is considered true iff it's not "0".
|
||||||
bool BoolFromGTestEnv(const char* flag, bool default_value) {
|
bool BoolFromGTestEnv(const char* flag, bool default_value) {
|
||||||
const String env_var = FlagToEnvVar(flag);
|
const std::string env_var = FlagToEnvVar(flag);
|
||||||
const char* const string_value = posix::GetEnv(env_var.c_str());
|
const char* const string_value = posix::GetEnv(env_var.c_str());
|
||||||
return string_value == NULL ?
|
return string_value == NULL ?
|
||||||
default_value : strcmp(string_value, "0") != 0;
|
default_value : strcmp(string_value, "0") != 0;
|
||||||
@ -715,7 +1153,7 @@ bool BoolFromGTestEnv(const char* flag, bool default_value) {
|
|||||||
// variable corresponding to the given flag; if it isn't set or
|
// variable corresponding to the given flag; if it isn't set or
|
||||||
// doesn't represent a valid 32-bit integer, returns default_value.
|
// doesn't represent a valid 32-bit integer, returns default_value.
|
||||||
Int32 Int32FromGTestEnv(const char* flag, Int32 default_value) {
|
Int32 Int32FromGTestEnv(const char* flag, Int32 default_value) {
|
||||||
const String env_var = FlagToEnvVar(flag);
|
const std::string env_var = FlagToEnvVar(flag);
|
||||||
const char* const string_value = posix::GetEnv(env_var.c_str());
|
const char* const string_value = posix::GetEnv(env_var.c_str());
|
||||||
if (string_value == NULL) {
|
if (string_value == NULL) {
|
||||||
// The environment variable is not set.
|
// The environment variable is not set.
|
||||||
@ -737,7 +1175,7 @@ Int32 Int32FromGTestEnv(const char* flag, Int32 default_value) {
|
|||||||
// Reads and returns the string environment variable corresponding to
|
// Reads and returns the string environment variable corresponding to
|
||||||
// the given flag; if it's not set, returns default_value.
|
// the given flag; if it's not set, returns default_value.
|
||||||
const char* StringFromGTestEnv(const char* flag, const char* default_value) {
|
const char* StringFromGTestEnv(const char* flag, const char* default_value) {
|
||||||
const String env_var = FlagToEnvVar(flag);
|
const std::string env_var = FlagToEnvVar(flag);
|
||||||
const char* const value = posix::GetEnv(env_var.c_str());
|
const char* const value = posix::GetEnv(env_var.c_str());
|
||||||
return value == NULL ? default_value : value;
|
return value == NULL ? default_value : value;
|
||||||
}
|
}
|
||||||
|
@ -45,6 +45,7 @@
|
|||||||
#include "gtest/gtest-printers.h"
|
#include "gtest/gtest-printers.h"
|
||||||
#include <ctype.h>
|
#include <ctype.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
#include <cwchar>
|
||||||
#include <ostream> // NOLINT
|
#include <ostream> // NOLINT
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "gtest/internal/gtest-port.h"
|
#include "gtest/internal/gtest-port.h"
|
||||||
@ -55,15 +56,10 @@ namespace {
|
|||||||
|
|
||||||
using ::std::ostream;
|
using ::std::ostream;
|
||||||
|
|
||||||
#if GTEST_OS_WINDOWS_MOBILE // Windows CE does not define _snprintf_s.
|
|
||||||
# define snprintf _snprintf
|
|
||||||
#elif _MSC_VER >= 1400 // VC 8.0 and later deprecate snprintf and _snprintf.
|
|
||||||
# define snprintf _snprintf_s
|
|
||||||
#elif _MSC_VER
|
|
||||||
# define snprintf _snprintf
|
|
||||||
#endif // GTEST_OS_WINDOWS_MOBILE
|
|
||||||
|
|
||||||
// Prints a segment of bytes in the given object.
|
// Prints a segment of bytes in the given object.
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_MEMORY_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_ADDRESS_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_THREAD_
|
||||||
void PrintByteSegmentInObjectTo(const unsigned char* obj_bytes, size_t start,
|
void PrintByteSegmentInObjectTo(const unsigned char* obj_bytes, size_t start,
|
||||||
size_t count, ostream* os) {
|
size_t count, ostream* os) {
|
||||||
char text[5] = "";
|
char text[5] = "";
|
||||||
@ -77,7 +73,7 @@ void PrintByteSegmentInObjectTo(const unsigned char* obj_bytes, size_t start,
|
|||||||
else
|
else
|
||||||
*os << '-';
|
*os << '-';
|
||||||
}
|
}
|
||||||
snprintf(text, sizeof(text), "%02X", obj_bytes[j]);
|
GTEST_SNPRINTF_(text, sizeof(text), "%02X", obj_bytes[j]);
|
||||||
*os << text;
|
*os << text;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -184,16 +180,16 @@ static CharFormat PrintAsCharLiteralTo(Char c, ostream* os) {
|
|||||||
*os << static_cast<char>(c);
|
*os << static_cast<char>(c);
|
||||||
return kAsIs;
|
return kAsIs;
|
||||||
} else {
|
} else {
|
||||||
*os << String::Format("\\x%X", static_cast<UnsignedChar>(c));
|
*os << "\\x" + String::FormatHexInt(static_cast<UnsignedChar>(c));
|
||||||
return kHexEscape;
|
return kHexEscape;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return kSpecialEscape;
|
return kSpecialEscape;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints a char c as if it's part of a string literal, escaping it when
|
// Prints a wchar_t c as if it's part of a string literal, escaping it when
|
||||||
// necessary; returns how c was formatted.
|
// necessary; returns how c was formatted.
|
||||||
static CharFormat PrintAsWideStringLiteralTo(wchar_t c, ostream* os) {
|
static CharFormat PrintAsStringLiteralTo(wchar_t c, ostream* os) {
|
||||||
switch (c) {
|
switch (c) {
|
||||||
case L'\'':
|
case L'\'':
|
||||||
*os << "'";
|
*os << "'";
|
||||||
@ -208,8 +204,9 @@ static CharFormat PrintAsWideStringLiteralTo(wchar_t c, ostream* os) {
|
|||||||
|
|
||||||
// Prints a char c as if it's part of a string literal, escaping it when
|
// Prints a char c as if it's part of a string literal, escaping it when
|
||||||
// necessary; returns how c was formatted.
|
// necessary; returns how c was formatted.
|
||||||
static CharFormat PrintAsNarrowStringLiteralTo(char c, ostream* os) {
|
static CharFormat PrintAsStringLiteralTo(char c, ostream* os) {
|
||||||
return PrintAsWideStringLiteralTo(static_cast<unsigned char>(c), os);
|
return PrintAsStringLiteralTo(
|
||||||
|
static_cast<wchar_t>(static_cast<unsigned char>(c)), os);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints a wide or narrow character c and its code. '\0' is printed
|
// Prints a wide or narrow character c and its code. '\0' is printed
|
||||||
@ -228,7 +225,7 @@ void PrintCharAndCodeTo(Char c, ostream* os) {
|
|||||||
// obvious).
|
// obvious).
|
||||||
if (c == 0)
|
if (c == 0)
|
||||||
return;
|
return;
|
||||||
*os << " (" << String::Format("%d", c).c_str();
|
*os << " (" << static_cast<int>(c);
|
||||||
|
|
||||||
// For more convenience, we print c's code again in hexidecimal,
|
// For more convenience, we print c's code again in hexidecimal,
|
||||||
// unless c was already printed in the form '\x##' or the code is in
|
// unless c was already printed in the form '\x##' or the code is in
|
||||||
@ -236,8 +233,7 @@ void PrintCharAndCodeTo(Char c, ostream* os) {
|
|||||||
if (format == kHexEscape || (1 <= c && c <= 9)) {
|
if (format == kHexEscape || (1 <= c && c <= 9)) {
|
||||||
// Do nothing.
|
// Do nothing.
|
||||||
} else {
|
} else {
|
||||||
*os << String::Format(", 0x%X",
|
*os << ", 0x" << String::FormatHexInt(static_cast<UnsignedChar>(c));
|
||||||
static_cast<UnsignedChar>(c)).c_str();
|
|
||||||
}
|
}
|
||||||
*os << ")";
|
*os << ")";
|
||||||
}
|
}
|
||||||
@ -255,48 +251,69 @@ void PrintTo(wchar_t wc, ostream* os) {
|
|||||||
PrintCharAndCodeTo<wchar_t>(wc, os);
|
PrintCharAndCodeTo<wchar_t>(wc, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints the given array of characters to the ostream.
|
// Prints the given array of characters to the ostream. CharType must be either
|
||||||
// The array starts at *begin, the length is len, it may include '\0' characters
|
// char or wchar_t.
|
||||||
// and may not be null-terminated.
|
// The array starts at begin, the length is len, it may include '\0' characters
|
||||||
static void PrintCharsAsStringTo(const char* begin, size_t len, ostream* os) {
|
// and may not be NUL-terminated.
|
||||||
*os << "\"";
|
template <typename CharType>
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_MEMORY_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_ADDRESS_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_THREAD_
|
||||||
|
static void PrintCharsAsStringTo(
|
||||||
|
const CharType* begin, size_t len, ostream* os) {
|
||||||
|
const char* const kQuoteBegin = sizeof(CharType) == 1 ? "\"" : "L\"";
|
||||||
|
*os << kQuoteBegin;
|
||||||
bool is_previous_hex = false;
|
bool is_previous_hex = false;
|
||||||
for (size_t index = 0; index < len; ++index) {
|
for (size_t index = 0; index < len; ++index) {
|
||||||
const char cur = begin[index];
|
const CharType cur = begin[index];
|
||||||
if (is_previous_hex && IsXDigit(cur)) {
|
if (is_previous_hex && IsXDigit(cur)) {
|
||||||
// Previous character is of '\x..' form and this character can be
|
// Previous character is of '\x..' form and this character can be
|
||||||
// interpreted as another hexadecimal digit in its number. Break string to
|
// interpreted as another hexadecimal digit in its number. Break string to
|
||||||
// disambiguate.
|
// disambiguate.
|
||||||
*os << "\" \"";
|
*os << "\" " << kQuoteBegin;
|
||||||
}
|
}
|
||||||
is_previous_hex = PrintAsNarrowStringLiteralTo(cur, os) == kHexEscape;
|
is_previous_hex = PrintAsStringLiteralTo(cur, os) == kHexEscape;
|
||||||
}
|
}
|
||||||
*os << "\"";
|
*os << "\"";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Prints a (const) char/wchar_t array of 'len' elements, starting at address
|
||||||
|
// 'begin'. CharType must be either char or wchar_t.
|
||||||
|
template <typename CharType>
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_MEMORY_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_ADDRESS_
|
||||||
|
GTEST_ATTRIBUTE_NO_SANITIZE_THREAD_
|
||||||
|
static void UniversalPrintCharArray(
|
||||||
|
const CharType* begin, size_t len, ostream* os) {
|
||||||
|
// The code
|
||||||
|
// const char kFoo[] = "foo";
|
||||||
|
// generates an array of 4, not 3, elements, with the last one being '\0'.
|
||||||
|
//
|
||||||
|
// Therefore when printing a char array, we don't print the last element if
|
||||||
|
// it's '\0', such that the output matches the string literal as it's
|
||||||
|
// written in the source code.
|
||||||
|
if (len > 0 && begin[len - 1] == '\0') {
|
||||||
|
PrintCharsAsStringTo(begin, len - 1, os);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// If, however, the last element in the array is not '\0', e.g.
|
||||||
|
// const char kFoo[] = { 'f', 'o', 'o' };
|
||||||
|
// we must print the entire array. We also print a message to indicate
|
||||||
|
// that the array is not NUL-terminated.
|
||||||
|
PrintCharsAsStringTo(begin, len, os);
|
||||||
|
*os << " (no terminating NUL)";
|
||||||
|
}
|
||||||
|
|
||||||
// Prints a (const) char array of 'len' elements, starting at address 'begin'.
|
// Prints a (const) char array of 'len' elements, starting at address 'begin'.
|
||||||
void UniversalPrintArray(const char* begin, size_t len, ostream* os) {
|
void UniversalPrintArray(const char* begin, size_t len, ostream* os) {
|
||||||
PrintCharsAsStringTo(begin, len, os);
|
UniversalPrintCharArray(begin, len, os);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints the given array of wide characters to the ostream.
|
// Prints a (const) wchar_t array of 'len' elements, starting at address
|
||||||
// The array starts at *begin, the length is len, it may include L'\0'
|
// 'begin'.
|
||||||
// characters and may not be null-terminated.
|
void UniversalPrintArray(const wchar_t* begin, size_t len, ostream* os) {
|
||||||
static void PrintWideCharsAsStringTo(const wchar_t* begin, size_t len,
|
UniversalPrintCharArray(begin, len, os);
|
||||||
ostream* os) {
|
|
||||||
*os << "L\"";
|
|
||||||
bool is_previous_hex = false;
|
|
||||||
for (size_t index = 0; index < len; ++index) {
|
|
||||||
const wchar_t cur = begin[index];
|
|
||||||
if (is_previous_hex && isascii(cur) && IsXDigit(static_cast<char>(cur))) {
|
|
||||||
// Previous character is of '\x..' form and this character can be
|
|
||||||
// interpreted as another hexadecimal digit in its number. Break string to
|
|
||||||
// disambiguate.
|
|
||||||
*os << "\" L\"";
|
|
||||||
}
|
|
||||||
is_previous_hex = PrintAsWideStringLiteralTo(cur, os) == kHexEscape;
|
|
||||||
}
|
|
||||||
*os << "\"";
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints the given C string to the ostream.
|
// Prints the given C string to the ostream.
|
||||||
@ -322,7 +339,7 @@ void PrintTo(const wchar_t* s, ostream* os) {
|
|||||||
*os << "NULL";
|
*os << "NULL";
|
||||||
} else {
|
} else {
|
||||||
*os << ImplicitCast_<const void*>(s) << " pointing to ";
|
*os << ImplicitCast_<const void*>(s) << " pointing to ";
|
||||||
PrintWideCharsAsStringTo(s, wcslen(s), os);
|
PrintCharsAsStringTo(s, std::wcslen(s), os);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#endif // wchar_t is native
|
#endif // wchar_t is native
|
||||||
@ -341,13 +358,13 @@ void PrintStringTo(const ::std::string& s, ostream* os) {
|
|||||||
// Prints a ::wstring object.
|
// Prints a ::wstring object.
|
||||||
#if GTEST_HAS_GLOBAL_WSTRING
|
#if GTEST_HAS_GLOBAL_WSTRING
|
||||||
void PrintWideStringTo(const ::wstring& s, ostream* os) {
|
void PrintWideStringTo(const ::wstring& s, ostream* os) {
|
||||||
PrintWideCharsAsStringTo(s.data(), s.size(), os);
|
PrintCharsAsStringTo(s.data(), s.size(), os);
|
||||||
}
|
}
|
||||||
#endif // GTEST_HAS_GLOBAL_WSTRING
|
#endif // GTEST_HAS_GLOBAL_WSTRING
|
||||||
|
|
||||||
#if GTEST_HAS_STD_WSTRING
|
#if GTEST_HAS_STD_WSTRING
|
||||||
void PrintWideStringTo(const ::std::wstring& s, ostream* os) {
|
void PrintWideStringTo(const ::std::wstring& s, ostream* os) {
|
||||||
PrintWideCharsAsStringTo(s.data(), s.size(), os);
|
PrintCharsAsStringTo(s.data(), s.size(), os);
|
||||||
}
|
}
|
||||||
#endif // GTEST_HAS_STD_WSTRING
|
#endif // GTEST_HAS_STD_WSTRING
|
||||||
|
|
||||||
|
@ -35,9 +35,9 @@
|
|||||||
|
|
||||||
// Indicates that this translation unit is part of Google Test's
|
// Indicates that this translation unit is part of Google Test's
|
||||||
// implementation. It must come before gtest-internal-inl.h is
|
// implementation. It must come before gtest-internal-inl.h is
|
||||||
// included, or there will be a compiler error. This trick is to
|
// included, or there will be a compiler error. This trick exists to
|
||||||
// prevent a user from accidentally including gtest-internal-inl.h in
|
// prevent the accidental inclusion of gtest-internal-inl.h in the
|
||||||
// his code.
|
// user's code.
|
||||||
#define GTEST_IMPLEMENTATION_ 1
|
#define GTEST_IMPLEMENTATION_ 1
|
||||||
#include "src/gtest-internal-inl.h"
|
#include "src/gtest-internal-inl.h"
|
||||||
#undef GTEST_IMPLEMENTATION_
|
#undef GTEST_IMPLEMENTATION_
|
||||||
@ -48,10 +48,10 @@ using internal::GetUnitTestImpl;
|
|||||||
|
|
||||||
// Gets the summary of the failure message by omitting the stack trace
|
// Gets the summary of the failure message by omitting the stack trace
|
||||||
// in it.
|
// in it.
|
||||||
internal::String TestPartResult::ExtractSummary(const char* message) {
|
std::string TestPartResult::ExtractSummary(const char* message) {
|
||||||
const char* const stack_trace = strstr(message, internal::kStackTraceMarker);
|
const char* const stack_trace = strstr(message, internal::kStackTraceMarker);
|
||||||
return stack_trace == NULL ? internal::String(message) :
|
return stack_trace == NULL ? message :
|
||||||
internal::String(message, stack_trace - message);
|
std::string(message, stack_trace);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prints a TestPartResult object.
|
// Prints a TestPartResult object.
|
||||||
|
@ -45,6 +45,15 @@ static const char* SkipSpaces(const char* str) {
|
|||||||
return str;
|
return str;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static std::vector<std::string> SplitIntoTestNames(const char* src) {
|
||||||
|
std::vector<std::string> name_vec;
|
||||||
|
src = SkipSpaces(src);
|
||||||
|
for (; src != NULL; src = SkipComma(src)) {
|
||||||
|
name_vec.push_back(StripTrailingSpaces(GetPrefixUntilComma(src)));
|
||||||
|
}
|
||||||
|
return name_vec;
|
||||||
|
}
|
||||||
|
|
||||||
// Verifies that registered_tests match the test names in
|
// Verifies that registered_tests match the test names in
|
||||||
// defined_test_names_; returns registered_tests if successful, or
|
// defined_test_names_; returns registered_tests if successful, or
|
||||||
// aborts the program otherwise.
|
// aborts the program otherwise.
|
||||||
@ -53,15 +62,14 @@ const char* TypedTestCasePState::VerifyRegisteredTestNames(
|
|||||||
typedef ::std::set<const char*>::const_iterator DefinedTestIter;
|
typedef ::std::set<const char*>::const_iterator DefinedTestIter;
|
||||||
registered_ = true;
|
registered_ = true;
|
||||||
|
|
||||||
// Skip initial whitespace in registered_tests since some
|
std::vector<std::string> name_vec = SplitIntoTestNames(registered_tests);
|
||||||
// preprocessors prefix stringizied literals with whitespace.
|
|
||||||
registered_tests = SkipSpaces(registered_tests);
|
|
||||||
|
|
||||||
Message errors;
|
Message errors;
|
||||||
::std::set<String> tests;
|
|
||||||
for (const char* names = registered_tests; names != NULL;
|
std::set<std::string> tests;
|
||||||
names = SkipComma(names)) {
|
for (std::vector<std::string>::const_iterator name_it = name_vec.begin();
|
||||||
const String name = GetPrefixUntilComma(names);
|
name_it != name_vec.end(); ++name_it) {
|
||||||
|
const std::string& name = *name_it;
|
||||||
if (tests.count(name) != 0) {
|
if (tests.count(name) != 0) {
|
||||||
errors << "Test " << name << " is listed more than once.\n";
|
errors << "Test " << name << " is listed more than once.\n";
|
||||||
continue;
|
continue;
|
||||||
@ -93,7 +101,7 @@ const char* TypedTestCasePState::VerifyRegisteredTestNames(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const String& errors_str = errors.GetString();
|
const std::string& errors_str = errors.GetString();
|
||||||
if (errors_str != "") {
|
if (errors_str != "") {
|
||||||
fprintf(stderr, "%s %s", FormatFileLocation(file, line).c_str(),
|
fprintf(stderr, "%s %s", FormatFileLocation(file, line).c_str(),
|
||||||
errors_str.c_str());
|
errors_str.c_str());
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -27,13 +27,12 @@
|
|||||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
#include <iostream>
|
#include <stdio.h>
|
||||||
|
|
||||||
#include "gtest/gtest.h"
|
#include "gtest/gtest.h"
|
||||||
|
|
||||||
GTEST_API_ int main(int argc, char **argv) {
|
GTEST_API_ int main(int argc, char **argv) {
|
||||||
std::cout << "Running main() from gtest_main.cc\n";
|
printf("Running main() from gtest_main.cc\n");
|
||||||
|
|
||||||
testing::InitGoogleTest(&argc, argv);
|
testing::InitGoogleTest(&argc, argv);
|
||||||
return RUN_ALL_TESTS();
|
return RUN_ALL_TESTS();
|
||||||
}
|
}
|
||||||
|
@ -86,7 +86,7 @@ TEST(AddressFromURL, Success)
|
|||||||
|
|
||||||
bool dnssec_result = false;
|
bool dnssec_result = false;
|
||||||
|
|
||||||
std::vector<std::string> addresses = tools::wallet2::addresses_from_url("donate.monero.cc", dnssec_result);
|
std::vector<std::string> addresses = tools::wallet2::addresses_from_url("donate.getmonero.org", dnssec_result);
|
||||||
|
|
||||||
EXPECT_EQ(1, addresses.size());
|
EXPECT_EQ(1, addresses.size());
|
||||||
if (addresses.size() == 1)
|
if (addresses.size() == 1)
|
||||||
|
@ -43,13 +43,13 @@ TEST(DNSResolver, IPv4Success)
|
|||||||
|
|
||||||
ASSERT_EQ(1, ips.size());
|
ASSERT_EQ(1, ips.size());
|
||||||
|
|
||||||
ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
//ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
||||||
|
|
||||||
ips = tools::DNSResolver::instance().get_ipv4("example.com", avail, valid);
|
ips = tools::DNSResolver::instance().get_ipv4("example.com", avail, valid);
|
||||||
|
|
||||||
ASSERT_EQ(1, ips.size());
|
ASSERT_EQ(1, ips.size());
|
||||||
|
|
||||||
ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
//ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST(DNSResolver, IPv4Failure)
|
TEST(DNSResolver, IPv4Failure)
|
||||||
@ -68,6 +68,38 @@ TEST(DNSResolver, IPv4Failure)
|
|||||||
ASSERT_EQ(0, ips.size());
|
ASSERT_EQ(0, ips.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST(DNSResolver, DNSSECSuccess)
|
||||||
|
{
|
||||||
|
tools::DNSResolver resolver;
|
||||||
|
|
||||||
|
bool avail, valid;
|
||||||
|
|
||||||
|
auto ips = resolver.get_ipv4("example.com", avail, valid);
|
||||||
|
|
||||||
|
ASSERT_EQ(1, ips.size());
|
||||||
|
|
||||||
|
//ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
||||||
|
|
||||||
|
ASSERT_TRUE(avail);
|
||||||
|
ASSERT_TRUE(valid);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(DNSResolver, DNSSECFailure)
|
||||||
|
{
|
||||||
|
tools::DNSResolver resolver;
|
||||||
|
|
||||||
|
bool avail, valid;
|
||||||
|
|
||||||
|
auto ips = resolver.get_ipv4("dnssec-failed.org", avail, valid);
|
||||||
|
|
||||||
|
ASSERT_EQ(1, ips.size());
|
||||||
|
|
||||||
|
//ASSERT_STREQ("93.184.216.119", ips[0].c_str());
|
||||||
|
|
||||||
|
ASSERT_TRUE(avail);
|
||||||
|
ASSERT_FALSE(valid);
|
||||||
|
}
|
||||||
|
|
||||||
// It would be great to include an IPv6 test and assume it'll pass, but not every ISP / resolver plays nicely with IPv6;)
|
// It would be great to include an IPv6 test and assume it'll pass, but not every ISP / resolver plays nicely with IPv6;)
|
||||||
/*TEST(DNSResolver, IPv6Success)
|
/*TEST(DNSResolver, IPv6Success)
|
||||||
{
|
{
|
||||||
@ -108,12 +140,12 @@ TEST(DNSResolver, GetTXTRecord)
|
|||||||
{
|
{
|
||||||
bool avail, valid;
|
bool avail, valid;
|
||||||
|
|
||||||
std::vector<std::string> records = tools::DNSResolver::instance().get_txt_record("donate.monero.cc", avail, valid);
|
std::vector<std::string> records = tools::DNSResolver::instance().get_txt_record("donate.getmonero.org", avail, valid);
|
||||||
|
|
||||||
EXPECT_NE(0, records.size());
|
EXPECT_NE(0, records.size());
|
||||||
|
|
||||||
for (auto& rec : records)
|
for (auto& rec : records)
|
||||||
{
|
{
|
||||||
std::cout << "TXT record for donate.monero.cc: " << rec << std::endl;
|
std::cout << "TXT record for donate.getmonero.org: " << rec << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user