mirror of
https://github.com/aler9/gortsplib
synced 2025-10-05 15:16:51 +08:00
cabfcd627b
this fixes the client-read-republish example and allows users to set their own media controls. Function medias.SetControls() is provided in order to automatically set media controls and must be called manually.
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/aler9/gortsplib/v2"
|
|
"github.com/aler9/gortsplib/v2/pkg/format"
|
|
"github.com/aler9/gortsplib/v2/pkg/media"
|
|
"github.com/pion/rtp"
|
|
)
|
|
|
|
// This example shows how to
|
|
// 1. set additional client options
|
|
// 2. generate RTP/H264 frames from a file with GStreamer
|
|
// 3. connect to a RTSP server, announce an H264 media
|
|
// 4. write the frames to the server
|
|
|
|
func main() {
|
|
// open a listener to receive RTP/H264 frames
|
|
pc, err := net.ListenPacket("udp", "localhost:9000")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer pc.Close()
|
|
|
|
log.Println("Waiting for a RTP/H264 stream on port 9000 - you can send one with GStreamer:\n" +
|
|
"gst-launch-1.0 filesrc location=video.mp4 ! qtdemux ! video/x-h264" +
|
|
" ! h264parse config-interval=1 ! rtph264pay ! udpsink host=127.0.0.1 port=9000")
|
|
|
|
// wait for first packet
|
|
buf := make([]byte, 2048)
|
|
n, _, err := pc.ReadFrom(buf)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
log.Println("stream connected")
|
|
|
|
// create a media that contains a H264 media
|
|
medias := media.Medias{&media.Media{
|
|
Type: media.TypeVideo,
|
|
Formats: []format.Format{&format.H264{
|
|
PayloadTyp: 96,
|
|
PacketizationMode: 1,
|
|
}},
|
|
}}
|
|
medias.SetControls()
|
|
|
|
// Client allows to set additional client options
|
|
c := &gortsplib.Client{
|
|
// the stream transport (UDP or TCP). If nil, it is chosen automatically
|
|
Transport: nil,
|
|
// timeout of read operations
|
|
ReadTimeout: 10 * time.Second,
|
|
// timeout of write operations
|
|
WriteTimeout: 10 * time.Second,
|
|
}
|
|
|
|
// connect to the server and start recording the media
|
|
err = c.StartRecording("rtsp://localhost:8554/mystream", medias)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer c.Close()
|
|
|
|
var pkt rtp.Packet
|
|
for {
|
|
// parse RTP packet
|
|
err = pkt.Unmarshal(buf[:n])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// route RTP packet to the server
|
|
err = c.WritePacketRTP(medias[0], &pkt)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// read another RTP packet from source
|
|
n, _, err = pc.ReadFrom(buf)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|