Signal/Android
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Android.git synced 2025-12-22 20:18:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix avatar provider buffer underflow.

Browse Source
This commit is contained in:
Clark
2023-07-25 11:31:21 -04:00
committed by Cody Henthorne
parent 557ef5820e
commit 5648fd2e91
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/src/main/java/org/thoughtcrime/securesms/providers/AvatarProvider.kt
View File

@@ -195,8 +195,10 @@ class AvatarProvider : BaseContentProvider() {
}
val memoryFile = MemoryFile("${recipient.id}-imf", outputStream.size())
StreamUtil.copy(ByteArrayInputStream(outputStream.toByteArray()), memoryFile.outputStream)
StreamUtil.close(memoryFile.outputStream)
val memoryFileOutputStream = memoryFile.outputStream
StreamUtil.copy(ByteArrayInputStream(outputStream.toByteArray()), memoryFileOutputStream)
StreamUtil.close(memoryFileOutputStream)
return MemoryFileUtil.getParcelFileDescriptor(memoryFile)
}
@@ -217,8 +219,8 @@ class AvatarProvider : BaseContentProvider() {
override fun onRead(offset: Long, size: Int, data: ByteArray?): Int {
ensureResourceLoaded()
return memoryFile!!.readBytes(data, offset.toInt(), 0, size)
val memoryFileSnapshot = memoryFile
return memoryFileSnapshot!!.readBytes(data, offset.toInt(), 0, size.coerceAtMost((memoryFileSnapshot.length() - offset).toInt()))
}
override fun onRelease() {
@@ -240,8 +242,10 @@ class AvatarProvider : BaseContentProvider() {
Log.i(TAG, "Writing ${recipient.id} icon to MemoryFile")
memoryFile = MemoryFile("${recipient.id}-imf", outputStream.size())
StreamUtil.copy(ByteArrayInputStream(outputStream.toByteArray()), memoryFile!!.outputStream)
StreamUtil.close(memoryFile!!.outputStream)
val memoryFileOutputStream = memoryFile!!.outputStream
StreamUtil.copy(ByteArrayInputStream(outputStream.toByteArray()), memoryFileOutputStream)
StreamUtil.close(memoryFileOutputStream)
}
}
}