| /* |
| * Copyright 2018 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 <forward_list> |
| #include <iterator> |
| #include <memory> |
| #include <vector> |
| |
| #include "base_packet_builder.h" |
| #include "hci/address.h" |
| #include "types/class_of_device.h" |
| |
| namespace test_vendor_lib { |
| namespace packets { |
| |
| using ::bluetooth::hci::Address; |
| |
| // Abstract base class that is subclassed to build specifc packets. |
| // The template parameter little_endian controls the generation of insert(). |
| template <bool little_endian> |
| class PacketBuilder : public BasePacketBuilder { |
| public: |
| PacketBuilder() = default; |
| virtual ~PacketBuilder() = default; |
| |
| // Classes which need fragmentation should define a function like this: |
| // std::forward_list<DerivedBuilder>& Fragment(size_t max_size); |
| |
| protected: |
| // Write sizeof(FixedWidthIntegerType) bytes using the iterator |
| template <typename FixedWidthIntegerType, |
| typename std::enable_if<std::is_integral<FixedWidthIntegerType>::value, int>::type = 0> |
| void insert(FixedWidthIntegerType value, std::back_insert_iterator<std::vector<uint8_t>> it) const { |
| for (size_t i = 0; i < sizeof(FixedWidthIntegerType); i++) { |
| if (little_endian == true) { |
| *it = static_cast<uint8_t>(value >> (i * 8)); |
| } else { |
| *it = static_cast<uint8_t>(value >> ((sizeof(FixedWidthIntegerType) - i - 1) * 8)); |
| } |
| it++; |
| } |
| } |
| |
| // Specialized insert that allows inserting enums without casting |
| template <typename Enum, typename std::enable_if<std::is_enum_v<Enum>, int>::type = 0> |
| inline void insert(Enum value, std::back_insert_iterator<std::vector<uint8_t>> it) const { |
| using enum_type = typename std::underlying_type_t<Enum>; |
| static_assert(std::is_unsigned_v<enum_type>, "Enum type is signed. Did you forget to specify the enum size?"); |
| insert<enum_type>(static_cast<enum_type>(value), it); |
| } |
| |
| // Write a vector of FixedWidthIntegerType using the iterator |
| template <typename FixedWidthIntegerType> |
| void insert_vector(const std::vector<FixedWidthIntegerType>& vec, |
| std::back_insert_iterator<std::vector<uint8_t>> it) const { |
| static_assert(std::is_integral<FixedWidthIntegerType>::value, |
| "PacketBuilder::insert requires an integral type vector."); |
| for (const auto& element : vec) { |
| insert(element, it); |
| } |
| } |
| |
| void insert_address(const Address& addr, std::back_insert_iterator<std::vector<uint8_t>> it) const { |
| for (const auto& element : addr.address) { |
| insert(element, it); |
| } |
| } |
| |
| void insert_class_of_device(const ClassOfDevice& cod, std::back_insert_iterator<std::vector<uint8_t>> it) const { |
| for (const auto& element : cod.cod) { |
| insert(element, it); |
| } |
| } |
| }; |
| |
| } // namespace packets |
| } // namespace test_vendor_lib |