citrun

commit f56ff53a98783743e33ad47b063efbe6cb5b4396
parent 3aa602c158f92d51cb251472c3b355826b9da8f3
Author: kyle <kyle@getaddrinfo.net>
Date:   Sun, 25 Oct 2015 23:48:25 -0600

instrument: get instrumented line numbers working

Diffstat:
M
instrument.cpp
 | 
47
+++++++++++++++++++++++++++++++++++++----------

1 file changed, 37 insertions(+), 10 deletions(-)
diff --git a/instrument.cpp b/instrument.cpp
@@ -35,23 +35,49 @@ public:
 	MyASTVisitor(Rewriter &R) : TheRewriter(R) {}
 
 	bool VisitStmt(Stmt *s) {
-		// Only care about If statements.
-		if (isa<IfStmt>(s)) {
-			IfStmt *IfStatement = cast<IfStmt>(s);
-			Stmt *Then = IfStatement->getThen();
+		std::stringstream ss;
+		SourceManager &SM = TheRewriter.getSourceMgr();
+		unsigned line = SM.getPresumedLineNumber(s->getLocStart());
 
-			TheRewriter.InsertText(Then->getLocStart(), "// the 'if' part\n", true,
-					true);
+		ss << "lines[" << line << "] = 1";
 
-			Stmt *Else = IfStatement->getElse();
-			if (Else)
-				TheRewriter.InsertText(Else->getLocStart(), "// the 'else' part\n",
-						true, true);
+		if (isa<IfStmt>(s)) {
+			IfStmt *IfStatement = cast<IfStmt>(s);
+			Stmt *Cond = IfStatement->getCond();
+			ss << ", ";
+			TheRewriter.InsertTextBefore(Cond->getLocStart(),
+					ss.str());
+		}
+		else if (isa<ForStmt>(s)) {
+			ForStmt *ForStatement = cast<ForStmt>(s);
+			Stmt *Cond = ForStatement->getCond();
+			ss << ", ";
+			TheRewriter.InsertTextAfter(Cond->getLocStart(),
+					ss.str());
+		}
+		else if (isa<ReturnStmt>(s)) {
+			ReturnStmt *ReturnStatement = cast<ReturnStmt>(s);
+			Expr *RetValue = ReturnStatement->getRetValue();
+			ss << ", ";
+			TheRewriter.InsertTextBefore(RetValue->getLocStart(),
+					ss.str());
+		}
+		else if (isa<BreakStmt>(s) || isa<ContinueStmt>(s) ||
+		    isa<DeclStmt>(s) || isa<SwitchStmt>(s) ||
+		    isa<SwitchCase>(s)) {
+			ss << "; ";
+			TheRewriter.InsertTextBefore(s->getLocStart(),
+					ss.str());
+		}
+		else if (isa<Expr>(s)) {
+			// TheRewriter.InsertTextBefore(s->getLocStart(),
+			//		"lines[xx] = 1; ");
 		}
 
 		return true;
 	}
 
+#if 0
 	bool VisitFunctionDecl(FunctionDecl *f) {
 		// Only function definitions (with bodies), not declarations.
 		if (f->hasBody()) {
@@ -81,6 +107,7 @@ public:
 
 		return true;
 	}
+#endif
 
 private:
 	Rewriter &TheRewriter;