manageiq-smartstate 0.1.0

data/lib/disk/modules/MSVSDynamicDisk.rb

@@ -0,0 +1,42 @@
+require 'disk/modules/MSCommon'
+
+module MSVSDynamicDisk
+  def d_init
+    self.diskType = "MSVS Dynamic"
+    self.blockSize = MSCommon::SECTOR_LENGTH
+    if dInfo.mountMode.nil? || dInfo.mountMode == "r"
+      dInfo.mountMode = "r"
+      fileMode = "r"
+    elsif dInfo.mountMode == "rw"
+      fileMode = "r+"
+    else
+      raise "Unrecognized mountMode: #{dInfo.mountMode}"
+    end
+    @ms_disk_file = if dInfo.hyperv_connection
+                      MSCommon.connect_to_hyperv(dInfo)
+                    else
+                      MiqLargeFile.open(dInfo.fileName, fileMode)
+                    end
+    MSCommon.d_init_common(dInfo, @ms_disk_file)
+  end
+
+  def getBase
+    self
+  end
+
+  def d_read(pos, len)
+    MSCommon.d_read_common(pos, len)
+  end
+
+  def d_write(pos, buf, len)
+    MSCommon.d_write_common(pos, buf, len)
+  end
+
+  def d_close
+    @ms_disk_file.close
+  end
+
+  def d_size
+    MSCommon.d_size_common
+  end
+end # module

data/lib/disk/modules/MSVSFixedDisk.rb

@@ -0,0 +1,45 @@
+require 'disk/modules/MiqLargeFile'
+
+module MSVSFixedDisk
+  def d_init
+    @diskType = "MSVSFixed"
+    @blockSize = 512
+
+    if dInfo.mountMode.nil? || dInfo.mountMode == "r"
+      dInfo.mountMode = "r"
+      fileMode = "r"
+    elsif dInfo.mountMode == "rw"
+      fileMode = "r+"
+    else
+      raise "Unrecognized mountMode: #{dInfo.mountMode}"
+    end
+
+    if dInfo.hyperv_connection
+      @ms_flat_disk_file = MSCommon.connect_to_hyperv(dInfo)
+    else
+      @ms_flat_disk_file = MiqLargeFile.open(dInfo.fileName, fileMode)
+    end
+  end
+
+  def d_read(pos, len)
+    @ms_flat_disk_file.seek(pos, IO::SEEK_SET)
+    @ms_flat_disk_file.read(len)
+  end
+
+  def getBase
+    self
+  end
+
+  def d_write(pos, buf, len)
+    @ms_flat_disk_file.seek(pos, IO::SEEK_SET)
+    @ms_flat_disk_file.write(buf, len)
+  end
+
+  def d_close
+    @ms_flat_disk_file.close
+  end
+
+  def d_size
+    File.size(dInfo.fileName)
+  end
+end

data/lib/disk/modules/MiqLargeFile.rb

@@ -0,0 +1,63 @@
+require 'sys-uname'
+require 'util/miq-system'
+
+if Sys::Platform::IMPL == :linux
+  if MiqSystem.arch == :x86_64
+    require 'linux_block_device'
+    require 'disk/modules/RawBlockIO'
+  end
+elsif Sys::Platform::OS == :windows
+  require 'disk/modules/MiqLargeFileWin32'
+end
+
+module MiqLargeFile
+  def self.open(file_name, flags)
+    case Sys::Platform::OS
+    when :windows
+      return(MiqLargeFileWin32.new(file_name, flags))
+    when :unix
+      if Sys::Platform::IMPL == :linux
+        return(RawBlockIO.new(file_name, flags)) if MiqLargeFileStat.new(file_name).blockdev?
+      end
+      return(MiqLargeFileOther.new(file_name, flags))
+    else
+      return(MiqLargeFileOther.new(file_name, flags))
+    end
+  end
+
+  def self.size(file_name)
+    case Sys::Platform::OS
+    when :windows
+      # The win32/file require is needed to support +2GB file sizes
+      require 'win32/file'
+      File.size(file_name)
+    else
+      f = open(file_name, "r")
+      s = f.size
+      f.close
+      return s
+    end
+  end
+
+  # For camcorder interposition.
+  class MiqLargeFileStat
+    def initialize(file_name)
+      @file_name = file_name
+    end
+
+    def blockdev?
+      File.stat(@file_name).blockdev?
+    end
+  end
+
+  class MiqLargeFileOther < File
+    def write(buf, _len)
+      super(buf)
+    end
+
+    def size
+      return stat.size unless stat.blockdev?
+      LinuxBlockDevice.size(fileno)
+    end
+  end
+end

data/lib/disk/modules/MiqLargeFileWin32.rb

