From 2ecdba92a5a4f94a009ef2e328863a30170c699b Mon Sep 17 00:00:00 2001
From: Malte Bargholz <malte@screenri.de>
Date: Mon, 14 Sep 2020 20:57:50 +0200
Subject: [PATCH] [generic-experiment] allow passing state directory via
 environment

---
 src/experiments/generic-experiment/experiment.cc | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/experiments/generic-experiment/experiment.cc b/src/experiments/generic-experiment/experiment.cc
index 2fcdeb6d..b4e227ca 100644
--- a/src/experiments/generic-experiment/experiment.cc
+++ b/src/experiments/generic-experiment/experiment.cc
@@ -249,7 +249,16 @@ bool GenericExperiment::cb_start_experiment() {
 	if (cmd[STATE_DIR]) {
 		std::string value(cmd[STATE_DIR].first()->arg);
 		m_state_dir = value;
-		m_log << "Set state dir to " << value << endl;
+		m_log << "Set state dir (from cmdline) to " << value << endl;
+	} else {
+		char *state_dir = getenv("FAIL_STATE_DIR");
+		if (state_dir == NULL) {
+			m_log << "ERROR: no FAIL_STATE_DIR set or --state-dir given. exiting!" << std::endl;
+			exit(1);
+		} else {
+			m_state_dir = std::string(state_dir);
+			m_log << "Set state dir (from enviroment) to " << m_state_dir << endl;
+		}
 	}
 
 	if (cmd[TIMEOUT]) {