gd/packet/parser/test/variable.h

/******************************************************************************
 *
 *  Copyright 2019 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 <optional>
#include <sstream>
#include <string>

#include "packet/bit_inserter.h"
#include "packet/iterator.h"

namespace bluetooth {
namespace packet {
namespace parser {
namespace test {

class Variable final {
 public:
  std::string data;

  Variable() = default;
  Variable(const Variable&) = default;
  Variable(const std::string& str);

  void Serialize(BitInserter& bi) const;

  size_t size() const;

  template <bool little_endian>
  static std::optional<Iterator<little_endian>> Parse(Variable* instance, Iterator<little_endian> it) {
    if (it.NumBytesRemaining() < 1) {
      return {};
    }
    size_t data_length = it.template extract<uint8_t>();
    if (data_length > 255) {
      return {};
    }
    if (it.NumBytesRemaining() < data_length) {
      return {};
    }
    std::stringstream ss;
    for (size_t i = 0; i < data_length; i++) {
      ss << it.template extract<char>();
    }
    *instance = ss.str();
    return it;
  }

  std::string ToString() const {
    return data;
  }
};

}  // namespace test
}  // namespace parser
}  // namespace packet
}  // namespace bluetooth