1package eu.siacs.conversations.ui;
2
3import android.app.ActionBar;
4import android.content.Intent;
5import android.os.Bundle;
6import android.util.Log;
7import android.view.Gravity;
8import android.view.Menu;
9import android.view.MenuItem;
10import android.view.View;
11import android.view.View.OnClickListener;
12import android.widget.Button;
13import android.widget.CompoundButton;
14import android.widget.LinearLayout;
15import android.widget.TextView;
16import android.widget.Toast;
17
18import com.google.zxing.integration.android.IntentIntegrator;
19
20import org.whispersystems.libsignal.IdentityKey;
21
22import java.util.ArrayList;
23import java.util.Arrays;
24import java.util.HashMap;
25import java.util.List;
26import java.util.Map;
27import java.util.Set;
28import java.util.concurrent.atomic.AtomicBoolean;
29
30import eu.siacs.conversations.Config;
31import eu.siacs.conversations.R;
32import eu.siacs.conversations.crypto.axolotl.AxolotlService;
33import eu.siacs.conversations.crypto.axolotl.FingerprintStatus;
34import eu.siacs.conversations.entities.Account;
35import eu.siacs.conversations.entities.Conversation;
36import eu.siacs.conversations.utils.CryptoHelper;
37import eu.siacs.conversations.utils.XmppUri;
38import eu.siacs.conversations.xmpp.OnKeyStatusUpdated;
39import eu.siacs.conversations.xmpp.jid.InvalidJidException;
40import eu.siacs.conversations.xmpp.jid.Jid;
41
42public class TrustKeysActivity extends OmemoActivity implements OnKeyStatusUpdated {
43 private List<Jid> contactJids;
44
45 private Account mAccount;
46 private Conversation mConversation;
47 private TextView keyErrorMessage;
48 private LinearLayout keyErrorMessageCard;
49 private TextView ownKeysTitle;
50 private LinearLayout ownKeys;
51 private LinearLayout ownKeysCard;
52 private LinearLayout foreignKeys;
53 private Button mSaveButton;
54 private Button mCancelButton;
55
56 private AtomicBoolean mUseCameraHintShown = new AtomicBoolean(false);
57
58 private AxolotlService.FetchStatus lastFetchReport = AxolotlService.FetchStatus.SUCCESS;
59
60 private final Map<String, Boolean> ownKeysToTrust = new HashMap<>();
61 private final Map<Jid,Map<String, Boolean>> foreignKeysToTrust = new HashMap<>();
62
63 private final OnClickListener mSaveButtonListener = new OnClickListener() {
64 @Override
65 public void onClick(View v) {
66 commitTrusts();
67 finishOk();
68 }
69 };
70
71 private final OnClickListener mCancelButtonListener = new OnClickListener() {
72 @Override
73 public void onClick(View v) {
74 setResult(RESULT_CANCELED);
75 finish();
76 }
77 };
78 private Toast mUseCameraHintToast = null;
79
80 @Override
81 protected void refreshUiReal() {
82 invalidateOptionsMenu();
83 populateView();
84 }
85
86 @Override
87 protected void onCreate(final Bundle savedInstanceState) {
88 super.onCreate(savedInstanceState);
89 setContentView(R.layout.activity_trust_keys);
90 this.contactJids = new ArrayList<>();
91 for(String jid : getIntent().getStringArrayExtra("contacts")) {
92 try {
93 this.contactJids.add(Jid.fromString(jid));
94 } catch (InvalidJidException e) {
95 e.printStackTrace();
96 }
97 }
98
99 keyErrorMessageCard = (LinearLayout) findViewById(R.id.key_error_message_card);
100 keyErrorMessage = (TextView) findViewById(R.id.key_error_message);
101 ownKeysTitle = (TextView) findViewById(R.id.own_keys_title);
102 ownKeys = (LinearLayout) findViewById(R.id.own_keys_details);
103 ownKeysCard = (LinearLayout) findViewById(R.id.own_keys_card);
104 foreignKeys = (LinearLayout) findViewById(R.id.foreign_keys);
105 mCancelButton = (Button) findViewById(R.id.cancel_button);
106 mCancelButton.setOnClickListener(mCancelButtonListener);
107 mSaveButton = (Button) findViewById(R.id.save_button);
108 mSaveButton.setOnClickListener(mSaveButtonListener);
109
110
111 if (getActionBar() != null) {
112 getActionBar().setHomeButtonEnabled(true);
113 getActionBar().setDisplayHomeAsUpEnabled(true);
114 }
115
116 if (savedInstanceState != null) {
117 mUseCameraHintShown.set(savedInstanceState.getBoolean("camera_hint_shown",false));
118 }
119 }
120
121 @Override
122 public void onSaveInstanceState(Bundle savedInstanceState) {
123 savedInstanceState.putBoolean("camera_hint_shown", mUseCameraHintShown.get());
124 super.onSaveInstanceState(savedInstanceState);
125 }
126
127 @Override
128 public boolean onCreateOptionsMenu(Menu menu) {
129 getMenuInflater().inflate(R.menu.trust_keys, menu);
130 MenuItem scanQrCode = menu.findItem(R.id.action_scan_qr_code);
131 scanQrCode.setVisible(ownKeysToTrust.size() > 0 || foreignActuallyHasKeys());
132 return super.onCreateOptionsMenu(menu);
133 }
134
135 private void showCameraToast() {
136 mUseCameraHintToast = Toast.makeText(this,R.string.use_camera_icon_to_scan_barcode,Toast.LENGTH_LONG);
137 ActionBar actionBar = getActionBar();
138 mUseCameraHintToast.setGravity(Gravity.TOP | Gravity.END, 0 ,actionBar == null ? 0 : actionBar.getHeight());
139 mUseCameraHintToast.show();
140 }
141
142 @Override
143 public boolean onOptionsItemSelected(MenuItem item) {
144 switch (item.getItemId()) {
145 case R.id.action_scan_qr_code:
146 if (hasPendingKeyFetches()) {
147 Toast.makeText(this, R.string.please_wait_for_keys_to_be_fetched, Toast.LENGTH_SHORT).show();
148 } else {
149 new IntentIntegrator(this).initiateScan(Arrays.asList("AZTEC","QR_CODE"));
150 return true;
151 }
152 }
153 return super.onOptionsItemSelected(item);
154 }
155
156 @Override
157 protected void onStop() {
158 super.onStop();
159 if (mUseCameraHintToast != null) {
160 mUseCameraHintToast.cancel();
161 }
162 }
163
164 @Override
165 protected void processFingerprintVerification(XmppUri uri) {
166 if (mConversation != null
167 && mAccount != null
168 && uri.hasFingerprints()
169 && mAccount.getAxolotlService().getCryptoTargets(mConversation).contains(uri.getJid())) {
170 boolean performedVerification = xmppConnectionService.verifyFingerprints(mAccount.getRoster().getContact(uri.getJid()),uri.getFingerprints());
171 boolean keys = reloadFingerprints();
172 if (performedVerification && !keys && !hasNoOtherTrustedKeys() && !hasPendingKeyFetches()) {
173 Toast.makeText(this,R.string.all_omemo_keys_have_been_verified, Toast.LENGTH_SHORT).show();
174 finishOk();
175 return;
176 } else if (performedVerification) {
177 Toast.makeText(this,R.string.verified_fingerprints,Toast.LENGTH_SHORT).show();
178 }
179 } else {
180 reloadFingerprints();
181 Log.d(Config.LOGTAG,"xmpp uri was: "+uri.getJid()+" has Fingerprints: "+Boolean.toString(uri.hasFingerprints()));
182 Toast.makeText(this,R.string.barcode_does_not_contain_fingerprints_for_this_conversation,Toast.LENGTH_SHORT).show();
183 }
184 populateView();
185 }
186
187 private void populateView() {
188 setTitle(getString(R.string.trust_omemo_fingerprints));
189 ownKeys.removeAllViews();
190 foreignKeys.removeAllViews();
191 boolean hasOwnKeys = false;
192 boolean hasForeignKeys = false;
193 for(final String fingerprint : ownKeysToTrust.keySet()) {
194 hasOwnKeys = true;
195 addFingerprintRowWithListeners(ownKeys, mAccount, fingerprint, false,
196 FingerprintStatus.createActive(ownKeysToTrust.get(fingerprint)), false, false,
197 new CompoundButton.OnCheckedChangeListener() {
198 @Override
199 public void onCheckedChanged(CompoundButton buttonView, boolean isChecked) {
200 ownKeysToTrust.put(fingerprint, isChecked);
201 // own fingerprints have no impact on locked status.
202 }
203 }
204 );
205 }
206
207 synchronized (this.foreignKeysToTrust) {
208 for (Map.Entry<Jid, Map<String, Boolean>> entry : foreignKeysToTrust.entrySet()) {
209 hasForeignKeys = true;
210 final LinearLayout layout = (LinearLayout) getLayoutInflater().inflate(R.layout.keys_card, foreignKeys, false);
211 final Jid jid = entry.getKey();
212 final TextView header = (TextView) layout.findViewById(R.id.foreign_keys_title);
213 final LinearLayout keysContainer = (LinearLayout) layout.findViewById(R.id.foreign_keys_details);
214 final TextView informNoKeys = (TextView) layout.findViewById(R.id.no_keys_to_accept);
215 header.setText(jid.toString());
216 final Map<String, Boolean> fingerprints = entry.getValue();
217 for (final String fingerprint : fingerprints.keySet()) {
218 addFingerprintRowWithListeners(keysContainer, mAccount, fingerprint, false,
219 FingerprintStatus.createActive(fingerprints.get(fingerprint)), false, false,
220 new CompoundButton.OnCheckedChangeListener() {
221 @Override
222 public void onCheckedChanged(CompoundButton buttonView, boolean isChecked) {
223 fingerprints.put(fingerprint, isChecked);
224 lockOrUnlockAsNeeded();
225 }
226 }
227 );
228 }
229 if (fingerprints.size() == 0) {
230 informNoKeys.setVisibility(View.VISIBLE);
231 informNoKeys.setText(getString(R.string.no_keys_just_confirm,mAccount.getRoster().getContact(jid).getDisplayName()));
232 } else {
233 informNoKeys.setVisibility(View.GONE);
234 }
235 foreignKeys.addView(layout);
236 }
237 }
238
239 if ((hasOwnKeys || foreignActuallyHasKeys()) && mUseCameraHintShown.compareAndSet(false,true)) {
240 showCameraToast();
241 }
242
243 ownKeysTitle.setText(mAccount.getJid().toBareJid().toString());
244 ownKeysCard.setVisibility(hasOwnKeys ? View.VISIBLE : View.GONE);
245 foreignKeys.setVisibility(hasForeignKeys ? View.VISIBLE : View.GONE);
246 if(hasPendingKeyFetches()) {
247 setFetching();
248 lock();
249 } else {
250 if (!hasForeignKeys && hasNoOtherTrustedKeys()) {
251 keyErrorMessageCard.setVisibility(View.VISIBLE);
252 if (lastFetchReport == AxolotlService.FetchStatus.ERROR
253 || mAccount.getAxolotlService().fetchMapHasErrors(contactJids)) {
254 keyErrorMessage.setText(R.string.error_no_keys_to_trust_server_error);
255 } else {
256 keyErrorMessage.setText(R.string.error_no_keys_to_trust);
257 }
258 ownKeys.removeAllViews();
259 ownKeysCard.setVisibility(View.GONE);
260 foreignKeys.removeAllViews();
261 foreignKeys.setVisibility(View.GONE);
262 }
263 lockOrUnlockAsNeeded();
264 setDone();
265 }
266 }
267
268 private boolean foreignActuallyHasKeys() {
269 synchronized (this.foreignKeysToTrust) {
270 for (Map.Entry<Jid, Map<String, Boolean>> entry : foreignKeysToTrust.entrySet()) {
271 if (entry.getValue().size() > 0) {
272 return true;
273 }
274 }
275 }
276 return false;
277 }
278
279 private boolean reloadFingerprints() {
280 List<Jid> acceptedTargets = mConversation == null ? new ArrayList<Jid>() : mConversation.getAcceptedCryptoTargets();
281 ownKeysToTrust.clear();
282 AxolotlService service = this.mAccount.getAxolotlService();
283 Set<IdentityKey> ownKeysSet = service.getKeysWithTrust(FingerprintStatus.createActiveUndecided());
284 for(final IdentityKey identityKey : ownKeysSet) {
285 final String fingerprint = CryptoHelper.bytesToHex(identityKey.getPublicKey().serialize());
286 if(!ownKeysToTrust.containsKey(fingerprint)) {
287 ownKeysToTrust.put(fingerprint, false);
288 }
289 }
290 synchronized (this.foreignKeysToTrust) {
291 foreignKeysToTrust.clear();
292 for (Jid jid : contactJids) {
293 Set<IdentityKey> foreignKeysSet = service.getKeysWithTrust(FingerprintStatus.createActiveUndecided(), jid);
294 if (hasNoOtherTrustedKeys(jid) && ownKeysSet.size() == 0) {
295 foreignKeysSet.addAll(service.getKeysWithTrust(FingerprintStatus.createActive(false), jid));
296 }
297 Map<String, Boolean> foreignFingerprints = new HashMap<>();
298 for (final IdentityKey identityKey : foreignKeysSet) {
299 final String fingerprint = CryptoHelper.bytesToHex(identityKey.getPublicKey().serialize());
300 if (!foreignFingerprints.containsKey(fingerprint)) {
301 foreignFingerprints.put(fingerprint, false);
302 }
303 }
304 if (foreignFingerprints.size() > 0 || !acceptedTargets.contains(jid)) {
305 foreignKeysToTrust.put(jid, foreignFingerprints);
306 }
307 }
308 }
309 return ownKeysSet.size() + foreignKeysToTrust.size() > 0;
310 }
311
312 public void onBackendConnected() {
313 Intent intent = getIntent();
314 this.mAccount = extractAccount(intent);
315 if (this.mAccount != null && intent != null) {
316 String uuid = intent.getStringExtra("conversation");
317 this.mConversation = xmppConnectionService.findConversationByUuid(uuid);
318 if (this.mPendingFingerprintVerificationUri != null) {
319 processFingerprintVerification(this.mPendingFingerprintVerificationUri);
320 this.mPendingFingerprintVerificationUri = null;
321 } else {
322 reloadFingerprints();
323 populateView();
324 invalidateOptionsMenu();
325 }
326 }
327 }
328
329 private boolean hasNoOtherTrustedKeys() {
330 return mAccount == null || mAccount.getAxolotlService().anyTargetHasNoTrustedKeys(contactJids);
331 }
332
333 private boolean hasNoOtherTrustedKeys(Jid contact) {
334 return mAccount == null || mAccount.getAxolotlService().getNumTrustedKeys(contact) == 0;
335 }
336
337 private boolean hasPendingKeyFetches() {
338 return mAccount != null && mAccount.getAxolotlService().hasPendingKeyFetches(mAccount, contactJids);
339 }
340
341
342 @Override
343 public void onKeyStatusUpdated(final AxolotlService.FetchStatus report) {
344 final boolean keysToTrust = reloadFingerprints();
345 if (report != null) {
346 lastFetchReport = report;
347 runOnUiThread(new Runnable() {
348 @Override
349 public void run() {
350 if (mUseCameraHintToast != null && !keysToTrust) {
351 mUseCameraHintToast.cancel();
352 }
353 switch (report) {
354 case ERROR:
355 Toast.makeText(TrustKeysActivity.this,R.string.error_fetching_omemo_key,Toast.LENGTH_SHORT).show();
356 break;
357 case SUCCESS_TRUSTED:
358 Toast.makeText(TrustKeysActivity.this,R.string.blindly_trusted_omemo_keys,Toast.LENGTH_LONG).show();
359 break;
360 case SUCCESS_VERIFIED:
361 Toast.makeText(TrustKeysActivity.this,
362 Config.X509_VERIFICATION ? R.string.verified_omemo_key_with_certificate : R.string.all_omemo_keys_have_been_verified,
363 Toast.LENGTH_LONG).show();
364 break;
365 }
366 }
367 });
368
369 }
370 if (keysToTrust || hasPendingKeyFetches() || hasNoOtherTrustedKeys()) {
371 refreshUi();
372 } else {
373 runOnUiThread(new Runnable() {
374 @Override
375 public void run() {
376 finishOk();
377 }
378 });
379
380 }
381 }
382
383 private void finishOk() {
384 Intent data = new Intent();
385 data.putExtra("choice", getIntent().getIntExtra("choice", ConversationActivity.ATTACHMENT_CHOICE_INVALID));
386 setResult(RESULT_OK, data);
387 finish();
388 }
389
390 private void commitTrusts() {
391 for(final String fingerprint :ownKeysToTrust.keySet()) {
392 mAccount.getAxolotlService().setFingerprintTrust(
393 fingerprint,
394 FingerprintStatus.createActive(ownKeysToTrust.get(fingerprint)));
395 }
396 List<Jid> acceptedTargets = mConversation == null ? new ArrayList<Jid>() : mConversation.getAcceptedCryptoTargets();
397 synchronized (this.foreignKeysToTrust) {
398 for (Map.Entry<Jid, Map<String, Boolean>> entry : foreignKeysToTrust.entrySet()) {
399 Jid jid = entry.getKey();
400 Map<String, Boolean> value = entry.getValue();
401 if (!acceptedTargets.contains(jid)) {
402 acceptedTargets.add(jid);
403 }
404 for (final String fingerprint : value.keySet()) {
405 mAccount.getAxolotlService().setFingerprintTrust(
406 fingerprint,
407 FingerprintStatus.createActive(value.get(fingerprint)));
408 }
409 }
410 }
411 if (mConversation != null && mConversation.getMode() == Conversation.MODE_MULTI) {
412 mConversation.setAcceptedCryptoTargets(acceptedTargets);
413 xmppConnectionService.updateConversation(mConversation);
414 }
415 }
416
417 private void unlock() {
418 mSaveButton.setEnabled(true);
419 mSaveButton.setTextColor(getPrimaryTextColor());
420 }
421
422 private void lock() {
423 mSaveButton.setEnabled(false);
424 mSaveButton.setTextColor(getSecondaryTextColor());
425 }
426
427 private void lockOrUnlockAsNeeded() {
428 synchronized (this.foreignKeysToTrust) {
429 for (Jid jid : contactJids) {
430 Map<String, Boolean> fingerprints = foreignKeysToTrust.get(jid);
431 if (hasNoOtherTrustedKeys(jid) && (fingerprints == null || !fingerprints.values().contains(true))) {
432 lock();
433 return;
434 }
435 }
436 }
437 unlock();
438
439 }
440
441 private void setDone() {
442 mSaveButton.setText(getString(R.string.done));
443 }
444
445 private void setFetching() {
446 mSaveButton.setText(getString(R.string.fetching_keys));
447 }
448}