conversations-classic/src/main/java/eu/siacs/conversations/entities/ServiceDiscoveryResult.java

349 lines
9 KiB
Java
Raw Normal View History

package eu.siacs.conversations.entities;
import android.content.ContentValues;
import android.database.Cursor;
import android.support.annotation.NonNull;
import android.util.Base64;
import android.util.Log;
import java.io.UnsupportedEncodingException;
import java.lang.Comparable;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.Collections;
2016-02-03 16:19:05 +00:00
import java.util.Comparator;
import java.util.List;
2018-02-16 14:57:02 +00:00
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import eu.siacs.conversations.Config;
import eu.siacs.conversations.xml.Element;
2018-02-16 14:57:02 +00:00
import eu.siacs.conversations.xml.Namespace;
2016-02-03 16:19:05 +00:00
import eu.siacs.conversations.xmpp.forms.Data;
import eu.siacs.conversations.xmpp.forms.Field;
import eu.siacs.conversations.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<String> features;
2016-02-03 16:19:05 +00:00
protected final List<Data> forms;
private final List<Identity> identities;
public ServiceDiscoveryResult(final IqPacket packet) {
this.identities = new ArrayList<>();
this.features = new ArrayList<>();
2016-02-03 16:19:05 +00:00
this.forms = new ArrayList<>();
this.hash = "sha-1"; // We only support sha-1 for now
final List<Element> 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"));
}
2018-02-16 14:57:02 +00:00
} else if (element.getName().equals("x") && element.getAttribute("xmlns").equals(Namespace.DATA)) {
2016-02-03 16:19:05 +00:00
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<>();
2016-02-03 16:19:05 +00:00
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("<","&lt;");
}
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<String> 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<Identity> getIdentities() {
return this.identities;
}
public List<String> 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<Identity> 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<String> 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()));
2016-02-03 16:19:05 +00:00
for (Data form : forms) {
s.append(clean(form.getFormType())).append("<");
List<Field> fields = form.getFields();
Collections.sort(fields, (lhs, rhs) -> lhs.getFieldName().compareTo(rhs.getFieldName()));
for (Field field : fields) {
s.append(clean(field.getFieldName())).append("<");
List<String> values = field.getValues();
Collections.sort(values);
for (String value : values) {
s.append(blankNull(value)).append("<");
}
}
2016-02-03 16:19:05 +00:00
}
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;
}
}
}
}