Synnefo » snf-image-creator

import string

BOOT_TIMEOUT = 300

        self._guest_exec('netsh interface ipv6 set global '

                         'randomizeidentifiers=disabled store=persistent')

        """Run the Microsoft System Preparation Tool on the Image. This will

        remove system-specific data and will make the image ready to be

        deployed. After this no other task may run.

        self._guest_exec(r'C:\Windows\system32\sysprep\sysprep '

                         r'/quiet /generalize /oobe /shutdown')

            token = self._enable_os_monitor()

        vm = None

        monitor = None

            monitorfd, monitor = tempfile.mkstemp()

            os.close(monitorfd)

            vm, display = self._create_vm(monitor)

            self.out.success("started (console on vnc display: %d)." % display)

            self.out.output("Waiting for OS to boot ...", False)

            if not self._wait_on_file(monitor, token):

                raise FatalError("Windows booting timed out.")

            else:

                self.out.success('done')

            self.out.output("Disabling automatic logon ...", False)

            self._disable_autologon()

            self.out.output('Preparing system from image creation:')

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

            self.out.success("done")

            if monitor is not None:

                os.unlink(monitor)

            if vm is not None:

                self._destroy_vm(vm)

    def _create_vm(self, monitor):

        """Create a VM with the image attached as the disk

            monitor: a file to be used to monitor when the OS is up

        """

        def random_mac():

            mac = [0x00, 0x16, 0x3e,

                   random.randint(0x00, 0x7f),

                   random.randint(0x00, 0xff),

                   random.randint(0x00, 0xff)]

            return ':'.join(map(lambda x: "%02x" % x, mac))

        # Use ganeti's VNC port range for a random vnc port

        vnc_port = random.randint(11000, 14999)

        display = vnc_port - 5900

        vm = kvm('-smp', '1', '-m', '1024', '-drive',

                 'file=%s,format=raw,cache=none,if=virtio' % self.image.device,

                 '-netdev', 'type=user,hostfwd=tcp::445-:445,id=netdev0',

                 '-device', 'virtio-net-pci,mac=%s,netdev=netdev0' %

                 random_mac(), '-vnc', ':%d' % display, '-serial',

                 'file:%s' % monitor, _bg=True)

        return vm, display

    def _destroy_vm(self, vm):

        """Destroy a VM previously created by _create_vm"""

        if vm.process.alive:

            vm.terminate()

        self._guest_exec(r'shutdown /s /t 5')

    def _wait_on_file(self, fname, msg):

        """Wait until a message appears on a file"""

        for i in range(BOOT_TIMEOUT):

            time.sleep(1)

            with open(fname) as f:

                for line in f:

                    if line.startswith(msg):

                        return True

        return False

    def _disable_autologon(self):

        """Disable automatic logon on the windows image"""

        winlogon = \

            r'"HKLM\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Winlogon"'

        self._guest_exec('REG DELETE %s /v DefaultUserName /f' % winlogon)

        self._guest_exec('REG DELETE %s /v DefaultPassword /f' % winlogon)

        self._guest_exec('REG DELETE %s /v AutoAdminLogon /f' % winlogon)

    def _enable_os_monitor(self):

        """Add a script in the registry that will send a random string to the

        first serial port when the windows image finishes booting.

        """

        token = "".join(random.choice(string.ascii_letters) for x in range(16))

        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)

            # Enable automatic logon.

            # This is needed because we need to execute a script that we add in

            # the RunOnce registry entry and those programs only get executed

            # when a user logs on. There is a RunServicesOnce registry entry

            # whose keys get executed in the background when the logon dialog

            # box first appears, but they seem to only work with services and

            # not arbitrary command line expressions :-(

            #

            # Instructions on how to turn on automatic logon in Windows can be

            # found here: http://support.microsoft.com/kb/324737

            #

            # Warning: Registry change will not work if the “Logon Banner” is

            # defined on the server either by a Group Policy object (GPO) or by

            # a local policy.

            winlogon = h.root()

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

                          'Winlogon'):

                winlogon = h.node_get_child(winlogon, child)

            h.node_set_value(

                winlogon,

                {'key': 'DefaultUserName', 't': 1,

                 'value': "Administrator".encode('utf-16le')})

            h.node_set_value(

                winlogon,

                {'key': 'DefaultPassword', 't': 1,

                 'value':  self.sysprep_params['password'].encode('utf-16le')})

            h.node_set_value(

                winlogon,

                {'key': 'AutoAdminLogon', 't': 1,

                 'value': "1".encode('utf-16le')})

            key = h.root()

            for child in ('Microsoft', 'Windows', 'CurrentVersion'):

                key = h.node_get_child(key, child)

            runonce = h.node_get_child(key, "RunOnce")

            if runonce is None:

                runonce = h.node_add_child(key, "RunOnce")

            value = (

                r'C:\Windows\System32\WindowsPowerShell\v1.0\powershell.exe '

                r'-ExecutionPolicy RemoteSigned '

                r'"&{$port=new-Object System.IO.Ports.SerialPort COM1,9600,'

                r'None,8,one;$port.open();$port.WriteLine(\"' + token + r'\");'

                r'$port.Close()}"').encode('utf-16le')

            h.node_set_value(runonce,

                             {'key': "BootMonitor", 't': 1, 'value': value})

            h.commit(None)

            self.g.upload(software, path)

        finally:

            os.unlink(software)

        return token

    def _guest_exec(self, command, fatal=True):

        """Execute a command on a windows VM"""

        stdout, stderr = winexe.communicate()

        if rc != 0 and fatal:

            reason = stderr if len(stderr) else stdout

            raise FatalError("Command: `%s' failed. Reason: %s" %

                             (command, reason))

        return (stdout, stderr, rc)