mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2025-04-03 08:11:52 +00:00
7be7a53496
We only considered the length of the operation and the length of the StreamRef without considered what it meant for the offset to be at a non-zero position. llvm-svn: 271496
81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
//===- StreamRef.h - A copyable reference to a stream -----------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_DEBUGINFO_CODEVIEW_STREAMREF_H
|
|
#define LLVM_DEBUGINFO_CODEVIEW_STREAMREF_H
|
|
|
|
#include "llvm/DebugInfo/CodeView/CodeViewError.h"
|
|
#include "llvm/DebugInfo/CodeView/StreamInterface.h"
|
|
|
|
namespace llvm {
|
|
namespace codeview {
|
|
|
|
class StreamRef : public StreamInterface {
|
|
public:
|
|
StreamRef() : Stream(nullptr), ViewOffset(0), Length(0) {}
|
|
StreamRef(const StreamInterface &Stream)
|
|
: Stream(&Stream), ViewOffset(0), Length(Stream.getLength()) {}
|
|
StreamRef(const StreamInterface &Stream, uint32_t Offset, uint32_t Length)
|
|
: Stream(&Stream), ViewOffset(Offset), Length(Length) {}
|
|
StreamRef(const StreamRef &Other)
|
|
: Stream(Other.Stream), ViewOffset(Other.ViewOffset),
|
|
Length(Other.Length) {}
|
|
|
|
Error readBytes(uint32_t Offset, uint32_t Size,
|
|
ArrayRef<uint8_t> &Buffer) const override {
|
|
if (Size + Offset > Length)
|
|
return make_error<CodeViewError>(cv_error_code::insufficient_buffer);
|
|
return Stream->readBytes(ViewOffset + Offset, Size, Buffer);
|
|
}
|
|
|
|
uint32_t getLength() const override { return Length; }
|
|
StreamRef drop_front(uint32_t N) const {
|
|
if (!Stream)
|
|
return StreamRef();
|
|
|
|
N = std::min(N, Length);
|
|
return StreamRef(*Stream, ViewOffset + N, Length - N);
|
|
}
|
|
|
|
StreamRef keep_front(uint32_t N) const {
|
|
if (!Stream)
|
|
return StreamRef();
|
|
N = std::min(N, Length);
|
|
return StreamRef(*Stream, ViewOffset, N);
|
|
}
|
|
|
|
bool operator==(const StreamRef &Other) const {
|
|
if (Stream != Other.Stream)
|
|
return false;
|
|
if (ViewOffset != Other.ViewOffset)
|
|
return false;
|
|
if (Length != Other.Length)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
bool operator!=(const StreamRef &Other) const { return !(*this == Other); }
|
|
|
|
StreamRef &operator=(const StreamRef &Other) {
|
|
Stream = Other.Stream;
|
|
ViewOffset = Other.ViewOffset;
|
|
Length = Other.Length;
|
|
return *this;
|
|
}
|
|
|
|
private:
|
|
const StreamInterface *Stream;
|
|
uint32_t ViewOffset;
|
|
uint32_t Length;
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif // LLVM_DEBUGINFO_CODEVIEW_STREAMREF_H
|