9416703f5b
- Use the CHECK macro everywhere that assert was used. - Remove the _debug version of the tests and leave the CHECK macro so it's always checking in the code. Bug: 23762183 Test: Ran unit tests. Change-Id: Ie705eedae393d0e95bb9d99f852687a11881aef1
146 lines
3.5 KiB
C++
146 lines
3.5 KiB
C++
/*
|
|
* Copyright (C) 2016 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef _LIBUNWINDSTACK_MEMORY_H
|
|
#define _LIBUNWINDSTACK_MEMORY_H
|
|
|
|
#include <stdint.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "Check.h"
|
|
|
|
class Memory {
|
|
public:
|
|
Memory() = default;
|
|
virtual ~Memory() = default;
|
|
|
|
virtual bool ReadString(uint64_t addr, std::string* string, uint64_t max_read = UINT64_MAX);
|
|
|
|
virtual bool Read(uint64_t addr, void* dst, size_t size) = 0;
|
|
|
|
inline bool ReadField(uint64_t addr, void* start, void* field, size_t size) {
|
|
if (reinterpret_cast<uintptr_t>(field) < reinterpret_cast<uintptr_t>(start)) {
|
|
return false;
|
|
}
|
|
uint64_t offset = reinterpret_cast<uintptr_t>(field) - reinterpret_cast<uintptr_t>(start);
|
|
if (__builtin_add_overflow(addr, offset, &offset)) {
|
|
return false;
|
|
}
|
|
// The read will check if offset + size overflows.
|
|
return Read(offset, field, size);
|
|
}
|
|
|
|
inline bool Read32(uint64_t addr, uint32_t* dst) {
|
|
return Read(addr, dst, sizeof(uint32_t));
|
|
}
|
|
|
|
inline bool Read64(uint64_t addr, uint64_t* dst) {
|
|
return Read(addr, dst, sizeof(uint64_t));
|
|
}
|
|
};
|
|
|
|
class MemoryBuffer : public Memory {
|
|
public:
|
|
MemoryBuffer() = default;
|
|
virtual ~MemoryBuffer() = default;
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
|
|
uint8_t* GetPtr(size_t offset);
|
|
|
|
void Resize(size_t size) { raw_.resize(size); }
|
|
|
|
uint64_t Size() { return raw_.size(); }
|
|
|
|
private:
|
|
std::vector<uint8_t> raw_;
|
|
};
|
|
|
|
class MemoryFileAtOffset : public Memory {
|
|
public:
|
|
MemoryFileAtOffset() = default;
|
|
virtual ~MemoryFileAtOffset();
|
|
|
|
bool Init(const std::string& file, uint64_t offset, uint64_t size = UINT64_MAX);
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
|
|
void Clear();
|
|
|
|
protected:
|
|
size_t size_ = 0;
|
|
size_t offset_ = 0;
|
|
uint8_t* data_ = nullptr;
|
|
};
|
|
|
|
class MemoryOffline : public MemoryFileAtOffset {
|
|
public:
|
|
MemoryOffline() = default;
|
|
virtual ~MemoryOffline() = default;
|
|
|
|
bool Init(const std::string& file, uint64_t offset);
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
|
|
private:
|
|
uint64_t start_;
|
|
};
|
|
|
|
class MemoryRemote : public Memory {
|
|
public:
|
|
MemoryRemote(pid_t pid) : pid_(pid) {}
|
|
virtual ~MemoryRemote() = default;
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
|
|
pid_t pid() { return pid_; }
|
|
|
|
protected:
|
|
virtual bool PtraceRead(uint64_t addr, long* value);
|
|
|
|
private:
|
|
pid_t pid_;
|
|
};
|
|
|
|
class MemoryLocal : public Memory {
|
|
public:
|
|
MemoryLocal() = default;
|
|
virtual ~MemoryLocal() = default;
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
};
|
|
|
|
class MemoryRange : public Memory {
|
|
public:
|
|
MemoryRange(Memory* memory, uint64_t begin, uint64_t end)
|
|
: memory_(memory), begin_(begin), length_(end - begin) {
|
|
CHECK(end > begin);
|
|
}
|
|
virtual ~MemoryRange() { delete memory_; }
|
|
|
|
bool Read(uint64_t addr, void* dst, size_t size) override;
|
|
|
|
private:
|
|
Memory* memory_;
|
|
uint64_t begin_;
|
|
uint64_t length_;
|
|
};
|
|
|
|
#endif // _LIBUNWINDSTACK_MEMORY_H
|