quic-go实现屏幕广播程序

🕗 发布于 2024-10-10 02:38 golang 后端 quic-go

最近在折腾quic-go, 突然想起屏广适合用udp实现，而http3基于quic-go，后者又基于udp, 所以玩一下。

先贴出本机运行效果图：
在这里插入图片描述

功能(实现)说明：

1.服务器先启动作为共享屏幕方，等待客户端连接上来
2.客户端连接
3.客户端和服务器建立连接后，服务器主动打开stream

在一个for 循环中：每秒操作30次下面操作：

4.服务器开始抓取本机屏幕内容，转换成Image
5.数据传输协议：Image字节长度 + Image内容

6.客户端按上述协议接收数据，解析成Image对象，放界面上展示

服务端代码：

package main

import (
"bytes"
"context"
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"encoding/binary"
"encoding/pem"
"fmt"
"github.com/quic-go/quic-go"
"image"
"image/png"
"log"
"math/big"
"os"
"time"

"crypto/tls"
"github.com/kbinani/screenshot"
)

const addr = "localhost:4000"

var currentDir, _ = os.Getwd()

var quicConf = &quic.Config{
Allow0RTT:                      true,
MaxIdleTimeout:                 40 * time.Second,
InitialStreamReceiveWindow:     1 << 20,  // 1 MB
MaxStreamReceiveWindow:         6 << 20,  // 6 MB
InitialConnectionReceiveWindow: 2 << 20,  // 2 MB
MaxConnectionReceiveWindow:     12 << 20, // 12 MB
}

func main() {
//listener, err := quic.ListenAddr(addr, generateTLSConfig(), quicConf)
listener, err := quic.ListenAddr(addr, generateTLSConfig2(), quicConf)
if err != nil {
log.Fatal(err)
}
fmt.Println("Server listening on", addr)

for {
// 接受客户端连接
sess, err := listener.Accept(context.Background())
if err != nil {
log.Fatal(err)
}
fmt.Println("New client connected")
go handleConnection(sess)
}
}

func handleConnection(sess quic.Connection) {
stream, err := sess.OpenStream()
if err != nil {
log.Fatal(err)
}
fmt.Println("New stream opened:", stream.StreamID())
defer stream.Close()
var b []byte
for {
// 捕获桌面屏幕
img, err := captureScreen()
if err != nil {
log.Fatal(err)
}

// 将图像编码为 PNG 格式
var buf bytes.Buffer
err = png.Encode(&buf, img)
if err != nil {
log.Fatal(err)
}

// magic校验
//n, err := stream.Write([]byte{0x05, 0x19})
//if err != nil {
//log.Fatal(err)
//}
b = buf.Bytes()
//var headLenBuf = make([]byte, 4)
//binary.BigEndian.PutUint32(headLenBuf, uint32(len(b)))
//_, err = stream.Write(headLenBuf)
err = binary.Write(stream, binary.BigEndian, uint32(len(b)))
if err != nil {
log.Fatal(err)
}
// 将图像数据发送到客户端
_, err = stream.Write(b)
if err != nil {
log.Fatal(err)
}
// 每秒捕获并传输一帧
time.Sleep(1 * time.Second / 30)
}
}

func captureScreen() (image.Image, error) {
bounds := screenshot.GetDisplayBounds(0) // 捕获主屏幕
img, err := screenshot.CaptureRect(bounds)
if err != nil {
return nil, err
}
return img, nil
}

/*
*
openssl req -x509 -newkey rsa:4096 -keyout privkey.pem -out cert.pem -days 365 -nodes
*/
func generateTLSConfig() *tls.Config {
// 使用自签名证书
// goland运行使用它
cert, err := tls.LoadX509KeyPair(currentDir+"/screenbroadcast/cert.pem", currentDir+"/screenbroadcast/privkey.pem")
// 命令行运行使用它
//cert, err := tls.LoadX509KeyPair("cert.pem", "privkey.pem")
if err != nil {
log.Fatal(err)
}
return &tls.Config{
Certificates: []tls.Certificate{cert},
NextProtos:   []string{"h3-29"},
}
}

