mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2024-12-15 20:29:48 +00:00
493bc7e300
The existing code for method StreamingMemoryObject.fetchToPos does not respect the corresonding call to setKnownObjectSize(). As a result, it allows the StreamingMemoryObject to read bytes past the object size. This patch provides a test case, and code to fix the problem. Patch by Karl Schimpf Differential Revision: http://reviews.llvm.org/D8931 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@237939 91177308-0d34-0410-b5e6-96231b3b80d8
39 lines
1.0 KiB
C++
39 lines
1.0 KiB
C++
//===- llvm/unittest/Support/StreamingMemoryObject.cpp - unit tests -------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/StreamingMemoryObject.h"
|
|
#include "gtest/gtest.h"
|
|
#include <string.h>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
class NullDataStreamer : public DataStreamer {
|
|
size_t GetBytes(unsigned char *buf, size_t len) override {
|
|
memset(buf, 0, len);
|
|
return len;
|
|
}
|
|
};
|
|
}
|
|
|
|
TEST(StreamingMemoryObject, Test) {
|
|
auto *DS = new NullDataStreamer();
|
|
StreamingMemoryObject O(DS);
|
|
EXPECT_TRUE(O.isValidAddress(32 * 1024));
|
|
}
|
|
|
|
TEST(StreamingMemoryObject, TestSetKnownObjectSize) {
|
|
auto *DS = new NullDataStreamer();
|
|
StreamingMemoryObject O(DS);
|
|
uint8_t Buf[32];
|
|
EXPECT_EQ((uint64_t) 16, O.readBytes(Buf, 16, 0));
|
|
O.setKnownObjectSize(24);
|
|
EXPECT_EQ((uint64_t) 8, O.readBytes(Buf, 16, 16));
|
|
}
|