Как распаковать различные формы целых чисел в байтовый буфер в Golang?

Мне нужно извлечь различные поля в байтовом буфере. Я придумал это решение:

func (fs *FileSystem) readSB() {
    // fs.f is a *os.File
    buf := make([]byte, 1024)
    fs.f.ReadAt(buf, 1024)

    // Offset: type
    var p *bytes.Buffer

    // 0: uint32
    p = bytes.NewBuffer(buf[0:])
    binary.Read(p, binary.LittleEndian, &fs.sb.inodeCount)
    // 4: uint32
    p = bytes.NewBuffer(buf[4:])
    binary.Read(p, binary.LittleEndian, &fs.sb.blockCount)
    // 20: uint32
    p = bytes.NewBuffer(buf[20:])
    binary.Read(p, binary.LittleEndian, &fs.sb.firstDataBlock)
    // 24: uint32
    p = bytes.NewBuffer(buf[24:])
    binary.Read(p, binary.LittleEndian, &fs.sb.blockSize)
    fs.sb.blockSize = 1024 << fs.sb.blockSize
    // 32: uint32
    p = bytes.NewBuffer(buf[32:])
    binary.Read(p, binary.LittleEndian, &fs.sb.blockPerGroup)
    // 40: uint32
    p = bytes.NewBuffer(buf[40:])
    binary.Read(p, binary.LittleEndian, &fs.sb.inodePerBlock)
}

Есть ли более лучший/идиоматический/прямой способ сделать это?

  • Я хочу, чтобы смещения были явными
  • Я хочу читать из буфера байтов, а не искать и читать из файла, когда это возможно.

go buffer unpack
person knarf    schedule 10.09.2012    source источник
comment
Вы смотрели на encoding/gob? Это не сработает с вашей целью явных смещений, но если ваша цель на самом деле просто сериализовать/десериализовать, то ее намного проще использовать.   -  person Running Wild    schedule 10.09.2012
comment
Я разбираю существующий формат (ext2fs).   -  person knarf    schedule 11.09.2012
comment
То, что у вас есть, довольно идиоматично. вы могли бы пофантазировать, если бы захотели с циклом for и кусочком указателей, но это, вероятно, не читалось бы так ясно, как то, что у вас есть здесь.   -  person Jeremy Wall    schedule 11.09.2012
comment
Необходимость создавать bytes.Buffer каждый раз кажется расточительной.   -  person knarf    schedule 11.09.2012

Ответы (2)


arrow_upward
33
arrow_downward

Вы можете избежать создания нового буфера каждый раз, используя .Next(), чтобы пропустить ненужные байты. не хочу читать:

{
    // Offset: type
    p := bytes.NewBuffer(buf)

    // 0: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.inodeCount)

    // 4: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.blockCount)

    // Skip [8:20)
    p.Next(12)

    // 20: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.firstDataBlock)

    // 24: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.blockSize)
    fs.sb.blockSize = 1024 << fs.sb.blockSize

    // Skip [28:32)
    p.Next(4)

    // 32: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.blockPerGroup)

    // Skip [36:40)
    p.Next(4)

    // 40: uint32
    binary.Read(p, binary.LittleEndian, &fs.sb.inodePerBlock)
}

Или вы можете не читать фрагмент за фрагментом и создать структуру заголовка, которую вы читаете напрямую, используя binary.Read:

type Head struct {
    InodeCount      uint32  //  0:4
    BlockCount      uint32  //  4:8
    Unknown1        uint32  //  8:12
    Unknown2        uint32  // 12:16
    Unknown3        uint32  // 16:20
    FirstBlock      uint32  // 20:24
    BlockSize       uint32  // 24:28
    Unknown4        uint32  // 28:32
    BlocksPerGroup  uint32  // 32:36
    Unknown5        uint32  // 36:40
    InodesPerBlock  uint32  // 40:44
}

func main() {
    var header Head

    err = binary.Read(file, binary.LittleEndian, &header)

    if err != nil {
        log.Fatal(err)
    }

    log.Printf("%#v\n", header)
}
person nemo    schedule 11.09.2012

arrow_upward
1
arrow_downward

У меня есть пакет binpacker для обработки таких ситуаций.

пример

пример данных:

buffer := new(bytes.Buffer)
packer := binpacker.NewPacker(buffer)
unpacker := binpacker.NewUnpacker(buffer)
packer.PushByte(0x01)
packer.PushUint16(math.MaxUint16)

распаковать его:

var val1 byte
var val2 uint16
var err error
val1, err = unpacker.ShiftByte()
val2, err = unpacker.ShiftUint16()

Or

var val1 byte
var val2 uint16
var err error
unpacker.FetchByte(&val1).FetchUint16(&val2)
unpacker.Error() // Make sure error is nil
person Sirui Zhuang    schedule 11.02.2016
comment
Обратите внимание, что вы создали пакет. - person majidarif; 17.01.2017