Retire ProvisioningAddress and WebsocketAddress

service/src/main/java/org/whispersystems/textsecuregcm/controllers/ProvisioningController.java

@@ -34,7 +34,6 @@ import org.whispersystems.textsecuregcm.entities.ProvisioningMessage;
 import org.whispersystems.textsecuregcm.limits.RateLimiters;
 import org.whispersystems.textsecuregcm.metrics.UserAgentTagUtil;
 import org.whispersystems.textsecuregcm.push.ProvisioningManager;
-import org.whispersystems.textsecuregcm.websocket.ProvisioningAddress;
 import org.whispersystems.websocket.auth.ReadOnly;
 
 /**
@@ -96,8 +95,10 @@ public class ProvisioningController {
 
     rateLimiters.getMessagesLimiter().validate(auth.getAccount().getUuid());
 
-    if (!provisioningManager.sendProvisioningMessage(ProvisioningAddress.create(provisioningAddress),
-        Base64.getMimeDecoder().decode(message.body()))) {
+    final boolean subscriberPresent =
+        provisioningManager.sendProvisioningMessage(provisioningAddress, Base64.getMimeDecoder().decode(message.body()));
+
+    if (!subscriberPresent) {
       throw new WebApplicationException(Response.Status.NOT_FOUND);
     }
   }

service/src/main/java/org/whispersystems/textsecuregcm/push/ProvisioningManager.java

@@ -28,8 +28,6 @@ import org.whispersystems.textsecuregcm.configuration.CircuitBreakerConfiguratio
 import org.whispersystems.textsecuregcm.redis.RedisOperation;
 import org.whispersystems.textsecuregcm.storage.PubSubProtos;
 import org.whispersystems.textsecuregcm.util.CircuitBreakerUtil;
-import org.whispersystems.textsecuregcm.websocket.InvalidWebsocketAddressException;
-import org.whispersystems.textsecuregcm.websocket.ProvisioningAddress;
 
 public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> implements Managed {
 
@@ -39,7 +37,7 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
 
   private final CircuitBreaker circuitBreaker;
 
-  private final Map<ProvisioningAddress, Consumer<PubSubProtos.PubSubMessage>> listenersByProvisioningAddress =
+  private final Map<String, Consumer<PubSubProtos.PubSubMessage>> listenersByProvisioningAddress =
       new ConcurrentHashMap<>();
 
   private static final String ACTIVE_LISTENERS_GAUGE_NAME = name(ProvisioningManager.class, "activeListeners");
@@ -82,21 +80,21 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
     redisClient.shutdown();
   }
 
-  public void addListener(final ProvisioningAddress address, final Consumer<PubSubProtos.PubSubMessage> listener) {
+  public void addListener(final String address, final Consumer<PubSubProtos.PubSubMessage> listener) {
     listenersByProvisioningAddress.put(address, listener);
 
     circuitBreaker.executeRunnable(
-        () -> subscriptionConnection.sync().subscribe(address.serialize().getBytes(StandardCharsets.UTF_8)));
+        () -> subscriptionConnection.sync().subscribe(address.getBytes(StandardCharsets.UTF_8)));
   }
 
-  public void removeListener(final ProvisioningAddress address) {
+  public void removeListener(final String address) {
     RedisOperation.unchecked(() -> circuitBreaker.executeRunnable(
-        () -> subscriptionConnection.sync().unsubscribe(address.serialize().getBytes(StandardCharsets.UTF_8))));
+        () -> subscriptionConnection.sync().unsubscribe(address.getBytes(StandardCharsets.UTF_8))));
 
     listenersByProvisioningAddress.remove(address);
   }
 
-  public boolean sendProvisioningMessage(final ProvisioningAddress address, final byte[] body) {
+  public boolean sendProvisioningMessage(final String address, final byte[] body) {
     final PubSubProtos.PubSubMessage pubSubMessage = PubSubProtos.PubSubMessage.newBuilder()
         .setType(PubSubProtos.PubSubMessage.Type.DELIVER)
         .setContent(ByteString.copyFrom(body))
@@ -104,7 +102,7 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
 
     final boolean receiverPresent = circuitBreaker.executeSupplier(
         () -> publicationConnection.sync()
-            .publish(address.serialize().getBytes(StandardCharsets.UTF_8), pubSubMessage.toByteArray()) > 0);
+            .publish(address.getBytes(StandardCharsets.UTF_8), pubSubMessage.toByteArray()) > 0);
 
     Metrics.counter(SEND_PROVISIONING_MESSAGE_COUNTER_NAME, "online", String.valueOf(receiverPresent)).increment();
 
@@ -114,7 +112,7 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
   @Override
   public void message(final byte[] channel, final byte[] message) {
     try {
-      final ProvisioningAddress address = new ProvisioningAddress(new String(channel, StandardCharsets.UTF_8));
+      final String address = new String(channel, StandardCharsets.UTF_8);
       final PubSubProtos.PubSubMessage pubSubMessage = PubSubProtos.PubSubMessage.parseFrom(message);
 
       if (pubSubMessage.getType() == PubSubProtos.PubSubMessage.Type.DELIVER) {
@@ -129,8 +127,6 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
 
         Metrics.counter(RECEIVE_PROVISIONING_MESSAGE_COUNTER_NAME, "listenerPresent", String.valueOf(listenerPresent)).increment();
       }
-    } catch (final InvalidWebsocketAddressException e) {
-      logger.warn("Failed to parse provisioning address", e);
     } catch (final InvalidProtocolBufferException e) {
       logger.warn("Failed to parse pub/sub message", e);
     }
@@ -138,10 +134,6 @@ public class ProvisioningManager extends RedisPubSubAdapter<byte[], byte[]> impl
 
   @Override
   public void unsubscribed(final byte[] channel, final long count) {
-    try {
-      listenersByProvisioningAddress.remove(new ProvisioningAddress(new String(channel)));
-    } catch (final InvalidWebsocketAddressException e) {
-      logger.warn("Failed to parse provisioning address for `unsubscribe` event", e);
-    }
+    listenersByProvisioningAddress.remove(new String(channel, StandardCharsets.UTF_8));
   }
 }

service/src/main/java/org/whispersystems/textsecuregcm/storage/PubSubAddress.java

@@ -1,11 +0,0 @@
-/*
- * Copyright 2013-2020 Signal Messenger, LLC
- * SPDX-License-Identifier: AGPL-3.0-only
- */
-
-package org.whispersystems.textsecuregcm.storage;
-
-public interface PubSubAddress {
-
-  String serialize();
-}