func generateTLSConfig2() *tls.Config {
key, err := rsa.GenerateKey(rand.Reader, 1024)
if err != nil {
panic(err)
}
template := x509.Certificate{SerialNumber: big.NewInt(1)}
certDER, err := x509.CreateCertificate(rand.Reader, &template, &template, &key.PublicKey, key)
if err != nil {
panic(err)
}
keyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(key)})
certPEM := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: certDER})

tlsCert, err := tls.X509KeyPair(certPEM, keyPEM)
if err != nil {
panic(err)
}
return &tls.Config{
Certificates: []tls.Certificate{tlsCert},
NextProtos:   []string{"h3-29"},
}
}

客户端代码：

package main

import (
"bytes"
"context"
"crypto/tls"
"encoding/binary"
"fmt"
"github.com/quic-go/quic-go"
"image"
"image/png"
"io"
"log"
"time"

"github.com/faiface/pixel"
"github.com/faiface/pixel/pixelgl"
)

const addr = "localhost:4000"

var headLenBuf = make([]byte, 4)

func main() {
pixelgl.Run(run)
}

func run() {
tlsConf := &tls.Config{
InsecureSkipVerify: true,
NextProtos:         []string{"h3-29"},
}

quicConfig := &quic.Config{
MaxIdleTimeout:  40 * time.Second,
KeepAlivePeriod: 30 * time.Second, // 使用quic的心跳机制
}
// 创建 QUIC 连接到服务器
sess, err := quic.DialAddr(context.Background(), addr, tlsConf, quicConfig)
if err != nil {
log.Fatal(err)
}

// 接收一个 QUIC stream：没错，是server主动推送数据过来，先发起的open stream
stream, err := sess.AcceptStream(context.Background())
if err != nil {
log.Fatal(err)
}

// 创建窗口显示接收的屏幕图像
cfg := pixelgl.WindowConfig{
Title:     "Screen Broadcast",
Bounds:    pixel.R(0, 0, 1024, 680),
VSync:     true,
Resizable: true,
}
win, err := pixelgl.NewWindow(cfg)
if err != nil {
log.Fatal(err)
}
for !win.Closed() {
// 接收图像数据
img, err := receiveImage(stream)
if err != nil {
if err == io.EOF {
break
}
log.Fatal(err)
}

// 将图像转换为 pixel.Picture
pic := pixel.PictureDataFromImage(img)

// 绘制图像
sprite := pixel.NewSprite(pic, pic.Bounds())
win.Clear(pixel.RGB(0, 0, 0))
sprite.Draw(win, pixel.IM.Moved(win.Bounds().Center()))
win.Update()
}
}

func receiveImage(stream quic.Stream) (image.Image, error) {
//_, err := io.ReadFull(stream, headLenBuf[:2])
//if err != nil {
//return nil, err
//}
//if headLenBuf[0] != 0x05 && headLenBuf[1] != 0x19 {
//return nil, errors.New("invalid magic")
//}
_, err := io.ReadFull(stream, headLenBuf)
if err != nil {
fmt.Println("video Error reading:", err.Error())
return nil, err
}
headLen := binary.BigEndian.Uint32(headLenBuf)
var buf bytes.Buffer
// 从 QUIC stream 读取图像数据
_, err = io.CopyN(&buf, stream, int64(headLen))
if err != nil {
return nil, err
}

// 解码 PNG 图像
img, err := png.Decode(&buf)
if err != nil {
return nil, err
}

return img, nil
}

下面开始说其中涉及到的坑：

当我本机(mac m1) OS版本为 12.1 时，运行服务器程序失败：

../../../../go/pkg/mod/github.com/kbinani/screenshot@v0.0.0-20240820160931-a8a2c5d0e191/darwin.go:9:10: fatal error:
'ScreenCaptureKit/ScreenCaptureKit.h' file not found
#include <ScreenCaptureKit/ScreenCaptureKit.h>

网上说升级系统到12.3+,因为ScreenCaptureKit 是 macOS 12.3 及更高版本中引入的 API，用于捕获屏幕内容。但是我升级到12.7.6后仍然报错…
在这里插入图片描述

然后看github.com/kbinani/screenshot源码：我当前下载的screenshot版本需要14.4+ ？

#if __ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ > MAC_OS_VERSION_14_4

FYI：我不敢升级到15版本，，，不敢。。。只是小版本升级

