checkpoint before merge

2 files changed, 187 insertions, 0 deletions

diff --git a/src/publisher/trex_publisher.cpp b/src/publisher/trex_publisher.cpp
new file mode 100644
index 00000000..49602708
--- /dev/null
+++ b/src/publisher/trex_publisher.cpp
@@ -0,0 +1,107 @@
+/*
+ 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_publisher.h"
+#include <zmq.h>
+#include <assert.h>
+#include <sstream>
+#include <iostream>
+
+/**
+ * create the publisher
+ * 
+ */
+bool 
+TrexPublisher::Create(uint16_t port, bool disable){
+
+    if (disable) {
+        return (true);
+    }
+
+    m_context = zmq_ctx_new();
+    if ( m_context == 0 ) {
+        show_zmq_last_error("can't connect to ZMQ library");
+    }
+
+    m_publisher = zmq_socket (m_context, ZMQ_PUB);
+    if ( m_context == 0 ) {
+        show_zmq_last_error("can't create ZMQ socket");
+    }
+
+    std::stringstream ss;
+    ss << "tcp://*:" << port;
+
+    int rc = zmq_bind (m_publisher, ss.str().c_str());
+    if (rc != 0 ) {
+        show_zmq_last_error("can't bind to ZMQ socket at " + ss.str());
+    }
+
+    std::cout << "zmq publisher at: " << ss.str() << "\n";
+    return (true);
+}
+
+
+void 
+TrexPublisher::Delete(){
+    if (m_publisher) {
+        zmq_close (m_publisher);
+        m_publisher = NULL;
+    }
+    if (m_context) {
+        zmq_ctx_destroy (m_context);
+        m_context = NULL;
+    }
+}
+
+
+void 
+TrexPublisher::publish_json(const std::string &s){
+    if (m_publisher) {
+        int size = zmq_send (m_publisher, s.c_str(), s.length(), 0);
+        assert(size == s.length());
+    }
+}
+
+void
+TrexPublisher::publish_event(TrexPublisherEvent *ev) {
+    Json::FastWriter writer;
+    Json::Value value;
+    std::string s;
+    
+    value["name"] = "event";
+    value["type"] = ev->get_type();
+    ev->to_json(value["data"]);
+
+    s = writer.write(value);
+    publish_json(s);
+}
+
+/**
+ * error handling
+ * 
+ */
+void 
+TrexPublisher::show_zmq_last_error(const std::string &err){
+    std::cout << " ERROR " << err << "\n";
+    std::cout << " ZMQ: " << zmq_strerror (zmq_errno ());
+    exit(-1);
+}
+
diff --git a/src/publisher/trex_publisher.h b/src/publisher/trex_publisher.h
new file mode 100644
index 00000000..7c3fff92
--- /dev/null
+++ b/src/publisher/trex_publisher.h
@@ -0,0 +1,80 @@
+/*
+ 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.
+*/
+#ifndef __TREX_PUBLISHER_H__
+#define __TREX_PUBLISHER_H__
+
+#include <stdint.h>
+#include <string>
+#include <json/json.h>
+
+/**
+ * base event type
+ * 
+ */
+class TrexPublisherEvent {
+public:
+    virtual void to_json(Json::Value &json) = 0;
+    virtual uint8_t get_type() = 0;
+
+protected:
+    enum {
+        EVENT_PORT_STOPPED = 1
+    };
+
+};
+
+/**
+ * port stopped transmitting
+ * 
+ */
+class TrexEventPortStopped : public TrexPublisherEvent {
+public:
+    TrexEventPortStopped(uint8_t port_id);
+    virtual void to_json(Json::Value &json);
+    virtual uint8_t get_type() {
+        return (EVENT_PORT_STOPPED);
+    }
+};
+
+
+class TrexPublisher {
+
+public:
+
+    TrexPublisher() {
+        m_context = NULL;
+        m_publisher = NULL;
+    }
+
+    bool Create(uint16_t port, bool disable);
+    void Delete();
+    void publish_json(const std::string &s);
+
+    void publish_event(TrexPublisherEvent *ev);
+
+private:
+    void show_zmq_last_error(const std::string &err);
+private:
+    void * m_context;
+    void * m_publisher;
+};
+
+#endif /* __TREX_PUBLISHER_H__ */