ceph/src/seastar/src/util/alloc_failure_injector.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 2017 ScyllaDB
  20  */
  21
  22 #include <seastar/util/alloc_failure_injector.hh>
  23 #include <seastar/util/backtrace.hh>
  24 #include <seastar/util/log.hh>
  25 #include <seastar/util/defer.hh>
  26
  27 namespace seastar {
  28 namespace memory {
  29
  30 static logger log("failure_injector");
  31
  32 thread_local alloc_failure_injector the_alloc_failure_injector;
  33
  34 void alloc_failure_injector::fail() {
  35     _failed = true;
  36     cancel();
  37     if (log.is_enabled(log_level::trace)) {
  38         log.trace("Failing at {}", current_backtrace());
  39     }
  40     _on_alloc_failure();
  41 }
  42
  43 void alloc_failure_injector::run_with_callback(noncopyable_function<void()> callback, noncopyable_function<void()> to_run) {
  44     auto restore = defer([this, prev = std::exchange(_on_alloc_failure, std::move(callback))] () mutable {
  45         _on_alloc_failure = std::move(prev);
  46     });
  47     to_run();
  48 }
  49
  50 void with_allocation_failures(noncopyable_function<void()> func) {
  51     auto& injector = memory::local_failure_injector();
  52     uint64_t i = 0;
  53     do {
  54         try {
  55             injector.fail_after(i++);
  56             func();
  57             injector.cancel();
  58         } catch (const std::bad_alloc&) {
  59             // expected
  60         }
  61     } while (injector.failed());
  62 }
  63
  64 }
  65 }