CheckedRegions understnad ret types

Author: Aaron Eline

clang/lib/3C/CheckedRegions.cpp

@@ -92,14 +92,19 @@ CheckedRegionAdder::findParentCompound(const ast_type_traits::DynTypedNode &N,
   return *Min;
 }
 
-bool CheckedRegionAdder::isFunctionBody(CompoundStmt *S) {
+
+bool isTopLevel(ASTContext *Context, CompoundStmt *S) {
   const auto &Parents = Context->getParents(*S);
   if (Parents.empty()) {
     return false;
   }
   return Parents[0].get<FunctionDecl>();
 }
 
+bool CheckedRegionAdder::isFunctionBody(CompoundStmt *S) {
+  return isTopLevel(Context, S);
+}
+
 bool CheckedRegionAdder::isParentChecked(
     const ast_type_traits::DynTypedNode &DTN) {
   if (const auto *Parent = findParentCompound(DTN).first) {
@@ -154,13 +159,26 @@ bool CheckedRegionFinder::VisitDoStmt(DoStmt *S) {
 
 bool CheckedRegionFinder::VisitCompoundStmt(CompoundStmt *S) {
   // Visit all subblocks, find all unchecked types
-  bool Localwild = 0;
+  bool Localwild = false;
   for (const auto &SubStmt : S->children()) {
     CheckedRegionFinder Sub(Context, Writer, Info, Seen, Map, EmitWarnings);
     Sub.TraverseStmt(SubStmt);
     Localwild |= Sub.Wild;
   }
 
+  // If we are a function def, need to check return type
+  if (isTopLevel(Context, S)) {
+    const auto &Parents = Context->getParents(*S);
+    assert(!Parents.empty());
+    FunctionDecl* Parent = const_cast<FunctionDecl*>(Parents[0].get<FunctionDecl>());
+    assert(Parent != NULL);
+    auto retType = Parent->getReturnType().getTypePtr();
+    if (retType->isPointerType()) {
+      CVarOption CV = Info.getVariable(Parent, Context);
+      Localwild |= isWild(CV) || containsUncheckedPtr(Parent->getReturnType());
+    }
+  }
+
   markChecked(S, Localwild);
 
   Wild = false;

clang/test/3C/checkedregions.c

@@ -80,3 +80,10 @@ void baz(void) {
     bad();
   }
 }
+
+int* g() { 
+	//CHECK: int* g(void) {
+	return 1;
+}
+
+

clang/test/3C/fn_sets.c

@@ -51,19 +51,19 @@ void foo1(int *z) {
 /* Testing Something with a larger set of functions */
 
 int *a() {
-	//CHECK: int *a(void) _Checked {
+	//CHECK: int *a(void)  {
   return 0;
 }
 int *b() {
-	//CHECK: int *b(void) _Checked {
+	//CHECK: int *b(void)  {
   return 0;
 }
 int *c() {
-	//CHECK: int *c(void) _Checked {
+	//CHECK: int *c(void)  {
   return 0;
 }
 int *d() {
-	//CHECK: int *d(void) _Checked {
+	//CHECK: int *d(void)  {
   return 0;
 }
 int *e() {
@@ -72,7 +72,7 @@ int *e() {
 	//CHECK: return (int*) 1;
 }
 int *i() {
-	//CHECK: _Ptr<int> i(void) _Checked {
+	//CHECK: _Ptr<int> i(void) _Checked  {
   return 0;
 }