Class: Macinbox::Actions::CreateVMDKFromImage

Inherits:
Object
  • Object
show all
Defined in:
lib/macinbox/actions/create_vmdk_from_image.rb

Instance Method Summary collapse

Constructor Details

#initialize(opts) ⇒ CreateVMDKFromImage

Returns a new instance of CreateVMDKFromImage.

Raises:



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 16

def initialize(opts)
  @input_image       = opts[:image_path]  or raise ArgumentError.new(":image_path not specified")
  @output_path       = opts[:vmdk_path]   or raise ArgumentError.new(":vmdk_path not specified")
  @vmware_fusion_app = opts[:vmware_path] or raise ArgumentError.new(":vmware_path not specified")
  @use_qemu          = opts[:use_qemu]

  @collector         = opts[:collector]   or raise ArgumentError.new(":collector not specified")

  raise Macinbox::Error.new("input image not found")   unless File.exist? @input_image
  raise Macinbox::Error.new("VMware Fusion not found") unless File.exist? @vmware_fusion_app

  if @use_qemu
    raise Macinbox::Error.new("qemu-img not found") unless File.exist? "/usr/local/bin/qemu-img"
  end
end

Instance Method Details

#attach_image ⇒ Object 



55
56
57
58
59
60
61
62
63
64
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 55

def attach_image
  Logger.info "Attaching the image..." do
    @disk = VirtualDisk.new(@image)
    @collector.on_cleanup { @disk.detach! }
    @image_mountpoint = "#{@temp_dir}/image_mountpoint"
    FileUtils.mkdir @image_mountpoint
    @disk.attach
    @disk.mount(at: @image_mountpoint, owners: true)
  end
end

#convert_image ⇒ Object 



134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 134

