1
2 // *****************************************************************************
3 //
4 // test rfcomm query tests
5 //
6 // *****************************************************************************
7
8
9 #include <stdint.h>
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <string.h>
13
14 #include "CppUTest/TestHarness.h"
15 #include "CppUTest/CommandLineTestRunner.h"
16 #include "CppUTestExt/MockSupport.h"
17
18 #include "hci_cmd.h"
19
20 #include "btstack_memory.h"
21 #include "hci.h"
22 #include "ble/gatt_client.h"
23 #include "btstack_event.h"
24
25 static uint8_t advertisement_received;
26 static uint8_t connected;
27 static uint8_t advertisement_packet[150];
28 static btstack_packet_callback_registration_t hci_event_callback_registration;
29
30 extern "C" void mock_simulate_hci_state_working(void);
31 extern "C" void mock_simulate_command_complete(const hci_cmd_t *cmd);
32 extern "C" void mock_simulate_scan_response(void);
33 extern "C" void mock_simulate_connected(void);
34
CHECK_EQUAL_ARRAY(const uint8_t * expected,uint8_t * actual,int size)35 void CHECK_EQUAL_ARRAY(const uint8_t * expected, uint8_t * actual, int size){
36 for (int i=0; i<size; i++){
37 BYTES_EQUAL(expected[i], actual[i]);
38 }
39 }
40
41 // -----------------------------------------------------
42
handle_hci_event(uint8_t packet_type,uint16_t channel,uint8_t * packet,uint16_t size)43 static void handle_hci_event(uint8_t packet_type, uint16_t channel, uint8_t *packet, uint16_t size){
44 if (packet_type != HCI_EVENT_PACKET) return;
45
46 bd_addr_t address;
47 uint8_t event = packet[0];
48 switch (event) {
49 case GAP_EVENT_ADVERTISING_REPORT:{
50 advertisement_received = 1;
51 memcpy(advertisement_packet, packet, size);
52 gap_event_advertising_report_get_address(packet, address);
53 gap_connect(address, (bd_addr_type_t)packet[3]);
54 break;
55 }
56 case HCI_EVENT_META_GAP:
57 // wait for connection complete
58 if (hci_event_le_meta_get_subevent_code(packet) != GAP_SUBEVENT_LE_CONNECTION_COMPLETE) break;
59 connected = 1;
60 break;
61 case HCI_EVENT_DISCONNECTION_COMPLETE:
62 exit(0);
63 break;
64 default:
65 break;
66 }
67 }
68
69
TEST_GROUP(LECentral)70 TEST_GROUP(LECentral){
71 void setup(void){
72 advertisement_received = 0;
73 connected = 0;
74 // register for HCI events
75 hci_event_callback_registration.callback = &handle_hci_event;
76 hci_add_event_handler(&hci_event_callback_registration);
77
78 l2cap_init();
79
80 mock().expectOneCall("hci_can_send_packet_now_using_packet_buffer").andReturnValue(1);
81 mock_simulate_hci_state_working();
82 }
83
84 void teardown(void){
85 mock().clear();
86 }
87 };
88
89
90
TEST(LECentral,TestScanningAndConnect)91 TEST(LECentral, TestScanningAndConnect){
92 uint8_t expected_bt_addr[] = {0x34, 0xB1, 0xF7, 0xD1, 0x77, 0x9B};
93
94 mock_simulate_scan_response();
95
96 CHECK(advertisement_received);
97 CHECK_EQUAL(0xE2, advertisement_packet[2]); // event type
98 CHECK_EQUAL(0x01, advertisement_packet[3]); // address type
99 CHECK_EQUAL_ARRAY(expected_bt_addr, &advertisement_packet[4], 6);
100 CHECK_EQUAL(0xCC, advertisement_packet[10]); // rssi
101 CHECK_EQUAL(0x09, advertisement_packet[11]); // data size
102
103 for (int i=0; i<0x09; i++){ // data
104 CHECK_EQUAL(i, advertisement_packet[12+i]);
105 }
106 mock_simulate_connected();
107 CHECK(connected);
108 }
109
main(int argc,const char * argv[])110 int main (int argc, const char * argv[]){
111 return CommandLineTestRunner::RunAllTests(argc, argv);
112 }
113