Class Iterator

lib/Bytecode/Reader/BytecodeReader.cpp:789–827

      : context(context), masterReader(masterReader) {}

  class Iterator {
  public:
    Iterator(DebugInfoReader &reader, uint64_t opIndex)
        : reader(reader), opIndex(opIndex) {}

    /// Return the next debug info attribute for the current operation.
    template <typename T>
    T next() {
      // Check if the index is reserved for special debug info attributes.
      if (opIndex == static_cast<uint64_t>(Bytecode::DebugReserved::UnknownLoc))
        return dyn_cast<T>(UnknownLoc::get(&reader.context));

      // Adjust the index to account for reserved indices.
      auto actualOpIndex =
          opIndex - static_cast<uint64_t>(Bytecode::DebugReserved::SIZE);

      // Calculate the offset for the current operation index.
      if (actualOpIndex >= reader.diIndexOffsets.size())
        return T();
      auto offset = reader.diIndexOffsets[actualOpIndex];

      // Validate size to prevent excessive memory allocation.
      if (reader.diIndices.size() > (std::numeric_limits<uint32_t>::max() - 1))
        return T();
      if (offset > (std::numeric_limits<uint32_t>::max() - 1))
        return T();
      if (offset + opIndexOffset >= reader.diIndices.size())
        return T();
      offset += opIndexOffset++;

      // Return the next debug info attribute for the current operation.
      return reader.getDebugInfo<T>(reader.diIndices[offset]);
    }

  private:
    DebugInfoReader &reader;
    uint64_t opIndex;
    uint64_t opIndexOffset = 0;
  };

  Iterator getIterator(uint64_t opIndex) { return Iterator(*this, opIndex); }