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

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.impl.Arguments;
   4 import net.sourceforge.argparse4j.inf.Namespace;
   5 import net.sourceforge.argparse4j.inf.Subparser;
   6
   7 import org.asamk.signal.OutputWriter;
   8 import org.asamk.signal.commands.exceptions.CommandException;
   9 import org.asamk.signal.commands.exceptions.UserErrorException;
  10 import org.asamk.signal.manager.Manager;
  11 import org.asamk.signal.util.Hex;
  12 import org.whispersystems.signalservice.api.util.InvalidNumberException;
  13
  14 import java.util.Base64;
  15 import java.util.Locale;
  16
  17 public class TrustCommand implements JsonRpcLocalCommand {
  18
  19     @Override
  20     public String getName() {
  21         return "trust";
  22     }
  23
  24     @Override
  25     public void attachToSubparser(final Subparser subparser) {
  26         subparser.help("Set the trust level of a given number.");
  27         subparser.addArgument("number").help("Specify the phone number, for which to set the trust.").required(true);
  28         var mutTrust = subparser.addMutuallyExclusiveGroup();
  29         mutTrust.addArgument("-a", "--trust-all-known-keys")
  30                 .help("Trust all known keys of this user, only use this for testing.")
  31                 .action(Arguments.storeTrue());
  32         mutTrust.addArgument("-v", "--verified-safety-number", "--verified-fingerprint")
  33                 .help("Specify the safety number of the key, only use this option if you have verified the safety number.");
  34     }
  35
  36     @Override
  37     public void handleCommand(
  38             final Namespace ns, final Manager m, final OutputWriter outputWriter
  39     ) throws CommandException {
  40         var number = ns.getString("number");
  41         if (ns.getBoolean("trust-all-known-keys")) {
  42             boolean res;
  43             try {
  44                 res = m.trustIdentityAllKeys(number);
  45             } catch (InvalidNumberException e) {
  46                 throw new UserErrorException("Failed to parse recipient: " + e.getMessage());
  47             }
  48             if (!res) {
  49                 throw new UserErrorException("Failed to set the trust for this number, make sure the number is correct.");
  50             }
  51         } else {
  52             var safetyNumber = ns.getString("verified-safety-number");
  53             if (safetyNumber == null) {
  54                 throw new UserErrorException(
  55                         "You need to specify the fingerprint/safety number you have verified with -v SAFETY_NUMBER");
  56             }
  57
  58             safetyNumber = safetyNumber.replaceAll(" ", "");
  59             if (safetyNumber.length() == 66) {
  60                 byte[] fingerprintBytes;
  61                 try {
  62                     fingerprintBytes = Hex.toByteArray(safetyNumber.toLowerCase(Locale.ROOT));
  63                 } catch (Exception e) {
  64                     throw new UserErrorException(
  65                             "Failed to parse the fingerprint, make sure the fingerprint is a correctly encoded hex string without additional characters.");
  66                 }
  67                 boolean res;
  68                 try {
  69                     res = m.trustIdentityVerified(number, fingerprintBytes);
  70                 } catch (InvalidNumberException e) {
  71                     throw new UserErrorException("Failed to parse recipient: " + e.getMessage());
  72                 }
  73                 if (!res) {
  74                     throw new UserErrorException(
  75                             "Failed to set the trust for the fingerprint of this number, make sure the number and the fingerprint are correct.");
  76                 }
  77             } else if (safetyNumber.length() == 60) {
  78                 boolean res;
  79                 try {
  80                     res = m.trustIdentityVerifiedSafetyNumber(number, safetyNumber);
  81                 } catch (InvalidNumberException e) {
  82                     throw new UserErrorException("Failed to parse recipient: " + e.getMessage());
  83                 }
  84                 if (!res) {
  85                     throw new UserErrorException(
  86                             "Failed to set the trust for the safety number of this phone number, make sure the phone number and the safety number are correct.");
  87                 }
  88             } else {
  89                 final byte[] scannableSafetyNumber;
  90                 try {
  91                     scannableSafetyNumber = Base64.getDecoder().decode(safetyNumber);
  92                 } catch (IllegalArgumentException e) {
  93                     throw new UserErrorException(
  94                             "Safety number has invalid format, either specify the old hex fingerprint or the new safety number");
  95                 }
  96                 boolean res;
  97                 try {
  98                     res = m.trustIdentityVerifiedSafetyNumber(number, scannableSafetyNumber);
  99                 } catch (InvalidNumberException e) {
 100                     throw new UserErrorException("Failed to parse recipient: " + e.getMessage());
 101                 }
 102                 if (!res) {
 103                     throw new UserErrorException(
 104                             "Failed to set the trust for the safety number of this phone number, make sure the phone number and the safety number are correct.");
 105                 }
 106             }
 107         }
 108     }
 109 }