mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-06 19:07:49 +08:00
350e262433
GCC Does not explicitly include stdint. This update includes them where needed. A very minor change that would allow GCC13 and above to compile correctly. In newer versions, cstdint is not implicitly included, hence it errors out during compilation for some files.
127 lines
2.9 KiB
C++
127 lines
2.9 KiB
C++
/*
|
|
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT license that can be found in the
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef __amf_h
|
|
#define __amf_h
|
|
|
|
#include <assert.h>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <functional>
|
|
#include <cstdint>
|
|
namespace toolkit {
|
|
class BufferLikeString;
|
|
}
|
|
enum AMFType {
|
|
AMF_NUMBER,
|
|
AMF_INTEGER,
|
|
AMF_BOOLEAN,
|
|
AMF_STRING,
|
|
AMF_OBJECT,
|
|
AMF_NULL,
|
|
AMF_UNDEFINED,
|
|
AMF_ECMA_ARRAY,
|
|
AMF_STRICT_ARRAY,
|
|
};
|
|
|
|
class AMFValue;
|
|
|
|
class AMFValue {
|
|
public:
|
|
friend class AMFEncoder;
|
|
|
|
using mapType = std::map<std::string, AMFValue>;
|
|
using arrayType = std::vector<AMFValue>;
|
|
|
|
~AMFValue();
|
|
AMFValue(AMFType type = AMF_NULL);
|
|
AMFValue(const char *s);
|
|
AMFValue(const std::string &s);
|
|
AMFValue(double n);
|
|
AMFValue(int i);
|
|
AMFValue(bool b);
|
|
AMFValue(const AMFValue &from);
|
|
AMFValue &operator = (const AMFValue &from);
|
|
|
|
void clear();
|
|
AMFType type() const ;
|
|
const std::string &as_string() const;
|
|
double as_number() const;
|
|
int as_integer() const;
|
|
bool as_boolean() const;
|
|
std::string to_string() const;
|
|
const AMFValue &operator[](const char *str) const;
|
|
void object_for_each(const std::function<void(const std::string &key, const AMFValue &val)> &fun) const ;
|
|
operator bool() const;
|
|
void set(const std::string &s, const AMFValue &val);
|
|
void add(const AMFValue &val);
|
|
|
|
private:
|
|
const mapType &getMap() const;
|
|
const arrayType &getArr() const;
|
|
void destroy();
|
|
void init();
|
|
|
|
private:
|
|
AMFType _type;
|
|
union {
|
|
std::string *string;
|
|
double number;
|
|
int integer;
|
|
bool boolean;
|
|
mapType *object;
|
|
arrayType *array;
|
|
} _value;
|
|
};
|
|
|
|
class AMFDecoder {
|
|
public:
|
|
AMFDecoder(const toolkit::BufferLikeString &buf, size_t pos, int version = 0);
|
|
template<typename TP>
|
|
TP load();
|
|
|
|
private:
|
|
std::string load_key();
|
|
AMFValue load_object();
|
|
AMFValue load_ecma();
|
|
AMFValue load_arr();
|
|
uint8_t front();
|
|
uint8_t pop_front();
|
|
|
|
private:
|
|
const toolkit::BufferLikeString &buf;
|
|
size_t pos;
|
|
int version;
|
|
};
|
|
|
|
class AMFEncoder {
|
|
public:
|
|
AMFEncoder & operator <<(const char *s);
|
|
AMFEncoder & operator <<(const std::string &s);
|
|
AMFEncoder & operator <<(std::nullptr_t);
|
|
AMFEncoder & operator <<(const int n);
|
|
AMFEncoder & operator <<(const double n);
|
|
AMFEncoder & operator <<(const bool b);
|
|
AMFEncoder & operator <<(const AMFValue &value);
|
|
const std::string& data() const ;
|
|
void clear() ;
|
|
|
|
private:
|
|
void write_key(const std::string &s);
|
|
AMFEncoder &write_undefined();
|
|
|
|
private:
|
|
std::string buf;
|
|
};
|
|
|
|
|
|
#endif
|