1 files changed, 85 insertions, 0 deletions
diff --git a/Websocket/communication-protocol.h b/Websocket/communication-protocol.h
new file mode 100644
index 00000000..f3c488ef
--- /dev/null
+++ b/Websocket/communication-protocol.h
@@ -0,0 +1,85 @@
+/*
+ * Copyright (c) 2017 Cisco and/or its affiliates.
+ * 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 COMMUNICATION_PROTOCOL_HPP
+#define COMMUNICATION_PROTOCOL_HPP
+
+#include <string>
+#include <map>
+#include <set>
+#include "query.h"
+#include "../UI/GraphDataSource.h"
+#include "websocket-server.h"
+
+enum ProtocolVersion
+{
+  CONTROL_PROTOCOL_V1 = 1,
+  CONTROL_PROTOCOL_V2 = 2
+};
+
+typedef struct ProtocolDetails {
+  static std::set<std::string> AllowedObjectName;
+  static std::set<std::string> AllowedFields;
+  static std::set<std::string> AllowedActions;
+//  static std::list<std::string> AllowedParameters; // Per action
+  static std::set<std::string> AllowedFilters;
+  static std::set<std::string> AllowedOperands; // Per filter
+} ProtocolDetails;
+
+class CommunicationProtocol
+{
+public:
+  CommunicationProtocol(ProtocolVersion version = ProtocolVersion::CONTROL_PROTOCOL_V1);
+
+  bool
+  checkFilter(const std::vector<std::string> &filter);
+
+  bool
+  checkFields(const std::string &field);
+
+  bool
+  checkObjectName(const std::string &objectName);
+
+  bool
+  checkParameters(const std::string &parameter);
+
+  bool
+  checkAction(const std::string &action);
+
+  std::string
+  evaluateFilters(const std::list<std::vector<std::string>> &filters);
+
+  void
+  processQuery(Server *s, websocketpp::connection_hdl hdl, message_ptr msg,
+               Query query);
+
+  Query
+  makeReplyQuery(const Query &request);
+
+  void
+  setGraphDataSource(GraphDataSource *graphDataSource);
+
+  static std::function<void(const boost::system::error_code&)> timerCallback;
+
+
+private:
+  GraphDataSource *graphDataSource;
+  ProtocolVersion  version;
+
+  std::shared_ptr<boost::asio::deadline_timer> subscribeTimer;
+};
+
+
+#endif //COMMUNICATION_PROTOCOL_HPP