1 /*
2  * Copyright (C) 2016 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <stdint.h>
20 
21 #include <unwindstack/Elf.h>
22 #include <unwindstack/Memory.h>
23 #include <unwindstack/Regs.h>
24 
25 #include "Check.h"
26 
27 namespace unwindstack {
28 
29 class RegsFake : public Regs {
30  public:
RegsFake(uint16_t total_regs)31   RegsFake(uint16_t total_regs) : Regs(total_regs, Regs::Location(Regs::LOCATION_UNKNOWN, 0)) {}
32   virtual ~RegsFake() = default;
33 
Arch()34   ArchEnum Arch() override { return fake_arch_; }
RawData()35   void* RawData() override { return nullptr; }
pc()36   uint64_t pc() override { return fake_pc_; }
sp()37   uint64_t sp() override { return fake_sp_; }
set_pc(uint64_t pc)38   void set_pc(uint64_t pc) override { fake_pc_ = pc; }
set_sp(uint64_t sp)39   void set_sp(uint64_t sp) override { fake_sp_ = sp; }
40 
SetPcFromReturnAddress(Memory *)41   bool SetPcFromReturnAddress(Memory*) override {
42     if (!fake_return_address_valid_) {
43       return false;
44     }
45     fake_pc_ = fake_return_address_;
46     return true;
47   }
48 
IterateRegisters(std::function<void (const char *,uint64_t)>)49   void IterateRegisters(std::function<void(const char*, uint64_t)>) override {}
50 
Is32Bit()51   bool Is32Bit() {
52     CHECK(fake_arch_ != ARCH_UNKNOWN);
53     return fake_arch_ == ARCH_ARM || fake_arch_ == ARCH_X86;
54   }
55 
StepIfSignalHandler(uint64_t,Elf *,Memory *)56   bool StepIfSignalHandler(uint64_t, Elf*, Memory*) override { return false; }
57 
FakeSetArch(ArchEnum arch)58   void FakeSetArch(ArchEnum arch) { fake_arch_ = arch; }
FakeSetDexPc(uint64_t dex_pc)59   void FakeSetDexPc(uint64_t dex_pc) { dex_pc_ = dex_pc; }
FakeSetReturnAddress(uint64_t return_address)60   void FakeSetReturnAddress(uint64_t return_address) { fake_return_address_ = return_address; }
FakeSetReturnAddressValid(bool valid)61   void FakeSetReturnAddressValid(bool valid) { fake_return_address_valid_ = valid; }
62 
Clone()63   Regs* Clone() override { return nullptr; }
64 
65  private:
66   ArchEnum fake_arch_ = ARCH_UNKNOWN;
67   uint64_t fake_pc_ = 0;
68   uint64_t fake_sp_ = 0;
69   bool fake_return_address_valid_ = false;
70   uint64_t fake_return_address_ = 0;
71 };
72 
73 template <typename TypeParam>
74 class RegsImplFake : public RegsImpl<TypeParam> {
75  public:
RegsImplFake(uint16_t total_regs)76   RegsImplFake(uint16_t total_regs)
77       : RegsImpl<TypeParam>(total_regs, Regs::Location(Regs::LOCATION_UNKNOWN, 0)) {}
78   virtual ~RegsImplFake() = default;
79 
Arch()80   ArchEnum Arch() override { return ARCH_UNKNOWN; }
pc()81   uint64_t pc() override { return fake_pc_; }
sp()82   uint64_t sp() override { return fake_sp_; }
set_pc(uint64_t pc)83   void set_pc(uint64_t pc) override { fake_pc_ = pc; }
set_sp(uint64_t sp)84   void set_sp(uint64_t sp) override { fake_sp_ = sp; }
set_pseudo_reg(uint64_t reg)85   void set_pseudo_reg(uint64_t reg) { fake_pseudo_reg_ = reg; }
86 
SetPcFromReturnAddress(Memory *)87   bool SetPcFromReturnAddress(Memory*) override { return false; }
StepIfSignalHandler(uint64_t,Elf *,Memory *)88   bool StepIfSignalHandler(uint64_t, Elf*, Memory*) override { return false; }
89 
SetPseudoRegister(uint16_t reg,uint64_t value)90   bool SetPseudoRegister(uint16_t reg, uint64_t value) override {
91     if (fake_pseudo_reg_ != reg) {
92       return false;
93     }
94     fake_pseudo_reg_value_ = value;
95     return true;
96   }
GetPseudoRegister(uint16_t reg,uint64_t * value)97   bool GetPseudoRegister(uint16_t reg, uint64_t* value) override {
98     if (fake_pseudo_reg_ != reg) {
99       return false;
100     }
101     *value = fake_pseudo_reg_value_;
102     return true;
103   }
104 
Clone()105   Regs* Clone() override { return nullptr; }
106 
107  private:
108   uint64_t fake_pc_ = 0;
109   uint64_t fake_sp_ = 0;
110   uint16_t fake_pseudo_reg_ = 0;
111   uint64_t fake_pseudo_reg_value_ = 0;
112 };
113 
114 }  // namespace unwindstack
115