ubsan_suppress.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "CxxUtils/ubsan_suppress.h"
#include <unistd.h>
#include <fcntl.h>
#include <dlfcn.h>
 
 
namespace CxxUtils {
 
 
class RedirStderr
{
public:
  RedirStderr();
  ~RedirStderr();
  RedirStderr (const RedirStderr&) = delete;
  RedirStderr& operator= (const RedirStderr&) = delete;
 
private:
  int m_nullfd;
  int m_stderr;
};
 
 
// Redirect stderr to /dev/null, remembering the original FD.
RedirStderr::RedirStderr()
{
  m_nullfd = open ("/dev/null", O_WRONLY);
  m_stderr = dup (2);
  if (m_nullfd >= 0) {
    dup2 (m_nullfd, 2);
  }
}
 
 
// Restore stderr to its original FD.
RedirStderr::~RedirStderr()
{
  dup2 (m_stderr, 2);
  close (m_nullfd);
  close (m_stderr);
}
 

void ubsan_suppress (void (*func)())
{
  if (dlsym (RTLD_DEFAULT, "__ubsan_handle_add_overflow") != NULL)
  {
    RedirStderr redir;
    func();
  }
  else {
    func();
  }
}
 
 
} // namespace CxxUtils