nem2: Handle Provision Namespace transactions

firmware/fsm.c

@@ -1154,12 +1154,13 @@ void fsm_msgNEMSignTx(NEMSignTx *msg) {
 	CHECK_PARAM(msg->has_transaction, _("No common provided"));
 
 	// Ensure exactly one transaction is provided
-	unsigned int provided = msg->has_transfer;
+	unsigned int provided = msg->has_transfer + msg->has_provision_namespace;
 	CHECK_PARAM(provided != 0, _("No transaction provided"));
 	CHECK_PARAM(provided == 1, _("More than one transaction provided"));
 
 	NEM_CHECK_PARAM(nem_validate_common(&msg->transaction, false));
 	NEM_CHECK_PARAM_WHEN(msg->has_transfer, nem_validate_transfer(&msg->transfer, msg->transaction.network));
+	NEM_CHECK_PARAM_WHEN(msg->has_provision_namespace, nem_validate_provision_namespace(&msg->provision_namespace, msg->transaction.network));
 
 	bool cosigning = msg->has_cosigning && msg->cosigning;
 	if (msg->has_multisig) {
@@ -1194,6 +1195,12 @@ void fsm_msgNEMSignTx(NEMSignTx *msg) {
 		return;
 	}
 
+	if (msg->has_provision_namespace && !nem_askProvisionNamespace(common, &msg->provision_namespace, network)) {
+		fsm_sendFailure(FailureType_Failure_ActionCancelled, _("Signing cancelled by user"));
+		layoutHome();
+		return;
+	}
+
 	RESP_INIT(NEMSignedTx);
 
 	HDNode *node = fsm_getDerivedNode(ED25519_KECCAK_NAME, msg->transaction.address_n, msg->transaction.address_n_count);
@@ -1215,6 +1222,11 @@ void fsm_msgNEMSignTx(NEMSignTx *msg) {
 			return;
 		}
 
+		if (msg->has_provision_namespace && !nem_fsmProvisionNamespace(&inner, &msg->multisig, &msg->provision_namespace)) {
+			layoutHome();
+			return;
+		}
+
 		if (!nem_fsmMultisig(&context, &msg->transaction, &inner, cosigning)) {
 			layoutHome();
 			return;
@@ -1224,6 +1236,11 @@ void fsm_msgNEMSignTx(NEMSignTx *msg) {
 			layoutHome();
 			return;
 		}
+
+		if (msg->has_provision_namespace && !nem_fsmProvisionNamespace(&context, &msg->transaction, &msg->provision_namespace)) {
+			layoutHome();
+			return;
+		}
 	}
 
 	resp->has_data = true;

firmware/nem2.c

@@ -80,6 +80,15 @@ const char *nem_validate_transfer(const NEMTransfer *transfer, uint8_t network)
 	return NULL;
 }
 
+const char *nem_validate_provision_namespace(const NEMProvisionNamespace *provision_namespace, uint8_t network) {
+	if (!provision_namespace->has_namespace) return _("No namespace provided");
+	if (!provision_namespace->has_sink) return _("No rental sink provided");
+	if (!provision_namespace->has_fee) return _("No rental sink fee provided");
+
+	if (!nem_validate_address(provision_namespace->sink, network)) return _("Invalid rental sink address");
+
+	return NULL;
+}
 
 bool nem_askTransfer(const NEMTransactionCommon *common, const NEMTransfer *transfer, const char *desc) {
 	if (transfer->mosaics_count) {
@@ -234,6 +243,42 @@ bool nem_fsmTransfer(nem_transaction_ctx *context, const HDNode *node, const NEM
 	return true;
 }
 
+bool nem_askProvisionNamespace(const NEMTransactionCommon *common, const NEMProvisionNamespace *provision_namespace, const char *desc) {
+	layoutDialog(&bmp_icon_question,
+		_("Cancel"),
+		_("Next"),
+		desc,
+		_("Create namespace"),
+		provision_namespace->namespace,
+		provision_namespace->has_parent ? _("under namespace") : NULL,
+		provision_namespace->has_parent ? provision_namespace->parent : NULL,
+		NULL,
+		NULL);
+	if (!protectButton(ButtonRequestType_ButtonRequest_ConfirmOutput, false)) {
+		return false;
+	}
+
+	layoutNEMNetworkFee(desc, true, _("Confirm rental fee of"), provision_namespace->fee, _("and network fee of"), common->fee);
+	if (!protectButton(ButtonRequestType_ButtonRequest_SignTx, false)) {
+		return false;
+	}
+
+	return true;
+}
+
+bool nem_fsmProvisionNamespace(nem_transaction_ctx *context, const NEMTransactionCommon *common, const NEMProvisionNamespace *provision_namespace) {
+	return nem_transaction_create_provision_namespace(context,
+		common->network,
+		common->timestamp,
+		NULL,
+		common->fee,
+		common->deadline,
+		provision_namespace->namespace,
+		provision_namespace->has_parent ? provision_namespace->parent : NULL,
+		provision_namespace->sink,
+		provision_namespace->fee);
+}
+
 bool nem_askMultisig(const char *address, const char *desc, bool cosigning, uint64_t fee) {
 	layoutNEMDialog(&bmp_icon_question,
 		_("Cancel"),

firmware/nem2.h

@@ -29,6 +29,7 @@
 
 const char *nem_validate_common(NEMTransactionCommon *common, bool inner);
 const char *nem_validate_transfer(const NEMTransfer *transfer, uint8_t network);
+const char *nem_validate_provision_namespace(const NEMProvisionNamespace *provision_namespace, uint8_t network);
 
 bool nem_askTransaction(const char *desc, const NEMTransactionCommon *common, const NEMSignTx *msg);
 bool nem_fsmTransaction(nem_transaction_ctx *context, const HDNode *node, const NEMTransactionCommon *common, const NEMSignTx *msg);
@@ -36,6 +37,9 @@ bool nem_fsmTransaction(nem_transaction_ctx *context, const HDNode *node, const
 bool nem_askTransfer(const NEMTransactionCommon *common, const NEMTransfer *transfer, const char *desc);
 bool nem_fsmTransfer(nem_transaction_ctx *context, const HDNode *node, const NEMTransactionCommon *common, const NEMTransfer *transfer);
 
+bool nem_askProvisionNamespace(const NEMTransactionCommon *common, const NEMProvisionNamespace *provision_namespace, const char *desc);
+bool nem_fsmProvisionNamespace(nem_transaction_ctx *context, const NEMTransactionCommon *common, const NEMProvisionNamespace *provision_namespace);
+
 bool nem_askMultisig(const char *address, const char *desc, bool cosigning, uint64_t fee);
 bool nem_fsmMultisig(nem_transaction_ctx *context, const NEMTransactionCommon *common, const nem_transaction_ctx *inner, bool cosigning);