Início / coding / Perguntas / 79125791
Accepted
Ming
Asked: 2024-10-25 20:56:52 +0800 CST

Substituir uma string em uma fatia de byte por uma nova string de comprimento diferente sem interferir em outros dados

  • 772

Estou trabalhando em um código onde sei a posição inicial de uma string em um pacote, e sei o terminador da string, que é 0x00 0x00 0x00. Preciso substituir a string antiga por uma nova, mas estou tendo um problema: a nova string pode ser maior ou menor que a antiga, e não posso interferir nos outros bytes do meu pacote. Alguém pode me ajudar com isso?

func replaceString(data []byte, newString string, startPos int) error {
    // Checks if the start position is valid
    if startPos >= len(data) || startPos < 0 {
        return fmt.Errorf("start position out of bounds")
    }

    terminatorPos := bytes.Index(data[startPos:], []byte{0x00, 0x00, 0x00})
    if terminatorPos == -1 {
        return fmt.Errorf("termination sequence not found")
    }

    buf := &bytes.Buffer{}
    packets.WriteS(buf, "testing long string")

    return nil
}

func WriteS(buf *bytes.Buffer, value string) {
    // Converter a string para UCS-2
    ucs2 := utf16.Encode([]rune(value))

    // Escrever os bytes da string em UCS-2 (little-endian)
    for _, char := range ucs2 {
        buf.WriteByte(byte(char & 0xff))        // Byte menos significativo
        buf.WriteByte(byte((char >> 8) & 0xff)) // Byte mais significativo
    }

    // Escrever o terminador null (2 bytes de 0x00)
    buf.WriteByte(0x00)
    buf.WriteByte(0x00)
    buf.WriteByte(0x00)
}
go

1 respostas

  1. Best Answer

    Aqui está uma solução que retorna um []byte, ou um error. Ele é projetado para tolerar NULs UTF-16 simples antes do terminador. Ele requer Go 1.21para slices.Replace.

    package main

import (
    "bytes"
    "fmt"
    "slices"
    "unicode/utf16"
)

func replaceString(data []byte, newString string, startPos int) ([]byte, error) {
    if startPos >= len(data) || startPos < 0 {
        return nil, fmt.Errorf("start position out of bounds")
    }
    zero := 0
    terminatorPos := -1
    for i, b := range data[startPos:] {
        // Assumes UTF-16 i.e. 2 bytes per character. The terminator must
        // begin on an even-numbered byte.
        if b != 0 {
            zero = 0
        } else if zero > 0 || i%2 == 0 {
            zero += 1
            if zero == 3 {
                terminatorPos = i + 1 - zero
                break
            }
        }
    }
    if terminatorPos == -1 {
        return nil, fmt.Errorf("termination sequence not found")
    }
    ucs2 := utf16.Encode([]rune(newString))
    buf := &bytes.Buffer{}
    for _, char := range ucs2 {
        buf.WriteByte(byte(char & 0xff))
        buf.WriteByte(byte((char >> 8) & 0xff))
    }
    return slices.Replace(data, startPos, startPos+terminatorPos, buf.Bytes()...), nil
}

func main() {
    // [42 97 0 98 0 99 0 0 0 0 42]
    fmt.Println(replaceString([]byte{42, 0, 0, 1, 2, 3, 0, 0, 0, 0, 42}, "abc", 1))
}
    • 3

relate perguntas