使用gortsplib实现Go语言RTSP服务器与客户端

gortsplib是一个纯Go语言实现的RTSP协议库，可以用于开发RTSP服务器和客户端。下面我将介绍如何使用这个库来实现基本的RTSP功能。

安装gortsplib

首先安装gortsplib库：

go get github.com/aler9/gortsplib/v2

RTSP服务器实现

下面是一个简单的RTSP服务器实现示例：

package main

import (
	"log"
	"time"

	"github.com/aler9/gortsplib/v2"
	"github.com/aler9/gortsplib/v2/pkg/base"
	"github.com/aler9/gortsplib/v2/pkg/format"
	"github.com/aler9/gortsplib/v2/pkg/media"
	"github.com/pion/rtp"
)

type customMedia struct {
	medias []*media.Media
}

func (m *customMedia) Medias() []*media.Media {
	return m.medias
}

func (m *customMedia) OnSetup(*gortsplib.ServerHandlerOnSetupCtx) (*base.Response, error) {
	return &base.Response{
		StatusCode: base.StatusOK,
	}, nil
}

func (m *customMedia) OnPlay(*gortsplib.ServerHandlerOnPlayCtx) (*base.Response, error) {
	return &base.Response{
		StatusCode: base.StatusOK,
	}, nil
}

func main() {
	// 创建H264视频格式
	h264Format := &format.H264{
		PayloadTyp:        96,
		SPS:               []byte{0x67, 0x42, 0xc0, 0x28, 0xd9, 0x00, 0x78, 0x02, 0x27, 0xe5, 0x84, 0x00, 0x00, 0x03, 0x00, 0x04, 0x00, 0x00, 0x03, 0x00, 0xf0, 0x3c, 0x60, 0xc9, 0x20},
		PPS:               []byte{0x68, 0xce, 0x38, 0x80},
		PacketizationMode: 1,
	}
	
	// 创建媒体流
	medi := &media.Media{
		Type:    media.TypeVideo,
		Formats: []format.Format{h264Format},
	}
	
	// 创建服务器
	s := &gortsplib.Server{
		Handler: &customMedia{
			medias: []*media.Media{medi},
		},
		RTSPAddress: ":8554",
	}
	
	// 启动服务器
	log.Println("RTSP server is running on rtsp://localhost:8554")
	err := s.Start()
	if err != nil {
		panic(err)
	}
	defer s.Close()

	// 模拟发送RTP包
	track := &gortsplib.TrackH264{
		PayloadType: 96,
		SPS:         h264Format.SPS,
		PPS:         h264Format.PPS,
	}
	
	// 创建会话
	session := gortsplib.NewServerSession(s, &gortsplib.Track{Media: medi})
	
	// 定期发送RTP包
	ticker := time.NewTicker(time.Millisecond * 33)
	defer ticker.Stop()
	
	for range ticker.C {
		pkt := &rtp.Packet{
			Header: rtp.Header{
				Version:        2,
				PayloadType:    96,
				SequenceNumber: 12345,
				Timestamp:      54321,
				SSRC:           uint32(time.Now().UnixNano()),
			},
			Payload: []byte{0x05, 0x00, 0x00, 0x00}, // 模拟H264 NALU
		}
		
		err := session.WritePacketRTP(medi, pkt)
		if err != nil {
			log.Println("write packet error:", err)
			break
		}
	}
}

RTSP客户端实现

下面是一个简单的RTSP客户端实现示例：

package main

import (
	"log"
	"time"

	"github.com/aler9/gortsplib/v2"
	"github.com/aler9/gortsplib/v2/pkg/base"
	"github.com/aler9/gortsplib/v2/pkg/format"
	"github.com/pion/rtp"
)

type clientHandler struct {
	client *gortsplib.Client
}

func (h *clientHandler) OnPacketRTP(medi *media.Media, forma format.Format, pkt *rtp.Packet) {
	log.Printf("RTP packet: media=%v format=%v packet=%v\n", medi, forma, pkt)
}

func main() {
	// RTSP服务器地址
	address := "rtsp://localhost:8554"

	// 创建客户端
	client := &gortsplib.Client{
		OnPacketRTP: (&clientHandler{}).OnPacketRTP,
	}

	// 连接到服务器
	u, err := base.ParseURL(address)
	if err != nil {
		panic(err)
	}

	err = client.Start(u.Scheme, u.Host)
	if err != nil {
		panic(err)
	}
	defer client.Close()

	// 获取可用的媒体描述
	desc, _, err := client.Describe(u)
	if err != nil {
		panic(err)
	}

	// 设置媒体格式
	err = client.SetupAll(desc.BaseURL, desc.Medias)
	if err != nil {
		panic(err)
	}

	// 开始播放
	_, err = client.Play(nil)
	if err != nil {
		panic(err)
	}

	log.Println("Client is receiving RTP packets...")

	// 保持运行
	time.Sleep(time.Minute)
}

高级功能

1. 认证支持

服务器端可以添加认证：

s := &gortsplib.Server{
	Handler: &customMedia{
		medias: []*media.Media{medi},
	},
	RTSPAddress: ":8554",
	AuthMethods: []base.AuthMethod{base.AuthBasic},
	AuthUser:    "admin",
	AuthPass:    "password",
}

客户端连接时需要提供认证信息：

client := &gortsplib.Client{
	OnPacketRTP: (&clientHandler{}).OnPacketRTP,
	AnyAuth: &base.Auth{
		Method: base.AuthBasic,
		User:   "admin",
		Pass:   "password",
	},
}

2. 支持TCP传输

默认情况下，gortsplib使用UDP传输RTP数据。如果需要使用TCP：

client := &gortsplib.Client{
	Transport: gortsplib.TransportTCP,
	OnPacketRTP: (&clientHandler{}).OnPacketRTP,
}

3. 录制RTSP流

可以将接收到的RTP包保存到文件：

type recorderHandler struct {
	file *os.File
}

func (h *recorderHandler) OnPacketRTP(medi *media.Media, forma format.Format, pkt *rtp.Packet) {
	// 写入RTP包头
	h.file.Write(pkt.Header.Marshal())
	// 写入RTP负载
	h.file.Write(pkt.Payload)
}

func main() {
	// 创建文件
	file, err := os.Create("stream.rtp")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	// 创建客户端
	client := &gortsplib.Client{
		OnPacketRTP: (&recorderHandler{file: file}).OnPacketRTP,
	}
	
	// ...其余客户端代码...
}

总结

gortsplib是一个功能强大且易于使用的RTSP库，完全用Go语言实现。通过上面的示例，你可以快速实现RTSP服务器和客户端的基本功能。该库支持：

1. RTSP协议的核心功能（DESCRIBE, SETUP, PLAY, TEARDOWN等）
2. 多种传输协议（UDP, TCP）
3. 多种媒体格式（H264, H265, MPEG-4, AAC等）
4. 认证支持

对于更复杂的需求，你可以参考gortsplib的官方文档和示例代码进行扩展。