2024-04-07 15:14:54 +10:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"log"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"gitea.drjosh.dev/josh/jrouter/atalk/rtmp"
|
|
|
|
"github.com/google/gopacket/pcap"
|
|
|
|
"github.com/sfiera/multitalk/pkg/ddp"
|
|
|
|
"github.com/sfiera/multitalk/pkg/ethertalk"
|
|
|
|
)
|
|
|
|
|
|
|
|
// RTMPMachine implements RTMP on an AppleTalk network attached to the router.
|
|
|
|
type RTMPMachine struct {
|
|
|
|
aarp *AARPMachine
|
|
|
|
cfg *config
|
|
|
|
pcapHandle *pcap.Handle
|
|
|
|
}
|
|
|
|
|
2024-04-07 15:41:27 +10:00
|
|
|
func (m *RTMPMachine) Run(ctx context.Context, incomingCh <-chan *ddp.ExtPacket) error {
|
2024-04-07 15:14:54 +10:00
|
|
|
bcastTicker := time.NewTicker(10 * time.Second)
|
|
|
|
defer bcastTicker.Stop()
|
|
|
|
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-ctx.Done():
|
|
|
|
return ctx.Err()
|
|
|
|
|
|
|
|
case <-bcastTicker.C:
|
|
|
|
// Broadcast an RTMP Data
|
|
|
|
myAddr, ok := m.aarp.Address()
|
|
|
|
if !ok {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
dataPkt := &rtmp.DataPacket{
|
|
|
|
RouterAddr: myAddr.Proto,
|
|
|
|
Extended: true,
|
|
|
|
NetworkTuples: []rtmp.NetworkTuple{
|
|
|
|
// "The first tuple in RTMP Data packets sent on extended
|
|
|
|
// networks ... indicates the network number range assigned
|
|
|
|
// to that network."
|
|
|
|
{
|
|
|
|
Extended: true,
|
|
|
|
RangeStart: m.cfg.EtherTalk.NetStart,
|
|
|
|
RangeEnd: m.cfg.EtherTalk.NetEnd,
|
|
|
|
Distance: 0,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
// TODO: append more networks!
|
|
|
|
|
|
|
|
dataPktRaw, err := dataPkt.Marshal()
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("RTMP: Couldn't marshal Data packet: %v", err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
ddpPkt := ddp.ExtPacket{
|
|
|
|
ExtHeader: ddp.ExtHeader{
|
|
|
|
Size: uint16(len(dataPktRaw)),
|
2024-04-07 15:25:30 +10:00
|
|
|
Cksum: 0,
|
|
|
|
DstNet: 0, // this network
|
2024-04-07 15:14:54 +10:00
|
|
|
DstNode: 0xff, // broadcast packet
|
|
|
|
DstSocket: 1, // the special RTMP socket
|
|
|
|
SrcNet: myAddr.Proto.Network,
|
|
|
|
SrcNode: myAddr.Proto.Node,
|
|
|
|
SrcSocket: 1, // the special RTMP socket
|
|
|
|
Proto: ddp.ProtoRTMPResp,
|
|
|
|
},
|
|
|
|
Data: dataPktRaw,
|
|
|
|
}
|
|
|
|
|
2024-04-07 15:25:30 +10:00
|
|
|
ethFrame, err := ethertalk.AppleTalk(myAddr.Hardware, ddpPkt)
|
2024-04-07 15:14:54 +10:00
|
|
|
if err != nil {
|
|
|
|
log.Printf("RTMP: Couldn't create EtherTalk frame: %v", err)
|
|
|
|
}
|
2024-04-07 15:25:30 +10:00
|
|
|
ethFrame.Dst = ethertalk.AppleTalkBroadcast
|
2024-04-07 15:14:54 +10:00
|
|
|
|
|
|
|
ethFrameRaw, err := ethertalk.Marshal(*ethFrame)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("RTMP: Couldn't marshal EtherTalk frame: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := m.pcapHandle.WritePacketData(ethFrameRaw); err != nil {
|
|
|
|
log.Printf("RTMP: Couldn't write frame: %v", err)
|
|
|
|
}
|
2024-04-07 15:41:27 +10:00
|
|
|
|
|
|
|
case pkt := <-incomingCh:
|
|
|
|
switch pkt.Proto {
|
|
|
|
case ddp.ProtoRTMPReq:
|
|
|
|
// I can answer RTMP requests!
|
|
|
|
req, err := rtmp.UnmarshalRequestPacket(pkt.Data)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("RTMP: Couldn't unmarshal Request packet: %v", err)
|
|
|
|
}
|
|
|
|
switch req.Function {
|
|
|
|
case 1: // RTMP Request
|
|
|
|
// TODO
|
|
|
|
log.Print("RTMP: Got Request")
|
|
|
|
|
|
|
|
case 2: // RTMP RDR with split-horizon processing
|
|
|
|
// TODO
|
|
|
|
log.Print("RTMP: Got RDR with split-horizon")
|
|
|
|
|
|
|
|
case 3: // RTMP RDR for whole table
|
|
|
|
// TODO
|
|
|
|
log.Print("RTMP: Got RDR without split-horizon")
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
case ddp.ProtoRTMPResp:
|
|
|
|
// It's a peer router on the AppleTalk network!
|
|
|
|
// TODO
|
|
|
|
log.Print("RTMP: Got Response or ")
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2024-04-07 15:14:54 +10:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|