reading from the stream in a loop

this ensures that desired number of bytes will be read,
as stream.read(byte[] buffer) will just make effort to read
at least 1 byte
This commit is contained in:
Kamil Bobrowski 2020-08-03 21:53:55 +02:00 committed by Marvin W
parent 5f70d943cb
commit 7d91551126
1 changed files with 9 additions and 1 deletions

View File

@ -159,7 +159,15 @@ class ExposureNotificationServiceImpl(private val context: Context, private val
val entry = stream.nextEntry ?: break
if (entry.name == "export.bin") {
val prefix = ByteArray(16)
if (stream.read(prefix) == prefix.size && String(prefix).trim() == "EK Export v1") {
var totalBytesRead = 0
var bytesRead = 0
while (bytesRead != -1 && totalBytesRead < prefix.size) {
bytesRead = stream.read(prefix, totalBytesRead, prefix.size - totalBytesRead)
if (bytesRead > 0) {
totalBytesRead += bytesRead
}
}
if (totalBytesRead == prefix.size && String(prefix).trim() == "EK Export v1") {
val fileKeys = storeDiagnosisKeyExport(params.token, TemporaryExposureKeyExport.ADAPTER.decode(stream))
keys + fileKeys
} else {