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 #if defined(HAVE_PTHREAD)
  38 # include <pthread.h>
  39 #endif
  40 #include <signal.h>
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string>
  44 #include "glog/logging.h"
  45
  46 #ifdef HAVE_LIB_GFLAGS
  47 #include <gflags/gflags.h>
  48 using namespace GFLAGS_NAMESPACE;
  49 #endif
  50
  51 using namespace GOOGLE_NAMESPACE;
  52
  53 void* DieInThread(void*) {
  54   // We assume pthread_t is an integral number or a pointer, rather
  55   // than a complex struct.  In some environments, pthread_self()
  56   // returns an uint64 but in some other environments pthread_self()
  57   // returns a pointer.  Hence we use C-style cast here, rather than
  58   // reinterpret/static_cast, to support both types of environments.
  59   fprintf(stderr, "0x%lx is dying\n", (long)pthread_self());
  60   // Use volatile to prevent from these to be optimized away.
  61   volatile int a = 0;
  62   volatile int b = 1 / a;
  63   fprintf(stderr, "We should have died: b=%d\n", b);
  64   return NULL;
  65 }
  66
  67 void WriteToStdout(const char* data, int size) {
  68   if (write(STDOUT_FILENO, data, size) < 0) {
  69     // Ignore errors.
  70   }
  71 }
  72
  73 int main(int argc, char **argv) {
  74 #if defined(HAVE_STACKTRACE) && defined(HAVE_SYMBOLIZE)
  75   InitGoogleLogging(argv[0]);
  76 #ifdef HAVE_LIB_GFLAGS
  77   ParseCommandLineFlags(&argc, &argv, true);
  78 #endif
  79   InstallFailureSignalHandler();
  80   const std::string command = argc > 1 ? argv[1] : "none";
  81   if (command == "segv") {
  82     // We'll check if this is outputted.
  83     LOG(INFO) << "create the log file";
  84     LOG(INFO) << "a message before segv";
  85     // We assume 0xDEAD is not writable.
  86     int *a = (int*)0xDEAD;
  87     *a = 0;
  88   } else if (command == "loop") {
  89     fprintf(stderr, "looping\n");
  90     while (true);
  91   } else if (command == "die_in_thread") {
  92 #if defined(HAVE_PTHREAD)
  93     pthread_t thread;
  94     pthread_create(&thread, NULL, &DieInThread, NULL);
  95     pthread_join(thread, NULL);
  96 #else
  97     fprintf(stderr, "no pthread\n");
  98     return 1;
  99 #endif
 100   } else if (command == "dump_to_stdout") {
 101     InstallFailureWriter(WriteToStdout);
 102     abort();
 103   } else if (command == "installed") {
 104     fprintf(stderr, "signal handler installed: %s\n",
 105         IsFailureSignalHandlerInstalled() ? "true" : "false");
 106   } else {
 107     // Tell the shell script
 108     puts("OK");
 109   }
 110 #endif
 111   return 0;
 112 }