最后解决办法：使用低版本的screenshot：
去官网：https://pkg.go.dev/github.com/kbinani/screenshot@v0.0.0-20240820160931-a8a2c5d0e191/example?tab=versions
在这里插入图片描述
使用低版本的2023试试：

jelex@jelexxudeMacBook-Pro screenbroadcast % go get github.com/kbinani/screenshot@v0.0.0-20230831090513-3e604f0f372a

最后果然没问题了！

坑二：client程序无法交叉编译打包

在这里插入图片描述
我没有在windows电脑上验证，如果有使用windows版本的golang使用者看到本篇后，是否可以帮忙打包验证？

坑三：打包服务端程序成exe，在另一台电脑上运行，本机mac 作为客户端连接后没反应，直到超时报错退出：

2024/10/09 15:29:43 timeout: no recent network activity

是否有道友愿意联调？FYI: 我周边没有golang开发者，他们电脑上没安装golang环境…

或者有大佬知道这个问题能直接赐教吗？

原文地址：https://blog.csdn.net/csdnfanguyinheng/article/details/142787806

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：关于AOT、JIT编译模式以及基于静态编译构建微服务应用
下一篇：【SQL】换座位

Haproxy搭建Web群集
HAProxy是可提供高可用性、负载均衡以及基于TCP和HTTP应用的代理，是免费、快速并且可靠的一种解决方案。HAProxy非常适用于并发大（并发达1w以上）web站点，这些站点通常又需要会话保持或
阅读更多2024-10-10
【Flask】Flask数据库
大多数的数据库引擎都有对应的 Python 包，包括开源包和商业包。Flask 并不限制你使用何种类型的数据库包，因此可以根据自己的喜好选择使用 MySQL、Postgres、SQLite、Redis
阅读更多2024-10-10
车载SerDes历史和发展概述
介绍车载摄像头方案的历史、现状和发展
阅读更多2024-10-10
ARMv7/v8m GCC Compilation
00-Guide01-Toolchains02-GNU binutils03-ELF file04-Compilation process05-CFLAGS options06- Linker Scr
阅读更多2024-10-10
阿里云物联网自有app创建之初始化SDK
步骤一、新建工程，配置gradle,导入.so文件，生成apk。建立emptyActivity随便起命，但是注意要包名要以com.开头打开app的gradle，建立存储**.so动态链接库文件**的l
阅读更多2024-10-10
阿里云NAS之间迁移实践
阿里云提供的在线迁移服务是一种存储产品数据通道，客户有时需要在阿里云NAS之间进行数据迁移。本文档详细介绍了针对这一场景的相关内容。
阅读更多2024-10-10
linux（ubuntu22.04) 设置网络通信相关-配置指令
修改发送（TX）或接收（RX）队列大小。首先得到每个网络接口的名称。修改缓存区大小为25M。
阅读更多2024-10-10
SpringBoot 整合阿里云 OSS图片上传
‌阿里云OSS（Object Storage Service）是一种基于云存储的产品，适用于存储和管理各种类型的文件，包括图片、视频、文档等。‌ 阿里云OSS具有高可靠性、高可用性和低成本等优点，因此
阅读更多2024-10-10
Ubuntu下的CUDA环境的安装与配置
安装 NVIDIA 驱动：确保显卡驱动正确安装。安装 CUDA 工具包：下载并配置好 CUDA 工具包。配置环境变量：设置 CUDA 的PATH和。验证安装：使用nvcc和示例代码来验证安装是否成功。
阅读更多2024-10-10
Python 与 Pycharm 的简易安装教程，包含Pycharm的修改
红色框是选择适合自己电脑系统和版本的部分，蓝色框是选择系统的部分，黄色框是版本号。如果电脑没有获取到Python的位置，自己也忘记Python的位置，可以 win + R 打开cmd，输入以下代码查看
阅读更多2024-10-10

quic-go实现屏幕广播程序

最近在折腾quic-go, 突然想起屏广适合用udp实现，而http3基于quic-go，后者又基于udp, 所以玩一下。

功能(实现)说明：

坑二：client程序无法交叉编译打包

坑三：打包服务端程序成exe，在另一台电脑上运行，本机mac 作为客户端连接后没反应，直到超时报错退出：

相关文章