folly/experimental/symbolizer/test/SignalHandlerTest.cpp

   1 /*
   2  * Copyright 2014 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <folly/experimental/symbolizer/test/SignalHandlerTest.h>
  18 #include <folly/experimental/symbolizer/SignalHandler.h>
  19
  20 #include <gtest/gtest.h>
  21
  22 #include <folly/FileUtil.h>
  23 #include <folly/Range.h>
  24 #include <folly/CPortability.h>
  25
  26 namespace folly { namespace symbolizer { namespace test {
  27
  28 namespace {
  29
  30 void print(StringPiece sp) {
  31   writeFull(STDERR_FILENO, sp.data(), sp.size());
  32 }
  33
  34
  35 void callback1() {
  36   print("Callback1\n");
  37 }
  38
  39 void callback2() {
  40   print("Callback2\n");
  41 }
  42
  43 }  // namespace
  44
  45 TEST(SignalHandler, Simple) {
  46   addFatalSignalCallback(callback1);
  47   addFatalSignalCallback(callback2);
  48   installFatalSignalHandler();
  49   installFatalSignalCallbacks();
  50
  51 #ifdef FOLLY_SANITIZE_ADDRESS
  52   EXPECT_DEATH(
  53       failHard(),
  54       // Testing an ASAN-enabled binary evokes a different diagnostic.
  55       // Use a regexp that requires only the first line of that output:
  56       "^ASAN:SIGSEGV\n.*");
  57 #else
  58   EXPECT_DEATH(
  59       failHard(),
  60       "^\\*\\*\\* Aborted at [0-9]+ \\(Unix time, try 'date -d @[0-9]+'\\) "
  61       "\\*\\*\\*\n"
  62       "\\*\\*\\* Signal 11 \\(SIGSEGV\\) \\(0x2a\\) received by PID [0-9]+ "
  63       "\\(TID 0x[0-9a-f]+\\), stack trace: \\*\\*\\*\n"
  64       ".*\n"
  65       "    @ [0-9a-f]+ folly::symbolizer::test::SignalHandler_Simple_Test"
  66       "::TestBody\\(\\)\n"
  67       ".*\n"
  68       "    @ [0-9a-f]+ main\n"
  69       ".*\n"
  70       "Callback1\n"
  71       "Callback2\n"
  72       );
  73 #endif
  74 }
  75
  76
  77 }}}  // namespaces