X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/af8a27e87f7844e733d1b42419c3976f0b41ae58..f60a10eb6e40921c32045c1e38843d6b87f3d274:/src/main/java/org/asamk/signal/Main.java diff --git a/src/main/java/org/asamk/signal/Main.java b/src/main/java/org/asamk/signal/Main.java index d562c4e1..a0820f24 100644 --- a/src/main/java/org/asamk/signal/Main.java +++ b/src/main/java/org/asamk/signal/Main.java @@ -1,59 +1,53 @@ -/** - * Copyright (C) 2015 AsamK - * - * This program is free software: you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation, either version 3 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program. If not, see . +/* + Copyright (C) 2015-2018 AsamK + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . */ package org.asamk.signal; import net.sourceforge.argparse4j.ArgumentParsers; import net.sourceforge.argparse4j.impl.Arguments; import net.sourceforge.argparse4j.inf.*; -import org.apache.commons.io.IOUtils; import org.apache.http.util.TextUtils; import org.asamk.Signal; +import org.asamk.signal.commands.*; +import org.asamk.signal.manager.BaseConfig; +import org.asamk.signal.manager.Manager; import org.freedesktop.dbus.DBusConnection; import org.freedesktop.dbus.exceptions.DBusException; -import org.freedesktop.dbus.exceptions.DBusExecutionException; -import org.whispersystems.signalservice.api.crypto.UntrustedIdentityException; -import org.whispersystems.signalservice.api.messages.*; -import org.whispersystems.signalservice.api.messages.multidevice.SignalServiceSyncMessage; -import org.whispersystems.signalservice.api.push.SignalServiceAddress; -import org.whispersystems.signalservice.api.push.exceptions.EncapsulatedExceptions; -import org.whispersystems.signalservice.api.push.exceptions.NetworkFailureException; -import org.whispersystems.signalservice.api.push.exceptions.UnregisteredUserException; import org.whispersystems.signalservice.api.util.PhoneNumberFormatter; import java.io.File; -import java.io.IOException; import java.security.Security; -import java.util.ArrayList; -import java.util.List; +import java.util.Map; public class Main { - public static final String SIGNAL_BUSNAME = "org.asamk.Signal"; - public static final String SIGNAL_OBJECTPATH = "/org/asamk/Signal"; - public static void main(String[] args) { // Workaround for BKS truststore - Security.insertProviderAt(new org.spongycastle.jce.provider.BouncyCastleProvider(), 1); + Security.insertProviderAt(new org.bouncycastle.jce.provider.BouncyCastleProvider(), 1); Namespace ns = parseArgs(args); if (ns == null) { System.exit(1); } + int res = handleCommands(ns); + System.exit(res); + } + + private static int handleCommands(Namespace ns) { final String username = ns.getString("username"); Manager m; Signal ts; @@ -69,16 +63,18 @@ public class Main { busType = DBusConnection.SESSION; } dBusConn = DBusConnection.getConnection(busType); - ts = (Signal) dBusConn.getRemoteObject( - SIGNAL_BUSNAME, SIGNAL_OBJECTPATH, + ts = dBusConn.getRemoteObject( + DbusConfig.SIGNAL_BUSNAME, DbusConfig.SIGNAL_OBJECTPATH, Signal.class); + } catch (UnsatisfiedLinkError e) { + System.err.println("Missing native library dependency for dbus service: " + e.getMessage()); + return 1; } catch (DBusException e) { e.printStackTrace(); if (dBusConn != null) { dBusConn.disconnect(); } - System.exit(3); - return; + return 3; } } else { String settingsPath = ns.getString("config"); @@ -94,241 +90,40 @@ public class Main { m = new Manager(username, settingsPath); ts = m; - if (m.userExists()) { - try { - m.load(); - } catch (Exception e) { - System.err.println("Error loading state file \"" + m.getFileName() + "\": " + e.getMessage()); - System.exit(2); - return; - } + try { + m.init(); + } catch (Exception e) { + System.err.println("Error loading state file: " + e.getMessage()); + return 2; } } - switch (ns.getString("command")) { - case "register": - if (dBusConn != null) { - System.err.println("register is not yet implementd via dbus"); - System.exit(1); - } - if (!m.userHasKeys()) { - m.createNewIdentity(); - } - try { - m.register(ns.getBoolean("voice")); - } catch (IOException e) { - System.err.println("Request verify error: " + e.getMessage()); - System.exit(3); - } - break; - case "verify": - if (dBusConn != null) { - System.err.println("verify is not yet implementd via dbus"); - System.exit(1); - } - if (!m.userHasKeys()) { - System.err.println("User has no keys, first call register."); - System.exit(1); - } - if (m.isRegistered()) { - System.err.println("User registration is already verified"); - System.exit(1); - } - try { - m.verifyAccount(ns.getString("verificationCode")); - } catch (IOException e) { - System.err.println("Verify error: " + e.getMessage()); - System.exit(3); - } - break; - case "send": - if (dBusConn == null && !m.isRegistered()) { - System.err.println("User is not registered."); - System.exit(1); - } + String commandKey = ns.getString("command"); + final Map commands = Commands.getCommands(); + if (commands.containsKey(commandKey)) { + Command command = commands.get(commandKey); - if (ns.getBoolean("endsession")) { - if (ns.getList("recipient") == null) { - System.err.println("No recipients given"); - System.err.println("Aborting sending."); - System.exit(1); - } - try { - ts.sendEndSessionMessage(ns.getList("recipient")); - } catch (IOException e) { - handleIOException(e); - } catch (EncapsulatedExceptions e) { - handleEncapsulatedExceptions(e); - } catch (AssertionError e) { - handleAssertionError(e); - } catch (DBusExecutionException e) { - handleDBusExecutionException(e); - } + if (dBusConn != null) { + if (command instanceof ExtendedDbusCommand) { + return ((ExtendedDbusCommand) command).handleCommand(ns, ts, dBusConn); + } else if (command instanceof DbusCommand) { + return ((DbusCommand) command).handleCommand(ns, ts); } else { - String messageText = ns.getString("message"); - if (messageText == null) { - try { - messageText = IOUtils.toString(System.in); - } catch (IOException e) { - System.err.println("Failed to read message from stdin: " + e.getMessage()); - System.err.println("Aborting sending."); - System.exit(1); - } - } - - try { - List attachments = ns.getList("attachment"); - if (attachments == null) { - attachments = new ArrayList<>(); - } - if (ns.getString("group") != null) { - byte[] groupId = decodeGroupId(ns.getString("group")); - ts.sendGroupMessage(messageText, attachments, groupId); - } else { - ts.sendMessage(messageText, attachments, ns.getList("recipient")); - } - } catch (IOException e) { - handleIOException(e); - } catch (EncapsulatedExceptions e) { - handleEncapsulatedExceptions(e); - } catch (AssertionError e) { - handleAssertionError(e); - } catch (GroupNotFoundException e) { - handleGroupNotFoundException(e); - } catch (AttachmentInvalidException e) { - System.err.println("Failed to add attachment: " + e.getMessage()); - System.err.println("Aborting sending."); - System.exit(1); - } catch (DBusExecutionException e) { - handleDBusExecutionException(e); - } - } - - break; - case "receive": - if (dBusConn != null) { - System.err.println("receive is not yet implementd via dbus"); - System.exit(1); - } - if (!m.isRegistered()) { - System.err.println("User is not registered."); - System.exit(1); - } - int timeout = 5; - if (ns.getInt("timeout") != null) { - timeout = ns.getInt("timeout"); - } - boolean returnOnTimeout = true; - if (timeout < 0) { - returnOnTimeout = false; - timeout = 3600; - } - try { - m.receiveMessages(timeout, returnOnTimeout, new ReceiveMessageHandler(m)); - } catch (IOException e) { - System.err.println("Error while receiving messages: " + e.getMessage()); - System.exit(3); - } catch (AssertionError e) { - handleAssertionError(e); - } - break; - case "quitGroup": - if (dBusConn != null) { - System.err.println("quitGroup is not yet implementd via dbus"); - System.exit(1); + System.err.println(commandKey + " is not yet implemented via dbus"); + return 1; } - if (!m.isRegistered()) { - System.err.println("User is not registered."); - System.exit(1); - } - - try { - m.sendQuitGroupMessage(decodeGroupId(ns.getString("group"))); - } catch (IOException e) { - handleIOException(e); - } catch (EncapsulatedExceptions e) { - handleEncapsulatedExceptions(e); - } catch (AssertionError e) { - handleAssertionError(e); - } catch (GroupNotFoundException e) { - handleGroupNotFoundException(e); - } - - break; - case "updateGroup": - if (dBusConn != null) { - System.err.println("updateGroup is not yet implementd via dbus"); - System.exit(1); - } - if (!m.isRegistered()) { - System.err.println("User is not registered."); - System.exit(1); - } - - try { - byte[] groupId = null; - if (ns.getString("group") != null) { - groupId = decodeGroupId(ns.getString("group")); - } - byte[] newGroupId = m.sendUpdateGroupMessage(groupId, ns.getString("name"), ns.getList("member"), ns.getString("avatar")); - if (groupId == null) { - System.out.println("Creating new group \"" + Base64.encodeBytes(newGroupId) + "\" …"); - } - } catch (IOException e) { - handleIOException(e); - } catch (AttachmentInvalidException e) { - System.err.println("Failed to add avatar attachment for group\": " + e.getMessage()); - System.err.println("Aborting sending."); - System.exit(1); - } catch (GroupNotFoundException e) { - handleGroupNotFoundException(e); - } catch (EncapsulatedExceptions e) { - handleEncapsulatedExceptions(e); - } - - break; - case "daemon": - if (dBusConn != null) { - System.err.println("Stop it."); - System.exit(1); - } - if (!m.isRegistered()) { - System.err.println("User is not registered."); - System.exit(1); - } - DBusConnection conn = null; - try { - try { - int busType; - if (ns.getBoolean("system")) { - busType = DBusConnection.SYSTEM; - } else { - busType = DBusConnection.SESSION; - } - conn = DBusConnection.getConnection(busType); - conn.exportObject(SIGNAL_OBJECTPATH, m); - conn.requestBusName(SIGNAL_BUSNAME); - } catch (DBusException e) { - e.printStackTrace(); - System.exit(3); - } - try { - m.receiveMessages(3600, false, new DbusReceiveMessageHandler(m, conn)); - } catch (IOException e) { - System.err.println("Error while receiving messages: " + e.getMessage()); - System.exit(3); - } catch (AssertionError e) { - handleAssertionError(e); - } - } finally { - if (conn != null) { - conn.disconnect(); - } + } else { + if (command instanceof LocalCommand) { + return ((LocalCommand) command).handleCommand(ns, m); + } else if (command instanceof DbusCommand) { + return ((DbusCommand) command).handleCommand(ns, ts); + } else { + System.err.println(commandKey + " is only works via dbus"); + return 1; } - - break; + } } - System.exit(0); + return 0; } finally { if (dBusConn != null) { dBusConn.disconnect(); @@ -336,40 +131,18 @@ public class Main { } } - private static void handleGroupNotFoundException(GroupNotFoundException e) { - System.err.println("Failed to send to group: " + e.getMessage()); - System.err.println("Aborting sending."); - System.exit(1); - } - - private static void handleDBusExecutionException(DBusExecutionException e) { - System.err.println("Cannot connect to dbus: " + e.getMessage()); - System.err.println("Aborting."); - System.exit(1); - } - - private static byte[] decodeGroupId(String groupId) { - try { - return Base64.decode(groupId); - } catch (IOException e) { - System.err.println("Failed to decode groupId (must be base64) \"" + groupId + "\": " + e.getMessage()); - System.err.println("Aborting sending."); - System.exit(1); - return null; - } - } - private static Namespace parseArgs(String[] args) { - ArgumentParser parser = ArgumentParsers.newArgumentParser("signal-cli") + ArgumentParser parser = ArgumentParsers.newFor("signal-cli") + .build() .defaultHelp(true) .description("Commandline interface for Signal.") - .version(Manager.PROJECT_NAME + " " + Manager.PROJECT_VERSION); + .version(BaseConfig.PROJECT_NAME + " " + BaseConfig.PROJECT_VERSION); parser.addArgument("-v", "--version") .help("Show package version.") .action(Arguments.version()); parser.addArgument("--config") - .help("Set the path, where to store the config (Default: $HOME/.config/signal-cli)."); + .help("Set the path, where to store the config (Default: $HOME/.config/signal)."); MutuallyExclusiveGroup mut = parser.addMutuallyExclusiveGroup(); mut.addArgument("-u", "--username") @@ -387,245 +160,41 @@ public class Main { .description("valid subcommands") .help("additional help"); - Subparser parserRegister = subparsers.addParser("register"); - parserRegister.addArgument("-v", "--voice") - .help("The verification should be done over voice, not sms.") - .action(Arguments.storeTrue()); - - Subparser parserVerify = subparsers.addParser("verify"); - parserVerify.addArgument("verificationCode") - .help("The verification code you received via sms or voice call."); - - Subparser parserSend = subparsers.addParser("send"); - parserSend.addArgument("-g", "--group") - .help("Specify the recipient group ID."); - parserSend.addArgument("recipient") - .help("Specify the recipients' phone number.") - .nargs("*"); - parserSend.addArgument("-m", "--message") - .help("Specify the message, if missing standard input is used."); - parserSend.addArgument("-a", "--attachment") - .nargs("*") - .help("Add file as attachment"); - parserSend.addArgument("-e", "--endsession") - .help("Clear session state and send end session message.") - .action(Arguments.storeTrue()); - - Subparser parserLeaveGroup = subparsers.addParser("quitGroup"); - parserLeaveGroup.addArgument("-g", "--group") - .required(true) - .help("Specify the recipient group ID."); - - Subparser parserUpdateGroup = subparsers.addParser("updateGroup"); - parserUpdateGroup.addArgument("-g", "--group") - .help("Specify the recipient group ID."); - parserUpdateGroup.addArgument("-n", "--name") - .help("Specify the new group name."); - parserUpdateGroup.addArgument("-a", "--avatar") - .help("Specify a new group avatar image file"); - parserUpdateGroup.addArgument("-m", "--member") - .nargs("*") - .help("Specify one or more members to add to the group"); - - Subparser parserReceive = subparsers.addParser("receive"); - parserReceive.addArgument("-t", "--timeout") - .type(int.class) - .help("Number of seconds to wait for new messages (negative values disable timeout)"); - - Subparser parserDaemon = subparsers.addParser("daemon"); - parserDaemon.addArgument("--system") - .action(Arguments.storeTrue()) - .help("Use DBus system bus instead of user bus."); + final Map commands = Commands.getCommands(); + for (Map.Entry entry : commands.entrySet()) { + Subparser subparser = subparsers.addParser(entry.getKey()); + entry.getValue().attachToSubparser(subparser); + } + Namespace ns; try { - Namespace ns = parser.parseArgs(args); - if (!ns.getBoolean("dbus") && !ns.getBoolean("dbus_system")) { - if (ns.getString("username") == null) { - parser.printUsage(); - System.err.println("You need to specify a username (phone number)"); - System.exit(2); - } - if (!PhoneNumberFormatter.isValidNumber(ns.getString("username"))) { - System.err.println("Invalid username (phone number), make sure you include the country code."); - System.exit(2); - } - } - if (ns.getList("recipient") != null && !ns.getList("recipient").isEmpty() && ns.getString("group") != null) { - System.err.println("You cannot specify recipients by phone number and groups a the same time"); - System.exit(2); - } - return ns; + ns = parser.parseArgs(args); } catch (ArgumentParserException e) { parser.handleError(e); return null; } - } - - private static void handleAssertionError(AssertionError e) { - System.err.println("Failed to send/receive message (Assertion): " + e.getMessage()); - e.printStackTrace(); - System.err.println("If you use an Oracle JRE please check if you have unlimited strength crypto enabled, see README"); - System.exit(1); - } - - private static void handleEncapsulatedExceptions(EncapsulatedExceptions e) { - System.err.println("Failed to send (some) messages:"); - for (NetworkFailureException n : e.getNetworkExceptions()) { - System.err.println("Network failure for \"" + n.getE164number() + "\": " + n.getMessage()); - } - for (UnregisteredUserException n : e.getUnregisteredUserExceptions()) { - System.err.println("Unregistered user \"" + n.getE164Number() + "\": " + n.getMessage()); - } - for (UntrustedIdentityException n : e.getUntrustedIdentityExceptions()) { - System.err.println("Untrusted Identity for \"" + n.getE164Number() + "\": " + n.getMessage()); - } - } - - private static void handleIOException(IOException e) { - System.err.println("Failed to send message: " + e.getMessage()); - } - - private static class ReceiveMessageHandler implements Manager.ReceiveMessageHandler { - final Manager m; - - public ReceiveMessageHandler(Manager m) { - this.m = m; - } - - @Override - public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, GroupInfo group) { - SignalServiceAddress source = envelope.getSourceAddress(); - System.out.println(String.format("Envelope from: %s (device: %d)", source.getNumber(), envelope.getSourceDevice())); - if (source.getRelay().isPresent()) { - System.out.println("Relayed by: " + source.getRelay().get()); - } - System.out.println("Timestamp: " + envelope.getTimestamp()); - - if (envelope.isReceipt()) { - System.out.println("Got receipt."); - } else if (envelope.isSignalMessage() | envelope.isPreKeySignalMessage()) { - if (content == null) { - System.out.println("Failed to decrypt message."); - } else { - if (content.getDataMessage().isPresent()) { - SignalServiceDataMessage message = content.getDataMessage().get(); - - System.out.println("Message timestamp: " + message.getTimestamp()); - if (message.getBody().isPresent()) { - System.out.println("Body: " + message.getBody().get()); - } - if (message.getGroupInfo().isPresent()) { - SignalServiceGroup groupInfo = message.getGroupInfo().get(); - System.out.println("Group info:"); - System.out.println(" Id: " + Base64.encodeBytes(groupInfo.getGroupId())); - if (groupInfo.getName().isPresent()) { - System.out.println(" Name: " + groupInfo.getName().get()); - } else if (group != null) { - System.out.println(" Name: " + group.name); - } else { - System.out.println(" Name: "); - } - System.out.println(" Type: " + groupInfo.getType()); - if (groupInfo.getMembers().isPresent()) { - for (String member : groupInfo.getMembers().get()) { - System.out.println(" Member: " + member); - } - } - if (groupInfo.getAvatar().isPresent()) { - System.out.println(" Avatar:"); - printAttachment(groupInfo.getAvatar().get()); - } - } - if (message.isEndSession()) { - System.out.println("Is end session"); - } - - if (message.getAttachments().isPresent()) { - System.out.println("Attachments: "); - for (SignalServiceAttachment attachment : message.getAttachments().get()) { - printAttachment(attachment); - } - } - } - if (content.getSyncMessage().isPresent()) { - SignalServiceSyncMessage syncMessage = content.getSyncMessage().get(); - System.out.println("Received sync message"); - } - } - } else { - System.out.println("Unknown message received."); + if ("link".equals(ns.getString("command"))) { + if (ns.getString("username") != null) { + parser.printUsage(); + System.err.println("You cannot specify a username (phone number) when linking"); + System.exit(2); } - System.out.println(); - } - - private void printAttachment(SignalServiceAttachment attachment) { - System.out.println("- " + attachment.getContentType() + " (" + (attachment.isPointer() ? "Pointer" : "") + (attachment.isStream() ? "Stream" : "") + ")"); - if (attachment.isPointer()) { - final SignalServiceAttachmentPointer pointer = attachment.asPointer(); - System.out.println(" Id: " + pointer.getId() + " Key length: " + pointer.getKey().length + (pointer.getRelay().isPresent() ? " Relay: " + pointer.getRelay().get() : "")); - System.out.println(" Size: " + (pointer.getSize().isPresent() ? pointer.getSize().get() + " bytes" : "") + (pointer.getPreview().isPresent() ? " (Preview is available: " + pointer.getPreview().get().length + " bytes)" : "")); - File file = m.getAttachmentFile(pointer.getId()); - if (file.exists()) { - System.out.println(" Stored plaintext in: " + file); - } + } else if (!ns.getBoolean("dbus") && !ns.getBoolean("dbus_system")) { + if (ns.getString("username") == null) { + parser.printUsage(); + System.err.println("You need to specify a username (phone number)"); + System.exit(2); } - } - } - - private static class DbusReceiveMessageHandler extends ReceiveMessageHandler { - final DBusConnection conn; - - public DbusReceiveMessageHandler(Manager m, DBusConnection conn) { - super(m); - this.conn = conn; - } - - @Override - public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, GroupInfo group) { - super.handleMessage(envelope, content, group); - - if (!envelope.isReceipt() && content != null && content.getDataMessage().isPresent()) { - SignalServiceDataMessage message = content.getDataMessage().get(); - - if (!message.isEndSession() && - !(message.getGroupInfo().isPresent() && - message.getGroupInfo().get().getType() != SignalServiceGroup.Type.DELIVER)) { - List attachments = new ArrayList<>(); - if (message.getAttachments().isPresent()) { - for (SignalServiceAttachment attachment : message.getAttachments().get()) { - if (attachment.isPointer()) { - attachments.add(m.getAttachmentFile(attachment.asPointer().getId()).getAbsolutePath()); - } - } - } - - try { - conn.sendSignal(new Signal.MessageReceived( - SIGNAL_OBJECTPATH, - message.getTimestamp(), - envelope.getSource(), - message.getGroupInfo().isPresent() ? message.getGroupInfo().get().getGroupId() : new byte[0], - message.getBody().isPresent() ? message.getBody().get() : "", - attachments)); - } catch (DBusException e) { - e.printStackTrace(); - } - } + if (!PhoneNumberFormatter.isValidNumber(ns.getString("username"))) { + System.err.println("Invalid username (phone number), make sure you include the country code."); + System.exit(2); } } - - private void printAttachment(SignalServiceAttachment attachment) { - System.out.println("- " + attachment.getContentType() + " (" + (attachment.isPointer() ? "Pointer" : "") + (attachment.isStream() ? "Stream" : "") + ")"); - if (attachment.isPointer()) { - final SignalServiceAttachmentPointer pointer = attachment.asPointer(); - System.out.println(" Id: " + pointer.getId() + " Key length: " + pointer.getKey().length + (pointer.getRelay().isPresent() ? " Relay: " + pointer.getRelay().get() : "")); - System.out.println(" Size: " + (pointer.getSize().isPresent() ? pointer.getSize().get() + " bytes" : "") + (pointer.getPreview().isPresent() ? " (Preview is available: " + pointer.getPreview().get().length + " bytes)" : "")); - File file = m.getAttachmentFile(pointer.getId()); - if (file.exists()) { - System.out.println(" Stored plaintext in: " + file); - } - } + if (ns.getList("recipient") != null && !ns.getList("recipient").isEmpty() && ns.getString("group") != null) { + System.err.println("You cannot specify recipients by phone number and groups a the same time"); + System.exit(2); } + return ns; } }