1 files changed, 318 insertions, 0 deletions
diff --git a/src/sim/trex_sim_stateless.cpp b/src/sim/trex_sim_stateless.cpp
new file mode 100644
index 00000000..2821644f
--- /dev/null
+++ b/src/sim/trex_sim_stateless.cpp
@@ -0,0 +1,318 @@
+/*
+ Itay Marom
+ Cisco Systems, Inc.
+*/
+
+/*
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+
+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.
+*/
+
+#include "trex_sim.h"
+#include <trex_stateless.h>
+#include <trex_stateless_messaging.h>
+#include <trex_rpc_cmd_api.h>
+#include <json/json.h>
+#include <stdexcept>
+#include <sstream>
+
+using namespace std;
+
+TrexStateless * get_stateless_obj() {
+    return SimStateless::get_instance().get_stateless_obj();
+}
+
+
+class SimRunException : public std::runtime_error 
+{
+public:
+    SimRunException() : std::runtime_error("") {
+
+    }
+    SimRunException(const std::string &what) : std::runtime_error(what) {
+    }
+};
+
+/*************** hook for platform API **************/
+class SimPlatformApi : public TrexPlatformApi {
+public:
+    SimPlatformApi(int dp_core_count) {
+        m_dp_core_count = dp_core_count;
+    }
+
+    virtual uint8_t get_dp_core_count() const {
+        return m_dp_core_count;
+    }
+
+    virtual void get_global_stats(TrexPlatformGlobalStats &stats) const {
+    }
+    virtual void get_interface_info(uint8_t interface_id, std::string &driver_name, driver_speed_e &speed) const {
+    }
+    virtual void get_interface_stats(uint8_t interface_id, TrexPlatformInterfaceStats &stats) const {
+    }
+
+    virtual void port_id_to_cores(uint8_t port_id, std::vector<std::pair<uint8_t, uint8_t>> &cores_id_list) const {
+        for (int i = 0; i < m_dp_core_count; i++) {
+             cores_id_list.push_back(std::make_pair(i, 0));
+        }
+    }
+
+private:
+    int m_dp_core_count;
+};
+
+/**
+ * handler for DP to CP messages
+ * 
+ * @author imarom (19-Nov-15)
+ */
+class DpToCpHandler {
+public:
+    virtual void handle(TrexStatelessDpToCpMsgBase *msg) = 0;
+};
+
+/************************ 
+ * 
+ *  stateless sim object
+ *  
+************************/
+class SimPublisher : public TrexPublisher {
+public:
+
+    /* override create */
+    bool Create(uint16_t port, bool disable) {
+        return true;
+    }
+
+    void Delete() {
+
+    }
+
+    void publish_json(const std::string &s) {
+    }
+
+    virtual ~SimPublisher() {
+    }
+};
+
+/************************ 
+ * 
+ *  stateless sim object
+ *  
+************************/
+
+SimStateless::SimStateless() {
+    m_trex_stateless    = NULL;
+    m_publisher         = NULL;
+    m_dp_to_cp_handler  = NULL;
+    m_verbose           = false;
+    m_dp_core_count     = -1;
+    m_dp_core_index     = -1;
+    m_port_count        = -1;
+
+    /* override ownership checks */
+    TrexRpcCommand::test_set_override_ownership(true);
+}
+
+
+int
+SimStateless::run(const string &json_filename,
+                  const string &out_filename,
+                  int port_count,
+                  int dp_core_count,
+                  int dp_core_index) {
+
+    assert(dp_core_count > 0);
+    assert(dp_core_index >= 0);
+    assert(dp_core_index < dp_core_count);
+
+    m_dp_core_count = dp_core_count;
+    m_dp_core_index = dp_core_index;
+    m_port_count    = port_count;
+
+    prepare_dataplane();
+    prepare_control_plane();
+
+    try {
+        execute_json(json_filename);
+    } catch (const SimRunException &e) {
+        std::cout << "*** test failed ***\n\n" << e.what() << "\n";
+        return (-1);
+    }
+
+    run_dp(out_filename);
+
+    return 0;
+}
+
+
+SimStateless::~SimStateless() {
+    if (m_trex_stateless) {
+        delete m_trex_stateless;
+        m_trex_stateless = NULL;
+    }
+
+    if (m_publisher) {
+        delete m_publisher;
+        m_publisher = NULL;
+    }
+
+    m_fl.Delete();
+}
+
+/**
+ * prepare control plane for test
+ * 
+ * @author imarom (28-Dec-15)
+ */
+void
+SimStateless::prepare_control_plane() {
+    TrexStatelessCfg cfg;
+    
+    m_publisher = new SimPublisher();
+
+    TrexRpcServerConfig rpc_req_resp_cfg(TrexRpcServerConfig::RPC_PROT_MOCK, 0);
+
+    cfg.m_port_count         = m_port_count;
+    cfg.m_rpc_req_resp_cfg   = &rpc_req_resp_cfg;
+    cfg.m_rpc_async_cfg      = NULL;
+    cfg.m_rpc_server_verbose = false;
+    cfg.m_platform_api       = new SimPlatformApi(m_dp_core_count);
+    cfg.m_publisher          = m_publisher;
+
+    m_trex_stateless = new TrexStateless(cfg);
+
+    m_trex_stateless->launch_control_plane();
+
+    for (auto &port : m_trex_stateless->get_port_list()) {
+        port->acquire("test", 0, true);
+    }
+
+}
+
+
+/**
+ * prepare the data plane for test
+ * 
+ */
+void
+SimStateless::prepare_dataplane() {
+    
+    CGlobalInfo::m_options.m_expected_portd = m_port_count;
+
+    assert(CMsgIns::Ins()->Create(m_dp_core_count));
+    m_fl.Create();
+    m_fl.generate_p_thread_info(m_dp_core_count);
+
+    for (int i = 0; i < m_dp_core_count; i++) {
+        m_fl.m_threads_info[i]->set_vif(&m_erf_vif);
+    }
+}
+
+
+
+void
+SimStateless::execute_json(const std::string &json_filename) {
+
+    std::ifstream test(json_filename);
+    std::stringstream buffer;
+    buffer << test.rdbuf();
+
+    std::string rep = m_trex_stateless->get_rpc_server()->test_inject_request(buffer.str());
+
+    Json::Value root;
+    Json::Reader reader;
+    reader.parse(rep, root, false);
+
+    if (is_verbose()) {
+        std::cout << "server response: \n\n" << root << "\n\n";
+    }
+
+    validate_response(root);
+
+}
+
+void
+SimStateless::validate_response(const Json::Value &resp) {
+    std::stringstream ss;
+
+    if (resp.isArray()) {
+        for (const auto &single : resp) {
+            if (single["error"] != Json::nullValue) {
+                ss << "failed with:\n\n" << single["error"] << "\n\n";
+                throw SimRunException(ss.str());
+            }
+        }
+    } else {
+        if (resp["error"] != Json::nullValue) {
+            ss << "failed with:\n\n" << resp["error"] << "\n\n";
+            throw SimRunException(ss.str());
+        }
+    }
+  
+}
+
+
+void
+SimStateless::run_dp(const std::string &out_filename) {
+
+    for (int i = 0; i < m_dp_core_count; i++) {
+        if (i == m_dp_core_index) {
+            run_dp_core(i, out_filename);
+        } else {
+            run_dp_core(i, "/dev/null");
+        }
+    }
+
+    CFlowGenListPerThread *lpt = m_fl.m_threads_info[m_dp_core_index];
+  
+    std::cout << "\n";
+    std::cout << "ports:        " << m_port_count << "\n";
+    std::cout << "cores:        " << m_dp_core_count << "\n";
+    std::cout << "core index:   " << m_dp_core_index << "\n";
+    std::cout << "\nwritten " << lpt->m_node_gen.m_cnt << " packets " << "to '" << out_filename << "'\n\n";
+}
+
+void
+SimStateless::run_dp_core(int core_index, const std::string &out_filename) {
+
+    CFlowGenListPerThread *lpt = m_fl.m_threads_info[core_index];
+
+    lpt->start_stateless_simulation_file((std::string)out_filename, CGlobalInfo::m_options.preview);
+    lpt->start_stateless_daemon_simulation();
+
+    flush_dp_to_cp_messages_core(core_index);
+}
+
+
+void
+SimStateless::flush_dp_to_cp_messages_core(int core_index) {
+
+    CNodeRing *ring = CMsgIns::Ins()->getCpDp()->getRingDpToCp(core_index);
+
+    while ( true ) {
+        CGenNode * node = NULL;
+        if (ring->Dequeue(node) != 0) {
+            break;
+        }
+        assert(node);
+
+        TrexStatelessDpToCpMsgBase * msg = (TrexStatelessDpToCpMsgBase *)node;
+        if (m_dp_to_cp_handler) {
+            m_dp_to_cp_handler->handle(msg);
+        }
+
+        delete msg;
+    }
+}