llvm-6502/unittests/Support/StreamingMemoryObject.cpp
Derek Schuff 493bc7e300 Fix StreamingMemoryObject to respect known object size.
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
2015-05-21 19:40:19 +00:00

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));
}