stream-server/rtmp/amf/command.go

127 lines
2.8 KiB
Go

package amf
import (
"errors"
)
func (amf_obj_root AMFObj) ProcessConnect() (err error) {
err = errors.New("Bad AMF connect command")
if !check_object(amf_obj_root, 0, "connect") {
return
}
if !check_object(amf_obj_root, 1, 1.0) {
return
}
if _, ok := amf_obj_root[2]; !ok {
return
} else if _, ok := amf_obj_root[2].(AMFObj); !ok {
return
}
if !check_object(amf_obj_root[2].(AMFObj), "app", "live") {
return
}
err = nil
return
}
func (amf_obj_root AMFObj) ProcessCreateStream(trans_id *float64) (err error) {
err = errors.New("Bad AMF create stream")
if !check_object(amf_obj_root, 0, "createStream") {
return
}
if _, ok := amf_obj_root[1]; !ok {
return
}
transac_id_float, ok := amf_obj_root[1].(float64)
if !ok {
return
}
*trans_id = transac_id_float
err = nil
return
}
func (amf_obj_root AMFObj) ProcessPublish(trans_id *float64, stream_key *string) (err error) {
err = errors.New("Bad publish")
if !check_object(amf_obj_root, 0, "publish") {
return
}
if _, ok := amf_obj_root[1]; !ok {
return
} else if transac_id_float, ok := amf_obj_root[1].(float64); ok {
*trans_id = transac_id_float
} else {
return
}
if _, ok := amf_obj_root[3]; !ok {
return
} else if stream_key_val, ok := amf_obj_root[3].(string); ok {
*stream_key = stream_key_val
} else {
return
}
if !check_object(amf_obj_root, 4, "live") {
return
}
err = nil
return
}
func EncodeConnectResponse() ([]byte, error) {
amf_root_obj := make(AMFObj)
amf_root_obj[0] = "_result"
amf_root_obj[1] = 1.0
amf_root_obj[2] = make(AMFObj)
amf_prop_obj := amf_root_obj[2].(AMFObj)
amf_prop_obj["fmsVer"] = "FMS/3,5,5,2004"
amf_prop_obj["capabilities"] = 31.0
amf_root_obj[3] = make(AMFObj)
amf_event_obj := amf_root_obj[3].(AMFObj)
amf_event_obj["level"] = "status"
amf_event_obj["code"] = "NetConnection.Connect.Success"
amf_event_obj["description"] = "Connection Succeeded"
amf_event_obj["objectEncoding"] = 0.0
return Encode(amf_root_obj)
}
func EncodeCreateStreamResponse(trans_id float64) ([]byte, error) {
amf_root_obj := make(AMFObj)
amf_root_obj[0] = "_result"
amf_root_obj[1] = trans_id
amf_root_obj[2] = nil
amf_root_obj[3] = 1.0
return Encode(amf_root_obj)
}
func EncodePublishResponse(trans_id float64) ([]byte, error) {
amf_root_obj := make(AMFObj)
amf_root_obj[0] = "onStatus"
amf_root_obj[1] = trans_id
amf_root_obj[2] = nil
amf_root_obj[3] = make(AMFObj)
amf_event_obj := amf_root_obj[3].(AMFObj)
amf_event_obj["level"] = "status"
amf_event_obj["code"] = "NetStream.Publish.Start"
amf_event_obj["description"] = "Start Publishing"
return Encode(amf_root_obj)
}
func check_object(amf_obj AMFObj, key interface{}, target interface{}) (bool) {
if val, ok := amf_obj[key]; ok && val == target{
return true
}
return false
}