add anchor checksum

JuliaHEP · Jul 1, 2024 · e668c84 · e668c84
1 parent dd3c06c
commit e668c84
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/RNTuple/bootstrap.jl b/src/RNTuple/bootstrap.jl
@@ -16,6 +16,9 @@ end
 function ROOT_3a3a_Experimental_3a3a_RNTuple(io, tkey::TKey, refs)
     local_io = datastream(io, tkey)
     skip(local_io, 6)
+    _before_anchor = position(local_io)
+    anchor_checksum = xxh3_64(read(local_io, 2*4 + 6*8))
+    seek(local_io, _before_anchor)
     anchor = ROOT_3a3a_Experimental_3a3a_RNTuple(;
                     fVersionEpoch = readtype(local_io, UInt16),
                     fVersionMajor = readtype(local_io, UInt16),
@@ -30,6 +33,9 @@ function ROOT_3a3a_Experimental_3a3a_RNTuple(io, tkey::TKey, refs)
                     fChecksum = readtype(local_io, UInt64),
                                        )
 
+    @assert anchor.fChecksum == anchor_checksum "RNtuple anchor checksum doesn't match"
+
+
     header_bytes = decompress_bytes(read_seek_nb(io, anchor.fSeekHeader, anchor.fNBytesHeader), anchor.fLenHeader)
     header_io = IOBuffer(header_bytes)
     header = _rntuple_read(header_io, RNTupleEnvelope{RNTupleHeader})