1    :  // Copyright 2015 Google Inc. All Rights Reserved.
   2    :  //
   3    :  // Licensed under the Apache License, Version 2.0 (the "License");
   4    :  // you may not use this file except in compliance with the License.
   5    :  // You may obtain a copy of the License at
   6    :  //
   7    :  //     http://www.apache.org/licenses/LICENSE-2.0
   8    :  //
   9    :  // Unless required by applicable law or agreed to in writing, software
  10    :  // distributed under the License is distributed on an "AS IS" BASIS,
  11    :  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12    :  // See the License for the specific language governing permissions and
  13    :  // limitations under the License.
  14    :  //
  15    :  // Implements a bit source that reads its own process' memory.
  16    :  #include "syzygy/refinery/testing/self_bit_source.h"
  17    :  
  18    :  #include <Windows.h>
  19    :  
  20  m :  namespace testing {
  21    :  
  22  m :  SelfBitSource::~SelfBitSource() {
  23  m :  }
  24    :  
  25  m :  bool SelfBitSource::GetAll(const refinery::AddressRange& range,
  26  m :                             void* data_ptr) {
  27  m :    DCHECK(range.IsValid());
  28  m :    DCHECK(data_ptr);
  29    :  
  30  m :    size_t read_bytes = 0;
  31  m :    if (!GetFrom(range, &read_bytes, data_ptr))
  32  m :      return false;
  33  m :    if (read_bytes != range.size())
  34  m :      return false;
  35    :  
  36  m :    return true;
  37  m :  }
  38    :  
  39  m :  bool SelfBitSource::GetFrom(const refinery::AddressRange& range,
  40  m :                              size_t* data_cnt,
  41  m :                              void* data_ptr) {
  42  m :    DCHECK(range.IsValid());
  43  m :    DCHECK(data_cnt);
  44  m :    DCHECK(data_ptr);
  45    :  
  46  m :    *data_cnt = 0;
  47    :  
  48  m :    DWORD read_bytes = 0;
  49  m :    BOOL succeeded = ::ReadProcessMemory(
  50  m :        ::GetCurrentProcess(), reinterpret_cast<const void*>(range.start()),
  51  m :        data_ptr, range.size(), &read_bytes);
  52  m :    if (!succeeded)
  53  m :      return false;
  54  m :    *data_cnt = read_bytes;
  55    :  
  56  m :    return read_bytes != 0;
  57  m :  }
  58    :  
  59  m :  bool SelfBitSource::HasSome(const refinery::AddressRange& range) {
  60    :    // TODO(siggi): Fixme!
  61  m :    return true;
  62  m :  }
  63    :  
  64  m :  }  // namespace testing