From 59aabb0e6970ddfca8312c5f476adc4e651d8189 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Tue, 21 Apr 2020 09:01:42 -0400
Subject: [PATCH] Fix apikey storage

Adapt to how crypted. values are handled.
---
 confluent_server/confluent/credserver.py  | 2 +-
 confluent_server/confluent/selfservice.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/confluent_server/confluent/credserver.py b/confluent_server/confluent/credserver.py
index dec553f8..f6632d00 100644
--- a/confluent_server/confluent/credserver.py
+++ b/confluent_server/confluent/credserver.py
@@ -69,7 +69,7 @@ class CredServer(object):
                 client.close()
                 return
             echotoken = util.stringify(client.recv(tlv[1]))
-            cfgupdate = {nodename: {'crypted.selfapikey': echotoken, 'deployment.apiarmed': ''}}
+            cfgupdate = {nodename: {'crypted.selfapikey': {'hashvalue': echotoken}, 'deployment.apiarmed': ''}}
             if apiarmed == 'continuous':
                 del cfgupdate[nodename]['deployment.apiarmed']
             self.cfm.set_node_attributes(cfgupdate)
diff --git a/confluent_server/confluent/selfservice.py b/confluent_server/confluent/selfservice.py
index d271174e..ee9f8f33 100644
--- a/confluent_server/confluent/selfservice.py
+++ b/confluent_server/confluent/selfservice.py
@@ -28,7 +28,7 @@ def handle_request(env, start_response):
         return
     cfg = configmanager.ConfigManager(None)
     eak = cfg.get_node_attributes(nodename, 'crypted.selfapikey').get(
-        nodename, {}).get('crypted.selfapikey', {}).get('value', None)
+        nodename, {}).get('crypted.selfapikey', {}).get('hashvalue', None)
     if not eak:
         start_response('401 Unauthorized', [])
         yield 'Unauthorized'