mirror of
https://github.com/github/codeql.git
synced 2025-12-20 10:46:30 +01:00
1b9dcac2dd
This turned out easier than expected previously. `llvm::PointerUnion` was also considered, which would have less memory footprint, but it would require more effort as it is lacking the same implicit conversions and operators that `std::variant` provides. Also renamed `ToTag<E>` to `TrapTagOf<E>` and introduced a derived convenience functor `TrapLabelOf<E>`.
79 lines
2.1 KiB
C++
79 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include "swift/extractor/trap/generated/TrapTags.h"
|
|
|
|
namespace codeql {
|
|
|
|
class UntypedTrapLabel {
|
|
uint64_t id_;
|
|
|
|
friend class std::hash<UntypedTrapLabel>;
|
|
template <typename Tag>
|
|
friend class TrapLabel;
|
|
|
|
protected:
|
|
UntypedTrapLabel() : id_{0xffffffffffffffff} {}
|
|
UntypedTrapLabel(uint64_t id) : id_{id} {}
|
|
|
|
public:
|
|
friend std::ostream& operator<<(std::ostream& out, UntypedTrapLabel l) {
|
|
out << '#' << std::hex << l.id_ << std::dec;
|
|
return out;
|
|
}
|
|
|
|
friend bool operator==(UntypedTrapLabel lhs, UntypedTrapLabel rhs) { return lhs.id_ == rhs.id_; }
|
|
};
|
|
|
|
template <typename TagParam>
|
|
class TrapLabel : public UntypedTrapLabel {
|
|
template <typename OtherTag>
|
|
friend class TrapLabel;
|
|
|
|
using UntypedTrapLabel::UntypedTrapLabel;
|
|
|
|
public:
|
|
using Tag = TagParam;
|
|
|
|
TrapLabel() = default;
|
|
|
|
// The caller is responsible for ensuring ID uniqueness.
|
|
static TrapLabel unsafeCreateFromExplicitId(uint64_t id) { return {id}; }
|
|
static TrapLabel unsafeCreateFromUntyped(UntypedTrapLabel label) { return {label.id_}; }
|
|
|
|
template <typename OtherTag>
|
|
TrapLabel(const TrapLabel<OtherTag>& other) : UntypedTrapLabel(other) {
|
|
// we temporarily need to bypass the label type system for unknown AST nodes and types
|
|
if constexpr (std::is_same_v<Tag, UnknownAstNodeTag>) {
|
|
static_assert(std::is_base_of_v<AstNodeTag, OtherTag>, "wrong label assignment!");
|
|
} else if constexpr (std::is_same_v<Tag, UnknownTypeTag>) {
|
|
static_assert(std::is_base_of_v<TypeTag, OtherTag>, "wrong label assignment!");
|
|
} else {
|
|
static_assert(std::is_base_of_v<Tag, OtherTag>, "wrong label assignment!");
|
|
}
|
|
}
|
|
};
|
|
|
|
inline auto trapQuoted(const std::string& s) {
|
|
return std::quoted(s, '"', '"');
|
|
}
|
|
|
|
template <typename Tag>
|
|
struct Binding {
|
|
TrapLabel<Tag> id;
|
|
};
|
|
|
|
} // namespace codeql
|
|
|
|
namespace std {
|
|
template <>
|
|
struct hash<codeql::UntypedTrapLabel> {
|
|
size_t operator()(const codeql::UntypedTrapLabel& l) const noexcept {
|
|
return std::hash<uint64_t>{}(l.id_);
|
|
}
|
|
};
|
|
} // namespace std
|