torrent/peer_protocol/int.go

package peer_protocol

import (
	"bytes"
	"encoding/binary"
	"io"
)

type Integer uint32

func (i *Integer) Read(r io.Reader) error {
	return binary.Read(r, binary.BigEndian, i)
}

func (i *Integer) UnmarshalBinary(b []byte) error {
	return i.Read(bytes.NewReader(b))
}

func (i *Integer) ReadFrom(r io.Reader) error {
	var b [4]byte
	n, err := r.Read(b[:])
	if n == 4 {
		*i = Integer(binary.BigEndian.Uint32(b[:]))
		return nil
	}
	if err == nil {
		return io.ErrUnexpectedEOF
	}
	return err
}

// It's perfectly fine to cast these to an int. TODO: Or is it?
func (i Integer) Int() int {
	return int(i)
}

func (i Integer) Uint64() uint64 {
	return uint64(i)
}

func (i Integer) Uint32() uint32 {
	return uint32(i)
}