#6449 Switch to removing specific invalid characters instead of including different character scripts for file names

This commit is contained in:
Christian Rowlands 2024-10-16 09:30:31 -04:00
parent f8b2bc0530
commit 686ca0512f
2 changed files with 26 additions and 18 deletions

View file

@ -1,5 +1,5 @@
/* /*
* Copyright 2020 The Matrix.org Foundation C.I.C. * Copyright (c) 2024 New Vector Ltd
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -16,15 +16,23 @@
package org.matrix.android.sdk.internal.util package org.matrix.android.sdk.internal.util
import androidx.test.ext.junit.runners.AndroidJUnit4
import org.junit.Assert.assertEquals import org.junit.Assert.assertEquals
import org.junit.Test import org.junit.Test
import org.junit.runner.RunWith
import org.matrix.android.sdk.InstrumentedTest
import org.matrix.android.sdk.internal.session.DefaultFileService.Companion.DEFAULT_FILENAME import org.matrix.android.sdk.internal.session.DefaultFileService.Companion.DEFAULT_FILENAME
import org.matrix.android.sdk.internal.util.file.safeFileName import org.matrix.android.sdk.internal.util.file.safeFileName
class FileUtilTest { /**
* These tests are run on an Android device because they need to use the static
* MimeTypeMap#getSingleton() method, which was failing in the unit test directory.
*/
@RunWith(AndroidJUnit4::class)
class FileUtilTest : InstrumentedTest {
@Test @Test
fun `should return original filename when valid characters are used`() { fun shouldReturnOriginalFilenameWhenValidCharactersAreUsed() {
val fileName = "validFileName.txt" val fileName = "validFileName.txt"
val mimeType = "text/plain" val mimeType = "text/plain"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -32,15 +40,15 @@ class FileUtilTest {
} }
@Test @Test
fun `should replace invalid characters with underscores`() { fun shouldReplaceInvalidCharactersWithUnderscores() {
val fileName = "invalid/filename:with*chars?.txt" val fileName = "invalid/filename:with*chars?.txt"
val mimeType = "text/plain" val mimeType = "text/plain"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
assertEquals("invalid_filename_with_chars_.txt", result) assertEquals("invalid/filename_with_chars_.txt", result)
} }
@Test @Test
fun `should allow Cyrillic characters in the filename`() { fun shouldAllowCyrillicCharactersInTheFilename() {
val fileName = "тестовыйФайл.txt" val fileName = "тестовыйФайл.txt"
val mimeType = "text/plain" val mimeType = "text/plain"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -48,7 +56,7 @@ class FileUtilTest {
} }
@Test @Test
fun `should allow Han characters in the filename`() { fun shouldAllowHanCharactersInTheFilename() {
val fileName = "测试文件.txt" val fileName = "测试文件.txt"
val mimeType = "text/plain" val mimeType = "text/plain"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -56,15 +64,15 @@ class FileUtilTest {
} }
@Test @Test
fun `should return default filename when input is null`() { fun shouldReturnDefaultFilenameWhenInputIsNull() {
val fileName = null val fileName = null
val mimeType = "text/plain" val mimeType = "text/plain"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
assertEquals(DEFAULT_FILENAME, result) assertEquals("$DEFAULT_FILENAME.txt", result)
} }
@Test @Test
fun `should add the correct extension when missing`() { fun shouldAddTheCorrectExtensionWhenMissing() {
val fileName = "myDocument" val fileName = "myDocument"
val mimeType = "application/pdf" val mimeType = "application/pdf"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -72,15 +80,15 @@ class FileUtilTest {
} }
@Test @Test
fun `should replace invalid characters and add the correct extension`() { fun shouldReplaceInvalidCharactersAndAddTheCorrectExtension() {
val fileName = "my*docu/ment" val fileName = "my*docu/ment"
val mimeType = "application/pdf" val mimeType = "application/pdf"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
assertEquals("my_docu_ment.pdf", result) assertEquals("my_docu/ment.pdf", result)
} }
@Test @Test
fun `should not modify the extension if it matches the mimeType`() { fun shouldNotModifyTheExtensionIfItMatchesTheMimeType() {
val fileName = "report.pdf" val fileName = "report.pdf"
val mimeType = "application/pdf" val mimeType = "application/pdf"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -88,7 +96,7 @@ class FileUtilTest {
} }
@Test @Test
fun `should replace spaces with underscores`() { fun shouldReplaceSpacesWithUnderscores() {
val fileName = "my report.doc" val fileName = "my report.doc"
val mimeType = "application/msword" val mimeType = "application/msword"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -96,7 +104,7 @@ class FileUtilTest {
} }
@Test @Test
fun `should append extension if file name has none and mimeType is valid`() { fun shouldAppendExtensionIfFileNameHasNoneAndMimeTypeIsValid() {
val fileName = "newfile" val fileName = "newfile"
val mimeType = "image/jpeg" val mimeType = "image/jpeg"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
@ -104,10 +112,10 @@ class FileUtilTest {
} }
@Test @Test
fun `should keep hyphenated names intact`() { fun shouldKeepHyphenatedNamesIntact() {
val fileName = "my-file-name" val fileName = "my-file-name"
val mimeType = "application/octet-stream" val mimeType = "application/octet-stream"
val result = safeFileName(fileName, mimeType) val result = safeFileName(fileName, mimeType)
assertEquals("my-file-name", result) assertEquals("my-file-name.bin", result)
} }
} }

View file

@ -29,7 +29,7 @@ fun safeFileName(fileName: String?, mimeType: String?): String {
// filename has to be safe for the Android System // filename has to be safe for the Android System
Timber.i("ISSUE: FileService: original fileName $fileName") Timber.i("ISSUE: FileService: original fileName $fileName")
val result = fileName val result = fileName
?.replace("[^\\p{sc=Cyrillic}\\p{sc=Han}a-z A-Z0-9\\\\.\\-]".toRegex(), "_") ?.replace("[\\\\?%*:|\"<>\\s]".toRegex(), "_")
?.takeIf { it.isNotEmpty() } ?.takeIf { it.isNotEmpty() }
?: DEFAULT_FILENAME ?: DEFAULT_FILENAME
Timber.i("ISSUE: FileService: safeFileName $result") Timber.i("ISSUE: FileService: safeFileName $result")