cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

lsvmbus (3555B)

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