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.MutuallyExclusiveGroup;
   5 import net.sourceforge.argparse4j.inf.Namespace;
   6 import net.sourceforge.argparse4j.inf.Subparser;
   7
   8 import org.asamk.signal.manager.Manager;
   9 import org.asamk.signal.util.Hex;
  10
  11 import java.util.Locale;
  12
  13 public class TrustCommand implements LocalCommand {
  14
  15     @Override
  16     public void attachToSubparser(final Subparser subparser) {
  17         subparser.addArgument("number")
  18                 .help("Specify the phone number, for which to set the trust.")
  19                 .required(true);
  20         MutuallyExclusiveGroup mutTrust = subparser.addMutuallyExclusiveGroup();
  21         mutTrust.addArgument("-a", "--trust-all-known-keys")
  22                 .help("Trust all known keys of this user, only use this for testing.")
  23                 .action(Arguments.storeTrue());
  24         mutTrust.addArgument("-v", "--verified-fingerprint")
  25                 .help("Specify the fingerprint of the key, only use this option if you have verified the fingerprint.");
  26     }
  27
  28     @Override
  29     public int handleCommand(final Namespace ns, final Manager m) {
  30         if (!m.isRegistered()) {
  31             System.err.println("User is not registered.");
  32             return 1;
  33         }
  34         String number = ns.getString("number");
  35         if (ns.getBoolean("trust_all_known_keys")) {
  36             boolean res = m.trustIdentityAllKeys(number);
  37             if (!res) {
  38                 System.err.println("Failed to set the trust for this number, make sure the number is correct.");
  39                 return 1;
  40             }
  41         } else {
  42             String fingerprint = ns.getString("verified_fingerprint");
  43             if (fingerprint != null) {
  44                 fingerprint = fingerprint.replaceAll(" ", "");
  45                 if (fingerprint.length() == 66) {
  46                     byte[] fingerprintBytes;
  47                     try {
  48                         fingerprintBytes = Hex.toByteArray(fingerprint.toLowerCase(Locale.ROOT));
  49                     } catch (Exception e) {
  50                         System.err.println("Failed to parse the fingerprint, make sure the fingerprint is a correctly encoded hex string without additional characters.");
  51                         return 1;
  52                     }
  53                     boolean res = m.trustIdentityVerified(number, fingerprintBytes);
  54                     if (!res) {
  55                         System.err.println("Failed to set the trust for the fingerprint of this number, make sure the number and the fingerprint are correct.");
  56                         return 1;
  57                     }
  58                 } else if (fingerprint.length() == 60) {
  59                     boolean res = m.trustIdentityVerifiedSafetyNumber(number, fingerprint);
  60                     if (!res) {
  61                         System.err.println("Failed to set the trust for the safety number of this phone number, make sure the phone number and the safety number are correct.");
  62                         return 1;
  63                     }
  64                 } else {
  65                     System.err.println("Fingerprint has invalid format, either specify the old hex fingerprint or the new safety number");
  66                     return 1;
  67                 }
  68             } else {
  69                 System.err.println("You need to specify the fingerprint you have verified with -v FINGERPRINT");
  70                 return 1;
  71             }
  72         }
  73         return 0;
  74     }
  75 }