service/src/main/java/org/whispersystems/textsecuregcm/websocket/ProvisioningAddress.java

@@ -1,37 +0,0 @@
-/*
- * Copyright 2013-2020 Signal Messenger, LLC
- * SPDX-License-Identifier: AGPL-3.0-only
- */
-
-package org.whispersystems.textsecuregcm.websocket;
-
-import java.security.SecureRandom;
-import java.util.Base64;
-
-public class ProvisioningAddress extends WebsocketAddress {
-
-  public static byte DEVICE_ID = 0;
-
-  public static ProvisioningAddress create(String address) {
-    return new ProvisioningAddress(address, DEVICE_ID);
-  }
-
-  private ProvisioningAddress(String address, byte deviceId) {
-    super(address, deviceId);
-  }
-
-  public ProvisioningAddress(String serialized) throws InvalidWebsocketAddressException {
-    super(serialized);
-  }
-
-  public String getAddress() {
-    return getNumber();
-  }
-
-  public static ProvisioningAddress generate() {
-    byte[] random = new byte[16];
-    new SecureRandom().nextBytes(random);
-
-    return create(Base64.getUrlEncoder().withoutPadding().encodeToString(random));
-  }
-}

service/src/main/java/org/whispersystems/textsecuregcm/websocket/ProvisioningConnectListener.java

@@ -5,6 +5,7 @@
 
 package org.whispersystems.textsecuregcm.websocket;
 
