ceph/src/seastar/src/core/on_internal_error.cc

   1 /*
   2  * This file is open source software, licensed to you under the terms
   3  * of the Apache License, Version 2.0 (the "License").  See the NOTICE file
   4  * distributed with this work for additional information regarding copyright
   5  * ownership.  You may not use this file except in compliance with the License.
   6  *
   7  * You may obtain a copy of the License at
   8  *
   9  *   http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing,
  12  * software distributed under the License is distributed on an
  13  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14  * KIND, either express or implied.  See the License for the
  15  * specific language governing permissions and limitations
  16  * under the License.
  17  */
  18 /*
  19  * Copyright 2020 ScyllaDB
  20  */
  21
  22 #include <seastar/core/on_internal_error.hh>
  23 #include <seastar/util/backtrace.hh>
  24 #include <seastar/util/log.hh>
  25
  26 #include <atomic>
  27
  28 static std::atomic<bool> abort_on_internal_error{false};
  29
  30 using namespace seastar;
  31
  32 void seastar::set_abort_on_internal_error(bool do_abort) {
  33     abort_on_internal_error.store(do_abort);
  34 }
  35
  36 void seastar::on_internal_error(logger& logger, std::string_view msg) {
  37     if (abort_on_internal_error.load()) {
  38         logger.error("{}, at: {}", msg, current_backtrace());
  39         abort();
  40     } else {
  41         throw_with_backtrace<std::runtime_error>(std::string(msg));
  42     }
  43 }
  44
  45 void seastar::on_internal_error(logger& logger, std::exception_ptr ex) {
  46     if (abort_on_internal_error.load()) {
  47         logger.error("{}", ex);
  48         abort();
  49     } else {
  50         std::rethrow_exception(std::move(ex));
  51     }
  52 }
  53
  54 void seastar::on_internal_error_noexcept(logger& logger, std::string_view msg) noexcept {
  55     logger.error("{}, at: {}", msg, current_backtrace());
  56     if (abort_on_internal_error.load()) {
  57         abort();
  58     }
  59 }