def add(name, path, osversion, servicepack, vm, ip, port, adapter, netmask,
gateway, cpus, ramsize, vramsize, mode):
if not os.path.exists(path):
log.error('Disk image not found: %s', path)
exit(1)
session = Session()
image = session.query(Image).filter_by(name=name).first()
if image:
log.error("Image already exists: %s", name)
exit(1)
session.add(
Image(name=str(name),
path=path,
osversion=osversion,
servicepack=str(servicepack),
mode=mode,
ipaddr=ip,
port=port,
adapter=adapter,
netmask=netmask,
gateway=gateway,
cpus=cpus,
ramsize=ramsize,
vramsize=vramsize,
vm=str(vm)))
session.commit()
def snapshot(name, vmname, ipaddr, resolution, ramsize, cpus, hostname,
adapter, vm_visible, count, vrde, vrde_port, interactive, debug):
if debug:
log.setLevel(logging.DEBUG)
session = Session()
if adapter:
log.error(
"Specifying a different adapter is not yet supported for "
"snapshots (this will require detaching the current adapter and "
"attaching a new one after the static IP address has been "
"updated or so)."
)
exit(1)
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
# From now on this image is multiattach.
image.mode = "multiattach"
session.commit()
if not count:
snapshot = do_snapshot(
image, vmname, ipaddr, resolution, ramsize, cpus,
hostname or random_string(8, 16), adapter, vm_visible,
vrde, vrde_port, interactive
)
session.add(snapshot)
else:
if hostname:
log.error(
"You specified a hostname, but this is not supported when "
"creating multiple snapshots at once."
)
exit(1)
for x in xrange(count):
snapshot = do_snapshot(
image, "%s%d" % (vmname, x + 1), ipaddr, resolution,
ramsize, cpus, hostname, adapter, vm_visible,
vrde, vrde_port, interactive
)
session.add(snapshot)
# TODO Implement some limits to make sure that the IP address does
# not "exceed" its provided subnet (and thus also require the user
# to specify an IP range, rather than an IP address).
ipaddr = ipaddr_increase(ipaddr)
hostname = random_string(8, 16)
session.commit()
def snapshot(name, vmname, ipaddr, resolution, ramsize, cpus, hostname,
adapter, vm_visible, count, vrde, vrde_port, interactive, debug):
if debug:
log.setLevel(logging.DEBUG)
session = Session()
if adapter:
log.error(
"Specifying a different adapter is not yet supported for "
"snapshots (this will require detaching the current adapter and "
"attaching a new one after the static IP address has been "
"updated or so)."
)
exit(1)
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
# From now on this image is multiattach.
image.mode = "multiattach"
session.commit()
if not count:
snapshot = do_snapshot(
image, vmname, ipaddr, resolution, ramsize, cpus,
hostname or random_string(8, 16), adapter, vm_visible,
vrde, vrde_port, interactive
)
session.add(snapshot)
else:
if hostname:
log.error(
"You specified a hostname, but this is not supported when "
"creating multiple snapshots at once."
)
exit(1)
for x in xrange(count):
snapshot = do_snapshot(
image, "%s%d" % (vmname, x + 1), ipaddr, resolution,
ramsize, cpus, hostname, adapter, vm_visible,
vrde, vrde_port, interactive
)
session.add(snapshot)
# TODO Implement some limits to make sure that the IP address does
# not "exceed" its provided subnet (and thus also require the user
# to specify an IP range, rather than an IP address).
ipaddr = ipaddr_increase(ipaddr)
hostname = random_string(8, 16)
session.commit()
def snapshot(name, vmname, ipaddr, resolution, ramsize, cpus, hostname,
adapter, vm_visible):
session = Session()
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
# From now on this image is multiattach.
image.mode = "multiattach"
session.commit()
m, h = initvm(image, name=vmname)
m.start_vm(visible=vm_visible)
wait_for_host(image.ipaddr, image.port)
a = Agent(image.ipaddr, image.port)
a.ping()
# Assign a new hostname.
a.hostname(hostname)
a.reboot()
a.kill()
# Wait for the reboot to kick in.
time.sleep(10)
wait_for_host(image.ipaddr, image.port)
a.ping()
if resolution:
width, height = resolution.split("x")
a.resolution(width, height)
a.remove("C:\\vmcloak")
a.static_ip(ipaddr, image.netmask, image.gateway, h.interface)
m.snapshot("vmcloak", "Snapshot created by VM Cloak.")
m.stopvm()
# Create a database entry for this snapshot.
snapshot = Snapshot(image_id=image.id, vmname=vmname, ipaddr=ipaddr,
port=image.port, hostname=hostname)
session.add(snapshot)
session.commit()
def add(name, path, osversion, servicepack, vm, ip, port, adapter,
netmask, gateway, cpus, ramsize, vramsize, mode):
if not os.path.exists(path):
log.error('Disk image not found: %s', path)
exit(1)
session = Session()
image = session.query(Image).filter_by(name=name).first()
if image:
log.error("Image already exists: %s", name)
exit(1)
session.add(Image(name=str(name), path=path, osversion=osversion,
servicepack=str(servicepack), mode=mode,
ipaddr=ip, port=port, adapter=adapter,
netmask=netmask, gateway=gateway,
cpus=cpus, ramsize=ramsize, vramsize=vramsize, vm=str(vm)))
session.commit()
def snapshot(name, vmname, ipaddr, resolution, ramsize, cpus, hostname,
adapter, vm_visible, count):
session = Session()
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
# From now on this image is multiattach.
image.mode = "multiattach"
session.commit()
if not count:
snapshot = do_snapshot(
image, vmname, ipaddr, resolution, ramsize, cpus,
hostname or random_string(8, 16), adapter, vm_visible
)
session.add(snapshot)
else:
if hostname:
log.error(
"You specified a hostname, but this is not supported when "
"creating multiple snapshots at once."
)
exit(1)
for x in xrange(count):
snapshot = do_snapshot(
image, "%s%d" % (vmname, x + 1), ipaddr, resolution,
ramsize, cpus, hostname, adapter, vm_visible
)
session.add(snapshot)
# TODO Implement some limits to make sure that the IP address does
# not "exceed" its provided subnet (and thus also require the user
# to specify an IP range, rather than an IP address).
ipaddr = ipaddr_increase(ipaddr)
hostname = random_string(8, 16)
session.commit()
def clone(name, outname):
session = Session()
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
outpath = os.path.join(image_path, "%s.vdi" % outname)
m = VirtualBox(None)
m.clone_hd(image.path, outpath)
# Retain all fields but update the mode, name & path.
make_transient(image)
image.id = None
image.mode = "normal"
image.name = outname
image.path = outpath
session.add(image)
session.commit()
def clone(name, outname):
session = Session()
image = session.query(Image).filter_by(name=name).first()
if not image:
log.error("Image not found: %s", name)
exit(1)
outpath = os.path.join(image_path, "%s.vdi" % outname)
m = VirtualBox(None)
m.clone_hd(image.path, outpath)
# Retain all fields but update the mode, name & path.
make_transient(image)
image.id = None
image.mode = "normal"
image.name = outname
image.path = outpath
session.add(image)
session.commit()
def init(name, winxp, win7x86, win7x64, win81x86, win81x64, win10x86, win10x64,
product, vm, iso_mount, serial_key, ip, port, adapter, netmask,
gateway, dns, cpus, ramsize, vramsize, tempdir, resolution,
vm_visible, debug, verbose):
if verbose:
log.setLevel(logging.INFO)
if debug:
log.setLevel(logging.DEBUG)
session = Session()
image = session.query(Image).filter_by(name=name).first()
if image:
log.error("Image already exists: %s", name)
exit(1)
if vm not in VMCLOAK_VM_MODES:
log.error(
"Only VirtualBox Machinery or iso is supported at this point.")
exit(1)
if winxp:
h = WindowsXP()
osversion = "winxp"
ramsize = ramsize or 1024
elif win7x86:
h = Windows7x86()
ramsize = ramsize or 1024
osversion = "win7x86"
elif win7x64:
h = Windows7x64()
ramsize = ramsize or 2048
osversion = "win7x64"
elif win81x86:
h = Windows81x86()
ramsize = ramsize or 2048
osversion = "win81x86"
elif win81x64:
h = Windows81x64()
ramsize = ramsize or 2048
osversion = "win81x64"
elif win10x86:
h = Windows10x86()
ramsize = ramsize or 2048
osversion = "win10x86"
elif win10x64:
h = Windows10x64()
ramsize = ramsize or 2048
osversion = "win10x64"
else:
log.error("Please provide one of --winxp, --win7x86, --win7x64, "
"--win81x86, --win81x64, --win10x86, --win10x64.")
exit(1)
mount = h.pickmount(iso_mount)
if not mount:
log.error("Please specify --iso-mount to a directory containing the "
"mounted Windows Installer ISO image.")
log.info("Refer to the documentation on mounting an .iso image.")
exit(1)
if not h.set_serial_key(serial_key):
exit(1)
h.configure(tempdir=tempdir, product=product)
reso_width, reso_height = resolution.split("x")
settings = dict(
GUEST_IP=ip,
AGENT_PORT=port,
GUEST_MASK=netmask,
GUEST_GATEWAY=gateway,
DNSSERVER=dns,
DEBUG="yes" if debug else "no",
RESO_WIDTH=reso_width,
RESO_HEIGHT=reso_height,
INTERFACE=h.interface,
)
bootstrap = tempfile.mkdtemp(dir=tempdir)
vmcloak_dir = os.path.join(bootstrap, "vmcloak")
os.mkdir(vmcloak_dir)
# Write the configuration values for bootstrap.bat.
with open(os.path.join(vmcloak_dir, "settings.bat"), "wb") as f:
for key, value in settings.items():
print >> f, "set %s=%s" % (key, value)
# Download the Python dependency and set it up for bootstrapping the VM.
d = Python27(i=Image(osversion=osversion))
d.download()
shutil.copy(d.filepath, vmcloak_dir)
iso_path = os.path.join(tempdir, "%s.iso" % name)
hdd_path = os.path.join(image_path, "%s.vdi" % name)
m = VirtualBox(name=name)
if not h.buildiso(mount, iso_path, bootstrap, tempdir):
shutil.rmtree(bootstrap)
exit(1)
shutil.rmtree(bootstrap)
if vm == "virtualbox":
m.create_vm()
m.os_type(osversion)
m.cpus(cpus)
m.mouse("usbtablet")
m.ramsize(ramsize)
m.vramsize(vramsize)
m.create_hd(hdd_path)
m.attach_iso(iso_path)
m.hostonly(nictype=h.nictype, adapter=adapter)
log.info("Starting the Virtual Machine %r to install Windows.", name)
m.start_vm(visible=vm_visible)
m.wait_for_state(shutdown=True)
m.detach_iso()
os.unlink(iso_path)
m.remove_hd()
m.compact_hd(hdd_path)
m.delete_vm()
else:
log.info("You can find your deployment ISO image from : %s" % iso_path)
log.info("Added image %r to the repository.", name)
session.add(
Image(name=name,
path=hdd_path,
osversion=osversion,
servicepack="%s" % h.service_pack,
mode="normal",
ipaddr=ip,
port=port,
adapter=adapter,
netmask=netmask,
gateway=gateway,
cpus=cpus,
ramsize=ramsize,
vramsize=vramsize,
vm="%s" % vm))
session.commit()
def init(name, winxp, win7x86, win7x64, win81x86, win81x64, win10x86,
win10x64, product, vm, iso_mount, serial_key, ip, port, adapter,
netmask, gateway, dns, cpus, ramsize, vramsize, tempdir, resolution,
vm_visible, debug, verbose):
if verbose:
log.setLevel(logging.INFO)
if debug:
log.setLevel(logging.DEBUG)
session = Session()
image = session.query(Image).filter_by(name=name).first()
if image:
log.error("Image already exists: %s", name)
exit(1)
if vm not in VMCLOAK_VM_MODES:
log.error("Only VirtualBox Machinery or iso is supported at this point.")
exit(1)
if winxp:
h = WindowsXP()
osversion = "winxp"
ramsize = ramsize or 1024
elif win7x86:
h = Windows7x86()
ramsize = ramsize or 1024
osversion = "win7x86"
elif win7x64:
h = Windows7x64()
ramsize = ramsize or 2048
osversion = "win7x64"
elif win81x86:
h = Windows81x86()
ramsize = ramsize or 2048
osversion = "win81x86"
elif win81x64:
h = Windows81x64()
ramsize = ramsize or 2048
osversion = "win81x64"
elif win10x86:
h = Windows10x86()
ramsize = ramsize or 2048
osversion = "win10x86"
elif win10x64:
h = Windows10x64()
ramsize = ramsize or 2048
osversion = "win10x64"
else:
log.error(
"Please provide one of --winxp, --win7x86, --win7x64, "
"--win81x86, --win81x64, --win10x86, --win10x64."
)
exit(1)
mount = h.pickmount(iso_mount)
if not mount:
log.error("Please specify --iso-mount to a directory containing the "
"mounted Windows Installer ISO image.")
log.info("Refer to the documentation on mounting an .iso image.")
exit(1)
if not h.set_serial_key(serial_key):
exit(1)
h.configure(tempdir=tempdir, product=product)
reso_width, reso_height = resolution.split("x")
settings = dict(
GUEST_IP=ip,
AGENT_PORT=port,
GUEST_MASK=netmask,
GUEST_GATEWAY=gateway,
DNSSERVER=dns,
DEBUG="yes" if debug else "no",
RESO_WIDTH=reso_width,
RESO_HEIGHT=reso_height,
INTERFACE=h.interface,
)
bootstrap = tempfile.mkdtemp(dir=tempdir)
vmcloak_dir = os.path.join(bootstrap, "vmcloak")
os.mkdir(vmcloak_dir)
# Write the configuration values for bootstrap.bat.
with open(os.path.join(vmcloak_dir, "settings.bat"), "wb") as f:
for key, value in settings.items():
print>>f, "set %s=%s" % (key, value)
# Download the Python dependency and set it up for bootstrapping the VM.
d = Python27(i=Image(osversion=osversion))
d.download()
shutil.copy(d.filepath, vmcloak_dir)
iso_path = os.path.join(tempdir, "%s.iso" % name)
hdd_path = os.path.join(image_path, "%s.vdi" % name)
m = VirtualBox(name=name)
if not h.buildiso(mount, iso_path, bootstrap, tempdir):
shutil.rmtree(bootstrap)
exit(1)
shutil.rmtree(bootstrap)
if vm == "virtualbox":
m.create_vm()
m.os_type(osversion)
m.cpus(cpus)
m.mouse("usbtablet")
m.ramsize(ramsize)
m.vramsize(vramsize)
m.create_hd(hdd_path)
m.attach_iso(iso_path)
m.hostonly(nictype=h.nictype, adapter=adapter)
log.info("Starting the Virtual Machine %r to install Windows.", name)
m.start_vm(visible=vm_visible)
m.wait_for_state(shutdown=True)
m.detach_iso()
os.unlink(iso_path)
m.remove_hd()
m.compact_hd(hdd_path)
m.delete_vm()
else:
log.info("You can find your deployment ISO image from : %s" % iso_path)
log.info("Added image %r to the repository.", name)
session.add(Image(name=name, path=hdd_path, osversion=osversion,
servicepack="%s" % h.service_pack, mode="normal",
ipaddr=ip, port=port, adapter=adapter,
netmask=netmask, gateway=gateway,
cpus=cpus, ramsize=ramsize, vramsize=vramsize, vm="%s" % vm))
session.commit()