+import com.google.common.annotations.VisibleForTesting;
 import org.whispersystems.textsecuregcm.auth.AuthenticatedDevice;
 import org.whispersystems.textsecuregcm.entities.MessageProtos;
 import org.whispersystems.textsecuregcm.entities.ProvisioningMessage;
@@ -13,8 +14,11 @@ import org.whispersystems.textsecuregcm.storage.PubSubProtos;
 import org.whispersystems.textsecuregcm.util.HeaderUtils;
 import org.whispersystems.websocket.session.WebSocketSessionContext;
 import org.whispersystems.websocket.setup.WebSocketConnectListener;
+import java.security.SecureRandom;
+import java.util.Base64;
 import java.util.List;
 import java.util.Optional;
+import java.util.UUID;
 
 /**
  * A "provisioning WebSocket" provides a mechanism for sending a caller-defined provisioning message from the primary
@@ -40,7 +44,7 @@ public class ProvisioningConnectListener implements WebSocketConnectListener {
 
   @Override
   public void onWebSocketConnect(WebSocketSessionContext context) {
-    final ProvisioningAddress provisioningAddress = ProvisioningAddress.generate();
+    final String provisioningAddress = UUID.randomUUID().toString();
     context.addWebsocketClosedListener((context1, statusCode, reason) -> provisioningManager.removeListener(provisioningAddress));
 
     provisioningManager.addListener(provisioningAddress, message -> {
@@ -53,9 +57,15 @@ public class ProvisioningConnectListener implements WebSocketConnectListener {
     });
 
     context.getClient().sendRequest("PUT", "/v1/address", List.of(HeaderUtils.getTimestampHeader()),
-        Optional.of(MessageProtos.ProvisioningUuid.newBuilder()
-            .setUuid(provisioningAddress.getAddress())
-            .build()
-            .toByteArray()));
+        Optional.of(generateProvisioningUuid().toByteArray()));
+  }
+
+  private static MessageProtos.ProvisioningUuid generateProvisioningUuid() {
+    final byte[] provisioningAddress = new byte[16];
+    new SecureRandom().nextBytes(provisioningAddress);
+
+    return MessageProtos.ProvisioningUuid.newBuilder()
+        .setUuid(Base64.getUrlEncoder().encodeToString(provisioningAddress))
+        .build();
   }
 }

service/src/main/java/org/whispersystems/textsecuregcm/websocket/WebsocketAddress.java

@@ -1,68 +0,0 @@
-/*
- * Copyright 2013-2020 Signal Messenger, LLC
- * SPDX-License-Identifier: AGPL-3.0-only
- */
-
-package org.whispersystems.textsecuregcm.websocket;
-
-import org.whispersystems.textsecuregcm.storage.PubSubAddress;
-
-public class WebsocketAddress implements PubSubAddress {
-
-  private final String number;
-  private final byte deviceId;
-
-  public WebsocketAddress(String number, byte deviceId) {
-    this.number    = number;
-    this.deviceId  = deviceId;
-  }
-
-  public WebsocketAddress(String serialized) throws InvalidWebsocketAddressException {
-    try {
-      String[] parts = serialized.split(":", 2);
-
-      if (parts.length != 2) {
-        throw new InvalidWebsocketAddressException("Bad address: " + serialized);
-      }
-
-      this.number   = parts[0];
-      this.deviceId = Byte.parseByte(parts[1]);
-    } catch (NumberFormatException e) {
-      throw new InvalidWebsocketAddressException(e);
-    }
-  }
-
-  public String getNumber() {
-    return number;
-  }
-
-  public byte getDeviceId() {
-    return deviceId;
-  }
-
-  public String serialize() {
-    return number + ":" + deviceId;
-  }
-
-  public String toString() {
-    return serialize();
-  }
-
-  @Override
-  public boolean equals(Object other) {
-    if (other == null) return false;
-    if (!(other instanceof WebsocketAddress)) return false;
-
-    WebsocketAddress that = (WebsocketAddress)other;
-
-    return
-        this.number.equals(that.number) &&
-        this.deviceId == that.deviceId;
-  }
-
-  @Override
-  public int hashCode() {
-    return number.hashCode() ^ (int)deviceId;
-  }
-
-}