Introducing irept_instrumenter.

Author: marek-trtik

src/taint-slicer/Makefile

@@ -4,6 +4,7 @@ SRC = slicer.cpp \
       instrumentation_props.cpp \
       instrumenter.cpp \
       slicing_tasks_builder.cpp \
+      irept_instrument.cpp \
       # end of SRC
 
 INCLUDES= -I .. -I ../../cbmc/src -I ../../boost

src/taint-slicer/irept_instrument.cpp

@@ -0,0 +1,129 @@
+/*******************************************************************\
+
+Module: irept_instrument
+
+Author: Marek Trtik
+
+Date: May 2017
+
+
+@ Copyright Diffblue, Ltd.
+
+\*******************************************************************/
+
+#include <taint-slicer/irept_instrument.h>
+
+irept make_shallow_copy(const irept irep)
+{
+  irept result(irep.id());
+  result.get_sub()=irep.get_sub();
+  result.get_named_sub()=irep.get_named_sub();
+  result.get_comments()=irep.get_comments();
+  return result;
+}
+
+irept instrument(const irept irep, const instrumenter_fnt &instrumenter)
+{
+  bool modified=false;
+
+  irept::subt new_sub;
+  for(const irept sub : irep.get_sub())
+  {
+    new_sub.push_back(instrument(sub, instrumenter));
+    if(new_sub.back()!=sub)
+      modified=true;
+  }
+
+  irept::named_subt new_named_sub;
+  for(const irept::named_subt::value_type name_sub : irep.get_named_sub())
+  {
+    const auto it_bool=
+      new_named_sub.insert({
+        name_sub.first,
+        instrument(name_sub.second, instrumenter)});
+    assert(it_bool.second);
+    if(it_bool.first->second!=name_sub.second)
+      modified=true;
+  }
+
+  irept::named_subt new_comments;
+  for(const irept::named_subt::value_type name_sub : irep.get_comments())
+  {
+    const auto it_bool=
+      new_comments.insert({
+        name_sub.first,
+        instrument(name_sub.second, instrumenter)});
+    assert(it_bool.second);
+    if(it_bool.first->second!=name_sub.second)
+      modified=true;
+  }
+
+  if(modified)
+  {
+    irept result(irep.id());
+    result.get_sub()=new_sub;
+    result.get_named_sub()=new_named_sub;
+    result.get_comments()=new_comments;
+    return instrumenter(result);
+  }
+  return instrumenter(irep);
+}
+
+irept instrument_using_pivot(
+  const irept pivot,
+  const irept irep,
+  const pivot_instrumenter_fnt &instrumenter)
+{
+  bool modified=false;
+
+  irept::subt new_sub;
+  for(auto pit=pivot.get_sub().cbegin(), iit=irep.get_sub().cbegin();
+      pit!=pivot.get_sub().cend() && iit!=irep.get_sub().cend(); ++pit, ++iit)
+  {
+    new_sub.push_back(instrument_using_pivot(*pit, *iit, instrumenter));
+    if(new_sub.back()!=*iit)
+      modified=true;
+  }
+
+  irept::named_subt new_named_sub;
+  for(auto pit=pivot.get_named_sub().cbegin(),
+           iit=irep.get_named_sub().cbegin();
+      pit!=pivot.get_named_sub().cend() &&
+      iit!=irep.get_named_sub().cend();
+      ++pit, ++iit)
+  {
+    const auto it_bool=
+      new_named_sub.insert({
+        iit->first,
+        instrument_using_pivot(pit->second, iit->second, instrumenter)});
+    assert(it_bool.second);
+    if(it_bool.first->second!=iit->second)
+      modified=true;
+  }
+
+  irept::named_subt new_comments;
+  for(auto pit=pivot.get_comments().cbegin(),
+           iit=irep.get_comments().cbegin();
+      pit!=pivot.get_comments().cend() &&
+      iit!=irep.get_comments().cend();
+      ++pit, ++iit)
+  {
+    const auto it_bool=
+      new_comments.insert({
+        iit->first,
+        instrument_using_pivot(pit->second, iit->second, instrumenter)});
+    assert(it_bool.second);
+    if(it_bool.first->second!=iit->second)
+      modified=true;
+  }
+
+  if(modified)
+  {
+    irept result(irep.id());
+    result.get_sub()=new_sub;
+    result.get_named_sub()=new_named_sub;
+    result.get_comments()=new_comments;
+    return instrumenter(pivot, result);
+  }
+  return instrumenter(pivot, irep);
+}

src/taint-slicer/irept_instrument.h

@@ -0,0 +1,67 @@
+/*******************************************************************\
+
+Module: irept_instrument
+
+Author: Marek Trtik
+
+Date: May 2017
+
+The module provides utility functuions simplifying the instrumentation
+proces of "irept" hierarchical and shared data structures.
+
+@ Copyright Diffblue, Ltd.
+
+\*******************************************************************/
+
+#ifndef CPROVER_TAINT_IREPT_INSTRUMENT_H
+#define CPROVER_TAINT_IREPT_INSTRUMENT_H
+
+#include <util/irep.h>
+#include <functional>
+
+typedef std::function<irept(irept)> instrumenter_fnt;
+typedef std::function<irept(irept, irept)> pivot_instrumenter_fnt;
+
+/*******************************************************************\
+
+Function:
+
+  Inputs:
+
+ Outputs:
+
+ Purpose:
+
+\*******************************************************************/
+irept make_shallow_copy(const irept irep);
+
+/*******************************************************************\
+
+Function:
+
+  Inputs:
+
+ Outputs:
+
+ Purpose:
+
+\*******************************************************************/
+irept instrument(const irept irep, const instrumenter_fnt &instrumenter);
+
+/*******************************************************************\
+
+Function:
+
+  Inputs:
+
+ Outputs:
+
+ Purpose:
+
+\*******************************************************************/
+irept instrument_using_pivot(
+  const irept pivot,
+  const irept irep,
+  const pivot_instrumenter_fnt &instrumenter);
+
+#endif