This is the next step towards supporting bitcode auto upgrade with opaque pointers. The ValueList now stores the Value* together with its associated type ID, which allows inspecting the original pointer element type of arbitrary values. This is a largely mechanical change threading the type ID through various places. I've left TODOTypeID placeholders in a number of places where determining the type ID is either non-trivial or requires allocating a new type ID not present in the original bitcode. For this reason, the new type IDs are also not used for anything yet (apart from propagation). They will get used once the TODOs are resolved. Differential Revision: https://reviews.llvm.org/D119821
105 lines
3.1 KiB
C++
105 lines
3.1 KiB
C++
//===-- Bitcode/Reader/ValueList.h - Number values --------------*- 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 class gives values and types Unique ID's.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_BITCODE_READER_VALUELIST_H
|
|
#define LLVM_LIB_BITCODE_READER_VALUELIST_H
|
|
|
|
#include "llvm/IR/ValueHandle.h"
|
|
#include <cassert>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
namespace llvm {
|
|
|
|
class Constant;
|
|
class LLVMContext;
|
|
class Type;
|
|
class Value;
|
|
|
|
class BitcodeReaderValueList {
|
|
/// Maps Value ID to pair of Value* and Type ID.
|
|
std::vector<std::pair<WeakTrackingVH, unsigned>> ValuePtrs;
|
|
|
|
/// As we resolve forward-referenced constants, we add information about them
|
|
/// to this vector. This allows us to resolve them in bulk instead of
|
|
/// resolving each reference at a time. See the code in
|
|
/// ResolveConstantForwardRefs for more information about this.
|
|
///
|
|
/// The key of this vector is the placeholder constant, the value is the slot
|
|
/// number that holds the resolved value.
|
|
using ResolveConstantsTy = std::vector<std::pair<Constant *, unsigned>>;
|
|
ResolveConstantsTy ResolveConstants;
|
|
LLVMContext &Context;
|
|
|
|
/// Maximum number of valid references. Forward references exceeding the
|
|
/// maximum must be invalid.
|
|
unsigned RefsUpperBound;
|
|
|
|
public:
|
|
BitcodeReaderValueList(LLVMContext &C, size_t RefsUpperBound)
|
|
: Context(C),
|
|
RefsUpperBound(std::min((size_t)std::numeric_limits<unsigned>::max(),
|
|
RefsUpperBound)) {}
|
|
|
|
~BitcodeReaderValueList() {
|
|
assert(ResolveConstants.empty() && "Constants not resolved?");
|
|
}
|
|
|
|
// vector compatibility methods
|
|
unsigned size() const { return ValuePtrs.size(); }
|
|
void resize(unsigned N) {
|
|
ValuePtrs.resize(N);
|
|
}
|
|
void push_back(Value *V, unsigned TypeID) {
|
|
ValuePtrs.emplace_back(V, TypeID);
|
|
}
|
|
|
|
void clear() {
|
|
assert(ResolveConstants.empty() && "Constants not resolved?");
|
|
ValuePtrs.clear();
|
|
}
|
|
|
|
Value *operator[](unsigned i) const {
|
|
assert(i < ValuePtrs.size());
|
|
return ValuePtrs[i].first;
|
|
}
|
|
|
|
unsigned getTypeID(unsigned ValNo) const {
|
|
assert(ValNo < ValuePtrs.size());
|
|
return ValuePtrs[ValNo].second;
|
|
}
|
|
|
|
Value *back() const { return ValuePtrs.back().first; }
|
|
void pop_back() {
|
|
ValuePtrs.pop_back();
|
|
}
|
|
bool empty() const { return ValuePtrs.empty(); }
|
|
|
|
void shrinkTo(unsigned N) {
|
|
assert(N <= size() && "Invalid shrinkTo request!");
|
|
ValuePtrs.resize(N);
|
|
}
|
|
|
|
Constant *getConstantFwdRef(unsigned Idx, Type *Ty, unsigned TyID);
|
|
Value *getValueFwdRef(unsigned Idx, Type *Ty, unsigned TyID);
|
|
|
|
void assignValue(unsigned Idx, Value *V, unsigned TypeID);
|
|
|
|
/// Once all constants are read, this method bulk resolves any forward
|
|
/// references.
|
|
void resolveConstantForwardRefs();
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_LIB_BITCODE_READER_VALUELIST_H
|