Synnefo » snf-image-creator

from image_creator.os_type import OSBase, sysprep

from image_creator.util import FatalError, check_guestfs_version

    @sysprep(enabled=False)

    def remove_user_accounts(self, print_header=True):

        """Remove all user accounts with id greater than 1000"""

        pass

    def do_sysprep(self):

        """Prepare system for image creation."""

        if getattr(self, 'syspreped', False):

            raise FatalError("Image is already syspreped!")

        self.mount(readonly=False)

        try:

            disabled_uac = self._update_uac_remote_setting(1)

        finally:

            self.umount()

        self.out.output("Shutting down helper VM ...", False)

        self.g.sync()

        # guestfs_shutdown which is the prefered way to shutdown the backend

        # process was introduced in version 1.19.16

        if check_guestfs_version(self.g, 1, 19, 16) >= 0:

            ret = self.g.shutdown()

        else:

            ret = self.g.kill_subprocess()

        self.out.success('done')

        self.out.output("Starting windows VM ...", False)

        try:

            pass

        finally:

            self.out.output("Relaunching helper VM (may take a while) ...",

                            False)

            self.g.launch()

            self.out.success('done')

        if disabled_uac:

            self._update_uac_remote_setting(0)

        self.syspreped = True

    def _registry_file_path(self, regfile):

        """Retrieves the case sensitive path to a registry file"""

        systemroot = self.g.inspect_get_windows_systemroot(self.root)

        path = "%s/system32/config/%s" % (systemroot, regfile)

        try:

            path = self.g.case_sensitive_path(path)

        except RuntimeError as e:

            raise FatalError("Unable to retrieve registry file: %s. Reason: %s"

                             % (regfile, str(e)))

        return path

    def _update_uac_remote_setting(self, value):

        """Updates the registry key value:

        [HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Policies

        \System]"LocalAccountTokenFilterPolicy"

        value = 1 will disable the UAC remote restrictions

        value = 0 will enable the UAC remote restrictions

        For more info see here: http://support.microsoft.com/kb/951016

        Returns:

            True if the key is changed

            False if the key is unchanged

        """

        if value not in (0, 1):

            raise ValueError("Valid values for value parameter are 0 and 1")

        path = self._registry_file_path('SOFTWARE')

        softwarefd, software = tempfile.mkstemp()

        try:

            os.close(softwarefd)

            self.g.download(path, software)

            h = hivex.Hivex(software, write=True)

            key = h.root()

            for child in ('Microsoft', 'Windows', 'CurrentVersion', 'Policies',

                          'System'):

                key = h.node_get_child(key, child)

            policy = None

            for val in h.node_values(key):

                if h.value_key(val) == "LocalAccountTokenFilterPolicy":

                    policy = val

            if policy is not None:

                dword = h.value_dword(policy)

                if dword == value:

                    return False

            elif value == 0:

                return False

            new_value = {

                'key': "LocalAccountTokenFilterPolicy", 't': 4L,

                'value': '%s\x00\x00\x00' % '\x00' if value == 0 else '\x01'}

            h.node_set_value(key, new_value)

            h.commit(None)

            self.g.upload(software, path)

        finally:

            os.unlink(software)

        return True

        path = self._registry_file_path('SAM')

            os.close(samfd)