122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package cpio
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
var files = []struct {
|
|
Name, Body string
|
|
}{
|
|
{"./gophers.txt", "Gopher names:\nGeorge\nGeoffrey\nGonzo"},
|
|
{"./readme.txt", "This archive contains some text files."},
|
|
{"./todo.txt", "Get animal handling license."},
|
|
}
|
|
|
|
func TestRead(t *testing.T) {
|
|
f, err := os.Open("testdata/test_svr4_crc.cpio")
|
|
if err != nil {
|
|
t.Fatalf("error opening test file: %v", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
r := NewReader(f)
|
|
for {
|
|
_, err := r.Next()
|
|
if err == io.EOF {
|
|
return
|
|
}
|
|
if err != nil {
|
|
t.Errorf("error moving to next header: %v", err)
|
|
return
|
|
}
|
|
// TODO: validate header fields
|
|
}
|
|
}
|
|
|
|
func TestSVR4CRC(t *testing.T) {
|
|
f, err := os.Open("testdata/test_svr4_crc.cpio")
|
|
if err != nil {
|
|
t.Fatalf("error opening test file: %v", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
w := NewHash()
|
|
r := NewReader(f)
|
|
for {
|
|
hdr, err := r.Next()
|
|
if err != nil {
|
|
if err != io.EOF {
|
|
t.Errorf("error moving to next header: %v", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
if hdr.Mode.IsRegular() {
|
|
w.Reset()
|
|
_, err = io.CopyN(w, r, hdr.Size)
|
|
if err != nil {
|
|
t.Fatalf("error writing to checksum hash: %v", err)
|
|
}
|
|
|
|
sum := Checksum(w.Sum32())
|
|
if sum != hdr.Checksum {
|
|
t.Errorf("expected checksum %v, got %v for %v", hdr.Checksum, sum, hdr.Name)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func ExampleNewHash() {
|
|
// Open the cpio archive for reading.
|
|
f, err := os.Open("testdata/test_svr4_crc.cpio")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
r := NewReader(f)
|
|
|
|
// create a Hash
|
|
h := NewHash()
|
|
|
|
// Iterate through the files in the archive.
|
|
for {
|
|
hdr, err := r.Next()
|
|
if err == io.EOF {
|
|
// end of cpio archive
|
|
return
|
|
}
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// skip symlinks, directories, etc.
|
|
if !hdr.Mode.IsRegular() {
|
|
continue
|
|
}
|
|
|
|
// read file into hash
|
|
h.Reset()
|
|
_, err = io.CopyN(h, r, hdr.Size)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// check hash matches header checksum
|
|
sum := Checksum(h.Sum32())
|
|
if sum == hdr.Checksum {
|
|
fmt.Printf("Checksum OK: %v (%v)\n", hdr.Name, hdr.Checksum)
|
|
} else {
|
|
fmt.Printf("Checksum FAIL: %v - expected %v, got %v\n", hdr.Name, hdr.Checksum, sum)
|
|
}
|
|
}
|
|
|
|
// Output:
|
|
// Checksum OK: gophers.txt (00000C98)
|
|
// Checksum OK: readme.txt (00000E3D)
|
|
// Checksum OK: todo.txt (00000A52)
|
|
}
|