源码分享-M3U8数据流ts的AES-128解密并合并---GoLang实现

news/2024/5/20 0:50:13 标签: M3U8, AES-128, ts, Go

之前使用C语言实现了一次,见M3U8数据流tsAES-128解密并合并。

学习了Go语言后,又用Go重新实现了一遍。源码如下,无第三方库依赖。

package main

import (
	"crypto/aes"
	"crypto/cipher"
	"encoding/binary"
	"fmt"
	"io"
	"os"
	"strconv"
)

func decodeAES128CBC(key []byte, index int, inFile string, out io.Writer) error {
	block, err := aes.NewCipher(key)
	if err != nil {
		return err
	}
	inBuf, err := os.ReadFile(inFile)
	if err != nil || len(inBuf) == 0 {
		return err
	}

	var iv [16]byte
	binary.BigEndian.PutUint32(iv[12:], uint32(index))

	outBuf := make([]byte, len(inBuf))
	mode := cipher.NewCBCDecrypter(block, iv[:])
	mode.CryptBlocks(outBuf, inBuf)

	pad := int(outBuf[len(outBuf)-1])
	_, err = out.Write(outBuf[:len(outBuf)-pad])
	return err
}

func decodeAesM3u8(key, inPath, outFile string) (int, error) {
	out, err := os.Create(outFile)
	if err != nil {
		return 0, err
	}
	defer out.Close()

	for i := 0; ; i++ {
		err = decodeAES128CBC([]byte(key), i, inPath+strconv.Itoa(i), out)
		if err != nil {
			if i == 0 {
				return 0, err
			}
			return i, nil
		}
	}
}

func mergeM3u8(inPath, outFile string) (int, error) {
	out, err := os.Create(outFile)
	if err != nil {
		return 0, err
	}
	defer out.Close()

	for i := 0; ; i++ {
		in, err := os.Open(inPath + strconv.Itoa(i))
		if err != nil {
			if i == 0 {
				return 0, err
			}
			return i, nil
		}

		_, err = io.Copy(out, in)
		if err != nil {
			_ = in.Close()
			return i, err
		}

		err = in.Close()
		if err != nil {
			return i, err
		}
	}
}

func main() {
	inPath, key, outFile := os.Args[1], os.Args[2], os.Args[3]

	var n int
	var err error
	if len(key) == 16 {
		n, err = decodeAesM3u8(key, inPath, outFile)
	} else {
		n, err = mergeM3u8(inPath, outFile)
	}
	if err != nil {
		fmt.Println(err)
		return
	}
	fmt.Println("successfully converted", n, "files")
}

http://www.niftyadmin.cn/n/5067999.html

相关文章

OA产品选型的指导原则

1、最关键的是OA能帮你做什么,切忌大而全 对于企业来说,OA系统毕竟是提高企业内部管理效率的一种"工具",因此该哪些功能是企业选择OA系统时所应最为关注的因素,切忌大而全。OA系统的开发技术是一个参考的因素&#xff0…

【JavaEE】多线程(五)- 基础知识完结篇

多线程(五) 文章目录 多线程(五)volatile关键字保证内存可见性JMM(Java Memory Model) 不保证原子性 wait 和 notifywait()notify()线程饿死 上文我们主要讲了 synchronized以及线程安全的一些话题 可重入…

制作 3 档可调灯程序编写

PWM 0~255 可以将数据映射到0 75 150 225 尽可能均匀电压间隔

【异常错误】WSL2设置为全核cpu和全部内存

今天偶尔发现 WSL占用的内存是真实内存的一半(通过htop命令查看即可) 现在需要修改配置到使用全部的CPU资源: Windows R 键, 输入 %UserProfile% 并运行进入用户文件夹, 新建文件 .wslconfig,文件内容如下&#xf…

洛谷_分支循环

p2433 问题 5 甲列火车长 260 米,每秒行 12 米;乙列火车长220 米,每秒行 20 米,两车相向而行,从两车车头相遇时开始计时,多长时间后两车车尾相离?已知答案是整数。 计算方式:两车车…

第八章 排序 四、冒泡排序

目录 一、算法思想 二、例子 三、代码实现 四、验证 五、算法性能分析 注意:要分清楚交换次数和移动次数 六、总结 一、算法思想 从后往前,两两比较相邻元素的值,若为逆序,则交换它们的值,直到全部比较完。 二…

SPI 通信协议

1. SPI通信 1. 什么是SPI通信协议 2. SPI的通信过程 在一开始会先把发送缓冲器的数据(8位)。一次性放到移位寄存器里。 移位寄存器会一位一位发送出去。但是要先放到锁存器里。然后从机来读取。从机的过程也一样。当移位寄存器的数据全部发送完。其实…

STM32CubeMX学习笔记-DAC接口使用(输出电压)

STM32CubeMX学习笔记-DAC接口使用(输出电压) 一、DAC简介二、DAC通道选择三、新建工程四、DAC14.1 参数配置4.2 生成代码 五、库函数六、修改main函数 原创链接 1 一、DAC简介 DAC(Digital-to-Analog Converter),即数字/模拟转换模块&#x…