def main():
args = get_args()
try:
if is_xenial_or_above():
ssl = __import__("ssl")
context = ssl._create_unverified_context()
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port, sslContext=context)
else:
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port)
si_content = si.RetrieveContent()
except:
print "Unable to connect to %s" % args.host
exit(1)
# get VM object
vm_obj = get_obj(si_content, [vim.VirtualMachine], args.vm_name)
if not vm_obj:
print "VM %s not pressent" %(args.vm_name)
exit(1)
if args.pci_nics:
task = add_pci_nics(args, vm_obj)
wait_for_task(task)
if args.sriov_nics:
task = add_sriov_nics(args, vm_obj, si_content)
wait_for_task(task)
connect.Disconnect(si)
def main():
args = get_args()
try:
if is_xenial_or_above():
ssl = __import__("ssl")
context = ssl._create_unverified_context()
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port, sslContext=context)
else:
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port)
si_content = si.RetrieveContent()
except:
print "Unable to connect to %s" % args.host
exit(1)
# get VM object
vm_obj = get_obj(si_content, [vim.VirtualMachine], args.vm_name)
if not vm_obj:
print "VM %s not pressent" %(args.vm_name)
exit(1)
#net_obj = get_obj(si_content, [vim.Network], args.network_name)
nics = args.nics.split(',')
for nic in nics:
if not nic:
continue
task = update_mac(nic, vm_obj, si_content)
wait_for_task(task)
connect.Disconnect(si)
def fix_sriov_pg(si, dvs, dvs_pg_name):
dvs_pg_obj = get_dvs_pg_obj(si, [vim.dvs.DistributedVirtualPortgroup],
dvs_pg_name, dvs.name)
dv_pg_spec = vim.dvs.DistributedVirtualPortgroup.ConfigSpec()
dv_pg_spec.name = dvs_pg_name
dv_pg_spec.configVersion = dvs_pg_obj.config.configVersion
dv_pg_spec.type = vim.dvs.DistributedVirtualPortgroup.PortgroupType.earlyBinding
dv_pg_spec.defaultPortConfig = vim.dvs.VmwareDistributedVirtualSwitch.VmwarePortConfigPolicy(
)
dv_pg_spec.defaultPortConfig.securityPolicy = vim.dvs.VmwareDistributedVirtualSwitch.SecurityPolicy(
)
dv_pg_spec.defaultPortConfig.securityPolicy.allowPromiscuous = vim.BoolPolicy(
value=False)
dv_pg_spec.defaultPortConfig.securityPolicy.macChanges = vim.BoolPolicy(
value=True)
dv_pg_spec.defaultPortConfig.securityPolicy.forgedTransmits = vim.BoolPolicy(
value=True)
dv_pg_spec.defaultPortConfig.securityPolicy.inherited = False
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy = vim.VmwareUplinkPortTeamingPolicy(
)
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy.uplinkPortOrder = vim.VMwareUplinkPortOrderPolicy(
)
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy.uplinkPortOrder.activeUplinkPort = None
task = dvs_pg_obj.ReconfigureDVPortgroup_Task(dv_pg_spec)
wait_for_task(task)
def main():
args = get_args()
try:
if is_xenial_or_above():
ssl = __import__("ssl")
context = ssl._create_unverified_context()
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port,
sslContext=context)
else:
si = connect.SmartConnect(host=args.host,
user=args.user,
pwd=args.password,
port=args.port)
si_content = si.RetrieveContent()
except:
print "Unable to connect to %s" % args.host
exit(1)
# get VM object
vm_obj = get_obj(si_content, [vim.VirtualMachine], args.vm_name)
if not vm_obj:
print "VM %s not pressent" % (args.vm_name)
exit(1)
if args.pci_nics:
task = add_pci_nics(args, vm_obj)
wait_for_task(task)
if args.sriov_nics:
task = add_sriov_nics(args, vm_obj, si_content)
wait_for_task(task)
connect.Disconnect(si)
def add_pci_nics(args, vm):
pci_id_list = args.pci_nics
pci_id_list.sort()
if vm.runtime.powerState == vim.VirtualMachinePowerState.poweredOn:
print "VM:%s is powered ON. Cannot do hot pci add now. Shutting it down" % (
args.vm_name)
poweroffvm(vm)
for pci_id in pci_id_list:
device_config_list = []
found = False
for device_list in vm.config.hardware.device:
if (isinstance(device_list, vim.vm.device.VirtualPCIPassthrough)) == True \
and device_list.backing.id == pci_id:
print "pci_device already present! Not adding the pci device."
found = True
break
if found == True:
continue
pci_passthroughs = vm.environmentBrowser.QueryConfigTarget(
host=None).pciPassthrough
for pci_entry in pci_passthroughs:
if pci_entry.pciDevice.id == pci_id:
found = True
print "Found the pci device %s in the host" % (pci_id)
break
if found == False:
print "Did not find the pci passthrough device %s on the host" % (
pci_id)
exit(1)
print "Adding PCI device to Contrail VM: %s" % (vm_name)
deviceId = hex(pci_entry.pciDevice.deviceId % 2**16).lstrip('0x')
backing = vim.VirtualPCIPassthroughDeviceBackingInfo(
deviceId=deviceId,
id=pci_entry.pciDevice.id,
systemId=pci_entry.systemId,
vendorId=pci_entry.pciDevice.vendorId,
deviceName=pci_entry.pciDevice.deviceName)
hba_object = vim.VirtualPCIPassthrough(key=-100, backing=backing)
new_device_config = vim.VirtualDeviceConfigSpec(device=hba_object)
new_device_config.operation = "add"
new_device_config.device.connectable = vim.vm.device.VirtualDevice.ConnectInfo(
)
new_device_config.device.connectable.startConnected = True
device_config_list.append(new_device_config)
vm_spec = vim.vm.ConfigSpec()
vm_spec.deviceChange = device_config_list
task = vm.ReconfigVM_Task(spec=vm_spec)
wait_for_task(task)
def fix_sriov_pg(si, dvs, dvs_pg_name):
dvs_pg_obj = get_dvs_pg_obj(si, [vim.dvs.DistributedVirtualPortgroup],
dvs_pg_name, dvs.name)
dv_pg_spec = vim.dvs.DistributedVirtualPortgroup.ConfigSpec()
dv_pg_spec.name = dvs_pg_name
dv_pg_spec.configVersion = dvs_pg_obj.config.configVersion
dv_pg_spec.type = vim.dvs.DistributedVirtualPortgroup.PortgroupType.earlyBinding
dv_pg_spec.defaultPortConfig = vim.dvs.VmwareDistributedVirtualSwitch.VmwarePortConfigPolicy()
dv_pg_spec.defaultPortConfig.securityPolicy = vim.dvs.VmwareDistributedVirtualSwitch.SecurityPolicy()
dv_pg_spec.defaultPortConfig.securityPolicy.allowPromiscuous = vim.BoolPolicy(value=False)
dv_pg_spec.defaultPortConfig.securityPolicy.macChanges = vim.BoolPolicy(value=True)
dv_pg_spec.defaultPortConfig.securityPolicy.forgedTransmits = vim.BoolPolicy(value=True)
dv_pg_spec.defaultPortConfig.securityPolicy.inherited = False
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy = vim.VmwareUplinkPortTeamingPolicy()
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy.uplinkPortOrder = vim.VMwareUplinkPortOrderPolicy()
dv_pg_spec.defaultPortConfig.uplinkTeamingPolicy.uplinkPortOrder.activeUplinkPort = None
task = dvs_pg_obj.ReconfigureDVPortgroup_Task(dv_pg_spec)
wait_for_task(task)
def add_pci_nics(args, vm):
pci_id_list = args.pci_nics
pci_id_list.sort()
if vm.runtime.powerState == vim.VirtualMachinePowerState.poweredOn:
print "VM:%s is powered ON. Cannot do hot pci add now. Shutting it down" %(args.vm_name)
poweroffvm(vm);
for pci_id in pci_id_list:
device_config_list = []
found = False
for device_list in vm.config.hardware.device:
if (isinstance(device_list, vim.vm.device.VirtualPCIPassthrough)) == True \
and device_list.backing.id == pci_id:
print "pci_device already present! Not adding the pci device."
found = True
break
if found == True:
continue
pci_passthroughs = vm.environmentBrowser.QueryConfigTarget(host=None).pciPassthrough
for pci_entry in pci_passthroughs:
if pci_entry.pciDevice.id == pci_id:
found = True
print "Found the pci device %s in the host" %(pci_id)
break
if found == False:
print "Did not find the pci passthrough device %s on the host" %(pci_id)
exit(1)
print "Adding PCI device to Contrail VM: %s" %(vm_name)
deviceId = hex(pci_entry.pciDevice.deviceId % 2**16).lstrip('0x')
backing = vim.VirtualPCIPassthroughDeviceBackingInfo(deviceId=deviceId,
id=pci_entry.pciDevice.id,
systemId=pci_entry.systemId,
vendorId=pci_entry.pciDevice.vendorId,
deviceName=pci_entry.pciDevice.deviceName)
hba_object = vim.VirtualPCIPassthrough(key=-100, backing=backing)
new_device_config = vim.VirtualDeviceConfigSpec(device=hba_object)
new_device_config.operation = "add"
new_device_config.device.connectable = vim.vm.device.VirtualDevice.ConnectInfo()
new_device_config.device.connectable.startConnected = True
device_config_list.append(new_device_config)
vm_spec=vim.vm.ConfigSpec()
vm_spec.deviceChange=device_config_list
task=vm.ReconfigVM_Task(spec=vm_spec)
wait_for_task(task)
def add_sriov_nics(args, vm, si_content):
dvs = get_obj(si_content, [vim.DistributedVirtualSwitch], args.sriov_dvs)
fix_sriov_pg(si_content, dvs, args.sriov_dvs_pg)
sr_iov_nic_list = args.sriov_nics
if vm.runtime.powerState == vim.VirtualMachinePowerState.poweredOn:
print "VM:%s is powered ON. Cannot do hot pci add now. Shutting it down" %(args.vm_name)
poweroff(vm)
# get pci id of the sriov nic
ssh_handle = paramiko.SSHClient()
ssh_handle.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh_handle.connect(args.esxi_host,
username = args.esxi_user,
password = args.esxi_password)
cmd = "vmware -v"
stdin, stdout, stderr = ssh_handle.exec_command(cmd)
err = stderr.read()
op = stdout.read()
if err:
self.log_and_raise_exception(err)
esxi_version = op.split()[2][:3]
for sr_iov_nic in sr_iov_nic_list:
cmd = "vmkchdev -l | grep %s" %sr_iov_nic
stdin, stdout, stderr = ssh_handle.exec_command(cmd)
err = stderr.read()
op = stdout.read()
if err:
self.log_and_raise_exception(err)
nic_info = str(op)
if len(nic_info) == 0:
raise Exception("Unable to add sriov interface for physical nic %s \
on esxi host %s" %(sr_iov_nic, args.esxi_host))
pci_id = nic_info.split()[0]
if (esxi_version == '5.5'):
pci_id = pci_id[5:]
mac_address = None
devices = []
nicspec = vim.vm.device.VirtualDeviceSpec()
nicspec.device = vim.vm.device.VirtualSriovEthernetCard()
nicspec.operation = vim.vm.device.VirtualDeviceSpec.Operation.add
nicspec.device.wakeOnLanEnabled = True
nicspec.device.allowGuestOSMtuChange = True
nicspec.device.deviceInfo = vim.Description()
pg_obj = get_obj([vim.dvs.DistributedVirtualPortgroup], args.sriov_dvs_pg)
dvs_port_connection = vim.dvs.PortConnection()
dvs_port_connection.portgroupKey = pg_obj.key
dvs_port_connection.switchUuid = pg_obj.config.distributedVirtualSwitch.uuid
nicspec.device.backing = vim.vm.device.VirtualEthernetCard.DistributedVirtualPortBackingInfo()
nicspec.device.backing.port = dvs_port_connection
nicspec.device.sriovBacking = vim.vm.device.VirtualSriovEthernetCard.SriovBackingInfo()
nicspec.device.sriovBacking.physicalFunctionBacking = vim.vm.device.VirtualPCIPassthrough.DeviceBackingInfo()
nicspec.device.sriovBacking.physicalFunctionBacking.id = pci_id
if (mac_address):
nicspec.device.addressType = "Manual"
nicspec.device.macAddress = mac_address
devices.append(nicspec)
vmconf = vim.vm.ConfigSpec(deviceChange=devices)
task = vm.ReconfigVM_Task(vmconf)
wait_for_task(task)
if not mac_address:
for device in vm.config.hardware.device:
if isinstance(device, vim.vm.device.VirtualSriovEthernetCard):
devices = []
mac_address = device.macAddress
nicspec = vim.vm.device.VirtualDeviceSpec()
nicspec.operation = vim.vm.device.VirtualDeviceSpec.Operation.edit
nicspec.device = device
nicspec.device.addressType = "Manual"
nicspec.device.macAddress = mac_address
devices.append(nicspec)
vmconf = vim.vm.ConfigSpec(deviceChange=devices)
task = vm.ReconfigVM_Task(vmconf)
wait_for_task(task)
def poweroffvm(vm_obj):
task = vm_obj.PowerOff()
wait_for_task(task)
def poweroffvm(vm_obj):
task = vm_obj.PowerOff()
wait_for_task(task)
def add_sriov_nics(args, vm, si_content):
dvs = get_obj(si_content, [vim.DistributedVirtualSwitch], args.sriov_dvs)
fix_sriov_pg(si_content, dvs, args.sriov_dvs_pg)
sr_iov_nic_list = args.sriov_nics
if vm.runtime.powerState == vim.VirtualMachinePowerState.poweredOn:
print "VM:%s is powered ON. Cannot do hot pci add now. Shutting it down" % (
args.vm_name)
poweroff(vm)
# get pci id of the sriov nic
ssh_handle = paramiko.SSHClient()
ssh_handle.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh_handle.connect(args.esxi_host,
username=args.esxi_user,
password=args.esxi_password)
cmd = "vmware -v"
stdin, stdout, stderr = ssh_handle.exec_command(cmd)
err = stderr.read()
op = stdout.read()
if err:
self.log_and_raise_exception(err)
esxi_version = op.split()[2][:3]
for sr_iov_nic in sr_iov_nic_list:
cmd = "vmkchdev -l | grep %s" % sr_iov_nic
stdin, stdout, stderr = ssh_handle.exec_command(cmd)
err = stderr.read()
op = stdout.read()
if err:
self.log_and_raise_exception(err)
nic_info = str(op)
if len(nic_info) == 0:
raise Exception(
"Unable to add sriov interface for physical nic %s \
on esxi host %s" % (sr_iov_nic, args.esxi_host))
pci_id = nic_info.split()[0]
if (esxi_version == '5.5'):
pci_id = pci_id[5:]
mac_address = None
devices = []
nicspec = vim.vm.device.VirtualDeviceSpec()
nicspec.device = vim.vm.device.VirtualSriovEthernetCard()
nicspec.operation = vim.vm.device.VirtualDeviceSpec.Operation.add
nicspec.device.wakeOnLanEnabled = True
nicspec.device.allowGuestOSMtuChange = True
nicspec.device.deviceInfo = vim.Description()
pg_obj = get_obj([vim.dvs.DistributedVirtualPortgroup],
args.sriov_dvs_pg)
dvs_port_connection = vim.dvs.PortConnection()
dvs_port_connection.portgroupKey = pg_obj.key
dvs_port_connection.switchUuid = pg_obj.config.distributedVirtualSwitch.uuid
nicspec.device.backing = vim.vm.device.VirtualEthernetCard.DistributedVirtualPortBackingInfo(
)
nicspec.device.backing.port = dvs_port_connection
nicspec.device.sriovBacking = vim.vm.device.VirtualSriovEthernetCard.SriovBackingInfo(
)
nicspec.device.sriovBacking.physicalFunctionBacking = vim.vm.device.VirtualPCIPassthrough.DeviceBackingInfo(
)
nicspec.device.sriovBacking.physicalFunctionBacking.id = pci_id
if (mac_address):
nicspec.device.addressType = "Manual"
nicspec.device.macAddress = mac_address
devices.append(nicspec)
vmconf = vim.vm.ConfigSpec(deviceChange=devices)
task = vm.ReconfigVM_Task(vmconf)
wait_for_task(task)
if not mac_address:
for device in vm.config.hardware.device:
if isinstance(device, vim.vm.device.VirtualSriovEthernetCard):
devices = []
mac_address = device.macAddress
nicspec = vim.vm.device.VirtualDeviceSpec()
nicspec.operation = vim.vm.device.VirtualDeviceSpec.Operation.edit
nicspec.device = device
nicspec.device.addressType = "Manual"
nicspec.device.macAddress = mac_address
devices.append(nicspec)
vmconf = vim.vm.ConfigSpec(deviceChange=devices)
task = vm.ReconfigVM_Task(vmconf)
wait_for_task(task)