src/main/kotlin/nbt/Nbt.kt diff

← → edited src/main/kotlin/nbt/Nbt.kt

Ignore whitespaces Open in IDE

⁠ joe: 2025

Copy Copy

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2024 minecraft-dev
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, version 3.0 only.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/

package com.demonwav.mcdev.nbt

import com.demonwav.mcdev.asset.MCDevBundle
import com.demonwav.mcdev.nbt.tags.NbtTag
import com.demonwav.mcdev.nbt.tags.NbtTypeId
import com.demonwav.mcdev.nbt.tags.RootCompound
import com.demonwav.mcdev.nbt.tags.TagByte
import com.demonwav.mcdev.nbt.tags.TagByteArray
import com.demonwav.mcdev.nbt.tags.TagCompound
import com.demonwav.mcdev.nbt.tags.TagDouble
import com.demonwav.mcdev.nbt.tags.TagEnd
import com.demonwav.mcdev.nbt.tags.TagFloat
import com.demonwav.mcdev.nbt.tags.TagInt
import com.demonwav.mcdev.nbt.tags.TagIntArray
import com.demonwav.mcdev.nbt.tags.TagList
import com.demonwav.mcdev.nbt.tags.TagLong
import com.demonwav.mcdev.nbt.tags.TagLongArray
import com.demonwav.mcdev.nbt.tags.TagShort
import com.demonwav.mcdev.nbt.tags.TagString
import java.io.DataInputStream
import java.io.InputStream
import java.util.zip.GZIPInputStream
import java.util.zip.ZipException

object Nbt {

private fun getActualInputStream(stream: InputStream): Pair<DataInputStream, Boolean> {
return try {
DataInputStream(GZIPInputStream(stream)) to true
} catch (e: ZipException) {
stream.reset()
DataInputStream(stream) to false
}
}

/**
* Parse the NBT file from the InputStream and return the root TagCompound for the NBT file. This method closes the stream when
* it is finished with it.
*/
@Throws(MalformedNbtFileException::class)
fun buildTagTree(inputStream: InputStream, timeout: Long): Pair<RootCompound, Boolean> {
try {
val (stream, isCompressed) = getActualInputStream(inputStream)

stream.use {
val tagIdByte = stream.readByte()
val tagId = NbtTypeId.getById(tagIdByte)
?: throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))

if (tagId != NbtTypeId.COMPOUND) {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.invalid_root"))
}

val start = System.currentTimeMillis()

return RootCompound(stream.readUTF(), stream.readCompoundTag(start, timeout).tagMap) to isCompressed
}
} catch (e: Throwable) {
if (e is MalformedNbtFileException) {
throw e
} else {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.reading"), e)
}
}
}

private fun DataInputStream.readCompoundTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val tagMap = HashMap<String, NbtTag>()

var tagIdByte = this.readByte()
var tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}
while (tagId != NbtTypeId.END) {
val name = this.readUTF()

tagMap[name] = this.readTag(tagId, start, timeout)

tagIdByte = this.readByte()
tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}
}

return@checkTimeout TagCompound(tagMap)
}

private fun DataInputStream.readByteTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagByte(this.readByte()) }

private fun DataInputStream.readShortTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagShort(this.readShort()) }

private fun DataInputStream.readIntTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagInt(this.readInt()) }

private fun DataInputStream.readLongTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagLong(this.readLong()) }

private fun DataInputStream.readFloatTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagFloat(this.readFloat()) }

private fun DataInputStream.readDoubleTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagDouble(this.readDouble()) }

private fun DataInputStream.readStringTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagString(this.readUTF()) }

private fun DataInputStream.readListTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val tagIdByte = this.readByte()
val tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}

val length = this.readInt()
if (length <= 0) {
return@checkTimeout TagList(tagId, emptyList())
}

val list = List(length) {
this.readTag(tagId, start, timeout)
}
return@checkTimeout TagList(tagId, list)
}

private fun DataInputStream.readByteArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val bytes = ByteArray(length)
this.readFully(bytes)
return@checkTimeout TagByteArray(bytes)
}

private fun DataInputStream.readIntArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val ints = IntArray(length) {
this.readInt()
}

return@checkTimeout TagIntArray(ints)
}

private fun DataInputStream.readLongArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val longs = LongArray(length) {
this.readLong()
}

return@checkTimeout TagLongArray(longs)
}

