1package eu.siacs.conversations.generator;
2
3
4import android.util.Base64;
5import android.util.Log;
6
7import org.whispersystems.libaxolotl.IdentityKey;
8import org.whispersystems.libaxolotl.ecc.ECPublicKey;
9import org.whispersystems.libaxolotl.state.PreKeyRecord;
10import org.whispersystems.libaxolotl.state.SignedPreKeyRecord;
11
12import java.security.cert.CertificateEncodingException;
13import java.security.cert.X509Certificate;
14import java.util.ArrayList;
15import java.util.List;
16import java.util.Set;
17
18import eu.siacs.conversations.Config;
19import eu.siacs.conversations.crypto.axolotl.AxolotlService;
20import eu.siacs.conversations.entities.Account;
21import eu.siacs.conversations.entities.Conversation;
22import eu.siacs.conversations.entities.DownloadableFile;
23import eu.siacs.conversations.services.MessageArchiveService;
24import eu.siacs.conversations.services.XmppConnectionService;
25import eu.siacs.conversations.utils.Xmlns;
26import eu.siacs.conversations.xml.Element;
27import eu.siacs.conversations.xmpp.forms.Data;
28import eu.siacs.conversations.xmpp.jid.Jid;
29import eu.siacs.conversations.xmpp.pep.Avatar;
30import eu.siacs.conversations.xmpp.stanzas.IqPacket;
31
32public class IqGenerator extends AbstractGenerator {
33
34 public IqGenerator(final XmppConnectionService service) {
35 super(service);
36 }
37
38 public IqPacket discoResponse(final IqPacket request) {
39 final IqPacket packet = new IqPacket(IqPacket.TYPE.RESULT);
40 packet.setId(request.getId());
41 packet.setTo(request.getFrom());
42 final Element query = packet.addChild("query",
43 "http://jabber.org/protocol/disco#info");
44 query.setAttribute("node", request.query().getAttribute("node"));
45 final Element identity = query.addChild("identity");
46 identity.setAttribute("category", "client");
47 identity.setAttribute("type", IDENTITY_TYPE);
48 identity.setAttribute("name", getIdentityName());
49 for (final String feature : getFeatures()) {
50 query.addChild("feature").setAttribute("var", feature);
51 }
52 return packet;
53 }
54
55 public IqPacket versionResponse(final IqPacket request) {
56 final IqPacket packet = request.generateResponse(IqPacket.TYPE.RESULT);
57 Element query = packet.query("jabber:iq:version");
58 query.addChild("name").setContent(IDENTITY_NAME);
59 query.addChild("version").setContent(getIdentityVersion());
60 return packet;
61 }
62
63 protected IqPacket publish(final String node, final Element item) {
64 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
65 final Element pubsub = packet.addChild("pubsub",
66 "http://jabber.org/protocol/pubsub");
67 final Element publish = pubsub.addChild("publish");
68 publish.setAttribute("node", node);
69 publish.addChild(item);
70 return packet;
71 }
72
73 protected IqPacket retrieve(String node, Element item) {
74 final IqPacket packet = new IqPacket(IqPacket.TYPE.GET);
75 final Element pubsub = packet.addChild("pubsub",
76 "http://jabber.org/protocol/pubsub");
77 final Element items = pubsub.addChild("items");
78 items.setAttribute("node", node);
79 if (item != null) {
80 items.addChild(item);
81 }
82 return packet;
83 }
84
85 public IqPacket publishAvatar(Avatar avatar) {
86 final Element item = new Element("item");
87 item.setAttribute("id", avatar.sha1sum);
88 final Element data = item.addChild("data", "urn:xmpp:avatar:data");
89 data.setContent(avatar.image);
90 return publish("urn:xmpp:avatar:data", item);
91 }
92
93 public IqPacket publishAvatarMetadata(final Avatar avatar) {
94 final Element item = new Element("item");
95 item.setAttribute("id", avatar.sha1sum);
96 final Element metadata = item
97 .addChild("metadata", "urn:xmpp:avatar:metadata");
98 final Element info = metadata.addChild("info");
99 info.setAttribute("bytes", avatar.size);
100 info.setAttribute("id", avatar.sha1sum);
101 info.setAttribute("height", avatar.height);
102 info.setAttribute("width", avatar.height);
103 info.setAttribute("type", avatar.type);
104 return publish("urn:xmpp:avatar:metadata", item);
105 }
106
107 public IqPacket retrievePepAvatar(final Avatar avatar) {
108 final Element item = new Element("item");
109 item.setAttribute("id", avatar.sha1sum);
110 final IqPacket packet = retrieve("urn:xmpp:avatar:data", item);
111 packet.setTo(avatar.owner);
112 return packet;
113 }
114
115 public IqPacket retrieveVcardAvatar(final Avatar avatar) {
116 final IqPacket packet = new IqPacket(IqPacket.TYPE.GET);
117 packet.setTo(avatar.owner);
118 packet.addChild("vCard", "vcard-temp");
119 return packet;
120 }
121
122 public IqPacket retrieveAvatarMetaData(final Jid to) {
123 final IqPacket packet = retrieve("urn:xmpp:avatar:metadata", null);
124 if (to != null) {
125 packet.setTo(to);
126 }
127 return packet;
128 }
129
130 public IqPacket retrieveDeviceIds(final Jid to) {
131 final IqPacket packet = retrieve(AxolotlService.PEP_DEVICE_LIST, null);
132 if(to != null) {
133 packet.setTo(to);
134 }
135 return packet;
136 }
137
138 public IqPacket retrieveBundlesForDevice(final Jid to, final int deviceid) {
139 final IqPacket packet = retrieve(AxolotlService.PEP_BUNDLES+":"+deviceid, null);
140 packet.setTo(to);
141 return packet;
142 }
143
144 public IqPacket retrieveVerificationForDevice(final Jid to, final int deviceid) {
145 final IqPacket packet = retrieve(AxolotlService.PEP_VERIFICATION+":"+deviceid, null);
146 packet.setTo(to);
147 return packet;
148 }
149
150 public IqPacket publishDeviceIds(final Set<Integer> ids) {
151 final Element item = new Element("item");
152 final Element list = item.addChild("list", AxolotlService.PEP_PREFIX);
153 for(Integer id:ids) {
154 final Element device = new Element("device");
155 device.setAttribute("id", id);
156 list.addChild(device);
157 }
158 return publish(AxolotlService.PEP_DEVICE_LIST, item);
159 }
160
161 public IqPacket publishBundles(final SignedPreKeyRecord signedPreKeyRecord, final IdentityKey identityKey,
162 final Set<PreKeyRecord> preKeyRecords, final int deviceId) {
163 final Element item = new Element("item");
164 final Element bundle = item.addChild("bundle", AxolotlService.PEP_PREFIX);
165 final Element signedPreKeyPublic = bundle.addChild("signedPreKeyPublic");
166 signedPreKeyPublic.setAttribute("signedPreKeyId", signedPreKeyRecord.getId());
167 ECPublicKey publicKey = signedPreKeyRecord.getKeyPair().getPublicKey();
168 signedPreKeyPublic.setContent(Base64.encodeToString(publicKey.serialize(),Base64.DEFAULT));
169 final Element signedPreKeySignature = bundle.addChild("signedPreKeySignature");
170 signedPreKeySignature.setContent(Base64.encodeToString(signedPreKeyRecord.getSignature(),Base64.DEFAULT));
171 final Element identityKeyElement = bundle.addChild("identityKey");
172 identityKeyElement.setContent(Base64.encodeToString(identityKey.serialize(), Base64.DEFAULT));
173
174 final Element prekeys = bundle.addChild("prekeys", AxolotlService.PEP_PREFIX);
175 for(PreKeyRecord preKeyRecord:preKeyRecords) {
176 final Element prekey = prekeys.addChild("preKeyPublic");
177 prekey.setAttribute("preKeyId", preKeyRecord.getId());
178 prekey.setContent(Base64.encodeToString(preKeyRecord.getKeyPair().getPublicKey().serialize(), Base64.DEFAULT));
179 }
180
181 return publish(AxolotlService.PEP_BUNDLES+":"+deviceId, item);
182 }
183
184 public IqPacket publishVerification(byte[] signature, X509Certificate[] certificates, final int deviceId) {
185 final Element item = new Element("item");
186 final Element verification = item.addChild("verification", AxolotlService.PEP_PREFIX);
187 final Element chain = verification.addChild("chain");
188 for(int i = 0; i < certificates.length; ++i) {
189 try {
190 Element certificate = chain.addChild("certificate");
191 certificate.setContent(Base64.encodeToString(certificates[i].getEncoded(), Base64.DEFAULT));
192 certificate.setAttribute("index",i);
193 } catch (CertificateEncodingException e) {
194 Log.d(Config.LOGTAG, "could not encode certificate");
195 }
196 }
197 verification.addChild("signature").setContent(Base64.encodeToString(signature, Base64.DEFAULT));
198 return publish(AxolotlService.PEP_VERIFICATION+":"+deviceId, item);
199 }
200
201 public IqPacket queryMessageArchiveManagement(final MessageArchiveService.Query mam) {
202 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
203 final Element query = packet.query("urn:xmpp:mam:0");
204 query.setAttribute("queryid", mam.getQueryId());
205 final Data data = new Data();
206 data.setFormType("urn:xmpp:mam:0");
207 if (mam.muc()) {
208 packet.setTo(mam.getWith());
209 } else if (mam.getWith()!=null) {
210 data.put("with", mam.getWith().toString());
211 }
212 data.put("start", getTimestamp(mam.getStart()));
213 data.put("end", getTimestamp(mam.getEnd()));
214 data.submit();
215 query.addChild(data);
216 if (mam.getPagingOrder() == MessageArchiveService.PagingOrder.REVERSE) {
217 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("before").setContent(mam.getReference());
218 } else if (mam.getReference() != null) {
219 query.addChild("set", "http://jabber.org/protocol/rsm").addChild("after").setContent(mam.getReference());
220 }
221 return packet;
222 }
223 public IqPacket generateGetBlockList() {
224 final IqPacket iq = new IqPacket(IqPacket.TYPE.GET);
225 iq.addChild("blocklist", Xmlns.BLOCKING);
226
227 return iq;
228 }
229
230 public IqPacket generateSetBlockRequest(final Jid jid) {
231 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
232 final Element block = iq.addChild("block", Xmlns.BLOCKING);
233 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
234 return iq;
235 }
236
237 public IqPacket generateSetUnblockRequest(final Jid jid) {
238 final IqPacket iq = new IqPacket(IqPacket.TYPE.SET);
239 final Element block = iq.addChild("unblock", Xmlns.BLOCKING);
240 block.addChild("item").setAttribute("jid", jid.toBareJid().toString());
241 return iq;
242 }
243
244 public IqPacket generateSetPassword(final Account account, final String newPassword) {
245 final IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
246 packet.setTo(account.getServer());
247 final Element query = packet.addChild("query", Xmlns.REGISTER);
248 final Jid jid = account.getJid();
249 query.addChild("username").setContent(jid.getLocalpart());
250 query.addChild("password").setContent(newPassword);
251 return packet;
252 }
253
254 public IqPacket changeAffiliation(Conversation conference, Jid jid, String affiliation) {
255 List<Jid> jids = new ArrayList<>();
256 jids.add(jid);
257 return changeAffiliation(conference,jids,affiliation);
258 }
259
260 public IqPacket changeAffiliation(Conversation conference, List<Jid> jids, String affiliation) {
261 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
262 packet.setTo(conference.getJid().toBareJid());
263 packet.setFrom(conference.getAccount().getJid());
264 Element query = packet.query("http://jabber.org/protocol/muc#admin");
265 for(Jid jid : jids) {
266 Element item = query.addChild("item");
267 item.setAttribute("jid", jid.toString());
268 item.setAttribute("affiliation", affiliation);
269 }
270 return packet;
271 }
272
273 public IqPacket changeRole(Conversation conference, String nick, String role) {
274 IqPacket packet = new IqPacket(IqPacket.TYPE.SET);
275 packet.setTo(conference.getJid().toBareJid());
276 packet.setFrom(conference.getAccount().getJid());
277 Element item = packet.query("http://jabber.org/protocol/muc#admin").addChild("item");
278 item.setAttribute("nick", nick);
279 item.setAttribute("role", role);
280 return packet;
281 }
282
283 public IqPacket requestHttpUploadSlot(Jid host, DownloadableFile file, String mime) {
284 IqPacket packet = new IqPacket(IqPacket.TYPE.GET);
285 packet.setTo(host);
286 Element request = packet.addChild("request",Xmlns.HTTP_UPLOAD);
287 request.addChild("filename").setContent(file.getName());
288 request.addChild("size").setContent(String.valueOf(file.getExpectedSize()));
289 if (mime != null) {
290 request.addChild("content-type").setContent(mime);
291 }
292 return packet;
293 }
294
295 public IqPacket generateCreateAccountWithCaptcha(Account account, String id, Data data) {
296 final IqPacket register = new IqPacket(IqPacket.TYPE.SET);
297
298 register.setTo(account.getServer());
299 register.setId(id);
300 register.query("jabber:iq:register").addChild(data);
301
302 return register;
303 }
304}