mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 02:41:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

python: fix magic number in writer, assert current spec version and magic number in reader

Browse Source
This commit is contained in:
Brandon Liu
2022-11-01 17:24:29 +08:00
parent 302eeab47e
commit cd1374cc33
2 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/pmtiles/tile.py
View File

@@ -191,8 +191,19 @@ def serialize_directory(entries):
return gzip.compress(b_io.getvalue())
class SpecVersionUnsupported(Exception):
pass
class MagicNumberNotFound(Exception):
pass
def deserialize_header(buf):
if buf[0:7].decode() != "PMTiles":
raise MagicNumberNotFound()
if buf[7] != 0x3:
raise SpecVersionUnsupported()
def read_uint64(pos):
return int.from_bytes(buf[pos : pos + 8], byteorder="little")
@@ -240,7 +251,7 @@ def serialize_header(h):
b_io.write(i.to_bytes(1, byteorder="little"))
b_io.write("PMTiles".encode())
b_io.write(b"3")
b_io.write(b"\x03")
write_uint64(h["root_offset"])
write_uint64(h["root_length"])
write_uint64(h["metadata_offset"])