confui_cbor.h

Go to the documentation of this file.

/*
 * 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 <stdint.h>
 
#include <memory>
#include <string>
#include <vector>
 
#include "cbor.h"
 
namespace cuttlefish {
namespace confui {
 
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<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<uint8_t>&& GetMessage();
 
  const uint32_t kMax = static_cast<uint32_t>(MessageSize::MAX);
 
 private:
  class CborDeleter {
   public:
    void operator()(cbor_item_t* ptr) { cbor_decref(&ptr); }
  };
 
  std::unique_ptr<cbor_item_t, CborDeleter> cb_map_;
  std::string prompt_text_;
  std::vector<uint8_t> extra_data_;
  Error buffer_status_;
  std::vector<uint8_t> buffer_;
 
  void Init();
  Error CheckUTF8Copy(const std::string& text);
};
 
}  // namespace confui
}  // end of namespace cuttlefish