package eu.faircode.email; /* This file is part of FairEmail. FairEmail is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. FairEmail is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with FairEmail. If not, see . Copyright 2018-2023 by Marcel Bokhorst (M66B) */ import android.app.Dialog; import android.content.Context; import android.content.Intent; import android.content.SharedPreferences; import android.graphics.Bitmap; import android.graphics.Typeface; import android.net.Uri; import android.os.Bundle; import android.text.TextUtils; import android.view.LayoutInflater; import android.view.Menu; import android.view.MenuItem; import android.view.View; import android.view.ViewGroup; import android.widget.Button; import android.widget.ImageButton; import android.widget.ImageView; import android.widget.ProgressBar; import android.widget.TextView; import android.widget.Toast; import androidx.annotation.NonNull; import androidx.annotation.Nullable; import androidx.appcompat.app.AlertDialog; import androidx.appcompat.widget.PopupMenu; import androidx.core.app.ShareCompat; import androidx.core.content.FileProvider; import androidx.fragment.app.Fragment; import androidx.lifecycle.Lifecycle; import androidx.lifecycle.LifecycleObserver; import androidx.lifecycle.LifecycleOwner; import androidx.lifecycle.OnLifecycleEvent; import androidx.localbroadcastmanager.content.LocalBroadcastManager; import androidx.preference.PreferenceManager; import androidx.recyclerview.widget.DiffUtil; import androidx.recyclerview.widget.LinearLayoutManager; import androidx.recyclerview.widget.ListUpdateCallback; import androidx.recyclerview.widget.RecyclerView; import java.io.File; import java.text.NumberFormat; import java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import java.util.List; import javax.mail.Part; public class AdapterAttachment extends RecyclerView.Adapter { private Fragment parentFragment; private Context context; private LifecycleOwner owner; private LayoutInflater inflater; private boolean readonly; private AdapterMessage.IProperties properties; private boolean vt_enabled; private String vt_apikey; private boolean debug; private int dp12; private int dp36; private int textColorTertiary; private int colorWarning; private List items = new ArrayList<>(); public class ViewHolder extends RecyclerView.ViewHolder implements View.OnClickListener, View.OnLongClickListener { private View view; private ImageButton ibDelete; private ImageView ivType; private ImageView ivDisposition; private TextView tvName; private TextView tvSize; private ImageView ivStatus; private ImageButton ibSave; private ImageButton ibScan; private TextView tvType; private TextView tvError; private ProgressBar progressbar; private TwoStateOwner powner = new TwoStateOwner(owner, "AttachmentPopup"); ViewHolder(View itemView) { super(itemView); view = itemView.findViewById(R.id.clItem); ibDelete = itemView.findViewById(R.id.ibDelete); ivType = itemView.findViewById(R.id.ivType); tvName = itemView.findViewById(R.id.tvName); tvSize = itemView.findViewById(R.id.tvSize); ivStatus = itemView.findViewById(R.id.ivStatus); ibSave = itemView.findViewById(R.id.ibSave); ibScan = itemView.findViewById(R.id.ibScan); tvType = itemView.findViewById(R.id.tvType); ivDisposition = itemView.findViewById(R.id.ivDisposition); tvError = itemView.findViewById(R.id.tvError); progressbar = itemView.findViewById(R.id.progressbar); } private void wire() { view.setOnClickListener(this); ibDelete.setOnClickListener(this); ibSave.setOnClickListener(this); ibScan.setOnClickListener(this); view.setOnLongClickListener(this); } private void unwire() { view.setOnClickListener(null); ibDelete.setOnClickListener(null); ibSave.setOnClickListener(null); ibScan.setOnClickListener(null); view.setOnLongClickListener(null); } private void bindTo(EntityAttachment attachment) { view.setAlpha(!attachment.isAttachment() ? Helper.LOW_LIGHT : 1.0f); ViewGroup.MarginLayoutParams lparam = (ViewGroup.MarginLayoutParams) view.getLayoutParams(); lparam.setMarginStart(attachment.subsequence == null ? 0 : dp12); view.setLayoutParams(lparam); ibDelete.setVisibility(readonly ? View.GONE : View.VISIBLE); if (!readonly && attachment.isImage()) { if (attachment.available) { Bitmap bm = ImageHelper.decodeImage( attachment.getFile(context), attachment.getMimeType(), dp36); if (bm == null) ivType.setImageResource(R.drawable.twotone_broken_image_24); else ivType.setImageBitmap(bm); } else ivType.setImageResource(R.drawable.twotone_hourglass_top_24); ivDisposition.setVisibility(View.GONE); } else { int resid = 0; String extension = Helper.guessExtension(attachment.getMimeType()); if (extension != null) resid = context.getResources().getIdentifier("file_" + extension, "drawable", context.getPackageName()); if (resid == 0) ivType.setImageDrawable(null); else ivType.setImageResource(resid); ivDisposition.setImageLevel(Part.INLINE.equals(attachment.disposition) ? 1 : 0); ivDisposition.setVisibility( Part.ATTACHMENT.equals(attachment.disposition) || Part.INLINE.equals(attachment.disposition) ? View.VISIBLE : View.INVISIBLE); } boolean dangerous = Helper.DANGEROUS_EXTENSIONS.contains(Helper.getExtension(attachment.name)); tvName.setText(attachment.name); tvName.setTextColor(dangerous ? colorWarning : textColorTertiary); tvName.setTypeface(null, dangerous ? Typeface.BOLD : Typeface.NORMAL); if (attachment.size != null) tvSize.setText(Helper.humanReadableByteCount(attachment.size)); tvSize.setVisibility(attachment.size == null ? View.GONE : View.VISIBLE); if (attachment.available) { ivStatus.setImageResource(R.drawable.twotone_visibility_24); ivStatus.setVisibility(View.VISIBLE); } else { if (attachment.progress == null) { ivStatus.setImageResource(R.drawable.twotone_cloud_download_24); ivStatus.setVisibility(View.VISIBLE); } else ivStatus.setVisibility(View.GONE); } ibSave.setVisibility(attachment.available ? View.VISIBLE : View.GONE); ibScan.setVisibility(attachment.available && vt_enabled && !TextUtils.isEmpty(vt_apikey) && !BuildConfig.PLAY_STORE_RELEASE ? View.VISIBLE : View.GONE); if (attachment.progress != null) progressbar.setProgress(attachment.progress); progressbar.setVisibility( attachment.progress == null || attachment.available ? View.GONE : View.VISIBLE); StringBuilder sb = new StringBuilder(); sb.append(attachment.type); if (debug || BuildConfig.DEBUG) { if (attachment.cid != null) { sb.append(' ').append(attachment.cid); if (attachment.related != null) sb.append(' ').append(attachment.related); } if (attachment.isEncryption()) sb.append(' ').append(attachment.encryption); } tvType.setText(sb.toString()); tvError.setText(attachment.error); tvError.setVisibility(attachment.error == null ? View.GONE : View.VISIBLE); if (properties != null) { String aid = properties.getValue("attachment"); if (aid != null) { if (attachment.id.equals(Long.parseLong(aid)) && attachment.available) { properties.setValue("attachment", null); onView(attachment); } } } } @Override public void onClick(View view) { int pos = getAdapterPosition(); if (pos == RecyclerView.NO_POSITION) return; EntityAttachment attachment = items.get(pos); if (attachment == null) return; int id = view.getId(); if (id == R.id.ibDelete) onDelete(attachment); else if (id == R.id.ibSave) onSave(attachment); else if (id == R.id.ibScan) onScan(attachment); else { if (attachment.available) onView(attachment); else { if (attachment.progress == null) if (attachment.subsequence == null) onDownload(attachment); else if (!TextUtils.isEmpty(attachment.error)) ToastEx.makeText(context, attachment.error, Toast.LENGTH_LONG).show(); } } } @Override public boolean onLongClick(View view) { int pos = getAdapterPosition(); if (pos == RecyclerView.NO_POSITION) return false; EntityAttachment attachment = items.get(pos); if (attachment == null || !attachment.available) return false; if (readonly) return onShare(attachment); else { PopupMenuLifecycle popupMenu = new PopupMenuLifecycle(context, powner, view); popupMenu.getMenu().add(Menu.NONE, R.string.title_share, 1, R.string.title_share); popupMenu.getMenu().add(Menu.NONE, R.string.title_zip, 2, R.string.title_zip) .setEnabled(!attachment.isInline() && !attachment.isCompressed()); popupMenu.setOnMenuItemClickListener(new PopupMenu.OnMenuItemClickListener() { @Override public boolean onMenuItemClick(MenuItem item) { int itemId = item.getItemId(); if (itemId == R.string.title_share) return onShare(attachment); else if (itemId == R.string.title_zip) return onZip(attachment); return false; } }); popupMenu.show(); return true; } } private boolean onShare(final EntityAttachment attachment) { try { new ShareCompat.IntentBuilder(context) .setType(attachment.getMimeType()) .addStream(attachment.getUri(context)) .setChooserTitle(R.string.title_select_app) .startChooser(); return true; } catch (Throwable ex) { /* java.lang.IllegalArgumentException: Failed to resolve canonical path for ... at androidx.core.content.FileProvider$SimplePathStrategy.getUriForFile(SourceFile:730) at androidx.core.content.FileProvider.getUriForFile(SourceFile:418) */ Log.unexpectedError(parentFragment.getParentFragmentManager(), ex); return false; } } private boolean onZip(final EntityAttachment attachment) { Bundle args = new Bundle(); args.putLong("id", attachment.id); new SimpleTask() { @Override protected Void onExecute(Context context, Bundle args) throws Throwable { long id = args.getLong("id"); DB db = DB.getInstance(context); EntityAttachment attachment = db.attachment().getAttachment(id); if (attachment != null) attachment.zip(context); return null; } @Override protected void onException(Bundle args, Throwable ex) { Log.unexpectedError(parentFragment.getParentFragmentManager(), ex); } }.execute(context, owner, args, "attachment:zip"); return true; } private void onDelete(final EntityAttachment attachment) { Bundle args = new Bundle(); args.putLong("id", attachment.id); new SimpleTask() { @Override protected Void onExecute(Context context, Bundle args) { long id = args.getLong("id"); EntityAttachment attachment; DB db = DB.getInstance(context); try { db.beginTransaction(); attachment = db.attachment().getAttachment(id); if (attachment == null) return null; db.attachment().deleteAttachment(attachment.id); db.setTransactionSuccessful(); } finally { db.endTransaction(); } attachment.getFile(context).delete(); return null; } @Override protected void onException(Bundle args, Throwable ex) { Log.unexpectedError(parentFragment.getParentFragmentManager(), ex); } }.execute(context, owner, args, "attachment:delete"); } private void onSave(EntityAttachment attachment) { LocalBroadcastManager lbm = LocalBroadcastManager.getInstance(context); lbm.sendBroadcast( new Intent(FragmentBase.ACTION_STORE_ATTACHMENT) .putExtra("id", attachment.id) .putExtra("name", Helper.sanitizeFilename(attachment.name)) .putExtra("type", attachment.getMimeType())); } private void onScan(EntityAttachment attachment) { Bundle args = new Bundle(); args.putString("apiKey", vt_apikey); args.putString("name", attachment.name); args.putSerializable("file", attachment.getFile(context)); FragmentDialogVirusTotal fragment = new FragmentDialogVirusTotal(); fragment.setArguments(args); fragment.show(parentFragment.getParentFragmentManager(), "attachment:scan"); } private void onView(EntityAttachment attachment) { try { String title = (attachment.name == null ? attachment.cid : attachment.name); Helper.share(context, attachment.getFile(context), attachment.getMimeType(), title); } catch (Throwable ex) { Log.unexpectedError(parentFragment.getParentFragmentManager(), ex); } } private void onDownload(EntityAttachment attachment) { if (properties != null) properties.setValue("attachment", Long.toString(attachment.id)); Bundle args = new Bundle(); args.putLong("id", attachment.id); args.putLong("message", attachment.message); args.putInt("sequence", attachment.sequence); new SimpleTask() { @Override protected Void onExecute(Context context, Bundle args) { long id = args.getLong("id"); long mid = args.getLong("message"); Long reload = null; DB db = DB.getInstance(context); try { db.beginTransaction(); EntityMessage message = db.message().getMessage(mid); if (message == null) return null; EntityAccount account = db.account().getAccount(message.account); if (account == null) return null; if (account.protocol == EntityAccount.TYPE_IMAP && message.uid == null) return null; if (!"connected".equals(account.state) && !account.isTransient(context)) reload = account.id; EntityAttachment attachment = db.attachment().getAttachment(id); if (attachment == null || attachment.progress != null || attachment.available) return null; EntityOperation.queue(context, message, EntityOperation.ATTACHMENT, id); db.setTransactionSuccessful(); } finally { db.endTransaction(); } if (reload == null) ServiceSynchronize.eval(context, "attachment"); else ServiceSynchronize.reload(context, reload, true, "attachment"); return null; } @Override protected void onException(Bundle args, Throwable ex) { Log.unexpectedError(parentFragment.getParentFragmentManager(), ex); } }.execute(context, owner, args, "attachment:fetch"); } } AdapterAttachment(Fragment parentFragment, boolean readonly, final AdapterMessage.IProperties properties) { this.parentFragment = parentFragment; this.readonly = readonly; this.properties = properties; this.context = parentFragment.getContext(); this.owner = parentFragment.getViewLifecycleOwner(); this.inflater = LayoutInflater.from(context); SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context); this.vt_enabled = prefs.getBoolean("vt_enabled", false); this.vt_apikey = prefs.getString("vt_apikey", null); this.debug = prefs.getBoolean("debug", false); this.dp12 = Helper.dp2pixels(context, 12); this.dp36 = Helper.dp2pixels(context, 36); this.textColorTertiary = Helper.resolveColor(context, android.R.attr.textColorTertiary); this.colorWarning = Helper.resolveColor(context, R.attr.colorWarning); setHasStableIds(true); owner.getLifecycle().addObserver(new LifecycleObserver() { @OnLifecycleEvent(Lifecycle.Event.ON_DESTROY) public void onDestroyed() { Log.d(AdapterAttachment.this + " parent destroyed"); AdapterAttachment.this.parentFragment = null; owner.getLifecycle().removeObserver(this); } }); } public void set(@NonNull List attachments) { Log.i("Set attachments=" + attachments.size()); Collections.sort(attachments, new Comparator() { @Override public int compare(EntityAttachment a1, EntityAttachment a2) { return a1.sequence.compareTo(a2.sequence); } }); DiffUtil.DiffResult diff = DiffUtil.calculateDiff(new DiffCallback(items, attachments), false); items = attachments; diff.dispatchUpdatesTo(new ListUpdateCallback() { @Override public void onInserted(int position, int count) { Log.d("Inserted @" + position + " #" + count); } @Override public void onRemoved(int position, int count) { Log.d("Removed @" + position + " #" + count); } @Override public void onMoved(int fromPosition, int toPosition) { Log.d("Moved " + fromPosition + ">" + toPosition); } @Override public void onChanged(int position, int count, Object payload) { Log.d("Changed @" + position + " #" + count); } }); try { diff.dispatchUpdatesTo(this); } catch (Throwable ex) { Log.e(ex); } } private static class DiffCallback extends DiffUtil.Callback { private List prev = new ArrayList<>(); private List next = new ArrayList<>(); DiffCallback(List prev, List next) { this.prev.addAll(prev); this.next.addAll(next); } @Override public int getOldListSize() { return prev.size(); } @Override public int getNewListSize() { return next.size(); } @Override public boolean areItemsTheSame(int oldItemPosition, int newItemPosition) { EntityAttachment a1 = prev.get(oldItemPosition); EntityAttachment a2 = next.get(newItemPosition); return a1.id.equals(a2.id); } @Override public boolean areContentsTheSame(int oldItemPosition, int newItemPosition) { EntityAttachment a1 = prev.get(oldItemPosition); EntityAttachment a2 = next.get(newItemPosition); return a1.equals(a2); } } @Override public long getItemId(int position) { return items.get(position).id; } @Override public int getItemCount() { return items.size(); } @Override @NonNull public ViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) { return new ViewHolder(inflater.inflate(R.layout.item_attachment, parent, false)); } @Override public void onBindViewHolder(@NonNull ViewHolder holder, int position) { holder.unwire(); EntityAttachment attachment = items.get(position); holder.powner.recreate(attachment == null ? null : attachment.id); holder.bindTo(attachment); holder.wire(); } public static class FragmentDialogVirusTotal extends FragmentDialogBase { @NonNull @Override public Dialog onCreateDialog(@Nullable Bundle savedInstanceState) { Bundle args = getArguments(); String apiKey = args.getString("apiKey"); String name = args.getString("name"); final Context context = getContext(); View view = LayoutInflater.from(context).inflate(R.layout.dialog_virus_total, null); final TextView tvName = view.findViewById(R.id.tvName); final TextView tvError = view.findViewById(R.id.tvError); final TextView tvUnknown = view.findViewById(R.id.tvUnknown); final TextView tvSummary = view.findViewById(R.id.tvSummary); final TextView tvLabel = view.findViewById(R.id.tvLabel); final TextView tvReport = view.findViewById(R.id.tvReport); final RecyclerView rvScan = view.findViewById(R.id.rvScan); final Button btnUpload = view.findViewById(R.id.btnUpload); final ProgressBar pbUpload = view.findViewById(R.id.pbUpload); final TextView tvAnalyzing = view.findViewById(R.id.tvAnalyzing); final TextView tvPrivacy = view.findViewById(R.id.tvPrivacy); final ProgressBar pbWait = view.findViewById(R.id.pbWait); tvName.setText(name); tvName.setVisibility(TextUtils.isEmpty(name) ? View.GONE : View.VISIBLE); tvError.setVisibility(View.GONE); tvUnknown.setVisibility(View.GONE); tvSummary.setVisibility(View.GONE); tvLabel.setVisibility(View.GONE); tvReport.setVisibility(View.GONE); tvReport.getPaint().setUnderlineText(true); rvScan.setHasFixedSize(false); LinearLayoutManager llm = new LinearLayoutManager(getContext()); rvScan.setLayoutManager(llm); final AdapterVirusTotal adapter = new AdapterVirusTotal(getContext(), getViewLifecycleOwner()); rvScan.setAdapter(adapter); rvScan.setVisibility(View.GONE); btnUpload.setVisibility(View.GONE); pbUpload.setVisibility(View.GONE); tvAnalyzing.setVisibility(View.GONE); tvPrivacy.setVisibility(View.GONE); tvPrivacy.getPaint().setUnderlineText(true); pbWait.setVisibility(View.GONE); final SimpleTask taskLookup = new SimpleTask() { @Override protected void onPreExecute(Bundle args) { tvError.setVisibility(View.GONE); pbWait.setVisibility(View.VISIBLE); } @Override protected void onPostExecute(Bundle args) { pbWait.setVisibility(View.GONE); } @Override protected Bundle onExecute(Context context, Bundle args) throws Throwable { String apiKey = args.getString("apiKey"); File file = (File) args.getSerializable("file"); return VirusTotal.lookup(context, file, apiKey); } @Override protected void onExecuted(Bundle args, Bundle result) { List scans = result.getParcelableArrayList("scans"); String label = result.getString("label"); String analysis = args.getString("analysis"); int malicious = 0; if (scans != null) for (VirusTotal.ScanResult scan : scans) if ("malicious".equals(scan.category)) malicious++; NumberFormat NF = NumberFormat.getNumberInstance(); tvUnknown.setVisibility(scans == null ? View.VISIBLE : View.GONE); tvSummary.setText(getString(R.string.title_vt_summary, NF.format(malicious))); tvSummary.setTextColor(Helper.resolveColor(context, malicious == 0 ? android.R.attr.textColorPrimary : R.attr.colorWarning)); tvSummary.setTypeface(malicious == 0 ? Typeface.DEFAULT : Typeface.DEFAULT_BOLD); tvSummary.setVisibility(scans == null ? View.GONE : View.VISIBLE); tvLabel.setText(label); tvReport.setVisibility(scans == null ? View.GONE : View.VISIBLE); adapter.set(scans == null ? new ArrayList<>() : scans); rvScan.setVisibility(scans == null ? View.GONE : View.VISIBLE); btnUpload.setVisibility(scans == null && !TextUtils.isEmpty(apiKey) ? View.VISIBLE : View.GONE); tvPrivacy.setVisibility(btnUpload.getVisibility()); if (analysis != null && args.getBoolean("init")) { args.remove("init"); btnUpload.callOnClick(); } } @Override protected void onException(Bundle args, Throwable ex) { tvError.setText(Log.formatThrowable(ex, false)); tvError.setVisibility(View.VISIBLE); } }; final SimpleTask taskUpload = new SimpleTask() { @Override protected void onPreExecute(Bundle args) { btnUpload.setEnabled(false); pbUpload.setVisibility(View.VISIBLE); } @Override protected void onPostExecute(Bundle args) { btnUpload.setEnabled(true); tvAnalyzing.setVisibility(View.GONE); pbUpload.setVisibility(View.GONE); } @Override protected Void onExecute(Context context, Bundle args) throws Throwable { String apiKey = args.getString("apiKey"); File file = (File) args.getSerializable("file"); String analysis = args.getString("analysis"); if (analysis == null) { analysis = VirusTotal.upload(context, file, apiKey); args.putString("analysis", analysis); } postProgress(analysis); VirusTotal.waitForAnalysis(context, analysis, apiKey); return null; } @Override protected void onProgress(CharSequence status, Bundle data) { tvAnalyzing.setVisibility(View.VISIBLE); } @Override protected void onExecuted(Bundle args, Void data) { taskLookup.execute(FragmentDialogVirusTotal.this, args, "attachment:lookup"); } @Override protected void onException(Bundle args, Throwable ex) { Log.unexpectedError(getParentFragmentManager(), ex); } }; final SimpleTask taskUrl = new SimpleTask() { @Override protected String onExecute(Context context, Bundle args) throws Throwable { File file = (File) args.getSerializable("file"); return VirusTotal.getUrl(file); } @Override protected void onExecuted(Bundle args, String uri) { Helper.view(context, Uri.parse(uri), true); } @Override protected void onException(Bundle args, Throwable ex) { Log.unexpectedError(getParentFragmentManager(), ex); } }; tvReport.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View view) { taskUrl.execute(FragmentDialogVirusTotal.this, args, "attachment:report"); } }); btnUpload.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View view) { taskUpload.execute(FragmentDialogVirusTotal.this, args, "attachment:upload"); } }); tvPrivacy.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { Helper.view(v.getContext(), Uri.parse(VirusTotal.URI_PRIVACY), true); } }); if (TextUtils.isEmpty(apiKey)) pbWait.setVisibility(View.GONE); else { args.putBoolean("init", true); taskLookup.execute(this, args, "attachment:lookup"); } return new AlertDialog.Builder(context) .setView(view) .setNegativeButton(R.string.title_setup_done, null) .create(); } } }