private fun DataInputStream.readTag(tagId: NbtTypeId, start: Long, timeout: Long): NbtTag {
return when (tagId) {
NbtTypeId.END -> TagEnd
NbtTypeId.BYTE -> this.readByteTag(start, timeout)
NbtTypeId.SHORT -> this.readShortTag(start, timeout)
NbtTypeId.INT -> this.readIntTag(start, timeout)
NbtTypeId.LONG -> this.readLongTag(start, timeout)
NbtTypeId.FLOAT -> this.readFloatTag(start, timeout)
NbtTypeId.DOUBLE -> this.readDoubleTag(start, timeout)
NbtTypeId.BYTE_ARRAY -> this.readByteArrayTag(start, timeout)
NbtTypeId.STRING -> this.readStringTag(start, timeout)
NbtTypeId.LIST -> this.readListTag(start, timeout)
NbtTypeId.COMPOUND -> this.readCompoundTag(start, timeout)
NbtTypeId.INT_ARRAY -> this.readIntArrayTag(start, timeout)
NbtTypeId.LONG_ARRAY -> this.readLongArrayTag(start, timeout)
}
}

private inline fun <T : Any> checkTimeout(start: Long, timeout: Long, action: () -> T): T {
val now = System.currentTimeMillis()
val took = now - start

if (took > timeout) {
throw NbtFileParseTimeoutException(MCDevBundle("nbt.lang.errors.parse_timeout", took, timeout))
}

return action()
}
}

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2025 minecraft-dev
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, version 3.0 only.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/

package com.demonwav.mcdev.nbt

import com.demonwav.mcdev.asset.MCDevBundle
import com.demonwav.mcdev.nbt.tags.NbtTag
import com.demonwav.mcdev.nbt.tags.NbtTypeId
import com.demonwav.mcdev.nbt.tags.RootCompound
import com.demonwav.mcdev.nbt.tags.TagByte
import com.demonwav.mcdev.nbt.tags.TagByteArray
import com.demonwav.mcdev.nbt.tags.TagCompound
import com.demonwav.mcdev.nbt.tags.TagDouble
import com.demonwav.mcdev.nbt.tags.TagEnd
import com.demonwav.mcdev.nbt.tags.TagFloat
import com.demonwav.mcdev.nbt.tags.TagInt
import com.demonwav.mcdev.nbt.tags.TagIntArray
import com.demonwav.mcdev.nbt.tags.TagList
import com.demonwav.mcdev.nbt.tags.TagLong
import com.demonwav.mcdev.nbt.tags.TagLongArray
import com.demonwav.mcdev.nbt.tags.TagShort
import com.demonwav.mcdev.nbt.tags.TagString
import java.io.DataInputStream
import java.io.InputStream
import java.util.zip.GZIPInputStream
import java.util.zip.ZipException

object Nbt {

private fun getActualInputStream(stream: InputStream): Pair<DataInputStream, Boolean> {
return try {
DataInputStream(GZIPInputStream(stream)) to true
} catch (e: ZipException) {
stream.reset()
DataInputStream(stream) to false
}
}

/**
* Parse the NBT file from the InputStream and return the root TagCompound for the NBT file. This method closes the stream when
* it is finished with it.
*/
@Throws(MalformedNbtFileException::class)
fun buildTagTree(inputStream: InputStream, timeout: Long): Pair<RootCompound, Boolean> {
try {
val (stream, isCompressed) = getActualInputStream(inputStream)

stream.use {
val tagIdByte = stream.readByte()
val tagId = NbtTypeId.getById(tagIdByte)
?: throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))

if (tagId != NbtTypeId.COMPOUND) {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.invalid_root"))
}

val start = System.currentTimeMillis()

return RootCompound(stream.readUTF(), stream.readCompoundTag(start, timeout).tagMap) to isCompressed
}
} catch (e: Throwable) {
if (e is MalformedNbtFileException) {
throw e
} else {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.reading"), e)
}
}
}

private fun DataInputStream.readCompoundTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val tagMap = HashMap<String, NbtTag>()

var tagIdByte = this.readByte()
var tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}
while (tagId != NbtTypeId.END) {
val name = this.readUTF()

tagMap[name] = this.readTag(tagId, start, timeout)

tagIdByte = this.readByte()
tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}
}

return@checkTimeout TagCompound(tagMap)
}

private fun DataInputStream.readByteTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagByte(this.readByte()) }

private fun DataInputStream.readShortTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagShort(this.readShort()) }

private fun DataInputStream.readIntTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagInt(this.readInt()) }

private fun DataInputStream.readLongTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagLong(this.readLong()) }

private fun DataInputStream.readFloatTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagFloat(this.readFloat()) }

private fun DataInputStream.readDoubleTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagDouble(this.readDouble()) }

