81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
// Copyright 2024 by Sascha L. Teichmann
|
|
// Use of this source code is governed by the MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package common
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
)
|
|
|
|
type bigEndianReader struct {
|
|
parent io.Reader
|
|
buf [4]byte
|
|
err error
|
|
}
|
|
|
|
func (ber *bigEndianReader) u8() (uint8, error) {
|
|
if ber.err != nil {
|
|
return 0, ber.err
|
|
}
|
|
_, err := io.ReadFull(ber.parent, ber.buf[:1])
|
|
if err != nil {
|
|
ber.err = err
|
|
return 0, err
|
|
}
|
|
return ber.buf[0], nil
|
|
}
|
|
|
|
func (ber *bigEndianReader) u16() (uint16, error) {
|
|
if ber.err != nil {
|
|
return 0, ber.err
|
|
}
|
|
two := ber.buf[:2]
|
|
_, err := io.ReadFull(ber.parent, two)
|
|
if err != nil {
|
|
ber.err = err
|
|
return 0, err
|
|
}
|
|
return binary.BigEndian.Uint16(two), nil
|
|
}
|
|
|
|
func (ber *bigEndianReader) u32() (uint32, error) {
|
|
if ber.err != nil {
|
|
return 0, ber.err
|
|
}
|
|
four := ber.buf[:2]
|
|
_, err := io.ReadFull(ber.parent, four)
|
|
if err != nil {
|
|
ber.err = err
|
|
return 0, err
|
|
}
|
|
return binary.BigEndian.Uint32(four), nil
|
|
}
|
|
|
|
func (ber *bigEndianReader) str(n int) (string, error) {
|
|
if ber.err != nil {
|
|
return "", ber.err
|
|
}
|
|
buf := make([]byte, n)
|
|
_, err := io.ReadFull(ber.parent, buf)
|
|
if err != nil {
|
|
ber.err = err
|
|
return "", err
|
|
}
|
|
return string(buf), nil
|
|
}
|
|
|
|
func (ber *bigEndianReader) full(n int) ([]byte, error) {
|
|
if ber.err != nil {
|
|
return nil, ber.err
|
|
}
|
|
buf := make([]byte, n)
|
|
_, err := io.ReadFull(ber.parent, buf)
|
|
if err != nil {
|
|
ber.err = err
|
|
return nil, err
|
|
}
|
|
return buf, nil
|
|
}
|