aos/die.cc

#include "aos/die.h"

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include <stdint.h>

#include <string>
#include <atomic>

namespace aos {

void Die(const char *format, ...) {
  va_list args;
  va_start(args, format);
  VDie(format, args);
  // va_end(args)  // not because VDie never returns
}

namespace {

// Calculates the filename to dump the message into.
const std::string GetFilename() {
#ifdef __VXWORKS__
  const char *name = taskName(0);  // get the name of this task
  if (name == NULL) name = "<unknown>";
  const std::string first_part = "/aos_fatal_error.";
  return first_part + std::string(name);
#else
  char *filename;
  if (asprintf(&filename, "/tmp/aos_fatal_error.%jd",
               static_cast<intmax_t>(getpid())) > 0) {
    std::string r(filename);
    free(filename);
    return r;
  } else {
    fprintf(stderr, "aos fatal: asprintf(%p, \"thingie with %%jd\", %jd)"
            " failed with %d\n", &filename,
            static_cast<intmax_t>(getpid()), errno);
    return std::string();
  }
#endif
}

::std::atomic_bool test_mode(false);

}  // namespace

void VDie(const char *format, va_list args_in) {
  // We don't bother va_ending either of these because we're going nowhere and
  // vxworks has some weird bugs that sometimes show up...
  va_list args1, args2;

  fputs("aos fatal: ERROR!! details following\n", stderr);
  va_copy(args1, args_in);
  vfprintf(stderr, format, args1);
  if (!test_mode.load()) {
    fputs("aos fatal: ERROR!! see stderr for details\n", stdout);

    const std::string filename = GetFilename();
    if (!filename.empty()) {
      FILE *error_file = fopen(filename.c_str(), "w");
      if (error_file != NULL) {
        va_copy(args2, args_in);
        vfprintf(error_file, format, args2);
        fclose(error_file);
      } else {
        fprintf(stderr, "aos fatal: fopen('%s', \"w\") failed with %d\n",
                filename.c_str(), errno);
      }
    }
  }

  abort();
}

void SetDieTestMode(bool new_test_mode) {
  test_mode.store(new_test_mode);
}

}  // namespace aos