From 3dba2d13e6ba7ced2a032b9fbb98984a44370f5f Mon Sep 17 00:00:00 2001
From: Daniil Baturin <daniil@baturin.org>
Date: Fri, 27 Jul 2018 01:17:25 +0200
Subject: T666, T616: new implementation of the VRRP CLI.
---
python/vyos/keepalived.py | 153 ++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 153 insertions(+)
create mode 100644 python/vyos/keepalived.py
(limited to 'python')
diff --git a/python/vyos/keepalived.py b/python/vyos/keepalived.py
new file mode 100644
index 000000000..4114aa736
--- /dev/null
+++ b/python/vyos/keepalived.py
@@ -0,0 +1,153 @@
+# Copyright 2018 VyOS maintainers and contributors <maintainers@vyos.io>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library. If not, see <http://www.gnu.org/licenses/>.
+
+import re
+import os
+import signal
+import json
+
+import vyos.util
+
+
+pid_file = '/var/run/keepalived.pid'
+state_file = '/tmp/keepalived.data'
+stats_file = '/tmp/keepalived.stats'
+json_file = '/tmp/keepalived.json'
+
+state_dir = '/var/run/vyos/vrrp/'
+
+def vrrp_running():
+ if not os.path.exists(vyos.keepalived.pid_file) \
+ or not vyos.util.process_running(vyos.keepalived.pid_file):
+ return False
+ else:
+ return True
+
+def keepalived_running():
+ return vyos.util.process_running(pid_file)
+
+def force_state_data_dump():
+ pid = vyos.util.read_file(pid_file)
+ os.kill(int(pid), signal.SIGUSR1)
+
+def force_stats_dump():
+ pid = vyos.util.read_file(pid_file)
+ os.kill(int(pid), signal.SIGUSR2)
+
+def force_json_dump():
+ pid = vyos.util.read_file(pid_file)
+ os.kill(int(pid), signal.SIGRTMIN+2)
+
+def get_json_data():
+ with open(json_file, 'r') as f:
+ j = json.load(f)
+ return j
+
+def get_statistics():
+ return vyos.util.read_file(stats_file)
+
+def get_state_data():
+ return vyos.util.read_file(state_file)
+
+def decode_state(code):
+ state = None
+ if code == 0:
+ state = "INIT"
+ elif code == 1:
+ state = "BACKUP"
+ elif code == 2:
+ state = "MASTER"
+ elif code == 3:
+ state = "FAULT"
+ else:
+ state = "UNKNOWN"
+
+ return state
+
+## The functions are mainly for transition script wrappers
+## to compensate for the fact that keepalived doesn't keep persistent
+## state between reloads.
+def get_old_state(group):
+ file = os.path.join(state_dir, "{0}.state".format(group))
+ if os.path.exists(file):
+ with open(file, 'r') as f:
+ data = f.read().strip()
+ return data
+ else:
+ return None
+
+def save_state(group, state):
+ if not os.path.exists(state_dir):
+ os.makedirs(state_dir)
+
+ file = os.path.join(state_dir, "{0}.state".format(group))
+ with open(file, 'w') as f:
+ f.write(state)
+
+## These functions are for the old, and hopefully obsolete plaintext
+## (non machine-readable) data format introduced by Vyatta back in the days
+## They are kept here just in case, if JSON output option turns out or becomes
+## insufficient.
+
+def read_state_data():
+ with open(state_file, 'r') as f:
+ lines = f.readlines()
+ return lines
+
+def parse_keepalived_data(data_lines):
+ vrrp_groups = {}
+
+ # Scratch variable
+ group_name = None
+
+ # Sadly there is no explicit end marker in that format, so we have
+ # only two states, one before the first VRRP instance is encountered
+ # and one after an instance/"group" was encountered
+ # We'll set group_name once the first group is encountered,
+ # and assume we are inside a group if it's set afterwards
+ #
+ # It may not be necessary since the keywords found inside groups and before
+ # the VRRP Topology section seem to have no intersection,
+ # but better safe than sorry.
+
+ for line in data_lines:
+ if re.match(r'^\s*VRRP Instance', line, re.IGNORECASE):
+ # Example: "VRRP Instance = Foo"
+ name = re.match(r'^\s*VRRP Instance\s+=\s+(.*)$', line, re.IGNORECASE).groups()[0].strip()
+ group_name = name
+ vrrp_groups[name] = {}
+ elif re.match(r'^\s*State', line, re.IGNORECASE) and group_name:
+ # Example: " State = MASTER"
+ group_state = re.match(r'^\s*State\s+=\s+(.*)$', line, re.IGNORECASE).groups()[0].strip()
+ vrrp_groups[group_name]["state"] = group_state
+ elif re.match(r'^\s*Last transition', line, re.IGNORECASE) and group_name:
+ # Example: " Last transition = 1532043820 (Thu Jul 19 23:43:40 2018)"
+ trans_time = re.match(r'^\s*Last transition\s+=\s+(\d+)\s', line, re.IGNORECASE).groups()[0]
+ vrrp_groups[group_name]["last_transition"] = trans_time
+ elif re.match(r'^\s*Interface', line, re.IGNORECASE) and group_name:
+ # Example: " Interface = eth0.30"
+ interface = re.match(r'\s*Interface\s+=\s+(.*)$', line, re.IGNORECASE).groups()[0].strip()
+ vrrp_groups[group_name]["interface"] = interface
+ elif re.match(r'^\s*Virtual Router ID', line, re.IGNORECASE) and group_name:
+ # Example: " Virtual Router ID = 14"
+ vrid = re.match(r'^\s*Virtual Router ID\s+=\s+(.*)$', line, re.IGNORECASE).groups()[0].strip()
+ vrrp_groups[group_name]["vrid"] = vrid
+ elif re.match(r'^\s*------< Interfaces', line, re.IGNORECASE):
+ # Interfaces section appears to always be present,
+ # and there's nothing of interest for us below that section,
+ # so we use it as an end of input marker
+ break
+
+ return vrrp_groups
--
cgit v1.2.3