From 2694c84ced41d921438fc859f12b3b54c57f4cd8 Mon Sep 17 00:00:00 2001
From: James Pickett <jamesandrewpickett@gmail.com>
Date: Fri, 20 Dec 2024 09:14:23 -0800
Subject: [PATCH] add logging on success

---
 ee/secureenclaverunner/secureenclaverunner.go |  8 ++++++++
 ee/tpmrunner/tpmrunner.go                     | 12 +++++++-----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/ee/secureenclaverunner/secureenclaverunner.go b/ee/secureenclaverunner/secureenclaverunner.go
index b928eb02c..ecda1304d 100644
--- a/ee/secureenclaverunner/secureenclaverunner.go
+++ b/ee/secureenclaverunner/secureenclaverunner.go
@@ -197,6 +197,10 @@ func (ser *secureEnclaveRunner) currentConsoleUserKey(ctx context.Context) (*ecd
 
 	key, ok := ser.uidPubKeyMap[cu.Uid]
 	if ok {
+		ser.slogger.Log(ctx, slog.LevelDebug,
+			"found existing key for console user",
+			"uid", cu.Uid,
+		)
 		span.AddEvent("found_existing_key_for_console_user")
 		return key, nil
 	}
@@ -207,6 +211,10 @@ func (ser *secureEnclaveRunner) currentConsoleUserKey(ctx context.Context) (*ecd
 		return nil, fmt.Errorf("creating key: %w", err)
 	}
 
+	ser.slogger.Log(ctx, slog.LevelInfo,
+		"created new key for console user",
+		"uid", cu.Uid,
+	)
 	span.AddEvent("created_new_key_for_console_user")
 
 	ser.uidPubKeyMap[cu.Uid] = key
diff --git a/ee/tpmrunner/tpmrunner.go b/ee/tpmrunner/tpmrunner.go
index 0b509df44..4e126bad0 100644
--- a/ee/tpmrunner/tpmrunner.go
+++ b/ee/tpmrunner/tpmrunner.go
@@ -98,7 +98,7 @@ func (tr *tpmRunner) Execute() error {
 			continue
 		case <-tr.interrupt:
 			tr.slogger.Log(context.TODO(), slog.LevelDebug,
-				"interrupt received, exiting secure enclave signer execute loop",
+				"interrupt received, exiting tpm signer execute loop",
 			)
 			return nil
 		}
@@ -208,10 +208,6 @@ func (tr *tpmRunner) loadOrCreateKeys(ctx context.Context) error {
 	}
 
 	if pubData == nil || priData == nil {
-		tr.slogger.Log(ctx, slog.LevelInfo,
-			"generating new tpm keys",
-		)
-
 		var err error
 		priData, pubData, err = tr.signerCreator.CreateKey()
 		if err != nil {
@@ -230,6 +226,9 @@ func (tr *tpmRunner) loadOrCreateKeys(ctx context.Context) error {
 			return thisErr
 		}
 
+		tr.slogger.Log(ctx, slog.LevelInfo,
+			"new tpm keys generated",
+		)
 		span.AddEvent("generated_new_tpm_keys")
 	}
 
@@ -242,6 +241,9 @@ func (tr *tpmRunner) loadOrCreateKeys(ctx context.Context) error {
 
 	tr.signer = k
 
+	tr.slogger.Log(ctx, slog.LevelDebug,
+		"tpm signer created",
+	)
 	span.AddEvent("created_tpm_signer")
 
 	return nil