1 /*
2  * Copyright 2020 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 <arpa/inet.h>  // htons
18 #include <dlfcn.h>
19 #include <gtest/gtest.h>
20 
21 #include "stack/include/avrc_api.h"
22 #include "stack/include/bt_types.h"
23 
24 class StackAvrcpTest : public ::testing::Test {
25 protected:
26   StackAvrcpTest() = default;
27 
28   virtual ~StackAvrcpTest() = default;
29 };
30 
TEST_F(StackAvrcpTest,test_avrcp_ctrl_parse_vendor_rsp)31 TEST_F(StackAvrcpTest, test_avrcp_ctrl_parse_vendor_rsp) {
32   uint8_t scratch_buf[512]{};
33   uint16_t scratch_buf_len = 512;
34   tAVRC_MSG msg{};
35   tAVRC_RESPONSE result{};
36   uint8_t vendor_rsp_buf[512]{};
37 
38   msg.hdr.opcode = AVRC_OP_VENDOR;
39   msg.hdr.ctype = AVRC_CMD_STATUS;
40 
41   memset(vendor_rsp_buf, 0, sizeof(vendor_rsp_buf));
42   vendor_rsp_buf[0] = AVRC_PDU_GET_ELEMENT_ATTR;
43   uint8_t* p = &vendor_rsp_buf[2];
44   UINT16_TO_BE_STREAM(p, 0x0009);   // parameter length
45   UINT8_TO_STREAM(p, 0x01);         // number of attributes
46   UINT32_TO_STREAM(p, 0x00000000);  // attribute ID
47   UINT16_TO_STREAM(p, 0x0000);      // character set ID
48   UINT16_TO_STREAM(p, 0xffff);      // attribute value length
49   msg.vendor.p_vendor_data = vendor_rsp_buf;
50   msg.vendor.vendor_len = 13;
51   EXPECT_EQ(AVRC_Ctrl_ParsResponse(&msg, &result, scratch_buf, &scratch_buf_len),
52             AVRC_STS_INTERNAL_ERR);
53 }
54 
TEST_F(StackAvrcpTest,test_avrcp_parse_browse_rsp)55 TEST_F(StackAvrcpTest, test_avrcp_parse_browse_rsp) {
56   uint8_t scratch_buf[512]{};
57   uint16_t scratch_buf_len = 512;
58   tAVRC_MSG msg{};
59   tAVRC_RESPONSE result{};
60   uint8_t browse_rsp_buf[512]{};
61 
62   msg.hdr.opcode = AVRC_OP_BROWSE;
63 
64   memset(browse_rsp_buf, 0, sizeof(browse_rsp_buf));
65   browse_rsp_buf[0] = AVRC_PDU_GET_ITEM_ATTRIBUTES;
66   uint8_t* p = &browse_rsp_buf[1];
67   UINT16_TO_BE_STREAM(p, 0x000a);   // parameter length;
68   UINT8_TO_STREAM(p, 0x04);         // status
69   UINT8_TO_STREAM(p, 0x01);         // number of attribute
70   UINT32_TO_STREAM(p, 0x00000000);  // attribute ID
71   UINT16_TO_STREAM(p, 0x0000);      // character set ID
72   UINT16_TO_STREAM(p, 0xffff);      // attribute value length
73   msg.browse.p_browse_data = browse_rsp_buf;
74   msg.browse.browse_len = 13;
75   EXPECT_EQ(AVRC_Ctrl_ParsResponse(&msg, &result, scratch_buf, &scratch_buf_len), AVRC_STS_BAD_CMD);
76 }
77 
TEST_F(StackAvrcpTest,test_avrcp_parse_browse_cmd)78 TEST_F(StackAvrcpTest, test_avrcp_parse_browse_cmd) {
79   uint8_t scratch_buf[512]{};
80   tAVRC_MSG msg{};
81   tAVRC_COMMAND result{};
82   uint8_t browse_cmd_buf[512]{};
83 
84   msg.hdr.opcode = AVRC_OP_BROWSE;
85   msg.browse.p_browse_data = browse_cmd_buf;
86   msg.browse.browse_len = 2;
87   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
88 
89   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
90   browse_cmd_buf[0] = AVRC_PDU_SET_BROWSED_PLAYER;
91   msg.browse.browse_len = 3;
92   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
93 
94   msg.browse.browse_len = 5;
95   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
96 
97   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
98   browse_cmd_buf[0] = AVRC_PDU_GET_FOLDER_ITEMS;
99   msg.browse.browse_len = 3;
100   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
101 
102   msg.browse.browse_len = 13;
103   uint8_t* p = &browse_cmd_buf[3];
104   UINT8_TO_STREAM(p, AVRC_SCOPE_NOW_PLAYING);  // scope
105   UINT32_TO_STREAM(p, 0x00000001);             // start_item
106   UINT32_TO_STREAM(p, 0x00000002);             // end_item
107   browse_cmd_buf[12] = 0;                      // attr_count
108   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
109 
110   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
111   browse_cmd_buf[0] = AVRC_PDU_CHANGE_PATH;
112   msg.browse.browse_len = 3;
113   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
114 
115   msg.browse.browse_len = 14;
116   p = &browse_cmd_buf[3];
117   UINT16_TO_STREAM(p, 0x1234);      // uid_counter
118   UINT8_TO_STREAM(p, AVRC_DIR_UP);  // direction
119   UINT8_TO_STREAM(p, 0);            // attr_count
120   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
121 
122   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
123   browse_cmd_buf[0] = AVRC_PDU_GET_ITEM_ATTRIBUTES;
124   msg.browse.browse_len = 3;
125   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
126 
127   msg.browse.browse_len = 15;
128   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
129 
130   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
131   browse_cmd_buf[0] = AVRC_PDU_GET_TOTAL_NUM_OF_ITEMS;
132   msg.browse.browse_len = 3;
133   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
134 
135   msg.browse.browse_len = 4;
136   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
137 
138   memset(browse_cmd_buf, 0, sizeof(browse_cmd_buf));
139   browse_cmd_buf[0] = AVRC_PDU_SEARCH;
140   msg.browse.browse_len = 3;
141   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_BAD_CMD);
142 
143   p = &browse_cmd_buf[3];
144   UINT16_TO_STREAM(p, 0x0000);  // charset_id
145   UINT16_TO_STREAM(p, 0x0000);  // str_len
146   msg.browse.browse_len = 7;
147   EXPECT_EQ(AVRC_ParsCommand(&msg, &result, scratch_buf, sizeof(scratch_buf)), AVRC_STS_NO_ERROR);
148 }
149 
TEST_F(StackAvrcpTest,test_avrcp_pdu_register_notification)150 TEST_F(StackAvrcpTest, test_avrcp_pdu_register_notification) {
151   ASSERT_EQ(htons(0x500), 5);
152 
153   struct {
154     uint8_t pdu;
155     uint8_t reserved;
156     uint16_t len;
157     struct {
158       uint8_t event_id;
159       uint32_t param;
160     } payload;
161   } data = {
162           AVRC_PDU_REGISTER_NOTIFICATION,
163           0,  // reserved
164           htons(sizeof(data.payload)),
165           .payload =
166                   {
167                           .event_id = 0,
168                           .param = 0x1234,
169                   },
170   };
171 
172   tAVRC_MSG msg = {
173           .vendor =
174                   {
175                           .hdr =
176                                   {
177                                           .ctype = AVRC_CMD_NOTIF,
178                                           .opcode = AVRC_OP_VENDOR,
179                                   },
180                           .p_vendor_data = (uint8_t*)&data,
181                           .vendor_len = sizeof(data),
182                   },
183   };
184   tAVRC_COMMAND result{};
185 
186   // Run through all possible event ids
187   uint8_t id = 0;
188   do {
189     data.payload.event_id = id;
190     ASSERT_EQ((id == 0 || id > AVRC_NUM_NOTIF_EVENTS) ? AVRC_STS_BAD_PARAM : AVRC_STS_NO_ERROR,
191               AVRC_Ctrl_ParsCommand(&msg, &result));
192   } while (++id != 0);
193 }
194