forked from OSchip/llvm-project
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
//===-- xray_utils.h --------------------------------------------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of XRay, a dynamic runtime instrumentation system.
|
|
//
|
|
// Some shared utilities for the XRay runtime implementation.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#ifndef XRAY_UTILS_H
|
|
#define XRAY_UTILS_H
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <sys/types.h>
|
|
#include <utility>
|
|
|
|
#include "sanitizer_common/sanitizer_common.h"
|
|
#if SANITIZER_FUCHSIA
|
|
#include <zircon/types.h>
|
|
#endif
|
|
|
|
namespace __xray {
|
|
|
|
class LogWriter {
|
|
public:
|
|
#if SANITIZER_FUCHSIA
|
|
LogWriter(zx_handle_t Vmo) : Vmo(Vmo) {}
|
|
#else
|
|
explicit LogWriter(int Fd) : Fd(Fd) {}
|
|
#endif
|
|
~LogWriter();
|
|
|
|
// Write a character range into a log.
|
|
void WriteAll(const char *Begin, const char *End);
|
|
|
|
void Flush();
|
|
|
|
// Returns a new log instance initialized using the flag-provided values.
|
|
static LogWriter *Open();
|
|
// Closes and deallocates the log instance.
|
|
static void Close(LogWriter *LogWriter);
|
|
|
|
private:
|
|
#if SANITIZER_FUCHSIA
|
|
zx_handle_t Vmo = ZX_HANDLE_INVALID;
|
|
uint64_t Offset = 0;
|
|
#else
|
|
int Fd = -1;
|
|
#endif
|
|
};
|
|
|
|
constexpr size_t gcd(size_t a, size_t b) {
|
|
return (b == 0) ? a : gcd(b, a % b);
|
|
}
|
|
|
|
constexpr size_t lcm(size_t a, size_t b) { return a * b / gcd(a, b); }
|
|
|
|
constexpr size_t nearest_boundary(size_t number, size_t multiple) {
|
|
return multiple * ((number / multiple) + (number % multiple ? 1 : 0));
|
|
}
|
|
|
|
constexpr size_t next_pow2_helper(size_t num, size_t acc) {
|
|
return (1u << acc) >= num ? (1u << acc) : next_pow2_helper(num, acc + 1);
|
|
}
|
|
|
|
constexpr size_t next_pow2(size_t number) {
|
|
return next_pow2_helper(number, 1);
|
|
}
|
|
|
|
template <class T> constexpr T &max(T &A, T &B) { return A > B ? A : B; }
|
|
|
|
template <class T> constexpr T &min(T &A, T &B) { return A <= B ? A : B; }
|
|
|
|
constexpr ptrdiff_t diff(uintptr_t A, uintptr_t B) {
|
|
return max(A, B) - min(A, B);
|
|
}
|
|
|
|
} // namespace __xray
|
|
|
|
#endif // XRAY_UTILS_H
|