From f74f52ae782c8a28eae286323efa4d11dc842682 Mon Sep 17 00:00:00 2001
From: Darshan Vandra <dvandra.ai@gmail.com>
Date: Thu, 31 Jan 2019 18:32:58 +0530
Subject: [PATCH] workflow: add ping-mesh workflow

---
 statics/js/components/workflow.js |   2 +-
 statics/workflows/ping-mesh.yaml  | 104 ++++++++++++++++++++++++++++++
 2 files changed, 105 insertions(+), 1 deletion(-)
 create mode 100644 statics/workflows/ping-mesh.yaml

diff --git a/statics/js/components/workflow.js b/statics/js/components/workflow.js
index 515cdcc2ab..5ed95599e9 100644
--- a/statics/js/components/workflow.js
+++ b/statics/js/components/workflow.js
@@ -17,7 +17,7 @@ Vue.component('item', {
   <div class="form-group">
     <div v-if="Type == 'string'" class="form-group">
       <label :for="Name">{{Description}}</label>
-      <textarea:id="Name" v-model="formData[Name]"></textarea>
+      <textarea :id="Name" v-model="formData[Name]" class="form-control input-sm"></textarea>
     </div>
     <div v-else-if="Type == 'date'" class="form-group">
     <label :for="Name">{{Description}}</label>
diff --git a/statics/workflows/ping-mesh.yaml b/statics/workflows/ping-mesh.yaml
new file mode 100644
index 0000000000..b7564ad363
--- /dev/null
+++ b/statics/workflows/ping-mesh.yaml
@@ -0,0 +1,104 @@
+---
+UUID: "784c3329-0f47-449b-5c58-2d207bcfb501"
+Name: "Ping Mesh (ICMP/TCP/UDP)"
+Description: "Check Connectivity from Multiple Source to Single Destination"
+Parameters:
+  - Name: protocol
+    Description: Protocol
+    Type:  choice
+    Default: icmp4
+    Values:
+      - Description: "Protocol : ICMPv4/Echo request"
+        Value: icmp4
+      - Description: "Protocol : TCP/IPv4"
+        Value: tcp4
+      - Description: "Protocol : UDP/IPv4"
+        Value: udp4
+  - Name: source
+    Description: Source Nodes (Enter a Gremlin Query to select Source Nodes)
+    Type: string
+  - Name: destination
+    Description: Destination Node (Select Destination Node)
+    Type: node
+Source: |
+  function PingMesh(protocol, src, dst) {
+    var sources = [];
+    var result = {};
+    var pktform = {};
+    client.gremlin.query(src).then(function(nodes) {
+      nodes.forEach(function(node) {
+        sources.push(node.Metadata.TID);
+      });
+    })
+    var capture = new Capture();
+    capture.GremlinQuery = "G.V().Has('TID', '" + dst + "')";
+    return client.captures.create(capture).then(function (c) {
+      capture = c
+    }).then(function () {
+      return sleep(1000)
+    }).then(function () {
+      sources.forEach(function(s) {
+        if (s == dst) { return; }
+        var packetInjection = new PacketInjection();
+        packetInjection.Src = "G.V().Has('TID', '" + s + "')"
+        packetInjection.Dst = "G.V().Has('TID', '" + dst + "')"
+        packetInjection.Count = 5
+        return client.G.V().Has("TID", dst).then(
+        function (nodes) {
+          if (nodes[0].Metadata.Neutron && nodes[0].Metadata.Neutron.IPV4) {
+            packetInjection.DstIP = nodes[0].Metadata.Neutron.IPV4[0]
+          }
+          if (nodes[0].Metadata.ExtID && nodes[0].Metadata.ExtID["attached-mac"]) {
+            packetInjection.DstMAC = nodes[0].Metadata.ExtID["attached-mac"]
+          }
+          if (protocol == "icmp4") {
+            packetInjection.Type = protocol;
+            packetInjection.ICMPID = Math.floor(Math.random() * 65535);
+          }
+          if (protocol == "tcp4" || protocol == "udp4") {
+            packetInjection.Type = protocol;
+            packetInjection.SrcPort = 1024 + Math.floor(Math.random() * (65535-1024));
+            packetInjection.DstPort = 1024 + Math.floor(Math.random() * (65535-1024));
+          }
+        }).then(function () {
+          return client.G.V().Has("TID", s)
+        }).then(function (nodes) {
+          if (nodes[0].Metadata.Neutron && nodes[0].Metadata.Neutron.IPV4) {
+            packetInjection.SrcIP = nodes[0].Metadata.Neutron.IPV4[0]
+          }
+          if (nodes[0].Metadata.ExtID && nodes[0].Metadata.ExtID["attached-mac"]) {
+            packetInjection.SrcMAC = nodes[0].Metadata.ExtID["attached-mac"]
+          } else {
+            packetInjection.SrcIP = nodes[0].Metadata.IPV4[0]
+          }
+          pktform[s] = packetInjection;
+          pktform[s].SrcIP = pktform[s].SrcIP.split("/")[0]
+          return client.packetInjections.create(packetInjection)
+        })
+      });
+    }).then(function () {
+      return sleep(1000)
+    }).then(function () {
+      sources.forEach(function(s) {
+        if (s == dst) { return; }
+        if (protocol == "icmp4") {
+          client.G.Flows().Has("ICMP.ID", pktform[s].ICMPID, "Network.A", pktform[s].SrcIP).then(function(flows) {
+            result[s] = {"Connected" : flows.length > 0 && flows[0].Metric.ABPackets > 0, "Replied" : flows.length > 0 && flows[0].Metric.BAPackets > 0};
+            return result
+          })
+        } else {
+          transport_protocol = protocol.toUpperCase().split(4)[0];
+          client.G.Flows().Has("Transport.A", pktform[s].SrcPort, "Transport.B", pktform[s].DstPort, "Transport.Protocol", transport_protocol, "Network.A", pktform[s].SrcIP).then(function(flows) {
+            result[s] = {"Connected" : flows.length > 0 && flows[0].Metric.ABPackets > 0, "Replied" : flows.length > 0 && flows[0].Metric.BAPackets > 0};
+            return result
+          })
+        }
+      });
+    }).then(function () {
+      return result
+    }).finally(function () {
+      return client.captures.delete(capture.UUID)
+    }).catch(function () {
+      return client.captures.delete(capture.UUID)
+    });
+  }