psycopg2: broaden exceptions (#8018)

2022-07-16 18:48:57 +01:00 · 2022-07-16 18:48:57 +01:00 · aabda0f17f
parent 5405d3b6e0
commit aabda0f17f
1 changed files with 8 additions and 2 deletions
--- a/projects/psycopg2/fuzz_sql.py
+++ b/projects/psycopg2/fuzz_sql.py
@ -31,8 +31,14 @@ def TestInput(data):
        q1.join([i1, i2])
        query.format(q1, i3,  i4)
        comp = sql.Composed([query, i5])
-    except (TypeError, ValueError) as e:
-       pass
+    except (
+        AttributeError,
+        TypeError,
+        ValueError,
+        KeyError,
+        IndexError
+    ) as e:
+        pass

 def main():
    atheris.instrument_all()