mahmad/stream-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added number, bool and string amf decoding

Browse source
This commit is contained in:
Muaz Ahmad 2023-08-11 15:51:46 +05:00
parent 74d95d4769
commit 369a7c40f4
1 changed files with 40 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
rtmp/amf/decode.go
View file

@ -12,7 +12,6 @@ func DecodeAMF(data *[]byte) (AMFObj, error) {
var byte_idx uint32 var byte_idx uint32
var root_obj_idx uint8 var root_obj_idx uint8
amf_root_obj := make(AMFObj) amf_root_obj := make(AMFObj)
for { for {
top_level_obj, err := read_next(data, &byte_idx) top_level_obj, err := read_next(data, &byte_idx)
if err != nil { if err != nil {
@ -20,6 +19,9 @@ func DecodeAMF(data *[]byte) (AMFObj, error) {
} }
amf_root_obj[root_obj_idx] = top_level_obj amf_root_obj[root_obj_idx] = top_level_obj
root_obj_idx += 1 root_obj_idx += 1
if byte_idx == uint32(len(*data)) {
break
}
} }
return amf_root_obj, nil return amf_root_obj, nil
} }
@ -28,8 +30,9 @@ func read_bytes(data *[]byte, byte_idx *uint32, n uint32) ([]byte, error) {
if int(*byte_idx + n) > len(*data) { if int(*byte_idx + n) > len(*data) {
return make([]byte, 0), errors.New("Read goes past end") return make([]byte, 0), errors.New("Read goes past end")
} }
read_slice := (*data)[*byte_idx:*byte_idx + n]
*byte_idx += n *byte_idx += n
return (*data)[*byte_idx:*byte_idx + n], nil return read_slice, nil
} }
func read_number(data *[]byte, byte_idx *uint32) (float64, error) { func read_number(data *[]byte, byte_idx *uint32) (float64, error) {
@ -40,20 +43,49 @@ func read_number(data *[]byte, byte_idx *uint32) (float64, error) {
return math.Float64frombits(binary.BigEndian.Uint64(float_bytes)), nil return math.Float64frombits(binary.BigEndian.Uint64(float_bytes)), nil
} }
func read_bool(data *[]byte, byte_idx *uint32) (bool, error) {
bool_byte, err := read_bytes(data, byte_idx, 1)
if err != nil {
return false, err
}
if bool_byte[0] > 1 {
return false, errors.New("bool byte must be 0 or 1")
}
return bool_byte[0] != 0, nil
}
func read_string(data *[]byte, byte_idx *uint32) (string, error) {
string_len, err := read_bytes(data, byte_idx, 2)
if err != nil {
return "", err
}
string_bytes, err := read_bytes(data, byte_idx, uint32(binary.BigEndian.Uint16(string_len)))
if err != nil {
return "", err
}
return string(string_bytes), err
}
func read_next(data *[]byte, byte_idx *uint32) (interface{}, error) { func read_next(data *[]byte, byte_idx *uint32) (interface{}, error) {
data_type, err := read_bytes(data, byte_idx, 1) data_type, err := read_bytes(data, byte_idx, 1)
var next_obj interface{}
if err != nil { if err != nil {
return nil, err return nil, err
} }
switch data_type[0] { switch data_type[0] {
case 0: case 0:
next_obj, err := read_number(data, byte_idx) next_obj, err = read_number(data, byte_idx)
case 1:
next_obj, err = read_bool(data, byte_idx)
case 2:
next_obj, err = read_string(data, byte_idx)
default:
return nil, errors.New("Unhandled data type")
}
if err != nil { if err != nil {
return nil, err return nil, err
} }
return next_obj, nil return next_obj, nil
} }
return nil, errors.New("tmp")
}