Merge pull request iovisor#1982 from iovisor/yhs_dev

prevent array subscript expression if base/index is not rewritable
linux-kern · Sep 21, 2018 · f7b1fb3 · f7b1fb3
2 parents c7ccd5b + a6d7e3a
commit f7b1fb3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/cc/frontends/clang/b_frontend_action.cc b/src/cc/frontends/clang/b_frontend_action.cc
@@ -505,6 +505,12 @@ bool ProbeVisitor::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
  Expr *idx = E->getIdx();
  memb_visited_.insert(E);
 
+ if (!rewriter_.isRewritable(GET_BEGINLOC(base)))
+ return true;
+ if (!rewriter_.isRewritable(GET_BEGINLOC(idx)))
+ return true;
+
+
  string pre, lbracket, rbracket;
  LangOptions opts;
  SourceLocation lbracket_start, lbracket_end;