Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore: Block Signing protobufs #16426

Merged
merged 10 commits into from
Nov 6, 2024
Original file line number Diff line number Diff line change
@@ -0,0 +1,88 @@
/**
* # Tss Share Signature
* Represents a transaction that submits a node's share signature on a block hash
* during the TSS (Threshold Signature Scheme) process.
*
* ### Keywords
* The key words "MUST", "MUST NOT", "REQUIRED", "SHALL", "SHALL NOT",
* "SHOULD", "SHOULD NOT", "RECOMMENDED", "MAY", and "OPTIONAL" in this
* document are to be interpreted as described in
* [RFC2119](https://www.ietf.org/rfc/rfc2119) and clarified in
* [RFC8174](https://www.ietf.org/rfc/rfc8174).
*/
syntax = "proto3";

package com.hedera.hapi.services.auxiliary.tss;

/*
* Copyright (C) 2024 Hedera Hashgraph, LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

option java_package = "com.hedera.hapi.services.auxiliary.tss.legacy";
// <<<pbj.java_package = "com.hedera.hapi.services.auxiliary.tss">>> This comment is special code for setting PBJ Compiler java package
option java_multiple_files = true;

/**
* A TSS Share Signature transaction Body.<br/>
* This transaction body communicates a node's signature of a block hash
* using its private share within the TSS process.
* This transaction MUST be prioritized for low latency gossip transmission.
*
* ### Block Stream Effects
* This transaction body will be present in the block stream. This will not have
* any state changes or transaction output or transaction result.
*/
thomas-swirlds-labs marked this conversation as resolved.
Show resolved Hide resolved
message TssShareSignatureTransactionBody {
/**
* A SHA2-384 Hash.<br/>
* This is the hash of the roster that includes the node whose
* share produced this share signature.
* <p>
* This value is REQUIRED.<br/>
* This value MUST identify the network roster active at the time this
* share signature was produced.<br/>
* This share signature MUST be produced from a share distributed during
* the re-keying process for the identified roster.
*/
bytes roster_hash = 1;

/**
* An index of the share from the node private shares.<br/>
* This is the index of the share that produced this share signature.
* <p>
* This value is REQUIRED.<br/>
* The share referred to by this index MUST exist.<br/>
* The share index MUST be greater than or equal to 0.
*/
uint64 share_index = 2;

/**
* A SHA2-384 hash.<br/>
* This is the hash of the message that was signed.
* <p>
* This value is REQUIRED.<br/>
* The message signed MUST be a block hash.
*/
bytes message_hash = 3;

/**
* The signature bytes.<br/>
* This is the signature generated by signing the block hash with the node's private share.
* <p>
* This value is REQUIRED.<br/>
* This value MUST be a valid signature of the message hash with the node's private share.
*/
bytes share_signature = 4;
}
5 changes: 5 additions & 0 deletions hapi/hedera-protobufs/services/basic_types.proto
Original file line number Diff line number Diff line change
Expand Up @@ -1259,6 +1259,11 @@ enum HederaFunctionality {
* Submit a vote as part of the Threshold Signature Scheme (TSS) processing.
*/
TssVote = 97;

/**
* Submit a node signature as part of the Threshold Signature Scheme (TSS) processing.
*/
TssShareSignature = 98;
}

/**
Expand Down
6 changes: 6 additions & 0 deletions hapi/hedera-protobufs/services/transaction_body.proto
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,7 @@ import "node_delete.proto";

import "auxiliary/tss/tss_message.proto";
import "auxiliary/tss/tss_vote.proto";
import "auxiliary/tss/tss_share_signature.proto";

/**
* A single transaction. All transaction types are possible here.
Expand Down Expand Up @@ -431,5 +432,10 @@ message TransactionBody {
* A transaction body for a `tssVote` request.
*/
com.hedera.hapi.services.auxiliary.tss.TssVoteTransactionBody tssVote = 62;

/**
* A transaction body for a 'tssShareSignature` request
*/
com.hedera.hapi.services.auxiliary.tss.TssShareSignatureTransactionBody tssShareSignature = 63;
}
}
1 change: 1 addition & 0 deletions hapi/src/main/java/com/hedera/hapi/util/HapiUtils.java
Original file line number Diff line number Diff line change
Expand Up @@ -236,6 +236,7 @@
case TOKEN_CLAIM_AIRDROP -> HederaFunctionality.TOKEN_CLAIM_AIRDROP;
case TSS_MESSAGE -> HederaFunctionality.TSS_MESSAGE;
case TSS_VOTE -> HederaFunctionality.TSS_VOTE;
case TSS_SHARE_SIGNATURE -> HederaFunctionality.TSS_SHARE_SIGNATURE;

