コード例 #1
0
ファイル: heimlich.go プロジェクト: robot0x/heim
func swallow(self string, binary string, paths []string) error {
	sf, err := os.Open(self)
	if err != nil {
		return fmt.Errorf("open %s: %s", self, err)
	}
	hzp := binary + ".hzp"
	f, err := os.OpenFile(hzp, os.O_RDWR|os.O_CREATE, 0755)
	if err != nil {
		return fmt.Errorf("open %s: %s", hzp, err)
	}
	n, err := io.Copy(f, sf)
	if err != nil {
		sf.Close()
		return fmt.Errorf("copy %s->%s: %s", self, hzp, err)
	}
	if err := sf.Close(); err != nil {
		return fmt.Errorf("close %s: %s", self, err)
	}
	zw := patchzip.NewWriterAt(f, n)
	if err := swallowFile(zw, binary); err != nil {
		f.Close()
		return fmt.Errorf("swallow %s: %s", binary, err)
	}
	for _, path := range paths {
		if err := swallowFile(zw, path); err != nil {
			f.Close()
			return fmt.Errorf("swallow %s: %s", path, err)
		}
	}
	if err := zw.Close(); err != nil {
		return err
	}
	return f.Close()
}
コード例 #2
0
func Zip(path string, dirs []string) (err error) {
	if IsZip(path) {
		return errors.New(path + " is already a zip file")
	}

	f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
	if err != nil {
		return
	}
	defer f.Close()

	startoffset, err := f.Seek(0, os.SEEK_END)
	if err != nil {
		return
	}

	w := patch.NewWriterAt(f, startoffset)

	for _, dir := range dirs {
		err = filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}

			fh, err := patch.FileInfoHeader(info)
			if err != nil {
				return err
			}
			fh.Name = path

			p, err := w.CreateHeader(fh)
			if err != nil {
				return err
			}
			if !info.IsDir() {
				content, err := ioutil.ReadFile(path)
				if err != nil {
					return err
				}
				_, err = p.Write(content)
				if err != nil {
					return err
				}
			}
			return err
		})
	}
	err = w.Close()
	return
}