1 //===-- dfsan_interceptors.cpp --------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file is a part of DataFlowSanitizer. 10 // 11 // Interceptors for standard library functions. 12 //===----------------------------------------------------------------------===// 13 14 #include "dfsan/dfsan.h" 15 #include "interception/interception.h" 16 #include "sanitizer_common/sanitizer_common.h" 17 18 using namespace __sanitizer; 19 20 INTERCEPTOR(void *, mmap, void *addr, SIZE_T length, int prot, int flags, 21 int fd, OFF_T offset) { 22 void *res = REAL(mmap)(addr, length, prot, flags, fd, offset); 23 if (res != (void*)-1) 24 dfsan_set_label(0, res, RoundUpTo(length, GetPageSize())); 25 return res; 26 } 27 28 INTERCEPTOR(void *, mmap64, void *addr, SIZE_T length, int prot, int flags, 29 int fd, OFF64_T offset) { 30 void *res = REAL(mmap64)(addr, length, prot, flags, fd, offset); 31 if (res != (void*)-1) 32 dfsan_set_label(0, res, RoundUpTo(length, GetPageSize())); 33 return res; 34 } 35 36 namespace __dfsan { 37 void InitializeInterceptors() { 38 static int inited = 0; 39 CHECK_EQ(inited, 0); 40 41 INTERCEPT_FUNCTION(mmap); 42 INTERCEPT_FUNCTION(mmap64); 43 inited = 1; 44 } 45 } // namespace __dfsan 46