Initial commit

1 files changed, 69 insertions, 0 deletions

diff --git a/pyarg-parsetuple.cocci b/pyarg-parsetuple.cocci
new file mode 100644
index 0000000..2bc454b
--- /dev/null
+++ b/pyarg-parsetuple.cocci
@@ -0,0 +1,69 @@
+@ParseTuple@
+expression args;
+expression fmt;
+type t1;
+t1 p1;
+@@
+
+PyArg_ParseTuple(args, fmt, p1)
+
+@script:python@
+args << ParseTuple.args;
+fmt << ParseTuple.fmt;
+t1 << ParseTuple.t1;
+
+@@
+
+"""
+Analyze
+"""
+# FIXME: generalize thisto varargs
+
+
+def get_types(strfmt):
+    """
+    Generate a list of C type names from a format string
+    """
+    result = []
+    i = 0
+    while i < len(strfmt):
+        c = strfmt[i]
+        if c == 'i':
+            result.append('int *')
+        if c in [':', ';']:
+            break
+        i += 1
+    return result
+
+class CExtensionError(Exception):
+    # Base class for errors discovered by static analysis in C extension code
+    pass
+
+class WrongNumberOfVars(CExtensionError):
+    pass
+
+class NotEnoughVars(WrongNumberOfVars):
+    pass
+
+class TooManyVars(WrongNumberOfVars):
+    pass
+
+class MismatchingType(WrongNumberOfVars):
+    pass
+
+def validate_types(format_string, actual_types):
+    exp_types = get_types(format_string)
+    if len(actual_types) < len(exp_types):
+        raise NotEnoughVars(actual_types, exp_types)
+    if len(actual_types) > len(exp_types):
+        raise TooManyVars(actual_types, exp_types)
+    for exp, actual in zip(exp_types, actual_types):
+        if exp != actual:
+            raise MismatchingType(exp, actual)
+
+print "args: %s" % args
+print "fmt: %s" % fmt
+print "var1: %s" % t1
+print get_types(fmt.expr)
+
+validate_types(fmt.expr, [t1])