133 lines
3.1 KiB
Go
133 lines
3.1 KiB
Go
package amf
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
// most are hard-coded checks to make sure the received AMF0 command is what was received
|
|
|
|
|
|
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
|
|
}
|
|
|
|
// Encodes do the reverse, pacakge some data into the corresponding AMF0 message to send
|
|
|
|
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)
|
|
}
|
|
|
|
// helper function, checks if the given key exists for the object and checks if the matching value is the same as the target
|
|
func check_object(amf_obj AMFObj, key interface{}, target interface{}) (bool) {
|
|
if val, ok := amf_obj[key]; ok && val == target{
|
|
return true
|
|
}
|
|
return false
|
|
}
|