diff options
| -rw-r--r-- | packfile/packfile.go | 1 | ||||
| -rw-r--r-- | packfile/reader.go | 9 |
2 files changed, 10 insertions, 0 deletions
diff --git a/packfile/packfile.go b/packfile/packfile.go index e670cd0..11ef969 100644 --- a/packfile/packfile.go +++ b/packfile/packfile.go @@ -4,6 +4,7 @@ import "fmt" type Packfile struct { Version uint32 + Size int64 ObjectCount int Checksum []byte Commits map[string]*Commit diff --git a/packfile/reader.go b/packfile/reader.go index 73afdb9..627fdfd 100644 --- a/packfile/reader.go +++ b/packfile/reader.go @@ -11,6 +11,10 @@ import ( const MaxObjectsLimit = 1000000 +const MaxSizeLimit = 300 * 1 << 20 + +var ErrMaxSize = fmt.Errorf("Max size exceeded for in-memory client") + type TrackingByteReader struct { r io.Reader n int @@ -24,6 +28,9 @@ func (t *TrackingByteReader) Read(p []byte) (n int, err error) { return 0, err } t.n += n + if t.n >= MaxSizeLimit { + return n, ErrMaxSize + } return n, err } @@ -103,6 +110,8 @@ func (pr *PackfileReader) Read() (*Packfile, error) { return nil, err } + packfile.Size = int64(pr.r.Pos()) + return packfile, nil } |