folly/experimental/symbolizer/test/SignalHandlerTest.cpp

   1 /*
   2  * Copyright 2017 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 <folly/FileUtil.h>
  21 #include <folly/Range.h>
  22 #include <folly/CPortability.h>
  23 #include <folly/portability/GTest.h>
  24
  25 namespace folly { namespace symbolizer { namespace test {
  26
  27 namespace {
  28
  29 void print(StringPiece sp) {
  30   writeFull(STDERR_FILENO, sp.data(), sp.size());
  31 }
  32
  33
  34 void callback1() {
  35   print("Callback1\n");
  36 }
  37
  38 void callback2() {
  39   print("Callback2\n");
  40 }
  41
  42 }  // namespace
  43
  44 TEST(SignalHandler, Simple) {
  45   addFatalSignalCallback(callback1);
  46   addFatalSignalCallback(callback2);
  47   installFatalSignalHandler();
  48   installFatalSignalCallbacks();
  49
  50 #ifdef FOLLY_SANITIZE_ADDRESS
  51   EXPECT_DEATH(
  52       failHard(),
  53       // Testing an ASAN-enabled binary evokes a different diagnostic.
  54       // Use a regexp that requires only the first line of that output:
  55       "^ASAN:SIGSEGV\n.*");
  56 #else
  57   EXPECT_DEATH(
  58       failHard(),
  59       "^\\*\\*\\* Aborted at [0-9]+ \\(Unix time, try 'date -d @[0-9]+'\\) "
  60       "\\*\\*\\*\n"
  61       "\\*\\*\\* Signal 11 \\(SIGSEGV\\) \\(0x2a\\) received by PID [0-9]+ "
  62       "\\(pthread TID 0x[0-9a-f]+\\) \\(linux TID [0-9]+\\) "
  63       "\\(maybe from PID [0-9]+, UID [0-9]+\\) "
  64       "\\(code: address not mapped to object\\), "
  65       "stack trace: \\*\\*\\*\n"
  66       ".*\n"
  67       ".*    @ [0-9a-f]+.* folly::symbolizer::test::SignalHandler_Simple_Test"
  68       "::TestBody\\(\\).*\n"
  69       ".*\n"
  70       ".*    @ [0-9a-f]+.* main.*\n"
  71       ".*\n"
  72       "Callback1\n"
  73       "Callback2\n");
  74 #endif
  75 }
  76
  77
  78 }}}  // namespaces
  79
  80 // Can't use initFacebookLight since that would install its own signal handlers
  81 // Can't use initFacebookNoSignals since we cannot depend on common
  82 int main(int argc, char** argv) {
  83   ::testing::InitGoogleTest(&argc, argv);
  84   return RUN_ALL_TESTS();
  85 }