62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package reseed
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"io/ioutil"
|
|
)
|
|
|
|
func zipSeeds(seeds []routerInfo) ([]byte, error) {
|
|
// Create a buffer to write our archive to.
|
|
buf := new(bytes.Buffer)
|
|
|
|
// Create a new zip archive.
|
|
zipWriter := zip.NewWriter(buf)
|
|
|
|
// Add some files to the archive.
|
|
for _, file := range seeds {
|
|
fileHeader := &zip.FileHeader{Name: file.Name, Method: zip.Deflate}
|
|
fileHeader.SetModTime(file.ModTime)
|
|
zipFile, err := zipWriter.CreateHeader(fileHeader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = zipFile.Write(file.Data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if err := zipWriter.Close(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
func uzipSeeds(c []byte) ([]routerInfo, error) {
|
|
input := bytes.NewReader(c)
|
|
zipReader, err := zip.NewReader(input, int64(len(c)))
|
|
if nil != err {
|
|
return nil, err
|
|
}
|
|
|
|
var seeds []routerInfo
|
|
for _, f := range zipReader.File {
|
|
rc, err := f.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
data, err := ioutil.ReadAll(rc)
|
|
rc.Close()
|
|
if nil != err {
|
|
return nil, err
|
|
}
|
|
|
|
seeds = append(seeds, routerInfo{Name: f.Name, Data: data})
|
|
}
|
|
|
|
return seeds, nil
|
|
}
|