AST: modify pretty printing of function pointer types

Author: sipma

chb/ast/ASTCPrettyPrinter.py

@@ -151,7 +151,7 @@ def write_local_declarations(self, referenced: Set[str]) -> None:
                     and not self.localsymboltable.is_formal(vinfo.vname)):
                 self.ccode.newline(indent=self.indent)
                 if vinfo.vtype is None:
-                    self.ccode.write("? " + vinfo.vname)
+                    self.ccode.write("? " + vinfo.vname + ";")
                     continue
 
                 if vinfo.vtype.is_array:
@@ -185,14 +185,19 @@ def write_global_declarations(self) -> None:
                 continue
             self.ccode.newline(indent=self.indent)
             if vinfo.vtype is None:
-                self.ccode.write("? " + vinfo.vname)
+                self.ccode.write("? " + vinfo.vname + ";")
                 continue
 
             if vinfo.vtype.is_function:
                 ftype = cast(AST.ASTTypFun, vinfo.vtype)
-                ftype.returntyp.accept(self)
-                self.ccode.write(" ")
-                self.ccode.write(vinfo.vname)
+                if ftype.returntyp.is_function_pointer:
+                    returntyp = cast(AST.ASTTypPtr, ftype.returntyp)
+                    fnptr = cast(AST.ASTTypFun, returntyp.tgttyp)
+                    self.funtypptr_with_name(fnptr, vinfo.vname)
+                else:
+                    ftype.returntyp.accept(self)
+                    self.ccode.write(" ")
+                    self.ccode.write(vinfo.vname)
                 self.ccode.write("(")
                 if ftype.argtypes is not None:
                     ftype.argtypes.accept(self)
@@ -608,20 +613,43 @@ def visit_fun_typ(self, t: AST.ASTTypFun) -> None:
             t.argtypes.accept(self)
         self.ccode.write(")")
 
+    def funtypptr_with_name(self, t: AST.ASTTypFun, name: str) -> None:
+        if t.returntyp.is_function_pointer:
+            returntyp = cast(AST.ASTTypPtr, t.returntyp)
+            retty = cast(AST.ASTTypFun, returntyp.tgttyp)
+            self.funtypptr_with_name(retty, name)
+            self.ccode.write("(")
+            if t.argtypes is not None:
+                t.argtypes.accept(self)
+            self.ccode.write(")")
+        else:
+            t.returntyp.accept(self)
+            self.ccode.write(" (*")
+            self.ccode.write(name)
+            self.ccode.write(")(")
+            if t.argtypes is not None:
+                t.argtypes.accept(self)
+                self.ccode.write(")")
+
     def visit_funargs(self, funargs: AST.ASTFunArgs) -> None:
         args = funargs.funargs
         if len(args) == 0:
-            pass
+            self.ccode.write("void")
         else:
             for arg in args[:-1]:
                 arg.accept(self)
                 self.ccode.write(", ")
             args[-1].accept(self)
 
     def visit_funarg(self, funarg: AST.ASTFunArg) -> None:
-        funarg.argtyp.accept(self)
-        self.ccode.write(" ")
-        self.ccode.write(funarg.argname)
+        if funarg.argtyp.is_function_pointer:
+            argtyp = cast(AST.ASTTypPtr, funarg.argtyp)
+            fnptr = cast(AST.ASTTypFun, argtyp.tgttyp)
+            self.funtypptr_with_name(fnptr, funarg.argname)
+        else:
+            funarg.argtyp.accept(self)
+            self.ccode.write(" ")
+            self.ccode.write(funarg.argname)
 
     def visit_named_typ(self, t: AST.ASTTypNamed) -> None:
         self.ccode.write("typedef ")
@@ -649,6 +677,11 @@ def visit_compinfo(self, cinfo: AST.ASTCompInfo) -> None:
                 if atype.size_expr is not None:
                     atype.size_expr.accept(self)
                 self.ccode.write("];")
+            elif finfo.fieldtype.is_function_pointer:
+                fieldtyp = cast(AST.ASTTypPtr, finfo.fieldtype)
+                funtyp = cast(AST.ASTTypFun, fieldtyp.tgttyp)
+                self.funtypptr_with_name(funtyp, finfo.fieldname)
+                self.ccode.write(";")
             else:
                 finfo.fieldtype.accept(self)
                 self.ccode.write(" ")

chb/ast/ASTNode.py

@@ -2242,6 +2242,10 @@ def is_pointer(self) -> bool:
     def is_void_pointer(self) -> bool:
         return False
 
+    @property
+    def is_function_pointer(self) -> bool:
+        return False
+
     @property
     def is_scalar(self) -> bool:
         return (
@@ -2394,6 +2398,10 @@ def is_pointer(self) -> bool:
     def is_void_pointer(self) -> bool:
         return self.tgttyp.is_void
 
+    @property
+    def is_function_pointer(self) -> bool:
+        return self.tgttyp.is_function
+
     def accept(self, visitor: "ASTVisitor") -> None:
         return visitor.visit_pointer_typ(self)
 

chb/ast/astutil.py

@@ -4,7 +4,7 @@
 # ------------------------------------------------------------------------------
 # The MIT License (MIT)
 #
-# Copyright (c) 2023  Aarno Labs LLC
+# Copyright (c) 2023-2025  Aarno Labs LLC
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -358,7 +358,7 @@ def parsecmd(args: argparse.Namespace) -> NoReturn:
     parsemanager.parse_ifile(ifile)
 
     for (faddr, dfn) in functions.items():
-        fname = faddr.replace("0x", "sub_")
+        fname = faddr
         xpath = os.path.join(cname, "functions")
         xpath = os.path.join(xpath, fname)
         xfile = os.path.join(xpath, cname + "_" + fname + "_cfun.xml")
@@ -381,6 +381,4 @@ def parsecmd(args: argparse.Namespace) -> NoReturn:
 
         xsbody = rootnode.find("sbody")
 
-
-
     exit(0)