mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-23 12:50:49 +02:00
* update github.com/blevesearch/bleve v2.0.2 -> v2.0.3 * github.com/denisenkom/go-mssqldb v0.9.0 -> v0.10.0 * github.com/editorconfig/editorconfig-core-go v2.4.1 -> v2.4.2 * github.com/go-chi/cors v1.1.1 -> v1.2.0 * github.com/go-git/go-billy v5.0.0 -> v5.1.0 * github.com/go-git/go-git v5.2.0 -> v5.3.0 * github.com/go-ldap/ldap v3.2.4 -> v3.3.0 * github.com/go-redis/redis v8.6.0 -> v8.8.2 * github.com/go-sql-driver/mysql v1.5.0 -> v1.6.0 * github.com/go-swagger/go-swagger v0.26.1 -> v0.27.0 * github.com/lib/pq v1.9.0 -> v1.10.1 * github.com/mattn/go-sqlite3 v1.14.6 -> v1.14.7 * github.com/go-testfixtures/testfixtures v3.5.0 -> v3.6.0 * github.com/issue9/identicon v1.0.1 -> v1.2.0 * github.com/klauspost/compress v1.11.8 -> v1.12.1 * github.com/mgechev/revive v1.0.3 -> v1.0.6 * github.com/microcosm-cc/bluemonday v1.0.7 -> v1.0.8 * github.com/niklasfasching/go-org v1.4.0 -> v1.5.0 * github.com/olivere/elastic v7.0.22 -> v7.0.24 * github.com/pelletier/go-toml v1.8.1 -> v1.9.0 * github.com/prometheus/client_golang v1.9.0 -> v1.10.0 * github.com/xanzy/go-gitlab v0.44.0 -> v0.48.0 * github.com/yuin/goldmark v1.3.3 -> v1.3.5 * github.com/6543/go-version v1.2.4 -> v1.3.1 * do github.com/lib/pq v1.10.0 -> v1.10.1 again ...
167 lines
3.4 KiB
Go
Vendored
167 lines
3.4 KiB
Go
Vendored
package internal
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
)
|
|
|
|
// ByteInput typed interface around io.Reader or raw bytes
|
|
type ByteInput interface {
|
|
// Next returns a slice containing the next n bytes from the buffer,
|
|
// advancing the buffer as if the bytes had been returned by Read.
|
|
Next(n int) ([]byte, error)
|
|
// ReadUInt32 reads uint32 with LittleEndian order
|
|
ReadUInt32() (uint32, error)
|
|
// ReadUInt16 reads uint16 with LittleEndian order
|
|
ReadUInt16() (uint16, error)
|
|
// GetReadBytes returns read bytes
|
|
GetReadBytes() int64
|
|
// SkipBytes skips exactly n bytes
|
|
SkipBytes(n int) error
|
|
}
|
|
|
|
// NewByteInputFromReader creates reader wrapper
|
|
func NewByteInputFromReader(reader io.Reader) ByteInput {
|
|
return &ByteInputAdapter{
|
|
r: reader,
|
|
readBytes: 0,
|
|
}
|
|
}
|
|
|
|
// NewByteInput creates raw bytes wrapper
|
|
func NewByteInput(buf []byte) ByteInput {
|
|
return &ByteBuffer{
|
|
buf: buf,
|
|
off: 0,
|
|
}
|
|
}
|
|
|
|
// ByteBuffer raw bytes wrapper
|
|
type ByteBuffer struct {
|
|
buf []byte
|
|
off int
|
|
}
|
|
|
|
// Next returns a slice containing the next n bytes from the reader
|
|
// If there are fewer bytes than the given n, io.ErrUnexpectedEOF will be returned
|
|
func (b *ByteBuffer) Next(n int) ([]byte, error) {
|
|
m := len(b.buf) - b.off
|
|
|
|
if n > m {
|
|
return nil, io.ErrUnexpectedEOF
|
|
}
|
|
|
|
data := b.buf[b.off : b.off+n]
|
|
b.off += n
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// ReadUInt32 reads uint32 with LittleEndian order
|
|
func (b *ByteBuffer) ReadUInt32() (uint32, error) {
|
|
if len(b.buf)-b.off < 4 {
|
|
return 0, io.ErrUnexpectedEOF
|
|
}
|
|
|
|
v := binary.LittleEndian.Uint32(b.buf[b.off:])
|
|
b.off += 4
|
|
|
|
return v, nil
|
|
}
|
|
|
|
// ReadUInt16 reads uint16 with LittleEndian order
|
|
func (b *ByteBuffer) ReadUInt16() (uint16, error) {
|
|
if len(b.buf)-b.off < 2 {
|
|
return 0, io.ErrUnexpectedEOF
|
|
}
|
|
|
|
v := binary.LittleEndian.Uint16(b.buf[b.off:])
|
|
b.off += 2
|
|
|
|
return v, nil
|
|
}
|
|
|
|
// GetReadBytes returns read bytes
|
|
func (b *ByteBuffer) GetReadBytes() int64 {
|
|
return int64(b.off)
|
|
}
|
|
|
|
// SkipBytes skips exactly n bytes
|
|
func (b *ByteBuffer) SkipBytes(n int) error {
|
|
m := len(b.buf) - b.off
|
|
|
|
if n > m {
|
|
return io.ErrUnexpectedEOF
|
|
}
|
|
|
|
b.off += n
|
|
|
|
return nil
|
|
}
|
|
|
|
// Reset resets the given buffer with a new byte slice
|
|
func (b *ByteBuffer) Reset(buf []byte) {
|
|
b.buf = buf
|
|
b.off = 0
|
|
}
|
|
|
|
// ByteInputAdapter reader wrapper
|
|
type ByteInputAdapter struct {
|
|
r io.Reader
|
|
readBytes int
|
|
}
|
|
|
|
// Next returns a slice containing the next n bytes from the buffer,
|
|
// advancing the buffer as if the bytes had been returned by Read.
|
|
func (b *ByteInputAdapter) Next(n int) ([]byte, error) {
|
|
buf := make([]byte, n)
|
|
m, err := io.ReadAtLeast(b.r, buf, n)
|
|
b.readBytes += m
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf, nil
|
|
}
|
|
|
|
// ReadUInt32 reads uint32 with LittleEndian order
|
|
func (b *ByteInputAdapter) ReadUInt32() (uint32, error) {
|
|
buf, err := b.Next(4)
|
|
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return binary.LittleEndian.Uint32(buf), nil
|
|
}
|
|
|
|
// ReadUInt16 reads uint16 with LittleEndian order
|
|
func (b *ByteInputAdapter) ReadUInt16() (uint16, error) {
|
|
buf, err := b.Next(2)
|
|
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return binary.LittleEndian.Uint16(buf), nil
|
|
}
|
|
|
|
// GetReadBytes returns read bytes
|
|
func (b *ByteInputAdapter) GetReadBytes() int64 {
|
|
return int64(b.readBytes)
|
|
}
|
|
|
|
// SkipBytes skips exactly n bytes
|
|
func (b *ByteInputAdapter) SkipBytes(n int) error {
|
|
_, err := b.Next(n)
|
|
|
|
return err
|
|
}
|
|
|
|
// Reset resets the given buffer with a new stream
|
|
func (b *ByteInputAdapter) Reset(stream io.Reader) {
|
|
b.r = stream
|
|
b.readBytes = 0
|
|
}
|