diff --git a/d2common/stream_writer.go b/d2common/stream_writer.go index 6c10530b..63bb578f 100644 --- a/d2common/stream_writer.go +++ b/d2common/stream_writer.go @@ -1,69 +1,63 @@ package d2common +import "bytes" + // StreamWriter allows you to create a byte array by streaming in writes of various sizes type StreamWriter struct { - data []byte + data *bytes.Buffer } // CreateStreamWriter creates a new StreamWriter instance func CreateStreamWriter() *StreamWriter { result := &StreamWriter{ - data: make([]byte, 0), + data: new(bytes.Buffer), } return result } // PushByte writes a byte to the stream func (v *StreamWriter) PushByte(val byte) { - v.data = append(v.data, val) + v.data.WriteByte(val) } // PushUint16 writes an uint16 word to the stream func (v *StreamWriter) PushUint16(val uint16) { - v.data = append(v.data, byte(val&0xFF)) - v.data = append(v.data, byte((val>>8)&0xFF)) + v.data.WriteByte(byte(val) & 0xFF) + v.data.WriteByte(byte(val>>8) & 0xFF) } // PushInt16 writes a int16 word to the stream func (v *StreamWriter) PushInt16(val int16) { - v.data = append(v.data, byte(val&0xFF)) - v.data = append(v.data, byte((val>>8)&0xFF)) + v.data.WriteByte(byte(val) & 0xFF) + v.data.WriteByte(byte(val>>8) & 0xFF) } // PushUint32 writes a uint32 dword to the stream func (v *StreamWriter) PushUint32(val uint32) { - v.data = append(v.data, byte(val&0xFF)) - v.data = append(v.data, byte((val>>8)&0xFF)) - v.data = append(v.data, byte((val>>16)&0xFF)) - v.data = append(v.data, byte((val>>24)&0xFF)) + v.data.WriteByte(byte(val) & 0xFF) + v.data.WriteByte(byte(val>>8) & 0xFF) + v.data.WriteByte(byte(val>>16) & 0xFF) + v.data.WriteByte(byte(val>>24) & 0xFF) } // PushUint64 writes a uint64 qword to the stream func (v *StreamWriter) PushUint64(val uint64) { - v.data = append(v.data, byte(val&0xFF)) - v.data = append(v.data, byte((val>>8)&0xFF)) - v.data = append(v.data, byte((val>>16)&0xFF)) - v.data = append(v.data, byte((val>>24)&0xFF)) - v.data = append(v.data, byte((val>>32)&0xFF)) - v.data = append(v.data, byte((val>>40)&0xFF)) - v.data = append(v.data, byte((val>>48)&0xFF)) - v.data = append(v.data, byte((val>>56)&0xFF)) + v.data.WriteByte(byte(val) & 0xFF) + v.data.WriteByte(byte(val>>8) & 0xFF) + v.data.WriteByte(byte(val>>16) & 0xFF) + v.data.WriteByte(byte(val>>24) & 0xFF) + v.data.WriteByte(byte(val>>32) & 0xFF) + v.data.WriteByte(byte(val>>40) & 0xFF) + v.data.WriteByte(byte(val>>48) & 0xFF) + v.data.WriteByte(byte(val>>56) & 0xFF) } // PushInt64 writes a uint64 qword to the stream func (v *StreamWriter) PushInt64(val int64) { - result := uint64(val) - v.data = append(v.data, byte(result&0xFF)) - v.data = append(v.data, byte((result>>8)&0xFF)) - v.data = append(v.data, byte((result>>16)&0xFF)) - v.data = append(v.data, byte((result>>24)&0xFF)) - v.data = append(v.data, byte((result>>32)&0xFF)) - v.data = append(v.data, byte((result>>40)&0xFF)) - v.data = append(v.data, byte((result>>48)&0xFF)) - v.data = append(v.data, byte((result>>56)&0xFF)) + v.PushUint64(uint64(val)) } // GetBytes returns the the byte slice of the underlying data func (v *StreamWriter) GetBytes() []byte { - return v.data + return v.data.Bytes() }