src/signalhandler_unittest.cc

   1 // Copyright (c) 2008, Google Inc.
   2 // All rights reserved.
   3 //
   4 // Redistribution and use in source and binary forms, with or without
   5 // modification, are permitted provided that the following conditions are
   6 // met:
   7 //
   8 //     * Redistributions of source code must retain the above copyright
   9 // notice, this list of conditions and the following disclaimer.
  10 //     * Redistributions in binary form must reproduce the above
  11 // copyright notice, this list of conditions and the following disclaimer
  12 // in the documentation and/or other materials provided with the
  13 // distribution.
  14 //     * Neither the name of Google Inc. nor the names of its
  15 // contributors may be used to endorse or promote products derived from
  16 // this software without specific prior written permission.
  17 //
  18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29 //
  30 // Author: Satoru Takabayashi
  31 //
  32 // This is a helper binary for testing signalhandler.cc.  The actual test
  33 // is done in signalhandler_unittest.sh.
  34
  35 #include "utilities.h"
  36
  37 #include <pthread.h>
  38 #include <signal.h>
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string>
  42 #include "glog/logging.h"
  43
  44 using namespace GOOGLE_NAMESPACE;
  45
  46 void* DieInThread(void*) {
  47   fprintf(stderr, "0x%lx is dying\n", pthread_self());
  48   // Use volatile to prevent from these to be optimized away.
  49   volatile int a = 0;
  50   volatile int b = 1 / a;
  51 }
  52
  53 void WriteToStdout(const char* data, int size) {
  54   write(STDOUT_FILENO, data, size);
  55 }
  56
  57 int main(int argc, char **argv) {
  58 #if defined(HAVE_STACKTRACE) && defined(HAVE_SYMBOLIZE)
  59   InitGoogleLogging(argv[0]);
  60 #ifdef HAVE_LIB_GFLAGS
  61   ParseCommandLineFlags(&argc, &argv, true);
  62 #endif
  63   InstallFailureSignalHandler();
  64   const std::string command = argc > 1 ? argv[1] : "none";
  65   if (command == "segv") {
  66     // We'll check if this is outputted.
  67     LOG(INFO) << "create the log file";
  68     LOG(INFO) << "a message before segv";
  69     // We assume 0xDEAD is not writable.
  70     int *a = (int*)0xDEAD;
  71     *a = 0;
  72   } else if (command == "loop") {
  73     fprintf(stderr, "looping\n");
  74     while (true);
  75   } else if (command == "die_in_thread") {
  76     pthread_t thread;
  77     pthread_create(&thread, NULL, &DieInThread, NULL);
  78     pthread_join(thread, NULL);
  79   } else if (command == "dump_to_stdout") {
  80     InstallFailureWriter(WriteToStdout);
  81     abort();
  82   } else {
  83     // Tell the shell script
  84     puts("OK");
  85   }
  86 #endif
  87   return 0;
  88 }