141 lines
3 KiB
Go
141 lines
3 KiB
Go
package libipcamera
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
// RTPRelay holds information on the relaying stream listener
|
|
type RTPRelay struct {
|
|
close bool
|
|
targetIP net.IP
|
|
targetPort int
|
|
listener net.PacketConn
|
|
context context.Context
|
|
}
|
|
|
|
// CreateRTPRelay creates a UDP listener that handles live data
|
|
// from the camera and forwards it as an RTP stream
|
|
func CreateRTPRelay(ctx context.Context, targetAddress net.IP, targetPort int) *RTPRelay {
|
|
conn, err := net.ListenPacket("udp", ":6669")
|
|
|
|
if err != nil {
|
|
log.Printf("ERROR: %s\n", err)
|
|
}
|
|
|
|
relay := RTPRelay{
|
|
targetIP: targetAddress,
|
|
targetPort: targetPort,
|
|
listener: conn,
|
|
context: ctx,
|
|
}
|
|
if err != nil {
|
|
log.Printf("ERROR: %s\n", err)
|
|
}
|
|
|
|
go handleCameraStream(relay, conn)
|
|
|
|
return &relay
|
|
}
|
|
|
|
func handleCameraStream(relay RTPRelay, conn net.PacketConn) {
|
|
buffer := make([]byte, 2048)
|
|
packetReader := bytes.NewReader(buffer)
|
|
|
|
header := streamHeader{}
|
|
var payload []byte
|
|
|
|
rtpTarget := net.UDPAddr{
|
|
IP: relay.targetIP,
|
|
Port: relay.targetPort,
|
|
}
|
|
rtpSource, _ := net.ResolveUDPAddr("udp", "127.0.0.1")
|
|
rtpConn, err := net.DialUDP("udp", rtpSource, &rtpTarget)
|
|
if err != nil {
|
|
log.Printf("ERROR creating RTP sender: %s\n", err)
|
|
}
|
|
|
|
var sequenceNumber uint16
|
|
var elapsed uint32
|
|
|
|
frameBuffer := bytes.Buffer{}
|
|
packetBuffer := bytes.Buffer{}
|
|
|
|
for {
|
|
conn.SetReadDeadline(time.Now().Add(10 * time.Second))
|
|
|
|
select {
|
|
case <-relay.context.Done():
|
|
log.Println("Context Done")
|
|
rtpConn.Close()
|
|
relay.listener.Close()
|
|
break
|
|
default:
|
|
if relay.close {
|
|
rtpConn.Close()
|
|
relay.listener.Close()
|
|
break
|
|
}
|
|
|
|
conn.ReadFrom(buffer)
|
|
packetReader.Reset(buffer)
|
|
|
|
binary.Read(packetReader, binary.BigEndian, &header)
|
|
|
|
if header.Magic != 0xBCDE {
|
|
log.Printf("Received message with invalid magic (%x).", header.Magic)
|
|
break
|
|
}
|
|
|
|
if header.Length > 0 {
|
|
payload = make([]byte, header.Length)
|
|
_, err := io.ReadFull(packetReader, payload)
|
|
if err != nil {
|
|
log.Printf("Read Error: %s\n", err)
|
|
break
|
|
}
|
|
} else {
|
|
payload = []byte{}
|
|
}
|
|
|
|
switch header.MessageType {
|
|
case 0x0001: // H.264 Data
|
|
frameBuffer.Write(payload)
|
|
case 0x0002: // Time
|
|
// Append the Framebuffer
|
|
packetBuffer.Write(frameBuffer.Bytes())
|
|
|
|
// Send out the packet
|
|
rtpConn.Write(packetBuffer.Bytes())
|
|
|
|
// Prepare the next packet
|
|
packetBuffer.Reset()
|
|
packetBuffer.Write([]byte{0x80, 0x63})
|
|
binary.Write(&packetBuffer, binary.BigEndian, sequenceNumber+1)
|
|
binary.Write(&packetBuffer, binary.BigEndian, (uint32)(elapsed)*90)
|
|
binary.Write(&packetBuffer, binary.BigEndian, (uint64(0)))
|
|
|
|
// Reset the Framebuffer
|
|
frameBuffer.Reset()
|
|
sequenceNumber++
|
|
|
|
elapsed = binary.LittleEndian.Uint32(payload[12:])
|
|
default:
|
|
log.Printf("Received Unknown Message: %+v\n", header)
|
|
log.Printf("Payload:\n%s\n", hex.Dump(payload))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Stop stops listening for packets
|
|
func (r *RTPRelay) Stop() {
|
|
r.listener.Close()
|
|
r.close = true
|
|
}
|