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

#ifndef CRYPTOHOME_FLATBUFFER_SECURE_ALLOCATOR_BRIDGE_H_
#define CRYPTOHOME_FLATBUFFER_SECURE_ALLOCATOR_BRIDGE_H_

#include <brillo/secure_allocator.h>
#include <flatbuffers/flatbuffers.h>

namespace cryptohome {

// This class wraps the SecureAllocator in a flatbuffers::Allocator interface,
// allowing flatbuffers in cryptohome to be put into eraseable memory.
class FlatbufferSecureAllocatorBridge : public flatbuffers::Allocator {
 public:
  uint8_t* allocate(size_t size) override { return allocator_.allocate(size); }

  void deallocate(uint8_t* p, size_t size) override {
    return allocator_.deallocate(p, size);
  }

 private:
  brillo::SecureAllocator<uint8_t> allocator_;
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_FLATBUFFER_SECURE_ALLOCATOR_BRIDGE_H_