def convert_image
  Logger.info "Converting the image to VMDK format#{@use_qemu ? " using QEMU" : ""}..." do
    if @use_qemu
      @disk.convert(outfile: "#{@temp_dir}/macinbox.dmg")
      Task.run %W[ /usr/local/bin/qemu-img convert -f dmg -O vmdk #{@temp_dir}/macinbox.dmg #{@temp_dir}/macinbox.vmdk ]
    else
      @disk.attach
      task_opts = $verbose ? {} : { :out => File::NULL }
      rawdiskCreator = "#{@vmware_fusion_app}/Contents/Library/vmware-rawdiskCreator"
      vdiskmanager = "#{@vmware_fusion_app}/Contents/Library/vmware-vdiskmanager"
      Dir.chdir(@temp_dir) do
        Task.run %W[ #{rawdiskCreator} create #{@disk.device} fullDevice rawdisk lsilogic ] + [task_opts]
        Task.run %W[ #{vdiskmanager} -t 0 -r rawdisk.vmdk macinbox.vmdk ] + [task_opts]
      end
      @disk.eject
    end
  end
end

#copy_input_image ⇒ Object 



48
49
50
51
52
53
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 48

def copy_input_image
  Logger.info "Copying the image..." do
    @image = "#{@temp_dir}/macinbox.sparseimage"
    Macinbox::copyfiles(from: @input_image, to: @image)
  end
end

#create_temp_dir ⇒ Object 



43
44
45
46
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 43

def create_temp_dir
  @temp_dir = Task.backtick %W[ /usr/bin/mktemp -d -t create_vmdk_from_image ]
  @collector.add_temp_dir @temp_dir
end

#eject_image ⇒ Object 



130
131
132
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 130

def eject_image
  @disk.eject
end

#install_vmware_tools ⇒ Object 



66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 66

def install_vmware_tools
  tools_image = "#{@vmware_fusion_app}/Contents/Library/isoimages/darwin.iso"

  unless File.exist? tools_image
    Logger.info "Downloading the VMware Tools..." do
      bundle_version = Task.backtick %W[ defaults read #{"/Applications/VMware Fusion.app/Contents/Info.plist"} CFBundleVersion ]
      bundle_short_version = Task.backtick %W[ defaults read #{"/Applications/VMware Fusion.app/Contents/Info.plist"} CFBundleShortVersionString ]
      darwin_iso_url = "http://softwareupdate.vmware.com/cds/vmw-desktop/fusion/#{bundle_short_version}/#{bundle_version}/packages/com.vmware.fusion.tools.darwin.zip.tar"
      Dir.chdir(@temp_dir) do
        Task.run %W[ /usr/bin/curl #{darwin_iso_url} -O ] + ($verbose ? [] : %W[ -s -S ])
        Task.run %W[ /usr/bin/tar -xf com.vmware.fusion.tools.darwin.zip.tar com.vmware.fusion.tools.darwin.zip ]
        Task.run %W[ /usr/bin/unzip ] + ($verbose ? [] : %W[ -qq ]) + %W[ com.vmware.fusion.tools.darwin.zip payload/darwin.iso ]
      end
      tools_image = "#{@temp_dir}/payload/darwin.iso"
    end
  end

  Logger.info "Installing the VMware Tools..." do
    tools_disk = VirtualDisk.new(tools_image)
    @collector.on_cleanup { tools_disk.detach! }
    tools_mountpoint = "#{@temp_dir}/tools_mountpoint"
    FileUtils.mkdir tools_mountpoint
    tools_disk.attach
    tools_disk.mount(at: tools_mountpoint)
    tools_package = "#{tools_mountpoint}/Install VMware Tools.app/Contents/Resources/VMware Tools.pkg"
    tools_package_dir = "#{@temp_dir}/tools_package"
    Task.run %W[ /usr/sbin/pkgutil --expand #{tools_package} #{tools_package_dir} ]
    Task.run %W[ /usr/bin/ditto -x -z #{tools_package_dir}/files.pkg/Payload #{@image_mountpoint} ]
    image_vmhgfs_filesystem_resources = "#{@image_mountpoint}/Library/Filesystems/vmhgfs.fs/Contents/Resources"
    FileUtils.mkdir_p image_vmhgfs_filesystem_resources
    FileUtils.ln_s "/Library/Application Support/VMware Tools/mount_vmhgfs", "#{image_vmhgfs_filesystem_resources}/"
    tools_disk.eject
  end
end

#run ⇒ Object 



32
33
34
35
36
37
38
39
40
41
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 32

def run
  create_temp_dir
  copy_input_image
  attach_image
  install_vmware_tools
  set_spc_kextpolicy
  eject_image
  convert_image
  save_image
end

#save_image ⇒ Object 



153
154
155
156
157
158
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 153

def save_image
  Logger.info "Moving the VMDK to the destination..." do
    FileUtils.chown ENV["SUDO_USER"], nil, "#{@temp_dir}/macinbox.vmdk"
    FileUtils.mv "#{@temp_dir}/macinbox.vmdk", @output_path
  end
end

#set_spc_kextpolicy ⇒ Object 



101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/macinbox/actions/create_vmdk_from_image.rb', line 101

def set_spc_kextpolicy
  Logger.info "Setting the KextPolicy to allow loading the VMware kernel extensions..." do
    image_spc_kextpolicy = "#{@image_mountpoint}/private/var/db/SystemPolicyConfiguration/KextPolicy"
    unless File.exist? image_spc_kextpolicy
      Task.run_with_input %W[ /usr/bin/sqlite3 #{image_spc_kextpolicy} ] do |pipe|
        pipe.write "          PRAGMA foreign_keys=OFF;\n          BEGIN TRANSACTION;\n          CREATE TABLE kext_load_history_v3 ( path TEXT PRIMARY KEY, team_id TEXT, bundle_id TEXT, boot_uuid TEXT, created_at TEXT, last_seen TEXT, flags INTEGER );\n          CREATE TABLE kext_policy ( team_id TEXT, bundle_id TEXT, allowed BOOLEAN, developer_name TEXT, flags INTEGER, PRIMARY KEY (team_id, bundle_id) );\n          CREATE TABLE kext_policy_mdm ( team_id TEXT, bundle_id TEXT, allowed BOOLEAN, payload_uuid TEXT, PRIMARY KEY (team_id, bundle_id) );\n          CREATE TABLE settings ( name TEXT, value TEXT, PRIMARY KEY (name) );\n          COMMIT;\n        EOF\n      end\n    end\n    Task.run_with_input %W[ /usr/bin/sqlite3 \#{image_spc_kextpolicy} ] do |pipe|\n      pipe.write <<~EOF\n        PRAGMA foreign_keys=OFF;\n        BEGIN TRANSACTION;\n        INSERT OR REPLACE INTO kext_policy VALUES('EG7KH642X6','com.vmware.kext.VMwareGfx',1,'VMware, Inc.',1);\n        INSERT OR REPLACE INTO kext_policy VALUES('EG7KH642X6','com.vmware.kext.vmmemctl',1,'VMware, Inc.',1);\n        INSERT OR REPLACE INTO kext_policy VALUES('EG7KH642X6','com.vmware.kext.vmhgfs',1,'VMware, Inc.',1);\n        COMMIT;\n      EOF\n    end\n  end\nend\n"