gravestench/OpenDiablo2
1
1
Fork 0
mirror of https://github.com/OpenDiablo2/OpenDiablo2 synced 2024-06-19 13:45:23 +00:00
Code Issues Releases Wiki Activity

stream writer: added warnings when bits count is greater then possible input size (in PUshBits... methods)

Browse Source
This commit is contained in:
M. Sz 2021-02-06 17:23:11 +01:00
parent b3a754a4a6
commit aadfbbc8a6
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
d2common/d2datautils/stream_writer.go
View File

@ -1,6 +1,9 @@
package d2datautils package d2datautils
import "bytes" import (
"bytes"
"log"
)
// StreamWriter allows you to create a byte array by streaming in writes of various sizes // StreamWriter allows you to create a byte array by streaming in writes of various sizes
type StreamWriter struct { type StreamWriter struct {
@ -49,7 +52,12 @@ func (v *StreamWriter) PushBit(b bool) {
} }
// PushBits pushes bits (with max range 8) // PushBits pushes bits (with max range 8)
//func (v *StreamWriter) PushBits(b byte, bits int) {
func (v *StreamWriter) PushBits(b byte, bits int) { func (v *StreamWriter) PushBits(b byte, bits int) {
if bits > bitsPerByte {
log.Print("input bits number must be less (or equal) then 8")
}
val := b val := b
for i := 0; i < bits; i++ { for i := 0; i < bits; i++ {
if val&1 == 1 { if val&1 == 1 {
@ -64,6 +72,10 @@ func (v *StreamWriter) PushBits(b byte, bits int) {
// PushBits16 pushes bits (with max range 16) // PushBits16 pushes bits (with max range 16)
func (v *StreamWriter) PushBits16(b uint16, bits int) { func (v *StreamWriter) PushBits16(b uint16, bits int) {
if bits > bitsPerByte*bytesPerint16 {
log.Print("input bits number must be less (or equal) then 16")
}
val := b val := b
for i := 0; i < bits; i++ { for i := 0; i < bits; i++ {
if val&1 == 1 { if val&1 == 1 {
@ -78,6 +90,10 @@ func (v *StreamWriter) PushBits16(b uint16, bits int) {
// PushBits32 pushes bits (with max range 32) // PushBits32 pushes bits (with max range 32)
func (v *StreamWriter) PushBits32(b uint32, bits int) { func (v *StreamWriter) PushBits32(b uint32, bits int) {
if bits > bitsPerByte*bytesPerint32 {
log.Print("input bits number must be less (or equal) then 32")
}
val := b val := b
for i := 0; i < bits; i++ { for i := 0; i < bits; i++ {
if val&1 == 1 { if val&1 == 1 {