Check warning on line 239 in hapi/src/main/java/com/hedera/hapi/util/HapiUtils.java

View check run for this annotation

Codecov / codecov/patch

hapi/src/main/java/com/hedera/hapi/util/HapiUtils.java#L239

Added line #L239 was not covered by tests
case UNSET -> throw new UnknownHederaFunctionality();
};
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ public String getServiceName(@NonNull final TransactionBody txBody) {
};

case NODE_CREATE, NODE_DELETE, NODE_UPDATE -> AddressBookService.NAME;
case TSS_MESSAGE, TSS_VOTE -> TssBaseService.NAME;
case TSS_MESSAGE, TSS_VOTE, TSS_SHARE_SIGNATURE -> TssBaseService.NAME;

default -> NON_EXISTING_SERVICE;
};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
import com.hedera.node.app.spi.AppContext;
import com.hedera.node.app.tss.api.TssLibrary;
import com.hedera.node.app.tss.handlers.TssMessageHandler;
import com.hedera.node.app.tss.handlers.TssShareSignatureHandler;
import com.hedera.node.app.tss.handlers.TssSubmissions;
import com.hedera.node.app.tss.handlers.TssVoteHandler;
import com.swirlds.metrics.api.Metrics;
Expand Down Expand Up @@ -46,5 +47,7 @@ TssBaseServiceComponent create(

TssVoteHandler tssVoteHandler();

TssShareSignatureHandler tssShareSignatureHandler();

TssSubmissions tssSubmissions();
}
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,8 @@ public TssBaseServiceImpl(
final var component = DaggerTssBaseServiceComponent.factory()
.create(tssLibrary, appContext.gossip(), submissionExecutor, tssLibraryExecutor, metrics);
this.tssMetrics = component.tssMetrics();
this.tssHandlers = new TssHandlers(component.tssMessageHandler(), component.tssVoteHandler());
this.tssHandlers = new TssHandlers(
component.tssMessageHandler(), component.tssVoteHandler(), component.tssShareSignatureHandler());
this.tssSubmissions = component.tssSubmissions();
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,9 @@

package com.hedera.node.app.tss.handlers;

import com.hedera.node.app.spi.workflows.TransactionHandler;
import edu.umd.cs.findbugs.annotations.NonNull;

public record TssHandlers(@NonNull TransactionHandler tssMessageHandler, @NonNull TransactionHandler tssVoteHandler) {}
public record TssHandlers(
@NonNull TssMessageHandler tssMessageHandler,
@NonNull TssVoteHandler tssVoteHandler,
@NonNull TssShareSignatureHandler tssShareSignatureHandler) {}
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
/*
* Copyright (C) 2024 Hedera Hashgraph, LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package com.hedera.node.app.tss.handlers;

import static java.util.Objects.requireNonNull;

import com.hedera.hapi.node.transaction.TransactionBody;
import com.hedera.node.app.spi.workflows.HandleContext;
import com.hedera.node.app.spi.workflows.HandleException;
import com.hedera.node.app.spi.workflows.PreCheckException;
import com.hedera.node.app.spi.workflows.PreHandleContext;
import com.hedera.node.app.spi.workflows.TransactionHandler;
import edu.umd.cs.findbugs.annotations.NonNull;
import javax.inject.Inject;
import javax.inject.Singleton;

/**
* Handles TSS share signature transactions.
* This is yet to be implemented.
*/
@Singleton
public class TssShareSignatureHandler implements TransactionHandler {

@Inject
public TssShareSignatureHandler() {}

@Override
public void preHandle(@NonNull final PreHandleContext context) throws PreCheckException {
requireNonNull(context);

Check warning on line 43 in hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java

View check run for this annotation

Codecov / codecov/patch

hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java#L43

Added line #L43 was not covered by tests
// TODO: Implement this in the later PRS
}

Check warning on line 45 in hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java

View check run for this annotation

Codecov / codecov/patch

hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java#L45

Added line #L45 was not covered by tests

@Override
public void pureChecks(@NonNull final TransactionBody txn) throws PreCheckException {
requireNonNull(txn);
}

Check warning on line 50 in hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java

View check run for this annotation

Codecov / codecov/patch

hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java#L49-L50

Added lines #L49 - L50 were not covered by tests

@Override
public void handle(@NonNull final HandleContext context) throws HandleException {
requireNonNull(context);
}

Check warning on line 55 in hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java

View check run for this annotation

Codecov / codecov/patch

hedera-node/hedera-app/src/main/java/com/hedera/node/app/tss/handlers/TssShareSignatureHandler.java#L54-L55

Added lines #L54 - L55 were not covered by tests
}
Original file line number Diff line number Diff line change
Expand Up @@ -224,6 +224,7 @@

case TSS_MESSAGE -> handlers.tssMessageHandler();
case TSS_VOTE -> handlers.tssVoteHandler();
case TSS_SHARE_SIGNATURE -> handlers.tssShareSignatureHandler();

Check warning on line 227 in hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/dispatcher/TransactionDispatcher.java

View check run for this annotation

Codecov / codecov/patch

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/dispatcher/TransactionDispatcher.java#L227

Added line #L227 was not covered by tests

default -> throw new UnsupportedOperationException(TYPE_NOT_SUPPORTED);
};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,9 @@
import com.hedera.node.app.service.token.impl.handlers.TokenUpdateHandler;
import com.hedera.node.app.service.token.impl.handlers.TokenUpdateNftsHandler;
import com.hedera.node.app.service.util.impl.handlers.UtilPrngHandler;
import com.hedera.node.app.spi.workflows.TransactionHandler;
import com.hedera.node.app.tss.handlers.TssMessageHandler;
import com.hedera.node.app.tss.handlers.TssShareSignatureHandler;
import com.hedera.node.app.tss.handlers.TssVoteHandler;
import edu.umd.cs.findbugs.annotations.NonNull;

