fix: limit AMF decoder nesting depth (#4787)

### Motivation
- Prevent unbounded recursive-descent parsing in the AMF decoder that
allowed deeply nested AMF objects/arrays to exhaust the process stack
and cause a pre-auth remote DoS.

### Description
- Add an RAII depth guard `AMFDecoder::DepthGuard` and a per-decoder
`depth` counter with `kMaxDepth = 64` to `AMFDecoder` in
`src/Rtmp/amf.h`/`src/Rtmp/amf.cpp`.
- Instrument `load_object()`, `load_ecma()`, and `load_arr()` to create
a `DepthGuard` at entry so nesting is incremented and reliably
decremented on return or exception, and throw
`std::runtime_error("Maximum AMF nesting depth exceeded")` when the
limit is exceeded.
- Add `tests/test_amf.cpp` to assert that a 64-level nested AMF
container is accepted and a 65-level nested container is rejected with
the expected error.
- Keep non-nested parsing behavior unchanged; the guard only enforces a
maximum container nesting depth.

### Testing
- Configured the project with tests enabled using `cmake -S . -B build
-DENABLE_TESTS=ON`, which generated the `test_amf` target successfully.
- Built the test target with `cmake --build build --target test_amf` and
performed smoke compiles of modified sources, with `src/Rtmp/amf.cpp`
compiling to object (`/tmp/amf.o`) successfully.
- Compiled the new test source with `c++ -std=gnu++11 -c
tests/test_amf.cpp` successfully and verified `git diff --cached
--check` passed.

------
[Codex
Task](https://chatgpt.com/codex/cloud/tasks/task_e_6a63c9326b9883209c857c0f4fa5d0cc)
This commit is contained in:
YuLi
2026-07-27 07:56:18 -07:00
committed by GitHub
parent 867fc9ce83
commit 6b16281ab0
3 changed files with 146 additions and 2 deletions

View File

@@ -452,6 +452,17 @@ const std::string& AMFEncoder::data() const {
//////////////////Decoder//////////////////
AMFDecoder::DepthGuard::DepthGuard(AMFDecoder &decoder) : _decoder(decoder) {
if (_decoder.depth >= AMFDecoder::kMaxDepth) {
throw std::runtime_error("Maximum AMF nesting depth exceeded");
}
++_decoder.depth;
}
AMFDecoder::DepthGuard::~DepthGuard() {
--_decoder.depth;
}
uint8_t AMFDecoder::front() {
if (pos >= buf.size()) {
throw std::runtime_error("Not enough data");
@@ -625,6 +636,7 @@ std::string AMFDecoder::load_key() {
}
AMFValue AMFDecoder::load_object() {
DepthGuard depth_guard(*this);
AMFValue object(AMF_OBJECT);
if (pop_front() != AMF0_OBJECT) {
throw std::runtime_error("Expected an object");
@@ -643,6 +655,7 @@ AMFValue AMFDecoder::load_object() {
}
AMFValue AMFDecoder::load_ecma() {
DepthGuard depth_guard(*this);
/* ECMA array is the same as object, with 4 extra zero bytes */
AMFValue object(AMF_ECMA_ARRAY);
if (pop_front() != AMF0_ECMA_ARRAY) {
@@ -665,7 +678,8 @@ AMFValue AMFDecoder::load_ecma() {
return object;
}
AMFValue AMFDecoder::load_arr() {
/* ECMA array is the same as object, with 4 extra zero bytes */
DepthGuard depth_guard(*this);
/* Strict array is a count-prefixed list of AMF values. */
AMFValue object(AMF_STRICT_ARRAY);
if (pop_front() != AMF0_STRICT_ARRAY) {
throw std::runtime_error("Expected an STRICT array");
@@ -689,4 +703,3 @@ AMFValue AMFDecoder::load_arr() {
AMFDecoder::AMFDecoder(const BufferLikeString &buf_in, size_t pos_in, int version_in) :
buf(buf_in), pos(pos_in), version(version_in) {
}

View File

@@ -89,6 +89,19 @@ public:
TP load();
private:
class DepthGuard {
public:
explicit DepthGuard(AMFDecoder &decoder);
~DepthGuard();
DepthGuard(const DepthGuard &) = delete;
DepthGuard(DepthGuard &&) = delete;
DepthGuard &operator=(const DepthGuard &) = delete;
DepthGuard &operator=(DepthGuard &&) = delete;
private:
AMFDecoder &_decoder;
};
std::string load_key();
AMFValue load_object();
AMFValue load_ecma();
@@ -100,6 +113,9 @@ private:
const toolkit::BufferLikeString &buf;
size_t pos;
int version;
size_t depth = 0;
static constexpr size_t kMaxDepth = 64;
};
class AMFEncoder {