Adds extra logs for HSM related classes

rsksmart · Nov 17, 2023 · 0a4d054 · 0a4d054
1 parent c0677a9
commit 0a4d054
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 8 deletions.
diff --git a/src/main/java/co/rsk/federate/signing/ECDSAHSMSigner.java b/src/main/java/co/rsk/federate/signing/ECDSAHSMSigner.java
@@ -90,7 +90,7 @@ public ECDSASignerCheckResult check() {
                 client.getPublicKey(mapping.getValue());
             } catch (HSMClientException e) {
                 messages.add(e.getMessage());
-                LOGGER.error("Unable to retrieve public key", e);
+                LOGGER.error("[check] Unable to retrieve public key", e);
             }
         }
 

diff --git a/src/main/java/co/rsk/federate/signing/hsm/advanceblockchain/HSMBookkeepingService.java b/src/main/java/co/rsk/federate/signing/hsm/advanceblockchain/HSMBookkeepingService.java
@@ -155,12 +155,16 @@ protected void informConfirmedBlockHeaders() {
                 informing = false;
                 return;
             }
+
             logger.debug(
-                "[informConfirmedBlockHeaders] Going to inform {} block headers. From {} to {}",
-                blocks.size(),
-                blocks.get(0).getHash(),
-                blocks.get(blocks.size() - 1).getHash()
+                    "[informConfirmedBlockHeaders] Going to inform {} block headers. From block number {} with hash {} to block number {} with hash {}",
+                    blocks.size(),
+                    blocks.get(0).getNumber(),
+                    blocks.get(0).getHash(),
+                    blocks.get(blocks.size() - 1).getNumber(),
+                    blocks.get(blocks.size() - 1).getHash()
             );
+
             hsmBookkeepingClient.advanceBlockchain(blocks);
             hsmCurrentBestBlock = getHsmBestBlock();
             logger.debug(

diff --git a/src/main/java/co/rsk/federate/signing/hsm/client/HSMClientProtocol.java b/src/main/java/co/rsk/federate/signing/hsm/client/HSMClientProtocol.java
@@ -104,11 +104,11 @@ public JsonNode send(ObjectNode command) throws HSMClientException {
                 client = clientProvider.acquire();
                 String commandName = command.get(COMMAND.getFieldName()).toString();
                 logger.trace("[send] Sending command to hsm: {}", commandName);
-                Future future = getExecutor().submit(new HSMRequest(client, command));
+                Future<JsonNode> future = getExecutor().submit(new HSMRequest(client, command));
                 JsonNode result = null;
                 try {
                     logger.trace("[send] Fetching response for command: {}", commandName);
-                    result = (JsonNode) future.get();
+                    result = future.get();
                     logger.trace("[send] Got response for command: {}", commandName);
                 } catch (ExecutionException e) {
                     Throwable cause = e.getCause();

diff --git a/src/main/java/co/rsk/federate/signing/hsm/client/HSMResponseHandlerBase.java b/src/main/java/co/rsk/federate/signing/hsm/client/HSMResponseHandlerBase.java
@@ -24,13 +24,17 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import java.util.ArrayList;
 import java.util.List;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import static co.rsk.federate.signing.HSMField.ERROR;
 import static co.rsk.federate.signing.HSMField.ERROR_CODE;
 import static co.rsk.federate.signing.hsm.client.HSMResponseCode.*;
 
 public class HSMResponseHandlerBase {
 
+    private static final Logger logger = LoggerFactory.getLogger(HSMResponseHandlerBase.class);
+
     //When a `getVersion` request is sent to HSM a `device error` can occur
     //Since the HSM version is yet undefined we need to handle error codes for both versions at this stage
     protected void handleErrorResponse(String methodName, int errorCode, JsonNode response) throws HSMClientException {
@@ -41,9 +45,11 @@ protected void handleErrorResponse(String methodName, int errorCode, JsonNode re
     }
 
     protected final int validateResponse(String methodName, JsonNode response) throws HSMClientException {
+        logger.trace("[validateResponse] validating response for HSM method {}. Response: {}", methodName, response.asText());
         validatePresenceOf(response, ERROR_CODE.getFieldName());
 
         int errorCode = response.get(ERROR_CODE.getFieldName()).asInt();
+        logger.trace("[validateResponse] response error code: {}", errorCode);
         if (getOkErrorCodes().contains(errorCode)) {
             return errorCode;
         }

diff --git a/src/main/java/co/rsk/federate/signing/hsm/client/HSMSigningClientProvider.java b/src/main/java/co/rsk/federate/signing/hsm/client/HSMSigningClientProvider.java
@@ -46,7 +46,7 @@ public HSMSigningClientProvider(HSMClientProtocol protocol, String keyId) {
     public HSMSigningClient getSigningClient() throws HSMClientException {
         int version = this.hsmClientProtocol.getVersion();
         HSMSigningClient client;
-        logger.debug("[getClient] version: {}, keyId: {}", version, keyId);
+        logger.debug("[getSigningClient] version: {}, keyId: {}", version, keyId);
         if (version == 1) {
             client = new HSMSigningClientV1(this.hsmClientProtocol);
         } else if (version >= 2) {