78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
// Copyright (c) 2013-2016 The btcsuite developers
|
||
|
|
// Use of this source code is governed by an ISC
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package wire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
// fixedWriter implements the io.Writer interface and intentially allows
|
||
|
|
// testing of error paths by forcing short writes.
|
||
|
|
type fixedWriter struct {
|
||
|
|
b []byte
|
||
|
|
pos int
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write writes the contents of p to w. When the contents of p would cause
|
||
|
|
// the writer to exceed the maximum allowed size of the fixed writer,
|
||
|
|
// io.ErrShortWrite is returned and the writer is left unchanged.
|
||
|
|
//
|
||
|
|
// This satisfies the io.Writer interface.
|
||
|
|
func (w *fixedWriter) Write(p []byte) (n int, err error) {
|
||
|
|
lenp := len(p)
|
||
|
|
if w.pos+lenp > cap(w.b) {
|
||
|
|
return 0, io.ErrShortWrite
|
||
|
|
}
|
||
|
|
n = lenp
|
||
|
|
w.pos += copy(w.b[w.pos:], p)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Bytes returns the bytes already written to the fixed writer.
|
||
|
|
func (w *fixedWriter) Bytes() []byte {
|
||
|
|
return w.b
|
||
|
|
}
|
||
|
|
|
||
|
|
// newFixedWriter returns a new io.Writer that will error once more bytes than
|
||
|
|
// the specified max have been written.
|
||
|
|
func newFixedWriter(max int) io.Writer {
|
||
|
|
b := make([]byte, max)
|
||
|
|
fw := fixedWriter{b, 0}
|
||
|
|
return &fw
|
||
|
|
}
|
||
|
|
|
||
|
|
// fixedReader implements the io.Reader interface and intentially allows
|
||
|
|
// testing of error paths by forcing short reads.
|
||
|
|
type fixedReader struct {
|
||
|
|
buf []byte
|
||
|
|
pos int
|
||
|
|
iobuf *bytes.Buffer
|
||
|
|
}
|
||
|
|
|
||
|
|
// Read reads the next len(p) bytes from the fixed reader. When the number of
|
||
|
|
// bytes read would exceed the maximum number of allowed bytes to be read from
|
||
|
|
// the fixed writer, an error is returned.
|
||
|
|
//
|
||
|
|
// This satisfies the io.Reader interface.
|
||
|
|
func (fr *fixedReader) Read(p []byte) (n int, err error) {
|
||
|
|
n, err = fr.iobuf.Read(p)
|
||
|
|
fr.pos += n
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// newFixedReader returns a new io.Reader that will error once more bytes than
|
||
|
|
// the specified max have been read.
|
||
|
|
func newFixedReader(max int, buf []byte) io.Reader {
|
||
|
|
b := make([]byte, max)
|
||
|
|
if buf != nil {
|
||
|
|
copy(b[:], buf)
|
||
|
|
}
|
||
|
|
|
||
|
|
iobuf := bytes.NewBuffer(b)
|
||
|
|
fr := fixedReader{b, 0, iobuf}
|
||
|
|
return &fr
|
||
|
|
}
|