mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
ba5785036a
安全修复（客户端鉴权）： - remotedevice 服务端移除授权状态存储（authorized map/SetAuthorized/handleDeviceAuth） - DeviceMeta.Authorized 改为设备 hello 自报，服务端仅透传展示 - device_ctl_* 工具移除服务端授权检查，无条件转发，设备端自行决定是否执行 - 共享设备桥库 Bridge 新增本地 authorized 状态，未授权收到 cmd 直接拒绝 - waiter: --device-authorized / device_authorized 配置控制本地授权 - GUI: 授权存 gui-prefs 本地文件；设备页仅本机可切换开关 - webui /device/auth 旧路径返回 410 Gone - 根因：agent 可经 config_set 篡改服务端授权配置自行授权设备 插件管理强化： - 内置插件禁止卸载（IsBuiltinPlugin + 409），外部插件卸载即时生效 - 卸载不存在插件返回 404；移除误导性 reload_required 提示 - webui 插件路由：名称白名单校验防路径穿越、保留字路径保护
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package client
|
||
|
||
// BinaryChunker 提供二进制数据分块传输功能。
|
||
// 用于将大体积数据（如录像 mp4、大图片）按分块协议发送。
|
||
// 协议:
|
||
// cmd_data_start {op, req_id, kind, total, chunk_size, mime} —— 文本帧
|
||
// <N 个二进制帧 0x2> —— data bytes
|
||
// cmd_data_end {op, req_id, status:ok|error, error?} —— 文本帧
|
||
|
||
const (
|
||
// DefaultChunkSize 默认分块大小（8KB）
|
||
DefaultChunkSize = 8192
|
||
|
||
// MaxBinaryFrameSize 二进制帧最大大小（8MB）
|
||
MaxBinaryFrameSize = 8 << 20
|
||
)
|
||
|
||
// ChunkCallback 分块发送回调，用于逐块处理。
|
||
type ChunkCallback func(chunk []byte) error
|
||
|
||
// ChunkData 将数据按指定大小分块。
|
||
func ChunkData(data []byte, chunkSize int) [][]byte {
|
||
if chunkSize <= 0 {
|
||
chunkSize = DefaultChunkSize
|
||
}
|
||
total := len(data)
|
||
var chunks [][]byte
|
||
for off := 0; off < total; off += chunkSize {
|
||
end := off + chunkSize
|
||
if end > total {
|
||
end = total
|
||
}
|
||
chunks = append(chunks, data[off:end])
|
||
}
|
||
return chunks
|
||
}
|
||
|
||
// SendChunked 使用回调逐块发送数据。
|
||
func SendChunked(data []byte, chunkSize int, fn ChunkCallback) error {
|
||
if chunkSize <= 0 {
|
||
chunkSize = DefaultChunkSize
|
||
}
|
||
chunks := ChunkData(data, chunkSize)
|
||
for _, chunk := range chunks {
|
||
if err := fn(chunk); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ===== 数据聚合（接收端） =====
|
||
|
||
// DataAccumulator 聚合从设备接收的二进制分块数据。
|
||
type DataAccumulator struct {
|
||
ReqID string
|
||
Kind string
|
||
MIME string
|
||
Total int
|
||
Got int
|
||
Chunks [][]byte
|
||
}
|
||
|
||
// NewDataAccumulator 创建数据聚合器。
|
||
func NewDataAccumulator(reqID, kind, mime string, total int) *DataAccumulator {
|
||
return &DataAccumulator{
|
||
ReqID: reqID,
|
||
Kind: kind,
|
||
MIME: mime,
|
||
Total: total,
|
||
Chunks: make([][]byte, 0),
|
||
}
|
||
}
|
||
|
||
// Append 追加一块数据。
|
||
func (da *DataAccumulator) Append(chunk []byte) {
|
||
da.Chunks = append(da.Chunks, chunk)
|
||
da.Got += len(chunk)
|
||
}
|
||
|
||
// Assemble 聚合所有分块为完整数据。
|
||
func (da *DataAccumulator) Assemble() []byte {
|
||
total := 0
|
||
for _, c := range da.Chunks {
|
||
total += len(c)
|
||
}
|
||
data := make([]byte, 0, total)
|
||
for _, c := range da.Chunks {
|
||
data = append(data, c...)
|
||
}
|
||
return data
|
||
}
|
||
|
||
// ExceededLimit 检查是否超出限制（声明的 2 倍或硬上限 64MB）。
|
||
func (da *DataAccumulator) ExceededLimit() bool {
|
||
limit := da.Total*2 + 1024
|
||
if limit < 64<<20 {
|
||
limit = 64 << 20
|
||
}
|
||
return da.Got > limit
|
||
}
|