/**
Expand Down Expand Up @@ -132,5 +134,6 @@ public record TransactionHandlers(
@NonNull NodeDeleteHandler nodeDeleteHandler,
@NonNull TokenClaimAirdropHandler tokenClaimAirdropHandler,
@NonNull UtilPrngHandler utilPrngHandler,
@NonNull TransactionHandler tssMessageHandler,
@NonNull TransactionHandler tssVoteHandler) {}
@NonNull TssMessageHandler tssMessageHandler,
@NonNull TssVoteHandler tssVoteHandler,
@NonNull TssShareSignatureHandler tssShareSignatureHandler) {}
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,7 @@ static TransactionHandlers provideTransactionHandlers(
tokenHandlers.tokenClaimAirdropHandler(),
utilHandlers.prngHandler(),
tssHandlers.get().tssMessageHandler(),
tssHandlers.get().tssVoteHandler());
tssHandlers.get().tssVoteHandler(),
tssHandlers.get().tssShareSignatureHandler());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,12 @@
import com.hedera.node.app.signature.AppSignatureVerifier;
import com.hedera.node.app.signature.impl.SignatureExpanderImpl;
import com.hedera.node.app.signature.impl.SignatureVerifierImpl;
import com.hedera.node.app.spi.workflows.TransactionHandler;
import com.hedera.node.app.state.recordcache.RecordCacheService;
import com.hedera.node.app.tss.TssBaseService;
import com.hedera.node.app.tss.handlers.TssHandlers;
import com.hedera.node.app.tss.handlers.TssMessageHandler;
import com.hedera.node.app.tss.handlers.TssShareSignatureHandler;
import com.hedera.node.app.tss.handlers.TssVoteHandler;
import com.hedera.node.config.data.HederaConfig;
import com.hedera.node.config.testfixtures.HederaTestConfigBuilder;
import com.hedera.pbj.runtime.io.buffer.Bytes;
Expand Down Expand Up @@ -79,7 +81,13 @@ class IngestComponentTest {
private TssBaseService tssBaseService;

@Mock
private TransactionHandler transactionHandler;
private TssMessageHandler tssMessageHandler;

@Mock
private TssVoteHandler tssVoteHandler;

@Mock
private TssShareSignatureHandler tssShareSignatureHandler;

private HederaInjectionComponent app;

Expand All @@ -105,7 +113,8 @@ void setUp() {
new SignatureExpanderImpl(),
new SignatureVerifierImpl(CryptographyHolder.get())),
UNAVAILABLE_GOSSIP);
given(tssBaseService.tssHandlers()).willReturn(new TssHandlers(transactionHandler, transactionHandler));
given(tssBaseService.tssHandlers())
.willReturn(new TssHandlers(tssMessageHandler, tssVoteHandler, tssShareSignatureHandler));
app = DaggerHederaInjectionComponent.builder()
.configProviderImpl(configProvider)
.bootstrapConfigProviderImpl(new BootstrapConfigProviderImpl())
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@
import com.hedera.node.app.service.util.impl.handlers.UtilPrngHandler;
import com.hedera.node.app.tss.handlers.TssHandlers;
import com.hedera.node.app.tss.handlers.TssMessageHandler;
import com.hedera.node.app.tss.handlers.TssShareSignatureHandler;
import com.hedera.node.app.tss.handlers.TssVoteHandler;
import com.hedera.node.app.workflows.dispatcher.TransactionHandlers;
import org.junit.jupiter.api.Test;
Expand Down Expand Up @@ -264,6 +265,9 @@ class HandleWorkflowModuleTest {
@Mock
private TssVoteHandler tssVoteHandler;

@Mock
private TssShareSignatureHandler tssShareSignatureHandler;

@Test
void usesComponentsToGetHandlers() {
given(consensusHandlers.consensusCreateTopicHandler()).willReturn(consensusCreateTopicHandler);
Expand Down Expand Up @@ -321,7 +325,7 @@ void usesComponentsToGetHandlers() {
consensusHandlers,
fileHandlers,
() -> contractHandlers,
() -> new TssHandlers(tssMessageHandler, tssVoteHandler),
() -> new TssHandlers(tssMessageHandler, tssVoteHandler, tssShareSignatureHandler),
scheduleHandlers,
tokenHandlers,
utilHandlers,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,7 @@
import static com.hedera.hapi.node.base.HederaFunctionality.TRANSACTION_GET_RECEIPT;
import static com.hedera.hapi.node.base.HederaFunctionality.TRANSACTION_GET_RECORD;
import static com.hedera.hapi.node.base.HederaFunctionality.TSS_MESSAGE;
import static com.hedera.hapi.node.base.HederaFunctionality.TSS_SHARE_SIGNATURE;
import static com.hedera.hapi.node.base.HederaFunctionality.TSS_VOTE;
import static com.hedera.hapi.node.base.HederaFunctionality.UTIL_PRNG;

Expand Down Expand Up @@ -263,7 +264,8 @@ public record ApiPermissionConfig(
@ConfigProperty(defaultValue = "2-55") PermissionedAccountsRange updateNode,
@ConfigProperty(defaultValue = "2-55") PermissionedAccountsRange deleteNode,
@ConfigProperty(defaultValue = "0-0") PermissionedAccountsRange tssMessage,
@ConfigProperty(defaultValue = "0-0") PermissionedAccountsRange tssVote) {
@ConfigProperty(defaultValue = "0-0") PermissionedAccountsRange tssVote,
@ConfigProperty(defaultValue = "0-0") PermissionedAccountsRange tssShareSignature) {

private static final EnumMap<HederaFunctionality, Function<ApiPermissionConfig, PermissionedAccountsRange>>
permissionKeys = new EnumMap<>(HederaFunctionality.class);
Expand Down Expand Up @@ -343,6 +345,7 @@ public record ApiPermissionConfig(
permissionKeys.put(NODE_DELETE, c -> c.deleteNode);
permissionKeys.put(TSS_MESSAGE, c -> c.tssMessage);
permissionKeys.put(TSS_VOTE, c -> c.tssVote);
permissionKeys.put(TSS_SHARE_SIGNATURE, c -> c.tssShareSignature);
}

/**
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,19 @@

/**
* Configuration for the TSS service.
* @param maxSharesPerNode The maximum number of shares that can be assigned to a node.
* @param timesToTrySubmission The number of times to retry a submission on getting an {@link IllegalStateException}
* @param distinctTxnIdsToTry The number of distinct transaction IDs to try in the event of a duplicate id
* @param keyActiveRoster A test-only configuration; set this to true to enable the process that will key the candidate roster with TSS key material.
*
* @param maxSharesPerNode The maximum number of shares that can be assigned to a node.
* @param timesToTrySubmission The number of times to retry a submission on getting an {@link IllegalStateException}
* @param retryDelay The delay between retries
* @param distinctTxnIdsToTry The number of distinct transaction IDs to try in the event of a duplicate id
* @param keyCandidateRoster A feature flag for TSS; set this to true to enable the process that will key
* @param keyActiveRoster A test-only configuration; set this to true to enable the process that will
* key the candidate roster with TSS key material, without waiting for upgrade
* boundary.
* @param signatureLivenessPeriodMinutes The amount of time a share signature is held in memory before being
* discarded in minutes
* @param ledgerSignatureFailureThreshold The number of consecutive failures to produce a ledger signature before
* logging an error
*/
@ConfigData("tss")
public record TssConfig(
Expand All @@ -36,4 +45,5 @@ public record TssConfig(
@ConfigProperty(defaultValue = "10") @NetworkProperty int distinctTxnIdsToTry,
@ConfigProperty(defaultValue = "false") @NetworkProperty boolean keyCandidateRoster,
@ConfigProperty(defaultValue = "false") @NetworkProperty boolean keyActiveRoster,
@ConfigProperty(defaultValue = "false") @NetworkProperty boolean enableLedgerId) {}
@ConfigProperty(defaultValue = "5") @NetworkProperty int signatureLivenessPeriodMinutes,
@ConfigProperty(defaultValue = "2") @NetworkProperty int ledgerSignatureFailureThreshold) {}
Loading