From nobody Thu Oct 2 09:22:08 2025 Received: from us-smtp-delivery-124.mimecast.com (us-smtp-delivery-124.mimecast.com [170.10.129.124]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 99A8E313D42 for ; Fri, 19 Sep 2025 14:10:56 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=170.10.129.124 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1758291058; cv=none; b=nKrE+m2zBI8HEhMD/ybXYhvLmUQFAM3gHeuENYIivSsifWzppYWPivsUqVkRFewKrXpLuMFncAQdXgshVq5SaQFENBX5EEuZsNls+BDRiXlpeRX0mz3cmfIeG+2iiNbcvqNFcdwUbvPGmm7D48GDYdpA+mtthTtCQXLcFi0ZUoM= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1758291058; c=relaxed/simple; bh=MOmACpdek2E3u4tO+qZ+5RnTnZZM+p2GztFDoML/iUg=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=Pz0CUMVFfOTGVY7PB8SKvlRhBFE/n/AjvM1bPuDmHTSaVua6GYvKQumCUPHVLCeawRym3bho56a2kQBTJiTgMqWmS9rwXSpOHu5POlhsXEPpnHxwheHCqNis64hX1l2Uq/MSXG6wd+kjNC/XYiO4etgRn3wWf3Ckz71dKfKEpPM= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=quarantine dis=none) header.from=redhat.com; spf=pass smtp.mailfrom=redhat.com; dkim=pass (1024-bit key) header.d=redhat.com header.i=@redhat.com header.b=YGHxm9ai; arc=none smtp.client-ip=170.10.129.124 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=quarantine dis=none) header.from=redhat.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=redhat.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (1024-bit key) header.d=redhat.com header.i=@redhat.com header.b="YGHxm9ai" DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=redhat.com; s=mimecast20190719; t=1758291055; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=FdFDRKaens7ItTY1mAwlhagHLPXzGZoYgrUtrNXmcik=; b=YGHxm9aif2IGS4rg6trzzWj4Tumxf/Psld+Fa03CVLZao+lgrHA6Pg7Q0CY8GZn2niF6AU 2S4qzhwIWNmLt23seCXM1zrz+9wh7muHEYnhDRxzj9Z3itbaE/1WjFTDAvHwJ3gA2IQ+W1 Xq9WpWh4SkiLeShyYoqgl6fN71nJeEw= Received: from mx-prod-mc-08.mail-002.prod.us-west-2.aws.redhat.com (ec2-35-165-154-97.us-west-2.compute.amazonaws.com [35.165.154.97]) by relay.mimecast.com with ESMTP with STARTTLS (version=TLSv1.3, cipher=TLS_AES_256_GCM_SHA384) id us-mta-581-3k2PXnJbNBa4b1P-nmTa1A-1; Fri, 19 Sep 2025 10:10:54 -0400 X-MC-Unique: 3k2PXnJbNBa4b1P-nmTa1A-1 X-Mimecast-MFC-AGG-ID: 3k2PXnJbNBa4b1P-nmTa1A_1758291053 Received: from mx-prod-int-03.mail-002.prod.us-west-2.aws.redhat.com (mx-prod-int-03.mail-002.prod.us-west-2.aws.redhat.com [10.30.177.12]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits) server-digest SHA256) (No client certificate requested) by mx-prod-mc-08.mail-002.prod.us-west-2.aws.redhat.com (Postfix) with ESMTPS id 2E8D81830740; Fri, 19 Sep 2025 14:10:41 +0000 (UTC) Received: from gmonaco-thinkpadt14gen3.rmtit.csb (unknown [10.45.224.13]) by mx-prod-int-03.mail-002.prod.us-west-2.aws.redhat.com (Postfix) with ESMTP id A81801956048; Fri, 19 Sep 2025 14:10:35 +0000 (UTC) From: Gabriele Monaco To: linux-kernel@vger.kernel.org, Steven Rostedt , linux-trace-kernel@vger.kernel.org Cc: Gabriele Monaco , Nam Cao , Tomas Glozar , Juri Lelli , Clark Williams , John Kacur Subject: [PATCH v2 06/20] verification/rvgen: Annotate DA functions with types Date: Fri, 19 Sep 2025 16:09:40 +0200 Message-ID: <20250919140954.104920-7-gmonaco@redhat.com> In-Reply-To: <20250919140954.104920-1-gmonaco@redhat.com> References: <20250919140954.104920-1-gmonaco@redhat.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable X-Scanned-By: MIMEDefang 3.0 on 10.30.177.12 Content-Type: text/plain; charset="utf-8" Functions in automata.py, dot2c.py and dot2k.py don't have type annotations and it can get complicated to remember how to use them. Add minimal type annotations. Signed-off-by: Gabriele Monaco --- tools/verification/rvgen/rvgen/automata.py | 20 +++++------ tools/verification/rvgen/rvgen/dot2c.py | 42 +++++++++++----------- tools/verification/rvgen/rvgen/dot2k.py | 20 +++++------ 3 files changed, 41 insertions(+), 41 deletions(-) diff --git a/tools/verification/rvgen/rvgen/automata.py b/tools/verificatio= n/rvgen/rvgen/automata.py index d9a3fe2b74bf..3f06aef8d4fd 100644 --- a/tools/verification/rvgen/rvgen/automata.py +++ b/tools/verification/rvgen/rvgen/automata.py @@ -28,7 +28,7 @@ class Automata: self.function =3D self.__create_matrix() self.events_start, self.events_start_run =3D self.__store_init_eve= nts() =20 - def __get_model_name(self): + def __get_model_name(self) -> str: basename =3D ntpath.basename(self.__dot_path) if not basename.endswith(".dot") and not basename.endswith(".gv"): print("not a dot file") @@ -40,7 +40,7 @@ class Automata: =20 return model_name =20 - def __open_dot(self): + def __open_dot(self) -> list[str]: cursor =3D 0 dot_lines =3D [] try: @@ -60,13 +60,13 @@ class Automata: cursor +=3D 1 return dot_lines =20 - def __get_cursor_begin_states(self): + def __get_cursor_begin_states(self) -> int: cursor =3D 0 while self.__dot_lines[cursor].split()[0] !=3D "{node": cursor +=3D 1 return cursor =20 - def __get_cursor_begin_events(self): + def __get_cursor_begin_events(self) -> int: cursor =3D 0 while self.__dot_lines[cursor].split()[0] !=3D "{node": cursor +=3D 1 @@ -76,7 +76,7 @@ class Automata: cursor +=3D 1 return cursor =20 - def __get_state_variables(self): + def __get_state_variables(self) -> tuple[list[str], str, list[str]]: # wait for node declaration states =3D [] final_states =3D [] @@ -116,7 +116,7 @@ class Automata: =20 return states, initial_state, final_states =20 - def __get_event_variables(self): + def __get_event_variables(self) -> list[str]: # here we are at the begin of transitions, take a note, we will re= turn later. cursor =3D self.__get_cursor_begin_events() =20 @@ -140,7 +140,7 @@ class Automata: =20 return sorted(set(events)) =20 - def __create_matrix(self): + def __create_matrix(self) -> list[list[str]]: # transform the array into a dictionary events =3D self.events states =3D self.states @@ -174,7 +174,7 @@ class Automata: =20 return matrix =20 - def __store_init_events(self): + def __store_init_events(self) -> tuple[list[bool], list[bool]]: events_start =3D [False] * len(self.events) events_start_run =3D [False] * len(self.events) for i, _ in enumerate(self.events): @@ -196,10 +196,10 @@ class Automata: events_start_run[i] =3D True return events_start, events_start_run =20 - def is_start_event(self, event): + def is_start_event(self, event: str) -> bool: return self.events_start[self.events.index(event)] =20 - def is_start_run_event(self, event): + def is_start_run_event(self, event: str) -> bool: # prefer handle_start_event if there if any(self.events_start): return False diff --git a/tools/verification/rvgen/rvgen/dot2c.py b/tools/verification/r= vgen/rvgen/dot2c.py index b9b6f14cc536..fd64174fcfad 100644 --- a/tools/verification/rvgen/rvgen/dot2c.py +++ b/tools/verification/rvgen/rvgen/dot2c.py @@ -35,7 +35,7 @@ class Dot2c(Automata): # cut off the last \n return string[:-1] =20 - def __get_enum_states_content(self): + def __get_enum_states_content(self) -> list[str]: buff =3D [] buff.append("\t%s%s =3D 0," % (self.initial_state, self.enum_suffi= x)) for state in self.states: @@ -49,7 +49,7 @@ class Dot2c(Automata): buff =3D self.__get_enum_states_content() return self.__buff_to_string(buff) =20 - def format_states_enum(self): + def format_states_enum(self) -> list[str]: buff =3D [] buff.append("enum %s {" % self.enum_states_def) buff.append(self.get_enum_states_string()) @@ -57,7 +57,7 @@ class Dot2c(Automata): =20 return buff =20 - def __get_enum_events_content(self): + def __get_enum_events_content(self) -> list[str]: buff =3D [] first =3D True for event in self.events: @@ -75,7 +75,7 @@ class Dot2c(Automata): buff =3D self.__get_enum_events_content() return self.__buff_to_string(buff) =20 - def format_events_enum(self): + def format_events_enum(self) -> list[str]: buff =3D [] buff.append("enum %s {" % self.enum_events_def) buff.append(self.get_enum_events_string()) @@ -83,7 +83,7 @@ class Dot2c(Automata): =20 return buff =20 - def get_minimun_type(self): + def get_minimun_type(self) -> str: min_type =3D "unsigned char" =20 if self.states.__len__() > 255: @@ -97,7 +97,7 @@ class Dot2c(Automata): =20 return min_type =20 - def format_automaton_definition(self): + def format_automaton_definition(self) -> list[str]: min_type =3D self.get_minimun_type() buff =3D [] buff.append("struct %s {" % self.struct_automaton_def) @@ -109,12 +109,12 @@ class Dot2c(Automata): buff.append("};\n") return buff =20 - def format_aut_init_header(self): + def format_aut_init_header(self) -> list[str]: buff =3D [] buff.append("static const struct %s %s =3D {" % (self.struct_autom= aton_def, self.var_automaton_def)) return buff =20 - def __get_string_vector_per_line_content(self, buff): + def __get_string_vector_per_line_content(self, buff: list[str]) -> str: first =3D True string =3D "" for entry in buff: @@ -133,14 +133,14 @@ class Dot2c(Automata): def get_aut_init_states_string(self): return self.__get_string_vector_per_line_content(self.states) =20 - def format_aut_init_events_string(self): + def format_aut_init_events_string(self) -> list[str]: buff =3D [] buff.append("\t.event_names =3D {") buff.append(self.get_aut_init_events_string()) buff.append("\t},") return buff =20 - def format_aut_init_states_string(self): + def format_aut_init_states_string(self) -> list[str]: buff =3D [] buff.append("\t.state_names =3D {") buff.append(self.get_aut_init_states_string()) @@ -148,11 +148,11 @@ class Dot2c(Automata): =20 return buff =20 - def __get_max_strlen_of_states(self): + def __get_max_strlen_of_states(self) -> int: max_state_name =3D max(self.states, key =3D len).__len__() return max(max_state_name, self.invalid_state_str.__len__()) =20 - def get_aut_init_function(self): + def get_aut_init_function(self) -> str: nr_states =3D self.states.__len__() nr_events =3D self.events.__len__() buff =3D [] @@ -180,7 +180,7 @@ class Dot2c(Automata): =20 return self.__buff_to_string(buff) =20 - def format_aut_init_function(self): + def format_aut_init_function(self) -> list[str]: buff =3D [] buff.append("\t.function =3D {") buff.append(self.get_aut_init_function()) @@ -188,17 +188,17 @@ class Dot2c(Automata): =20 return buff =20 - def get_aut_init_initial_state(self): + def get_aut_init_initial_state(self) -> str: return self.initial_state =20 - def format_aut_init_initial_state(self): + def format_aut_init_initial_state(self) -> list[str]: buff =3D [] initial_state =3D self.get_aut_init_initial_state() buff.append("\t.initial_state =3D " + initial_state + self.enum_su= ffix + ",") =20 return buff =20 - def get_aut_init_final_states(self): + def get_aut_init_final_states(self) -> str: line =3D "" first =3D True for state in self.states: @@ -213,29 +213,29 @@ class Dot2c(Automata): line =3D line + '0' return line =20 - def format_aut_init_final_states(self): + def format_aut_init_final_states(self) -> list[str]: buff =3D [] buff.append("\t.final_states =3D { %s }," % self.get_aut_init_final= _states()) =20 return buff =20 - def __get_automaton_initialization_footer_string(self): + def __get_automaton_initialization_footer_string(self) -> str: footer =3D "};\n" return footer =20 - def format_aut_init_footer(self): + def format_aut_init_footer(self) -> list[str]: buff =3D [] buff.append(self.__get_automaton_initialization_footer_string()) =20 return buff =20 - def format_invalid_state(self): + def format_invalid_state(self) -> list[str]: buff =3D [] buff.append("#define %s state_max%s\n" % (self.invalid_state_str, = self.enum_suffix)) =20 return buff =20 - def format_model(self): + def format_model(self) -> list[str]: buff =3D [] buff +=3D self.format_states_enum() buff +=3D self.format_invalid_state() diff --git a/tools/verification/rvgen/rvgen/dot2k.py b/tools/verification/r= vgen/rvgen/dot2k.py index d618a842fc52..6128fe238430 100644 --- a/tools/verification/rvgen/rvgen/dot2k.py +++ b/tools/verification/rvgen/rvgen/dot2k.py @@ -21,10 +21,10 @@ class dot2k(Monitor, Dot2c): Dot2c.__init__(self, file_path, extra_params.get("model_name")) self.enum_suffix =3D "_%s" % self.name =20 - def fill_monitor_type(self): + def fill_monitor_type(self) -> str: return self.monitor_type.upper() =20 - def fill_tracepoint_handlers_skel(self): + def fill_tracepoint_handlers_skel(self) -> str: buff =3D [] for event in self.events: buff.append("static void handle_%s(void *data, /* XXX: fill he= ader */)" % event) @@ -45,19 +45,19 @@ class dot2k(Monitor, Dot2c): buff.append("") return '\n'.join(buff) =20 - def fill_tracepoint_attach_probe(self): + def fill_tracepoint_attach_probe(self) -> str: buff =3D [] for event in self.events: buff.append("\trv_attach_trace_probe(\"%s\", /* XXX: tracepoin= t */, handle_%s);" % (self.name, event)) return '\n'.join(buff) =20 - def fill_tracepoint_detach_helper(self): + def fill_tracepoint_detach_helper(self) -> str: buff =3D [] for event in self.events: buff.append("\trv_detach_trace_probe(\"%s\", /* XXX: tracepoin= t */, handle_%s);" % (self.name, event)) return '\n'.join(buff) =20 - def fill_model_h_header(self): + def fill_model_h_header(self) -> list[str]: buff =3D [] buff.append("/* SPDX-License-Identifier: GPL-2.0 */") buff.append("/*") @@ -71,7 +71,7 @@ class dot2k(Monitor, Dot2c): =20 return buff =20 - def fill_model_h(self): + def fill_model_h(self) -> str: # # Adjust the definition names # @@ -85,17 +85,17 @@ class dot2k(Monitor, Dot2c): =20 return '\n'.join(buff) =20 - def fill_monitor_class_type(self): + def fill_monitor_class_type(self) -> str: if self.monitor_type =3D=3D "per_task": return "DA_MON_EVENTS_ID" return "DA_MON_EVENTS_IMPLICIT" =20 - def fill_monitor_class(self): + def fill_monitor_class(self) -> str: if self.monitor_type =3D=3D "per_task": return "da_monitor_id" return "da_monitor" =20 - def fill_tracepoint_args_skel(self, tp_type): + def fill_tracepoint_args_skel(self, tp_type: str) -> str: buff =3D [] tp_args_event =3D [ ("char *", "state"), @@ -117,7 +117,7 @@ class dot2k(Monitor, Dot2c): buff.append(" TP_ARGS(%s)" % tp_args_c) return '\n'.join(buff) =20 - def fill_main_c(self): + def fill_main_c(self) -> str: main_c =3D super().fill_main_c() =20 min_type =3D self.get_minimun_type() --=20 2.51.0