1 /******************************************************************************
2  *
3  *  Copyright 2015 Google Inc.
4  *
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at:
8  *
9  *  http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an "AS IS" BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  ******************************************************************************/
18 
19 #include <base/logging.h>
20 
21 #include "gd/common/init_flags.h"
22 #include "hci/include/btsnoop_mem.h"
23 
24 static btsnoop_data_cb data_callback = NULL;
25 static activity_attribution_cb attribution_callback = NULL;
26 
btsnoop_mem_set_callback(btsnoop_data_cb cb)27 void btsnoop_mem_set_callback(btsnoop_data_cb cb) { data_callback = cb; }
28 
activity_attribution_set_callback(activity_attribution_cb cb)29 void activity_attribution_set_callback(activity_attribution_cb cb) {
30   attribution_callback = cb;
31 }
32 
btsnoop_mem_capture(const BT_HDR * packet,uint64_t timestamp_us)33 void btsnoop_mem_capture(const BT_HDR* packet, uint64_t timestamp_us) {
34   if (!data_callback && !attribution_callback) return;
35 
36   CHECK(packet);
37 
38   const uint8_t* data = &packet->data[packet->offset];
39   const uint16_t type = packet->event & BT_EVT_MASK;
40   size_t length = 0;
41 
42   switch (type) {
43     case BT_EVT_TO_LM_HCI_CMD:
44       if (packet->len > 2) length = data[2] + 3;
45       break;
46 
47     case BT_EVT_TO_BTU_HCI_EVT:
48       if (packet->len > 1) length = data[1] + 2;
49       break;
50 
51     case BT_EVT_TO_LM_HCI_ACL:
52     case BT_EVT_TO_BTU_HCI_ACL:
53       if (packet->len > 3) length = (data[2] | (data[3] << 8)) + 4;
54       break;
55 
56     case BT_EVT_TO_LM_HCI_SCO:
57     case BT_EVT_TO_BTU_HCI_SCO:
58       if (packet->len > 2) length = data[2] + 3;
59       break;
60 
61     case BT_EVT_TO_LM_HCI_ISO:
62     case BT_EVT_TO_BTU_HCI_ISO:
63       if (packet->len > 3) length = (data[2] | ((data[3] & 0x3f) << 8)) + 4;
64       break;
65   }
66 
67   if (length && data_callback)
68     (*data_callback)(type, data, length, timestamp_us);
69   if (length && attribution_callback &&
70       bluetooth::common::init_flags::btaa_hci_is_enabled()) {
71     (*attribution_callback)(type, data, length, timestamp_us);
72   }
73 }
74