// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef BASE_MAC_SCOPED_MACH_MSG_DESTROY_H_
#define BASE_MAC_SCOPED_MACH_MSG_DESTROY_H_

#include <mach/message.h>

#include "base/memory/raw_ptr.h"

namespace base {

// Calls mach_msg_destroy on the specified message when the object goes out
// of scope.
class ScopedMachMsgDestroy {
 public:
  explicit ScopedMachMsgDestroy(mach_msg_header_t* header) : header_(header) {}

  ScopedMachMsgDestroy(const ScopedMachMsgDestroy&) = delete;
  ScopedMachMsgDestroy& operator=(const ScopedMachMsgDestroy&) = delete;

  ~ScopedMachMsgDestroy() {
    if (header_) {
      mach_msg_destroy(header_);
    }
  }

  // Prevents the message from being destroyed when it goes out of scope.
  void Disarm() { header_ = nullptr; }

 private:
  raw_ptr<mach_msg_header_t> header_;
};

}  // namespace base

#endif  // BASE_MAC_SCOPED_MACH_MSG_DESTROY_H_