mirror of
https://github.com/Monibuca/plugin-webrtc.git
synced 2025-10-31 02:26:49 +08:00
初步连接成功
This commit is contained in:
langhuihui
223
main.go
223
main.go
@@ -4,11 +4,11 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
. "github.com/Monibuca/engine/v2"
|
||||
rtsp "github.com/Monibuca/plugin-rtsp"
|
||||
"github.com/pion/rtcp"
|
||||
. "github.com/pion/webrtc/v2"
|
||||
)
|
||||
@@ -16,20 +16,15 @@ import (
|
||||
var config = &struct {
|
||||
ICEServers []string
|
||||
}{[]string{
|
||||
"stun.l.google.com:19302",
|
||||
"stun1.l.google.com:19302",
|
||||
"stun2.l.google.com:19302",
|
||||
"stun3.l.google.com:19302",
|
||||
"stun4.l.google.com:19302",
|
||||
"stun.ekiga.net",
|
||||
"stun.ideasip.com",
|
||||
"stun.schlund.de",
|
||||
"stun.stunprotocol.org:3478",
|
||||
"stun.voiparound.com",
|
||||
"stun.voipbuster.com",
|
||||
"stun.voipstunt.com",
|
||||
"stun.voxgratia.org",
|
||||
"stun.services.mozilla.com",
|
||||
"stun:stun.ekiga.net",
|
||||
"stun:stun.ideasip.com",
|
||||
"stun:stun.schlund.de",
|
||||
"stun:stun.stunprotocol.org:3478",
|
||||
"stun:stun.voiparound.com",
|
||||
"stun:stun.voipbuster.com",
|
||||
"stun:stun.voipstunt.com",
|
||||
"stun:stun.voxgratia.org",
|
||||
"stun:stun.services.mozilla.com",
|
||||
"stun:stun.xten.com",
|
||||
"stun:stun.softjoys.com",
|
||||
"stun:stunserver.org",
|
||||
@@ -42,12 +37,18 @@ var config = &struct {
|
||||
"stun:stun01.sipphone.com",
|
||||
}}
|
||||
|
||||
type udpConn struct {
|
||||
conn *net.UDPConn
|
||||
port int
|
||||
}
|
||||
// type udpConn struct {
|
||||
// conn *net.UDPConn
|
||||
// port int
|
||||
// }
|
||||
|
||||
var m = MediaEngine{}
|
||||
var api *API
|
||||
|
||||
func init() {
|
||||
m.RegisterCodec(NewRTPH264Codec(DefaultPayloadTypeH264, 90000))
|
||||
//m.RegisterCodec(NewRTPPCMUCodec(DefaultPayloadTypePCMU, 8000))
|
||||
api = NewAPI(WithMediaEngine(m))
|
||||
InstallPlugin(&PluginConfig{
|
||||
Config: config,
|
||||
Name: "WebRTC",
|
||||
@@ -55,10 +56,13 @@ func init() {
|
||||
Run: run,
|
||||
})
|
||||
}
|
||||
func run() {
|
||||
m := MediaEngine{}
|
||||
m.RegisterCodec(NewRTPH264Codec(DefaultPayloadTypeH264, 90000))
|
||||
api := NewAPI(WithMediaEngine(m))
|
||||
|
||||
type WebRTC struct {
|
||||
rtsp.RTSP
|
||||
*PeerConnection
|
||||
}
|
||||
|
||||
func (rtc *WebRTC) Publish(streamPath string) bool {
|
||||
peerConnection, err := api.NewPeerConnection(Configuration{
|
||||
ICEServers: []ICEServer{
|
||||
{
|
||||
@@ -66,96 +70,15 @@ func run() {
|
||||
},
|
||||
},
|
||||
})
|
||||
if _, err = peerConnection.AddTransceiverFromKind(RTPCodecTypeVideo); err != nil {
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return false
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return
|
||||
return false
|
||||
}
|
||||
// Allow us to receive 1 audio track, and 1 video track
|
||||
if _, err = peerConnection.AddTransceiverFromKind(RTPCodecTypeAudio); err != nil {
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
} else if _, err = peerConnection.AddTransceiverFromKind(RTPCodecTypeVideo); err != nil {
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
}
|
||||
var laddr *net.UDPAddr
|
||||
if laddr, err = net.ResolveUDPAddr("udp", "127.0.0.1:"); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Prepare udp conns
|
||||
udpConns := map[string]*udpConn{
|
||||
"audio": {port: 4000},
|
||||
"video": {port: 4002},
|
||||
}
|
||||
for _, c := range udpConns {
|
||||
// Create remote addr
|
||||
var raddr *net.UDPAddr
|
||||
if raddr, err = net.ResolveUDPAddr("udp", fmt.Sprintf("127.0.0.1:%d", c.port)); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Dial udp
|
||||
if c.conn, err = net.DialUDP("udp", laddr, raddr); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer func(conn net.PacketConn) {
|
||||
if closeErr := conn.Close(); closeErr != nil {
|
||||
panic(closeErr)
|
||||
}
|
||||
}(c.conn)
|
||||
}
|
||||
|
||||
// Set a handler for when a new remote track starts, this handler will forward data to
|
||||
// our UDP listeners.
|
||||
// In your application this is where you would handle/process audio/video
|
||||
peerConnection.OnTrack(func(track *Track, receiver *RTPReceiver) {
|
||||
// Retrieve udp connection
|
||||
c, ok := udpConns[track.Kind().String()]
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
// Send a PLI on an interval so that the publisher is pushing a keyframe every rtcpPLIInterval
|
||||
go func() {
|
||||
ticker := time.NewTicker(time.Second * 2)
|
||||
for range ticker.C {
|
||||
if rtcpErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: track.SSRC()}}); rtcpErr != nil {
|
||||
fmt.Println(rtcpErr)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
b := make([]byte, 1500)
|
||||
for {
|
||||
// Read
|
||||
n, readErr := track.Read(b)
|
||||
if readErr != nil {
|
||||
panic(readErr)
|
||||
}
|
||||
|
||||
// Write
|
||||
if _, err = c.conn.Write(b[:n]); err != nil {
|
||||
// For this particular example, third party applications usually timeout after a short
|
||||
// amount of time during which the user doesn't have enough time to provide the answer
|
||||
// to the browser.
|
||||
// That's why, for this particular example, the user first needs to provide the answer
|
||||
// to the browser then open the third party application. Therefore we must not kill
|
||||
// the forward on "connection refused" errors
|
||||
if opError, ok := err.(*net.OpError); ok && opError.Err.Error() == "write: connection refused" {
|
||||
continue
|
||||
}
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
// Set the handler for ICE connection state
|
||||
// This will notify you when the peer has connected/disconnected
|
||||
peerConnection.OnICEConnectionStateChange(func(connectionState ICEConnectionState) {
|
||||
fmt.Printf("Connection State has changed %s \n", connectionState.String())
|
||||
|
||||
@@ -167,33 +90,75 @@ func run() {
|
||||
//TODO
|
||||
}
|
||||
})
|
||||
rtc.PeerConnection = peerConnection
|
||||
if rtc.RTSP.Publish(streamPath) {
|
||||
rtc.Stream.Type = "WebRTC"
|
||||
peerConnection.OnTrack(func(track *Track, receiver *RTPReceiver) {
|
||||
defer rtc.Close()
|
||||
go func() {
|
||||
ticker := time.NewTicker(time.Second * 2)
|
||||
select {
|
||||
case <-ticker.C:
|
||||
if rtcpErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: track.SSRC()}}); rtcpErr != nil {
|
||||
fmt.Println(rtcpErr)
|
||||
}
|
||||
case <-rtc.Done():
|
||||
return
|
||||
}
|
||||
}()
|
||||
pack := &rtsp.RTPPack{
|
||||
Type: rtsp.RTPType(track.Kind() - 1),
|
||||
}
|
||||
for b := make([]byte, 1460); ; rtc.HandleRTP(pack) {
|
||||
i, err := track.Read(b)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if err = pack.Unmarshal(b[:i]); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
})
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
func run() {
|
||||
http.HandleFunc("/webrtc/answer", func(w http.ResponseWriter, r *http.Request) {
|
||||
// Wait for the offer to be pasted
|
||||
streamPath := r.URL.Query().Get("streamPath")
|
||||
offer := SessionDescription{}
|
||||
bytes, err := ioutil.ReadAll(r.Body)
|
||||
err = json.Unmarshal(bytes, &offer)
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
// Set the remote SessionDescription
|
||||
if err = peerConnection.SetRemoteDescription(offer); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
rtc := new(WebRTC)
|
||||
if rtc.Publish(streamPath) {
|
||||
// Set the remote SessionDescription
|
||||
if err = rtc.SetRemoteDescription(offer); err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Create answer
|
||||
answer, err := peerConnection.CreateAnswer(nil)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
// Create answer
|
||||
answer, err := rtc.CreateAnswer(nil)
|
||||
if err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
|
||||
// Sets the LocalDescription, and starts our UDP listeners
|
||||
if err = peerConnection.SetLocalDescription(answer); err != nil {
|
||||
panic(err)
|
||||
// Sets the LocalDescription, and starts our UDP listeners
|
||||
if err = rtc.SetLocalDescription(answer); err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
if bytes, err = json.Marshal(answer); err != nil {
|
||||
Println(err)
|
||||
return
|
||||
}
|
||||
w.Write(bytes)
|
||||
}
|
||||
bytes, err = json.Marshal(answer)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
w.Write(bytes)
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user