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