@@ -0,0 +1,107 @@
+require 'memory_buffer'
+require 'Win32API'
+
+class MiqLargeFileWin32
+  # Access specifiers.
+  GENERIC_READ  = 0x80000000
+  GENERIC_WRITE = 0x40000000
+
+  # Share specifiers.
+  FILE_SHARE_READ   = 0x00000001
+  FILE_SHARE_WRITE  = 0x00000002
+
+  # Creation disposition specifiers.
+  OPEN_EXISTING = 3
+
+  # Seek origin.
+  FILE_BEGIN    = 0
+  FILE_CURRENT  = 1
+  FILE_END      = 2
+
+  # Misc.
+  INVALID_HANDLE_VALUE      = -1
+
+  # API.
+  @@createFile        = Win32API.new('kernel32', 'CreateFileA',       'PLLLLLL',  'L')
+  @@setFilePointerEx  = Win32API.new('kernel32', 'SetFilePointerEx',  'LLLPL',      'L')
+  @@readFile          = Win32API.new('kernel32', 'ReadFile',          'LPLPL',    'L')
+  @@writeFile         = Win32API.new('kernel32', 'WriteFile',         'LPLPL',    'L')
+  @@closeHandle       = Win32API.new('kernel32', 'CloseHandle',       'L',        'L')
+  @@getFileSize       = Win32API.new('kernel32', 'GetFileSizeEx',     'LP',       'L')
+  @@getLastError      = Win32API.new('kernel32', 'GetLastError',      nil,        'L')
+
+  # Default constructor.
+  def initialize(file, access = "r")
+    accessOptions = 0
+    accessOptions |= GENERIC_READ if access.include?("r")
+    accessOptions |= GENERIC_WRITE if access.include?("+")
+    @hFile = @@createFile.call(file, accessOptions, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, OPEN_EXISTING, 0, 0)
+    raise "Couldn't open file #{file}" if @hFile == INVALID_HANDLE_VALUE
+    $log.debug("MiqLargeFile<#{object_id}> Opening #{file}, handle 0x#{'%08x' % @hFile}") if $log
+  end
+
+  # Alternative initialization.
+  def self.open(file, access = "r")
+    initialize(file, access)
+  end
+
+  def size
+    siz = MemoryBuffer.create_quad
+    err = @@getFileSize.call(@hFile, siz)
+    siz.unpack('Q')[0]
+  end
+
+  def seek(offset, whence)
+    method = case whence
+             when IO::SEEK_SET then FILE_BEGIN
+             when IO::SEEK_CUR then FILE_CURRENT
+             when IO::SEEK_END then FILE_END
+             else raise "Invalid seek method: #{whence}"
+             end
+    offHi, offLo = offset.divmod(4294967296) # 2 ** 32
+    new_pos = MemoryBuffer.create_quad
+    res = @@setFilePointerEx.call(@hFile, offLo, offHi, new_pos, method)
+    if res == 0
+      last_error = @@getLastError.call
+      raise "SetFilePointerEx failed - Offset:[#{offset}] method:[#{whence}] Handle:[0x#{'%08x' % @hFile}] GetLastErrorNum:[#{last_error}]"
+    end
+    new_pos = new_pos.unpack('Q')[0]
+    new_pos
+  end
+
+  def read(bytes)
+    # puts "#{getFilePos}, #{bytes}" if $track_pos
+    buf = MemoryBuffer.create(bytes)
+    bytesRead = MemoryBuffer.create_long
+    res = @@readFile.call(@hFile, buf, bytes, bytesRead, 0)
+    bytesRead = bytesRead.unpack('L')[0]
+    if bytesRead == 0 && bytes != 0 || res == 0
+      last_error = @@getLastError.call
+      raise "Read from disk file failed - result:[#{res}] bytesRead:[#{bytesRead}] expected:[#{bytes}] file pointer:[#{getFilePos}] file size:[#{size}] Handle:[0x#{'%08x' % @hFile}] GetLastErrorNum:[#{last_error}]."
+    end
+    buf = buf[0...bytesRead] if bytesRead != bytes
+    buf
+  end
+
+  def write(buf, len)
+    # NOTE: len must be a double word.
+    bytesWritten = MemoryBuffer.create_long
+    res = @@writeFile.call(@hFile, buf, len, bytesWritten, 0)
+    raise "WriteFile failed" if res == 0
+    bytesWritten = bytesWritten.unpack('L')[0]
+    raise "Not all data written" if bytesWritten != len
+    bytesWritten
+  end
+
+  def close
+    $log.debug("MiqLargeFile<#{object_id}> Closing handle 0x#{'%08x' % @hFile}") if $log
+    h = @hFile
+    res = @@closeHandle.call(@hFile)
+    @hFile = INVALID_HANDLE_VALUE if res != 0
+    return res, h
+  end
+
+  def getFilePos
+    seek(0, IO::SEEK_CUR)
+  end
+end