1package eu.siacs.conversations.parser;
  2
  3import android.util.Log;
  4import android.util.Pair;
  5
  6import net.java.otr4j.session.Session;
  7import net.java.otr4j.session.SessionStatus;
  8
  9import java.util.List;
 10import java.util.Set;
 11
 12import eu.siacs.conversations.Config;
 13import eu.siacs.conversations.crypto.axolotl.AxolotlService;
 14import eu.siacs.conversations.crypto.axolotl.XmppAxolotlMessage;
 15import eu.siacs.conversations.entities.Account;
 16import eu.siacs.conversations.entities.Contact;
 17import eu.siacs.conversations.entities.Conversation;
 18import eu.siacs.conversations.entities.Message;
 19import eu.siacs.conversations.entities.MucOptions;
 20import eu.siacs.conversations.http.HttpConnectionManager;
 21import eu.siacs.conversations.services.MessageArchiveService;
 22import eu.siacs.conversations.services.XmppConnectionService;
 23import eu.siacs.conversations.utils.CryptoHelper;
 24import eu.siacs.conversations.xml.Element;
 25import eu.siacs.conversations.xmpp.OnMessagePacketReceived;
 26import eu.siacs.conversations.xmpp.chatstate.ChatState;
 27import eu.siacs.conversations.xmpp.jid.Jid;
 28import eu.siacs.conversations.xmpp.pep.Avatar;
 29import eu.siacs.conversations.xmpp.stanzas.MessagePacket;
 30
 31public class MessageParser extends AbstractParser implements
 32		OnMessagePacketReceived {
 33	public MessageParser(XmppConnectionService service) {
 34		super(service);
 35	}
 36
 37	private boolean extractChatState(Conversation conversation, final MessagePacket packet) {
 38		ChatState state = ChatState.parse(packet);
 39		if (state != null && conversation != null) {
 40			final Account account = conversation.getAccount();
 41			Jid from = packet.getFrom();
 42			if (from.toBareJid().equals(account.getJid().toBareJid())) {
 43				conversation.setOutgoingChatState(state);
 44				return false;
 45			} else {
 46				return conversation.setIncomingChatState(state);
 47			}
 48		}
 49		return false;
 50	}
 51
 52	private Message parseOtrChat(String body, Jid from, String id, Conversation conversation) {
 53		String presence;
 54		if (from.isBareJid()) {
 55			presence = "";
 56		} else {
 57			presence = from.getResourcepart();
 58		}
 59		if (body.matches("^\\?OTRv\\d{1,2}\\?.*")) {
 60			conversation.endOtrIfNeeded();
 61		}
 62		if (!conversation.hasValidOtrSession()) {
 63			conversation.startOtrSession(presence,false);
 64		} else {
 65			String foreignPresence = conversation.getOtrSession().getSessionID().getUserID();
 66			if (!foreignPresence.equals(presence)) {
 67				conversation.endOtrIfNeeded();
 68				conversation.startOtrSession(presence, false);
 69			}
 70		}
 71		try {
 72			conversation.setLastReceivedOtrMessageId(id);
 73			Session otrSession = conversation.getOtrSession();
 74			SessionStatus before = otrSession.getSessionStatus();
 75			body = otrSession.transformReceiving(body);
 76			SessionStatus after = otrSession.getSessionStatus();
 77			if ((before != after) && (after == SessionStatus.ENCRYPTED)) {
 78				conversation.setNextEncryption(Message.ENCRYPTION_OTR);
 79				mXmppConnectionService.onOtrSessionEstablished(conversation);
 80			} else if ((before != after) && (after == SessionStatus.FINISHED)) {
 81				conversation.setNextEncryption(Message.ENCRYPTION_NONE);
 82				conversation.resetOtrSession();
 83				mXmppConnectionService.updateConversationUi();
 84			}
 85			if ((body == null) || (body.isEmpty())) {
 86				return null;
 87			}
 88			if (body.startsWith(CryptoHelper.FILETRANSFER)) {
 89				String key = body.substring(CryptoHelper.FILETRANSFER.length());
 90				conversation.setSymmetricKey(CryptoHelper.hexToBytes(key));
 91				return null;
 92			}
 93			Message finishedMessage = new Message(conversation, body, Message.ENCRYPTION_OTR, Message.STATUS_RECEIVED);
 94			conversation.setLastReceivedOtrMessageId(null);
 95			return finishedMessage;
 96		} catch (Exception e) {
 97			conversation.resetOtrSession();
 98			return null;
 99		}
100	}
101
102	private Message parseAxolotlChat(Element axolotlMessage, Jid from, String id, Conversation conversation, int status) {
103		Message finishedMessage = null;
104		AxolotlService service = conversation.getAccount().getAxolotlService();
105		XmppAxolotlMessage xmppAxolotlMessage = new XmppAxolotlMessage(from.toBareJid(), axolotlMessage);
106		XmppAxolotlMessage.XmppAxolotlPlaintextMessage plaintextMessage = service.processReceiving(xmppAxolotlMessage);
107		if(plaintextMessage != null) {
108			finishedMessage = new Message(conversation, plaintextMessage.getPlaintext(), Message.ENCRYPTION_AXOLOTL, status);
109			finishedMessage.setAxolotlSession(plaintextMessage.getSession());
110		}
111
112		return finishedMessage;
113	}
114
115	private class Invite {
116		Jid jid;
117		String password;
118		Invite(Jid jid, String password) {
119			this.jid = jid;
120			this.password = password;
121		}
122
123		public boolean execute(Account account) {
124			if (jid != null) {
125				Conversation conversation = mXmppConnectionService.findOrCreateConversation(account, jid, true);
126				if (!conversation.getMucOptions().online()) {
127					conversation.getMucOptions().setPassword(password);
128					mXmppConnectionService.databaseBackend.updateConversation(conversation);
129					mXmppConnectionService.joinMuc(conversation);
130					mXmppConnectionService.updateConversationUi();
131				}
132				return true;
133			}
134			return false;
135		}
136	}
137
138	private Invite extractInvite(Element message) {
139		Element x = message.findChild("x", "http://jabber.org/protocol/muc#user");
140		if (x != null) {
141			Element invite = x.findChild("invite");
142			if (invite != null) {
143				Element pw = x.findChild("password");
144				return new Invite(message.getAttributeAsJid("from"), pw != null ? pw.getContent(): null);
145			}
146		} else {
147			x = message.findChild("x","jabber:x:conference");
148			if (x != null) {
149				return new Invite(x.getAttributeAsJid("jid"),x.getAttribute("password"));
150			}
151		}
152		return null;
153	}
154
155	private void parseEvent(final Element event, final Jid from, final Account account) {
156		Element items = event.findChild("items");
157		String node = items == null ? null : items.getAttribute("node");
158		if ("urn:xmpp:avatar:metadata".equals(node)) {
159			Avatar avatar = Avatar.parseMetadata(items);
160			if (avatar != null) {
161				avatar.owner = from;
162				if (mXmppConnectionService.getFileBackend().isAvatarCached(avatar)) {
163					if (account.getJid().toBareJid().equals(from)) {
164						if (account.setAvatar(avatar.getFilename())) {
165							mXmppConnectionService.databaseBackend.updateAccount(account);
166						}
167						mXmppConnectionService.getAvatarService().clear(account);
168						mXmppConnectionService.updateConversationUi();
169						mXmppConnectionService.updateAccountUi();
170					} else {
171						Contact contact = account.getRoster().getContact(from);
172						contact.setAvatar(avatar);
173						mXmppConnectionService.getAvatarService().clear(contact);
174						mXmppConnectionService.updateConversationUi();
175						mXmppConnectionService.updateRosterUi();
176					}
177				} else {
178					mXmppConnectionService.fetchAvatar(account, avatar);
179				}
180			}
181		} else if ("http://jabber.org/protocol/nick".equals(node)) {
182			Element i = items.findChild("item");
183			Element nick = i == null ? null : i.findChild("nick", "http://jabber.org/protocol/nick");
184			if (nick != null) {
185				Contact contact = account.getRoster().getContact(from);
186				contact.setPresenceName(nick.getContent());
187				mXmppConnectionService.getAvatarService().clear(account);
188				mXmppConnectionService.updateConversationUi();
189				mXmppConnectionService.updateAccountUi();
190			}
191		} else if (AxolotlService.PEP_DEVICE_LIST.equals(node)) {
192			Log.d(Config.LOGTAG, "Received PEP device list update from "+ from + ", processing...");
193			Element item = items.findChild("item");
194			Set<Integer> deviceIds = mXmppConnectionService.getIqParser().deviceIds(item);
195			AxolotlService axolotlService = account.getAxolotlService();
196			axolotlService.registerDevices(from, deviceIds);
197		}
198	}
199
200	private boolean handleErrorMessage(Account account, MessagePacket packet) {
201		if (packet.getType() == MessagePacket.TYPE_ERROR) {
202			Jid from = packet.getFrom();
203			if (from != null) {
204				mXmppConnectionService.markMessage(account, from.toBareJid(), packet.getId(), Message.STATUS_SEND_FAILED);
205			}
206			return true;
207		}
208		return false;
209	}
210
211	@Override
212	public void onMessagePacketReceived(Account account, MessagePacket original) {
213		if (handleErrorMessage(account, original)) {
214			return;
215		}
216		final MessagePacket packet;
217		Long timestamp = null;
218		final boolean isForwarded;
219		String serverMsgId = null;
220		final Element fin = original.findChild("fin", "urn:xmpp:mam:0");
221		if (fin != null) {
222			mXmppConnectionService.getMessageArchiveService().processFin(fin,original.getFrom());
223			return;
224		}
225		final Element result = original.findChild("result","urn:xmpp:mam:0");
226		final MessageArchiveService.Query query = result == null ? null : mXmppConnectionService.getMessageArchiveService().findQuery(result.getAttribute("queryid"));
227		if (query != null && query.validFrom(original.getFrom())) {
228			Pair<MessagePacket, Long> f = original.getForwardedMessagePacket("result", "urn:xmpp:mam:0");
229			if (f == null) {
230				return;
231			}
232			timestamp = f.second;
233			packet = f.first;
234			isForwarded = true;
235			serverMsgId = result.getAttribute("id");
236			query.incrementTotalCount();
237		} else if (query != null) {
238			Log.d(Config.LOGTAG,account.getJid().toBareJid()+": received mam result from invalid sender");
239			return;
240		} else if (original.fromServer(account)) {
241			Pair<MessagePacket, Long> f;
242			f = original.getForwardedMessagePacket("received", "urn:xmpp:carbons:2");
243			f = f == null ? original.getForwardedMessagePacket("sent", "urn:xmpp:carbons:2") : f;
244			packet = f != null ? f.first : original;
245			if (handleErrorMessage(account, packet)) {
246				return;
247			}
248			timestamp = f != null ? f.second : null;
249			isForwarded = f != null;
250		} else {
251			packet = original;
252			isForwarded = false;
253		}
254
255		if (timestamp == null) {
256			timestamp = AbstractParser.getTimestamp(packet, System.currentTimeMillis());
257		}
258		final String body = packet.getBody();
259		final Element mucUserElement = packet.findChild("x","http://jabber.org/protocol/muc#user");
260		final String pgpEncrypted = packet.findChildContent("x", "jabber:x:encrypted");
261		final Element axolotlEncrypted = packet.findChild("axolotl_message", AxolotlService.PEP_PREFIX);
262		int status;
263		final Jid counterpart;
264		final Jid to = packet.getTo();
265		final Jid from = packet.getFrom();
266		final String remoteMsgId = packet.getId();
267
268		if (from == null || to == null) {
269			Log.d(Config.LOGTAG,"no to or from in: "+packet.toString());
270			return;
271		}
272		
273		boolean isTypeGroupChat = packet.getType() == MessagePacket.TYPE_GROUPCHAT;
274		boolean isProperlyAddressed = !to.isBareJid() || account.countPresences() == 1;
275		boolean isMucStatusMessage = from.isBareJid() && mucUserElement != null && mucUserElement.hasChild("status");
276		if (packet.fromAccount(account)) {
277			status = Message.STATUS_SEND;
278			counterpart = to;
279		} else {
280			status = Message.STATUS_RECEIVED;
281			counterpart = from;
282		}
283
284		Invite invite = extractInvite(packet);
285		if (invite != null && invite.execute(account)) {
286			return;
287		}
288
289		if (extractChatState(mXmppConnectionService.find(account, from), packet)) {
290			mXmppConnectionService.updateConversationUi();
291		}
292
293		if ((body != null || pgpEncrypted != null || axolotlEncrypted != null) && !isMucStatusMessage) {
294			Conversation conversation = mXmppConnectionService.findOrCreateConversation(account, counterpart.toBareJid(), isTypeGroupChat);
295			if (isTypeGroupChat) {
296				if (counterpart.getResourcepart().equals(conversation.getMucOptions().getActualNick())) {
297					status = Message.STATUS_SEND_RECEIVED;
298					if (mXmppConnectionService.markMessage(conversation, remoteMsgId, status)) {
299						return;
300					} else {
301						Message message = conversation.findSentMessageWithBody(body);
302						if (message != null) {
303							message.setRemoteMsgId(remoteMsgId);
304							mXmppConnectionService.markMessage(message, status);
305							return;
306						}
307					}
308				} else {
309					status = Message.STATUS_RECEIVED;
310				}
311			}
312			Message message;
313			if (body != null && body.startsWith("?OTR")) {
314				if (!isForwarded && !isTypeGroupChat && isProperlyAddressed) {
315					message = parseOtrChat(body, from, remoteMsgId, conversation);
316					if (message == null) {
317						return;
318					}
319				} else {
320					message = new Message(conversation, body, Message.ENCRYPTION_NONE, status);
321				}
322			} else if (pgpEncrypted != null) {
323				message = new Message(conversation, pgpEncrypted, Message.ENCRYPTION_PGP, status);
324			} else if (axolotlEncrypted != null) {
325				message = parseAxolotlChat(axolotlEncrypted, from, remoteMsgId, conversation, status);
326				if (message == null) {
327					return;
328				}
329			} else {
330				message = new Message(conversation, body, Message.ENCRYPTION_NONE, status);
331			}
332			message.setCounterpart(counterpart);
333			message.setRemoteMsgId(remoteMsgId);
334			message.setServerMsgId(serverMsgId);
335			message.setTime(timestamp);
336			message.markable = packet.hasChild("markable", "urn:xmpp:chat-markers:0");
337			if (conversation.getMode() == Conversation.MODE_MULTI) {
338				message.setTrueCounterpart(conversation.getMucOptions().getTrueCounterpart(counterpart.getResourcepart()));
339				if (!isTypeGroupChat) {
340					message.setType(Message.TYPE_PRIVATE);
341				}
342			}
343			updateLastseen(packet,account,true);
344			boolean checkForDuplicates = serverMsgId != null
345					|| (isTypeGroupChat && packet.hasChild("delay","urn:xmpp:delay"))
346					|| message.getType() == Message.TYPE_PRIVATE;
347			if (checkForDuplicates && conversation.hasDuplicateMessage(message)) {
348				Log.d(Config.LOGTAG,"skipping duplicate message from "+message.getCounterpart().toString()+" "+message.getBody());
349				return;
350			}
351			if (query != null) {
352				query.incrementMessageCount();
353			}
354			conversation.add(message);
355			if (serverMsgId == null) {
356				if (status == Message.STATUS_SEND || status == Message.STATUS_SEND_RECEIVED) {
357					mXmppConnectionService.markRead(conversation);
358					account.activateGracePeriod();
359				} else {
360					message.markUnread();
361				}
362				mXmppConnectionService.updateConversationUi();
363			}
364
365			if (mXmppConnectionService.confirmMessages() && remoteMsgId != null && !isForwarded) {
366				if (packet.hasChild("markable", "urn:xmpp:chat-markers:0")) {
367					MessagePacket receipt = mXmppConnectionService
368							.getMessageGenerator().received(account, packet, "urn:xmpp:chat-markers:0");
369					mXmppConnectionService.sendMessagePacket(account, receipt);
370				}
371				if (packet.hasChild("request", "urn:xmpp:receipts")) {
372					MessagePacket receipt = mXmppConnectionService
373							.getMessageGenerator().received(account, packet, "urn:xmpp:receipts");
374					mXmppConnectionService.sendMessagePacket(account, receipt);
375				}
376			}
377			if (account.getXmppConnection() != null && account.getXmppConnection().getFeatures().advancedStreamFeaturesLoaded()) {
378				if (conversation.setLastMessageTransmitted(System.currentTimeMillis())) {
379					mXmppConnectionService.updateConversation(conversation);
380				}
381			}
382
383			if (message.getStatus() == Message.STATUS_RECEIVED
384					&& conversation.getOtrSession() != null
385					&& !conversation.getOtrSession().getSessionID().getUserID()
386					.equals(message.getCounterpart().getResourcepart())) {
387				conversation.endOtrIfNeeded();
388			}
389
390			if (message.getEncryption() == Message.ENCRYPTION_NONE || mXmppConnectionService.saveEncryptedMessages()) {
391				mXmppConnectionService.databaseBackend.createMessage(message);
392			}
393			final HttpConnectionManager manager = this.mXmppConnectionService.getHttpConnectionManager();
394			if (message.trusted() && message.treatAsDownloadable() != Message.Decision.NEVER && manager.getAutoAcceptFileSize() > 0) {
395				manager.createNewDownloadConnection(message);
396			} else if (!message.isRead()) {
397				mXmppConnectionService.getNotificationService().push(message);
398			}
399		} else { //no body
400			if (isTypeGroupChat) {
401				Conversation conversation = mXmppConnectionService.find(account, from.toBareJid());
402				if (packet.hasChild("subject")) {
403					if (conversation != null && conversation.getMode() == Conversation.MODE_MULTI) {
404						conversation.setHasMessagesLeftOnServer(conversation.countMessages() > 0);
405						conversation.getMucOptions().setSubject(packet.findChildContent("subject"));
406						mXmppConnectionService.updateConversationUi();
407						return;
408					}
409				}
410
411				if (conversation != null && isMucStatusMessage) {
412					for (Element child : mucUserElement.getChildren()) {
413						if (child.getName().equals("status")
414								&& MucOptions.STATUS_CODE_ROOM_CONFIG_CHANGED.equals(child.getAttribute("code"))) {
415							mXmppConnectionService.fetchConferenceConfiguration(conversation);
416						}
417					}
418				}
419			}
420		}
421
422		Element received = packet.findChild("received", "urn:xmpp:chat-markers:0");
423		if (received == null) {
424			received = packet.findChild("received", "urn:xmpp:receipts");
425		}
426		if (received != null && !packet.fromAccount(account)) {
427			mXmppConnectionService.markMessage(account, from.toBareJid(), received.getAttribute("id"), Message.STATUS_SEND_RECEIVED);
428		}
429		Element displayed = packet.findChild("displayed", "urn:xmpp:chat-markers:0");
430		if (displayed != null) {
431			if (packet.fromAccount(account)) {
432				Conversation conversation = mXmppConnectionService.find(account,counterpart.toBareJid());
433				if (conversation != null) {
434					mXmppConnectionService.markRead(conversation);
435				}
436			} else {
437				updateLastseen(packet, account, true);
438				final Message displayedMessage = mXmppConnectionService.markMessage(account, from.toBareJid(), displayed.getAttribute("id"), Message.STATUS_SEND_DISPLAYED);
439				Message message = displayedMessage == null ? null : displayedMessage.prev();
440				while (message != null
441						&& message.getStatus() == Message.STATUS_SEND_RECEIVED
442						&& message.getTimeSent() < displayedMessage.getTimeSent()) {
443					mXmppConnectionService.markMessage(message, Message.STATUS_SEND_DISPLAYED);
444					message = message.prev();
445				}
446			}
447		}
448
449		Element event = packet.findChild("event", "http://jabber.org/protocol/pubsub#event");
450		if (event != null) {
451			parseEvent(event, from, account);
452		}
453
454		String nick = packet.findChildContent("nick", "http://jabber.org/protocol/nick");
455		if (nick != null) {
456			Contact contact = account.getRoster().getContact(from);
457			contact.setPresenceName(nick);
458		}
459	}
460}