package de.pixart.messenger.entities; import android.content.ContentValues; import android.database.Cursor; import android.support.annotation.NonNull; import android.util.Base64; import org.json.JSONArray; import org.json.JSONException; import org.json.JSONObject; import java.io.UnsupportedEncodingException; import java.security.MessageDigest; import java.security.NoSuchAlgorithmException; import java.util.ArrayList; import java.util.Collections; import java.util.List; import de.pixart.messenger.utils.Namespace; import de.pixart.messenger.xml.Element; import de.pixart.messenger.xmpp.forms.Data; import de.pixart.messenger.xmpp.forms.Field; import de.pixart.messenger.xmpp.stanzas.IqPacket; public class ServiceDiscoveryResult { public static final String TABLENAME = "discovery_results"; public static final String HASH = "hash"; public static final String VER = "ver"; public static final String RESULT = "result"; protected final String hash; protected final byte[] ver; protected final List features; protected final List forms; private final List identities; public ServiceDiscoveryResult(final IqPacket packet) { this.identities = new ArrayList<>(); this.features = new ArrayList<>(); this.forms = new ArrayList<>(); this.hash = "sha-1"; // We only support sha-1 for now final List elements = packet.query().getChildren(); for (final Element element : elements) { if (element.getName().equals("identity")) { Identity id = new Identity(element); if (id.getType() != null && id.getCategory() != null) { identities.add(id); } } else if (element.getName().equals("feature")) { if (element.getAttribute("var") != null) { features.add(element.getAttribute("var")); } } else if (element.getName().equals("x") && element.getAttribute("xmlns").equals(Namespace.DATA)) { forms.add(Data.parse(element)); } } this.ver = this.mkCapHash(); } private ServiceDiscoveryResult(String hash, byte[] ver, JSONObject o) throws JSONException { this.identities = new ArrayList<>(); this.features = new ArrayList<>(); this.forms = new ArrayList<>(); this.hash = hash; this.ver = ver; JSONArray identities = o.optJSONArray("identities"); if (identities != null) { for (int i = 0; i < identities.length(); i++) { this.identities.add(new Identity(identities.getJSONObject(i))); } } JSONArray features = o.optJSONArray("features"); if (features != null) { for (int i = 0; i < features.length(); i++) { this.features.add(features.getString(i)); } } JSONArray forms = o.optJSONArray("forms"); if (forms != null) { for (int i = 0; i < forms.length(); i++) { this.forms.add(createFormFromJSONObject(forms.getJSONObject(i))); } } } public ServiceDiscoveryResult(Cursor cursor) throws JSONException { this( cursor.getString(cursor.getColumnIndex(HASH)), Base64.decode(cursor.getString(cursor.getColumnIndex(VER)), Base64.DEFAULT), new JSONObject(cursor.getString(cursor.getColumnIndex(RESULT))) ); } private static String clean(String s) { return s.replace("<", "<"); } private static String blankNull(String s) { return s == null ? "" : clean(s); } private static Data createFormFromJSONObject(JSONObject o) { Data data = new Data(); JSONArray names = o.names(); for (int i = 0; i < names.length(); ++i) { try { String name = names.getString(i); JSONArray jsonValues = o.getJSONArray(name); ArrayList values = new ArrayList<>(jsonValues.length()); for (int j = 0; j < jsonValues.length(); ++j) { values.add(jsonValues.getString(j)); } data.put(name, values); } catch (Exception e) { e.printStackTrace(); } } return data; } private static JSONObject createJSONFromForm(Data data) { JSONObject object = new JSONObject(); for (Field field : data.getFields()) { try { JSONArray jsonValues = new JSONArray(); for (String value : field.getValues()) { jsonValues.put(value); } object.put(field.getFieldName(), jsonValues); } catch (Exception e) { e.printStackTrace(); } } try { JSONArray jsonValues = new JSONArray(); jsonValues.put(data.getFormType()); object.put(Data.FORM_TYPE, jsonValues); } catch (Exception e) { e.printStackTrace(); } return object; } public String getVer() { return new String(Base64.encode(this.ver, Base64.DEFAULT)).trim(); } public List getIdentities() { return this.identities; } public List getFeatures() { return this.features; } public boolean hasIdentity(String category, String type) { for (Identity id : this.getIdentities()) { if ((category == null || id.getCategory().equals(category)) && (type == null || id.getType().equals(type))) { return true; } } return false; } public String getExtendedDiscoInformation(String formType, String name) { for (Data form : this.forms) { if (formType.equals(form.getFormType())) { for (Field field : form.getFields()) { if (name.equals(field.getFieldName())) { return field.getValue(); } } } } return null; } private byte[] mkCapHash() { StringBuilder s = new StringBuilder(); List identities = this.getIdentities(); Collections.sort(identities); for (Identity id : identities) { s.append(blankNull(id.getCategory())) .append("/") .append(blankNull(id.getType())) .append("/") .append(blankNull(id.getLang())) .append("/") .append(blankNull(id.getName())) .append("<"); } List features = this.getFeatures(); Collections.sort(features); for (String feature : features) { s.append(clean(feature)).append("<"); } Collections.sort(forms, (lhs, rhs) -> lhs.getFormType().compareTo(rhs.getFormType())); for (Data form : forms) { s.append(clean(form.getFormType())).append("<"); List fields = form.getFields(); Collections.sort(fields, (lhs, rhs) -> lhs.getFieldName().compareTo(rhs.getFieldName())); for (Field field : fields) { s.append(clean(field.getFieldName())).append("<"); List values = field.getValues(); Collections.sort(values); for (String value : values) { s.append(blankNull(value)).append("<"); } } } MessageDigest md; try { md = MessageDigest.getInstance("SHA-1"); } catch (NoSuchAlgorithmException e) { return null; } try { return md.digest(s.toString().getBytes("UTF-8")); } catch (UnsupportedEncodingException e) { return null; } } private JSONObject toJSON() { try { JSONObject o = new JSONObject(); JSONArray ids = new JSONArray(); for (Identity id : this.getIdentities()) { ids.put(id.toJSON()); } o.put("identities", ids); o.put("features", new JSONArray(this.getFeatures())); JSONArray forms = new JSONArray(); for (Data data : this.forms) { forms.put(createJSONFromForm(data)); } o.put("forms", forms); return o; } catch (JSONException e) { return null; } } public ContentValues getContentValues() { final ContentValues values = new ContentValues(); values.put(HASH, this.hash); values.put(VER, getVer()); JSONObject jsonObject = toJSON(); values.put(RESULT, jsonObject == null ? "" : jsonObject.toString()); return values; } public static class Identity implements Comparable { protected final String type; protected final String lang; protected final String name; final String category; Identity(final String category, final String type, final String lang, final String name) { this.category = category; this.type = type; this.lang = lang; this.name = name; } Identity(final Element el) { this( el.getAttribute("category"), el.getAttribute("type"), el.getAttribute("xml:lang"), el.getAttribute("name") ); } Identity(final JSONObject o) { this( o.optString("category", null), o.optString("type", null), o.optString("lang", null), o.optString("name", null) ); } public String getCategory() { return this.category; } public String getType() { return this.type; } public String getLang() { return this.lang; } public String getName() { return this.name; } public int compareTo(@NonNull Object other) { Identity o = (Identity) other; int r = blankNull(this.getCategory()).compareTo(blankNull(o.getCategory())); if (r == 0) { r = blankNull(this.getType()).compareTo(blankNull(o.getType())); } if (r == 0) { r = blankNull(this.getLang()).compareTo(blankNull(o.getLang())); } if (r == 0) { r = blankNull(this.getName()).compareTo(blankNull(o.getName())); } return r; } JSONObject toJSON() { try { JSONObject o = new JSONObject(); o.put("category", this.getCategory()); o.put("type", this.getType()); o.put("lang", this.getLang()); o.put("name", this.getName()); return o; } catch (JSONException e) { return null; } } } }