Merge Starkware branch

src_features/getPublicKey/cmd_getPublicKey.c

@@ -0,0 +1,76 @@
+#include "shared_context.h"
+#include "apdu_constants.h"
+#ifdef TARGET_BLUE
+#include "ui_blue.h"
+#endif
+#ifdef HAVE_UX_FLOW
+#include "ui_flow.h"
+#endif
+#include "feature_getPublicKey.h"
+
+void handleGetPublicKey(uint8_t p1, uint8_t p2, uint8_t *dataBuffer, uint16_t dataLength, unsigned int *flags, unsigned int *tx) {
+  UNUSED(dataLength);
+  uint8_t privateKeyData[32];
+  uint32_t bip32Path[MAX_BIP32_PATH];
+  uint32_t i;
+  uint8_t bip32PathLength = *(dataBuffer++);
+  cx_ecfp_private_key_t privateKey;
+  reset_app_context();
+  if ((bip32PathLength < 0x01) ||
+      (bip32PathLength > MAX_BIP32_PATH)) {
+    PRINTF("Invalid path\n");
+    THROW(0x6a80);
+  }
+  if ((p1 != P1_CONFIRM) && (p1 != P1_NON_CONFIRM)) {
+    THROW(0x6B00);
+  }
+  if ((p2 != P2_CHAINCODE) && (p2 != P2_NO_CHAINCODE)) {
+    THROW(0x6B00);
+  }
+  for (i = 0; i < bip32PathLength; i++) {
+    bip32Path[i] = U4BE(dataBuffer, 0);
+    dataBuffer += 4;
+  }
+  tmpCtx.publicKeyContext.getChaincode = (p2 == P2_CHAINCODE);
+  io_seproxyhal_io_heartbeat();
+  os_perso_derive_node_bip32(CX_CURVE_256K1, bip32Path, bip32PathLength, privateKeyData, (tmpCtx.publicKeyContext.getChaincode ? tmpCtx.publicKeyContext.chainCode : NULL));
+  cx_ecfp_init_private_key(CX_CURVE_256K1, privateKeyData, 32, &privateKey);
+  io_seproxyhal_io_heartbeat();
+  cx_ecfp_generate_pair(CX_CURVE_256K1, &tmpCtx.publicKeyContext.publicKey, &privateKey, 1);
+  os_memset(&privateKey, 0, sizeof(privateKey));
+  os_memset(privateKeyData, 0, sizeof(privateKeyData));
+  io_seproxyhal_io_heartbeat();
+  getEthAddressStringFromKey(&tmpCtx.publicKeyContext.publicKey, tmpCtx.publicKeyContext.address, &sha3);
+#ifndef NO_CONSENT
+  if (p1 == P1_NON_CONFIRM)
+#endif // NO_CONSENT
+  {
+    *tx = set_result_get_publicKey();
+    THROW(0x9000);
+  }
+#ifndef NO_CONSENT
+  else
+  {
+    /*
+    addressSummary[0] = '0';
+    addressSummary[1] = 'x';
+    os_memmove((unsigned char *)(addressSummary + 2), tmpCtx.publicKeyContext.address, 4);
+    os_memmove((unsigned char *)(addressSummary + 6), "...", 3);
+    os_memmove((unsigned char *)(addressSummary + 9), tmpCtx.publicKeyContext.address + 40 - 4, 4);
+    addressSummary[13] = '\0';
+    */
+
+    // prepare for a UI based reply
+#if defined(TARGET_BLUE)
+    snprintf(strings.common.fullAddress, sizeof(strings.common.fullAddress), "0x%.*s", 40, tmpCtx.publicKeyContext.address);
+    UX_DISPLAY(ui_address_blue, ui_address_blue_prepro);
+#else
+    snprintf(strings.common.fullAddress, sizeof(strings.common.fullAddress), "0x%.*s", 40, tmpCtx.publicKeyContext.address);
+    ux_flow_init(0, ux_display_public_flow, NULL);
+#endif // #if TARGET_ID
+
+    *flags |= IO_ASYNCH_REPLY;
+  }
+#endif // NO_CONSENT
+}
+

