基于WebRTC实现安全文件传输的gfile库

WebRTC (Web Real-Time Communication) 是一个支持浏览器之间进行实时音视频和数据通信的开源项目。在Go语言中，我们可以利用WebRTC技术实现安全的P2P文件传输。以下是关于如何使用gfile库实现这一功能的详细介绍。

gfile库简介

gfile是一个基于WebRTC的Go语言库，专为安全文件传输设计。它提供以下特性：

• 端到端加密传输
• P2P直连，减少服务器带宽消耗
• 支持大文件分块传输
• 跨平台兼容性

安装gfile

go get github.com/giongto35/gfile

基本使用示例

1. 发送文件

package main

import (
	"fmt"
	"github.com/giongto35/gfile"
	"log"
)

func main() {
	// 创建发送端
	sender, err := gfile.NewSender(gfile.Config{
		StunServers: []string{"stun:stun.l.google.com:19302"},
	})
	if err != nil {
		log.Fatal(err)
	}

	// 生成分享码
	code, err := sender.GenerateCode()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("分享码: %s\n", code)

	// 添加要发送的文件
	err = sender.AddFile("example.txt")
	if err != nil {
		log.Fatal(err)
	}

	// 等待接收方连接并传输文件
	err = sender.Send()
	if err != nil {
		log.Fatal(err)
	}
	
	fmt.Println("文件传输完成")
}

2. 接收文件

package main

import (
	"fmt"
	"github.com/giongto35/gfile"
	"log"
)

func main() {
	// 创建接收端
	receiver, err := gfile.NewReceiver(gfile.Config{
		StunServers: []string{"stun:stun.l.google.com:19302"},
	})
	if err != nil {
		log.Fatal(err)
	}

	// 输入分享码
	var code string
	fmt.Print("请输入分享码: ")
	fmt.Scanln(&code)

	// 连接发送方
	err = receiver.Connect(code)
	if err != nil {
		log.Fatal(err)
	}

	// 设置保存目录
	receiver.SetOutputDir("./downloads")

	// 接收文件
	err = receiver.Receive()
	if err != nil {
		log.Fatal(err)
	}
	
	fmt.Println("文件接收完成")
}

高级功能

1. 自定义配置

config := gfile.Config{
	StunServers: []string{
		"stun:stun1.l.google.com:19302",
		"stun:stun2.l.google.com:19302",
	},
	TurnServers: []gfile.TurnServer{
		{
			URLs:       []string{"turn:turn.example.com:3478"},
			Username:   "username",
			Credential: "password",
		},
	},
	ChunkSize:    1024 * 64, // 64KB的块大小
	Encryption:   true,      // 启用加密
	DataChannel:  "gfile",   // 数据通道名称
}

2. 进度监控

// 发送端进度监控
sender.SetProgressFunc(func(progress float64, speed float64) {
	fmt.Printf("进度: %.2f%%, 速度: %.2f KB/s\n", progress*100, speed/1024)
})

// 接收端进度监控
receiver.SetProgressFunc(func(progress float64, speed float64) {
	fmt.Printf("进度: %.2f%%, 速度: %.2f KB/s\n", progress*100, speed/1024)
})

3. 多文件传输

// 添加多个文件
err = sender.AddFiles([]string{"file1.txt", "file2.jpg", "document.pdf"})
if err != nil {
	log.Fatal(err)
}

// 或者添加整个目录
err = sender.AddDir("documents")
if err != nil {
	log.Fatal(err)
}

安全注意事项

1. 加密传输：gfile默认使用DTLS-SRTP进行加密，确保传输安全
2. 验证机制：分享码机制防止中间人攻击
3. 文件验证：传输完成后可进行哈希校验

// 启用SHA256校验
sender.SetVerifyHash(true)

// 接收端验证文件
if receiver.VerifyFile("example.txt") {
	fmt.Println("文件校验通过")
} else {
	fmt.Println("文件校验失败")
}

性能优化建议

1. 调整ChunkSize以适应不同网络环境
2. 使用可靠的TURN服务器作为NAT穿透后备方案
3. 对于大文件传输，考虑启用压缩

config := gfile.Config{
	ChunkSize:  1024 * 128, // 增大块大小提升传输效率
	Compress:   true,       // 启用压缩
	BufferSize: 1024 * 1024 * 5, // 5MB的缓冲区
}

错误处理

err = sender.Send()
if err != nil {
	switch e := err.(type) {
	case *gfile.ConnectionError:
		fmt.Println("连接错误:", e.Details)
	case *gfile.TransferError:
		fmt.Println("传输错误:", e.File, e.Details)
	default:
		fmt.Println("未知错误:", err)
	}
	return
}

gfile库为Go开发者提供了简单而强大的WebRTC文件传输能力，适用于需要安全P2P文件共享的场景。通过适当的配置和错误处理，可以构建出稳定可靠的文件传输应用。