nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/manager/DeviceLinkInfo.java

   1 package org.asamk.signal.manager;
   2
   3 import org.whispersystems.libsignal.InvalidKeyException;
   4 import org.whispersystems.libsignal.ecc.Curve;
   5 import org.whispersystems.libsignal.ecc.ECPublicKey;
   6 import org.whispersystems.util.Base64;
   7
   8 import java.io.IOException;
   9 import java.net.URI;
  10 import java.net.URLDecoder;
  11 import java.net.URLEncoder;
  12 import java.nio.charset.StandardCharsets;
  13 import java.util.HashMap;
  14 import java.util.Map;
  15
  16 import static org.whispersystems.signalservice.internal.util.Util.isEmpty;
  17
  18 public class DeviceLinkInfo {
  19
  20     final String deviceIdentifier;
  21     final ECPublicKey deviceKey;
  22
  23     public static DeviceLinkInfo parseDeviceLinkUri(URI linkUri) throws IOException, InvalidKeyException {
  24         Map<String, String> query = getQueryMap(linkUri.getRawQuery());
  25         String deviceIdentifier = query.get("uuid");
  26         String publicKeyEncoded = query.get("pub_key");
  27
  28         if (isEmpty(deviceIdentifier) || isEmpty(publicKeyEncoded)) {
  29             throw new RuntimeException("Invalid device link uri");
  30         }
  31
  32         ECPublicKey deviceKey = Curve.decodePoint(Base64.decode(publicKeyEncoded), 0);
  33
  34         return new DeviceLinkInfo(deviceIdentifier, deviceKey);
  35     }
  36
  37     private static Map<String, String> getQueryMap(String query) {
  38         String[] params = query.split("&");
  39         Map<String, String> map = new HashMap<>();
  40         for (String param : params) {
  41             final String[] paramParts = param.split("=");
  42             String name = URLDecoder.decode(paramParts[0], StandardCharsets.UTF_8);
  43             String value = URLDecoder.decode(paramParts[1], StandardCharsets.UTF_8);
  44             map.put(name, value);
  45         }
  46         return map;
  47     }
  48
  49     public DeviceLinkInfo(final String deviceIdentifier, final ECPublicKey deviceKey) {
  50         this.deviceIdentifier = deviceIdentifier;
  51         this.deviceKey = deviceKey;
  52     }
  53
  54     public String createDeviceLinkUri() {
  55         return "tsdevice:/?uuid="
  56                 + URLEncoder.encode(deviceIdentifier, StandardCharsets.UTF_8)
  57                 + "&pub_key="
  58                 + URLEncoder.encode(Base64.encodeBytesWithoutPadding(deviceKey.serialize()), StandardCharsets.UTF_8);
  59     }
  60 }