io.ReadFull导致文件损坏吗？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《io.ReadFull导致文件损坏吗？》，聊聊，希望可以帮助到正在努力赚钱的你。

我正在尝试使用 io.readfull 来下载和解密 hls 流，以分块处理数据以节省内存：

为了简单起见，省略了不相关的代码部分。

func main() {
    f, _ := os.create(out.ts)

    for _, v := range mediaplaylist {
        resp, _ := http.get(v.uri)
        for {
            r, err := decrypthls(key, iv, resp.body)
            if err != nil && err == io.eof {
                break
            else if err != nil && err != io.errunexpectedeof {
                panic(err)
            }
            io.copy(f, r)
        }
    }
}

func decrypthls(key []byte, iv []byte, r io.reader) (io.reader, error) {
    block, _ := aes.newcipher(key)

    buf := make([]byte, 8192)

    mode := cipher.newcbcdecrypter(block, iv)

        n, err := io.readfull(r, buf)
        if err != nil && err != io.errunexpectedeof {
                return nil, err
        }

    mode.cryptblocks(buf, buf)

    return bytes.newreader(buf[:n]), err
}

起初这似乎有效，因为文件大小正确并且下载过程中没有错误， 但视频已损坏。不完全，因为文件仍然被识别为视频，但图像和声音失真。

如果我更改代码以使用 ioutil.readall 代替，最终的视频文件将不再损坏：

func main() {
    f, _ := os.Create(out.ts)

    for _, v := range mediaPlaylist {
        resp, _ := http.Get(v.URI)
        segment, _ := ioutil.ReadAll(resp.Body)
        r, _ := decryptHLS(key, iv, &segment)
        io.Copy(f, r)
    }
}

func decryptHLS(key []byte, iv []byte, s *[]byte) io.Reader {
    block, _ := aes.NewCipher(key)

    mode := cipher.NewCBCDecrypter(block, iv)

    mode.CryptBlocks(*s, *s)

    return bytes.NewReader(*s)
}

有什么想法为什么它在将整个段读入内存时工作正常，而不是在使用 io.readfull 并分块处理它时工作正常吗？

解决方案

在内部，CBCDecrypter 会生成 iv 的副本，因此后续块会以初始 IV 开始，而不是因之前的解密而发生变异的 IV。

创建解密器一次，您应该能够继续重复使用它来逐块解密（假设块大小是此加密算法期望的块大小的倍数）。

今天关于《io.ReadFull导致文件损坏吗？》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！