The higher the sleep score, the more likely the fuzzer makes a thread sleep

[c11tester.git] / predicate.cc

diff --git a/predicate.cc b/predicate.cc
index 94eedfd52253382a46e659488346bd596fa397dd..ab64a34ea085a9b01b1b67efe6798933b4c1f841 100644 (file)
--- a/predicate.cc
+++ b/predicate.cc
@@ -1,13 +1,19 @@
+#include "funcinst.h"
 #include "predicate.h"
 #include "concretepredicate.h"
 
-Predicate::Predicate(FuncInst * func_inst, bool is_entry) :
+Predicate::Predicate(FuncInst * func_inst, bool is_entry, bool is_exit) :
        func_inst(func_inst),
        entry_predicate(is_entry),
+       exit_predicate(is_exit),
        does_write(false),
+       exploration_count(0),
+       failure_count(0),
+       sleep_score(100),
        pred_expressions(16),
        children(),
        parent(NULL),
+       exit(NULL),
        backedges(16)
 {}
 
@@ -92,6 +98,31 @@ ConcretePredicate * Predicate::evaluate(inst_act_map_t * inst_act_map, thread_id
        return concrete;
 }
 
+void Predicate::incr_expl_count()
+{
+       exploration_count++;
+}
+
+void Predicate::incr_fail_count()
+{
+       failure_count++;
+}
+
+void Predicate::incr_sleep_score(uint32_t amount)
+{
+       if (sleep_score + amount > 100)
+               sleep_score = 100;
+       else
+               sleep_score += amount;
+}
+
+void Predicate::decr_sleep_score(uint32_t amount)
+{
+       if (sleep_score > amount)
+               sleep_score -= amount;
+       else
+               sleep_score = 0;
+}
 
 void Predicate::print_predicate()
 {
@@ -101,13 +132,17 @@ void Predicate::print_predicate()
                return;
        }
 
-       func_inst->print();
+       if (exit_predicate) {
+               model_print("exit node\"];\n");
+               return;
+       }
 
-       PredExprSetIter * it = pred_expressions.iterator();
+       func_inst->print();
 
        if (pred_expressions.getSize() == 0)
                model_print("predicate unset\n");
 
+       PredExprSetIter * it = pred_expressions.iterator();
        while (it->hasNext()) {
                struct pred_expr * expr = it->next();
                FuncInst * inst = expr->func_inst;
@@ -129,6 +164,7 @@ void Predicate::print_predicate()
        if (does_write) {
                model_print("Does write\n");
        }
+       model_print("Count: %d, failed count: %d\n", exploration_count, failure_count);
        model_print("\"];\n");
 }
 
@@ -146,4 +182,8 @@ void Predicate::print_pred_subtree()
                Predicate * backedge = it->next();
                model_print("\"%p\" -> \"%p\"[style=dashed, color=grey]\n", this, backedge);
        }
+
+       if (exit) {
+               model_print("\"%p\" -> \"%p\"[style=dashed, color=red]\n", this, exit);
+       }
 }