|
|
|
@ -1,41 +1,49 @@ |
|
|
|
package webrtc |
|
|
|
|
|
|
|
import ( |
|
|
|
"bytes" |
|
|
|
"encoding/json" |
|
|
|
"fmt" |
|
|
|
"io/ioutil" |
|
|
|
"net/http" |
|
|
|
"sync" |
|
|
|
"time" |
|
|
|
|
|
|
|
. "github.com/Monibuca/engine/v2" |
|
|
|
"github.com/Monibuca/engine/v2/avformat" |
|
|
|
"github.com/Monibuca/engine/v2/util" |
|
|
|
. "github.com/Monibuca/plugin-rtp" |
|
|
|
"github.com/pion/rtcp" |
|
|
|
"github.com/pion/rtp" |
|
|
|
"github.com/pion/rtp/codecs" |
|
|
|
. "github.com/pion/webrtc/v2" |
|
|
|
) |
|
|
|
|
|
|
|
var config = &struct { |
|
|
|
var config struct { |
|
|
|
ICEServers []string |
|
|
|
}{[]string{ |
|
|
|
"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", |
|
|
|
"stun:stun.schlund.de", |
|
|
|
"stun:stun.rixtelecom.se", |
|
|
|
"stun:stun.iptel.org", |
|
|
|
"stun:stun.ideasip.com", |
|
|
|
"stun:stun.fwdnet.net", |
|
|
|
"stun:stun.ekiga.net", |
|
|
|
"stun:stun01.sipphone.com", |
|
|
|
}} |
|
|
|
} |
|
|
|
|
|
|
|
// }{[]string{
|
|
|
|
// "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",
|
|
|
|
// "stun:stun.schlund.de",
|
|
|
|
// "stun:stun.rixtelecom.se",
|
|
|
|
// "stun:stun.iptel.org",
|
|
|
|
// "stun:stun.ideasip.com",
|
|
|
|
// "stun:stun.fwdnet.net",
|
|
|
|
// "stun:stun.ekiga.net",
|
|
|
|
// "stun:stun01.sipphone.com",
|
|
|
|
// }}
|
|
|
|
|
|
|
|
// type udpConn struct {
|
|
|
|
// conn *net.UDPConn
|
|
|
|
@ -44,13 +52,16 @@ var config = &struct { |
|
|
|
|
|
|
|
var m MediaEngine |
|
|
|
var api *API |
|
|
|
var SSRC uint32 |
|
|
|
var SSRCMap = make(map[string]uint32) |
|
|
|
var ssrcLock sync.Mutex |
|
|
|
|
|
|
|
func init() { |
|
|
|
m.RegisterCodec(NewRTPH264Codec(DefaultPayloadTypeH264, 90000)) |
|
|
|
//m.RegisterCodec(NewRTPPCMUCodec(DefaultPayloadTypePCMU, 8000))
|
|
|
|
api = NewAPI(WithMediaEngine(m)) |
|
|
|
InstallPlugin(&PluginConfig{ |
|
|
|
Config: config, |
|
|
|
Config: &config, |
|
|
|
Name: "WebRTC", |
|
|
|
Type: PLUGIN_PUBLISHER | PLUGIN_SUBSCRIBER, |
|
|
|
Run: run, |
|
|
|
@ -60,9 +71,10 @@ func init() { |
|
|
|
type WebRTC struct { |
|
|
|
RTP |
|
|
|
*PeerConnection |
|
|
|
RemoteAddr string |
|
|
|
} |
|
|
|
|
|
|
|
func (rtc *WebRTC) Publish(streamPath string) bool { |
|
|
|
func (rtc *WebRTC) Play(streamPath string) bool { |
|
|
|
peerConnection, err := api.NewPeerConnection(Configuration{ |
|
|
|
ICEServers: []ICEServer{ |
|
|
|
{ |
|
|
|
@ -79,15 +91,170 @@ func (rtc *WebRTC) Publish(streamPath string) bool { |
|
|
|
if err != nil { |
|
|
|
return false |
|
|
|
} |
|
|
|
|
|
|
|
rtc.PeerConnection = peerConnection |
|
|
|
// Create a video track, using the same SSRC as the incoming RTP Packet
|
|
|
|
ssrcLock.Lock() |
|
|
|
ssrc, ok := SSRCMap[streamPath] |
|
|
|
if !ok { |
|
|
|
SSRC++ |
|
|
|
ssrc = SSRC |
|
|
|
SSRCMap[streamPath] = SSRC |
|
|
|
} |
|
|
|
ssrcLock.Unlock() |
|
|
|
videoTrack, err := peerConnection.NewTrack(DefaultPayloadTypeH264, ssrc, "video", "monibuca") |
|
|
|
if err != nil { |
|
|
|
Println(err) |
|
|
|
return false |
|
|
|
} |
|
|
|
if _, err = peerConnection.AddTrack(videoTrack); err != nil { |
|
|
|
Println(err) |
|
|
|
return false |
|
|
|
} |
|
|
|
peerConnection.OnICEConnectionStateChange(func(connectionState ICEConnectionState) { |
|
|
|
Printf("%s Connection State has changed %s ", streamPath, connectionState.String()) |
|
|
|
switch connectionState { |
|
|
|
case ICEConnectionStateDisconnected: |
|
|
|
rtc.Stream.Close() |
|
|
|
if rtc.Stream != nil { |
|
|
|
rtc.Stream.Close() |
|
|
|
} |
|
|
|
case ICEConnectionStateConnected: |
|
|
|
var sequence uint16 |
|
|
|
var sub Subscriber |
|
|
|
sub.ID = rtc.RemoteAddr |
|
|
|
sub.Type = "WebRTC" |
|
|
|
nextHeader := func(ts uint32, marker bool) rtp.Header { |
|
|
|
sequence++ |
|
|
|
return rtp.Header{ |
|
|
|
Version: 2, |
|
|
|
SSRC: ssrc, |
|
|
|
PayloadType: DefaultPayloadTypeH264, |
|
|
|
SequenceNumber: sequence, |
|
|
|
Timestamp: ts, |
|
|
|
Marker: marker, |
|
|
|
} |
|
|
|
} |
|
|
|
stapA := func(naul ...[]byte) []byte { |
|
|
|
var buffer bytes.Buffer |
|
|
|
buffer.WriteByte(24) |
|
|
|
for _, n := range naul { |
|
|
|
l := len(n) |
|
|
|
buffer.WriteByte(byte(l >> 8)) |
|
|
|
buffer.WriteByte(byte(l)) |
|
|
|
buffer.Write(n) |
|
|
|
} |
|
|
|
return buffer.Bytes() |
|
|
|
} |
|
|
|
|
|
|
|
// aud := []byte{0x09, 0x30}
|
|
|
|
sub.OnData = func(packet *avformat.SendPacket) error { |
|
|
|
if packet.Type == avformat.FLV_TAG_TYPE_AUDIO { |
|
|
|
return nil |
|
|
|
} |
|
|
|
if packet.IsSequence { |
|
|
|
payload := packet.Payload[11:] |
|
|
|
spsLen := int(payload[0])<<8 + int(payload[1]) |
|
|
|
sps := payload[2:spsLen] |
|
|
|
payload = payload[3+spsLen:] |
|
|
|
ppsLen := int(payload[0])<<8 + int(payload[1]) |
|
|
|
pps := payload[2:ppsLen] |
|
|
|
if err := videoTrack.WriteRTP(&rtp.Packet{ |
|
|
|
Header: nextHeader(0, false), |
|
|
|
Payload: stapA(sps, pps), |
|
|
|
}); err != nil { |
|
|
|
return err |
|
|
|
} |
|
|
|
} else { |
|
|
|
payload := packet.Payload[5:] |
|
|
|
for { |
|
|
|
var naulLen = int(util.BigEndian.Uint32(payload)) |
|
|
|
payload = payload[4:] |
|
|
|
_payload := payload[:naulLen] |
|
|
|
if naulLen > 1000 { |
|
|
|
part := _payload[:1000] |
|
|
|
indicator := ((part[0] >> 5) << 5) | 28 |
|
|
|
nalutype := part[0] & 31 |
|
|
|
header := 128 | nalutype |
|
|
|
part = part[1:] |
|
|
|
marker := false |
|
|
|
for { |
|
|
|
if err := videoTrack.WriteRTP(&rtp.Packet{ |
|
|
|
Header: nextHeader(packet.Timestamp*90, marker), |
|
|
|
Payload: append([]byte{indicator, header}, part...), |
|
|
|
}); err != nil { |
|
|
|
return err |
|
|
|
} |
|
|
|
marker = true |
|
|
|
if _payload == nil { |
|
|
|
break |
|
|
|
} |
|
|
|
if len(_payload[1000:]) <= 1000 { |
|
|
|
header = 64 | nalutype |
|
|
|
part = _payload[1000:] |
|
|
|
_payload = nil |
|
|
|
} else { |
|
|
|
header = nalutype |
|
|
|
part = _payload[1000:] |
|
|
|
_payload = part |
|
|
|
} |
|
|
|
} |
|
|
|
} else { |
|
|
|
if err := videoTrack.WriteRTP(&rtp.Packet{ |
|
|
|
Header: nextHeader(packet.Timestamp*90, false), |
|
|
|
Payload: _payload, |
|
|
|
}); err != nil { |
|
|
|
return err |
|
|
|
} |
|
|
|
} |
|
|
|
if len(payload) < naulLen+4 { |
|
|
|
break |
|
|
|
} |
|
|
|
payload = payload[naulLen:] |
|
|
|
} |
|
|
|
// if err := videoTrack.WriteRTP(&rtp.Packet{
|
|
|
|
// Header: nextHeader(packet.Timestamp * 90),
|
|
|
|
// Payload: aud,
|
|
|
|
// }); err != nil {
|
|
|
|
// return err
|
|
|
|
// }
|
|
|
|
} |
|
|
|
return nil |
|
|
|
} |
|
|
|
sub.Subscribe(streamPath) |
|
|
|
} |
|
|
|
}) |
|
|
|
return true |
|
|
|
} |
|
|
|
func (rtc *WebRTC) Publish(streamPath string) bool { |
|
|
|
peerConnection, err := api.NewPeerConnection(Configuration{ |
|
|
|
ICEServers: []ICEServer{ |
|
|
|
{ |
|
|
|
URLs: config.ICEServers, |
|
|
|
}, |
|
|
|
}, |
|
|
|
}) |
|
|
|
if _, err = peerConnection.AddTransceiverFromKind(RTPCodecTypeVideo); err != nil { |
|
|
|
if err != nil { |
|
|
|
Println(err) |
|
|
|
return false |
|
|
|
} |
|
|
|
} |
|
|
|
if err != nil { |
|
|
|
return false |
|
|
|
} |
|
|
|
peerConnection.OnICEConnectionStateChange(func(connectionState ICEConnectionState) { |
|
|
|
Printf("%s Connection State has changed %s ", streamPath, connectionState.String()) |
|
|
|
switch connectionState { |
|
|
|
case ICEConnectionStateDisconnected, ICEConnectionStateFailed: |
|
|
|
if rtc.Stream != nil { |
|
|
|
rtc.Stream.Close() |
|
|
|
} |
|
|
|
} |
|
|
|
}) |
|
|
|
rtc.PeerConnection = peerConnection |
|
|
|
if rtc.RTP.Publish(streamPath) { |
|
|
|
//f, _ := os.OpenFile("resource/live/rtc.h264", os.O_TRUNC|os.O_WRONLY, 0666)
|
|
|
|
var h264 codecs.H264Packet |
|
|
|
rtc.Stream.Type = "WebRTC" |
|
|
|
peerConnection.OnTrack(func(track *Track, receiver *RTPReceiver) { |
|
|
|
defer rtc.Stream.Close() |
|
|
|
@ -113,6 +280,8 @@ func (rtc *WebRTC) Publish(streamPath string) bool { |
|
|
|
if err = pack.Unmarshal(b[:i]); err != nil { |
|
|
|
return |
|
|
|
} |
|
|
|
h264.Unmarshal(pack.Payload) |
|
|
|
// f.Write(bytes)
|
|
|
|
} |
|
|
|
}) |
|
|
|
} else { |
|
|
|
@ -120,8 +289,32 @@ func (rtc *WebRTC) Publish(streamPath string) bool { |
|
|
|
} |
|
|
|
return true |
|
|
|
} |
|
|
|
func (rtc *WebRTC) GetAnswer(offer SessionDescription) ([]byte, error) { |
|
|
|
if err := rtc.SetRemoteDescription(offer); err != nil { |
|
|
|
Println(err) |
|
|
|
return nil, err |
|
|
|
} |
|
|
|
// Create answer
|
|
|
|
answer, err := rtc.CreateAnswer(nil) |
|
|
|
if err != nil { |
|
|
|
Println(err) |
|
|
|
return nil, err |
|
|
|
} |
|
|
|
|
|
|
|
// Sets the LocalDescription, and starts our UDP listeners
|
|
|
|
if err = rtc.SetLocalDescription(answer); err != nil { |
|
|
|
Println(err) |
|
|
|
return nil, err |
|
|
|
} |
|
|
|
if bytes, err := json.Marshal(answer); err != nil { |
|
|
|
Println(err) |
|
|
|
return bytes, err |
|
|
|
} else { |
|
|
|
return bytes, nil |
|
|
|
} |
|
|
|
} |
|
|
|
func run() { |
|
|
|
http.HandleFunc("/webrtc/answer", func(w http.ResponseWriter, r *http.Request) { |
|
|
|
http.HandleFunc("/webrtc/play", func(w http.ResponseWriter, r *http.Request) { |
|
|
|
streamPath := r.URL.Query().Get("streamPath") |
|
|
|
offer := SessionDescription{} |
|
|
|
bytes, err := ioutil.ReadAll(r.Body) |
|
|
|
@ -131,30 +324,38 @@ func run() { |
|
|
|
return |
|
|
|
} |
|
|
|
rtc := new(WebRTC) |
|
|
|
if rtc.Publish(streamPath) { |
|
|
|
// Set the remote SessionDescription
|
|
|
|
if err = rtc.SetRemoteDescription(offer); err != nil { |
|
|
|
rtc.RemoteAddr = r.RemoteAddr |
|
|
|
if rtc.Play(streamPath) { |
|
|
|
if bytes, err = rtc.GetAnswer(offer); err == nil { |
|
|
|
w.Write(bytes) |
|
|
|
} else { |
|
|
|
Println(err) |
|
|
|
w.Write([]byte(err.Error())) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// Create answer
|
|
|
|
answer, err := rtc.CreateAnswer(nil) |
|
|
|
if err != nil { |
|
|
|
Println(err) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// 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 { |
|
|
|
} else { |
|
|
|
w.Write([]byte(`{"errmsg":"bad name"}`)) |
|
|
|
} |
|
|
|
}) |
|
|
|
http.HandleFunc("/webrtc/publish", func(w http.ResponseWriter, r *http.Request) { |
|
|
|
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) |
|
|
|
rtc.RemoteAddr = r.RemoteAddr |
|
|
|
if rtc.Publish(streamPath) { |
|
|
|
if bytes, err = rtc.GetAnswer(offer); err == nil { |
|
|
|
w.Write(bytes) |
|
|
|
} else { |
|
|
|
Println(err) |
|
|
|
w.Write([]byte(err.Error())) |
|
|
|
return |
|
|
|
} |
|
|
|
w.Write(bytes) |
|
|
|
} else { |
|
|
|
w.Write([]byte(`{"errmsg":"bad name"}`)) |
|
|
|
} |
|
|
|
|
langhuihui
5 years ago