From 0e5b6d8ee9b7cd3acf56b232f54254d5c46e0c3d Mon Sep 17 00:00:00 2001 From: Mateusz Malicki Date: Tue, 17 Feb 2015 15:49:35 +0100 Subject: [PATCH 01/16] GVariant visitor test [Bug/Feature] GVariant visitor test [Cause] N/A [Solution] Test convert to and from GVariant [Verification] Build, run tests: vasum-server-unit-tests -t ConfigurationSuite Change-Id: I8e7573bc3d44724b17560bfaa9a040922530ff40 --- tests/unit_tests/config/ut-configuration.cpp | 51 ++++++++++++++++++++++++++-- 1 file changed, 48 insertions(+), 3 deletions(-) diff --git a/tests/unit_tests/config/ut-configuration.cpp b/tests/unit_tests/config/ut-configuration.cpp index 2e3b487..8049975 100644 --- a/tests/unit_tests/config/ut-configuration.cpp +++ b/tests/unit_tests/config/ut-configuration.cpp @@ -52,7 +52,7 @@ struct Fixture { BOOST_FIXTURE_TEST_SUITE(ConfigurationSuite, Fixture) -BOOST_AUTO_TEST_CASE(FromStringTest) +BOOST_AUTO_TEST_CASE(FromJsonStringTest) { TestConfig testConfig; @@ -96,10 +96,28 @@ BOOST_AUTO_TEST_CASE(FromStringTest) BOOST_CHECK_EQUAL(4, testConfig.subVector[0].intVector[1]); BOOST_CHECK_EQUAL(6, testConfig.subVector[1].intVector[1]); + BOOST_CHECK(testConfig.union1.is()); + BOOST_CHECK_EQUAL(2, testConfig.union1.as()); + + BOOST_CHECK(testConfig.union2.is()); + BOOST_CHECK_EQUAL(54321, testConfig.union2.as().intVal); + BOOST_REQUIRE_EQUAL(1, testConfig.union2.as().intVector.size()); + BOOST_CHECK_EQUAL(1, testConfig.union2.as().intVector[0]); + BOOST_CHECK_EQUAL(234, testConfig.union2.as().subSubObj.intVal); + + BOOST_REQUIRE_EQUAL(2, testConfig.unions.size()); + BOOST_CHECK(testConfig.unions[0].is()); + BOOST_CHECK_EQUAL(2, testConfig.unions[0].as()); + + BOOST_CHECK(testConfig.unions[1].is()); + BOOST_CHECK_EQUAL(54321, testConfig.unions[1].as().intVal); + BOOST_REQUIRE_EQUAL(1, testConfig.unions[1].as().intVector.size()); + BOOST_CHECK_EQUAL(1, testConfig.unions[1].as().intVector[0]); + BOOST_CHECK_EQUAL(234, testConfig.unions[1].as().subSubObj.intVal); } -BOOST_AUTO_TEST_CASE(ToStringTest) +BOOST_AUTO_TEST_CASE(ToJsonStringTest) { TestConfig testConfig; BOOST_REQUIRE_NO_THROW(loadFromJsonString(jsonTestString, testConfig)); @@ -135,7 +153,7 @@ struct UnionConfig { } // namespace loadErrorsTest -BOOST_AUTO_TEST_CASE(LoadErrorsTest) +BOOST_AUTO_TEST_CASE(JsonLoadErrorsTest) { using namespace loadErrorsTest; @@ -351,6 +369,14 @@ BOOST_AUTO_TEST_CASE(PartialConfigTest) partialConfig.intVector = {7}; config::saveToKVStore(DB_PATH, partialConfig, DB_PREFIX); } + + // from gvariant (partial is not supported!) + { + PartialTestConfig partialConfig; + std::unique_ptr v(saveToGVariant(config), + g_variant_unref); + BOOST_CHECK_THROW(loadFromGVariant(v.get(), partialConfig), ConfigException); + } } BOOST_AUTO_TEST_CASE(ConfigUnionTest) @@ -399,4 +425,23 @@ BOOST_AUTO_TEST_CASE(ConfigUnionTest) BOOST_CHECK_EQUAL(out, jsonTestString); } + +BOOST_AUTO_TEST_CASE(GVariantVisitorTest) +{ + TestConfig testConfig; + BOOST_REQUIRE_NO_THROW(loadFromJsonString(jsonTestString, testConfig)); + std::unique_ptr v(saveToGVariant(testConfig), + g_variant_unref); + TestConfig testConfig2; + loadFromGVariant(v.get(), testConfig2); + std::string out = saveToJsonString(testConfig2); + BOOST_CHECK_EQUAL(out, jsonTestString); + + PartialTestConfig partialConfig; + partialConfig.stringVal = testConfig.stringVal; + partialConfig.intVector = testConfig.intVector; + v.reset(saveToGVariant(partialConfig)); + BOOST_CHECK_THROW(loadFromGVariant(v.get(), testConfig), ConfigException); +} + BOOST_AUTO_TEST_SUITE_END() -- 2.7.4 From b094b5574e100d45577a7271f827f057a947c81e Mon Sep 17 00:00:00 2001 From: Mateusz Malicki Date: Thu, 12 Feb 2015 11:43:56 +0100 Subject: [PATCH 02/16] Stubs for creating netdev [Feature] Stubs (cli, client, server) for creating netdev [Cause] N/A [Solution] N/A [Verification] Build, run cli commnads Change-Id: Ibf2631d0f6e42823793d3f5bd971586348811891 --- cli/command-line-interface.cpp | 59 ++++++++++++++++++++++++++++++++++ cli/command-line-interface.hpp | 20 ++++++++++++ cli/main.cpp | 30 +++++++++++++++++ client/vasum-client-impl.cpp | 55 +++++++++++++++++++++----------- client/vasum-client-impl.hpp | 27 ++++++++++++---- client/vasum-client.cpp | 29 ++++++++++++----- client/vasum-client.h | 48 ++++++++++++++++++++-------- common/base-exception.cpp | 7 +++- common/base-exception.hpp | 1 + server/host-connection.cpp | 46 +++++++++++++++++++++++++++ server/host-connection.hpp | 33 +++++++++++++++++++ server/host-dbus-definitions.hpp | 64 +++++++++++++++++++++++-------------- server/zone-admin.cpp | 17 ++++++++++ server/zone-admin.hpp | 18 +++++++++++ server/zone.cpp | 21 ++++++++++++ server/zone.hpp | 18 +++++++++++ server/zones-manager.cpp | 69 ++++++++++++++++++++++++++++++++++++++++ server/zones-manager.hpp | 12 +++++++ 18 files changed, 503 insertions(+), 71 deletions(-) diff --git a/cli/command-line-interface.cpp b/cli/command-line-interface.cpp index 06be364..a189a92 100644 --- a/cli/command-line-interface.cpp +++ b/cli/command-line-interface.cpp @@ -37,6 +37,7 @@ #include #include #include +#include using namespace std; @@ -150,6 +151,22 @@ ostream& operator<<(ostream& out, const Table& table) return out; } +enum macvlan_mode macvlanFromString(const std::string& mode) { + if (mode == "private") { + return MACVLAN_MODE_PRIVATE; + } + if (mode == "vepa") { + return MACVLAN_MODE_VEPA; + } + if (mode == "bridge") { + return MACVLAN_MODE_BRIDGE; + } + if (mode == "passthru") { + return MACVLAN_MODE_PASSTHRU; + } + throw runtime_error("Unsupported macvlan mode"); +} + } // namespace void CommandLineInterface::printUsage(std::ostream& out) const @@ -335,5 +352,47 @@ void revoke_device(int pos, int argc, const char** argv) one_shot(bind(vsm_revoke_device, _1, argv[pos + 1], argv[pos + 2])); } +void create_netdev_veth(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 3) { + throw runtime_error("Not enough parameters"); + } + one_shot(bind(vsm_create_netdev_veth, + _1, + argv[pos + 1], + argv[pos + 2], + argv[pos + 3])); +} + +void create_netdev_macvlan(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 4) { + throw runtime_error("Not enough parameters"); + } + one_shot(bind(vsm_create_netdev_macvlan, + _1, + argv[pos + 1], + argv[pos + 2], + argv[pos + 3], + macvlanFromString(argv[pos + 4]))); +} + +void create_netdev_phys(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 2) { + throw runtime_error("Not enough parameters"); + } + one_shot(bind(vsm_create_netdev_phys, + _1, + argv[pos + 1], + argv[pos + 2])); +} + } // namespace cli } // namespace vasum diff --git a/cli/command-line-interface.hpp b/cli/command-line-interface.hpp index 055bb1e..34ac013 100644 --- a/cli/command-line-interface.hpp +++ b/cli/command-line-interface.hpp @@ -187,6 +187,26 @@ void grant_device(int pos, int argc, const char** argv); */ void revoke_device(int pos, int argc, const char** argv); +/** + * Parses command line arguments and call vsm_create_netdev_veth + * + * @see vsm_create_netdev_veth + */ +void create_netdev_veth(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_create_netdev_macvlan + * + * @see vsm_create_netdev_macvlan + */ +void create_netdev_macvlan(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_create_netdev_phys + * + * @see vsm_create_netdev_phys + */ +void create_netdev_phys(int pos, int argc, const char** argv); } // namespace cli } // namespace vasum diff --git a/cli/main.cpp b/cli/main.cpp index d003d10..38a8ae5 100644 --- a/cli/main.cpp +++ b/cli/main.cpp @@ -140,6 +140,36 @@ std::map commands = { {{"zone_id", "id zone name"}, {"device_name", " device name"}} } + }, + { + "create_netdev_veth", { + create_netdev_veth, + "create_netdev_veth zone_id zoneDev hostDev", + "Create netdev in zone", + {{"zone_id", "id zone name"}, + {"zoneDev", "network device id"}, + {"hostDev", "host bridge id"}} + } + }, + { + "create_netdev_macvlan", { + create_netdev_macvlan, + "create_netdev_macvlan zone_id zoneDev hostDev mode", + "Create netdev in zone", + {{"zone_id", "id zone name"}, + {"zoneDev", "network device id"}, + {"hostDev", "host bridge id"}, + {"mode", "macvlan mode (private, vepa, bridge, passthru)"}} + } + }, + { + "create_netdev_phys", { + create_netdev_phys, + "create_netdev_phys zone_id devId", + "Create/move netdev to zone", + {{"zone_id", "id zone name"}, + {"devId", "network device id"}} + } } }; diff --git a/client/vasum-client-impl.cpp b/client/vasum-client-impl.cpp index d27fad3..d6f4217 100644 --- a/client/vasum-client-impl.cpp +++ b/client/vasum-client-impl.cpp @@ -152,7 +152,7 @@ void toArray(GVariant* in, T** scArray) *scArray = ids; } -VsmStatus toStatus(const std::exception& ex) +VsmStatus toStatus(const exception& ex) { if (typeid(DbusCustomException) == typeid(ex)) { return VSMCLIENT_CUSTOM_ERROR; @@ -168,14 +168,14 @@ VsmStatus toStatus(const std::exception& ex) return VSMCLIENT_OTHER_ERROR; } -bool readFirstLineOfFile(const std::string& path, std::string& ret) +bool readFirstLineOfFile(const string& path, string& ret) { - std::ifstream file(path); + ifstream file(path); if (!file) { return false; } - std::getline(file, ret); + getline(file, ret); return true; } @@ -209,7 +209,7 @@ Client::Status::Status() { } -Client::Status::Status(VsmStatus status, const std::string& msg) +Client::Status::Status(VsmStatus status, const string& msg) : mVsmStatus(status), mMsg(msg) { } @@ -272,10 +272,10 @@ VsmStatus Client::signalSubscribe(const DbusInterfaceInfo& info, SignalCallback signalCallback, VsmSubscriptionId* subscriptionId) { - auto onSignal = [=](const std::string& /*senderBusName*/, - const std::string & objectPath, - const std::string & interface, - const std::string & signalName, + auto onSignal = [=](const string& /*senderBusName*/, + const string & objectPath, + const string & interface, + const string & signalName, GVariant * parameters) { if (objectPath == info.objectPath && interface == info.interface && @@ -290,7 +290,7 @@ VsmStatus Client::signalSubscribe(const DbusInterfaceInfo& info, *subscriptionId = id; } mStatus = Status(); - } catch (const std::exception& ex) { + } catch (const exception& ex) { mStatus = Status(toStatus(ex), ex.what()); } return vsm_get_status(); @@ -301,7 +301,7 @@ VsmStatus Client::signalUnsubscribe(VsmSubscriptionId id) try { mConnection->signalUnsubscribe(id); mStatus = Status(); - } catch (const std::exception& ex) { + } catch (const exception& ex) { mStatus = Status(toStatus(ex), ex.what()); } return vsm_get_status(); @@ -385,15 +385,15 @@ VsmStatus Client::vsm_lookup_zone_by_pid(int pid, VsmString* id) noexcept { assert(id); - const std::string path = "/proc/" + std::to_string(pid) + "/cpuset"; + const string path = "/proc/" + to_string(pid) + "/cpuset"; - std::string cpuset; + string cpuset; if (!readFirstLineOfFile(path, cpuset)) { mStatus = Status(VSMCLIENT_INVALID_ARGUMENT, "Process not found"); return vsm_get_status(); } - std::string zoneId; + string zoneId; if (!parseZoneIdFromCpuSet(cpuset, zoneId)) { mStatus = Status(VSMCLIENT_OTHER_ERROR, "unknown format of cpuset"); return vsm_get_status(); @@ -560,19 +560,36 @@ VsmStatus Client::vsm_netdev_set_ipv6_addr(const char*, const char*, struct in6_ return vsm_get_status(); } -VsmStatus Client::vsm_create_netdev(const char*, VsmNetdevType, const char*, const char*) noexcept +VsmStatus Client::vsm_create_netdev_veth(const char* zone, + const char* zoneDev, + const char* hostDev) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); - return vsm_get_status(); + GVariant* args_in = g_variant_new("(sss)", zone, zoneDev, hostDev); + return callMethod(HOST_INTERFACE, api::host::METHOD_CREATE_NETDEV_VETH, args_in); } -VsmStatus Client::vsm_destroy_netdev(const char*, const char*) noexcept +VsmStatus Client::vsm_create_netdev_macvlan(const char* zone, + const char* zoneDev, + const char* hostDev, + enum macvlan_mode mode) noexcept +{ + GVariant* args_in = g_variant_new("(sssu)", zone, zoneDev, hostDev, mode); + return callMethod(HOST_INTERFACE, api::host::METHOD_CREATE_NETDEV_MACVLAN, args_in); +} + +VsmStatus Client::vsm_create_netdev_phys(const char* zone, const char* devId) noexcept +{ + GVariant* args_in = g_variant_new("(ss)", zone, devId); + return callMethod(HOST_INTERFACE, api::host::METHOD_CREATE_NETDEV_PHYS, args_in); +} + +VsmStatus Client::vsm_lookup_netdev_by_name(const char*, const char*, VsmNetdev*) noexcept { mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); return vsm_get_status(); } -VsmStatus Client::vsm_lookup_netdev_by_name(const char*, const char*, VsmNetdev*) noexcept +VsmStatus Client::vsm_destroy_netdev(const char*, const char*) noexcept { mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); return vsm_get_status(); diff --git a/client/vasum-client-impl.hpp b/client/vasum-client-impl.hpp index b5acee7..193f140 100644 --- a/client/vasum-client-impl.hpp +++ b/client/vasum-client-impl.hpp @@ -29,6 +29,7 @@ #include "vasum-client.h" #include #include +#include /** * Structure which defines the dbus interface. @@ -228,17 +229,24 @@ public: int prefix) noexcept; /** - * @see ::vsm_create_netdev + * @see ::vsm_create_netdev_veth */ - VsmStatus vsm_create_netdev(const char* zone, - VsmNetdevType netdevType, - const char* target, - const char* netdevId) noexcept; + VsmStatus vsm_create_netdev_veth(const char* zone, + const char* zoneDev, + const char* hostDev) noexcept; /** - * @see ::vsm_destroy_netdev + * @see ::vsm_create_netdev_macvlan + */ + VsmStatus vsm_create_netdev_macvlan(const char* zone, + const char* zoneDev, + const char* hostDev, + enum macvlan_mode mode) noexcept; + + /** + * @see ::vsm_create_netdev_phys */ - VsmStatus vsm_destroy_netdev(const char* zone, const char* netdevId) noexcept; + VsmStatus vsm_create_netdev_phys(const char* zone, const char* devId) noexcept; /** * @see ::vsm_lookup_netdev_by_name @@ -248,6 +256,11 @@ public: VsmNetdev* netdev) noexcept; /** + * @see ::vsm_destroy_netdev + */ + VsmStatus vsm_destroy_netdev(const char* zone, const char* devId) noexcept; + + /** * @see ::vsm_declare_file */ VsmStatus vsm_declare_file(const char* zone, diff --git a/client/vasum-client.cpp b/client/vasum-client.cpp index 5310b42..1f011c1 100644 --- a/client/vasum-client.cpp +++ b/client/vasum-client.cpp @@ -250,18 +250,26 @@ API VsmStatus vsm_netdev_set_ipv6_addr(VsmClient client, return getClient(client).vsm_netdev_set_ipv6_addr(zone, netdevId, addr, prefix); } -API VsmStatus vsm_create_netdev(VsmClient client, - const char* zone, - VsmNetdevType netdevType, - const char* target, - const char* netdevId) +API VsmStatus vsm_create_netdev_veth(VsmClient client, + const char* zone, + const char* zoneDev, + const char* hostDev) +{ + return getClient(client).vsm_create_netdev_veth(zone, zoneDev, hostDev); +} + +API VsmStatus vsm_create_netdev_macvlan(VsmClient client, + const char* zone, + const char* zoneDev, + const char* hostDev, + enum macvlan_mode mode) { - return getClient(client).vsm_create_netdev(zone, netdevType, target, netdevId); + return getClient(client).vsm_create_netdev_macvlan(zone, zoneDev, hostDev, mode); } -API VsmStatus vsm_destroy_netdev(VsmClient client, const char* zone, const char* netdevId) +API VsmStatus vsm_create_netdev_phys(VsmClient client, const char* zone, const char* devId) { - return getClient(client).vsm_destroy_netdev(zone, netdevId); + return getClient(client).vsm_create_netdev_phys(zone, devId); } API VsmStatus vsm_lookup_netdev_by_name(VsmClient client, @@ -272,6 +280,11 @@ API VsmStatus vsm_lookup_netdev_by_name(VsmClient client, return getClient(client).vsm_lookup_netdev_by_name(zone, netdevId, netdev); } +API VsmStatus vsm_destroy_netdev(VsmClient client, const char* zone, const char* devId) +{ + return getClient(client).vsm_destroy_netdev(zone, devId); +} + API VsmStatus vsm_declare_file(VsmClient client, const char* zone, VsmFileType type, diff --git a/client/vasum-client.h b/client/vasum-client.h index ba21449..1fb48b1 100644 --- a/client/vasum-client.h +++ b/client/vasum-client.h @@ -81,6 +81,7 @@ finish: #include #include #include +#include #ifdef __cplusplus extern "C" @@ -551,30 +552,41 @@ VsmStatus vsm_netdev_set_ipv6_addr(VsmClient client, int prefix); /** - * Create netdev in zone + * Create veth netdev in zone * * @param[in] client vasum-server's client * @param[in] zone zone name - * @param[in] netdevType netdev type - * @param[in] target TODO: this is taken form zone-control - * @param[in] netdevId network device id + * @param[in] zoneDev in host network device id + * @param[in] hostDev in zone network device id * @return status of this function call */ -VsmStatus vsm_create_netdev(VsmClient client, - const char* zone, - VsmNetdevType netdevType, - const char* target, - const char* netdevId); - +VsmStatus vsm_create_netdev_veth(VsmClient client, + const char* zone, + const char* zoneDev, + const char* hostDev); /** - * Remove netdev from zone + * Create macvlab in zone * * @param[in] client vasum-server's client * @param[in] zone zone name - * @param[in] netdevId network device id + * @param[in] zoneDev in host network device id + * @param[in] hostDev in zone network device id + * @return status of this function call + */ +VsmStatus vsm_create_netdev_macvlan(VsmClient client, + const char* zone, + const char* zoneDev, + const char* hostDev, + enum macvlan_mode mode); +/** + * Create/move phys netdev in/to zone + * + * @param[in] client vasum-server's client + * @param[in] zone zone name + * @param[in] devId network device id * @return status of this function call */ -VsmStatus vsm_destroy_netdev(VsmClient client, const char* zone, const char* netdevId); +VsmStatus vsm_create_netdev_phys(VsmClient client, const char* zone, const char* devId); /** * Get netdev informations @@ -592,6 +604,16 @@ VsmStatus vsm_lookup_netdev_by_name(VsmClient client, VsmNetdev* netdev); /** + * Remove netdev from zone + * + * @param[in] client vasum-server's client + * @param[in] zone zone name + * @param[in] devId network device id + * @return status of this function call + */ +VsmStatus vsm_destroy_netdev(VsmClient client, const char* zone, const char* devId); + +/** * Create file, directory or pipe in zone * * Declare file, directory or pipe that will be created while zone startup diff --git a/common/base-exception.cpp b/common/base-exception.cpp index bcfe600..628296d 100644 --- a/common/base-exception.cpp +++ b/common/base-exception.cpp @@ -34,8 +34,13 @@ const int ERROR_MESSAGE_BUFFER_CAPACITY = 256; std::string getSystemErrorMessage() { + return getSystemErrorMessage(errno); +} + +std::string getSystemErrorMessage(int err) +{ char buf[ERROR_MESSAGE_BUFFER_CAPACITY]; - return strerror_r(errno, buf, sizeof(buf)); + return strerror_r(err, buf, sizeof(buf)); } } // namespace vasum diff --git a/common/base-exception.hpp b/common/base-exception.hpp index 3969404..5637e76 100644 --- a/common/base-exception.hpp +++ b/common/base-exception.hpp @@ -46,6 +46,7 @@ struct VasumException: public std::runtime_error { * it is wrapper for strerror_r */ std::string getSystemErrorMessage(); +std::string getSystemErrorMessage(int err); } // namespace vasum diff --git a/server/host-connection.cpp b/server/host-connection.cpp index 4a8a2ac..75c4433 100644 --- a/server/host-connection.cpp +++ b/server/host-connection.cpp @@ -130,6 +130,21 @@ void HostConnection::setGetZoneInfoCallback(const GetZoneInfoCallback& callback) mGetZoneInfoCallback = callback; } +void HostConnection::setCreateNetdevVethCallback(const CreateNetdevVethCallback& callback) +{ + mCreateNetdevVethCallback = callback; +} + +void HostConnection::setCreateNetdevMacvlanCallback(const CreateNetdevMacvlanCallback& callback) +{ + mCreateNetdevMacvlanCallback = callback; +} + +void HostConnection::setCreateNetdevPhysCallback(const CreateNetdevPhysCallback& callback) +{ + mCreateNetdevPhysCallback = callback; +} + void HostConnection::setDeclareFileCallback(const DeclareFileCallback& callback) { mDeclareFileCallback = callback; @@ -281,6 +296,37 @@ void HostConnection::onMessageCall(const std::string& objectPath, return; } + if (methodName == api::host::METHOD_CREATE_NETDEV_VETH) { + const gchar* id = NULL; + const gchar* zoneDev = NULL; + const gchar* hostDev = NULL; + g_variant_get(parameters, "(&s&s&s)", &id, &zoneDev, &hostDev); + if (mCreateNetdevVethCallback) { + mCreateNetdevVethCallback(id, zoneDev, hostDev, result); + } + return; + } + + if (methodName == api::host::METHOD_CREATE_NETDEV_MACVLAN) { + const gchar* id = NULL; + const gchar* zoneDev = NULL; + const gchar* hostDev = NULL; + guint32 mode; + g_variant_get(parameters, "(&s&s&su)", &id, &zoneDev, &hostDev, &mode); + if (mCreateNetdevMacvlanCallback) { + mCreateNetdevMacvlanCallback(id, zoneDev, hostDev, mode, result); + } + } + + if (methodName == api::host::METHOD_CREATE_NETDEV_PHYS) { + const gchar* id = NULL; + const gchar* devId = NULL; + g_variant_get(parameters, "(&s&s)", &id, &devId); + if (mCreateNetdevPhysCallback) { + mCreateNetdevPhysCallback(id, devId, result); + } + } + if (methodName == api::host::METHOD_DECLARE_FILE) { const gchar* zone; int32_t type; diff --git a/server/host-connection.hpp b/server/host-connection.hpp index 2d029c1..649ed15 100644 --- a/server/host-connection.hpp +++ b/server/host-connection.hpp @@ -60,6 +60,21 @@ public: typedef std::function GetZoneInfoCallback; + typedef std::function CreateNetdevVethCallback; + typedef std::function CreateNetdevMacvlanCallback; + typedef std::function CreateNetdevPhysCallback; typedef std::function" " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " " " " " " " diff --git a/server/zone-admin.cpp b/server/zone-admin.cpp index 32a18d2..fa465bc 100644 --- a/server/zone-admin.cpp +++ b/server/zone-admin.cpp @@ -297,5 +297,22 @@ std::int64_t ZoneAdmin::getSchedulerQuota() return 0; } +void ZoneAdmin::createNetdevVeth(const std::string& /* zoneDev */, + const std::string& /* hostDev */) +{ + throw ZoneOperationException("Not implemented"); +} + +void ZoneAdmin::createNetdevMacvlan(const std::string& /* zoneDev */, + const std::string& /* hostDev */, + const uint32_t& /* mode */) +{ + throw ZoneOperationException("Not implemented"); +} + +void ZoneAdmin::moveNetdev(const std::string& /* devId */) +{ + throw ZoneOperationException("Not implemented"); +} } // namespace vasum diff --git a/server/zone-admin.hpp b/server/zone-admin.hpp index 10b7f02..9125a6d 100644 --- a/server/zone-admin.hpp +++ b/server/zone-admin.hpp @@ -130,6 +130,24 @@ public: */ std::int64_t getSchedulerQuota(); + /** + * Create veth network device + */ + void createNetdevVeth(const std::string& zoneDev, + const std::string& hostDev); + + /** + * Create macvlan network device + */ + void createNetdevMacvlan(const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode); + + /** + * Move network device to zone + */ + void moveNetdev(const std::string& devId); + private: const ZoneConfig& mConfig; lxc::LxcZone mZone; diff --git a/server/zone.cpp b/server/zone.cpp index e8580e1..5c15f21 100644 --- a/server/zone.cpp +++ b/server/zone.cpp @@ -207,6 +207,27 @@ bool Zone::activateVT() return true; } +void Zone::createNetdevVeth(const std::string& zoneDev, + const std::string& hostDev) +{ + Lock lock(mReconnectMutex); + mAdmin->createNetdevVeth(zoneDev, hostDev); +} + +void Zone::createNetdevMacvlan(const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode) +{ + Lock lock(mReconnectMutex); + mAdmin->createNetdevMacvlan(zoneDev, hostDev, mode); +} + +void Zone::moveNetdev(const std::string& devId) +{ + Lock lock(mReconnectMutex); + mAdmin->moveNetdev(devId); +} + void Zone::goForeground() { Lock lock(mReconnectMutex); diff --git a/server/zone.hpp b/server/zone.hpp index b708811..759faa3 100644 --- a/server/zone.hpp +++ b/server/zone.hpp @@ -267,6 +267,24 @@ public: */ std::string getRootPath() const; + /** + * Create veth network device + */ + void createNetdevVeth(const std::string& zoneDev, + const std::string& hostDev); + + /** + * Create macvlan network device + */ + void createNetdevMacvlan(const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode); + + /** + * Move network device to zone + */ + void moveNetdev(const std::string& devId); + private: utils::Worker::Pointer mWorker; ZoneConfig mConfig; diff --git a/server/zones-manager.cpp b/server/zones-manager.cpp index bccfd21..fd8501f 100644 --- a/server/zones-manager.cpp +++ b/server/zones-manager.cpp @@ -141,6 +141,15 @@ ZonesManager::ZonesManager(const std::string& configPath) mHostConnection.setGetZoneInfoCallback(bind(&ZonesManager::handleGetZoneInfoCall, this, _1, _2)); + mHostConnection.setCreateNetdevVethCallback(bind(&ZonesManager::handleCreateNetdevVethCall, + this, _1, _2, _3, _4)); + + mHostConnection.setCreateNetdevMacvlanCallback(bind(&ZonesManager::handleCreateNetdevMacvlanCall, + this, _1, _2, _3, _4, _5)); + + mHostConnection.setCreateNetdevPhysCallback(bind(&ZonesManager::handleCreateNetdevPhysCall, + this, _1, _2, _3)); + mHostConnection.setDeclareFileCallback(bind(&ZonesManager::handleDeclareFileCall, this, _1, _2, _3, _4, _5, _6)); @@ -792,6 +801,66 @@ void ZonesManager::handleGetZoneInfoCall(const std::string& id, zone.getRootPath().c_str())); } +void ZonesManager::handleCreateNetdevVethCall(const std::string& zone, + const std::string& zoneDev, + const std::string& hostDev, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("CreateNetdevVeth call"); + try { + Lock lock(mMutex); + + getZone(zone).createNetdevVeth(zoneDev, hostDev); + result->setVoid(); + } catch (const std::out_of_range&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't create veth: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + +void ZonesManager::handleCreateNetdevMacvlanCall(const std::string& zone, + const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("CreateNetdevMacvlan call"); + try { + Lock lock(mMutex); + + getZone(zone).createNetdevMacvlan(zoneDev, hostDev, mode); + result->setVoid(); + } catch (const std::out_of_range&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't create macvlan: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + +void ZonesManager::handleCreateNetdevPhysCall(const std::string& zone, + const std::string& devId, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("CreateNetdevPhys call"); + try { + Lock lock(mMutex); + + getZone(zone).moveNetdev(devId); + result->setVoid(); + } catch (const std::out_of_range&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't create netdev: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + void ZonesManager::handleDeclareFileCall(const std::string& zone, const int32_t& type, const std::string& path, diff --git a/server/zones-manager.hpp b/server/zones-manager.hpp index 6453278..df5e12b 100644 --- a/server/zones-manager.hpp +++ b/server/zones-manager.hpp @@ -161,6 +161,18 @@ private: void handleGetZoneIdsCall(dbus::MethodResultBuilder::Pointer result); void handleGetActiveZoneIdCall(dbus::MethodResultBuilder::Pointer result); void handleGetZoneInfoCall(const std::string& id, dbus::MethodResultBuilder::Pointer result); + void handleCreateNetdevVethCall(const std::string& zone, + const std::string& zoneDev, + const std::string& hostDev, + dbus::MethodResultBuilder::Pointer result); + void handleCreateNetdevMacvlanCall(const std::string& zone, + const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode, + dbus::MethodResultBuilder::Pointer result); + void handleCreateNetdevPhysCall(const std::string& zone, + const std::string& devId, + dbus::MethodResultBuilder::Pointer result); void handleDeclareFileCall(const std::string& zone, const int32_t& type, const std::string& path, -- 2.7.4 From 3099bfee950f1f9d83e9262326031295fbc70692 Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Wed, 18 Feb 2015 16:16:43 +0100 Subject: [PATCH 03/16] Better exception testing [Bug/Feature] N/A [Cause] N/A [Solution] N/A [Verification] Run tests Change-Id: I814deebc0f8e4fdecd5760eb2b1dce1fe6e7df15 --- common/base-exception.hpp | 2 +- common/utils/exception.hpp | 2 +- common/utils/fs.cpp | 2 +- common/utils/value-latch.hpp | 6 ++-- server/exception.hpp | 12 +++---- server/input-monitor.cpp | 12 +++---- tests/unit_tests/dbus/ut-connection.cpp | 7 ++-- tests/unit_tests/log/ut-logger.cpp | 6 ++-- tests/unit_tests/lxc/ut-zone.cpp | 2 +- tests/unit_tests/server/ut-input-monitor.cpp | 18 +++++----- tests/unit_tests/server/ut-server.cpp | 4 ++- tests/unit_tests/server/ut-zone-admin.cpp | 12 ++++--- tests/unit_tests/server/ut-zone-connection.cpp | 21 ++++-------- tests/unit_tests/server/ut-zone-provision.cpp | 10 ++---- tests/unit_tests/server/ut-zone.cpp | 8 +++-- tests/unit_tests/server/ut-zones-manager.cpp | 47 +++++++++++++------------- tests/unit_tests/ut.hpp | 23 +++++++++++++ tests/unit_tests/utils/ut-fs.cpp | 7 ++-- tests/unit_tests/utils/ut-value-latch.cpp | 12 +++++-- zone-daemon/exception.hpp | 2 +- 20 files changed, 122 insertions(+), 93 deletions(-) diff --git a/common/base-exception.hpp b/common/base-exception.hpp index 5637e76..89b79bc 100644 --- a/common/base-exception.hpp +++ b/common/base-exception.hpp @@ -38,7 +38,7 @@ namespace vasum { */ struct VasumException: public std::runtime_error { - VasumException(const std::string& error = "") : std::runtime_error(error) {} + VasumException(const std::string& error) : std::runtime_error(error) {} }; /** diff --git a/common/utils/exception.hpp b/common/utils/exception.hpp index 56fd0bd..32f416c 100644 --- a/common/utils/exception.hpp +++ b/common/utils/exception.hpp @@ -37,7 +37,7 @@ namespace vasum { */ struct UtilsException: public VasumException { - UtilsException(const std::string& error = "") : VasumException(error) {} + UtilsException(const std::string& error) : VasumException(error) {} }; diff --git a/common/utils/fs.cpp b/common/utils/fs.cpp index b2260cc..d7ae2dd 100644 --- a/common/utils/fs.cpp +++ b/common/utils/fs.cpp @@ -53,7 +53,7 @@ std::string readFileContent(const std::string& path) { std::string result; if (!readFileContent(path, result)) { - throw UtilsException(); + throw UtilsException("Read failed"); } return result; } diff --git a/common/utils/value-latch.hpp b/common/utils/value-latch.hpp index 72c7873..0e06f6f 100644 --- a/common/utils/value-latch.hpp +++ b/common/utils/value-latch.hpp @@ -78,7 +78,7 @@ void ValueLatch::set(const T& value) { std::unique_lock lock(mMutex); if (mValue) { - throw UtilsException("Cannot set ValueLatch multiple times!"); + throw UtilsException("Cannot set value multiple times"); } mValue.reset(new T(value)); mCondition.notify_one(); @@ -89,7 +89,7 @@ void ValueLatch::set(T&& value) { std::unique_lock lock(mMutex); if (mValue) { - throw UtilsException("Cannot set ValueLatch multiple times!"); + throw UtilsException("Cannot set value multiple times"); } mValue.reset(new T(std::move(value))); mCondition.notify_one(); @@ -116,7 +116,7 @@ T ValueLatch::get(const unsigned int timeoutMs) std::unique_ptr retValue(std::move(mValue)); return T(std::move(*retValue)); } else { - throw UtilsException("Timeout occured."); + throw UtilsException("Timeout occured"); } } diff --git a/server/exception.hpp b/server/exception.hpp index cb358b1..bf5a106 100644 --- a/server/exception.hpp +++ b/server/exception.hpp @@ -37,7 +37,7 @@ namespace vasum { */ struct ServerException: public VasumException { - ServerException(const std::string& error = "") : VasumException(error) {} + ServerException(const std::string& error) : VasumException(error) {} }; /** @@ -46,7 +46,7 @@ struct ServerException: public VasumException { */ struct ZoneOperationException: public ServerException { - ZoneOperationException(const std::string& error = "") : ServerException(error) {} + ZoneOperationException(const std::string& error) : ServerException(error) {} }; /** @@ -54,7 +54,7 @@ struct ZoneOperationException: public ServerException { */ struct InvalidZoneIdException : public ServerException { - InvalidZoneIdException(const std::string& error = "") : ServerException(error) {} + InvalidZoneIdException(const std::string& error) : ServerException(error) {} }; /** @@ -62,7 +62,7 @@ struct InvalidZoneIdException : public ServerException { */ struct ZoneConnectionException: public ServerException { - ZoneConnectionException(const std::string& error = "") : ServerException(error) {} + ZoneConnectionException(const std::string& error) : ServerException(error) {} }; /** @@ -70,7 +70,7 @@ struct ZoneConnectionException: public ServerException { */ struct HostConnectionException: public ServerException { - HostConnectionException(const std::string& error = "") : ServerException(error) {} + HostConnectionException(const std::string& error) : ServerException(error) {} }; /** @@ -79,7 +79,7 @@ struct HostConnectionException: public ServerException { */ struct InputMonitorException: public ServerException { - InputMonitorException(const std::string& error = "") : ServerException(error) {} + InputMonitorException(const std::string& error) : ServerException(error) {} }; diff --git a/server/input-monitor.cpp b/server/input-monitor.cpp index 7937840..68e63c0 100644 --- a/server/input-monitor.cpp +++ b/server/input-monitor.cpp @@ -71,12 +71,12 @@ InputMonitor::InputMonitor(const InputConfig& inputConfig, { if (mConfig.timeWindowMs > MAX_TIME_WINDOW_SEC * 1000L) { LOGE("Time window exceeds maximum: " << MAX_TIME_WINDOW_SEC); - throw InputMonitorException(); + throw InputMonitorException("Time window exceeds maximum"); } if (mConfig.numberOfEvents > MAX_NUMBER_OF_EVENTS) { LOGE("Number of events exceeds maximum: " << MAX_NUMBER_OF_EVENTS); - throw InputMonitorException(); + throw InputMonitorException("Number of events exceeds maximum"); } std::string devicePath = getDevicePath(); @@ -168,7 +168,7 @@ std::string InputMonitor::getDevicePath() const if (it == end) { LOGE("None of the files under '" << DEVICE_DIR << "' represents device named: " << device); - throw InputMonitorException(); + throw InputMonitorException("Cannot find a device"); } return it->path().string(); @@ -183,7 +183,7 @@ void InputMonitor::createGIOChannel(const std::string& devicePath) if (fd < 0) { LOGE("Cannot create input monitor channel. Device file: " << devicePath << " doesn't exist"); - throw InputMonitorException(); + throw InputMonitorException("Device does not exist"); } mChannelPtr = g_io_channel_unix_new(fd); @@ -193,7 +193,7 @@ void InputMonitor::createGIOChannel(const std::string& devicePath) NULL, NULL) != G_IO_STATUS_NORMAL) { LOGE("Cannot set encoding for input monitor channel "); - throw InputMonitorException(); + throw InputMonitorException("Cannot set encoding"); } using namespace std::placeholders; @@ -207,7 +207,7 @@ void InputMonitor::createGIOChannel(const std::string& devicePath) &utils::deleteCallbackWrapper); if (!mSourceId) { LOGE("Cannot add watch on device input file"); - throw InputMonitorException(); + throw InputMonitorException("Cannot add watch"); } } diff --git a/tests/unit_tests/dbus/ut-connection.cpp b/tests/unit_tests/dbus/ut-connection.cpp index 94b02b7..5182225 100644 --- a/tests/unit_tests/dbus/ut-connection.cpp +++ b/tests/unit_tests/dbus/ut-connection.cpp @@ -574,10 +574,9 @@ BOOST_AUTO_TEST_CASE(DbusApiTest) BOOST_CHECK_EQUAL("Processed: arg", client.process("arg")); BOOST_CHECK_NO_THROW(client.throwException(0)); - auto checkException = [](const DbusCustomException& e) { - return e.what() == std::string("Argument: 666"); - }; - BOOST_CHECK_EXCEPTION(client.throwException(666), DbusCustomException, checkException); + BOOST_CHECK_EXCEPTION(client.throwException(666), + DbusCustomException, + WhatEquals("Argument: 666")); } BOOST_AUTO_TEST_CASE(DbusApiNotifyTest) diff --git a/tests/unit_tests/log/ut-logger.cpp b/tests/unit_tests/log/ut-logger.cpp index 74028db..ef3634e 100644 --- a/tests/unit_tests/log/ut-logger.cpp +++ b/tests/unit_tests/log/ut-logger.cpp @@ -33,7 +33,7 @@ #include -BOOST_AUTO_TEST_SUITE(LogSuite) +BOOST_AUTO_TEST_SUITE(LoggerSuite) using namespace logger; @@ -135,7 +135,9 @@ BOOST_AUTO_TEST_CASE(StringLogLevelSetandGet) Logger::setLogLevel("ERROR"); BOOST_CHECK(LogLevel::ERROR == Logger::getLogLevel()); - BOOST_REQUIRE_THROW(Logger::setLogLevel("UNKNOWN"), std::runtime_error); + BOOST_REQUIRE_EXCEPTION(Logger::setLogLevel("UNKNOWN"), + std::runtime_error, + WhatEquals("Invalid LogLevel to parse")); //TODO change message } BOOST_AUTO_TEST_CASE(TestLogsError) diff --git a/tests/unit_tests/lxc/ut-zone.cpp b/tests/unit_tests/lxc/ut-zone.cpp index 6840e31..3b375bd 100644 --- a/tests/unit_tests/lxc/ut-zone.cpp +++ b/tests/unit_tests/lxc/ut-zone.cpp @@ -94,7 +94,7 @@ BOOST_AUTO_TEST_CASE(CreateDestroyTest) BOOST_CHECK(lxc.isDefined()); BOOST_CHECK_EQUAL(lxc.getConfigItem("lxc.rootfs"), LXC_PATH + ZONE_NAME + "/rootfs"); - BOOST_CHECK_THROW(lxc.getConfigItem("xxx"), LxcException); + BOOST_CHECK_EXCEPTION(lxc.getConfigItem("xxx"), LxcException, WhatEquals("Key not found")); BOOST_CHECK(lxc.destroy()); diff --git a/tests/unit_tests/server/ut-input-monitor.cpp b/tests/unit_tests/server/ut-input-monitor.cpp index 241da6c..9b4292f 100644 --- a/tests/unit_tests/server/ut-input-monitor.cpp +++ b/tests/unit_tests/server/ut-input-monitor.cpp @@ -94,23 +94,25 @@ BOOST_FIXTURE_TEST_SUITE(InputMonitorSuite, Fixture) BOOST_AUTO_TEST_CASE(Config_OK) { - BOOST_REQUIRE_NO_THROW(InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback())); + InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback()); } BOOST_AUTO_TEST_CASE(Config_timeWindowMsTooHigh) { inputConfig.timeWindowMs = 50000; - BOOST_REQUIRE_THROW(InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback()), - InputMonitorException); + BOOST_REQUIRE_EXCEPTION(InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback()), + InputMonitorException, + WhatEquals("Time window exceeds maximum")); } BOOST_AUTO_TEST_CASE(Config_deviceFilePathNotExisting) { inputConfig.device = TEST_INPUT_DEVICE + "notExisting"; - BOOST_REQUIRE_THROW(InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback()), - InputMonitorException); + BOOST_REQUIRE_EXCEPTION(InputMonitor inputMonitor(inputConfig, InputMonitor::NotifyCallback()), + InputMonitorException, + WhatEquals("Cannot find a device")); } namespace { @@ -119,8 +121,7 @@ void sendNEvents(Fixture& f, unsigned int noOfEventsToSend) { Latch eventLatch; - std::unique_ptr inputMonitor; - BOOST_REQUIRE_NO_THROW(inputMonitor.reset(new InputMonitor(f.inputConfig, [&] {eventLatch.set();}))); + InputMonitor inputMonitor(f.inputConfig, [&] {eventLatch.set();}); int fd = ::open(TEST_INPUT_DEVICE.c_str(), O_WRONLY); BOOST_REQUIRE(fd >= 0); @@ -166,8 +167,7 @@ void sendNEventsWithPauses(Fixture& f, unsigned int noOfEventsToSend) { Latch eventLatch; - std::unique_ptr inputMonitor; - BOOST_REQUIRE_NO_THROW(inputMonitor.reset(new InputMonitor(f.inputConfig, [&] {eventLatch.set();}))); + InputMonitor inputMonitor(f.inputConfig, [&] {eventLatch.set();}); int fd = ::open(TEST_INPUT_DEVICE.c_str(), O_WRONLY); BOOST_REQUIRE(fd >= 0); diff --git a/tests/unit_tests/server/ut-server.cpp b/tests/unit_tests/server/ut-server.cpp index c65e652..4df5d89 100644 --- a/tests/unit_tests/server/ut-server.cpp +++ b/tests/unit_tests/server/ut-server.cpp @@ -82,7 +82,9 @@ BOOST_AUTO_TEST_CASE(ConstructorDestructorTest) BOOST_AUTO_TEST_CASE(MissingConfigTest) { - BOOST_REQUIRE_THROW(Server(MISSING_CONFIG_PATH).run(AS_ROOT), ConfigException);//TODO check message + BOOST_REQUIRE_EXCEPTION(Server(MISSING_CONFIG_PATH).run(AS_ROOT), + ConfigException, + WhatEquals("Could not load " + MISSING_CONFIG_PATH)); } BOOST_AUTO_TEST_CASE(TerminateTest) diff --git a/tests/unit_tests/server/ut-zone-admin.cpp b/tests/unit_tests/server/ut-zone-admin.cpp index 21cc850..255e75e 100644 --- a/tests/unit_tests/server/ut-zone-admin.cpp +++ b/tests/unit_tests/server/ut-zone-admin.cpp @@ -87,7 +87,9 @@ BOOST_AUTO_TEST_CASE(ConstructorDestructorTest) BOOST_AUTO_TEST_CASE(MissingConfigTest) { - BOOST_REQUIRE_THROW(create(MISSING_CONFIG_PATH), ZoneOperationException);//TODO check message + BOOST_REQUIRE_EXCEPTION(create(MISSING_CONFIG_PATH), + ZoneOperationException, + WhatEquals("Could not create zone")); } BOOST_AUTO_TEST_CASE(StartTest) @@ -103,7 +105,9 @@ BOOST_AUTO_TEST_CASE(StartTest) BOOST_AUTO_TEST_CASE(StartBuggyTest) { auto admin = create(BUGGY_CONFIG_PATH); - BOOST_REQUIRE_THROW(admin->start(), ZoneOperationException);//TODO check message + BOOST_REQUIRE_EXCEPTION(admin->start(), + ZoneOperationException, + WhatEquals("Could not start zone")); } BOOST_AUTO_TEST_CASE(StopShutdownTest) @@ -158,9 +162,9 @@ BOOST_AUTO_TEST_CASE(SuspendResumeTest) // // admin->start(); // ensureStarted(); -// BOOST_REQUIRE_NO_THROW(admin->setSchedulerLevel(SchedulerLevel::FOREGROUND)); +// admin->setSchedulerLevel(SchedulerLevel::FOREGROUND); // BOOST_REQUIRE(admin->getSchedulerQuota() == config.cpuQuotaForeground); -// BOOST_REQUIRE_NO_THROW(admin->setSchedulerLevel(SchedulerLevel::BACKGROUND)); +// admin->setSchedulerLevel(SchedulerLevel::BACKGROUND); // BOOST_REQUIRE(admin->getSchedulerQuota() == config.cpuQuotaBackground); //} diff --git a/tests/unit_tests/server/ut-zone-connection.cpp b/tests/unit_tests/server/ut-zone-connection.cpp index a93b469..6d3e8e4 100644 --- a/tests/unit_tests/server/ut-zone-connection.cpp +++ b/tests/unit_tests/server/ut-zone-connection.cpp @@ -132,7 +132,7 @@ BOOST_AUTO_TEST_CASE(ConstructorDestructorConnectTest) ScopedGlibLoop loop; ScopedDbusDaemon dbus; - BOOST_REQUIRE_NO_THROW(ZoneConnection(dbus.acquireAddress(), nullptr)); + ZoneConnection(dbus.acquireAddress(), nullptr); } BOOST_AUTO_TEST_CASE(NotifyActiveZoneApiTest) @@ -141,16 +141,14 @@ BOOST_AUTO_TEST_CASE(NotifyActiveZoneApiTest) ScopedDbusDaemon dbus; Latch notifyCalled; - std::unique_ptr connection; - - BOOST_REQUIRE_NO_THROW(connection.reset(new ZoneConnection(dbus.acquireAddress(), nullptr))); + ZoneConnection connection(dbus.acquireAddress(), nullptr); auto callback = [&](const std::string& application, const std::string& message) { if (application == "testapp" && message == "testmessage") { notifyCalled.set(); } }; - connection->setNotifyActiveZoneCallback(callback); + connection.setNotifyActiveZoneCallback(callback); DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); client->callMethod(api::zone::BUS_NAME, @@ -168,9 +166,7 @@ BOOST_AUTO_TEST_CASE(SignalNotificationApiTest) ScopedDbusDaemon dbus; Latch signalEmitted; - std::unique_ptr connection; - - BOOST_REQUIRE_NO_THROW(connection.reset(new ZoneConnection(dbus.acquireAddress(), nullptr))); + ZoneConnection connection(dbus.acquireAddress(), nullptr); DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); @@ -197,7 +193,7 @@ BOOST_AUTO_TEST_CASE(SignalNotificationApiTest) }; client->signalSubscribe(handler, api::zone::BUS_NAME); - connection->sendNotification("testzone", "testapp", "testmessage"); + connection.sendNotification("testzone", "testapp", "testmessage"); BOOST_CHECK(signalEmitted.wait(EVENT_TIMEOUT)); } @@ -208,10 +204,7 @@ BOOST_AUTO_TEST_CASE(SignalDisplayOffApiTest) ScopedDbusDaemon dbus; Latch displayOffCalled; - std::unique_ptr connection; - - BOOST_REQUIRE_NO_THROW(connection.reset(new ZoneConnection(dbus.acquireAddress(), - nullptr))); + ZoneConnection connection(dbus.acquireAddress(), nullptr); DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); @@ -219,7 +212,7 @@ BOOST_AUTO_TEST_CASE(SignalDisplayOffApiTest) displayOffCalled.set(); }; - connection->setDisplayOffCallback(callback); + connection.setDisplayOffCallback(callback); client->emitSignal(fake_power_manager_api::OBJECT_PATH, fake_power_manager_api::INTERFACE, diff --git a/tests/unit_tests/server/ut-zone-provision.cpp b/tests/unit_tests/server/ut-zone-provision.cpp index 9038fd4..4109cb2 100644 --- a/tests/unit_tests/server/ut-zone-provision.cpp +++ b/tests/unit_tests/server/ut-zone-provision.cpp @@ -89,12 +89,6 @@ struct Fixture { } }; -std::function expectedMessage(const std::string& message) { - return [=](const std::exception& e) { - return e.what() == message; - }; -} - } // namespace @@ -265,7 +259,7 @@ BOOST_AUTO_TEST_CASE(DeclareMountTest) zoneProvision.declareMount("/fake/path2", "/fake/path2", "tmpfs", 077, "fake"); BOOST_CHECK_EXCEPTION(zoneProvision.declareMount("/fake/path2", "/fake/path2", "tmpfs", 077, "fake"), UtilsException, - expectedMessage("Provision already exists")); + WhatEquals("Provision already exists")); ZoneProvisioningConfig config; load(config); @@ -397,7 +391,7 @@ BOOST_AUTO_TEST_CASE(RemoveTest) zoneProvision.remove("link /fake/path2 /fake/path4"); BOOST_CHECK_EXCEPTION(zoneProvision.remove("link /fake/path_fake /fake/path2"), UtilsException, - expectedMessage("Can't find provision")); + WhatEquals("Can't find provision")); const std::vector provisions = zoneProvision.list(); BOOST_REQUIRE_EQUAL(provisions.size(), expected.size()); diff --git a/tests/unit_tests/server/ut-zone.cpp b/tests/unit_tests/server/ut-zone.cpp index dbdf65a..3cd6031 100644 --- a/tests/unit_tests/server/ut-zone.cpp +++ b/tests/unit_tests/server/ut-zone.cpp @@ -94,12 +94,16 @@ BOOST_AUTO_TEST_CASE(ConstructorDestructorTest) BOOST_AUTO_TEST_CASE(BuggyConfigTest) { - BOOST_REQUIRE_THROW(create(BUGGY_CONFIG_PATH), ZoneOperationException);//TODO check message + BOOST_REQUIRE_EXCEPTION(create(BUGGY_CONFIG_PATH), + ZoneOperationException, + WhatEquals("Could not create zone")); } BOOST_AUTO_TEST_CASE(MissingConfigTest) { - BOOST_REQUIRE_THROW(create(MISSING_CONFIG_PATH), ConfigException);//TODO check message + BOOST_REQUIRE_EXCEPTION(create(MISSING_CONFIG_PATH), + ConfigException, + WhatEquals("Could not load " + MISSING_CONFIG_PATH)); } BOOST_AUTO_TEST_CASE(StartStopTest) diff --git a/tests/unit_tests/server/ut-zones-manager.cpp b/tests/unit_tests/server/ut-zones-manager.cpp index 58ac060..e93ab84 100644 --- a/tests/unit_tests/server/ut-zones-manager.cpp +++ b/tests/unit_tests/server/ut-zones-manager.cpp @@ -451,13 +451,6 @@ private: } }; -std::function expectedMessage(const std::string& message) -{ - return [=](const std::exception& e) { - return e.what() == message; - }; -} - template bool spinWaitFor(int timeoutMs, Predicate pred) { @@ -497,7 +490,9 @@ BOOST_AUTO_TEST_CASE(ConstructorDestructorTest) BOOST_AUTO_TEST_CASE(MissingConfigTest) { - BOOST_REQUIRE_THROW(ZonesManager cm(MISSING_CONFIG_PATH), ConfigException); + BOOST_REQUIRE_EXCEPTION(ZonesManager{MISSING_CONFIG_PATH}, + ConfigException, + WhatEquals("Could not load " + MISSING_CONFIG_PATH)); } BOOST_AUTO_TEST_CASE(CreateTest) @@ -867,12 +862,12 @@ BOOST_AUTO_TEST_CASE(ProxyCallTest) // host -> unknown BOOST_CHECK_EXCEPTION(dbuses.at(0)->testApiProxyCall("unknown", "param"), DbusCustomException, - expectedMessage("Unknown proxy call target")); + WhatEquals("Unknown proxy call target")); // forwarding error BOOST_CHECK_EXCEPTION(dbuses.at(0)->testApiProxyCall("host", ""), DbusCustomException, - expectedMessage("Test error")); + WhatEquals("Test error")); // forbidden call BOOST_CHECK_EXCEPTION(dbuses.at(0)->proxyCall("host", @@ -882,7 +877,7 @@ BOOST_AUTO_TEST_CASE(ProxyCallTest) "foo", g_variant_new("(s)", "arg")), DbusCustomException, - expectedMessage("Proxy call forbidden")); + WhatEquals("Proxy call forbidden")); } namespace { @@ -1039,12 +1034,14 @@ BOOST_AUTO_TEST_CASE(SetActiveZoneTest) BOOST_CHECK(dbus.callMethodGetActiveZoneId() == zoneId); } - BOOST_REQUIRE_THROW(dbus.callMethodSetActiveZone(NON_EXISTANT_ZONE_ID), - DbusException); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodSetActiveZone(NON_EXISTANT_ZONE_ID), + DbusException, + WhatEquals("No such zone id")); cm.stopAll(); - BOOST_REQUIRE_THROW(dbus.callMethodSetActiveZone("zone1"), - DbusException); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodSetActiveZone("zone1"), + DbusException, + WhatEquals("Could not activate stopped or paused zone")); } BOOST_AUTO_TEST_CASE(CreateDestroyZoneTest) @@ -1204,16 +1201,20 @@ BOOST_AUTO_TEST_CASE(LockUnlockZoneTest) BOOST_CHECK(cm.isRunning(zoneId)); } - BOOST_REQUIRE_THROW(dbus.callMethodLockZone(NON_EXISTANT_ZONE_ID), - DbusException); - BOOST_REQUIRE_THROW(dbus.callMethodUnlockZone(NON_EXISTANT_ZONE_ID), - DbusException); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodLockZone(NON_EXISTANT_ZONE_ID), + DbusException, + WhatEquals("No such zone id")); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodUnlockZone(NON_EXISTANT_ZONE_ID), + DbusException, + WhatEquals("No such zone id")); cm.stopAll(); - BOOST_REQUIRE_THROW(dbus.callMethodLockZone("zone1"), - DbusException); - BOOST_REQUIRE_THROW(dbus.callMethodUnlockZone("zone1"), - DbusException); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodLockZone("zone1"), + DbusException, + WhatEquals("Zone is not running")); + BOOST_REQUIRE_EXCEPTION(dbus.callMethodUnlockZone("zone1"), + DbusException, + WhatEquals("Zone is not paused")); } BOOST_AUTO_TEST_SUITE_END() diff --git a/tests/unit_tests/ut.hpp b/tests/unit_tests/ut.hpp index cbc322a..d8b4b94 100644 --- a/tests/unit_tests/ut.hpp +++ b/tests/unit_tests/ut.hpp @@ -28,4 +28,27 @@ #define BOOST_TEST_DYN_LINK #include +#include + +/** + * An exception message checker + * + * Usage example: + * BOOST_CHECK_EXCEPTION(foo(), SomeException, WhatEquals("oops")) + */ +class WhatEquals { +public: + explicit WhatEquals(const std::string& message) + : mMessage(message) {} + + template + bool operator()(const T& e) + { + BOOST_WARN_EQUAL(e.what(), mMessage); // additional failure info + return e.what() == mMessage; + } +private: + std::string mMessage; +}; + #endif // UNIT_TESTS_UT_HPP diff --git a/tests/unit_tests/utils/ut-fs.cpp b/tests/unit_tests/utils/ut-fs.cpp index 63b4f63..365c1e1 100644 --- a/tests/unit_tests/utils/ut-fs.cpp +++ b/tests/unit_tests/utils/ut-fs.cpp @@ -76,7 +76,9 @@ const std::string FILE_NAME_RANDOM_2 = BOOST_AUTO_TEST_CASE(ReadFileContentTest) { BOOST_CHECK_EQUAL(FILE_CONTENT, readFileContent(FILE_PATH)); - BOOST_CHECK_THROW(readFileContent(BUGGY_FILE_PATH), UtilsException); + BOOST_CHECK_EXCEPTION(readFileContent(BUGGY_FILE_PATH), + UtilsException, + WhatEquals("Read failed")); } BOOST_AUTO_TEST_CASE(SaveFileContentTest) @@ -217,8 +219,7 @@ BOOST_AUTO_TEST_CASE(CopyDirContentsTest) BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_RANDOM_1), FILE_CONTENT_3); BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_RANDOM_2), FILE_CONTENT_2); - fs::file_status st; - BOOST_REQUIRE_NO_THROW(st = fs::status(fs::path(dst_inner2))); + fs::file_status st = fs::status(fs::path(dst_inner2)); BOOST_CHECK(fs::owner_read == st.permissions()); } diff --git a/tests/unit_tests/utils/ut-value-latch.cpp b/tests/unit_tests/utils/ut-value-latch.cpp index a16128e..711c8e6 100644 --- a/tests/unit_tests/utils/ut-value-latch.cpp +++ b/tests/unit_tests/utils/ut-value-latch.cpp @@ -105,7 +105,9 @@ BOOST_AUTO_TEST_CASE(TimeoutTest) { ValueLatch testLatch; - BOOST_REQUIRE_THROW(testLatch.get(EXPECTED_TIMEOUT), vasum::UtilsException); + BOOST_REQUIRE_EXCEPTION(testLatch.get(EXPECTED_TIMEOUT), + vasum::UtilsException, + WhatEquals("Timeout occured")); } BOOST_AUTO_TEST_CASE(MultipleSetTest) @@ -113,7 +115,9 @@ BOOST_AUTO_TEST_CASE(MultipleSetTest) ValueLatch testLatch; testLatch.set(3); - BOOST_REQUIRE_THROW(testLatch.set(2), vasum::UtilsException); + BOOST_REQUIRE_EXCEPTION(testLatch.set(2), + vasum::UtilsException, + WhatEquals("Cannot set value multiple times")); } BOOST_AUTO_TEST_CASE(MultipleGetTest) @@ -122,7 +126,9 @@ BOOST_AUTO_TEST_CASE(MultipleGetTest) testLatch.set(3); testLatch.get(TIMEOUT); - BOOST_REQUIRE_THROW(testLatch.get(EXPECTED_TIMEOUT), vasum::UtilsException); + BOOST_REQUIRE_EXCEPTION(testLatch.get(EXPECTED_TIMEOUT), + vasum::UtilsException, + WhatEquals("Timeout occured")); } BOOST_AUTO_TEST_SUITE_END() diff --git a/zone-daemon/exception.hpp b/zone-daemon/exception.hpp index a281910..5fcf54d 100644 --- a/zone-daemon/exception.hpp +++ b/zone-daemon/exception.hpp @@ -37,7 +37,7 @@ namespace zone_daemon { */ struct ZoneDaemonException: public VasumException { - ZoneDaemonException(const std::string& error = "") : VasumException(error) {} + ZoneDaemonException(const std::string& error) : VasumException(error) {} }; -- 2.7.4 From cce992d41fef71d7725b3dbebc54a24db61f1f7e Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Thu, 19 Feb 2015 12:28:27 +0100 Subject: [PATCH 04/16] Unify unit tests cleanup [Bug/Feature] N/A [Cause] N/A [Solution] N/A [Verification] Run tests Change-Id: I2a49450b3991355656028194b14f9eae5a222456 --- .../dbus/configs/ut-connection/ut-dbus.conf | 2 +- tests/unit_tests/dbus/test-common.hpp | 5 +- tests/unit_tests/dbus/ut-connection.cpp | 7 +- tests/unit_tests/ipc/ut-ipc.cpp | 105 +++++++------- .../server/configs/ut-zone-connection/ut-dbus.conf | 2 +- tests/unit_tests/server/ut-input-monitor.cpp | 15 +- tests/unit_tests/server/ut-zone-connection.cpp | 42 +++--- tests/unit_tests/utils/ut-fs.cpp | 160 ++++++++++----------- 8 files changed, 162 insertions(+), 176 deletions(-) diff --git a/tests/unit_tests/dbus/configs/ut-connection/ut-dbus.conf b/tests/unit_tests/dbus/configs/ut-connection/ut-dbus.conf index f3bf58c..d6091a2 100644 --- a/tests/unit_tests/dbus/configs/ut-connection/ut-dbus.conf +++ b/tests/unit_tests/dbus/configs/ut-connection/ut-dbus.conf @@ -5,7 +5,7 @@ custom - unix:path=/tmp/zone_socket + unix:path=/tmp/ut-dbus/test-socket diff --git a/tests/unit_tests/dbus/test-common.hpp b/tests/unit_tests/dbus/test-common.hpp index ae76d82..7b6262b 100644 --- a/tests/unit_tests/dbus/test-common.hpp +++ b/tests/unit_tests/dbus/test-common.hpp @@ -31,8 +31,9 @@ namespace vasum { -const std::string DBUS_SOCKET_FILE = "/tmp/zone_socket"; -const std::string DBUS_ADDRESS = "unix:path=" + DBUS_SOCKET_FILE; +const std::string DBUS_SOCKET_DIR = "/tmp/ut-dbus"; +const std::string DBUS_SOCKET_PATH = DBUS_SOCKET_DIR + "/test-socket"; +const std::string DBUS_ADDRESS = "unix:path=" + DBUS_SOCKET_PATH; const std::string TESTAPI_BUS_NAME = "org.tizen.tests"; const std::string TESTAPI_OBJECT_PATH = "/org/tizen/tests"; diff --git a/tests/unit_tests/dbus/ut-connection.cpp b/tests/unit_tests/dbus/ut-connection.cpp index 5182225..4948cd7 100644 --- a/tests/unit_tests/dbus/ut-connection.cpp +++ b/tests/unit_tests/dbus/ut-connection.cpp @@ -28,6 +28,7 @@ #include "dbus/test-client.hpp" #include "dbus/test-common.hpp" #include "utils/scoped-daemon.hpp" +#include "utils/scoped-dir.hpp" #include "dbus/connection.hpp" #include "dbus/exception.hpp" @@ -37,7 +38,6 @@ #include "utils/fs.hpp" #include "logger/logger.hpp" -#include #include #include #include @@ -65,16 +65,17 @@ const int EVENT_TIMEOUT = 1000; class ScopedDbusDaemon { public: ScopedDbusDaemon() + : mTestPathGuard(DBUS_SOCKET_DIR) { - boost::filesystem::remove("/tmp/zone_socket"); mDaemon.start(DBUS_DAEMON_PROC, DBUS_DAEMON_ARGS); - waitForFile(DBUS_SOCKET_FILE, DBUS_DAEMON_TIMEOUT); + waitForFile(DBUS_SOCKET_PATH, DBUS_DAEMON_TIMEOUT); } void stop() { mDaemon.stop(); } private: + ScopedDir mTestPathGuard; ScopedDaemon mDaemon; }; diff --git a/tests/unit_tests/ipc/ut-ipc.cpp b/tests/unit_tests/ipc/ut-ipc.cpp index 56c94cb..193eec9 100644 --- a/tests/unit_tests/ipc/ut-ipc.cpp +++ b/tests/unit_tests/ipc/ut-ipc.cpp @@ -37,6 +37,7 @@ #include "utils/glib-loop.hpp" #include "utils/latch.hpp" #include "utils/value-latch.hpp" +#include "utils/scoped-dir.hpp" #include "config/fields.hpp" #include "logger/logger.hpp" @@ -47,13 +48,11 @@ #include #include #include -#include using namespace vasum; using namespace vasum::ipc; using namespace vasum::utils; using namespace std::placeholders; -namespace fs = boost::filesystem; namespace { @@ -66,16 +65,16 @@ const int SHORT_OPERATION_TIME = TIMEOUT / 100; // Time that will cause "TIMEOUT" methods to throw const int LONG_OPERATION_TIME = 1000 + TIMEOUT; +const std::string TEST_DIR = "/tmp/ut-ipc"; +const std::string SOCKET_PATH = TEST_DIR + "/test.socket"; + struct Fixture { - std::string socketPath; + ScopedDir mTestPathGuard; + std::string SOCKET_PATH; Fixture() - : socketPath(fs::unique_path("/tmp/ipc-%%%%.socket").string()) - { - } - ~Fixture() + : mTestPathGuard(TEST_DIR) { - fs::remove(socketPath); } }; @@ -224,13 +223,13 @@ BOOST_FIXTURE_TEST_SUITE(IPCSuite, Fixture) BOOST_AUTO_TEST_CASE(ConstructorDestructor) { - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); } BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, returnEmptyCallback); s.setMethodHandler(1, returnDataCallback); @@ -239,7 +238,7 @@ BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) s.setMethodHandler(1, echoCallback); s.setMethodHandler(2, returnDataCallback); - Client c(socketPath); + Client c(SOCKET_PATH); connect(s, c); testEcho(c, 1); @@ -251,8 +250,8 @@ BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) BOOST_AUTO_TEST_CASE(ClientAddRemoveMethod) { - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); c.setMethodHandler(1, returnEmptyCallback); c.setMethodHandler(1, returnDataCallback); @@ -271,7 +270,7 @@ BOOST_AUTO_TEST_CASE(ClientAddRemoveMethod) BOOST_AUTO_TEST_CASE(ServiceStartStop) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, returnDataCallback); @@ -286,8 +285,8 @@ BOOST_AUTO_TEST_CASE(ServiceStartStop) BOOST_AUTO_TEST_CASE(ClientStartStop) { - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); c.setMethodHandler(1, returnDataCallback); c.start(); @@ -304,11 +303,11 @@ BOOST_AUTO_TEST_CASE(ClientStartStop) BOOST_AUTO_TEST_CASE(SyncClientToServiceEcho) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.setMethodHandler(2, echoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); connect(s, c); testEcho(c, 1); @@ -317,12 +316,12 @@ BOOST_AUTO_TEST_CASE(SyncClientToServiceEcho) BOOST_AUTO_TEST_CASE(Restart) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); s.setMethodHandler(2, echoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); c.start(); testEcho(c, 1); testEcho(c, 2); @@ -342,8 +341,8 @@ BOOST_AUTO_TEST_CASE(Restart) BOOST_AUTO_TEST_CASE(SyncServiceToClientEcho) { - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); c.setMethodHandler(1, echoCallback); PeerID peerID = connect(s, c); @@ -359,10 +358,10 @@ BOOST_AUTO_TEST_CASE(AsyncClientToServiceEcho) ValueLatch> recvDataLatch; // Setup Service and Client - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(socketPath); + Client c(SOCKET_PATH); c.start(); //Async call @@ -381,8 +380,8 @@ BOOST_AUTO_TEST_CASE(AsyncServiceToClientEcho) std::shared_ptr sentData(new SendData(56)); ValueLatch> recvDataLatch; - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); c.setMethodHandler(1, echoCallback); PeerID peerID = connect(s, c); @@ -401,10 +400,10 @@ BOOST_AUTO_TEST_CASE(AsyncServiceToClientEcho) BOOST_AUTO_TEST_CASE(SyncTimeout) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, longEchoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); connect(s, c); std::shared_ptr sentData(new SendData(78)); @@ -413,10 +412,10 @@ BOOST_AUTO_TEST_CASE(SyncTimeout) BOOST_AUTO_TEST_CASE(SerializationError) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); connect(s, c); std::shared_ptr throwingData(new ThrowOnAcceptData()); @@ -427,11 +426,11 @@ BOOST_AUTO_TEST_CASE(SerializationError) BOOST_AUTO_TEST_CASE(ParseError) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(socketPath); + Client c(SOCKET_PATH); c.start(); std::shared_ptr sentData(new SendData(78)); @@ -441,7 +440,7 @@ BOOST_AUTO_TEST_CASE(ParseError) BOOST_AUTO_TEST_CASE(DisconnectedPeerError) { ValueLatch> retStatusLatch; - Service s(socketPath); + Service s(SOCKET_PATH); auto method = [](const PeerID, std::shared_ptr&, MethodResult::Pointer methodResult) { auto resultData = std::make_shared(1); @@ -452,7 +451,7 @@ BOOST_AUTO_TEST_CASE(DisconnectedPeerError) s.setMethodHandler(1, method); s.start(); - Client c(socketPath); + Client c(SOCKET_PATH); c.start(); auto dataBack = [&retStatusLatch](Result && r) { @@ -474,14 +473,14 @@ BOOST_AUTO_TEST_CASE(DisconnectedPeerError) BOOST_AUTO_TEST_CASE(ReadTimeout) { - Service s(socketPath); + Service s(SOCKET_PATH); auto longEchoCallback = [](const PeerID, std::shared_ptr& data, MethodResult::Pointer methodResult) { auto resultData = std::make_shared(data->intVal, LONG_OPERATION_TIME); methodResult->set(resultData); }; s.setMethodHandler(1, longEchoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); connect(s, c); // Test timeout on read @@ -492,11 +491,11 @@ BOOST_AUTO_TEST_CASE(ReadTimeout) BOOST_AUTO_TEST_CASE(WriteTimeout) { - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(socketPath); + Client c(SOCKET_PATH); c.start(); // Test echo with a minimal timeout @@ -516,8 +515,8 @@ BOOST_AUTO_TEST_CASE(AddSignalInRuntime) ValueLatch> recvDataLatchA; ValueLatch> recvDataLatchB; - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); connect(s, c); auto handlerA = [&recvDataLatchA](const PeerID, std::shared_ptr& data) { @@ -552,8 +551,8 @@ BOOST_AUTO_TEST_CASE(AddSignalOffline) ValueLatch> recvDataLatchA; ValueLatch> recvDataLatchB; - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); auto handlerA = [&recvDataLatchA](const PeerID, std::shared_ptr& data) { recvDataLatchA.set(data); @@ -594,10 +593,10 @@ BOOST_AUTO_TEST_CASE(ServiceGSource) }; IPCGSource::Pointer serviceGSource; - Service s(socketPath); + Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); - Client c(socketPath); + Client c(SOCKET_PATH); s.setSignalHandler(2, signalHandler); connectServiceGSource(s, c); @@ -620,11 +619,11 @@ BOOST_AUTO_TEST_CASE(ClientGSource) l.set(); }; - Service s(socketPath); + Service s(SOCKET_PATH); s.start(); IPCGSource::Pointer clientGSource; - Client c(socketPath); + Client c(SOCKET_PATH); c.setMethodHandler(1, echoCallback); c.setSignalHandler(2, signalHandler); @@ -643,8 +642,8 @@ BOOST_AUTO_TEST_CASE(UsersError) const int TEST_ERROR_CODE = -234; const std::string TEST_ERROR_MESSAGE = "Ay, caramba!"; - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); auto clientID = connect(s, c); auto throwingMethodHandler = [&](const PeerID, std::shared_ptr&, MethodResult::Pointer) { @@ -677,8 +676,8 @@ BOOST_AUTO_TEST_CASE(AsyncResult) const int TEST_ERROR_CODE = -567; const std::string TEST_ERROR_MESSAGE = "Ooo jooo!"; - Service s(socketPath); - Client c(socketPath); + Service s(SOCKET_PATH); + Client c(SOCKET_PATH); auto clientID = connect(s, c); auto errorMethodHandler = [&](const PeerID, std::shared_ptr&, MethodResult::Pointer methodResult) { @@ -734,14 +733,14 @@ BOOST_AUTO_TEST_CASE(AsyncResult) // ipc::setMaxFDNumber(50); // // Setup Service and many Clients -// Service s(socketPath); +// Service s(SOCKET_PATH); // s.setMethodHandler(1, echoCallback); // s.start(); // std::list clients; // for (int i = 0; i < 100; ++i) { // try { -// clients.push_back(Client(socketPath)); +// clients.push_back(Client(SOCKET_PATH)); // clients.back().start(); // } catch (...) {} // } diff --git a/tests/unit_tests/server/configs/ut-zone-connection/ut-dbus.conf b/tests/unit_tests/server/configs/ut-zone-connection/ut-dbus.conf index d60e774..66ee3f9 100644 --- a/tests/unit_tests/server/configs/ut-zone-connection/ut-dbus.conf +++ b/tests/unit_tests/server/configs/ut-zone-connection/ut-dbus.conf @@ -5,7 +5,7 @@ custom - unix:path=/tmp/ut-zone-connection/dbus/system_bus_socket + unix:path=/tmp/ut-zones/mount-point/dbus/system_bus_socket diff --git a/tests/unit_tests/server/ut-input-monitor.cpp b/tests/unit_tests/server/ut-input-monitor.cpp index 9b4292f..3560b40 100644 --- a/tests/unit_tests/server/ut-input-monitor.cpp +++ b/tests/unit_tests/server/ut-input-monitor.cpp @@ -32,6 +32,7 @@ #include "utils/glib-loop.hpp" #include "utils/latch.hpp" +#include "utils/scoped-dir.hpp" #include #include @@ -39,7 +40,6 @@ #include #include -#include #include #include #include @@ -50,8 +50,8 @@ using namespace vasum::utils; namespace { -std::string TEST_INPUT_DEVICE = - boost::filesystem::unique_path("/tmp/testInputDevice-%%%%").string(); +const std::string TEST_DIR = "/tmp/ut-input-monitor"; +const std::string TEST_INPUT_DEVICE = TEST_DIR + "/input-device"; const int EVENT_TYPE = 1; const int EVENT_CODE = 139; @@ -61,11 +61,13 @@ const int EVENT_BUTTON_PRESSED = 1; const int SINGLE_EVENT_TIMEOUT = 1000; struct Fixture { - InputConfig inputConfig; utils::ScopedGlibLoop mLoop; + ScopedDir mTestPathGuard; + InputConfig inputConfig; struct input_event ie; Fixture() + : mTestPathGuard(TEST_DIR) { inputConfig.numberOfEvents = 2; inputConfig.device = TEST_INPUT_DEVICE; @@ -80,13 +82,8 @@ struct Fixture { ie.code = EVENT_CODE; ie.value = EVENT_BUTTON_RELEASED; - ::remove(TEST_INPUT_DEVICE.c_str()); BOOST_CHECK(::mkfifo(TEST_INPUT_DEVICE.c_str(), S_IWUSR | S_IRUSR) >= 0); } - ~Fixture() - { - ::remove(TEST_INPUT_DEVICE.c_str()); - } }; } // namespace diff --git a/tests/unit_tests/server/ut-zone-connection.cpp b/tests/unit_tests/server/ut-zone-connection.cpp index 6d3e8e4..fb6df88 100644 --- a/tests/unit_tests/server/ut-zone-connection.cpp +++ b/tests/unit_tests/server/ut-zone-connection.cpp @@ -39,10 +39,9 @@ #include "utils/glib-loop.hpp" #include "utils/latch.hpp" #include "utils/fs.hpp" +#include "utils/scoped-dir.hpp" -BOOST_AUTO_TEST_SUITE(ZoneConnectionSuite) - using namespace vasum; using namespace vasum::utils; using namespace dbus; @@ -57,13 +56,15 @@ const char* const DBUS_DAEMON_ARGS[] = { NULL }; -const std::string TRANSPORT_MOUNT_POINT = "/tmp/ut-zone-connection"; +const std::string ZONES_PATH = "/tmp/ut-zones"; +const std::string TRANSPORT_MOUNT_POINT = ZONES_PATH + "/mount-point"; const int EVENT_TIMEOUT = 1000; -class ScopedDbusDaemon { +class Fixture { public: - ScopedDbusDaemon() - : mTransport(TRANSPORT_MOUNT_POINT) + Fixture() + : mZonesPathGuard(ZONES_PATH) + , mTransport(TRANSPORT_MOUNT_POINT) { mDaemon.start(DBUS_DAEMON_PROC, DBUS_DAEMON_ARGS); } @@ -73,6 +74,8 @@ public: return mTransport.acquireAddress(); } private: + ScopedGlibLoop mLoop; + ScopedDir mZonesPathGuard; ZoneConnectionTransport mTransport; ScopedDaemon mDaemon; }; @@ -126,22 +129,17 @@ private: } // namespace +BOOST_FIXTURE_TEST_SUITE(ZoneConnectionSuite, Fixture) BOOST_AUTO_TEST_CASE(ConstructorDestructorConnectTest) { - ScopedGlibLoop loop; - ScopedDbusDaemon dbus; - - ZoneConnection(dbus.acquireAddress(), nullptr); + ZoneConnection(acquireAddress(), nullptr); } BOOST_AUTO_TEST_CASE(NotifyActiveZoneApiTest) { - ScopedGlibLoop loop; - ScopedDbusDaemon dbus; - Latch notifyCalled; - ZoneConnection connection(dbus.acquireAddress(), nullptr); + ZoneConnection connection(acquireAddress(), nullptr); auto callback = [&](const std::string& application, const std::string& message) { if (application == "testapp" && message == "testmessage") { @@ -150,7 +148,7 @@ BOOST_AUTO_TEST_CASE(NotifyActiveZoneApiTest) }; connection.setNotifyActiveZoneCallback(callback); - DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); + DbusConnection::Pointer client = DbusConnection::create(acquireAddress()); client->callMethod(api::zone::BUS_NAME, api::zone::OBJECT_PATH, api::zone::INTERFACE, @@ -162,13 +160,10 @@ BOOST_AUTO_TEST_CASE(NotifyActiveZoneApiTest) BOOST_AUTO_TEST_CASE(SignalNotificationApiTest) { - ScopedGlibLoop loop; - ScopedDbusDaemon dbus; - Latch signalEmitted; - ZoneConnection connection(dbus.acquireAddress(), nullptr); + ZoneConnection connection(acquireAddress(), nullptr); - DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); + DbusConnection::Pointer client = DbusConnection::create(acquireAddress()); auto handler = [&](const std::string& /*senderBusName*/, const std::string& objectPath, @@ -200,13 +195,10 @@ BOOST_AUTO_TEST_CASE(SignalNotificationApiTest) BOOST_AUTO_TEST_CASE(SignalDisplayOffApiTest) { - ScopedGlibLoop loop; - ScopedDbusDaemon dbus; - Latch displayOffCalled; - ZoneConnection connection(dbus.acquireAddress(), nullptr); + ZoneConnection connection(acquireAddress(), nullptr); - DbusConnection::Pointer client = DbusConnection::create(dbus.acquireAddress()); + DbusConnection::Pointer client = DbusConnection::create(acquireAddress()); auto callback = [&]() { displayOffCalled.set(); diff --git a/tests/unit_tests/utils/ut-fs.cpp b/tests/unit_tests/utils/ut-fs.cpp index 365c1e1..b2defba 100644 --- a/tests/unit_tests/utils/ut-fs.cpp +++ b/tests/unit_tests/utils/ut-fs.cpp @@ -28,54 +28,53 @@ #include "utils/fs.hpp" #include "utils/exception.hpp" +#include "utils/scoped-dir.hpp" #include #include #include -BOOST_AUTO_TEST_SUITE(UtilsFSSuite) - using namespace vasum; using namespace vasum::utils; namespace { -const std::string FILE_PATH = VSM_TEST_CONFIG_INSTALL_DIR "/utils/ut-fs/file.txt"; -const std::string FILE_CONTENT = "File content\n" - "Line 1\n" - "Line 2\n"; +const std::string TEST_PATH = "/tmp/ut-fsutils"; +const std::string REFERENCE_FILE_PATH = VSM_TEST_CONFIG_INSTALL_DIR "/utils/ut-fs/file.txt"; +const std::string REFERENCE_FILE_CONTENT = "File content\n" + "Line 1\n" + "Line 2\n"; const std::string FILE_CONTENT_2 = "Some other content\n" "Just to see if\n" "everything is copied correctly\n"; const std::string FILE_CONTENT_3 = "More content\n" "More and more content\n" "That's a lot of data to test\n"; -const std::string BUGGY_FILE_PATH = "/some/missing/file/path/file.txt"; -const std::string TMP_PATH = "/tmp"; -const std::string FILE_PATH_RANDOM = - boost::filesystem::unique_path("/tmp/testFile-%%%%").string(); -const std::string MOUNT_POINT_RANDOM_1 = - boost::filesystem::unique_path("/tmp/mountPoint-%%%%").string(); -const std::string MOUNT_POINT_RANDOM_2 = - boost::filesystem::unique_path("/tmp/mountPoint-%%%%").string(); -const std::string FILE_DIR_RANDOM_1 = - boost::filesystem::unique_path("testDir-%%%%").string(); -const std::string FILE_DIR_RANDOM_2 = - boost::filesystem::unique_path("testDir-%%%%").string(); -const std::string FILE_DIR_RANDOM_3 = - boost::filesystem::unique_path("testDir-%%%%").string(); -const std::string FILE_DIR_RANDOM_4 = - boost::filesystem::unique_path("testDir-%%%%").string(); -const std::string FILE_NAME_RANDOM_1 = - boost::filesystem::unique_path("testFile-%%%%").string(); -const std::string FILE_NAME_RANDOM_2 = - boost::filesystem::unique_path("testFile-%%%%").string(); +const std::string BUGGY_FILE_PATH = TEST_PATH + "/missing/file.txt"; +const std::string FILE_PATH = TEST_PATH + "/testFile"; +const std::string MOUNT_POINT_1 = TEST_PATH + "/mountPoint-1"; +const std::string MOUNT_POINT_2 = TEST_PATH + "/mountPoint-2"; +const std::string FILE_DIR_1 = "testDir-1"; +const std::string FILE_DIR_2 = "testDir-2"; +const std::string FILE_DIR_3 = "testDir-3"; +const std::string FILE_DIR_4 = "testDir-4"; +const std::string FILE_NAME_1 = "testFile-1"; +const std::string FILE_NAME_2 = "testFile-2"; + +struct Fixture { + utils::ScopedDir mTestPathGuard; + Fixture() + : mTestPathGuard(TEST_PATH) + {} +}; } // namespace +BOOST_FIXTURE_TEST_SUITE(UtilsFSSuite, Fixture) + BOOST_AUTO_TEST_CASE(ReadFileContentTest) { - BOOST_CHECK_EQUAL(FILE_CONTENT, readFileContent(FILE_PATH)); + BOOST_CHECK_EQUAL(REFERENCE_FILE_CONTENT, readFileContent(REFERENCE_FILE_PATH)); BOOST_CHECK_EXCEPTION(readFileContent(BUGGY_FILE_PATH), UtilsException, WhatEquals("Read failed")); @@ -83,18 +82,15 @@ BOOST_AUTO_TEST_CASE(ReadFileContentTest) BOOST_AUTO_TEST_CASE(SaveFileContentTest) { - BOOST_REQUIRE(saveFileContent(FILE_PATH_RANDOM, FILE_CONTENT)); - BOOST_CHECK_EQUAL(FILE_CONTENT, readFileContent(FILE_PATH)); - - boost::system::error_code ec; - boost::filesystem::remove(FILE_PATH_RANDOM, ec); + BOOST_REQUIRE(saveFileContent(FILE_PATH, REFERENCE_FILE_CONTENT)); + BOOST_CHECK_EQUAL(REFERENCE_FILE_CONTENT, readFileContent(FILE_PATH)); } BOOST_AUTO_TEST_CASE(RemoveFileTest) { - BOOST_REQUIRE(saveFileContent(FILE_PATH_RANDOM, FILE_CONTENT)); - BOOST_REQUIRE(removeFile(FILE_PATH_RANDOM)); - BOOST_REQUIRE(!boost::filesystem::exists(FILE_PATH_RANDOM)); + BOOST_REQUIRE(saveFileContent(FILE_PATH, REFERENCE_FILE_CONTENT)); + BOOST_REQUIRE(removeFile(FILE_PATH)); + BOOST_REQUIRE(!boost::filesystem::exists(FILE_PATH)); } BOOST_AUTO_TEST_CASE(MountPointTest) @@ -103,20 +99,20 @@ BOOST_AUTO_TEST_CASE(MountPointTest) namespace fs = boost::filesystem; boost::system::error_code ec; - BOOST_REQUIRE(fs::create_directory(MOUNT_POINT_RANDOM_1, ec)); - BOOST_REQUIRE(isMountPoint(MOUNT_POINT_RANDOM_1, result)); + BOOST_REQUIRE(fs::create_directory(MOUNT_POINT_1, ec)); + BOOST_REQUIRE(isMountPoint(MOUNT_POINT_1, result)); BOOST_CHECK_EQUAL(result, false); - BOOST_REQUIRE(hasSameMountPoint(TMP_PATH, MOUNT_POINT_RANDOM_1, result)); + BOOST_REQUIRE(hasSameMountPoint(TEST_PATH, MOUNT_POINT_1, result)); BOOST_CHECK_EQUAL(result, true); - BOOST_REQUIRE(mountRun(MOUNT_POINT_RANDOM_1)); - BOOST_REQUIRE(isMountPoint(MOUNT_POINT_RANDOM_1, result)); + BOOST_REQUIRE(mountRun(MOUNT_POINT_1)); + BOOST_REQUIRE(isMountPoint(MOUNT_POINT_1, result)); BOOST_CHECK_EQUAL(result, true); - BOOST_REQUIRE(hasSameMountPoint(TMP_PATH, MOUNT_POINT_RANDOM_1, result)); + BOOST_REQUIRE(hasSameMountPoint(TEST_PATH, MOUNT_POINT_1, result)); BOOST_CHECK_EQUAL(result, false); - BOOST_REQUIRE(umount(MOUNT_POINT_RANDOM_1)); - BOOST_REQUIRE(fs::remove(MOUNT_POINT_RANDOM_1, ec)); + BOOST_REQUIRE(umount(MOUNT_POINT_1)); + BOOST_REQUIRE(fs::remove(MOUNT_POINT_1, ec)); } BOOST_AUTO_TEST_CASE(MoveFileTest) @@ -126,32 +122,32 @@ BOOST_AUTO_TEST_CASE(MoveFileTest) std::string src, dst; // same mount point - src = TMP_PATH + "/" + FILE_NAME_RANDOM_1; - dst = TMP_PATH + "/" + FILE_NAME_RANDOM_2; + src = TEST_PATH + "/" + FILE_NAME_1; + dst = TEST_PATH + "/" + FILE_NAME_2; - BOOST_REQUIRE(saveFileContent(src, FILE_CONTENT)); + BOOST_REQUIRE(saveFileContent(src, REFERENCE_FILE_CONTENT)); BOOST_CHECK(moveFile(src, dst)); BOOST_CHECK(!fs::exists(src)); - BOOST_CHECK_EQUAL(readFileContent(dst), FILE_CONTENT); + BOOST_CHECK_EQUAL(readFileContent(dst), REFERENCE_FILE_CONTENT); BOOST_REQUIRE(fs::remove(dst)); // different mount point - src = TMP_PATH + "/" + FILE_NAME_RANDOM_1; - dst = MOUNT_POINT_RANDOM_2 + "/" + FILE_NAME_RANDOM_2; + src = TEST_PATH + "/" + FILE_NAME_1; + dst = MOUNT_POINT_2 + "/" + FILE_NAME_2; - BOOST_REQUIRE(fs::create_directory(MOUNT_POINT_RANDOM_2, ec)); - BOOST_REQUIRE(mountRun(MOUNT_POINT_RANDOM_2)); - BOOST_REQUIRE(saveFileContent(src, FILE_CONTENT)); + BOOST_REQUIRE(fs::create_directory(MOUNT_POINT_2, ec)); + BOOST_REQUIRE(mountRun(MOUNT_POINT_2)); + BOOST_REQUIRE(saveFileContent(src, REFERENCE_FILE_CONTENT)); BOOST_CHECK(moveFile(src, dst)); BOOST_CHECK(!fs::exists(src)); - BOOST_CHECK_EQUAL(readFileContent(dst), FILE_CONTENT); + BOOST_CHECK_EQUAL(readFileContent(dst), REFERENCE_FILE_CONTENT); BOOST_REQUIRE(fs::remove(dst)); - BOOST_REQUIRE(umount(MOUNT_POINT_RANDOM_2)); - BOOST_REQUIRE(fs::remove(MOUNT_POINT_RANDOM_2, ec)); + BOOST_REQUIRE(umount(MOUNT_POINT_2)); + BOOST_REQUIRE(fs::remove(MOUNT_POINT_2, ec)); } BOOST_AUTO_TEST_CASE(CopyDirContentsTest) @@ -160,24 +156,24 @@ BOOST_AUTO_TEST_CASE(CopyDirContentsTest) std::string src, src_inner, src_inner2, dst, dst_inner, dst_inner2; boost::system::error_code ec; - src = TMP_PATH + "/" + FILE_DIR_RANDOM_1; - src_inner = src + "/" + FILE_DIR_RANDOM_3; - src_inner2 = src + "/" + FILE_DIR_RANDOM_4; + src = TEST_PATH + "/" + FILE_DIR_1; + src_inner = src + "/" + FILE_DIR_3; + src_inner2 = src + "/" + FILE_DIR_4; - dst = TMP_PATH + "/" + FILE_DIR_RANDOM_2; - dst_inner = dst + "/" + FILE_DIR_RANDOM_3; - dst_inner2 = dst + "/" + FILE_DIR_RANDOM_4; + dst = TEST_PATH + "/" + FILE_DIR_2; + dst_inner = dst + "/" + FILE_DIR_3; + dst_inner2 = dst + "/" + FILE_DIR_4; // template dir structure: // |-src - // |-FILE_NAME_RANDOM_1 - // |-FILE_NAME_RANDOM_2 + // |-FILE_NAME_1 + // |-FILE_NAME_2 // |-src_inner (rw directory) - // | |-FILE_NAME_RANDOM_1 + // | |-FILE_NAME_1 // | // |-src_inner2 (ro directory) - // |-FILE_NAME_RANDOM_1 - // |-FILE_NAME_RANDOM_2 + // |-FILE_NAME_1 + // |-FILE_NAME_2 // create entire structure with files BOOST_REQUIRE(fs::create_directory(src, ec)); @@ -187,11 +183,11 @@ BOOST_AUTO_TEST_CASE(CopyDirContentsTest) BOOST_REQUIRE(fs::create_directory(src_inner2, ec)); BOOST_REQUIRE(ec.value() == 0); - BOOST_REQUIRE(saveFileContent(src + "/" + FILE_NAME_RANDOM_1, FILE_CONTENT)); - BOOST_REQUIRE(saveFileContent(src + "/" + FILE_NAME_RANDOM_2, FILE_CONTENT_2)); - BOOST_REQUIRE(saveFileContent(src_inner + "/" + FILE_NAME_RANDOM_1, FILE_CONTENT_3)); - BOOST_REQUIRE(saveFileContent(src_inner2 + "/" + FILE_NAME_RANDOM_1, FILE_CONTENT_3)); - BOOST_REQUIRE(saveFileContent(src_inner2 + "/" + FILE_NAME_RANDOM_2, FILE_CONTENT_2)); + BOOST_REQUIRE(saveFileContent(src + "/" + FILE_NAME_1, REFERENCE_FILE_CONTENT)); + BOOST_REQUIRE(saveFileContent(src + "/" + FILE_NAME_2, FILE_CONTENT_2)); + BOOST_REQUIRE(saveFileContent(src_inner + "/" + FILE_NAME_1, FILE_CONTENT_3)); + BOOST_REQUIRE(saveFileContent(src_inner2 + "/" + FILE_NAME_1, FILE_CONTENT_3)); + BOOST_REQUIRE(saveFileContent(src_inner2 + "/" + FILE_NAME_2, FILE_CONTENT_2)); // change permissions of src_inner2 directory fs::permissions(src_inner2, fs::owner_read, ec); @@ -205,19 +201,19 @@ BOOST_AUTO_TEST_CASE(CopyDirContentsTest) BOOST_CHECK(copyDirContents(src, dst)); // check if copy is successful - BOOST_CHECK(fs::exists(dst + "/" + FILE_NAME_RANDOM_1)); - BOOST_CHECK(fs::exists(dst + "/" + FILE_NAME_RANDOM_2)); + BOOST_CHECK(fs::exists(dst + "/" + FILE_NAME_1)); + BOOST_CHECK(fs::exists(dst + "/" + FILE_NAME_2)); BOOST_CHECK(fs::exists(dst_inner)); - BOOST_CHECK(fs::exists(dst_inner + "/" + FILE_NAME_RANDOM_1)); + BOOST_CHECK(fs::exists(dst_inner + "/" + FILE_NAME_1)); BOOST_CHECK(fs::exists(dst_inner2)); - BOOST_CHECK(fs::exists(dst_inner2 + "/" + FILE_NAME_RANDOM_1)); - BOOST_CHECK(fs::exists(dst_inner2 + "/" + FILE_NAME_RANDOM_2)); - - BOOST_CHECK_EQUAL(readFileContent(dst + "/" + FILE_NAME_RANDOM_1), FILE_CONTENT); - BOOST_CHECK_EQUAL(readFileContent(dst + "/" + FILE_NAME_RANDOM_2), FILE_CONTENT_2); - BOOST_CHECK_EQUAL(readFileContent(dst_inner + "/" + FILE_NAME_RANDOM_1), FILE_CONTENT_3); - BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_RANDOM_1), FILE_CONTENT_3); - BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_RANDOM_2), FILE_CONTENT_2); + BOOST_CHECK(fs::exists(dst_inner2 + "/" + FILE_NAME_1)); + BOOST_CHECK(fs::exists(dst_inner2 + "/" + FILE_NAME_2)); + + BOOST_CHECK_EQUAL(readFileContent(dst + "/" + FILE_NAME_1), REFERENCE_FILE_CONTENT); + BOOST_CHECK_EQUAL(readFileContent(dst + "/" + FILE_NAME_2), FILE_CONTENT_2); + BOOST_CHECK_EQUAL(readFileContent(dst_inner + "/" + FILE_NAME_1), FILE_CONTENT_3); + BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_1), FILE_CONTENT_3); + BOOST_CHECK_EQUAL(readFileContent(dst_inner2 + "/" + FILE_NAME_2), FILE_CONTENT_2); fs::file_status st = fs::status(fs::path(dst_inner2)); BOOST_CHECK(fs::owner_read == st.permissions()); -- 2.7.4 From 61a908d7f44dca36a5e6759e786c2d86abec33c1 Mon Sep 17 00:00:00 2001 From: Lukasz Kostyra Date: Thu, 5 Feb 2015 08:57:28 +0100 Subject: [PATCH 05/16] Add systemd socket support [Feature] Systemd unit files for Vasum socket [Cause] IPC works with Vasum socket [Solution] Add required systemd unit files [Verification] Build, install, run tests Change-Id: I552010681204a61e873d419b430bab8556080345 --- CMakeLists.txt | 1 - common/ipc/internals/utils.cpp | 10 ++-- packaging/vasum.spec | 16 ++++++ server/configs/CMakeLists.txt | 4 +- server/configs/systemd/vasum.service.in | 1 + server/configs/systemd/vasum.socket | 9 +++ tests/CMakeLists.txt | 4 +- tests/unit_tests/CMakeLists.txt | 20 ++++++- tests/unit_tests/ipc/ut-socket.cpp | 51 +++++++++++++++++ .../socket_test_service/configs/CMakeLists.txt | 26 +++++++++ .../configs/systemd/vasum-socket-test.service | 10 ++++ .../configs/systemd/vasum-socket-test.socket | 9 +++ .../unit_tests/socket_test_service/socket-test.cpp | 66 ++++++++++++++++++++++ .../unit_tests/socket_test_service/socket-test.hpp | 39 +++++++++++++ 14 files changed, 257 insertions(+), 9 deletions(-) create mode 100644 server/configs/systemd/vasum.socket create mode 100644 tests/unit_tests/ipc/ut-socket.cpp create mode 100644 tests/unit_tests/socket_test_service/configs/CMakeLists.txt create mode 100644 tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.service create mode 100644 tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.socket create mode 100644 tests/unit_tests/socket_test_service/socket-test.cpp create mode 100644 tests/unit_tests/socket_test_service/socket-test.hpp diff --git a/CMakeLists.txt b/CMakeLists.txt index 06a1064..67702a0 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -126,7 +126,6 @@ SET(SERVER_FOLDER ${PROJECT_SOURCE_DIR}/server) SET(ZONE_SUPPORT_FOLDER ${PROJECT_SOURCE_DIR}/zone-support) SET(ZONE_DAEMON_FOLDER ${PROJECT_SOURCE_DIR}/zone-daemon) SET(TESTS_FOLDER ${PROJECT_SOURCE_DIR}/tests) -SET(UNIT_TESTS_FOLDER ${TESTS_FOLDER}/unit_tests) SET(CLI_FOLDER ${PROJECT_SOURCE_DIR}/cli) IF(NOT DEFINED SYSCONF_INSTALL_DIR) diff --git a/common/ipc/internals/utils.cpp b/common/ipc/internals/utils.cpp index bd98c1b..cb9c93c 100644 --- a/common/ipc/internals/utils.cpp +++ b/common/ipc/internals/utils.cpp @@ -75,13 +75,15 @@ void waitForEvent(int fd, throw IPCException("Timeout in read"); } + if (fds[0].revents & event) { + // Here Comes the Sun + break; + } + if (fds[0].revents & POLLHUP) { - LOGE("Peer disconnected"); + LOGW("Peer disconnected"); throw IPCException("Peer disconnected"); } - - // Here Comes the Sun - break; } } diff --git a/packaging/vasum.spec b/packaging/vasum.spec index 113348b..db8d6be 100644 --- a/packaging/vasum.spec +++ b/packaging/vasum.spec @@ -48,6 +48,7 @@ between them. A process from inside a zone can request a switch of context %attr(755,root,root) /etc/vasum/lxc-templates/*.sh %config /etc/vasum/templates/*.conf %{_unitdir}/vasum.service +%{_unitdir}/vasum.socket %{_unitdir}/multi-user.target.wants/vasum.service /etc/dbus-1/system.d/org.tizen.vasum.host.conf %dir %{_datadir}/.zones @@ -211,10 +212,23 @@ Requires: boost-test %description tests Unit tests for both: server and client and integration tests. +%post tests +systemctl daemon-reload +systemctl enable vasum-socket-test.socket +systemctl start vasum-socket-test.socket + +%preun tests +systemctl stop vasum-socket-test.socket +systemctl disable vasum-socket-test.socket + +%postun tests +systemctl daemon-reload + %files tests %manifest packaging/vasum-server-tests.manifest %defattr(644,root,root,755) %attr(755,root,root) %{_bindir}/vasum-server-unit-tests +%attr(755,root,root) %{_bindir}/vasum-socket-test %attr(755,root,root) %{script_dir}/vsm_all_tests.py %attr(755,root,root) %{script_dir}/vsm_int_tests.py %attr(755,root,root) %{script_dir}/vsm_launch_test.py @@ -223,3 +237,5 @@ Unit tests for both: server and client and integration tests. %attr(755,root,root) %{_datadir}/vasum/lxc-templates %{python_sitelib}/vsm_integration_tests /etc/dbus-1/system.d/org.tizen.vasum.tests.conf +%{_unitdir}/vasum-socket-test.socket +%{_unitdir}/vasum-socket-test.service diff --git a/server/configs/CMakeLists.txt b/server/configs/CMakeLists.txt index f5f82c5..e34137e 100644 --- a/server/configs/CMakeLists.txt +++ b/server/configs/CMakeLists.txt @@ -49,5 +49,7 @@ INSTALL(PROGRAMS ${admin_CONF} INSTALL(PROGRAMS ${template_CONF} DESTINATION ${VSM_CONFIG_INSTALL_DIR}/templates) -INSTALL(FILES ${CMAKE_BINARY_DIR}/systemd/vasum.service +INSTALL(FILES + ${CMAKE_BINARY_DIR}/systemd/vasum.service + systemd/vasum.socket DESTINATION ${SYSTEMD_UNIT_DIR}) diff --git a/server/configs/systemd/vasum.service.in b/server/configs/systemd/vasum.service.in index 8131ed5..29ebbc8 100644 --- a/server/configs/systemd/vasum.service.in +++ b/server/configs/systemd/vasum.service.in @@ -7,6 +7,7 @@ Type=simple ExecStart=${CMAKE_INSTALL_PREFIX}/bin/vasum-server -r Restart=on-failure ExecReload=/bin/kill -HUP $MAINPID +Sockets=vasum.socket [Install] WantedBy=multi-user.target diff --git a/server/configs/systemd/vasum.socket b/server/configs/systemd/vasum.socket new file mode 100644 index 0000000..792935d --- /dev/null +++ b/server/configs/systemd/vasum.socket @@ -0,0 +1,9 @@ +[Socket] +ListenStream=/run/vasum.socket +SocketMode=0755 +SmackLabelIPIn=* +SmackLabelIPOut=@ +Service=vasum.service + +[Install] +WantedBy=sockets.target diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 60214f9..7b7434b 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -17,7 +17,9 @@ # @author Lukasz Kostyra (l.kostyra@samsung.com) # +SET(UNIT_TESTS_FOLDER ${TESTS_FOLDER}/unit_tests) +SET(SOCKET_TEST_FOLDER ${UNIT_TESTS_FOLDER}/socket_test_service) + ADD_SUBDIRECTORY(scripts) ADD_SUBDIRECTORY(integration_tests) ADD_SUBDIRECTORY(unit_tests) - diff --git a/tests/unit_tests/CMakeLists.txt b/tests/unit_tests/CMakeLists.txt index 96ab943..fc96586 100644 --- a/tests/unit_tests/CMakeLists.txt +++ b/tests/unit_tests/CMakeLists.txt @@ -23,25 +23,34 @@ FILE(GLOB_RECURSE project_SRCS *.cpp *.hpp) FILE(GLOB_RECURSE common_SRCS ${COMMON_FOLDER}/*.cpp ${COMMON_FOLDER}/*.hpp) FILE(GLOB server_SRCS ${SERVER_FOLDER}/*.cpp ${SERVER_FOLDER}/*.hpp) FILE(GLOB client_SRCS ${CLIENT_FOLDER}/*.cpp ${CLIENT_FOLDER}/*.h) +FILE(GLOB socket_test_SRCS ${SOCKET_TEST_FOLDER}/*.cpp ${SOCKET_TEST_FOLDER}/*.hpp) FILE(GLOB main_SRC ${SERVER_FOLDER}/main.cpp) LIST(REMOVE_ITEM server_SRCS ${main_SRC}) +# We must compile socket-test separately, exclude it from unit-test build +LIST(REMOVE_ITEM project_SRCS ${socket_test_SRCS}) + ## Setup target ################################################################ SET(UT_SERVER_CODENAME "${PROJECT_NAME}-server-unit-tests") +SET(SOCKET_TEST_CODENAME "${PROJECT_NAME}-socket-test") ADD_EXECUTABLE(${UT_SERVER_CODENAME} ${project_SRCS} ${common_SRCS} ${server_SRCS} ${client_SRCS}) ## A fake target to test vasum-client C API ADD_EXECUTABLE("vasum-client-c-api-compile-test" client/client-c-api-compile-test.c) +## A stub mini-service to test socket functionality +ADD_EXECUTABLE(${SOCKET_TEST_CODENAME} ${socket_test_SRCS} ${common_SRCS}) + ## Link libraries ############################################################## FIND_PACKAGE (Boost COMPONENTS unit_test_framework system filesystem regex) PKG_CHECK_MODULES(UT_SERVER_DEPS REQUIRED lxc json gio-2.0 libsystemd-daemon libsystemd-journal libcap-ng libLogger libSimpleDbus libConfig) -INCLUDE_DIRECTORIES(${COMMON_FOLDER} ${SERVER_FOLDER} ${UNIT_TESTS_FOLDER} ${CLIENT_FOLDER}) +INCLUDE_DIRECTORIES(${COMMON_FOLDER} ${SERVER_FOLDER} ${UNIT_TESTS_FOLDER} ${CLIENT_FOLDER} + ${SOCKET_TEST_FOLDER}) INCLUDE_DIRECTORIES(SYSTEM ${UT_SERVER_DEPS_INCLUDE_DIRS} ${Boost_INCLUDE_DIRS}) SET_TARGET_PROPERTIES(${UT_SERVER_CODENAME} PROPERTIES @@ -49,8 +58,13 @@ SET_TARGET_PROPERTIES(${UT_SERVER_CODENAME} PROPERTIES LINK_FLAGS "-pthread" ) -TARGET_LINK_LIBRARIES(${UT_SERVER_CODENAME} ${UT_SERVER_DEPS_LIBRARIES} ${Boost_LIBRARIES}) +SET_TARGET_PROPERTIES(${SOCKET_TEST_CODENAME} PROPERTIES + COMPILE_FLAGS "-pthread" + LINK_FLAGS "-pthread" +) +TARGET_LINK_LIBRARIES(${UT_SERVER_CODENAME} ${UT_SERVER_DEPS_LIBRARIES} ${Boost_LIBRARIES}) +TARGET_LINK_LIBRARIES(${SOCKET_TEST_CODENAME} ${UT_SERVER_DEPS_LIBRARIES} ${Boost_LIBRARIES}) ## Subdirectories ############################################################## SET(VSM_TEST_CONFIG_INSTALL_DIR ${VSM_DATA_INSTALL_DIR}/tests) @@ -63,7 +77,9 @@ ADD_SUBDIRECTORY(server/configs) ADD_SUBDIRECTORY(utils/configs) ADD_SUBDIRECTORY(client/configs) ADD_SUBDIRECTORY(lxc/templates) +ADD_SUBDIRECTORY(socket_test_service/configs) ## Install ##################################################################### INSTALL(TARGETS ${UT_SERVER_CODENAME} DESTINATION bin) +INSTALL(TARGETS ${SOCKET_TEST_CODENAME} DESTINATION bin) diff --git a/tests/unit_tests/ipc/ut-socket.cpp b/tests/unit_tests/ipc/ut-socket.cpp new file mode 100644 index 0000000..6f78f90 --- /dev/null +++ b/tests/unit_tests/ipc/ut-socket.cpp @@ -0,0 +1,51 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Lukasz Kostyra + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Lukasz Kostyra (l.kostyra@samsung.com) + * @brief Socket IPC module tests + */ + +#include "config.hpp" +#include "ut.hpp" + +#include "ipc/internals/socket.hpp" + +#include + +using namespace vasum::ipc; + +BOOST_AUTO_TEST_SUITE(SocketSuite) + +BOOST_AUTO_TEST_CASE(SystemdSocket) +{ + std::string readMessage; + + { + Socket socket = Socket::connectSocket(vasum::socket_test::SOCKET_PATH); + BOOST_REQUIRE_GT(socket.getFD(), -1); + + readMessage.resize(vasum::socket_test::TEST_MESSAGE.size()); + socket.read(&readMessage.front(), readMessage.size()); + } + + BOOST_REQUIRE_EQUAL(readMessage, vasum::socket_test::TEST_MESSAGE); +} + +BOOST_AUTO_TEST_SUITE_END() diff --git a/tests/unit_tests/socket_test_service/configs/CMakeLists.txt b/tests/unit_tests/socket_test_service/configs/CMakeLists.txt new file mode 100644 index 0000000..17e79b9 --- /dev/null +++ b/tests/unit_tests/socket_test_service/configs/CMakeLists.txt @@ -0,0 +1,26 @@ +# Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# +# @file CMakeLists.txt +# @author Lukasz Kostyra (l.kostyra@samsung.com) +# + +MESSAGE(STATUS "Installing configs for Vasum sycket test service to " ${VSM_TEST_CONFIG_INSTALL_DIR}) + +## Install ##################################################################### +INSTALL(FILES systemd/vasum-socket-test.socket + DESTINATION ${SYSTEMD_UNIT_DIR}) +INSTALL(FILES systemd/vasum-socket-test.service + DESTINATION ${SYSTEMD_UNIT_DIR}) diff --git a/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.service b/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.service new file mode 100644 index 0000000..1d43529 --- /dev/null +++ b/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.service @@ -0,0 +1,10 @@ +[Unit] +Description=Vasum Socket tests mini-service +ConditionVirtualization=no + +[Service] +Type=simple +ExecStart=/usr/bin/vasum-socket-test +Sockets=vasum-socket-test.socket +StartLimitInterval=0 +StartLimitBurst=0 diff --git a/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.socket b/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.socket new file mode 100644 index 0000000..bcaeaa8 --- /dev/null +++ b/tests/unit_tests/socket_test_service/configs/systemd/vasum-socket-test.socket @@ -0,0 +1,9 @@ +[Socket] +ListenStream=/run/vasum-socket-test.socket +SocketMode=0755 +SmackLabelIPIn=* +SmackLabelIPOut=@ +Service=vasum-socket-test.service + +[Install] +WantedBy=sockets.target diff --git a/tests/unit_tests/socket_test_service/socket-test.cpp b/tests/unit_tests/socket_test_service/socket-test.cpp new file mode 100644 index 0000000..609bf87 --- /dev/null +++ b/tests/unit_tests/socket_test_service/socket-test.cpp @@ -0,0 +1,66 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Lukasz Kostyra + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Lukasz Kostyra (l.kostyra@samsung.com) + * @brief Mini-service for IPC Socket mechanism tests + */ + +#include +#include "socket-test.hpp" + +#include +#include +#include + +#include +#include + +#include +#include + +using namespace vasum::socket_test; +using namespace vasum::ipc; +using namespace logger; + +// NOTE this is a single-usage program, only meant to test vasum::ipc::Socket module. +// It's purpose is to be activated when needed by systemd socket activation mechanism. +int main() +{ + Logger::setLogLevel(LogLevel::TRACE); + Logger::setLogBackend(new SystemdJournalBackend()); + + try { + Socket listeningSocket(Socket::createSocket(SOCKET_PATH)); + if (listeningSocket.getFD() < 0) { + LOGE("Failed to connect to socket!"); + return 1; + } + + std::shared_ptr clientSocket = listeningSocket.accept(); + LOGI("Connected! Emitting message to client."); + clientSocket->write(TEST_MESSAGE.c_str(), TEST_MESSAGE.size()); + LOGI("Message sent through socket! Exiting."); + } catch (const IPCException& e) { + LOGE("IPC exception caught! " << e.what()); + return 1; + } + + return 0; +} diff --git a/tests/unit_tests/socket_test_service/socket-test.hpp b/tests/unit_tests/socket_test_service/socket-test.hpp new file mode 100644 index 0000000..f7a292e --- /dev/null +++ b/tests/unit_tests/socket_test_service/socket-test.hpp @@ -0,0 +1,39 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Lukasz Kostyra + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Lukasz Kostyra (l.kostyra@samsung.com) + * @brief Mini-service for IPC Socket mechanism tests + */ + +#ifndef TESTS_SOCKET_TEST_SERVICE_HPP +#define TESTS_SOCKET_TEST_SERVICE_HPP + +#include + +namespace vasum { +namespace socket_test { + +const std::string SOCKET_PATH = "/run/vasum-socket-test.socket"; +const std::string TEST_MESSAGE = "Some great messages, ey!"; + +} // namespace socket_test +} // namespace vasum + +#endif // TESTS_SOCKET_TEST_SERVICE_HPP -- 2.7.4 From 3a22b40dee63c508cd6d98911e5b6f1771cc10c2 Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Thu, 26 Feb 2015 16:30:11 +0100 Subject: [PATCH 06/16] Enable cpuQuota options, fix cpuQuota logic [Bug/Feature] Options were disabled after migration from libvirt [Cause] N/A [Solution] N/A [Verification] Run tests, run two containers use 100% cpu on first and switch to second and check total cpu usage in host. Change-Id: I6c79f639f7cf0def3ad0ea73550940d8f5edba9e --- common/lxc/cgroup.cpp | 26 ++++++++++- common/lxc/cgroup.hpp | 6 +++ common/utils/fs.cpp | 16 +++++++ common/utils/fs.hpp | 6 +++ server/zone-admin.cpp | 75 ++++++++++--------------------- server/zone.cpp | 12 +++-- server/zones-manager.cpp | 17 ++++--- tests/unit_tests/server/ut-zone-admin.cpp | 26 ++++++----- 8 files changed, 109 insertions(+), 75 deletions(-) diff --git a/common/lxc/cgroup.cpp b/common/lxc/cgroup.cpp index 62f3184..0ec6da2 100644 --- a/common/lxc/cgroup.cpp +++ b/common/lxc/cgroup.cpp @@ -27,6 +27,7 @@ #include "lxc/cgroup.hpp" #include "logger/logger.hpp" #include "utils/fs.hpp" +#include "utils/paths.hpp" #include #include @@ -54,17 +55,38 @@ std::string flagsToPermissions(bool grant, uint32_t flags) } } +std::string getCgroupPath(const std::string& zoneName, + const std::string& cgroupController, + const std::string& cgroupName) +{ + return utils::createFilePath("/sys/fs/cgroup", + cgroupController, + "lxc", + zoneName, + cgroupName); +} + } // namespace bool setCgroup(const std::string& zoneName, + const std::string& cgroupController, const std::string& cgroupName, const std::string& value) { - const std::string path = "/sys/fs/cgroup/devices/lxc/" + zoneName + "/" + cgroupName; + const std::string path = getCgroupPath(zoneName, cgroupController, cgroupName); LOGD("Set '" << value << "' to " << path); return utils::saveFileContent(path, value); } +bool getCgroup(const std::string& zoneName, + const std::string& cgroupController, + const std::string& cgroupName, + std::string& value) +{ + const std::string path = getCgroupPath(zoneName, cgroupController, cgroupName); + return utils::readFirstLineOfFile(path, value); +} + bool isDevice(const std::string& device) { struct stat devStat; @@ -113,7 +135,7 @@ bool setDeviceAccess(const std::string& zoneName, snprintf(value, sizeof(value), "%c %u:%u %s", type, major, minor, perm.c_str()); std::string name = grant ? "devices.allow" : "devices.deny"; - return setCgroup(zoneName, name, value); + return setCgroup(zoneName, "devices", name, value); } diff --git a/common/lxc/cgroup.hpp b/common/lxc/cgroup.hpp index 2592054..23f0e9b 100644 --- a/common/lxc/cgroup.hpp +++ b/common/lxc/cgroup.hpp @@ -31,9 +31,15 @@ namespace vasum { namespace lxc { bool setCgroup(const std::string& zoneName, + const std::string& cgroupController, const std::string& cgroupName, const std::string& value); +bool getCgroup(const std::string& zoneName, + const std::string& cgroupController, + const std::string& cgroupName, + std::string& value); + bool isDevice(const std::string& device); bool setDeviceAccess(const std::string& zoneName, diff --git a/common/utils/fs.cpp b/common/utils/fs.cpp index d7ae2dd..e3a59b3 100644 --- a/common/utils/fs.cpp +++ b/common/utils/fs.cpp @@ -101,6 +101,22 @@ bool saveFileContent(const std::string& path, const std::string& content) return true; } +bool readFirstLineOfFile(const std::string& path, std::string& ret) +{ + std::ifstream file(path); + if (!file) { + LOGD(path << ": could not open for reading"); + return false; + } + + std::getline(file, ret); + if (!file) { + LOGD(path << ": read error"); + return false; + } + return true; +} + bool removeFile(const std::string& path) { LOGD(path << ": exists, removing."); diff --git a/common/utils/fs.hpp b/common/utils/fs.hpp index b00f95a..b2e05b6 100644 --- a/common/utils/fs.hpp +++ b/common/utils/fs.hpp @@ -50,6 +50,12 @@ bool readFileContent(const std::string& path, std::string& content); bool saveFileContent(const std::string& path, const std::string& content); /** + * Read a line from file + * Its goal is to read a kernel config files (eg. from /proc, /sys/) + */ +bool readFirstLineOfFile(const std::string& path, std::string& ret); + +/** * Remove file */ bool removeFile(const std::string& path); diff --git a/server/zone-admin.cpp b/server/zone-admin.cpp index fa465bc..c5e994d 100644 --- a/server/zone-admin.cpp +++ b/server/zone-admin.cpp @@ -30,8 +30,10 @@ #include "logger/logger.hpp" #include "utils/paths.hpp" #include "utils/c-array.hpp" +#include "lxc/cgroup.hpp" #include +#include namespace vasum { @@ -210,6 +212,8 @@ bool ZoneAdmin::isPaused() void ZoneAdmin::setSchedulerLevel(SchedulerLevel sched) { + assert(isRunning()); + switch (sched) { case SchedulerLevel::FOREGROUND: LOGD(mId << ": Setting SchedulerLevel::FOREGROUND"); @@ -229,27 +233,21 @@ void ZoneAdmin::setSchedulerLevel(SchedulerLevel sched) } -void ZoneAdmin::setSchedulerParams(std::uint64_t, std::uint64_t, std::int64_t) -//void ZoneAdmin::setSchedulerParams(std::uint64_t cpuShares, std::uint64_t vcpuPeriod, std::int64_t vcpuQuota) +void ZoneAdmin::setSchedulerParams(std::uint64_t cpuShares, + std::uint64_t vcpuPeriod, + std::int64_t vcpuQuota) { -// assert(mZone); -// -// int maxParams = 3; -// int numParamsBuff = 0; -// -// std::unique_ptr params(new virTypedParameter[maxParams]); -// -// virTypedParameterPtr paramsTmp = params.get(); -// -// virTypedParamsAddULLong(¶msTmp, &numParamsBuff, &maxParams, VIR_DOMAIN_SCHEDULER_CPU_SHARES, cpuShares); -// virTypedParamsAddULLong(¶msTmp, &numParamsBuff, &maxParams, VIR_DOMAIN_SCHEDULER_VCPU_PERIOD, vcpuPeriod); -// virTypedParamsAddLLong(¶msTmp, &numParamsBuff, &maxParams, VIR_DOMAIN_SCHEDULER_VCPU_QUOTA, vcpuQuota); -// -// if (virDomainSetSchedulerParameters(mZone.get(), params.get(), numParamsBuff) < 0) { -// LOGE(mId << ": Error while setting the zone's scheduler params:\n" -// << libvirt::libvirtFormatError()); -// throw ZoneOperationException(); -// } + assert(vcpuPeriod >= 1000 && vcpuPeriod <= 1000000); + assert(vcpuQuota == -1 || + (vcpuQuota >= 1000 && vcpuQuota <= static_cast(ULLONG_MAX / 1000))); + + if (!lxc::setCgroup(mId, "cpu", "cpu.shares", std::to_string(cpuShares)) || + !lxc::setCgroup(mId, "cpu", "cpu.cfs_period_us", std::to_string(vcpuPeriod)) || + !lxc::setCgroup(mId, "cpu", "cpu.cfs_quota_us", std::to_string(vcpuQuota))) { + + LOGE(mId << ": Error while setting the zone's scheduler params"); + throw ZoneOperationException("Could not set scheduler params"); + } } void ZoneAdmin::setDetachOnExit() @@ -264,37 +262,12 @@ void ZoneAdmin::setDestroyOnExit() std::int64_t ZoneAdmin::getSchedulerQuota() { -// assert(mZone); -// -// int numParamsBuff; -// std::unique_ptr type(virDomainGetSchedulerType(mZone.get(), &numParamsBuff), free); -// -// if (type == NULL || numParamsBuff <= 0 || strcmp(type.get(), "posix") != 0) { -// LOGE(mId << ": Error while getting the zone's scheduler type:\n" -// << libvirt::libvirtFormatError()); -// throw ZoneOperationException(); -// } -// -// std::unique_ptr params(new virTypedParameter[numParamsBuff]); -// -// if (virDomainGetSchedulerParameters(mZone.get(), params.get(), &numParamsBuff) < 0) { -// LOGE(mId << ": Error while getting the zone's scheduler params:\n" -// << libvirt::libvirtFormatError()); -// throw ZoneOperationException(); -// } -// -// long long quota; -// if (virTypedParamsGetLLong(params.get(), -// numParamsBuff, -// VIR_DOMAIN_SCHEDULER_VCPU_QUOTA, -// "a) <= 0) { -// LOGE(mId << ": Error while getting the zone's scheduler quota param:\n" -// << libvirt::libvirtFormatError()); -// throw ZoneOperationException(); -// } -// -// return quota; - return 0; + std::string ret; + if (!lxc::getCgroup(mId, "cpu", "cpu.cfs_quota_us", ret)) { + LOGE(mId << ": Error while getting the zone's scheduler quota param"); + throw ZoneOperationException("Could not get scheduler quota param"); + } + return std::stoll(ret); } void ZoneAdmin::createNetdevVeth(const std::string& /* zoneDev */, diff --git a/server/zone.cpp b/server/zone.cpp index 5c15f21..0607fda 100644 --- a/server/zone.cpp +++ b/server/zone.cpp @@ -125,19 +125,23 @@ void Zone::start() if (mConfig.enableDbusIntegration) { mConnectionTransport.reset(new ZoneConnectionTransport(mRunMountPoint)); } + mAdmin->start(); if (mConfig.enableDbusIntegration) { + // Increase cpu quota before connect, otherwise it'd take ages. + goForeground(); connect(); } - - // Send to the background only after we're connected, otherwise it'd take ages. - LOGD(getId() << ": sending to the background"); - goBackground(); + // refocus in ZonesManager will adjust cpu quota after all } void Zone::stop() { Lock lock(mReconnectMutex); + if (mAdmin->isRunning()) { + // boost stopping + goForeground(); + } disconnect(); mAdmin->stop(); mConnectionTransport.reset(); diff --git a/server/zones-manager.cpp b/server/zones-manager.cpp index fd8501f..5f0dca4 100644 --- a/server/zones-manager.cpp +++ b/server/zones-manager.cpp @@ -392,13 +392,15 @@ void ZonesManager::focusInternal(Zones::iterator iter) } for (auto& zone : mZones) { - std::string id = zone->getId(); - if (id == idToFocus) { - LOGD(id << ": being sent to foreground"); - zone->goForeground(); - } else { - LOGD(id << ": being sent to background"); - zone->goBackground(); + if (zone->isRunning()) { + std::string id = zone->getId(); + if (id == idToFocus) { + LOGD(id << ": being sent to foreground"); + zone->goForeground(); + } else { + LOGD(id << ": being sent to background"); + zone->goBackground(); + } } } mActiveZoneId = idToFocus; @@ -1245,6 +1247,7 @@ void ZonesManager::handleLockZoneCall(const std::string& id, LOGT("Lock zone"); try { + zone.goBackground();// make sure it will be in background after unlock zone.suspend(); refocus(); } catch (ZoneOperationException& e) { diff --git a/tests/unit_tests/server/ut-zone-admin.cpp b/tests/unit_tests/server/ut-zone-admin.cpp index 255e75e..4dd08db 100644 --- a/tests/unit_tests/server/ut-zone-admin.cpp +++ b/tests/unit_tests/server/ut-zone-admin.cpp @@ -156,16 +156,20 @@ BOOST_AUTO_TEST_CASE(SuspendResumeTest) BOOST_CHECK(admin->isRunning()); } -//BOOST_AUTO_TEST_CASE(SchedulerLevelTest) -//{ -// auto admin = create(TEST_CONFIG_PATH); -// -// admin->start(); -// ensureStarted(); -// admin->setSchedulerLevel(SchedulerLevel::FOREGROUND); -// BOOST_REQUIRE(admin->getSchedulerQuota() == config.cpuQuotaForeground); -// admin->setSchedulerLevel(SchedulerLevel::BACKGROUND); -// BOOST_REQUIRE(admin->getSchedulerQuota() == config.cpuQuotaBackground); -//} +BOOST_AUTO_TEST_CASE(SchedulerLevelTest) +{ + BOOST_REQUIRE(mConfig.cpuQuotaForeground != mConfig.cpuQuotaBackground); + + auto admin = create(TEST_CONFIG_PATH); + + admin->start(); + ensureStarted(); + + admin->setSchedulerLevel(SchedulerLevel::FOREGROUND); + BOOST_CHECK_EQUAL(admin->getSchedulerQuota(), mConfig.cpuQuotaForeground); + + admin->setSchedulerLevel(SchedulerLevel::BACKGROUND); + BOOST_CHECK_EQUAL(admin->getSchedulerQuota(), mConfig.cpuQuotaBackground); +} BOOST_AUTO_TEST_SUITE_END() -- 2.7.4 From 633f086ac82e192354b061ffed1682f4fd1d2a1e Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Mon, 23 Feb 2015 18:03:32 +0100 Subject: [PATCH 07/16] Epoll wrapper, epoll dispatchers (glib and threaded) [Bug/Feature] N/A [Cause] N/A [Solution] Epoll is the only way, using no additional threads, to aggregate descriptors to one epoll descriptor. [Verification] Run tests Change-Id: I3863129a8b947c467615b2e9d352fce3bd1cda9a --- common/ipc/client.cpp | 2 +- common/ipc/internals/acceptor.cpp | 2 +- common/ipc/internals/event-queue.hpp | 4 +- common/ipc/internals/processor.cpp | 10 +- common/ipc/internals/request-queue.hpp | 4 +- common/ipc/internals/socket.cpp | 18 +- common/ipc/service.cpp | 2 +- common/utils/event-poll.cpp | 151 +++++++++++++++ common/utils/event-poll.hpp | 68 +++++++ common/{ipc/internals => utils}/eventfd.cpp | 26 ++- common/{ipc/internals => utils}/eventfd.hpp | 11 +- .../internals/utils.cpp => utils/fd-utils.cpp} | 76 ++++---- .../internals/utils.hpp => utils/fd-utils.hpp} | 13 +- common/utils/glib-poll-dispatcher.cpp | 61 ++++++ common/utils/glib-poll-dispatcher.hpp | 53 ++++++ common/utils/thread-poll-dispatcher.cpp | 53 ++++++ common/utils/thread-poll-dispatcher.hpp | 52 ++++++ tests/unit_tests/ipc/ut-ipc.cpp | 2 - tests/unit_tests/utils/ut-event-poll.cpp | 205 +++++++++++++++++++++ 19 files changed, 726 insertions(+), 87 deletions(-) create mode 100644 common/utils/event-poll.cpp create mode 100644 common/utils/event-poll.hpp rename common/{ipc/internals => utils}/eventfd.cpp (67%) rename common/{ipc/internals => utils}/eventfd.hpp (88%) rename common/{ipc/internals/utils.cpp => utils/fd-utils.cpp} (68%) rename common/{ipc/internals/utils.hpp => utils/fd-utils.hpp} (90%) create mode 100644 common/utils/glib-poll-dispatcher.cpp create mode 100644 common/utils/glib-poll-dispatcher.hpp create mode 100644 common/utils/thread-poll-dispatcher.cpp create mode 100644 common/utils/thread-poll-dispatcher.hpp create mode 100644 tests/unit_tests/utils/ut-event-poll.cpp diff --git a/common/ipc/client.cpp b/common/ipc/client.cpp index 51e59f8..6669d8a 100644 --- a/common/ipc/client.cpp +++ b/common/ipc/client.cpp @@ -45,7 +45,7 @@ Client::~Client() LOGS("Client Destructor"); try { stop(); - } catch (IPCException& e) { + } catch (std::exception& e) { LOGE("Error in Client's destructor: " << e.what()); } } diff --git a/common/ipc/internals/acceptor.cpp b/common/ipc/internals/acceptor.cpp index 627e1fe..ecb2210 100644 --- a/common/ipc/internals/acceptor.cpp +++ b/common/ipc/internals/acceptor.cpp @@ -49,7 +49,7 @@ Acceptor::~Acceptor() LOGT("Destroying Acceptor"); try { stop(); - } catch (IPCException& e) { + } catch (std::exception& e) { LOGE("Error in destructor: " << e.what()); } LOGT("Destroyed Acceptor"); diff --git a/common/ipc/internals/event-queue.hpp b/common/ipc/internals/event-queue.hpp index 2c591f7..b4532fd 100644 --- a/common/ipc/internals/event-queue.hpp +++ b/common/ipc/internals/event-queue.hpp @@ -26,7 +26,7 @@ #define COMMON_IPC_INTERNALS_EVENT_QUEUE_HPP #include "ipc/exception.hpp" -#include "ipc/internals/eventfd.hpp" +#include "utils/eventfd.hpp" #include "logger/logger.hpp" #include @@ -82,7 +82,7 @@ private: std::mutex mCommunicationMutex; std::queue mMessages; - EventFD mEventFD; + utils::EventFD mEventFD; }; template diff --git a/common/ipc/internals/processor.cpp b/common/ipc/internals/processor.cpp index 79c1a22..93f721f 100644 --- a/common/ipc/internals/processor.cpp +++ b/common/ipc/internals/processor.cpp @@ -26,8 +26,8 @@ #include "ipc/exception.hpp" #include "ipc/internals/processor.hpp" -#include "ipc/internals/utils.hpp" #include "utils/signal.hpp" +#include "utils/exception.hpp" #include #include @@ -79,21 +79,21 @@ Processor::~Processor() LOGS(mLogPrefix + "Processor Destructor"); try { stop(); - } catch (IPCException& e) { + } catch (std::exception& e) { LOGE(mLogPrefix + "Error in Processor's destructor: " << e.what()); } } Processor::Peers::iterator Processor::getPeerInfoIterator(const FileDescriptor fd) { - return std::find_if(mPeerInfo.begin(), mPeerInfo.end(), [&fd](const PeerInfo & peerInfo) { + return std::find_if(mPeerInfo.begin(), mPeerInfo.end(), [fd](const PeerInfo & peerInfo) { return fd == peerInfo.socketPtr->getFD(); }); } Processor::Peers::iterator Processor::getPeerInfoIterator(const PeerID peerID) { - return std::find_if(mPeerInfo.begin(), mPeerInfo.end(), [&peerID](const PeerInfo & peerInfo) { + return std::find_if(mPeerInfo.begin(), mPeerInfo.end(), [peerID](const PeerInfo & peerInfo) { return peerID == peerInfo.peerID; }); } @@ -411,7 +411,7 @@ bool Processor::handleInput(const FileDescriptor fd) Socket::Guard guard = socket.getGuard(); socket.read(&methodID, sizeof(methodID)); socket.read(&messageID, sizeof(messageID)); - } catch (const IPCException& e) { + } catch (const UtilsException& e) { LOGE(mLogPrefix + "Error during reading the socket"); removePeerInternal(peerIt, std::make_exception_ptr(IPCNaughtyPeerException())); diff --git a/common/ipc/internals/request-queue.hpp b/common/ipc/internals/request-queue.hpp index e1ad46b..44306ee 100644 --- a/common/ipc/internals/request-queue.hpp +++ b/common/ipc/internals/request-queue.hpp @@ -26,7 +26,7 @@ #define COMMON_IPC_INTERNALS_MESSAGE_QUEUE_HPP #include "ipc/exception.hpp" -#include "ipc/internals/eventfd.hpp" +#include "utils/eventfd.hpp" #include "logger/logger.hpp" #include @@ -115,7 +115,7 @@ private: std::list mRequests; std::mutex mStateMutex; - EventFD mEventFD; + utils::EventFD mEventFD; }; template diff --git a/common/ipc/internals/socket.cpp b/common/ipc/internals/socket.cpp index 2469af6..a66456e 100644 --- a/common/ipc/internals/socket.cpp +++ b/common/ipc/internals/socket.cpp @@ -26,7 +26,7 @@ #include "ipc/exception.hpp" #include "ipc/internals/socket.hpp" -#include "ipc/internals/utils.hpp" +#include "utils/fd-utils.hpp" #include "logger/logger.hpp" #include @@ -60,8 +60,8 @@ Socket::Socket(Socket&& socket) noexcept Socket::~Socket() noexcept { try { - ipc::close(mFD); - } catch (IPCException& e) { + utils::close(mFD); + } catch (std::exception& e) { LOGE("Error in Socket's destructor: " << e.what()); } } @@ -89,13 +89,13 @@ std::shared_ptr Socket::accept() void Socket::write(const void* bufferPtr, const size_t size) const { Guard guard(mCommunicationMutex); - ipc::write(mFD, bufferPtr, size); + utils::write(mFD, bufferPtr, size); } void Socket::read(void* bufferPtr, const size_t size) const { Guard guard(mCommunicationMutex); - ipc::read(mFD, bufferPtr, size); + utils::read(mFD, bufferPtr, size); } int Socket::getSystemdSocket(const std::string& path) @@ -142,7 +142,7 @@ int Socket::createZoneSocket(const std::string& path) reinterpret_cast(&serverAddress), sizeof(struct sockaddr_un))) { std::string message = strerror(errno); - ::close(sockfd); + utils::close(sockfd); LOGE("Error in bind: " << message); IPCException("Error in bind: " + message); } @@ -150,7 +150,7 @@ int Socket::createZoneSocket(const std::string& path) if (-1 == ::listen(sockfd, MAX_QUEUE_LENGTH)) { std::string message = strerror(errno); - ::close(sockfd); + utils::close(sockfd); LOGE("Error in listen: " << message); IPCException("Error in listen: " + message); } @@ -188,7 +188,7 @@ Socket Socket::connectSocket(const std::string& path) if (-1 == connect(fd, reinterpret_cast(&serverAddress), sizeof(struct sockaddr_un))) { - ::close(fd); + utils::close(fd); LOGE("Error in connect: " + std::string(strerror(errno))); throw IPCException("Error in connect: " + std::string(strerror(errno))); } @@ -196,7 +196,7 @@ Socket Socket::connectSocket(const std::string& path) // Nonblock socket int flags = fcntl(fd, F_GETFL, 0); if (-1 == fcntl(fd, F_SETFL, flags | O_NONBLOCK)) { - ::close(fd); + utils::close(fd); LOGE("Error in fcntl: " + std::string(strerror(errno))); throw IPCException("Error in fcntl: " + std::string(strerror(errno))); } diff --git a/common/ipc/service.cpp b/common/ipc/service.cpp index d945f73..442d804 100644 --- a/common/ipc/service.cpp +++ b/common/ipc/service.cpp @@ -50,7 +50,7 @@ Service::~Service() LOGS("Service Destructor"); try { stop(); - } catch (IPCException& e) { + } catch (std::exception& e) { LOGE("Error in Service's destructor: " << e.what()); } } diff --git a/common/utils/event-poll.cpp b/common/utils/event-poll.cpp new file mode 100644 index 0000000..b37d727 --- /dev/null +++ b/common/utils/event-poll.cpp @@ -0,0 +1,151 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief C++ epoll wrapper + */ + +#include "config.hpp" +#include "utils/event-poll.hpp" +#include "utils/fd-utils.hpp" +#include "utils/exception.hpp" +#include "logger/logger.hpp" + +#include +#include +#include +#include + +namespace vasum { +namespace utils { + +EventPoll::EventPoll() + : mPollFD(::epoll_create1(EPOLL_CLOEXEC)) +{ + if (mPollFD == -1) { + LOGE("Failed to create epoll: " << getSystemErrorMessage()); + throw UtilsException("Could not create epoll"); + } +} + +EventPoll::~EventPoll() +{ + if (!mCallbacks.empty()) { + LOGW("Not removed callbacks: " << mCallbacks.size()); + assert(0 && "Not removed callbacks left"); + } + utils::close(mPollFD); +} + +int EventPoll::getPollFD() const +{ + return mPollFD; +} + +void EventPoll::addFD(const int fd, const Events events, Callback&& callback) +{ + std::lock_guard lock(mMutex); + + if (mCallbacks.find(fd) != mCallbacks.end()) { + LOGW("Already added fd: " << fd); + throw UtilsException("FD already added"); + } + + if (!addFDInternal(fd, events)) { + throw UtilsException("Could not add fd"); + } + + mCallbacks.insert({fd, std::make_shared(std::move(callback))}); + LOGT("Callback added for " << fd); +} + +void EventPoll::removeFD(const int fd) +{ + std::lock_guard lock(mMutex); + + auto iter = mCallbacks.find(fd); + if (iter == mCallbacks.end()) { + LOGW("Failed to remove nonexistent fd: " << fd); + throw UtilsException("FD does not exist"); + } + mCallbacks.erase(iter); + removeFDInternal(fd); + LOGT("Callback removed for " << fd); +} + +bool EventPoll::dispatchIteration(const int timeoutMs) +{ + for (;;) { + epoll_event event; + int num = epoll_wait(mPollFD, &event, 1, timeoutMs); + if (num == 0) { + return false; // timeout + } + if (num < 0) { + if (errno == EINTR) { + continue; + } + LOGE("Failed to wait on epoll: " << getSystemErrorMessage()); + throw UtilsException("Could not wait for event"); + } + + // callback could be removed in the meantime, so be careful, find it inside lock + std::lock_guard lock(mMutex); + auto iter = mCallbacks.find(event.data.fd); + if (iter == mCallbacks.end()) { + continue; + } + + // add ref because removeFD(self) can be called inside callback + std::shared_ptr callback(iter->second); + return (*callback)(event.data.fd, event.events); + } +} + +void EventPoll::dispatchLoop() +{ + while (dispatchIteration(-1)) {} +} + +bool EventPoll::addFDInternal(const int fd, const Events events) +{ + epoll_event event; + memset(&event, 0, sizeof(event)); + event.events = events; + event.data.fd = fd; + + if (epoll_ctl(mPollFD, EPOLL_CTL_ADD, fd, &event) == -1) { + LOGE("Failed to add fd to poll: " << getSystemErrorMessage()); + return false; + } + return true; +} + +void EventPoll::removeFDInternal(const int fd) +{ + if (epoll_ctl(mPollFD, EPOLL_CTL_DEL, fd, NULL) == -1) { + assert(errno != EBADF); // always removeFD before closing fd locally! + // this is important because linux will reuse this fd number + LOGE("Failed to remove fd from poll: " << getSystemErrorMessage()); + } +} + +} // namespace utils +} // namespace vasum diff --git a/common/utils/event-poll.hpp b/common/utils/event-poll.hpp new file mode 100644 index 0000000..1a7ae2c --- /dev/null +++ b/common/utils/event-poll.hpp @@ -0,0 +1,68 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief C++ epoll wrapper + */ + +#ifndef COMMON_UTILS_EVENT_POLL_HPP +#define COMMON_UTILS_EVENT_POLL_HPP + +#include +#include +#include +#include + +namespace vasum { +namespace utils { + +class EventPoll { +public: + + typedef unsigned int Events; + typedef std::function Callback; + + EventPoll(); + ~EventPoll(); + + int getPollFD() const; + + void addFD(const int fd, const Events events, Callback&& callback); + void removeFD(const int fd); + + bool dispatchIteration(const int timeoutMs); + void dispatchLoop(); + +private: + typedef std::recursive_mutex Mutex; + + const int mPollFD; + Mutex mMutex; + std::unordered_map> mCallbacks; + + bool addFDInternal(const int fd, const Events events); + void removeFDInternal(const int fd); +}; + + +} // namespace utils +} // namespace vasum + +#endif // COMMON_UTILS_EVENT_POLL_HPP diff --git a/common/ipc/internals/eventfd.cpp b/common/utils/eventfd.cpp similarity index 67% rename from common/ipc/internals/eventfd.cpp rename to common/utils/eventfd.cpp index 729243d..2fd8d62 100644 --- a/common/ipc/internals/eventfd.cpp +++ b/common/utils/eventfd.cpp @@ -24,9 +24,9 @@ #include "config.hpp" -#include "ipc/internals/eventfd.hpp" -#include "ipc/internals/utils.hpp" -#include "ipc/exception.hpp" +#include "utils/eventfd.hpp" +#include "utils/exception.hpp" +#include "utils/fd-utils.hpp" #include "logger/logger.hpp" #include @@ -35,24 +35,20 @@ #include namespace vasum { -namespace ipc { +namespace utils { EventFD::EventFD() { - mFD = ::eventfd(0, EFD_SEMAPHORE | EFD_NONBLOCK); + mFD = ::eventfd(0, EFD_SEMAPHORE | EFD_CLOEXEC); if (mFD == -1) { - LOGE("Error in eventfd: " << std::string(strerror(errno))); - throw IPCException("Error in eventfd: " + std::string(strerror(errno))); + LOGE("Error in eventfd: " << getSystemErrorMessage()); + throw UtilsException("Error in eventfd: " + getSystemErrorMessage()); } } EventFD::~EventFD() { - try { - ipc::close(mFD); - } catch (IPCException& e) { - LOGE("Error in Event's destructor: " << e.what()); - } + utils::close(mFD); } int EventFD::getFD() const @@ -63,15 +59,15 @@ int EventFD::getFD() const void EventFD::send() { const std::uint64_t toSend = 1; - ipc::write(mFD, &toSend, sizeof(toSend)); + utils::write(mFD, &toSend, sizeof(toSend)); } void EventFD::receive() { std::uint64_t readBuffer; - ipc::read(mFD, &readBuffer, sizeof(readBuffer)); + utils::read(mFD, &readBuffer, sizeof(readBuffer)); } -} // namespace ipc +} // namespace utils } // namespace vasum diff --git a/common/ipc/internals/eventfd.hpp b/common/utils/eventfd.hpp similarity index 88% rename from common/ipc/internals/eventfd.hpp rename to common/utils/eventfd.hpp index cf3094c..b3d5027 100644 --- a/common/ipc/internals/eventfd.hpp +++ b/common/utils/eventfd.hpp @@ -22,17 +22,18 @@ * @brief Eventfd wrapper */ -#ifndef COMMON_IPC_INTERNALS_EVENTFD_HPP -#define COMMON_IPC_INTERNALS_EVENTFD_HPP +#ifndef COMMON_UTILS_EVENTFD_HPP +#define COMMON_UTILS_EVENTFD_HPP namespace vasum { -namespace ipc { +namespace utils { class EventFD { public: EventFD(); ~EventFD(); + EventFD(const EventFD& eventfd) = delete; EventFD& operator=(const EventFD&) = delete; @@ -56,7 +57,7 @@ private: int mFD; }; -} // namespace ipc +} // namespace utils } // namespace vasum -#endif // COMMON_IPC_INTERNALS_EVENTFD_HPP +#endif // COMMON_UTILS_EVENTFD_HPP diff --git a/common/ipc/internals/utils.cpp b/common/utils/fd-utils.cpp similarity index 68% rename from common/ipc/internals/utils.cpp rename to common/utils/fd-utils.cpp index cb9c93c..59bad36 100644 --- a/common/ipc/internals/utils.cpp +++ b/common/utils/fd-utils.cpp @@ -19,13 +19,13 @@ /** * @file * @author Jan Olszak (j.olszak@samsung.com) - * @brief Utility functions + * @brief File descriptor utility functions */ #include "config.hpp" -#include "ipc/exception.hpp" -#include "ipc/internals/utils.hpp" +#include "utils/fd-utils.hpp" +#include "utils/exception.hpp" #include "logger/logger.hpp" #include @@ -40,7 +40,7 @@ namespace fs = boost::filesystem; namespace chr = std::chrono; namespace vasum { -namespace ipc { +namespace utils { namespace { @@ -51,13 +51,14 @@ void waitForEvent(int fd, // Wait for the rest of the data struct pollfd fds[1]; fds[0].fd = fd; - fds[0].events = event | POLLHUP; + fds[0].events = event; for (;;) { chr::milliseconds timeoutMS = chr::duration_cast(deadline - chr::high_resolution_clock::now()); if (timeoutMS.count() < 0) { - LOGE("Timeout in read"); - throw IPCException("Timeout in read"); + LOGE("Timeout while waiting for event: " << std::hex << event << + " on fd: " << std::dec << fd); + throw UtilsException("Timeout"); } int ret = ::poll(fds, 1 /*fds size*/, timeoutMS.count()); @@ -66,13 +67,13 @@ void waitForEvent(int fd, if (errno == EINTR) { continue; } - LOGE("Error in poll: " + std::string(strerror(errno))); - throw IPCException("Error in poll: " + std::string(strerror(errno))); + LOGE("Error in poll: " + getSystemErrorMessage()); + throw UtilsException("Error in poll: " + getSystemErrorMessage()); } if (ret == 0) { LOGE("Timeout in read"); - throw IPCException("Timeout in read"); + throw UtilsException("Timeout in read"); } if (fds[0].revents & event) { @@ -82,7 +83,7 @@ void waitForEvent(int fd, if (fds[0].revents & POLLHUP) { LOGW("Peer disconnected"); - throw IPCException("Peer disconnected"); + throw UtilsException("Peer disconnected"); } } } @@ -98,11 +99,10 @@ void close(int fd) for (;;) { if (-1 == ::close(fd)) { if (errno == EINTR) { - LOGD("Close interrupted by a signal, retrying"); + LOGT("Close interrupted by a signal, retrying"); continue; } - LOGE("Error in close: " << std::string(strerror(errno))); - throw IPCException("Error in close: " + std::string(strerror(errno))); + LOGE("Error in close: " << getSystemErrorMessage()); } break; } @@ -118,22 +118,21 @@ void write(int fd, const void* bufferPtr, const size_t size, int timeoutMS) int n = ::write(fd, reinterpret_cast(bufferPtr) + nTotal, size - nTotal); - if (n > 0) { + if (n >= 0) { nTotal += n; + if (nTotal == size) { + // All data is written, break loop + break; + } } else if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR) { // Neglected errors LOGD("Retrying write"); } else { - LOGE("Error during writing: " + std::string(strerror(errno))); - throw IPCException("Error during writing: " + std::string(strerror(errno))); + LOGE("Error during writing: " + getSystemErrorMessage()); + throw UtilsException("Error during writing: " + getSystemErrorMessage()); } - if (nTotal >= size) { - // All data is written, break loop - break; - } else { - waitForEvent(fd, POLLOUT, deadline); - } + waitForEvent(fd, POLLOUT, deadline); } } @@ -147,22 +146,25 @@ void read(int fd, void* bufferPtr, const size_t size, int timeoutMS) int n = ::read(fd, reinterpret_cast(bufferPtr) + nTotal, size - nTotal); - if (n > 0) { + if (n >= 0) { nTotal += n; + if (nTotal == size) { + // All data is read, break loop + break; + } + if (n == 0) { + LOGW("Peer disconnected"); + throw UtilsException("Peer disconnected"); + } } else if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR) { // Neglected errors LOGD("Retrying read"); } else { - LOGE("Error during reading: " + std::string(strerror(errno))); - throw IPCException("Error during reading: " + std::string(strerror(errno))); + LOGE("Error during reading: " + getSystemErrorMessage()); + throw UtilsException("Error during reading: " + getSystemErrorMessage()); } - if (nTotal >= size) { - // All data is read, break loop - break; - } else { - waitForEvent(fd, POLLIN, deadline); - } + waitForEvent(fd, POLLIN, deadline); } } @@ -170,8 +172,8 @@ unsigned int getMaxFDNumber() { struct rlimit rlim; if (-1 == getrlimit(RLIMIT_NOFILE, &rlim)) { - LOGE("Error during getrlimit: " + std::string(strerror(errno))); - throw IPCException("Error during getrlimit: " + std::string(strerror(errno))); + LOGE("Error during getrlimit: " + getSystemErrorMessage()); + throw UtilsException("Error during getrlimit: " + getSystemErrorMessage()); } return rlim.rlim_cur; } @@ -182,8 +184,8 @@ void setMaxFDNumber(unsigned int limit) rlim.rlim_cur = limit; rlim.rlim_max = limit; if (-1 == setrlimit(RLIMIT_NOFILE, &rlim)) { - LOGE("Error during setrlimit: " + std::string(strerror(errno))); - throw IPCException("Error during setrlimit: " + std::string(strerror(errno))); + LOGE("Error during setrlimit: " + getSystemErrorMessage()); + throw UtilsException("Error during setrlimit: " + getSystemErrorMessage()); } } @@ -194,6 +196,6 @@ unsigned int getFDNumber() fs::directory_iterator()); } -} // namespace ipc +} // namespace utils } // namespace vasum diff --git a/common/ipc/internals/utils.hpp b/common/utils/fd-utils.hpp similarity index 90% rename from common/ipc/internals/utils.hpp rename to common/utils/fd-utils.hpp index c3bcaf1..ac32b0a 100644 --- a/common/ipc/internals/utils.hpp +++ b/common/utils/fd-utils.hpp @@ -19,20 +19,19 @@ /** * @file * @author Jan Olszak (j.olszak@samsung.com) - * @brief Utility functions + * @brief File descriptor utility functions */ -#ifndef COMMON_IPC_INTERNALS_UTILS_HPP -#define COMMON_IPC_INTERNALS_UTILS_HPP +#ifndef COMMON_UTILS_FD_HPP +#define COMMON_UTILS_FD_HPP #include namespace vasum { -namespace ipc { +namespace utils { /** * Close the file descriptor. - * Repeat until */ void close(int fd); @@ -73,7 +72,7 @@ void setMaxFDNumber(unsigned int limit); */ unsigned int getFDNumber(); -} // namespace ipc +} // namespace utils } // namespace vasum -#endif // COMMON_IPC_INTERNALS_UTILS_HPP +#endif // COMMON_UTILS_FD_HPP diff --git a/common/utils/glib-poll-dispatcher.cpp b/common/utils/glib-poll-dispatcher.cpp new file mode 100644 index 0000000..5c66d16 --- /dev/null +++ b/common/utils/glib-poll-dispatcher.cpp @@ -0,0 +1,61 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief glib epoll dispatcher + */ + +#include "config.hpp" +#include "utils/glib-poll-dispatcher.hpp" +#include "utils/callback-wrapper.hpp" + +namespace vasum { +namespace utils { + +GlibPollDispatcher::GlibPollDispatcher(EventPoll& poll) +{ + mChannel = g_io_channel_unix_new(poll.getPollFD()); + + auto dispatchCallback = [&]() { + poll.dispatchIteration(0); + }; + + auto cCallback = [](GIOChannel*, GIOCondition, gpointer data) -> gboolean { + getCallbackFromPointer(data)(); + return TRUE; + }; + + mWatchId = g_io_add_watch_full(mChannel, + G_PRIORITY_DEFAULT, + G_IO_IN, + cCallback, + createCallbackWrapper(dispatchCallback, mGuard.spawn()), + &deleteCallbackWrapper); +} + +GlibPollDispatcher::~GlibPollDispatcher() +{ + g_source_remove(mWatchId); + g_io_channel_unref(mChannel); + // mGuard destructor will wait for full unregister of dispatchCallback +} + +} // namespace utils +} // namespace vasum diff --git a/common/utils/glib-poll-dispatcher.hpp b/common/utils/glib-poll-dispatcher.hpp new file mode 100644 index 0000000..07da0c3 --- /dev/null +++ b/common/utils/glib-poll-dispatcher.hpp @@ -0,0 +1,53 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief glib epoll dispatcher + */ + +#ifndef COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP +#define COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP + +#include "utils/event-poll.hpp" +#include "utils/callback-guard.hpp" + +#include + +namespace vasum { +namespace utils { + +/** + * Will dispatch poll events in glib thread + */ +class GlibPollDispatcher { +public: + GlibPollDispatcher(EventPoll& poll); + ~GlibPollDispatcher(); +private: + CallbackGuard mGuard; + GIOChannel* mChannel; + guint mWatchId; +}; + + +} // namespace utils +} // namespace vasum + +#endif // COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP diff --git a/common/utils/thread-poll-dispatcher.cpp b/common/utils/thread-poll-dispatcher.cpp new file mode 100644 index 0000000..f350587 --- /dev/null +++ b/common/utils/thread-poll-dispatcher.cpp @@ -0,0 +1,53 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief Thread epoll dispatcher + */ + +#include "config.hpp" +#include "utils/thread-poll-dispatcher.hpp" + +#include + +namespace vasum { +namespace utils { + +ThreadPollDispatcher::ThreadPollDispatcher(EventPoll& poll) + : mPoll(poll) + , mThread([&]{ poll.dispatchLoop(); }) +{ + auto controlCallback = [this](int, EventPoll::Events) -> bool { + mStopEvent.receive(); + return false; // break the loop + }; + + poll.addFD(mStopEvent.getFD(), EPOLLIN, std::move(controlCallback)); +} + +ThreadPollDispatcher::~ThreadPollDispatcher() +{ + mStopEvent.send(); + mThread.join(); + mPoll.removeFD(mStopEvent.getFD()); +} + +} // namespace utils +} // namespace vasum diff --git a/common/utils/thread-poll-dispatcher.hpp b/common/utils/thread-poll-dispatcher.hpp new file mode 100644 index 0000000..e54cb4e --- /dev/null +++ b/common/utils/thread-poll-dispatcher.hpp @@ -0,0 +1,52 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief Thread epoll dispatcher + */ + +#ifndef COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP +#define COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP + +#include "utils/event-poll.hpp" +#include "utils/eventfd.hpp" + +#include + +namespace vasum { +namespace utils { + +/** + * Will dispatch poll events in a newly created thread + */ +class ThreadPollDispatcher { +public: + ThreadPollDispatcher(EventPoll& poll); + ~ThreadPollDispatcher(); +private: + EventPoll& mPoll; + EventFD mStopEvent; + std::thread mThread; +}; + +} // namespace utils +} // namespace vasum + +#endif // COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP diff --git a/tests/unit_tests/ipc/ut-ipc.cpp b/tests/unit_tests/ipc/ut-ipc.cpp index 193eec9..fa0de22 100644 --- a/tests/unit_tests/ipc/ut-ipc.cpp +++ b/tests/unit_tests/ipc/ut-ipc.cpp @@ -592,7 +592,6 @@ BOOST_AUTO_TEST_CASE(ServiceGSource) l.set(); }; - IPCGSource::Pointer serviceGSource; Service s(SOCKET_PATH); s.setMethodHandler(1, echoCallback); @@ -622,7 +621,6 @@ BOOST_AUTO_TEST_CASE(ClientGSource) Service s(SOCKET_PATH); s.start(); - IPCGSource::Pointer clientGSource; Client c(SOCKET_PATH); c.setMethodHandler(1, echoCallback); c.setSignalHandler(2, signalHandler); diff --git a/tests/unit_tests/utils/ut-event-poll.cpp b/tests/unit_tests/utils/ut-event-poll.cpp new file mode 100644 index 0000000..e387393 --- /dev/null +++ b/tests/unit_tests/utils/ut-event-poll.cpp @@ -0,0 +1,205 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief Unit tests of event poll + */ + +#include "config.hpp" +#include "ut.hpp" + +#include "utils/event-poll.hpp" +#include "logger/logger.hpp" +#include "ipc/internals/socket.hpp" +#include "utils/latch.hpp" +#include "utils/glib-loop.hpp" +#include "utils/glib-poll-dispatcher.hpp" +#include "utils/thread-poll-dispatcher.hpp" + +#include +#include + +using namespace vasum::utils; +using namespace vasum::ipc; + +namespace { + +const int unsigned TIMEOUT = 1000; +#define ADD_EVENT(e) {EPOLL##e, #e} +const std::map EVENT_NAMES = { + ADD_EVENT(IN), + ADD_EVENT(OUT), + ADD_EVENT(ERR), + ADD_EVENT(HUP), + ADD_EVENT(RDHUP), +}; +#undef ADD_EVENT + +std::string strEvents(EventPoll::Events events) +{ + if (events == 0) { + return ""; + } + std::ostringstream ss; + for (const auto& p : EVENT_NAMES) { + if (events & p.first) { + ss << p.second << ", "; + events &= ~p.first; + } + } + if (events != 0) { + ss << std::hex << events; + return ss.str(); + } else { + std::string ret = ss.str(); + ret.resize(ret.size() - 2); + return ret; + } +} + +} // namespace + +BOOST_AUTO_TEST_SUITE(EventPollSuite) + +BOOST_AUTO_TEST_CASE(EmptyPoll) +{ + EventPoll poll; + BOOST_CHECK(!poll.dispatchIteration(0)); +} + +BOOST_AUTO_TEST_CASE(ThreadedPoll) +{ + EventPoll poll; + ThreadPollDispatcher dispatcher(poll); +} + +BOOST_AUTO_TEST_CASE(GlibPoll) +{ + ScopedGlibLoop loop; + + EventPoll poll; + GlibPollDispatcher dispatcher(poll); +} + +void doSocketTest(EventPoll& poll, Latch& goodMessage, Latch& remoteClosed) +{ + const std::string PATH = "/tmp/ut-poll.sock"; + const std::string MESSAGE = "This is a test message"; + + Socket listen = Socket::createSocket(PATH); + std::shared_ptr server; + + auto serverCallback = [&](int, EventPoll::Events events) -> bool { + LOGD("Server events: " << strEvents(events)); + + if (events & EPOLLOUT) { + server->write(MESSAGE.data(), MESSAGE.size()); + poll.removeFD(server->getFD()); + server.reset(); + } + return true; + }; + + auto listenCallback = [&](int, EventPoll::Events events) -> bool { + LOGD("Listen events: " << strEvents(events)); + if (events & EPOLLIN) { + server = listen.accept(); + poll.addFD(server->getFD(), EPOLLHUP | EPOLLRDHUP | EPOLLOUT, serverCallback); + } + return true; + }; + + poll.addFD(listen.getFD(), EPOLLIN, listenCallback); + + Socket client = Socket::connectSocket(PATH); + + auto clientCallback = [&](int, EventPoll::Events events) -> bool { + LOGD("Client events: " << strEvents(events)); + + if (events & EPOLLIN) { + std::string ret(MESSAGE.size(), 'x'); + client.read(&ret.front(), ret.size()); + if (ret == MESSAGE) { + goodMessage.set(); + } + } + if (events & EPOLLRDHUP) { + poll.removeFD(client.getFD()); + remoteClosed.set(); + } + return true; + }; + + poll.addFD(client.getFD(), EPOLLHUP | EPOLLRDHUP | EPOLLIN, clientCallback); + + BOOST_CHECK(goodMessage.wait(TIMEOUT)); + BOOST_CHECK(remoteClosed.wait(TIMEOUT)); + + poll.removeFD(listen.getFD()); +} + +BOOST_AUTO_TEST_CASE(ThreadedPollSocket) +{ + Latch goodMessage; + Latch remoteClosed; + + EventPoll poll; + ThreadPollDispatcher dispatcher(poll); + + doSocketTest(poll, goodMessage, remoteClosed); +} + +BOOST_AUTO_TEST_CASE(GlibPollSocket) +{ + Latch goodMessage; + Latch remoteClosed; + + ScopedGlibLoop loop; + + EventPoll poll; + GlibPollDispatcher dispatcher(poll); + + doSocketTest(poll, goodMessage, remoteClosed); +} + +BOOST_AUTO_TEST_CASE(PollStacking) +{ + Latch goodMessage; + Latch remoteClosed; + + EventPoll outer; + EventPoll inner; + + auto dispatchInner = [&](int, EventPoll::Events) -> bool { + inner.dispatchIteration(0); + return true; + }; + + outer.addFD(inner.getPollFD(), EPOLLIN, dispatchInner); + + ThreadPollDispatcher dispatcher(outer); + doSocketTest(inner, goodMessage, remoteClosed); + + outer.removeFD(inner.getPollFD()); +} + +BOOST_AUTO_TEST_SUITE_END() + -- 2.7.4 From a2bfdea7f5bb1e2126701e0cf7d691f326afa9c4 Mon Sep 17 00:00:00 2001 From: Mateusz Malicki Date: Thu, 12 Feb 2015 16:39:58 +0100 Subject: [PATCH 08/16] Creating netdev [Feature] Filled stubs for creating netdev (create_netdev_phys, create_netdev_macvlan, create_netdev_veth) [Cause] N/A [Solution] Manage netdev through netlink interface [Verification] Build, install, run vasum-server, create zone (vasum-cli create_zone zone1), start zone (vasum-cli start_zone zone1), move interface f.e. p2p1 to zone (vasum-cli create_netdev_phys zone1 p2p1) enter to zone (lxc-console -n zone1 --lxcpath=/usr/local/share/.zones -t 0) check interfaces (ifconfig -a) do analogous for create_netdev_macvlan and create_netdev_veth Change-Id: I299d3ebeb8f101a386f5b156dbe79d7779600ef6 --- common/lxc/zone.cpp | 5 + common/lxc/zone.hpp | 7 ++ common/netlink/netlink-message.cpp | 164 +++++++++++++++++++++++++ common/netlink/netlink-message.hpp | 116 ++++++++++++++++++ common/netlink/netlink.cpp | 155 ++++++++++++++++++++++++ common/netlink/netlink.hpp | 79 ++++++++++++ server/netdev.cpp | 242 +++++++++++++++++++++++++++++++++++++ server/netdev.hpp | 45 +++++++ server/zone-admin.cpp | 19 +-- 9 files changed, 823 insertions(+), 9 deletions(-) create mode 100644 common/netlink/netlink-message.cpp create mode 100644 common/netlink/netlink-message.hpp create mode 100644 common/netlink/netlink.cpp create mode 100644 common/netlink/netlink.hpp create mode 100644 server/netdev.cpp create mode 100644 server/netdev.hpp diff --git a/common/lxc/zone.cpp b/common/lxc/zone.cpp index dc148e4..bbb1e02 100644 --- a/common/lxc/zone.cpp +++ b/common/lxc/zone.cpp @@ -313,6 +313,11 @@ bool LxcZone::waitForState(State state, int timeout) return true; } +pid_t LxcZone::getInitPid() const +{ + return mLxcContainer->init_pid(mLxcContainer); +} + bool LxcZone::setRunLevel(int runLevel) { auto callback = [](void* param) -> int { diff --git a/common/lxc/zone.hpp b/common/lxc/zone.hpp index 1573c7d..0ed19a3 100644 --- a/common/lxc/zone.hpp +++ b/common/lxc/zone.hpp @@ -26,6 +26,7 @@ #define COMMON_LXC_ZONE_HPP #include +#include // fwd declaration of lxc internals struct lxc_container; @@ -136,6 +137,12 @@ public: * Unfreeze zone */ bool unfreeze(); + + /** + * Get pid of init process + */ + pid_t getInitPid() const; + private: lxc_container* mLxcContainer; diff --git a/common/netlink/netlink-message.cpp b/common/netlink/netlink-message.cpp new file mode 100644 index 0000000..72d3597 --- /dev/null +++ b/common/netlink/netlink-message.cpp @@ -0,0 +1,164 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Netlink message class definition + */ + +#include "config.hpp" +#include "netlink-message.hpp" +#include "netlink.hpp" +#include "base-exception.hpp" + +#include + +#include +#include +#include + +#include +#include +#include +#include + +#ifndef PAGE_SIZE +#define PAGE_SIZE 4096 +#endif + +const int NLMSG_GOOD_SIZE = 2*PAGE_SIZE; +constexpr rtattr* NLMSG_TAIL(nlmsghdr* nmsg) +{ + return reinterpret_cast(reinterpret_cast(nmsg) + NLMSG_ALIGN(nmsg->nlmsg_len)); +} + +namespace vasum { +namespace netlink { + +NetlinkMessage::NetlinkMessage(uint16_t type, uint16_t flags) +{ + static uint32_t seq = 0; + mNlmsg.resize(NLMSG_GOOD_SIZE, 0); + hdr().nlmsg_len = NLMSG_HDRLEN; + hdr().nlmsg_flags = flags | NLM_F_ACK; + hdr().nlmsg_type = type; + hdr().nlmsg_seq = ++seq; + hdr().nlmsg_pid = getpid(); +} + +NetlinkMessage& NetlinkMessage::beginNested(int ifla) +{ + struct rtattr *nest = NLMSG_TAIL(&hdr()); + put(ifla, NULL, 0); + mNested.push(nest); + return *this; +} + +NetlinkMessage& NetlinkMessage::endNested() +{ + assert(!mNested.empty()); + rtattr *nest = reinterpret_cast(mNested.top()); + nest->rta_len = std::distance(reinterpret_cast(nest), + reinterpret_cast(NLMSG_TAIL(&hdr()))); + mNested.pop(); + return *this; +} + +NetlinkMessage& NetlinkMessage::put(int ifla, const std::string& value) +{ + return put(ifla, value.c_str(), value.size() + 1); +} + +NetlinkMessage& NetlinkMessage::put(int ifla, const void* data, int len) +{ + struct rtattr *rta; + size_t rtalen = RTA_LENGTH(len); + int newLen = NLMSG_ALIGN(hdr().nlmsg_len) + RTA_ALIGN(rtalen); + + setMinCapacity(newLen); + rta = NLMSG_TAIL(&hdr()); + rta->rta_type = ifla; + rta->rta_len = rtalen; + memcpy(RTA_DATA(rta), data, len); + hdr().nlmsg_len = newLen; + return *this; +} + +NetlinkMessage& NetlinkMessage::put(const void* data, int len) +{ + setMinCapacity(hdr().nlmsg_len + len); + memcpy((reinterpret_cast(&hdr()) + hdr().nlmsg_len), data, len); + hdr().nlmsg_len += len; + return *this; +} + +nlmsghdr& NetlinkMessage::hdr() { + return *reinterpret_cast(mNlmsg.data()); +} + +const nlmsghdr& NetlinkMessage::hdr() const { + return *reinterpret_cast(mNlmsg.data()); +} + +void NetlinkMessage::setMinCapacity(unsigned int size) +{ + if (mNlmsg.size() < size) { + mNlmsg.resize(size, 0); + } +} + +void send(const NetlinkMessage& msg) +{ + //TODO: Handle messages with responses + assert(msg.hdr().nlmsg_flags & NLM_F_ACK); + + const int answerLen = NLMSG_ALIGN(msg.hdr().nlmsg_len + sizeof(nlmsgerr)); + std::unique_ptr answerBuff(new char[answerLen]); + nlmsghdr* answer = reinterpret_cast(answerBuff.get()); + answer->nlmsg_len = answerLen; + + Netlink nl; + nl.open(); + try { + nl.send(&msg.hdr()); + //Receive ACK Netlink Message + do { + nl.rcv(answer); + } while (answer->nlmsg_type == NLMSG_NOOP); + } catch (const std::exception& ex) { + LOGE("Sending failed (" << ex.what() << ")"); + nl.close(); + throw; + } + nl.close(); + if (answer->nlmsg_type != NLMSG_ERROR) { + // It is not NACK/ACK message + throw VasumException("Sending failed ( unrecognized message type )"); + } + nlmsgerr *err = reinterpret_cast(NLMSG_DATA(answer)); + if (answer->nlmsg_seq != msg.hdr().nlmsg_seq) { + throw VasumException("Sending failed ( answer message was mismatched )"); + } + if (err->error) { + throw VasumException("Sending failed (" + getSystemErrorMessage(-err->error) + ")"); + } +} + +} // namespace netlink +} // namespace vasum diff --git a/common/netlink/netlink-message.hpp b/common/netlink/netlink-message.hpp new file mode 100644 index 0000000..e828feb --- /dev/null +++ b/common/netlink/netlink-message.hpp @@ -0,0 +1,116 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Netlink message class declaration + */ + +#ifndef COMMON_NETLINK_NETLINK_MESSAGE_HPP +#define COMMON_NETLINK_NETLINK_MESSAGE_HPP + +#include +#include +#include +#include +#include +#include + +namespace vasum { +namespace netlink { + +/** + * NetlinkMessage is used to creatie a netlink messages + */ +class NetlinkMessage { +public: + /** + * Create netlink message + * + * @param type rtnetlink message type (see man 7 rtnetlink) + * @param flags nlmsg flags (see man 7 netlink) + */ + NetlinkMessage(std::uint16_t type, std::uint16_t flags); + + /** + * Add nested atribute type + * + * All future attributes will be nested in this attribute (till call to endNested) + * + * @param ifla attribute name + */ + NetlinkMessage& beginNested(int ifla); + + /** + * End nested atribute + */ + NetlinkMessage& endNested(); + + ///@{ + /*Add sample attribute */ + NetlinkMessage& put(int ifla, const std::string& value); + template + NetlinkMessage& put(int ifla, const T& value); + ///@} + + /** + * Add raw data + * + * Add raw data to end of netlink message + */ + template + NetlinkMessage& put(const T& value); + + /** + * Send netlink message + * + * It is not thread safe + */ + friend void send(const NetlinkMessage& msg); +private: + std::vector mNlmsg; + std::stack mNested; + + NetlinkMessage& put(int ifla, const void* data, int len); + NetlinkMessage& put(const void* data, int len); + nlmsghdr& hdr(); + const nlmsghdr& hdr() const; + void setMinCapacity(unsigned int size); + + +}; + +template +NetlinkMessage& NetlinkMessage::put(int ifla, const T& value) +{ + static_assert(std::is_pod::value, "Require trivial and standard-layout"); + return put(ifla, &value, sizeof(value)); +} + +template +NetlinkMessage& NetlinkMessage::put(const T& value) +{ + static_assert(std::is_pod::value, "Require trivial and standard-layout structure"); + return put(&value, sizeof(value)); +} + +} // namespace netlink +} // namespace vasum + +#endif // COMMON_NETLINK_NETLINK_MESSAGE_HPP diff --git a/common/netlink/netlink.cpp b/common/netlink/netlink.cpp new file mode 100644 index 0000000..fa86b45 --- /dev/null +++ b/common/netlink/netlink.cpp @@ -0,0 +1,155 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Netlink class definition + */ + +#include "config.hpp" +#include "netlink.hpp" +#include "utils.hpp" +#include "base-exception.hpp" + +#include +#include +#include +#include +#include +#include + +namespace vasum { + +namespace { + +template +T make_clean() +{ + static_assert(std::is_pod::value, "make_clean require trivial and standard-layout"); + T value; + std::fill_n(reinterpret_cast(&value), sizeof(value), 0); + return value; +} + +} // namespace + +Netlink::Netlink() : mFd(-1) +{ +} + +Netlink::~Netlink() +{ + close(); +} + +void Netlink::open() +{ + assert(mFd == -1); + mFd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE); + if (mFd == -1) { + LOGE("Can't open socket (" << getSystemErrorMessage() << ")"); + throw VasumException("Can't open netlink connection"); + } + + sockaddr_nl local = make_clean(); + local.nl_family = AF_NETLINK; + + if (bind(mFd, (struct sockaddr *)&local, sizeof(local)) < 0) { + int err = errno; + close(); + LOGE("Can't bind to socket (" << getSystemErrorMessage(err) << ")"); + throw VasumException("Can't set up netlink connection"); + } +} + +void Netlink::close() +{ + if (mFd != -1) { + ::close(mFd); + mFd = -1; + } +} + +void Netlink::send(const nlmsghdr *nlmsg) +{ + msghdr msg = make_clean(); + sockaddr_nl nladdr = make_clean(); + iovec iov = make_clean(); + + iov.iov_base = (void *)nlmsg; + iov.iov_len = nlmsg->nlmsg_len; + msg.msg_name = &nladdr; + msg.msg_namelen = sizeof(nladdr); + msg.msg_iov = &iov; + msg.msg_iovlen = 1; + nladdr.nl_family = AF_NETLINK; + + int ret = sendmsg(mFd, &msg, 0); + if (ret < 0) { + LOGE("Can't send message (" << getSystemErrorMessage() << ")"); + throw VasumException("Can't send netlink message"); + } +} + +int Netlink::rcv(nlmsghdr *answer) +{ + //TODO: Handle too small buffer situation (buffer resizing) + msghdr msg = make_clean(); + sockaddr_nl nladdr = make_clean(); + iovec iov = make_clean(); + + iov.iov_base = answer; + iov.iov_len = answer->nlmsg_len; + msg.msg_name = &nladdr; + msg.msg_namelen = sizeof(nladdr); + msg.msg_iov = &iov; + msg.msg_iovlen = 1; + nladdr.nl_family = AF_NETLINK; + + int ret = recvmsg(mFd, &msg, 0); + if (ret < 0) { + LOGE("Can't receive message (" + getSystemErrorMessage() + ")"); + throw VasumException("Can't receive netlink message"); + } + if (ret == 0) { + LOGE("Peer has performed an orderly shutdown"); + throw VasumException("Can't receive netlink message"); + } + if (msg.msg_flags & MSG_TRUNC) { + LOGE("Can't receive message (" + getSystemErrorMessage(EMSGSIZE) + ")"); + throw VasumException("Can't receive netlink message"); + } + if (msg.msg_flags & MSG_ERRQUEUE) { + LOGE("No data was received but an extended error"); + throw VasumException("Can't receive netlink message"); + } + if (msg.msg_flags & MSG_OOB) { + LOGE("Internal error (expedited or out-of-band data were received)"); + throw VasumException("Can't receive netlink message"); + } + if (msg.msg_flags & (MSG_EOR | MSG_CTRUNC)) { + assert(!"This should not happen!"); + LOGE("Internal error (" << std::to_string(msg.msg_flags) << ")"); + throw VasumException("Internal error while recaiving netlink message"); + } + + return ret; +} + +} //namespace vasum diff --git a/common/netlink/netlink.hpp b/common/netlink/netlink.hpp new file mode 100644 index 0000000..2e73ce8 --- /dev/null +++ b/common/netlink/netlink.hpp @@ -0,0 +1,79 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Netlink class declaration + */ + +#ifndef COMMON_NETLINK_NETLINK_HPP +#define COMMON_NETLINK_NETLINK_HPP + +#include + +namespace vasum { + +/** + * Netlink class is responsible for communicating + * with kernel through netlink interface + */ +class Netlink { +public: + Netlink(); + ~Netlink(); + + Netlink(const Netlink& other) = delete; + Netlink& operator=(const Netlink& other) = delete; + + /** + * Open connnection + */ + void open(); + + /** + * Close connection + */ + void close(); + + /** + * Send message + * + * It is not thread safe and even you shouldn't call this function on + * different instances at the same time + * + * @param nlmsg pointer to message + */ + void send(const nlmsghdr *nlmsg); + + /** + * Receive message + * + * It is not thread safe and even you shouldn't call this function on + * different instances at the same time + * + * @param answer pointer to answer buffer + */ + int rcv(nlmsghdr *answer); +private: + int mFd; +}; + +} // namespace vasum + +#endif /* COMMON_NETLINK_NETLINK_HPP */ diff --git a/server/netdev.cpp b/server/netdev.cpp new file mode 100644 index 0000000..b4a2d21 --- /dev/null +++ b/server/netdev.cpp @@ -0,0 +1,242 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Network devices management functions definition + */ + +#include "config.hpp" +#include "netdev.hpp" +#include "netlink/netlink-message.hpp" +#include "utils.hpp" +#include "exception.hpp" + +#include + +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +using namespace std; +using namespace vasum; +using namespace vasum::netlink; + +namespace vasum { +namespace netdev { + +namespace { + +template +T make_clean() +{ + static_assert(std::is_pod::value, "make_clean require trivial and standard-layout"); + T value; + std::fill_n(reinterpret_cast(&value), sizeof(value), 0); + return value; +} + +string getUniqueVethName() +{ + auto find = [](const ifaddrs* ifaddr, const string& name) -> bool { + for (const ifaddrs* ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) { + if (name == ifa->ifa_name) { + return true; + } + } + return false; + }; + + ifaddrs* ifaddr; + getifaddrs(&ifaddr); + string newName; + int i = 0; + do { + newName = "veth0" + to_string(++i); + } while (find(ifaddr, newName)); + + freeifaddrs(ifaddr); + return newName; +} + +uint32_t getInterfaceIndex(const string& name) { + uint32_t index = if_nametoindex(name.c_str()); + if (!index) { + LOGE("Can't get " << name << " interface index (" << getSystemErrorMessage() << ")"); + throw ZoneOperationException("Can't find interface"); + } + return index; +} + +void validateNetdevName(const string& name) +{ + if (name.size() <= 1 || name.size() >= IFNAMSIZ) { + throw ZoneOperationException("Invalid netdev name format"); + } +} + +void createPipedNetdev(const string& netdev1, const string& netdev2) +{ + validateNetdevName(netdev1); + validateNetdevName(netdev2); + + NetlinkMessage nlm(RTM_NEWLINK, NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL | NLM_F_ACK); + ifinfomsg infoPeer = make_clean(); + infoPeer.ifi_family = AF_UNSPEC; + infoPeer.ifi_change = 0xFFFFFFFF; + nlm.put(infoPeer) + .beginNested(IFLA_LINKINFO) + .put(IFLA_INFO_KIND, "veth") + .beginNested(IFLA_INFO_DATA) + .beginNested(VETH_INFO_PEER) + .put(infoPeer) + .put(IFLA_IFNAME, netdev2) + .endNested() + .endNested() + .endNested() + .put(IFLA_IFNAME, netdev1); + send(nlm); +} + +void attachToBridge(const string& bridge, const string& netdev) +{ + validateNetdevName(bridge); + validateNetdevName(netdev); + + uint32_t index = getInterfaceIndex(netdev); + int fd = socket(AF_LOCAL, SOCK_STREAM, 0); + if (fd < 0) { + LOGE("Can't open socket (" << getSystemErrorMessage() << ")"); + throw ZoneOperationException("Can't attach to bridge"); + } + + struct ifreq ifr = make_clean(); + strncpy(ifr.ifr_name, bridge.c_str(), IFNAMSIZ); + ifr.ifr_ifindex = index; + int err = ioctl(fd, SIOCBRADDIF, &ifr); + if (err < 0) { + int error = errno; + //TODO: Close can be interrupted. Move util functions from ipc + ::close(fd); + LOGE("Can't attach to bridge (" + getSystemErrorMessage(error) + ")"); + throw ZoneOperationException("Can't attach to bridge"); + } + close(fd); +} + +int setFlags(const string& name, uint32_t mask, uint32_t flags) +{ + uint32_t index = getInterfaceIndex(name); + NetlinkMessage nlm(RTM_NEWLINK, NLM_F_REQUEST | NLM_F_ACK); + ifinfomsg infoPeer = make_clean(); + infoPeer.ifi_family = AF_UNSPEC; + infoPeer.ifi_index = index; + infoPeer.ifi_flags = flags; + // since kernel v2.6.22 ifi_change is used to change only selected flags; + infoPeer.ifi_change = mask; + nlm.put(infoPeer); + send(nlm); + return 0; +} + +void up(const string& netdev) +{ + setFlags(netdev, IFF_UP, IFF_UP); +} + +void moveToNS(const string& netdev, pid_t pid) +{ + uint32_t index = getInterfaceIndex(netdev); + NetlinkMessage nlm(RTM_NEWLINK, NLM_F_REQUEST | NLM_F_ACK); + ifinfomsg infopeer = make_clean(); + infopeer.ifi_family = AF_UNSPEC; + infopeer.ifi_index = index; + nlm.put(infopeer) + .put(IFLA_NET_NS_PID, pid); + send(nlm); +} + +void createMacvlan(const string& master, const string& slave, const macvlan_mode& mode) +{ + validateNetdevName(master); + validateNetdevName(slave); + + uint32_t index = getInterfaceIndex(master); + NetlinkMessage nlm(RTM_NEWLINK, NLM_F_REQUEST|NLM_F_CREATE|NLM_F_EXCL|NLM_F_ACK); + ifinfomsg infopeer = make_clean(); + infopeer.ifi_family = AF_UNSPEC; + infopeer.ifi_change = 0xFFFFFFFF; + nlm.put(infopeer) + .beginNested(IFLA_LINKINFO) + .put(IFLA_INFO_KIND, "macvlan") + .beginNested(IFLA_INFO_DATA) + .put(IFLA_MACVLAN_MODE, static_cast(mode)) + .endNested() + .endNested() + .put(IFLA_LINK, index) + .put(IFLA_IFNAME, slave); + send(nlm); +} + +} // namespace + +void createVeth(const pid_t& nsPid, const string& nsDev, const string& hostDev) +{ + string hostVeth = getUniqueVethName(); + LOGT("Creating veth: bridge: " << hostDev << ", port: " << hostVeth << ", zone: " << nsDev); + createPipedNetdev(nsDev, hostVeth); + //TODO: clean up if following instructions fail + attachToBridge(hostDev, hostVeth); + up(hostVeth); + moveToNS(nsDev, nsPid); +} + +void createMacvlan(const pid_t& nsPid, + const string& nsDev, + const string& hostDev, + const macvlan_mode& mode) +{ + LOGT("Creating macvlan: host: " << hostDev << ", zone: " << nsDev << ", mode: " << mode); + createMacvlan(hostDev, nsDev, mode); + //TODO: clean up if following instructions fail + up(nsDev); + moveToNS(nsDev, nsPid); +} + +void movePhys(const pid_t& nsPid, const string& devId) +{ + LOGT("Creating phys: dev: " << devId); + moveToNS(devId, nsPid); +} + +} //namespace netdev +} //namespace vasum + diff --git a/server/netdev.hpp b/server/netdev.hpp new file mode 100644 index 0000000..de761ca --- /dev/null +++ b/server/netdev.hpp @@ -0,0 +1,45 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Mateusz Malicki + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Mateusz Malicki (m.malicki2@samsung.com) + * @brief Network devices management functions declaration + */ + +#ifndef SERVER_NETDEV_HPP +#define SERVER_NETDEV_HPP + +#include +#include +#include + +namespace vasum { +namespace netdev { + +void createVeth(const pid_t& nsPid, const std::string& nsDev, const std::string& hostDev); +void createMacvlan(const pid_t& nsPid, + const std::string& nsDev, + const std::string& hostDev, + const macvlan_mode& mode); +void movePhys(const pid_t& nsPid, const std::string& devId); + +} //namespace netdev +} //namespace vasum + +#endif // SERVER_NETDEV_HPP diff --git a/server/zone-admin.cpp b/server/zone-admin.cpp index c5e994d..5a4adfa 100644 --- a/server/zone-admin.cpp +++ b/server/zone-admin.cpp @@ -26,6 +26,7 @@ #include "zone-admin.hpp" #include "exception.hpp" +#include "netdev.hpp" #include "logger/logger.hpp" #include "utils/paths.hpp" @@ -270,22 +271,22 @@ std::int64_t ZoneAdmin::getSchedulerQuota() return std::stoll(ret); } -void ZoneAdmin::createNetdevVeth(const std::string& /* zoneDev */, - const std::string& /* hostDev */) +void ZoneAdmin::createNetdevVeth(const std::string& zoneDev, + const std::string& hostDev) { - throw ZoneOperationException("Not implemented"); + netdev::createVeth(mZone.getInitPid(), zoneDev, hostDev); } -void ZoneAdmin::createNetdevMacvlan(const std::string& /* zoneDev */, - const std::string& /* hostDev */, - const uint32_t& /* mode */) +void ZoneAdmin::createNetdevMacvlan(const std::string& zoneDev, + const std::string& hostDev, + const uint32_t& mode) { - throw ZoneOperationException("Not implemented"); + netdev::createMacvlan(mZone.getInitPid(), zoneDev, hostDev, static_cast(mode)); } -void ZoneAdmin::moveNetdev(const std::string& /* devId */) +void ZoneAdmin::moveNetdev(const std::string& devId) { - throw ZoneOperationException("Not implemented"); + netdev::movePhys(mZone.getInitPid(), devId); } } // namespace vasum -- 2.7.4 From d8256e79f51e16d2d35fbf3b4c12f714cc47e215 Mon Sep 17 00:00:00 2001 From: Mateusz Malicki Date: Thu, 19 Feb 2015 15:44:34 +0100 Subject: [PATCH 09/16] Stubs for get/set ipv4/ipv6, destroy netdev and list netdev [Feature] Stubs (cli, client, server) for get/set ipv4/ipv6, destroy netdev and list netdev netdev [Cause] N/A [Solution] N/A [Verification] Build, run cli commnads Change-Id: Ia1dc21e5fb252b2f83d617abec785bce7d6ffc9e --- cli/command-line-interface.cpp | 129 +++++++++++++++++++++++ cli/command-line-interface.hpp | 49 +++++++++ cli/main.cpp | 66 ++++++++++++ client/CMakeLists.txt | 4 +- client/vasum-client-impl.cpp | 205 ++++++++++++++++++++++++++++++++++--- client/vasum-client-impl.hpp | 10 ++ client/vasum-client.cpp | 14 +++ client/vasum-client.h | 25 +++++ common/netlink/netlink-message.cpp | 6 +- server/host-connection.cpp | 53 ++++++++++ server/host-connection.hpp | 36 ++++++- server/host-dbus-definitions.hpp | 17 +++ server/zone-admin.cpp | 15 +++ server/zone-admin.hpp | 16 +++ server/zone.cpp | 18 ++++ server/zone.hpp | 15 +++ server/zones-manager.cpp | 87 +++++++++++++++- server/zones-manager.hpp | 9 ++ 18 files changed, 754 insertions(+), 20 deletions(-) diff --git a/cli/command-line-interface.cpp b/cli/command-line-interface.cpp index a189a92..77e9dc5 100644 --- a/cli/command-line-interface.cpp +++ b/cli/command-line-interface.cpp @@ -38,6 +38,7 @@ #include #include #include +#include using namespace std; @@ -394,5 +395,133 @@ void create_netdev_phys(int pos, int argc, const char** argv) argv[pos + 2])); } +void zone_get_netdevs(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 1) { + throw runtime_error("Not enough parameters"); + } + VsmArrayString ids; + one_shot(bind(vsm_zone_get_netdevs, + _1, + argv[pos + 1], + &ids)); + string delim; + for (VsmString* id = ids; *id; ++id) { + cout << delim << *id; + delim = ", "; + } + if (delim.empty()) { + cout << "There is no network device in zone"; + } + cout << endl; + vsm_array_string_free(ids); +} + +void netdev_get_ipv4_addr(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 2) { + throw runtime_error("Not enough parameters"); + } + in_addr addr; + one_shot(bind(vsm_netdev_get_ipv4_addr, + _1, + argv[pos + 1], + argv[pos + 2], + &addr)); + char buf[INET_ADDRSTRLEN]; + if (inet_ntop(AF_INET, &addr, buf, INET_ADDRSTRLEN) == NULL) { + throw runtime_error("Server gave the wrong address format"); + } + cout << buf << endl; +} + +void netdev_get_ipv6_addr(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 2) { + throw runtime_error("Not enough parameters"); + } + in6_addr addr; + one_shot(bind(vsm_netdev_get_ipv6_addr, + _1, + argv[pos + 1], + argv[pos + 2], + &addr)); + char buf[INET6_ADDRSTRLEN]; + if (inet_ntop(AF_INET6, &addr, buf, INET6_ADDRSTRLEN) == NULL) { + throw runtime_error("Server gave the wrong address format"); + } + cout << buf << endl; +} + +void netdev_set_ipv4_addr(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 4) { + throw runtime_error("Not enough parameters"); + } + in_addr addr; + if (inet_pton(AF_INET, argv[pos + 3], &addr) != 1) { + throw runtime_error("Wrong address format"); + }; + one_shot(bind(vsm_netdev_set_ipv4_addr, + _1, + argv[pos + 1], + argv[pos + 2], + &addr, + stoi(argv[pos + 4]))); +} + +void netdev_set_ipv6_addr(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 4) { + throw runtime_error("Not enough parameters"); + } + in6_addr addr; + if (inet_pton(AF_INET6, argv[pos + 3], &addr) != 1) { + throw runtime_error("Wrong address format"); + }; + one_shot(bind(vsm_netdev_set_ipv6_addr, + _1, + argv[pos + 1], + argv[pos + 2], + &addr, + stoi(argv[pos + 4]))); +} + +void netdev_up(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 2) { + throw runtime_error("Not enough parameters"); + } + one_shot(bind(vsm_netdev_up, + _1, + argv[pos + 1], + argv[pos + 2])); +} + +void netdev_down(int pos, int argc, const char** argv) +{ + using namespace std::placeholders; + + if (argc <= pos + 2) { + throw runtime_error("Not enough parameters"); + } + one_shot(bind(vsm_netdev_down, + _1, + argv[pos + 1], + argv[pos + 2])); +} + } // namespace cli } // namespace vasum diff --git a/cli/command-line-interface.hpp b/cli/command-line-interface.hpp index 34ac013..eb33428 100644 --- a/cli/command-line-interface.hpp +++ b/cli/command-line-interface.hpp @@ -208,6 +208,55 @@ void create_netdev_macvlan(int pos, int argc, const char** argv); */ void create_netdev_phys(int pos, int argc, const char** argv); +/** + * Parses command line arguments and prints result of vsm_zone_get_netdevs + * + * @see vsm_zone_get_netdevs + */ +void zone_get_netdevs(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and prints result of vsm_netdev_get_ipv4_addr + * + * @see vsm_netdev_get_ipv4_addr + */ +void netdev_get_ipv4_addr(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and and prints result of vsm_netdev_get_ipv6_addr + * + * @see vsm_netdev_get_ipv6_addr + */ +void netdev_get_ipv6_addr(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_netdev_set_ipv4_addr + * + * @see vsm_netdev_set_ipv4_addr + */ +void netdev_set_ipv4_addr(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_netdev_set_ipv6_addr + * + * @see vsm_netdev_set_ipv6_addr + */ +void netdev_set_ipv6_addr(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_netdev_up + * + * @see vsm_netdev_up + */ +void netdev_up(int pos, int argc, const char** argv); + +/** + * Parses command line arguments and call vsm_netdev_down + * + * @see vsm_netdev_down + */ +void netdev_down(int pos, int argc, const char** argv); + } // namespace cli } // namespace vasum diff --git a/cli/main.cpp b/cli/main.cpp index 38a8ae5..5f073be 100644 --- a/cli/main.cpp +++ b/cli/main.cpp @@ -170,6 +170,72 @@ std::map commands = { {{"zone_id", "id zone name"}, {"devId", "network device id"}} } + }, + { + "zone_get_netdevs", { + zone_get_netdevs, + "zone_get_netdevs zone_id", + "List network devices in the zone", + {{"zone_id", "id zone name"}} + } + }, + { + "netdev_get_ipv4_addr", { + netdev_get_ipv4_addr, + "netdev_get_ipv4_addr zone_id netdev_id", + "Get ipv4 address", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device name"}} + } + }, + { + "netdev_get_ipv6_addr", { + netdev_get_ipv6_addr, + "netdev_get_ipv6_addr zone_id netdev_id", + "Get ipv6 address", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device name"}} + } + }, + { + "netdev_set_ipv4_addr", { + netdev_set_ipv4_addr, + "netdev_set_ipv4_addr zone_id netdev_id address prefix_len", + "Set ipv4 address", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device name"}, + {"address", "ipv4 address"}, + {"prefix_len", "bit length of prefix"}} + } + }, + { + "netdev_set_ipv6_addr", { + netdev_set_ipv6_addr, + "netdev_set_ipv6_addr zone_id netdev_id address prefix_len", + "Set ipv6 address", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device name"}, + {"address", "ipv6 address"}, + {"prefix_len", "bit length of prefix"}} + } + }, + { + "netdev_up", { + netdev_up, + "netdev_up zone_id netdev_id", + "Turn up a network device in the zone", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device id"}} + } + }, + { + "zone_get_netdevs", { + zone_get_netdevs, + "zone_get_netdevs zone_id", + "Turn down a network device in the zone", + {{"zone_id", "id zone name"}, + {"netdev_id", "network device id"}} + } } }; diff --git a/client/CMakeLists.txt b/client/CMakeLists.txt index 7060b7b..558f3a7 100644 --- a/client/CMakeLists.txt +++ b/client/CMakeLists.txt @@ -23,7 +23,9 @@ FILE(GLOB project_SRCS *.cpp *.hpp *.h) FILE(GLOB common_SRCS ${COMMON_FOLDER}/utils/callback-guard.hpp ${COMMON_FOLDER}/utils/callback-guard.cpp ${COMMON_FOLDER}/utils/glib-loop.cpp - ${COMMON_FOLDER}/utils/glib-loop.hpp) + ${COMMON_FOLDER}/utils/glib-loop.hpp + ${COMMON_FOLDER}/base-exception.hpp + ${COMMON_FOLDER}/base-exception.cpp) SET(_LIB_VERSION_ "0.0.1") SET(_LIB_SOVERSION_ "0") diff --git a/client/vasum-client-impl.cpp b/client/vasum-client-impl.cpp index d6f4217..d8ba562 100644 --- a/client/vasum-client-impl.cpp +++ b/client/vasum-client-impl.cpp @@ -31,10 +31,15 @@ #include #include #include +#include +#include +#include +#include #include #include #include +#include using namespace std; using namespace dbus; @@ -77,6 +82,21 @@ void toDict(GVariant* in, VsmArrayString* keys, VsmArrayString* values) *values = outv; } +vector> toDict(GVariant* in) +{ + assert(in); + + const gchar* key; + const gchar* value; + vector> dict; + GVariantIter iter; + g_variant_iter_init(&iter, in); + while (g_variant_iter_loop(&iter, "(&s&s)", &key, &value)) { + dict.push_back(make_tuple(key, value)); + } + return dict; +} + void toBasic(GVariant* in, char** str) { assert(in); @@ -152,6 +172,37 @@ void toArray(GVariant* in, T** scArray) *scArray = ids; } +string toString(const in_addr* addr) +{ + char buf[INET_ADDRSTRLEN]; + const char* ret = inet_ntop(AF_INET, addr, buf, INET_ADDRSTRLEN); + if (ret == NULL) { + throw runtime_error(getSystemErrorMessage()); + } + return ret; +} + +string toString(const in6_addr* addr) +{ + char buf[INET6_ADDRSTRLEN]; + const char* ret = inet_ntop(AF_INET6, addr, buf, INET6_ADDRSTRLEN); + if (ret == NULL) { + throw runtime_error(getSystemErrorMessage()); + } + return ret; +} + +GVariant* createTupleArray(const vector>& dict) +{ + GVariantBuilder builder; + g_variant_builder_init(&builder, G_VARIANT_TYPE_ARRAY); + for (const auto entry : dict) { + g_variant_builder_add(&builder, "(ss)", get<0>(entry).c_str(), get<1>(entry).c_str()); + } + return g_variant_builder_end(&builder); +} + + VsmStatus toStatus(const exception& ex) { if (typeid(DbusCustomException) == typeid(ex)) { @@ -530,34 +581,162 @@ VsmStatus Client::vsm_revoke_device(const char* id, const char* device) noexcept return callMethod(HOST_INTERFACE, api::host::METHOD_REVOKE_DEVICE, args_in); } -VsmStatus Client::vsm_zone_get_netdevs(const char*, VsmArrayString*) noexcept +VsmStatus Client::vsm_zone_get_netdevs(const char* zone, VsmArrayString* netdevIds) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); - return vsm_get_status(); + assert(zone); + assert(netdevIds); + + GVariant* out = NULL; + GVariant* args_in = g_variant_new("(s)", zone); + VsmStatus ret = callMethod(HOST_INTERFACE, + api::host::METHOD_GET_NETDEV_LIST, + args_in, + "(as)", + &out); + if (ret != VSMCLIENT_SUCCESS) { + return ret; + } + GVariant* unpacked; + g_variant_get(out, "(*)", &unpacked); + toArray(unpacked, netdevIds); + g_variant_unref(unpacked); + g_variant_unref(out); + return ret; } -VsmStatus Client::vsm_netdev_get_ipv4_addr(const char*, const char*, struct in_addr*) noexcept +VsmStatus Client::vsm_netdev_get_ipv4_addr(const char* zone, + const char* netdevId, + struct in_addr* addr) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); + assert(zone); + assert(netdevId); + assert(addr); + + GVariant* out = NULL; + GVariant* args_in = g_variant_new("(ss)", zone, netdevId); + VsmStatus ret = callMethod(HOST_INTERFACE, + api::host::METHOD_GET_NETDEV_ATTRS, + args_in, + "(a(ss))", + &out); + if (ret != VSMCLIENT_SUCCESS) { + return ret; + } + GVariant* unpacked; + g_variant_get(out, "(*)", &unpacked); + vector> attrs = toDict(unpacked); + g_variant_unref(unpacked); + g_variant_unref(out); + + auto it = find_if(attrs.begin(), attrs.end(), [](const tuple& entry) { + return get<0>(entry) == "ipv4"; + }); + if (it != attrs.end()) { + if (inet_pton(AF_INET, get<1>(*it).c_str(), addr) != 1) { + mStatus = Status(VSMCLIENT_CUSTOM_ERROR, "Invalid response data"); + return vsm_get_status(); + } + mStatus = Status(); + return vsm_get_status(); + } + mStatus = Status(VSMCLIENT_CUSTOM_ERROR, "Address not found"); return vsm_get_status(); } -VsmStatus Client::vsm_netdev_get_ipv6_addr(const char*, const char*, struct in6_addr*) noexcept +VsmStatus Client::vsm_netdev_get_ipv6_addr(const char* zone, + const char* netdevId, + struct in6_addr* addr) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); + assert(zone); + assert(netdevId); + assert(addr); + + GVariant* out = NULL; + GVariant* args_in = g_variant_new("(ss)", zone, netdevId); + VsmStatus ret = callMethod(HOST_INTERFACE, + api::host::METHOD_GET_NETDEV_ATTRS, + args_in, + "(a(ss))", + &out); + if (ret != VSMCLIENT_SUCCESS) { + return ret; + } + GVariant* unpacked; + g_variant_get(out, "(*)", &unpacked); + vector> attrs = toDict(unpacked); + g_variant_unref(unpacked); + g_variant_unref(out); + + //XXX: return only one address + auto it = find_if(attrs.begin(), attrs.end(), [](const tuple& entry) { + return get<0>(entry) == "ipv6"; + }); + if (it != attrs.end()) { + if (inet_pton(AF_INET6, get<1>(*it).c_str(), addr) != 1) { + mStatus = Status(VSMCLIENT_CUSTOM_ERROR, "Invalid response data"); + return vsm_get_status(); + } + mStatus = Status(); + return vsm_get_status(); + } + mStatus = Status(VSMCLIENT_CUSTOM_ERROR, "Address not found"); return vsm_get_status(); } -VsmStatus Client::vsm_netdev_set_ipv4_addr(const char*, const char*, struct in_addr*, int) noexcept +VsmStatus Client::vsm_netdev_set_ipv4_addr(const char* zone, + const char* netdevId, + struct in_addr* addr, + int mask) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); - return vsm_get_status(); + try { + GVariant* dict = createTupleArray({make_tuple("ipv4", toString(addr)), + make_tuple("mask", to_string(mask))}); + GVariant* args_in = g_variant_new("(ss@a(ss))", zone, netdevId, dict); + return callMethod(HOST_INTERFACE, api::host::METHOD_SET_NETDEV_ATTRS, args_in); + } catch (exception& ex) { + mStatus = Status(VSMCLIENT_INVALID_ARGUMENT, ex.what()); + return vsm_get_status(); + } } -VsmStatus Client::vsm_netdev_set_ipv6_addr(const char*, const char*, struct in6_addr*, int) noexcept +VsmStatus Client::vsm_netdev_set_ipv6_addr(const char* zone, + const char* netdevId, + struct in6_addr* addr, + int mask) noexcept { - mStatus = Status(VSMCLIENT_OTHER_ERROR, "Not implemented"); - return vsm_get_status(); + try { + GVariant* dict = createTupleArray({make_tuple("ipv6", toString(addr)), + make_tuple("mask", to_string(mask))}); + GVariant* args_in = g_variant_new("(ss@a(ss))", zone, netdevId, dict); + return callMethod(HOST_INTERFACE, api::host::METHOD_SET_NETDEV_ATTRS, args_in); + } catch (exception& ex) { + mStatus = Status(VSMCLIENT_INVALID_ARGUMENT, ex.what()); + return vsm_get_status(); + } +} + +VsmStatus Client::vsm_netdev_up(const char* zone, const char* netdevId) noexcept +{ + try { + GVariant* dict = createTupleArray({make_tuple("up", "true")}); + GVariant* args_in = g_variant_new("(ss@a(ss))", zone, netdevId, dict); + return callMethod(HOST_INTERFACE, api::host::METHOD_SET_NETDEV_ATTRS, args_in); + } catch (exception& ex) { + mStatus = Status(VSMCLIENT_INVALID_ARGUMENT, ex.what()); + return vsm_get_status(); + } +} + +VsmStatus Client::vsm_netdev_down(const char* zone, const char* netdevId) noexcept +{ + try { + GVariant* dict = createTupleArray({make_tuple("up", "false")}); + GVariant* args_in = g_variant_new("(ss@a(ss))", zone, netdevId, dict); + return callMethod(HOST_INTERFACE, api::host::METHOD_SET_NETDEV_ATTRS, args_in); + } catch (exception& ex) { + mStatus = Status(VSMCLIENT_INVALID_ARGUMENT, ex.what()); + return vsm_get_status(); + } } VsmStatus Client::vsm_create_netdev_veth(const char* zone, diff --git a/client/vasum-client-impl.hpp b/client/vasum-client-impl.hpp index 193f140..9399839 100644 --- a/client/vasum-client-impl.hpp +++ b/client/vasum-client-impl.hpp @@ -229,6 +229,16 @@ public: int prefix) noexcept; /** + * @see ::vsm_netdev_up + */ + VsmStatus vsm_netdev_up(const char* zone, const char* netdevId) noexcept; + + /** + * @see ::vsm_netdev_down + */ + VsmStatus vsm_netdev_down(const char* zone, const char* netdevId) noexcept; + + /** * @see ::vsm_create_netdev_veth */ VsmStatus vsm_create_netdev_veth(const char* zone, diff --git a/client/vasum-client.cpp b/client/vasum-client.cpp index 1f011c1..9c0ca60 100644 --- a/client/vasum-client.cpp +++ b/client/vasum-client.cpp @@ -250,6 +250,20 @@ API VsmStatus vsm_netdev_set_ipv6_addr(VsmClient client, return getClient(client).vsm_netdev_set_ipv6_addr(zone, netdevId, addr, prefix); } +API VsmStatus vsm_netdev_up(VsmClient client, + const char* zone, + const char* netdevId) +{ + return getClient(client).vsm_netdev_up(zone, netdevId); +} + +API VsmStatus vsm_netdev_down(VsmClient client, + const char* zone, + const char* netdevId) +{ + return getClient(client).vsm_netdev_down(zone, netdevId); +} + API VsmStatus vsm_create_netdev_veth(VsmClient client, const char* zone, const char* zoneDev, diff --git a/client/vasum-client.h b/client/vasum-client.h index 1fb48b1..0bd40ad 100644 --- a/client/vasum-client.h +++ b/client/vasum-client.h @@ -552,6 +552,31 @@ VsmStatus vsm_netdev_set_ipv6_addr(VsmClient client, int prefix); /** + * Turn up a network device in the zone + * + * @param[in] client vasum-server's client + * @param[in] zone zone name + * @param[in] netdevId netdev id + * @return status of this function call + */ +VsmStatus vsm_netdev_up(VsmClient client, + const char* zone, + const char* netdevId); + +/** + * Turn down a network device in the zone + * + * @param[in] client vasum-server's client + * @param[in] zone zone name + * @param[in] netdevId netdev id + * @return status of this function call + */ +VsmStatus vsm_netdev_down(VsmClient client, + const char* zone, + const char* netdevId); + + +/** * Create veth netdev in zone * * @param[in] client vasum-server's client diff --git a/common/netlink/netlink-message.cpp b/common/netlink/netlink-message.cpp index 72d3597..9f33450 100644 --- a/common/netlink/netlink-message.cpp +++ b/common/netlink/netlink-message.cpp @@ -42,12 +42,16 @@ #define PAGE_SIZE 4096 #endif +namespace { + const int NLMSG_GOOD_SIZE = 2*PAGE_SIZE; -constexpr rtattr* NLMSG_TAIL(nlmsghdr* nmsg) +inline rtattr* NLMSG_TAIL(nlmsghdr* nmsg) { return reinterpret_cast(reinterpret_cast(nmsg) + NLMSG_ALIGN(nmsg->nlmsg_len)); } +} // namespace + namespace vasum { namespace netlink { diff --git a/server/host-connection.cpp b/server/host-connection.cpp index 75c4433..6790480 100644 --- a/server/host-connection.cpp +++ b/server/host-connection.cpp @@ -130,6 +130,21 @@ void HostConnection::setGetZoneInfoCallback(const GetZoneInfoCallback& callback) mGetZoneInfoCallback = callback; } +void HostConnection::setSetNetdevAttrsCallback(const SetNetdevAttrsCallback& callback) +{ + mSetNetdevAttrsCallback = callback; +} + +void HostConnection::setGetNetdevAttrsCallback(const GetNetdevAttrsCallback& callback) +{ + mGetNetdevAttrsCallback = callback; +} + +void HostConnection::setGetNetdevListCallback(const GetNetdevListCallback& callback) +{ + mGetNetdevListCallback = callback; +} + void HostConnection::setCreateNetdevVethCallback(const CreateNetdevVethCallback& callback) { mCreateNetdevVethCallback = callback; @@ -296,6 +311,44 @@ void HostConnection::onMessageCall(const std::string& objectPath, return; } + if (methodName == api::host::METHOD_SET_NETDEV_ATTRS){ + const gchar* zone = NULL; + const gchar* netdev = NULL; + GVariantIter* iter; + g_variant_get(parameters, "(&s&sa(ss))", &zone, &netdev, &iter); + gchar* key = NULL; + gchar* value = NULL; + std::vector> attrs; + while (g_variant_iter_loop(iter, "(&s&s)", &key, &value)) { + attrs.push_back(std::make_tuple(key, value)); + } + g_variant_iter_free(iter); + if (mSetNetdevAttrsCallback) { + mSetNetdevAttrsCallback(zone, netdev, attrs, result); + } + return; + } + + if (methodName == api::host::METHOD_GET_NETDEV_ATTRS){ + const gchar* zone = NULL; + const gchar* netdev = NULL; + g_variant_get(parameters, "(&s&s)", &zone, &netdev); + if (mGetNetdevAttrsCallback) { + mGetNetdevAttrsCallback(zone, netdev, result); + } + return; + } + + if (methodName == api::host::METHOD_GET_NETDEV_LIST){ + const gchar* zone = NULL; + g_variant_get(parameters, "(&s)", &zone); + if (mGetNetdevListCallback) { + mGetNetdevListCallback(zone, result); + } + return; + } + + if (methodName == api::host::METHOD_CREATE_NETDEV_VETH) { const gchar* id = NULL; const gchar* zoneDev = NULL; diff --git a/server/host-connection.hpp b/server/host-connection.hpp index 649ed15..569c68f 100644 --- a/server/host-connection.hpp +++ b/server/host-connection.hpp @@ -30,6 +30,8 @@ #include #include +#include +#include namespace vasum { @@ -60,8 +62,20 @@ public: typedef std::function GetZoneInfoCallback; - typedef std::function>& attrs, + dbus::MethodResultBuilder::Pointer result + )> SetNetdevAttrsCallback; + typedef std::function GetNetdevAttrsCallback; + typedef std::function GetNetdevListCallback; + typedef std::function CreateNetdevVethCallback; @@ -165,6 +179,21 @@ public: void setGetZoneInfoCallback(const GetZoneInfoCallback& callback); /** + * Register a callback called to set network device attributes + */ + void setSetNetdevAttrsCallback(const SetNetdevAttrsCallback& callback); + + /** + * Register a callback called to get network device attributes + */ + void setGetNetdevAttrsCallback(const GetNetdevAttrsCallback& callback); + + /** + * Register a callback called to get network device list + */ + void setGetNetdevListCallback(const GetNetdevListCallback& callback); + + /** * Register a callback called to create veth */ void setCreateNetdevVethCallback(const CreateNetdevVethCallback& callback); @@ -270,6 +299,9 @@ private: GetZoneIdsCallback mGetZoneIdsCallback; GetActiveZoneIdCallback mGetActiveZoneIdCallback; GetZoneInfoCallback mGetZoneInfoCallback; + SetNetdevAttrsCallback mSetNetdevAttrsCallback; + GetNetdevAttrsCallback mGetNetdevAttrsCallback; + GetNetdevListCallback mGetNetdevListCallback; CreateNetdevVethCallback mCreateNetdevVethCallback; CreateNetdevMacvlanCallback mCreateNetdevMacvlanCallback; CreateNetdevPhysCallback mCreateNetdevPhysCallback; diff --git a/server/host-dbus-definitions.hpp b/server/host-dbus-definitions.hpp index 6d25bb1..2c2a201 100644 --- a/server/host-dbus-definitions.hpp +++ b/server/host-dbus-definitions.hpp @@ -42,6 +42,9 @@ const std::string METHOD_GET_ZONE_DBUSES = "GetZoneDbuses"; const std::string METHOD_GET_ZONE_ID_LIST = "GetZoneIds"; const std::string METHOD_GET_ACTIVE_ZONE_ID = "GetActiveZoneId"; const std::string METHOD_GET_ZONE_INFO = "GetZoneInfo"; +const std::string METHOD_SET_NETDEV_ATTRS = "SetNetdevAttrs"; +const std::string METHOD_GET_NETDEV_ATTRS = "GetNetdevAttrs"; +const std::string METHOD_GET_NETDEV_LIST = "GetNetdevList"; const std::string METHOD_CREATE_NETDEV_VETH = "CreateNetdevVeth"; const std::string METHOD_CREATE_NETDEV_MACVLAN = "CreateNetdevMacvlan"; const std::string METHOD_CREATE_NETDEV_PHYS = "CreateNetdevPhys"; @@ -88,6 +91,20 @@ const std::string DEFINITION = " " " " " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " " " " " " " diff --git a/server/zone-admin.cpp b/server/zone-admin.cpp index 5a4adfa..519b172 100644 --- a/server/zone-admin.cpp +++ b/server/zone-admin.cpp @@ -289,4 +289,19 @@ void ZoneAdmin::moveNetdev(const std::string& devId) netdev::movePhys(mZone.getInitPid(), devId); } +void ZoneAdmin::setNetdevAttrs(const std::string& /* netdev */, const NetdevAttrs& /* attrs */) +{ + throw ZoneOperationException("Not implemented"); +} + +ZoneAdmin::NetdevAttrs ZoneAdmin::getNetdevAttrs(const std::string& /* netdev */) +{ + throw ZoneOperationException("Not implemented"); +} + +std::vector ZoneAdmin::getNetdevList() +{ + throw ZoneOperationException("Not implemented"); +} + } // namespace vasum diff --git a/server/zone-admin.hpp b/server/zone-admin.hpp index 9125a6d..44d2613 100644 --- a/server/zone-admin.hpp +++ b/server/zone-admin.hpp @@ -41,6 +41,7 @@ enum class SchedulerLevel { class ZoneAdmin { public: + typedef std::vector> NetdevAttrs; /** * ZoneAdmin constructor @@ -148,6 +149,21 @@ public: */ void moveNetdev(const std::string& devId); + /** + * Set network device attributes + */ + void setNetdevAttrs(const std::string& netdev, const NetdevAttrs& attrs); + + /** + * Get network device attributes + */ + NetdevAttrs getNetdevAttrs(const std::string& netdev); + + /** + * Get network device list + */ + std::vector getNetdevList(); + private: const ZoneConfig& mConfig; lxc::LxcZone mZone; diff --git a/server/zone.cpp b/server/zone.cpp index 0607fda..fed66fa 100644 --- a/server/zone.cpp +++ b/server/zone.cpp @@ -441,4 +441,22 @@ void Zone::removeDeclaration(const std::string& declarationId) mProvision->remove(declarationId); } +void Zone::setNetdevAttrs(const std::string& netdev, const ZoneAdmin::NetdevAttrs& attrs) +{ + Lock lock(mReconnectMutex); + mAdmin->setNetdevAttrs(netdev, attrs); +} + +ZoneAdmin::NetdevAttrs Zone::getNetdevAttrs(const std::string& netdev) +{ + Lock lock(mReconnectMutex); + return mAdmin->getNetdevAttrs(netdev); +} + +std::vector Zone::getNetdevList() +{ + Lock lock(mReconnectMutex); + return mAdmin->getNetdevList(); +} + } // namespace vasum diff --git a/server/zone.hpp b/server/zone.hpp index 759faa3..832fcec 100644 --- a/server/zone.hpp +++ b/server/zone.hpp @@ -285,6 +285,21 @@ public: */ void moveNetdev(const std::string& devId); + /** + * Set network device attributes + */ + void setNetdevAttrs(const std::string& netdev, const ZoneAdmin::NetdevAttrs& attrs); + + /** + * Get network device attributes + */ + ZoneAdmin::NetdevAttrs getNetdevAttrs(const std::string& netdev); + + /** + * Get network device list + */ + std::vector getNetdevList(); + private: utils::Worker::Pointer mWorker; ZoneConfig mConfig; diff --git a/server/zones-manager.cpp b/server/zones-manager.cpp index 5f0dca4..283f57f 100644 --- a/server/zones-manager.cpp +++ b/server/zones-manager.cpp @@ -141,6 +141,15 @@ ZonesManager::ZonesManager(const std::string& configPath) mHostConnection.setGetZoneInfoCallback(bind(&ZonesManager::handleGetZoneInfoCall, this, _1, _2)); + mHostConnection.setSetNetdevAttrsCallback(bind(&ZonesManager::handleSetNetdevAttrsCall, + this, _1, _2, _3, _4)); + + mHostConnection.setGetNetdevAttrsCallback(bind(&ZonesManager::handleGetNetdevAttrsCall, + this, _1, _2, _3)); + + mHostConnection.setGetNetdevListCallback(bind(&ZonesManager::handleGetNetdevListCall, + this, _1, _2)); + mHostConnection.setCreateNetdevVethCallback(bind(&ZonesManager::handleCreateNetdevVethCall, this, _1, _2, _3, _4)); @@ -803,6 +812,78 @@ void ZonesManager::handleGetZoneInfoCall(const std::string& id, zone.getRootPath().c_str())); } +void ZonesManager::handleSetNetdevAttrsCall(const std::string& zone, + const std::string& netdev, + const std::vector< + std::tuple>& attrs, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("SetNetdevAttrs call"); + try { + Lock lock(mMutex); + + getZone(zone).setNetdevAttrs(netdev, attrs); + result->setVoid(); + } catch (const InvalidZoneIdException&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't set attributes: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + +void ZonesManager::handleGetNetdevAttrsCall(const std::string& zone, + const std::string& netdev, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("GetNetdevAttrs call"); + try { + Lock lock(mMutex); + + const auto attrs = getZone(zone).getNetdevAttrs(netdev); + + GVariantBuilder builder; + g_variant_builder_init(&builder, G_VARIANT_TYPE_ARRAY); + for (const auto entry : attrs) { + g_variant_builder_add(&builder, + "(ss)", + std::get<0>(entry).c_str(), + std::get<1>(entry).c_str()); + } + result->set(g_variant_builder_end(&builder)); + } catch (const InvalidZoneIdException&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't set attributes: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + +void ZonesManager::handleGetNetdevListCall(const std::string& zone, + dbus::MethodResultBuilder::Pointer result) +{ + LOGI("GetNetdevList call"); + try { + Lock lock(mMutex); + std::vector netdevs; + for(auto& netdev: getZone(zone).getNetdevList()){ + netdevs.push_back(g_variant_new_string(netdev.c_str())); + } + GVariant* array = g_variant_new_array(G_VARIANT_TYPE("s"), + netdevs.data(), + netdevs.size()); + result->set(g_variant_new("(@as)", array)); + } catch (const InvalidZoneIdException&) { + LOGE("No zone with id=" << zone); + result->setError(api::ERROR_INVALID_ID, "No such zone id"); + } catch (const VasumException& ex) { + LOGE("Can't set attributes: " << ex.what()); + result->setError(api::ERROR_INTERNAL, ex.what()); + } +} + void ZonesManager::handleCreateNetdevVethCall(const std::string& zone, const std::string& zoneDev, const std::string& hostDev, @@ -814,7 +895,7 @@ void ZonesManager::handleCreateNetdevVethCall(const std::string& zone, getZone(zone).createNetdevVeth(zoneDev, hostDev); result->setVoid(); - } catch (const std::out_of_range&) { + } catch (const InvalidZoneIdException&) { LOGE("No zone with id=" << zone); result->setError(api::ERROR_INVALID_ID, "No such zone id"); } catch (const VasumException& ex) { @@ -835,7 +916,7 @@ void ZonesManager::handleCreateNetdevMacvlanCall(const std::string& zone, getZone(zone).createNetdevMacvlan(zoneDev, hostDev, mode); result->setVoid(); - } catch (const std::out_of_range&) { + } catch (const InvalidZoneIdException&) { LOGE("No zone with id=" << zone); result->setError(api::ERROR_INVALID_ID, "No such zone id"); } catch (const VasumException& ex) { @@ -854,7 +935,7 @@ void ZonesManager::handleCreateNetdevPhysCall(const std::string& zone, getZone(zone).moveNetdev(devId); result->setVoid(); - } catch (const std::out_of_range&) { + } catch (const InvalidZoneIdException&) { LOGE("No zone with id=" << zone); result->setError(api::ERROR_INVALID_ID, "No such zone id"); } catch (const VasumException& ex) { diff --git a/server/zones-manager.hpp b/server/zones-manager.hpp index df5e12b..ee1606d 100644 --- a/server/zones-manager.hpp +++ b/server/zones-manager.hpp @@ -161,6 +161,15 @@ private: void handleGetZoneIdsCall(dbus::MethodResultBuilder::Pointer result); void handleGetActiveZoneIdCall(dbus::MethodResultBuilder::Pointer result); void handleGetZoneInfoCall(const std::string& id, dbus::MethodResultBuilder::Pointer result); + void handleSetNetdevAttrsCall(const std::string& zone, + const std::string& netdev, + const std::vector>& attrs, + dbus::MethodResultBuilder::Pointer result); + void handleGetNetdevAttrsCall(const std::string& zone, + const std::string& netdev, + dbus::MethodResultBuilder::Pointer result); + void handleGetNetdevListCall(const std::string& zone, + dbus::MethodResultBuilder::Pointer result); void handleCreateNetdevVethCall(const std::string& zone, const std::string& zoneDev, const std::string& hostDev, -- 2.7.4 From 0ca217568de95a695483c92db178355a739a77b9 Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Mon, 2 Mar 2015 11:57:50 +0100 Subject: [PATCH 10/16] Persistent state of zone [Bug/Feature] Zone was always started after vasum restart regardless of the zone state before restart [Cause] N/A [Solution] N/A [Verification] Run tests. Run vasum, pause one container, restart vasum, check this container is still paused. The same test with stopped and started state. Change-Id: I0426e3e4f2af9b2bc0c110a7702b9c742a2747e6 --- server/configs/templates/default.conf | 1 + server/server.cpp | 4 +- server/zone-config.hpp | 6 ++ server/zone.cpp | 52 ++++++++- server/zone.hpp | 11 +- server/zones-manager.cpp | 37 +++---- server/zones-manager.hpp | 13 ++- .../ut-client/templates/console-dbus.conf.in | 1 + tests/unit_tests/client/ut-client.cpp | 4 +- .../configs/ut-server/templates/default.conf | 1 + .../configs/ut-zone-admin/templates/buggy.conf | 1 + .../configs/ut-zone-admin/templates/missing.conf | 1 + .../ut-zone-admin/templates/test-no-shutdown.conf | 1 + .../configs/ut-zone-admin/templates/test.conf | 1 + .../server/configs/ut-zone/templates/buggy.conf | 1 + .../configs/ut-zone/templates/test-dbus.conf.in | 1 + .../server/configs/ut-zone/templates/test.conf | 1 + .../templates/console-dbus.conf.in | 1 + .../ut-zones-manager/templates/console.conf | 1 + tests/unit_tests/server/ut-server.cpp | 2 +- tests/unit_tests/server/ut-zone-admin.cpp | 4 +- tests/unit_tests/server/ut-zone.cpp | 4 +- tests/unit_tests/server/ut-zones-manager.cpp | 119 ++++++++++++++++----- 23 files changed, 206 insertions(+), 62 deletions(-) diff --git a/server/configs/templates/default.conf b/server/configs/templates/default.conf index 2aa7a66..9eb0e08 100644 --- a/server/configs/templates/default.conf +++ b/server/configs/templates/default.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "tizen-common-wayland.sh", "initWithArgs" : [], + "requestedState" : "stopped", "ipv4Gateway" : "10.0.~IP~.1", "ipv4" : "10.0.~IP~.2", "cpuQuotaForeground" : -1, diff --git a/server/server.cpp b/server/server.cpp index b4e2344..c2ca778 100644 --- a/server/server.cpp +++ b/server/server.cpp @@ -107,7 +107,7 @@ void Server::run(bool asRoot) utils::ScopedGlibLoop loop; ZonesManager manager(mConfigPath); - manager.startAll(); + manager.restoreAll(); LOGI("Daemon started"); gSignalLatch.wait(); @@ -118,7 +118,7 @@ void Server::run(bool asRoot) } LOGI("Stopping daemon..."); - // manager.stopAll() will be called in destructor + // manager.shutdownAll() will be called in destructor } LOGI("Daemon stopped"); } diff --git a/server/zone-config.hpp b/server/zone-config.hpp index da4b570..f5d81e5 100644 --- a/server/zone-config.hpp +++ b/server/zone-config.hpp @@ -110,6 +110,11 @@ struct ZoneConfig { struct ZoneDynamicConfig { /** + * Requested zone state after restore + */ + std::string requestedState; + + /** * IP v4 gateway address */ std::string ipv4Gateway; @@ -131,6 +136,7 @@ struct ZoneDynamicConfig { CONFIG_REGISTER ( + requestedState, ipv4Gateway, ipv4, vt, diff --git a/server/zone.cpp b/server/zone.cpp index fed66fa..1bc2d09 100644 --- a/server/zone.cpp +++ b/server/zone.cpp @@ -50,6 +50,10 @@ typedef std::lock_guard Lock; const int RECONNECT_RETRIES = 15; const int RECONNECT_DELAY = 1 * 1000; +const std::string STATE_STOPPED = "stopped"; +const std::string STATE_RUNNING = "running"; +const std::string STATE_PAUSED = "paused"; + } // namespace Zone::Zone(const utils::Worker::Pointer& worker, @@ -60,6 +64,7 @@ Zone::Zone(const utils::Worker::Pointer& worker, const std::string& lxcTemplatePrefix, const std::string& baseRunMountPointPath) : mWorker(worker) + , mDbPath(dbPath) { const std::string dbPrefix = getZoneDbPrefix(zoneId); config::loadFromKVStoreWithJsonFile(dbPath, zoneTemplatePath, mConfig, dbPrefix); @@ -118,9 +123,46 @@ int Zone::getPrivilege() const return mConfig.privilege; } +void Zone::saveDynamicConfig() +{ + config::saveToKVStore(mDbPath, mDynamicConfig, getZoneDbPrefix(getId())); +} + +void Zone::updateRequestedState(const std::string& state) +{ + // assume mutex is locked + if (state != mDynamicConfig.requestedState) { + LOGT("Set requested state of " << getId() << " to " << state); + mDynamicConfig.requestedState = state; + saveDynamicConfig(); + } +} + +void Zone::restore() +{ + std::string requestedState; + { + Lock lock(mReconnectMutex); + requestedState = mDynamicConfig.requestedState; + LOGT("Requested state of " << getId() << ": " << requestedState); + } + + if (requestedState == STATE_RUNNING) { + start(); + } else if (requestedState == STATE_STOPPED) { + } else if (requestedState == STATE_PAUSED) { + start(); + suspend(); + } else { + LOGE("Invalid requested state: " << requestedState); + assert(0 && "invalid requested state"); + } +} + void Zone::start() { Lock lock(mReconnectMutex); + updateRequestedState(STATE_RUNNING); mProvision->start(); if (mConfig.enableDbusIntegration) { mConnectionTransport.reset(new ZoneConnectionTransport(mRunMountPoint)); @@ -135,9 +177,12 @@ void Zone::start() // refocus in ZonesManager will adjust cpu quota after all } -void Zone::stop() +void Zone::stop(bool saveState) { Lock lock(mReconnectMutex); + if (saveState) { + updateRequestedState(STATE_STOPPED); + } if (mAdmin->isRunning()) { // boost stopping goForeground(); @@ -192,6 +237,7 @@ std::string Zone::getDbusAddress() const int Zone::getVT() const { + Lock lock(mReconnectMutex); return mDynamicConfig.vt; } @@ -275,12 +321,14 @@ void Zone::suspend() { Lock lock(mReconnectMutex); mAdmin->suspend(); + updateRequestedState(STATE_PAUSED); } void Zone::resume() { Lock lock(mReconnectMutex); mAdmin->resume(); + updateRequestedState(STATE_RUNNING); } bool Zone::isPaused() @@ -329,7 +377,7 @@ void Zone::reconnectHandler() } LOGE(getId() << ": Reconnecting to the DBUS has failed, stopping the zone"); - stop(); + stop(false); } void Zone::setNotifyActiveZoneCallback(const NotifyActiveZoneCallback& callback) diff --git a/server/zone.hpp b/server/zone.hpp index 832fcec..69d0324 100644 --- a/server/zone.hpp +++ b/server/zone.hpp @@ -98,14 +98,20 @@ public: int getPrivilege() const; /** + * Restore zone to the previous state + */ + void restore(); + + /** * Boot the zone to the background. */ void start(); /** * Try to shutdown the zone, if failed, destroy it. + * @param saveState save zone's state */ - void stop(); + void stop(bool saveState); /** * Activate this zone's VT @@ -319,11 +325,14 @@ private: std::string mDbusAddress; std::string mRunMountPoint; std::string mRootPath; + std::string mDbPath; void onNameLostCallback(); void reconnectHandler(); void connect(); void disconnect(); + void saveDynamicConfig(); + void updateRequestedState(const std::string& state); }; diff --git a/server/zones-manager.cpp b/server/zones-manager.cpp index 283f57f..d90bab7 100644 --- a/server/zones-manager.cpp +++ b/server/zones-manager.cpp @@ -226,9 +226,9 @@ ZonesManager::~ZonesManager() if (!mDetachOnExit) { try { - stopAll(); + shutdownAll(); } catch (ServerException&) { - LOGE("Failed to stop all of the zones"); + LOGE("Failed to shutdown all of the zones"); } } // wait for all tasks to complete @@ -434,27 +434,27 @@ void ZonesManager::refocus() focusInternal(iter); } -void ZonesManager::startAll() +void ZonesManager::restoreAll() { - LOGI("Starting all zones"); + LOGI("Restoring all zones"); Lock lock(mMutex); for (auto& zone : mZones) { - zone->start(); + zone->restore(); } refocus(); } -void ZonesManager::stopAll() +void ZonesManager::shutdownAll() { LOGI("Stopping all zones"); Lock lock(mMutex); for (auto& zone : mZones) { - zone->stop(); + zone->stop(false); } refocus(); @@ -463,26 +463,19 @@ void ZonesManager::stopAll() bool ZonesManager::isPaused(const std::string& zoneId) { Lock lock(mMutex); - - auto iter = findZone(zoneId); - if (iter == mZones.end()) { - LOGE("No such zone id: " << zoneId); - throw InvalidZoneIdException("No such zone"); - } - - return get(iter).isPaused(); + return getZone(zoneId).isPaused(); } bool ZonesManager::isRunning(const std::string& zoneId) { Lock lock(mMutex); + return getZone(zoneId).isRunning(); +} - auto iter = findZone(zoneId); - if (iter == mZones.end()) { - LOGE("No such zone id: " << zoneId); - throw InvalidZoneIdException("No such zone"); - } - return get(iter).isRunning(); +bool ZonesManager::isStopped(const std::string& zoneId) +{ + Lock lock(mMutex); + return getZone(zoneId).isStopped(); } std::string ZonesManager::getRunningForegroundZoneId() @@ -1265,7 +1258,7 @@ void ZonesManager::handleShutdownZoneCall(const std::string& id, result->setError(api::ERROR_INVALID_ID, "No such zone id"); return; } - get(iter).stop(); + get(iter).stop(true); refocus(); result->setVoid(); } catch (ZoneOperationException& e) { diff --git a/server/zones-manager.hpp b/server/zones-manager.hpp index ee1606d..cb730e3 100644 --- a/server/zones-manager.hpp +++ b/server/zones-manager.hpp @@ -70,14 +70,14 @@ public: void focus(const std::string& zoneId); /** - * Start up all the configured zones + * Restore all the configured zones to the saved state */ - void startAll(); + void restoreAll(); /** - * Stop all managed zones + * Shutdown all managed zones without changing the saved state */ - void stopAll(); + void shutdownAll(); /** * @return Is the zone in a paused state? @@ -90,6 +90,11 @@ public: bool isRunning(const std::string& zoneId); /** + * @return Is the zone stopped? + */ + bool isStopped(const std::string& zoneId); + + /** * @return id of the currently focused/foreground zone */ std::string getRunningForegroundZoneId(); diff --git a/tests/unit_tests/client/configs/ut-client/templates/console-dbus.conf.in b/tests/unit_tests/client/configs/ut-client/templates/console-dbus.conf.in index 0bd0e03..7778208 100644 --- a/tests/unit_tests/client/configs/ut-client/templates/console-dbus.conf.in +++ b/tests/unit_tests/client/configs/ut-client/templates/console-dbus.conf.in @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal-dbus.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; /usr/bin/dbus-daemon --config-file=@VSM_TEST_CONFIG_INSTALL_DIR@/server/ut-zones-manager/ut-dbus.conf --fork; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 20, diff --git a/tests/unit_tests/client/ut-client.cpp b/tests/unit_tests/client/ut-client.cpp index 1c55668..ce31795 100644 --- a/tests/unit_tests/client/ut-client.cpp +++ b/tests/unit_tests/client/ut-client.cpp @@ -77,7 +77,7 @@ struct Fixture { cm.createZone("zone1", TEMPLATE_NAME); cm.createZone("zone2", TEMPLATE_NAME); cm.createZone("zone3", TEMPLATE_NAME); - cm.startAll(); + cm.restoreAll(); LOGI("------- setup complete --------"); } @@ -153,7 +153,7 @@ BOOST_FIXTURE_TEST_SUITE(ClientSuite, Fixture) BOOST_AUTO_TEST_CASE(NotRunningServerTest) { - cm.stopAll(); + cm.shutdownAll(); VsmClient client = vsm_client_create(); VsmStatus status = vsm_connect_custom(client, diff --git a/tests/unit_tests/server/configs/ut-server/templates/default.conf b/tests/unit_tests/server/configs/ut-server/templates/default.conf index 875cdc4..a2f703e 100644 --- a/tests/unit_tests/server/configs/ut-server/templates/default.conf +++ b/tests/unit_tests/server/configs/ut-server/templates/default.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 20, diff --git a/tests/unit_tests/server/configs/ut-zone-admin/templates/buggy.conf b/tests/unit_tests/server/configs/ut-zone-admin/templates/buggy.conf index 07569fe..f585ae8 100644 --- a/tests/unit_tests/server/configs/ut-zone-admin/templates/buggy.conf +++ b/tests/unit_tests/server/configs/ut-zone-admin/templates/buggy.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/foo"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone-admin/templates/missing.conf b/tests/unit_tests/server/configs/ut-zone-admin/templates/missing.conf index e0b5e4a..a3f305b 100644 --- a/tests/unit_tests/server/configs/ut-zone-admin/templates/missing.conf +++ b/tests/unit_tests/server/configs/ut-zone-admin/templates/missing.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "missing.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone-admin/templates/test-no-shutdown.conf b/tests/unit_tests/server/configs/ut-zone-admin/templates/test-no-shutdown.conf index 1579851..98e95ab 100644 --- a/tests/unit_tests/server/configs/ut-zone-admin/templates/test-no-shutdown.conf +++ b/tests/unit_tests/server/configs/ut-zone-admin/templates/test-no-shutdown.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/bin/sh"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone-admin/templates/test.conf b/tests/unit_tests/server/configs/ut-zone-admin/templates/test.conf index d74f221..a16b3fa 100644 --- a/tests/unit_tests/server/configs/ut-zone-admin/templates/test.conf +++ b/tests/unit_tests/server/configs/ut-zone-admin/templates/test.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone/templates/buggy.conf b/tests/unit_tests/server/configs/ut-zone/templates/buggy.conf index 1b449bc..79436c0 100644 --- a/tests/unit_tests/server/configs/ut-zone/templates/buggy.conf +++ b/tests/unit_tests/server/configs/ut-zone/templates/buggy.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "/buggy/path", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone/templates/test-dbus.conf.in b/tests/unit_tests/server/configs/ut-zone/templates/test-dbus.conf.in index 9e9a1f8..5d76429 100644 --- a/tests/unit_tests/server/configs/ut-zone/templates/test-dbus.conf.in +++ b/tests/unit_tests/server/configs/ut-zone/templates/test-dbus.conf.in @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal-dbus.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; /usr/bin/dbus-daemon --config-file=@VSM_TEST_CONFIG_INSTALL_DIR@/server/ut-zone/ut-dbus.conf --fork; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zone/templates/test.conf b/tests/unit_tests/server/configs/ut-zone/templates/test.conf index 3bc3277..168aa21 100644 --- a/tests/unit_tests/server/configs/ut-zone/templates/test.conf +++ b/tests/unit_tests/server/configs/ut-zone/templates/test.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 10, diff --git a/tests/unit_tests/server/configs/ut-zones-manager/templates/console-dbus.conf.in b/tests/unit_tests/server/configs/ut-zones-manager/templates/console-dbus.conf.in index 0bd0e03..7778208 100644 --- a/tests/unit_tests/server/configs/ut-zones-manager/templates/console-dbus.conf.in +++ b/tests/unit_tests/server/configs/ut-zones-manager/templates/console-dbus.conf.in @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal-dbus.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; /usr/bin/dbus-daemon --config-file=@VSM_TEST_CONFIG_INSTALL_DIR@/server/ut-zones-manager/ut-dbus.conf --fork; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 20, diff --git a/tests/unit_tests/server/configs/ut-zones-manager/templates/console.conf b/tests/unit_tests/server/configs/ut-zones-manager/templates/console.conf index 9b3013c..6e49c5b 100644 --- a/tests/unit_tests/server/configs/ut-zones-manager/templates/console.conf +++ b/tests/unit_tests/server/configs/ut-zones-manager/templates/console.conf @@ -1,6 +1,7 @@ { "lxcTemplate" : "minimal.sh", "initWithArgs" : ["/bin/sh", "-c", "trap exit SIGTERM; read"], + "requestedState" : "running", "ipv4Gateway" : "", "ipv4" : "", "privilege" : 20, diff --git a/tests/unit_tests/server/ut-server.cpp b/tests/unit_tests/server/ut-server.cpp index 4df5d89..eccea8a 100644 --- a/tests/unit_tests/server/ut-server.cpp +++ b/tests/unit_tests/server/ut-server.cpp @@ -63,7 +63,7 @@ struct Fixture { vasum::ZonesManager manager(TEST_CONFIG_PATH); manager.createZone("zone1", TEMPLATE_NAME); manager.createZone("zone2", TEMPLATE_NAME); - manager.startAll(); + manager.restoreAll(); } }; } // namespace diff --git a/tests/unit_tests/server/ut-zone-admin.cpp b/tests/unit_tests/server/ut-zone-admin.cpp index 4dd08db..c711169 100644 --- a/tests/unit_tests/server/ut-zone-admin.cpp +++ b/tests/unit_tests/server/ut-zone-admin.cpp @@ -158,10 +158,10 @@ BOOST_AUTO_TEST_CASE(SuspendResumeTest) BOOST_AUTO_TEST_CASE(SchedulerLevelTest) { - BOOST_REQUIRE(mConfig.cpuQuotaForeground != mConfig.cpuQuotaBackground); - auto admin = create(TEST_CONFIG_PATH); + BOOST_REQUIRE(mConfig.cpuQuotaForeground != mConfig.cpuQuotaBackground); + admin->start(); ensureStarted(); diff --git a/tests/unit_tests/server/ut-zone.cpp b/tests/unit_tests/server/ut-zone.cpp index 3cd6031..37f65be 100644 --- a/tests/unit_tests/server/ut-zone.cpp +++ b/tests/unit_tests/server/ut-zone.cpp @@ -111,7 +111,7 @@ BOOST_AUTO_TEST_CASE(StartStopTest) auto c = create(TEST_CONFIG_PATH); c->start(); ensureStarted(); - c->stop(); + c->stop(true); } BOOST_AUTO_TEST_CASE(DbusConnectionTest) @@ -121,7 +121,7 @@ BOOST_AUTO_TEST_CASE(DbusConnectionTest) auto c = create(TEST_DBUS_CONFIG_PATH); c->start(); ensureStarted(); - c->stop(); + c->stop(true); } // TODO: DbusReconnectionTest diff --git a/tests/unit_tests/server/ut-zones-manager.cpp b/tests/unit_tests/server/ut-zones-manager.cpp index e93ab84..0e40093 100644 --- a/tests/unit_tests/server/ut-zones-manager.cpp +++ b/tests/unit_tests/server/ut-zones-manager.cpp @@ -508,9 +508,9 @@ BOOST_AUTO_TEST_CASE(StartStopTest) cm.createZone("zone1", SIMPLE_TEMPLATE); cm.createZone("zone2", SIMPLE_TEMPLATE); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), "zone1"); - cm.stopAll(); + cm.shutdownAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), ""); } @@ -520,13 +520,13 @@ BOOST_AUTO_TEST_CASE(DetachOnExitTest) ZonesManager cm(TEST_CONFIG_PATH); cm.createZone("zone1", SIMPLE_TEMPLATE); cm.createZone("zone2", SIMPLE_TEMPLATE); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), "zone1"); cm.setZonesDetachOnExit(); } { ZonesManager cm(TEST_CONFIG_PATH); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), "zone1"); } } @@ -537,7 +537,7 @@ BOOST_AUTO_TEST_CASE(FocusTest) cm.createZone("zone1", SIMPLE_TEMPLATE); cm.createZone("zone2", SIMPLE_TEMPLATE); cm.createZone("zone3", SIMPLE_TEMPLATE); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK(cm.getRunningForegroundZoneId() == "zone1"); cm.focus("zone2"); @@ -554,7 +554,7 @@ BOOST_AUTO_TEST_CASE(NotifyActiveZoneTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); Latch signalReceivedLatch; std::map> signalReceivedSourcesMap; @@ -624,7 +624,7 @@ BOOST_AUTO_TEST_CASE(DisplayOffTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); std::vector> clients; for (int i = 1; i <= TEST_DBUS_CONNECTION_ZONES_COUNT; ++i) { @@ -661,7 +661,7 @@ BOOST_AUTO_TEST_CASE(MoveFileTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); Latch notificationLatch; std::string notificationSource; @@ -763,7 +763,7 @@ BOOST_AUTO_TEST_CASE(AllowSwitchToDefaultTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); std::vector> clients; for (int i = 1; i <= TEST_DBUS_CONNECTION_ZONES_COUNT; ++i) { @@ -812,7 +812,7 @@ BOOST_AUTO_TEST_CASE(ProxyCallTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); std::map> dbuses; for (int i = 0; i <= TEST_DBUS_CONNECTION_ZONES_COUNT; ++i) { @@ -904,9 +904,9 @@ BOOST_AUTO_TEST_CASE(GetZoneDbusesTest) cm.createZone("zone3", DBUS_TEMPLATE); BOOST_CHECK(EXPECTED_DBUSES_NONE == host.callMethodGetZoneDbuses()); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK(EXPECTED_DBUSES_ALL == host.callMethodGetZoneDbuses()); - cm.stopAll(); + cm.shutdownAll(); BOOST_CHECK(EXPECTED_DBUSES_NONE == host.callMethodGetZoneDbuses()); } @@ -919,9 +919,9 @@ BOOST_AUTO_TEST_CASE(GetZoneDbusesNoDbusTest) cm.createZone("zone3", SIMPLE_TEMPLATE); BOOST_CHECK(EXPECTED_DBUSES_NONE == host.callMethodGetZoneDbuses()); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK(EXPECTED_DBUSES_NONE == host.callMethodGetZoneDbuses()); - cm.stopAll(); + cm.shutdownAll(); BOOST_CHECK(EXPECTED_DBUSES_NONE == host.callMethodGetZoneDbuses()); } @@ -961,7 +961,7 @@ BOOST_AUTO_TEST_CASE(ZoneDbusesSignalsTest) BOOST_CHECK(signalLatch.empty()); BOOST_CHECK(collectedDbuses.empty()); - cm.startAll(); + cm.restoreAll(); BOOST_REQUIRE(signalLatch.waitForN(TEST_DBUS_CONNECTION_ZONES_COUNT, EVENT_TIMEOUT)); BOOST_CHECK(signalLatch.empty()); @@ -998,7 +998,7 @@ BOOST_AUTO_TEST_CASE(GetActiveZoneIdTest) cm.createZone("zone1", SIMPLE_TEMPLATE); cm.createZone("zone2", SIMPLE_TEMPLATE); cm.createZone("zone3", SIMPLE_TEMPLATE); - cm.startAll(); + cm.restoreAll(); DbusAccessory dbus(DbusAccessory::HOST_ID); @@ -1011,7 +1011,7 @@ BOOST_AUTO_TEST_CASE(GetActiveZoneIdTest) BOOST_CHECK(dbus.callMethodGetActiveZoneId() == zoneId); } - cm.stopAll(); + cm.shutdownAll(); BOOST_CHECK(dbus.callMethodGetActiveZoneId() == ""); } @@ -1021,7 +1021,7 @@ BOOST_AUTO_TEST_CASE(SetActiveZoneTest) cm.createZone("zone1", SIMPLE_TEMPLATE); cm.createZone("zone2", SIMPLE_TEMPLATE); cm.createZone("zone3", SIMPLE_TEMPLATE); - cm.startAll(); + cm.restoreAll(); DbusAccessory dbus(DbusAccessory::HOST_ID); @@ -1038,7 +1038,7 @@ BOOST_AUTO_TEST_CASE(SetActiveZoneTest) DbusException, WhatEquals("No such zone id")); - cm.stopAll(); + cm.shutdownAll(); BOOST_REQUIRE_EXCEPTION(dbus.callMethodSetActiveZone("zone1"), DbusException, WhatEquals("Could not activate stopped or paused zone")); @@ -1051,7 +1051,7 @@ BOOST_AUTO_TEST_CASE(CreateDestroyZoneTest) const std::string zone3 = "test3"; ZonesManager cm(TEST_CONFIG_PATH); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), ""); @@ -1074,7 +1074,7 @@ BOOST_AUTO_TEST_CASE(CreateDestroyZoneTest) dbus.callAsyncMethodCreateZone(zone3, SIMPLE_TEMPLATE, resultCallback); BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); - cm.startAll(); + cm.restoreAll(); BOOST_CHECK_EQUAL(cm.getRunningForegroundZoneId(), zone1); cm.focus(zone3); @@ -1107,7 +1107,7 @@ BOOST_AUTO_TEST_CASE(CreateDestroyZonePersistenceTest) auto getZoneIds = []() -> std::vector { ZonesManager cm(TEST_CONFIG_PATH); - cm.startAll(); + cm.restoreAll(); DbusAccessory dbus(DbusAccessory::HOST_ID); return dbus.callMethodGetZoneIds(); @@ -1140,6 +1140,77 @@ BOOST_AUTO_TEST_CASE(CreateDestroyZonePersistenceTest) BOOST_CHECK(getZoneIds().empty()); } +BOOST_AUTO_TEST_CASE(ZoneStatePersistenceTest) +{ + const std::string zone1 = "zone1"; + const std::string zone2 = "zone2"; + const std::string zone3 = "zone3"; + const std::string zone4 = "zone4"; + const std::string zone5 = "zone5"; + + Latch callDone; + auto resultCallback = [&]() { + callDone.set(); + }; + + // firts run + { + ZonesManager cm(TEST_CONFIG_PATH); + DbusAccessory dbus(DbusAccessory::HOST_ID); + + // zone1 - created + dbus.callAsyncMethodCreateZone(zone1, SIMPLE_TEMPLATE, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + + // zone2 - started + dbus.callAsyncMethodCreateZone(zone2, SIMPLE_TEMPLATE, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodStartZone(zone2, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + BOOST_CHECK(cm.isRunning(zone2)); + + // zone3 - started then paused + dbus.callAsyncMethodCreateZone(zone3, SIMPLE_TEMPLATE, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodStartZone(zone3, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callMethodLockZone(zone3); + BOOST_CHECK(cm.isPaused(zone3)); + + // zone4 - started then stopped + dbus.callAsyncMethodCreateZone(zone4, SIMPLE_TEMPLATE, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodStartZone(zone4, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodShutdownZone(zone4, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + BOOST_CHECK(cm.isStopped(zone4)); + + // zone5 - started then stopped then started + dbus.callAsyncMethodCreateZone(zone5, SIMPLE_TEMPLATE, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodStartZone(zone5, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodShutdownZone(zone5, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + dbus.callAsyncMethodStartZone(zone5, resultCallback); + BOOST_REQUIRE(callDone.wait(EVENT_TIMEOUT)); + BOOST_CHECK(cm.isRunning(zone5)); + } + + // second run + { + ZonesManager cm(TEST_CONFIG_PATH); + cm.restoreAll(); + + BOOST_CHECK(cm.isRunning(zone1)); // because the default json value + BOOST_CHECK(cm.isRunning(zone2)); + BOOST_CHECK(cm.isPaused(zone3)); + BOOST_CHECK(cm.isStopped(zone4)); + BOOST_CHECK(cm.isRunning(zone5)); + } +} + BOOST_AUTO_TEST_CASE(StartShutdownZoneTest) { const std::string zone1 = "zone1"; @@ -1186,7 +1257,7 @@ BOOST_AUTO_TEST_CASE(LockUnlockZoneTest) cm.createZone("zone1", DBUS_TEMPLATE); cm.createZone("zone2", DBUS_TEMPLATE); cm.createZone("zone3", DBUS_TEMPLATE); - cm.startAll(); + cm.restoreAll(); DbusAccessory dbus(DbusAccessory::HOST_ID); @@ -1208,7 +1279,7 @@ BOOST_AUTO_TEST_CASE(LockUnlockZoneTest) DbusException, WhatEquals("No such zone id")); - cm.stopAll(); + cm.shutdownAll(); BOOST_REQUIRE_EXCEPTION(dbus.callMethodLockZone("zone1"), DbusException, WhatEquals("Zone is not running")); -- 2.7.4 From ff4f6693266208b5e9f5bfc69b256347ccca8d35 Mon Sep 17 00:00:00 2001 From: Jan Olszak Date: Wed, 4 Mar 2015 17:54:51 +0100 Subject: [PATCH 11/16] Coloring compiler's output [Feature] N/A [Cause] N/A [Solution] N/A [Verification] Build Change-Id: Ie417e1b7db193798b1f560bac33475be94f1ad4e --- CMakeLists.txt | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 67702a0..d5cb3e3 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -28,12 +28,24 @@ INCLUDE(FindPkgConfig) ## default CMAKE_INSTALL_* variables ########################################### INCLUDE(GNUInstallDirs) +## Color output if it's possible: +IF (( "${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU" AND NOT (CMAKE_CXX_COMPILER_VERSION VERSION_LESS 4.8.3)) + OR ("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang" )) + + ADD_DEFINITIONS("-fdiagnostics-color=always") +ENDIF() ## Compiler flags, depending on the build type ################################# IF(NOT CMAKE_BUILD_TYPE) SET(CMAKE_BUILD_TYPE "DEBUG") ENDIF(NOT CMAKE_BUILD_TYPE) -MESSAGE(STATUS "Build type: ${CMAKE_BUILD_TYPE}") + +## Print build information ##################################################### +MESSAGE(STATUS "-------------------------------------------------") +MESSAGE(STATUS "Compiler: " ${CMAKE_CXX_COMPILER_ID}) +MESSAGE(STATUS "Compiler version: " ${CMAKE_CXX_COMPILER_VERSION}) +MESSAGE(STATUS "Build type: " ${CMAKE_BUILD_TYPE}) +MESSAGE(STATUS "-------------------------------------------------") # special case for a GCC < 4.7, assume rest is fine IF("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU" AND CMAKE_CXX_COMPILER_VERSION VERSION_LESS 4.7) @@ -111,14 +123,12 @@ ADD_DEFINITIONS(-DDISK_GROUP="${DISK_GROUP}") ADD_DEFINITIONS(-DTTY_GROUP="${TTY_GROUP}") ## Python packages directory ################################################### - IF(NOT DEFINED PYTHON_SITELIB) EXECUTE_PROCESS(COMMAND python -c "from distutils.sysconfig import get_python_lib; import sys; sys.stdout.write(get_python_lib())" OUTPUT_VARIABLE PYTHON_SITELIB) ENDIF(NOT DEFINED PYTHON_SITELIB) - ## Subdirectories ############################################################## SET(COMMON_FOLDER ${PROJECT_SOURCE_DIR}/common) SET(CLIENT_FOLDER ${PROJECT_SOURCE_DIR}/client) -- 2.7.4 From 7396882735e82ef5f034c46286b0499d8f5437e7 Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Mon, 9 Mar 2015 13:34:27 +0100 Subject: [PATCH 12/16] Epoll refactor [Bug/Feature] N/A [Cause] N/A [Solution] N/A [Verification] Run tests Change-Id: Iabb0557aa42ff09d0a2b3f9c36b451cf5fdad10f --- common/{utils => epoll}/event-poll.cpp | 6 +- common/{utils => epoll}/event-poll.hpp | 14 +-- common/epoll/events.cpp | 71 +++++++++++++++ common/epoll/events.hpp | 41 +++++++++ common/{utils => epoll}/glib-poll-dispatcher.cpp | 25 +++--- common/{utils => epoll}/glib-poll-dispatcher.hpp | 17 ++-- common/{utils => epoll}/thread-poll-dispatcher.cpp | 24 ++--- common/{utils => epoll}/thread-poll-dispatcher.hpp | 20 +++-- .../unit_tests/{utils => epoll}/ut-event-poll.cpp | 100 ++++++--------------- tests/unit_tests/ipc/ut-ipc.cpp | 1 - 10 files changed, 199 insertions(+), 120 deletions(-) rename common/{utils => epoll}/event-poll.cpp (98%) rename common/{utils => epoll}/event-poll.hpp (89%) create mode 100644 common/epoll/events.cpp create mode 100644 common/epoll/events.hpp rename common/{utils => epoll}/glib-poll-dispatcher.cpp (70%) rename common/{utils => epoll}/glib-poll-dispatcher.hpp (80%) rename common/{utils => epoll}/thread-poll-dispatcher.cpp (73%) rename common/{utils => epoll}/thread-poll-dispatcher.hpp (77%) rename tests/unit_tests/{utils => epoll}/ut-event-poll.cpp (58%) diff --git a/common/utils/event-poll.cpp b/common/epoll/event-poll.cpp similarity index 98% rename from common/utils/event-poll.cpp rename to common/epoll/event-poll.cpp index b37d727..dc1c1bc 100644 --- a/common/utils/event-poll.cpp +++ b/common/epoll/event-poll.cpp @@ -23,7 +23,7 @@ */ #include "config.hpp" -#include "utils/event-poll.hpp" +#include "epoll/event-poll.hpp" #include "utils/fd-utils.hpp" #include "utils/exception.hpp" #include "logger/logger.hpp" @@ -34,7 +34,7 @@ #include namespace vasum { -namespace utils { +namespace epoll { EventPoll::EventPoll() : mPollFD(::epoll_create1(EPOLL_CLOEXEC)) @@ -147,5 +147,5 @@ void EventPoll::removeFDInternal(const int fd) } } -} // namespace utils +} // namespace epoll } // namespace vasum diff --git a/common/utils/event-poll.hpp b/common/epoll/event-poll.hpp similarity index 89% rename from common/utils/event-poll.hpp rename to common/epoll/event-poll.hpp index 1a7ae2c..2d37aaa 100644 --- a/common/utils/event-poll.hpp +++ b/common/epoll/event-poll.hpp @@ -22,8 +22,10 @@ * @brief C++ epoll wrapper */ -#ifndef COMMON_UTILS_EVENT_POLL_HPP -#define COMMON_UTILS_EVENT_POLL_HPP +#ifndef COMMON_EPOLL_EVENT_POLL_HPP +#define COMMON_EPOLL_EVENT_POLL_HPP + +#include "epoll/events.hpp" #include #include @@ -31,12 +33,10 @@ #include namespace vasum { -namespace utils { +namespace epoll { class EventPoll { public: - - typedef unsigned int Events; typedef std::function Callback; EventPoll(); @@ -62,7 +62,7 @@ private: }; -} // namespace utils +} // namespace epoll } // namespace vasum -#endif // COMMON_UTILS_EVENT_POLL_HPP +#endif // COMMON_EPOLL_EVENT_POLL_HPP diff --git a/common/epoll/events.cpp b/common/epoll/events.cpp new file mode 100644 index 0000000..1ee27e0 --- /dev/null +++ b/common/epoll/events.cpp @@ -0,0 +1,71 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief Epoll events + */ + +#include "config.hpp" +#include "epoll/events.hpp" + +#include + +namespace vasum { +namespace epoll { + +namespace { + +std::string eventToString(Events event) +{ + switch (event) { + case EPOLLIN: return "IN"; + case EPOLLOUT: return "OUT"; + case EPOLLERR: return "ERR"; + case EPOLLHUP: return "HUP"; + case EPOLLRDHUP: return "RDHUP"; + default: + std::ostringstream ss; + ss << "0x" << std::hex << event; + return ss.str(); + } +} + +} // namespace + +std::string eventsToString(Events events) +{ + if (events == 0) { + return ""; + } + std::string ret; + for (unsigned int i = 0; i<32; ++i) { + Events event = 1u << i; + if (events & event) { + if (!ret.empty()) { + ret.append(", "); + } + ret.append(eventToString(event)); + } + } + return ret; +} + +} // namespace epoll +} // namespace vasum diff --git a/common/epoll/events.hpp b/common/epoll/events.hpp new file mode 100644 index 0000000..62eb00a --- /dev/null +++ b/common/epoll/events.hpp @@ -0,0 +1,41 @@ +/* + * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved + * + * Contact: Piotr Bartosiewicz + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License + */ + +/** + * @file + * @author Piotr Bartosiewicz (p.bartosiewi@partner.samsung.com) + * @brief Epoll events + */ + +#ifndef COMMON_EPOLL_EVENTS_HPP +#define COMMON_EPOLL_EVENTS_HPP + +#include +#include // for EPOLL* constatnts + +namespace vasum { +namespace epoll { + +typedef unsigned int Events; ///< bitmask of EPOLL* constants + +std::string eventsToString(Events events); + +} // namespace epoll +} // namespace vasum + +#endif // COMMON_EPOLL_EVENTS_HPP diff --git a/common/utils/glib-poll-dispatcher.cpp b/common/epoll/glib-poll-dispatcher.cpp similarity index 70% rename from common/utils/glib-poll-dispatcher.cpp rename to common/epoll/glib-poll-dispatcher.cpp index 5c66d16..cde535d 100644 --- a/common/utils/glib-poll-dispatcher.cpp +++ b/common/epoll/glib-poll-dispatcher.cpp @@ -23,22 +23,22 @@ */ #include "config.hpp" -#include "utils/glib-poll-dispatcher.hpp" +#include "epoll/glib-poll-dispatcher.hpp" #include "utils/callback-wrapper.hpp" namespace vasum { -namespace utils { +namespace epoll { -GlibPollDispatcher::GlibPollDispatcher(EventPoll& poll) +GlibPollDispatcher::GlibPollDispatcher() { - mChannel = g_io_channel_unix_new(poll.getPollFD()); + mChannel = g_io_channel_unix_new(mPoll.getPollFD()); - auto dispatchCallback = [&]() { - poll.dispatchIteration(0); + auto dispatchCallback = [this]() { + mPoll.dispatchIteration(0); }; auto cCallback = [](GIOChannel*, GIOCondition, gpointer data) -> gboolean { - getCallbackFromPointer(data)(); + utils::getCallbackFromPointer(data)(); return TRUE; }; @@ -46,8 +46,8 @@ GlibPollDispatcher::GlibPollDispatcher(EventPoll& poll) G_PRIORITY_DEFAULT, G_IO_IN, cCallback, - createCallbackWrapper(dispatchCallback, mGuard.spawn()), - &deleteCallbackWrapper); + utils::createCallbackWrapper(dispatchCallback, mGuard.spawn()), + &utils::deleteCallbackWrapper); } GlibPollDispatcher::~GlibPollDispatcher() @@ -57,5 +57,10 @@ GlibPollDispatcher::~GlibPollDispatcher() // mGuard destructor will wait for full unregister of dispatchCallback } -} // namespace utils +EventPoll& GlibPollDispatcher::getPoll() +{ + return mPoll; +} + +} // namespace epoll } // namespace vasum diff --git a/common/utils/glib-poll-dispatcher.hpp b/common/epoll/glib-poll-dispatcher.hpp similarity index 80% rename from common/utils/glib-poll-dispatcher.hpp rename to common/epoll/glib-poll-dispatcher.hpp index 07da0c3..cf300bb 100644 --- a/common/utils/glib-poll-dispatcher.hpp +++ b/common/epoll/glib-poll-dispatcher.hpp @@ -22,32 +22,35 @@ * @brief glib epoll dispatcher */ -#ifndef COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP -#define COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP +#ifndef COMMON_EPOLL_GLIB_POLL_DISPATCHER_HPP +#define COMMON_EPOLL_GLIB_POLL_DISPATCHER_HPP -#include "utils/event-poll.hpp" +#include "epoll/event-poll.hpp" #include "utils/callback-guard.hpp" #include namespace vasum { -namespace utils { +namespace epoll { /** * Will dispatch poll events in glib thread */ class GlibPollDispatcher { public: - GlibPollDispatcher(EventPoll& poll); + GlibPollDispatcher(); ~GlibPollDispatcher(); + + EventPoll& getPoll(); private: - CallbackGuard mGuard; + EventPoll mPoll; // before mGuard! + utils::CallbackGuard mGuard; GIOChannel* mChannel; guint mWatchId; }; -} // namespace utils +} // namespace epoll } // namespace vasum #endif // COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP diff --git a/common/utils/thread-poll-dispatcher.cpp b/common/epoll/thread-poll-dispatcher.cpp similarity index 73% rename from common/utils/thread-poll-dispatcher.cpp rename to common/epoll/thread-poll-dispatcher.cpp index f350587..797a8c4 100644 --- a/common/utils/thread-poll-dispatcher.cpp +++ b/common/epoll/thread-poll-dispatcher.cpp @@ -23,23 +23,22 @@ */ #include "config.hpp" -#include "utils/thread-poll-dispatcher.hpp" - -#include +#include "epoll/thread-poll-dispatcher.hpp" namespace vasum { -namespace utils { +namespace epoll { -ThreadPollDispatcher::ThreadPollDispatcher(EventPoll& poll) - : mPoll(poll) - , mThread([&]{ poll.dispatchLoop(); }) +ThreadPollDispatcher::ThreadPollDispatcher() { - auto controlCallback = [this](int, EventPoll::Events) -> bool { + auto controlCallback = [this](int, Events) -> bool { mStopEvent.receive(); return false; // break the loop }; - poll.addFD(mStopEvent.getFD(), EPOLLIN, std::move(controlCallback)); + mPoll.addFD(mStopEvent.getFD(), EPOLLIN, std::move(controlCallback)); + mThread = std::thread([this] { + mPoll.dispatchLoop(); + }); } ThreadPollDispatcher::~ThreadPollDispatcher() @@ -49,5 +48,10 @@ ThreadPollDispatcher::~ThreadPollDispatcher() mPoll.removeFD(mStopEvent.getFD()); } -} // namespace utils +EventPoll& ThreadPollDispatcher::getPoll() +{ + return mPoll; +} + +} // namespace epoll } // namespace vasum diff --git a/common/utils/thread-poll-dispatcher.hpp b/common/epoll/thread-poll-dispatcher.hpp similarity index 77% rename from common/utils/thread-poll-dispatcher.hpp rename to common/epoll/thread-poll-dispatcher.hpp index e54cb4e..af6b278 100644 --- a/common/utils/thread-poll-dispatcher.hpp +++ b/common/epoll/thread-poll-dispatcher.hpp @@ -22,31 +22,33 @@ * @brief Thread epoll dispatcher */ -#ifndef COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP -#define COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP +#ifndef COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP +#define COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP -#include "utils/event-poll.hpp" +#include "epoll/event-poll.hpp" #include "utils/eventfd.hpp" #include namespace vasum { -namespace utils { +namespace epoll { /** * Will dispatch poll events in a newly created thread */ class ThreadPollDispatcher { public: - ThreadPollDispatcher(EventPoll& poll); + ThreadPollDispatcher(); ~ThreadPollDispatcher(); + + EventPoll& getPoll(); private: - EventPoll& mPoll; - EventFD mStopEvent; + EventPoll mPoll; + utils::EventFD mStopEvent; std::thread mThread; }; -} // namespace utils +} // namespace epoll } // namespace vasum -#endif // COMMON_UTILS_THREAD_POLL_DISPATCHER_HPP +#endif // COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP diff --git a/tests/unit_tests/utils/ut-event-poll.cpp b/tests/unit_tests/epoll/ut-event-poll.cpp similarity index 58% rename from tests/unit_tests/utils/ut-event-poll.cpp rename to tests/unit_tests/epoll/ut-event-poll.cpp index e387393..0bcbe81 100644 --- a/tests/unit_tests/utils/ut-event-poll.cpp +++ b/tests/unit_tests/epoll/ut-event-poll.cpp @@ -26,54 +26,21 @@ #include "config.hpp" #include "ut.hpp" -#include "utils/event-poll.hpp" +#include "epoll/event-poll.hpp" #include "logger/logger.hpp" #include "ipc/internals/socket.hpp" #include "utils/latch.hpp" #include "utils/glib-loop.hpp" -#include "utils/glib-poll-dispatcher.hpp" -#include "utils/thread-poll-dispatcher.hpp" - -#include -#include +#include "epoll/glib-poll-dispatcher.hpp" +#include "epoll/thread-poll-dispatcher.hpp" using namespace vasum::utils; +using namespace vasum::epoll; using namespace vasum::ipc; namespace { const int unsigned TIMEOUT = 1000; -#define ADD_EVENT(e) {EPOLL##e, #e} -const std::map EVENT_NAMES = { - ADD_EVENT(IN), - ADD_EVENT(OUT), - ADD_EVENT(ERR), - ADD_EVENT(HUP), - ADD_EVENT(RDHUP), -}; -#undef ADD_EVENT - -std::string strEvents(EventPoll::Events events) -{ - if (events == 0) { - return ""; - } - std::ostringstream ss; - for (const auto& p : EVENT_NAMES) { - if (events & p.first) { - ss << p.second << ", "; - events &= ~p.first; - } - } - if (events != 0) { - ss << std::hex << events; - return ss.str(); - } else { - std::string ret = ss.str(); - ret.resize(ret.size() - 2); - return ret; - } -} } // namespace @@ -87,28 +54,29 @@ BOOST_AUTO_TEST_CASE(EmptyPoll) BOOST_AUTO_TEST_CASE(ThreadedPoll) { - EventPoll poll; - ThreadPollDispatcher dispatcher(poll); + ThreadPollDispatcher dispatcher; } BOOST_AUTO_TEST_CASE(GlibPoll) { ScopedGlibLoop loop; - EventPoll poll; - GlibPollDispatcher dispatcher(poll); + GlibPollDispatcher dispatcher; } -void doSocketTest(EventPoll& poll, Latch& goodMessage, Latch& remoteClosed) +void doSocketTest(EventPoll& poll) { const std::string PATH = "/tmp/ut-poll.sock"; const std::string MESSAGE = "This is a test message"; + Latch goodMessage; + Latch remoteClosed; + Socket listen = Socket::createSocket(PATH); std::shared_ptr server; - auto serverCallback = [&](int, EventPoll::Events events) -> bool { - LOGD("Server events: " << strEvents(events)); + auto serverCallback = [&](int, Events events) -> bool { + LOGD("Server events: " << eventsToString(events)); if (events & EPOLLOUT) { server->write(MESSAGE.data(), MESSAGE.size()); @@ -118,8 +86,8 @@ void doSocketTest(EventPoll& poll, Latch& goodMessage, Latch& remoteClosed) return true; }; - auto listenCallback = [&](int, EventPoll::Events events) -> bool { - LOGD("Listen events: " << strEvents(events)); + auto listenCallback = [&](int, Events events) -> bool { + LOGD("Listen events: " << eventsToString(events)); if (events & EPOLLIN) { server = listen.accept(); poll.addFD(server->getFD(), EPOLLHUP | EPOLLRDHUP | EPOLLOUT, serverCallback); @@ -131,8 +99,8 @@ void doSocketTest(EventPoll& poll, Latch& goodMessage, Latch& remoteClosed) Socket client = Socket::connectSocket(PATH); - auto clientCallback = [&](int, EventPoll::Events events) -> bool { - LOGD("Client events: " << strEvents(events)); + auto clientCallback = [&](int, Events events) -> bool { + LOGD("Client events: " << eventsToString(events)); if (events & EPOLLIN) { std::string ret(MESSAGE.size(), 'x'); @@ -158,47 +126,33 @@ void doSocketTest(EventPoll& poll, Latch& goodMessage, Latch& remoteClosed) BOOST_AUTO_TEST_CASE(ThreadedPollSocket) { - Latch goodMessage; - Latch remoteClosed; + ThreadPollDispatcher dispatcher; - EventPoll poll; - ThreadPollDispatcher dispatcher(poll); - - doSocketTest(poll, goodMessage, remoteClosed); + doSocketTest(dispatcher.getPoll()); } BOOST_AUTO_TEST_CASE(GlibPollSocket) { - Latch goodMessage; - Latch remoteClosed; - ScopedGlibLoop loop; - EventPoll poll; - GlibPollDispatcher dispatcher(poll); + GlibPollDispatcher dispatcher; - doSocketTest(poll, goodMessage, remoteClosed); + doSocketTest(dispatcher.getPoll()); } BOOST_AUTO_TEST_CASE(PollStacking) { - Latch goodMessage; - Latch remoteClosed; + ThreadPollDispatcher dispatcher; - EventPoll outer; - EventPoll inner; + EventPoll innerPoll; - auto dispatchInner = [&](int, EventPoll::Events) -> bool { - inner.dispatchIteration(0); + auto dispatchInner = [&](int, Events) -> bool { + innerPoll.dispatchIteration(0); return true; }; - - outer.addFD(inner.getPollFD(), EPOLLIN, dispatchInner); - - ThreadPollDispatcher dispatcher(outer); - doSocketTest(inner, goodMessage, remoteClosed); - - outer.removeFD(inner.getPollFD()); + dispatcher.getPoll().addFD(innerPoll.getPollFD(), EPOLLIN, dispatchInner); + doSocketTest(innerPoll); + dispatcher.getPoll().removeFD(innerPoll.getPollFD()); } BOOST_AUTO_TEST_SUITE_END() diff --git a/tests/unit_tests/ipc/ut-ipc.cpp b/tests/unit_tests/ipc/ut-ipc.cpp index fa0de22..088f576 100644 --- a/tests/unit_tests/ipc/ut-ipc.cpp +++ b/tests/unit_tests/ipc/ut-ipc.cpp @@ -70,7 +70,6 @@ const std::string SOCKET_PATH = TEST_DIR + "/test.socket"; struct Fixture { ScopedDir mTestPathGuard; - std::string SOCKET_PATH; Fixture() : mTestPathGuard(TEST_DIR) -- 2.7.4 From 02b066a8079eed2704dfc8afa1762bbc7faf29d4 Mon Sep 17 00:00:00 2001 From: Dariusz Michaluk Date: Tue, 3 Mar 2015 09:05:00 +0100 Subject: [PATCH 13/16] Disable dbus integration. [Bug/Feature] Disable dbus integration. [Cause] It is not used yet. Server and tests require a different dbus policy user, switching is inconvenient. [Solution] N/A [Verification] Build, run server, run tests Change-Id: I4e28b3b21d5eac6d23462ba2c4eb32aa7e32d4ed Signed-off-by: Dariusz Michaluk --- server/configs/lxc-templates/tizen-common-wayland.sh | 2 +- server/configs/templates/default.conf | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/server/configs/lxc-templates/tizen-common-wayland.sh b/server/configs/lxc-templates/tizen-common-wayland.sh index 9424dc4..5c572b7 100755 --- a/server/configs/lxc-templates/tizen-common-wayland.sh +++ b/server/configs/lxc-templates/tizen-common-wayland.sh @@ -297,6 +297,6 @@ ${path}/weston.ini etc/xdg/weston/weston.ini none ro,bind 0 0 devtmpfs dev devtmpfs rw,relatime,mode=755 0 0 devpts dev/pts devpts rw,relatime,gid=5,mode=620,ptmxmode=000 0 0 /sys/fs/smackfs sys/fs/smackfs none rw,bind 0 0 -/var/run/zones/${name}/run var/run none rw,bind 0 0 +#/var/run/zones/${name}/run var/run none rw,bind 0 0 #tmpfs run tmpfs rw,nosuid,nodev,mode=755 0 0 EOF diff --git a/server/configs/templates/default.conf b/server/configs/templates/default.conf index 9eb0e08..42f587b 100644 --- a/server/configs/templates/default.conf +++ b/server/configs/templates/default.conf @@ -9,7 +9,7 @@ "privilege" : 10, "vt" : 0, "switchToDefaultAfterTimeout" : true, - "enableDbusIntegration" : true, + "enableDbusIntegration" : false, "runMountPoint" : "~NAME~/run", "provisions" : [], "permittedToSend" : [ "/tmp/.*" ], -- 2.7.4 From 8c8d3f3e21999353f84edb837b7a161dda71d9b3 Mon Sep 17 00:00:00 2001 From: Dariusz Michaluk Date: Wed, 4 Mar 2015 10:26:16 +0100 Subject: [PATCH 14/16] Add timeout after zone start. [Bug/Feature] Add timeout after zone start. [Cause] N/A [Solution] N/A [Verification] Build, run server, run tests Change-Id: Ie6adace3d36ad98756d8efc823a056356f41d02d Signed-off-by: Dariusz Michaluk --- server/zone-admin.cpp | 13 +++++++++++++ server/zone-admin.hpp | 1 + 2 files changed, 14 insertions(+) diff --git a/server/zone-admin.cpp b/server/zone-admin.cpp index 519b172..8005fb4 100644 --- a/server/zone-admin.cpp +++ b/server/zone-admin.cpp @@ -35,6 +35,8 @@ #include #include +#include +#include namespace vasum { @@ -55,6 +57,7 @@ ZoneAdmin::ZoneAdmin(const std::string& zoneId, const ZoneConfig& config, const ZoneDynamicConfig& dynamicConfig) : mConfig(config), + mDynamicConfig(dynamicConfig), mZone(zonesPath, zoneId), mId(zoneId), mDetachOnExit(false), @@ -138,6 +141,16 @@ void ZoneAdmin::start() throw ZoneOperationException("Could not start zone"); } + // Wait until the full platform launch with graphical stack. + // VT should be activated by a graphical stack. + // If we do it with 'zoneToFocus.activateVT' before starting the graphical stack, + // graphical stack initialization failed and we finally switch to the black screen. + // Skip waiting when graphical stack is not running (unit tests). + if (mDynamicConfig.vt > 0) { + // TODO, timeout is a temporary solution + std::this_thread::sleep_for(std::chrono::milliseconds(4000)); + } + LOGD(mId << ": Started"); } diff --git a/server/zone-admin.hpp b/server/zone-admin.hpp index 44d2613..ade4585 100644 --- a/server/zone-admin.hpp +++ b/server/zone-admin.hpp @@ -166,6 +166,7 @@ public: private: const ZoneConfig& mConfig; + const ZoneDynamicConfig& mDynamicConfig; lxc::LxcZone mZone; const std::string mId; bool mDetachOnExit; -- 2.7.4 From 0e37ce4957c03d1c4077ef1211f3e3799d50ad2e Mon Sep 17 00:00:00 2001 From: Dariusz Michaluk Date: Wed, 4 Mar 2015 14:28:11 +0100 Subject: [PATCH 15/16] Fix building rpm packages. [Bug/Feature] Not deleted configs after rpm packages erasing. [Cause] Configs are not marked in spec file. [Solution] Move configs to a single directory. Mark all configs in spec file. [Verification] Build, install, uninstall Change-Id: I9728eb67e756ceca1d5cad8078d0040e7018975b Signed-off-by: Dariusz Michaluk --- CMakeLists.txt | 1 - packaging/vasum.spec | 12 ++++++------ tests/unit_tests/CMakeLists.txt | 4 ++-- 3 files changed, 8 insertions(+), 9 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index d5cb3e3..743860b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -167,7 +167,6 @@ IF(NOT DEFINED DATA_DIR) ENDIF(NOT DEFINED DATA_DIR) SET(VSM_CONFIG_INSTALL_DIR ${SYSCONF_INSTALL_DIR}/vasum) -SET(VSM_DATA_INSTALL_DIR ${SHARE_INSTALL_PREFIX}/vasum) ADD_SUBDIRECTORY(${CLIENT_FOLDER}) ADD_SUBDIRECTORY(${SERVER_FOLDER}) diff --git a/packaging/vasum.spec b/packaging/vasum.spec index db8d6be..bd7e9f1 100644 --- a/packaging/vasum.spec +++ b/packaging/vasum.spec @@ -50,7 +50,7 @@ between them. A process from inside a zone can request a switch of context %{_unitdir}/vasum.service %{_unitdir}/vasum.socket %{_unitdir}/multi-user.target.wants/vasum.service -/etc/dbus-1/system.d/org.tizen.vasum.host.conf +%config /etc/dbus-1/system.d/org.tizen.vasum.host.conf %dir %{_datadir}/.zones %prep @@ -165,7 +165,7 @@ Zones support installed inside every zone. %files zone-support %manifest packaging/vasum-zone-support.manifest %defattr(644,root,root,755) -/etc/dbus-1/system.d/org.tizen.vasum.zone.conf +%config /etc/dbus-1/system.d/org.tizen.vasum.zone.conf ## Zone Daemon Package #################################################### @@ -182,7 +182,7 @@ Daemon running inside every zone. %manifest packaging/vasum-zone-daemon.manifest %defattr(644,root,root,755) %attr(755,root,root) %{_bindir}/vasum-zone-daemon -/etc/dbus-1/system.d/org.tizen.vasum.zone.daemon.conf +%config /etc/dbus-1/system.d/org.tizen.vasum.zone.daemon.conf ## Command Line Interface ###################################################### @@ -233,9 +233,9 @@ systemctl daemon-reload %attr(755,root,root) %{script_dir}/vsm_int_tests.py %attr(755,root,root) %{script_dir}/vsm_launch_test.py %{script_dir}/vsm_test_parser.py -%{_datadir}/vasum/tests -%attr(755,root,root) %{_datadir}/vasum/lxc-templates +%config /etc/vasum/tests +%attr(755,root,root) /etc/vasum/tests/lxc-templates %{python_sitelib}/vsm_integration_tests -/etc/dbus-1/system.d/org.tizen.vasum.tests.conf +%config /etc/dbus-1/system.d/org.tizen.vasum.tests.conf %{_unitdir}/vasum-socket-test.socket %{_unitdir}/vasum-socket-test.service diff --git a/tests/unit_tests/CMakeLists.txt b/tests/unit_tests/CMakeLists.txt index fc96586..2a561ed 100644 --- a/tests/unit_tests/CMakeLists.txt +++ b/tests/unit_tests/CMakeLists.txt @@ -67,8 +67,8 @@ TARGET_LINK_LIBRARIES(${UT_SERVER_CODENAME} ${UT_SERVER_DEPS_LIBRARIES} ${Boost_ TARGET_LINK_LIBRARIES(${SOCKET_TEST_CODENAME} ${UT_SERVER_DEPS_LIBRARIES} ${Boost_LIBRARIES}) ## Subdirectories ############################################################## -SET(VSM_TEST_CONFIG_INSTALL_DIR ${VSM_DATA_INSTALL_DIR}/tests) -SET(VSM_TEST_LXC_TEMPLATES_INSTALL_DIR ${VSM_DATA_INSTALL_DIR}/lxc-templates/tests) +SET(VSM_TEST_CONFIG_INSTALL_DIR ${VSM_CONFIG_INSTALL_DIR}/tests) +SET(VSM_TEST_LXC_TEMPLATES_INSTALL_DIR ${VSM_CONFIG_INSTALL_DIR}/tests/lxc-templates) ADD_DEFINITIONS(-DVSM_TEST_CONFIG_INSTALL_DIR="${VSM_TEST_CONFIG_INSTALL_DIR}") ADD_DEFINITIONS(-DVSM_TEST_LXC_TEMPLATES_INSTALL_DIR="${VSM_TEST_LXC_TEMPLATES_INSTALL_DIR}") -- 2.7.4 From 99314f1553605b796e6ae5b90466113f98e79360 Mon Sep 17 00:00:00 2001 From: Piotr Bartosiewicz Date: Tue, 10 Mar 2015 12:01:48 +0100 Subject: [PATCH 16/16] IPC works on epoll [Bug/Feature] N/A [Cause] N/A [Solution] N/A [Verification] Run tests Change-Id: I6f6b5a7a70cecbecbdf0c502d0f8618577892a48 --- common/epoll/event-poll.cpp | 15 +- ...lib-poll-dispatcher.cpp => glib-dispatcher.cpp} | 8 +- ...lib-poll-dispatcher.hpp => glib-dispatcher.hpp} | 12 +- ...d-poll-dispatcher.cpp => thread-dispatcher.cpp} | 8 +- ...d-poll-dispatcher.hpp => thread-dispatcher.hpp} | 12 +- common/ipc/client.cpp | 71 +++--- common/ipc/client.hpp | 35 +-- common/ipc/internals/acceptor.cpp | 92 +------- common/ipc/internals/acceptor.hpp | 34 --- common/ipc/internals/processor.cpp | 148 ++---------- common/ipc/internals/processor.hpp | 18 +- common/ipc/service.cpp | 97 +++----- common/ipc/service.hpp | 35 +-- tests/scripts/vsm_test_parser.py | 2 +- tests/unit_tests/epoll/ut-event-poll.cpp | 14 +- tests/unit_tests/ipc/ut-ipc.cpp | 250 ++++++++++----------- tests/unit_tests/ut.hpp | 14 ++ 17 files changed, 271 insertions(+), 594 deletions(-) rename common/epoll/{glib-poll-dispatcher.cpp => glib-dispatcher.cpp} (91%) rename common/epoll/{glib-poll-dispatcher.hpp => glib-dispatcher.hpp} (84%) rename common/epoll/{thread-poll-dispatcher.cpp => thread-dispatcher.cpp} (88%) rename common/epoll/{thread-poll-dispatcher.hpp => thread-dispatcher.hpp} (83%) diff --git a/common/epoll/event-poll.cpp b/common/epoll/event-poll.cpp index dc1c1bc..9eb9fa0 100644 --- a/common/epoll/event-poll.cpp +++ b/common/epoll/event-poll.cpp @@ -49,6 +49,9 @@ EventPoll::~EventPoll() { if (!mCallbacks.empty()) { LOGW("Not removed callbacks: " << mCallbacks.size()); + for (const auto& item : mCallbacks) { + LOGT("Not removed fd: " << item.first); + } assert(0 && "Not removed callbacks left"); } utils::close(mPollFD); @@ -73,7 +76,7 @@ void EventPoll::addFD(const int fd, const Events events, Callback&& callback) } mCallbacks.insert({fd, std::make_shared(std::move(callback))}); - LOGT("Callback added for " << fd); + LOGT("Callback added for fd: " << fd); } void EventPoll::removeFD(const int fd) @@ -87,7 +90,7 @@ void EventPoll::removeFD(const int fd) } mCallbacks.erase(iter); removeFDInternal(fd); - LOGT("Callback removed for " << fd); + LOGT("Callback removed for fd: " << fd); } bool EventPoll::dispatchIteration(const int timeoutMs) @@ -115,7 +118,13 @@ bool EventPoll::dispatchIteration(const int timeoutMs) // add ref because removeFD(self) can be called inside callback std::shared_ptr callback(iter->second); - return (*callback)(event.data.fd, event.events); + try { + LOGT("Dispatch fd: " << event.data.fd << ", events: " << eventsToString(event.events)); + return (*callback)(event.data.fd, event.events); + } catch (std::exception& e) { + LOGE("Got unexpected exception: " << e.what()); + assert(0 && "Callback should not throw any exceptions"); + } } } diff --git a/common/epoll/glib-poll-dispatcher.cpp b/common/epoll/glib-dispatcher.cpp similarity index 91% rename from common/epoll/glib-poll-dispatcher.cpp rename to common/epoll/glib-dispatcher.cpp index cde535d..9567abb 100644 --- a/common/epoll/glib-poll-dispatcher.cpp +++ b/common/epoll/glib-dispatcher.cpp @@ -23,13 +23,13 @@ */ #include "config.hpp" -#include "epoll/glib-poll-dispatcher.hpp" +#include "epoll/glib-dispatcher.hpp" #include "utils/callback-wrapper.hpp" namespace vasum { namespace epoll { -GlibPollDispatcher::GlibPollDispatcher() +GlibDispatcher::GlibDispatcher() { mChannel = g_io_channel_unix_new(mPoll.getPollFD()); @@ -50,14 +50,14 @@ GlibPollDispatcher::GlibPollDispatcher() &utils::deleteCallbackWrapper); } -GlibPollDispatcher::~GlibPollDispatcher() +GlibDispatcher::~GlibDispatcher() { g_source_remove(mWatchId); g_io_channel_unref(mChannel); // mGuard destructor will wait for full unregister of dispatchCallback } -EventPoll& GlibPollDispatcher::getPoll() +EventPoll& GlibDispatcher::getPoll() { return mPoll; } diff --git a/common/epoll/glib-poll-dispatcher.hpp b/common/epoll/glib-dispatcher.hpp similarity index 84% rename from common/epoll/glib-poll-dispatcher.hpp rename to common/epoll/glib-dispatcher.hpp index cf300bb..0fa7b3a 100644 --- a/common/epoll/glib-poll-dispatcher.hpp +++ b/common/epoll/glib-dispatcher.hpp @@ -22,8 +22,8 @@ * @brief glib epoll dispatcher */ -#ifndef COMMON_EPOLL_GLIB_POLL_DISPATCHER_HPP -#define COMMON_EPOLL_GLIB_POLL_DISPATCHER_HPP +#ifndef COMMON_EPOLL_GLIB_DISPATCHER_HPP +#define COMMON_EPOLL_GLIB_DISPATCHER_HPP #include "epoll/event-poll.hpp" #include "utils/callback-guard.hpp" @@ -36,10 +36,10 @@ namespace epoll { /** * Will dispatch poll events in glib thread */ -class GlibPollDispatcher { +class GlibDispatcher { public: - GlibPollDispatcher(); - ~GlibPollDispatcher(); + GlibDispatcher(); + ~GlibDispatcher(); EventPoll& getPoll(); private: @@ -53,4 +53,4 @@ private: } // namespace epoll } // namespace vasum -#endif // COMMON_UTILS_GLIB_POLL_DISPATCHER_HPP +#endif // COMMON_UTILS_GLIB_DISPATCHER_HPP diff --git a/common/epoll/thread-poll-dispatcher.cpp b/common/epoll/thread-dispatcher.cpp similarity index 88% rename from common/epoll/thread-poll-dispatcher.cpp rename to common/epoll/thread-dispatcher.cpp index 797a8c4..a82cad8 100644 --- a/common/epoll/thread-poll-dispatcher.cpp +++ b/common/epoll/thread-dispatcher.cpp @@ -23,12 +23,12 @@ */ #include "config.hpp" -#include "epoll/thread-poll-dispatcher.hpp" +#include "epoll/thread-dispatcher.hpp" namespace vasum { namespace epoll { -ThreadPollDispatcher::ThreadPollDispatcher() +ThreadDispatcher::ThreadDispatcher() { auto controlCallback = [this](int, Events) -> bool { mStopEvent.receive(); @@ -41,14 +41,14 @@ ThreadPollDispatcher::ThreadPollDispatcher() }); } -ThreadPollDispatcher::~ThreadPollDispatcher() +ThreadDispatcher::~ThreadDispatcher() { mStopEvent.send(); mThread.join(); mPoll.removeFD(mStopEvent.getFD()); } -EventPoll& ThreadPollDispatcher::getPoll() +EventPoll& ThreadDispatcher::getPoll() { return mPoll; } diff --git a/common/epoll/thread-poll-dispatcher.hpp b/common/epoll/thread-dispatcher.hpp similarity index 83% rename from common/epoll/thread-poll-dispatcher.hpp rename to common/epoll/thread-dispatcher.hpp index af6b278..7d0f30d 100644 --- a/common/epoll/thread-poll-dispatcher.hpp +++ b/common/epoll/thread-dispatcher.hpp @@ -22,8 +22,8 @@ * @brief Thread epoll dispatcher */ -#ifndef COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP -#define COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP +#ifndef COMMON_EPOLL_THREAD_DISPATCHER_HPP +#define COMMON_EPOLL_THREAD_DISPATCHER_HPP #include "epoll/event-poll.hpp" #include "utils/eventfd.hpp" @@ -36,10 +36,10 @@ namespace epoll { /** * Will dispatch poll events in a newly created thread */ -class ThreadPollDispatcher { +class ThreadDispatcher { public: - ThreadPollDispatcher(); - ~ThreadPollDispatcher(); + ThreadDispatcher(); + ~ThreadDispatcher(); EventPoll& getPoll(); private: @@ -51,4 +51,4 @@ private: } // namespace epoll } // namespace vasum -#endif // COMMON_EPOLL_THREAD_POLL_DISPATCHER_HPP +#endif // COMMON_EPOLL_THREAD_DISPATCHER_HPP diff --git a/common/ipc/client.cpp b/common/ipc/client.cpp index 6669d8a..abecb71 100644 --- a/common/ipc/client.cpp +++ b/common/ipc/client.cpp @@ -31,8 +31,9 @@ namespace vasum { namespace ipc { -Client::Client(const std::string& socketPath) - : mProcessor("[CLIENT] "), +Client::Client(epoll::EventPoll& eventPoll, const std::string& socketPath) + : mEventPoll(eventPoll), + mProcessor("[CLIENT] "), mSocketPath(socketPath) { LOGS("Client Constructor"); @@ -50,14 +51,19 @@ Client::~Client() } } -void Client::start(const bool usesExternalPolling) +void Client::start() { + if (mProcessor.isStarted()) { + return; + } LOGS("Client start"); // Initialize the connection with the server - if (usesExternalPolling) { - startPoll(); - } - mProcessor.start(usesExternalPolling); + auto handleEvent = [&](int, epoll::Events) -> bool { + mProcessor.handleEvent(); + return true; + }; + mEventPoll.addFD(mProcessor.getEventFD(), EPOLLIN, handleEvent); + mProcessor.start(); LOGD("Connecting to " + mSocketPath); auto socketPtr = std::make_shared(Socket::connectSocket(mSocketPath)); @@ -71,34 +77,18 @@ bool Client::isStarted() void Client::stop() { + if (!mProcessor.isStarted()) { + return; + } LOGS("Client stop"); mProcessor.stop(); - if (mIPCGSourcePtr) { - stopPoll(); - } -} - -void Client::startPoll() -{ - LOGS("Client startPoll"); - using namespace std::placeholders; - mIPCGSourcePtr = IPCGSource::create(std::bind(&Client::handle, this, _1, _2)); - mIPCGSourcePtr->addFD(mProcessor.getEventFD()); - mIPCGSourcePtr->attach(); + mEventPoll.removeFD(mProcessor.getEventFD()); } -void Client::stopPoll() -{ - LOGS("Client stopPoll"); - - mIPCGSourcePtr->removeFD(mProcessor.getEventFD()); - mIPCGSourcePtr->detach(); - mIPCGSourcePtr.reset(); -} - -void Client::handle(const FileDescriptor fd, const short pollEvent) +void Client::handle(const FileDescriptor fd, const epoll::Events pollEvents) { + //TODO remove handle method LOGS("Client handle"); if (!isStarted()) { @@ -106,17 +96,12 @@ void Client::handle(const FileDescriptor fd, const short pollEvent) return; } - if (fd == mProcessor.getEventFD() && (pollEvent & POLLIN)) { - mProcessor.handleEvent(); - return; - - } else if (pollEvent & POLLIN) { + if (pollEvents & EPOLLIN) { mProcessor.handleInput(fd); - return; + } - } else if (pollEvent & POLLHUP) { + if ((pollEvents & EPOLLHUP) || (pollEvents & EPOLLRDHUP)) { mProcessor.handleLostConnection(fd); - return; } } @@ -124,9 +109,11 @@ void Client::setNewPeerCallback(const PeerCallback& newPeerCallback) { LOGS("Client setNewPeerCallback"); auto callback = [newPeerCallback, this](PeerID peerID, FileDescriptor fd) { - if (mIPCGSourcePtr) { - mIPCGSourcePtr->addFD(fd); - } + auto handleFd = [&](FileDescriptor fd, epoll::Events events) -> bool { + handle(fd, events); + return true; + }; + mEventPoll.addFD(fd, EPOLLIN | EPOLLHUP | EPOLLRDHUP, handleFd); if (newPeerCallback) { newPeerCallback(peerID, fd); } @@ -138,9 +125,7 @@ void Client::setRemovedPeerCallback(const PeerCallback& removedPeerCallback) { LOGS("Client setRemovedPeerCallback"); auto callback = [removedPeerCallback, this](PeerID peerID, FileDescriptor fd) { - if (mIPCGSourcePtr) { - mIPCGSourcePtr->removeFD(fd); - } + mEventPoll.removeFD(fd); if (removedPeerCallback) { removedPeerCallback(peerID, fd); } diff --git a/common/ipc/client.hpp b/common/ipc/client.hpp index 321b4da..a3342b6 100644 --- a/common/ipc/client.hpp +++ b/common/ipc/client.hpp @@ -26,9 +26,9 @@ #define COMMON_IPC_CLIENT_HPP #include "ipc/internals/processor.hpp" -#include "ipc/ipc-gsource.hpp" #include "ipc/types.hpp" #include "ipc/result.hpp" +#include "epoll/event-poll.hpp" #include "logger/logger.hpp" #include @@ -40,28 +40,24 @@ namespace ipc { * This class wraps communication via UX sockets for client applications. * It uses serialization mechanism from libConfig. * - * There is one additional thread: - * - PROCESSOR is responsible for the communication and calling the callbacks - * * For message format @see ipc::Processor */ class Client { public: /** + * @param eventPoll event poll * @param serverPath path to the server's socket */ - Client(const std::string& serverPath); + Client(epoll::EventPoll& eventPoll, const std::string& serverPath); ~Client(); Client(const Client&) = delete; Client& operator=(const Client&) = delete; /** - * Starts the worker thread - * - * @param usesExternalPolling internal or external polling is used + * Starts processing */ - void start(const bool usesExternalPolling = false); + void start(); /** * @return is the communication thread running @@ -69,21 +65,11 @@ public: bool isStarted(); /** - * Stops all worker thread + * Stops processing */ void stop(); /** - * Used with an external polling loop. - * Handles one event from the file descriptor. - * - * @param fd file descriptor - * @param pollEvent event on the fd. Defined in poll.h - * - */ - void handle(const FileDescriptor fd, const short pollEvent); - - /** * Set the callback called for each new connection to a peer * * @param newPeerCallback the callback @@ -170,14 +156,13 @@ public: const std::shared_ptr& data); private: - - void startPoll(); - void stopPoll(); - + epoll::EventPoll& mEventPoll; PeerID mServiceID; Processor mProcessor; std::string mSocketPath; - IPCGSource::Pointer mIPCGSourcePtr; + + void handle(const FileDescriptor fd, const epoll::Events pollEvents); + }; template diff --git a/common/ipc/internals/acceptor.cpp b/common/ipc/internals/acceptor.cpp index ecb2210..15b9ad8 100644 --- a/common/ipc/internals/acceptor.cpp +++ b/common/ipc/internals/acceptor.cpp @@ -24,21 +24,14 @@ #include "config.hpp" -#include "ipc/exception.hpp" #include "ipc/internals/acceptor.hpp" #include "logger/logger.hpp" -#include -#include -#include -#include - namespace vasum { namespace ipc { Acceptor::Acceptor(const std::string& socketPath, const NewConnectionCallback& newConnectionCallback) - : mIsRunning(false), - mNewConnectionCallback(newConnectionCallback), + : mNewConnectionCallback(newConnectionCallback), mSocket(Socket::createSocket(socketPath)) { LOGT("Creating Acceptor for socket " << socketPath); @@ -46,98 +39,15 @@ Acceptor::Acceptor(const std::string& socketPath, const NewConnectionCallback& n Acceptor::~Acceptor() { - LOGT("Destroying Acceptor"); - try { - stop(); - } catch (std::exception& e) { - LOGE("Error in destructor: " << e.what()); - } LOGT("Destroyed Acceptor"); } -void Acceptor::start() -{ - LOGT("Starting Acceptor"); - if (!mThread.joinable()) { - mThread = std::thread(&Acceptor::run, this); - } - LOGT("Started Acceptor"); -} - -void Acceptor::stop() -{ - LOGT("Stopping Acceptor"); - - if (mThread.joinable()) { - mEventQueue.send(Event::FINISH); - LOGT("Waiting for Acceptor to finish"); - mThread.join(); - } - - LOGT("Stopped Acceptor"); -} - -void Acceptor::run() -{ - // Setup polling structure - std::vector fds(2); - - fds[0].fd = mEventQueue.getFD(); - fds[0].events = POLLIN; - - fds[1].fd = mSocket.getFD(); - fds[1].events = POLLIN; - - mIsRunning = true; - while (mIsRunning) { - LOGT("Waiting for new connections..."); - - int ret = ::poll(fds.data(), fds.size(), -1 /*blocking call*/); - - LOGT("...Incoming connection!"); - - if (ret == -1 || ret == 0) { - if (errno == EINTR) { - continue; - } - LOGE("Error in poll: " << std::string(strerror(errno))); - throw IPCException("Error in poll: " + std::string(strerror(errno))); - } - - // Check for incoming connections - if (fds[1].revents & POLLIN) { - fds[1].revents = 0; - handleConnection(); - } - - // Check for incoming events - if (fds[0].revents & POLLIN) { - fds[0].revents = 0; - handleEvent(); - } - } - LOGT("Exiting run"); -} - void Acceptor::handleConnection() { std::shared_ptr tmpSocket = mSocket.accept(); mNewConnectionCallback(tmpSocket); } -void Acceptor::handleEvent() -{ - if (mEventQueue.receive() == Event::FINISH) { - LOGD("Event FINISH"); - mIsRunning = false; - } -} - -FileDescriptor Acceptor::getEventFD() -{ - return mEventQueue.getFD(); -} - FileDescriptor Acceptor::getConnectionFD() { return mSocket.getFD(); diff --git a/common/ipc/internals/acceptor.hpp b/common/ipc/internals/acceptor.hpp index f87a0bb..a6e9ec8 100644 --- a/common/ipc/internals/acceptor.hpp +++ b/common/ipc/internals/acceptor.hpp @@ -28,11 +28,9 @@ #include "config.hpp" #include "ipc/internals/socket.hpp" -#include "ipc/internals/event-queue.hpp" #include "ipc/types.hpp" #include -#include namespace vasum { namespace ipc { @@ -59,51 +57,19 @@ public: Acceptor& operator=(const Acceptor&) = delete; /** - * Starts the thread accepting the new connections. - */ - void start(); - - /** - * Stops the accepting thread. - */ - void stop(); - - /** * Handle one incoming connection. * Used with external polling */ void handleConnection(); /** - * Handle one event from the internal event's queue - * Used with external polling - */ - void handleEvent(); - - /** - * @return file descriptor of internal event's queue - */ - FileDescriptor getEventFD(); - - /** * @return file descriptor for the connection socket */ FileDescriptor getConnectionFD(); private: - enum class Event : int { - FINISH // Shutdown request - }; - - bool mIsRunning; - NewConnectionCallback mNewConnectionCallback; Socket mSocket; - - EventQueue mEventQueue; - std::thread mThread; - - void run(); }; } // namespace ipc diff --git a/common/ipc/internals/processor.cpp b/common/ipc/internals/processor.cpp index 93f721f..587723c 100644 --- a/common/ipc/internals/processor.cpp +++ b/common/ipc/internals/processor.cpp @@ -26,7 +26,6 @@ #include "ipc/exception.hpp" #include "ipc/internals/processor.hpp" -#include "utils/signal.hpp" #include "utils/exception.hpp" #include @@ -65,8 +64,6 @@ Processor::Processor(const std::string& logName, { LOGS(mLogPrefix + "Processor Constructor"); - utils::signalBlock(SIGPIPE); - using namespace std::placeholders; setSignalHandlerInternal(REGISTER_SIGNAL_METHOD_ID, std::bind(&Processor::onNewSignals, this, _1, _2)); @@ -104,7 +101,7 @@ bool Processor::isStarted() return mIsRunning; } -void Processor::start(bool usesExternalPolling) +void Processor::start() { LOGS(mLogPrefix + "Processor start"); @@ -112,10 +109,6 @@ void Processor::start(bool usesExternalPolling) if (!mIsRunning) { LOGI(mLogPrefix + "Processor start"); mIsRunning = true; - mUsesExternalPolling = usesExternalPolling; - if (!usesExternalPolling) { - mThread = std::thread(&Processor::run, this); - } } } @@ -133,15 +126,12 @@ void Processor::stop() LOGD(mLogPrefix + "Waiting for the Processor to stop"); - if (mThread.joinable()) { - mThread.join(); - } else { - // Wait till the FINISH request is served - Lock lock(mStateMutex); - conditionPtr->wait(lock, [this]() { - return !mIsRunning; - }); - } + // Wait till the FINISH request is served + Lock lock(mStateMutex); + conditionPtr->wait(lock, [this]() { + return !mIsRunning; + }); + assert(mPeerInfo.empty()); } } @@ -204,7 +194,8 @@ PeerID Processor::addPeer(const std::shared_ptr& socketPtr) auto requestPtr = std::make_shared(socketPtr); mRequestQueue.pushBack(Event::ADD_PEER, requestPtr); - LOGI(mLogPrefix + "Add Peer Request. Id: " << requestPtr->peerID); + LOGI(mLogPrefix + "Add Peer Request. Id: " << requestPtr->peerID + << ", fd: " << socketPtr->getFD()); return requestPtr->peerID; } @@ -233,14 +224,14 @@ void Processor::removePeerSyncInternal(const PeerID peerID, Lock& lock) void Processor::removePeerInternal(Peers::iterator peerIt, const std::exception_ptr& exceptionPtr) { - LOGS(mLogPrefix + "Processor removePeerInternal peerID: " << peerIt->peerID); - LOGI(mLogPrefix + "Removing peer. peerID: " << peerIt->peerID); - if (peerIt == mPeerInfo.end()) { LOGW("Peer already removed"); return; } + LOGS(mLogPrefix + "Processor removePeerInternal peerID: " << peerIt->peerID); + LOGI(mLogPrefix + "Removing peer. peerID: " << peerIt->peerID); + // Remove from signal addressees for (auto it = mSignalsPeers.begin(); it != mSignalsPeers.end();) { it->second.remove(peerIt->peerID); @@ -270,100 +261,6 @@ void Processor::removePeerInternal(Peers::iterator peerIt, const std::exception_ mPeerInfo.erase(peerIt); } -void Processor::resetPolling() -{ - LOGS(mLogPrefix + "Processor resetPolling"); - - if (mUsesExternalPolling) { - return; - } - - // Setup polling on eventfd and sockets - mFDs.resize(mPeerInfo.size() + 1); - LOGI(mLogPrefix + "Reseting mFDS.size: " << mFDs.size()); - - mFDs[0].fd = mRequestQueue.getFD(); - mFDs[0].events = POLLIN; - - for (unsigned int i = 1; i < mFDs.size(); ++i) { - auto fd = mPeerInfo[i - 1].socketPtr->getFD(); - - LOGI(mLogPrefix + "Reseting fd: " << fd); - - mFDs[i].fd = fd; - mFDs[i].events = POLLIN | POLLHUP; // Listen for input events - // TODO: It's possible to block on writing to fd. Maybe listen for POLLOUT too? - } -} - -void Processor::run() -{ - LOGS(mLogPrefix + "Processor run"); - - { - Lock lock(mStateMutex); - resetPolling(); - } - - while (isStarted()) { - LOGT(mLogPrefix + "Waiting for communication..."); - int ret = poll(mFDs.data(), mFDs.size(), -1 /*blocking call*/); - LOGT(mLogPrefix + "... incoming communication!"); - if (ret == -1 || ret == 0) { - if (errno == EINTR) { - continue; - } - LOGE(mLogPrefix + "Error in poll: " << std::string(strerror(errno))); - throw IPCException("Error in poll: " + std::string(strerror(errno))); - } - - // Check for lost connections: - if (handleLostConnections()) { - // mFDs changed - resetPolling(); - continue; - } - - // Check for incoming data. - if (handleInputs()) { - // mFDs changed - resetPolling(); - continue; - } - - // Check for incoming events - if (mFDs[0].revents & POLLIN) { - mFDs[0].revents &= ~(POLLIN); - if (handleEvent()) { - // mFDs changed - resetPolling(); - continue; - } - } - - } -} - -bool Processor::handleLostConnections() -{ - Lock lock(mStateMutex); - - bool isPeerRemoved = false; - - for (unsigned int i = 1; i < mFDs.size(); ++i) { - if (mFDs[i].revents & POLLHUP) { - auto peerIt = getPeerInfoIterator(mFDs[i].fd); - LOGI(mLogPrefix + "Lost connection to peer: " << peerIt->peerID); - mFDs[i].revents &= ~(POLLHUP); - removePeerInternal(peerIt, - std::make_exception_ptr(IPCPeerDisconnectedException())); - isPeerRemoved = true; - } - } - - return isPeerRemoved; -} - bool Processor::handleLostConnection(const FileDescriptor fd) { Lock lock(mStateMutex); @@ -373,21 +270,6 @@ bool Processor::handleLostConnection(const FileDescriptor fd) return true; } -bool Processor::handleInputs() -{ - // Lock not needed, mFDs won't be changed by handleInput - - bool pollChanged = false; - for (unsigned int i = 1; i < mFDs.size(); ++i) { - if (mFDs[i].revents & POLLIN) { - mFDs[i].revents &= ~(POLLIN); - pollChanged = pollChanged || handleInput(mFDs[i].fd); - } - } - - return pollChanged; -} - bool Processor::handleInput(const FileDescriptor fd) { LOGS(mLogPrefix + "Processor handleInput fd: " << fd); @@ -795,6 +677,12 @@ bool Processor::onFinishRequest(FinishRequest& request) } } + // Close peers + while (!mPeerInfo.empty()) { + removePeerInternal(--mPeerInfo.end(), + std::make_exception_ptr(IPCClosingException())); + } + mIsRunning = false; request.conditionPtr->notify_all(); diff --git a/common/ipc/internals/processor.hpp b/common/ipc/internals/processor.hpp index 86a3f42..6e0ae81 100644 --- a/common/ipc/internals/processor.hpp +++ b/common/ipc/internals/processor.hpp @@ -43,7 +43,6 @@ #include "logger/logger-scope.hpp" #include -#include #include #include #include @@ -136,12 +135,9 @@ public: /** - * Start the processing thread. - * Quits immediately after starting the thread. - * - * @param usesExternalPolling internal or external polling is used + * Start processing. */ - void start(const bool usesExternalPolling); + void start(); /** * @return is processor running @@ -432,14 +428,12 @@ private: RequestQueue mRequestQueue; bool mIsRunning; - bool mUsesExternalPolling; std::unordered_map> mMethodsCallbacks; std::unordered_map> mSignalsCallbacks; std::unordered_map> mSignalsPeers; Peers mPeerInfo; - std::vector mFDs; std::unordered_map mReturnCallbacks; @@ -451,8 +445,6 @@ private: unsigned int mMaxNumberOfPeers; - std::thread mThread; - template void setMethodHandlerInternal(const MethodID methodID, const typename MethodHandler::type& process); @@ -466,8 +458,6 @@ private: const PeerID peerID, const std::shared_ptr& data); - void run(); - // Request handlers bool onMethodRequest(MethodRequest& request); bool onSignalRequest(SignalRequest& request); @@ -476,9 +466,6 @@ private: bool onSendResultRequest(SendResultRequest& request); bool onFinishRequest(FinishRequest& request); - bool handleLostConnections(); - bool handleInputs(); - bool onReturnValue(Peers::iterator& peerIt, const MessageID messageID); bool onRemoteMethod(Peers::iterator& peerIt, @@ -489,7 +476,6 @@ private: const MethodID methodID, const MessageID messageID, std::shared_ptr signalCallbacks); - void resetPolling(); void removePeerInternal(Peers::iterator peerIt, const std::exception_ptr& exceptionPtr); diff --git a/common/ipc/service.cpp b/common/ipc/service.cpp index 442d804..c436726 100644 --- a/common/ipc/service.cpp +++ b/common/ipc/service.cpp @@ -33,10 +33,12 @@ using namespace std::placeholders; namespace vasum { namespace ipc { -Service::Service(const std::string& socketPath, +Service::Service(epoll::EventPoll& eventPoll, + const std::string& socketPath, const PeerCallback& addPeerCallback, const PeerCallback& removePeerCallback) - : mProcessor("[SERVICE] "), + : mEventPoll(eventPoll), + mProcessor("[SERVICE] "), mAcceptor(socketPath, std::bind(&Processor::addPeer, &mProcessor, _1)) { @@ -55,19 +57,23 @@ Service::~Service() } } -void Service::start(const bool usesExternalPolling) +void Service::start() { - LOGS("Service start"); - if (usesExternalPolling) { - startPoll(); - } - mProcessor.start(usesExternalPolling); - - // There can be an incoming connection from mAcceptor before mProcessor is listening, - // but it's OK. It will handle the connection when ready. So no need to wait for mProcessor. - if (!usesExternalPolling) { - mAcceptor.start(); + if (mProcessor.isStarted()) { + return; } + LOGS("Service start"); + auto handleConnection = [&](int, epoll::Events) -> bool { + mAcceptor.handleConnection(); + return true; + }; + auto handleProcessorEvent = [&](int, epoll::Events) -> bool { + mProcessor.handleEvent(); + return true; + }; + mEventPoll.addFD(mAcceptor.getConnectionFD(), EPOLLIN, handleConnection); + mEventPoll.addFD(mProcessor.getEventFD(), EPOLLIN, handleProcessorEvent); + mProcessor.start(); } bool Service::isStarted() @@ -77,39 +83,19 @@ bool Service::isStarted() void Service::stop() { + if (!mProcessor.isStarted()) { + return; + } LOGS("Service stop"); - mAcceptor.stop(); mProcessor.stop(); - if (mIPCGSourcePtr) { - stopPoll(); - } -} - -void Service::startPoll() -{ - LOGS("Service startPoll"); - - mIPCGSourcePtr = IPCGSource::create(std::bind(&Service::handle, this, _1, _2)); - mIPCGSourcePtr->addFD(mAcceptor.getEventFD()); - mIPCGSourcePtr->addFD(mAcceptor.getConnectionFD()); - mIPCGSourcePtr->addFD(mProcessor.getEventFD()); - mIPCGSourcePtr->attach(); + mEventPoll.removeFD(mAcceptor.getConnectionFD()); + mEventPoll.removeFD(mProcessor.getEventFD()); } -void Service::stopPoll() -{ - LOGS("Service stopPoll"); - - mIPCGSourcePtr->removeFD(mAcceptor.getEventFD()); - mIPCGSourcePtr->removeFD(mAcceptor.getConnectionFD()); - mIPCGSourcePtr->removeFD(mProcessor.getEventFD()); - mIPCGSourcePtr->detach(); - mIPCGSourcePtr.reset(); -} - -void Service::handle(const FileDescriptor fd, const short pollEvent) +void Service::handle(const FileDescriptor fd, const epoll::Events pollEvents) { + //TODO remove handle method LOGS("Service handle"); if (!isStarted()) { @@ -117,25 +103,12 @@ void Service::handle(const FileDescriptor fd, const short pollEvent) return; } - if (fd == mProcessor.getEventFD() && (pollEvent & POLLIN)) { - mProcessor.handleEvent(); - return; - - } else if (fd == mAcceptor.getConnectionFD() && (pollEvent & POLLIN)) { - mAcceptor.handleConnection(); - return; - - } else if (fd == mAcceptor.getEventFD() && (pollEvent & POLLIN)) { - mAcceptor.handleEvent(); - return; - - } else if (pollEvent & POLLIN) { + if (pollEvents & EPOLLIN) { mProcessor.handleInput(fd); - return; + } - } else if (pollEvent & POLLHUP) { + if ((pollEvents & EPOLLHUP) || (pollEvents & EPOLLRDHUP)) { mProcessor.handleLostConnection(fd); - return; } } @@ -143,9 +116,11 @@ void Service::setNewPeerCallback(const PeerCallback& newPeerCallback) { LOGS("Service setNewPeerCallback"); auto callback = [newPeerCallback, this](PeerID peerID, FileDescriptor fd) { - if (mIPCGSourcePtr) { - mIPCGSourcePtr->addFD(fd); - } + auto handleFd = [&](FileDescriptor fd, epoll::Events events) -> bool { + handle(fd, events); + return true; + }; + mEventPoll.addFD(fd, EPOLLIN | EPOLLHUP | EPOLLRDHUP, handleFd); if (newPeerCallback) { newPeerCallback(peerID, fd); } @@ -157,9 +132,7 @@ void Service::setRemovedPeerCallback(const PeerCallback& removedPeerCallback) { LOGS("Service setRemovedPeerCallback"); auto callback = [removedPeerCallback, this](PeerID peerID, FileDescriptor fd) { - if (mIPCGSourcePtr) { - mIPCGSourcePtr->removeFD(fd); - } + mEventPoll.removeFD(fd); if (removedPeerCallback) { removedPeerCallback(peerID, fd); } diff --git a/common/ipc/service.hpp b/common/ipc/service.hpp index 022c9c9..880c137 100644 --- a/common/ipc/service.hpp +++ b/common/ipc/service.hpp @@ -27,9 +27,9 @@ #include "ipc/internals/processor.hpp" #include "ipc/internals/acceptor.hpp" -#include "ipc/ipc-gsource.hpp" #include "ipc/types.hpp" #include "ipc/result.hpp" +#include "epoll/event-poll.hpp" #include "logger/logger.hpp" #include @@ -42,18 +42,16 @@ namespace ipc { * This class wraps communication via UX sockets. * It uses serialization mechanism from libConfig. * - * There are two working threads: - * - ACCEPTOR accepts incoming connections and passes them to PROCESSOR - * - PROCESSOR is responsible for the communication and calling the callbacks - * * For message format @see ipc::Processor */ class Service { public: /** + * @param eventPoll event poll * @param path path to the socket */ - Service(const std::string& path, + Service(epoll::EventPoll& eventPoll, + const std::string& path, const PeerCallback& addPeerCallback = nullptr, const PeerCallback& removePeerCallback = nullptr); ~Service(); @@ -62,11 +60,9 @@ public: Service& operator=(const Service&) = delete; /** - * Starts the worker and acceptor threads - * - * @param usesExternalPolling internal or external polling is used + * Starts processing */ - void start(const bool usesExternalPolling = false); + void start(); /** * @return is the communication thread running @@ -79,16 +75,6 @@ public: void stop(); /** - * Used with an external polling loop. - * Handles one event from the file descriptor. - * - * @param fd file descriptor - * @param pollEvent event on the fd. Defined in poll.h - * - */ - void handle(const FileDescriptor fd, const short pollEvent); - - /** * Set the callback called for each new connection to a peer * * @param newPeerCallback the callback @@ -175,14 +161,11 @@ public: void signal(const MethodID methodID, const std::shared_ptr& data); private: - - void startPoll(); - void stopPoll(); - - typedef std::lock_guard Lock; + epoll::EventPoll& mEventPoll; Processor mProcessor; Acceptor mAcceptor; - IPCGSource::Pointer mIPCGSourcePtr; + + void handle(const FileDescriptor fd, const epoll::Events pollEvents); }; diff --git a/tests/scripts/vsm_test_parser.py b/tests/scripts/vsm_test_parser.py index 41dd495..d0d4fa7 100644 --- a/tests/scripts/vsm_test_parser.py +++ b/tests/scripts/vsm_test_parser.py @@ -35,7 +35,7 @@ class Logger(object): __indentChar = " " def testCaseSummary(self, testSuite, testName, testResult, recLevel): - msg = self.__indentChar * recLevel + BOLD + "{:<50}".format(testName) + msg = self.__indentChar * recLevel + BOLD + "{:<50} ".format(testName) if testResult == "passed": msg += GREEN diff --git a/tests/unit_tests/epoll/ut-event-poll.cpp b/tests/unit_tests/epoll/ut-event-poll.cpp index 0bcbe81..00cb385 100644 --- a/tests/unit_tests/epoll/ut-event-poll.cpp +++ b/tests/unit_tests/epoll/ut-event-poll.cpp @@ -31,8 +31,8 @@ #include "ipc/internals/socket.hpp" #include "utils/latch.hpp" #include "utils/glib-loop.hpp" -#include "epoll/glib-poll-dispatcher.hpp" -#include "epoll/thread-poll-dispatcher.hpp" +#include "epoll/glib-dispatcher.hpp" +#include "epoll/thread-dispatcher.hpp" using namespace vasum::utils; using namespace vasum::epoll; @@ -54,14 +54,14 @@ BOOST_AUTO_TEST_CASE(EmptyPoll) BOOST_AUTO_TEST_CASE(ThreadedPoll) { - ThreadPollDispatcher dispatcher; + ThreadDispatcher dispatcher; } BOOST_AUTO_TEST_CASE(GlibPoll) { ScopedGlibLoop loop; - GlibPollDispatcher dispatcher; + GlibDispatcher dispatcher; } void doSocketTest(EventPoll& poll) @@ -126,7 +126,7 @@ void doSocketTest(EventPoll& poll) BOOST_AUTO_TEST_CASE(ThreadedPollSocket) { - ThreadPollDispatcher dispatcher; + ThreadDispatcher dispatcher; doSocketTest(dispatcher.getPoll()); } @@ -135,14 +135,14 @@ BOOST_AUTO_TEST_CASE(GlibPollSocket) { ScopedGlibLoop loop; - GlibPollDispatcher dispatcher; + GlibDispatcher dispatcher; doSocketTest(dispatcher.getPoll()); } BOOST_AUTO_TEST_CASE(PollStacking) { - ThreadPollDispatcher dispatcher; + ThreadDispatcher dispatcher; EventPoll innerPoll; diff --git a/tests/unit_tests/ipc/ut-ipc.cpp b/tests/unit_tests/ipc/ut-ipc.cpp index 088f576..8f47a65 100644 --- a/tests/unit_tests/ipc/ut-ipc.cpp +++ b/tests/unit_tests/ipc/ut-ipc.cpp @@ -34,6 +34,8 @@ #include "ipc/client.hpp" #include "ipc/types.hpp" #include "ipc/result.hpp" +#include "epoll/thread-dispatcher.hpp" +#include "epoll/glib-dispatcher.hpp" #include "utils/glib-loop.hpp" #include "utils/latch.hpp" #include "utils/value-latch.hpp" @@ -51,11 +53,10 @@ using namespace vasum; using namespace vasum::ipc; +using namespace vasum::epoll; using namespace vasum::utils; using namespace std::placeholders; -namespace { - // Timeout for sending one message const int TIMEOUT = 1000 /*ms*/; @@ -68,15 +69,28 @@ const int LONG_OPERATION_TIME = 1000 + TIMEOUT; const std::string TEST_DIR = "/tmp/ut-ipc"; const std::string SOCKET_PATH = TEST_DIR + "/test.socket"; -struct Fixture { +struct FixtureBase { ScopedDir mTestPathGuard; - Fixture() + FixtureBase() : mTestPathGuard(TEST_DIR) { } }; +struct ThreadedFixture : FixtureBase { + ThreadDispatcher dispatcher; + + EventPoll& getPoll() { return dispatcher.getPoll(); } +}; + +struct GlibFixture : FixtureBase { + ScopedGlibLoop glibLoop; + GlibDispatcher dispatcher; + + EventPoll& getPoll() { return dispatcher.getPoll(); } +}; + struct SendData { int intVal; SendData(int i): intVal(i) {} @@ -167,7 +181,7 @@ void longEchoCallback(const PeerID, methodResult->set(returnData); } -PeerID connect(Service& s, Client& c, bool isServiceGlib = false, bool isClientGlib = false) +PeerID connect(Service& s, Client& c) { // Connects the Client to the Service and returns Clients PeerID ValueLatch peerIDLatch; @@ -178,10 +192,10 @@ PeerID connect(Service& s, Client& c, bool isServiceGlib = false, bool isClientG s.setNewPeerCallback(newPeerCallback); if (!s.isStarted()) { - s.start(isServiceGlib); + s.start(); } - c.start(isClientGlib); + c.start(); PeerID peerID = peerIDLatch.get(TIMEOUT); s.setNewPeerCallback(nullptr); @@ -189,16 +203,6 @@ PeerID connect(Service& s, Client& c, bool isServiceGlib = false, bool isClientG return peerID; } -PeerID connectServiceGSource(Service& s, Client& c) -{ - return connect(s, c, true, false); -} - -PeerID connectClientGSource(Service& s, Client& c) -{ - return connect(s, c, false, true); -} - void testEcho(Client& c, const MethodID methodID) { std::shared_ptr sentData(new SendData(34)); @@ -215,20 +219,17 @@ void testEcho(Service& s, const MethodID methodID, const PeerID peerID) BOOST_CHECK_EQUAL(recvData->intVal, sentData->intVal); } -} // namespace - - -BOOST_FIXTURE_TEST_SUITE(IPCSuite, Fixture) +BOOST_AUTO_TEST_SUITE(IPCSuite) -BOOST_AUTO_TEST_CASE(ConstructorDestructor) +MULTI_FIXTURE_TEST_CASE(ConstructorDestructor, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); } -BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) +MULTI_FIXTURE_TEST_CASE(ServiceAddRemoveMethod, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, returnEmptyCallback); s.setMethodHandler(1, returnDataCallback); @@ -237,7 +238,7 @@ BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) s.setMethodHandler(1, echoCallback); s.setMethodHandler(2, returnDataCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); testEcho(c, 1); @@ -247,10 +248,10 @@ BOOST_AUTO_TEST_CASE(ServiceAddRemoveMethod) BOOST_CHECK_THROW(testEcho(c, 2), IPCException); } -BOOST_AUTO_TEST_CASE(ClientAddRemoveMethod) +MULTI_FIXTURE_TEST_CASE(ClientAddRemoveMethod, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.setMethodHandler(1, returnEmptyCallback); c.setMethodHandler(1, returnDataCallback); @@ -267,9 +268,9 @@ BOOST_AUTO_TEST_CASE(ClientAddRemoveMethod) BOOST_CHECK_THROW(testEcho(s, 1, peerID), IPCException); } -BOOST_AUTO_TEST_CASE(ServiceStartStop) +MULTI_FIXTURE_TEST_CASE(ServiceStartStop, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, returnDataCallback); @@ -282,10 +283,10 @@ BOOST_AUTO_TEST_CASE(ServiceStartStop) s.start(); } -BOOST_AUTO_TEST_CASE(ClientStartStop) +MULTI_FIXTURE_TEST_CASE(ClientStartStop, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.setMethodHandler(1, returnDataCallback); c.start(); @@ -300,27 +301,27 @@ BOOST_AUTO_TEST_CASE(ClientStartStop) c.stop(); } -BOOST_AUTO_TEST_CASE(SyncClientToServiceEcho) +MULTI_FIXTURE_TEST_CASE(SyncClientToServiceEcho, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.setMethodHandler(2, echoCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); testEcho(c, 1); testEcho(c, 2); } -BOOST_AUTO_TEST_CASE(Restart) +MULTI_FIXTURE_TEST_CASE(Restart, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); s.setMethodHandler(2, echoCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.start(); testEcho(c, 1); testEcho(c, 2); @@ -334,14 +335,19 @@ BOOST_AUTO_TEST_CASE(Restart) s.stop(); s.start(); + BOOST_CHECK_THROW(testEcho(c, 2), IPCException); + + c.stop(); + c.start(); + testEcho(c, 1); testEcho(c, 2); } -BOOST_AUTO_TEST_CASE(SyncServiceToClientEcho) +MULTI_FIXTURE_TEST_CASE(SyncServiceToClientEcho, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.setMethodHandler(1, echoCallback); PeerID peerID = connect(s, c); @@ -351,16 +357,16 @@ BOOST_AUTO_TEST_CASE(SyncServiceToClientEcho) BOOST_CHECK_EQUAL(recvData->intVal, sentData->intVal); } -BOOST_AUTO_TEST_CASE(AsyncClientToServiceEcho) +MULTI_FIXTURE_TEST_CASE(AsyncClientToServiceEcho, F, ThreadedFixture, GlibFixture) { std::shared_ptr sentData(new SendData(34)); ValueLatch> recvDataLatch; // Setup Service and Client - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.start(); //Async call @@ -374,13 +380,13 @@ BOOST_AUTO_TEST_CASE(AsyncClientToServiceEcho) BOOST_CHECK_EQUAL(recvData->intVal, sentData->intVal); } -BOOST_AUTO_TEST_CASE(AsyncServiceToClientEcho) +MULTI_FIXTURE_TEST_CASE(AsyncServiceToClientEcho, F, ThreadedFixture, GlibFixture) { std::shared_ptr sentData(new SendData(56)); ValueLatch> recvDataLatch; - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.setMethodHandler(1, echoCallback); PeerID peerID = connect(s, c); @@ -397,24 +403,24 @@ BOOST_AUTO_TEST_CASE(AsyncServiceToClientEcho) } -BOOST_AUTO_TEST_CASE(SyncTimeout) +MULTI_FIXTURE_TEST_CASE(SyncTimeout, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, longEchoCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); std::shared_ptr sentData(new SendData(78)); BOOST_REQUIRE_THROW((c.callSync(1, sentData, TIMEOUT)), IPCException); } -BOOST_AUTO_TEST_CASE(SerializationError) +MULTI_FIXTURE_TEST_CASE(SerializationError, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); std::shared_ptr throwingData(new ThrowOnAcceptData()); @@ -423,23 +429,24 @@ BOOST_AUTO_TEST_CASE(SerializationError) } -BOOST_AUTO_TEST_CASE(ParseError) +MULTI_FIXTURE_TEST_CASE(ParseError, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.start(); std::shared_ptr sentData(new SendData(78)); BOOST_CHECK_THROW((c.callSync(1, sentData, 10000)), IPCParsingException); } -BOOST_AUTO_TEST_CASE(DisconnectedPeerError) +MULTI_FIXTURE_TEST_CASE(DisconnectedPeerError, F, ThreadedFixture, GlibFixture) { ValueLatch> retStatusLatch; - Service s(SOCKET_PATH); + + Service s(F::getPoll(), SOCKET_PATH); auto method = [](const PeerID, std::shared_ptr&, MethodResult::Pointer methodResult) { auto resultData = std::make_shared(1); @@ -450,7 +457,7 @@ BOOST_AUTO_TEST_CASE(DisconnectedPeerError) s.setMethodHandler(1, method); s.start(); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.start(); auto dataBack = [&retStatusLatch](Result && r) { @@ -470,16 +477,16 @@ BOOST_AUTO_TEST_CASE(DisconnectedPeerError) } -BOOST_AUTO_TEST_CASE(ReadTimeout) +MULTI_FIXTURE_TEST_CASE(ReadTimeout, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); auto longEchoCallback = [](const PeerID, std::shared_ptr& data, MethodResult::Pointer methodResult) { auto resultData = std::make_shared(data->intVal, LONG_OPERATION_TIME); methodResult->set(resultData); }; s.setMethodHandler(1, longEchoCallback); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); // Test timeout on read @@ -488,13 +495,13 @@ BOOST_AUTO_TEST_CASE(ReadTimeout) } -BOOST_AUTO_TEST_CASE(WriteTimeout) +MULTI_FIXTURE_TEST_CASE(WriteTimeout, F, ThreadedFixture, GlibFixture) { - Service s(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); s.setMethodHandler(1, echoCallback); s.start(); - Client c(SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); c.start(); // Test echo with a minimal timeout @@ -509,13 +516,13 @@ BOOST_AUTO_TEST_CASE(WriteTimeout) } -BOOST_AUTO_TEST_CASE(AddSignalInRuntime) +MULTI_FIXTURE_TEST_CASE(AddSignalInRuntime, F, ThreadedFixture, GlibFixture) { ValueLatch> recvDataLatchA; ValueLatch> recvDataLatchB; - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); connect(s, c); auto handlerA = [&recvDataLatchA](const PeerID, std::shared_ptr& data) { @@ -545,13 +552,13 @@ BOOST_AUTO_TEST_CASE(AddSignalInRuntime) } -BOOST_AUTO_TEST_CASE(AddSignalOffline) +MULTI_FIXTURE_TEST_CASE(AddSignalOffline, F, ThreadedFixture, GlibFixture) { ValueLatch> recvDataLatchA; ValueLatch> recvDataLatchB; - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); auto handlerA = [&recvDataLatchA](const PeerID, std::shared_ptr& data) { recvDataLatchA.set(data); @@ -581,66 +588,13 @@ BOOST_AUTO_TEST_CASE(AddSignalOffline) BOOST_CHECK_EQUAL(recvDataB->intVal, sendDataB->intVal); } - -BOOST_AUTO_TEST_CASE(ServiceGSource) -{ - utils::Latch l; - ScopedGlibLoop loop; - - auto signalHandler = [&l](const PeerID, std::shared_ptr&) { - l.set(); - }; - - Service s(SOCKET_PATH); - s.setMethodHandler(1, echoCallback); - - Client c(SOCKET_PATH); - s.setSignalHandler(2, signalHandler); - - connectServiceGSource(s, c); - - testEcho(c, 1); - - auto data = std::make_shared(1); - c.signal(2, data); - - BOOST_CHECK(l.wait(TIMEOUT)); -} - - -BOOST_AUTO_TEST_CASE(ClientGSource) -{ - utils::Latch l; - ScopedGlibLoop loop; - - auto signalHandler = [&l](const PeerID, std::shared_ptr&) { - l.set(); - }; - - Service s(SOCKET_PATH); - s.start(); - - Client c(SOCKET_PATH); - c.setMethodHandler(1, echoCallback); - c.setSignalHandler(2, signalHandler); - - PeerID peerID = connectClientGSource(s, c); - - testEcho(s, 1, peerID); - - auto data = std::make_shared(1); - s.signal(2, data); - - BOOST_CHECK(l.wait(TIMEOUT)); -} - -BOOST_AUTO_TEST_CASE(UsersError) +MULTI_FIXTURE_TEST_CASE(UsersError, F, ThreadedFixture, GlibFixture) { const int TEST_ERROR_CODE = -234; const std::string TEST_ERROR_MESSAGE = "Ay, caramba!"; - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); auto clientID = connect(s, c); auto throwingMethodHandler = [&](const PeerID, std::shared_ptr&, MethodResult::Pointer) { @@ -668,13 +622,13 @@ BOOST_AUTO_TEST_CASE(UsersError) BOOST_CHECK_EXCEPTION((s.callSync(2, clientID, sentData, TIMEOUT)), IPCUserException, hasProperData); } -BOOST_AUTO_TEST_CASE(AsyncResult) +MULTI_FIXTURE_TEST_CASE(AsyncResult, F, ThreadedFixture, GlibFixture) { const int TEST_ERROR_CODE = -567; const std::string TEST_ERROR_MESSAGE = "Ooo jooo!"; - Service s(SOCKET_PATH); - Client c(SOCKET_PATH); + Service s(F::getPoll(), SOCKET_PATH); + Client c(F::getPoll(), SOCKET_PATH); auto clientID = connect(s, c); auto errorMethodHandler = [&](const PeerID, std::shared_ptr&, MethodResult::Pointer methodResult) { @@ -724,20 +678,44 @@ BOOST_AUTO_TEST_CASE(AsyncResult) BOOST_CHECK_EQUAL(recvData->intVal, sentData->intVal); } -// BOOST_AUTO_TEST_CASE(ConnectionLimitTest) +MULTI_FIXTURE_TEST_CASE(MixOperations, F, ThreadedFixture, GlibFixture) +{ + utils::Latch l; + + auto signalHandler = [&l](const PeerID, std::shared_ptr&) { + l.set(); + }; + + Service s(F::getPoll(), SOCKET_PATH); + s.setMethodHandler(1, echoCallback); + + Client c(F::getPoll(), SOCKET_PATH); + s.setSignalHandler(2, signalHandler); + + connect(s, c); + + testEcho(c, 1); + + auto data = std::make_shared(1); + c.signal(2, data); + + BOOST_CHECK(l.wait(TIMEOUT)); +} + +// MULTI_FIXTURE_TEST_CASE(ConnectionLimitTest, F, ThreadedFixture, GlibFixture) // { // unsigned oldLimit = ipc::getMaxFDNumber(); // ipc::setMaxFDNumber(50); // // Setup Service and many Clients -// Service s(SOCKET_PATH); +// Service s(F::getPoll(), SOCKET_PATH); // s.setMethodHandler(1, echoCallback); // s.start(); // std::list clients; // for (int i = 0; i < 100; ++i) { // try { -// clients.push_back(Client(SOCKET_PATH)); +// clients.push_back(Client(F::getPoll(), SOCKET_PATH)); // clients.back().start(); // } catch (...) {} // } diff --git a/tests/unit_tests/ut.hpp b/tests/unit_tests/ut.hpp index d8b4b94..e96d363 100644 --- a/tests/unit_tests/ut.hpp +++ b/tests/unit_tests/ut.hpp @@ -28,9 +28,23 @@ #define BOOST_TEST_DYN_LINK #include +#include + #include /** + * Usage example: + * + * MULTI_FIXTURE_TEST_CASE(Test, T, Fixture1, Fixture2, Fixture3) { + * std::cout << T::i << "\n"; + * } + */ +#define MULTI_FIXTURE_TEST_CASE(NAME, TPARAM, ...) \ + typedef boost::mpl::vector<__VA_ARGS__> NAME##_fixtures; \ + BOOST_FIXTURE_TEST_CASE_TEMPLATE(NAME, TPARAM, NAME##_fixtures, TPARAM) + + +/** * An exception message checker * * Usage example: -- 2.7.4