forked from TrueCloudLab/distribution
74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
package sha512
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto"
|
|
"crypto/rand" // To register the stdlib sha224 and sha256 algs.
|
|
"crypto/sha512"
|
|
"hash"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stevvooe/resumable"
|
|
)
|
|
|
|
func compareResumableHash(t *testing.T, newResumable func() hash.Hash, newStdlib func() hash.Hash) {
|
|
// Read 3 Kilobytes of random data into a buffer.
|
|
buf := make([]byte, 3*1024)
|
|
if _, err := io.ReadFull(rand.Reader, buf); err != nil {
|
|
t.Fatalf("unable to load random data: %s", err)
|
|
}
|
|
|
|
// Use two Hash objects to consume prefixes of the data. One will be
|
|
// snapshotted and resumed with each additional byte, then both will write
|
|
// that byte. The digests should be equal after each byte is digested.
|
|
resumableHasher := newResumable().(resumable.Hash)
|
|
stdlibHasher := newStdlib()
|
|
|
|
// First, assert that the initial distest is the same.
|
|
if !bytes.Equal(resumableHasher.Sum(nil), stdlibHasher.Sum(nil)) {
|
|
t.Fatalf("initial digests do not match: got %x, expected %x", resumableHasher.Sum(nil), stdlibHasher.Sum(nil))
|
|
}
|
|
|
|
multiWriter := io.MultiWriter(resumableHasher, stdlibHasher)
|
|
|
|
for i := 1; i <= len(buf); i++ {
|
|
|
|
// Write the next byte.
|
|
multiWriter.Write(buf[i-1 : i])
|
|
|
|
if !bytes.Equal(resumableHasher.Sum(nil), stdlibHasher.Sum(nil)) {
|
|
t.Fatalf("digests do not match: got %x, expected %x", resumableHasher.Sum(nil), stdlibHasher.Sum(nil))
|
|
}
|
|
|
|
// Snapshot, reset, and restore the chunk hasher.
|
|
hashState, err := resumableHasher.State()
|
|
if err != nil {
|
|
t.Fatalf("unable to get state of hash function: %s", err)
|
|
}
|
|
resumableHasher.Reset()
|
|
if err := resumableHasher.Restore(hashState); err != nil {
|
|
t.Fatalf("unable to restorte state of hash function: %s", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestResumable(t *testing.T) {
|
|
compareResumableHash(t, New384, sha512.New384)
|
|
compareResumableHash(t, New, sha512.New)
|
|
}
|
|
|
|
func TestResumableRegistered(t *testing.T) {
|
|
|
|
for _, hf := range []crypto.Hash{crypto.SHA384, crypto.SHA512} {
|
|
// make sure that the hash gets the resumable version from the global
|
|
// registry in crypto library.
|
|
h := hf.New()
|
|
|
|
if rh, ok := h.(resumable.Hash); !ok {
|
|
t.Fatalf("non-resumable hash function registered: %#v %#v", rh, crypto.SHA256)
|
|
}
|
|
|
|
}
|
|
|
|
}
|