func buildImage(objectClient *objectclient.ObjectClient, tarReader *tar.Reader,
filter *filter.Filter) (*filesystem.FileSystem, error) {
var dh dataHandler
dh.objQ = objectclient.NewObjectAdderQueue(objectClient, 1024*1024*128)
fs, err := untar.Decode(tarReader, &dh, filter)
if err != nil {
return nil, err
}
err = dh.objQ.Flush()
if err != nil {
return nil, err
}
return fs, nil
}
func buildImageWithHasher(imageSClient *srpc.Client, filter *filter.Filter,
imageFilename string, h *hasher) (*filesystem.FileSystem, error) {
fi, err := os.Lstat(imageFilename)
if err != nil {
return nil, err
}
if fi.IsDir() {
sfs, err := scanner.ScanFileSystem(imageFilename, nil, filter, nil, h,
nil)
if err != nil {
return nil, err
}
return &sfs.FileSystem, nil
}
imageFile, err := os.Open(imageFilename)
if err != nil {
return nil, errors.New("error opening image file: " + err.Error())
}
defer imageFile.Close()
if partitionTable, err := mbr.Decode(imageFile); err != nil {
if err != io.EOF {
return nil, err
} // Else perhaps a tiny tarfile, definitely not a partition table.
} else if partitionTable != nil {
return buildImageFromRaw(imageSClient, filter, imageFile,
partitionTable, h)
}
var imageReader io.Reader
if strings.HasSuffix(imageFilename, ".tar") {
imageReader = imageFile
} else if strings.HasSuffix(imageFilename, ".tar.gz") ||
strings.HasSuffix(imageFilename, ".tgz") {
gzipReader, err := gzip.NewReader(imageFile)
if err != nil {
return nil, errors.New(
"error creating gzip reader: " + err.Error())
}
defer gzipReader.Close()
imageReader = gzipReader
} else {
return nil, errors.New("unrecognised image type")
}
tarReader := tar.NewReader(imageReader)
fs, err := untar.Decode(tarReader, h, filter)
if err != nil {
return nil, errors.New("error building image: " + err.Error())
}
return fs, nil
}