moved 'owning states' to port granularity instead of machine

9 files changed, 174 insertions, 111 deletions

diff --git a/src/gtest/rpc_test.cpp b/src/gtest/rpc_test.cpp
index 015da2f8..168ee936 100644
--- a/src/gtest/rpc_test.cpp
+++ b/src/gtest/rpc_test.cpp
@@ -113,10 +113,11 @@ public:
 class RpcTestOwned : public RpcTest {
 public:
 
-    void create_request(Json::Value &request, const string &method, int id = 1, bool owned = true)  {
+    void create_request(Json::Value &request, const string &method, int id = 1, int port_id = 1, bool owned = true)  {
         RpcTest::create_request(request, method, id);
         if (owned) {
-            request["params"]["handler"] = m_ownership_handler; 
+            request["params"]["port_id"] = port_id;
+            request["params"]["handler"] = m_ownership_handler[port_id];
         }
     }
 
@@ -124,37 +125,43 @@ protected:
 
     virtual void SetUp() {
         RpcTest::SetUp();
-        take_ownership();
+
+        for (int i = 0 ; i < 4; i++) {
+            m_ownership_handler[i] = take_ownership(i);
+        }
     }
 
 
-    void take_ownership(void) {
+    string take_ownership(uint8_t port_id) {
         Json::Value request;
         Json::Value response;
 
         RpcTest::create_request(request, "acquire", 1);
 
+        request["params"]["port_id"] = port_id;
         request["params"]["user"] = "test";
         request["params"]["force"] = true;
 
         send_request(request, response);
 
         EXPECT_TRUE(response["result"] != Json::nullValue);
-        m_ownership_handler = response["result"].asString();
+        return response["result"].asString();
     }
 
-    void release_ownership() {
+    void release_ownership(uint8_t port_id) {
         Json::Value request;
         Json::Value response;
 
         RpcTest::create_request(request, "release", 1);
+
         request["params"]["handler"] = m_ownership_handler;
+        request["params"]["port_id"] = port_id;
 
         send_request(request, response);
         EXPECT_TRUE(response["result"] == "ACK");
     }
 
-    string m_ownership_handler;
+    string m_ownership_handler[4];
 };
 
 TEST_F(RpcTest, basic_rpc_negative_cases) {
@@ -383,6 +390,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
 
     /* no user before acquring */
     create_request(request, "get_owner");
+    request["params"]["port_id"] = 1;
     send_request(request, response);
     EXPECT_TRUE(response["result"] != Json::nullValue);
 
@@ -390,6 +398,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
 
     /* soft acquire */
     create_request(request, "acquire");
+    request["params"]["port_id"] = 1;
     request["params"]["user"] = "itay";
     request["params"]["force"] = false;
 
@@ -397,6 +406,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
     EXPECT_TRUE(response["result"] != Json::nullValue);
 
     create_request(request, "get_owner");
+    request["params"]["port_id"] = 1;
     send_request(request, response);
     EXPECT_TRUE(response["result"] != Json::nullValue);
 
@@ -404,6 +414,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
 
     /* hard acquire */
     create_request(request, "acquire");
+    request["params"]["port_id"] = 1;
     request["params"]["user"] = "moshe";
     request["params"]["force"] = false;
 
@@ -419,6 +430,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
 
     /* make sure */
     create_request(request, "get_owner");
+    request["params"]["port_id"] = 1;
     send_request(request, response);
     EXPECT_TRUE(response["result"] != Json::nullValue);
 
@@ -426,6 +438,7 @@ TEST_F(RpcTest, get_owner_acquire_release) {
 
     /* release */
     create_request(request, "release");
+    request["params"]["port_id"] = 1;
     request["params"]["handler"] = handler;
     send_request(request, response);
 
@@ -463,9 +476,8 @@ TEST_F(RpcTestOwned, add_remove_stream) {
     Json::Value response;
 
     /* verify no such stream */
-    create_request(request, "get_stream", 1);
+    create_request(request, "get_stream", 1, 1);
 
-    request["params"]["port_id"] = 1;
     request["params"]["stream_id"] = 5;
 
     send_request(request, response);
@@ -475,8 +487,7 @@ TEST_F(RpcTestOwned, add_remove_stream) {
     EXPECT_EQ(response["error"]["code"], -32000);
 
     /* add it */
-    create_request(request, "add_stream", 1);
-    request["params"]["port_id"] = 1;
+    create_request(request, "add_stream", 1, 1);
     request["params"]["stream_id"] = 5;
 
     Json::Value stream;
@@ -488,9 +499,8 @@ TEST_F(RpcTestOwned, add_remove_stream) {
     EXPECT_EQ(response["result"], "ACK");
 
     /* get it */
-    create_request(request, "get_stream", 1);
+    create_request(request, "get_stream", 1, 1);
 
-    request["params"]["port_id"] = 1;
     request["params"]["stream_id"] = 5;
 
     send_request(request, response);
@@ -498,9 +508,8 @@ TEST_F(RpcTestOwned, add_remove_stream) {
     EXPECT_TRUE(compare_streams(stream, response["result"]["stream"]));
 
     // remove it
-    create_request(request, "remove_stream", 1);
+    create_request(request, "remove_stream", 1, 1);
 
-    request["params"]["port_id"] = 1;
     request["params"]["stream_id"] = 5;
 
     send_request(request, response);
@@ -520,7 +529,7 @@ TEST_F(RpcTestOwned, get_stream_id_list) {
     Json::Value response;
     
      /* add stream 1 */
-    create_request(request, "add_stream");
+    create_request(request, "add_stream", 1);
     request["params"]["port_id"] = 1;
 
     Json::Value stream;
@@ -578,8 +587,7 @@ TEST_F(RpcTestOwned, start_stop_traffic) {
     Json::Value response;
 
     /* add stream #1 */
-    create_request(request, "add_stream");
-    request["params"]["port_id"] = 1;
+    create_request(request, "add_stream", 1, 1);
     request["params"]["stream_id"] = 5;
 
     Json::Value stream;
@@ -590,9 +598,8 @@ TEST_F(RpcTestOwned, start_stop_traffic) {
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
-      /* add stream #1 */
-    create_request(request, "add_stream");
-    request["params"]["port_id"] = 3;
+    /* add stream #1 */
+    create_request(request, "add_stream", 1, 3);
     request["params"]["stream_id"] = 12;
     request["params"]["stream"] = stream;
     
@@ -600,52 +607,53 @@ TEST_F(RpcTestOwned, start_stop_traffic) {
     EXPECT_EQ(response["result"], "ACK");
 
     /* start port 1 */
-    create_request(request, "start_traffic");
-    request["params"]["port_id"] = 1;
+    create_request(request, "start_traffic", 1, 1);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
 
     /* start port 3 */
-    request["params"]["port_id"] = 3;
+    create_request(request, "start_traffic", 1, 3);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
     /* start not configured port */
-
-    request["params"]["port_id"] = 2;
+    create_request(request, "start_traffic", 1, 2);
     send_request(request, response);
     EXPECT_EQ(response["error"]["code"], -32000);
 
     /* stop port 1 */
-    create_request(request, "stop_traffic");
-    request["params"]["port_id"] = 1;
+    create_request(request, "stop_traffic", 1, 1);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
     /* stop port 3 */
-    request["params"]["port_id"] = 3;
+    create_request(request, "stop_traffic", 1, 3);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
     /* start 1 again */
-    create_request(request, "start_traffic");
-    request["params"]["port_id"] = 1;
+    create_request(request, "start_traffic", 1, 1);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 
     /* start 1 twice (error) */
+    create_request(request, "start_traffic", 1, 1);
     send_request(request, response);
     EXPECT_EQ(response["error"]["code"], -32000);
 
     /* make sure you cannot release while traffic is active */
-    //create_request(request, "release");
-    //send_request(request, response);
-    //EXPECT_EQ(response["error"]["code"], -32000);
+    create_request(request, "release", 1, 1);
+    send_request(request, response);
+    EXPECT_EQ(response["error"]["code"], -32000);
 
-    /* done */
-    create_request(request, "stop_traffic");
-    request["params"]["port_id"] = 1;
+    /* stop traffic on port #1 */
+    create_request(request, "stop_traffic",1 ,1);
+    send_request(request, response);
+    EXPECT_EQ(response["result"], "ACK");
+
+    /* release */
+    create_request(request, "release", 1, 1);
     send_request(request, response);
     EXPECT_EQ(response["result"], "ACK");
 }
diff --git a/src/rpc-server/commands/trex_rpc_cmd_general.cpp b/src/rpc-server/commands/trex_rpc_cmd_general.cpp
index 97ccae06..106a167a 100644
--- a/src/rpc-server/commands/trex_rpc_cmd_general.cpp
+++ b/src/rpc-server/commands/trex_rpc_cmd_general.cpp
@@ -175,6 +175,8 @@ TrexRpcCmdGetSysInfo::_run(const Json::Value &params, Json::Value &result) {
         section["ports"][i]["driver"]  = driver;
         section["ports"][i]["speed"]   = speed;
 
+        section["ports"][i]["owner"] = port->get_owner();
+
         switch (port->get_state()) {
         case TrexStatelessPort::PORT_STATE_DOWN:
             section["ports"][i]["status"] = "down";
@@ -208,7 +210,10 @@ trex_rpc_cmd_rc_e
 TrexRpcCmdGetOwner::_run(const Json::Value &params, Json::Value &result) {
     Json::Value &section = result["result"];
 
-    section["owner"] = TrexRpcServer::get_owner();
+    uint8_t port_id = parse_port(params, result);
+
+    TrexStatelessPort *port = TrexStateless::get_instance().get_port_by_id(port_id);
+    section["owner"] = port->get_owner();
 
     return (TREX_RPC_CMD_OK);
 }
@@ -220,17 +225,21 @@ TrexRpcCmdGetOwner::_run(const Json::Value &params, Json::Value &result) {
 trex_rpc_cmd_rc_e
 TrexRpcCmdAcquire::_run(const Json::Value &params, Json::Value &result) {
 
+    uint8_t port_id = parse_port(params, result);
+
     const string &new_owner = parse_string(params, "user", result);
     bool force = parse_bool(params, "force", result);
 
     /* if not free and not you and not force - fail */
-    if ( (!TrexRpcServer::is_free_to_aquire()) && (TrexRpcServer::get_owner() != new_owner) && (!force)) {
-        generate_execute_err(result, "device is already taken by '" + TrexRpcServer::get_owner() + "'");
+    TrexStatelessPort *port = TrexStateless::get_instance().get_port_by_id(port_id);
+
+    if ( (!port->is_free_to_aquire()) && (port->get_owner() != new_owner) && (!force)) {
+        generate_execute_err(result, "device is already taken by '" + port->get_owner() + "'");
     }
 
-    string handle = TrexRpcServer::set_owner(new_owner);
+    port->set_owner(new_owner);
 
-    result["result"] = handle;
+    result["result"] = port->get_owner_handler();
 
     return (TREX_RPC_CMD_OK);
 }
@@ -242,7 +251,15 @@ TrexRpcCmdAcquire::_run(const Json::Value &params, Json::Value &result) {
 trex_rpc_cmd_rc_e
 TrexRpcCmdRelease::_run(const Json::Value &params, Json::Value &result) {
 
-    TrexRpcServer::clear_owner();
+    uint8_t port_id = parse_port(params, result);
+
+    TrexStatelessPort *port = TrexStateless::get_instance().get_port_by_id(port_id);
+
+    if (port->get_state() == TrexStatelessPort::PORT_STATE_TRANSMITTING) {
+        generate_execute_err(result, "cannot release a port during transmission");
+    }
+
+    port->clear_owner();
 
     result["result"] = "ACK";
 
diff --git a/src/rpc-server/commands/trex_rpc_cmds.h b/src/rpc-server/commands/trex_rpc_cmds.h
index 643aa22d..e261d1c6 100644
--- a/src/rpc-server/commands/trex_rpc_cmds.h
+++ b/src/rpc-server/commands/trex_rpc_cmds.h
@@ -69,9 +69,9 @@ void get_hostname(std::string &hostname);
 /**
  * ownership
  */
-TREX_RPC_CMD_DEFINE(TrexRpcCmdGetOwner,   "get_owner",       0, false);
-TREX_RPC_CMD_DEFINE(TrexRpcCmdAcquire,    "acquire",         2, false);
-TREX_RPC_CMD_DEFINE(TrexRpcCmdRelease,    "release",         0, true);
+TREX_RPC_CMD_DEFINE(TrexRpcCmdGetOwner,   "get_owner",       1, false);
+TREX_RPC_CMD_DEFINE(TrexRpcCmdAcquire,    "acquire",         3, false);
+TREX_RPC_CMD_DEFINE(TrexRpcCmdRelease,    "release",         1, true);
 
 
 /**
diff --git a/src/rpc-server/trex_rpc_cmd.cpp b/src/rpc-server/trex_rpc_cmd.cpp
index 437e8b1a..6c355e70 100644
--- a/src/rpc-server/trex_rpc_cmd.cpp
+++ b/src/rpc-server/trex_rpc_cmd.cpp
@@ -20,6 +20,7 @@ limitations under the License.
 */
 #include <trex_rpc_cmd_api.h>
 #include <trex_rpc_server_api.h>
+#include <trex_stateless_api.h>
 
 trex_rpc_cmd_rc_e 
 TrexRpcCommand::run(const Json::Value &params, Json::Value &result) {
@@ -57,12 +58,32 @@ TrexRpcCommand::check_param_count(const Json::Value &params, int expected, Json:
 void
 TrexRpcCommand::verify_ownership(const Json::Value &params, Json::Value &result) {
     std::string handler = parse_string(params, "handler", result);
+    uint8_t port_id = parse_port(params, result);
 
-    if (!TrexRpcServer::verify_owner_handler(handler)) {
+    TrexStatelessPort *port = TrexStateless::get_instance().get_port_by_id(port_id);
+
+    if (!port->verify_owner_handler(handler)) {
         generate_execute_err(result, "invalid handler provided. please pass the handler given when calling 'acquire' or take ownership");
     }
 }
 
+uint8_t 
+TrexRpcCommand::parse_port(const Json::Value &params, Json::Value &result) {
+    uint8_t port_id = parse_byte(params, "port_id", result);
+    validate_port_id(port_id, result);
+
+    return (port_id);
+}
+
+void 
+TrexRpcCommand::validate_port_id(uint8_t port_id, Json::Value &result) {
+    if (port_id >= TrexStateless::get_instance().get_port_count()) {
+        std::stringstream ss;
+        ss << "invalid port id - should be between 0 and " << (int)TrexStateless::get_instance().get_port_count() - 1;
+        generate_execute_err(result, ss.str());
+    }
+}
+
 const char *
 TrexRpcCommand::type_to_str(field_type_e type) {
     switch (type) {
diff --git a/src/rpc-server/trex_rpc_cmd_api.h b/src/rpc-server/trex_rpc_cmd_api.h
index cab50cfd..3c718eaa 100644
--- a/src/rpc-server/trex_rpc_cmd_api.h
+++ b/src/rpc-server/trex_rpc_cmd_api.h
@@ -125,6 +125,12 @@ protected:
     void verify_ownership(const Json::Value &params, Json::Value &result);
 
     /**
+     * validate port id
+     * 
+     */
+    void validate_port_id(uint8_t port_id, Json::Value &result);
+
+    /**
      * parse functions
      * 
      */
@@ -146,6 +152,9 @@ protected:
     const Json::Value & parse_object(const Json::Value &parent, int index, Json::Value &result);
     const Json::Value & parse_array(const Json::Value &parent, int index, Json::Value &result);
 
+    /* shortcut for parsing port id */
+    uint8_t parse_port(const Json::Value &params, Json::Value &result);
+
     /**
      * parse a field from choices 
      * 
diff --git a/src/rpc-server/trex_rpc_server.cpp b/src/rpc-server/trex_rpc_server.cpp
index a5988cab..6b8c200d 100644
--- a/src/rpc-server/trex_rpc_server.cpp
+++ b/src/rpc-server/trex_rpc_server.cpp
@@ -111,8 +111,6 @@ get_current_date_time() {
 }
 
 const std::string TrexRpcServer::s_server_uptime = get_current_date_time();
-std::string TrexRpcServer::s_owner = "none";
-std::string TrexRpcServer::s_owner_handler = "";
 
 TrexRpcServer::TrexRpcServer(const TrexRpcServerConfig &req_resp_cfg) {
 
@@ -158,22 +156,3 @@ void TrexRpcServer::set_verbose(bool verbose) {
     }
 }
 
-/**
- * generate a random connection handler
- * 
- */
-std::string TrexRpcServer::generate_handler() {
-    std::stringstream ss;
-
-    static const char alphanum[] =
-        "0123456789"
-        "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
-        "abcdefghijklmnopqrstuvwxyz";
-
-    /* generate 8 bytes of random handler */
-    for (int i = 0; i < 8; ++i) {
-        ss << alphanum[rand() % (sizeof(alphanum) - 1)];
-    }
-
-    return (ss.str());
-}
diff --git a/src/rpc-server/trex_rpc_server_api.h b/src/rpc-server/trex_rpc_server_api.h
index c34ac0f8..06bbe10c 100644
--- a/src/rpc-server/trex_rpc_server_api.h
+++ b/src/rpc-server/trex_rpc_server_api.h
@@ -164,49 +164,7 @@ public:
     }
 
 
-    /**
-     * query for ownership
-     * 
-     */
-    static const std::string &get_owner() {
-        return s_owner;
-    }
-
-    /**
-     * owner handler 
-     * for the connection 
-     * 
-     */
-    static const std::string &get_owner_handler() {
-        return s_owner_handler;
-    }
-
-    static bool is_free_to_aquire() {
-        return (s_owner == "none");
-    }
-
-    /**
-    * take ownership of the server array 
-    * this is static 
-    * ownership is total 
-    * 
-    */
-    static std::string set_owner(const std::string &owner) {
-        s_owner = owner;
-        s_owner_handler = generate_handler();
-        return (s_owner_handler);
-    }
-
-    static void clear_owner() {
-        s_owner = "none";
-        s_owner_handler = "";
-    }
-
-    static bool verify_owner_handler(const std::string &handler) {
-
-        return ( (s_owner != "none") && (s_owner_handler == handler) );
-
-    }
+   
 
 private:
     static std::string generate_handler();
diff --git a/src/stateless/trex_stateless.cpp b/src/stateless/trex_stateless.cpp
index dcad1514..6a3169d4 100644
--- a/src/stateless/trex_stateless.cpp
+++ b/src/stateless/trex_stateless.cpp
@@ -79,6 +79,7 @@ uint8_t TrexStateless::get_port_count() {
  **************************/
 TrexStatelessPort::TrexStatelessPort(uint8_t port_id) : m_port_id(port_id) {
     m_port_state = PORT_STATE_UP_IDLE;
+    clear_owner();
 }
 
 
@@ -127,3 +128,25 @@ TrexStatelessPort::get_properties(string &driver, string &speed) {
     driver = "Unknown Driver";
     speed  = "Unknown Speed";
 }
+
+
+/**
+ * generate a random connection handler
+ * 
+ */
+std::string 
+TrexStatelessPort::generate_handler() {
+    std::stringstream ss;
+
+    static const char alphanum[] =
+        "0123456789"
+        "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        "abcdefghijklmnopqrstuvwxyz";
+
+    /* generate 8 bytes of random handler */
+    for (int i = 0; i < 8; ++i) {
+        ss << alphanum[rand() % (sizeof(alphanum) - 1)];
+    }
+
+    return (ss.str());
+}
diff --git a/src/stateless/trex_stateless_api.h b/src/stateless/trex_stateless_api.h
index 60d26878..e02e93da 100644
--- a/src/stateless/trex_stateless_api.h
+++ b/src/stateless/trex_stateless_api.h
@@ -106,10 +106,58 @@ public:
      */
     void get_properties(std::string &driver, std::string &speed);
 
+    /**
+    * query for ownership
+    * 
+    */
+    const std::string &get_owner() {
+        return m_owner;
+    }
+
+    /**
+     * owner handler 
+     * for the connection 
+     * 
+     */
+    const std::string &get_owner_handler() {
+        return m_owner_handler;
+    }
+
+    bool is_free_to_aquire() {
+        return (m_owner == "none");
+    }
+
+    /**
+    * take ownership of the server array 
+    * this is static 
+    * ownership is total 
+    * 
+    */
+    void set_owner(const std::string &owner) {
+        m_owner = owner;
+        m_owner_handler = generate_handler();
+    }
+
+    void clear_owner() {
+        m_owner = "none";
+        m_owner_handler = "";
+    }
+
+    bool verify_owner_handler(const std::string &handler) {
+
+        return ( (m_owner != "none") && (m_owner_handler == handler) );
+
+    }
+
 private:
+
+    std::string generate_handler();
+
     TrexStreamTable  m_stream_table;
     uint8_t          m_port_id;
     port_state_e     m_port_state;
+    std::string      m_owner;
+    std::string      m_owner_handler;
 };
 
 /**