152 lines
4.9 KiB
Scheme
152 lines
4.9 KiB
Scheme
(import
|
|
(fmt fmt)
|
|
(matchable))
|
|
|
|
;;;;---------------------------------------------------
|
|
;;;; Constants
|
|
;;;;---------------------------------------------------
|
|
|
|
;; FIXME: duplicate with main.scm
|
|
(define (current-metadata)
|
|
"./metadata.bin")
|
|
|
|
(define metadata-block-size 4096)
|
|
(define superblock-magic 27022010)
|
|
(define superblock-salt 160774)
|
|
(define uuid-size 16)
|
|
(define space-map-root-size 128)
|
|
|
|
;;;;---------------------------------------------------
|
|
;;;; Metadata IO
|
|
;;;;---------------------------------------------------
|
|
|
|
(define (open-metadata path)
|
|
(open-file-input-port path (file-options) (buffer-mode none)))
|
|
|
|
(define-syntax with-metadata
|
|
(syntax-rules ()
|
|
((_ (port path) body ...) (let ((port (open-metadata path)))
|
|
(dynamic-wind
|
|
(lambda () #f)
|
|
(lambda () body ...)
|
|
(lambda () (close-port port)))))))
|
|
|
|
;; FIXME: return our own condition?
|
|
(define (io-error msg)
|
|
(raise (condition
|
|
(make-error)
|
|
(make-message-condition msg))))
|
|
|
|
;;; Returns a 4k bytevector
|
|
(define (read-block port b)
|
|
(set-port-position! port (* b metadata-block-size))
|
|
(let ((data (get-bytevector-n port metadata-block-size)))
|
|
(unless (and (not (eof-object? data))
|
|
(= metadata-block-size (bytevector-length data)))
|
|
(io-error (fmt #f (dsp "unable to read metadata block: ") (num b))))
|
|
data))
|
|
|
|
;;;;---------------------------------------------------
|
|
;;;; CRC32
|
|
;;;;---------------------------------------------------
|
|
|
|
;; FIXME: move to own library
|
|
(load-shared-object "libz.so")
|
|
(define crc32
|
|
(foreign-procedure "crc32" (unsigned-long u8* unsigned-int) unsigned-long))
|
|
|
|
(define crc32-combine
|
|
(foreign-procedure "crc32_combine" (unsigned-long unsigned-long unsigned-long) unsigned-long))
|
|
|
|
;; FIXME: stop copying the bytevector. I'm not sure how to pass an offset into
|
|
;; the bv.
|
|
(define (crc32-region salt bv start end)
|
|
(assert (< start end))
|
|
(let ((len (- end start)))
|
|
(let ((copy (make-bytevector len)))
|
|
(bytevector-copy! bv start copy 0 len)
|
|
(let ((crc (crc32 salt copy 0)))
|
|
(crc32 crc copy len)))))
|
|
|
|
;;;;---------------------------------------------------
|
|
;;;; Decoding
|
|
;;;;---------------------------------------------------
|
|
|
|
(define-syntax unpack-type
|
|
(syntax-rules (le32 le64 bytes)
|
|
((_ bv offset le32)
|
|
(bytevector-u32-ref bv offset (endianness little)))
|
|
|
|
((_ bv offset le64)
|
|
(bytevector-u64-ref bv offset (endianness little)))
|
|
|
|
((_ bv offset (bytes count))
|
|
(let ((copy (make-bytevector count)))
|
|
(bytevector-copy! bv offset copy 0 count)
|
|
copy))))
|
|
|
|
(define (size-type t)
|
|
(syntax->datum
|
|
(syntax-case t (le32 le64 bytes)
|
|
(le32 #'4)
|
|
(le64 #'8)
|
|
((bytes count) #'count))))
|
|
|
|
(define-syntax binary-format
|
|
(lambda (x)
|
|
(syntax-case x ()
|
|
((_ (name pack-name unpack-name) (field type) ...)
|
|
(with-syntax ((((t o) ...)
|
|
(let f ((acc 0) (types #'(type ...)))
|
|
(if (null? types)
|
|
'()
|
|
(cons (list (car types) acc)
|
|
(f (+ acc (size-type (car types))) (cdr types)))))))
|
|
#'(begin
|
|
(define-record-type name (fields field ...))
|
|
|
|
(define (unpack-name bv offset)
|
|
((record-constructor (record-type-descriptor name))
|
|
(unpack-type bv (+ offset o) t) ...))))))))
|
|
|
|
(binary-format (superblock pack-superblock unpack-superblock)
|
|
(csum le32)
|
|
(flags le32)
|
|
(block-nr le64)
|
|
(uuid (bytes 16))
|
|
(magic le64)
|
|
(version le32)
|
|
(time le32)
|
|
(trans-id le64)
|
|
(metadata-snap le64)
|
|
(data-space-map-root (bytes 128))
|
|
(metadata-space-map-root (bytes 128))
|
|
(data-mapping-root le64)
|
|
(device-details-root le64)
|
|
(data-block-size le32)
|
|
(metadata-block-size le32)
|
|
(metadata-nr-blocks le64)
|
|
(compat-flags le32)
|
|
(compat-ro-flags le32)
|
|
(incompat-flags le32))
|
|
|
|
;;;;---------------------------------------------------
|
|
;;;; Top level
|
|
;;;;---------------------------------------------------
|
|
|
|
(define (check-magic sb)
|
|
((let ((m (bytevector-u32-ref sb 32 (endianness little))))
|
|
(fmt #t (dsp "on disk magic: ") (num m) nl)
|
|
)))
|
|
|
|
(define (read-superblock)
|
|
(with-metadata (md (current-metadata))
|
|
(unpack-superblock (read-block md 0) 0)))
|
|
|
|
(define (check-superblock)
|
|
(with-metadata (md (current-metadata))
|
|
(let ((superblock (read-block md 0)))
|
|
(fmt #t (dsp "checksum on disk: ") (dsp (bytevector-u32-ref superblock 0 (endianness little))) nl)
|
|
(fmt #t (dsp "calculated checksum: ") (dsp (crc32-region superblock-salt superblock 4 4092)) nl)
|
|
(check-magic superblock))))
|