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/CPortability.h>
  21 #include <folly/FileUtil.h>
  22 #include <folly/Range.h>
  23 #include <folly/portability/GTest.h>
  24
  25 namespace folly {
  26 namespace symbolizer {
  27 namespace test {
  28
  29 namespace {
  30
  31 void print(StringPiece sp) {
  32   writeFull(STDERR_FILENO, sp.data(), sp.size());
  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       "\\(pthread TID 0x[0-9a-f]+\\) \\(linux TID [0-9]+\\) "
  64       "\\(maybe from PID [0-9]+, UID [0-9]+\\) "
  65       "\\(code: address not mapped to object\\), "
  66       "stack trace: \\*\\*\\*\n"
  67       ".*\n"
  68       ".*    @ [0-9a-f]+.* folly::symbolizer::test::SignalHandler_Simple_Test"
  69       "::TestBody\\(\\).*\n"
  70       ".*\n"
  71       ".*    @ [0-9a-f]+.* main.*\n"
  72       ".*\n"
  73       "Callback1\n"
  74       "Callback2\n");
  75 #endif
  76 }
  77 } // namespace test
  78 } // namespace symbolizer
  79 } // namespace folly
  80
  81 // Can't use initFacebookLight since that would install its own signal handlers
  82 // Can't use initFacebookNoSignals since we cannot depend on common
  83 int main(int argc, char** argv) {
  84   ::testing::InitGoogleTest(&argc, argv);
  85   return RUN_ALL_TESTS();
  86 }