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   EXPECT_DEATH(
  52       failHard(),
  53       "^\\*\\*\\* Aborted at [0-9]+ \\(Unix time, try 'date -d @[0-9]+'\\) "
  54       "\\*\\*\\*\n"
  55       "\\*\\*\\* Signal 11 \\(SIGSEGV\\) \\(0x2a\\) received by PID [0-9]+ "
  56       "\\(pthread TID 0x[0-9a-f]+\\) \\(linux TID [0-9]+\\) "
  57       "\\(maybe from PID [0-9]+, UID [0-9]+\\) "
  58       "\\(code: address not mapped to object\\), "
  59       "stack trace: \\*\\*\\*\n"
  60       ".*\n"
  61       ".*    @ [0-9a-f]+.* folly::symbolizer::test::SignalHandler_Simple_Test"
  62       "::TestBody\\(\\).*\n"
  63       ".*\n"
  64       ".*    @ [0-9a-f]+.* main.*\n"
  65       ".*\n"
  66       "Callback1\n"
  67       "Callback2\n"
  68       ".*");
  69 }
  70 } // namespace test
  71 } // namespace symbolizer
  72 } // namespace folly
  73
  74 // Can't use initFacebookLight since that would install its own signal handlers
  75 // Can't use initFacebookNoSignals since we cannot depend on common
  76 int main(int argc, char** argv) {
  77   ::testing::InitGoogleTest(&argc, argv);
  78   return RUN_ALL_TESTS();
  79 }