Add container upgrade script

1 files changed, 198 insertions, 0 deletions

diff --git a/scripts/upgrade.py b/scripts/upgrade.py
new file mode 100644
index 0000000..98efc70
--- /dev/null
+++ b/scripts/upgrade.py
@@ -0,0 +1,198 @@
+#!/usr/bin/python3
+
+import sys
+import time
+import pylxd
+from pylxd import Client
+
+class ExecuteError(RuntimeError):
+        def __init__(self, command, exit_code):
+                self.command = command
+                self.exit_code = exit_code
+
+        def __str__(self):
+                return 'Command "%s" exit code %d' % (' '.join(self.command), self.exit_code)
+
+def find_source_image(client, image):
+        try:
+                client.images.get_by_alias(image)
+                return {'type': 'image', 'alias': image}
+        except pylxd.exceptions.NotFound as e:
+                if len(image) >= 12:
+                        client.images.get(image)
+                        return {'type': 'image', 'fingerprint': image}
+                else:
+                        raise
+
+def copy_config(old, new):
+        new.devices = old.devices
+        new.description = old.description
+        new.profiles = old.profiles
+        new_config = new.config
+
+        for key, value in old.config.items():
+                if key.endswith("hwaddr"):
+                        new_config[key] = value
+
+#        for key, value in new_config.items():
+#                print("%s %s" % (key, value))
+
+        new.config = new_config
+
+def log_stdout(message):
+        print(message, end='', flush=True)
+
+def log_stderr(message):
+        print(message, end='', file=sys.stderr, flush=True)
+
+class Container:
+        def __init__(self, container):
+                self.container = container
+
+        def __getattr__(self, name):
+                return self.container.__getattribute__(name)
+
+        def __setattr__(self, name, value):
+                if name not in ('container'):
+                        self.container.__setattr__(name, value)
+                else:
+                        super(Container, self).__setattr__(name, value)
+
+        def execute_with_output(self, command, *args, **kwargs):
+                extra_args={}
+                if 'stderr_handler' not in kwargs:
+                        extra_args['stderr_handler'] = log_stderr
+                (exit_code, stdout, stderr) = self.container.execute(command, *args, **kwargs, **extra_args)
+                if exit_code != 0:
+                        raise ExecuteError(command, exit_code)
+                return stdout
+
+        def execute(self, command, *args, **kwargs):
+                extra_args={}
+                if 'stdout_handler' not in kwargs:
+                        extra_args['stdout_handler'] = log_stdout
+                self.execute_with_output(command, *args, **kwargs, **extra_args)
+
+        def execute_retry(self, command, retries, *args, **kwargs):
+                for i in range(retries + 1):
+                        try:
+                                self.execute(command, *args, **kwargs)
+                        except ExecuteError as e:
+                                if i == retries:
+                                        raise
+                                continue
+                        return
+                raise NotImplementedError()
+
+        def ping(self, dest):
+                self.execute_retry(['ping', '-c', '1', '-q', dest], 2,
+                                   stdout_handler=None)
+
+        def sysupgrade_backup(self, ):
+                return self.execute_with_output(['sysupgrade', '-b', '-'],
+                                                decode=False)
+
+        def sysupgrade_restore(self, data):
+                self.execute(['sysupgrade', '-r', '-'],
+                        stdin_payload=data, decode=False)
+
+        def opkg_list_installed(self, ):
+                return self.execute_with_output(['opkg', 'list-installed'])
+
+        def opkg_update(self):
+                print("Update")
+                self.execute(['opkg', 'update'])
+
+        def opkg_install(self, packages):
+                print("Installing %s" % packages)
+                self.execute(['opkg', 'install'] + packages)
+
+        def package_set(self):
+                old_list = self.opkg_list_installed().split('\n')
+                old_packages = []
+                i = 1
+                for l in old_list:
+                        i = i + 1
+                        res = l.split(' ')
+                        if len(res) == 3:
+                                (name, _, version) = res
+                                old_packages.append(name)
+                return frozenset(old_packages)
+
+def usage(argv):
+        print("Usage:", argv[0], "<old container> <new container> <image>")
+        exit(1)
+
+def main(argv):
+        is_allow_existing = False
+
+        if len(argv) == 4:
+                pos = 1
+        else:
+                usage(argv)
+
+        old_name = argv[pos]; pos=pos+1
+        new_name = argv[pos]; pos=pos+1
+        new_image = argv[pos]; pos=pos+1
+        client = Client()
+
+        old = Container(client.containers.get(old_name))
+
+        if old.status == 'Stopped':
+                print("Start", old_name)
+                old.start(wait=True)
+
+        new_source = find_source_image(client, new_image)
+        new_config = {'name': new_name, 'source': new_source}
+
+        if is_allow_existing and client.containers.exists(new_name):
+                new = Container(client.containers.get(new_name))
+        else:
+                print("Create", new_name, new_config)
+                new = Container(client.containers.create(new_config, wait=True))
+
+        if new.status == 'Stopped':
+                print("Start", new_name)
+                new.start(wait=True)
+
+                print("Ping downloads.openwrt.org")
+                new.ping('downloads.openwrt.org')
+
+                print("Update package list")
+                new.opkg_update()
+
+        old_set = old.package_set()
+        new_set = new.package_set()
+        add_packages = list(old_set.difference(new_set).difference(['iw']))
+
+        if len(add_packages) > 0:
+                print("Install", add_packages)
+                new.opkg_install(add_packages)
+        else:
+                print("No packages installed")
+
+        print("Backup", old_name)
+        backup_data = old.sysupgrade_backup()
+
+        print("Restore", new_name)
+        new.sysupgrade_restore(backup_data)
+
+        print("Stop", old_name)
+        old.stop(wait=True)
+
+        print("Stop", new_name)
+        new.stop(wait=True)
+
+        print("Copy config")
+        copy_config(old, new)
+        new.save()
+
+        print("Wait 2s")
+        time.sleep(2)
+
+        print("Start", new_name)
+        new.start(wait=True)
+        print("Finished")
+
+if __name__ == '__main__':
+        main(sys.argv)