src_features/getPublicKey/feature_getPublicKey.h

@@ -0,0 +1,4 @@
+#include "shared_context.h"
+
+uint32_t set_result_get_publicKey(void);
+

src_features/getPublicKey/logic_getPublicKey.c

@@ -0,0 +1,17 @@
+#include "shared_context.h"
+
+uint32_t set_result_get_publicKey() {
+    uint32_t tx = 0;
+    G_io_apdu_buffer[tx++] = 65;
+    os_memmove(G_io_apdu_buffer + tx, tmpCtx.publicKeyContext.publicKey.W, 65);
+    tx += 65;
+    G_io_apdu_buffer[tx++] = 40;
+    os_memmove(G_io_apdu_buffer + tx, tmpCtx.publicKeyContext.address, 40);
+    tx += 40;
+    if (tmpCtx.publicKeyContext.getChaincode) {
+      os_memmove(G_io_apdu_buffer + tx, tmpCtx.publicKeyContext.chainCode, 32);
+      tx += 32;
+    }
+    return tx;
+}
+

src_features/getPublicKey/ui_common_getPublicKey.c

@@ -0,0 +1,27 @@
+#include "shared_context.h"
+#include "feature_getPublicKey.h"
+#include "ui_callbacks.h"
+
+unsigned int io_seproxyhal_touch_address_ok(const bagl_element_t *e) {
+    uint32_t tx = set_result_get_publicKey();
+    G_io_apdu_buffer[tx++] = 0x90;
+    G_io_apdu_buffer[tx++] = 0x00;
+    reset_app_context();
+    // Send back the response, do not restart the event loop
+    io_exchange(CHANNEL_APDU | IO_RETURN_AFTER_TX, tx);
+    // Display back the original UX
+    ui_idle();
+    return 0; // do not redraw the widget
+}
+
+unsigned int io_seproxyhal_touch_address_cancel(const bagl_element_t *e) {
+    G_io_apdu_buffer[0] = 0x69;
+    G_io_apdu_buffer[1] = 0x85;
+    reset_app_context();
+    // Send back the response, do not restart the event loop
+    io_exchange(CHANNEL_APDU | IO_RETURN_AFTER_TX, 2);
+    // Display back the original UX
+    ui_idle();
+    return 0; // do not redraw the widget
+}
+

src_features/getPublicKey/ui_flow_getPublicKey.c

@@ -0,0 +1,47 @@
+#include "shared_context.h"
+#include "ui_callbacks.h"
+
+#ifdef HAVE_UX_FLOW
+
+UX_FLOW_DEF_NOCB(
+    ux_display_public_flow_1_step,
+    pnn,
+    {
+      &C_icon_eye,
+      "Verify",
+      "address",
+    });
+UX_FLOW_DEF_NOCB(
+    ux_display_public_flow_2_step,
+    bnnn_paging,
+    {
+      .title = "Address",
+      .text = strings.common.fullAddress,
+    });
+UX_FLOW_DEF_VALID(
+    ux_display_public_flow_3_step,
+    pb,
+    io_seproxyhal_touch_address_ok(NULL),
+    {
+      &C_icon_validate_14,
+      "Approve",
+    });
+UX_FLOW_DEF_VALID(
+    ux_display_public_flow_4_step,
+    pb,
+    io_seproxyhal_touch_address_cancel(NULL),
+    {
+      &C_icon_crossmark,
+      "Reject",
+    });
+
+const ux_flow_step_t *        const ux_display_public_flow [] = {
+  &ux_display_public_flow_1_step,
+  &ux_display_public_flow_2_step,
+  &ux_display_public_flow_3_step,
+  &ux_display_public_flow_4_step,
+  FLOW_END_STEP,
+};
+
+#endif
+