mirror of https://github.com/go-gitea/gitea.git
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package packfile
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"sync"
|
|
|
|
"gopkg.in/src-d/go-git.v4/plumbing/storer"
|
|
"gopkg.in/src-d/go-git.v4/utils/ioutil"
|
|
)
|
|
|
|
var signature = []byte{'P', 'A', 'C', 'K'}
|
|
|
|
const (
|
|
// VersionSupported is the packfile version supported by this package
|
|
VersionSupported uint32 = 2
|
|
|
|
firstLengthBits = uint8(4) // the first byte into object header has 4 bits to store the length
|
|
lengthBits = uint8(7) // subsequent bytes has 7 bits to store the length
|
|
maskFirstLength = 15 // 0000 1111
|
|
maskContinue = 0x80 // 1000 0000
|
|
maskLength = uint8(127) // 0111 1111
|
|
maskType = uint8(112) // 0111 0000
|
|
)
|
|
|
|
// UpdateObjectStorage updates the storer with the objects in the given
|
|
// packfile.
|
|
func UpdateObjectStorage(s storer.Storer, packfile io.Reader) error {
|
|
if pw, ok := s.(storer.PackfileWriter); ok {
|
|
return WritePackfileToObjectStorage(pw, packfile)
|
|
}
|
|
|
|
p, err := NewParserWithStorage(NewScanner(packfile), s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = p.Parse()
|
|
return err
|
|
}
|
|
|
|
// WritePackfileToObjectStorage writes all the packfile objects into the given
|
|
// object storage.
|
|
func WritePackfileToObjectStorage(
|
|
sw storer.PackfileWriter,
|
|
packfile io.Reader,
|
|
) (err error) {
|
|
w, err := sw.PackfileWriter()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer ioutil.CheckClose(w, &err)
|
|
|
|
var n int64
|
|
n, err = io.Copy(w, packfile)
|
|
if err == nil && n == 0 {
|
|
return ErrEmptyPackfile
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
var bufPool = sync.Pool{
|
|
New: func() interface{} {
|
|
return bytes.NewBuffer(nil)
|
|
},
|
|
}
|