Caleb Gardner
70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package squashfs
|
|
|
|
type fragment struct {
|
|
w *Writer
|
|
files []*fileHolder
|
|
sizes []uint32
|
|
}
|
|
|
|
func (f *fragment) SizeLeft() uint32 {
|
|
totalSize := uint32(0)
|
|
for _, siz := range f.sizes {
|
|
totalSize += siz
|
|
}
|
|
return f.w.BlockSize - uint32(totalSize)
|
|
}
|
|
|
|
func (f *fragment) AddFragment(fil *fileHolder) {
|
|
//SizeLeft should already be checked
|
|
fil.fragOffset = len(f.files)
|
|
f.files = append(f.files, fil)
|
|
f.sizes = append(f.sizes, fil.blockSizes[len(fil.blockSizes)-1])
|
|
}
|
|
|
|
func (w *Writer) addToFragments(fil *fileHolder) {
|
|
fragSize := fil.blockSizes[len(fil.blockSizes)-1]
|
|
//only fragment if the final block is less then 80% of a full block or AlwaysFragment
|
|
if w.Flags.AlwaysFragment || fragSize < uint32(float32(w.BlockSize)*0.8) {
|
|
var possibleFrags []int
|
|
for i := range w.frags {
|
|
left := w.frags[i].SizeLeft()
|
|
if left == fragSize {
|
|
fil.fragIndex = i
|
|
w.frags[i].AddFragment(fil)
|
|
return
|
|
} else if left > fragSize {
|
|
possibleFrags = append(possibleFrags, i)
|
|
}
|
|
}
|
|
if len(possibleFrags) > 0 {
|
|
fil.fragIndex = possibleFrags[0]
|
|
} else {
|
|
fil.fragIndex = len(w.frags)
|
|
w.frags = append(w.frags, fragment{
|
|
w: w,
|
|
files: []*fileHolder{fil},
|
|
sizes: []uint32{fragSize},
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *Writer) calculateFragsAndBlockSizes() {
|
|
for _, files := range w.structure {
|
|
for i := range files {
|
|
files[i].fragIndex = -1
|
|
files[i].blockSizes = make([]uint32, files[i].size/uint64(w.BlockSize))
|
|
for j := range files[i].blockSizes {
|
|
files[i].blockSizes[j] = w.BlockSize
|
|
}
|
|
fragSize := uint32(files[i].size % uint64(w.BlockSize))
|
|
if fragSize > 0 {
|
|
files[i].blockSizes = append(files[i].blockSizes, fragSize)
|
|
if !w.Flags.NoFragments {
|
|
w.addToFragments(files[i])
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|