Merge pull request #1982 from iovisor/yhs_dev

prevent array subscript expression if base/index is not rewritable

Merge pull request #1982 from iovisor/yhs_dev
prevent array subscript expression if base/index is not rewritable
f7b1fb34 · Teng Qin · GitHub · c7ccd5b7 · a6d7e3af · f7b1fb34
Commit f7b1fb34 authored Sep 21, 2018 by Teng Qin Committed by GitHub Sep 21, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/cc/frontends/clang/b_frontend_action.cc src/cc/frontends/clang/b_frontend_action.cc +6 -0

No files found.
--- 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;