blob: 43cf8352f4b4600c781217d029d28c692b81eea2 [file] [log] [blame]
//===-- Holder Class for manipulating va_lists ------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_LIBC_SRC_SUPPORT_ARG_LIST_H
#define LLVM_LIBC_SRC_SUPPORT_ARG_LIST_H
#include "src/__support/common.h"
#include <stdarg.h>
#include <stddef.h>
namespace __llvm_libc {
namespace internal {
class ArgList {
va_list vlist;
public:
LIBC_INLINE ArgList(va_list vlist) { va_copy(this->vlist, vlist); }
LIBC_INLINE ArgList(ArgList &other) { va_copy(this->vlist, other.vlist); }
LIBC_INLINE ~ArgList() { va_end(this->vlist); }
LIBC_INLINE ArgList &operator=(ArgList &rhs) {
va_copy(vlist, rhs.vlist);
return *this;
}
template <class T> LIBC_INLINE T next_var() { return va_arg(vlist, T); }
};
// Used for testing things that use an ArgList when it's impossible to know what
// the arguments should be ahead of time. An example of this would be fuzzing,
// since a function passed a random input could request unpredictable arguments.
class MockArgList {
size_t arg_counter = 0;
public:
LIBC_INLINE MockArgList() = default;
LIBC_INLINE MockArgList(va_list) { ; }
LIBC_INLINE MockArgList(MockArgList &other) {
arg_counter = other.arg_counter;
}
LIBC_INLINE ~MockArgList() = default;
LIBC_INLINE MockArgList &operator=(MockArgList &rhs) {
arg_counter = rhs.arg_counter;
return *this;
}
template <class T> LIBC_INLINE T next_var() {
++arg_counter;
return T(arg_counter);
}
size_t read_count() const { return arg_counter; }
};
} // namespace internal
} // namespace __llvm_libc
#endif // LLVM_LIBC_SRC_SUPPORT_ARG_LIST_H