1043 lines
23 KiB
Go
1043 lines
23 KiB
Go
// Copyright 2015 Google Inc. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package zip
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/flate"
|
|
"errors"
|
|
"fmt"
|
|
"hash/crc32"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
"unicode"
|
|
|
|
"github.com/google/blueprint/pathtools"
|
|
|
|
"android/soong/jar"
|
|
"android/soong/third_party/zip"
|
|
)
|
|
|
|
// Block size used during parallel compression of a single file.
|
|
const parallelBlockSize = 1 * 1024 * 1024 // 1MB
|
|
|
|
// Minimum file size to use parallel compression. It requires more
|
|
// flate.Writer allocations, since we can't change the dictionary
|
|
// during Reset
|
|
const minParallelFileSize = parallelBlockSize * 6
|
|
|
|
// Size of the ZIP compression window (32KB)
|
|
const windowSize = 32 * 1024
|
|
|
|
type nopCloser struct {
|
|
io.Writer
|
|
}
|
|
|
|
func (nopCloser) Close() error {
|
|
return nil
|
|
}
|
|
|
|
type byteReaderCloser struct {
|
|
*bytes.Reader
|
|
io.Closer
|
|
}
|
|
|
|
type pathMapping struct {
|
|
dest, src string
|
|
zipMethod uint16
|
|
}
|
|
|
|
type FileArg struct {
|
|
PathPrefixInZip, SourcePrefixToStrip string
|
|
SourceFiles []string
|
|
JunkPaths bool
|
|
GlobDir string
|
|
}
|
|
|
|
type FileArgsBuilder struct {
|
|
state FileArg
|
|
err error
|
|
fs pathtools.FileSystem
|
|
|
|
fileArgs []FileArg
|
|
}
|
|
|
|
func NewFileArgsBuilder() *FileArgsBuilder {
|
|
return &FileArgsBuilder{
|
|
fs: pathtools.OsFs,
|
|
}
|
|
}
|
|
|
|
func (b *FileArgsBuilder) JunkPaths(v bool) *FileArgsBuilder {
|
|
b.state.JunkPaths = v
|
|
b.state.SourcePrefixToStrip = ""
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) SourcePrefixToStrip(prefixToStrip string) *FileArgsBuilder {
|
|
b.state.JunkPaths = false
|
|
b.state.SourcePrefixToStrip = prefixToStrip
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) PathPrefixInZip(rootPrefix string) *FileArgsBuilder {
|
|
b.state.PathPrefixInZip = rootPrefix
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) File(name string) *FileArgsBuilder {
|
|
if b.err != nil {
|
|
return b
|
|
}
|
|
|
|
arg := b.state
|
|
arg.SourceFiles = []string{name}
|
|
b.fileArgs = append(b.fileArgs, arg)
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) Dir(name string) *FileArgsBuilder {
|
|
if b.err != nil {
|
|
return b
|
|
}
|
|
|
|
arg := b.state
|
|
arg.GlobDir = name
|
|
b.fileArgs = append(b.fileArgs, arg)
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) List(name string) *FileArgsBuilder {
|
|
if b.err != nil {
|
|
return b
|
|
}
|
|
|
|
f, err := b.fs.Open(name)
|
|
if err != nil {
|
|
b.err = err
|
|
return b
|
|
}
|
|
defer f.Close()
|
|
|
|
list, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
b.err = err
|
|
return b
|
|
}
|
|
|
|
arg := b.state
|
|
arg.SourceFiles = strings.Fields(string(list))
|
|
b.fileArgs = append(b.fileArgs, arg)
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) RspFile(name string) *FileArgsBuilder {
|
|
if b.err != nil {
|
|
return b
|
|
}
|
|
|
|
f, err := b.fs.Open(name)
|
|
if err != nil {
|
|
b.err = err
|
|
return b
|
|
}
|
|
defer f.Close()
|
|
|
|
list, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
b.err = err
|
|
return b
|
|
}
|
|
|
|
arg := b.state
|
|
arg.SourceFiles = ReadRespFile(list)
|
|
b.fileArgs = append(b.fileArgs, arg)
|
|
return b
|
|
}
|
|
|
|
func (b *FileArgsBuilder) Error() error {
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
return b.err
|
|
}
|
|
|
|
func (b *FileArgsBuilder) FileArgs() []FileArg {
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
return b.fileArgs
|
|
}
|
|
|
|
type IncorrectRelativeRootError struct {
|
|
RelativeRoot string
|
|
Path string
|
|
}
|
|
|
|
func (x IncorrectRelativeRootError) Error() string {
|
|
return fmt.Sprintf("path %q is outside relative root %q", x.Path, x.RelativeRoot)
|
|
}
|
|
|
|
type ZipWriter struct {
|
|
time time.Time
|
|
createdFiles map[string]string
|
|
createdDirs map[string]string
|
|
directories bool
|
|
|
|
errors chan error
|
|
writeOps chan chan *zipEntry
|
|
|
|
cpuRateLimiter *CPURateLimiter
|
|
memoryRateLimiter *MemoryRateLimiter
|
|
|
|
compressorPool sync.Pool
|
|
compLevel int
|
|
|
|
followSymlinks pathtools.ShouldFollowSymlinks
|
|
ignoreMissingFiles bool
|
|
|
|
stderr io.Writer
|
|
fs pathtools.FileSystem
|
|
}
|
|
|
|
type zipEntry struct {
|
|
fh *zip.FileHeader
|
|
|
|
// List of delayed io.Reader
|
|
futureReaders chan chan io.Reader
|
|
|
|
// Only used for passing into the MemoryRateLimiter to ensure we
|
|
// release as much memory as much as we request
|
|
allocatedSize int64
|
|
}
|
|
|
|
type ZipArgs struct {
|
|
FileArgs []FileArg
|
|
OutputFilePath string
|
|
EmulateJar bool
|
|
SrcJar bool
|
|
AddDirectoryEntriesToZip bool
|
|
CompressionLevel int
|
|
ManifestSourcePath string
|
|
NumParallelJobs int
|
|
NonDeflatedFiles map[string]bool
|
|
WriteIfChanged bool
|
|
StoreSymlinks bool
|
|
IgnoreMissingFiles bool
|
|
|
|
Stderr io.Writer
|
|
Filesystem pathtools.FileSystem
|
|
}
|
|
|
|
const NOQUOTE = '\x00'
|
|
|
|
func ReadRespFile(bytes []byte) []string {
|
|
var args []string
|
|
var arg []rune
|
|
|
|
isEscaping := false
|
|
quotingStart := NOQUOTE
|
|
for _, c := range string(bytes) {
|
|
switch {
|
|
case isEscaping:
|
|
if quotingStart == '"' {
|
|
if !(c == '"' || c == '\\') {
|
|
// '\"' or '\\' will be escaped under double quoting.
|
|
arg = append(arg, '\\')
|
|
}
|
|
}
|
|
arg = append(arg, c)
|
|
isEscaping = false
|
|
case c == '\\' && quotingStart != '\'':
|
|
isEscaping = true
|
|
case quotingStart == NOQUOTE && (c == '\'' || c == '"'):
|
|
quotingStart = c
|
|
case quotingStart != NOQUOTE && c == quotingStart:
|
|
quotingStart = NOQUOTE
|
|
case quotingStart == NOQUOTE && unicode.IsSpace(c):
|
|
// Current character is a space outside quotes
|
|
if len(arg) != 0 {
|
|
args = append(args, string(arg))
|
|
}
|
|
arg = arg[:0]
|
|
default:
|
|
arg = append(arg, c)
|
|
}
|
|
}
|
|
|
|
if len(arg) != 0 {
|
|
args = append(args, string(arg))
|
|
}
|
|
|
|
return args
|
|
}
|
|
|
|
func ZipTo(args ZipArgs, w io.Writer) error {
|
|
if args.EmulateJar {
|
|
args.AddDirectoryEntriesToZip = true
|
|
}
|
|
|
|
// Have Glob follow symlinks if they are not being stored as symlinks in the zip file.
|
|
followSymlinks := pathtools.ShouldFollowSymlinks(!args.StoreSymlinks)
|
|
|
|
z := &ZipWriter{
|
|
time: jar.DefaultTime,
|
|
createdDirs: make(map[string]string),
|
|
createdFiles: make(map[string]string),
|
|
directories: args.AddDirectoryEntriesToZip,
|
|
compLevel: args.CompressionLevel,
|
|
followSymlinks: followSymlinks,
|
|
ignoreMissingFiles: args.IgnoreMissingFiles,
|
|
stderr: args.Stderr,
|
|
fs: args.Filesystem,
|
|
}
|
|
|
|
if z.fs == nil {
|
|
z.fs = pathtools.OsFs
|
|
}
|
|
|
|
if z.stderr == nil {
|
|
z.stderr = os.Stderr
|
|
}
|
|
|
|
pathMappings := []pathMapping{}
|
|
|
|
noCompression := args.CompressionLevel == 0
|
|
|
|
for _, fa := range args.FileArgs {
|
|
var srcs []string
|
|
for _, s := range fa.SourceFiles {
|
|
s = strings.TrimSpace(s)
|
|
if s == "" {
|
|
continue
|
|
}
|
|
|
|
globbed, _, err := z.fs.Glob(s, nil, followSymlinks)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(globbed) == 0 {
|
|
err := &os.PathError{
|
|
Op: "lstat",
|
|
Path: s,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
if args.IgnoreMissingFiles {
|
|
fmt.Fprintln(z.stderr, "warning:", err)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
srcs = append(srcs, globbed...)
|
|
}
|
|
if fa.GlobDir != "" {
|
|
if exists, isDir, err := z.fs.Exists(fa.GlobDir); err != nil {
|
|
return err
|
|
} else if !exists && !args.IgnoreMissingFiles {
|
|
err := &os.PathError{
|
|
Op: "lstat",
|
|
Path: fa.GlobDir,
|
|
Err: os.ErrNotExist,
|
|
}
|
|
if args.IgnoreMissingFiles {
|
|
fmt.Fprintln(z.stderr, "warning:", err)
|
|
} else {
|
|
return err
|
|
}
|
|
} else if !isDir && !args.IgnoreMissingFiles {
|
|
err := &os.PathError{
|
|
Op: "lstat",
|
|
Path: fa.GlobDir,
|
|
Err: syscall.ENOTDIR,
|
|
}
|
|
if args.IgnoreMissingFiles {
|
|
fmt.Fprintln(z.stderr, "warning:", err)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
globbed, _, err := z.fs.Glob(filepath.Join(fa.GlobDir, "**/*"), nil, followSymlinks)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
srcs = append(srcs, globbed...)
|
|
}
|
|
for _, src := range srcs {
|
|
err := fillPathPairs(fa, src, &pathMappings, args.NonDeflatedFiles, noCompression)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return z.write(w, pathMappings, args.ManifestSourcePath, args.EmulateJar, args.SrcJar, args.NumParallelJobs)
|
|
}
|
|
|
|
func Zip(args ZipArgs) error {
|
|
if args.OutputFilePath == "" {
|
|
return fmt.Errorf("output file path must be nonempty")
|
|
}
|
|
|
|
buf := &bytes.Buffer{}
|
|
var out io.Writer = buf
|
|
|
|
if !args.WriteIfChanged {
|
|
f, err := os.Create(args.OutputFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer f.Close()
|
|
defer func() {
|
|
if err != nil {
|
|
os.Remove(args.OutputFilePath)
|
|
}
|
|
}()
|
|
|
|
out = f
|
|
}
|
|
|
|
err := ZipTo(args, out)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if args.WriteIfChanged {
|
|
err := pathtools.WriteFileIfChanged(args.OutputFilePath, buf.Bytes(), 0666)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func fillPathPairs(fa FileArg, src string, pathMappings *[]pathMapping,
|
|
nonDeflatedFiles map[string]bool, noCompression bool) error {
|
|
|
|
var dest string
|
|
|
|
if fa.JunkPaths {
|
|
dest = filepath.Base(src)
|
|
} else {
|
|
var err error
|
|
dest, err = filepath.Rel(fa.SourcePrefixToStrip, src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if strings.HasPrefix(dest, "../") {
|
|
return IncorrectRelativeRootError{
|
|
Path: src,
|
|
RelativeRoot: fa.SourcePrefixToStrip,
|
|
}
|
|
}
|
|
|
|
}
|
|
dest = filepath.Join(fa.PathPrefixInZip, dest)
|
|
|
|
zipMethod := zip.Deflate
|
|
if _, found := nonDeflatedFiles[dest]; found || noCompression {
|
|
zipMethod = zip.Store
|
|
}
|
|
*pathMappings = append(*pathMappings,
|
|
pathMapping{dest: dest, src: src, zipMethod: zipMethod})
|
|
|
|
return nil
|
|
}
|
|
|
|
func jarSort(mappings []pathMapping) {
|
|
less := func(i int, j int) (smaller bool) {
|
|
return jar.EntryNamesLess(mappings[i].dest, mappings[j].dest)
|
|
}
|
|
sort.SliceStable(mappings, less)
|
|
}
|
|
|
|
func (z *ZipWriter) write(f io.Writer, pathMappings []pathMapping, manifest string, emulateJar, srcJar bool,
|
|
parallelJobs int) error {
|
|
|
|
z.errors = make(chan error)
|
|
defer close(z.errors)
|
|
|
|
// This channel size can be essentially unlimited -- it's used as a fifo
|
|
// queue decouple the CPU and IO loads. Directories don't require any
|
|
// compression time, but still cost some IO. Similar with small files that
|
|
// can be very fast to compress. Some files that are more difficult to
|
|
// compress won't take a corresponding longer time writing out.
|
|
//
|
|
// The optimum size here depends on your CPU and IO characteristics, and
|
|
// the the layout of your zip file. 1000 was chosen mostly at random as
|
|
// something that worked reasonably well for a test file.
|
|
//
|
|
// The RateLimit object will put the upper bounds on the number of
|
|
// parallel compressions and outstanding buffers.
|
|
z.writeOps = make(chan chan *zipEntry, 1000)
|
|
z.cpuRateLimiter = NewCPURateLimiter(int64(parallelJobs))
|
|
z.memoryRateLimiter = NewMemoryRateLimiter(0)
|
|
defer func() {
|
|
z.cpuRateLimiter.Stop()
|
|
z.memoryRateLimiter.Stop()
|
|
}()
|
|
|
|
if manifest != "" && !emulateJar {
|
|
return errors.New("must specify --jar when specifying a manifest via -m")
|
|
}
|
|
|
|
if emulateJar {
|
|
// manifest may be empty, in which case addManifest will fill in a default
|
|
pathMappings = append(pathMappings, pathMapping{jar.ManifestFile, manifest, zip.Deflate})
|
|
|
|
jarSort(pathMappings)
|
|
}
|
|
|
|
go func() {
|
|
var err error
|
|
defer close(z.writeOps)
|
|
|
|
for _, ele := range pathMappings {
|
|
if emulateJar && ele.dest == jar.ManifestFile {
|
|
err = z.addManifest(ele.dest, ele.src, ele.zipMethod)
|
|
} else {
|
|
err = z.addFile(ele.dest, ele.src, ele.zipMethod, emulateJar, srcJar)
|
|
}
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
zipw := zip.NewWriter(f)
|
|
|
|
var currentWriteOpChan chan *zipEntry
|
|
var currentWriter io.WriteCloser
|
|
var currentReaders chan chan io.Reader
|
|
var currentReader chan io.Reader
|
|
var done bool
|
|
|
|
for !done {
|
|
var writeOpsChan chan chan *zipEntry
|
|
var writeOpChan chan *zipEntry
|
|
var readersChan chan chan io.Reader
|
|
|
|
if currentReader != nil {
|
|
// Only read and process errors
|
|
} else if currentReaders != nil {
|
|
readersChan = currentReaders
|
|
} else if currentWriteOpChan != nil {
|
|
writeOpChan = currentWriteOpChan
|
|
} else {
|
|
writeOpsChan = z.writeOps
|
|
}
|
|
|
|
select {
|
|
case writeOp, ok := <-writeOpsChan:
|
|
if !ok {
|
|
done = true
|
|
}
|
|
|
|
currentWriteOpChan = writeOp
|
|
|
|
case op := <-writeOpChan:
|
|
currentWriteOpChan = nil
|
|
|
|
var err error
|
|
if op.fh.Method == zip.Deflate {
|
|
currentWriter, err = zipw.CreateCompressedHeader(op.fh)
|
|
} else {
|
|
var zw io.Writer
|
|
|
|
op.fh.CompressedSize64 = op.fh.UncompressedSize64
|
|
|
|
zw, err = zipw.CreateHeaderAndroid(op.fh)
|
|
currentWriter = nopCloser{zw}
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
currentReaders = op.futureReaders
|
|
if op.futureReaders == nil {
|
|
currentWriter.Close()
|
|
currentWriter = nil
|
|
}
|
|
z.memoryRateLimiter.Finish(op.allocatedSize)
|
|
|
|
case futureReader, ok := <-readersChan:
|
|
if !ok {
|
|
// Done with reading
|
|
currentWriter.Close()
|
|
currentWriter = nil
|
|
currentReaders = nil
|
|
}
|
|
|
|
currentReader = futureReader
|
|
|
|
case reader := <-currentReader:
|
|
_, err := io.Copy(currentWriter, reader)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
currentReader = nil
|
|
|
|
case err := <-z.errors:
|
|
return err
|
|
}
|
|
}
|
|
|
|
// One last chance to catch an error
|
|
select {
|
|
case err := <-z.errors:
|
|
return err
|
|
default:
|
|
zipw.Close()
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// imports (possibly with compression) <src> into the zip at sub-path <dest>
|
|
func (z *ZipWriter) addFile(dest, src string, method uint16, emulateJar, srcJar bool) error {
|
|
var fileSize int64
|
|
var executable bool
|
|
|
|
var s os.FileInfo
|
|
var err error
|
|
if z.followSymlinks {
|
|
s, err = z.fs.Stat(src)
|
|
} else {
|
|
s, err = z.fs.Lstat(src)
|
|
}
|
|
|
|
if err != nil {
|
|
if os.IsNotExist(err) && z.ignoreMissingFiles {
|
|
fmt.Fprintln(z.stderr, "warning:", err)
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
createParentDirs := func(dest, src string) error {
|
|
if err := z.writeDirectory(filepath.Dir(dest), src, emulateJar); err != nil {
|
|
return err
|
|
}
|
|
|
|
if prev, exists := z.createdDirs[dest]; exists {
|
|
return fmt.Errorf("destination %q is both a directory %q and a file %q", dest, prev, src)
|
|
}
|
|
if prev, exists := z.createdFiles[dest]; exists {
|
|
return fmt.Errorf("destination %q has two files %q and %q", dest, prev, src)
|
|
}
|
|
|
|
z.createdFiles[dest] = src
|
|
|
|
return nil
|
|
}
|
|
|
|
if s.IsDir() {
|
|
if z.directories {
|
|
return z.writeDirectory(dest, src, emulateJar)
|
|
}
|
|
return nil
|
|
} else if s.Mode()&os.ModeSymlink != 0 {
|
|
err = createParentDirs(dest, src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return z.writeSymlink(dest, src)
|
|
} else if s.Mode().IsRegular() {
|
|
r, err := z.fs.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if srcJar && filepath.Ext(src) == ".java" {
|
|
// rewrite the destination using the package path if it can be determined
|
|
pkg, err := jar.JavaPackage(r, src)
|
|
if err != nil {
|
|
// ignore errors for now, leaving the file at in its original location in the zip
|
|
} else {
|
|
dest = filepath.Join(filepath.Join(strings.Split(pkg, ".")...), filepath.Base(src))
|
|
}
|
|
|
|
_, err = r.Seek(0, io.SeekStart)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
fileSize = s.Size()
|
|
executable = s.Mode()&0100 != 0
|
|
|
|
header := &zip.FileHeader{
|
|
Name: dest,
|
|
Method: method,
|
|
UncompressedSize64: uint64(fileSize),
|
|
}
|
|
|
|
if executable {
|
|
header.SetMode(0700)
|
|
}
|
|
|
|
err = createParentDirs(dest, src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return z.writeFileContents(header, r)
|
|
} else {
|
|
return fmt.Errorf("%s is not a file, directory, or symlink", src)
|
|
}
|
|
}
|
|
|
|
func (z *ZipWriter) addManifest(dest string, src string, method uint16) error {
|
|
if prev, exists := z.createdDirs[dest]; exists {
|
|
return fmt.Errorf("destination %q is both a directory %q and a file %q", dest, prev, src)
|
|
}
|
|
if prev, exists := z.createdFiles[dest]; exists {
|
|
return fmt.Errorf("destination %q has two files %q and %q", dest, prev, src)
|
|
}
|
|
|
|
if err := z.writeDirectory(filepath.Dir(dest), src, true); err != nil {
|
|
return err
|
|
}
|
|
|
|
var contents []byte
|
|
if src != "" {
|
|
f, err := z.fs.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
contents, err = ioutil.ReadAll(f)
|
|
f.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
fh, buf, err := jar.ManifestFileContents(contents)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
reader := &byteReaderCloser{bytes.NewReader(buf), ioutil.NopCloser(nil)}
|
|
|
|
return z.writeFileContents(fh, reader)
|
|
}
|
|
|
|
func (z *ZipWriter) writeFileContents(header *zip.FileHeader, r pathtools.ReaderAtSeekerCloser) (err error) {
|
|
|
|
header.SetModTime(z.time)
|
|
|
|
compressChan := make(chan *zipEntry, 1)
|
|
z.writeOps <- compressChan
|
|
|
|
// Pre-fill a zipEntry, it will be sent in the compressChan once
|
|
// we're sure about the Method and CRC.
|
|
ze := &zipEntry{
|
|
fh: header,
|
|
}
|
|
|
|
ze.allocatedSize = int64(header.UncompressedSize64)
|
|
z.cpuRateLimiter.Request()
|
|
z.memoryRateLimiter.Request(ze.allocatedSize)
|
|
|
|
fileSize := int64(header.UncompressedSize64)
|
|
if fileSize == 0 {
|
|
fileSize = int64(header.UncompressedSize)
|
|
}
|
|
|
|
if header.Method == zip.Deflate && fileSize >= minParallelFileSize {
|
|
wg := new(sync.WaitGroup)
|
|
|
|
// Allocate enough buffer to hold all readers. We'll limit
|
|
// this based on actual buffer sizes in RateLimit.
|
|
ze.futureReaders = make(chan chan io.Reader, (fileSize/parallelBlockSize)+1)
|
|
|
|
// Calculate the CRC in the background, since reading the entire
|
|
// file could take a while.
|
|
//
|
|
// We could split this up into chunks as well, but it's faster
|
|
// than the compression. Due to the Go Zip API, we also need to
|
|
// know the result before we can begin writing the compressed
|
|
// data out to the zipfile.
|
|
wg.Add(1)
|
|
go z.crcFile(r, ze, compressChan, wg)
|
|
|
|
for start := int64(0); start < fileSize; start += parallelBlockSize {
|
|
sr := io.NewSectionReader(r, start, parallelBlockSize)
|
|
resultChan := make(chan io.Reader, 1)
|
|
ze.futureReaders <- resultChan
|
|
|
|
z.cpuRateLimiter.Request()
|
|
|
|
last := !(start+parallelBlockSize < fileSize)
|
|
var dict []byte
|
|
if start >= windowSize {
|
|
dict, err = ioutil.ReadAll(io.NewSectionReader(r, start-windowSize, windowSize))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
wg.Add(1)
|
|
go z.compressPartialFile(sr, dict, last, resultChan, wg)
|
|
}
|
|
|
|
close(ze.futureReaders)
|
|
|
|
// Close the file handle after all readers are done
|
|
go func(wg *sync.WaitGroup, closer io.Closer) {
|
|
wg.Wait()
|
|
closer.Close()
|
|
}(wg, r)
|
|
} else {
|
|
go func() {
|
|
z.compressWholeFile(ze, r, compressChan)
|
|
r.Close()
|
|
}()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (z *ZipWriter) crcFile(r io.Reader, ze *zipEntry, resultChan chan *zipEntry, wg *sync.WaitGroup) {
|
|
defer wg.Done()
|
|
defer z.cpuRateLimiter.Finish()
|
|
|
|
crc := crc32.NewIEEE()
|
|
_, err := io.Copy(crc, r)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
|
|
ze.fh.CRC32 = crc.Sum32()
|
|
resultChan <- ze
|
|
close(resultChan)
|
|
}
|
|
|
|
func (z *ZipWriter) compressPartialFile(r io.Reader, dict []byte, last bool, resultChan chan io.Reader, wg *sync.WaitGroup) {
|
|
defer wg.Done()
|
|
|
|
result, err := z.compressBlock(r, dict, last)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
|
|
z.cpuRateLimiter.Finish()
|
|
|
|
resultChan <- result
|
|
}
|
|
|
|
func (z *ZipWriter) compressBlock(r io.Reader, dict []byte, last bool) (*bytes.Buffer, error) {
|
|
buf := new(bytes.Buffer)
|
|
var fw *flate.Writer
|
|
var err error
|
|
if len(dict) > 0 {
|
|
// There's no way to Reset a Writer with a new dictionary, so
|
|
// don't use the Pool
|
|
fw, err = flate.NewWriterDict(buf, z.compLevel, dict)
|
|
} else {
|
|
var ok bool
|
|
if fw, ok = z.compressorPool.Get().(*flate.Writer); ok {
|
|
fw.Reset(buf)
|
|
} else {
|
|
fw, err = flate.NewWriter(buf, z.compLevel)
|
|
}
|
|
defer z.compressorPool.Put(fw)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = io.Copy(fw, r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if last {
|
|
fw.Close()
|
|
} else {
|
|
fw.Flush()
|
|
}
|
|
|
|
return buf, nil
|
|
}
|
|
|
|
func (z *ZipWriter) compressWholeFile(ze *zipEntry, r io.ReadSeeker, compressChan chan *zipEntry) {
|
|
|
|
crc := crc32.NewIEEE()
|
|
_, err := io.Copy(crc, r)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
|
|
ze.fh.CRC32 = crc.Sum32()
|
|
|
|
_, err = r.Seek(0, 0)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
|
|
readFile := func(reader io.ReadSeeker) ([]byte, error) {
|
|
_, err := reader.Seek(0, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
buf, err := ioutil.ReadAll(reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf, nil
|
|
}
|
|
|
|
ze.futureReaders = make(chan chan io.Reader, 1)
|
|
futureReader := make(chan io.Reader, 1)
|
|
ze.futureReaders <- futureReader
|
|
close(ze.futureReaders)
|
|
|
|
if ze.fh.Method == zip.Deflate {
|
|
compressed, err := z.compressBlock(r, nil, true)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
if uint64(compressed.Len()) < ze.fh.UncompressedSize64 {
|
|
futureReader <- compressed
|
|
} else {
|
|
buf, err := readFile(r)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
ze.fh.Method = zip.Store
|
|
futureReader <- bytes.NewReader(buf)
|
|
}
|
|
} else {
|
|
buf, err := readFile(r)
|
|
if err != nil {
|
|
z.errors <- err
|
|
return
|
|
}
|
|
ze.fh.Method = zip.Store
|
|
futureReader <- bytes.NewReader(buf)
|
|
}
|
|
|
|
z.cpuRateLimiter.Finish()
|
|
|
|
close(futureReader)
|
|
|
|
compressChan <- ze
|
|
close(compressChan)
|
|
}
|
|
|
|
// writeDirectory annotates that dir is a directory created for the src file or directory, and adds
|
|
// the directory entry to the zip file if directories are enabled.
|
|
func (z *ZipWriter) writeDirectory(dir string, src string, emulateJar bool) error {
|
|
// clean the input
|
|
dir = filepath.Clean(dir)
|
|
|
|
// discover any uncreated directories in the path
|
|
zipDirs := []string{}
|
|
for dir != "" && dir != "." {
|
|
if _, exists := z.createdDirs[dir]; exists {
|
|
break
|
|
}
|
|
|
|
if prev, exists := z.createdFiles[dir]; exists {
|
|
return fmt.Errorf("destination %q is both a directory %q and a file %q", dir, src, prev)
|
|
}
|
|
|
|
z.createdDirs[dir] = src
|
|
// parent directories precede their children
|
|
zipDirs = append([]string{dir}, zipDirs...)
|
|
|
|
dir = filepath.Dir(dir)
|
|
}
|
|
|
|
if z.directories {
|
|
// make a directory entry for each uncreated directory
|
|
for _, cleanDir := range zipDirs {
|
|
var dirHeader *zip.FileHeader
|
|
|
|
if emulateJar && cleanDir+"/" == jar.MetaDir {
|
|
dirHeader = jar.MetaDirFileHeader()
|
|
} else {
|
|
dirHeader = &zip.FileHeader{
|
|
Name: cleanDir + "/",
|
|
}
|
|
dirHeader.SetMode(0700 | os.ModeDir)
|
|
}
|
|
|
|
dirHeader.SetModTime(z.time)
|
|
|
|
ze := make(chan *zipEntry, 1)
|
|
ze <- &zipEntry{
|
|
fh: dirHeader,
|
|
}
|
|
close(ze)
|
|
z.writeOps <- ze
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (z *ZipWriter) writeSymlink(rel, file string) error {
|
|
fileHeader := &zip.FileHeader{
|
|
Name: rel,
|
|
}
|
|
fileHeader.SetModTime(z.time)
|
|
fileHeader.SetMode(0777 | os.ModeSymlink)
|
|
|
|
dest, err := z.fs.Readlink(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fileHeader.UncompressedSize64 = uint64(len(dest))
|
|
fileHeader.CRC32 = crc32.ChecksumIEEE([]byte(dest))
|
|
|
|
ze := make(chan *zipEntry, 1)
|
|
futureReaders := make(chan chan io.Reader, 1)
|
|
futureReader := make(chan io.Reader, 1)
|
|
futureReaders <- futureReader
|
|
close(futureReaders)
|
|
futureReader <- bytes.NewBufferString(dest)
|
|
close(futureReader)
|
|
|
|
ze <- &zipEntry{
|
|
fh: fileHeader,
|
|
futureReaders: futureReaders,
|
|
}
|
|
close(ze)
|
|
z.writeOps <- ze
|
|
|
|
return nil
|
|
}
|