def enter()

in VMEncryption/main/oscrypto/ubuntu_1404/encryptstates/UnmountOldrootState.py [0:0]

• 74 lines of code
• 23 McCabe index (conditional complexity)

    def enter(self):
        if not self.should_enter():
            return

        self.context.logger.log("Entering unmount_oldroot state")
        
        self.command_executor.Execute('service ssh restart', True)
        
        proc_comm = ProcessCommunicator()
        self.command_executor.Execute(command_to_execute="initctl list",
                                      raise_exception_on_failure=True,
                                      communicator=proc_comm)

        for line in proc_comm.stdout.split('\n'):
            if not "running" in line:
                continue

            if "walinuxagent" in line or "ssh" in line or "cryptdisks" in line:
                continue

            splitted = line.split()
            if len(splitted):
                service = splitted[0]
                self.command_executor.Execute('service {0} restart'.format(service))

        self.command_executor.Execute('swapoff -a', True)

        self.bek_util.umount_azure_passhprase(self.encryption_config, force=True)

        if os.path.exists("/oldroot/mnt"):
            self.command_executor.Execute('umount /oldroot/mnt')

        if os.path.exists("/oldroot/mnt/azure_bek_disk"):
            self.command_executor.Execute('umount /oldroot/mnt/azure_bek_disk')

        if os.path.exists("/mnt"):
            self.command_executor.Execute('umount /mnt')

        if os.path.exists("/mnt/azure_bek_disk"):
            self.command_executor.Execute('umount /mnt/azure_bek_disk')

        proc_comm = ProcessCommunicator()

        self.command_executor.Execute(command_to_execute="fuser -vm /oldroot",
                                      raise_exception_on_failure=True,
                                      communicator=proc_comm)

        self.context.logger.log("Processes using oldroot:\n{0}".format(proc_comm.stdout))

        procs_to_kill = filter(lambda p: p.isdigit(), proc_comm.stdout.split())
        procs_to_kill = reversed(sorted(procs_to_kill))

        for victim in procs_to_kill:
            proc_name = ""

            try:
                with open("/proc/{0}/cmdline".format(victim)) as f:
                    proc_name = f.read()
            except IOError as e:
                self.context.logger.log("Proc {0} is already dead".format(victim))

            self.context.logger.log("Killing process: {0} ({1})".format(proc_name, victim))

            if int(victim) == os.getpid():
                self.context.logger.log("Restarting WALA in before committing suicide")

                # Kill any other daemons that are blocked and would be executed after this process commits
                # suicide
                self.command_executor.Execute('at -f /restart-wala.sh now + 1 minutes', True)
                self.command_executor.ExecuteInBash('pkill -f .*ForLinux.*handle.py.*daemon.*', True)

            if int(victim) == 1:
                self.context.logger.log("Skipping init")
                continue

            self.command_executor.Execute('kill -9 {0}'.format(victim))

        self.command_executor.Execute('telinit u', True)

        sleep(3)

        self.command_executor.Execute('umount -a', False)

        sleep(3)

        for mount_item in self.disk_util.get_mount_items():
            if "/oldroot/" in mount_item["dest"]:
                self.command_executor.Execute('umount ' + mount_item["dest"], True)


        if self.command_executor.Execute('mountpoint /oldroot', False):
            self.should_exit()
            return
        
        self.command_executor.Execute('umount /oldroot', True)

        sleep(3)
        
        attempt = 1

        while True:
            if attempt > 10:
                raise Exception("Block device {0} did not appear in 10 restart attempts".format(self.rootfs_block_device))

            self.context.logger.log("Restarting udev")
            self.command_executor.Execute('service udev restart')

            sleep(10)

            if self.command_executor.ExecuteInBash('[ -b {0} ]'.format(self.rootfs_block_device), False) == 0:
                break

            attempt += 1

        self.command_executor.Execute('e2fsck -yf {0}'.format(self.rootfs_block_device), True)