Subversion Repositories configs

"""
Copyright 2008-2020 VMware, Inc.  All rights reserved. -- VMware Confidential

VMware VMX component installer.
"""

GCONF_DEFAULTS = 'xml:readwrite:/etc/gconf/gconf.xml.defaults'
DEST = LIBDIR/'vmware'
CONFIG = DEST/'setup/vmware-config'
CUPSLIBDIR = LIBDIR/'cups'
SETTINGS = \
    {'libdir': DEST,
     'bindir': BINDIR,
     'initscriptdir': INITSCRIPTDIR,
     'gksu.rootMethod': 'sudo' if 'SUDO_USER' in ENV else 'su',
     'NETWORKING': 'yes',
     'authd.fullpath': SBINDIR/'vmware-authd',
    }

vmwareSentinel = '# Automatically generated by the VMware Installer - DO NOT REMOVE\n'

# Present requirement is a RHEL7-like environment.
MIN_GLIBC_VERSION = (2, 17)
MIN_KERNEL_VERSION = (3, 10)

# Player and Workstation both depend on some configuration living
# in /etc/vmware
ETCDIR = Destination('/etc/vmware')

class VMX(Installer):
   def InitializeQuestions(self, old, new, upgrade):
      self.AddQuestion('ClosePrograms', key='ClosePrograms', text='',
                       required=True, default='Yes', level='REQUIRED')

      # Check whether we're in a VM or on real hardware.
      # Nothing currently depends on this, but it might be useful to log.
      self.checkVMTempFile = self.GetFilePath('extra/checkvm')
      # Get the directory above our installer and write checkvm to it so we can run it
      tmpdir = path(ENV['VMWARE_INSTALLER']).dirname().dirname()
      self.checkvmBin = tmpdir/'checkvm'
      path(self.checkVMTempFile).copy(self.checkvmBin)
      self.checkvmBin.chmod(0o755)
      path(self.checkVMTempFile).remove()

      ret = self.RunCommand(self.checkvmBin, ignoreErrors=True)
      if ret.retCode == 0:
         # We are running in a virtual machine.
         log.Info('Running inside a virtual machine!')
      elif ret.retCode == 1:
         # We are on normal hardware.
         log.Info('Running on a real machine!')
      else:
         log.Warn('Something went wrong detecting whether we are in a VM or not.  Assuming we are not.')


   def InitializeInstall(self, old, new, upgrade):
      if self._checkXenPresence():
         log.Warn('This system is running a Xen kernel. You cannot run VMs under the Xen kernel.')

      if path('/dev/kvm').exists():
         log.Warn('This system has KVM enabled. You cannot run VMs with KVM enabled.')

      self.AddTarget('File', 'bin/*', BINDIR)
      self.AddTarget('File', 'sbin/*', SBINDIR)
      self.AddTarget('File', 'lib/*', DEST)
      self.AddTarget('File', 'roms/*', DEST/'roms')
      self.AddTarget('File', 'etc/init.d/vmware', SYSCONFDIR/'init.d/vmware')

      # Install the proper modules.xml file
      self.AddTarget('File', 'extra/modules.xml', DEST/'modules/modules.xml')

      # Symlink all binaries to appLoader.
      for i in ('vmware-modconfig', 'vmware-modconfig-console',
                'vmware-gksu', 'vmware-vmblock-fuse'):
        self.AddTarget('Link', DEST/'bin/appLoader', DEST/'bin'/i)

      self.SetPermission(DEST/'bin/*', BINARY)
      self.SetPermission(SYSCONFDIR/'init.d/vmware', BINARY)
      self.SetPermission(DEST/'bin/vmware-vmx*', SETUID)
      self.SetPermission(DEST/'lib/libvmware-gksu.so/gksu-run-helper', BINARY)
      self.SetPermission(SBINDIR/'vmware-authd', SETUID)

      # modprobe.d script.  Only install this on systems where modprobe.d
      # exists.
      if path('/etc/modprobe.d').exists():
         self.AddTarget('File', 'etc/modprobe.d/modprobe-vmware-fuse.conf',
                        '/etc/modprobe.d/vmware-fuse.conf')

      # Add a link for internationalization directory
      self.AddTarget('Link', DEST/'icu', SYSCONFDIR/'vmware/icu')

   def PreTransactionUninstall(self, old, new, upgrade):
      self.AddQuestion('ClosePrograms', key='ClosePrograms', text='',
                       required=True, default='Yes', level='REQUIRED')

   def PreInstall(self, old, new, upgrade):
      # Remove all modules in case some were left behind somehow.  For
      # example, the installation database could have been blown away.
      ret, kvers, _ = self.RunCommand('uname', '-r')
      kvers = kvers.strip()
      modules = ('vmmon', 'vmnet')
      base = path('/lib/modules/%s/misc' % kvers)

      for module in modules:
         for ext in ('o', 'ko'):
            mod = '%s.%s' % (module, ext)
            (base/mod).remove(ignore_errors=True)

   def PreUninstall(self, old, new, upgrade):
      script = INITSCRIPTDIR/'vmware'

      # Stop and deconfigure services
      if ENV.get('VMWARE_SKIP_SERVICES'):
         log.Info('Skipping stopping services')
      elif INITSCRIPTDIR and self._scriptRunnable(script) and self.RunCommand(script, 'stop', ignoreErrors=True).retCode != 0:
         log.Error('Unable to stop VMware services')

      # Deconfigure services
      inits = self.LoadInclude('initscript')
      inits.DeconfigureService('vmware')

      # This must be the last thing done since the actions above may
      # depend on it.
      for key in list(SETTINGS.keys()):
         self.RunCommand(CONFIG, '-d', key)

      # Remove prelink appLoader exclusion.
      self._configurePrelink(False)

   def PostUninstall(self, old, new, upgrade):
      # Modules have been removed during uninstall.  We need to be sure to
      # run depmod to pick up that change.  Don't fail if for some reason
      # depmod can't be executed though.
      self.RunCommand('depmod', '-a', ignoreErrors=True)

   def PostInstall(self, old, new, upgrade):
      for key, val in list(SETTINGS.items()):
         self.RunCommand(CONFIG, '-s', key, val)

      bootstrap = ETCDIR/'bootstrap'
      # Create the bootstrap file.
      bootstrap.unlink(ignore_errors=True)
      # Fill it.
      for i in ('PREFIX', 'BINDIR', 'SBINDIR', 'LIBDIR', 'DATADIR',
                'SYSCONFDIR', 'DOCDIR', 'MANDIR', 'INCLUDEDIR', 'INITDIR',
                'INITSCRIPTDIR'):
         bootstrap.write_text('%s="%s"\n' % (i, globals()[i]), append=True)
      # Register it.
      # XXX: Skip registration for this change.  Allow the installer to handle
      # deletion.  A later change will correct this.
      # self.RegisterFile(bootstrap, fileType='ConfigFile')

      # Configure Gtk+.
      # @todo: make it its own component
      libconf = DEST/'libconf'
      replace = libconf/'etc/gtk-3.0/gdk-pixbuf.loaders'
      template = '@@LIBCONF_DIR@@'

      self.RunCommand('sed', '-e', 's,%s,%s,g' % (template, libconf),
                      '-i', replace)

      # Add prelink appLoader exclusion.
      self._configurePrelink(True)

      # Set up virtual networking
      self._configureNetworking(old, new)

      # Set up the vmware service script
      inits = self.LoadInclude('initscript')
      inits.ConfigureService('vmware',
                             'This service starts and stops VMware services',
                             '$network $syslog', # Start
                             '$network $syslog', # Stop
                             '',
                             '',
                             19,
                             8)

      # Make sure to start services
      script = INITSCRIPTDIR/'vmware'
      if INITSCRIPTDIR and script.exists():
         self.RunCommand(script, 'stop', ignoreErrors=True)
         self.RunCommand(script, 'start')

      # If no INITDIR was given, notify the user that the vmware service must
      # be manually set up
      if not INITDIR:
         self.UserMessage('No rc*.d style init script directories were given to the installer. '
                          'You must manually add the necessary links to ensure that the vmware '
                          'service at %s is automatically started and stopped on startup and shutdown.' % str(INITSCRIPTDIR/'vmware'))

   def _scriptRunnable(self, script):
      """ Returns True if the script exists and is in a runnable state """
      return script.isexe() and script.isfile() and self.RunCommand(script, 'validate').retCode == 100

   def _AddLineToFile(self, fil, text, addToEnd=True):
      """
      Add a line/consecutive lines to a file, surrounded by the VMware Sentinel.
      ### This method only adds a single line to a file ###
      ### You cannot make more than one modification per file ###

      @param fil: The file name.  Either a string or path object
      @param text: The text to add.  This function appends a \n
      @param addToEnd: Add to the end of the file?  If false, to the beginning.

      @return: True on successful modification, False if the file did not exist.
      """
      fil = path(fil) # Make sure it's a path object
      if fil.exists():
         txt = fil.text()
         # Modify the text
         if addToEnd:
            txt = ''.join([txt, vmwareSentinel, text, '\n', vmwareSentinel])
         else:
            txt = ''.join([vmwareSentinel, text, '\n', vmwareSentinel, txt])
         # Write it back to the file
         fil.write_text(txt)
         return True
      else:
         log.Info('Attempted to modify file %s, does not exist.' % fil)
         return False

   def _RemoveLineFromFile(self, fil):
      """
      Remove a line bracketed by the VMware Sentinel

      @param fil: The file name.  Either a string or path object

      @return: True on successful modification, False if the file did not exist.
      """
      fil = path(fil) # Make sure it's a path object
      if fil.exists():
         txt = fil.text()
         m = re.sub(vmwareSentinel + '.*\n' + vmwareSentinel,
                    '', txt, re.DOTALL)
         fil.write_text(m)
         return True
      else:
         log.Info('Attempted to modify file %s, does not exist.' % fil)
         return False

   def _configurePrelink(self, enable):
      """
      Configures prelinking by adding appLoader exclusion.

      @param enable: True if to add exclusion, False if to remove it.
      """
      prelink = path('/etc/prelink.conf')

      if prelink.exists():
         # XXX: It would probably be good to refactor this into some
         # sort of helper function that can add and remove lines from
         # files.
         skipPrelink = [ '# appLoader will segfault if prelinked.',
                         '-b %s' % (DEST/'bin/appLoader') ]

         lines = prelink.lines(encoding='utf-8', retain=False)

         # Strip whitespace from lines so that trailing whitespace
         # won't impact matching lines.
         lines = [line.strip() for line in lines]

         if enable:
            if skipPrelink[-1] not in lines:
               lines += skipPrelink
               log.Info('Added appLoader prelink exclusion to %s.', prelink)
            else:
               log.Warn('appLoader skip prelinking already present.')
         else:
            found = False

            for line in skipPrelink:
               if line in lines:
                  found = True
                  lines.remove(line)

            if found:
               log.Info('Removed appLoader prelink exclusion from %s.', prelink)

         # XXX: This can technically fail.  Actually, there are a
         # whole host of things that can fail in this function.  On
         # one hand we would like to catch errors and correct them
         # while being fairly resilient at installation time.
         #
         # One option might be to have a @failok decoration for things
         # that can fail loudly for internal builds but do not cause
         # installations to fail in release builds.
         prelink.write_lines(lines, encoding='utf-8')
      else:
         log.Info('Prelink not present, skipping configuration.')

   def _validate_cpu_flags(self, reqFlags):
      """
      Checks to ensure that every CPU on the given machine has all the
      flags passed in the reqFlags list.
      @param: A list of the flags that must be present.
      @returns: True on all necessary flags present, False othewise.
      """
      flagSec = False
      cpuInfoFile = '/proc/cpuinfo'
      try:
         cpuInfo = open(cpuInfoFile, mode='r')
         for line in cpuInfo:
            if line.startswith('flags'):
               flagSec = True
               flagsFound = line.split(' ')
               for flag in reqFlags:
                  if flag not in flagsFound:
                     return False
         return flagSec
      except IOError:
         raise Exception('Could not open %s' % (cpuInfoFile))
      finally:
         if cpuInfo:
            cpuInfo.close()

   def PreTransactionInstall(self, old, new, upgrade):
      if ENV.get('VMWARE_SKIP_VERSION_CHECKS'):
         return  # skip all checks

      # CPU flags check
      # TODO: also check for vt / ept / (AMD equivalent)
      reqFlags = ['lm']
      cpuOk = self._validate_cpu_flags(reqFlags)
      kernelOk = self.isVersionOkay(self.getHostKernelVersion(),
                                    MIN_KERNEL_VERSION)
      glibcOk = self.isVersionOkay(self.getHostGlibcVersion(),
                                   MIN_GLIBC_VERSION)
      if not cpuOk:
         self.UserMessage('One or more of your processors does not have the '
                          'necessary 64bit extensions to run VMware '
                          'virtual machines.')
      if not kernelOk:
         self.UserMessage('The Linux kernel version on this system is '
                          'insufficient for running VMware virtual machines. '
                          'Linux kernel version %s or later is required.'
                          % str(MIN_KERNEL_VERSION))
      if not glibcOk:
         self.UserMessage('The glibc library version on this system is '
                          'insufficient for running VMware virtual machines. '
                          'Glibc library version %s or later is required.'
                          % str(MIN_GLIBC_VERSION))
      if not cpuOk or not glibcOk or not kernelOk:
         raise Exception('Host not supported by this product.')

   def PostTransactionInstall(self, old, new, upgrade):
      # Build modules with modconfig.  This can't happen during the other install
      # phases because VMIS has locked the database and modconfig invokes another
      # version of VMIS to register the compiled modules.
      if ENV.get('VMWARE_SKIP_MODULES'):
         log.Info('Skipping kernel module installation')
      else:
         # run depmod -a before modconfig in case we are upgrading from a version
         # that did not properly do this after uninstalling modules. modconfig
         # uses modules.dep to determine upstream status of modules, therefore
         # it might get confused if modules.dep is not up-to-date.
         self.RunCommand('depmod', '-a', ignoreErrors=True)
         ret = self.RunCommand(BINDIR/'vmware-modconfig', '--console', '--install-all')
         if ret.retCode == 0:
            log.Info('Successfully installed kernel modules')
         else:
            log.Info('Unable to install kernel modules')
            log.Info('stdout: %s' % ret.stdout)
            log.Info('stderr: %s' % ret.stderr)

   def _checkXenPresence(self):
      """
      Checks whether this install is within a Xen domain,
      that is running on a Xen kernel.  Returns True if so.
      """
      xenTest = path('/proc/xen/capabilities')
      return xenTest.exists() and len(xenTest.text())

   def _escape(self, string):
      """ Escapes a string for use in a shell context """
      # XXX: Borrowed from util/shell.py: Escape.  Break that into a component-side
      # include file and remove this method once that's done.
      return "'%s'" % string.replace("'", '"\'"')

   def _configureNetworking(self, old, new):
      if 'VMWARE_SKIP_NETWORKING' in ENV:
         return

      vnetlib = BINDIR/'vmware-networks'

      backup = ENV.get('VMWARE_RESTORE_NETWORKING')
      backup = backup and path(backup)

      locations = ENV.get('VMWARE_MIGRATE_NETWORKING')
      locations = locations and path(locations)

      if backup and backup.owner == 'root':
         log.Info('Restoring previous network settings from %s', backup)
         backup.copy('/etc/vmware/networking')
         backup.remove()
         old = 1
      elif locations and locations.owner == 'root' and not old:
         log.Info('Migrating old network settings from %s', locations)
         if self.RunCommand(vnetlib, '--migrate-network-settings', locations, ignoreErrors=True).retCode == 0:
            locations.remove()
            return True
         else:
            log.Info('Migrating network settings failed, forcing new network settings')
            old = 0
            new = 1

      # old might be None so make it 0 for the purposes of vnetlib.
      old = old or 0

      return self.RunCommand(vnetlib, '--postinstall', '%s,%s,%s' % ('vmware-player', old, new),
                             ignoreErrors=True).retCode == 0

   def getHostGlibcVersion(self):
      """Return host's glibc version as a tuple."""
      try:
         import subprocess
         out = subprocess.check_output(["ldd", "--version"]).decode()
         brand = out.split('\n', 1)[0]
      except subprocess.CalledProcessError:
         log.Info('Unable to query glibc version.')
         return None
      # brand = full ldd brand string, e.g. 'ldd (Distro GLIBC N.NN) 2.17'
      log.Info('Glibc brand string ' + brand)
      try:
          # versionStr = glibc version, e.g. '2.17'
          versionStr = brand.split(' ')[-1]
          glibcVersion = tuple([int(x) for x in versionStr.split(".")])
          log.Info('Found glibc version ' + str(glibcVersion))
          return glibcVersion
      except Exception as e:
          log.Info('Unable to parse glibc version: ' + str(e))
          return None

   def getHostKernelVersion(self):
      """Return host's kernel version as a tuple."""
      import platform
      # releaseStr = full kernel version, e.g. '3.10.0-327.el7.x86_64'
      releaseStr = platform.release()
      log.Info('Kernel release string ' + releaseStr)
      # versionStr = kernel numeric version, e.g. '3.10.0'
      try:
          versionStr = releaseStr.split("-")[0]
          kernelVersion = tuple([int(x) for x in versionStr.split(".")])
          log.Info('Found Linux kernel version ' + str(kernelVersion))
          return kernelVersion
      except Exception as e:
          log.Info('Unable to parse kernel version: ' + str(e))
          return None

   def isVersionOkay(self, version, minVersion):
      if not version:
         return True  # Do not error out on parse failure
      elif version < minVersion:
         return False
      else:
         return True