import random
import re
import sys
+import typing as T
from dataclasses import dataclass
__all__ = [ 'KconfigDataError', 'KconfigParserError',
'KconfigData', 'KconfigParser' ,
'defconfig', 'allyesconfig', 'allnoconfig', 'randconfig' ]
+Mangler = T.Callable[[bool], bool]
+
@dataclass
class IncludeInfo:
file: str
line: int
parent: IncludeInfo | None
- def __iter__(self):
- inf = self
+ def __iter__(self) -> T.Iterator[str]:
+ inf: IncludeInfo | None = self
while inf is not None:
yield "%s:%d" % (inf.file, inf.line)
inf = inf.parent
- def error_path(self):
+ def error_path(self) -> str:
res = ""
for loc in self:
res = "In file included from %s:\n" % loc + res
return res
-def debug_print(*args):
+def debug_print(*args: object) -> None:
#print('# ' + (' '.join(str(x) for x in args)))
pass
# -------------------------------------------
class KconfigDataError(Exception):
- def __init__(self, msg):
+ def __init__(self, msg: str) -> None:
self.msg = msg
- def __str__(self):
+ def __str__(self) -> str:
return self.msg
-allyesconfig = lambda x: True
-allnoconfig = lambda x: False
-defconfig = lambda x: x
-randconfig = lambda x: random.randint(0, 1) == 1
+allyesconfig: Mangler = lambda x: True
+allnoconfig: Mangler = lambda x: False
+defconfig: Mangler = lambda x: x
+randconfig: Mangler = lambda x: random.randint(0, 1) == 1
class KconfigData:
class Expr:
- def __and__(self, rhs):
+ def __and__(self, rhs: KconfigData.Expr) -> KconfigData.Expr:
return KconfigData.AND(self, rhs)
- def __or__(self, rhs):
+ def __or__(self, rhs: KconfigData.Expr) -> KconfigData.Expr:
return KconfigData.OR(self, rhs)
- def __invert__(self):
+ def __invert__(self) -> KconfigData.Expr:
return KconfigData.NOT(self)
# Abstract methods
- def add_edges_to(self, var):
+ def add_edges_to(self, var: KconfigData.Var) -> None:
pass
- def evaluate(self):
+ def evaluate(self) -> bool:
assert False
class AND(Expr):
- def __init__(self, lhs, rhs):
+ def __init__(self, lhs: KconfigData.Expr, rhs: KconfigData.Expr) -> None:
self.lhs = lhs
self.rhs = rhs
- def __str__(self):
+ def __str__(self) -> str:
return "(%s && %s)" % (self.lhs, self.rhs)
- def add_edges_to(self, var):
+ def add_edges_to(self, var: KconfigData.Var) -> None:
self.lhs.add_edges_to(var)
self.rhs.add_edges_to(var)
- def evaluate(self):
+ def evaluate(self) -> bool:
return self.lhs.evaluate() and self.rhs.evaluate()
class OR(Expr):
- def __init__(self, lhs, rhs):
+ def __init__(self, lhs: KconfigData.Expr, rhs: KconfigData.Expr) -> None:
self.lhs = lhs
self.rhs = rhs
- def __str__(self):
+ def __str__(self) -> str:
return "(%s || %s)" % (self.lhs, self.rhs)
- def add_edges_to(self, var):
+ def add_edges_to(self, var: KconfigData.Var) -> None:
self.lhs.add_edges_to(var)
self.rhs.add_edges_to(var)
- def evaluate(self):
+ def evaluate(self) -> bool:
return self.lhs.evaluate() or self.rhs.evaluate()
class NOT(Expr):
- def __init__(self, lhs):
+ def __init__(self, lhs: KconfigData.Expr) -> None:
self.lhs = lhs
- def __str__(self):
+ def __str__(self) -> str:
return "!%s" % (self.lhs)
- def add_edges_to(self, var):
+ def add_edges_to(self, var: KconfigData.Var) -> None:
self.lhs.add_edges_to(var)
- def evaluate(self):
+ def evaluate(self) -> bool:
return not self.lhs.evaluate()
class Var(Expr):
- def __init__(self, name):
+ def __init__(self, name: str) -> None:
self.name = name
- self.value = None
- self.outgoing = set()
- self.clauses_for_var = list()
- def __str__(self):
+ self.value: bool | None = None
+ self.outgoing: set[KconfigData.Var] = set()
+ self.clauses_for_var: list[KconfigData.Clause] = []
+ def __str__(self) -> str:
return self.name
- def has_value(self):
+ def has_value(self) -> bool:
return self.value is not None
- def set_value(self, val, clause):
+ def set_value(self, val: bool, clause: KconfigData.Clause) -> None:
self.clauses_for_var.append(clause)
if self.has_value() and self.value != val:
print("The following clauses were found for " + self.name, file=sys.stderr)
self.value = val
# depth first search of the dependency graph
- def dfs(self, visited, f):
+ def dfs(self, visited: set[KconfigData.Var],
+ f: T.Callable[[KconfigData.Var], None]) -> None:
if self in visited:
return
visited.add(self)
v.dfs(visited, f)
f(self)
- def add_edges_to(self, var):
+ def add_edges_to(self, var: KconfigData.Var) -> None:
self.outgoing.add(var)
- def evaluate(self):
+ def evaluate(self) -> bool:
if not self.has_value():
raise KconfigDataError('cycle found including %s' % self)
+ assert self.value is not None
return self.value
class Clause:
- def __init__(self, dest):
+ def __init__(self, dest: KconfigData.Var) -> None:
self.dest = dest
- def priority(self):
+ def priority(self) -> int:
return 0
- def process(self):
+ def process(self) -> None:
pass
class AssignmentClause(Clause):
- def __init__(self, dest, value):
+ def __init__(self, dest: KconfigData.Var, value: bool) -> None:
KconfigData.Clause.__init__(self, dest)
self.value = value
- def __str__(self):
+ def __str__(self) -> str:
return "CONFIG_%s=%s" % (self.dest, 'y' if self.value else 'n')
- def process(self):
+ def process(self) -> None:
self.dest.set_value(self.value, self)
class DefaultClause(Clause):
- def __init__(self, dest, value, cond=None):
+ def __init__(self, dest: KconfigData.Var, value: bool,
+ cond: KconfigData.Expr | None = None) -> None:
KconfigData.Clause.__init__(self, dest)
self.value = value
self.cond = cond
if self.cond is not None:
self.cond.add_edges_to(self.dest)
- def __str__(self):
+ def __str__(self) -> str:
value = 'y' if self.value else 'n'
if self.cond is None:
return "config %s default %s" % (self.dest, value)
else:
return "config %s default %s if %s" % (self.dest, value, self.cond)
- def priority(self):
+ def priority(self) -> int:
# Defaults are processed just before leaving the variable
return -1
- def process(self):
+ def process(self) -> None:
if not self.dest.has_value() and \
(self.cond is None or self.cond.evaluate()):
self.dest.set_value(self.value, self)
class DependsOnClause(Clause):
- def __init__(self, dest, expr):
+ def __init__(self, dest: KconfigData.Var, expr: KconfigData.Expr) -> None:
KconfigData.Clause.__init__(self, dest)
self.expr = expr
self.expr.add_edges_to(self.dest)
- def __str__(self):
+ def __str__(self) -> str:
return "config %s depends on %s" % (self.dest, self.expr)
- def process(self):
+ def process(self) -> None:
if not self.expr.evaluate():
self.dest.set_value(False, self)
class SelectClause(Clause):
- def __init__(self, dest, cond):
+ def __init__(self, dest: KconfigData.Var, cond: KconfigData.Expr) -> None:
KconfigData.Clause.__init__(self, dest)
self.cond = cond
self.cond.add_edges_to(self.dest)
- def __str__(self):
+ def __str__(self) -> str:
return "select %s if %s" % (self.dest, self.cond)
- def process(self):
+ def process(self) -> None:
if self.cond.evaluate():
self.dest.set_value(True, self)
- def __init__(self, value_mangler=defconfig):
+ def __init__(self, value_mangler: Mangler = defconfig) -> None:
self.value_mangler = value_mangler
- self.previously_included = []
- self.defined_vars = set()
- self.referenced_vars = dict()
- self.clauses = list()
+ self.previously_included: list[str] = []
+ self.defined_vars: set[str] = set()
+ self.referenced_vars: dict[str, KconfigData.Var] = {}
+ self.clauses: list[KconfigData.Clause] = []
# semantic analysis -------------
- def check_undefined(self):
+ def check_undefined(self) -> bool:
undef = False
for i in self.referenced_vars:
if i not in self.defined_vars:
undef = True
return undef
- def compute_config(self):
+ def compute_config(self) -> dict[str, bool]:
if self.check_undefined():
raise KconfigDataError("there were undefined symbols")
debug_print(source, "->", [str(x) for x in edges.outgoing])
# The reverse of the depth-first order is the topological sort
- dfo = dict()
- visited = set()
+ dfo: dict[KconfigData.Var, int] = {}
+ visited: set[KconfigData.Var] = set()
debug_print("\n")
- def visit_fn(var):
+ def visit_fn(var: KconfigData.Var) -> None:
debug_print(var, "has DFS number", len(dfo))
dfo[var] = len(dfo)
clause.process()
debug_print("")
- values = dict()
+ values: dict[str, bool] = {}
for name, v in self.referenced_vars.items():
debug_print("Evaluating", name)
values[name] = v.evaluate()
# semantic actions -------------
- def do_declaration(self, var):
+ def do_declaration(self, var: KconfigData.Var) -> None:
if var.name in self.defined_vars:
raise KconfigDataError('variable "%s" defined twice' % var.name)
-
self.defined_vars.add(var.name)
# var is a string with the variable's name.
- def do_var(self, var):
+ def do_var(self, var: str) -> KconfigData.Var:
if var in self.referenced_vars:
return self.referenced_vars[var]
var_obj = self.referenced_vars[var] = KconfigData.Var(var)
return var_obj
- def do_assignment(self, var, val):
+ def do_assignment(self, var: KconfigData.Var, val: bool) -> None:
self.clauses.append(KconfigData.AssignmentClause(var, val))
- def do_cmdline_assignment(self, var, val):
+ def do_cmdline_assignment(self, var: str, val: bool) -> None:
assert var.startswith("CONFIG_")
self.do_assignment(self.do_var(var[7:]), val)
- def do_default(self, var, val, cond=None):
+ def do_default(self, var: KconfigData.Var, val: bool,
+ cond: KconfigData.Expr | None = None) -> None:
val = self.value_mangler(val)
self.clauses.append(KconfigData.DefaultClause(var, val, cond))
- def do_depends_on(self, var, expr):
+ def do_depends_on(self, var: KconfigData.Var,
+ expr: KconfigData.Expr) -> None:
self.clauses.append(KconfigData.DependsOnClause(var, expr))
- def do_select(self, var, symbol, cond=None):
+ def do_select(self, var: KconfigData.Var, symbol: KconfigData.Var,
+ cond: KconfigData.Expr | None = None) -> None:
cond = (cond & var) if cond is not None else var
self.clauses.append(KconfigData.SelectClause(symbol, cond))
- def do_imply(self, var, symbol, cond=None):
+ def do_imply(self, var: KconfigData.Var, symbol: KconfigData.Var,
+ cond: KconfigData.Expr | None = None) -> None:
# "config X imply Y [if COND]" is the same as
# "config Y default y if X [&& COND]"
cond = (cond & var) if cond is not None else var
# -------------------------------------------
# tokens table
-TOKENS = {}
+TOKENS: dict[int, str] = {}
TOK_NONE = -1
TOK_LPAREN = 0; TOKENS[TOK_LPAREN] = '"("'
TOK_RPAREN = 1; TOKENS[TOK_RPAREN] = '")"'
TOK_EOF = 18; TOKENS[TOK_EOF] = 'end of file'
class KconfigParserError(Exception):
- def __init__(self, parser, msg, tok=None):
+ def __init__(self, parser: KconfigParser, msg: str,
+ tok: int | str | None = None) -> None:
self.loc = parser.location()
tok = tok if tok is not None else parser.tok
if tok != TOK_NONE:
msg = '%s before %s' % (msg, location)
self.msg = msg
- def __str__(self):
+ def __str__(self) -> str:
return "%s: %s" % (self.loc, self.msg)
class KconfigParser:
@classmethod
- def parse(cls, fp, data, incl_info=None):
+ def parse(cls, fp: T.TextIO, data: KconfigData, incl_info: IncludeInfo | None = None) -> None:
cls(fp, data, incl_info).parse_config()
- def __init__(self, fp, data, incl_info):
+ def __init__(self, fp: T.TextIO, data: KconfigData, incl_info: IncludeInfo | None = None):
self.data = data
self.incl_info = incl_info
self.abs_fname = os.path.abspath(fp.name)
self.fname = fp.name
self.data.previously_included.append(self.abs_fname)
+
src = fp.read()
if src == '' or src[-1] != '\n':
src += '\n'
self.src = src
- self.cursor = 0
- self.line = 1
- self.line_pos = 0
+ self.cursor: int = 0
+ self.line: int = 1
+ self.line_pos: int = 0
+ self.pos: int = 0
+ self.tok: int = TOK_NONE
+ self.val: str | None = None
self.get_token()
# file management -----
- def location(self):
+ def location(self) -> str:
col = 1
for ch in self.src[self.line_pos:self.pos]:
if ch == '\t':
incl_chain = inf.error_path() if inf is not None else ""
return '%s%s:%d:%d' % (incl_chain, self.fname, self.line, col)
- def do_include(self, include):
+ def do_include(self, include: str) -> None:
incl_abs_fname = os.path.join(os.path.dirname(self.abs_fname),
include)
# catch inclusion cycle
# recursive descent parser -----
# y_or_n: Y | N
- def parse_y_or_n(self):
+ def parse_y_or_n(self) -> bool:
if self.tok == TOK_Y:
self.get_token()
return True
raise KconfigParserError(self, 'Expected "y" or "n"')
# var: ID
- def parse_var(self):
+ def parse_var(self) -> KconfigData.Var:
if self.tok != TOK_ID:
raise KconfigParserError(self, 'Expected identifier')
val = self.val
return self.data.do_var(val)
# assignment_var: ID (starting with "CONFIG_")
- def parse_assignment_var(self):
+ def parse_assignment_var(self) -> KconfigData.Var:
if self.tok != TOK_ID:
raise KconfigParserError(self, 'Expected identifier')
val = self.val
return self.data.do_var(val[7:])
# assignment: var EQUAL y_or_n
- def parse_assignment(self):
+ def parse_assignment(self) -> None:
var = self.parse_assignment_var()
if self.tok != TOK_EQUAL:
raise KconfigParserError(self, 'Expected "="')
# primary: NOT primary
# | LPAREN expr RPAREN
# | var
- def parse_primary(self):
+ def parse_primary(self) -> KconfigData.Expr:
if self.tok == TOK_NOT:
self.get_token()
val = ~self.parse_primary()
return val
# disj: primary (OR primary)*
- def parse_disj(self):
+ def parse_disj(self) -> KconfigData.Expr:
lhs = self.parse_primary()
while self.tok == TOK_OR:
self.get_token()
return lhs
# expr: disj (AND disj)*
- def parse_expr(self):
+ def parse_expr(self) -> KconfigData.Expr:
lhs = self.parse_disj()
while self.tok == TOK_AND:
self.get_token()
# condition: IF expr
# | empty
- def parse_condition(self):
+ def parse_condition(self) -> KconfigData.Expr | None:
if self.tok != TOK_IF:
return None
self.get_token()
# | DEPENDS ON expr
# | SELECT var condition
# | BOOL
- def parse_property(self, var):
+ def parse_property(self, var: KconfigData.Var) -> None:
if self.tok == TOK_DEFAULT:
self.get_token()
val = self.parse_y_or_n()
# properties: properties property
# | /* empty */
- def parse_properties(self, var):
+ def parse_properties(self, var: KconfigData.Var) -> None:
while self.tok == TOK_DEFAULT or self.tok == TOK_DEPENDS or \
self.tok == TOK_SELECT or self.tok == TOK_BOOL or \
self.tok == TOK_IMPLY:
+ '"default", "depends on", "imply" or "select"')
# declaration: config var properties
- def parse_declaration(self):
+ def parse_declaration(self) -> None:
if self.tok == TOK_CONFIG:
self.get_token()
var = self.parse_var()
# clause: SOURCE
# | declaration
# | assignment
- def parse_clause(self):
+ def parse_clause(self) -> None:
if self.tok == TOK_SOURCE:
val = self.val
+ assert val is not None
self.get_token()
self.do_include(val)
elif self.tok == TOK_CONFIG:
raise KconfigParserError(self, 'expected "source", "config" or identifier')
# config: clause+ EOF
- def parse_config(self):
+ def parse_config(self) -> KconfigData:
while self.tok != TOK_EOF:
self.parse_clause()
return self.data
# scanner -----
- def get_token(self):
+ def get_token(self) -> None:
+ assert self.src is not None
while True:
ch = self.src[self.cursor]
self.pos = self.cursor
self.tok = tok
return
- def check_keyword(self, rest):
+ def check_keyword(self, rest: str) -> bool:
+ assert self.src is not None
if not self.src.startswith(rest, self.cursor):
return False
length = len(rest)
self.cursor += length
return True
- def scan_token(self, ch):
+ def scan_token(self, ch: str) -> int | None:
+ assert self.src is not None
if ch == '#':
self.cursor = self.src.find('\n', self.cursor)
return None
def main() -> None:
argv = sys.argv
- mode = defconfig
+ mode: Mangler = defconfig
if len(sys.argv) > 1:
if argv[1] == '--defconfig':
del argv[1]
sys.exit(1)
data = KconfigData(mode)
- external_vars = set()
+ external_vars: set[str] = set()
for arg in argv[3:]:
m = re.match(r'^(CONFIG_[A-Z0-9_]+)=([yn]?)$', arg)
if m is not None: