From a5337989b2b4d06e6d55716fc78d480b61f71aa8 Mon Sep 17 00:00:00 2001
From: Eric Chou <echou@yahoo.com>
Date: Tue, 12 Jul 2016 13:41:31 -0700
Subject: [PATCH] Add json_config and modified fastnetmon_a10_v02.py file

---
 .gitignore                                 |   3 +++
 src/a10_plugin/a10.pyc                     | Bin 1561 -> 1561 bytes
 src/a10_plugin/fastnetmon_a10_v0.2.py      |  16 +++++++++-------
 src/a10_plugin/json_config/__init__.py     |   0
 src/a10_plugin/json_config/logoff.py       |   3 +++
 src/a10_plugin/json_config/write_memory.py |   1 +
 6 files changed, 16 insertions(+), 7 deletions(-)
 create mode 100644 .gitignore
 create mode 100644 src/a10_plugin/json_config/__init__.py
 create mode 100644 src/a10_plugin/json_config/logoff.py
 create mode 100644 src/a10_plugin/json_config/write_memory.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..f14285f
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+*.pyc
+__pycache__
+*.DS_Store
diff --git a/src/a10_plugin/a10.pyc b/src/a10_plugin/a10.pyc
index 116396cb82b079aba1fbcefcb96e9151ac785172..17a8f468dadc3ebfd28cd92a6e45be8434b02e65 100644
GIT binary patch
delta 15
WcmbQqGn0pn`7<w<0NX}3VKx9C90NoE

delta 15
WcmbQqGn0pn`7<xq71oVx!fXH|uLK7G

diff --git a/src/a10_plugin/fastnetmon_a10_v0.2.py b/src/a10_plugin/fastnetmon_a10_v0.2.py
index 60816a1..a2e9e37 100755
--- a/src/a10_plugin/fastnetmon_a10_v0.2.py
+++ b/src/a10_plugin/fastnetmon_a10_v0.2.py
@@ -1,7 +1,7 @@
 #!/usr/bin/python
 
 #
-# v0.2 created [ban |unban] [on ramp | off ramp action] for A10 TPS 
+# v0.2 created [ban | unban] [on ramp | off ramp action] for A10 TPS 
 # Eric Chou (ericc@a10networks.com)
 #
 
@@ -10,6 +10,8 @@ from sys import stdin
 import optparse
 import logging, json
 from a10 import axapi_auth, axapi_action
+from json_config.logoff import logoff_path
+from json_config.write_memory import write_mem_path
 
 LOG_FILE = "/var/log/fastnetmon-notify.log"
 
@@ -45,7 +47,10 @@ if action == "unban":
     except Exception as e: 
         logger.info("route not removed in unban, returned: " + str(e))
 
-    axapi_action(mitigator_base_url+'/axapi/v3/logoff', signature=signature)
+    # Commit config
+    axapi_action(mitigator_base_url+write_mem_path, signature=signature)
+    # Logoff
+    axapi_action(mitigator_base_url+logoff_path, signature=signature)
  
     sys.exit(0)
 
@@ -55,10 +60,8 @@ elif action == "ban":
     if zone_name in [i['zone-name'] for i in json.loads(r)['zone-list']]:
         r = axapi_action(mitigator_base_url+'/axapi/v3/ddos/dst/zone/'+zone_name, method="DELETE", signature=signature)
         logger.info(str(r))
-    axapi_action(mitigator_base_url+'/axapi/v3/logoff', signature=signature)
 
     # A10 Mitigation On Ramp 
-    mitigator_base_url, signature = axapi_auth(mitigator_ip, "admin", "a10")
     zone_name = client_ip_as_string + "_zone"
     ip_addr = client_ip_as_string
     port_num = 53
@@ -87,7 +90,6 @@ elif action == "ban":
                         "type":"pkt-rate",
                         "score":50,
                         "zone-threshold-num":1,
-                        "zone-violation-actions":"bmf_a10_script",
                       }
                     ],
                   },
@@ -124,9 +126,9 @@ elif action == "ban":
         logger("route not added: " + str(e))
 
     # Commit changes
-    axapi_action(mitigator_base_url+'/axapi/v3/write/memory', signature=signature)
+    axapi_action(mitigator_base_url+write_mem_path, signature=signature)
     # Log off
-    axapi_action(mitigator_base_url+'/axapi/v3/logoff', signature=signature)
+    axapi_action(mitigator_base_url+logoff_path, signature=signature)
     
     sys.exit(0)
 
diff --git a/src/a10_plugin/json_config/__init__.py b/src/a10_plugin/json_config/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/a10_plugin/json_config/logoff.py b/src/a10_plugin/json_config/logoff.py
new file mode 100644
index 0000000..35c7f27
--- /dev/null
+++ b/src/a10_plugin/json_config/logoff.py
@@ -0,0 +1,3 @@
+
+logoff_path = '/axapi/v3/logoff'
+
diff --git a/src/a10_plugin/json_config/write_memory.py b/src/a10_plugin/json_config/write_memory.py
new file mode 100644
index 0000000..4634b16
--- /dev/null
+++ b/src/a10_plugin/json_config/write_memory.py
@@ -0,0 +1 @@
+write_mem_path = '/axapi/v3/write/memory'