You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
208 lines
7.9 KiB
208 lines
7.9 KiB
4 months ago
|
/*
|
||
|
|
* Copyright (C) 2016 The Android Open Source Project
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <nvram/messages/nvram_messages.h>
|
||
|
|
|
||
|
|
#include <nvram/messages/blob.h>
|
||
|
|
#include <nvram/messages/io.h>
|
||
|
|
#include <nvram/messages/proto.hpp>
|
||
|
|
|
||
|
|
namespace nvram {
|
||
|
|
|
||
|
|
// Descriptors for the message types.
|
||
|
|
|
||
|
|
// IMPORTANT: The field numbers specified here correspond to protocol buffer
|
||
|
|
// fields on the wire. While they are arbitrary, they should never be
|
||
|
|
// reordered, reassigned, or overloaded once defined.
|
||
|
|
template<> struct DescriptorForType<GetInfoRequest> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<GetInfoResponse> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &GetInfoResponse::total_size),
|
||
|
|
MakeField(2, &GetInfoResponse::available_size),
|
||
|
|
MakeField(3, &GetInfoResponse::max_spaces),
|
||
|
|
MakeField(4, &GetInfoResponse::space_list),
|
||
|
|
MakeField(5, &GetInfoResponse::max_space_size),
|
||
|
|
MakeField(6, &GetInfoResponse::wipe_disabled));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<CreateSpaceRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &CreateSpaceRequest::index),
|
||
|
|
MakeField(2, &CreateSpaceRequest::size),
|
||
|
|
MakeField(3, &CreateSpaceRequest::controls),
|
||
|
|
MakeField(4, &CreateSpaceRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<CreateSpaceResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<GetSpaceInfoRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &GetSpaceInfoRequest::index));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<GetSpaceInfoResponse> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &GetSpaceInfoResponse::size),
|
||
|
|
MakeField(2, &GetSpaceInfoResponse::controls),
|
||
|
|
MakeField(3, &GetSpaceInfoResponse::read_locked),
|
||
|
|
MakeField(4, &GetSpaceInfoResponse::write_locked));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DeleteSpaceRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &DeleteSpaceRequest::index),
|
||
|
|
MakeField(2, &DeleteSpaceRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DeleteSpaceResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DisableCreateRequest> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DisableCreateResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<WriteSpaceRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &WriteSpaceRequest::index),
|
||
|
|
MakeField(2, &WriteSpaceRequest::buffer),
|
||
|
|
MakeField(3, &WriteSpaceRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<WriteSpaceResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<ReadSpaceRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &ReadSpaceRequest::index),
|
||
|
|
MakeField(2, &ReadSpaceRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<ReadSpaceResponse> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &ReadSpaceResponse::buffer));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<LockSpaceWriteRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &LockSpaceWriteRequest::index),
|
||
|
|
MakeField(2, &LockSpaceWriteRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<LockSpaceWriteResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<LockSpaceReadRequest> {
|
||
|
|
static constexpr auto kFields =
|
||
|
|
MakeFieldList(MakeField(1, &LockSpaceReadRequest::index),
|
||
|
|
MakeField(2, &LockSpaceReadRequest::authorization_value));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<LockSpaceReadResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<WipeStorageRequest> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<WipeStorageResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DisableWipeRequest> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<DisableWipeResponse> {
|
||
|
|
static constexpr auto kFields = MakeFieldList();
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<Request> {
|
||
|
|
static constexpr auto kFields = MakeFieldList(
|
||
|
|
MakeOneOfField(1, &Request::payload, COMMAND_GET_INFO),
|
||
|
|
MakeOneOfField(2, &Request::payload, COMMAND_CREATE_SPACE),
|
||
|
|
MakeOneOfField(3, &Request::payload, COMMAND_GET_SPACE_INFO),
|
||
|
|
MakeOneOfField(4, &Request::payload, COMMAND_DELETE_SPACE),
|
||
|
|
MakeOneOfField(5, &Request::payload, COMMAND_DISABLE_CREATE),
|
||
|
|
MakeOneOfField(6, &Request::payload, COMMAND_WRITE_SPACE),
|
||
|
|
MakeOneOfField(7, &Request::payload, COMMAND_READ_SPACE),
|
||
|
|
MakeOneOfField(8, &Request::payload, COMMAND_LOCK_SPACE_WRITE),
|
||
|
|
MakeOneOfField(9, &Request::payload, COMMAND_LOCK_SPACE_READ),
|
||
|
|
MakeOneOfField(10, &Request::payload, COMMAND_WIPE_STORAGE),
|
||
|
|
MakeOneOfField(11, &Request::payload, COMMAND_DISABLE_WIPE));
|
||
|
|
};
|
||
|
|
|
||
|
|
template<> struct DescriptorForType<Response> {
|
||
|
|
static constexpr auto kFields = MakeFieldList(
|
||
|
|
MakeField(1, &Response::result),
|
||
|
|
MakeOneOfField(2, &Response::payload, COMMAND_GET_INFO),
|
||
|
|
MakeOneOfField(3, &Response::payload, COMMAND_CREATE_SPACE),
|
||
|
|
MakeOneOfField(4, &Response::payload, COMMAND_GET_SPACE_INFO),
|
||
|
|
MakeOneOfField(5, &Response::payload, COMMAND_DELETE_SPACE),
|
||
|
|
MakeOneOfField(6, &Response::payload, COMMAND_DISABLE_CREATE),
|
||
|
|
MakeOneOfField(7, &Response::payload, COMMAND_WRITE_SPACE),
|
||
|
|
MakeOneOfField(8, &Response::payload, COMMAND_READ_SPACE),
|
||
|
|
MakeOneOfField(9, &Response::payload, COMMAND_LOCK_SPACE_WRITE),
|
||
|
|
MakeOneOfField(10, &Response::payload, COMMAND_LOCK_SPACE_READ),
|
||
|
|
MakeOneOfField(11, &Response::payload, COMMAND_WIPE_STORAGE),
|
||
|
|
MakeOneOfField(12, &Response::payload, COMMAND_DISABLE_WIPE));
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename Message>
|
||
|
|
bool Encode(const Message& msg, Blob* blob) {
|
||
|
|
BlobOutputStreamBuffer stream(blob);
|
||
|
|
return nvram::proto::Encode(msg, &stream) && stream.Truncate();
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename Message>
|
||
|
|
bool Encode(const Message& msg, void* buffer, size_t* size) {
|
||
|
|
ArrayOutputStreamBuffer stream(buffer, *size);
|
||
|
|
if (!nvram::proto::Encode(msg, &stream)) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
*size = stream.bytes_written();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename Message>
|
||
|
|
bool Decode(const uint8_t* data, size_t size, Message* msg) {
|
||
|
|
InputStreamBuffer stream(data, size);
|
||
|
|
return nvram::proto::Decode(msg, &stream) && stream.Done();
|
||
|
|
}
|
||
|
|
|
||
|
|
// Instantiate the templates for the |Request| and |Response| message types.
|
||
|
|
template NVRAM_EXPORT bool Encode<Request>(const Request&, Blob*);
|
||
|
|
template NVRAM_EXPORT bool Encode<Request>(const Request&, void*, size_t*);
|
||
|
|
template NVRAM_EXPORT bool Decode<Request>(const uint8_t*, size_t, Request*);
|
||
|
|
|
||
|
|
template NVRAM_EXPORT bool Encode<Response>(const Response&, Blob*);
|
||
|
|
template NVRAM_EXPORT bool Encode<Response>(const Response&, void*, size_t*);
|
||
|
|
template NVRAM_EXPORT bool Decode<Response>(const uint8_t*, size_t, Response*);
|
||
|
|
|
||
|
|
} // namespace nvram
|