1 /*
2  * Copyright 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "capabilities_packet.h"
18 
19 #include <bluetooth/log.h>
20 
21 #include "internal_include/bt_trace.h"
22 
23 namespace bluetooth {
24 namespace avrcp {
25 
MakeBuilder(Capability capability)26 std::unique_ptr<GetCapabilitiesRequestBuilder> GetCapabilitiesRequestBuilder::MakeBuilder(
27         Capability capability) {
28   std::unique_ptr<GetCapabilitiesRequestBuilder> builder(
29           new GetCapabilitiesRequestBuilder(capability));
30 
31   return builder;
32 }
33 
size() const34 size_t GetCapabilitiesRequestBuilder::size() const { return GetCapabilitiesRequest::kMinSize(); }
35 
Serialize(const std::shared_ptr<::bluetooth::Packet> & pkt)36 bool GetCapabilitiesRequestBuilder::Serialize(const std::shared_ptr<::bluetooth::Packet>& pkt) {
37   ReserveSpace(pkt, size());
38 
39   // Push the standard avrcp headers
40   PacketBuilder::PushHeader(pkt);
41 
42   // Push the avrcp vendor command headers, the parameter length is always 1
43   VendorPacketBuilder::PushHeader(pkt, 1);
44 
45   // Push the capability
46   return AddPayloadOctets1(pkt, static_cast<uint8_t>(capability_));
47 }
48 
GetCapabilityRequested() const49 Capability GetCapabilitiesRequest::GetCapabilityRequested() const {
50   auto value = *(begin() + VendorPacket::kMinSize());
51   return static_cast<Capability>(value);
52 }
53 
IsValid() const54 bool GetCapabilitiesRequest::IsValid() const {
55   if (!VendorPacket::IsValid()) {
56     return false;
57   }
58   return size() == VendorPacket::kMinSize() + 1;
59 }
60 
ToString() const61 std::string GetCapabilitiesRequest::ToString() const {
62   std::stringstream ss;
63   ss << "AvrcpCapabilityReqPacket: " << std::endl;
64   ss << "  └ cType = " << GetCType() << std::endl;
65   ss << "  └ Subunit Type = " << loghex(GetSubunitType()) << std::endl;
66   ss << "  └ Subunit ID = " << loghex(GetSubunitId()) << std::endl;
67   ss << "  └ OpCode = " << GetOpcode() << std::endl;
68   ss << "  └ Company ID = " << loghex(GetCompanyId()) << std::endl;
69   ss << "  └ Command PDU = " << GetCommandPdu() << std::endl;
70   ss << "  └ PacketType = " << GetPacketType() << std::endl;
71   ss << "  └ Parameter Length = " << loghex(GetParameterLength()) << std::endl;
72   ss << "  └ Capabilities Request = " << GetCapabilityRequested() << std::endl;
73   return ss.str();
74 }
75 
76 std::unique_ptr<GetCapabilitiesResponseBuilder>
MakeCompanyIdBuilder(uint32_t company_id_element)77 GetCapabilitiesResponseBuilder::MakeCompanyIdBuilder(uint32_t company_id_element) {
78   std::unique_ptr<GetCapabilitiesResponseBuilder> builder(
79           new GetCapabilitiesResponseBuilder(Capability::COMPANY_ID));
80 
81   company_id_element &= 0x00FFFFFF;
82   builder->elements_.insert(company_id_element);
83 
84   return builder;
85 }
86 
87 std::unique_ptr<GetCapabilitiesResponseBuilder>
MakeEventsSupportedBuilder(Event event)88 GetCapabilitiesResponseBuilder::MakeEventsSupportedBuilder(Event event) {
89   std::unique_ptr<GetCapabilitiesResponseBuilder> builder(
90           new GetCapabilitiesResponseBuilder(Capability::EVENTS_SUPPORTED));
91 
92   builder->elements_.insert(static_cast<uint8_t>(event));
93 
94   return builder;
95 }
96 
AddCompanyId(uint32_t company_id)97 GetCapabilitiesResponseBuilder* GetCapabilitiesResponseBuilder::AddCompanyId(uint32_t company_id) {
98   log::assert_that(capability_ == Capability::COMPANY_ID,
99                    "assert failed: capability_ == Capability::COMPANY_ID");
100   log::assert_that(elements_.size() < size_t(0xFF), "maximum capability count reached");
101 
102   company_id &= 0x00FFFFFF;
103   elements_.insert(company_id);
104 
105   return this;
106 }
107 
AddEvent(Event event)108 GetCapabilitiesResponseBuilder* GetCapabilitiesResponseBuilder::AddEvent(Event event) {
109   log::assert_that(capability_ == Capability::EVENTS_SUPPORTED,
110                    "assert failed: capability_ == Capability::EVENTS_SUPPORTED");
111   log::assert_that(elements_.size() < size_t(0xFF), "maximum capability count reached");
112 
113   elements_.insert(static_cast<uint8_t>(event));
114 
115   return this;
116 }
117 
size() const118 size_t GetCapabilitiesResponseBuilder::size() const {
119   // Company ID's have a size of 3 while Events have a size of 1
120   size_t capability_count = elements_.size();
121   size_t capability_size = capability_ == Capability::COMPANY_ID ? 3 : 1;
122 
123   return GetCapabilitiesResponse::kMinSize() + (capability_count * capability_size);
124 }
125 
Serialize(const std::shared_ptr<::bluetooth::Packet> & pkt)126 bool GetCapabilitiesResponseBuilder::Serialize(const std::shared_ptr<::bluetooth::Packet>& pkt) {
127   ReserveSpace(pkt, size());
128 
129   // Push the standard avrcp headers
130   PacketBuilder::PushHeader(pkt);
131 
132   // Push the avrcp vendor command headers
133   uint16_t parameter_count = size() - VendorPacket::kMinSize();
134   VendorPacketBuilder::PushHeader(pkt, parameter_count);
135 
136   // Push the capability, capability count, and elements
137   AddPayloadOctets1(pkt, static_cast<uint8_t>(capability_));
138   AddPayloadOctets1(pkt, elements_.size());
139   for (auto it = elements_.begin(); it != elements_.end(); it++) {
140     if (capability_ == Capability::COMPANY_ID) {
141       PushCompanyId(pkt, *it);
142     } else {
143       AddPayloadOctets1(pkt, *it);
144     }
145   }
146 
147   return true;
148 }
149 
150 }  // namespace avrcp
151 }  // namespace bluetooth
152