|
|
|
@ -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,134 +70,95 @@ func run() { |
|
|
|
}, |
|
|
|
}, |
|
|
|
}) |
|
|
|
if _, err = peerConnection.AddTransceiverFromKind(RTPCodecTypeVideo); err != nil { |
|
|
|
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) |
|
|
|
return false |
|
|
|
} |
|
|
|
peerConnection.OnICEConnectionStateChange(func(connectionState ICEConnectionState) { |
|
|
|
fmt.Printf("Connection State has changed %s \n", connectionState.String()) |
|
|
|
|
|
|
|
// 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) |
|
|
|
if connectionState == ICEConnectionStateConnected { |
|
|
|
fmt.Println("Ctrl+C the remote client to stop the demo") |
|
|
|
} else if connectionState == ICEConnectionStateFailed || |
|
|
|
connectionState == ICEConnectionStateDisconnected { |
|
|
|
fmt.Println("Done forwarding") |
|
|
|
//TODO
|
|
|
|
} |
|
|
|
|
|
|
|
// 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
|
|
|
|
}) |
|
|
|
rtc.PeerConnection = peerConnection |
|
|
|
if rtc.RTSP.Publish(streamPath) { |
|
|
|
rtc.Stream.Type = "WebRTC" |
|
|
|
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
|
|
|
|
defer rtc.Close() |
|
|
|
go func() { |
|
|
|
ticker := time.NewTicker(time.Second * 2) |
|
|
|
for range ticker.C { |
|
|
|
select { |
|
|
|
case <-ticker.C: |
|
|
|
if rtcpErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: track.SSRC()}}); rtcpErr != nil { |
|
|
|
fmt.Println(rtcpErr) |
|
|
|
} |
|
|
|
case <-rtc.Done(): |
|
|
|
return |
|
|
|
} |
|
|
|
}() |
|
|
|
|
|
|
|
b := make([]byte, 1500) |
|
|
|
for { |
|
|
|
// Read
|
|
|
|
n, readErr := track.Read(b) |
|
|
|
if readErr != nil { |
|
|
|
panic(readErr) |
|
|
|
pack := &rtsp.RTPPack{ |
|
|
|
Type: rtsp.RTPType(track.Kind() - 1), |
|
|
|
} |
|
|
|
|
|
|
|
// 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 |
|
|
|
for b := make([]byte, 1460); ; rtc.HandleRTP(pack) { |
|
|
|
i, err := track.Read(b) |
|
|
|
if err != nil { |
|
|
|
return |
|
|
|
} |
|
|
|
panic(err) |
|
|
|
if err = pack.Unmarshal(b[:i]); err != nil { |
|
|
|
return |
|
|
|
} |
|
|
|
} |
|
|
|
}) |
|
|
|
|
|
|
|
// 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()) |
|
|
|
|
|
|
|
if connectionState == ICEConnectionStateConnected { |
|
|
|
fmt.Println("Ctrl+C the remote client to stop the demo") |
|
|
|
} else if connectionState == ICEConnectionStateFailed || |
|
|
|
connectionState == ICEConnectionStateDisconnected { |
|
|
|
fmt.Println("Done forwarding") |
|
|
|
//TODO
|
|
|
|
} 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 |
|
|
|
} |
|
|
|
rtc := new(WebRTC) |
|
|
|
if rtc.Publish(streamPath) { |
|
|
|
// Set the remote SessionDescription
|
|
|
|
if err = peerConnection.SetRemoteDescription(offer); err != nil { |
|
|
|
panic(err) |
|
|
|
if err = rtc.SetRemoteDescription(offer); err != nil { |
|
|
|
Println(err) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// Create answer
|
|
|
|
answer, err := peerConnection.CreateAnswer(nil) |
|
|
|
answer, err := rtc.CreateAnswer(nil) |
|
|
|
if err != nil { |
|
|
|
panic(err) |
|
|
|
Println(err) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// Sets the LocalDescription, and starts our UDP listeners
|
|
|
|
if err = peerConnection.SetLocalDescription(answer); err != nil { |
|
|
|
panic(err) |
|
|
|
if err = rtc.SetLocalDescription(answer); err != nil { |
|
|
|
Println(err) |
|
|
|
return |
|
|
|
} |
|
|
|
bytes, err = json.Marshal(answer) |
|
|
|
if err != nil { |
|
|
|
panic(err) |
|
|
|
if bytes, err = json.Marshal(answer); err != nil { |
|
|
|
Println(err) |
|
|
|
return |
|
|
|
} |
|
|
|
w.Write(bytes) |
|
|
|
} |
|
|
|
}) |
|
|
|
} |
|
|
|
|
langhuihui
5 years ago