Merge Starkware branch

src_features/stark_getPublicKey/cmd_stark_getPublicKey.c

@@ -0,0 +1,65 @@
+#ifdef HAVE_STARKWARE
+
+#include "shared_context.h"
+#include "apdu_constants.h"
+#include "stark_utils.h"
+#include "feature_stark_getPublicKey.h"
+#ifdef TARGET_BLUE
+#include "ui_blue.h"
+#endif
+#ifdef HAVE_UX_FLOW
+#include "ui_flow.h"
+#endif
+
+void handleStarkwareGetPublicKey(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 != 0) {
+    THROW(0x6B00);
+  }
+  for (i = 0; i < bip32PathLength; i++) {
+    bip32Path[i] = U4BE(dataBuffer, 0);
+    dataBuffer += 4;
+  }
+  io_seproxyhal_io_heartbeat();
+  starkDerivePrivateKey(bip32Path, bip32PathLength, privateKeyData);
+  cx_ecfp_init_private_key(CX_CURVE_Stark256, privateKeyData, 32, &privateKey);
+  io_seproxyhal_io_heartbeat();
+  cx_ecfp_generate_pair(CX_CURVE_Stark256, &tmpCtx.publicKeyContext.publicKey, &privateKey, 1);
+  os_memset(&privateKey, 0, sizeof(privateKey));
+  os_memset(privateKeyData, 0, sizeof(privateKeyData));
+  io_seproxyhal_io_heartbeat();
+#ifndef NO_CONSENT
+  if (p1 == P1_NON_CONFIRM)
+#endif // NO_CONSENT
+  {
+    *tx = set_result_get_stark_publicKey();
+    THROW(0x9000);
+  }
+#ifndef NO_CONSENT
+  else
+  {
+    // prepare for a UI based reply
+    snprintf(strings.tmp.tmp, sizeof(strings.tmp.tmp), "0x%.*H", 32, tmpCtx.publicKeyContext.publicKey.W + 1);
+    ux_flow_init(0, ux_display_stark_public_flow, NULL);
+
+    *flags |= IO_ASYNCH_REPLY;
+  }
+#endif // NO_CONSENT
+}
+
+
+#endif

src_features/stark_getPublicKey/feature_stark_getPublicKey.h

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

src_features/stark_getPublicKey/logic_stark_getPublicKey.c

@@ -0,0 +1,15 @@
+#ifdef HAVE_STARKWARE
+
+#include "shared_context.h"
+#include "feature_stark_getPublicKey.h"
+
+uint32_t set_result_get_stark_publicKey() {
+    uint32_t tx = 0;
+    os_memmove(G_io_apdu_buffer + tx, tmpCtx.publicKeyContext.publicKey.W, 65);
+    tx += 65;
+    return tx;
+}
+
+#endif
+
+

src_features/stark_getPublicKey/ui_common_stark_getPublicKey.c

@@ -0,0 +1,20 @@
+#ifdef HAVE_STARKWARE
+
+#include "shared_context.h"
+#include "ui_callbacks.h"
+#include "feature_stark_getPublicKey.h"
+
+unsigned int io_seproxyhal_touch_stark_pubkey_ok(const bagl_element_t *e) {
+    uint32_t tx = set_result_get_stark_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
+}
+
+#endif
+

src_features/stark_getPublicKey/ui_flow_stark_getPublicKey.c

@@ -0,0 +1,49 @@
+#ifdef HAVE_STARKWARE
+
+#include "shared_context.h"
+#include "ui_callbacks.h"
+
+unsigned int io_seproxyhal_touch_stark_pubkey_ok(const bagl_element_t *e);
+
+UX_FLOW_DEF_NOCB(
+    ux_display_stark_public_flow_1_step,
+    pnn,
+    {
+      &C_icon_eye,
+      "Verify",
+      "Stark key",
+    });
+UX_FLOW_DEF_NOCB(
+    ux_display_stark_public_flow_2_step,
+    bnnn_paging,
+    {
+      .title = "Stark Key",
+      .text = strings.tmp.tmp,
+    });
+UX_FLOW_DEF_VALID(
+    ux_display_stark_public_flow_3_step,
+    pb,
+    io_seproxyhal_touch_stark_pubkey_ok(NULL),
+    {
+      &C_icon_validate_14,
+      "Approve",
+    });
+UX_FLOW_DEF_VALID(
+    ux_display_stark_public_flow_4_step,
+    pb,
+    io_seproxyhal_touch_address_cancel(NULL),
+    {
+      &C_icon_crossmark,
+      "Reject",
+    });
+
+const ux_flow_step_t *        const ux_display_stark_public_flow [] = {
+  &ux_display_stark_public_flow_1_step,
+  &ux_display_stark_public_flow_2_step,
+  &ux_display_stark_public_flow_3_step,
+  &ux_display_stark_public_flow_4_step,
+  FLOW_END_STEP,
+};
+
+#endif
+