additional approved states for agent auto-reactivate upon verifier re…

…start Signed-off-by: Apoorve Mohan <[email protected]>
Gasu16 · Feb 1, 2021 · 4442806 · 4442806
1 parent 2b217e9
commit 4442806
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/keylime/cloud_verifier_tornado.py b/keylime/cloud_verifier_tornado.py
@@ -649,7 +649,7 @@ def main():
     try:
         query_all = session.query(VerfierMain).all()
         for row in query_all:
-            if row.operational_state == states.GET_QUOTE:
+            if row.operational_state in states.APPROVED_REACTIVATE_STATES:
                 row.operational_state = states.START
         session.commit()
     except SQLAlchemyError as e:

diff --git a/keylime/common/states.py b/keylime/common/states.py
@@ -49,6 +49,7 @@
                 PROVIDE_V, PROVIDE_V_RETRY, FAILED, TERMINATED, INVALID_QUOTE,
                 TENANT_FAILED)
 
+APPROVED_REACTIVATE_STATES = [START, GET_QUOTE, GET_QUOTE_RETRY, PROVIDE_V, PROVIDE_V_RETRY]
 
 STATE_REPRESENTATIONS = {
     REGISTERED: "Registered",