private fun DataInputStream.readStringTag(start: Long, timeout: Long) =
checkTimeout(start, timeout) { TagString(this.readUTF()) }

private fun DataInputStream.readListTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val tagIdByte = this.readByte()
val tagId =
NbtTypeId.getById(tagIdByte) ?: run {
throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte))
}

val length = this.readInt()
if (length <= 0) {
return@checkTimeout TagList(tagId, emptyList())
}

val list = List(length) {
this.readTag(tagId, start, timeout)
}
return@checkTimeout TagList(tagId, list)
}

private fun DataInputStream.readByteArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val bytes = ByteArray(length)
this.readFully(bytes)
return@checkTimeout TagByteArray(bytes)
}

private fun DataInputStream.readIntArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val ints = IntArray(length) {
this.readInt()
}

return@checkTimeout TagIntArray(ints)
}

private fun DataInputStream.readLongArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) {
val length = this.readInt()

val longs = LongArray(length) {
this.readLong()
}

return@checkTimeout TagLongArray(longs)
}

private fun DataInputStream.readTag(tagId: NbtTypeId, start: Long, timeout: Long): NbtTag {
return when (tagId) {
NbtTypeId.END -> TagEnd
NbtTypeId.BYTE -> this.readByteTag(start, timeout)
NbtTypeId.SHORT -> this.readShortTag(start, timeout)
NbtTypeId.INT -> this.readIntTag(start, timeout)
NbtTypeId.LONG -> this.readLongTag(start, timeout)
NbtTypeId.FLOAT -> this.readFloatTag(start, timeout)
NbtTypeId.DOUBLE -> this.readDoubleTag(start, timeout)
NbtTypeId.BYTE_ARRAY -> this.readByteArrayTag(start, timeout)
NbtTypeId.STRING -> this.readStringTag(start, timeout)
NbtTypeId.LIST -> this.readListTag(start, timeout)
NbtTypeId.COMPOUND -> this.readCompoundTag(start, timeout)
NbtTypeId.INT_ARRAY -> this.readIntArrayTag(start, timeout)
NbtTypeId.LONG_ARRAY -> this.readLongArrayTag(start, timeout)
}
}

private inline fun <T : Any> checkTimeout(start: Long, timeout: Long, action: () -> T): T {
val now = System.currentTimeMillis()
val took = now - start

if (took > timeout) {
throw NbtFileParseTimeoutException(MCDevBundle("nbt.lang.errors.parse_timeout", took, timeout))
}

return action()
}
}

/* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2024 minecraft-dev * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published * by the Free Software Foundation, version 3.0 only. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program. If not, see <https://www.gnu.org/licenses/>. */ package com.demonwav.mcdev.nbt import com.demonwav.mcdev.asset.MCDevBundle import com.demonwav.mcdev.nbt.tags.NbtTag import com.demonwav.mcdev.nbt.tags.NbtTypeId import com.demonwav.mcdev.nbt.tags.RootCompound import com.demonwav.mcdev.nbt.tags.TagByte import com.demonwav.mcdev.nbt.tags.TagByteArray import com.demonwav.mcdev.nbt.tags.TagCompound import com.demonwav.mcdev.nbt.tags.TagDouble import com.demonwav.mcdev.nbt.tags.TagEnd import com.demonwav.mcdev.nbt.tags.TagFloat import com.demonwav.mcdev.nbt.tags.TagInt import com.demonwav.mcdev.nbt.tags.TagIntArray import com.demonwav.mcdev.nbt.tags.TagList import com.demonwav.mcdev.nbt.tags.TagLong import com.demonwav.mcdev.nbt.tags.TagLongArray import com.demonwav.mcdev.nbt.tags.TagShort import com.demonwav.mcdev.nbt.tags.TagString import java.io.DataInputStream import java.io.InputStream import java.util.zip.GZIPInputStream import java.util.zip.ZipException object Nbt { private fun getActualInputStream(stream: InputStream): Pair<DataInputStream, Boolean> { return try { DataInputStream(GZIPInputStream(stream)) to true } catch (e: ZipException) { stream.reset() DataInputStream(stream) to false } } /** * Parse the NBT file from the InputStream and return the root TagCompound for the NBT file. This method closes the stream when * it is finished with it. */ @Throws(MalformedNbtFileException::class) fun buildTagTree(inputStream: InputStream, timeout: Long): Pair<RootCompound, Boolean> { try { val (stream, isCompressed) = getActualInputStream(inputStream) stream.use { val tagIdByte = stream.readByte() val tagId = NbtTypeId.getById(tagIdByte) ?: throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) if (tagId != NbtTypeId.COMPOUND) { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.invalid_root")) } val start = System.currentTimeMillis() return RootCompound(stream.readUTF(), stream.readCompoundTag(start, timeout).tagMap) to isCompressed } } catch (e: Throwable) { if (e is MalformedNbtFileException) { throw e } else { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.reading"), e) } } } private fun DataInputStream.readCompoundTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val tagMap = HashMap<String, NbtTag>() var tagIdByte = this.readByte() var tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } while (tagId != NbtTypeId.END) { val name = this.readUTF() tagMap[name] = this.readTag(tagId, start, timeout) tagIdByte = this.readByte() tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } } return@checkTimeout TagCompound(tagMap) } private fun DataInputStream.readByteTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagByte(this.readByte()) } private fun DataInputStream.readShortTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagShort(this.readShort()) } private fun DataInputStream.readIntTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagInt(this.readInt()) } private fun DataInputStream.readLongTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagLong(this.readLong()) } private fun DataInputStream.readFloatTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagFloat(this.readFloat()) } private fun DataInputStream.readDoubleTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagDouble(this.readDouble()) } private fun DataInputStream.readStringTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagString(this.readUTF()) } private fun DataInputStream.readListTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val tagIdByte = this.readByte() val tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } val length = this.readInt() if (length <= 0) { return@checkTimeout TagList(tagId, emptyList()) } val list = List(length) { this.readTag(tagId, start, timeout) } return@checkTimeout TagList(tagId, list) } private fun DataInputStream.readByteArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val bytes = ByteArray(length) this.readFully(bytes) return@checkTimeout TagByteArray(bytes) } private fun DataInputStream.readIntArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val ints = IntArray(length) { this.readInt() } return@checkTimeout TagIntArray(ints) } private fun DataInputStream.readLongArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val longs = LongArray(length) { this.readLong() } return@checkTimeout TagLongArray(longs) } private fun DataInputStream.readTag(tagId: NbtTypeId, start: Long, timeout: Long): NbtTag { return when (tagId) { NbtTypeId.END -> TagEnd NbtTypeId.BYTE -> this.readByteTag(start, timeout) NbtTypeId.SHORT -> this.readShortTag(start, timeout) NbtTypeId.INT -> this.readIntTag(start, timeout) NbtTypeId.LONG -> this.readLongTag(start, timeout) NbtTypeId.FLOAT -> this.readFloatTag(start, timeout) NbtTypeId.DOUBLE -> this.readDoubleTag(start, timeout) NbtTypeId.BYTE_ARRAY -> this.readByteArrayTag(start, timeout) NbtTypeId.STRING -> this.readStringTag(start, timeout) NbtTypeId.LIST -> this.readListTag(start, timeout) NbtTypeId.COMPOUND -> this.readCompoundTag(start, timeout) NbtTypeId.INT_ARRAY -> this.readIntArrayTag(start, timeout) NbtTypeId.LONG_ARRAY -> this.readLongArrayTag(start, timeout) } } private inline fun <T : Any> checkTimeout(start: Long, timeout: Long, action: () -> T): T { val now = System.currentTimeMillis() val took = now - start if (took > timeout) { throw NbtFileParseTimeoutException(MCDevBundle("nbt.lang.errors.parse_timeout", took, timeout)) } return action() } } /* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2025 minecraft-dev * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published * by the Free Software Foundation, version 3.0 only. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program. If not, see <https://www.gnu.org/licenses/>. */ package com.demonwav.mcdev.nbt import com.demonwav.mcdev.asset.MCDevBundle import com.demonwav.mcdev.nbt.tags.NbtTag import com.demonwav.mcdev.nbt.tags.NbtTypeId import com.demonwav.mcdev.nbt.tags.RootCompound import com.demonwav.mcdev.nbt.tags.TagByte import com.demonwav.mcdev.nbt.tags.TagByteArray import com.demonwav.mcdev.nbt.tags.TagCompound import com.demonwav.mcdev.nbt.tags.TagDouble import com.demonwav.mcdev.nbt.tags.TagEnd import com.demonwav.mcdev.nbt.tags.TagFloat import com.demonwav.mcdev.nbt.tags.TagInt import com.demonwav.mcdev.nbt.tags.TagIntArray import com.demonwav.mcdev.nbt.tags.TagList import com.demonwav.mcdev.nbt.tags.TagLong import com.demonwav.mcdev.nbt.tags.TagLongArray import com.demonwav.mcdev.nbt.tags.TagShort import com.demonwav.mcdev.nbt.tags.TagString import java.io.DataInputStream import java.io.InputStream import java.util.zip.GZIPInputStream import java.util.zip.ZipException object Nbt { private fun getActualInputStream(stream: InputStream): Pair<DataInputStream, Boolean> { return try { DataInputStream(GZIPInputStream(stream)) to true } catch (e: ZipException) { stream.reset() DataInputStream(stream) to false } } /** * Parse the NBT file from the InputStream and return the root TagCompound for the NBT file. This method closes the stream when * it is finished with it. */ @Throws(MalformedNbtFileException::class) fun buildTagTree(inputStream: InputStream, timeout: Long): Pair<RootCompound, Boolean> { try { val (stream, isCompressed) = getActualInputStream(inputStream) stream.use { val tagIdByte = stream.readByte() val tagId = NbtTypeId.getById(tagIdByte) ?: throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) if (tagId != NbtTypeId.COMPOUND) { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.invalid_root")) } val start = System.currentTimeMillis() return RootCompound(stream.readUTF(), stream.readCompoundTag(start, timeout).tagMap) to isCompressed } } catch (e: Throwable) { if (e is MalformedNbtFileException) { throw e } else { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.reading"), e) } } } private fun DataInputStream.readCompoundTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val tagMap = HashMap<String, NbtTag>() var tagIdByte = this.readByte() var tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } while (tagId != NbtTypeId.END) { val name = this.readUTF() tagMap[name] = this.readTag(tagId, start, timeout) tagIdByte = this.readByte() tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } } return@checkTimeout TagCompound(tagMap) } private fun DataInputStream.readByteTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagByte(this.readByte()) } private fun DataInputStream.readShortTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagShort(this.readShort()) } private fun DataInputStream.readIntTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagInt(this.readInt()) } private fun DataInputStream.readLongTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagLong(this.readLong()) } private fun DataInputStream.readFloatTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagFloat(this.readFloat()) } private fun DataInputStream.readDoubleTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagDouble(this.readDouble()) } private fun DataInputStream.readStringTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { TagString(this.readUTF()) } private fun DataInputStream.readListTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val tagIdByte = this.readByte() val tagId = NbtTypeId.getById(tagIdByte) ?: run { throw MalformedNbtFileException(MCDevBundle("nbt.lang.errors.wrong_tag_id", tagIdByte)) } val length = this.readInt() if (length <= 0) { return@checkTimeout TagList(tagId, emptyList()) } val list = List(length) { this.readTag(tagId, start, timeout) } return@checkTimeout TagList(tagId, list) } private fun DataInputStream.readByteArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val bytes = ByteArray(length) this.readFully(bytes) return@checkTimeout TagByteArray(bytes) } private fun DataInputStream.readIntArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val ints = IntArray(length) { this.readInt() } return@checkTimeout TagIntArray(ints) } private fun DataInputStream.readLongArrayTag(start: Long, timeout: Long) = checkTimeout(start, timeout) { val length = this.readInt() val longs = LongArray(length) { this.readLong() } return@checkTimeout TagLongArray(longs) } private fun DataInputStream.readTag(tagId: NbtTypeId, start: Long, timeout: Long): NbtTag { return when (tagId) { NbtTypeId.END -> TagEnd NbtTypeId.BYTE -> this.readByteTag(start, timeout) NbtTypeId.SHORT -> this.readShortTag(start, timeout) NbtTypeId.INT -> this.readIntTag(start, timeout) NbtTypeId.LONG -> this.readLongTag(start, timeout) NbtTypeId.FLOAT -> this.readFloatTag(start, timeout) NbtTypeId.DOUBLE -> this.readDoubleTag(start, timeout) NbtTypeId.BYTE_ARRAY -> this.readByteArrayTag(start, timeout) NbtTypeId.STRING -> this.readStringTag(start, timeout) NbtTypeId.LIST -> this.readListTag(start, timeout) NbtTypeId.COMPOUND -> this.readCompoundTag(start, timeout) NbtTypeId.INT_ARRAY -> this.readIntArrayTag(start, timeout) NbtTypeId.LONG_ARRAY -> this.readLongArrayTag(start, timeout) } } private inline fun <T : Any> checkTimeout(start: Long, timeout: Long, action: () -> T): T { val now = System.currentTimeMillis() val took = now - start if (took > timeout) { throw NbtFileParseTimeoutException(MCDevBundle("nbt.lang.errors.parse_timeout", took, timeout)) } return action() } }

Add build comment

Mute test

Server communication failure

Run Custom Build

TODO

Responsibility

Edit tags

Loading...

Are you sure?