nutomic/ensichat
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Do not crash when receiving invalid message.

Browse source
This commit is contained in:
Felix Ableitner 2015-01-29 17:46:57 +01:00
parent 5a460c9527
commit 2f08ec0124
2 changed files with 28 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/src/main/scala/com/nutomic/ensichat/bluetooth/TransferThread.scala
View file

@ -5,6 +5,7 @@ import java.io._
import android.bluetooth.BluetoothSocket import android.bluetooth.BluetoothSocket
import android.util.Log import android.util.Log
import com.nutomic.ensichat.protocol._ import com.nutomic.ensichat.protocol._
import com.nutomic.ensichat.protocol.messages.Message.ReadMessageException
import com.nutomic.ensichat.protocol.messages.{ConnectionInfo, Message, MessageHeader} import com.nutomic.ensichat.protocol.messages.{ConnectionInfo, Message, MessageHeader}
/** /**
@ -54,8 +55,8 @@ class TransferThread(device: Device, socket: BluetoothSocket, Handler: Bluetooth
onReceive(msg, device.Id) onReceive(msg, device.Id)
} }
} catch { } catch {
case e: RuntimeException => case e: ReadMessageException =>
Log.i(Tag, "Received invalid message", e) Log.i(Tag, "Failed to read message", e)
case e: IOException => case e: IOException =>
Log.w(Tag, "Failed to read incoming message", e) Log.w(Tag, "Failed to read incoming message", e)
close() close()

9
app/src/main/scala/com/nutomic/ensichat/protocol/messages/Message.scala
View file

@ -1,6 +1,7 @@
package com.nutomic.ensichat.protocol.messages package com.nutomic.ensichat.protocol.messages
import java.io.InputStream import java.io.InputStream
import java.security.spec.InvalidKeySpecException
object Message { object Message {
@ -16,7 +17,11 @@ object Message {
val Charset = "UTF-8" val Charset = "UTF-8"
class ReadMessageException(throwable: Throwable)
extends RuntimeException(throwable)
def read(stream: InputStream): Message = { def read(stream: InputStream): Message = {
try {
val headerBytes = new Array[Byte](MessageHeader.Length) val headerBytes = new Array[Byte](MessageHeader.Length)
stream.read(headerBytes, 0, MessageHeader.Length) stream.read(headerBytes, 0, MessageHeader.Length)
val header = MessageHeader.read(headerBytes) val header = MessageHeader.read(headerBytes)
@ -37,6 +42,10 @@ object Message {
} }
new Message(header, crypto, body) new Message(header, crypto, body)
} catch {
case e @ (_ : OutOfMemoryError | _ : InvalidKeySpecException) =>
throw new ReadMessageException(e)
}
} }
} }