1*502d84bdSDexuan Cui#!/usr/bin/env python 2*502d84bdSDexuan Cui 3*502d84bdSDexuan Cuiimport os 4*502d84bdSDexuan Cuifrom optparse import OptionParser 5*502d84bdSDexuan Cui 6*502d84bdSDexuan Cuiparser = OptionParser() 7*502d84bdSDexuan Cuiparser.add_option("-v", "--verbose", dest="verbose", 8*502d84bdSDexuan Cui help="print verbose messages. Try -vv, -vvv for \ 9*502d84bdSDexuan Cui more verbose messages", action="count") 10*502d84bdSDexuan Cui 11*502d84bdSDexuan Cui(options, args) = parser.parse_args() 12*502d84bdSDexuan Cui 13*502d84bdSDexuan Cuiverbose = 0 14*502d84bdSDexuan Cuiif options.verbose is not None: 15*502d84bdSDexuan Cui verbose = options.verbose 16*502d84bdSDexuan Cui 17*502d84bdSDexuan Cuivmbus_sys_path = '/sys/bus/vmbus/devices' 18*502d84bdSDexuan Cuiif not os.path.isdir(vmbus_sys_path): 19*502d84bdSDexuan Cui print "%s doesn't exist: exiting..." % vmbus_sys_path 20*502d84bdSDexuan Cui exit(-1) 21*502d84bdSDexuan Cui 22*502d84bdSDexuan Cuivmbus_dev_dict = { 23*502d84bdSDexuan Cui '{0e0b6031-5213-4934-818b-38d90ced39db}' : '[Operating system shutdown]', 24*502d84bdSDexuan Cui '{9527e630-d0ae-497b-adce-e80ab0175caf}' : '[Time Synchronization]', 25*502d84bdSDexuan Cui '{57164f39-9115-4e78-ab55-382f3bd5422d}' : '[Heartbeat]', 26*502d84bdSDexuan Cui '{a9a0f4e7-5a45-4d96-b827-8a841e8c03e6}' : '[Data Exchange]', 27*502d84bdSDexuan Cui '{35fa2e29-ea23-4236-96ae-3a6ebacba440}' : '[Backup (volume checkpoint)]', 28*502d84bdSDexuan Cui '{34d14be3-dee4-41c8-9ae7-6b174977c192}' : '[Guest services]', 29*502d84bdSDexuan Cui '{525074dc-8985-46e2-8057-a307dc18a502}' : '[Dynamic Memory]', 30*502d84bdSDexuan Cui '{cfa8b69e-5b4a-4cc0-b98b-8ba1a1f3f95a}' : 'Synthetic mouse', 31*502d84bdSDexuan Cui '{f912ad6d-2b17-48ea-bd65-f927a61c7684}' : 'Synthetic keyboard', 32*502d84bdSDexuan Cui '{da0a7802-e377-4aac-8e77-0558eb1073f8}' : 'Synthetic framebuffer adapter', 33*502d84bdSDexuan Cui '{f8615163-df3e-46c5-913f-f2d2f965ed0e}' : 'Synthetic network adapter', 34*502d84bdSDexuan Cui '{32412632-86cb-44a2-9b5c-50d1417354f5}' : 'Synthetic IDE Controller', 35*502d84bdSDexuan Cui '{ba6163d9-04a1-4d29-b605-72e2ffb1dc7f}' : 'Synthetic SCSI Controller', 36*502d84bdSDexuan Cui '{2f9bcc4a-0069-4af3-b76b-6fd0be528cda}' : 'Synthetic fiber channel adapter', 37*502d84bdSDexuan Cui '{8c2eaf3d-32a7-4b09-ab99-bd1f1c86b501}' : 'Synthetic RDMA adapter', 38*502d84bdSDexuan Cui '{276aacf4-ac15-426c-98dd-7521ad3f01fe}' : '[Reserved system device]', 39*502d84bdSDexuan Cui '{f8e65716-3cb3-4a06-9a60-1889c5cccab5}' : '[Reserved system device]', 40*502d84bdSDexuan Cui '{3375baf4-9e15-4b30-b765-67acb10d607b}' : '[Reserved system device]', 41*502d84bdSDexuan Cui} 42*502d84bdSDexuan Cui 43*502d84bdSDexuan Cuidef get_vmbus_dev_attr(dev_name, attr): 44*502d84bdSDexuan Cui try: 45*502d84bdSDexuan Cui f = open('%s/%s/%s' % (vmbus_sys_path, dev_name, attr), 'r') 46*502d84bdSDexuan Cui lines = f.readlines() 47*502d84bdSDexuan Cui f.close() 48*502d84bdSDexuan Cui except IOError: 49*502d84bdSDexuan Cui lines = [] 50*502d84bdSDexuan Cui 51*502d84bdSDexuan Cui return lines 52*502d84bdSDexuan Cui 53*502d84bdSDexuan Cuiclass VMBus_Dev: 54*502d84bdSDexuan Cui pass 55*502d84bdSDexuan Cui 56*502d84bdSDexuan Cui 57*502d84bdSDexuan Cuivmbus_dev_list = [] 58*502d84bdSDexuan Cui 59*502d84bdSDexuan Cuifor f in os.listdir(vmbus_sys_path): 60*502d84bdSDexuan Cui vmbus_id = get_vmbus_dev_attr(f, 'id')[0].strip() 61*502d84bdSDexuan Cui class_id = get_vmbus_dev_attr(f, 'class_id')[0].strip() 62*502d84bdSDexuan Cui device_id = get_vmbus_dev_attr(f, 'device_id')[0].strip() 63*502d84bdSDexuan Cui dev_desc = vmbus_dev_dict.get(class_id, 'Unknown') 64*502d84bdSDexuan Cui 65*502d84bdSDexuan Cui chn_vp_mapping = get_vmbus_dev_attr(f, 'channel_vp_mapping') 66*502d84bdSDexuan Cui chn_vp_mapping = [c.strip() for c in chn_vp_mapping] 67*502d84bdSDexuan Cui chn_vp_mapping = sorted(chn_vp_mapping, 68*502d84bdSDexuan Cui key = lambda c : int(c.split(':')[0])) 69*502d84bdSDexuan Cui 70*502d84bdSDexuan Cui chn_vp_mapping = ['\tRel_ID=%s, target_cpu=%s' % 71*502d84bdSDexuan Cui (c.split(':')[0], c.split(':')[1]) 72*502d84bdSDexuan Cui for c in chn_vp_mapping] 73*502d84bdSDexuan Cui d = VMBus_Dev() 74*502d84bdSDexuan Cui d.sysfs_path = '%s/%s' % (vmbus_sys_path, f) 75*502d84bdSDexuan Cui d.vmbus_id = vmbus_id 76*502d84bdSDexuan Cui d.class_id = class_id 77*502d84bdSDexuan Cui d.device_id = device_id 78*502d84bdSDexuan Cui d.dev_desc = dev_desc 79*502d84bdSDexuan Cui d.chn_vp_mapping = '\n'.join(chn_vp_mapping) 80*502d84bdSDexuan Cui if d.chn_vp_mapping: 81*502d84bdSDexuan Cui d.chn_vp_mapping += '\n' 82*502d84bdSDexuan Cui 83*502d84bdSDexuan Cui vmbus_dev_list.append(d) 84*502d84bdSDexuan Cui 85*502d84bdSDexuan Cui 86*502d84bdSDexuan Cuivmbus_dev_list = sorted(vmbus_dev_list, key = lambda d : int(d.vmbus_id)) 87*502d84bdSDexuan Cui 88*502d84bdSDexuan Cuiformat0 = '%2s: %s' 89*502d84bdSDexuan Cuiformat1 = '%2s: Class_ID = %s - %s\n%s' 90*502d84bdSDexuan Cuiformat2 = '%2s: Class_ID = %s - %s\n\tDevice_ID = %s\n\tSysfs path: %s\n%s' 91*502d84bdSDexuan Cui 92*502d84bdSDexuan Cuifor d in vmbus_dev_list: 93*502d84bdSDexuan Cui if verbose == 0: 94*502d84bdSDexuan Cui print ('VMBUS ID ' + format0) % (d.vmbus_id, d.dev_desc) 95*502d84bdSDexuan Cui elif verbose == 1: 96*502d84bdSDexuan Cui print ('VMBUS ID ' + format1) % \ 97*502d84bdSDexuan Cui (d.vmbus_id, d.class_id, d.dev_desc, d.chn_vp_mapping) 98*502d84bdSDexuan Cui else: 99*502d84bdSDexuan Cui print ('VMBUS ID ' + format2) % \ 100*502d84bdSDexuan Cui (d.vmbus_id, d.class_id, d.dev_desc, \ 101*502d84bdSDexuan Cui d.device_id, d.sysfs_path, d.chn_vp_mapping) 102