mirror of https://github.com/yeka/zip.git
702 lines
19 KiB
Go
702 lines
19 KiB
Go
// Copyright 2010 The Go Authors. All rights reserved.
|
||
// Use of this source code is governed by a BSD-style
|
||
// license that can be found in the LICENSE file.
|
||
|
||
package zip
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/binary"
|
||
"encoding/hex"
|
||
"io"
|
||
"io/ioutil"
|
||
"os"
|
||
"path/filepath"
|
||
"regexp"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
type ZipTest struct {
|
||
Name string
|
||
Source func() (r io.ReaderAt, size int64) // if non-nil, used instead of testdata/<Name> file
|
||
Comment string
|
||
File []ZipTestFile
|
||
Error error // the error that Opening this file should return
|
||
}
|
||
|
||
type ZipTestFile struct {
|
||
Name string
|
||
Content []byte // if blank, will attempt to compare against File
|
||
ContentErr error
|
||
File string // name of file to compare to (relative to testdata/)
|
||
Mtime string // modified time in format "mm-dd-yy hh:mm:ss"
|
||
Mode os.FileMode
|
||
}
|
||
|
||
// Caution: The Mtime values found for the test files should correspond to
|
||
// the values listed with unzip -l <zipfile>. However, the values
|
||
// listed by unzip appear to be off by some hours. When creating
|
||
// fresh test files and testing them, this issue is not present.
|
||
// The test files were created in Sydney, so there might be a time
|
||
// zone issue. The time zone information does have to be encoded
|
||
// somewhere, because otherwise unzip -l could not provide a different
|
||
// time from what the archive/zip package provides, but there appears
|
||
// to be no documentation about this.
|
||
|
||
var tests = []ZipTest{
|
||
{
|
||
Name: "test.zip",
|
||
Comment: "This is a zipfile comment.",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "test.txt",
|
||
Content: []byte("This is a test text file.\n"),
|
||
Mtime: "09-05-10 12:12:02",
|
||
Mode: 0644,
|
||
},
|
||
{
|
||
Name: "gophercolor16x16.png",
|
||
File: "gophercolor16x16.png",
|
||
Mtime: "09-05-10 15:52:58",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "test-trailing-junk.zip",
|
||
Comment: "This is a zipfile comment.",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "test.txt",
|
||
Content: []byte("This is a test text file.\n"),
|
||
Mtime: "09-05-10 12:12:02",
|
||
Mode: 0644,
|
||
},
|
||
{
|
||
Name: "gophercolor16x16.png",
|
||
File: "gophercolor16x16.png",
|
||
Mtime: "09-05-10 15:52:58",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "r.zip",
|
||
Source: returnRecursiveZip,
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "r/r.zip",
|
||
Content: rZipBytes(),
|
||
Mtime: "03-04-10 00:24:16",
|
||
Mode: 0666,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "symlink.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "symlink",
|
||
Content: []byte("../target"),
|
||
Mode: 0777 | os.ModeSymlink,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "readme.zip",
|
||
},
|
||
{
|
||
Name: "readme.notzip",
|
||
Error: ErrFormat,
|
||
},
|
||
{
|
||
Name: "dd.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "filename",
|
||
Content: []byte("This is a test textfile.\n"),
|
||
Mtime: "02-02-11 13:06:20",
|
||
Mode: 0666,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
// created in windows XP file manager.
|
||
Name: "winxp.zip",
|
||
File: crossPlatform,
|
||
},
|
||
{
|
||
// created by Zip 3.0 under Linux
|
||
Name: "unix.zip",
|
||
File: crossPlatform,
|
||
},
|
||
{
|
||
// created by Go, before we wrote the "optional" data
|
||
// descriptor signatures (which are required by OS X)
|
||
Name: "go-no-datadesc-sig.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "foo.txt",
|
||
Content: []byte("foo\n"),
|
||
Mtime: "03-08-12 16:59:10",
|
||
Mode: 0644,
|
||
},
|
||
{
|
||
Name: "bar.txt",
|
||
Content: []byte("bar\n"),
|
||
Mtime: "03-08-12 16:59:12",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
// created by Go, after we wrote the "optional" data
|
||
// descriptor signatures (which are required by OS X)
|
||
Name: "go-with-datadesc-sig.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "foo.txt",
|
||
Content: []byte("foo\n"),
|
||
Mode: 0666,
|
||
},
|
||
{
|
||
Name: "bar.txt",
|
||
Content: []byte("bar\n"),
|
||
Mode: 0666,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "Bad-CRC32-in-data-descriptor",
|
||
Source: returnCorruptCRC32Zip,
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "foo.txt",
|
||
Content: []byte("foo\n"),
|
||
Mode: 0666,
|
||
ContentErr: ErrChecksum,
|
||
},
|
||
{
|
||
Name: "bar.txt",
|
||
Content: []byte("bar\n"),
|
||
Mode: 0666,
|
||
},
|
||
},
|
||
},
|
||
// Tests that we verify (and accept valid) crc32s on files
|
||
// with crc32s in their file header (not in data descriptors)
|
||
{
|
||
Name: "crc32-not-streamed.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "foo.txt",
|
||
Content: []byte("foo\n"),
|
||
Mtime: "03-08-12 16:59:10",
|
||
Mode: 0644,
|
||
},
|
||
{
|
||
Name: "bar.txt",
|
||
Content: []byte("bar\n"),
|
||
Mtime: "03-08-12 16:59:12",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
// Tests that we verify (and reject invalid) crc32s on files
|
||
// with crc32s in their file header (not in data descriptors)
|
||
{
|
||
Name: "crc32-not-streamed.zip",
|
||
Source: returnCorruptNotStreamedZip,
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "foo.txt",
|
||
Content: []byte("foo\n"),
|
||
Mtime: "03-08-12 16:59:10",
|
||
Mode: 0644,
|
||
ContentErr: ErrChecksum,
|
||
},
|
||
{
|
||
Name: "bar.txt",
|
||
Content: []byte("bar\n"),
|
||
Mtime: "03-08-12 16:59:12",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "zip64.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "README",
|
||
Content: []byte("This small file is in ZIP64 format.\n"),
|
||
Mtime: "08-10-12 14:33:32",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
// Another zip64 file with different Extras fields. (golang.org/issue/7069)
|
||
{
|
||
Name: "zip64-2.zip",
|
||
File: []ZipTestFile{
|
||
{
|
||
Name: "README",
|
||
Content: []byte("This small file is in ZIP64 format.\n"),
|
||
Mtime: "08-10-12 14:33:32",
|
||
Mode: 0644,
|
||
},
|
||
},
|
||
},
|
||
}
|
||
|
||
var crossPlatform = []ZipTestFile{
|
||
{
|
||
Name: "hello",
|
||
Content: []byte("world \r\n"),
|
||
Mode: 0666,
|
||
},
|
||
{
|
||
Name: "dir/bar",
|
||
Content: []byte("foo \r\n"),
|
||
Mode: 0666,
|
||
},
|
||
{
|
||
Name: "dir/empty/",
|
||
Content: []byte{},
|
||
Mode: os.ModeDir | 0777,
|
||
},
|
||
{
|
||
Name: "readonly",
|
||
Content: []byte("important \r\n"),
|
||
Mode: 0444,
|
||
},
|
||
}
|
||
|
||
func TestReader(t *testing.T) {
|
||
for _, zt := range tests {
|
||
readTestZip(t, zt)
|
||
}
|
||
}
|
||
|
||
func readTestZip(t *testing.T, zt ZipTest) {
|
||
var z *Reader
|
||
var err error
|
||
if zt.Source != nil {
|
||
rat, size := zt.Source()
|
||
z, err = NewReader(rat, size)
|
||
} else {
|
||
var rc *ReadCloser
|
||
rc, err = OpenReader(filepath.Join("testdata", zt.Name))
|
||
if err == nil {
|
||
defer rc.Close()
|
||
z = &rc.Reader
|
||
}
|
||
}
|
||
if err != zt.Error {
|
||
t.Errorf("%s: error=%v, want %v", zt.Name, err, zt.Error)
|
||
return
|
||
}
|
||
|
||
// bail if file is not zip
|
||
if err == ErrFormat {
|
||
return
|
||
}
|
||
|
||
// bail here if no Files expected to be tested
|
||
// (there may actually be files in the zip, but we don't care)
|
||
if zt.File == nil {
|
||
return
|
||
}
|
||
|
||
if z.Comment != zt.Comment {
|
||
t.Errorf("%s: comment=%q, want %q", zt.Name, z.Comment, zt.Comment)
|
||
}
|
||
if len(z.File) != len(zt.File) {
|
||
t.Fatalf("%s: file count=%d, want %d", zt.Name, len(z.File), len(zt.File))
|
||
}
|
||
|
||
// test read of each file
|
||
for i, ft := range zt.File {
|
||
readTestFile(t, zt, ft, z.File[i])
|
||
}
|
||
|
||
// test simultaneous reads
|
||
n := 0
|
||
done := make(chan bool)
|
||
for i := 0; i < 5; i++ {
|
||
for j, ft := range zt.File {
|
||
go func(j int, ft ZipTestFile) {
|
||
readTestFile(t, zt, ft, z.File[j])
|
||
done <- true
|
||
}(j, ft)
|
||
n++
|
||
}
|
||
}
|
||
for ; n > 0; n-- {
|
||
<-done
|
||
}
|
||
}
|
||
|
||
func readTestFile(t *testing.T, zt ZipTest, ft ZipTestFile, f *File) {
|
||
if f.Name != ft.Name {
|
||
t.Errorf("%s: name=%q, want %q", zt.Name, f.Name, ft.Name)
|
||
}
|
||
|
||
if ft.Mtime != "" {
|
||
mtime, err := time.Parse("01-02-06 15:04:05", ft.Mtime)
|
||
if err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
if ft := f.ModTime(); !ft.Equal(mtime) {
|
||
t.Errorf("%s: %s: mtime=%s, want %s", zt.Name, f.Name, ft, mtime)
|
||
}
|
||
}
|
||
|
||
testFileMode(t, zt.Name, f, ft.Mode)
|
||
|
||
var b bytes.Buffer
|
||
r, err := f.Open()
|
||
if err != nil {
|
||
t.Errorf("%s: %v", zt.Name, err)
|
||
return
|
||
}
|
||
|
||
_, err = io.Copy(&b, r)
|
||
if err != ft.ContentErr {
|
||
t.Errorf("%s: copying contents: %v (want %v)", zt.Name, err, ft.ContentErr)
|
||
}
|
||
if err != nil {
|
||
return
|
||
}
|
||
r.Close()
|
||
|
||
size := uint64(f.UncompressedSize)
|
||
if size == uint32max {
|
||
size = f.UncompressedSize64
|
||
}
|
||
if g := uint64(b.Len()); g != size {
|
||
t.Errorf("%v: read %v bytes but f.UncompressedSize == %v", f.Name, g, size)
|
||
}
|
||
|
||
var c []byte
|
||
if ft.Content != nil {
|
||
c = ft.Content
|
||
} else if c, err = ioutil.ReadFile("testdata/" + ft.File); err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
|
||
if b.Len() != len(c) {
|
||
t.Errorf("%s: len=%d, want %d", f.Name, b.Len(), len(c))
|
||
return
|
||
}
|
||
|
||
for i, b := range b.Bytes() {
|
||
if b != c[i] {
|
||
t.Errorf("%s: content[%d]=%q want %q", f.Name, i, b, c[i])
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func testFileMode(t *testing.T, zipName string, f *File, want os.FileMode) {
|
||
mode := f.Mode()
|
||
if want == 0 {
|
||
t.Errorf("%s: %s mode: got %v, want none", zipName, f.Name, mode)
|
||
} else if mode != want {
|
||
t.Errorf("%s: %s mode: want %v, got %v", zipName, f.Name, want, mode)
|
||
}
|
||
}
|
||
|
||
func TestInvalidFiles(t *testing.T) {
|
||
const size = 1024 * 70 // 70kb
|
||
b := make([]byte, size)
|
||
|
||
// zeroes
|
||
_, err := NewReader(bytes.NewReader(b), size)
|
||
if err != ErrFormat {
|
||
t.Errorf("zeroes: error=%v, want %v", err, ErrFormat)
|
||
}
|
||
|
||
// repeated directoryEndSignatures
|
||
sig := make([]byte, 4)
|
||
binary.LittleEndian.PutUint32(sig, directoryEndSignature)
|
||
for i := 0; i < size-4; i += 4 {
|
||
copy(b[i:i+4], sig)
|
||
}
|
||
_, err = NewReader(bytes.NewReader(b), size)
|
||
if err != ErrFormat {
|
||
t.Errorf("sigs: error=%v, want %v", err, ErrFormat)
|
||
}
|
||
}
|
||
|
||
func messWith(fileName string, corrupter func(b []byte)) (r io.ReaderAt, size int64) {
|
||
data, err := ioutil.ReadFile(filepath.Join("testdata", fileName))
|
||
if err != nil {
|
||
panic("Error reading " + fileName + ": " + err.Error())
|
||
}
|
||
corrupter(data)
|
||
return bytes.NewReader(data), int64(len(data))
|
||
}
|
||
|
||
func returnCorruptCRC32Zip() (r io.ReaderAt, size int64) {
|
||
return messWith("go-with-datadesc-sig.zip", func(b []byte) {
|
||
// Corrupt one of the CRC32s in the data descriptor:
|
||
b[0x2d]++
|
||
})
|
||
}
|
||
|
||
func returnCorruptNotStreamedZip() (r io.ReaderAt, size int64) {
|
||
return messWith("crc32-not-streamed.zip", func(b []byte) {
|
||
// Corrupt foo.txt's final crc32 byte, in both
|
||
// the file header and TOC. (0x7e -> 0x7f)
|
||
b[0x11]++
|
||
b[0x9d]++
|
||
|
||
// TODO(bradfitz): add a new test that only corrupts
|
||
// one of these values, and verify that that's also an
|
||
// error. Currently, the reader code doesn't verify the
|
||
// fileheader and TOC's crc32 match if they're both
|
||
// non-zero and only the second line above, the TOC,
|
||
// is what matters.
|
||
})
|
||
}
|
||
|
||
// rZipBytes returns the bytes of a recursive zip file, without
|
||
// putting it on disk and triggering certain virus scanners.
|
||
func rZipBytes() []byte {
|
||
s := `
|
||
0000000 50 4b 03 04 14 00 00 00 08 00 08 03 64 3c f9 f4
|
||
0000010 89 64 48 01 00 00 b8 01 00 00 07 00 00 00 72 2f
|
||
0000020 72 2e 7a 69 70 00 25 00 da ff 50 4b 03 04 14 00
|
||
0000030 00 00 08 00 08 03 64 3c f9 f4 89 64 48 01 00 00
|
||
0000040 b8 01 00 00 07 00 00 00 72 2f 72 2e 7a 69 70 00
|
||
0000050 2f 00 d0 ff 00 25 00 da ff 50 4b 03 04 14 00 00
|
||
0000060 00 08 00 08 03 64 3c f9 f4 89 64 48 01 00 00 b8
|
||
0000070 01 00 00 07 00 00 00 72 2f 72 2e 7a 69 70 00 2f
|
||
0000080 00 d0 ff c2 54 8e 57 39 00 05 00 fa ff c2 54 8e
|
||
0000090 57 39 00 05 00 fa ff 00 05 00 fa ff 00 14 00 eb
|
||
00000a0 ff c2 54 8e 57 39 00 05 00 fa ff 00 05 00 fa ff
|
||
00000b0 00 14 00 eb ff 42 88 21 c4 00 00 14 00 eb ff 42
|
||
00000c0 88 21 c4 00 00 14 00 eb ff 42 88 21 c4 00 00 14
|
||
00000d0 00 eb ff 42 88 21 c4 00 00 14 00 eb ff 42 88 21
|
||
00000e0 c4 00 00 00 00 ff ff 00 00 00 ff ff 00 34 00 cb
|
||
00000f0 ff 42 88 21 c4 00 00 00 00 ff ff 00 00 00 ff ff
|
||
0000100 00 34 00 cb ff 42 e8 21 5e 0f 00 00 00 ff ff 0a
|
||
0000110 f0 66 64 12 61 c0 15 dc e8 a0 48 bf 48 af 2a b3
|
||
0000120 20 c0 9b 95 0d c4 67 04 42 53 06 06 06 40 00 06
|
||
0000130 00 f9 ff 6d 01 00 00 00 00 42 e8 21 5e 0f 00 00
|
||
0000140 00 ff ff 0a f0 66 64 12 61 c0 15 dc e8 a0 48 bf
|
||
0000150 48 af 2a b3 20 c0 9b 95 0d c4 67 04 42 53 06 06
|
||
0000160 06 40 00 06 00 f9 ff 6d 01 00 00 00 00 50 4b 01
|
||
0000170 02 14 00 14 00 00 00 08 00 08 03 64 3c f9 f4 89
|
||
0000180 64 48 01 00 00 b8 01 00 00 07 00 00 00 00 00 00
|
||
0000190 00 00 00 00 00 00 00 00 00 00 00 72 2f 72 2e 7a
|
||
00001a0 69 70 50 4b 05 06 00 00 00 00 01 00 01 00 35 00
|
||
00001b0 00 00 6d 01 00 00 00 00`
|
||
s = regexp.MustCompile(`[0-9a-f]{7}`).ReplaceAllString(s, "")
|
||
s = regexp.MustCompile(`\s+`).ReplaceAllString(s, "")
|
||
b, err := hex.DecodeString(s)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return b
|
||
}
|
||
|
||
func returnRecursiveZip() (r io.ReaderAt, size int64) {
|
||
b := rZipBytes()
|
||
return bytes.NewReader(b), int64(len(b))
|
||
}
|
||
|
||
func TestIssue8186(t *testing.T) {
|
||
// Directory headers & data found in the TOC of a JAR file.
|
||
dirEnts := []string{
|
||
"PK\x01\x02\n\x00\n\x00\x00\b\x00\x004\x9d3?\xaa\x1b\x06\xf0\x81\x02\x00\x00\x81\x02\x00\x00-\x00\x05\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00res/drawable-xhdpi-v4/ic_actionbar_accept.png\xfe\xca\x00\x00\x00",
|
||
"PK\x01\x02\n\x00\n\x00\x00\b\x00\x004\x9d3?\x90K\x89\xc7t\n\x00\x00t\n\x00\x00\x0e\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xd1\x02\x00\x00resources.arsc\x00\x00\x00",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?\xff$\x18\xed3\x03\x00\x00\xb4\b\x00\x00\x13\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00t\r\x00\x00AndroidManifest.xml",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?\x14\xc5K\xab\x192\x02\x00\xc8\xcd\x04\x00\v\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xe8\x10\x00\x00classes.dex",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?E\x96\nD\xac\x01\x00\x00P\x03\x00\x00&\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00:C\x02\x00res/layout/actionbar_set_wallpaper.xml",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?Ļ\x14\xe3\xd8\x01\x00\x00\xd8\x03\x00\x00 \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00:E\x02\x00res/layout/wallpaper_cropper.xml",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?}\xc1\x15\x9eZ\x01\x00\x00!\x02\x00\x00\x14\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00`G\x02\x00META-INF/MANIFEST.MF",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?\xe6\x98Ьo\x01\x00\x00\x84\x02\x00\x00\x10\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xfcH\x02\x00META-INF/CERT.SF",
|
||
"PK\x01\x02\x14\x00\x14\x00\b\b\b\x004\x9d3?\xbfP\x96b\x86\x04\x00\x00\xb2\x06\x00\x00\x11\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xa9J\x02\x00META-INF/CERT.RSA",
|
||
}
|
||
for i, s := range dirEnts {
|
||
var f File
|
||
err := readDirectoryHeader(&f, strings.NewReader(s))
|
||
if err != nil {
|
||
t.Errorf("error reading #%d: %v", i, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// Verify we return ErrUnexpectedEOF when length is short.
|
||
func TestIssue10957(t *testing.T) {
|
||
data := []byte("PK\x03\x040000000PK\x01\x0200000" +
|
||
"0000000000000000000\x00" +
|
||
"\x00\x00\x00\x00\x00000000000000PK\x01" +
|
||
"\x020000000000000000000" +
|
||
"00000\v\x00\x00\x00\x00\x00000000000" +
|
||
"00000000000000PK\x01\x0200" +
|
||
"00000000000000000000" +
|
||
"00\v\x00\x00\x00\x00\x00000000000000" +
|
||
"00000000000PK\x01\x020000<" +
|
||
"0\x00\x0000000000000000\v\x00\v" +
|
||
"\x00\x00\x00\x00\x0000000000\x00\x00\x00\x00000" +
|
||
"00000000PK\x01\x0200000000" +
|
||
"0000000000000000\v\x00\x00\x00" +
|
||
"\x00\x0000PK\x05\x06000000\x05\x000000" +
|
||
"\v\x00\x00\x00\x00\x00")
|
||
z, err := NewReader(bytes.NewReader(data), int64(len(data)))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for i, f := range z.File {
|
||
r, err := f.Open()
|
||
if err != nil {
|
||
continue
|
||
}
|
||
if f.UncompressedSize64 < 1e6 {
|
||
n, err := io.Copy(ioutil.Discard, r)
|
||
if i == 3 && err != io.ErrUnexpectedEOF {
|
||
t.Errorf("File[3] error = %v; want io.ErrUnexpectedEOF", err)
|
||
}
|
||
if err == nil && uint64(n) != f.UncompressedSize64 {
|
||
t.Errorf("file %d: bad size: copied=%d; want=%d", i, n, f.UncompressedSize64)
|
||
}
|
||
}
|
||
r.Close()
|
||
}
|
||
}
|
||
|
||
// Verify the number of files is sane.
|
||
func TestIssue10956(t *testing.T) {
|
||
data := []byte("PK\x06\x06PK\x06\a0000\x00\x00\x00\x00\x00\x00\x00\x00" +
|
||
"0000PK\x05\x06000000000000" +
|
||
"0000\v\x00000\x00\x00\x00\x00\x00\x00\x000")
|
||
_, err := NewReader(bytes.NewReader(data), int64(len(data)))
|
||
const want = "TOC declares impossible 3472328296227680304 files in 57 byte"
|
||
if err == nil && !strings.Contains(err.Error(), want) {
|
||
t.Errorf("error = %v; want %q", err, want)
|
||
}
|
||
}
|
||
|
||
// Verify we return ErrUnexpectedEOF when reading truncated data descriptor.
|
||
func TestIssue11146(t *testing.T) {
|
||
data := []byte("PK\x03\x040000000000000000" +
|
||
"000000\x01\x00\x00\x000\x01\x00\x00\xff\xff0000" +
|
||
"0000000000000000PK\x01\x02" +
|
||
"0000\b0\b\x00000000000000" +
|
||
"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x000000PK\x05\x06\x00\x00" +
|
||
"\x00\x0000\x01\x0000008\x00\x00\x00\x00\x00")
|
||
z, err := NewReader(bytes.NewReader(data), int64(len(data)))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
r, err := z.File[0].Open()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
_, err = ioutil.ReadAll(r)
|
||
if err != io.ErrUnexpectedEOF {
|
||
t.Errorf("File[0] error = %v; want io.ErrUnexpectedEOF", err)
|
||
}
|
||
r.Close()
|
||
}
|
||
|
||
func TestSimplePassword(t *testing.T) {
|
||
file := "hello-aes.zip"
|
||
var buf bytes.Buffer
|
||
r, err := OpenReader(filepath.Join("testdata", file))
|
||
if err != nil {
|
||
t.Errorf("Expected %s to open: %v.", file, err)
|
||
}
|
||
defer r.Close()
|
||
if len(r.File) != 1 {
|
||
t.Errorf("Expected %s to contain one file.", file)
|
||
}
|
||
f := r.File[0]
|
||
if f.FileInfo().Name() != "hello.txt" {
|
||
t.Errorf("Expected %s to have a file named hello.txt", file)
|
||
}
|
||
if f.Method != 0 {
|
||
t.Errorf("Expected %s to have its Method set to 0.", file)
|
||
}
|
||
f.SetPassword([]byte("golang"))
|
||
rc, err := f.Open()
|
||
if err != nil {
|
||
t.Errorf("Expected to open the readcloser: %v.", err)
|
||
}
|
||
_, err = io.Copy(&buf, rc)
|
||
if err != nil {
|
||
t.Errorf("Expected to copy bytes: %v.", err)
|
||
}
|
||
if !bytes.Contains(buf.Bytes(), []byte("Hello World\r\n")) {
|
||
t.Errorf("Expected contents were not found.")
|
||
}
|
||
}
|
||
|
||
func TestHelloWorldAes(t *testing.T) {
|
||
file := "world-aes.zip"
|
||
expecting := "helloworld"
|
||
r, err := OpenReader(filepath.Join("testdata", file))
|
||
if err != nil {
|
||
t.Errorf("Expected %s to open: %v", file, err)
|
||
}
|
||
defer r.Close()
|
||
if len(r.File) != 2 {
|
||
t.Errorf("Expected %s to contain two files.", file)
|
||
}
|
||
var b bytes.Buffer
|
||
for _, f := range r.File {
|
||
if !f.IsEncrypted() {
|
||
t.Errorf("Expected %s to be encrypted.", f.FileInfo().Name)
|
||
}
|
||
f.SetPassword([]byte("golang"))
|
||
rc, err := f.Open()
|
||
if err != nil {
|
||
t.Errorf("Expected to open readcloser: %v", err)
|
||
}
|
||
defer rc.Close()
|
||
if _, err := io.Copy(&b, rc); err != nil {
|
||
t.Errorf("Expected to copy bytes to buffer: %v", err)
|
||
}
|
||
}
|
||
if !bytes.Equal([]byte(expecting), b.Bytes()) {
|
||
t.Errorf("Expected ending content to be %s instead of %s", expecting, b.Bytes())
|
||
}
|
||
}
|
||
|
||
func TestMacbethAct1(t *testing.T) {
|
||
file := "macbeth-act1.zip"
|
||
expecting := "Exeunt"
|
||
var b bytes.Buffer
|
||
r, err := OpenReader(filepath.Join("testdata", file))
|
||
if err != nil {
|
||
t.Errorf("Expected %s to open: %v", file, err)
|
||
}
|
||
defer r.Close()
|
||
for _, f := range r.File {
|
||
if !f.IsEncrypted() {
|
||
t.Errorf("Expected %s to be encrypted.", f.Name)
|
||
}
|
||
f.SetPassword([]byte("golang"))
|
||
rc, err := f.Open()
|
||
if err != nil {
|
||
t.Errorf("Expected to open readcloser: %v", err)
|
||
}
|
||
defer rc.Close()
|
||
if _, err := io.Copy(&b, rc); err != nil {
|
||
t.Errorf("Expected to copy bytes to buffer: %v", err)
|
||
}
|
||
}
|
||
if !bytes.Contains(b.Bytes(), []byte(expecting)) {
|
||
t.Errorf("Expected to find %s in the buffer %v", expecting, b.Bytes())
|
||
}
|
||
}
|
||
|
||
// Test for AE-1 vs AE-2
|
||
// Test for tampered data payload, use messWith
|