Add contract address and selector to setExternalPlugin

src_features/setExternalPlugin/cmd_setExternalPlugin.c

@@ -2,6 +2,9 @@
 #include "apdu_constants.h"
 #include "ui_flow.h"
 
+#define CONTRACT_ADDR_SIZE 20
+#define SELECTOR_SIZE      4
+
 void handleSetExternalPlugin(uint8_t p1,
                              uint8_t p2,
                              uint8_t *workBuffer,
@@ -11,9 +14,9 @@ void handleSetExternalPlugin(uint8_t p1,
     UNUSED(p1);
     UNUSED(p2);
     UNUSED(flags);
-    uint8_t pluginNameLength = dataLength;
+    uint8_t pluginNameLength = *workBuffer++;
 
-    if (dataLength < 1) {
+    if (dataLength < 1 || dataLength != 1 + pluginNameLength + CONTRACT_ADDR_SIZE + SELECTOR_SIZE) {
         THROW(0x6A80);
     }
 
@@ -23,6 +26,7 @@ void handleSetExternalPlugin(uint8_t p1,
 
     memmove(dataContext.tokenContext.pluginName, workBuffer, pluginNameLength);
     dataContext.tokenContext.pluginName[pluginNameLength] = '\0';
+    workBuffer += pluginNameLength;
 
     PRINTF("Check external plugin %s\n", dataContext.tokenContext.pluginName);
 
@@ -36,6 +40,9 @@ void handleSetExternalPlugin(uint8_t p1,
         }
         CATCH_OTHER(e) {
             PRINTF("%s external plugin is not present\n", dataContext.tokenContext.pluginName);
+            memset(dataContext.tokenContext.pluginName,
+                   sizeof(dataContext.tokenContext.pluginName),
+                   0);
             THROW(0x6984);
         }
         FINALLY {
@@ -43,9 +50,13 @@ void handleSetExternalPlugin(uint8_t p1,
     }
     END_TRY;
 
-    tmpCtx.transactionContext.externalPluginIsSet = true;
     PRINTF("Plugin found\n");
 
+    memmove(dataContext.tokenContext.contract_address, workBuffer, CONTRACT_ADDR_SIZE);
+    workBuffer += 20;
+    memmove(dataContext.tokenContext.method_selector, workBuffer, SELECTOR_SIZE);
+    tmpCtx.transactionContext.externalPluginIsSet = true;
+
     G_io_apdu_buffer[(*tx)++] = 0x90;
     G_io_apdu_buffer[(*tx)++] = 0x00;
 }