| From 7b98bdda0ba286bcfe0d97fd206fd203bc32d3b4 Mon Sep 17 00:00:00 2001 |
| From: Sasha Levin <sashal@kernel.org> |
| Date: Thu, 17 Oct 2024 08:42:39 +0200 |
| Subject: verification/dot2: Improve dot parser robustness |
| |
| From: Gabriele Monaco <gmonaco@redhat.com> |
| |
| [ Upstream commit 571f8b3f866a6d990a50fe5c89fe0ea78784d70b ] |
| |
| This patch makes the dot parser used by dot2c and dot2k slightly more |
| robust, namely: |
| * allows parsing files with the gv extension (GraphViz) |
| * correctly parses edges with any indentation |
| * used to work only with a single character (e.g. '\t') |
| Additionally it fixes a couple of warnings reported by pylint such as |
| wrong indentation and comparison to False instead of `not ...` |
| |
| Link: https://lore.kernel.org/20241017064238.41394-2-gmonaco@redhat.com |
| Signed-off-by: Gabriele Monaco <gmonaco@redhat.com> |
| Signed-off-by: Steven Rostedt (Google) <rostedt@goodmis.org> |
| Signed-off-by: Sasha Levin <sashal@kernel.org> |
| --- |
| tools/verification/dot2/automata.py | 18 +++++++++--------- |
| 1 file changed, 9 insertions(+), 9 deletions(-) |
| |
| diff --git a/tools/verification/dot2/automata.py b/tools/verification/dot2/automata.py |
| index baffeb960ff0b..bdeb98baa8b06 100644 |
| --- a/tools/verification/dot2/automata.py |
| +++ b/tools/verification/dot2/automata.py |
| @@ -29,11 +29,11 @@ class Automata: |
| |
| def __get_model_name(self): |
| basename = ntpath.basename(self.__dot_path) |
| - if basename.endswith(".dot") == False: |
| + if not basename.endswith(".dot") and not basename.endswith(".gv"): |
| print("not a dot file") |
| raise Exception("not a dot file: %s" % self.__dot_path) |
| |
| - model_name = basename[0:-4] |
| + model_name = ntpath.splitext(basename)[0] |
| if model_name.__len__() == 0: |
| raise Exception("not a dot file: %s" % self.__dot_path) |
| |
| @@ -68,9 +68,9 @@ class Automata: |
| def __get_cursor_begin_events(self): |
| cursor = 0 |
| while self.__dot_lines[cursor].split()[0] != "{node": |
| - cursor += 1 |
| + cursor += 1 |
| while self.__dot_lines[cursor].split()[0] == "{node": |
| - cursor += 1 |
| + cursor += 1 |
| # skip initial state transition |
| cursor += 1 |
| return cursor |
| @@ -94,11 +94,11 @@ class Automata: |
| initial_state = state[7:] |
| else: |
| states.append(state) |
| - if self.__dot_lines[cursor].__contains__("doublecircle") == True: |
| + if "doublecircle" in self.__dot_lines[cursor]: |
| final_states.append(state) |
| has_final_states = True |
| |
| - if self.__dot_lines[cursor].__contains__("ellipse") == True: |
| + if "ellipse" in self.__dot_lines[cursor]: |
| final_states.append(state) |
| has_final_states = True |
| |
| @@ -110,7 +110,7 @@ class Automata: |
| # Insert the initial state at the bein og the states |
| states.insert(0, initial_state) |
| |
| - if has_final_states == False: |
| + if not has_final_states: |
| final_states.append(initial_state) |
| |
| return states, initial_state, final_states |
| @@ -120,7 +120,7 @@ class Automata: |
| cursor = self.__get_cursor_begin_events() |
| |
| events = [] |
| - while self.__dot_lines[cursor][1] == '"': |
| + while self.__dot_lines[cursor].lstrip()[0] == '"': |
| # transitions have the format: |
| # "all_fired" -> "both_fired" [ label = "disable_irq" ]; |
| # ------------ event is here ------------^^^^^ |
| @@ -161,7 +161,7 @@ class Automata: |
| # and we are back! Let's fill the matrix |
| cursor = self.__get_cursor_begin_events() |
| |
| - while self.__dot_lines[cursor][1] == '"': |
| + while self.__dot_lines[cursor].lstrip()[0] == '"': |
| if self.__dot_lines[cursor].split()[1] == "->": |
| line = self.__dot_lines[cursor].split() |
| origin_state = line[0].replace('"','').replace(',','_') |
| -- |
| 2.43.0 |
| |