MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add fail-fast HMAC init to RegistrationServiceClient constructor

This commit is contained in:
Chris Eager 2025-01-22 18:04:26 -06:00 committed by Chris Eager
parent 47550d48e7
commit c9e192564c
1 changed files with 30 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
service/src/main/java/org/whispersystems/textsecuregcm/registration/RegistrationServiceClient.java
View File

@ -14,6 +14,7 @@ import io.grpc.TlsChannelCredentials;
import java.io.ByteArrayInputStream; import java.io.ByteArrayInputStream;
import java.io.IOException; import java.io.IOException;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException; import java.security.NoSuchAlgorithmException;
import java.time.Duration; import java.time.Duration;
import java.util.Base64; import java.util.Base64;
@ -23,6 +24,7 @@ import java.util.concurrent.CompletionException;
import java.util.concurrent.Executor; import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import javax.crypto.Mac; import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.checkerframework.checker.nullness.qual.Nullable; import org.checkerframework.checker.nullness.qual.Nullable;
import org.signal.registration.rpc.CheckVerificationCodeRequest; import org.signal.registration.rpc.CheckVerificationCodeRequest;
@ -82,6 +84,13 @@ public class RegistrationServiceClient implements Managed {
this.stub = RegistrationServiceGrpc.newFutureStub(channel).withCallCredentials(callCredentials); this.stub = RegistrationServiceGrpc.newFutureStub(channel).withCallCredentials(callCredentials);
this.collationKeySalt = collationKeySalt; this.collationKeySalt = collationKeySalt;
this.callbackExecutor = callbackExecutor; this.callbackExecutor = callbackExecutor;
// Fail fast: reject bad keys
try {
getInitializedMac(collationKeySalt);
} catch (final InvalidKeyException e) {
throw new IllegalArgumentException(e);
}
} }
public CompletableFuture<RegistrationServiceSession> createRegistrationSession( public CompletableFuture<RegistrationServiceSession> createRegistrationSession(
@ -89,7 +98,7 @@ public class RegistrationServiceClient implements Managed {
final long e164 = Long.parseLong( final long e164 = Long.parseLong(
PhoneNumberUtil.getInstance().format(phoneNumber, PhoneNumberUtil.PhoneNumberFormat.E164).substring(1)); PhoneNumberUtil.getInstance().format(phoneNumber, PhoneNumberUtil.PhoneNumberFormat.E164).substring(1));
final String rateLimitCollationKey = hmac(sourceHost, collationKeySalt); final String rateLimitCollationKey = hmac(sourceHost);
return CompletableFutureUtil.toCompletableFuture(stub.withDeadline(toDeadline(timeout)) return CompletableFutureUtil.toCompletableFuture(stub.withDeadline(toDeadline(timeout))
.createSession(CreateRegistrationSessionRequest.newBuilder() .createSession(CreateRegistrationSessionRequest.newBuilder()
@ -270,17 +279,29 @@ public class RegistrationServiceClient implements Managed {
} }
} }
private static String hmac(String sourceHost, byte[] collationKeySalt) { private String hmac(String sourceHost) {
final Mac hmacSha256; final Mac hmacSha256 = getInitializedMac();
hmacSha256.update(sourceHost.getBytes(StandardCharsets.UTF_8));
return BASE64_UNPADDED_ENCODER.encodeToString(hmacSha256.doFinal());
}
private Mac getInitializedMac() {
try { try {
hmacSha256 = Mac.getInstance("HmacSHA256"); return getInitializedMac(collationKeySalt);
} catch (final InvalidKeyException e) {
throw new AssertionError(e);
}
}
private static Mac getInitializedMac(byte[] key) throws InvalidKeyException {
try {
final Mac hmacSha256 = Mac.getInstance("HmacSHA256");
hmacSha256.init(new SecretKeySpec(key, "HmacSHA256"));
return hmacSha256;
} catch (NoSuchAlgorithmException e) { } catch (NoSuchAlgorithmException e) {
throw new AssertionError(e); throw new AssertionError(e);
} }
hmacSha256.update(sourceHost.getBytes(StandardCharsets.UTF_8));
hmacSha256.update(collationKeySalt);
return BASE64_UNPADDED_ENCODER.encodeToString(hmacSha256.doFinal());
} }
} }