xref: /freebsd/contrib/llvm-project/compiler-rt/lib/hwasan/hwasan_memintrinsics.cpp (revision 06c3fb2749bda94cb5201f81ffdb8fa6c3161b2e)
10b57cec5SDimitry Andric //===-- hwasan_memintrinsics.cpp --------------------------------*- C++ -*-===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric ///
90b57cec5SDimitry Andric /// \file
100b57cec5SDimitry Andric /// This file is a part of HWAddressSanitizer and contains HWASAN versions of
110b57cec5SDimitry Andric /// memset, memcpy and memmove
120b57cec5SDimitry Andric ///
130b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
140b57cec5SDimitry Andric 
150b57cec5SDimitry Andric #include <string.h>
160b57cec5SDimitry Andric #include "hwasan.h"
170b57cec5SDimitry Andric #include "hwasan_checks.h"
180b57cec5SDimitry Andric #include "hwasan_flags.h"
190b57cec5SDimitry Andric #include "hwasan_interface_internal.h"
200b57cec5SDimitry Andric #include "sanitizer_common/sanitizer_libc.h"
210b57cec5SDimitry Andric 
220b57cec5SDimitry Andric using namespace __hwasan;
230b57cec5SDimitry Andric 
__hwasan_memset(void * block,int c,uptr size)240b57cec5SDimitry Andric void *__hwasan_memset(void *block, int c, uptr size) {
250b57cec5SDimitry Andric   CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
260b57cec5SDimitry Andric       reinterpret_cast<uptr>(block), size);
27fe6060f1SDimitry Andric   return memset(block, c, size);
280b57cec5SDimitry Andric }
290b57cec5SDimitry Andric 
__hwasan_memcpy(void * to,const void * from,uptr size)300b57cec5SDimitry Andric void *__hwasan_memcpy(void *to, const void *from, uptr size) {
310b57cec5SDimitry Andric   CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
320b57cec5SDimitry Andric       reinterpret_cast<uptr>(to), size);
330b57cec5SDimitry Andric   CheckAddressSized<ErrorAction::Recover, AccessType::Load>(
340b57cec5SDimitry Andric       reinterpret_cast<uptr>(from), size);
35fe6060f1SDimitry Andric   return memcpy(to, from, size);
360b57cec5SDimitry Andric }
370b57cec5SDimitry Andric 
__hwasan_memmove(void * to,const void * from,uptr size)380b57cec5SDimitry Andric void *__hwasan_memmove(void *to, const void *from, uptr size) {
390b57cec5SDimitry Andric   CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
400b57cec5SDimitry Andric       reinterpret_cast<uptr>(to), size);
410b57cec5SDimitry Andric   CheckAddressSized<ErrorAction::Recover, AccessType::Load>(
420b57cec5SDimitry Andric       reinterpret_cast<uptr>(from), size);
431fd87a68SDimitry Andric   return memmove(to, from, size);
440b57cec5SDimitry Andric }
45*06c3fb27SDimitry Andric 
__hwasan_memset_match_all(void * block,int c,uptr size,u8 match_all_tag)46*06c3fb27SDimitry Andric void *__hwasan_memset_match_all(void *block, int c, uptr size,
47*06c3fb27SDimitry Andric                                 u8 match_all_tag) {
48*06c3fb27SDimitry Andric   if (GetTagFromPointer(reinterpret_cast<uptr>(block)) != match_all_tag)
49*06c3fb27SDimitry Andric     CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
50*06c3fb27SDimitry Andric         reinterpret_cast<uptr>(block), size);
51*06c3fb27SDimitry Andric   return memset(block, c, size);
52*06c3fb27SDimitry Andric }
53*06c3fb27SDimitry Andric 
__hwasan_memcpy_match_all(void * to,const void * from,uptr size,u8 match_all_tag)54*06c3fb27SDimitry Andric void *__hwasan_memcpy_match_all(void *to, const void *from, uptr size,
55*06c3fb27SDimitry Andric                                 u8 match_all_tag) {
56*06c3fb27SDimitry Andric   if (GetTagFromPointer(reinterpret_cast<uptr>(to)) != match_all_tag)
57*06c3fb27SDimitry Andric     CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
58*06c3fb27SDimitry Andric         reinterpret_cast<uptr>(to), size);
59*06c3fb27SDimitry Andric   if (GetTagFromPointer(reinterpret_cast<uptr>(from)) != match_all_tag)
60*06c3fb27SDimitry Andric     CheckAddressSized<ErrorAction::Recover, AccessType::Load>(
61*06c3fb27SDimitry Andric         reinterpret_cast<uptr>(from), size);
62*06c3fb27SDimitry Andric   return memcpy(to, from, size);
63*06c3fb27SDimitry Andric }
64*06c3fb27SDimitry Andric 
__hwasan_memmove_match_all(void * to,const void * from,uptr size,u8 match_all_tag)65*06c3fb27SDimitry Andric void *__hwasan_memmove_match_all(void *to, const void *from, uptr size,
66*06c3fb27SDimitry Andric                                  u8 match_all_tag) {
67*06c3fb27SDimitry Andric   if (GetTagFromPointer(reinterpret_cast<uptr>(to)) != match_all_tag)
68*06c3fb27SDimitry Andric     CheckAddressSized<ErrorAction::Recover, AccessType::Store>(
69*06c3fb27SDimitry Andric         reinterpret_cast<uptr>(to), size);
70*06c3fb27SDimitry Andric   if (GetTagFromPointer(reinterpret_cast<uptr>(from)) != match_all_tag)
71*06c3fb27SDimitry Andric     CheckAddressSized<ErrorAction::Recover, AccessType::Load>(
72*06c3fb27SDimitry Andric         reinterpret_cast<uptr>(from), size);
73*06c3fb27SDimitry Andric   return memmove(to, from, size);
74*06c3fb27SDimitry Andric }
75