From 8ad087c40679c200c168f06dddd00b2166194cab Mon Sep 17 00:00:00 2001 From: Him188 Date: Fri, 7 Feb 2020 12:32:41 +0800 Subject: [PATCH] Add fast-fail size check --- .../kotlin/net.mamoe.mirai/utils/ExternalImage.kt | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/ExternalImage.kt b/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/ExternalImage.kt index 011346d81..ea89adfb4 100644 --- a/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/ExternalImage.kt +++ b/mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/ExternalImage.kt @@ -25,9 +25,13 @@ class ExternalImage( val md5: ByteArray, imageFormat: String, val input: Input, - val inputSize: Long, + val inputSize: Long, // dont be greater than Int.MAX val filename: String ) { + init { + check(inputSize in Int.MIN_VALUE.toLong()..Int.MAX_VALUE.toLong()) { "file is too big" } + } + companion object { operator fun invoke( width: Int,