78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"os"
|
|
)
|
|
|
|
// series of messages sent from the client and responses
|
|
// window ack and peer bw seem to be completely useless to obs
|
|
func NegotiateConnect(chnk_wrp_ptr *ChunkWrapper) (bool) {
|
|
if err := chnk_wrp_ptr.ReadPeerChunkSize(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.ReadConnectCommand(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WriteWindowAckSize(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WritePeerBandwidth(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WriteChunkSize(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WriteConnectResponse(); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// next sequence
|
|
func CreateStream(chnk_wrp_ptr *ChunkWrapper) (bool) {
|
|
// first 2 messages are "releaseStream" and "FCPublish"
|
|
// no idea why the second exists since the next command is Publish
|
|
// the first has no docs since flash server docs are gone
|
|
// assuming it just signals a delete on any streams the current connection had
|
|
// it can be ignored for a pure stream intake server
|
|
if _, err := chnk_wrp_ptr.ReadChunk(); err != nil {
|
|
return false
|
|
}
|
|
if _, err := chnk_wrp_ptr.ReadChunk(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.ReadCreateStream(); err != nil {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WriteCreateStreamResponse(); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// Publish command handling, this is where stream keys are exchanged which means
|
|
// auth checking happens here
|
|
func PublishAsKey(chnk_wrp_ptr *ChunkWrapper) (bool) {
|
|
if err := chnk_wrp_ptr.ReadPublish(); err != nil {
|
|
return false
|
|
}
|
|
if !check_stream_key(chnk_wrp_ptr.params.stream_key) {
|
|
return false
|
|
}
|
|
if err := chnk_wrp_ptr.WritePublishResponse(); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// auth checking. again db would be the goal, for now just using a dir system
|
|
// pre generate a directory under ~/live/___/. Set the stream key to the folder name
|
|
// and this just checks if the folder exists (like a known public key)
|
|
// not secure at all but works for a demo
|
|
func check_stream_key(stream_key string) (bool) {
|
|
base_dir, _ := os.UserHomeDir()
|
|
if fileinfo, err := os.Stat(base_dir + "/live/" + stream_key); err == nil && fileinfo.IsDir() {
|
|
return true
|
|
}
|
|
return false
|
|
}
|