/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <iostream>

#include <folly/init/Init.h>
#include <folly/logging/xlog.h>
#include <folly/portability/Stdlib.h>

DEFINE_string(
    category,
    "",
    "Crash with a message to this category instead of the default");
DEFINE_bool(crash, true, "Crash with a fatal log message.");
DEFINE_bool(
    check_debug,
    false,
    "Print whether this binary was built in debug mode "
    "and then exit successfully");

DEFINE_bool(fail_fatal_xlog_if, false, "Fail an XLOG_IF(FATAL) check.");
DEFINE_bool(fail_dfatal_xlog_if, false, "Fail an XLOG_IF(DFATAL) check.");
DEFINE_bool(fail_xcheck, false, "Fail an XCHECK() test.");
DEFINE_bool(
    fail_xcheck_nomsg,
    false,
    "Fail an XCHECK() test with no additional message.");
DEFINE_bool(fail_xdcheck, false, "Fail an XDCHECK() test.");

DEFINE_int32(xcheck_eq0, 0, "Check this value using XCHECK_EQ(value, 0)");
DEFINE_int32(xcheck_ne0, 1, "Check this value using XCHECK_NE 0)");
DEFINE_int32(xcheck_lt0, -1, "Check this value using XCHECK_LT(value, 0)");
DEFINE_int32(xcheck_le0, 0, "Check this value using XCHECK_LE(value, 0)");
DEFINE_int32(xcheck_gt0, 1, "Check this value using XCHECK_GT(value, 0)");
DEFINE_int32(xcheck_ge0, 0, "Check this value using XCHECK_GE(value, 0)");

DEFINE_int32(xdcheck_eq0, 0, "Check this value using XDCHECK_EQ(value, 0)");
DEFINE_int32(xdcheck_ne0, 1, "Check this value using XDCHECK_NE 0)");
DEFINE_int32(xdcheck_lt0, -1, "Check this value using XDCHECK_LT(value, 0)");
DEFINE_int32(xdcheck_le0, 0, "Check this value using XDCHECK_LE(value, 0)");
DEFINE_int32(xdcheck_gt0, 1, "Check this value using XDCHECK_GT(value, 0)");
DEFINE_int32(xdcheck_ge0, 0, "Check this value using XDCHECK_GE(value, 0)");

DEFINE_bool(
    test_xcheck_eq_evalutates_once,
    false,
    "Test an XCHECK_EQ() statement where the arguments have side effects");
DEFINE_bool(
    xcheck_eq_custom_struct,
    false,
    "Test an XCHECK_EQ() statement with a custom structure, "
    "to test log message formatting");
DEFINE_bool(
    xcheck_eq_pointers,
    false,
    "Test an XCHECK_EQ() statement with pointer arguments");

namespace {
/**
 * Helper class to optionally log a fatal message during static initialization
 * or destruction.
 *
 * Since command line arguments have not been processed during static
 * initialization, we check an environment variable.
 */
class InitChecker {
 public:
  InitChecker() : value_{getenv("CRASH_DURING_INIT")} {
    if (value_ && strcmp(value_, "shutdown") != 0) {
      XLOG(FATAL) << "crashing during static initialization";
    }
  }
  ~InitChecker() {
    if (value_) {
      XLOG(FATAL) << "crashing during static destruction";
    }
  }

  const char* value_{nullptr};
};

static InitChecker initChecker;
} // namespace

namespace {
int runHelper() {
  if (!FLAGS_category.empty()) {
    folly::Logger logger{FLAGS_category};
    FB_LOG(logger, FATAL, "crashing to category ", FLAGS_category);
  }

  if (!FLAGS_crash) {
    return 0;
  }

  XLOG(FATAL) << "test program crashing!";
  // Even though this function is defined to return an integer, the compiler
  // should be able to detect that XLOG(FATAL) never returns.  It shouldn't
  // complain that we don't return an integer here.
}
} // namespace

std::string fbLogFatalCheck() {
  folly::Logger logger("some.category");
  FB_LOG(logger, FATAL) << "we always crash";
  // This function mostly exists to make sure the compiler does not warn
  // about a missing return statement here.
}

std::string xcheckFatalCheck() {
  XCHECK(false);
  // This function mostly exists to make sure the compiler does not warn
  // about a missing return statement here.
}

struct MyStruct {
  MyStruct(uint32_t a_, uint32_t b_) : a(a_), b(b_) {}
  uint32_t a;
  uint32_t b;
};
bool operator==(const MyStruct& s1, const MyStruct& s2) {
  return (s1.a == s2.a) && (s1.b == s2.b);
}
bool operator<=(const MyStruct& s1, const MyStruct& s2) {
  return !(s1 == s2);
}

/*
 * This is a simple helper program to exercise the LOG(FATAL) functionality.
 */
int main(int argc, char* argv[]) {
  folly::Init init(&argc, &argv);

  if (FLAGS_check_debug) {
    std::cout << "DEBUG=" << static_cast<int>(folly::kIsDebug) << "\n";
    return 0;
  }

  XLOG_IF(FATAL, FLAGS_fail_fatal_xlog_if) << "--fail_fatal_xlog_if specified!";
  XLOG_IF(DFATAL, FLAGS_fail_dfatal_xlog_if)
      << "--fail_dfatal_xlog_if specified!";
  XCHECK(!FLAGS_fail_xcheck) << ": --fail_xcheck specified!";
  XCHECK(!FLAGS_fail_xcheck_nomsg);
  XDCHECK(!FLAGS_fail_xdcheck) << ": --fail_xdcheck specified!";

  XCHECK_EQ(FLAGS_xcheck_eq0, 0) << " extra user args";
  XCHECK_NE(FLAGS_xcheck_ne0, 0, " extra user args");
  XCHECK_LT(FLAGS_xcheck_lt0, 0, " extra ", "user", " args");
  XCHECK_LE(FLAGS_xcheck_le0, 0, " extra ", "user") << " args";
  XCHECK_GT(FLAGS_xcheck_gt0, 0) << " extra user args";
  XCHECK_GE(FLAGS_xcheck_ge0, 0) << " extra user args";
  XDCHECK_EQ(FLAGS_xdcheck_eq0, 0) << " extra user args";
  XDCHECK_NE(FLAGS_xdcheck_ne0, 0, " extra user args");
  XDCHECK_LT(FLAGS_xdcheck_lt0, 0) << " extra user args";
  XDCHECK_LE(FLAGS_xdcheck_le0, 0) << " extra user args";
  XDCHECK_GT(FLAGS_xdcheck_gt0, 0) << " extra user args";
  XDCHECK_GE(FLAGS_xdcheck_ge0, 0) << " extra user args";

  if (FLAGS_test_xcheck_eq_evalutates_once) {
    // Make sure XCHECK_EQ() only evaluates "++x" once,
    // and logs that it equals 6 and not 7.
    int x = 5;
    XCHECK_EQ(++x, 7);
  }
  if (FLAGS_xcheck_eq_custom_struct) {
    auto m = MyStruct(1, 0x12abcdef);
    XCHECK_EQ(MyStruct(1, 2), m);
  }
  if (FLAGS_xcheck_eq_pointers) {
    int localInt = 5;
    XCHECK_EQ(&argc, &localInt);
  }

  // Do the remainder of the work in a separate helper function.
  //
  // The main reason for putting this in a helper function is to ensure that
  // the compiler does not warn about missing return statements on XLOG(FATAL)
  // code paths.  Unfortunately it appears like some compilers always suppress
  // this warning for main().
  return runHelper();
}
