mirror of
https://github.com/octoleo/restic.git
synced 2024-11-26 06:46:34 +00:00
Return hdrSize in ListPack
This commit is contained in:
parent
cac481634c
commit
c3ddde9e7d
@ -749,7 +749,7 @@ func checkPack(ctx context.Context, r restic.Repository, id restic.ID, size int6
|
|||||||
return errors.Errorf("Pack size does not match, want %v, got %v", size, realSize)
|
return errors.Errorf("Pack size does not match, want %v, got %v", size, realSize)
|
||||||
}
|
}
|
||||||
|
|
||||||
blobs, err := pack.List(r.Key(), packfile, size)
|
blobs, _, err := pack.List(r.Key(), packfile, size)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -252,21 +252,24 @@ func (e InvalidFileError) Error() string {
|
|||||||
return e.Message
|
return e.Message
|
||||||
}
|
}
|
||||||
|
|
||||||
// List returns the list of entries found in a pack file.
|
// List returns the list of entries found in a pack file and the length of the
|
||||||
func List(k *crypto.Key, rd io.ReaderAt, size int64) (entries []restic.Blob, err error) {
|
// header (including header size and crypto overhead)
|
||||||
|
func List(k *crypto.Key, rd io.ReaderAt, size int64) (entries []restic.Blob, hdrSize uint32, err error) {
|
||||||
buf, err := readHeader(rd, size)
|
buf, err := readHeader(rd, size)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(buf) < k.NonceSize()+k.Overhead() {
|
if len(buf) < k.NonceSize()+k.Overhead() {
|
||||||
return nil, errors.New("invalid header, too small")
|
return nil, 0, errors.New("invalid header, too small")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
hdrSize = headerLengthSize + uint32(len(buf))
|
||||||
|
|
||||||
nonce, buf := buf[:k.NonceSize()], buf[k.NonceSize():]
|
nonce, buf := buf[:k.NonceSize()], buf[k.NonceSize():]
|
||||||
buf, err = k.Open(buf[:0], nonce, buf, nil)
|
buf, err = k.Open(buf[:0], nonce, buf, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
entries = make([]restic.Blob, 0, uint(len(buf))/entrySize)
|
entries = make([]restic.Blob, 0, uint(len(buf))/entrySize)
|
||||||
@ -275,7 +278,7 @@ func List(k *crypto.Key, rd io.ReaderAt, size int64) (entries []restic.Blob, err
|
|||||||
for len(buf) > 0 {
|
for len(buf) > 0 {
|
||||||
entry, err := parseHeaderEntry(buf)
|
entry, err := parseHeaderEntry(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, 0, err
|
||||||
}
|
}
|
||||||
entry.Offset = pos
|
entry.Offset = pos
|
||||||
|
|
||||||
@ -284,7 +287,7 @@ func List(k *crypto.Key, rd io.ReaderAt, size int64) (entries []restic.Blob, err
|
|||||||
buf = buf[entrySize:]
|
buf = buf[entrySize:]
|
||||||
}
|
}
|
||||||
|
|
||||||
return entries, nil
|
return entries, hdrSize, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// PackedSizeOfBlob returns the size a blob actually uses when saved in a pack
|
// PackedSizeOfBlob returns the size a blob actually uses when saved in a pack
|
||||||
|
@ -53,19 +53,18 @@ func verifyBlobs(t testing.TB, bufs []Buf, k *crypto.Key, rd io.ReaderAt, packSi
|
|||||||
for _, buf := range bufs {
|
for _, buf := range bufs {
|
||||||
written += len(buf.data)
|
written += len(buf.data)
|
||||||
}
|
}
|
||||||
// header length
|
// header length + header + header crypto
|
||||||
written += binary.Size(uint32(0))
|
headerSize := binary.Size(uint32(0)) + restic.CiphertextLength(len(bufs)*int(pack.EntrySize))
|
||||||
// header + header crypto
|
written += headerSize
|
||||||
headerSize := len(bufs) * (binary.Size(restic.BlobType(0)) + binary.Size(uint32(0)) + len(restic.ID{}))
|
|
||||||
written += restic.CiphertextLength(headerSize)
|
|
||||||
|
|
||||||
// check length
|
// check length
|
||||||
rtest.Equals(t, uint(written), packSize)
|
rtest.Equals(t, uint(written), packSize)
|
||||||
|
|
||||||
// read and parse it again
|
// read and parse it again
|
||||||
entries, err := pack.List(k, rd, int64(packSize))
|
entries, hdrSize, err := pack.List(k, rd, int64(packSize))
|
||||||
rtest.OK(t, err)
|
rtest.OK(t, err)
|
||||||
rtest.Equals(t, len(entries), len(bufs))
|
rtest.Equals(t, len(entries), len(bufs))
|
||||||
|
rtest.Equals(t, headerSize, int(hdrSize))
|
||||||
|
|
||||||
var buf []byte
|
var buf []byte
|
||||||
for i, b := range bufs {
|
for i, b := range bufs {
|
||||||
|
@ -92,7 +92,7 @@ func Repack(ctx context.Context, repo restic.Repository, packs restic.IDSet, kee
|
|||||||
for job := range processQueue {
|
for job := range processQueue {
|
||||||
tempfile, packID, packLength := job.tempfile, job.hash, job.packLength
|
tempfile, packID, packLength := job.tempfile, job.hash, job.packLength
|
||||||
|
|
||||||
blobs, err := pack.List(repo.Key(), tempfile, packLength)
|
blobs, _, err := pack.List(repo.Key(), tempfile, packLength)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -740,16 +740,11 @@ func (r *Repository) List(ctx context.Context, t restic.FileType, fn func(restic
|
|||||||
}
|
}
|
||||||
|
|
||||||
// ListPack returns the list of blobs saved in the pack id and the length of
|
// ListPack returns the list of blobs saved in the pack id and the length of
|
||||||
// the file as stored in the backend.
|
// the the pack header.
|
||||||
func (r *Repository) ListPack(ctx context.Context, id restic.ID, size int64) ([]restic.Blob, int64, error) {
|
func (r *Repository) ListPack(ctx context.Context, id restic.ID, size int64) ([]restic.Blob, uint32, error) {
|
||||||
h := restic.Handle{Type: restic.PackFile, Name: id.String()}
|
h := restic.Handle{Type: restic.PackFile, Name: id.String()}
|
||||||
|
|
||||||
blobs, err := pack.List(r.Key(), restic.ReaderAt(ctx, r.Backend(), h), size)
|
return pack.List(r.Key(), restic.ReaderAt(ctx, r.Backend(), h), size)
|
||||||
if err != nil {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return blobs, size, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete calls backend.Delete() if implemented, and returns an error
|
// Delete calls backend.Delete() if implemented, and returns an error
|
||||||
|
@ -32,7 +32,10 @@ type Repository interface {
|
|||||||
//
|
//
|
||||||
// The function fn is called in the same Goroutine List() was called from.
|
// The function fn is called in the same Goroutine List() was called from.
|
||||||
List(ctx context.Context, t FileType, fn func(ID, int64) error) error
|
List(ctx context.Context, t FileType, fn func(ID, int64) error) error
|
||||||
ListPack(context.Context, ID, int64) ([]Blob, int64, error)
|
|
||||||
|
// ListPack returns the list of blobs saved in the pack id and the length of
|
||||||
|
// the the pack header.
|
||||||
|
ListPack(context.Context, ID, int64) ([]Blob, uint32, error)
|
||||||
|
|
||||||
Flush(context.Context) error
|
Flush(context.Context) error
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user