452 lines
21 KiB
Java
452 lines
21 KiB
Java
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Scanner;
|
|
|
|
import org.jivesoftware.smack.AbstractXMPPConnection;
|
|
import org.jivesoftware.smack.SmackConfiguration;
|
|
import org.jivesoftware.smack.SmackException;
|
|
import org.jivesoftware.smack.XMPPException;
|
|
import org.jivesoftware.smack.chat.Chat;
|
|
import org.jivesoftware.smack.chat.ChatManager;
|
|
import org.jivesoftware.smack.packet.Message;
|
|
import org.jivesoftware.smack.packet.Presence;
|
|
import org.jivesoftware.smack.packet.Stanza;
|
|
import org.jivesoftware.smack.roster.Roster;
|
|
import org.jivesoftware.smack.roster.RosterEntry;
|
|
import org.jivesoftware.smack.tcp.XMPPTCPConnection;
|
|
import org.jivesoftware.smackx.carbons.CarbonManager;
|
|
import org.jivesoftware.smackx.carbons.packet.CarbonExtension;
|
|
import org.jivesoftware.smackx.muc.MultiUserChat;
|
|
import org.jivesoftware.smackx.muc.MultiUserChatException;
|
|
import org.jivesoftware.smackx.muc.MultiUserChatManager;
|
|
import org.jivesoftware.smackx.omemo.OmemoConfiguration;
|
|
import org.jivesoftware.smackx.omemo.OmemoManager;
|
|
import org.jivesoftware.smackx.omemo.OmemoMessage;
|
|
import org.jivesoftware.smackx.omemo.exceptions.UndecidedOmemoIdentityException;
|
|
import org.jivesoftware.smackx.omemo.internal.OmemoCachedDeviceList;
|
|
import org.jivesoftware.smackx.omemo.internal.OmemoDevice;
|
|
import org.jivesoftware.smackx.omemo.listener.OmemoMessageListener;
|
|
import org.jivesoftware.smackx.omemo.listener.OmemoMucMessageListener;
|
|
import org.jivesoftware.smackx.omemo.signal.SignalCachingOmemoStore;
|
|
import org.jivesoftware.smackx.omemo.signal.SignalFileBasedOmemoStore;
|
|
import org.jivesoftware.smackx.omemo.signal.SignalOmemoService;
|
|
import org.jivesoftware.smackx.omemo.trust.OmemoFingerprint;
|
|
import org.jivesoftware.smackx.omemo.trust.OmemoTrustCallback;
|
|
import org.jivesoftware.smackx.omemo.trust.TrustState;
|
|
|
|
import org.jline.reader.EndOfFileException;
|
|
import org.jline.reader.LineReader;
|
|
import org.jline.reader.LineReaderBuilder;
|
|
import org.jline.reader.UserInterruptException;
|
|
import org.jline.terminal.Terminal;
|
|
import org.jline.terminal.TerminalBuilder;
|
|
import org.jxmpp.jid.BareJid;
|
|
import org.jxmpp.jid.EntityBareJid;
|
|
import org.jxmpp.jid.impl.JidCreate;
|
|
import org.jxmpp.jid.parts.Resourcepart;
|
|
import org.jxmpp.stringprep.XmppStringprepException;
|
|
import org.whispersystems.libsignal.IdentityKey;
|
|
|
|
|
|
/**
|
|
* Test
|
|
* Created by vanitas on 28.11.16.
|
|
*/
|
|
public class Main {
|
|
private AbstractXMPPConnection connection;
|
|
private OmemoManager omemoManager;
|
|
private OmemoManager.LoggedInOmemoManager managerGuard;
|
|
|
|
private Main() throws XmppStringprepException {
|
|
//*
|
|
SmackConfiguration.DEBUG = true;
|
|
/*/
|
|
SmackConfiguration.DEBUG = false;
|
|
//*/
|
|
OmemoConfiguration.setAddOmemoHintBody(false);
|
|
}
|
|
|
|
public void start() throws Exception {
|
|
Terminal terminal = TerminalBuilder.terminal();
|
|
LineReader reader = LineReaderBuilder.builder()
|
|
.terminal(terminal)
|
|
.build();
|
|
String prompt = "> ";
|
|
|
|
Scanner scanner = new Scanner(System.in);
|
|
String jidname = null, password = null;
|
|
while(jidname == null) {
|
|
System.out.println("Enter username:");
|
|
jidname = scanner.nextLine();
|
|
}
|
|
while (password == null) {
|
|
System.out.println("Enter password:");
|
|
password = scanner.nextLine();
|
|
}
|
|
connection = new XMPPTCPConnection(jidname, password);
|
|
|
|
SignalOmemoService.acknowledgeLicense();
|
|
SignalOmemoService.setup();
|
|
SignalOmemoService service = (SignalOmemoService) SignalOmemoService.getInstance();
|
|
service.setOmemoStoreBackend(new SignalCachingOmemoStore(new SignalFileBasedOmemoStore(new File("store"))));
|
|
|
|
omemoManager = OmemoManager.getInstanceFor(connection);
|
|
omemoManager.setTrustCallback(new OmemoTrustCallback() {
|
|
@Override
|
|
public TrustState getTrust(OmemoDevice device, OmemoFingerprint fingerprint) {
|
|
return TrustState.trusted;
|
|
}
|
|
|
|
@Override
|
|
public void setTrust(OmemoDevice device, OmemoFingerprint fingerprint, TrustState state) {
|
|
|
|
}
|
|
});
|
|
connection.setPacketReplyTimeout(10000);
|
|
connection = connection.connect();
|
|
connection.login();
|
|
omemoManager.initialize();
|
|
managerGuard = new OmemoManager.LoggedInOmemoManager(omemoManager);
|
|
|
|
System.out.println("Logged in. Begin setting up OMEMO...");
|
|
|
|
OmemoMessageListener messageListener = new OmemoMessageListener() {
|
|
@Override
|
|
public void onOmemoMessageReceived(Stanza stanza, OmemoMessage.Received received) {
|
|
BareJid sender = stanza.getFrom().asBareJid();
|
|
if (received.isKeyTransportMessage()) {
|
|
return;
|
|
}
|
|
String decryptedBody = received.getBody();
|
|
if (sender != null && decryptedBody != null) {
|
|
reader.callWidget(LineReader.CLEAR);
|
|
reader.getTerminal().writer().println("\033[34m" + sender + ": " + decryptedBody);
|
|
reader.callWidget(LineReader.REDRAW_LINE);
|
|
reader.callWidget(LineReader.REDISPLAY);
|
|
reader.getTerminal().writer().flush();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onOmemoCarbonCopyReceived(CarbonExtension.Direction direction, Message carbonCopy, Message wrappingMessage, OmemoMessage.Received decryptedCarbonCopy) {
|
|
|
|
}
|
|
};
|
|
OmemoMucMessageListener mucMessageListener = (multiUserChat, stanza, received) -> {
|
|
BareJid bareJid = received.getSenderDevice().getJid();
|
|
if (received.isKeyTransportMessage()) {
|
|
return;
|
|
}
|
|
String s = received.getBody();
|
|
if (multiUserChat != null && bareJid != null && s != null) {
|
|
reader.callWidget(LineReader.CLEAR);
|
|
reader.getTerminal().writer().println("\033[36m" + multiUserChat.getRoom() + ": " + bareJid + ": " + s);
|
|
reader.callWidget(LineReader.REDRAW_LINE);
|
|
reader.callWidget(LineReader.REDISPLAY);
|
|
reader.getTerminal().writer().flush();
|
|
}
|
|
};
|
|
|
|
CarbonManager.getInstanceFor(connection).enableCarbons();
|
|
|
|
omemoManager.addOmemoMessageListener(messageListener);
|
|
omemoManager.addOmemoMucMessageListener(mucMessageListener);
|
|
|
|
Roster roster = Roster.getInstanceFor(connection);
|
|
roster.setSubscriptionMode(Roster.SubscriptionMode.accept_all);
|
|
|
|
ChatManager cm = ChatManager.getInstanceFor(connection);
|
|
cm.addChatListener((chat, b) -> chat.addMessageListener((chat1, message) -> {
|
|
if(message.getBody() != null && chat1 != null) {
|
|
System.out.println("Message received: " + chat1.getParticipant().toString() + ": " + message.getBody());
|
|
}
|
|
}));
|
|
|
|
MultiUserChatManager mucm = MultiUserChatManager.getInstanceFor(connection);
|
|
mucm.setAutoJoinOnReconnect(true);
|
|
mucm.addInvitationListener((xmppConnection, multiUserChat, entityFullJid, s, s1, message, invite) -> {
|
|
try {
|
|
multiUserChat.join(Resourcepart.from("OMEMO"));
|
|
multiUserChat.addMessageListener(message1 -> {
|
|
System.out.println("MUC: "+message1.getFrom()+": "+message1.getBody());
|
|
});
|
|
System.out.println("Joined Room "+multiUserChat.getRoom().asBareJid().toString());
|
|
} catch (SmackException.NoResponseException | XMPPException.XMPPErrorException | InterruptedException | MultiUserChatException.NotAMucServiceException | SmackException.NotConnectedException | XmppStringprepException e) {
|
|
e.printStackTrace();
|
|
}
|
|
});
|
|
System.out.println("OMEMO setup complete. You can now start chatting.");
|
|
Chat current = null;
|
|
boolean omemo = false;
|
|
|
|
while (true) {
|
|
String line = null;
|
|
try {
|
|
line = reader.readLine(prompt);
|
|
} catch (UserInterruptException e) {
|
|
// Ignore
|
|
} catch (EndOfFileException e) {
|
|
return;
|
|
}
|
|
String [] split = line.split(" ");
|
|
if(line.startsWith("/chat ")) {
|
|
String l = line.substring("/chat ".length());
|
|
if(l.length() == 0) {
|
|
System.out.println(current != null ? current.getParticipant() : "null");
|
|
} else {
|
|
String id = split[1];
|
|
BareJid jid = getJid(id);
|
|
if(jid != null) {
|
|
current = cm.createChat(jid.asEntityJidIfPossible());
|
|
current.sendMessage(l.substring(id.length() + 1));
|
|
}
|
|
}
|
|
} else if (line.startsWith("/quit")) {
|
|
scanner.close();
|
|
connection.disconnect(new Presence(Presence.Type.unavailable, "Smack is still alive :D", 100, Presence.Mode.away));
|
|
break;
|
|
} else if (line.startsWith("/add")) {
|
|
String jid = split.length == 4 ? split[1] : null;
|
|
if(jid != null) {
|
|
BareJid b = JidCreate.bareFrom(jid);
|
|
roster.createEntry(b, split[2], new String[]{split[3]});
|
|
} else {
|
|
System.out.println("Usage: /add jid@server nick group");
|
|
}
|
|
} else if(line.startsWith("/remove")) {
|
|
if(split.length == 2) {
|
|
BareJid b = getJid(split[1]);
|
|
roster.removeEntry(roster.getEntry(b));
|
|
System.out.println("Removed contact from roster");
|
|
}
|
|
} else if(line.startsWith("/list")){
|
|
|
|
if(split.length == 1) {
|
|
for (RosterEntry r : roster.getEntries()) {
|
|
System.out.println(r.getName() + " (" + r.getJid() + ") Can I see? " + r.canSeeHisPresence() + ". Can they see? " + r.canSeeMyPresence() + ". Online? " + roster.getPresence(r.getJid()).isAvailable());
|
|
}
|
|
for (EntityBareJid r : mucm.getJoinedRooms()) {
|
|
System.out.println(r.asBareJid().toString());
|
|
}
|
|
} else {
|
|
BareJid jid = getJid(split[1]);
|
|
try {
|
|
List<Presence> presences = roster.getAllPresences(jid);
|
|
for(Presence p : presences) {
|
|
System.out.println(p.getFrom());
|
|
}
|
|
} catch (Exception e) {}
|
|
omemoManager.requestDeviceListUpdateFor(jid);
|
|
OmemoCachedDeviceList list = service.getOmemoStoreBackend().loadCachedDeviceList(omemoManager.getOwnDevice(), jid);
|
|
if(list == null) {
|
|
list = new OmemoCachedDeviceList();
|
|
}
|
|
ArrayList<String> fps = new ArrayList<>();
|
|
for(int id : list.getActiveDevices()) {
|
|
OmemoDevice d = new OmemoDevice(jid, id);
|
|
IdentityKey idk = service.getOmemoStoreBackend().loadOmemoIdentityKey(omemoManager.getOwnDevice(), d);
|
|
if(idk == null) {
|
|
System.out.println("No identityKey for "+d);
|
|
} else {
|
|
OmemoFingerprint fp = service.getOmemoStoreBackend().getFingerprint(omemoManager.getOwnDevice(), d);
|
|
if (fp != null) {
|
|
fps.add(fp.blocksOf8Chars());
|
|
}
|
|
}
|
|
}
|
|
for(int i=0; i<fps.size(); i++) {
|
|
System.out.println(i+": "+fps.get(i));
|
|
}
|
|
}
|
|
} else if(line.startsWith("/trust")) {
|
|
if(split.length == 2) {
|
|
System.out.println("Usage: \n0: Untrusted, 1: Trusted, otherwise: Undecided");
|
|
BareJid jid = getJid(split[1]);
|
|
|
|
if(jid == null) {
|
|
continue;
|
|
}
|
|
|
|
System.out.println(jid);
|
|
|
|
omemoManager.requestDeviceListUpdateFor(jid);
|
|
|
|
|
|
for (OmemoDevice device : omemoManager.getDevicesOf(jid)) {
|
|
OmemoFingerprint fp = omemoManager.getFingerprint(device);
|
|
|
|
if (omemoManager.isDecidedOmemoIdentity(device, fp)) {
|
|
if (omemoManager.isTrustedOmemoIdentity(device, fp)) {
|
|
System.out.println("Status: Trusted");
|
|
} else {
|
|
System.out.println("Status: Untrusted");
|
|
}
|
|
} else {
|
|
System.out.println("Status: Undecided");
|
|
}
|
|
|
|
System.out.println(fp.blocksOf8Chars());
|
|
String decision = scanner.nextLine();
|
|
if (decision.equals("0")) {
|
|
omemoManager.distrustOmemoIdentity(device, fp);
|
|
System.out.println("Identity has been untrusted.");
|
|
} else if (decision.equals("1")) {
|
|
omemoManager.trustOmemoIdentity(device, fp);
|
|
System.out.println("Identity has been trusted.");
|
|
}
|
|
}
|
|
}
|
|
|
|
} else if(line.startsWith("/purge")) {
|
|
omemoManager.purgeDeviceList();
|
|
System.out.println("Purge successful.");
|
|
// } else if(line.startsWith("/regenerate")) {
|
|
// omemoManager.regenerateIdentity();
|
|
// System.out.println("Regeneration successful.");
|
|
} else if(line.startsWith("/omemo")) {
|
|
if(split.length == 1) {
|
|
} else {
|
|
BareJid recipient = getJid(split[1]);
|
|
if (recipient != null) {
|
|
String message = "";
|
|
for (int i = 2; i < split.length; i++) {
|
|
message += split[i] + " ";
|
|
}
|
|
OmemoMessage.Sent encrypted = null;
|
|
try {
|
|
encrypted = omemoManager.encrypt(recipient, message.trim());
|
|
} catch (UndecidedOmemoIdentityException e) {
|
|
System.out.println("There are undecided identities:");
|
|
for(OmemoDevice d : e.getUndecidedDevices()) {
|
|
System.out.println(d.toString());
|
|
}
|
|
}
|
|
if(encrypted != null) {
|
|
current = cm.createChat(recipient.asEntityJidIfPossible());
|
|
Message m = new Message();
|
|
m.addExtension(encrypted.getElement());
|
|
current.sendMessage(m);
|
|
}
|
|
}
|
|
}
|
|
omemo = true;
|
|
}
|
|
else if(line.startsWith("/mucomemo")) {
|
|
if(split.length >= 3) {
|
|
BareJid mucJid = getJid(split[1]);
|
|
if (mucJid != null) {
|
|
String message = "";
|
|
for (int i = 2; i < split.length; i++) {
|
|
message += split[i] + " ";
|
|
}
|
|
MultiUserChat muc = mucm.getMultiUserChat(mucJid.asEntityBareJidIfPossible());
|
|
OmemoMessage.Sent encrypted = null;
|
|
try {
|
|
encrypted = omemoManager.encrypt(muc, message.trim());
|
|
} catch (UndecidedOmemoIdentityException e) {
|
|
System.out.println("There are undecided identities:");
|
|
for(OmemoDevice d : e.getUndecidedDevices()) {
|
|
System.out.println(d.toString());
|
|
}
|
|
}
|
|
|
|
if(encrypted != null) {
|
|
Message m = new Message();
|
|
m.addExtension(encrypted.getElement());
|
|
muc.sendMessage(m);
|
|
}
|
|
}
|
|
}
|
|
} else if(line.startsWith("/fingerprint")) {
|
|
OmemoFingerprint fingerprint = omemoManager.getOwnFingerprint();
|
|
System.out.println(fingerprint.blocksOf8Chars());
|
|
} else if(line.startsWith("/help")) {
|
|
if(split.length == 1) {
|
|
System.out.println("Available options: \n" +
|
|
"/chat <Nickname/Jid> <Message>: Send a normal unencrypted chat message to a user. \n" +
|
|
"/omemo <Nickname/Jid> <Message>: Send an OMEMO encrypted message to a user. \n" +
|
|
"/mucomemo <MUC-Jid> <Message>: Send an OMEMO encrypted message to a group chat. \n" +
|
|
"/list: List your roster. \n" +
|
|
"/list <Nickname/Jid>: List all devices of a user. \n" +
|
|
"/fingerprint: Show your OMEMO fingerprint. \n" +
|
|
"/purge: Remove all other devices from your list of active devices. \n" +
|
|
"/regenerate: Create a new OMEMO identity. \n" +
|
|
"/add <jid> <Nickname> <group>: Add a new contact to your roster. \n" +
|
|
"/remove <jid>: Remove a contact from your roster. \n" +
|
|
"/quit: Quit the application.");
|
|
}
|
|
// } else if(line.startsWith("/mam")) {
|
|
// MamManager mamManager = MamManager.getInstanceFor(connection);
|
|
// MamManager.MamQueryResult result = mamManager.queryArchive(new Date(System.currentTimeMillis()-1000*60*60*24), new Date(System.currentTimeMillis()));
|
|
// for(ClearTextMessage d : omemoManager.decryptMamQueryResult(result)) {
|
|
// messageListener.onOmemoMessageReceived(d.getBody(), d.getOriginalMessage(), null, d.getMessageInformation());
|
|
// }
|
|
// System.out.println("Query finished");
|
|
} else if(line.startsWith("/ratchetUpdate")) {
|
|
if(split.length == 2) {
|
|
BareJid jid = getJid(split[1]);
|
|
OmemoCachedDeviceList cachedDeviceList = service.getOmemoStoreBackend().loadCachedDeviceList(omemoManager.getOwnDevice(), jid);
|
|
for(int id : cachedDeviceList.getActiveDevices()) {
|
|
OmemoDevice d = new OmemoDevice(jid, id);
|
|
omemoManager.sendRatchetUpdateMessage(d);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
if(current != null) {
|
|
if(!omemo) {
|
|
current.sendMessage(line);
|
|
} else {
|
|
try {
|
|
OmemoMessage.Sent e = omemoManager.encrypt(current.getParticipant().asEntityBareJid(), line.trim());
|
|
Message m = new Message();
|
|
m.addExtension(e.getElement());
|
|
current.sendMessage(m);
|
|
} catch (UndecidedOmemoIdentityException e) {
|
|
System.out.println("There are undecided identities:");
|
|
for(OmemoDevice d : e.getUndecidedDevices()) {
|
|
System.out.println(d.toString());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
System.out.println("please open a chat");
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
try {
|
|
Main main = new Main();
|
|
main.start();
|
|
} catch (Exception ignored) {
|
|
ignored.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public BareJid getJid(String user) {
|
|
Roster roster = Roster.getInstanceFor(connection);
|
|
RosterEntry r = null;
|
|
for(RosterEntry s : roster.getEntries()) {
|
|
if(s.getName() != null && s.getName().equals(user)) {
|
|
r = s;
|
|
break;
|
|
}
|
|
}
|
|
if(r != null) {
|
|
return r.getJid();
|
|
} else {
|
|
try {
|
|
return JidCreate.bareFrom(user);
|
|
} catch (XmppStringprepException e) {
|
|
e.printStackTrace();
|
|
return null;
|
|
}
|
|
}
|
|
}
|
|
}
|