drgn/util.py
Omar Sandoval 3b6c02107e vmtest: get SYS_kexec_file_load from util.SYS
This still isn't enough to be fully generic since since some
architectures don't support kexec_file_load, especially on older
kernels, and some architectures can't use the vmlinuz directly (e.g.,
AArch64 may need it to be decompressed). But, this is at least cleaner
than hard-coding the number for x86-64.

Signed-off-by: Omar Sandoval <osandov@osandov.com>
2022-10-11 16:08:20 -07:00

244 lines
7.7 KiB
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: GPL-3.0-or-later
from functools import total_ordering
import os
from pathlib import Path
import platform
import re
import sys
from typing import Union
def nproc() -> int:
return len(os.sched_getaffinity(0))
def out_of_date(path: Union[str, Path], *deps: Union[str, Path]) -> bool:
try:
mtime = os.stat(path).st_mtime
except FileNotFoundError:
return True
return any(os.stat(dep).st_mtime > mtime for dep in deps)
def _c_isdigit(c: int) -> bool:
# '0' <= c <= '9'
return 0x30 <= c <= 0x39
def _c_isalpha(c: int) -> bool:
# ('A' <= c <= 'Z') or ('a' <= c <= 'z')
return (0x41 <= c <= 0x5A) or (0x61 <= c <= 0x7A)
def _order(c: int) -> int:
if _c_isdigit(c):
return 0
elif _c_isalpha(c):
return c
elif c == 0x7E: # '~'
return -1
else:
return c + 0x100
def verrevcmp(v1: str, v2: str) -> int:
"""
Compare two versions according to the coreutils version sort rules
(https://www.gnu.org/software/coreutils/manual/html_node/Version_002dsort-ordering-rules.html).
Returns 0 if v1 == v2 by this definition, < 0 if v1 < v2, and > 0 if v1 >
v2.
Adapted from
https://git.savannah.gnu.org/cgit/gnulib.git/tree/lib/filevercmp.c.
"""
# By definition, version sort compares ASCII, not Unicode:
# https://www.gnu.org/software/coreutils/manual/html_node/Version-sort-ignores-locale.html.
s1 = bytearray(v1, "utf-8")
s2 = bytearray(v2, "utf-8")
s1_len = len(s1)
s2_len = len(s2)
# Add sentinels to avoid some length checks.
s1.append(0)
s2.append(0)
s1_pos = s2_pos = 0
while s1_pos < s1_len or s2_pos < s2_len:
while (s1_pos < s1_len and not _c_isdigit(s1[s1_pos])) or (
s2_pos < s2_len and not _c_isdigit(s2[s2_pos])
):
s1_c = _order(s1[s1_pos]) if s1_pos < s1_len else 0
s2_c = _order(s2[s2_pos]) if s2_pos < s2_len else 0
if s1_c != s2_c:
return s1_c - s2_c
s1_pos += 1
s2_pos += 1
while s1[s1_pos] == 0x30: # '0'
s1_pos += 1
while s2[s2_pos] == 0x30: # '0'
s2_pos += 1
first_diff = 0
while _c_isdigit(s1[s1_pos]) and _c_isdigit(s2[s2_pos]):
if not first_diff:
first_diff = s1[s1_pos] - s2[s2_pos]
s1_pos += 1
s2_pos += 1
if _c_isdigit(s1[s1_pos]):
return 1
if _c_isdigit(s2[s2_pos]):
return -1
if first_diff:
return first_diff
return 0
@total_ordering
class KernelVersion:
"""
Version ordered by verrevcmp(), with -rc releases before the final release.
"""
def __init__(self, release: str) -> None:
self._release = release
# ~ sorts before anything, including the end of the version.
self._key = re.sub(r"-(rc[0-9])", r"~\1", release)
def __eq__(self, other: object) -> bool:
if not isinstance(other, KernelVersion):
return NotImplemented
return self._key == other._key
def __lt__(self, other: object) -> bool:
if not isinstance(other, KernelVersion):
return NotImplemented
return verrevcmp(self._key, other._key) < 0
def __str__(self) -> str:
return self._release
NORMALIZED_MACHINE_NAME = platform.machine()
if NORMALIZED_MACHINE_NAME.startswith("aarch64") or NORMALIZED_MACHINE_NAME == "arm64":
NORMALIZED_MACHINE_NAME = "aarch64"
elif NORMALIZED_MACHINE_NAME.startswith("arm") or NORMALIZED_MACHINE_NAME == "sa110":
NORMALIZED_MACHINE_NAME = "arm"
elif re.fullmatch(r"i.86", NORMALIZED_MACHINE_NAME):
NORMALIZED_MACHINE_NAME = "i386"
elif NORMALIZED_MACHINE_NAME.startswith("ppc64"):
NORMALIZED_MACHINE_NAME = "ppc64"
elif NORMALIZED_MACHINE_NAME.startswith("ppc"):
NORMALIZED_MACHINE_NAME = "ppc"
elif NORMALIZED_MACHINE_NAME == "riscv":
NORMALIZED_MACHINE_NAME = "riscv32"
elif re.match(r"sh[0-9]", NORMALIZED_MACHINE_NAME):
NORMALIZED_MACHINE_NAME = "sh"
elif NORMALIZED_MACHINE_NAME == "sun4u":
NORMALIZED_MACHINE_NAME = "sparc64"
if NORMALIZED_MACHINE_NAME == "x86_64":
if sys.maxsize > 2**32:
SYS = {"bpf": 321, "kexec_file_load": 320, "rt_sigtimedwait": 128}
else: # x32
SYS = {"bpf": 321, "kexec_file_load": 320, "rt_sigtimedwait": 523}
else:
SYS = {
"aarch64": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"alpha": {"bpf": 515, "rt_sigtimedwait": 355},
"arc": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"arm": {
"bpf": 386,
"kexec_file_load": 401,
"rt_sigtimedwait": 177,
"rt_sigtimedwait_time64": 421,
},
"csky": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"hexagon": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"i386": {"bpf": 357, "rt_sigtimedwait": 177, "rt_sigtimedwait_time64": 421},
"ia64": {"bpf": 317, "rt_sigtimedwait": 159},
"loongarch": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"loongarch64": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"m68k": {"bpf": 354, "rt_sigtimedwait": 177, "rt_sigtimedwait_time64": 421},
"microblaze": {
"bpf": 387,
"rt_sigtimedwait": 177,
"rt_sigtimedwait_time64": 421,
},
# TODO: mips is missing here because I don't know how to distinguish
# between the o32 and n32 ABIs.
"mips64": {"bpf": 315, "rt_sigtimedwait": 126},
"nios2": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"openrisc": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"parisc": {
"bpf": 341,
"kexec_file_load": 355,
"rt_sigtimedwait": 177,
"rt_sigtimedwait_time64": 421,
},
"parisc64": {"bpf": 341, "kexec_file_load": 355, "rt_sigtimedwait": 177},
"ppc": {"bpf": 361, "rt_sigtimedwait": 176, "rt_sigtimedwait_time64": 421},
"ppc64": {"bpf": 361, "rt_sigtimedwait": 176},
"riscv32": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"riscv64": {
"bpf": 280,
"kexec_file_load": 294,
"rt_sigtimedwait": 137,
"rt_sigtimedwait_time64": 421,
},
"s390": {
"bpf": 351,
"kexec_file_load": 381,
"rt_sigtimedwait": 177,
"rt_sigtimedwait_time64": 421,
},
"s390x": {"bpf": 351, "kexec_file_load": 381, "rt_sigtimedwait": 177},
"sh": {"bpf": 375, "rt_sigtimedwait": 177, "rt_sigtimedwait_time64": 421},
"sparc": {"bpf": 349, "rt_sigtimedwait": 105, "rt_sigtimedwait_time64": 421},
"sparc64": {"bpf": 349, "rt_sigtimedwait": 105},
"xtensa": {"bpf": 340, "rt_sigtimedwait": 229, "rt_sigtimedwait_time64": 421},
}.get(NORMALIZED_MACHINE_NAME, {})