From 6348a43612459694501db5a5ce043c8d1dcbec87 Mon Sep 17 00:00:00 2001
From: Richard Ipsum <richard.ipsum@codethink.co.uk>
Date: Wed, 28 Jan 2015 17:28:00 +0000
Subject: Add check for virtual networks

An exception will be raised if any needed networks are not started
---
 morphlib/exts/kvm.check | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/morphlib/exts/kvm.check b/morphlib/exts/kvm.check
index 3c6accbf..b8877a89 100755
--- a/morphlib/exts/kvm.check
+++ b/morphlib/exts/kvm.check
@@ -17,6 +17,7 @@
 '''Preparatory checks for Morph 'kvm' write extension'''
 
 import cliapp
+import os
 import re
 import urlparse
 
@@ -46,6 +47,7 @@ class KvmPlusSshCheckExtension(morphlib.writeexts.WriteExtension):
         self.check_can_create_file_at_given_path(ssh_host, vm_path)
         self.check_no_existing_libvirt_vm(ssh_host, vm_name)
         self.check_extra_disks_exist(ssh_host, self.parse_attach_disks())
+        self.check_virtual_networks_are_started(ssh_host)
 
     def check_and_parse_location(self, location):
         '''Check and parse the location argument to get relevant data.'''
@@ -102,4 +104,50 @@ class KvmPlusSshCheckExtension(morphlib.writeexts.WriteExtension):
                 raise cliapp.AppException('Did not find file %s on host %s' %
                                           (filename, ssh_host))
 
+    def check_virtual_networks_are_started(self, ssh_host):
+
+        def check_virtual_network_is_started(network_name):
+            cmd = ['virsh', '-c', 'qemu:///system', 'net-info', network_name]
+            net_info = cliapp.ssh_runcmd(ssh_host, cmd).split('\n')
+
+            def pretty_concat(lines):
+                return '\n'.join(['\t%s' % line for line in lines])
+
+            for line in net_info:
+                m = re.match('^Active:\W*(\w+)\W*', line)
+                if m:
+                    break
+            else:
+                raise cliapp.AppException(
+                        "Got unexpected output parsing output of `%s':\n%s"
+                        % (' '.join(cmd), pretty_concat(net_info)))
+
+            network_active = m.group(1) == 'yes'
+
+            if not network_active:
+                raise cliapp.AppException("Network '%s' is not started"
+                                          % network_name)
+
+        def name(nic_entry):
+            if ',' in nic_entry:
+                # NETWORK_NAME,mac=12:34,model=e1000...
+                return nic_entry[:nic_entry.find(',')]
+            else:
+                return nic_entry    # NETWORK_NAME
+
+        if 'NIC_CONFIG' in os.environ:
+            nics = os.environ['NIC_CONFIG'].split()
+
+            # --network bridge= is used to specify a bridge
+            # --network user is used to specify a form of NAT
+            # (see the virt-install(1) man page)
+            networks = [name(n) for n in nics if not n.startswith('bridge=')
+                                                and not n.startswith('user')]
+        else:
+            networks = ['default']
+
+        for network in networks:
+            check_virtual_network_is_started(network)
+
+
 KvmPlusSshCheckExtension().run()
-- 
cgit v1.2.1