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

mem-phys-addr.py (2446B)

      1# mem-phys-addr.py: Resolve physical address samples
      2# SPDX-License-Identifier: GPL-2.0
      3#
      4# Copyright (c) 2018, Intel Corporation.
      5
      6from __future__ import division
      7from __future__ import print_function
      8
      9import os
     10import sys
     11import struct
     12import re
     13import bisect
     14import collections
     15
     16sys.path.append(os.environ['PERF_EXEC_PATH'] + \
     17	'/scripts/python/Perf-Trace-Util/lib/Perf/Trace')
     18
     19#physical address ranges for System RAM
     20system_ram = []
     21#physical address ranges for Persistent Memory
     22pmem = []
     23#file object for proc iomem
     24f = None
     25#Count for each type of memory
     26load_mem_type_cnt = collections.Counter()
     27#perf event name
     28event_name = None
     29
     30def parse_iomem():
     31	global f
     32	f = open('/proc/iomem', 'r')
     33	for i, j in enumerate(f):
     34		m = re.split('-|:',j,2)
     35		if m[2].strip() == 'System RAM':
     36			system_ram.append(int(m[0], 16))
     37			system_ram.append(int(m[1], 16))
     38		if m[2].strip() == 'Persistent Memory':
     39			pmem.append(int(m[0], 16))
     40			pmem.append(int(m[1], 16))
     41
     42def print_memory_type():
     43	print("Event: %s" % (event_name))
     44	print("%-40s  %10s  %10s\n" % ("Memory type", "count", "percentage"), end='')
     45	print("%-40s  %10s  %10s\n" % ("----------------------------------------",
     46					"-----------", "-----------"),
     47					end='');
     48	total = sum(load_mem_type_cnt.values())
     49	for mem_type, count in sorted(load_mem_type_cnt.most_common(), \
     50					key = lambda kv: (kv[1], kv[0]), reverse = True):
     51		print("%-40s  %10d  %10.1f%%\n" %
     52			(mem_type, count, 100 * count / total),
     53			end='')
     54
     55def trace_begin():
     56	parse_iomem()
     57
     58def trace_end():
     59	print_memory_type()
     60	f.close()
     61
     62def is_system_ram(phys_addr):
     63	#/proc/iomem is sorted
     64	position = bisect.bisect(system_ram, phys_addr)
     65	if position % 2 == 0:
     66		return False
     67	return True
     68
     69def is_persistent_mem(phys_addr):
     70	position = bisect.bisect(pmem, phys_addr)
     71	if position % 2 == 0:
     72		return False
     73	return True
     74
     75def find_memory_type(phys_addr):
     76	if phys_addr == 0:
     77		return "N/A"
     78	if is_system_ram(phys_addr):
     79		return "System RAM"
     80
     81	if is_persistent_mem(phys_addr):
     82		return "Persistent Memory"
     83
     84	#slow path, search all
     85	f.seek(0, 0)
     86	for j in f:
     87		m = re.split('-|:',j,2)
     88		if int(m[0], 16) <= phys_addr <= int(m[1], 16):
     89			return m[2]
     90	return "N/A"
     91
     92def process_event(param_dict):
     93	name       = param_dict["ev_name"]
     94	sample     = param_dict["sample"]
     95	phys_addr  = sample["phys_addr"]
     96
     97	global event_name
     98	if event_name == None:
     99		event_name = name
    100	load_mem_type_cnt[find_memory_type(phys_addr)] += 1