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