athaliar/APKDownloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DownloadWorker: improve file verification

Browse Source
This commit is contained in:
Rahul Patel
2025-02-26 02:29:35 +05:30
parent b867d4a527
commit 8af3fde374
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
app/src/main/java/com/aurora/store/data/work/DownloadWorker.kt
View File

@@ -358,6 +358,7 @@ class DownloadWorker @AssistedInject constructor(
when (status) { when (status) {
DownloadStatus.VERIFYING, DownloadStatus.VERIFYING,
DownloadStatus.CANCELLED -> return DownloadStatus.CANCELLED -> return
DownloadStatus.COMPLETED -> { DownloadStatus.COMPLETED -> {
// Mark progress as 100 manually to avoid race conditions // Mark progress as 100 manually to avoid race conditions
download.progress = 100 download.progress = 100
@@ -379,20 +380,28 @@ class DownloadWorker @AssistedInject constructor(
@OptIn(ExperimentalStdlibApi::class) @OptIn(ExperimentalStdlibApi::class)
private suspend fun verifyFile(gFile: GPlayFile): Boolean { private suspend fun verifyFile(gFile: GPlayFile): Boolean {
val file = PathUtil.getLocalFile(appContext, gFile, download) val file = PathUtil.getLocalFile(appContext, gFile, download)
val algorithm = if (gFile.sha256.isBlank()) Algorithm.SHA1 else Algorithm.SHA256 val algorithm = if (gFile.sha256.isBlank()) Algorithm.SHA1 else Algorithm.SHA256
val expectedSha = if (algorithm == Algorithm.SHA1) gFile.sha1 else gFile.sha256 val expectedSha = if (algorithm == Algorithm.SHA1) gFile.sha1 else gFile.sha256
if (expectedSha.isBlank()) return false
return withContext(Dispatchers.IO) { return withContext(Dispatchers.IO) {
val messageDigest = MessageDigest.getInstance(algorithm.value) try {
DigestInputStream(file.inputStream(), messageDigest).use { input -> val messageDigest = MessageDigest.getInstance(algorithm.value)
val buffer = ByteArray(DEFAULT_BUFFER_SIZE) file.inputStream().use { fis ->
var read = input.read(buffer, 0, DEFAULT_BUFFER_SIZE) DigestInputStream(fis, messageDigest).use { dis ->
while (read > -1) { val buffer = ByteArray(DEFAULT_BUFFER_SIZE)
read = input.read(buffer, 0, DEFAULT_BUFFER_SIZE) while (dis.read(buffer) != -1) { /* Just read, digest updates automatically */
}
}
} }
messageDigest.digest().toHexString() == expectedSha
} catch (e: Exception) {
Log.e(TAG, "Failed to verify $file", e)
false
} }
val sha = messageDigest.digest().toHexString()
return@withContext sha == expectedSha
} }
} }
} }