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 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 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 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 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 110 int main (int argc, const char * argv[]){ 111 return CommandLineTestRunner::RunAllTests(argc, argv); 112 } 113