102 lines
2.5 KiB
C++
102 lines
2.5 KiB
C++
/*
|
|
* Copyright 2021, 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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <android/hardware/keymaster/4.0/types.h>
|
|
|
|
#include <cn-cbor/cn-cbor.h>
|
|
|
|
namespace cuttlefish {
|
|
namespace confui {
|
|
|
|
/** take prompt_text_, extra_data
|
|
* returns CBOR map, created with the two
|
|
*
|
|
* Usage:
|
|
* if (IsOk()) GetMessage()
|
|
*
|
|
* The CBOR map is used to create signed confirmation
|
|
*/
|
|
class Cbor {
|
|
enum class Error : uint32_t {
|
|
OK = 0,
|
|
OUT_OF_DATA = 1,
|
|
MALFORMED = 2,
|
|
MALFORMED_UTF8 = 3,
|
|
};
|
|
|
|
enum class MessageSize : uint32_t { MAX = 6144u };
|
|
|
|
enum class Type : uint8_t {
|
|
NUMBER = 0,
|
|
NEGATIVE = 1,
|
|
BYTE_STRING = 2,
|
|
TEXT_STRING = 3,
|
|
ARRAY = 4,
|
|
MAP = 5,
|
|
TAG = 6,
|
|
FLOAT = 7,
|
|
};
|
|
|
|
public:
|
|
Cbor(const std::string& prompt_text,
|
|
const std::vector<std::uint8_t>& extra_data)
|
|
: prompt_text_(prompt_text),
|
|
extra_data_(extra_data),
|
|
buffer_status_{Error::OK},
|
|
buffer_(kMax + 1) {
|
|
Init();
|
|
}
|
|
|
|
bool IsOk() const { return buffer_status_ == Error::OK; }
|
|
Error GetErrorCode() const { return buffer_status_; }
|
|
bool IsMessageTooLong() const { return buffer_status_ == Error::OUT_OF_DATA; }
|
|
bool IsMalformedUtf8() const {
|
|
return buffer_status_ == Error::MALFORMED_UTF8;
|
|
}
|
|
// call this only when IsOk() returns true
|
|
std::vector<std::uint8_t>&& GetMessage();
|
|
|
|
/** When encoded, the Cbor object should not exceed this limit in terms of
|
|
* size in bytes
|
|
*/
|
|
const std::uint32_t kMax = static_cast<std::uint32_t>(MessageSize::MAX);
|
|
|
|
private:
|
|
class CborDeleter {
|
|
public:
|
|
void operator()(cn_cbor* ptr) { cn_cbor_free(ptr); }
|
|
};
|
|
|
|
std::unique_ptr<cn_cbor, CborDeleter> cb_map_;
|
|
std::string prompt_text_;
|
|
std::vector<std::uint8_t> extra_data_;
|
|
Error buffer_status_;
|
|
std::vector<std::uint8_t> buffer_;
|
|
|
|
void Init();
|
|
Error CheckUTF8Copy(const std::string& text);
|
|
};
|
|
|
|
} // namespace confui
|
|
} // end of namespace cuttlefish
|