OmemoVAxolotlProvider.java

  1. /**
  2.  *
  3.  * Copyright 2017 Paul Schaub
  4.  *
  5.  * Licensed under the Apache License, Version 2.0 (the "License");
  6.  * you may not use this file except in compliance with the License.
  7.  * You may obtain a copy of the License at
  8.  *
  9.  *     http://www.apache.org/licenses/LICENSE-2.0
  10.  *
  11.  * Unless required by applicable law or agreed to in writing, software
  12.  * distributed under the License is distributed on an "AS IS" BASIS,
  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14.  * See the License for the specific language governing permissions and
  15.  * limitations under the License.
  16.  */
  17. package org.jivesoftware.smackx.omemo.provider;

  18. import static org.jivesoftware.smackx.omemo.element.OmemoElement.ATTR_PAYLOAD;

  19. import java.io.IOException;
  20. import java.util.ArrayList;

  21. import org.jivesoftware.smack.packet.XmlEnvironment;
  22. import org.jivesoftware.smack.provider.ExtensionElementProvider;
  23. import org.jivesoftware.smack.util.stringencoder.Base64;
  24. import org.jivesoftware.smack.xml.XmlPullParser;
  25. import org.jivesoftware.smack.xml.XmlPullParserException;

  26. import org.jivesoftware.smackx.omemo.element.OmemoElement_VAxolotl;
  27. import org.jivesoftware.smackx.omemo.element.OmemoHeaderElement;
  28. import org.jivesoftware.smackx.omemo.element.OmemoHeaderElement_VAxolotl;
  29. import org.jivesoftware.smackx.omemo.element.OmemoKeyElement;

  30. /**
  31.  * Smack ExtensionProvider that parses incoming OMEMO Message element into OmemoMessageElement objects.
  32.  *
  33.  * @author Paul Schaub
  34.  */
  35. public class OmemoVAxolotlProvider extends ExtensionElementProvider<OmemoElement_VAxolotl> {

  36.     @Override
  37.     public OmemoElement_VAxolotl parse(XmlPullParser parser, int initialDepth, XmlEnvironment xmlEnvironment) throws XmlPullParserException, IOException {
  38.         int sid = -1;
  39.         ArrayList<OmemoKeyElement> keys = new ArrayList<>();
  40.         byte[] iv = null;
  41.         byte[] payload = null;

  42.         outerloop: while (true) {
  43.             XmlPullParser.Event tag = parser.next();
  44.             switch (tag) {
  45.                 case START_ELEMENT:
  46.                     String name = parser.getName();
  47.                     switch (name) {
  48.                         case OmemoHeaderElement.ELEMENT:
  49.                             for (int i = 0; i < parser.getAttributeCount(); i++) {
  50.                                 if (parser.getAttributeName(i).equals(OmemoHeaderElement.ATTR_SID)) {
  51.                                     sid = Integer.parseInt(parser.getAttributeValue(i));
  52.                                 }
  53.                             }
  54.                             break;
  55.                         case OmemoKeyElement.ELEMENT:
  56.                             boolean prekey = false;
  57.                             int rid = -1;
  58.                             for (int i = 0; i < parser.getAttributeCount(); i++) {
  59.                                 if (parser.getAttributeName(i).equals(OmemoKeyElement.ATTR_PREKEY)) {
  60.                                     prekey = Boolean.parseBoolean(parser.getAttributeValue(i));
  61.                                 } else if (parser.getAttributeName(i).equals(OmemoKeyElement.ATTR_RID)) {
  62.                                     rid = Integer.parseInt(parser.getAttributeValue(i));
  63.                                 }
  64.                             }
  65.                             keys.add(new OmemoKeyElement(Base64.decode(parser.nextText()), rid, prekey));
  66.                             break;
  67.                         case OmemoHeaderElement.ATTR_IV:
  68.                             iv = Base64.decode(parser.nextText());
  69.                             break;
  70.                         case ATTR_PAYLOAD:
  71.                             payload = Base64.decode(parser.nextText());
  72.                             break;
  73.                     }
  74.                     break;
  75.                 case END_ELEMENT:
  76.                     if (parser.getDepth() == initialDepth) {
  77.                         break outerloop;
  78.                     }
  79.                     break;
  80.                 default:
  81.                     // Catch all for incomplete switch (MissingCasesInEnumSwitch) statement.
  82.                     break;
  83.             }
  84.         }
  85.         OmemoHeaderElement_VAxolotl header = new OmemoHeaderElement_VAxolotl(sid, keys, iv);
  86.         return new OmemoElement_VAxolotl(header, payload);
  87.     }
  88. }