mirror of
https://github.com/nextcloud/android.git
synced 2024-11-26 23:28:42 +03:00
Cleanup UriUploader a bit, and add tests for private paths
Signed-off-by: Álvaro Brey <alvaro.brey@nextcloud.com>
This commit is contained in:
parent
cd3bd0845a
commit
8f8274cab2
2 changed files with 76 additions and 25 deletions
|
@ -0,0 +1,52 @@
|
|||
package com.owncloud.android.ui.helpers
|
||||
|
||||
import android.net.Uri
|
||||
import androidx.test.core.app.launchActivity
|
||||
import com.nextcloud.client.TestActivity
|
||||
import com.owncloud.android.AbstractIT
|
||||
import com.owncloud.android.files.services.FileUploader
|
||||
import org.junit.Assert
|
||||
import org.junit.Test
|
||||
|
||||
class UriUploaderIT : AbstractIT() {
|
||||
|
||||
@Test
|
||||
fun testUploadPrivatePathSharedPreferences() {
|
||||
launchActivity<TestActivity>().use { scenario ->
|
||||
scenario.onActivity { activity ->
|
||||
val packageName = activity.packageName
|
||||
val path = "file:///data/data/$packageName/shared_prefs/com.nextcloud.client_preferences.xml"
|
||||
testPrivatePath(activity, path)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testUploadPrivatePathUserFile() {
|
||||
launchActivity<TestActivity>().use { scenario ->
|
||||
scenario.onActivity { activity ->
|
||||
val packageName = activity.packageName
|
||||
val path = "file:///storage/emulated/0/Android/media/$packageName/nextcloud/test/welcome.txt"
|
||||
testPrivatePath(activity, path)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun testPrivatePath(activity: TestActivity, path: String) {
|
||||
val sut = UriUploader(
|
||||
activity,
|
||||
listOf(Uri.parse(path)),
|
||||
"",
|
||||
activity.user.orElseThrow(::RuntimeException),
|
||||
FileUploader.LOCAL_BEHAVIOUR_MOVE,
|
||||
false,
|
||||
null
|
||||
)
|
||||
val uploadResult = sut.uploadUris()
|
||||
Assert.assertEquals(
|
||||
"Wrong result code",
|
||||
UriUploader.UriUploaderResultCode.ERROR_SENSITIVE_PATH,
|
||||
uploadResult
|
||||
)
|
||||
}
|
||||
}
|
|
@ -49,8 +49,7 @@ import com.owncloud.android.utils.UriUtils.getDisplayNameForUri
|
|||
@Suppress(
|
||||
"Detekt.LongParameterList",
|
||||
"Detekt.SpreadOperator",
|
||||
"Detekt.TooGenericExceptionCaught",
|
||||
"Detekt.NestedBlockDepth"
|
||||
"Detekt.TooGenericExceptionCaught"
|
||||
) // legacy code
|
||||
class UriUploader(
|
||||
private val mActivity: FileActivity,
|
||||
|
@ -76,29 +75,23 @@ class UriUploader(
|
|||
Log_OC.e(TAG, "Sensitive URI detected, aborting upload.")
|
||||
code = UriUploaderResultCode.ERROR_SENSITIVE_PATH
|
||||
} else {
|
||||
val contentUris: MutableList<Uri> = ArrayList()
|
||||
val contentRemotePaths: MutableList<String> = ArrayList()
|
||||
var schemeFileCounter = 0
|
||||
for (sourceStream in mUrisToUpload) {
|
||||
val sourceUri = sourceStream as Uri?
|
||||
if (sourceUri != null) {
|
||||
val displayName = getDisplayNameForUri(sourceUri, mActivity)
|
||||
require(displayName != null) { "Display name cannot be null" }
|
||||
val remotePath = mUploadPath + displayName
|
||||
if (ContentResolver.SCHEME_CONTENT == sourceUri.scheme) {
|
||||
contentUris.add(sourceUri)
|
||||
contentRemotePaths.add(remotePath)
|
||||
} else if (ContentResolver.SCHEME_FILE == sourceUri.scheme) {
|
||||
// file: uris should point to a local file, should be safe let FileUploader handle them
|
||||
requestUpload(sourceUri.path, remotePath)
|
||||
schemeFileCounter++
|
||||
}
|
||||
}
|
||||
val uris = mUrisToUpload.filterNotNull()
|
||||
.map { it as Uri }
|
||||
.map { Pair(it, getRemotePathForUri(it)) }
|
||||
|
||||
val fileUris = uris
|
||||
.filter { it.first.scheme == ContentResolver.SCHEME_FILE }
|
||||
fileUris.forEach {
|
||||
requestUpload(it.first.path, it.second)
|
||||
}
|
||||
if (contentUris.isNotEmpty()) {
|
||||
// content: uris will be copied to temporary files before calling {@link FileUploader}
|
||||
copyThenUpload(contentUris.toTypedArray(), *contentRemotePaths.toTypedArray())
|
||||
} else if (schemeFileCounter == 0) {
|
||||
|
||||
val contentUrisNew = uris
|
||||
.filter { it.first.scheme == ContentResolver.SCHEME_CONTENT }
|
||||
|
||||
if (contentUrisNew.isNotEmpty()) {
|
||||
val (contentUris, contentRemotePaths) = contentUrisNew.unzip()
|
||||
copyThenUpload(contentUris.toTypedArray(), contentRemotePaths.toTypedArray())
|
||||
} else if (fileUris.isEmpty()) {
|
||||
code = UriUploaderResultCode.ERROR_NO_FILE_TO_UPLOAD
|
||||
}
|
||||
}
|
||||
|
@ -112,6 +105,12 @@ class UriUploader(
|
|||
return code
|
||||
}
|
||||
|
||||
private fun getRemotePathForUri(sourceUri: Uri): String {
|
||||
val displayName = getDisplayNameForUri(sourceUri, mActivity)
|
||||
require(displayName != null) { "Display name cannot be null" }
|
||||
return mUploadPath + displayName
|
||||
}
|
||||
|
||||
private fun isSensitiveUri(uri: Uri): Boolean = uri.toString().contains(mActivity.packageName)
|
||||
|
||||
/**
|
||||
|
@ -146,7 +145,7 @@ class UriUploader(
|
|||
* @param sourceUris Array of content:// URIs to the files to upload
|
||||
* @param remotePaths Array of absolute paths to set to the uploaded files
|
||||
*/
|
||||
private fun copyThenUpload(sourceUris: Array<Uri>, vararg remotePaths: String) {
|
||||
private fun copyThenUpload(sourceUris: Array<Uri>, remotePaths: Array<String>) {
|
||||
if (mShowWaitingDialog) {
|
||||
mActivity.showLoadingDialog(mActivity.resources.getString(R.string.wait_for_tmp_copy_from_private_storage))
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue