fix quokka-cli, following symlinks can cause infinite loop

quarkslab · Jan 12, 2025 · abccaf3 · abccaf3
1 parent 1278e1f
commit abccaf3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bindings/python/quokka/__main__.py b/bindings/python/quokka/__main__.py
@@ -37,15 +37,15 @@ class Bcolors:
 
 
 def recursive_file_iter(p: Path) -> Generator[Path, None, None]:
-    if p.is_file():
+    if p.is_file() and not p.is_symlink():
         mime_type = magic.from_file(p, mime=True)
         if mime_type not in BINARY_FORMAT and p.suffix not in EXTENSIONS_WHITELIST.get(
             mime_type, []
         ):
             pass
         else:
             yield p
-    elif p.is_dir():
+    elif p.is_dir() and not p.is_symlink():
         for f in p.iterdir():
             yield from recursive_file_iter(f)