Commit 0181812b authored by Viktor Kutuzov's avatar Viktor Kutuzov
Browse files

[Msan] Fix strlen() and strnlen() interceptors to work on FreeBSD 

Differential Revision: http://reviews.llvm.org/D6928

llvm-svn: 225986
parent b9d33419

compiler-rt/lib/msan/msan_interceptors.cc

+4 −0
Original line number Diff line number Diff line
@@ -318,6 +318,8 @@ INTERCEPTOR(char *, stpcpy, char *dest, const char *src) { // NOLINT 
INTERCEPTOR(char *, strdup, char *src) {

  ENSURE_MSAN_INITED();

  GET_STORE_STACK_TRACE;

  // On FreeBSD strdup() leverages strlen().

  InterceptorScope interceptor_scope;

  SIZE_T n = REAL(strlen)(src);

  char *res = REAL(strdup)(src);

  CopyPoison(res, src, n + 1, &stack);
@@ -341,6 +343,8 @@ INTERCEPTOR(char *, __strdup, char *src) { 
INTERCEPTOR(char *, strndup, char *src, SIZE_T n) {

  ENSURE_MSAN_INITED();

  GET_STORE_STACK_TRACE;

  // On FreeBSD strndup() leverages strnlen().

  InterceptorScope interceptor_scope;

  SIZE_T copy_size = REAL(strnlen)(src, n);

  char *res = REAL(strndup)(src, n);

  CopyPoison(res, src, copy_size, &stack);