Caught invalid error

capuanob · capuanob · commit d38535b33630 · 2022-12-30T21:39:10.000-05:00
diff --git a/mayhem/fuzz_twb.py b/mayhem/fuzz_twb.py
@@ -12,14 +12,14 @@
     from tableaudocumentapi import Workbook
 
 from lxml.etree import XMLSyntaxError
-from tableaudocumentapi.xfile import TableauVersionNotSupportedException
+from tableaudocumentapi.xfile import TableauVersionNotSupportedException, TableauInvalidFileException
 
 def TestOneInput(data):
     fdp = fuzz_helpers.EnhancedFuzzedDataProvider(data)
     try:
         with fdp.ConsumeTemporaryFile('.twb', all_data=True, as_bytes=True) as filename:
             Workbook(filename)
-    except (XMLSyntaxError, TableauVersionNotSupportedException, ValueError, AttributeError):
+    except (XMLSyntaxError, TableauVersionNotSupportedException, ValueError, AttributeError, TableauInvalidFileException):
         return -1
     except TypeError:
         if random.random() > .99:
