Add probes for x64.

syzygy/agent/asan/memory_interceptors_impl_x64.cc

+// Copyright 2016 Google Inc. All Rights Reserved.
+//
+// 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.
+
+// The C++ implementation of the memory interceptors, intended to work on win64,
+// unlike the win32 implementation, written (generated) in pure assembly.
+// This implementation provides less functions than the original win32 one,
+// as it's intended to be used outside of Syzygy.
+
+#include "syzygy/agent/asan/error_info.h"
+#include "syzygy/agent/asan/rtl_utils.h"
+#include "syzygy/agent/asan/runtime.h"
+#include "syzygy/agent/asan/shadow.h"
+
+// The template function that performs the checks.
+// @param access_size Access size in bytes.
+// @param heap_size The virtual address space size limit in bytes.
+//     It's 8 TB for Win7 and Win8 and 128 TB for Win8.1+.
+// @param access_mode The access mode, which can be any of AccessMode values,
+//     allthough this file only exports the probes for read and write accesses.
+// @param addr The address being accessed.
+template<size_t access_size, size_t heap_size,
+         agent::asan::AccessMode access_mode>
+void asan_check(const void* addr) {
+  using namespace agent::asan;
+  if (reinterpret_cast<uintptr_t>(addr) >= heap_size ||
+      !AsanRuntime::runtime()->shadow()->IsRangeAccessible(addr, access_size)) {
+    CONTEXT ctx = {};
+    ::RtlCaptureContext(&ctx);
+    AsanContext asan_ctx = {};
+    ContextToAsanContext(ctx, &asan_ctx);
+    ReportBadMemoryAccess(addr, access_mode, access_size, asan_ctx);
+  }
+}
+
+// A few macros to instantiate 'asan_check' and export the instantiations
+// with appropriate names.
+
+#define EXPORT_INTERCEPTOR_READ(access_size, suffix, heap_size)               \
+void asan_check_##access_size##_byte_read_access_no_flags_##suffix(           \
+         const void* addr) {                                                  \
+  return asan_check<access_size, heap_size,                                   \
+                    agent::asan::ASAN_READ_ACCESS>(addr);                     \
+}                                                                             \
+void asan_check_##access_size##_byte_read_access_##suffix(const void* addr) { \
+  return asan_check<access_size, heap_size,                                   \
+                    agent::asan::ASAN_READ_ACCESS>(addr);                     \
+}
+
+#define EXPORT_INTERCEPTOR_WRITE(access_size, suffix, heap_size)               \
+void asan_check_##access_size##_byte_write_access_no_flags_##suffix(           \
+         const void* addr) {                                                   \
+  return asan_check<access_size, heap_size,                                    \
+                    agent::asan::ASAN_WRITE_ACCESS>(addr);                     \
+}                                                                              \
+void asan_check_##access_size##_byte_write_access_##suffix(const void* addr) { \
+  return asan_check<access_size, heap_size,                                    \
+                    agent::asan::ASAN_WRITE_ACCESS>(addr);                     \
+}
+
+#define EXPORT_INTERCEPTOR(access_size, suffix, heap_size) \
+  EXPORT_INTERCEPTOR_READ(access_size, suffix, heap_size)  \
+  EXPORT_INTERCEPTOR_WRITE(access_size, suffix, heap_size)
+
+#define EXPORT_INTERCEPTORS_ALL_SIZES(suffix, heap_size) \

[Sébastien Marchand, 2016/09/13 16:03:09]

One more comment, I don't think that |heap_size| is an appropriate name, maybe
address_space_size ?

+  EXPORT_INTERCEPTOR(1,  suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(2,  suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(4,  suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(8,  suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(10, suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(16, suffix, heap_size)              \
+  EXPORT_INTERCEPTOR(32, suffix, heap_size)
+
+extern "C" {
+  void asan_no_check() { return; }

[Sébastien Marchand, 2016/09/13 15:48:16]

I don't think that we want to export the function like this, instead we should
update system_interceptor_generator.py to generate the proper def file depending
on the bitness.

+  void asan_string_no_check() { return; }
+  void* asan_shadow_references[] = { nullptr };
+  EXPORT_INTERCEPTORS_ALL_SIZES(8tb,   static_cast<size_t>(8)   << 40)
+  EXPORT_INTERCEPTORS_ALL_SIZES(128tb, static_cast<size_t>(128) << 40)
+}