terribleplan/mangadex_at_home
1
0
Fork 0
forked from gitlab-com-mangadex-pub/mangadex_at_home
Code Issues Pull requests Releases Wiki Activity

Properly close snapshot if referer invalid

Browse source
This commit is contained in:
Amos Ng 2020-06-29 06:42:44 +08:00
parent d24c061cc6
commit 620a859d5c
No known key found for this signature in database
GPG key ID: 89086414F634D123
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/main/kotlin/mdnet/base/server/ImageServer.kt
View file

@ -105,6 +105,10 @@ class ImageServer(private val cache: DiskLruCache, private val statistics: Atomi
handled.set(true) handled.set(true)
if (referer != null && !referer.startsWith("https://mangadex.org")) { if (referer != null && !referer.startsWith("https://mangadex.org")) {
if (snapshot != null) {
snapshot.close()
}
Response(Status.FORBIDDEN) Response(Status.FORBIDDEN)
} else if (snapshot != null && imageDatum != null) { } else if (snapshot != null && imageDatum != null) {
request.handleCacheHit(sanitizedUri, getRc4(rc4Bytes), snapshot, imageDatum) request.handleCacheHit(sanitizedUri, getRc4(rc4Bytes), snapshot, imageDatum)