mirror of
https://github.com/oxen-io/session-android.git
synced 2024-11-28 20:45:17 +00:00
803 lines
33 KiB
Java
803 lines
33 KiB
Java
/*
|
|
* Copyright (C) 2011 Whisper Systems
|
|
*
|
|
* This program 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.
|
|
*
|
|
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package org.thoughtcrime.securesms.database;
|
|
|
|
import android.annotation.SuppressLint;
|
|
import android.content.ContentValues;
|
|
import android.content.Context;
|
|
import android.database.Cursor;
|
|
import android.graphics.Bitmap;
|
|
import android.media.MediaMetadataRetriever;
|
|
import android.net.Uri;
|
|
import android.os.Build;
|
|
import android.support.annotation.NonNull;
|
|
import android.support.annotation.Nullable;
|
|
import android.support.annotation.VisibleForTesting;
|
|
import android.text.TextUtils;
|
|
import org.thoughtcrime.securesms.logging.Log;
|
|
import android.util.Pair;
|
|
|
|
import com.bumptech.glide.Glide;
|
|
|
|
import net.sqlcipher.database.SQLiteDatabase;
|
|
|
|
import org.json.JSONArray;
|
|
import org.json.JSONException;
|
|
import org.thoughtcrime.securesms.attachments.Attachment;
|
|
import org.thoughtcrime.securesms.attachments.AttachmentId;
|
|
import org.thoughtcrime.securesms.attachments.DatabaseAttachment;
|
|
import org.thoughtcrime.securesms.crypto.AttachmentSecret;
|
|
import org.thoughtcrime.securesms.crypto.ClassicDecryptingPartInputStream;
|
|
import org.thoughtcrime.securesms.crypto.ModernDecryptingPartInputStream;
|
|
import org.thoughtcrime.securesms.crypto.ModernEncryptingPartOutputStream;
|
|
import org.thoughtcrime.securesms.database.helpers.SQLCipherOpenHelper;
|
|
import org.thoughtcrime.securesms.mms.MediaStream;
|
|
import org.thoughtcrime.securesms.mms.MmsException;
|
|
import org.thoughtcrime.securesms.mms.PartAuthority;
|
|
import org.thoughtcrime.securesms.util.BitmapDecodingException;
|
|
import org.thoughtcrime.securesms.util.BitmapUtil;
|
|
import org.thoughtcrime.securesms.util.JsonUtils;
|
|
import org.thoughtcrime.securesms.util.MediaUtil;
|
|
import org.thoughtcrime.securesms.util.MediaUtil.ThumbnailData;
|
|
import org.thoughtcrime.securesms.util.StorageUtil;
|
|
import org.thoughtcrime.securesms.util.Util;
|
|
import org.thoughtcrime.securesms.video.EncryptedMediaDataSource;
|
|
|
|
import java.io.File;
|
|
import java.io.FileNotFoundException;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.util.Collections;
|
|
import java.util.HashMap;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.concurrent.Callable;
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.concurrent.ExecutorService;
|
|
|
|
public class AttachmentDatabase extends Database {
|
|
|
|
private static final String TAG = AttachmentDatabase.class.getSimpleName();
|
|
|
|
public static final String TABLE_NAME = "part";
|
|
public static final String ROW_ID = "_id";
|
|
static final String ATTACHMENT_JSON_ALIAS = "attachment_json";
|
|
public static final String MMS_ID = "mid";
|
|
static final String CONTENT_TYPE = "ct";
|
|
static final String NAME = "name";
|
|
static final String CONTENT_DISPOSITION = "cd";
|
|
static final String CONTENT_LOCATION = "cl";
|
|
public static final String DATA = "_data";
|
|
static final String TRANSFER_STATE = "pending_push";
|
|
public static final String SIZE = "data_size";
|
|
static final String FILE_NAME = "file_name";
|
|
public static final String THUMBNAIL = "thumbnail";
|
|
static final String THUMBNAIL_ASPECT_RATIO = "aspect_ratio";
|
|
public static final String UNIQUE_ID = "unique_id";
|
|
static final String DIGEST = "digest";
|
|
static final String VOICE_NOTE = "voice_note";
|
|
static final String QUOTE = "quote";
|
|
static final String FAST_PREFLIGHT_ID = "fast_preflight_id";
|
|
public static final String DATA_RANDOM = "data_random";
|
|
private static final String THUMBNAIL_RANDOM = "thumbnail_random";
|
|
static final String WIDTH = "width";
|
|
static final String HEIGHT = "height";
|
|
|
|
public static final String DIRECTORY = "parts";
|
|
|
|
public static final int TRANSFER_PROGRESS_DONE = 0;
|
|
public static final int TRANSFER_PROGRESS_STARTED = 1;
|
|
public static final int TRANSFER_PROGRESS_PENDING = 2;
|
|
public static final int TRANSFER_PROGRESS_FAILED = 3;
|
|
|
|
private static final String PART_ID_WHERE = ROW_ID + " = ? AND " + UNIQUE_ID + " = ?";
|
|
|
|
private static final String[] PROJECTION = new String[] {ROW_ID,
|
|
MMS_ID, CONTENT_TYPE, NAME, CONTENT_DISPOSITION,
|
|
CONTENT_LOCATION, DATA, THUMBNAIL, TRANSFER_STATE,
|
|
SIZE, FILE_NAME, THUMBNAIL, THUMBNAIL_ASPECT_RATIO,
|
|
UNIQUE_ID, DIGEST, FAST_PREFLIGHT_ID, VOICE_NOTE,
|
|
QUOTE, DATA_RANDOM, THUMBNAIL_RANDOM, WIDTH, HEIGHT};
|
|
|
|
public static final String CREATE_TABLE = "CREATE TABLE " + TABLE_NAME + " (" + ROW_ID + " INTEGER PRIMARY KEY, " +
|
|
MMS_ID + " INTEGER, " + "seq" + " INTEGER DEFAULT 0, " +
|
|
CONTENT_TYPE + " TEXT, " + NAME + " TEXT, " + "chset" + " INTEGER, " +
|
|
CONTENT_DISPOSITION + " TEXT, " + "fn" + " TEXT, " + "cid" + " TEXT, " +
|
|
CONTENT_LOCATION + " TEXT, " + "ctt_s" + " INTEGER, " +
|
|
"ctt_t" + " TEXT, " + "encrypted" + " INTEGER, " +
|
|
TRANSFER_STATE + " INTEGER, "+ DATA + " TEXT, " + SIZE + " INTEGER, " +
|
|
FILE_NAME + " TEXT, " + THUMBNAIL + " TEXT, " + THUMBNAIL_ASPECT_RATIO + " REAL, " +
|
|
UNIQUE_ID + " INTEGER NOT NULL, " + DIGEST + " BLOB, " + FAST_PREFLIGHT_ID + " TEXT, " +
|
|
VOICE_NOTE + " INTEGER DEFAULT 0, " + DATA_RANDOM + " BLOB, " + THUMBNAIL_RANDOM + " BLOB, " +
|
|
QUOTE + " INTEGER DEFAULT 0, " + WIDTH + " INTEGER DEFAULT 0, " + HEIGHT + " INTEGER DEFAULT 0);";
|
|
|
|
public static final String[] CREATE_INDEXS = {
|
|
"CREATE INDEX IF NOT EXISTS part_mms_id_index ON " + TABLE_NAME + " (" + MMS_ID + ");",
|
|
"CREATE INDEX IF NOT EXISTS pending_push_index ON " + TABLE_NAME + " (" + TRANSFER_STATE + ");",
|
|
};
|
|
|
|
private final ExecutorService thumbnailExecutor = Util.newSingleThreadedLifoExecutor();
|
|
|
|
private final AttachmentSecret attachmentSecret;
|
|
|
|
public AttachmentDatabase(Context context, SQLCipherOpenHelper databaseHelper, AttachmentSecret attachmentSecret) {
|
|
super(context, databaseHelper);
|
|
this.attachmentSecret = attachmentSecret;
|
|
}
|
|
|
|
public @NonNull InputStream getAttachmentStream(AttachmentId attachmentId, long offset)
|
|
throws IOException
|
|
{
|
|
InputStream dataStream = getDataStream(attachmentId, DATA, offset);
|
|
|
|
if (dataStream == null) throw new IOException("No stream for: " + attachmentId);
|
|
else return dataStream;
|
|
}
|
|
|
|
public @NonNull InputStream getThumbnailStream(@NonNull AttachmentId attachmentId)
|
|
throws IOException
|
|
{
|
|
Log.d(TAG, "getThumbnailStream(" + attachmentId + ")");
|
|
InputStream dataStream = getDataStream(attachmentId, THUMBNAIL, 0);
|
|
|
|
if (dataStream != null) {
|
|
return dataStream;
|
|
}
|
|
|
|
try {
|
|
InputStream generatedStream = thumbnailExecutor.submit(new ThumbnailFetchCallable(attachmentId)).get();
|
|
|
|
if (generatedStream == null) throw new FileNotFoundException("No thumbnail stream available: " + attachmentId);
|
|
else return generatedStream;
|
|
} catch (InterruptedException ie) {
|
|
throw new AssertionError("interrupted");
|
|
} catch (ExecutionException ee) {
|
|
Log.w(TAG, ee);
|
|
throw new IOException(ee);
|
|
}
|
|
}
|
|
|
|
public void setTransferProgressFailed(AttachmentId attachmentId, long mmsId)
|
|
throws MmsException
|
|
{
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
ContentValues values = new ContentValues();
|
|
values.put(TRANSFER_STATE, TRANSFER_PROGRESS_FAILED);
|
|
|
|
database.update(TABLE_NAME, values, PART_ID_WHERE, attachmentId.toStrings());
|
|
notifyConversationListeners(DatabaseFactory.getMmsDatabase(context).getThreadIdForMessage(mmsId));
|
|
}
|
|
|
|
public @Nullable DatabaseAttachment getAttachment(@NonNull AttachmentId attachmentId)
|
|
{
|
|
SQLiteDatabase database = databaseHelper.getReadableDatabase();
|
|
Cursor cursor = null;
|
|
|
|
try {
|
|
cursor = database.query(TABLE_NAME, PROJECTION, PART_ID_WHERE, attachmentId.toStrings(), null, null, null);
|
|
|
|
if (cursor != null && cursor.moveToFirst()) {
|
|
List<DatabaseAttachment> list = getAttachment(cursor);
|
|
|
|
if (list != null && list.size() > 0) {
|
|
return list.get(0);
|
|
}
|
|
}
|
|
|
|
return null;
|
|
} finally {
|
|
if (cursor != null)
|
|
cursor.close();
|
|
}
|
|
}
|
|
|
|
public @NonNull List<DatabaseAttachment> getAttachmentsForMessage(long mmsId) {
|
|
SQLiteDatabase database = databaseHelper.getReadableDatabase();
|
|
List<DatabaseAttachment> results = new LinkedList<>();
|
|
Cursor cursor = null;
|
|
|
|
try {
|
|
cursor = database.query(TABLE_NAME, PROJECTION, MMS_ID + " = ?", new String[] {mmsId+""},
|
|
null, null, null);
|
|
|
|
while (cursor != null && cursor.moveToNext()) {
|
|
results.addAll(getAttachment(cursor));
|
|
}
|
|
|
|
return results;
|
|
} finally {
|
|
if (cursor != null)
|
|
cursor.close();
|
|
}
|
|
}
|
|
|
|
public @NonNull List<DatabaseAttachment> getPendingAttachments() {
|
|
final SQLiteDatabase database = databaseHelper.getReadableDatabase();
|
|
final List<DatabaseAttachment> attachments = new LinkedList<>();
|
|
|
|
Cursor cursor = null;
|
|
try {
|
|
cursor = database.query(TABLE_NAME, PROJECTION, TRANSFER_STATE + " = ?", new String[] {String.valueOf(TRANSFER_PROGRESS_STARTED)}, null, null, null);
|
|
while (cursor != null && cursor.moveToNext()) {
|
|
attachments.addAll(getAttachment(cursor));
|
|
}
|
|
} finally {
|
|
if (cursor != null) cursor.close();
|
|
}
|
|
|
|
return attachments;
|
|
}
|
|
|
|
@SuppressWarnings("ResultOfMethodCallIgnored")
|
|
void deleteAttachmentsForMessage(long mmsId) {
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
Cursor cursor = null;
|
|
|
|
try {
|
|
cursor = database.query(TABLE_NAME, new String[] {DATA, THUMBNAIL, CONTENT_TYPE}, MMS_ID + " = ?",
|
|
new String[] {mmsId+""}, null, null, null);
|
|
|
|
while (cursor != null && cursor.moveToNext()) {
|
|
deleteAttachmentOnDisk(cursor.getString(0), cursor.getString(1), cursor.getString(2));
|
|
}
|
|
} finally {
|
|
if (cursor != null)
|
|
cursor.close();
|
|
}
|
|
|
|
database.delete(TABLE_NAME, MMS_ID + " = ?", new String[] {mmsId + ""});
|
|
notifyAttachmentListeners();
|
|
}
|
|
|
|
public void deleteAttachment(@NonNull AttachmentId id) {
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
|
|
try (Cursor cursor = database.query(TABLE_NAME,
|
|
new String[]{DATA, THUMBNAIL, CONTENT_TYPE},
|
|
PART_ID_WHERE,
|
|
id.toStrings(),
|
|
null,
|
|
null,
|
|
null))
|
|
{
|
|
if (cursor == null || !cursor.moveToNext()) {
|
|
Log.w(TAG, "Tried to delete an attachment, but it didn't exist.");
|
|
return;
|
|
}
|
|
String data = cursor.getString(0);
|
|
String thumbnail = cursor.getString(1);
|
|
String contentType = cursor.getString(2);
|
|
|
|
database.delete(TABLE_NAME, PART_ID_WHERE, id.toStrings());
|
|
deleteAttachmentOnDisk(data, thumbnail, contentType);
|
|
notifyAttachmentListeners();
|
|
}
|
|
}
|
|
|
|
@SuppressWarnings("ResultOfMethodCallIgnored")
|
|
void deleteAllAttachments() {
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
database.delete(TABLE_NAME, null, null);
|
|
|
|
File attachmentsDirectory = context.getDir(DIRECTORY, Context.MODE_PRIVATE);
|
|
File[] attachments = attachmentsDirectory.listFiles();
|
|
|
|
for (File attachment : attachments) {
|
|
attachment.delete();
|
|
}
|
|
|
|
notifyAttachmentListeners();
|
|
}
|
|
|
|
@SuppressWarnings("ResultOfMethodCallIgnored")
|
|
private void deleteAttachmentOnDisk(@Nullable String data, @Nullable String thumbnail, @Nullable String contentType) {
|
|
if (!TextUtils.isEmpty(data)) {
|
|
new File(data).delete();
|
|
}
|
|
|
|
if (!TextUtils.isEmpty(thumbnail)) {
|
|
new File(thumbnail).delete();
|
|
}
|
|
|
|
if (MediaUtil.isImageType(contentType) || thumbnail != null) {
|
|
Glide.get(context).clearDiskCache();
|
|
}
|
|
}
|
|
|
|
public void insertAttachmentsForPlaceholder(long mmsId, @NonNull AttachmentId attachmentId, @NonNull InputStream inputStream)
|
|
throws MmsException
|
|
{
|
|
DatabaseAttachment placeholder = getAttachment(attachmentId);
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
ContentValues values = new ContentValues();
|
|
DataInfo dataInfo = setAttachmentData(inputStream);
|
|
|
|
if (placeholder != null && placeholder.isQuote() && !placeholder.getContentType().startsWith("image")) {
|
|
values.put(THUMBNAIL, dataInfo.file.getAbsolutePath());
|
|
values.put(THUMBNAIL_RANDOM, dataInfo.random);
|
|
} else {
|
|
values.put(DATA, dataInfo.file.getAbsolutePath());
|
|
values.put(SIZE, dataInfo.length);
|
|
values.put(DATA_RANDOM, dataInfo.random);
|
|
}
|
|
|
|
values.put(TRANSFER_STATE, TRANSFER_PROGRESS_DONE);
|
|
values.put(CONTENT_LOCATION, (String)null);
|
|
values.put(CONTENT_DISPOSITION, (String)null);
|
|
values.put(DIGEST, (byte[])null);
|
|
values.put(NAME, (String) null);
|
|
values.put(FAST_PREFLIGHT_ID, (String)null);
|
|
|
|
if (database.update(TABLE_NAME, values, PART_ID_WHERE, attachmentId.toStrings()) == 0) {
|
|
//noinspection ResultOfMethodCallIgnored
|
|
dataInfo.file.delete();
|
|
} else {
|
|
notifyConversationListeners(DatabaseFactory.getMmsDatabase(context).getThreadIdForMessage(mmsId));
|
|
notifyConversationListListeners();
|
|
}
|
|
|
|
thumbnailExecutor.submit(new ThumbnailFetchCallable(attachmentId));
|
|
}
|
|
|
|
@NonNull Map<Attachment, AttachmentId> insertAttachmentsForMessage(long mmsId, @NonNull List<Attachment> attachments, @NonNull List<Attachment> quoteAttachment)
|
|
throws MmsException
|
|
{
|
|
Log.d(TAG, "insertParts(" + attachments.size() + ")");
|
|
|
|
Map<Attachment, AttachmentId> insertedAttachments = new HashMap<>();
|
|
|
|
for (Attachment attachment : attachments) {
|
|
AttachmentId attachmentId = insertAttachment(mmsId, attachment, attachment.isQuote());
|
|
insertedAttachments.put(attachment, attachmentId);
|
|
Log.i(TAG, "Inserted attachment at ID: " + attachmentId);
|
|
}
|
|
|
|
for (Attachment attachment : quoteAttachment) {
|
|
AttachmentId attachmentId = insertAttachment(mmsId, attachment, true);
|
|
insertedAttachments.put(attachment, attachmentId);
|
|
Log.i(TAG, "Inserted quoted attachment at ID: " + attachmentId);
|
|
}
|
|
|
|
return insertedAttachments;
|
|
}
|
|
|
|
public @NonNull Attachment updateAttachmentData(@NonNull Attachment attachment,
|
|
@NonNull MediaStream mediaStream)
|
|
throws MmsException
|
|
{
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
DatabaseAttachment databaseAttachment = (DatabaseAttachment) attachment;
|
|
DataInfo dataInfo = getAttachmentDataFileInfo(databaseAttachment.getAttachmentId(), DATA);
|
|
|
|
if (dataInfo == null) {
|
|
throw new MmsException("No attachment data found!");
|
|
}
|
|
|
|
dataInfo = setAttachmentData(dataInfo.file, mediaStream.getStream());
|
|
|
|
ContentValues contentValues = new ContentValues();
|
|
contentValues.put(SIZE, dataInfo.length);
|
|
contentValues.put(CONTENT_TYPE, mediaStream.getMimeType());
|
|
contentValues.put(WIDTH, mediaStream.getWidth());
|
|
contentValues.put(HEIGHT, mediaStream.getHeight());
|
|
contentValues.put(DATA_RANDOM, dataInfo.random);
|
|
|
|
database.update(TABLE_NAME, contentValues, PART_ID_WHERE, databaseAttachment.getAttachmentId().toStrings());
|
|
|
|
return new DatabaseAttachment(databaseAttachment.getAttachmentId(),
|
|
databaseAttachment.getMmsId(),
|
|
databaseAttachment.hasData(),
|
|
databaseAttachment.hasThumbnail(),
|
|
mediaStream.getMimeType(),
|
|
databaseAttachment.getTransferState(),
|
|
dataInfo.length,
|
|
databaseAttachment.getFileName(),
|
|
databaseAttachment.getLocation(),
|
|
databaseAttachment.getKey(),
|
|
databaseAttachment.getRelay(),
|
|
databaseAttachment.getDigest(),
|
|
databaseAttachment.getFastPreflightId(),
|
|
databaseAttachment.isVoiceNote(),
|
|
mediaStream.getWidth(),
|
|
mediaStream.getHeight(),
|
|
databaseAttachment.isQuote());
|
|
}
|
|
|
|
|
|
public void updateAttachmentFileName(@NonNull AttachmentId attachmentId,
|
|
@Nullable String fileName)
|
|
{
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
|
|
ContentValues contentValues = new ContentValues(1);
|
|
contentValues.put(FILE_NAME, StorageUtil.getCleanFileName(fileName));
|
|
|
|
database.update(TABLE_NAME, contentValues, PART_ID_WHERE, attachmentId.toStrings());
|
|
}
|
|
|
|
public void markAttachmentUploaded(long messageId, Attachment attachment) {
|
|
ContentValues values = new ContentValues(1);
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
|
|
values.put(TRANSFER_STATE, TRANSFER_PROGRESS_DONE);
|
|
database.update(TABLE_NAME, values, PART_ID_WHERE, ((DatabaseAttachment)attachment).getAttachmentId().toStrings());
|
|
|
|
notifyConversationListeners(DatabaseFactory.getMmsDatabase(context).getThreadIdForMessage(messageId));
|
|
}
|
|
|
|
public void setTransferState(long messageId, @NonNull Attachment attachment, int transferState) {
|
|
if (!(attachment instanceof DatabaseAttachment)) {
|
|
throw new AssertionError("Attempt to update attachment that doesn't belong to DB!");
|
|
}
|
|
|
|
setTransferState(messageId, ((DatabaseAttachment) attachment).getAttachmentId(), transferState);
|
|
}
|
|
|
|
public void setTransferState(long messageId, @NonNull AttachmentId attachmentId, int transferState) {
|
|
final ContentValues values = new ContentValues(1);
|
|
final SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
|
|
values.put(TRANSFER_STATE, transferState);
|
|
database.update(TABLE_NAME, values, PART_ID_WHERE, attachmentId.toStrings());
|
|
notifyConversationListeners(DatabaseFactory.getMmsDatabase(context).getThreadIdForMessage(messageId));
|
|
}
|
|
|
|
@SuppressWarnings("WeakerAccess")
|
|
@VisibleForTesting
|
|
protected @Nullable InputStream getDataStream(AttachmentId attachmentId, String dataType, long offset)
|
|
{
|
|
DataInfo dataInfo = getAttachmentDataFileInfo(attachmentId, dataType);
|
|
|
|
if (dataInfo == null) {
|
|
return null;
|
|
}
|
|
|
|
try {
|
|
if (dataInfo.random != null && dataInfo.random.length == 32) {
|
|
return ModernDecryptingPartInputStream.createFor(attachmentSecret, dataInfo.random, dataInfo.file, offset);
|
|
} else {
|
|
InputStream stream = ClassicDecryptingPartInputStream.createFor(attachmentSecret, dataInfo.file);
|
|
long skipped = stream.skip(offset);
|
|
|
|
if (skipped != offset) {
|
|
Log.w(TAG, "Skip failed: " + skipped + " vs " + offset);
|
|
return null;
|
|
}
|
|
|
|
return stream;
|
|
}
|
|
} catch (IOException e) {
|
|
Log.w(TAG, e);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private @Nullable DataInfo getAttachmentDataFileInfo(@NonNull AttachmentId attachmentId, @NonNull String dataType)
|
|
{
|
|
SQLiteDatabase database = databaseHelper.getReadableDatabase();
|
|
Cursor cursor = null;
|
|
|
|
String randomColumn;
|
|
|
|
switch (dataType) {
|
|
case DATA: randomColumn = DATA_RANDOM; break;
|
|
case THUMBNAIL: randomColumn = THUMBNAIL_RANDOM; break;
|
|
default:throw new AssertionError("Unknown data type: " + dataType);
|
|
}
|
|
|
|
try {
|
|
cursor = database.query(TABLE_NAME, new String[]{dataType, SIZE, randomColumn}, PART_ID_WHERE, attachmentId.toStrings(),
|
|
null, null, null);
|
|
|
|
if (cursor != null && cursor.moveToFirst()) {
|
|
if (cursor.isNull(0)) {
|
|
return null;
|
|
}
|
|
|
|
return new DataInfo(new File(cursor.getString(0)),
|
|
cursor.getLong(1),
|
|
cursor.getBlob(2));
|
|
} else {
|
|
return null;
|
|
}
|
|
} finally {
|
|
if (cursor != null)
|
|
cursor.close();
|
|
}
|
|
|
|
}
|
|
|
|
private @NonNull DataInfo setAttachmentData(@NonNull Uri uri)
|
|
throws MmsException
|
|
{
|
|
try {
|
|
InputStream inputStream = PartAuthority.getAttachmentStream(context, uri);
|
|
return setAttachmentData(inputStream);
|
|
} catch (IOException e) {
|
|
throw new MmsException(e);
|
|
}
|
|
}
|
|
|
|
private @NonNull DataInfo setAttachmentData(@NonNull InputStream in)
|
|
throws MmsException
|
|
{
|
|
try {
|
|
File partsDirectory = context.getDir(DIRECTORY, Context.MODE_PRIVATE);
|
|
File dataFile = File.createTempFile("part", ".mms", partsDirectory);
|
|
return setAttachmentData(dataFile, in);
|
|
} catch (IOException e) {
|
|
throw new MmsException(e);
|
|
}
|
|
}
|
|
|
|
private @NonNull DataInfo setAttachmentData(@NonNull File destination, @NonNull InputStream in)
|
|
throws MmsException
|
|
{
|
|
try {
|
|
Pair<byte[], OutputStream> out = ModernEncryptingPartOutputStream.createFor(attachmentSecret, destination, false);
|
|
long length = Util.copy(in, out.second);
|
|
|
|
return new DataInfo(destination, length, out.first);
|
|
} catch (IOException e) {
|
|
throw new MmsException(e);
|
|
}
|
|
}
|
|
|
|
public List<DatabaseAttachment> getAttachment(@NonNull Cursor cursor) {
|
|
try {
|
|
if (cursor.getColumnIndex(AttachmentDatabase.ATTACHMENT_JSON_ALIAS) != -1) {
|
|
if (cursor.isNull(cursor.getColumnIndexOrThrow(ATTACHMENT_JSON_ALIAS))) {
|
|
return new LinkedList<>();
|
|
}
|
|
|
|
List<DatabaseAttachment> result = new LinkedList<>();
|
|
JSONArray array = new JSONArray(cursor.getString(cursor.getColumnIndexOrThrow(ATTACHMENT_JSON_ALIAS)));
|
|
|
|
for (int i=0;i<array.length();i++) {
|
|
JsonUtils.SaneJSONObject object = new JsonUtils.SaneJSONObject(array.getJSONObject(i));
|
|
|
|
if (!object.isNull(ROW_ID)) {
|
|
result.add(new DatabaseAttachment(new AttachmentId(object.getLong(ROW_ID), object.getLong(UNIQUE_ID)),
|
|
object.getLong(MMS_ID),
|
|
!TextUtils.isEmpty(object.getString(DATA)),
|
|
!TextUtils.isEmpty(object.getString(THUMBNAIL)),
|
|
object.getString(CONTENT_TYPE),
|
|
object.getInt(TRANSFER_STATE),
|
|
object.getLong(SIZE),
|
|
object.getString(FILE_NAME),
|
|
object.getString(CONTENT_LOCATION),
|
|
object.getString(CONTENT_DISPOSITION),
|
|
object.getString(NAME),
|
|
null,
|
|
object.getString(FAST_PREFLIGHT_ID),
|
|
object.getInt(VOICE_NOTE) == 1,
|
|
object.getInt(WIDTH),
|
|
object.getInt(HEIGHT),
|
|
object.getInt(QUOTE) == 1));
|
|
}
|
|
}
|
|
|
|
return result;
|
|
} else {
|
|
return Collections.singletonList(new DatabaseAttachment(new AttachmentId(cursor.getLong(cursor.getColumnIndexOrThrow(ROW_ID)),
|
|
cursor.getLong(cursor.getColumnIndexOrThrow(UNIQUE_ID))),
|
|
cursor.getLong(cursor.getColumnIndexOrThrow(MMS_ID)),
|
|
!cursor.isNull(cursor.getColumnIndexOrThrow(DATA)),
|
|
!cursor.isNull(cursor.getColumnIndexOrThrow(THUMBNAIL)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(CONTENT_TYPE)),
|
|
cursor.getInt(cursor.getColumnIndexOrThrow(TRANSFER_STATE)),
|
|
cursor.getLong(cursor.getColumnIndexOrThrow(SIZE)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(FILE_NAME)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(CONTENT_LOCATION)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(CONTENT_DISPOSITION)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(NAME)),
|
|
cursor.getBlob(cursor.getColumnIndexOrThrow(DIGEST)),
|
|
cursor.getString(cursor.getColumnIndexOrThrow(FAST_PREFLIGHT_ID)),
|
|
cursor.getInt(cursor.getColumnIndexOrThrow(VOICE_NOTE)) == 1,
|
|
cursor.getInt(cursor.getColumnIndexOrThrow(WIDTH)),
|
|
cursor.getInt(cursor.getColumnIndexOrThrow(HEIGHT)),
|
|
cursor.getInt(cursor.getColumnIndexOrThrow(QUOTE)) == 1));
|
|
}
|
|
} catch (JSONException e) {
|
|
throw new AssertionError(e);
|
|
}
|
|
}
|
|
|
|
|
|
private AttachmentId insertAttachment(long mmsId, Attachment attachment, boolean quote)
|
|
throws MmsException
|
|
{
|
|
Log.d(TAG, "Inserting attachment for mms id: " + mmsId);
|
|
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
DataInfo dataInfo = null;
|
|
long uniqueId = System.currentTimeMillis();
|
|
|
|
if (attachment.getDataUri() != null) {
|
|
dataInfo = setAttachmentData(attachment.getDataUri());
|
|
Log.d(TAG, "Wrote part to file: " + dataInfo.file.getAbsolutePath());
|
|
}
|
|
|
|
ContentValues contentValues = new ContentValues();
|
|
contentValues.put(MMS_ID, mmsId);
|
|
contentValues.put(CONTENT_TYPE, attachment.getContentType());
|
|
contentValues.put(TRANSFER_STATE, attachment.getTransferState());
|
|
contentValues.put(UNIQUE_ID, uniqueId);
|
|
contentValues.put(CONTENT_LOCATION, attachment.getLocation());
|
|
contentValues.put(DIGEST, attachment.getDigest());
|
|
contentValues.put(CONTENT_DISPOSITION, attachment.getKey());
|
|
contentValues.put(NAME, attachment.getRelay());
|
|
contentValues.put(FILE_NAME, StorageUtil.getCleanFileName(attachment.getFileName()));
|
|
contentValues.put(SIZE, attachment.getSize());
|
|
contentValues.put(FAST_PREFLIGHT_ID, attachment.getFastPreflightId());
|
|
contentValues.put(VOICE_NOTE, attachment.isVoiceNote() ? 1 : 0);
|
|
contentValues.put(WIDTH, attachment.getWidth());
|
|
contentValues.put(HEIGHT, attachment.getHeight());
|
|
contentValues.put(QUOTE, quote);
|
|
|
|
if (dataInfo != null) {
|
|
contentValues.put(DATA, dataInfo.file.getAbsolutePath());
|
|
contentValues.put(SIZE, dataInfo.length);
|
|
contentValues.put(DATA_RANDOM, dataInfo.random);
|
|
}
|
|
|
|
long rowId = database.insert(TABLE_NAME, null, contentValues);
|
|
AttachmentId attachmentId = new AttachmentId(rowId, uniqueId);
|
|
Uri thumbnailUri = attachment.getThumbnailUri();
|
|
boolean hasThumbnail = false;
|
|
|
|
if (thumbnailUri != null) {
|
|
try (InputStream attachmentStream = PartAuthority.getAttachmentStream(context, thumbnailUri)) {
|
|
Pair<Integer, Integer> dimens = BitmapUtil.getDimensions(attachmentStream);
|
|
updateAttachmentThumbnail(attachmentId,
|
|
PartAuthority.getAttachmentStream(context, thumbnailUri),
|
|
(float) dimens.first / (float) dimens.second);
|
|
hasThumbnail = true;
|
|
} catch (IOException | BitmapDecodingException e) {
|
|
Log.w(TAG, "Failed to save existing thumbnail.", e);
|
|
}
|
|
}
|
|
|
|
if (!hasThumbnail && dataInfo != null) {
|
|
if (MediaUtil.hasVideoThumbnail(attachment.getDataUri())) {
|
|
Bitmap bitmap = MediaUtil.getVideoThumbnail(context, attachment.getDataUri());
|
|
|
|
if (bitmap != null) {
|
|
ThumbnailData thumbnailData = new ThumbnailData(bitmap);
|
|
updateAttachmentThumbnail(attachmentId, thumbnailData.toDataStream(), thumbnailData.getAspectRatio());
|
|
} else {
|
|
Log.w(TAG, "Retrieving video thumbnail failed, submitting thumbnail generation job...");
|
|
thumbnailExecutor.submit(new ThumbnailFetchCallable(attachmentId));
|
|
}
|
|
} else {
|
|
Log.i(TAG, "Submitting thumbnail generation job...");
|
|
thumbnailExecutor.submit(new ThumbnailFetchCallable(attachmentId));
|
|
}
|
|
}
|
|
|
|
return attachmentId;
|
|
}
|
|
|
|
@SuppressWarnings("WeakerAccess")
|
|
@VisibleForTesting
|
|
protected void updateAttachmentThumbnail(AttachmentId attachmentId, InputStream in, float aspectRatio)
|
|
throws MmsException
|
|
{
|
|
Log.i(TAG, "updating part thumbnail for #" + attachmentId);
|
|
|
|
DataInfo thumbnailFile = setAttachmentData(in);
|
|
|
|
SQLiteDatabase database = databaseHelper.getWritableDatabase();
|
|
ContentValues values = new ContentValues(2);
|
|
|
|
values.put(THUMBNAIL, thumbnailFile.file.getAbsolutePath());
|
|
values.put(THUMBNAIL_ASPECT_RATIO, aspectRatio);
|
|
values.put(THUMBNAIL_RANDOM, thumbnailFile.random);
|
|
|
|
database.update(TABLE_NAME, values, PART_ID_WHERE, attachmentId.toStrings());
|
|
|
|
Cursor cursor = database.query(TABLE_NAME, new String[] {MMS_ID}, PART_ID_WHERE, attachmentId.toStrings(), null, null, null);
|
|
|
|
try {
|
|
if (cursor != null && cursor.moveToFirst()) {
|
|
notifyConversationListeners(DatabaseFactory.getMmsDatabase(context).getThreadIdForMessage(cursor.getLong(cursor.getColumnIndexOrThrow(MMS_ID))));
|
|
}
|
|
} finally {
|
|
if (cursor != null) cursor.close();
|
|
}
|
|
}
|
|
|
|
|
|
@VisibleForTesting
|
|
class ThumbnailFetchCallable implements Callable<InputStream> {
|
|
|
|
private final AttachmentId attachmentId;
|
|
|
|
ThumbnailFetchCallable(AttachmentId attachmentId) {
|
|
this.attachmentId = attachmentId;
|
|
}
|
|
|
|
@Override
|
|
public @Nullable InputStream call() throws Exception {
|
|
Log.d(TAG, "Executing thumbnail job...");
|
|
final InputStream stream = getDataStream(attachmentId, THUMBNAIL, 0);
|
|
|
|
if (stream != null) {
|
|
return stream;
|
|
}
|
|
|
|
DatabaseAttachment attachment = getAttachment(attachmentId);
|
|
|
|
if (attachment == null || !attachment.hasData()) {
|
|
return null;
|
|
}
|
|
|
|
ThumbnailData data = null;
|
|
|
|
if (MediaUtil.isVideoType(attachment.getContentType())) {
|
|
data = generateVideoThumbnail(attachmentId);
|
|
}
|
|
|
|
if (data == null) {
|
|
return null;
|
|
}
|
|
|
|
updateAttachmentThumbnail(attachmentId, data.toDataStream(), data.getAspectRatio());
|
|
|
|
return getDataStream(attachmentId, THUMBNAIL, 0);
|
|
}
|
|
|
|
@SuppressLint("NewApi")
|
|
private ThumbnailData generateVideoThumbnail(AttachmentId attachmentId) {
|
|
if (Build.VERSION.SDK_INT < Build.VERSION_CODES.M) {
|
|
Log.w(TAG, "Video thumbnails not supported...");
|
|
return null;
|
|
}
|
|
|
|
DataInfo dataInfo = getAttachmentDataFileInfo(attachmentId, DATA);
|
|
|
|
if (dataInfo == null) {
|
|
Log.w(TAG, "No data file found for video thumbnail...");
|
|
return null;
|
|
}
|
|
|
|
EncryptedMediaDataSource dataSource = new EncryptedMediaDataSource(attachmentSecret, dataInfo.file, dataInfo.random, dataInfo.length);
|
|
MediaMetadataRetriever retriever = new MediaMetadataRetriever();
|
|
retriever.setDataSource(dataSource);
|
|
|
|
Bitmap bitmap = retriever.getFrameAtTime(1000);
|
|
|
|
Log.i(TAG, "Generated video thumbnail...");
|
|
return new ThumbnailData(bitmap);
|
|
}
|
|
}
|
|
|
|
private static class DataInfo {
|
|
private final File file;
|
|
private final long length;
|
|
private final byte[] random;
|
|
|
|
private DataInfo(File file, long length, byte[] random) {
|
|
this.file = file;
|
|
this.length = length;
|
|
this.random = random;
|
|
}
|
|
}
|
|
}
|