Blame vendor/github.com/Azure/azure-storage-blob-go/azblob/zc_util_validate.go

Packit 63bb0d
package azblob
Packit 63bb0d
Packit 63bb0d
import (
Packit 63bb0d
	"errors"
Packit 63bb0d
	"fmt"
Packit 63bb0d
	"io"
Packit 63bb0d
	"strconv"
Packit 63bb0d
)
Packit 63bb0d
Packit 63bb0d
// httpRange defines a range of bytes within an HTTP resource, starting at offset and
Packit 63bb0d
// ending at offset+count. A zero-value httpRange indicates the entire resource. An httpRange
Packit 63bb0d
// which has an offset but na zero value count indicates from the offset to the resource's end.
Packit 63bb0d
type httpRange struct {
Packit 63bb0d
	offset int64
Packit 63bb0d
	count  int64
Packit 63bb0d
}
Packit 63bb0d
Packit 63bb0d
func (r httpRange) pointers() *string {
Packit 63bb0d
	if r.offset == 0 && r.count == CountToEnd { // Do common case first for performance
Packit 63bb0d
		return nil // No specified range
Packit 63bb0d
	}
Packit 63bb0d
	endOffset := "" // if count == CountToEnd (0)
Packit 63bb0d
	if r.count > 0 {
Packit 63bb0d
		endOffset = strconv.FormatInt((r.offset+r.count)-1, 10)
Packit 63bb0d
	}
Packit 63bb0d
	dataRange := fmt.Sprintf("bytes=%v-%s", r.offset, endOffset)
Packit 63bb0d
	return &dataRange
Packit 63bb0d
}
Packit 63bb0d
Packit 63bb0d
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
Packit 63bb0d
Packit 63bb0d
func validateSeekableStreamAt0AndGetCount(body io.ReadSeeker) (int64, error) {
Packit 63bb0d
	if body == nil { // nil body's are "logically" seekable to 0 and are 0 bytes long
Packit 63bb0d
		return 0, nil
Packit 63bb0d
	}
Packit 63bb0d
Packit 63bb0d
	err := validateSeekableStreamAt0(body)
Packit 63bb0d
	if err != nil {
Packit 63bb0d
		return 0, err
Packit 63bb0d
	}
Packit 63bb0d
Packit 63bb0d
	count, err := body.Seek(0, io.SeekEnd)
Packit 63bb0d
	if err != nil {
Packit 63bb0d
		return 0, errors.New("body stream must be seekable")
Packit 63bb0d
	}
Packit 63bb0d
Packit 63bb0d
	body.Seek(0, io.SeekStart)
Packit 63bb0d
	return count, nil
Packit 63bb0d
}
Packit 63bb0d
Packit 63bb0d
// return an error if body is not a valid seekable stream at 0
Packit 63bb0d
func validateSeekableStreamAt0(body io.ReadSeeker) error {
Packit 63bb0d
	if body == nil { // nil body's are "logically" seekable to 0
Packit 63bb0d
		return nil
Packit 63bb0d
	}
Packit 63bb0d
	if pos, err := body.Seek(0, io.SeekCurrent); pos != 0 || err != nil {
Packit 63bb0d
		// Help detect programmer error
Packit 63bb0d
		if err != nil {
Packit 63bb0d
			return errors.New("body stream must be seekable")
Packit 63bb0d
		}
Packit 63bb0d
		return errors.New("body stream must be set to position 0")
Packit 63bb0d
	}
Packit 63bb0d
	return nil
Packit 63bb0d
}