blob: 850e8c27aadb3556c56ceb283c0bd07695957613 [file] [log] [blame]
// Copyright 2018 The Fuchsia 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 LIB_FIDL_CPP_ENCODER_H_
#define LIB_FIDL_CPP_ENCODER_H_
#include <lib/fidl/cpp/message.h>
#ifdef __Fuchsia__
#include <lib/zx/object.h>
#endif
#include <zircon/fidl.h>
#include <vector>
namespace fidl {
class Encoder final {
public:
enum NoHeader { NO_HEADER };
explicit Encoder(uint64_t ordinal);
explicit Encoder(NoHeader marker) {}
~Encoder();
size_t Alloc(size_t size);
template <typename T>
T* GetPtr(size_t offset) {
return reinterpret_cast<T*>(bytes_.data() + offset);
}
template <typename T>
const T* GetPtr(size_t offset) const {
return reinterpret_cast<const T*>(bytes_.data() + offset);
}
#ifdef __Fuchsia__
void EncodeHandle(zx::object_base* value, size_t offset);
// Add a handle to the encoder's handles without encoding it into the bytes.
// This is used to re-encode unknown handles, since their "encoded form" is
// already in the unknown bytes somewhere.
void EncodeUnknownHandle(zx::object_base* value);
#endif
HLCPPOutgoingMessage GetMessage();
void Reset(uint64_t ordinal);
size_t CurrentLength() const { return bytes_.size(); }
size_t CurrentHandleCount() const { return handles_.size(); }
std::vector<uint8_t> TakeBytes() { return std::move(bytes_); }
private:
void EncodeMessageHeader(uint64_t ordinal);
std::vector<uint8_t> bytes_;
std::vector<zx_handle_t> handles_;
};
} // namespace fidl
#endif // LIB_FIDL_CPP_ENCODER_H_