xref: /btstack/src/ble/sm.c (revision 68437d838a1a2268d4694d9ae6ee218951273a21)
13deb3ec6SMatthias Ringwald /*
23deb3ec6SMatthias Ringwald  * Copyright (C) 2014 BlueKitchen GmbH
33deb3ec6SMatthias Ringwald  *
43deb3ec6SMatthias Ringwald  * Redistribution and use in source and binary forms, with or without
53deb3ec6SMatthias Ringwald  * modification, are permitted provided that the following conditions
63deb3ec6SMatthias Ringwald  * are met:
73deb3ec6SMatthias Ringwald  *
83deb3ec6SMatthias Ringwald  * 1. Redistributions of source code must retain the above copyright
93deb3ec6SMatthias Ringwald  *    notice, this list of conditions and the following disclaimer.
103deb3ec6SMatthias Ringwald  * 2. Redistributions in binary form must reproduce the above copyright
113deb3ec6SMatthias Ringwald  *    notice, this list of conditions and the following disclaimer in the
123deb3ec6SMatthias Ringwald  *    documentation and/or other materials provided with the distribution.
133deb3ec6SMatthias Ringwald  * 3. Neither the name of the copyright holders nor the names of
143deb3ec6SMatthias Ringwald  *    contributors may be used to endorse or promote products derived
153deb3ec6SMatthias Ringwald  *    from this software without specific prior written permission.
163deb3ec6SMatthias Ringwald  * 4. Any redistribution, use, or modification is done solely for
173deb3ec6SMatthias Ringwald  *    personal benefit and not for any commercial purpose or for
183deb3ec6SMatthias Ringwald  *    monetary gain.
193deb3ec6SMatthias Ringwald  *
203deb3ec6SMatthias Ringwald  * THIS SOFTWARE IS PROVIDED BY BLUEKITCHEN GMBH AND CONTRIBUTORS
213deb3ec6SMatthias Ringwald  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
223deb3ec6SMatthias Ringwald  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
233deb3ec6SMatthias Ringwald  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL MATTHIAS
243deb3ec6SMatthias Ringwald  * RINGWALD OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
253deb3ec6SMatthias Ringwald  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
263deb3ec6SMatthias Ringwald  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
273deb3ec6SMatthias Ringwald  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
283deb3ec6SMatthias Ringwald  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
293deb3ec6SMatthias Ringwald  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
303deb3ec6SMatthias Ringwald  * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
313deb3ec6SMatthias Ringwald  * SUCH DAMAGE.
323deb3ec6SMatthias Ringwald  *
333deb3ec6SMatthias Ringwald  * Please inquire about commercial licensing options at
343deb3ec6SMatthias Ringwald  * [email protected]
353deb3ec6SMatthias Ringwald  *
363deb3ec6SMatthias Ringwald  */
373deb3ec6SMatthias Ringwald 
383deb3ec6SMatthias Ringwald #include <stdio.h>
393deb3ec6SMatthias Ringwald #include <string.h>
403deb3ec6SMatthias Ringwald #include <inttypes.h>
413deb3ec6SMatthias Ringwald 
423edc84c5SMatthias Ringwald #include "ble/le_device_db.h"
4359c6af15SMatthias Ringwald #include "ble/core.h"
443edc84c5SMatthias Ringwald #include "ble/sm.h"
450e2df43fSMatthias Ringwald #include "btstack_debug.h"
460e2df43fSMatthias Ringwald #include "btstack_event.h"
470e2df43fSMatthias Ringwald #include "btstack_linked_list.h"
480e2df43fSMatthias Ringwald #include "btstack_memory.h"
49f7a05cdaSMatthias Ringwald #include "gap.h"
500e2df43fSMatthias Ringwald #include "hci.h"
510e2df43fSMatthias Ringwald #include "l2cap.h"
523deb3ec6SMatthias Ringwald 
53e53be891SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
547df18c15SMatthias Ringwald #ifdef HAVE_HCI_CONTROLLER_DHKEY_SUPPORT
557df18c15SMatthias Ringwald #error "Support for DHKEY Support in HCI Controller not implemented yet. Please use software implementation"
567df18c15SMatthias Ringwald #else
577df18c15SMatthias Ringwald #define USE_MBEDTLS_FOR_ECDH
587df18c15SMatthias Ringwald #endif
597df18c15SMatthias Ringwald #endif
607df18c15SMatthias Ringwald 
617df18c15SMatthias Ringwald 
6227c32905SMatthias Ringwald // Software ECDH implementation provided by mbedtls
6327c32905SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
6427c32905SMatthias Ringwald #include "mbedtls/config.h"
6527c32905SMatthias Ringwald #include "mbedtls/platform.h"
6627c32905SMatthias Ringwald #include "mbedtls/ecp.h"
67*68437d83SMatthias Ringwald #include "sm_mbedtls_allocator.h"
68d3bd9600SMatthias Ringwald #endif
69d3bd9600SMatthias Ringwald 
703deb3ec6SMatthias Ringwald //
713deb3ec6SMatthias Ringwald // SM internal types and globals
723deb3ec6SMatthias Ringwald //
733deb3ec6SMatthias Ringwald 
743deb3ec6SMatthias Ringwald typedef enum {
753deb3ec6SMatthias Ringwald     DKG_W4_WORKING,
763deb3ec6SMatthias Ringwald     DKG_CALC_IRK,
773deb3ec6SMatthias Ringwald     DKG_W4_IRK,
783deb3ec6SMatthias Ringwald     DKG_CALC_DHK,
793deb3ec6SMatthias Ringwald     DKG_W4_DHK,
803deb3ec6SMatthias Ringwald     DKG_READY
813deb3ec6SMatthias Ringwald } derived_key_generation_t;
823deb3ec6SMatthias Ringwald 
833deb3ec6SMatthias Ringwald typedef enum {
843deb3ec6SMatthias Ringwald     RAU_W4_WORKING,
853deb3ec6SMatthias Ringwald     RAU_IDLE,
863deb3ec6SMatthias Ringwald     RAU_GET_RANDOM,
873deb3ec6SMatthias Ringwald     RAU_W4_RANDOM,
883deb3ec6SMatthias Ringwald     RAU_GET_ENC,
893deb3ec6SMatthias Ringwald     RAU_W4_ENC,
903deb3ec6SMatthias Ringwald     RAU_SET_ADDRESS,
913deb3ec6SMatthias Ringwald } random_address_update_t;
923deb3ec6SMatthias Ringwald 
933deb3ec6SMatthias Ringwald typedef enum {
943deb3ec6SMatthias Ringwald     CMAC_IDLE,
953deb3ec6SMatthias Ringwald     CMAC_CALC_SUBKEYS,
963deb3ec6SMatthias Ringwald     CMAC_W4_SUBKEYS,
973deb3ec6SMatthias Ringwald     CMAC_CALC_MI,
983deb3ec6SMatthias Ringwald     CMAC_W4_MI,
993deb3ec6SMatthias Ringwald     CMAC_CALC_MLAST,
1003deb3ec6SMatthias Ringwald     CMAC_W4_MLAST
1013deb3ec6SMatthias Ringwald } cmac_state_t;
1023deb3ec6SMatthias Ringwald 
1033deb3ec6SMatthias Ringwald typedef enum {
1043deb3ec6SMatthias Ringwald     JUST_WORKS,
10527c32905SMatthias Ringwald     PK_RESP_INPUT,  // Initiator displays PK, responder inputs PK
10627c32905SMatthias Ringwald     PK_INIT_INPUT,  // Responder displays PK, initiator inputs PK
1073deb3ec6SMatthias Ringwald     OK_BOTH_INPUT,  // Only input on both, both input PK
10827c32905SMatthias Ringwald     NK_BOTH_INPUT,  // Only numerical compparison (yes/no) on on both sides
1093deb3ec6SMatthias Ringwald     OOB             // OOB available on both sides
1103deb3ec6SMatthias Ringwald } stk_generation_method_t;
1113deb3ec6SMatthias Ringwald 
1123deb3ec6SMatthias Ringwald typedef enum {
1133deb3ec6SMatthias Ringwald     SM_USER_RESPONSE_IDLE,
1143deb3ec6SMatthias Ringwald     SM_USER_RESPONSE_PENDING,
1153deb3ec6SMatthias Ringwald     SM_USER_RESPONSE_CONFIRM,
1163deb3ec6SMatthias Ringwald     SM_USER_RESPONSE_PASSKEY,
1173deb3ec6SMatthias Ringwald     SM_USER_RESPONSE_DECLINE
1183deb3ec6SMatthias Ringwald } sm_user_response_t;
1193deb3ec6SMatthias Ringwald 
1203deb3ec6SMatthias Ringwald typedef enum {
1213deb3ec6SMatthias Ringwald     SM_AES128_IDLE,
1223deb3ec6SMatthias Ringwald     SM_AES128_ACTIVE
1233deb3ec6SMatthias Ringwald } sm_aes128_state_t;
1243deb3ec6SMatthias Ringwald 
1253deb3ec6SMatthias Ringwald typedef enum {
1263deb3ec6SMatthias Ringwald     ADDRESS_RESOLUTION_IDLE,
1273deb3ec6SMatthias Ringwald     ADDRESS_RESOLUTION_GENERAL,
1283deb3ec6SMatthias Ringwald     ADDRESS_RESOLUTION_FOR_CONNECTION,
1293deb3ec6SMatthias Ringwald } address_resolution_mode_t;
1303deb3ec6SMatthias Ringwald 
1313deb3ec6SMatthias Ringwald typedef enum {
1323deb3ec6SMatthias Ringwald     ADDRESS_RESOLUTION_SUCEEDED,
1333deb3ec6SMatthias Ringwald     ADDRESS_RESOLUTION_FAILED,
1343deb3ec6SMatthias Ringwald } address_resolution_event_t;
135901c000fSMatthias Ringwald 
136901c000fSMatthias Ringwald typedef enum {
1377df18c15SMatthias Ringwald     EC_KEY_GENERATION_IDLE,
1387df18c15SMatthias Ringwald     EC_KEY_GENERATION_ACTIVE,
1397df18c15SMatthias Ringwald     EC_KEY_GENERATION_DONE,
1407df18c15SMatthias Ringwald } ec_key_generation_state_t;
1417df18c15SMatthias Ringwald 
1427df18c15SMatthias Ringwald typedef enum {
143901c000fSMatthias Ringwald     SM_STATE_VAR_DHKEY_COMMAND_RECEIVED = 1 << 0
144901c000fSMatthias Ringwald } sm_state_var_t;
145901c000fSMatthias Ringwald 
1463deb3ec6SMatthias Ringwald //
1473deb3ec6SMatthias Ringwald // GLOBAL DATA
1483deb3ec6SMatthias Ringwald //
1493deb3ec6SMatthias Ringwald 
1503deb3ec6SMatthias Ringwald static uint8_t test_use_fixed_local_csrk;
1517df18c15SMatthias Ringwald static uint8_t test_use_fixed_ec_keypair;
1523deb3ec6SMatthias Ringwald 
1533deb3ec6SMatthias Ringwald // configuration
1543deb3ec6SMatthias Ringwald static uint8_t sm_accepted_stk_generation_methods;
1553deb3ec6SMatthias Ringwald static uint8_t sm_max_encryption_key_size;
1563deb3ec6SMatthias Ringwald static uint8_t sm_min_encryption_key_size;
1573deb3ec6SMatthias Ringwald static uint8_t sm_auth_req = 0;
1583deb3ec6SMatthias Ringwald static uint8_t sm_io_capabilities = IO_CAPABILITY_NO_INPUT_NO_OUTPUT;
1593deb3ec6SMatthias Ringwald static uint8_t sm_slave_request_security;
1603deb3ec6SMatthias Ringwald 
1613deb3ec6SMatthias Ringwald // Security Manager Master Keys, please use sm_set_er(er) and sm_set_ir(ir) with your own 128 bit random values
1623deb3ec6SMatthias Ringwald static sm_key_t sm_persistent_er;
1633deb3ec6SMatthias Ringwald static sm_key_t sm_persistent_ir;
1643deb3ec6SMatthias Ringwald 
1653deb3ec6SMatthias Ringwald // derived from sm_persistent_ir
1663deb3ec6SMatthias Ringwald static sm_key_t sm_persistent_dhk;
1673deb3ec6SMatthias Ringwald static sm_key_t sm_persistent_irk;
1683deb3ec6SMatthias Ringwald static uint8_t  sm_persistent_irk_ready = 0;    // used for testing
1693deb3ec6SMatthias Ringwald static derived_key_generation_t dkg_state;
1703deb3ec6SMatthias Ringwald 
1713deb3ec6SMatthias Ringwald // derived from sm_persistent_er
1723deb3ec6SMatthias Ringwald // ..
1733deb3ec6SMatthias Ringwald 
1743deb3ec6SMatthias Ringwald // random address update
1753deb3ec6SMatthias Ringwald static random_address_update_t rau_state;
1763deb3ec6SMatthias Ringwald static bd_addr_t sm_random_address;
1773deb3ec6SMatthias Ringwald 
178514d35fcSMatthias Ringwald // CMAC Calculation: General
1793deb3ec6SMatthias Ringwald static cmac_state_t sm_cmac_state;
1803deb3ec6SMatthias Ringwald static uint16_t     sm_cmac_message_len;
181514d35fcSMatthias Ringwald static sm_key_t     sm_cmac_k;
1823deb3ec6SMatthias Ringwald static sm_key_t     sm_cmac_x;
183514d35fcSMatthias Ringwald static sm_key_t     sm_cmac_m_last;
1843deb3ec6SMatthias Ringwald static uint8_t      sm_cmac_block_current;
1853deb3ec6SMatthias Ringwald static uint8_t      sm_cmac_block_count;
186514d35fcSMatthias Ringwald static uint8_t      (*sm_cmac_get_byte)(uint16_t offset);
187514d35fcSMatthias Ringwald static void         (*sm_cmac_done_handler)(uint8_t * hash);
188514d35fcSMatthias Ringwald 
189514d35fcSMatthias Ringwald // CMAC for ATT Signed Writes
190514d35fcSMatthias Ringwald static uint8_t      sm_cmac_header[3];
191aec94140SMatthias Ringwald static const uint8_t * sm_cmac_message;
192514d35fcSMatthias Ringwald static uint8_t      sm_cmac_sign_counter[4];
193514d35fcSMatthias Ringwald 
194514d35fcSMatthias Ringwald // CMAC for Secure Connection functions
195514d35fcSMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
196aec94140SMatthias Ringwald static sm_connection_t * sm_cmac_connection;
197514d35fcSMatthias Ringwald static uint8_t           sm_cmac_sc_buffer[80];
198514d35fcSMatthias Ringwald #endif
1993deb3ec6SMatthias Ringwald 
2003deb3ec6SMatthias Ringwald // resolvable private address lookup / CSRK calculation
2013deb3ec6SMatthias Ringwald static int       sm_address_resolution_test;
2023deb3ec6SMatthias Ringwald static int       sm_address_resolution_ah_calculation_active;
2033deb3ec6SMatthias Ringwald static uint8_t   sm_address_resolution_addr_type;
2043deb3ec6SMatthias Ringwald static bd_addr_t sm_address_resolution_address;
2053deb3ec6SMatthias Ringwald static void *    sm_address_resolution_context;
2063deb3ec6SMatthias Ringwald static address_resolution_mode_t sm_address_resolution_mode;
2078f2a52f4SMatthias Ringwald static btstack_linked_list_t sm_address_resolution_general_queue;
2083deb3ec6SMatthias Ringwald 
2093deb3ec6SMatthias Ringwald // aes128 crypto engine. store current sm_connection_t in sm_aes128_context
2103deb3ec6SMatthias Ringwald static sm_aes128_state_t  sm_aes128_state;
2113deb3ec6SMatthias Ringwald static void *             sm_aes128_context;
2123deb3ec6SMatthias Ringwald 
2133deb3ec6SMatthias Ringwald // random engine. store context (ususally sm_connection_t)
2143deb3ec6SMatthias Ringwald static void * sm_random_context;
2153deb3ec6SMatthias Ringwald 
216e03e489aSMatthias Ringwald // to receive hci events
217e03e489aSMatthias Ringwald static btstack_packet_callback_registration_t hci_event_callback_registration;
218e03e489aSMatthias Ringwald 
21989a78d34SMatthias Ringwald /* to dispatch sm event */
22089a78d34SMatthias Ringwald static btstack_linked_list_t sm_event_handlers;
22189a78d34SMatthias Ringwald 
22227c32905SMatthias Ringwald // Software ECDH implementation provided by mbedtls
22327c32905SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
224e722521aSMatthias Ringwald // group is always valid
225e722521aSMatthias Ringwald static mbedtls_ecp_group   mbedtls_ec_group;
2267df18c15SMatthias Ringwald static ec_key_generation_state_t ec_key_generation_state;
22705299751SMatthias Ringwald static uint8_t ec_qx[32];
22805299751SMatthias Ringwald static uint8_t ec_qy[32];
22905299751SMatthias Ringwald static uint8_t ec_d[32];
230*68437d83SMatthias Ringwald #ifndef HAVE_MALLOC
231*68437d83SMatthias Ringwald static uint8_t mbedtls_memory_buffer[4500+70*sizeof(void *)]; // experimental value on 64-bit system
232d3bd9600SMatthias Ringwald #endif
23327c32905SMatthias Ringwald #endif
23427c32905SMatthias Ringwald 
2353deb3ec6SMatthias Ringwald //
2363deb3ec6SMatthias Ringwald // Volume 3, Part H, Chapter 24
2373deb3ec6SMatthias Ringwald // "Security shall be initiated by the Security Manager in the device in the master role.
2383deb3ec6SMatthias Ringwald // The device in the slave role shall be the responding device."
2393deb3ec6SMatthias Ringwald // -> master := initiator, slave := responder
2403deb3ec6SMatthias Ringwald //
2413deb3ec6SMatthias Ringwald 
2423deb3ec6SMatthias Ringwald // data needed for security setup
2433deb3ec6SMatthias Ringwald typedef struct sm_setup_context {
2443deb3ec6SMatthias Ringwald 
245ec820d77SMatthias Ringwald     btstack_timer_source_t sm_timeout;
2463deb3ec6SMatthias Ringwald 
2473deb3ec6SMatthias Ringwald     // used in all phases
2483deb3ec6SMatthias Ringwald     uint8_t   sm_pairing_failed_reason;
2493deb3ec6SMatthias Ringwald 
2503deb3ec6SMatthias Ringwald     // user response, (Phase 1 and/or 2)
2513deb3ec6SMatthias Ringwald     uint8_t   sm_user_response;
2523deb3ec6SMatthias Ringwald 
2533deb3ec6SMatthias Ringwald     // defines which keys will be send after connection is encrypted - calculated during Phase 1, used Phase 3
2543deb3ec6SMatthias Ringwald     int       sm_key_distribution_send_set;
2553deb3ec6SMatthias Ringwald     int       sm_key_distribution_received_set;
2563deb3ec6SMatthias Ringwald 
2573deb3ec6SMatthias Ringwald     // Phase 2 (Pairing over SMP)
2583deb3ec6SMatthias Ringwald     stk_generation_method_t sm_stk_generation_method;
2593deb3ec6SMatthias Ringwald     sm_key_t  sm_tk;
26027c32905SMatthias Ringwald     uint8_t   sm_use_secure_connections;
2613deb3ec6SMatthias Ringwald 
2623deb3ec6SMatthias Ringwald     sm_key_t  sm_c1_t3_value;   // c1 calculation
2633deb3ec6SMatthias Ringwald     sm_pairing_packet_t sm_m_preq; // pairing request - needed only for c1
2643deb3ec6SMatthias Ringwald     sm_pairing_packet_t sm_s_pres; // pairing response - needed only for c1
2653deb3ec6SMatthias Ringwald     sm_key_t  sm_local_random;
2663deb3ec6SMatthias Ringwald     sm_key_t  sm_local_confirm;
2673deb3ec6SMatthias Ringwald     sm_key_t  sm_peer_random;
2683deb3ec6SMatthias Ringwald     sm_key_t  sm_peer_confirm;
2693deb3ec6SMatthias Ringwald     uint8_t   sm_m_addr_type;   // address and type can be removed
2703deb3ec6SMatthias Ringwald     uint8_t   sm_s_addr_type;   //  ''
2713deb3ec6SMatthias Ringwald     bd_addr_t sm_m_address;     //  ''
2723deb3ec6SMatthias Ringwald     bd_addr_t sm_s_address;     //  ''
2733deb3ec6SMatthias Ringwald     sm_key_t  sm_ltk;
2743deb3ec6SMatthias Ringwald 
275*68437d83SMatthias Ringwald     uint8_t   sm_state_vars;
276e53be891SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
2777df18c15SMatthias Ringwald     uint8_t   sm_peer_qx[32];   // also stores random for EC key generation during init
2787df18c15SMatthias Ringwald     uint8_t   sm_peer_qy[32];   //  ''
279446a8c36SMatthias Ringwald     sm_key_t  sm_peer_nonce;    // might be combined with sm_peer_random
280446a8c36SMatthias Ringwald     sm_key_t  sm_local_nonce;   // might be combined with sm_local_random
281e53be891SMatthias Ringwald     sm_key_t  sm_peer_dhkey_check;
282e53be891SMatthias Ringwald     sm_key_t  sm_local_dhkey_check;
283446a8c36SMatthias Ringwald     sm_key_t  sm_ra;
284446a8c36SMatthias Ringwald     sm_key_t  sm_rb;
2850346c37cSMatthias Ringwald     sm_key_t  sm_t;             // used for f5
286a9f29768SMatthias Ringwald     sm_key_t  sm_mackey;
2877df18c15SMatthias Ringwald     uint8_t   sm_passkey_bit;   // also stores number of generated random bytes for EC key generation
288e53be891SMatthias Ringwald #endif
28927c32905SMatthias Ringwald 
2903deb3ec6SMatthias Ringwald     // Phase 3
2913deb3ec6SMatthias Ringwald 
2923deb3ec6SMatthias Ringwald     // key distribution, we generate
2933deb3ec6SMatthias Ringwald     uint16_t  sm_local_y;
2943deb3ec6SMatthias Ringwald     uint16_t  sm_local_div;
2953deb3ec6SMatthias Ringwald     uint16_t  sm_local_ediv;
2963deb3ec6SMatthias Ringwald     uint8_t   sm_local_rand[8];
2973deb3ec6SMatthias Ringwald     sm_key_t  sm_local_ltk;
2983deb3ec6SMatthias Ringwald     sm_key_t  sm_local_csrk;
2993deb3ec6SMatthias Ringwald     sm_key_t  sm_local_irk;
3003deb3ec6SMatthias Ringwald     // sm_local_address/addr_type not needed
3013deb3ec6SMatthias Ringwald 
3023deb3ec6SMatthias Ringwald     // key distribution, received from peer
3033deb3ec6SMatthias Ringwald     uint16_t  sm_peer_y;
3043deb3ec6SMatthias Ringwald     uint16_t  sm_peer_div;
3053deb3ec6SMatthias Ringwald     uint16_t  sm_peer_ediv;
3063deb3ec6SMatthias Ringwald     uint8_t   sm_peer_rand[8];
3073deb3ec6SMatthias Ringwald     sm_key_t  sm_peer_ltk;
3083deb3ec6SMatthias Ringwald     sm_key_t  sm_peer_irk;
3093deb3ec6SMatthias Ringwald     sm_key_t  sm_peer_csrk;
3103deb3ec6SMatthias Ringwald     uint8_t   sm_peer_addr_type;
3113deb3ec6SMatthias Ringwald     bd_addr_t sm_peer_address;
3123deb3ec6SMatthias Ringwald 
3133deb3ec6SMatthias Ringwald } sm_setup_context_t;
3143deb3ec6SMatthias Ringwald 
3153deb3ec6SMatthias Ringwald //
3163deb3ec6SMatthias Ringwald static sm_setup_context_t the_setup;
3173deb3ec6SMatthias Ringwald static sm_setup_context_t * setup = &the_setup;
3183deb3ec6SMatthias Ringwald 
3193deb3ec6SMatthias Ringwald // active connection - the one for which the_setup is used for
3203deb3ec6SMatthias Ringwald static uint16_t sm_active_connection = 0;
3213deb3ec6SMatthias Ringwald 
3223deb3ec6SMatthias Ringwald // @returns 1 if oob data is available
3233deb3ec6SMatthias Ringwald // stores oob data in provided 16 byte buffer if not null
3243deb3ec6SMatthias Ringwald static int (*sm_get_oob_data)(uint8_t addres_type, bd_addr_t addr, uint8_t * oob_data) = NULL;
3253deb3ec6SMatthias Ringwald 
3263deb3ec6SMatthias Ringwald // used to notify applicationss that user interaction is neccessary, see sm_notify_t below
3273deb3ec6SMatthias Ringwald static btstack_packet_handler_t sm_client_packet_handler = NULL;
3283deb3ec6SMatthias Ringwald 
3293deb3ec6SMatthias Ringwald // horizontal: initiator capabilities
3303deb3ec6SMatthias Ringwald // vertial:    responder capabilities
3313deb3ec6SMatthias Ringwald static const stk_generation_method_t stk_generation_method [5] [5] = {
3323deb3ec6SMatthias Ringwald     { JUST_WORKS,      JUST_WORKS,       PK_INIT_INPUT,   JUST_WORKS,    PK_INIT_INPUT },
3333deb3ec6SMatthias Ringwald     { JUST_WORKS,      JUST_WORKS,       PK_INIT_INPUT,   JUST_WORKS,    PK_INIT_INPUT },
3343deb3ec6SMatthias Ringwald     { PK_RESP_INPUT,   PK_RESP_INPUT,    OK_BOTH_INPUT,   JUST_WORKS,    PK_RESP_INPUT },
3353deb3ec6SMatthias Ringwald     { JUST_WORKS,      JUST_WORKS,       JUST_WORKS,      JUST_WORKS,    JUST_WORKS    },
3363deb3ec6SMatthias Ringwald     { PK_RESP_INPUT,   PK_RESP_INPUT,    PK_INIT_INPUT,   JUST_WORKS,    PK_RESP_INPUT },
3373deb3ec6SMatthias Ringwald };
3383deb3ec6SMatthias Ringwald 
33927c32905SMatthias Ringwald // uses numeric comparison if one side has DisplayYesNo and KeyboardDisplay combinations
34027c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
34127c32905SMatthias Ringwald static const stk_generation_method_t stk_generation_method_with_secure_connection[5][5] = {
34227c32905SMatthias Ringwald     { JUST_WORKS,      JUST_WORKS,       PK_INIT_INPUT,   JUST_WORKS,    PK_INIT_INPUT },
34327c32905SMatthias Ringwald     { JUST_WORKS,      NK_BOTH_INPUT,    PK_INIT_INPUT,   JUST_WORKS,    NK_BOTH_INPUT },
34427c32905SMatthias Ringwald     { PK_RESP_INPUT,   PK_RESP_INPUT,    OK_BOTH_INPUT,   JUST_WORKS,    PK_RESP_INPUT },
34527c32905SMatthias Ringwald     { JUST_WORKS,      JUST_WORKS,       JUST_WORKS,      JUST_WORKS,    JUST_WORKS    },
34627c32905SMatthias Ringwald     { PK_RESP_INPUT,   NK_BOTH_INPUT,    PK_INIT_INPUT,   JUST_WORKS,    NK_BOTH_INPUT },
34727c32905SMatthias Ringwald };
34827c32905SMatthias Ringwald #endif
34927c32905SMatthias Ringwald 
3503deb3ec6SMatthias Ringwald static void sm_run(void);
351711e6c80SMatthias Ringwald static void sm_done_for_handle(hci_con_handle_t con_handle);
352711e6c80SMatthias Ringwald static sm_connection_t * sm_get_connection_for_handle(hci_con_handle_t con_handle);
3533deb3ec6SMatthias Ringwald static inline int sm_calc_actual_encryption_key_size(int other);
3543deb3ec6SMatthias Ringwald static int sm_validate_stk_generation_method(void);
355e53be891SMatthias Ringwald static void sm_shift_left_by_one_bit_inplace(int len, uint8_t * data);
3563deb3ec6SMatthias Ringwald 
3573deb3ec6SMatthias Ringwald static void log_info_hex16(const char * name, uint16_t value){
3583deb3ec6SMatthias Ringwald     log_info("%-6s 0x%04x", name, value);
3593deb3ec6SMatthias Ringwald }
3603deb3ec6SMatthias Ringwald 
3613deb3ec6SMatthias Ringwald // @returns 1 if all bytes are 0
3623764b551SMatthias Ringwald static int sm_is_null(uint8_t * data, int size){
3633deb3ec6SMatthias Ringwald     int i;
3643764b551SMatthias Ringwald     for (i=0; i < size ; i++){
3653764b551SMatthias Ringwald         if (data[i]) return 0;
3663deb3ec6SMatthias Ringwald     }
3673deb3ec6SMatthias Ringwald     return 1;
3683deb3ec6SMatthias Ringwald }
3693deb3ec6SMatthias Ringwald 
3703764b551SMatthias Ringwald static int sm_is_null_random(uint8_t random[8]){
3713764b551SMatthias Ringwald     return sm_is_null(random, 8);
3723764b551SMatthias Ringwald }
3733764b551SMatthias Ringwald 
3743764b551SMatthias Ringwald static int sm_is_null_key(uint8_t * key){
3753764b551SMatthias Ringwald     return sm_is_null(key, 16);
3763764b551SMatthias Ringwald }
3773764b551SMatthias Ringwald 
3783deb3ec6SMatthias Ringwald // Key utils
3793deb3ec6SMatthias Ringwald static void sm_reset_tk(void){
3803deb3ec6SMatthias Ringwald     int i;
3813deb3ec6SMatthias Ringwald     for (i=0;i<16;i++){
3823deb3ec6SMatthias Ringwald         setup->sm_tk[i] = 0;
3833deb3ec6SMatthias Ringwald     }
3843deb3ec6SMatthias Ringwald }
3853deb3ec6SMatthias Ringwald 
3863deb3ec6SMatthias Ringwald // "For example, if a 128-bit encryption key is 0x123456789ABCDEF0123456789ABCDEF0
3873deb3ec6SMatthias Ringwald // and it is reduced to 7 octets (56 bits), then the resulting key is 0x0000000000000000003456789ABCDEF0.""
3883deb3ec6SMatthias Ringwald static void sm_truncate_key(sm_key_t key, int max_encryption_size){
3893deb3ec6SMatthias Ringwald     int i;
3903deb3ec6SMatthias Ringwald     for (i = max_encryption_size ; i < 16 ; i++){
3913deb3ec6SMatthias Ringwald         key[15-i] = 0;
3923deb3ec6SMatthias Ringwald     }
3933deb3ec6SMatthias Ringwald }
3943deb3ec6SMatthias Ringwald 
3953deb3ec6SMatthias Ringwald // SMP Timeout implementation
3963deb3ec6SMatthias Ringwald 
3973deb3ec6SMatthias Ringwald // Upon transmission of the Pairing Request command or reception of the Pairing Request command,
3983deb3ec6SMatthias Ringwald // the Security Manager Timer shall be reset and started.
3993deb3ec6SMatthias Ringwald //
4003deb3ec6SMatthias Ringwald // The Security Manager Timer shall be reset when an L2CAP SMP command is queued for transmission.
4013deb3ec6SMatthias Ringwald //
4023deb3ec6SMatthias Ringwald // If the Security Manager Timer reaches 30 seconds, the procedure shall be considered to have failed,
4033deb3ec6SMatthias Ringwald // and the local higher layer shall be notified. No further SMP commands shall be sent over the L2CAP
4043deb3ec6SMatthias Ringwald // Security Manager Channel. A new SM procedure shall only be performed when a new physical link has been
4053deb3ec6SMatthias Ringwald // established.
4063deb3ec6SMatthias Ringwald 
407ec820d77SMatthias Ringwald static void sm_timeout_handler(btstack_timer_source_t * timer){
4083deb3ec6SMatthias Ringwald     log_info("SM timeout");
409c5b64319SMatthias Ringwald     sm_connection_t * sm_conn = (sm_connection_t*) btstack_run_loop_get_timer_context(timer);
4103deb3ec6SMatthias Ringwald     sm_conn->sm_engine_state = SM_GENERAL_TIMEOUT;
4113deb3ec6SMatthias Ringwald     sm_done_for_handle(sm_conn->sm_handle);
4123deb3ec6SMatthias Ringwald 
4133deb3ec6SMatthias Ringwald     // trigger handling of next ready connection
4143deb3ec6SMatthias Ringwald     sm_run();
4153deb3ec6SMatthias Ringwald }
4163deb3ec6SMatthias Ringwald static void sm_timeout_start(sm_connection_t * sm_conn){
417528a4a3bSMatthias Ringwald     btstack_run_loop_remove_timer(&setup->sm_timeout);
41891a977e8SMatthias Ringwald     btstack_run_loop_set_timer_context(&setup->sm_timeout, sm_conn);
419528a4a3bSMatthias Ringwald     btstack_run_loop_set_timer_handler(&setup->sm_timeout, sm_timeout_handler);
420528a4a3bSMatthias Ringwald     btstack_run_loop_set_timer(&setup->sm_timeout, 30000); // 30 seconds sm timeout
421528a4a3bSMatthias Ringwald     btstack_run_loop_add_timer(&setup->sm_timeout);
4223deb3ec6SMatthias Ringwald }
4233deb3ec6SMatthias Ringwald static void sm_timeout_stop(void){
424528a4a3bSMatthias Ringwald     btstack_run_loop_remove_timer(&setup->sm_timeout);
4253deb3ec6SMatthias Ringwald }
4263deb3ec6SMatthias Ringwald static void sm_timeout_reset(sm_connection_t * sm_conn){
4273deb3ec6SMatthias Ringwald     sm_timeout_stop();
4283deb3ec6SMatthias Ringwald     sm_timeout_start(sm_conn);
4293deb3ec6SMatthias Ringwald }
4303deb3ec6SMatthias Ringwald 
4313deb3ec6SMatthias Ringwald // end of sm timeout
4323deb3ec6SMatthias Ringwald 
4333deb3ec6SMatthias Ringwald // GAP Random Address updates
4343deb3ec6SMatthias Ringwald static gap_random_address_type_t gap_random_adress_type;
435ec820d77SMatthias Ringwald static btstack_timer_source_t gap_random_address_update_timer;
4363deb3ec6SMatthias Ringwald static uint32_t gap_random_adress_update_period;
4373deb3ec6SMatthias Ringwald 
4383deb3ec6SMatthias Ringwald static void gap_random_address_trigger(void){
4393deb3ec6SMatthias Ringwald     if (rau_state != RAU_IDLE) return;
4403deb3ec6SMatthias Ringwald     log_info("gap_random_address_trigger");
4413deb3ec6SMatthias Ringwald     rau_state = RAU_GET_RANDOM;
4423deb3ec6SMatthias Ringwald     sm_run();
4433deb3ec6SMatthias Ringwald }
4443deb3ec6SMatthias Ringwald 
445ec820d77SMatthias Ringwald static void gap_random_address_update_handler(btstack_timer_source_t * timer){
4463deb3ec6SMatthias Ringwald     log_info("GAP Random Address Update due");
447528a4a3bSMatthias Ringwald     btstack_run_loop_set_timer(&gap_random_address_update_timer, gap_random_adress_update_period);
448528a4a3bSMatthias Ringwald     btstack_run_loop_add_timer(&gap_random_address_update_timer);
4493deb3ec6SMatthias Ringwald     gap_random_address_trigger();
4503deb3ec6SMatthias Ringwald }
4513deb3ec6SMatthias Ringwald 
4523deb3ec6SMatthias Ringwald static void gap_random_address_update_start(void){
453528a4a3bSMatthias Ringwald     btstack_run_loop_set_timer_handler(&gap_random_address_update_timer, gap_random_address_update_handler);
454528a4a3bSMatthias Ringwald     btstack_run_loop_set_timer(&gap_random_address_update_timer, gap_random_adress_update_period);
455528a4a3bSMatthias Ringwald     btstack_run_loop_add_timer(&gap_random_address_update_timer);
4563deb3ec6SMatthias Ringwald }
4573deb3ec6SMatthias Ringwald 
4583deb3ec6SMatthias Ringwald static void gap_random_address_update_stop(void){
459528a4a3bSMatthias Ringwald     btstack_run_loop_remove_timer(&gap_random_address_update_timer);
4603deb3ec6SMatthias Ringwald }
4613deb3ec6SMatthias Ringwald 
4623deb3ec6SMatthias Ringwald 
4633deb3ec6SMatthias Ringwald static void sm_random_start(void * context){
4643deb3ec6SMatthias Ringwald     sm_random_context = context;
4653deb3ec6SMatthias Ringwald     hci_send_cmd(&hci_le_rand);
4663deb3ec6SMatthias Ringwald }
4673deb3ec6SMatthias Ringwald 
4683deb3ec6SMatthias Ringwald // pre: sm_aes128_state != SM_AES128_ACTIVE, hci_can_send_command == 1
4693deb3ec6SMatthias Ringwald // context is made availabe to aes128 result handler by this
4703deb3ec6SMatthias Ringwald static void sm_aes128_start(sm_key_t key, sm_key_t plaintext, void * context){
4713deb3ec6SMatthias Ringwald     sm_aes128_state = SM_AES128_ACTIVE;
4723deb3ec6SMatthias Ringwald     sm_key_t key_flipped, plaintext_flipped;
4739c80e4ccSMatthias Ringwald     reverse_128(key, key_flipped);
4749c80e4ccSMatthias Ringwald     reverse_128(plaintext, plaintext_flipped);
4753deb3ec6SMatthias Ringwald     sm_aes128_context = context;
4763deb3ec6SMatthias Ringwald     hci_send_cmd(&hci_le_encrypt, key_flipped, plaintext_flipped);
4773deb3ec6SMatthias Ringwald }
4783deb3ec6SMatthias Ringwald 
4793deb3ec6SMatthias Ringwald // ah(k,r) helper
4803deb3ec6SMatthias Ringwald // r = padding || r
4813deb3ec6SMatthias Ringwald // r - 24 bit value
4823deb3ec6SMatthias Ringwald static void sm_ah_r_prime(uint8_t r[3], sm_key_t r_prime){
4833deb3ec6SMatthias Ringwald     // r'= padding || r
4843deb3ec6SMatthias Ringwald     memset(r_prime, 0, 16);
4853deb3ec6SMatthias Ringwald     memcpy(&r_prime[13], r, 3);
4863deb3ec6SMatthias Ringwald }
4873deb3ec6SMatthias Ringwald 
4883deb3ec6SMatthias Ringwald // d1 helper
4893deb3ec6SMatthias Ringwald // d' = padding || r || d
4903deb3ec6SMatthias Ringwald // d,r - 16 bit values
4913deb3ec6SMatthias Ringwald static void sm_d1_d_prime(uint16_t d, uint16_t r, sm_key_t d1_prime){
4923deb3ec6SMatthias Ringwald     // d'= padding || r || d
4933deb3ec6SMatthias Ringwald     memset(d1_prime, 0, 16);
494f8fbdce0SMatthias Ringwald     big_endian_store_16(d1_prime, 12, r);
495f8fbdce0SMatthias Ringwald     big_endian_store_16(d1_prime, 14, d);
4963deb3ec6SMatthias Ringwald }
4973deb3ec6SMatthias Ringwald 
4983deb3ec6SMatthias Ringwald // dm helper
4993deb3ec6SMatthias Ringwald // r’ = padding || r
5003deb3ec6SMatthias Ringwald // r - 64 bit value
5013deb3ec6SMatthias Ringwald static void sm_dm_r_prime(uint8_t r[8], sm_key_t r_prime){
5023deb3ec6SMatthias Ringwald     memset(r_prime, 0, 16);
5033deb3ec6SMatthias Ringwald     memcpy(&r_prime[8], r, 8);
5043deb3ec6SMatthias Ringwald }
5053deb3ec6SMatthias Ringwald 
5063deb3ec6SMatthias Ringwald // calculate arguments for first AES128 operation in C1 function
5073deb3ec6SMatthias Ringwald static void sm_c1_t1(sm_key_t r, uint8_t preq[7], uint8_t pres[7], uint8_t iat, uint8_t rat, sm_key_t t1){
5083deb3ec6SMatthias Ringwald 
5093deb3ec6SMatthias Ringwald     // p1 = pres || preq || rat’ || iat’
5103deb3ec6SMatthias Ringwald     // "The octet of iat’ becomes the least significant octet of p1 and the most signifi-
5113deb3ec6SMatthias Ringwald     // cant octet of pres becomes the most significant octet of p1.
5123deb3ec6SMatthias Ringwald     // For example, if the 8-bit iat’ is 0x01, the 8-bit rat’ is 0x00, the 56-bit preq
5133deb3ec6SMatthias Ringwald     // is 0x07071000000101 and the 56 bit pres is 0x05000800000302 then
5143deb3ec6SMatthias Ringwald     // p1 is 0x05000800000302070710000001010001."
5153deb3ec6SMatthias Ringwald 
5163deb3ec6SMatthias Ringwald     sm_key_t p1;
5179c80e4ccSMatthias Ringwald     reverse_56(pres, &p1[0]);
5189c80e4ccSMatthias Ringwald     reverse_56(preq, &p1[7]);
5193deb3ec6SMatthias Ringwald     p1[14] = rat;
5203deb3ec6SMatthias Ringwald     p1[15] = iat;
5218314c363SMatthias Ringwald     log_info_key("p1", p1);
5228314c363SMatthias Ringwald     log_info_key("r", r);
5233deb3ec6SMatthias Ringwald 
5243deb3ec6SMatthias Ringwald     // t1 = r xor p1
5253deb3ec6SMatthias Ringwald     int i;
5263deb3ec6SMatthias Ringwald     for (i=0;i<16;i++){
5273deb3ec6SMatthias Ringwald         t1[i] = r[i] ^ p1[i];
5283deb3ec6SMatthias Ringwald     }
5298314c363SMatthias Ringwald     log_info_key("t1", t1);
5303deb3ec6SMatthias Ringwald }
5313deb3ec6SMatthias Ringwald 
5323deb3ec6SMatthias Ringwald // calculate arguments for second AES128 operation in C1 function
5333deb3ec6SMatthias Ringwald static void sm_c1_t3(sm_key_t t2, bd_addr_t ia, bd_addr_t ra, sm_key_t t3){
5343deb3ec6SMatthias Ringwald      // p2 = padding || ia || ra
5353deb3ec6SMatthias Ringwald     // "The least significant octet of ra becomes the least significant octet of p2 and
5363deb3ec6SMatthias Ringwald     // the most significant octet of padding becomes the most significant octet of p2.
5373deb3ec6SMatthias Ringwald     // For example, if 48-bit ia is 0xA1A2A3A4A5A6 and the 48-bit ra is
5383deb3ec6SMatthias Ringwald     // 0xB1B2B3B4B5B6 then p2 is 0x00000000A1A2A3A4A5A6B1B2B3B4B5B6.
5393deb3ec6SMatthias Ringwald 
5403deb3ec6SMatthias Ringwald     sm_key_t p2;
5413deb3ec6SMatthias Ringwald     memset(p2, 0, 16);
5423deb3ec6SMatthias Ringwald     memcpy(&p2[4],  ia, 6);
5433deb3ec6SMatthias Ringwald     memcpy(&p2[10], ra, 6);
5448314c363SMatthias Ringwald     log_info_key("p2", p2);
5453deb3ec6SMatthias Ringwald 
5463deb3ec6SMatthias Ringwald     // c1 = e(k, t2_xor_p2)
5473deb3ec6SMatthias Ringwald     int i;
5483deb3ec6SMatthias Ringwald     for (i=0;i<16;i++){
5493deb3ec6SMatthias Ringwald         t3[i] = t2[i] ^ p2[i];
5503deb3ec6SMatthias Ringwald     }
5518314c363SMatthias Ringwald     log_info_key("t3", t3);
5523deb3ec6SMatthias Ringwald }
5533deb3ec6SMatthias Ringwald 
5543deb3ec6SMatthias Ringwald static void sm_s1_r_prime(sm_key_t r1, sm_key_t r2, sm_key_t r_prime){
5558314c363SMatthias Ringwald     log_info_key("r1", r1);
5568314c363SMatthias Ringwald     log_info_key("r2", r2);
5573deb3ec6SMatthias Ringwald     memcpy(&r_prime[8], &r2[8], 8);
5583deb3ec6SMatthias Ringwald     memcpy(&r_prime[0], &r1[8], 8);
5593deb3ec6SMatthias Ringwald }
5603deb3ec6SMatthias Ringwald 
561e53be891SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
562e53be891SMatthias Ringwald // Software implementations of crypto toolbox for LE Secure Connection
563e53be891SMatthias Ringwald // TODO: replace with code to use AES Engine of HCI Controller
564e53be891SMatthias Ringwald typedef uint8_t sm_key24_t[3];
565e53be891SMatthias Ringwald typedef uint8_t sm_key56_t[7];
566e53be891SMatthias Ringwald typedef uint8_t sm_key256_t[32];
567e53be891SMatthias Ringwald 
568bd57ffebSMatthias Ringwald #if 0
569e53be891SMatthias Ringwald static void aes128_calc_cyphertext(const uint8_t key[16], const uint8_t plaintext[16], uint8_t cyphertext[16]){
570e53be891SMatthias Ringwald     uint32_t rk[RKLENGTH(KEYBITS)];
571e53be891SMatthias Ringwald     int nrounds = rijndaelSetupEncrypt(rk, &key[0], KEYBITS);
572e53be891SMatthias Ringwald     rijndaelEncrypt(rk, nrounds, plaintext, cyphertext);
573e53be891SMatthias Ringwald }
574e53be891SMatthias Ringwald 
575e53be891SMatthias Ringwald static void calc_subkeys(sm_key_t k0, sm_key_t k1, sm_key_t k2){
576e53be891SMatthias Ringwald     memcpy(k1, k0, 16);
577e53be891SMatthias Ringwald     sm_shift_left_by_one_bit_inplace(16, k1);
578e53be891SMatthias Ringwald     if (k0[0] & 0x80){
579e53be891SMatthias Ringwald         k1[15] ^= 0x87;
580e53be891SMatthias Ringwald     }
581e53be891SMatthias Ringwald     memcpy(k2, k1, 16);
582e53be891SMatthias Ringwald     sm_shift_left_by_one_bit_inplace(16, k2);
583e53be891SMatthias Ringwald     if (k1[0] & 0x80){
584e53be891SMatthias Ringwald         k2[15] ^= 0x87;
585e53be891SMatthias Ringwald     }
586e53be891SMatthias Ringwald }
587e53be891SMatthias Ringwald 
588e53be891SMatthias Ringwald static void aes_cmac(sm_key_t aes_cmac, const sm_key_t key, const uint8_t * data, int cmac_message_len){
589e53be891SMatthias Ringwald     sm_key_t k0, k1, k2, zero;
590e53be891SMatthias Ringwald     memset(zero, 0, 16);
591e53be891SMatthias Ringwald 
592e53be891SMatthias Ringwald     aes128_calc_cyphertext(key, zero, k0);
593e53be891SMatthias Ringwald     calc_subkeys(k0, k1, k2);
594e53be891SMatthias Ringwald 
595e53be891SMatthias Ringwald     int cmac_block_count = (cmac_message_len + 15) / 16;
596e53be891SMatthias Ringwald 
597e53be891SMatthias Ringwald     // step 3: ..
598e53be891SMatthias Ringwald     if (cmac_block_count==0){
599e53be891SMatthias Ringwald         cmac_block_count = 1;
600e53be891SMatthias Ringwald     }
601e53be891SMatthias Ringwald 
602e53be891SMatthias Ringwald     // step 4: set m_last
603e53be891SMatthias Ringwald     sm_key_t cmac_m_last;
604e53be891SMatthias Ringwald     int sm_cmac_last_block_complete = cmac_message_len != 0 && (cmac_message_len & 0x0f) == 0;
605e53be891SMatthias Ringwald     int i;
606e53be891SMatthias Ringwald     if (sm_cmac_last_block_complete){
607e53be891SMatthias Ringwald         for (i=0;i<16;i++){
608e53be891SMatthias Ringwald             cmac_m_last[i] = data[cmac_message_len - 16 + i] ^ k1[i];
609e53be891SMatthias Ringwald         }
610e53be891SMatthias Ringwald     } else {
611e53be891SMatthias Ringwald         int valid_octets_in_last_block = cmac_message_len & 0x0f;
612e53be891SMatthias Ringwald         for (i=0;i<16;i++){
613e53be891SMatthias Ringwald             if (i < valid_octets_in_last_block){
614e53be891SMatthias Ringwald                 cmac_m_last[i] = data[(cmac_message_len & 0xfff0) + i] ^ k2[i];
615e53be891SMatthias Ringwald                 continue;
616e53be891SMatthias Ringwald             }
617e53be891SMatthias Ringwald             if (i == valid_octets_in_last_block){
618e53be891SMatthias Ringwald                 cmac_m_last[i] = 0x80 ^ k2[i];
619e53be891SMatthias Ringwald                 continue;
620e53be891SMatthias Ringwald             }
621e53be891SMatthias Ringwald             cmac_m_last[i] = k2[i];
622e53be891SMatthias Ringwald         }
623e53be891SMatthias Ringwald     }
624e53be891SMatthias Ringwald 
625e53be891SMatthias Ringwald     // printf("sm_cmac_start: len %u, block count %u\n", cmac_message_len, cmac_block_count);
626e53be891SMatthias Ringwald     // LOG_KEY(cmac_m_last);
627e53be891SMatthias Ringwald 
628e53be891SMatthias Ringwald     // Step 5
629e53be891SMatthias Ringwald     sm_key_t cmac_x;
630e53be891SMatthias Ringwald     memset(cmac_x, 0, 16);
631e53be891SMatthias Ringwald 
632e53be891SMatthias Ringwald     // Step 6
633e53be891SMatthias Ringwald     sm_key_t sm_cmac_y;
634e53be891SMatthias Ringwald     for (int block = 0 ; block < cmac_block_count-1 ; block++){
635e53be891SMatthias Ringwald         for (i=0;i<16;i++){
636e53be891SMatthias Ringwald             sm_cmac_y[i] = cmac_x[i] ^ data[block * 16 + i];
637e53be891SMatthias Ringwald         }
638e53be891SMatthias Ringwald         aes128_calc_cyphertext(key, sm_cmac_y, cmac_x);
639e53be891SMatthias Ringwald     }
640e53be891SMatthias Ringwald     for (i=0;i<16;i++){
641e53be891SMatthias Ringwald         sm_cmac_y[i] = cmac_x[i] ^ cmac_m_last[i];
642e53be891SMatthias Ringwald     }
643e53be891SMatthias Ringwald 
644e53be891SMatthias Ringwald     // Step 7
645e53be891SMatthias Ringwald     aes128_calc_cyphertext(key, sm_cmac_y, aes_cmac);
646e53be891SMatthias Ringwald }
647bd57ffebSMatthias Ringwald #endif
648e53be891SMatthias Ringwald 
649446a8c36SMatthias Ringwald #if 0
650f92edc8eSMatthias Ringwald //
651f92edc8eSMatthias Ringwald // Link Key Conversion Function h6
652f92edc8eSMatthias Ringwald //
653e53be891SMatthias Ringwald // h6(W, keyID) = AES-CMACW(keyID)
654e53be891SMatthias Ringwald // - W is 128 bits
655e53be891SMatthias Ringwald // - keyID is 32 bits
656e53be891SMatthias Ringwald static void h6(sm_key_t res, const sm_key_t w, const uint32_t key_id){
657e53be891SMatthias Ringwald     uint8_t key_id_buffer[4];
658e53be891SMatthias Ringwald     big_endian_store_32(key_id_buffer, 0, key_id);
659e53be891SMatthias Ringwald     aes_cmac(res, w, key_id_buffer, 4);
660e53be891SMatthias Ringwald }
661e53be891SMatthias Ringwald #endif
662e53be891SMatthias Ringwald #endif
663e53be891SMatthias Ringwald 
664711e6c80SMatthias Ringwald static void sm_setup_event_base(uint8_t * event, int event_size, uint8_t type, hci_con_handle_t con_handle, uint8_t addr_type, bd_addr_t address){
6653deb3ec6SMatthias Ringwald     event[0] = type;
6663deb3ec6SMatthias Ringwald     event[1] = event_size - 2;
667711e6c80SMatthias Ringwald     little_endian_store_16(event, 2, con_handle);
6683deb3ec6SMatthias Ringwald     event[4] = addr_type;
669724d70a2SMatthias Ringwald     reverse_bd_addr(address, &event[5]);
6703deb3ec6SMatthias Ringwald }
6713deb3ec6SMatthias Ringwald 
67289a78d34SMatthias Ringwald static void sm_dispatch_event(uint8_t packet_type, uint16_t channel, uint8_t * packet, uint16_t size){
67389a78d34SMatthias Ringwald     if (sm_client_packet_handler) {
67489a78d34SMatthias Ringwald         sm_client_packet_handler(HCI_EVENT_PACKET, 0, packet, size);
67589a78d34SMatthias Ringwald     }
67689a78d34SMatthias Ringwald     // dispatch to all event handlers
67789a78d34SMatthias Ringwald     btstack_linked_list_iterator_t it;
67889a78d34SMatthias Ringwald     btstack_linked_list_iterator_init(&it, &sm_event_handlers);
67989a78d34SMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)){
68089a78d34SMatthias Ringwald         btstack_packet_callback_registration_t * entry = (btstack_packet_callback_registration_t*) btstack_linked_list_iterator_next(&it);
681d9a7306aSMatthias Ringwald         entry->callback(packet_type, 0, packet, size);
68289a78d34SMatthias Ringwald     }
68389a78d34SMatthias Ringwald }
68489a78d34SMatthias Ringwald 
685711e6c80SMatthias Ringwald static void sm_notify_client_base(uint8_t type, hci_con_handle_t con_handle, uint8_t addr_type, bd_addr_t address){
6863deb3ec6SMatthias Ringwald     uint8_t event[11];
687711e6c80SMatthias Ringwald     sm_setup_event_base(event, sizeof(event), type, con_handle, addr_type, address);
68889a78d34SMatthias Ringwald     sm_dispatch_event(HCI_EVENT_PACKET, 0, event, sizeof(event));
6893deb3ec6SMatthias Ringwald }
6903deb3ec6SMatthias Ringwald 
691711e6c80SMatthias Ringwald static void sm_notify_client_passkey(uint8_t type, hci_con_handle_t con_handle, uint8_t addr_type, bd_addr_t address, uint32_t passkey){
6923deb3ec6SMatthias Ringwald     uint8_t event[15];
693711e6c80SMatthias Ringwald     sm_setup_event_base(event, sizeof(event), type, con_handle, addr_type, address);
694f8fbdce0SMatthias Ringwald     little_endian_store_32(event, 11, passkey);
69589a78d34SMatthias Ringwald     sm_dispatch_event(HCI_EVENT_PACKET, 0, event, sizeof(event));
6963deb3ec6SMatthias Ringwald }
6973deb3ec6SMatthias Ringwald 
698711e6c80SMatthias Ringwald static void sm_notify_client_index(uint8_t type, hci_con_handle_t con_handle, uint8_t addr_type, bd_addr_t address, uint16_t index){
6993deb3ec6SMatthias Ringwald     uint8_t event[13];
700711e6c80SMatthias Ringwald     sm_setup_event_base(event, sizeof(event), type, con_handle, addr_type, address);
701f8fbdce0SMatthias Ringwald     little_endian_store_16(event, 11, index);
70289a78d34SMatthias Ringwald     sm_dispatch_event(HCI_EVENT_PACKET, 0, event, sizeof(event));
7033deb3ec6SMatthias Ringwald }
7043deb3ec6SMatthias Ringwald 
705711e6c80SMatthias Ringwald static void sm_notify_client_authorization(uint8_t type, hci_con_handle_t con_handle, uint8_t addr_type, bd_addr_t address, uint8_t result){
7063deb3ec6SMatthias Ringwald 
7073deb3ec6SMatthias Ringwald     uint8_t event[18];
708711e6c80SMatthias Ringwald     sm_setup_event_base(event, sizeof(event), type, con_handle, addr_type, address);
7093deb3ec6SMatthias Ringwald     event[11] = result;
71089a78d34SMatthias Ringwald     sm_dispatch_event(HCI_EVENT_PACKET, 0, (uint8_t*) &event, sizeof(event));
7113deb3ec6SMatthias Ringwald }
7123deb3ec6SMatthias Ringwald 
7133deb3ec6SMatthias Ringwald // decide on stk generation based on
7143deb3ec6SMatthias Ringwald // - pairing request
7153deb3ec6SMatthias Ringwald // - io capabilities
7163deb3ec6SMatthias Ringwald // - OOB data availability
7173deb3ec6SMatthias Ringwald static void sm_setup_tk(void){
7183deb3ec6SMatthias Ringwald 
7193deb3ec6SMatthias Ringwald     // default: just works
7203deb3ec6SMatthias Ringwald     setup->sm_stk_generation_method = JUST_WORKS;
7213deb3ec6SMatthias Ringwald 
72227c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
72327c32905SMatthias Ringwald     setup->sm_use_secure_connections = ( sm_pairing_packet_get_auth_req(setup->sm_m_preq)
72427c32905SMatthias Ringwald                                        & sm_pairing_packet_get_auth_req(setup->sm_s_pres)
72527c32905SMatthias Ringwald                                        & SM_AUTHREQ_SECURE_CONNECTION ) != 0;
726446a8c36SMatthias Ringwald     memset(setup->sm_ra, 0, 16);
727446a8c36SMatthias Ringwald     memset(setup->sm_rb, 0, 16);
72827c32905SMatthias Ringwald #else
72927c32905SMatthias Ringwald     setup->sm_use_secure_connections = 0;
73027c32905SMatthias Ringwald #endif
73127c32905SMatthias Ringwald 
73227c32905SMatthias Ringwald     // If both devices have not set the MITM option in the Authentication Requirements
73327c32905SMatthias Ringwald     // Flags, then the IO capabilities shall be ignored and the Just Works association
73427c32905SMatthias Ringwald     // model shall be used.
73527c32905SMatthias Ringwald     if (((sm_pairing_packet_get_auth_req(setup->sm_m_preq) & SM_AUTHREQ_MITM_PROTECTION) == 0)
73627c32905SMatthias Ringwald     &&  ((sm_pairing_packet_get_auth_req(setup->sm_s_pres) & SM_AUTHREQ_MITM_PROTECTION) == 0)){
73727c32905SMatthias Ringwald         log_info("SM: MITM not required by both -> JUST WORKS");
73827c32905SMatthias Ringwald         return;
73927c32905SMatthias Ringwald     }
74027c32905SMatthias Ringwald 
74127c32905SMatthias Ringwald     // TODO: with LE SC, OOB is used to transfer data OOB during pairing, single device with OOB is sufficient
74227c32905SMatthias Ringwald 
7433deb3ec6SMatthias Ringwald     // If both devices have out of band authentication data, then the Authentication
7443deb3ec6SMatthias Ringwald     // Requirements Flags shall be ignored when selecting the pairing method and the
7453deb3ec6SMatthias Ringwald     // Out of Band pairing method shall be used.
7461ad129beSMatthias Ringwald     if (sm_pairing_packet_get_oob_data_flag(setup->sm_m_preq)
7471ad129beSMatthias Ringwald     &&  sm_pairing_packet_get_oob_data_flag(setup->sm_s_pres)){
7483deb3ec6SMatthias Ringwald         log_info("SM: have OOB data");
7498314c363SMatthias Ringwald         log_info_key("OOB", setup->sm_tk);
7503deb3ec6SMatthias Ringwald         setup->sm_stk_generation_method = OOB;
7513deb3ec6SMatthias Ringwald         return;
7523deb3ec6SMatthias Ringwald     }
7533deb3ec6SMatthias Ringwald 
7543deb3ec6SMatthias Ringwald     // Reset TK as it has been setup in sm_init_setup
7553deb3ec6SMatthias Ringwald     sm_reset_tk();
7563deb3ec6SMatthias Ringwald 
7573deb3ec6SMatthias Ringwald     // Also use just works if unknown io capabilites
7588da2e96dSMatthias Ringwald     if ((sm_pairing_packet_get_io_capability(setup->sm_m_preq) > IO_CAPABILITY_KEYBOARD_DISPLAY) || (sm_pairing_packet_get_io_capability(setup->sm_s_pres) > IO_CAPABILITY_KEYBOARD_DISPLAY)){
7593deb3ec6SMatthias Ringwald         return;
7603deb3ec6SMatthias Ringwald     }
7613deb3ec6SMatthias Ringwald 
7623deb3ec6SMatthias Ringwald     // Otherwise the IO capabilities of the devices shall be used to determine the
7633deb3ec6SMatthias Ringwald     // pairing method as defined in Table 2.4.
76427c32905SMatthias Ringwald     // see http://stackoverflow.com/a/1052837/393697 for how to specify pointer to 2-dimensional array
76527c32905SMatthias Ringwald     const stk_generation_method_t (*generation_method)[5] = stk_generation_method;
76627c32905SMatthias Ringwald 
76727c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
768c6b7cbd9SMatthias Ringwald     // table not define by default
76927c32905SMatthias Ringwald     if (setup->sm_use_secure_connections){
77027c32905SMatthias Ringwald         generation_method = stk_generation_method_with_secure_connection;
77127c32905SMatthias Ringwald     }
77227c32905SMatthias Ringwald #endif
77327c32905SMatthias Ringwald     setup->sm_stk_generation_method = generation_method[sm_pairing_packet_get_io_capability(setup->sm_s_pres)][sm_pairing_packet_get_io_capability(setup->sm_m_preq)];
77427c32905SMatthias Ringwald 
7753deb3ec6SMatthias Ringwald     log_info("sm_setup_tk: master io cap: %u, slave io cap: %u -> method %u",
7761ad129beSMatthias Ringwald         sm_pairing_packet_get_io_capability(setup->sm_m_preq), sm_pairing_packet_get_io_capability(setup->sm_s_pres), setup->sm_stk_generation_method);
7773deb3ec6SMatthias Ringwald }
7783deb3ec6SMatthias Ringwald 
7793deb3ec6SMatthias Ringwald static int sm_key_distribution_flags_for_set(uint8_t key_set){
7803deb3ec6SMatthias Ringwald     int flags = 0;
7813deb3ec6SMatthias Ringwald     if (key_set & SM_KEYDIST_ENC_KEY){
7823deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_FLAG_ENCRYPTION_INFORMATION;
7833deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_FLAG_MASTER_IDENTIFICATION;
7843deb3ec6SMatthias Ringwald     }
7853deb3ec6SMatthias Ringwald     if (key_set & SM_KEYDIST_ID_KEY){
7863deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_FLAG_IDENTITY_INFORMATION;
7873deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_FLAG_IDENTITY_ADDRESS_INFORMATION;
7883deb3ec6SMatthias Ringwald     }
7893deb3ec6SMatthias Ringwald     if (key_set & SM_KEYDIST_SIGN){
7903deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION;
7913deb3ec6SMatthias Ringwald     }
7923deb3ec6SMatthias Ringwald     return flags;
7933deb3ec6SMatthias Ringwald }
7943deb3ec6SMatthias Ringwald 
7953deb3ec6SMatthias Ringwald static void sm_setup_key_distribution(uint8_t key_set){
7963deb3ec6SMatthias Ringwald     setup->sm_key_distribution_received_set = 0;
7973deb3ec6SMatthias Ringwald     setup->sm_key_distribution_send_set = sm_key_distribution_flags_for_set(key_set);
7983deb3ec6SMatthias Ringwald }
7993deb3ec6SMatthias Ringwald 
8003deb3ec6SMatthias Ringwald // CSRK Key Lookup
8013deb3ec6SMatthias Ringwald 
8023deb3ec6SMatthias Ringwald 
8033deb3ec6SMatthias Ringwald static int sm_address_resolution_idle(void){
8043deb3ec6SMatthias Ringwald     return sm_address_resolution_mode == ADDRESS_RESOLUTION_IDLE;
8053deb3ec6SMatthias Ringwald }
8063deb3ec6SMatthias Ringwald 
807711e6c80SMatthias Ringwald static void sm_address_resolution_start_lookup(uint8_t addr_type, hci_con_handle_t con_handle, bd_addr_t addr, address_resolution_mode_t mode, void * context){
8083deb3ec6SMatthias Ringwald     memcpy(sm_address_resolution_address, addr, 6);
8093deb3ec6SMatthias Ringwald     sm_address_resolution_addr_type = addr_type;
8103deb3ec6SMatthias Ringwald     sm_address_resolution_test = 0;
8113deb3ec6SMatthias Ringwald     sm_address_resolution_mode = mode;
8123deb3ec6SMatthias Ringwald     sm_address_resolution_context = context;
813711e6c80SMatthias Ringwald     sm_notify_client_base(SM_EVENT_IDENTITY_RESOLVING_STARTED, con_handle, addr_type, addr);
8143deb3ec6SMatthias Ringwald }
8153deb3ec6SMatthias Ringwald 
8163deb3ec6SMatthias Ringwald int sm_address_resolution_lookup(uint8_t address_type, bd_addr_t address){
8173deb3ec6SMatthias Ringwald     // check if already in list
818665d90f2SMatthias Ringwald     btstack_linked_list_iterator_t it;
8193deb3ec6SMatthias Ringwald     sm_lookup_entry_t * entry;
820665d90f2SMatthias Ringwald     btstack_linked_list_iterator_init(&it, &sm_address_resolution_general_queue);
821665d90f2SMatthias Ringwald     while(btstack_linked_list_iterator_has_next(&it)){
822665d90f2SMatthias Ringwald         entry = (sm_lookup_entry_t *) btstack_linked_list_iterator_next(&it);
8233deb3ec6SMatthias Ringwald         if (entry->address_type != address_type) continue;
8243deb3ec6SMatthias Ringwald         if (memcmp(entry->address, address, 6))  continue;
8253deb3ec6SMatthias Ringwald         // already in list
8263deb3ec6SMatthias Ringwald         return BTSTACK_BUSY;
8273deb3ec6SMatthias Ringwald     }
8283deb3ec6SMatthias Ringwald     entry = btstack_memory_sm_lookup_entry_get();
8293deb3ec6SMatthias Ringwald     if (!entry) return BTSTACK_MEMORY_ALLOC_FAILED;
8303deb3ec6SMatthias Ringwald     entry->address_type = (bd_addr_type_t) address_type;
8313deb3ec6SMatthias Ringwald     memcpy(entry->address, address, 6);
832665d90f2SMatthias Ringwald     btstack_linked_list_add(&sm_address_resolution_general_queue, (btstack_linked_item_t *) entry);
8333deb3ec6SMatthias Ringwald     sm_run();
8343deb3ec6SMatthias Ringwald     return 0;
8353deb3ec6SMatthias Ringwald }
8363deb3ec6SMatthias Ringwald 
8373deb3ec6SMatthias Ringwald // CMAC Implementation using AES128 engine
8383deb3ec6SMatthias Ringwald static void sm_shift_left_by_one_bit_inplace(int len, uint8_t * data){
8393deb3ec6SMatthias Ringwald     int i;
8403deb3ec6SMatthias Ringwald     int carry = 0;
8413deb3ec6SMatthias Ringwald     for (i=len-1; i >= 0 ; i--){
8423deb3ec6SMatthias Ringwald         int new_carry = data[i] >> 7;
8433deb3ec6SMatthias Ringwald         data[i] = data[i] << 1 | carry;
8443deb3ec6SMatthias Ringwald         carry = new_carry;
8453deb3ec6SMatthias Ringwald     }
8463deb3ec6SMatthias Ringwald }
8473deb3ec6SMatthias Ringwald 
8483deb3ec6SMatthias Ringwald // while x_state++ for an enum is possible in C, it isn't in C++. we use this helpers to avoid compile errors for now
8493deb3ec6SMatthias Ringwald static inline void sm_next_responding_state(sm_connection_t * sm_conn){
8503deb3ec6SMatthias Ringwald     sm_conn->sm_engine_state = (security_manager_state_t) (((int)sm_conn->sm_engine_state) + 1);
8513deb3ec6SMatthias Ringwald }
8523deb3ec6SMatthias Ringwald static inline void dkg_next_state(void){
8533deb3ec6SMatthias Ringwald     dkg_state = (derived_key_generation_t) (((int)dkg_state) + 1);
8543deb3ec6SMatthias Ringwald }
8553deb3ec6SMatthias Ringwald static inline void rau_next_state(void){
8563deb3ec6SMatthias Ringwald     rau_state = (random_address_update_t) (((int)rau_state) + 1);
8573deb3ec6SMatthias Ringwald }
858514d35fcSMatthias Ringwald 
859514d35fcSMatthias Ringwald // CMAC calculation using AES Engine
860514d35fcSMatthias Ringwald 
8613deb3ec6SMatthias Ringwald static inline void sm_cmac_next_state(void){
8623deb3ec6SMatthias Ringwald     sm_cmac_state = (cmac_state_t) (((int)sm_cmac_state) + 1);
8633deb3ec6SMatthias Ringwald }
864514d35fcSMatthias Ringwald 
8653deb3ec6SMatthias Ringwald static int sm_cmac_last_block_complete(void){
8663deb3ec6SMatthias Ringwald     if (sm_cmac_message_len == 0) return 0;
8673deb3ec6SMatthias Ringwald     return (sm_cmac_message_len & 0x0f) == 0;
8683deb3ec6SMatthias Ringwald }
869514d35fcSMatthias Ringwald 
870514d35fcSMatthias Ringwald static inline uint8_t sm_cmac_message_get_byte(uint16_t offset){
8713deb3ec6SMatthias Ringwald     if (offset >= sm_cmac_message_len) {
8723deb3ec6SMatthias Ringwald         log_error("sm_cmac_message_get_byte. out of bounds, access %u, len %u", offset, sm_cmac_message_len);
8733deb3ec6SMatthias Ringwald         return 0;
8743deb3ec6SMatthias Ringwald     }
8753deb3ec6SMatthias Ringwald 
8763deb3ec6SMatthias Ringwald     offset = sm_cmac_message_len - 1 - offset;
8773deb3ec6SMatthias Ringwald 
8783deb3ec6SMatthias Ringwald     // sm_cmac_header[3] | message[] | sm_cmac_sign_counter[4]
8793deb3ec6SMatthias Ringwald     if (offset < 3){
8803deb3ec6SMatthias Ringwald         return sm_cmac_header[offset];
8813deb3ec6SMatthias Ringwald     }
8823deb3ec6SMatthias Ringwald     int actual_message_len_incl_header = sm_cmac_message_len - 4;
8833deb3ec6SMatthias Ringwald     if (offset <  actual_message_len_incl_header){
8843deb3ec6SMatthias Ringwald         return sm_cmac_message[offset - 3];
8853deb3ec6SMatthias Ringwald     }
8863deb3ec6SMatthias Ringwald     return sm_cmac_sign_counter[offset - actual_message_len_incl_header];
8873deb3ec6SMatthias Ringwald }
8883deb3ec6SMatthias Ringwald 
889514d35fcSMatthias Ringwald // generic cmac calculation
890aec94140SMatthias Ringwald void sm_cmac_general_start(const sm_key_t key, uint16_t message_len, uint8_t (*get_byte_callback)(uint16_t offset), void (*done_callback)(uint8_t hash[8])){
891514d35fcSMatthias Ringwald     // Generalized CMAC
892514d35fcSMatthias Ringwald     memcpy(sm_cmac_k, key, 16);
8933deb3ec6SMatthias Ringwald     memset(sm_cmac_x, 0, 16);
894514d35fcSMatthias Ringwald     sm_cmac_block_current = 0;
895514d35fcSMatthias Ringwald     sm_cmac_message_len  = message_len;
896514d35fcSMatthias Ringwald     sm_cmac_done_handler = done_callback;
897514d35fcSMatthias Ringwald     sm_cmac_get_byte     = get_byte_callback;
8983deb3ec6SMatthias Ringwald 
8993deb3ec6SMatthias Ringwald     // step 2: n := ceil(len/const_Bsize);
9003deb3ec6SMatthias Ringwald     sm_cmac_block_count = (sm_cmac_message_len + 15) / 16;
9013deb3ec6SMatthias Ringwald 
9023deb3ec6SMatthias Ringwald     // step 3: ..
9033deb3ec6SMatthias Ringwald     if (sm_cmac_block_count==0){
9043deb3ec6SMatthias Ringwald         sm_cmac_block_count = 1;
9053deb3ec6SMatthias Ringwald     }
906514d35fcSMatthias Ringwald     log_info("sm_cmac_general_start: len %u, block count %u", sm_cmac_message_len, sm_cmac_block_count);
9073deb3ec6SMatthias Ringwald 
9083deb3ec6SMatthias Ringwald     // first, we need to compute l for k1, k2, and m_last
9093deb3ec6SMatthias Ringwald     sm_cmac_state = CMAC_CALC_SUBKEYS;
9103deb3ec6SMatthias Ringwald 
9113deb3ec6SMatthias Ringwald     // let's go
9123deb3ec6SMatthias Ringwald     sm_run();
9133deb3ec6SMatthias Ringwald }
9143deb3ec6SMatthias Ringwald 
915514d35fcSMatthias Ringwald // cmac for ATT Message signing
916aec94140SMatthias Ringwald void sm_cmac_start(const sm_key_t k, uint8_t opcode, hci_con_handle_t con_handle, uint16_t message_len, const uint8_t * message, uint32_t sign_counter, void (*done_handler)(uint8_t * hash)){
917514d35fcSMatthias Ringwald     // ATT Message Signing
918514d35fcSMatthias Ringwald     sm_cmac_header[0] = opcode;
919514d35fcSMatthias Ringwald     little_endian_store_16(sm_cmac_header, 1, con_handle);
920514d35fcSMatthias Ringwald     little_endian_store_32(sm_cmac_sign_counter, 0, sign_counter);
921514d35fcSMatthias Ringwald     uint16_t total_message_len = 3 + message_len + 4;  // incl. virtually prepended att opcode, handle and appended sign_counter in LE
922514d35fcSMatthias Ringwald     sm_cmac_message = message;
923514d35fcSMatthias Ringwald     sm_cmac_general_start(k, total_message_len, &sm_cmac_message_get_byte, done_handler);
924514d35fcSMatthias Ringwald }
925514d35fcSMatthias Ringwald 
9263deb3ec6SMatthias Ringwald int sm_cmac_ready(void){
9273deb3ec6SMatthias Ringwald     return sm_cmac_state == CMAC_IDLE;
9283deb3ec6SMatthias Ringwald }
9293deb3ec6SMatthias Ringwald 
9303deb3ec6SMatthias Ringwald static void sm_cmac_handle_aes_engine_ready(void){
9313deb3ec6SMatthias Ringwald     switch (sm_cmac_state){
9323deb3ec6SMatthias Ringwald         case CMAC_CALC_SUBKEYS: {
9333deb3ec6SMatthias Ringwald             sm_key_t const_zero;
9343deb3ec6SMatthias Ringwald             memset(const_zero, 0, 16);
9353deb3ec6SMatthias Ringwald             sm_cmac_next_state();
9363deb3ec6SMatthias Ringwald             sm_aes128_start(sm_cmac_k, const_zero, NULL);
9373deb3ec6SMatthias Ringwald             break;
9383deb3ec6SMatthias Ringwald         }
9393deb3ec6SMatthias Ringwald         case CMAC_CALC_MI: {
9403deb3ec6SMatthias Ringwald             int j;
9413deb3ec6SMatthias Ringwald             sm_key_t y;
9423deb3ec6SMatthias Ringwald             for (j=0;j<16;j++){
943514d35fcSMatthias Ringwald                 y[j] = sm_cmac_x[j] ^ sm_cmac_get_byte(sm_cmac_block_current*16 + j);
9443deb3ec6SMatthias Ringwald             }
9453deb3ec6SMatthias Ringwald             sm_cmac_block_current++;
9463deb3ec6SMatthias Ringwald             sm_cmac_next_state();
9473deb3ec6SMatthias Ringwald             sm_aes128_start(sm_cmac_k, y, NULL);
9483deb3ec6SMatthias Ringwald             break;
9493deb3ec6SMatthias Ringwald         }
9503deb3ec6SMatthias Ringwald         case CMAC_CALC_MLAST: {
9513deb3ec6SMatthias Ringwald             int i;
9523deb3ec6SMatthias Ringwald             sm_key_t y;
9533deb3ec6SMatthias Ringwald             for (i=0;i<16;i++){
9543deb3ec6SMatthias Ringwald                 y[i] = sm_cmac_x[i] ^ sm_cmac_m_last[i];
9553deb3ec6SMatthias Ringwald             }
9568314c363SMatthias Ringwald             log_info_key("Y", y);
9573deb3ec6SMatthias Ringwald             sm_cmac_block_current++;
9583deb3ec6SMatthias Ringwald             sm_cmac_next_state();
9593deb3ec6SMatthias Ringwald             sm_aes128_start(sm_cmac_k, y, NULL);
9603deb3ec6SMatthias Ringwald             break;
9613deb3ec6SMatthias Ringwald         }
9623deb3ec6SMatthias Ringwald         default:
9633deb3ec6SMatthias Ringwald             log_info("sm_cmac_handle_aes_engine_ready called in state %u", sm_cmac_state);
9643deb3ec6SMatthias Ringwald             break;
9653deb3ec6SMatthias Ringwald     }
9663deb3ec6SMatthias Ringwald }
9673deb3ec6SMatthias Ringwald 
9683deb3ec6SMatthias Ringwald static void sm_cmac_handle_encryption_result(sm_key_t data){
9693deb3ec6SMatthias Ringwald     switch (sm_cmac_state){
9703deb3ec6SMatthias Ringwald         case CMAC_W4_SUBKEYS: {
9713deb3ec6SMatthias Ringwald             sm_key_t k1;
9723deb3ec6SMatthias Ringwald             memcpy(k1, data, 16);
9733deb3ec6SMatthias Ringwald             sm_shift_left_by_one_bit_inplace(16, k1);
9743deb3ec6SMatthias Ringwald             if (data[0] & 0x80){
9753deb3ec6SMatthias Ringwald                 k1[15] ^= 0x87;
9763deb3ec6SMatthias Ringwald             }
9773deb3ec6SMatthias Ringwald             sm_key_t k2;
9783deb3ec6SMatthias Ringwald             memcpy(k2, k1, 16);
9793deb3ec6SMatthias Ringwald             sm_shift_left_by_one_bit_inplace(16, k2);
9803deb3ec6SMatthias Ringwald             if (k1[0] & 0x80){
9813deb3ec6SMatthias Ringwald                 k2[15] ^= 0x87;
9823deb3ec6SMatthias Ringwald             }
9833deb3ec6SMatthias Ringwald 
9848314c363SMatthias Ringwald             log_info_key("k", sm_cmac_k);
9858314c363SMatthias Ringwald             log_info_key("k1", k1);
9868314c363SMatthias Ringwald             log_info_key("k2", k2);
9873deb3ec6SMatthias Ringwald 
9883deb3ec6SMatthias Ringwald             // step 4: set m_last
9893deb3ec6SMatthias Ringwald             int i;
9903deb3ec6SMatthias Ringwald             if (sm_cmac_last_block_complete()){
9913deb3ec6SMatthias Ringwald                 for (i=0;i<16;i++){
992514d35fcSMatthias Ringwald                     sm_cmac_m_last[i] = sm_cmac_get_byte(sm_cmac_message_len - 16 + i) ^ k1[i];
9933deb3ec6SMatthias Ringwald                 }
9943deb3ec6SMatthias Ringwald             } else {
9953deb3ec6SMatthias Ringwald                 int valid_octets_in_last_block = sm_cmac_message_len & 0x0f;
9963deb3ec6SMatthias Ringwald                 for (i=0;i<16;i++){
9973deb3ec6SMatthias Ringwald                     if (i < valid_octets_in_last_block){
998514d35fcSMatthias Ringwald                         sm_cmac_m_last[i] = sm_cmac_get_byte((sm_cmac_message_len & 0xfff0) + i) ^ k2[i];
9993deb3ec6SMatthias Ringwald                         continue;
10003deb3ec6SMatthias Ringwald                     }
10013deb3ec6SMatthias Ringwald                     if (i == valid_octets_in_last_block){
10023deb3ec6SMatthias Ringwald                         sm_cmac_m_last[i] = 0x80 ^ k2[i];
10033deb3ec6SMatthias Ringwald                         continue;
10043deb3ec6SMatthias Ringwald                     }
10053deb3ec6SMatthias Ringwald                     sm_cmac_m_last[i] = k2[i];
10063deb3ec6SMatthias Ringwald                 }
10073deb3ec6SMatthias Ringwald             }
10083deb3ec6SMatthias Ringwald 
10093deb3ec6SMatthias Ringwald             // next
10103deb3ec6SMatthias Ringwald             sm_cmac_state = sm_cmac_block_current < sm_cmac_block_count - 1 ? CMAC_CALC_MI : CMAC_CALC_MLAST;
10113deb3ec6SMatthias Ringwald             break;
10123deb3ec6SMatthias Ringwald         }
10133deb3ec6SMatthias Ringwald         case CMAC_W4_MI:
10143deb3ec6SMatthias Ringwald             memcpy(sm_cmac_x, data, 16);
10153deb3ec6SMatthias Ringwald             sm_cmac_state = sm_cmac_block_current < sm_cmac_block_count - 1 ? CMAC_CALC_MI : CMAC_CALC_MLAST;
10163deb3ec6SMatthias Ringwald             break;
10173deb3ec6SMatthias Ringwald         case CMAC_W4_MLAST:
10183deb3ec6SMatthias Ringwald             // done
10190346c37cSMatthias Ringwald             log_info("Setting CMAC Engine to IDLE");
10200346c37cSMatthias Ringwald             sm_cmac_state = CMAC_IDLE;
10218314c363SMatthias Ringwald             log_info_key("CMAC", data);
10223deb3ec6SMatthias Ringwald             sm_cmac_done_handler(data);
10233deb3ec6SMatthias Ringwald             break;
10243deb3ec6SMatthias Ringwald         default:
10253deb3ec6SMatthias Ringwald             log_info("sm_cmac_handle_encryption_result called in state %u", sm_cmac_state);
10263deb3ec6SMatthias Ringwald             break;
10273deb3ec6SMatthias Ringwald     }
10283deb3ec6SMatthias Ringwald }
10293deb3ec6SMatthias Ringwald 
10303deb3ec6SMatthias Ringwald static void sm_trigger_user_response(sm_connection_t * sm_conn){
1031446a8c36SMatthias Ringwald     // notify client for: JUST WORKS confirm, Numeric comparison confirm, PASSKEY display or input
10323deb3ec6SMatthias Ringwald     setup->sm_user_response = SM_USER_RESPONSE_IDLE;
10333deb3ec6SMatthias Ringwald     switch (setup->sm_stk_generation_method){
10343deb3ec6SMatthias Ringwald         case PK_RESP_INPUT:
10353deb3ec6SMatthias Ringwald             if (sm_conn->sm_role){
10363deb3ec6SMatthias Ringwald                 setup->sm_user_response = SM_USER_RESPONSE_PENDING;
10375611a760SMatthias Ringwald                 sm_notify_client_base(SM_EVENT_PASSKEY_INPUT_NUMBER, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address);
10383deb3ec6SMatthias Ringwald             } else {
1039c9b8fdd9SMatthias Ringwald                 sm_notify_client_passkey(SM_EVENT_PASSKEY_DISPLAY_NUMBER, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, big_endian_read_32(setup->sm_tk, 12));
10403deb3ec6SMatthias Ringwald             }
10413deb3ec6SMatthias Ringwald             break;
10423deb3ec6SMatthias Ringwald         case PK_INIT_INPUT:
10433deb3ec6SMatthias Ringwald             if (sm_conn->sm_role){
1044c9b8fdd9SMatthias Ringwald                 sm_notify_client_passkey(SM_EVENT_PASSKEY_DISPLAY_NUMBER, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, big_endian_read_32(setup->sm_tk, 12));
10453deb3ec6SMatthias Ringwald             } else {
10463deb3ec6SMatthias Ringwald                 setup->sm_user_response = SM_USER_RESPONSE_PENDING;
10475611a760SMatthias Ringwald                 sm_notify_client_base(SM_EVENT_PASSKEY_INPUT_NUMBER, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address);
10483deb3ec6SMatthias Ringwald             }
10493deb3ec6SMatthias Ringwald             break;
10503deb3ec6SMatthias Ringwald         case OK_BOTH_INPUT:
10513deb3ec6SMatthias Ringwald             setup->sm_user_response = SM_USER_RESPONSE_PENDING;
10525611a760SMatthias Ringwald             sm_notify_client_base(SM_EVENT_PASSKEY_INPUT_NUMBER, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address);
10533deb3ec6SMatthias Ringwald             break;
105427c32905SMatthias Ringwald         case NK_BOTH_INPUT:
1055446a8c36SMatthias Ringwald             setup->sm_user_response = SM_USER_RESPONSE_PENDING;
1056c8c46d51SMatthias Ringwald             sm_notify_client_passkey(SM_EVENT_NUMERIC_COMPARISON_REQUEST, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, big_endian_read_32(setup->sm_tk, 12));
105727c32905SMatthias Ringwald             break;
10583deb3ec6SMatthias Ringwald         case JUST_WORKS:
10593deb3ec6SMatthias Ringwald             setup->sm_user_response = SM_USER_RESPONSE_PENDING;
10605611a760SMatthias Ringwald             sm_notify_client_base(SM_EVENT_JUST_WORKS_REQUEST, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address);
10613deb3ec6SMatthias Ringwald             break;
10623deb3ec6SMatthias Ringwald         case OOB:
10633deb3ec6SMatthias Ringwald             // client already provided OOB data, let's skip notification.
10643deb3ec6SMatthias Ringwald             break;
10653deb3ec6SMatthias Ringwald     }
10663deb3ec6SMatthias Ringwald }
10673deb3ec6SMatthias Ringwald 
10683deb3ec6SMatthias Ringwald static int sm_key_distribution_all_received(sm_connection_t * sm_conn){
10693deb3ec6SMatthias Ringwald     int recv_flags;
10703deb3ec6SMatthias Ringwald     if (sm_conn->sm_role){
107152f9cf63SMatthias Ringwald         // slave / responder
10721ad129beSMatthias Ringwald         recv_flags = sm_key_distribution_flags_for_set(sm_pairing_packet_get_initiator_key_distribution(setup->sm_s_pres));
10733deb3ec6SMatthias Ringwald     } else {
10743deb3ec6SMatthias Ringwald         // master / initiator
10751ad129beSMatthias Ringwald         recv_flags = sm_key_distribution_flags_for_set(sm_pairing_packet_get_responder_key_distribution(setup->sm_s_pres));
10763deb3ec6SMatthias Ringwald     }
10773deb3ec6SMatthias Ringwald     log_debug("sm_key_distribution_all_received: received 0x%02x, expecting 0x%02x", setup->sm_key_distribution_received_set, recv_flags);
10783deb3ec6SMatthias Ringwald     return recv_flags == setup->sm_key_distribution_received_set;
10793deb3ec6SMatthias Ringwald }
10803deb3ec6SMatthias Ringwald 
1081711e6c80SMatthias Ringwald static void sm_done_for_handle(hci_con_handle_t con_handle){
1082711e6c80SMatthias Ringwald     if (sm_active_connection == con_handle){
10833deb3ec6SMatthias Ringwald         sm_timeout_stop();
10843deb3ec6SMatthias Ringwald         sm_active_connection = 0;
1085711e6c80SMatthias Ringwald         log_info("sm: connection 0x%x released setup context", con_handle);
10863deb3ec6SMatthias Ringwald     }
10873deb3ec6SMatthias Ringwald }
10883deb3ec6SMatthias Ringwald 
10893deb3ec6SMatthias Ringwald static int sm_key_distribution_flags_for_auth_req(void){
10903deb3ec6SMatthias Ringwald     int flags = SM_KEYDIST_ID_KEY | SM_KEYDIST_SIGN;
10913deb3ec6SMatthias Ringwald     if (sm_auth_req & SM_AUTHREQ_BONDING){
10923deb3ec6SMatthias Ringwald         // encryption information only if bonding requested
10933deb3ec6SMatthias Ringwald         flags |= SM_KEYDIST_ENC_KEY;
10943deb3ec6SMatthias Ringwald     }
10953deb3ec6SMatthias Ringwald     return flags;
10963deb3ec6SMatthias Ringwald }
10973deb3ec6SMatthias Ringwald 
10983deb3ec6SMatthias Ringwald static void sm_init_setup(sm_connection_t * sm_conn){
10993deb3ec6SMatthias Ringwald 
11003deb3ec6SMatthias Ringwald     // fill in sm setup
1101901c000fSMatthias Ringwald     setup->sm_state_vars = 0;
11023deb3ec6SMatthias Ringwald     sm_reset_tk();
11033deb3ec6SMatthias Ringwald     setup->sm_peer_addr_type = sm_conn->sm_peer_addr_type;
11043deb3ec6SMatthias Ringwald     memcpy(setup->sm_peer_address, sm_conn->sm_peer_address, 6);
11053deb3ec6SMatthias Ringwald 
11063deb3ec6SMatthias Ringwald     // query client for OOB data
11073deb3ec6SMatthias Ringwald     int have_oob_data = 0;
11083deb3ec6SMatthias Ringwald     if (sm_get_oob_data) {
11093deb3ec6SMatthias Ringwald         have_oob_data = (*sm_get_oob_data)(sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, setup->sm_tk);
11103deb3ec6SMatthias Ringwald     }
11113deb3ec6SMatthias Ringwald 
11123deb3ec6SMatthias Ringwald     sm_pairing_packet_t * local_packet;
11133deb3ec6SMatthias Ringwald     if (sm_conn->sm_role){
11143deb3ec6SMatthias Ringwald         // slave
11153deb3ec6SMatthias Ringwald         local_packet = &setup->sm_s_pres;
111615a95bd5SMatthias Ringwald         gap_advertisements_get_address(&setup->sm_s_addr_type, setup->sm_s_address);
11173deb3ec6SMatthias Ringwald         setup->sm_m_addr_type = sm_conn->sm_peer_addr_type;
11183deb3ec6SMatthias Ringwald         memcpy(setup->sm_m_address, sm_conn->sm_peer_address, 6);
11193deb3ec6SMatthias Ringwald     } else {
11203deb3ec6SMatthias Ringwald         // master
11213deb3ec6SMatthias Ringwald         local_packet = &setup->sm_m_preq;
112215a95bd5SMatthias Ringwald         gap_advertisements_get_address(&setup->sm_m_addr_type, setup->sm_m_address);
11233deb3ec6SMatthias Ringwald         setup->sm_s_addr_type = sm_conn->sm_peer_addr_type;
11243deb3ec6SMatthias Ringwald         memcpy(setup->sm_s_address, sm_conn->sm_peer_address, 6);
11253deb3ec6SMatthias Ringwald 
11263deb3ec6SMatthias Ringwald         int key_distribution_flags = sm_key_distribution_flags_for_auth_req();
11271ad129beSMatthias Ringwald         sm_pairing_packet_set_initiator_key_distribution(setup->sm_m_preq, key_distribution_flags);
11281ad129beSMatthias Ringwald         sm_pairing_packet_set_responder_key_distribution(setup->sm_m_preq, key_distribution_flags);
11293deb3ec6SMatthias Ringwald     }
11303deb3ec6SMatthias Ringwald 
11311ad129beSMatthias Ringwald     sm_pairing_packet_set_io_capability(*local_packet, sm_io_capabilities);
11321ad129beSMatthias Ringwald     sm_pairing_packet_set_oob_data_flag(*local_packet, have_oob_data);
11331ad129beSMatthias Ringwald     sm_pairing_packet_set_auth_req(*local_packet, sm_auth_req);
11341ad129beSMatthias Ringwald     sm_pairing_packet_set_max_encryption_key_size(*local_packet, sm_max_encryption_key_size);
11353deb3ec6SMatthias Ringwald }
11363deb3ec6SMatthias Ringwald 
11373deb3ec6SMatthias Ringwald static int sm_stk_generation_init(sm_connection_t * sm_conn){
11383deb3ec6SMatthias Ringwald 
11393deb3ec6SMatthias Ringwald     sm_pairing_packet_t * remote_packet;
11403deb3ec6SMatthias Ringwald     int                   remote_key_request;
11413deb3ec6SMatthias Ringwald     if (sm_conn->sm_role){
114252f9cf63SMatthias Ringwald         // slave / responder
11433deb3ec6SMatthias Ringwald         remote_packet      = &setup->sm_m_preq;
11441ad129beSMatthias Ringwald         remote_key_request = sm_pairing_packet_get_responder_key_distribution(setup->sm_m_preq);
11453deb3ec6SMatthias Ringwald     } else {
11463deb3ec6SMatthias Ringwald         // master / initiator
11473deb3ec6SMatthias Ringwald         remote_packet      = &setup->sm_s_pres;
11481ad129beSMatthias Ringwald         remote_key_request = sm_pairing_packet_get_initiator_key_distribution(setup->sm_s_pres);
11493deb3ec6SMatthias Ringwald     }
11503deb3ec6SMatthias Ringwald 
11513deb3ec6SMatthias Ringwald     // check key size
11521ad129beSMatthias Ringwald     sm_conn->sm_actual_encryption_key_size = sm_calc_actual_encryption_key_size(sm_pairing_packet_get_max_encryption_key_size(*remote_packet));
11533deb3ec6SMatthias Ringwald     if (sm_conn->sm_actual_encryption_key_size == 0) return SM_REASON_ENCRYPTION_KEY_SIZE;
11543deb3ec6SMatthias Ringwald 
11553deb3ec6SMatthias Ringwald     // decide on STK generation method
11563deb3ec6SMatthias Ringwald     sm_setup_tk();
11573deb3ec6SMatthias Ringwald     log_info("SMP: generation method %u", setup->sm_stk_generation_method);
11583deb3ec6SMatthias Ringwald 
11593deb3ec6SMatthias Ringwald     // check if STK generation method is acceptable by client
11603deb3ec6SMatthias Ringwald     if (!sm_validate_stk_generation_method()) return SM_REASON_AUTHENTHICATION_REQUIREMENTS;
11613deb3ec6SMatthias Ringwald 
116252f9cf63SMatthias Ringwald     // identical to responder
116352f9cf63SMatthias Ringwald     sm_setup_key_distribution(remote_key_request);
116452f9cf63SMatthias Ringwald 
11653deb3ec6SMatthias Ringwald     // JUST WORKS doens't provide authentication
11663deb3ec6SMatthias Ringwald     sm_conn->sm_connection_authenticated = setup->sm_stk_generation_method == JUST_WORKS ? 0 : 1;
11673deb3ec6SMatthias Ringwald 
11683deb3ec6SMatthias Ringwald     return 0;
11693deb3ec6SMatthias Ringwald }
11703deb3ec6SMatthias Ringwald 
11713deb3ec6SMatthias Ringwald static void sm_address_resolution_handle_event(address_resolution_event_t event){
11723deb3ec6SMatthias Ringwald 
11733deb3ec6SMatthias Ringwald     // cache and reset context
11743deb3ec6SMatthias Ringwald     int matched_device_id = sm_address_resolution_test;
11753deb3ec6SMatthias Ringwald     address_resolution_mode_t mode = sm_address_resolution_mode;
11763deb3ec6SMatthias Ringwald     void * context = sm_address_resolution_context;
11773deb3ec6SMatthias Ringwald 
11783deb3ec6SMatthias Ringwald     // reset context
11793deb3ec6SMatthias Ringwald     sm_address_resolution_mode = ADDRESS_RESOLUTION_IDLE;
11803deb3ec6SMatthias Ringwald     sm_address_resolution_context = NULL;
11813deb3ec6SMatthias Ringwald     sm_address_resolution_test = -1;
1182711e6c80SMatthias Ringwald     hci_con_handle_t con_handle = 0;
11833deb3ec6SMatthias Ringwald 
11843deb3ec6SMatthias Ringwald     sm_connection_t * sm_connection;
11853deb3ec6SMatthias Ringwald     uint16_t ediv;
11863deb3ec6SMatthias Ringwald     switch (mode){
11873deb3ec6SMatthias Ringwald         case ADDRESS_RESOLUTION_GENERAL:
11883deb3ec6SMatthias Ringwald             break;
11893deb3ec6SMatthias Ringwald         case ADDRESS_RESOLUTION_FOR_CONNECTION:
11903deb3ec6SMatthias Ringwald             sm_connection = (sm_connection_t *) context;
1191711e6c80SMatthias Ringwald             con_handle = sm_connection->sm_handle;
11923deb3ec6SMatthias Ringwald             switch (event){
11933deb3ec6SMatthias Ringwald                 case ADDRESS_RESOLUTION_SUCEEDED:
11943deb3ec6SMatthias Ringwald                     sm_connection->sm_irk_lookup_state = IRK_LOOKUP_SUCCEEDED;
11953deb3ec6SMatthias Ringwald                     sm_connection->sm_le_db_index = matched_device_id;
11963deb3ec6SMatthias Ringwald                     log_info("ADDRESS_RESOLUTION_SUCEEDED, index %d", sm_connection->sm_le_db_index);
11973deb3ec6SMatthias Ringwald                     if (sm_connection->sm_role) break;
11983deb3ec6SMatthias Ringwald                     if (!sm_connection->sm_bonding_requested && !sm_connection->sm_security_request_received) break;
11993deb3ec6SMatthias Ringwald                     sm_connection->sm_security_request_received = 0;
12003deb3ec6SMatthias Ringwald                     sm_connection->sm_bonding_requested = 0;
12013deb3ec6SMatthias Ringwald                     le_device_db_encryption_get(sm_connection->sm_le_db_index, &ediv, NULL, NULL, NULL, NULL, NULL);
12023deb3ec6SMatthias Ringwald                     if (ediv){
12033deb3ec6SMatthias Ringwald                         sm_connection->sm_engine_state = SM_INITIATOR_PH0_HAS_LTK;
12043deb3ec6SMatthias Ringwald                     } else {
12053deb3ec6SMatthias Ringwald                         sm_connection->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
12063deb3ec6SMatthias Ringwald                     }
12073deb3ec6SMatthias Ringwald                     break;
12083deb3ec6SMatthias Ringwald                 case ADDRESS_RESOLUTION_FAILED:
12093deb3ec6SMatthias Ringwald                     sm_connection->sm_irk_lookup_state = IRK_LOOKUP_FAILED;
12103deb3ec6SMatthias Ringwald                     if (sm_connection->sm_role) break;
12113deb3ec6SMatthias Ringwald                     if (!sm_connection->sm_bonding_requested && !sm_connection->sm_security_request_received) break;
12123deb3ec6SMatthias Ringwald                     sm_connection->sm_security_request_received = 0;
12133deb3ec6SMatthias Ringwald                     sm_connection->sm_bonding_requested = 0;
12143deb3ec6SMatthias Ringwald                     sm_connection->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
12153deb3ec6SMatthias Ringwald                     break;
12163deb3ec6SMatthias Ringwald             }
12173deb3ec6SMatthias Ringwald             break;
12183deb3ec6SMatthias Ringwald         default:
12193deb3ec6SMatthias Ringwald             break;
12203deb3ec6SMatthias Ringwald     }
12213deb3ec6SMatthias Ringwald 
12223deb3ec6SMatthias Ringwald     switch (event){
12233deb3ec6SMatthias Ringwald         case ADDRESS_RESOLUTION_SUCEEDED:
1224711e6c80SMatthias Ringwald             sm_notify_client_index(SM_EVENT_IDENTITY_RESOLVING_SUCCEEDED, con_handle, sm_address_resolution_addr_type, sm_address_resolution_address, matched_device_id);
12253deb3ec6SMatthias Ringwald             break;
12263deb3ec6SMatthias Ringwald         case ADDRESS_RESOLUTION_FAILED:
1227711e6c80SMatthias Ringwald             sm_notify_client_base(SM_EVENT_IDENTITY_RESOLVING_FAILED, con_handle, sm_address_resolution_addr_type, sm_address_resolution_address);
12283deb3ec6SMatthias Ringwald             break;
12293deb3ec6SMatthias Ringwald     }
12303deb3ec6SMatthias Ringwald }
12313deb3ec6SMatthias Ringwald 
12323deb3ec6SMatthias Ringwald static void sm_key_distribution_handle_all_received(sm_connection_t * sm_conn){
12333deb3ec6SMatthias Ringwald 
12343deb3ec6SMatthias Ringwald     int le_db_index = -1;
12353deb3ec6SMatthias Ringwald 
12363deb3ec6SMatthias Ringwald     // lookup device based on IRK
12373deb3ec6SMatthias Ringwald     if (setup->sm_key_distribution_received_set & SM_KEYDIST_FLAG_IDENTITY_INFORMATION){
12383deb3ec6SMatthias Ringwald         int i;
12393deb3ec6SMatthias Ringwald         for (i=0; i < le_device_db_count(); i++){
12403deb3ec6SMatthias Ringwald             sm_key_t irk;
12413deb3ec6SMatthias Ringwald             bd_addr_t address;
12423deb3ec6SMatthias Ringwald             int address_type;
12433deb3ec6SMatthias Ringwald             le_device_db_info(i, &address_type, address, irk);
12443deb3ec6SMatthias Ringwald             if (memcmp(irk, setup->sm_peer_irk, 16) == 0){
12453deb3ec6SMatthias Ringwald                 log_info("sm: device found for IRK, updating");
12463deb3ec6SMatthias Ringwald                 le_db_index = i;
12473deb3ec6SMatthias Ringwald                 break;
12483deb3ec6SMatthias Ringwald             }
12493deb3ec6SMatthias Ringwald         }
12503deb3ec6SMatthias Ringwald     }
12513deb3ec6SMatthias Ringwald 
12523deb3ec6SMatthias Ringwald     // if not found, lookup via public address if possible
12533deb3ec6SMatthias Ringwald     log_info("sm peer addr type %u, peer addres %s", setup->sm_peer_addr_type, bd_addr_to_str(setup->sm_peer_address));
12543deb3ec6SMatthias Ringwald     if (le_db_index < 0 && setup->sm_peer_addr_type == BD_ADDR_TYPE_LE_PUBLIC){
12553deb3ec6SMatthias Ringwald         int i;
12563deb3ec6SMatthias Ringwald         for (i=0; i < le_device_db_count(); i++){
12573deb3ec6SMatthias Ringwald             bd_addr_t address;
12583deb3ec6SMatthias Ringwald             int address_type;
12593deb3ec6SMatthias Ringwald             le_device_db_info(i, &address_type, address, NULL);
12603deb3ec6SMatthias Ringwald             log_info("device %u, sm peer addr type %u, peer addres %s", i, address_type, bd_addr_to_str(address));
12613deb3ec6SMatthias Ringwald             if (address_type == BD_ADDR_TYPE_LE_PUBLIC && memcmp(address, setup->sm_peer_address, 6) == 0){
12623deb3ec6SMatthias Ringwald                 log_info("sm: device found for public address, updating");
12633deb3ec6SMatthias Ringwald                 le_db_index = i;
12643deb3ec6SMatthias Ringwald                 break;
12653deb3ec6SMatthias Ringwald             }
12663deb3ec6SMatthias Ringwald         }
12673deb3ec6SMatthias Ringwald     }
12683deb3ec6SMatthias Ringwald 
12693deb3ec6SMatthias Ringwald     // if not found, add to db
12703deb3ec6SMatthias Ringwald     if (le_db_index < 0) {
12713deb3ec6SMatthias Ringwald         le_db_index = le_device_db_add(setup->sm_peer_addr_type, setup->sm_peer_address, setup->sm_peer_irk);
12723deb3ec6SMatthias Ringwald     }
12733deb3ec6SMatthias Ringwald 
12743deb3ec6SMatthias Ringwald     if (le_db_index >= 0){
12753deb3ec6SMatthias Ringwald         le_device_db_local_counter_set(le_db_index, 0);
12763deb3ec6SMatthias Ringwald 
12773deb3ec6SMatthias Ringwald         // store local CSRK
12783deb3ec6SMatthias Ringwald         if (setup->sm_key_distribution_send_set & SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION){
12793deb3ec6SMatthias Ringwald             log_info("sm: store local CSRK");
12803deb3ec6SMatthias Ringwald             le_device_db_local_csrk_set(le_db_index, setup->sm_local_csrk);
12813deb3ec6SMatthias Ringwald             le_device_db_local_counter_set(le_db_index, 0);
12823deb3ec6SMatthias Ringwald         }
12833deb3ec6SMatthias Ringwald 
12843deb3ec6SMatthias Ringwald         // store remote CSRK
12853deb3ec6SMatthias Ringwald         if (setup->sm_key_distribution_received_set & SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION){
12863deb3ec6SMatthias Ringwald             log_info("sm: store remote CSRK");
12873deb3ec6SMatthias Ringwald             le_device_db_remote_csrk_set(le_db_index, setup->sm_peer_csrk);
12883deb3ec6SMatthias Ringwald             le_device_db_remote_counter_set(le_db_index, 0);
12893deb3ec6SMatthias Ringwald         }
12903deb3ec6SMatthias Ringwald 
12913deb3ec6SMatthias Ringwald         // store encryption information
12923deb3ec6SMatthias Ringwald         if (setup->sm_key_distribution_received_set & SM_KEYDIST_FLAG_ENCRYPTION_INFORMATION
12933deb3ec6SMatthias Ringwald             && setup->sm_key_distribution_received_set &  SM_KEYDIST_FLAG_MASTER_IDENTIFICATION){
12943deb3ec6SMatthias Ringwald             log_info("sm: set encryption information (key size %u, authenticatd %u)", sm_conn->sm_actual_encryption_key_size, sm_conn->sm_connection_authenticated);
12953deb3ec6SMatthias Ringwald             le_device_db_encryption_set(le_db_index, setup->sm_peer_ediv, setup->sm_peer_rand, setup->sm_peer_ltk,
12963deb3ec6SMatthias Ringwald                 sm_conn->sm_actual_encryption_key_size, sm_conn->sm_connection_authenticated, sm_conn->sm_connection_authorization_state == AUTHORIZATION_GRANTED);
12973deb3ec6SMatthias Ringwald         }
12983deb3ec6SMatthias Ringwald     }
12993deb3ec6SMatthias Ringwald 
13003deb3ec6SMatthias Ringwald     // keep le_db_index
13013deb3ec6SMatthias Ringwald     sm_conn->sm_le_db_index = le_db_index;
13023deb3ec6SMatthias Ringwald }
13033deb3ec6SMatthias Ringwald 
1304688a08f9SMatthias Ringwald static void sm_pairing_error(sm_connection_t * sm_conn, uint8_t reason){
1305688a08f9SMatthias Ringwald     setup->sm_pairing_failed_reason = reason;
1306688a08f9SMatthias Ringwald     sm_conn->sm_engine_state = SM_GENERAL_SEND_PAIRING_FAILED;
1307688a08f9SMatthias Ringwald }
1308688a08f9SMatthias Ringwald 
1309688a08f9SMatthias Ringwald static inline void sm_pdu_received_in_wrong_state(sm_connection_t * sm_conn){
1310688a08f9SMatthias Ringwald     sm_pairing_error(sm_conn, SM_REASON_UNSPECIFIED_REASON);
1311688a08f9SMatthias Ringwald }
1312688a08f9SMatthias Ringwald 
13139af0f475SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
1314688a08f9SMatthias Ringwald 
1315dc300847SMatthias Ringwald static void sm_sc_prepare_dhkey_check(sm_connection_t * sm_conn);
1316945888f5SMatthias Ringwald static int sm_passkey_used(stk_generation_method_t method);
1317f1c1783eSMatthias Ringwald static int sm_just_works_or_numeric_comparison(stk_generation_method_t method);
1318dc300847SMatthias Ringwald 
1319*68437d83SMatthias Ringwald static void sm_log_ec_keypair(void){
1320*68437d83SMatthias Ringwald     log_info("d: %s", ec_d);
1321*68437d83SMatthias Ringwald     log_info("X: %s", ec_qx);
1322*68437d83SMatthias Ringwald     log_info("Y: %s", ec_qy);
1323*68437d83SMatthias Ringwald }
1324*68437d83SMatthias Ringwald 
1325b35a3de2SMatthias Ringwald static void sm_sc_start_calculating_local_confirm(sm_connection_t * sm_conn){
1326b35a3de2SMatthias Ringwald     if (sm_passkey_used(setup->sm_stk_generation_method)){
1327b35a3de2SMatthias Ringwald         sm_conn->sm_engine_state = SM_SC_W2_GET_RANDOM_A;
1328b35a3de2SMatthias Ringwald     } else {
13291f9d84e9SMatthias Ringwald         sm_conn->sm_engine_state = SM_SC_W2_CMAC_FOR_CONFIRMATION;
1330b35a3de2SMatthias Ringwald     }
1331b35a3de2SMatthias Ringwald }
1332b35a3de2SMatthias Ringwald 
1333688a08f9SMatthias Ringwald static void sm_sc_state_after_receiving_random(sm_connection_t * sm_conn){
1334688a08f9SMatthias Ringwald     if (sm_conn->sm_role){
1335688a08f9SMatthias Ringwald         // Responder
1336688a08f9SMatthias Ringwald         sm_conn->sm_engine_state = SM_SC_SEND_PAIRING_RANDOM;
1337688a08f9SMatthias Ringwald     } else {
1338688a08f9SMatthias Ringwald         // Initiator role
1339688a08f9SMatthias Ringwald         switch (setup->sm_stk_generation_method){
1340688a08f9SMatthias Ringwald             case JUST_WORKS:
1341dc300847SMatthias Ringwald                 sm_sc_prepare_dhkey_check(sm_conn);
1342688a08f9SMatthias Ringwald                 break;
1343688a08f9SMatthias Ringwald 
1344f92edc8eSMatthias Ringwald             case NK_BOTH_INPUT:
1345bd57ffebSMatthias Ringwald                 sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_G2;
1346688a08f9SMatthias Ringwald                 break;
1347688a08f9SMatthias Ringwald             case PK_INIT_INPUT:
1348688a08f9SMatthias Ringwald             case PK_RESP_INPUT:
1349688a08f9SMatthias Ringwald             case OK_BOTH_INPUT:
1350688a08f9SMatthias Ringwald                 if (setup->sm_passkey_bit < 20) {
1351b35a3de2SMatthias Ringwald                     sm_sc_start_calculating_local_confirm(sm_conn);
1352688a08f9SMatthias Ringwald                 } else {
1353dc300847SMatthias Ringwald                     sm_sc_prepare_dhkey_check(sm_conn);
1354688a08f9SMatthias Ringwald                 }
1355688a08f9SMatthias Ringwald                 break;
1356688a08f9SMatthias Ringwald             case OOB:
1357688a08f9SMatthias Ringwald                 // TODO: implement SC OOB
1358688a08f9SMatthias Ringwald                 break;
1359688a08f9SMatthias Ringwald         }
1360688a08f9SMatthias Ringwald     }
1361688a08f9SMatthias Ringwald }
1362688a08f9SMatthias Ringwald 
1363aec94140SMatthias Ringwald static uint8_t sm_sc_cmac_get_byte(uint16_t offset){
1364aec94140SMatthias Ringwald     return sm_cmac_sc_buffer[offset];
1365aec94140SMatthias Ringwald }
1366688a08f9SMatthias Ringwald 
1367aec94140SMatthias Ringwald static void sm_sc_cmac_done(uint8_t * hash){
1368688a08f9SMatthias Ringwald     log_info("sm_sc_cmac_done: ");
1369688a08f9SMatthias Ringwald     log_info_hexdump(hash, 16);
1370688a08f9SMatthias Ringwald 
1371bd57ffebSMatthias Ringwald     sm_connection_t * sm_conn = sm_cmac_connection;
1372bd57ffebSMatthias Ringwald     sm_cmac_connection = NULL;
1373bd57ffebSMatthias Ringwald 
1374bd57ffebSMatthias Ringwald     switch (sm_conn->sm_engine_state){
1375aec94140SMatthias Ringwald         case SM_SC_W4_CMAC_FOR_CONFIRMATION:
1376aec94140SMatthias Ringwald             memcpy(setup->sm_local_confirm, hash, 16);
1377bd57ffebSMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_SEND_CONFIRMATION;
1378aec94140SMatthias Ringwald             break;
1379688a08f9SMatthias Ringwald         case SM_SC_W4_CMAC_FOR_CHECK_CONFIRMATION:
1380688a08f9SMatthias Ringwald             // check
1381688a08f9SMatthias Ringwald             if (0 != memcmp(hash, setup->sm_peer_confirm, 16)){
1382bd57ffebSMatthias Ringwald                 sm_pairing_error(sm_conn, SM_REASON_CONFIRM_VALUE_FAILED);
1383688a08f9SMatthias Ringwald                 break;
1384688a08f9SMatthias Ringwald             }
1385bd57ffebSMatthias Ringwald             sm_sc_state_after_receiving_random(sm_conn);
1386688a08f9SMatthias Ringwald             break;
1387901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_G2: {
1388901c000fSMatthias Ringwald             uint32_t vab = big_endian_read_32(hash, 12) % 1000000;
1389901c000fSMatthias Ringwald             big_endian_store_32(setup->sm_tk, 12, vab);
1390901c000fSMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_W4_USER_RESPONSE;
1391901c000fSMatthias Ringwald             sm_trigger_user_response(sm_conn);
1392019005a0SMatthias Ringwald             break;
1393019005a0SMatthias Ringwald         }
13940346c37cSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_SALT:
13950346c37cSMatthias Ringwald             memcpy(setup->sm_t, hash, 16);
1396bd57ffebSMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F5_MACKEY;
13970346c37cSMatthias Ringwald             break;
13980346c37cSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_MACKEY:
13990346c37cSMatthias Ringwald             memcpy(setup->sm_mackey, hash, 16);
1400bd57ffebSMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F5_LTK;
14010346c37cSMatthias Ringwald             break;
14020346c37cSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_LTK:
14030346c37cSMatthias Ringwald             memcpy(setup->sm_ltk, hash, 16);
1404bd57ffebSMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F6_FOR_DHKEY_CHECK;
1405019005a0SMatthias Ringwald             break;
1406901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F6_FOR_DHKEY_CHECK:
1407901c000fSMatthias Ringwald             memcpy(setup->sm_local_dhkey_check, hash, 16);
1408901c000fSMatthias Ringwald             if (sm_conn->sm_role){
1409901c000fSMatthias Ringwald                 // responder
1410901c000fSMatthias Ringwald                 if (setup->sm_state_vars & SM_STATE_VAR_DHKEY_COMMAND_RECEIVED){
1411901c000fSMatthias Ringwald                     sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F6_TO_VERIFY_DHKEY_CHECK;
1412901c000fSMatthias Ringwald                 } else {
1413901c000fSMatthias Ringwald                     sm_conn->sm_engine_state = SM_SC_W4_DHKEY_CHECK_COMMAND;
1414901c000fSMatthias Ringwald                 }
1415901c000fSMatthias Ringwald             } else {
1416901c000fSMatthias Ringwald                 sm_conn->sm_engine_state = SM_SC_SEND_DHKEY_CHECK_COMMAND;
1417901c000fSMatthias Ringwald             }
1418901c000fSMatthias Ringwald             break;
1419901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F6_TO_VERIFY_DHKEY_CHECK:
1420901c000fSMatthias Ringwald             if (0 != memcmp(hash, setup->sm_peer_dhkey_check, 16) ){
1421901c000fSMatthias Ringwald                 sm_pairing_error(sm_conn, SM_REASON_DHKEY_CHECK_FAILED);
1422aec94140SMatthias Ringwald                 break;
1423aec94140SMatthias Ringwald             }
1424901c000fSMatthias Ringwald             if (sm_conn->sm_role){
1425901c000fSMatthias Ringwald                 // responder
1426901c000fSMatthias Ringwald                 sm_conn->sm_engine_state = SM_SC_SEND_DHKEY_CHECK_COMMAND;
1427901c000fSMatthias Ringwald             } else {
1428901c000fSMatthias Ringwald                 // initiator
1429901c000fSMatthias Ringwald                 sm_conn->sm_engine_state = SM_INITIATOR_PH3_SEND_START_ENCRYPTION;
1430bd57ffebSMatthias Ringwald             }
1431901c000fSMatthias Ringwald             break;
1432bd57ffebSMatthias Ringwald         default:
1433bd57ffebSMatthias Ringwald             log_error("sm_sc_cmac_done in state %u", sm_conn->sm_engine_state);
1434bd57ffebSMatthias Ringwald             break;
1435bd57ffebSMatthias Ringwald     }
1436aec94140SMatthias Ringwald     sm_run();
1437aec94140SMatthias Ringwald }
1438aec94140SMatthias Ringwald 
1439688a08f9SMatthias Ringwald static void f4_engine(sm_connection_t * sm_conn, const sm_key256_t u, const sm_key256_t v, const sm_key_t x, uint8_t z){
1440dc300847SMatthias Ringwald     const uint16_t message_len = 65;
1441aec94140SMatthias Ringwald     sm_cmac_connection = sm_conn;
1442aec94140SMatthias Ringwald     memcpy(sm_cmac_sc_buffer, u, 32);
1443aec94140SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+32, v, 32);
1444aec94140SMatthias Ringwald     sm_cmac_sc_buffer[64] = z;
1445aec94140SMatthias Ringwald     log_info("f4 key");
1446aec94140SMatthias Ringwald     log_info_hexdump(x, 16);
1447aec94140SMatthias Ringwald     log_info("f4 message");
1448dc300847SMatthias Ringwald     log_info_hexdump(sm_cmac_sc_buffer, message_len);
1449dc300847SMatthias Ringwald     sm_cmac_general_start(x, message_len, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
1450aec94140SMatthias Ringwald }
1451aec94140SMatthias Ringwald 
14520346c37cSMatthias Ringwald static const sm_key_t f5_salt = { 0x6C ,0x88, 0x83, 0x91, 0xAA, 0xF5, 0xA5, 0x38, 0x60, 0x37, 0x0B, 0xDB, 0x5A, 0x60, 0x83, 0xBE};
14530346c37cSMatthias Ringwald static const uint8_t f5_key_id[] = { 0x62, 0x74, 0x6c, 0x65 };
14540346c37cSMatthias Ringwald static const uint8_t f5_length[] = { 0x01, 0x00};
14550346c37cSMatthias Ringwald 
14560346c37cSMatthias Ringwald static void sm_sc_calculate_dhkey(sm_key256_t dhkey){
14570346c37cSMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
14580346c37cSMatthias Ringwald     // da * Pb
1459e722521aSMatthias Ringwald     mbedtls_mpi d;
14600346c37cSMatthias Ringwald     mbedtls_ecp_point Q;
1461e722521aSMatthias Ringwald     mbedtls_mpi_init(&d);
14620346c37cSMatthias Ringwald     mbedtls_ecp_point_init(&Q);
1463e722521aSMatthias Ringwald     mbedtls_mpi_read_binary(&d, ec_d, 32);
14640346c37cSMatthias Ringwald     mbedtls_mpi_read_binary(&Q.X, setup->sm_peer_qx, 32);
14650346c37cSMatthias Ringwald     mbedtls_mpi_read_binary(&Q.Y, setup->sm_peer_qy, 32);
14660346c37cSMatthias Ringwald     mbedtls_mpi_read_string(&Q.Z, 16, "1" );
14670346c37cSMatthias Ringwald     mbedtls_ecp_point DH;
14680346c37cSMatthias Ringwald     mbedtls_ecp_point_init(&DH);
1469e722521aSMatthias Ringwald     mbedtls_ecp_mul(&mbedtls_ec_group, &DH, &d, &Q, NULL, NULL);
14700346c37cSMatthias Ringwald     mbedtls_mpi_write_binary(&DH.X, dhkey, 32);
1471891bb64aSMatthias Ringwald     mbedtls_ecp_point_free(&DH);
1472891bb64aSMatthias Ringwald     mbedtls_ecp_point_free(&Q);
14730346c37cSMatthias Ringwald #endif
14740346c37cSMatthias Ringwald     log_info("dhkey");
14750346c37cSMatthias Ringwald     log_info_hexdump(dhkey, 32);
14760346c37cSMatthias Ringwald }
14770346c37cSMatthias Ringwald 
14780346c37cSMatthias Ringwald static void f5_calculate_salt(sm_connection_t * sm_conn){
14790346c37cSMatthias Ringwald     // calculate DHKEY
14800346c37cSMatthias Ringwald     sm_key256_t dhkey;
14810346c37cSMatthias Ringwald     sm_sc_calculate_dhkey(dhkey);
14820346c37cSMatthias Ringwald 
14830346c37cSMatthias Ringwald     // calculate salt for f5
14840346c37cSMatthias Ringwald     const uint16_t message_len = 32;
14850346c37cSMatthias Ringwald     sm_cmac_connection = sm_conn;
14860346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer, dhkey, message_len);
14870346c37cSMatthias Ringwald     sm_cmac_general_start(f5_salt, message_len, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
14880346c37cSMatthias Ringwald }
14890346c37cSMatthias Ringwald 
14900346c37cSMatthias Ringwald static inline void f5_mackkey(sm_connection_t * sm_conn, sm_key_t t, const sm_key_t n1, const sm_key_t n2, const sm_key56_t a1, const sm_key56_t a2){
14910346c37cSMatthias Ringwald     const uint16_t message_len = 53;
14920346c37cSMatthias Ringwald     sm_cmac_connection = sm_conn;
14930346c37cSMatthias Ringwald 
14940346c37cSMatthias Ringwald     // f5(W, N1, N2, A1, A2) = AES-CMACT (Counter = 0 || keyID || N1 || N2|| A1|| A2 || Length = 256) -- this is the MacKey
14950346c37cSMatthias Ringwald     sm_cmac_sc_buffer[0] = 0;
14960346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+01, f5_key_id, 4);
14970346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+05, n1, 16);
14980346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+21, n2, 16);
14990346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+37, a1, 7);
15000346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+44, a2, 7);
15010346c37cSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+51, f5_length, 2);
15020346c37cSMatthias Ringwald     log_info("f5 key");
15030346c37cSMatthias Ringwald     log_info_hexdump(t, 16);
15040346c37cSMatthias Ringwald     log_info("f5 message for MacKey");
15050346c37cSMatthias Ringwald     log_info_hexdump(sm_cmac_sc_buffer, message_len);
15060346c37cSMatthias Ringwald     sm_cmac_general_start(t, message_len, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
15070346c37cSMatthias Ringwald }
15080346c37cSMatthias Ringwald 
15090346c37cSMatthias Ringwald static void f5_calculate_mackey(sm_connection_t * sm_conn){
15100346c37cSMatthias Ringwald     sm_key56_t bd_addr_master, bd_addr_slave;
15110346c37cSMatthias Ringwald     bd_addr_master[0] =  setup->sm_m_addr_type;
15120346c37cSMatthias Ringwald     bd_addr_slave[0]  =  setup->sm_s_addr_type;
15130346c37cSMatthias Ringwald     memcpy(&bd_addr_master[1], setup->sm_m_address, 6);
15140346c37cSMatthias Ringwald     memcpy(&bd_addr_slave[1],  setup->sm_s_address, 6);
15150346c37cSMatthias Ringwald     if (sm_conn->sm_role){
15160346c37cSMatthias Ringwald         // responder
15170346c37cSMatthias Ringwald         f5_mackkey(sm_conn, setup->sm_t, setup->sm_peer_nonce, setup->sm_local_nonce, bd_addr_master, bd_addr_slave);
15180346c37cSMatthias Ringwald     } else {
15190346c37cSMatthias Ringwald         // initiator
15200346c37cSMatthias Ringwald         f5_mackkey(sm_conn, setup->sm_t, setup->sm_local_nonce, setup->sm_peer_nonce, bd_addr_master, bd_addr_slave);
15210346c37cSMatthias Ringwald     }
15220346c37cSMatthias Ringwald }
15230346c37cSMatthias Ringwald 
15240346c37cSMatthias Ringwald // note: must be called right after f5_mackey, as sm_cmac_buffer[1..52] will be reused
15250346c37cSMatthias Ringwald static inline void f5_ltk(sm_connection_t * sm_conn, sm_key_t t){
15260346c37cSMatthias Ringwald     const uint16_t message_len = 53;
15270346c37cSMatthias Ringwald     sm_cmac_connection = sm_conn;
15280346c37cSMatthias Ringwald     sm_cmac_sc_buffer[0] = 1;
15290346c37cSMatthias Ringwald     // 1..52 setup before
15300346c37cSMatthias Ringwald     log_info("f5 key");
15310346c37cSMatthias Ringwald     log_info_hexdump(t, 16);
15320346c37cSMatthias Ringwald     log_info("f5 message for LTK");
15330346c37cSMatthias Ringwald     log_info_hexdump(sm_cmac_sc_buffer, message_len);
15340346c37cSMatthias Ringwald     sm_cmac_general_start(t, message_len, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
15350346c37cSMatthias Ringwald }
1536f92edc8eSMatthias Ringwald 
15370346c37cSMatthias Ringwald static void f5_calculate_ltk(sm_connection_t * sm_conn){
15380346c37cSMatthias Ringwald     f5_ltk(sm_conn, setup->sm_t);
15390346c37cSMatthias Ringwald }
15400346c37cSMatthias Ringwald 
154127163002SMatthias Ringwald static void f6_engine(sm_connection_t * sm_conn, const sm_key_t w, const sm_key_t n1, const sm_key_t n2, const sm_key_t r, const sm_key24_t io_cap, const sm_key56_t a1, const sm_key56_t a2){
1542dc300847SMatthias Ringwald     const uint16_t message_len = 65;
154327163002SMatthias Ringwald     sm_cmac_connection = sm_conn;
1544dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer, n1, 16);
1545dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+16, n2, 16);
1546dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+32, r, 16);
1547dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+48, io_cap, 3);
1548dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+51, a1, 7);
1549dc300847SMatthias Ringwald     memcpy(sm_cmac_sc_buffer+58, a2, 7);
1550dc300847SMatthias Ringwald     log_info("f6 key");
1551dc300847SMatthias Ringwald     log_info_hexdump(w, 16);
1552dc300847SMatthias Ringwald     log_info("f6 message");
1553dc300847SMatthias Ringwald     log_info_hexdump(sm_cmac_sc_buffer, message_len);
1554dc300847SMatthias Ringwald     sm_cmac_general_start(w, 65, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
1555dc300847SMatthias Ringwald }
1556dc300847SMatthias Ringwald 
1557f92edc8eSMatthias Ringwald // g2(U, V, X, Y) = AES-CMACX(U || V || Y) mod 2^32
1558f92edc8eSMatthias Ringwald // - U is 256 bits
1559f92edc8eSMatthias Ringwald // - V is 256 bits
1560f92edc8eSMatthias Ringwald // - X is 128 bits
1561f92edc8eSMatthias Ringwald // - Y is 128 bits
1562bd57ffebSMatthias Ringwald static void g2_engine(sm_connection_t * sm_conn, const sm_key256_t u, const sm_key256_t v, const sm_key_t x, const sm_key_t y){
1563bd57ffebSMatthias Ringwald     const uint16_t message_len = 80;
1564bd57ffebSMatthias Ringwald     sm_cmac_connection = sm_conn;
1565bd57ffebSMatthias Ringwald     memcpy(sm_cmac_sc_buffer, u, 32);
1566bd57ffebSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+32, v, 32);
1567bd57ffebSMatthias Ringwald     memcpy(sm_cmac_sc_buffer+64, y, 16);
1568f92edc8eSMatthias Ringwald     log_info("g2 key");
1569f92edc8eSMatthias Ringwald     log_info_hexdump(x, 16);
1570f92edc8eSMatthias Ringwald     log_info("g2 message");
1571bd57ffebSMatthias Ringwald     log_info_hexdump(sm_cmac_sc_buffer, sizeof(sm_cmac_sc_buffer));
1572bd57ffebSMatthias Ringwald     sm_cmac_general_start(x, message_len, &sm_sc_cmac_get_byte, &sm_sc_cmac_done);
1573f92edc8eSMatthias Ringwald }
1574f92edc8eSMatthias Ringwald 
1575b35a3de2SMatthias Ringwald static void g2_calculate(sm_connection_t * sm_conn) {
1576f92edc8eSMatthias Ringwald     // calc Va if numeric comparison
1577f92edc8eSMatthias Ringwald     if (sm_conn->sm_role){
1578f92edc8eSMatthias Ringwald         // responder
157905299751SMatthias Ringwald         g2_engine(sm_conn, setup->sm_peer_qx, ec_qx, setup->sm_peer_nonce, setup->sm_local_nonce);;
1580f92edc8eSMatthias Ringwald     } else {
1581f92edc8eSMatthias Ringwald         // initiator
158205299751SMatthias Ringwald         g2_engine(sm_conn, ec_qx, setup->sm_peer_qx, setup->sm_local_nonce, setup->sm_peer_nonce);
1583f92edc8eSMatthias Ringwald     }
1584f92edc8eSMatthias Ringwald }
1585f92edc8eSMatthias Ringwald 
1586945888f5SMatthias Ringwald static void sm_sc_calculate_local_confirm(sm_connection_t * sm_conn){
15879af0f475SMatthias Ringwald     uint8_t z = 0;
15889af0f475SMatthias Ringwald     if (setup->sm_stk_generation_method != JUST_WORKS && setup->sm_stk_generation_method != NK_BOTH_INPUT){
15899af0f475SMatthias Ringwald         // some form of passkey
15909af0f475SMatthias Ringwald         uint32_t pk = big_endian_read_32(setup->sm_tk, 12);
15919af0f475SMatthias Ringwald         z = 0x80 | ((pk >> setup->sm_passkey_bit) & 1);
15929af0f475SMatthias Ringwald         setup->sm_passkey_bit++;
15939af0f475SMatthias Ringwald     }
159405299751SMatthias Ringwald     f4_engine(sm_conn, ec_qx, setup->sm_peer_qx, setup->sm_local_nonce, z);
15959af0f475SMatthias Ringwald }
1596688a08f9SMatthias Ringwald 
1597688a08f9SMatthias Ringwald static void sm_sc_calculate_remote_confirm(sm_connection_t * sm_conn){
1598688a08f9SMatthias Ringwald     uint8_t z = 0;
1599688a08f9SMatthias Ringwald     if (setup->sm_stk_generation_method != JUST_WORKS && setup->sm_stk_generation_method != NK_BOTH_INPUT){
1600688a08f9SMatthias Ringwald         // some form of passkey
1601688a08f9SMatthias Ringwald         uint32_t pk = big_endian_read_32(setup->sm_tk, 12);
1602688a08f9SMatthias Ringwald         // sm_passkey_bit was increased before sending confirm value
1603688a08f9SMatthias Ringwald         z = 0x80 | ((pk >> (setup->sm_passkey_bit-1)) & 1);
1604688a08f9SMatthias Ringwald     }
160505299751SMatthias Ringwald     f4_engine(sm_conn, setup->sm_peer_qx, ec_qx, setup->sm_peer_nonce, z);
1606688a08f9SMatthias Ringwald }
1607688a08f9SMatthias Ringwald 
16080346c37cSMatthias Ringwald static void sm_sc_prepare_dhkey_check(sm_connection_t * sm_conn){
16090346c37cSMatthias Ringwald     sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F5_SALT;
1610dc300847SMatthias Ringwald }
1611dc300847SMatthias Ringwald 
1612dc300847SMatthias Ringwald static void sm_sc_calculate_f6_for_dhkey_check(sm_connection_t * sm_conn){
1613dc300847SMatthias Ringwald     // calculate DHKCheck
1614dc300847SMatthias Ringwald     sm_key56_t bd_addr_master, bd_addr_slave;
1615dc300847SMatthias Ringwald     bd_addr_master[0] =  setup->sm_m_addr_type;
1616dc300847SMatthias Ringwald     bd_addr_slave[0]  =  setup->sm_s_addr_type;
1617dc300847SMatthias Ringwald     memcpy(&bd_addr_master[1], setup->sm_m_address, 6);
1618dc300847SMatthias Ringwald     memcpy(&bd_addr_slave[1],  setup->sm_s_address, 6);
1619dc300847SMatthias Ringwald     uint8_t iocap_a[3];
1620dc300847SMatthias Ringwald     iocap_a[0] = sm_pairing_packet_get_auth_req(setup->sm_m_preq);
1621dc300847SMatthias Ringwald     iocap_a[1] = sm_pairing_packet_get_oob_data_flag(setup->sm_m_preq);
1622dc300847SMatthias Ringwald     iocap_a[2] = sm_pairing_packet_get_io_capability(setup->sm_m_preq);
1623dc300847SMatthias Ringwald     uint8_t iocap_b[3];
1624dc300847SMatthias Ringwald     iocap_b[0] = sm_pairing_packet_get_auth_req(setup->sm_s_pres);
1625dc300847SMatthias Ringwald     iocap_b[1] = sm_pairing_packet_get_oob_data_flag(setup->sm_s_pres);
1626dc300847SMatthias Ringwald     iocap_b[2] = sm_pairing_packet_get_io_capability(setup->sm_s_pres);
1627dc300847SMatthias Ringwald     if (sm_conn->sm_role){
1628dc300847SMatthias Ringwald         // responder
162927163002SMatthias Ringwald         f6_engine(sm_conn, setup->sm_mackey, setup->sm_local_nonce, setup->sm_peer_nonce, setup->sm_ra, iocap_b, bd_addr_slave, bd_addr_master);
1630dc300847SMatthias Ringwald     } else {
1631dc300847SMatthias Ringwald         // initiator
163227163002SMatthias Ringwald         f6_engine(sm_conn, setup->sm_mackey, setup->sm_local_nonce, setup->sm_peer_nonce, setup->sm_rb, iocap_a, bd_addr_master, bd_addr_slave);
1633dc300847SMatthias Ringwald     }
1634dc300847SMatthias Ringwald }
1635dc300847SMatthias Ringwald 
1636019005a0SMatthias Ringwald static void sm_sc_calculate_f6_to_verify_dhkey_check(sm_connection_t * sm_conn){
1637019005a0SMatthias Ringwald     // validate E = f6()
1638019005a0SMatthias Ringwald     sm_key56_t bd_addr_master, bd_addr_slave;
1639019005a0SMatthias Ringwald     bd_addr_master[0] =  setup->sm_m_addr_type;
1640019005a0SMatthias Ringwald     bd_addr_slave[0]  =  setup->sm_s_addr_type;
1641019005a0SMatthias Ringwald     memcpy(&bd_addr_master[1], setup->sm_m_address, 6);
1642019005a0SMatthias Ringwald     memcpy(&bd_addr_slave[1],  setup->sm_s_address, 6);
1643019005a0SMatthias Ringwald 
1644019005a0SMatthias Ringwald     uint8_t iocap_a[3];
1645019005a0SMatthias Ringwald     iocap_a[0] = sm_pairing_packet_get_auth_req(setup->sm_m_preq);
1646019005a0SMatthias Ringwald     iocap_a[1] = sm_pairing_packet_get_oob_data_flag(setup->sm_m_preq);
1647019005a0SMatthias Ringwald     iocap_a[2] = sm_pairing_packet_get_io_capability(setup->sm_m_preq);
1648019005a0SMatthias Ringwald     uint8_t iocap_b[3];
1649019005a0SMatthias Ringwald     iocap_b[0] = sm_pairing_packet_get_auth_req(setup->sm_s_pres);
1650019005a0SMatthias Ringwald     iocap_b[1] = sm_pairing_packet_get_oob_data_flag(setup->sm_s_pres);
1651019005a0SMatthias Ringwald     iocap_b[2] = sm_pairing_packet_get_io_capability(setup->sm_s_pres);
1652019005a0SMatthias Ringwald     if (sm_conn->sm_role){
1653019005a0SMatthias Ringwald         // responder
1654019005a0SMatthias Ringwald         f6_engine(sm_conn, setup->sm_mackey, setup->sm_peer_nonce, setup->sm_local_nonce, setup->sm_rb, iocap_a, bd_addr_master, bd_addr_slave);
1655019005a0SMatthias Ringwald     } else {
1656019005a0SMatthias Ringwald         // initiator
1657019005a0SMatthias Ringwald         f6_engine(sm_conn, setup->sm_mackey, setup->sm_peer_nonce, setup->sm_local_nonce, setup->sm_ra, iocap_b, bd_addr_slave, bd_addr_master);
1658019005a0SMatthias Ringwald     }
1659019005a0SMatthias Ringwald }
16609af0f475SMatthias Ringwald #endif
16619af0f475SMatthias Ringwald 
16625829ebe2SMatthias Ringwald static void sm_load_security_info(sm_connection_t * sm_connection){
16635829ebe2SMatthias Ringwald     int encryption_key_size;
16645829ebe2SMatthias Ringwald     int authenticated;
16655829ebe2SMatthias Ringwald     int authorized;
16665829ebe2SMatthias Ringwald 
16675829ebe2SMatthias Ringwald     // fetch data from device db - incl. authenticated/authorized/key size. Note all sm_connection_X require encryption enabled
16685829ebe2SMatthias Ringwald     le_device_db_encryption_get(sm_connection->sm_le_db_index, &setup->sm_peer_ediv, setup->sm_peer_rand, setup->sm_peer_ltk,
16695829ebe2SMatthias Ringwald                                 &encryption_key_size, &authenticated, &authorized);
16705829ebe2SMatthias Ringwald     log_info("db index %u, key size %u, authenticated %u, authorized %u", sm_connection->sm_le_db_index, encryption_key_size, authenticated, authorized);
16715829ebe2SMatthias Ringwald     sm_connection->sm_actual_encryption_key_size = encryption_key_size;
16725829ebe2SMatthias Ringwald     sm_connection->sm_connection_authenticated = authenticated;
16735829ebe2SMatthias Ringwald     sm_connection->sm_connection_authorization_state = authorized ? AUTHORIZATION_GRANTED : AUTHORIZATION_UNKNOWN;
16745829ebe2SMatthias Ringwald }
1675bd57ffebSMatthias Ringwald 
16763deb3ec6SMatthias Ringwald static void sm_run(void){
16773deb3ec6SMatthias Ringwald 
1678665d90f2SMatthias Ringwald     btstack_linked_list_iterator_t it;
16793deb3ec6SMatthias Ringwald 
16803deb3ec6SMatthias Ringwald     // assert that we can send at least commands
16813deb3ec6SMatthias Ringwald     if (!hci_can_send_command_packet_now()) return;
16823deb3ec6SMatthias Ringwald 
16833deb3ec6SMatthias Ringwald     //
16843deb3ec6SMatthias Ringwald     // non-connection related behaviour
16853deb3ec6SMatthias Ringwald     //
16863deb3ec6SMatthias Ringwald 
16873deb3ec6SMatthias Ringwald     // distributed key generation
16883deb3ec6SMatthias Ringwald     switch (dkg_state){
16893deb3ec6SMatthias Ringwald         case DKG_CALC_IRK:
16903deb3ec6SMatthias Ringwald             // already busy?
16913deb3ec6SMatthias Ringwald             if (sm_aes128_state == SM_AES128_IDLE) {
16923deb3ec6SMatthias Ringwald                 // IRK = d1(IR, 1, 0)
16933deb3ec6SMatthias Ringwald                 sm_key_t d1_prime;
16943deb3ec6SMatthias Ringwald                 sm_d1_d_prime(1, 0, d1_prime);  // plaintext
16953deb3ec6SMatthias Ringwald                 dkg_next_state();
16963deb3ec6SMatthias Ringwald                 sm_aes128_start(sm_persistent_ir, d1_prime, NULL);
16973deb3ec6SMatthias Ringwald                 return;
16983deb3ec6SMatthias Ringwald             }
16993deb3ec6SMatthias Ringwald             break;
17003deb3ec6SMatthias Ringwald         case DKG_CALC_DHK:
17013deb3ec6SMatthias Ringwald             // already busy?
17023deb3ec6SMatthias Ringwald             if (sm_aes128_state == SM_AES128_IDLE) {
17033deb3ec6SMatthias Ringwald                 // DHK = d1(IR, 3, 0)
17043deb3ec6SMatthias Ringwald                 sm_key_t d1_prime;
17053deb3ec6SMatthias Ringwald                 sm_d1_d_prime(3, 0, d1_prime);  // plaintext
17063deb3ec6SMatthias Ringwald                 dkg_next_state();
17073deb3ec6SMatthias Ringwald                 sm_aes128_start(sm_persistent_ir, d1_prime, NULL);
17083deb3ec6SMatthias Ringwald                 return;
17093deb3ec6SMatthias Ringwald             }
17103deb3ec6SMatthias Ringwald             break;
17113deb3ec6SMatthias Ringwald         default:
17123deb3ec6SMatthias Ringwald             break;
17133deb3ec6SMatthias Ringwald     }
17143deb3ec6SMatthias Ringwald 
17157df18c15SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
17167df18c15SMatthias Ringwald     if (ec_key_generation_state == EC_KEY_GENERATION_ACTIVE){
17177df18c15SMatthias Ringwald         sm_random_start(NULL);
17187df18c15SMatthias Ringwald         return;
17197df18c15SMatthias Ringwald     }
17207df18c15SMatthias Ringwald #endif
17217df18c15SMatthias Ringwald 
17223deb3ec6SMatthias Ringwald     // random address updates
17233deb3ec6SMatthias Ringwald     switch (rau_state){
17243deb3ec6SMatthias Ringwald         case RAU_GET_RANDOM:
17253deb3ec6SMatthias Ringwald             rau_next_state();
17263deb3ec6SMatthias Ringwald             sm_random_start(NULL);
17273deb3ec6SMatthias Ringwald             return;
17283deb3ec6SMatthias Ringwald         case RAU_GET_ENC:
17293deb3ec6SMatthias Ringwald             // already busy?
17303deb3ec6SMatthias Ringwald             if (sm_aes128_state == SM_AES128_IDLE) {
17313deb3ec6SMatthias Ringwald                 sm_key_t r_prime;
17323deb3ec6SMatthias Ringwald                 sm_ah_r_prime(sm_random_address, r_prime);
17333deb3ec6SMatthias Ringwald                 rau_next_state();
17343deb3ec6SMatthias Ringwald                 sm_aes128_start(sm_persistent_irk, r_prime, NULL);
17353deb3ec6SMatthias Ringwald                 return;
17363deb3ec6SMatthias Ringwald             }
17373deb3ec6SMatthias Ringwald             break;
17383deb3ec6SMatthias Ringwald         case RAU_SET_ADDRESS:
17393deb3ec6SMatthias Ringwald             log_info("New random address: %s", bd_addr_to_str(sm_random_address));
17403deb3ec6SMatthias Ringwald             rau_state = RAU_IDLE;
17413deb3ec6SMatthias Ringwald             hci_send_cmd(&hci_le_set_random_address, sm_random_address);
17423deb3ec6SMatthias Ringwald             return;
17433deb3ec6SMatthias Ringwald         default:
17443deb3ec6SMatthias Ringwald             break;
17453deb3ec6SMatthias Ringwald     }
17463deb3ec6SMatthias Ringwald 
17473deb3ec6SMatthias Ringwald     // CMAC
17483deb3ec6SMatthias Ringwald     switch (sm_cmac_state){
17493deb3ec6SMatthias Ringwald         case CMAC_CALC_SUBKEYS:
17503deb3ec6SMatthias Ringwald         case CMAC_CALC_MI:
17513deb3ec6SMatthias Ringwald         case CMAC_CALC_MLAST:
17523deb3ec6SMatthias Ringwald             // already busy?
17533deb3ec6SMatthias Ringwald             if (sm_aes128_state == SM_AES128_ACTIVE) break;
17543deb3ec6SMatthias Ringwald             sm_cmac_handle_aes_engine_ready();
17553deb3ec6SMatthias Ringwald             return;
17563deb3ec6SMatthias Ringwald         default:
17573deb3ec6SMatthias Ringwald             break;
17583deb3ec6SMatthias Ringwald     }
17593deb3ec6SMatthias Ringwald 
17603deb3ec6SMatthias Ringwald     // CSRK Lookup
17613deb3ec6SMatthias Ringwald     // -- if csrk lookup ready, find connection that require csrk lookup
17623deb3ec6SMatthias Ringwald     if (sm_address_resolution_idle()){
17633deb3ec6SMatthias Ringwald         hci_connections_get_iterator(&it);
1764665d90f2SMatthias Ringwald         while(btstack_linked_list_iterator_has_next(&it)){
1765665d90f2SMatthias Ringwald             hci_connection_t * hci_connection = (hci_connection_t *) btstack_linked_list_iterator_next(&it);
17663deb3ec6SMatthias Ringwald             sm_connection_t  * sm_connection  = &hci_connection->sm_connection;
17673deb3ec6SMatthias Ringwald             if (sm_connection->sm_irk_lookup_state == IRK_LOOKUP_W4_READY){
17683deb3ec6SMatthias Ringwald                 // and start lookup
17693deb3ec6SMatthias Ringwald                 sm_address_resolution_start_lookup(sm_connection->sm_peer_addr_type, sm_connection->sm_handle, sm_connection->sm_peer_address, ADDRESS_RESOLUTION_FOR_CONNECTION, sm_connection);
17703deb3ec6SMatthias Ringwald                 sm_connection->sm_irk_lookup_state = IRK_LOOKUP_STARTED;
17713deb3ec6SMatthias Ringwald                 break;
17723deb3ec6SMatthias Ringwald             }
17733deb3ec6SMatthias Ringwald         }
17743deb3ec6SMatthias Ringwald     }
17753deb3ec6SMatthias Ringwald 
17763deb3ec6SMatthias Ringwald     // -- if csrk lookup ready, resolved addresses for received addresses
17773deb3ec6SMatthias Ringwald     if (sm_address_resolution_idle()) {
1778665d90f2SMatthias Ringwald         if (!btstack_linked_list_empty(&sm_address_resolution_general_queue)){
17793deb3ec6SMatthias Ringwald             sm_lookup_entry_t * entry = (sm_lookup_entry_t *) sm_address_resolution_general_queue;
1780665d90f2SMatthias Ringwald             btstack_linked_list_remove(&sm_address_resolution_general_queue, (btstack_linked_item_t *) entry);
17813deb3ec6SMatthias Ringwald             sm_address_resolution_start_lookup(entry->address_type, 0, entry->address, ADDRESS_RESOLUTION_GENERAL, NULL);
17823deb3ec6SMatthias Ringwald             btstack_memory_sm_lookup_entry_free(entry);
17833deb3ec6SMatthias Ringwald         }
17843deb3ec6SMatthias Ringwald     }
17853deb3ec6SMatthias Ringwald 
17863deb3ec6SMatthias Ringwald     // -- Continue with CSRK device lookup by public or resolvable private address
17873deb3ec6SMatthias Ringwald     if (!sm_address_resolution_idle()){
17883deb3ec6SMatthias Ringwald         log_info("LE Device Lookup: device %u/%u", sm_address_resolution_test, le_device_db_count());
17893deb3ec6SMatthias Ringwald         while (sm_address_resolution_test < le_device_db_count()){
17903deb3ec6SMatthias Ringwald             int addr_type;
17913deb3ec6SMatthias Ringwald             bd_addr_t addr;
17923deb3ec6SMatthias Ringwald             sm_key_t irk;
17933deb3ec6SMatthias Ringwald             le_device_db_info(sm_address_resolution_test, &addr_type, addr, irk);
17943deb3ec6SMatthias Ringwald             log_info("device type %u, addr: %s", addr_type, bd_addr_to_str(addr));
17953deb3ec6SMatthias Ringwald 
17963deb3ec6SMatthias Ringwald             if (sm_address_resolution_addr_type == addr_type && memcmp(addr, sm_address_resolution_address, 6) == 0){
17973deb3ec6SMatthias Ringwald                 log_info("LE Device Lookup: found CSRK by { addr_type, address} ");
17983deb3ec6SMatthias Ringwald                 sm_address_resolution_handle_event(ADDRESS_RESOLUTION_SUCEEDED);
17993deb3ec6SMatthias Ringwald                 break;
18003deb3ec6SMatthias Ringwald             }
18013deb3ec6SMatthias Ringwald 
18023deb3ec6SMatthias Ringwald             if (sm_address_resolution_addr_type == 0){
18033deb3ec6SMatthias Ringwald                 sm_address_resolution_test++;
18043deb3ec6SMatthias Ringwald                 continue;
18053deb3ec6SMatthias Ringwald             }
18063deb3ec6SMatthias Ringwald 
18073deb3ec6SMatthias Ringwald             if (sm_aes128_state == SM_AES128_ACTIVE) break;
18083deb3ec6SMatthias Ringwald 
18093deb3ec6SMatthias Ringwald             log_info("LE Device Lookup: calculate AH");
18108314c363SMatthias Ringwald             log_info_key("IRK", irk);
18113deb3ec6SMatthias Ringwald 
18123deb3ec6SMatthias Ringwald             sm_key_t r_prime;
18133deb3ec6SMatthias Ringwald             sm_ah_r_prime(sm_address_resolution_address, r_prime);
18143deb3ec6SMatthias Ringwald             sm_address_resolution_ah_calculation_active = 1;
18153deb3ec6SMatthias Ringwald             sm_aes128_start(irk, r_prime, sm_address_resolution_context);   // keep context
18163deb3ec6SMatthias Ringwald             return;
18173deb3ec6SMatthias Ringwald         }
18183deb3ec6SMatthias Ringwald 
18193deb3ec6SMatthias Ringwald         if (sm_address_resolution_test >= le_device_db_count()){
18203deb3ec6SMatthias Ringwald             log_info("LE Device Lookup: not found");
18213deb3ec6SMatthias Ringwald             sm_address_resolution_handle_event(ADDRESS_RESOLUTION_FAILED);
18223deb3ec6SMatthias Ringwald         }
18233deb3ec6SMatthias Ringwald     }
18243deb3ec6SMatthias Ringwald 
18253deb3ec6SMatthias Ringwald 
18263deb3ec6SMatthias Ringwald     //
18273deb3ec6SMatthias Ringwald     // active connection handling
18283deb3ec6SMatthias Ringwald     // -- use loop to handle next connection if lock on setup context is released
18293deb3ec6SMatthias Ringwald 
18303deb3ec6SMatthias Ringwald     while (1) {
18313deb3ec6SMatthias Ringwald 
18323deb3ec6SMatthias Ringwald         // Find connections that requires setup context and make active if no other is locked
18333deb3ec6SMatthias Ringwald         hci_connections_get_iterator(&it);
1834665d90f2SMatthias Ringwald         while(!sm_active_connection && btstack_linked_list_iterator_has_next(&it)){
1835665d90f2SMatthias Ringwald             hci_connection_t * hci_connection = (hci_connection_t *) btstack_linked_list_iterator_next(&it);
18363deb3ec6SMatthias Ringwald             sm_connection_t  * sm_connection = &hci_connection->sm_connection;
18373deb3ec6SMatthias Ringwald             // - if no connection locked and we're ready/waiting for setup context, fetch it and start
18383deb3ec6SMatthias Ringwald             int done = 1;
18393deb3ec6SMatthias Ringwald             int err;
18403deb3ec6SMatthias Ringwald             switch (sm_connection->sm_engine_state) {
18413deb3ec6SMatthias Ringwald                 case SM_RESPONDER_SEND_SECURITY_REQUEST:
18423deb3ec6SMatthias Ringwald                     // send packet if possible,
1843adbe29e8SMatthias Ringwald                     if (l2cap_can_send_fixed_channel_packet_now(sm_connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL)){
1844b170b20fSMatthias Ringwald                         const uint8_t buffer[2] = { SM_CODE_SECURITY_REQUEST, SM_AUTHREQ_BONDING};
18453deb3ec6SMatthias Ringwald                         sm_connection->sm_engine_state = SM_RESPONDER_PH1_W4_PAIRING_REQUEST;
18463deb3ec6SMatthias Ringwald                         l2cap_send_connectionless(sm_connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
1847b170b20fSMatthias Ringwald                     } else {
1848b170b20fSMatthias Ringwald                         l2cap_request_can_send_fix_channel_now_event(sm_connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL);
18493deb3ec6SMatthias Ringwald                     }
18505829ebe2SMatthias Ringwald                     // don't lock sxetup context yet
18513deb3ec6SMatthias Ringwald                     done = 0;
18523deb3ec6SMatthias Ringwald                     break;
18533deb3ec6SMatthias Ringwald                 case SM_RESPONDER_PH1_PAIRING_REQUEST_RECEIVED:
18543deb3ec6SMatthias Ringwald                     sm_init_setup(sm_connection);
18553deb3ec6SMatthias Ringwald                     // recover pairing request
18563deb3ec6SMatthias Ringwald                     memcpy(&setup->sm_m_preq, &sm_connection->sm_m_preq, sizeof(sm_pairing_packet_t));
18573deb3ec6SMatthias Ringwald                     err = sm_stk_generation_init(sm_connection);
18583deb3ec6SMatthias Ringwald                     if (err){
18593deb3ec6SMatthias Ringwald                         setup->sm_pairing_failed_reason = err;
18603deb3ec6SMatthias Ringwald                         sm_connection->sm_engine_state = SM_GENERAL_SEND_PAIRING_FAILED;
18613deb3ec6SMatthias Ringwald                         break;
18623deb3ec6SMatthias Ringwald                     }
18633deb3ec6SMatthias Ringwald                     sm_timeout_start(sm_connection);
18643deb3ec6SMatthias Ringwald                     // generate random number first, if we need to show passkey
18653deb3ec6SMatthias Ringwald                     if (setup->sm_stk_generation_method == PK_INIT_INPUT){
18663deb3ec6SMatthias Ringwald                         sm_connection->sm_engine_state = SM_PH2_GET_RANDOM_TK;
18673deb3ec6SMatthias Ringwald                         break;
18683deb3ec6SMatthias Ringwald                     }
18693deb3ec6SMatthias Ringwald                     sm_connection->sm_engine_state = SM_RESPONDER_PH1_SEND_PAIRING_RESPONSE;
18703deb3ec6SMatthias Ringwald                     break;
18713deb3ec6SMatthias Ringwald                 case SM_INITIATOR_PH0_HAS_LTK:
18725829ebe2SMatthias Ringwald                     sm_load_security_info(sm_connection);
18733deb3ec6SMatthias Ringwald                     sm_connection->sm_engine_state = SM_INITIATOR_PH0_SEND_START_ENCRYPTION;
18743deb3ec6SMatthias Ringwald                     break;
18753deb3ec6SMatthias Ringwald                 case SM_RESPONDER_PH0_RECEIVED_LTK:
18765829ebe2SMatthias Ringwald                     switch (sm_connection->sm_irk_lookup_state){
18775829ebe2SMatthias Ringwald                         case IRK_LOOKUP_SUCCEEDED:{
18785829ebe2SMatthias Ringwald                                 sm_load_security_info(sm_connection);
18795829ebe2SMatthias Ringwald                                 sm_connection->sm_engine_state = SM_RESPONDER_PH2_SEND_LTK_REPLY;
18805829ebe2SMatthias Ringwald                                 break;
18815829ebe2SMatthias Ringwald                             }
18825829ebe2SMatthias Ringwald                         case IRK_LOOKUP_FAILED:
18835829ebe2SMatthias Ringwald                             // assume that we don't have a LTK for ediv == 0 and random == null
18845829ebe2SMatthias Ringwald                             if (sm_connection->sm_local_ediv == 0 && sm_is_null_random(sm_connection->sm_local_rand)){
18855829ebe2SMatthias Ringwald                                 log_info("LTK Request: ediv & random are empty");
18865829ebe2SMatthias Ringwald                                 sm_connection->sm_engine_state = SM_RESPONDER_PH0_SEND_LTK_REQUESTED_NEGATIVE_REPLY;
18875829ebe2SMatthias Ringwald                                 // TODO: no need to lock context yet -> done = 0;
18885829ebe2SMatthias Ringwald                                 break;
18895829ebe2SMatthias Ringwald                             }
18903deb3ec6SMatthias Ringwald                             // re-establish previously used LTK using Rand and EDIV
18913deb3ec6SMatthias Ringwald                             memcpy(setup->sm_local_rand, sm_connection->sm_local_rand, 8);
18923deb3ec6SMatthias Ringwald                             setup->sm_local_ediv = sm_connection->sm_local_ediv;
18933deb3ec6SMatthias Ringwald                             // re-establish used key encryption size
18943deb3ec6SMatthias Ringwald                             // no db for encryption size hack: encryption size is stored in lowest nibble of setup->sm_local_rand
18953deb3ec6SMatthias Ringwald                             sm_connection->sm_actual_encryption_key_size = (setup->sm_local_rand[7] & 0x0f) + 1;
18963deb3ec6SMatthias Ringwald                             // no db for authenticated flag hack: flag is stored in bit 4 of LSB
18973deb3ec6SMatthias Ringwald                             sm_connection->sm_connection_authenticated = (setup->sm_local_rand[7] & 0x10) >> 4;
18983deb3ec6SMatthias Ringwald                             log_info("sm: received ltk request with key size %u, authenticated %u",
18993deb3ec6SMatthias Ringwald                                     sm_connection->sm_actual_encryption_key_size, sm_connection->sm_connection_authenticated);
19003deb3ec6SMatthias Ringwald                             sm_connection->sm_engine_state = SM_RESPONDER_PH4_Y_GET_ENC;
19013deb3ec6SMatthias Ringwald                             break;
19025829ebe2SMatthias Ringwald                         default:
19035829ebe2SMatthias Ringwald                             // just wait until IRK lookup is completed
19045829ebe2SMatthias Ringwald                             // don't lock sxetup context yet
19055829ebe2SMatthias Ringwald                             done = 0;
19065829ebe2SMatthias Ringwald                             break;
19075829ebe2SMatthias Ringwald                     }
19085829ebe2SMatthias Ringwald                     break;
19093deb3ec6SMatthias Ringwald                 case SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST:
19103deb3ec6SMatthias Ringwald                     sm_init_setup(sm_connection);
19113deb3ec6SMatthias Ringwald                     sm_timeout_start(sm_connection);
19123deb3ec6SMatthias Ringwald                     sm_connection->sm_engine_state = SM_INITIATOR_PH1_SEND_PAIRING_REQUEST;
19133deb3ec6SMatthias Ringwald                     break;
19143deb3ec6SMatthias Ringwald                 default:
19153deb3ec6SMatthias Ringwald                     done = 0;
19163deb3ec6SMatthias Ringwald                     break;
19173deb3ec6SMatthias Ringwald             }
19183deb3ec6SMatthias Ringwald             if (done){
19193deb3ec6SMatthias Ringwald                 sm_active_connection = sm_connection->sm_handle;
19203deb3ec6SMatthias Ringwald                 log_info("sm: connection 0x%04x locked setup context as %s", sm_active_connection, sm_connection->sm_role ? "responder" : "initiator");
19213deb3ec6SMatthias Ringwald             }
19223deb3ec6SMatthias Ringwald         }
19233deb3ec6SMatthias Ringwald 
19243deb3ec6SMatthias Ringwald         //
19253deb3ec6SMatthias Ringwald         // active connection handling
19263deb3ec6SMatthias Ringwald         //
19273deb3ec6SMatthias Ringwald 
19283deb3ec6SMatthias Ringwald         if (sm_active_connection == 0) return;
19293deb3ec6SMatthias Ringwald 
19303deb3ec6SMatthias Ringwald         // assert that we could send a SM PDU - not needed for all of the following
1931b170b20fSMatthias Ringwald         if (!l2cap_can_send_fixed_channel_packet_now(sm_active_connection, L2CAP_CID_SECURITY_MANAGER_PROTOCOL)) {
1932b170b20fSMatthias Ringwald             l2cap_request_can_send_fix_channel_now_event(sm_active_connection, L2CAP_CID_SECURITY_MANAGER_PROTOCOL);
1933b170b20fSMatthias Ringwald             return;
1934b170b20fSMatthias Ringwald         }
19353deb3ec6SMatthias Ringwald 
19363deb3ec6SMatthias Ringwald         sm_connection_t * connection = sm_get_connection_for_handle(sm_active_connection);
19373deb3ec6SMatthias Ringwald         if (!connection) return;
19383deb3ec6SMatthias Ringwald 
19393deb3ec6SMatthias Ringwald         sm_key_t plaintext;
19403deb3ec6SMatthias Ringwald         int key_distribution_flags;
19413deb3ec6SMatthias Ringwald 
19423deb3ec6SMatthias Ringwald         log_info("sm_run: state %u", connection->sm_engine_state);
19433deb3ec6SMatthias Ringwald 
19443deb3ec6SMatthias Ringwald         // responding state
19453deb3ec6SMatthias Ringwald         switch (connection->sm_engine_state){
19463deb3ec6SMatthias Ringwald 
19473deb3ec6SMatthias Ringwald             // general
19483deb3ec6SMatthias Ringwald             case SM_GENERAL_SEND_PAIRING_FAILED: {
19493deb3ec6SMatthias Ringwald                 uint8_t buffer[2];
19503deb3ec6SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_FAILED;
19513deb3ec6SMatthias Ringwald                 buffer[1] = setup->sm_pairing_failed_reason;
19523deb3ec6SMatthias Ringwald                 connection->sm_engine_state = connection->sm_role ? SM_RESPONDER_IDLE : SM_INITIATOR_CONNECTED;
19533deb3ec6SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
19543deb3ec6SMatthias Ringwald                 sm_done_for_handle(connection->sm_handle);
19553deb3ec6SMatthias Ringwald                 break;
19563deb3ec6SMatthias Ringwald             }
19573deb3ec6SMatthias Ringwald 
1958aec94140SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
1959f1c1783eSMatthias Ringwald             case SM_SC_W2_GET_RANDOM_A:
1960f1c1783eSMatthias Ringwald                 sm_random_start(connection);
1961f1c1783eSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_GET_RANDOM_A;
1962f1c1783eSMatthias Ringwald                 break;
1963f1c1783eSMatthias Ringwald             case SM_SC_W2_GET_RANDOM_B:
1964f1c1783eSMatthias Ringwald                 sm_random_start(connection);
1965f1c1783eSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_GET_RANDOM_B;
1966f1c1783eSMatthias Ringwald                 break;
1967aec94140SMatthias Ringwald             case SM_SC_W2_CMAC_FOR_CONFIRMATION:
1968aec94140SMatthias Ringwald                 if (!sm_cmac_ready()) break;
1969aec94140SMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CMAC_FOR_CONFIRMATION;
1970aec94140SMatthias Ringwald                 sm_sc_calculate_local_confirm(connection);
1971aec94140SMatthias Ringwald                 break;
1972688a08f9SMatthias Ringwald             case SM_SC_W2_CMAC_FOR_CHECK_CONFIRMATION:
1973688a08f9SMatthias Ringwald                 if (!sm_cmac_ready()) break;
1974688a08f9SMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CMAC_FOR_CHECK_CONFIRMATION;
1975688a08f9SMatthias Ringwald                 sm_sc_calculate_remote_confirm(connection);
1976688a08f9SMatthias Ringwald                 break;
1977dc300847SMatthias Ringwald             case SM_SC_W2_CALCULATE_F6_FOR_DHKEY_CHECK:
1978dc300847SMatthias Ringwald                 if (!sm_cmac_ready()) break;
1979dc300847SMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_F6_FOR_DHKEY_CHECK;
1980dc300847SMatthias Ringwald                 sm_sc_calculate_f6_for_dhkey_check(connection);
1981dc300847SMatthias Ringwald                 break;
1982019005a0SMatthias Ringwald             case SM_SC_W2_CALCULATE_F6_TO_VERIFY_DHKEY_CHECK:
1983b35a3de2SMatthias Ringwald                 if (!sm_cmac_ready()) break;
1984019005a0SMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_F6_TO_VERIFY_DHKEY_CHECK;
19850346c37cSMatthias Ringwald                 sm_sc_calculate_f6_to_verify_dhkey_check(connection);
19860346c37cSMatthias Ringwald                 break;
19870346c37cSMatthias Ringwald             case SM_SC_W2_CALCULATE_F5_SALT:
1988b35a3de2SMatthias Ringwald                 if (!sm_cmac_ready()) break;
19890346c37cSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_F5_SALT;
19900346c37cSMatthias Ringwald                 f5_calculate_salt(connection);
19910346c37cSMatthias Ringwald                 break;
19920346c37cSMatthias Ringwald             case SM_SC_W2_CALCULATE_F5_MACKEY:
1993b35a3de2SMatthias Ringwald                 if (!sm_cmac_ready()) break;
19940346c37cSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_F5_MACKEY;
19950346c37cSMatthias Ringwald                 f5_calculate_mackey(connection);
19960346c37cSMatthias Ringwald                 break;
19970346c37cSMatthias Ringwald             case SM_SC_W2_CALCULATE_F5_LTK:
1998b35a3de2SMatthias Ringwald                 if (!sm_cmac_ready()) break;
19990346c37cSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_F5_LTK;
20000346c37cSMatthias Ringwald                 f5_calculate_ltk(connection);
2001019005a0SMatthias Ringwald                 break;
2002bd57ffebSMatthias Ringwald             case SM_SC_W2_CALCULATE_G2:
2003b35a3de2SMatthias Ringwald                 if (!sm_cmac_ready()) break;
2004bd57ffebSMatthias Ringwald                 connection->sm_engine_state = SM_SC_W4_CALCULATE_G2;
2005b35a3de2SMatthias Ringwald                 g2_calculate(connection);
2006bd57ffebSMatthias Ringwald                 break;
2007bd57ffebSMatthias Ringwald 
2008aec94140SMatthias Ringwald #endif
20093deb3ec6SMatthias Ringwald             // initiator side
20103deb3ec6SMatthias Ringwald             case SM_INITIATOR_PH0_SEND_START_ENCRYPTION: {
20113deb3ec6SMatthias Ringwald                 sm_key_t peer_ltk_flipped;
20129c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_peer_ltk, peer_ltk_flipped);
20133deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_INITIATOR_PH0_W4_CONNECTION_ENCRYPTED;
20143deb3ec6SMatthias Ringwald                 log_info("sm: hci_le_start_encryption ediv 0x%04x", setup->sm_peer_ediv);
2015c9b8fdd9SMatthias Ringwald                 uint32_t rand_high = big_endian_read_32(setup->sm_peer_rand, 0);
2016c9b8fdd9SMatthias Ringwald                 uint32_t rand_low  = big_endian_read_32(setup->sm_peer_rand, 4);
20173deb3ec6SMatthias Ringwald                 hci_send_cmd(&hci_le_start_encryption, connection->sm_handle,rand_low, rand_high, setup->sm_peer_ediv, peer_ltk_flipped);
20183deb3ec6SMatthias Ringwald                 return;
20193deb3ec6SMatthias Ringwald             }
20203deb3ec6SMatthias Ringwald 
20213deb3ec6SMatthias Ringwald             case SM_INITIATOR_PH1_SEND_PAIRING_REQUEST:
20221ad129beSMatthias Ringwald                 sm_pairing_packet_set_code(setup->sm_m_preq, SM_CODE_PAIRING_REQUEST);
20233deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_INITIATOR_PH1_W4_PAIRING_RESPONSE;
20243deb3ec6SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) &setup->sm_m_preq, sizeof(sm_pairing_packet_t));
20253deb3ec6SMatthias Ringwald                 sm_timeout_reset(connection);
20263deb3ec6SMatthias Ringwald                 break;
20273deb3ec6SMatthias Ringwald 
20283deb3ec6SMatthias Ringwald             // responder side
20293deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH0_SEND_LTK_REQUESTED_NEGATIVE_REPLY:
20303deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_RESPONDER_IDLE;
20313deb3ec6SMatthias Ringwald                 hci_send_cmd(&hci_le_long_term_key_negative_reply, connection->sm_handle);
20327dbe5dcfSMatthias Ringwald                 sm_done_for_handle(connection->sm_handle);
20333deb3ec6SMatthias Ringwald                 return;
20343deb3ec6SMatthias Ringwald 
203527c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
2036c6b7cbd9SMatthias Ringwald             case SM_SC_SEND_PUBLIC_KEY_COMMAND: {
203727c32905SMatthias Ringwald                 uint8_t buffer[65];
203827c32905SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_PUBLIC_KEY;
203927c32905SMatthias Ringwald                 //
204005299751SMatthias Ringwald                 reverse_256(ec_qx, &buffer[1]);
204105299751SMatthias Ringwald                 reverse_256(ec_qy, &buffer[33]);
2042e53be891SMatthias Ringwald 
204345a61d50SMatthias Ringwald                 // stk generation method
204445a61d50SMatthias Ringwald                 // passkey entry: notify app to show passkey or to request passkey
204545a61d50SMatthias Ringwald                 switch (setup->sm_stk_generation_method){
204645a61d50SMatthias Ringwald                     case JUST_WORKS:
204745a61d50SMatthias Ringwald                     case NK_BOTH_INPUT:
204827c32905SMatthias Ringwald                         if (connection->sm_role){
204907036a04SMatthias Ringwald                             // responder
2050b35a3de2SMatthias Ringwald                             sm_sc_start_calculating_local_confirm(connection);
205127c32905SMatthias Ringwald                         } else {
205207036a04SMatthias Ringwald                             // initiator
2053c6b7cbd9SMatthias Ringwald                             connection->sm_engine_state = SM_SC_W4_PUBLIC_KEY_COMMAND;
205427c32905SMatthias Ringwald                         }
205545a61d50SMatthias Ringwald                         break;
205645a61d50SMatthias Ringwald                     case PK_INIT_INPUT:
205745a61d50SMatthias Ringwald                     case PK_RESP_INPUT:
205845a61d50SMatthias Ringwald                     case OK_BOTH_INPUT:
205907036a04SMatthias Ringwald                         // use random TK for display
206045a61d50SMatthias Ringwald                         memcpy(setup->sm_ra, setup->sm_tk, 16);
206145a61d50SMatthias Ringwald                         memcpy(setup->sm_rb, setup->sm_tk, 16);
206245a61d50SMatthias Ringwald                         setup->sm_passkey_bit = 0;
206307036a04SMatthias Ringwald 
206445a61d50SMatthias Ringwald                         if (connection->sm_role){
206545a61d50SMatthias Ringwald                             // responder
2066c6b7cbd9SMatthias Ringwald                             connection->sm_engine_state = SM_SC_W4_CONFIRMATION;
206745a61d50SMatthias Ringwald                         } else {
206845a61d50SMatthias Ringwald                             // initiator
2069c6b7cbd9SMatthias Ringwald                             connection->sm_engine_state = SM_SC_W4_PUBLIC_KEY_COMMAND;
207045a61d50SMatthias Ringwald                         }
207145a61d50SMatthias Ringwald                         sm_trigger_user_response(connection);
207245a61d50SMatthias Ringwald                         break;
207345a61d50SMatthias Ringwald                     case OOB:
207445a61d50SMatthias Ringwald                         // TODO: implement SC OOB
207545a61d50SMatthias Ringwald                         break;
207645a61d50SMatthias Ringwald                 }
207745a61d50SMatthias Ringwald 
207827c32905SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
207927c32905SMatthias Ringwald                 sm_timeout_reset(connection);
208027c32905SMatthias Ringwald                 break;
208127c32905SMatthias Ringwald             }
2082c6b7cbd9SMatthias Ringwald             case SM_SC_SEND_CONFIRMATION: {
2083e53be891SMatthias Ringwald                 uint8_t buffer[17];
2084e53be891SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_CONFIRM;
20859af0f475SMatthias Ringwald                 reverse_128(setup->sm_local_confirm, &buffer[1]);
2086e53be891SMatthias Ringwald                 if (connection->sm_role){
2087c6b7cbd9SMatthias Ringwald                     connection->sm_engine_state = SM_SC_W4_PAIRING_RANDOM;
2088e53be891SMatthias Ringwald                 } else {
2089c6b7cbd9SMatthias Ringwald                     connection->sm_engine_state = SM_SC_W4_CONFIRMATION;
2090e53be891SMatthias Ringwald                 }
2091e53be891SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
2092e53be891SMatthias Ringwald                 sm_timeout_reset(connection);
2093e53be891SMatthias Ringwald                 break;
2094e53be891SMatthias Ringwald             }
2095c6b7cbd9SMatthias Ringwald             case SM_SC_SEND_PAIRING_RANDOM: {
2096e53be891SMatthias Ringwald                 uint8_t buffer[17];
2097e53be891SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_RANDOM;
2098e53be891SMatthias Ringwald                 reverse_128(setup->sm_local_nonce, &buffer[1]);
209945a61d50SMatthias Ringwald                 if (setup->sm_stk_generation_method != JUST_WORKS && setup->sm_stk_generation_method != NK_BOTH_INPUT && setup->sm_passkey_bit < 20){
210045a61d50SMatthias Ringwald                     if (connection->sm_role){
210145a61d50SMatthias Ringwald                         // responder
2102c6b7cbd9SMatthias Ringwald                         connection->sm_engine_state = SM_SC_W4_CONFIRMATION;
210345a61d50SMatthias Ringwald                     } else {
210445a61d50SMatthias Ringwald                         // initiator
2105c6b7cbd9SMatthias Ringwald                         connection->sm_engine_state = SM_SC_W4_PAIRING_RANDOM;
210645a61d50SMatthias Ringwald                     }
210745a61d50SMatthias Ringwald                 } else {
2108e53be891SMatthias Ringwald                     if (connection->sm_role){
2109e53be891SMatthias Ringwald                         // responder
2110446a8c36SMatthias Ringwald                         if (setup->sm_stk_generation_method == NK_BOTH_INPUT){
2111901c000fSMatthias Ringwald                             connection->sm_engine_state = SM_SC_W2_CALCULATE_G2;
21122886623dSMatthias Ringwald                         } else {
21132886623dSMatthias Ringwald                             sm_sc_prepare_dhkey_check(connection);
2114446a8c36SMatthias Ringwald                         }
2115e53be891SMatthias Ringwald                     } else {
2116136d331aSMatthias Ringwald                         // initiator
2117c6b7cbd9SMatthias Ringwald                         connection->sm_engine_state = SM_SC_W4_PAIRING_RANDOM;
2118e53be891SMatthias Ringwald                     }
211945a61d50SMatthias Ringwald                 }
2120e53be891SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
2121e53be891SMatthias Ringwald                 sm_timeout_reset(connection);
2122e53be891SMatthias Ringwald                 break;
2123e53be891SMatthias Ringwald             }
2124c6b7cbd9SMatthias Ringwald             case SM_SC_SEND_DHKEY_CHECK_COMMAND: {
2125e083ca23SMatthias Ringwald                 uint8_t buffer[17];
2126e083ca23SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_DHKEY_CHECK;
2127e53be891SMatthias Ringwald                 reverse_128(setup->sm_local_dhkey_check, &buffer[1]);
2128dc300847SMatthias Ringwald 
2129e53be891SMatthias Ringwald                 if (connection->sm_role){
2130c6b7cbd9SMatthias Ringwald                     connection->sm_engine_state = SM_SC_W4_LTK_REQUEST_SC;
2131e53be891SMatthias Ringwald                 } else {
2132c6b7cbd9SMatthias Ringwald                     connection->sm_engine_state = SM_SC_W4_DHKEY_CHECK_COMMAND;
2133e53be891SMatthias Ringwald                 }
2134e083ca23SMatthias Ringwald 
2135e53be891SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
2136e53be891SMatthias Ringwald                 sm_timeout_reset(connection);
2137e53be891SMatthias Ringwald                 break;
2138e53be891SMatthias Ringwald             }
2139e53be891SMatthias Ringwald 
2140e53be891SMatthias Ringwald #endif
21413deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH1_SEND_PAIRING_RESPONSE:
21423deb3ec6SMatthias Ringwald                 // echo initiator for now
21431ad129beSMatthias Ringwald                 sm_pairing_packet_set_code(setup->sm_s_pres,SM_CODE_PAIRING_RESPONSE);
21443deb3ec6SMatthias Ringwald                 key_distribution_flags = sm_key_distribution_flags_for_auth_req();
21451ad129beSMatthias Ringwald 
21463deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_RESPONDER_PH1_W4_PAIRING_CONFIRM;
214727c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
214827c32905SMatthias Ringwald                 if (setup->sm_use_secure_connections){
2149c6b7cbd9SMatthias Ringwald                     connection->sm_engine_state = SM_SC_W4_PUBLIC_KEY_COMMAND;
215052f9cf63SMatthias Ringwald                     // skip LTK/EDIV for SC
2151bbf8db22SMatthias Ringwald                     log_info("sm: dropping encryption information flag");
215252f9cf63SMatthias Ringwald                     key_distribution_flags &= ~SM_KEYDIST_ENC_KEY;
215327c32905SMatthias Ringwald                 }
215427c32905SMatthias Ringwald #endif
215552f9cf63SMatthias Ringwald                 sm_pairing_packet_set_initiator_key_distribution(setup->sm_s_pres, sm_pairing_packet_get_initiator_key_distribution(setup->sm_m_preq) & key_distribution_flags);
215652f9cf63SMatthias Ringwald                 sm_pairing_packet_set_responder_key_distribution(setup->sm_s_pres, sm_pairing_packet_get_responder_key_distribution(setup->sm_m_preq) & key_distribution_flags);
2157bbf8db22SMatthias Ringwald                 // update key distribution after ENC was dropped
2158cfd5ff74SMatthias Ringwald                 sm_setup_key_distribution(sm_pairing_packet_get_responder_key_distribution(setup->sm_s_pres));
215952f9cf63SMatthias Ringwald 
21603deb3ec6SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) &setup->sm_s_pres, sizeof(sm_pairing_packet_t));
21613deb3ec6SMatthias Ringwald                 sm_timeout_reset(connection);
2162446a8c36SMatthias Ringwald                 // SC Numeric Comparison will trigger user response after public keys & nonces have been exchanged
216345a61d50SMatthias Ringwald                 if (setup->sm_stk_generation_method == JUST_WORKS){
21643deb3ec6SMatthias Ringwald                     sm_trigger_user_response(connection);
2165446a8c36SMatthias Ringwald                 }
21663deb3ec6SMatthias Ringwald                 return;
21673deb3ec6SMatthias Ringwald 
21683deb3ec6SMatthias Ringwald             case SM_PH2_SEND_PAIRING_RANDOM: {
21693deb3ec6SMatthias Ringwald                 uint8_t buffer[17];
21703deb3ec6SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_RANDOM;
21719c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_local_random, &buffer[1]);
21723deb3ec6SMatthias Ringwald                 if (connection->sm_role){
21733deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_RESPONDER_PH2_W4_LTK_REQUEST;
21743deb3ec6SMatthias Ringwald                 } else {
21753deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_INITIATOR_PH2_W4_PAIRING_RANDOM;
21763deb3ec6SMatthias Ringwald                 }
21773deb3ec6SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
21783deb3ec6SMatthias Ringwald                 sm_timeout_reset(connection);
21793deb3ec6SMatthias Ringwald                 break;
21803deb3ec6SMatthias Ringwald             }
21813deb3ec6SMatthias Ringwald 
21823deb3ec6SMatthias Ringwald             case SM_PH2_GET_RANDOM_TK:
21833deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_RANDOM_A:
21843deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_RANDOM_B:
21853deb3ec6SMatthias Ringwald             case SM_PH3_GET_RANDOM:
21863deb3ec6SMatthias Ringwald             case SM_PH3_GET_DIV:
21873deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
21883deb3ec6SMatthias Ringwald                 sm_random_start(connection);
21893deb3ec6SMatthias Ringwald                 return;
21903deb3ec6SMatthias Ringwald 
21913deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_ENC_B:
21923deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_ENC_D:
21933deb3ec6SMatthias Ringwald                 // already busy?
21943deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
21953deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
21963deb3ec6SMatthias Ringwald                 sm_aes128_start(setup->sm_tk, setup->sm_c1_t3_value, connection);
21973deb3ec6SMatthias Ringwald                 return;
21983deb3ec6SMatthias Ringwald 
21993deb3ec6SMatthias Ringwald             case SM_PH3_LTK_GET_ENC:
22003deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH4_LTK_GET_ENC:
22013deb3ec6SMatthias Ringwald                 // already busy?
22023deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_IDLE) {
22033deb3ec6SMatthias Ringwald                     sm_key_t d_prime;
22043deb3ec6SMatthias Ringwald                     sm_d1_d_prime(setup->sm_local_div, 0, d_prime);
22053deb3ec6SMatthias Ringwald                     sm_next_responding_state(connection);
22063deb3ec6SMatthias Ringwald                     sm_aes128_start(sm_persistent_er, d_prime, connection);
22073deb3ec6SMatthias Ringwald                     return;
22083deb3ec6SMatthias Ringwald                 }
22093deb3ec6SMatthias Ringwald                 break;
22103deb3ec6SMatthias Ringwald 
22113deb3ec6SMatthias Ringwald             case SM_PH3_CSRK_GET_ENC:
22123deb3ec6SMatthias Ringwald                 // already busy?
22133deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_IDLE) {
22143deb3ec6SMatthias Ringwald                     sm_key_t d_prime;
22153deb3ec6SMatthias Ringwald                     sm_d1_d_prime(setup->sm_local_div, 1, d_prime);
22163deb3ec6SMatthias Ringwald                     sm_next_responding_state(connection);
22173deb3ec6SMatthias Ringwald                     sm_aes128_start(sm_persistent_er, d_prime, connection);
22183deb3ec6SMatthias Ringwald                     return;
22193deb3ec6SMatthias Ringwald                 }
22203deb3ec6SMatthias Ringwald                 break;
22213deb3ec6SMatthias Ringwald 
22223deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_ENC_C:
22233deb3ec6SMatthias Ringwald                 // already busy?
22243deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
22253deb3ec6SMatthias Ringwald                 // calculate m_confirm using aes128 engine - step 1
22263deb3ec6SMatthias Ringwald                 sm_c1_t1(setup->sm_peer_random, (uint8_t*) &setup->sm_m_preq, (uint8_t*) &setup->sm_s_pres, setup->sm_m_addr_type, setup->sm_s_addr_type, plaintext);
22273deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
22283deb3ec6SMatthias Ringwald                 sm_aes128_start(setup->sm_tk, plaintext, connection);
22293deb3ec6SMatthias Ringwald                 break;
22303deb3ec6SMatthias Ringwald             case SM_PH2_C1_GET_ENC_A:
22313deb3ec6SMatthias Ringwald                 // already busy?
22323deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
22333deb3ec6SMatthias Ringwald                 // calculate confirm using aes128 engine - step 1
22343deb3ec6SMatthias Ringwald                 sm_c1_t1(setup->sm_local_random, (uint8_t*) &setup->sm_m_preq, (uint8_t*) &setup->sm_s_pres, setup->sm_m_addr_type, setup->sm_s_addr_type, plaintext);
22353deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
22363deb3ec6SMatthias Ringwald                 sm_aes128_start(setup->sm_tk, plaintext, connection);
22373deb3ec6SMatthias Ringwald                 break;
22383deb3ec6SMatthias Ringwald             case SM_PH2_CALC_STK:
22393deb3ec6SMatthias Ringwald                 // already busy?
22403deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
22413deb3ec6SMatthias Ringwald                 // calculate STK
22423deb3ec6SMatthias Ringwald                 if (connection->sm_role){
22433deb3ec6SMatthias Ringwald                     sm_s1_r_prime(setup->sm_local_random, setup->sm_peer_random, plaintext);
22443deb3ec6SMatthias Ringwald                 } else {
22453deb3ec6SMatthias Ringwald                     sm_s1_r_prime(setup->sm_peer_random, setup->sm_local_random, plaintext);
22463deb3ec6SMatthias Ringwald                 }
22473deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
22483deb3ec6SMatthias Ringwald                 sm_aes128_start(setup->sm_tk, plaintext, connection);
22493deb3ec6SMatthias Ringwald                 break;
22503deb3ec6SMatthias Ringwald             case SM_PH3_Y_GET_ENC:
22513deb3ec6SMatthias Ringwald                 // already busy?
22523deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
22533deb3ec6SMatthias Ringwald                 // PH3B2 - calculate Y from      - enc
22543deb3ec6SMatthias Ringwald                 // Y = dm(DHK, Rand)
22553deb3ec6SMatthias Ringwald                 sm_dm_r_prime(setup->sm_local_rand, plaintext);
22563deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
22573deb3ec6SMatthias Ringwald                 sm_aes128_start(sm_persistent_dhk, plaintext, connection);
22583deb3ec6SMatthias Ringwald                 return;
22593deb3ec6SMatthias Ringwald             case SM_PH2_C1_SEND_PAIRING_CONFIRM: {
22603deb3ec6SMatthias Ringwald                 uint8_t buffer[17];
22613deb3ec6SMatthias Ringwald                 buffer[0] = SM_CODE_PAIRING_CONFIRM;
22629c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_local_confirm, &buffer[1]);
22633deb3ec6SMatthias Ringwald                 if (connection->sm_role){
22643deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_RESPONDER_PH2_W4_PAIRING_RANDOM;
22653deb3ec6SMatthias Ringwald                 } else {
22663deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_INITIATOR_PH2_W4_PAIRING_CONFIRM;
22673deb3ec6SMatthias Ringwald                 }
22683deb3ec6SMatthias Ringwald                 l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
22693deb3ec6SMatthias Ringwald                 sm_timeout_reset(connection);
22703deb3ec6SMatthias Ringwald                 return;
22713deb3ec6SMatthias Ringwald             }
22723deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH2_SEND_LTK_REPLY: {
22733deb3ec6SMatthias Ringwald                 sm_key_t stk_flipped;
22749c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_ltk, stk_flipped);
22753deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH2_W4_CONNECTION_ENCRYPTED;
22763deb3ec6SMatthias Ringwald                 hci_send_cmd(&hci_le_long_term_key_request_reply, connection->sm_handle, stk_flipped);
22773deb3ec6SMatthias Ringwald                 return;
22783deb3ec6SMatthias Ringwald             }
22793deb3ec6SMatthias Ringwald             case SM_INITIATOR_PH3_SEND_START_ENCRYPTION: {
22803deb3ec6SMatthias Ringwald                 sm_key_t stk_flipped;
22819c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_ltk, stk_flipped);
22823deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH2_W4_CONNECTION_ENCRYPTED;
22833deb3ec6SMatthias Ringwald                 hci_send_cmd(&hci_le_start_encryption, connection->sm_handle, 0, 0, 0, stk_flipped);
22843deb3ec6SMatthias Ringwald                 return;
22853deb3ec6SMatthias Ringwald             }
22863deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH4_SEND_LTK: {
22873deb3ec6SMatthias Ringwald                 sm_key_t ltk_flipped;
22889c80e4ccSMatthias Ringwald                 reverse_128(setup->sm_ltk, ltk_flipped);
22893deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_RESPONDER_IDLE;
22903deb3ec6SMatthias Ringwald                 hci_send_cmd(&hci_le_long_term_key_request_reply, connection->sm_handle, ltk_flipped);
22913deb3ec6SMatthias Ringwald                 return;
22923deb3ec6SMatthias Ringwald             }
22933deb3ec6SMatthias Ringwald             case SM_RESPONDER_PH4_Y_GET_ENC:
22943deb3ec6SMatthias Ringwald                 // already busy?
22953deb3ec6SMatthias Ringwald                 if (sm_aes128_state == SM_AES128_ACTIVE) break;
22963deb3ec6SMatthias Ringwald                 log_info("LTK Request: recalculating with ediv 0x%04x", setup->sm_local_ediv);
22973deb3ec6SMatthias Ringwald                 // Y = dm(DHK, Rand)
22983deb3ec6SMatthias Ringwald                 sm_dm_r_prime(setup->sm_local_rand, plaintext);
22993deb3ec6SMatthias Ringwald                 sm_next_responding_state(connection);
23003deb3ec6SMatthias Ringwald                 sm_aes128_start(sm_persistent_dhk, plaintext, connection);
23013deb3ec6SMatthias Ringwald                 return;
23023deb3ec6SMatthias Ringwald 
23033deb3ec6SMatthias Ringwald             case SM_PH3_DISTRIBUTE_KEYS:
23043deb3ec6SMatthias Ringwald                 if (setup->sm_key_distribution_send_set &   SM_KEYDIST_FLAG_ENCRYPTION_INFORMATION){
23053deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_send_set &= ~SM_KEYDIST_FLAG_ENCRYPTION_INFORMATION;
23063deb3ec6SMatthias Ringwald                     uint8_t buffer[17];
23073deb3ec6SMatthias Ringwald                     buffer[0] = SM_CODE_ENCRYPTION_INFORMATION;
23089c80e4ccSMatthias Ringwald                     reverse_128(setup->sm_ltk, &buffer[1]);
23093deb3ec6SMatthias Ringwald                     l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
23103deb3ec6SMatthias Ringwald                     sm_timeout_reset(connection);
23113deb3ec6SMatthias Ringwald                     return;
23123deb3ec6SMatthias Ringwald                 }
23133deb3ec6SMatthias Ringwald                 if (setup->sm_key_distribution_send_set &   SM_KEYDIST_FLAG_MASTER_IDENTIFICATION){
23143deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_send_set &= ~SM_KEYDIST_FLAG_MASTER_IDENTIFICATION;
23153deb3ec6SMatthias Ringwald                     uint8_t buffer[11];
23163deb3ec6SMatthias Ringwald                     buffer[0] = SM_CODE_MASTER_IDENTIFICATION;
2317f8fbdce0SMatthias Ringwald                     little_endian_store_16(buffer, 1, setup->sm_local_ediv);
23189c80e4ccSMatthias Ringwald                     reverse_64(setup->sm_local_rand, &buffer[3]);
23193deb3ec6SMatthias Ringwald                     l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
23203deb3ec6SMatthias Ringwald                     sm_timeout_reset(connection);
23213deb3ec6SMatthias Ringwald                     return;
23223deb3ec6SMatthias Ringwald                 }
23233deb3ec6SMatthias Ringwald                 if (setup->sm_key_distribution_send_set &   SM_KEYDIST_FLAG_IDENTITY_INFORMATION){
23243deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_send_set &= ~SM_KEYDIST_FLAG_IDENTITY_INFORMATION;
23253deb3ec6SMatthias Ringwald                     uint8_t buffer[17];
23263deb3ec6SMatthias Ringwald                     buffer[0] = SM_CODE_IDENTITY_INFORMATION;
23279c80e4ccSMatthias Ringwald                     reverse_128(sm_persistent_irk, &buffer[1]);
23283deb3ec6SMatthias Ringwald                     l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
23293deb3ec6SMatthias Ringwald                     sm_timeout_reset(connection);
23303deb3ec6SMatthias Ringwald                     return;
23313deb3ec6SMatthias Ringwald                 }
23323deb3ec6SMatthias Ringwald                 if (setup->sm_key_distribution_send_set &   SM_KEYDIST_FLAG_IDENTITY_ADDRESS_INFORMATION){
23333deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_send_set &= ~SM_KEYDIST_FLAG_IDENTITY_ADDRESS_INFORMATION;
23343deb3ec6SMatthias Ringwald                     bd_addr_t local_address;
23353deb3ec6SMatthias Ringwald                     uint8_t buffer[8];
23363deb3ec6SMatthias Ringwald                     buffer[0] = SM_CODE_IDENTITY_ADDRESS_INFORMATION;
233715a95bd5SMatthias Ringwald                     gap_advertisements_get_address(&buffer[1], local_address);
2338724d70a2SMatthias Ringwald                     reverse_bd_addr(local_address, &buffer[2]);
23393deb3ec6SMatthias Ringwald                     l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
23403deb3ec6SMatthias Ringwald                     sm_timeout_reset(connection);
23413deb3ec6SMatthias Ringwald                     return;
23423deb3ec6SMatthias Ringwald                 }
23433deb3ec6SMatthias Ringwald                 if (setup->sm_key_distribution_send_set &   SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION){
23443deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_send_set &= ~SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION;
23453deb3ec6SMatthias Ringwald 
23463deb3ec6SMatthias Ringwald                     // hack to reproduce test runs
23473deb3ec6SMatthias Ringwald                     if (test_use_fixed_local_csrk){
23483deb3ec6SMatthias Ringwald                         memset(setup->sm_local_csrk, 0xcc, 16);
23493deb3ec6SMatthias Ringwald                     }
23503deb3ec6SMatthias Ringwald 
23513deb3ec6SMatthias Ringwald                     uint8_t buffer[17];
23523deb3ec6SMatthias Ringwald                     buffer[0] = SM_CODE_SIGNING_INFORMATION;
23539c80e4ccSMatthias Ringwald                     reverse_128(setup->sm_local_csrk, &buffer[1]);
23543deb3ec6SMatthias Ringwald                     l2cap_send_connectionless(connection->sm_handle, L2CAP_CID_SECURITY_MANAGER_PROTOCOL, (uint8_t*) buffer, sizeof(buffer));
23553deb3ec6SMatthias Ringwald                     sm_timeout_reset(connection);
23563deb3ec6SMatthias Ringwald                     return;
23573deb3ec6SMatthias Ringwald                 }
23583deb3ec6SMatthias Ringwald 
23593deb3ec6SMatthias Ringwald                 // keys are sent
23603deb3ec6SMatthias Ringwald                 if (connection->sm_role){
23613deb3ec6SMatthias Ringwald                     // slave -> receive master keys if any
23623deb3ec6SMatthias Ringwald                     if (sm_key_distribution_all_received(connection)){
23633deb3ec6SMatthias Ringwald                         sm_key_distribution_handle_all_received(connection);
23643deb3ec6SMatthias Ringwald                         connection->sm_engine_state = SM_RESPONDER_IDLE;
23653deb3ec6SMatthias Ringwald                         sm_done_for_handle(connection->sm_handle);
23663deb3ec6SMatthias Ringwald                     } else {
23673deb3ec6SMatthias Ringwald                         connection->sm_engine_state = SM_PH3_RECEIVE_KEYS;
23683deb3ec6SMatthias Ringwald                     }
23693deb3ec6SMatthias Ringwald                 } else {
23703deb3ec6SMatthias Ringwald                     // master -> all done
23713deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_INITIATOR_CONNECTED;
23723deb3ec6SMatthias Ringwald                     sm_done_for_handle(connection->sm_handle);
23733deb3ec6SMatthias Ringwald                 }
23743deb3ec6SMatthias Ringwald                 break;
23753deb3ec6SMatthias Ringwald 
23763deb3ec6SMatthias Ringwald             default:
23773deb3ec6SMatthias Ringwald                 break;
23783deb3ec6SMatthias Ringwald         }
23793deb3ec6SMatthias Ringwald 
23803deb3ec6SMatthias Ringwald         // check again if active connection was released
23813deb3ec6SMatthias Ringwald         if (sm_active_connection) break;
23823deb3ec6SMatthias Ringwald     }
23833deb3ec6SMatthias Ringwald }
23843deb3ec6SMatthias Ringwald 
23853deb3ec6SMatthias Ringwald // note: aes engine is ready as we just got the aes result
23863deb3ec6SMatthias Ringwald static void sm_handle_encryption_result(uint8_t * data){
23873deb3ec6SMatthias Ringwald 
23883deb3ec6SMatthias Ringwald     sm_aes128_state = SM_AES128_IDLE;
23893deb3ec6SMatthias Ringwald 
23903deb3ec6SMatthias Ringwald     if (sm_address_resolution_ah_calculation_active){
23913deb3ec6SMatthias Ringwald         sm_address_resolution_ah_calculation_active = 0;
23923deb3ec6SMatthias Ringwald         // compare calulated address against connecting device
23933deb3ec6SMatthias Ringwald         uint8_t hash[3];
23949c80e4ccSMatthias Ringwald         reverse_24(data, hash);
23953deb3ec6SMatthias Ringwald         if (memcmp(&sm_address_resolution_address[3], hash, 3) == 0){
23963deb3ec6SMatthias Ringwald             log_info("LE Device Lookup: matched resolvable private address");
23973deb3ec6SMatthias Ringwald             sm_address_resolution_handle_event(ADDRESS_RESOLUTION_SUCEEDED);
23983deb3ec6SMatthias Ringwald             return;
23993deb3ec6SMatthias Ringwald         }
24003deb3ec6SMatthias Ringwald         // no match, try next
24013deb3ec6SMatthias Ringwald         sm_address_resolution_test++;
24023deb3ec6SMatthias Ringwald         return;
24033deb3ec6SMatthias Ringwald     }
24043deb3ec6SMatthias Ringwald 
24053deb3ec6SMatthias Ringwald     switch (dkg_state){
24063deb3ec6SMatthias Ringwald         case DKG_W4_IRK:
24079c80e4ccSMatthias Ringwald             reverse_128(data, sm_persistent_irk);
24088314c363SMatthias Ringwald             log_info_key("irk", sm_persistent_irk);
24093deb3ec6SMatthias Ringwald             dkg_next_state();
24103deb3ec6SMatthias Ringwald             return;
24113deb3ec6SMatthias Ringwald         case DKG_W4_DHK:
24129c80e4ccSMatthias Ringwald             reverse_128(data, sm_persistent_dhk);
24138314c363SMatthias Ringwald             log_info_key("dhk", sm_persistent_dhk);
24143deb3ec6SMatthias Ringwald             dkg_next_state();
24156f792faaSMatthias Ringwald             // SM Init Finished
24163deb3ec6SMatthias Ringwald             return;
24173deb3ec6SMatthias Ringwald         default:
24183deb3ec6SMatthias Ringwald             break;
24193deb3ec6SMatthias Ringwald     }
24203deb3ec6SMatthias Ringwald 
24213deb3ec6SMatthias Ringwald     switch (rau_state){
24223deb3ec6SMatthias Ringwald         case RAU_W4_ENC:
24239c80e4ccSMatthias Ringwald             reverse_24(data, &sm_random_address[3]);
24243deb3ec6SMatthias Ringwald             rau_next_state();
24253deb3ec6SMatthias Ringwald             return;
24263deb3ec6SMatthias Ringwald         default:
24273deb3ec6SMatthias Ringwald             break;
24283deb3ec6SMatthias Ringwald     }
24293deb3ec6SMatthias Ringwald 
24303deb3ec6SMatthias Ringwald     switch (sm_cmac_state){
24313deb3ec6SMatthias Ringwald         case CMAC_W4_SUBKEYS:
24323deb3ec6SMatthias Ringwald         case CMAC_W4_MI:
24333deb3ec6SMatthias Ringwald         case CMAC_W4_MLAST:
24343deb3ec6SMatthias Ringwald             {
24353deb3ec6SMatthias Ringwald             sm_key_t t;
24369c80e4ccSMatthias Ringwald             reverse_128(data, t);
24373deb3ec6SMatthias Ringwald             sm_cmac_handle_encryption_result(t);
24383deb3ec6SMatthias Ringwald             }
24393deb3ec6SMatthias Ringwald             return;
24403deb3ec6SMatthias Ringwald         default:
24413deb3ec6SMatthias Ringwald             break;
24423deb3ec6SMatthias Ringwald     }
24433deb3ec6SMatthias Ringwald 
24443deb3ec6SMatthias Ringwald     // retrieve sm_connection provided to sm_aes128_start_encryption
24453deb3ec6SMatthias Ringwald     sm_connection_t * connection = (sm_connection_t*) sm_aes128_context;
24463deb3ec6SMatthias Ringwald     if (!connection) return;
24473deb3ec6SMatthias Ringwald     switch (connection->sm_engine_state){
24483deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_ENC_A:
24493deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_ENC_C:
24503deb3ec6SMatthias Ringwald             {
24513deb3ec6SMatthias Ringwald             sm_key_t t2;
24529c80e4ccSMatthias Ringwald             reverse_128(data, t2);
24533deb3ec6SMatthias Ringwald             sm_c1_t3(t2, setup->sm_m_address, setup->sm_s_address, setup->sm_c1_t3_value);
24543deb3ec6SMatthias Ringwald             }
24553deb3ec6SMatthias Ringwald             sm_next_responding_state(connection);
24563deb3ec6SMatthias Ringwald             return;
24573deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_ENC_B:
24589c80e4ccSMatthias Ringwald             reverse_128(data, setup->sm_local_confirm);
24598314c363SMatthias Ringwald             log_info_key("c1!", setup->sm_local_confirm);
24603deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH2_C1_SEND_PAIRING_CONFIRM;
24613deb3ec6SMatthias Ringwald             return;
24623deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_ENC_D:
24633deb3ec6SMatthias Ringwald             {
24643deb3ec6SMatthias Ringwald             sm_key_t peer_confirm_test;
24659c80e4ccSMatthias Ringwald             reverse_128(data, peer_confirm_test);
24668314c363SMatthias Ringwald             log_info_key("c1!", peer_confirm_test);
24673deb3ec6SMatthias Ringwald             if (memcmp(setup->sm_peer_confirm, peer_confirm_test, 16) != 0){
24683deb3ec6SMatthias Ringwald                 setup->sm_pairing_failed_reason = SM_REASON_CONFIRM_VALUE_FAILED;
24693deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_GENERAL_SEND_PAIRING_FAILED;
24703deb3ec6SMatthias Ringwald                 return;
24713deb3ec6SMatthias Ringwald             }
24723deb3ec6SMatthias Ringwald             if (connection->sm_role){
24733deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH2_SEND_PAIRING_RANDOM;
24743deb3ec6SMatthias Ringwald             } else {
24753deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH2_CALC_STK;
24763deb3ec6SMatthias Ringwald             }
24773deb3ec6SMatthias Ringwald             }
24783deb3ec6SMatthias Ringwald             return;
24793deb3ec6SMatthias Ringwald         case SM_PH2_W4_STK:
24809c80e4ccSMatthias Ringwald             reverse_128(data, setup->sm_ltk);
24813deb3ec6SMatthias Ringwald             sm_truncate_key(setup->sm_ltk, connection->sm_actual_encryption_key_size);
24828314c363SMatthias Ringwald             log_info_key("stk", setup->sm_ltk);
24833deb3ec6SMatthias Ringwald             if (connection->sm_role){
24843deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_RESPONDER_PH2_SEND_LTK_REPLY;
24853deb3ec6SMatthias Ringwald             } else {
24863deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_INITIATOR_PH3_SEND_START_ENCRYPTION;
24873deb3ec6SMatthias Ringwald             }
24883deb3ec6SMatthias Ringwald             return;
24893deb3ec6SMatthias Ringwald         case SM_PH3_Y_W4_ENC:{
24903deb3ec6SMatthias Ringwald             sm_key_t y128;
24919c80e4ccSMatthias Ringwald             reverse_128(data, y128);
2492f8fbdce0SMatthias Ringwald             setup->sm_local_y = big_endian_read_16(y128, 14);
24933deb3ec6SMatthias Ringwald             log_info_hex16("y", setup->sm_local_y);
24943deb3ec6SMatthias Ringwald             // PH3B3 - calculate EDIV
24953deb3ec6SMatthias Ringwald             setup->sm_local_ediv = setup->sm_local_y ^ setup->sm_local_div;
24963deb3ec6SMatthias Ringwald             log_info_hex16("ediv", setup->sm_local_ediv);
24973deb3ec6SMatthias Ringwald             // PH3B4 - calculate LTK         - enc
24983deb3ec6SMatthias Ringwald             // LTK = d1(ER, DIV, 0))
24993deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH3_LTK_GET_ENC;
25003deb3ec6SMatthias Ringwald             return;
25013deb3ec6SMatthias Ringwald         }
25023deb3ec6SMatthias Ringwald         case SM_RESPONDER_PH4_Y_W4_ENC:{
25033deb3ec6SMatthias Ringwald             sm_key_t y128;
25049c80e4ccSMatthias Ringwald             reverse_128(data, y128);
2505f8fbdce0SMatthias Ringwald             setup->sm_local_y = big_endian_read_16(y128, 14);
25063deb3ec6SMatthias Ringwald             log_info_hex16("y", setup->sm_local_y);
25073deb3ec6SMatthias Ringwald 
25083deb3ec6SMatthias Ringwald             // PH3B3 - calculate DIV
25093deb3ec6SMatthias Ringwald             setup->sm_local_div = setup->sm_local_y ^ setup->sm_local_ediv;
25103deb3ec6SMatthias Ringwald             log_info_hex16("ediv", setup->sm_local_ediv);
25113deb3ec6SMatthias Ringwald             // PH3B4 - calculate LTK         - enc
25123deb3ec6SMatthias Ringwald             // LTK = d1(ER, DIV, 0))
25133deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_RESPONDER_PH4_LTK_GET_ENC;
25143deb3ec6SMatthias Ringwald             return;
25153deb3ec6SMatthias Ringwald         }
25163deb3ec6SMatthias Ringwald         case SM_PH3_LTK_W4_ENC:
25179c80e4ccSMatthias Ringwald             reverse_128(data, setup->sm_ltk);
25188314c363SMatthias Ringwald             log_info_key("ltk", setup->sm_ltk);
25193deb3ec6SMatthias Ringwald             // calc CSRK next
25203deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH3_CSRK_GET_ENC;
25213deb3ec6SMatthias Ringwald             return;
25223deb3ec6SMatthias Ringwald         case SM_PH3_CSRK_W4_ENC:
25239c80e4ccSMatthias Ringwald             reverse_128(data, setup->sm_local_csrk);
25248314c363SMatthias Ringwald             log_info_key("csrk", setup->sm_local_csrk);
25253deb3ec6SMatthias Ringwald             if (setup->sm_key_distribution_send_set){
25263deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH3_DISTRIBUTE_KEYS;
25273deb3ec6SMatthias Ringwald             } else {
25283deb3ec6SMatthias Ringwald                 // no keys to send, just continue
25293deb3ec6SMatthias Ringwald                 if (connection->sm_role){
25303deb3ec6SMatthias Ringwald                     // slave -> receive master keys
25313deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_PH3_RECEIVE_KEYS;
25323deb3ec6SMatthias Ringwald                 } else {
25333deb3ec6SMatthias Ringwald                     // master -> all done
25343deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_INITIATOR_CONNECTED;
25353deb3ec6SMatthias Ringwald                     sm_done_for_handle(connection->sm_handle);
25363deb3ec6SMatthias Ringwald                 }
25373deb3ec6SMatthias Ringwald             }
25383deb3ec6SMatthias Ringwald             return;
25393deb3ec6SMatthias Ringwald         case SM_RESPONDER_PH4_LTK_W4_ENC:
25409c80e4ccSMatthias Ringwald             reverse_128(data, setup->sm_ltk);
25413deb3ec6SMatthias Ringwald             sm_truncate_key(setup->sm_ltk, connection->sm_actual_encryption_key_size);
25428314c363SMatthias Ringwald             log_info_key("ltk", setup->sm_ltk);
25433deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_RESPONDER_PH4_SEND_LTK;
25443deb3ec6SMatthias Ringwald             return;
25453deb3ec6SMatthias Ringwald         default:
25463deb3ec6SMatthias Ringwald             break;
25473deb3ec6SMatthias Ringwald     }
25483deb3ec6SMatthias Ringwald }
25493deb3ec6SMatthias Ringwald 
2550e722521aSMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
2551e722521aSMatthias Ringwald 
25527df18c15SMatthias Ringwald static int sm_generate_f_rng(void * context, unsigned char * buffer, size_t size){
25537df18c15SMatthias Ringwald     int offset = setup->sm_passkey_bit;
25547df18c15SMatthias Ringwald     log_info("sm_generate_f_rng: size %u - offset %u", (int) size, offset);
25557df18c15SMatthias Ringwald     while (size) {
25567df18c15SMatthias Ringwald         if (offset < 32){
25577df18c15SMatthias Ringwald             *buffer++ = setup->sm_peer_qx[offset++];
25587df18c15SMatthias Ringwald         } else {
25597df18c15SMatthias Ringwald             *buffer++ = setup->sm_peer_qx[offset++ - 32];
25607df18c15SMatthias Ringwald         }
25617df18c15SMatthias Ringwald         size--;
25627df18c15SMatthias Ringwald     }
25637df18c15SMatthias Ringwald     setup->sm_passkey_bit = offset;
25647df18c15SMatthias Ringwald     return 0;
25657df18c15SMatthias Ringwald }
25667df18c15SMatthias Ringwald #endif
25677df18c15SMatthias Ringwald 
25683deb3ec6SMatthias Ringwald // note: random generator is ready. this doesn NOT imply that aes engine is unused!
25693deb3ec6SMatthias Ringwald static void sm_handle_random_result(uint8_t * data){
25703deb3ec6SMatthias Ringwald 
25717df18c15SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
25727df18c15SMatthias Ringwald     if (ec_key_generation_state == EC_KEY_GENERATION_ACTIVE){
25737df18c15SMatthias Ringwald         int num_bytes = setup->sm_passkey_bit;
25747df18c15SMatthias Ringwald         if (num_bytes < 32){
25757df18c15SMatthias Ringwald             memcpy(&setup->sm_peer_qx[num_bytes], data, 8);
25767df18c15SMatthias Ringwald         } else {
25777df18c15SMatthias Ringwald             memcpy(&setup->sm_peer_qx[num_bytes-32], data, 8);
25787df18c15SMatthias Ringwald         }
25797df18c15SMatthias Ringwald         num_bytes += 8;
25807df18c15SMatthias Ringwald         setup->sm_passkey_bit = num_bytes;
25817df18c15SMatthias Ringwald 
25827df18c15SMatthias Ringwald         if (num_bytes >= 64){
25837df18c15SMatthias Ringwald             // generate EC key
25847df18c15SMatthias Ringwald             setup->sm_passkey_bit = 0;
2585e722521aSMatthias Ringwald             mbedtls_mpi d;
2586e722521aSMatthias Ringwald             mbedtls_ecp_point P;
2587e722521aSMatthias Ringwald             mbedtls_mpi_init(&d);
2588e722521aSMatthias Ringwald             mbedtls_ecp_point_init(&P);
2589e722521aSMatthias Ringwald             mbedtls_ecp_gen_keypair(&mbedtls_ec_group, &d, &P, &sm_generate_f_rng, NULL);
2590e722521aSMatthias Ringwald             mbedtls_mpi_write_binary(&P.X, ec_qx, 16);
2591e722521aSMatthias Ringwald             mbedtls_mpi_write_binary(&P.Y, ec_qy, 16);
2592e722521aSMatthias Ringwald             mbedtls_mpi_write_binary(&d, ec_d, 16);
25937df18c15SMatthias Ringwald             sm_log_ec_keypair();
2594e722521aSMatthias Ringwald             mbedtls_ecp_point_free(&P);
2595e722521aSMatthias Ringwald             mbedtls_mpi_free(&d);
25967df18c15SMatthias Ringwald             ec_key_generation_state = EC_KEY_GENERATION_DONE;
25977df18c15SMatthias Ringwald         }
25987df18c15SMatthias Ringwald     }
25997df18c15SMatthias Ringwald #endif
26007df18c15SMatthias Ringwald 
26013deb3ec6SMatthias Ringwald     switch (rau_state){
26023deb3ec6SMatthias Ringwald         case RAU_W4_RANDOM:
26033deb3ec6SMatthias Ringwald             // non-resolvable vs. resolvable
26043deb3ec6SMatthias Ringwald             switch (gap_random_adress_type){
26053deb3ec6SMatthias Ringwald                 case GAP_RANDOM_ADDRESS_RESOLVABLE:
26063deb3ec6SMatthias Ringwald                     // resolvable: use random as prand and calc address hash
26073deb3ec6SMatthias Ringwald                     // "The two most significant bits of prand shall be equal to ‘0’ and ‘1"
26083deb3ec6SMatthias Ringwald                     memcpy(sm_random_address, data, 3);
26093deb3ec6SMatthias Ringwald                     sm_random_address[0] &= 0x3f;
26103deb3ec6SMatthias Ringwald                     sm_random_address[0] |= 0x40;
26113deb3ec6SMatthias Ringwald                     rau_state = RAU_GET_ENC;
26123deb3ec6SMatthias Ringwald                     break;
26133deb3ec6SMatthias Ringwald                 case GAP_RANDOM_ADDRESS_NON_RESOLVABLE:
26143deb3ec6SMatthias Ringwald                 default:
26153deb3ec6SMatthias Ringwald                     // "The two most significant bits of the address shall be equal to ‘0’""
26163deb3ec6SMatthias Ringwald                     memcpy(sm_random_address, data, 6);
26173deb3ec6SMatthias Ringwald                     sm_random_address[0] &= 0x3f;
26183deb3ec6SMatthias Ringwald                     rau_state = RAU_SET_ADDRESS;
26193deb3ec6SMatthias Ringwald                     break;
26203deb3ec6SMatthias Ringwald             }
26213deb3ec6SMatthias Ringwald             return;
26223deb3ec6SMatthias Ringwald         default:
26233deb3ec6SMatthias Ringwald             break;
26243deb3ec6SMatthias Ringwald     }
26253deb3ec6SMatthias Ringwald 
26263deb3ec6SMatthias Ringwald     // retrieve sm_connection provided to sm_random_start
26273deb3ec6SMatthias Ringwald     sm_connection_t * connection = (sm_connection_t *) sm_random_context;
26283deb3ec6SMatthias Ringwald     if (!connection) return;
26293deb3ec6SMatthias Ringwald     switch (connection->sm_engine_state){
2630f1c1783eSMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
2631f1c1783eSMatthias Ringwald         case SM_SC_W4_GET_RANDOM_A:
2632f1c1783eSMatthias Ringwald             memcpy(&setup->sm_local_nonce[0], data, 8);
2633f1c1783eSMatthias Ringwald             connection->sm_engine_state = SM_SC_W2_GET_RANDOM_B;
2634f1c1783eSMatthias Ringwald             break;
2635f1c1783eSMatthias Ringwald         case SM_SC_W4_GET_RANDOM_B:
2636f1c1783eSMatthias Ringwald             memcpy(&setup->sm_local_nonce[8], data, 8);
2637f1c1783eSMatthias Ringwald             // initiator & jw/nc -> send pairing random
2638f1c1783eSMatthias Ringwald             if (connection->sm_role == 0 && sm_just_works_or_numeric_comparison(setup->sm_stk_generation_method)){
2639f1c1783eSMatthias Ringwald                 connection->sm_engine_state = SM_SC_SEND_PAIRING_RANDOM;
2640f1c1783eSMatthias Ringwald                 break;
2641b35a3de2SMatthias Ringwald             } else {
2642b35a3de2SMatthias Ringwald                 connection->sm_engine_state = SM_SC_W2_CMAC_FOR_CONFIRMATION;
2643f1c1783eSMatthias Ringwald             }
2644f1c1783eSMatthias Ringwald             break;
2645f1c1783eSMatthias Ringwald #endif
2646f1c1783eSMatthias Ringwald 
26473deb3ec6SMatthias Ringwald         case SM_PH2_W4_RANDOM_TK:
26483deb3ec6SMatthias Ringwald         {
26493deb3ec6SMatthias Ringwald             // map random to 0-999999 without speding much cycles on a modulus operation
2650f8fbdce0SMatthias Ringwald             uint32_t tk = little_endian_read_32(data,0);
26513deb3ec6SMatthias Ringwald             tk = tk & 0xfffff;  // 1048575
26523deb3ec6SMatthias Ringwald             if (tk >= 999999){
26533deb3ec6SMatthias Ringwald                 tk = tk - 999999;
26543deb3ec6SMatthias Ringwald             }
26553deb3ec6SMatthias Ringwald             sm_reset_tk();
2656f8fbdce0SMatthias Ringwald             big_endian_store_32(setup->sm_tk, 12, tk);
26573deb3ec6SMatthias Ringwald             if (connection->sm_role){
26583deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_RESPONDER_PH1_SEND_PAIRING_RESPONSE;
26593deb3ec6SMatthias Ringwald             } else {
26603deb3ec6SMatthias Ringwald                 connection->sm_engine_state = SM_PH1_W4_USER_RESPONSE;
26613deb3ec6SMatthias Ringwald                 sm_trigger_user_response(connection);
26623deb3ec6SMatthias Ringwald                 // response_idle == nothing <--> sm_trigger_user_response() did not require response
26633deb3ec6SMatthias Ringwald                 if (setup->sm_user_response == SM_USER_RESPONSE_IDLE){
26643deb3ec6SMatthias Ringwald                     connection->sm_engine_state = SM_PH2_C1_GET_RANDOM_A;
26653deb3ec6SMatthias Ringwald                 }
26663deb3ec6SMatthias Ringwald             }
26673deb3ec6SMatthias Ringwald             return;
26683deb3ec6SMatthias Ringwald         }
26693deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_RANDOM_A:
26703deb3ec6SMatthias Ringwald             memcpy(&setup->sm_local_random[0], data, 8); // random endinaness
26713deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH2_C1_GET_RANDOM_B;
26723deb3ec6SMatthias Ringwald             return;
26733deb3ec6SMatthias Ringwald         case SM_PH2_C1_W4_RANDOM_B:
26743deb3ec6SMatthias Ringwald             memcpy(&setup->sm_local_random[8], data, 8); // random endinaness
26753deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH2_C1_GET_ENC_A;
26763deb3ec6SMatthias Ringwald             return;
26773deb3ec6SMatthias Ringwald         case SM_PH3_W4_RANDOM:
26789c80e4ccSMatthias Ringwald             reverse_64(data, setup->sm_local_rand);
26793deb3ec6SMatthias Ringwald             // no db for encryption size hack: encryption size is stored in lowest nibble of setup->sm_local_rand
26803deb3ec6SMatthias Ringwald             setup->sm_local_rand[7] = (setup->sm_local_rand[7] & 0xf0) + (connection->sm_actual_encryption_key_size - 1);
26813deb3ec6SMatthias Ringwald             // no db for authenticated flag hack: store flag in bit 4 of LSB
26823deb3ec6SMatthias Ringwald             setup->sm_local_rand[7] = (setup->sm_local_rand[7] & 0xef) + (connection->sm_connection_authenticated << 4);
26833deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH3_GET_DIV;
26843deb3ec6SMatthias Ringwald             return;
26853deb3ec6SMatthias Ringwald         case SM_PH3_W4_DIV:
26863deb3ec6SMatthias Ringwald             // use 16 bit from random value as div
2687f8fbdce0SMatthias Ringwald             setup->sm_local_div = big_endian_read_16(data, 0);
26883deb3ec6SMatthias Ringwald             log_info_hex16("div", setup->sm_local_div);
26893deb3ec6SMatthias Ringwald             connection->sm_engine_state = SM_PH3_Y_GET_ENC;
26903deb3ec6SMatthias Ringwald             return;
26913deb3ec6SMatthias Ringwald         default:
26923deb3ec6SMatthias Ringwald             break;
26933deb3ec6SMatthias Ringwald     }
26943deb3ec6SMatthias Ringwald }
26953deb3ec6SMatthias Ringwald 
2696d9a7306aSMatthias Ringwald static void sm_event_packet_handler (uint8_t packet_type, uint16_t channel, uint8_t *packet, uint16_t size){
26973deb3ec6SMatthias Ringwald 
26983deb3ec6SMatthias Ringwald     sm_connection_t  * sm_conn;
2699711e6c80SMatthias Ringwald     hci_con_handle_t con_handle;
27003deb3ec6SMatthias Ringwald 
27013deb3ec6SMatthias Ringwald     switch (packet_type) {
27023deb3ec6SMatthias Ringwald 
27033deb3ec6SMatthias Ringwald 		case HCI_EVENT_PACKET:
27040e2df43fSMatthias Ringwald 			switch (hci_event_packet_get_type(packet)) {
27053deb3ec6SMatthias Ringwald 
27063deb3ec6SMatthias Ringwald                 case BTSTACK_EVENT_STATE:
27073deb3ec6SMatthias Ringwald 					// bt stack activated, get started
2708be7cc9a0SMilanka Ringwald 					if (btstack_event_state_get_state(packet) == HCI_STATE_WORKING){
27093deb3ec6SMatthias Ringwald                         log_info("HCI Working!");
27103deb3ec6SMatthias Ringwald                         dkg_state = sm_persistent_irk_ready ? DKG_CALC_DHK : DKG_CALC_IRK;
27113deb3ec6SMatthias Ringwald                         rau_state = RAU_IDLE;
27127df18c15SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
27137df18c15SMatthias Ringwald                         if (!test_use_fixed_ec_keypair){
27147df18c15SMatthias Ringwald                             setup->sm_passkey_bit = 0;
27157df18c15SMatthias Ringwald                             ec_key_generation_state = EC_KEY_GENERATION_ACTIVE;
27167df18c15SMatthias Ringwald                         }
27177df18c15SMatthias Ringwald #endif
27183deb3ec6SMatthias Ringwald                         sm_run();
27193deb3ec6SMatthias Ringwald 					}
27203deb3ec6SMatthias Ringwald 					break;
27213deb3ec6SMatthias Ringwald 
27223deb3ec6SMatthias Ringwald                 case HCI_EVENT_LE_META:
27233deb3ec6SMatthias Ringwald                     switch (packet[2]) {
27243deb3ec6SMatthias Ringwald                         case HCI_SUBEVENT_LE_CONNECTION_COMPLETE:
27253deb3ec6SMatthias Ringwald 
27263deb3ec6SMatthias Ringwald                             log_info("sm: connected");
27273deb3ec6SMatthias Ringwald 
27283deb3ec6SMatthias Ringwald                             if (packet[3]) return; // connection failed
27293deb3ec6SMatthias Ringwald 
2730711e6c80SMatthias Ringwald                             con_handle = little_endian_read_16(packet, 4);
2731711e6c80SMatthias Ringwald                             sm_conn = sm_get_connection_for_handle(con_handle);
27323deb3ec6SMatthias Ringwald                             if (!sm_conn) break;
27333deb3ec6SMatthias Ringwald 
2734711e6c80SMatthias Ringwald                             sm_conn->sm_handle = con_handle;
27353deb3ec6SMatthias Ringwald                             sm_conn->sm_role = packet[6];
27363deb3ec6SMatthias Ringwald                             sm_conn->sm_peer_addr_type = packet[7];
2737724d70a2SMatthias Ringwald                             reverse_bd_addr(&packet[8],
2738724d70a2SMatthias Ringwald                                             sm_conn->sm_peer_address);
27393deb3ec6SMatthias Ringwald 
27403deb3ec6SMatthias Ringwald                             log_info("New sm_conn, role %s", sm_conn->sm_role ? "slave" : "master");
27413deb3ec6SMatthias Ringwald 
27423deb3ec6SMatthias Ringwald                             // reset security properties
27433deb3ec6SMatthias Ringwald                             sm_conn->sm_connection_encrypted = 0;
27443deb3ec6SMatthias Ringwald                             sm_conn->sm_connection_authenticated = 0;
27453deb3ec6SMatthias Ringwald                             sm_conn->sm_connection_authorization_state = AUTHORIZATION_UNKNOWN;
27463deb3ec6SMatthias Ringwald                             sm_conn->sm_le_db_index = -1;
27473deb3ec6SMatthias Ringwald 
27483deb3ec6SMatthias Ringwald                             // prepare CSRK lookup (does not involve setup)
27493deb3ec6SMatthias Ringwald                             sm_conn->sm_irk_lookup_state = IRK_LOOKUP_W4_READY;
27503deb3ec6SMatthias Ringwald 
27513deb3ec6SMatthias Ringwald                             // just connected -> everything else happens in sm_run()
27523deb3ec6SMatthias Ringwald                             if (sm_conn->sm_role){
27533deb3ec6SMatthias Ringwald                                 // slave - state already could be SM_RESPONDER_SEND_SECURITY_REQUEST instead
27543deb3ec6SMatthias Ringwald                                 if (sm_conn->sm_engine_state == SM_GENERAL_IDLE){
27553deb3ec6SMatthias Ringwald                                     if (sm_slave_request_security) {
27563deb3ec6SMatthias Ringwald                                         // request security if requested by app
27573deb3ec6SMatthias Ringwald                                         sm_conn->sm_engine_state = SM_RESPONDER_SEND_SECURITY_REQUEST;
27583deb3ec6SMatthias Ringwald                                     } else {
27593deb3ec6SMatthias Ringwald                                         // otherwise, wait for pairing request
27603deb3ec6SMatthias Ringwald                                         sm_conn->sm_engine_state = SM_RESPONDER_IDLE;
27613deb3ec6SMatthias Ringwald                                     }
27623deb3ec6SMatthias Ringwald                                 }
27633deb3ec6SMatthias Ringwald                                 break;
27643deb3ec6SMatthias Ringwald                             } else {
27653deb3ec6SMatthias Ringwald                                 // master
27663deb3ec6SMatthias Ringwald                                 sm_conn->sm_engine_state = SM_INITIATOR_CONNECTED;
27673deb3ec6SMatthias Ringwald                             }
27683deb3ec6SMatthias Ringwald                             break;
27693deb3ec6SMatthias Ringwald 
27703deb3ec6SMatthias Ringwald                         case HCI_SUBEVENT_LE_LONG_TERM_KEY_REQUEST:
2771711e6c80SMatthias Ringwald                             con_handle = little_endian_read_16(packet, 3);
2772711e6c80SMatthias Ringwald                             sm_conn = sm_get_connection_for_handle(con_handle);
27733deb3ec6SMatthias Ringwald                             if (!sm_conn) break;
27743deb3ec6SMatthias Ringwald 
27753deb3ec6SMatthias Ringwald                             log_info("LTK Request: state %u", sm_conn->sm_engine_state);
27763deb3ec6SMatthias Ringwald                             if (sm_conn->sm_engine_state == SM_RESPONDER_PH2_W4_LTK_REQUEST){
27773deb3ec6SMatthias Ringwald                                 sm_conn->sm_engine_state = SM_PH2_CALC_STK;
27783deb3ec6SMatthias Ringwald                                 break;
27793deb3ec6SMatthias Ringwald                             }
2780c6b7cbd9SMatthias Ringwald                             if (sm_conn->sm_engine_state == SM_SC_W4_LTK_REQUEST_SC){
2781e53be891SMatthias Ringwald                                 sm_conn->sm_engine_state = SM_RESPONDER_PH2_SEND_LTK_REPLY;
2782e53be891SMatthias Ringwald                                 break;
2783e53be891SMatthias Ringwald                             }
27843deb3ec6SMatthias Ringwald 
27853deb3ec6SMatthias Ringwald                             // store rand and ediv
27869c80e4ccSMatthias Ringwald                             reverse_64(&packet[5], sm_conn->sm_local_rand);
2787f8fbdce0SMatthias Ringwald                             sm_conn->sm_local_ediv   = little_endian_read_16(packet, 13);
27883deb3ec6SMatthias Ringwald                             sm_conn->sm_engine_state = SM_RESPONDER_PH0_RECEIVED_LTK;
27893deb3ec6SMatthias Ringwald                             break;
27903deb3ec6SMatthias Ringwald 
27913deb3ec6SMatthias Ringwald                         default:
27923deb3ec6SMatthias Ringwald                             break;
27933deb3ec6SMatthias Ringwald                     }
27943deb3ec6SMatthias Ringwald                     break;
27953deb3ec6SMatthias Ringwald 
27963deb3ec6SMatthias Ringwald                 case HCI_EVENT_ENCRYPTION_CHANGE:
2797711e6c80SMatthias Ringwald                     con_handle = little_endian_read_16(packet, 3);
2798711e6c80SMatthias Ringwald                     sm_conn = sm_get_connection_for_handle(con_handle);
27993deb3ec6SMatthias Ringwald                     if (!sm_conn) break;
28003deb3ec6SMatthias Ringwald 
28013deb3ec6SMatthias Ringwald                     sm_conn->sm_connection_encrypted = packet[5];
28023deb3ec6SMatthias Ringwald                     log_info("Encryption state change: %u, key size %u", sm_conn->sm_connection_encrypted,
28033deb3ec6SMatthias Ringwald                         sm_conn->sm_actual_encryption_key_size);
28043deb3ec6SMatthias Ringwald                     log_info("event handler, state %u", sm_conn->sm_engine_state);
28053deb3ec6SMatthias Ringwald                     if (!sm_conn->sm_connection_encrypted) break;
28063deb3ec6SMatthias Ringwald                     // continue if part of initial pairing
28073deb3ec6SMatthias Ringwald                     switch (sm_conn->sm_engine_state){
28083deb3ec6SMatthias Ringwald                         case SM_INITIATOR_PH0_W4_CONNECTION_ENCRYPTED:
28093deb3ec6SMatthias Ringwald                             sm_conn->sm_engine_state = SM_INITIATOR_CONNECTED;
28103deb3ec6SMatthias Ringwald                             sm_done_for_handle(sm_conn->sm_handle);
28113deb3ec6SMatthias Ringwald                             break;
28123deb3ec6SMatthias Ringwald                         case SM_PH2_W4_CONNECTION_ENCRYPTED:
28133deb3ec6SMatthias Ringwald                             if (sm_conn->sm_role){
28143deb3ec6SMatthias Ringwald                                 // slave
2815bbf8db22SMatthias Ringwald                                 if (setup->sm_use_secure_connections){
2816bbf8db22SMatthias Ringwald                                     sm_conn->sm_engine_state = SM_PH3_DISTRIBUTE_KEYS;
2817bbf8db22SMatthias Ringwald                                 } else {
28183deb3ec6SMatthias Ringwald                                     sm_conn->sm_engine_state = SM_PH3_GET_RANDOM;
2819bbf8db22SMatthias Ringwald                                 }
28203deb3ec6SMatthias Ringwald                             } else {
28213deb3ec6SMatthias Ringwald                                 // master
28223deb3ec6SMatthias Ringwald                                 if (sm_key_distribution_all_received(sm_conn)){
28233deb3ec6SMatthias Ringwald                                     // skip receiving keys as there are none
28243deb3ec6SMatthias Ringwald                                     sm_key_distribution_handle_all_received(sm_conn);
28253deb3ec6SMatthias Ringwald                                     sm_conn->sm_engine_state = SM_PH3_GET_RANDOM;
28263deb3ec6SMatthias Ringwald                                 } else {
28273deb3ec6SMatthias Ringwald                                     sm_conn->sm_engine_state = SM_PH3_RECEIVE_KEYS;
28283deb3ec6SMatthias Ringwald                                 }
28293deb3ec6SMatthias Ringwald                             }
28303deb3ec6SMatthias Ringwald                             break;
28313deb3ec6SMatthias Ringwald                         default:
28323deb3ec6SMatthias Ringwald                             break;
28333deb3ec6SMatthias Ringwald                     }
28343deb3ec6SMatthias Ringwald                     break;
28353deb3ec6SMatthias Ringwald 
28363deb3ec6SMatthias Ringwald                 case HCI_EVENT_ENCRYPTION_KEY_REFRESH_COMPLETE:
2837711e6c80SMatthias Ringwald                     con_handle = little_endian_read_16(packet, 3);
2838711e6c80SMatthias Ringwald                     sm_conn = sm_get_connection_for_handle(con_handle);
28393deb3ec6SMatthias Ringwald                     if (!sm_conn) break;
28403deb3ec6SMatthias Ringwald 
28413deb3ec6SMatthias Ringwald                     log_info("Encryption key refresh complete, key size %u", sm_conn->sm_actual_encryption_key_size);
28423deb3ec6SMatthias Ringwald                     log_info("event handler, state %u", sm_conn->sm_engine_state);
28433deb3ec6SMatthias Ringwald                     // continue if part of initial pairing
28443deb3ec6SMatthias Ringwald                     switch (sm_conn->sm_engine_state){
28453deb3ec6SMatthias Ringwald                         case SM_INITIATOR_PH0_W4_CONNECTION_ENCRYPTED:
28463deb3ec6SMatthias Ringwald                             sm_conn->sm_engine_state = SM_INITIATOR_CONNECTED;
28473deb3ec6SMatthias Ringwald                             sm_done_for_handle(sm_conn->sm_handle);
28483deb3ec6SMatthias Ringwald                             break;
28493deb3ec6SMatthias Ringwald                         case SM_PH2_W4_CONNECTION_ENCRYPTED:
28503deb3ec6SMatthias Ringwald                             if (sm_conn->sm_role){
28513deb3ec6SMatthias Ringwald                                 // slave
28523deb3ec6SMatthias Ringwald                                 sm_conn->sm_engine_state = SM_PH3_GET_RANDOM;
28533deb3ec6SMatthias Ringwald                             } else {
28543deb3ec6SMatthias Ringwald                                 // master
28553deb3ec6SMatthias Ringwald                                 sm_conn->sm_engine_state = SM_PH3_RECEIVE_KEYS;
28563deb3ec6SMatthias Ringwald                             }
28573deb3ec6SMatthias Ringwald                             break;
28583deb3ec6SMatthias Ringwald                         default:
28593deb3ec6SMatthias Ringwald                             break;
28603deb3ec6SMatthias Ringwald                     }
28613deb3ec6SMatthias Ringwald                     break;
28623deb3ec6SMatthias Ringwald 
28633deb3ec6SMatthias Ringwald 
28643deb3ec6SMatthias Ringwald                 case HCI_EVENT_DISCONNECTION_COMPLETE:
2865711e6c80SMatthias Ringwald                     con_handle = little_endian_read_16(packet, 3);
2866711e6c80SMatthias Ringwald                     sm_done_for_handle(con_handle);
2867711e6c80SMatthias Ringwald                     sm_conn = sm_get_connection_for_handle(con_handle);
28683deb3ec6SMatthias Ringwald                     if (!sm_conn) break;
28693deb3ec6SMatthias Ringwald 
28703deb3ec6SMatthias Ringwald                     // delete stored bonding on disconnect with authentication failure in ph0
28713deb3ec6SMatthias Ringwald                     if (sm_conn->sm_role == 0
28723deb3ec6SMatthias Ringwald                         && sm_conn->sm_engine_state == SM_INITIATOR_PH0_W4_CONNECTION_ENCRYPTED
28733deb3ec6SMatthias Ringwald                         && packet[2] == ERROR_CODE_AUTHENTICATION_FAILURE){
28743deb3ec6SMatthias Ringwald                         le_device_db_remove(sm_conn->sm_le_db_index);
28753deb3ec6SMatthias Ringwald                     }
28763deb3ec6SMatthias Ringwald 
28773deb3ec6SMatthias Ringwald                     sm_conn->sm_engine_state = SM_GENERAL_IDLE;
28783deb3ec6SMatthias Ringwald                     sm_conn->sm_handle = 0;
28793deb3ec6SMatthias Ringwald                     break;
28803deb3ec6SMatthias Ringwald 
28813deb3ec6SMatthias Ringwald 				case HCI_EVENT_COMMAND_COMPLETE:
2882073bd0faSMatthias Ringwald                     if (HCI_EVENT_IS_COMMAND_COMPLETE(packet, hci_le_encrypt)){
28833deb3ec6SMatthias Ringwald                         sm_handle_encryption_result(&packet[6]);
28843deb3ec6SMatthias Ringwald                         break;
28853deb3ec6SMatthias Ringwald                     }
2886073bd0faSMatthias Ringwald                     if (HCI_EVENT_IS_COMMAND_COMPLETE(packet, hci_le_rand)){
28873deb3ec6SMatthias Ringwald                         sm_handle_random_result(&packet[6]);
28883deb3ec6SMatthias Ringwald                         break;
28893deb3ec6SMatthias Ringwald                     }
289065b44ffdSMatthias Ringwald                     break;
289165b44ffdSMatthias Ringwald                 default:
289265b44ffdSMatthias Ringwald                     break;
28933deb3ec6SMatthias Ringwald 			}
289465b44ffdSMatthias Ringwald             break;
289565b44ffdSMatthias Ringwald         default:
289665b44ffdSMatthias Ringwald             break;
28973deb3ec6SMatthias Ringwald 	}
28983deb3ec6SMatthias Ringwald 
28993deb3ec6SMatthias Ringwald     sm_run();
29003deb3ec6SMatthias Ringwald }
29013deb3ec6SMatthias Ringwald 
29023deb3ec6SMatthias Ringwald static inline int sm_calc_actual_encryption_key_size(int other){
29033deb3ec6SMatthias Ringwald     if (other < sm_min_encryption_key_size) return 0;
29043deb3ec6SMatthias Ringwald     if (other < sm_max_encryption_key_size) return other;
29053deb3ec6SMatthias Ringwald     return sm_max_encryption_key_size;
29063deb3ec6SMatthias Ringwald }
29073deb3ec6SMatthias Ringwald 
2908945888f5SMatthias Ringwald 
2909945888f5SMatthias Ringwald static int sm_just_works_or_numeric_comparison(stk_generation_method_t method){
2910945888f5SMatthias Ringwald     switch (method){
2911945888f5SMatthias Ringwald         case JUST_WORKS:
2912945888f5SMatthias Ringwald         case NK_BOTH_INPUT:
2913945888f5SMatthias Ringwald             return 1;
2914945888f5SMatthias Ringwald         default:
2915945888f5SMatthias Ringwald             return 0;
2916945888f5SMatthias Ringwald     }
2917945888f5SMatthias Ringwald }
291807036a04SMatthias Ringwald // responder
2919945888f5SMatthias Ringwald 
2920688a08f9SMatthias Ringwald static int sm_passkey_used(stk_generation_method_t method){
2921688a08f9SMatthias Ringwald     switch (method){
2922688a08f9SMatthias Ringwald         case PK_RESP_INPUT:
2923688a08f9SMatthias Ringwald             return 1;
2924688a08f9SMatthias Ringwald         default:
2925688a08f9SMatthias Ringwald             return 0;
2926688a08f9SMatthias Ringwald     }
2927688a08f9SMatthias Ringwald }
2928688a08f9SMatthias Ringwald 
29293deb3ec6SMatthias Ringwald /**
29303deb3ec6SMatthias Ringwald  * @return ok
29313deb3ec6SMatthias Ringwald  */
29323deb3ec6SMatthias Ringwald static int sm_validate_stk_generation_method(void){
29333deb3ec6SMatthias Ringwald     // check if STK generation method is acceptable by client
29343deb3ec6SMatthias Ringwald     switch (setup->sm_stk_generation_method){
29353deb3ec6SMatthias Ringwald         case JUST_WORKS:
29363deb3ec6SMatthias Ringwald             return (sm_accepted_stk_generation_methods & SM_STK_GENERATION_METHOD_JUST_WORKS) != 0;
29373deb3ec6SMatthias Ringwald         case PK_RESP_INPUT:
29383deb3ec6SMatthias Ringwald         case PK_INIT_INPUT:
29393deb3ec6SMatthias Ringwald         case OK_BOTH_INPUT:
29403deb3ec6SMatthias Ringwald             return (sm_accepted_stk_generation_methods & SM_STK_GENERATION_METHOD_PASSKEY) != 0;
29413deb3ec6SMatthias Ringwald         case OOB:
29423deb3ec6SMatthias Ringwald             return (sm_accepted_stk_generation_methods & SM_STK_GENERATION_METHOD_OOB) != 0;
2943446a8c36SMatthias Ringwald         case NK_BOTH_INPUT:
2944b4343428SMatthias Ringwald             return (sm_accepted_stk_generation_methods & SM_STK_GENERATION_METHOD_NUMERIC_COMPARISON) != 0;
2945446a8c36SMatthias Ringwald             return 1;
29463deb3ec6SMatthias Ringwald         default:
29473deb3ec6SMatthias Ringwald             return 0;
29483deb3ec6SMatthias Ringwald     }
29493deb3ec6SMatthias Ringwald }
29503deb3ec6SMatthias Ringwald 
2951711e6c80SMatthias Ringwald static void sm_pdu_handler(uint8_t packet_type, hci_con_handle_t con_handle, uint8_t *packet, uint16_t size){
29523deb3ec6SMatthias Ringwald 
2953b170b20fSMatthias Ringwald     if (packet_type == HCI_EVENT_PACKET && packet[0] == L2CAP_EVENT_CAN_SEND_NOW){
2954b170b20fSMatthias Ringwald         sm_run();
2955b170b20fSMatthias Ringwald     }
2956b170b20fSMatthias Ringwald 
29573deb3ec6SMatthias Ringwald     if (packet_type != SM_DATA_PACKET) return;
29583deb3ec6SMatthias Ringwald 
2959711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
29603deb3ec6SMatthias Ringwald     if (!sm_conn) return;
29613deb3ec6SMatthias Ringwald 
29623deb3ec6SMatthias Ringwald     if (packet[0] == SM_CODE_PAIRING_FAILED){
29633deb3ec6SMatthias Ringwald         sm_conn->sm_engine_state = sm_conn->sm_role ? SM_RESPONDER_IDLE : SM_INITIATOR_CONNECTED;
29643deb3ec6SMatthias Ringwald         return;
29653deb3ec6SMatthias Ringwald     }
29663deb3ec6SMatthias Ringwald 
2967e03e489aSMatthias Ringwald     log_debug("sm_pdu_handler: state %u, pdu 0x%02x", sm_conn->sm_engine_state, packet[0]);
29683deb3ec6SMatthias Ringwald 
29693deb3ec6SMatthias Ringwald     int err;
29703deb3ec6SMatthias Ringwald 
29713deb3ec6SMatthias Ringwald     switch (sm_conn->sm_engine_state){
29723deb3ec6SMatthias Ringwald 
29733deb3ec6SMatthias Ringwald         // a sm timeout requries a new physical connection
29743deb3ec6SMatthias Ringwald         case SM_GENERAL_TIMEOUT:
29753deb3ec6SMatthias Ringwald             return;
29763deb3ec6SMatthias Ringwald 
29773deb3ec6SMatthias Ringwald         // Initiator
29783deb3ec6SMatthias Ringwald         case SM_INITIATOR_CONNECTED:
29793deb3ec6SMatthias Ringwald             if ((packet[0] != SM_CODE_SECURITY_REQUEST) || (sm_conn->sm_role)){
29803deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
29813deb3ec6SMatthias Ringwald                 break;
29823deb3ec6SMatthias Ringwald             }
29833deb3ec6SMatthias Ringwald             if (sm_conn->sm_irk_lookup_state == IRK_LOOKUP_FAILED){
29843deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
29853deb3ec6SMatthias Ringwald                 break;
29863deb3ec6SMatthias Ringwald             }
29873deb3ec6SMatthias Ringwald             if (sm_conn->sm_irk_lookup_state == IRK_LOOKUP_SUCCEEDED){
29883deb3ec6SMatthias Ringwald                 uint16_t ediv;
29893764b551SMatthias Ringwald                 sm_key_t ltk;
29903764b551SMatthias Ringwald                 le_device_db_encryption_get(sm_conn->sm_le_db_index, &ediv, NULL, ltk, NULL, NULL, NULL);
29913764b551SMatthias Ringwald                 if (!sm_is_null_key(ltk) || ediv){
29923deb3ec6SMatthias Ringwald                     log_info("sm: Setting up previous ltk/ediv/rand for device index %u", sm_conn->sm_le_db_index);
29933deb3ec6SMatthias Ringwald                     sm_conn->sm_engine_state = SM_INITIATOR_PH0_HAS_LTK;
29943deb3ec6SMatthias Ringwald                 } else {
29953deb3ec6SMatthias Ringwald                     sm_conn->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
29963deb3ec6SMatthias Ringwald                 }
29973deb3ec6SMatthias Ringwald                 break;
29983deb3ec6SMatthias Ringwald             }
29993deb3ec6SMatthias Ringwald             // otherwise, store security request
30003deb3ec6SMatthias Ringwald             sm_conn->sm_security_request_received = 1;
30013deb3ec6SMatthias Ringwald             break;
30023deb3ec6SMatthias Ringwald 
30033deb3ec6SMatthias Ringwald         case SM_INITIATOR_PH1_W4_PAIRING_RESPONSE:
30043deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_RESPONSE){
30053deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
30063deb3ec6SMatthias Ringwald                 break;
30073deb3ec6SMatthias Ringwald             }
30083deb3ec6SMatthias Ringwald             // store pairing request
30093deb3ec6SMatthias Ringwald             memcpy(&setup->sm_s_pres, packet, sizeof(sm_pairing_packet_t));
30103deb3ec6SMatthias Ringwald             err = sm_stk_generation_init(sm_conn);
30113deb3ec6SMatthias Ringwald             if (err){
30123deb3ec6SMatthias Ringwald                 setup->sm_pairing_failed_reason = err;
30133deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_GENERAL_SEND_PAIRING_FAILED;
30143deb3ec6SMatthias Ringwald                 break;
30153deb3ec6SMatthias Ringwald             }
3016136d331aSMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
3017136d331aSMatthias Ringwald             if (setup->sm_use_secure_connections){
30188cba5ca3SMatthias Ringwald                 // SC Numeric Comparison will trigger user response after public keys & nonces have been exchanged
30198cba5ca3SMatthias Ringwald                 if (setup->sm_stk_generation_method == JUST_WORKS){
3020136d331aSMatthias Ringwald                     sm_conn->sm_engine_state = SM_PH1_W4_USER_RESPONSE;
3021136d331aSMatthias Ringwald                     sm_trigger_user_response(sm_conn);
3022136d331aSMatthias Ringwald                     if (setup->sm_user_response == SM_USER_RESPONSE_IDLE){
3023c6b7cbd9SMatthias Ringwald                         sm_conn->sm_engine_state = SM_SC_SEND_PUBLIC_KEY_COMMAND;
3024136d331aSMatthias Ringwald                     }
30258cba5ca3SMatthias Ringwald                 } else {
3026c6b7cbd9SMatthias Ringwald                     sm_conn->sm_engine_state = SM_SC_SEND_PUBLIC_KEY_COMMAND;
30278cba5ca3SMatthias Ringwald                 }
3028136d331aSMatthias Ringwald                 break;
3029136d331aSMatthias Ringwald             }
3030136d331aSMatthias Ringwald #endif
30313deb3ec6SMatthias Ringwald             // generate random number first, if we need to show passkey
30323deb3ec6SMatthias Ringwald             if (setup->sm_stk_generation_method == PK_RESP_INPUT){
30333deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_PH2_GET_RANDOM_TK;
30343deb3ec6SMatthias Ringwald                 break;
30353deb3ec6SMatthias Ringwald             }
30363deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH1_W4_USER_RESPONSE;
30373deb3ec6SMatthias Ringwald             sm_trigger_user_response(sm_conn);
30383deb3ec6SMatthias Ringwald             // response_idle == nothing <--> sm_trigger_user_response() did not require response
30393deb3ec6SMatthias Ringwald             if (setup->sm_user_response == SM_USER_RESPONSE_IDLE){
30403deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_PH2_C1_GET_RANDOM_A;
30413deb3ec6SMatthias Ringwald             }
30423deb3ec6SMatthias Ringwald             break;
30433deb3ec6SMatthias Ringwald 
30443deb3ec6SMatthias Ringwald         case SM_INITIATOR_PH2_W4_PAIRING_CONFIRM:
30453deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_CONFIRM){
30463deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
30473deb3ec6SMatthias Ringwald                 break;
30483deb3ec6SMatthias Ringwald             }
30493deb3ec6SMatthias Ringwald 
30503deb3ec6SMatthias Ringwald             // store s_confirm
30519c80e4ccSMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_confirm);
30523deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH2_SEND_PAIRING_RANDOM;
30533deb3ec6SMatthias Ringwald             break;
30543deb3ec6SMatthias Ringwald 
30553deb3ec6SMatthias Ringwald         case SM_INITIATOR_PH2_W4_PAIRING_RANDOM:
30563deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_RANDOM){
30573deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
30583deb3ec6SMatthias Ringwald                 break;;
30593deb3ec6SMatthias Ringwald             }
30603deb3ec6SMatthias Ringwald 
30613deb3ec6SMatthias Ringwald             // received random value
30629c80e4ccSMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_random);
30633deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH2_C1_GET_ENC_C;
30643deb3ec6SMatthias Ringwald             break;
30653deb3ec6SMatthias Ringwald 
30663deb3ec6SMatthias Ringwald         // Responder
30673deb3ec6SMatthias Ringwald         case SM_RESPONDER_IDLE:
30683deb3ec6SMatthias Ringwald         case SM_RESPONDER_SEND_SECURITY_REQUEST:
30693deb3ec6SMatthias Ringwald         case SM_RESPONDER_PH1_W4_PAIRING_REQUEST:
30703deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_REQUEST){
30713deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
30723deb3ec6SMatthias Ringwald                 break;;
30733deb3ec6SMatthias Ringwald             }
30743deb3ec6SMatthias Ringwald 
30753deb3ec6SMatthias Ringwald             // store pairing request
30763deb3ec6SMatthias Ringwald             memcpy(&sm_conn->sm_m_preq, packet, sizeof(sm_pairing_packet_t));
30773deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_RESPONDER_PH1_PAIRING_REQUEST_RECEIVED;
30783deb3ec6SMatthias Ringwald             break;
30793deb3ec6SMatthias Ringwald 
308027c32905SMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
3081c6b7cbd9SMatthias Ringwald         case SM_SC_W4_PUBLIC_KEY_COMMAND:
308227c32905SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_PUBLIC_KEY){
308327c32905SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
308427c32905SMatthias Ringwald                 break;
308527c32905SMatthias Ringwald             }
3086bccf5e67SMatthias Ringwald 
3087e53be891SMatthias Ringwald             // store public key for DH Key calculation
3088e53be891SMatthias Ringwald             reverse_256(&packet[01], setup->sm_peer_qx);
3089e53be891SMatthias Ringwald             reverse_256(&packet[33], setup->sm_peer_qy);
3090bccf5e67SMatthias Ringwald 
3091bccf5e67SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
3092bccf5e67SMatthias Ringwald             // validate public key
3093bccf5e67SMatthias Ringwald             mbedtls_ecp_point Q;
3094bccf5e67SMatthias Ringwald             mbedtls_ecp_point_init( &Q );
3095bccf5e67SMatthias Ringwald             mbedtls_mpi_read_binary(&Q.X, setup->sm_peer_qx, 32);
3096bccf5e67SMatthias Ringwald             mbedtls_mpi_read_binary(&Q.Y, setup->sm_peer_qy, 32);
3097bccf5e67SMatthias Ringwald             mbedtls_mpi_read_string(&Q.Z, 16, "1" );
3098e722521aSMatthias Ringwald             err = mbedtls_ecp_check_pubkey(&mbedtls_ec_group, &Q);
3099891bb64aSMatthias Ringwald             mbedtls_ecp_point_free( & Q);
3100bccf5e67SMatthias Ringwald             if (err){
3101bccf5e67SMatthias Ringwald                 log_error("sm: peer public key invalid %x", err);
3102bccf5e67SMatthias Ringwald                 // uses "unspecified reason", there is no "public key invalid" error code
3103bccf5e67SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
3104bccf5e67SMatthias Ringwald                 break;
3105bccf5e67SMatthias Ringwald             }
3106891bb64aSMatthias Ringwald 
3107bccf5e67SMatthias Ringwald #endif
3108136d331aSMatthias Ringwald             if (sm_conn->sm_role){
3109136d331aSMatthias Ringwald                 // responder
3110c6b7cbd9SMatthias Ringwald                 sm_conn->sm_engine_state = SM_SC_SEND_PUBLIC_KEY_COMMAND;
3111136d331aSMatthias Ringwald             } else {
3112136d331aSMatthias Ringwald                 // initiator
3113a1e31e9cSMatthias Ringwald                 // stk generation method
3114a1e31e9cSMatthias Ringwald                 // passkey entry: notify app to show passkey or to request passkey
3115a1e31e9cSMatthias Ringwald                 switch (setup->sm_stk_generation_method){
3116a1e31e9cSMatthias Ringwald                     case JUST_WORKS:
3117a1e31e9cSMatthias Ringwald                     case NK_BOTH_INPUT:
3118c6b7cbd9SMatthias Ringwald                         sm_conn->sm_engine_state = SM_SC_W4_CONFIRMATION;
3119a1e31e9cSMatthias Ringwald                         break;
3120a1e31e9cSMatthias Ringwald                     case PK_RESP_INPUT:
312107036a04SMatthias Ringwald                         sm_sc_start_calculating_local_confirm(sm_conn);
312207036a04SMatthias Ringwald                         break;
312307036a04SMatthias Ringwald                     case PK_INIT_INPUT:
3124a1e31e9cSMatthias Ringwald                     case OK_BOTH_INPUT:
312507036a04SMatthias Ringwald                         if (setup->sm_user_response != SM_USER_RESPONSE_PASSKEY){
312607036a04SMatthias Ringwald                             sm_conn->sm_engine_state = SM_SC_W4_USER_RESPONSE;
312707036a04SMatthias Ringwald                             break;
312807036a04SMatthias Ringwald                         }
3129b35a3de2SMatthias Ringwald                         sm_sc_start_calculating_local_confirm(sm_conn);
3130a1e31e9cSMatthias Ringwald                         break;
3131a1e31e9cSMatthias Ringwald                     case OOB:
3132a1e31e9cSMatthias Ringwald                         // TODO: implement SC OOB
3133a1e31e9cSMatthias Ringwald                         break;
3134a1e31e9cSMatthias Ringwald                 }
3135136d331aSMatthias Ringwald             }
313627c32905SMatthias Ringwald             break;
3137e53be891SMatthias Ringwald 
3138c6b7cbd9SMatthias Ringwald         case SM_SC_W4_CONFIRMATION:
313945a61d50SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_CONFIRM){
314045a61d50SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
314145a61d50SMatthias Ringwald                 break;
314245a61d50SMatthias Ringwald             }
314345a61d50SMatthias Ringwald             // received confirm value
314445a61d50SMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_confirm);
314545a61d50SMatthias Ringwald 
314645a61d50SMatthias Ringwald             if (sm_conn->sm_role){
314745a61d50SMatthias Ringwald                 // responder
314807036a04SMatthias Ringwald                 if (sm_passkey_used(setup->sm_stk_generation_method)){
314907036a04SMatthias Ringwald                     if (setup->sm_user_response != SM_USER_RESPONSE_PASSKEY){
315007036a04SMatthias Ringwald                         // still waiting for passkey
315107036a04SMatthias Ringwald                         sm_conn->sm_engine_state = SM_SC_W4_USER_RESPONSE;
315207036a04SMatthias Ringwald                         break;
315307036a04SMatthias Ringwald                     }
315407036a04SMatthias Ringwald                 }
3155b35a3de2SMatthias Ringwald                 sm_sc_start_calculating_local_confirm(sm_conn);
315645a61d50SMatthias Ringwald             } else {
315745a61d50SMatthias Ringwald                 // initiator
3158945888f5SMatthias Ringwald                 if (sm_just_works_or_numeric_comparison(setup->sm_stk_generation_method)){
3159f1c1783eSMatthias Ringwald                     sm_conn->sm_engine_state = SM_SC_W2_GET_RANDOM_A;
3160f1c1783eSMatthias Ringwald                 } else {
3161c6b7cbd9SMatthias Ringwald                     sm_conn->sm_engine_state = SM_SC_SEND_PAIRING_RANDOM;
316245a61d50SMatthias Ringwald                 }
3163f1c1783eSMatthias Ringwald             }
316445a61d50SMatthias Ringwald             break;
316545a61d50SMatthias Ringwald 
3166c6b7cbd9SMatthias Ringwald         case SM_SC_W4_PAIRING_RANDOM:
3167e53be891SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_RANDOM){
3168e53be891SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
3169136d331aSMatthias Ringwald                 break;
3170e53be891SMatthias Ringwald             }
3171e53be891SMatthias Ringwald 
3172e53be891SMatthias Ringwald             // received random value
3173e53be891SMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_nonce);
3174e53be891SMatthias Ringwald 
31755a293e6eSMatthias Ringwald             // validate confirm value if Cb = f4(Pkb, Pka, Nb, z)
31765a293e6eSMatthias Ringwald             // only check for JUST WORK/NC in initiator role AND passkey entry
3177688a08f9SMatthias Ringwald             int passkey_entry = sm_passkey_used(setup->sm_stk_generation_method);
31785a293e6eSMatthias Ringwald             if (sm_conn->sm_role || passkey_entry) {
3179688a08f9SMatthias Ringwald                  sm_conn->sm_engine_state = SM_SC_W2_CMAC_FOR_CHECK_CONFIRMATION;
31805a293e6eSMatthias Ringwald             }
31816f52a196SMatthias Ringwald 
3182688a08f9SMatthias Ringwald             sm_sc_state_after_receiving_random(sm_conn);
3183e53be891SMatthias Ringwald             break;
3184e53be891SMatthias Ringwald 
3185901c000fSMatthias Ringwald         case SM_SC_W2_CALCULATE_G2:
3186901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_G2:
3187901c000fSMatthias Ringwald         case SM_SC_W2_CALCULATE_F5_SALT:
3188901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_SALT:
3189901c000fSMatthias Ringwald         case SM_SC_W2_CALCULATE_F5_MACKEY:
3190901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_MACKEY:
3191901c000fSMatthias Ringwald         case SM_SC_W2_CALCULATE_F5_LTK:
3192901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F5_LTK:
3193901c000fSMatthias Ringwald         case SM_SC_W2_CALCULATE_F6_FOR_DHKEY_CHECK:
3194c6b7cbd9SMatthias Ringwald         case SM_SC_W4_DHKEY_CHECK_COMMAND:
3195901c000fSMatthias Ringwald         case SM_SC_W4_CALCULATE_F6_FOR_DHKEY_CHECK:
3196e53be891SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_DHKEY_CHECK){
3197e53be891SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
3198e53be891SMatthias Ringwald                 break;
3199e53be891SMatthias Ringwald             }
3200e53be891SMatthias Ringwald             // store DHKey Check
3201901c000fSMatthias Ringwald             setup->sm_state_vars |= SM_STATE_VAR_DHKEY_COMMAND_RECEIVED;
3202e53be891SMatthias Ringwald             reverse_128(&packet[01], setup->sm_peer_dhkey_check);
3203446a8c36SMatthias Ringwald 
3204901c000fSMatthias Ringwald             // have we been only waiting for dhkey check command?
3205901c000fSMatthias Ringwald             if (sm_conn->sm_engine_state == SM_SC_W4_DHKEY_CHECK_COMMAND){
3206019005a0SMatthias Ringwald                 sm_conn->sm_engine_state = SM_SC_W2_CALCULATE_F6_TO_VERIFY_DHKEY_CHECK;
3207bd57ffebSMatthias Ringwald             }
3208bd57ffebSMatthias Ringwald             break;
320927c32905SMatthias Ringwald #endif
321027c32905SMatthias Ringwald 
32113deb3ec6SMatthias Ringwald         case SM_RESPONDER_PH1_W4_PAIRING_CONFIRM:
32123deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_CONFIRM){
32133deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
321427c32905SMatthias Ringwald                 break;
32153deb3ec6SMatthias Ringwald             }
32163deb3ec6SMatthias Ringwald 
32173deb3ec6SMatthias Ringwald             // received confirm value
32189c80e4ccSMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_confirm);
32193deb3ec6SMatthias Ringwald 
32203deb3ec6SMatthias Ringwald             // notify client to hide shown passkey
32213deb3ec6SMatthias Ringwald             if (setup->sm_stk_generation_method == PK_INIT_INPUT){
32225611a760SMatthias Ringwald                 sm_notify_client_base(SM_EVENT_PASSKEY_DISPLAY_CANCEL, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address);
32233deb3ec6SMatthias Ringwald             }
32243deb3ec6SMatthias Ringwald 
32253deb3ec6SMatthias Ringwald             // handle user cancel pairing?
32263deb3ec6SMatthias Ringwald             if (setup->sm_user_response == SM_USER_RESPONSE_DECLINE){
32273deb3ec6SMatthias Ringwald                 setup->sm_pairing_failed_reason = SM_REASON_PASSKEYT_ENTRY_FAILED;
32283deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_GENERAL_SEND_PAIRING_FAILED;
32293deb3ec6SMatthias Ringwald                 break;
32303deb3ec6SMatthias Ringwald             }
32313deb3ec6SMatthias Ringwald 
32323deb3ec6SMatthias Ringwald             // wait for user action?
32333deb3ec6SMatthias Ringwald             if (setup->sm_user_response == SM_USER_RESPONSE_PENDING){
32343deb3ec6SMatthias Ringwald                 sm_conn->sm_engine_state = SM_PH1_W4_USER_RESPONSE;
32353deb3ec6SMatthias Ringwald                 break;
32363deb3ec6SMatthias Ringwald             }
32373deb3ec6SMatthias Ringwald 
32383deb3ec6SMatthias Ringwald             // calculate and send local_confirm
32393deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH2_C1_GET_RANDOM_A;
32403deb3ec6SMatthias Ringwald             break;
32413deb3ec6SMatthias Ringwald 
32423deb3ec6SMatthias Ringwald         case SM_RESPONDER_PH2_W4_PAIRING_RANDOM:
32433deb3ec6SMatthias Ringwald             if (packet[0] != SM_CODE_PAIRING_RANDOM){
32443deb3ec6SMatthias Ringwald                 sm_pdu_received_in_wrong_state(sm_conn);
32453deb3ec6SMatthias Ringwald                 break;;
32463deb3ec6SMatthias Ringwald             }
32473deb3ec6SMatthias Ringwald 
32483deb3ec6SMatthias Ringwald             // received random value
32499c80e4ccSMatthias Ringwald             reverse_128(&packet[1], setup->sm_peer_random);
32503deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH2_C1_GET_ENC_C;
32513deb3ec6SMatthias Ringwald             break;
32523deb3ec6SMatthias Ringwald 
32533deb3ec6SMatthias Ringwald         case SM_PH3_RECEIVE_KEYS:
32543deb3ec6SMatthias Ringwald             switch(packet[0]){
32553deb3ec6SMatthias Ringwald                 case SM_CODE_ENCRYPTION_INFORMATION:
32563deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_received_set |= SM_KEYDIST_FLAG_ENCRYPTION_INFORMATION;
32579c80e4ccSMatthias Ringwald                     reverse_128(&packet[1], setup->sm_peer_ltk);
32583deb3ec6SMatthias Ringwald                     break;
32593deb3ec6SMatthias Ringwald 
32603deb3ec6SMatthias Ringwald                 case SM_CODE_MASTER_IDENTIFICATION:
32613deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_received_set |= SM_KEYDIST_FLAG_MASTER_IDENTIFICATION;
3262f8fbdce0SMatthias Ringwald                     setup->sm_peer_ediv = little_endian_read_16(packet, 1);
32639c80e4ccSMatthias Ringwald                     reverse_64(&packet[3], setup->sm_peer_rand);
32643deb3ec6SMatthias Ringwald                     break;
32653deb3ec6SMatthias Ringwald 
32663deb3ec6SMatthias Ringwald                 case SM_CODE_IDENTITY_INFORMATION:
32673deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_received_set |= SM_KEYDIST_FLAG_IDENTITY_INFORMATION;
32689c80e4ccSMatthias Ringwald                     reverse_128(&packet[1], setup->sm_peer_irk);
32693deb3ec6SMatthias Ringwald                     break;
32703deb3ec6SMatthias Ringwald 
32713deb3ec6SMatthias Ringwald                 case SM_CODE_IDENTITY_ADDRESS_INFORMATION:
32723deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_received_set |= SM_KEYDIST_FLAG_IDENTITY_ADDRESS_INFORMATION;
32733deb3ec6SMatthias Ringwald                     setup->sm_peer_addr_type = packet[1];
3274724d70a2SMatthias Ringwald                     reverse_bd_addr(&packet[2], setup->sm_peer_address);
32753deb3ec6SMatthias Ringwald                     break;
32763deb3ec6SMatthias Ringwald 
32773deb3ec6SMatthias Ringwald                 case SM_CODE_SIGNING_INFORMATION:
32783deb3ec6SMatthias Ringwald                     setup->sm_key_distribution_received_set |= SM_KEYDIST_FLAG_SIGNING_IDENTIFICATION;
32799c80e4ccSMatthias Ringwald                     reverse_128(&packet[1], setup->sm_peer_csrk);
32803deb3ec6SMatthias Ringwald                     break;
32813deb3ec6SMatthias Ringwald                 default:
32823deb3ec6SMatthias Ringwald                     // Unexpected PDU
32833deb3ec6SMatthias Ringwald                     log_info("Unexpected PDU %u in SM_PH3_RECEIVE_KEYS", packet[0]);
32843deb3ec6SMatthias Ringwald                     break;
32853deb3ec6SMatthias Ringwald             }
32863deb3ec6SMatthias Ringwald             // done with key distribution?
32873deb3ec6SMatthias Ringwald             if (sm_key_distribution_all_received(sm_conn)){
32883deb3ec6SMatthias Ringwald 
32893deb3ec6SMatthias Ringwald                 sm_key_distribution_handle_all_received(sm_conn);
32903deb3ec6SMatthias Ringwald 
32913deb3ec6SMatthias Ringwald                 if (sm_conn->sm_role){
32923deb3ec6SMatthias Ringwald                     sm_conn->sm_engine_state = SM_RESPONDER_IDLE;
32933deb3ec6SMatthias Ringwald                     sm_done_for_handle(sm_conn->sm_handle);
32943deb3ec6SMatthias Ringwald                 } else {
3295625f00b2SMatthias Ringwald                     if (setup->sm_use_secure_connections){
3296625f00b2SMatthias Ringwald                         sm_conn->sm_engine_state = SM_PH3_DISTRIBUTE_KEYS;
3297bbf8db22SMatthias Ringwald                     } else {
3298bbf8db22SMatthias Ringwald                         sm_conn->sm_engine_state = SM_PH3_GET_RANDOM;
3299625f00b2SMatthias Ringwald                     }
33003deb3ec6SMatthias Ringwald                 }
33013deb3ec6SMatthias Ringwald             }
33023deb3ec6SMatthias Ringwald             break;
33033deb3ec6SMatthias Ringwald         default:
33043deb3ec6SMatthias Ringwald             // Unexpected PDU
33053deb3ec6SMatthias Ringwald             log_info("Unexpected PDU %u in state %u", packet[0], sm_conn->sm_engine_state);
33063deb3ec6SMatthias Ringwald             break;
33073deb3ec6SMatthias Ringwald     }
33083deb3ec6SMatthias Ringwald 
33093deb3ec6SMatthias Ringwald     // try to send preparared packet
33103deb3ec6SMatthias Ringwald     sm_run();
33113deb3ec6SMatthias Ringwald }
33123deb3ec6SMatthias Ringwald 
33133deb3ec6SMatthias Ringwald // Security Manager Client API
33143deb3ec6SMatthias Ringwald void sm_register_oob_data_callback( int (*get_oob_data_callback)(uint8_t addres_type, bd_addr_t addr, uint8_t * oob_data)){
33153deb3ec6SMatthias Ringwald     sm_get_oob_data = get_oob_data_callback;
33163deb3ec6SMatthias Ringwald }
33173deb3ec6SMatthias Ringwald 
331889a78d34SMatthias Ringwald void sm_add_event_handler(btstack_packet_callback_registration_t * callback_handler){
331989a78d34SMatthias Ringwald     btstack_linked_list_add_tail(&sm_event_handlers, (btstack_linked_item_t*) callback_handler);
332089a78d34SMatthias Ringwald }
332189a78d34SMatthias Ringwald 
33223deb3ec6SMatthias Ringwald void sm_set_accepted_stk_generation_methods(uint8_t accepted_stk_generation_methods){
33233deb3ec6SMatthias Ringwald     sm_accepted_stk_generation_methods = accepted_stk_generation_methods;
33243deb3ec6SMatthias Ringwald }
33253deb3ec6SMatthias Ringwald 
33263deb3ec6SMatthias Ringwald void sm_set_encryption_key_size_range(uint8_t min_size, uint8_t max_size){
33273deb3ec6SMatthias Ringwald 	sm_min_encryption_key_size = min_size;
33283deb3ec6SMatthias Ringwald 	sm_max_encryption_key_size = max_size;
33293deb3ec6SMatthias Ringwald }
33303deb3ec6SMatthias Ringwald 
33313deb3ec6SMatthias Ringwald void sm_set_authentication_requirements(uint8_t auth_req){
33323deb3ec6SMatthias Ringwald     sm_auth_req = auth_req;
33333deb3ec6SMatthias Ringwald }
33343deb3ec6SMatthias Ringwald 
33353deb3ec6SMatthias Ringwald void sm_set_io_capabilities(io_capability_t io_capability){
33363deb3ec6SMatthias Ringwald     sm_io_capabilities = io_capability;
33373deb3ec6SMatthias Ringwald }
33383deb3ec6SMatthias Ringwald 
33393deb3ec6SMatthias Ringwald void sm_set_request_security(int enable){
33403deb3ec6SMatthias Ringwald     sm_slave_request_security = enable;
33413deb3ec6SMatthias Ringwald }
33423deb3ec6SMatthias Ringwald 
33433deb3ec6SMatthias Ringwald void sm_set_er(sm_key_t er){
33443deb3ec6SMatthias Ringwald     memcpy(sm_persistent_er, er, 16);
33453deb3ec6SMatthias Ringwald }
33463deb3ec6SMatthias Ringwald 
33473deb3ec6SMatthias Ringwald void sm_set_ir(sm_key_t ir){
33483deb3ec6SMatthias Ringwald     memcpy(sm_persistent_ir, ir, 16);
33493deb3ec6SMatthias Ringwald }
33503deb3ec6SMatthias Ringwald 
33513deb3ec6SMatthias Ringwald // Testing support only
33523deb3ec6SMatthias Ringwald void sm_test_set_irk(sm_key_t irk){
33533deb3ec6SMatthias Ringwald     memcpy(sm_persistent_irk, irk, 16);
33543deb3ec6SMatthias Ringwald     sm_persistent_irk_ready = 1;
33553deb3ec6SMatthias Ringwald }
33563deb3ec6SMatthias Ringwald 
33573deb3ec6SMatthias Ringwald void sm_test_use_fixed_local_csrk(void){
33583deb3ec6SMatthias Ringwald     test_use_fixed_local_csrk = 1;
33593deb3ec6SMatthias Ringwald }
33603deb3ec6SMatthias Ringwald 
33613deb3ec6SMatthias Ringwald void sm_init(void){
33623deb3ec6SMatthias Ringwald     // set some (BTstack default) ER and IR
33633deb3ec6SMatthias Ringwald     int i;
33643deb3ec6SMatthias Ringwald     sm_key_t er;
33653deb3ec6SMatthias Ringwald     sm_key_t ir;
33663deb3ec6SMatthias Ringwald     for (i=0;i<16;i++){
33673deb3ec6SMatthias Ringwald         er[i] = 0x30 + i;
33683deb3ec6SMatthias Ringwald         ir[i] = 0x90 + i;
33693deb3ec6SMatthias Ringwald     }
33703deb3ec6SMatthias Ringwald     sm_set_er(er);
33713deb3ec6SMatthias Ringwald     sm_set_ir(ir);
33723deb3ec6SMatthias Ringwald     // defaults
33733deb3ec6SMatthias Ringwald     sm_accepted_stk_generation_methods = SM_STK_GENERATION_METHOD_JUST_WORKS
33743deb3ec6SMatthias Ringwald                                        | SM_STK_GENERATION_METHOD_OOB
3375b4343428SMatthias Ringwald                                        | SM_STK_GENERATION_METHOD_PASSKEY
3376b4343428SMatthias Ringwald                                        | SM_STK_GENERATION_METHOD_NUMERIC_COMPARISON;
3377b4343428SMatthias Ringwald 
33783deb3ec6SMatthias Ringwald     sm_max_encryption_key_size = 16;
33793deb3ec6SMatthias Ringwald     sm_min_encryption_key_size = 7;
33803deb3ec6SMatthias Ringwald 
33813deb3ec6SMatthias Ringwald     sm_cmac_state  = CMAC_IDLE;
33823deb3ec6SMatthias Ringwald     dkg_state = DKG_W4_WORKING;
33833deb3ec6SMatthias Ringwald     rau_state = RAU_W4_WORKING;
33843deb3ec6SMatthias Ringwald     sm_aes128_state = SM_AES128_IDLE;
33853deb3ec6SMatthias Ringwald     sm_address_resolution_test = -1;    // no private address to resolve yet
33863deb3ec6SMatthias Ringwald     sm_address_resolution_ah_calculation_active = 0;
33873deb3ec6SMatthias Ringwald     sm_address_resolution_mode = ADDRESS_RESOLUTION_IDLE;
33883deb3ec6SMatthias Ringwald     sm_address_resolution_general_queue = NULL;
33893deb3ec6SMatthias Ringwald 
33903deb3ec6SMatthias Ringwald     gap_random_adress_update_period = 15 * 60 * 1000L;
33913deb3ec6SMatthias Ringwald 
33923deb3ec6SMatthias Ringwald     sm_active_connection = 0;
33933deb3ec6SMatthias Ringwald 
33943deb3ec6SMatthias Ringwald     test_use_fixed_local_csrk = 0;
33953deb3ec6SMatthias Ringwald 
3396e03e489aSMatthias Ringwald     // register for HCI Events from HCI
3397e03e489aSMatthias Ringwald     hci_event_callback_registration.callback = &sm_event_packet_handler;
3398e03e489aSMatthias Ringwald     hci_add_event_handler(&hci_event_callback_registration);
3399e03e489aSMatthias Ringwald 
3400b170b20fSMatthias Ringwald     // and L2CAP PDUs + L2CAP_EVENT_CAN_SEND_NOW
3401e03e489aSMatthias Ringwald     l2cap_register_fixed_channel(sm_pdu_handler, L2CAP_CID_SECURITY_MANAGER_PROTOCOL);
340227c32905SMatthias Ringwald 
340327c32905SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
34047df18c15SMatthias Ringwald     ec_key_generation_state = EC_KEY_GENERATION_IDLE;
3405e722521aSMatthias Ringwald     mbedtls_ecp_group_init(&mbedtls_ec_group);
3406e722521aSMatthias Ringwald     mbedtls_ecp_group_load(&mbedtls_ec_group, MBEDTLS_ECP_DP_SECP256R1);
3407*68437d83SMatthias Ringwald #ifndef HAVE_MALLOC
3408*68437d83SMatthias Ringwald     sm_mbedtls_allocator_init(mbedtls_memory_buffer, sizeof(mbedtls_memory_buffer));
3409d3bd9600SMatthias Ringwald #endif
3410*68437d83SMatthias Ringwald 
3411*68437d83SMatthias Ringwald #if 0
3412e722521aSMatthias Ringwald     // test
3413e722521aSMatthias Ringwald     printf("test dhkey check\n");
3414e722521aSMatthias Ringwald     sm_key256_t dhkey;
3415e722521aSMatthias Ringwald     memcpy(setup->sm_peer_qx, ec_qx, 32);
3416e722521aSMatthias Ringwald     memcpy(setup->sm_peer_qy, ec_qy, 32);
3417e722521aSMatthias Ringwald     sm_sc_calculate_dhkey(dhkey);
34187df18c15SMatthias Ringwald #endif
3419*68437d83SMatthias Ringwald #endif
34207df18c15SMatthias Ringwald }
34217df18c15SMatthias Ringwald 
34227df18c15SMatthias Ringwald void sm_test_use_fixed_ec_keypair(void){
34237df18c15SMatthias Ringwald     test_use_fixed_ec_keypair = 1;
34247df18c15SMatthias Ringwald #ifdef USE_MBEDTLS_FOR_ECDH
34257df18c15SMatthias Ringwald     // use test keypair from spec
3426e722521aSMatthias Ringwald     mbedtls_mpi x;
3427e722521aSMatthias Ringwald     mbedtls_mpi_read_string( &x, 16, "3f49f6d4a3c55f3874c9b3e3d2103f504aff607beb40b7995899b8a6cd3c1abd");
3428e722521aSMatthias Ringwald     mbedtls_mpi_write_binary(&x, ec_qx, 16);
3429e722521aSMatthias Ringwald     mbedtls_mpi_read_string( &x, 16, "20b003d2f297be2c5e2c83a7e9f9a5b9eff49111acf4fddbcc0301480e359de6");
3430e722521aSMatthias Ringwald     mbedtls_mpi_write_binary(&x, ec_qy, 16);
3431e722521aSMatthias Ringwald     mbedtls_mpi_read_string( &x, 16, "dc809c49652aeb6d63329abf5a52155c766345c28fed3024741c8ed01589d28b");
3432e722521aSMatthias Ringwald     mbedtls_mpi_write_binary(&x, ec_d, 16);
343327c32905SMatthias Ringwald #endif
34343deb3ec6SMatthias Ringwald }
34353deb3ec6SMatthias Ringwald 
3436711e6c80SMatthias Ringwald static sm_connection_t * sm_get_connection_for_handle(hci_con_handle_t con_handle){
3437711e6c80SMatthias Ringwald     hci_connection_t * hci_con = hci_connection_for_handle(con_handle);
34383deb3ec6SMatthias Ringwald     if (!hci_con) return NULL;
34393deb3ec6SMatthias Ringwald     return &hci_con->sm_connection;
34403deb3ec6SMatthias Ringwald }
34413deb3ec6SMatthias Ringwald 
34423deb3ec6SMatthias Ringwald // @returns 0 if not encrypted, 7-16 otherwise
3443711e6c80SMatthias Ringwald int sm_encryption_key_size(hci_con_handle_t con_handle){
3444711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
34453deb3ec6SMatthias Ringwald     if (!sm_conn) return 0;     // wrong connection
34463deb3ec6SMatthias Ringwald     if (!sm_conn->sm_connection_encrypted) return 0;
34473deb3ec6SMatthias Ringwald     return sm_conn->sm_actual_encryption_key_size;
34483deb3ec6SMatthias Ringwald }
34493deb3ec6SMatthias Ringwald 
3450711e6c80SMatthias Ringwald int sm_authenticated(hci_con_handle_t con_handle){
3451711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
34523deb3ec6SMatthias Ringwald     if (!sm_conn) return 0;     // wrong connection
34533deb3ec6SMatthias Ringwald     if (!sm_conn->sm_connection_encrypted) return 0; // unencrypted connection cannot be authenticated
34543deb3ec6SMatthias Ringwald     return sm_conn->sm_connection_authenticated;
34553deb3ec6SMatthias Ringwald }
34563deb3ec6SMatthias Ringwald 
3457711e6c80SMatthias Ringwald authorization_state_t sm_authorization_state(hci_con_handle_t con_handle){
3458711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
34593deb3ec6SMatthias Ringwald     if (!sm_conn) return AUTHORIZATION_UNKNOWN;     // wrong connection
34603deb3ec6SMatthias Ringwald     if (!sm_conn->sm_connection_encrypted)               return AUTHORIZATION_UNKNOWN; // unencrypted connection cannot be authorized
34613deb3ec6SMatthias Ringwald     if (!sm_conn->sm_connection_authenticated)           return AUTHORIZATION_UNKNOWN; // unauthenticatd connection cannot be authorized
34623deb3ec6SMatthias Ringwald     return sm_conn->sm_connection_authorization_state;
34633deb3ec6SMatthias Ringwald }
34643deb3ec6SMatthias Ringwald 
34653deb3ec6SMatthias Ringwald static void sm_send_security_request_for_connection(sm_connection_t * sm_conn){
34663deb3ec6SMatthias Ringwald     switch (sm_conn->sm_engine_state){
34673deb3ec6SMatthias Ringwald         case SM_GENERAL_IDLE:
34683deb3ec6SMatthias Ringwald         case SM_RESPONDER_IDLE:
34693deb3ec6SMatthias Ringwald             sm_conn->sm_engine_state = SM_RESPONDER_SEND_SECURITY_REQUEST;
34703deb3ec6SMatthias Ringwald             sm_run();
34713deb3ec6SMatthias Ringwald             break;
34723deb3ec6SMatthias Ringwald         default:
34733deb3ec6SMatthias Ringwald             break;
34743deb3ec6SMatthias Ringwald     }
34753deb3ec6SMatthias Ringwald }
34763deb3ec6SMatthias Ringwald 
34773deb3ec6SMatthias Ringwald /**
34783deb3ec6SMatthias Ringwald  * @brief Trigger Security Request
34793deb3ec6SMatthias Ringwald  */
3480711e6c80SMatthias Ringwald void sm_send_security_request(hci_con_handle_t con_handle){
3481711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
34823deb3ec6SMatthias Ringwald     if (!sm_conn) return;
34833deb3ec6SMatthias Ringwald     sm_send_security_request_for_connection(sm_conn);
34843deb3ec6SMatthias Ringwald }
34853deb3ec6SMatthias Ringwald 
34863deb3ec6SMatthias Ringwald // request pairing
3487711e6c80SMatthias Ringwald void sm_request_pairing(hci_con_handle_t con_handle){
3488711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
34893deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
34903deb3ec6SMatthias Ringwald 
34913deb3ec6SMatthias Ringwald     log_info("sm_request_pairing in role %u, state %u", sm_conn->sm_role, sm_conn->sm_engine_state);
34923deb3ec6SMatthias Ringwald     if (sm_conn->sm_role){
34933deb3ec6SMatthias Ringwald         sm_send_security_request_for_connection(sm_conn);
34943deb3ec6SMatthias Ringwald     } else {
34953deb3ec6SMatthias Ringwald         // used as a trigger to start central/master/initiator security procedures
34963deb3ec6SMatthias Ringwald         uint16_t ediv;
34973764b551SMatthias Ringwald         sm_key_t ltk;
34983deb3ec6SMatthias Ringwald         if (sm_conn->sm_engine_state == SM_INITIATOR_CONNECTED){
34993deb3ec6SMatthias Ringwald             switch (sm_conn->sm_irk_lookup_state){
35003deb3ec6SMatthias Ringwald                 case IRK_LOOKUP_FAILED:
35013deb3ec6SMatthias Ringwald                     sm_conn->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
35023deb3ec6SMatthias Ringwald                     break;
35033deb3ec6SMatthias Ringwald                 case IRK_LOOKUP_SUCCEEDED:
35043764b551SMatthias Ringwald                         le_device_db_encryption_get(sm_conn->sm_le_db_index, &ediv, NULL, ltk, NULL, NULL, NULL);
35053764b551SMatthias Ringwald                         if (!sm_is_null_key(ltk) || ediv){
35063deb3ec6SMatthias Ringwald                             log_info("sm: Setting up previous ltk/ediv/rand for device index %u", sm_conn->sm_le_db_index);
35073deb3ec6SMatthias Ringwald                             sm_conn->sm_engine_state = SM_INITIATOR_PH0_HAS_LTK;
35083deb3ec6SMatthias Ringwald                         } else {
35093deb3ec6SMatthias Ringwald                             sm_conn->sm_engine_state = SM_INITIATOR_PH1_W2_SEND_PAIRING_REQUEST;
35103deb3ec6SMatthias Ringwald                         }
35113deb3ec6SMatthias Ringwald                         break;
35123deb3ec6SMatthias Ringwald                 default:
35133deb3ec6SMatthias Ringwald                     sm_conn->sm_bonding_requested = 1;
35143deb3ec6SMatthias Ringwald                     break;
35153deb3ec6SMatthias Ringwald             }
3516e88b2961SMatthias Ringwald         } else if (sm_conn->sm_engine_state == SM_GENERAL_IDLE){
3517e88b2961SMatthias Ringwald             sm_conn->sm_bonding_requested = 1;
35183deb3ec6SMatthias Ringwald         }
35193deb3ec6SMatthias Ringwald     }
35203deb3ec6SMatthias Ringwald     sm_run();
35213deb3ec6SMatthias Ringwald }
35223deb3ec6SMatthias Ringwald 
35233deb3ec6SMatthias Ringwald // called by client app on authorization request
3524711e6c80SMatthias Ringwald void sm_authorization_decline(hci_con_handle_t con_handle){
3525711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
35263deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
35273deb3ec6SMatthias Ringwald     sm_conn->sm_connection_authorization_state = AUTHORIZATION_DECLINED;
35285611a760SMatthias Ringwald     sm_notify_client_authorization(SM_EVENT_AUTHORIZATION_RESULT, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, 0);
35293deb3ec6SMatthias Ringwald }
35303deb3ec6SMatthias Ringwald 
3531711e6c80SMatthias Ringwald void sm_authorization_grant(hci_con_handle_t con_handle){
3532711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
35333deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
35343deb3ec6SMatthias Ringwald     sm_conn->sm_connection_authorization_state = AUTHORIZATION_GRANTED;
35355611a760SMatthias Ringwald     sm_notify_client_authorization(SM_EVENT_AUTHORIZATION_RESULT, sm_conn->sm_handle, sm_conn->sm_peer_addr_type, sm_conn->sm_peer_address, 1);
35363deb3ec6SMatthias Ringwald }
35373deb3ec6SMatthias Ringwald 
35383deb3ec6SMatthias Ringwald // GAP Bonding API
35393deb3ec6SMatthias Ringwald 
3540711e6c80SMatthias Ringwald void sm_bonding_decline(hci_con_handle_t con_handle){
3541711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
35423deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
35433deb3ec6SMatthias Ringwald     setup->sm_user_response = SM_USER_RESPONSE_DECLINE;
35443deb3ec6SMatthias Ringwald 
35453deb3ec6SMatthias Ringwald     if (sm_conn->sm_engine_state == SM_PH1_W4_USER_RESPONSE){
3546de2fd182SMatthias Ringwald         switch (setup->sm_stk_generation_method){
3547de2fd182SMatthias Ringwald             case PK_RESP_INPUT:
3548de2fd182SMatthias Ringwald             case PK_INIT_INPUT:
3549de2fd182SMatthias Ringwald             case OK_BOTH_INPUT:
3550de2fd182SMatthias Ringwald                 sm_pairing_error(sm_conn, SM_GENERAL_SEND_PAIRING_FAILED);
3551de2fd182SMatthias Ringwald                 break;
3552de2fd182SMatthias Ringwald             case NK_BOTH_INPUT:
3553de2fd182SMatthias Ringwald                 sm_pairing_error(sm_conn, SM_REASON_NUMERIC_COMPARISON_FAILED);
3554de2fd182SMatthias Ringwald                 break;
3555de2fd182SMatthias Ringwald             case JUST_WORKS:
3556de2fd182SMatthias Ringwald             case OOB:
3557de2fd182SMatthias Ringwald                 sm_pairing_error(sm_conn, SM_REASON_UNSPECIFIED_REASON);
3558de2fd182SMatthias Ringwald                 break;
3559de2fd182SMatthias Ringwald         }
35603deb3ec6SMatthias Ringwald     }
35613deb3ec6SMatthias Ringwald     sm_run();
35623deb3ec6SMatthias Ringwald }
35633deb3ec6SMatthias Ringwald 
3564711e6c80SMatthias Ringwald void sm_just_works_confirm(hci_con_handle_t con_handle){
3565711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
35663deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
35673deb3ec6SMatthias Ringwald     setup->sm_user_response = SM_USER_RESPONSE_CONFIRM;
35683deb3ec6SMatthias Ringwald     if (sm_conn->sm_engine_state == SM_PH1_W4_USER_RESPONSE){
3569136d331aSMatthias Ringwald         if (setup->sm_use_secure_connections){
3570c6b7cbd9SMatthias Ringwald             sm_conn->sm_engine_state = SM_SC_SEND_PUBLIC_KEY_COMMAND;
3571bbf8db22SMatthias Ringwald         } else {
3572bbf8db22SMatthias Ringwald             sm_conn->sm_engine_state = SM_PH2_C1_GET_RANDOM_A;
3573136d331aSMatthias Ringwald         }
35743deb3ec6SMatthias Ringwald     }
35750346c37cSMatthias Ringwald 
35760346c37cSMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
3577c6b7cbd9SMatthias Ringwald     if (sm_conn->sm_engine_state == SM_SC_W4_USER_RESPONSE){
3578dc300847SMatthias Ringwald         sm_sc_prepare_dhkey_check(sm_conn);
3579446a8c36SMatthias Ringwald     }
35800346c37cSMatthias Ringwald #endif
35810346c37cSMatthias Ringwald 
35823deb3ec6SMatthias Ringwald     sm_run();
35833deb3ec6SMatthias Ringwald }
35843deb3ec6SMatthias Ringwald 
3585c8c46d51SMatthias Ringwald void sm_numeric_comparison_confirm(hci_con_handle_t con_handle){
3586c8c46d51SMatthias Ringwald     // for now, it's the same
3587c8c46d51SMatthias Ringwald     sm_just_works_confirm(con_handle);
3588c8c46d51SMatthias Ringwald }
3589c8c46d51SMatthias Ringwald 
3590711e6c80SMatthias Ringwald void sm_passkey_input(hci_con_handle_t con_handle, uint32_t passkey){
3591711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
35923deb3ec6SMatthias Ringwald     if (!sm_conn) return;     // wrong connection
35933deb3ec6SMatthias Ringwald     sm_reset_tk();
3594f8fbdce0SMatthias Ringwald     big_endian_store_32(setup->sm_tk, 12, passkey);
35953deb3ec6SMatthias Ringwald     setup->sm_user_response = SM_USER_RESPONSE_PASSKEY;
35963deb3ec6SMatthias Ringwald     if (sm_conn->sm_engine_state == SM_PH1_W4_USER_RESPONSE){
35973deb3ec6SMatthias Ringwald         sm_conn->sm_engine_state = SM_PH2_C1_GET_RANDOM_A;
35983deb3ec6SMatthias Ringwald     }
35991c516d8fSMatthias Ringwald #ifdef ENABLE_LE_SECURE_CONNECTIONS
360007036a04SMatthias Ringwald     memcpy(setup->sm_ra, setup->sm_tk, 16);
360107036a04SMatthias Ringwald     memcpy(setup->sm_rb, setup->sm_tk, 16);
360207036a04SMatthias Ringwald     if (sm_conn->sm_engine_state == SM_SC_W4_USER_RESPONSE){
360307036a04SMatthias Ringwald         sm_sc_start_calculating_local_confirm(sm_conn);
360407036a04SMatthias Ringwald     }
36051c516d8fSMatthias Ringwald #endif
36063deb3ec6SMatthias Ringwald     sm_run();
36073deb3ec6SMatthias Ringwald }
36083deb3ec6SMatthias Ringwald 
36093deb3ec6SMatthias Ringwald /**
36103deb3ec6SMatthias Ringwald  * @brief Identify device in LE Device DB
36113deb3ec6SMatthias Ringwald  * @param handle
36123deb3ec6SMatthias Ringwald  * @returns index from le_device_db or -1 if not found/identified
36133deb3ec6SMatthias Ringwald  */
3614711e6c80SMatthias Ringwald int sm_le_device_index(hci_con_handle_t con_handle ){
3615711e6c80SMatthias Ringwald     sm_connection_t * sm_conn = sm_get_connection_for_handle(con_handle);
36163deb3ec6SMatthias Ringwald     if (!sm_conn) return -1;
36173deb3ec6SMatthias Ringwald     return sm_conn->sm_le_db_index;
36183deb3ec6SMatthias Ringwald }
36193deb3ec6SMatthias Ringwald 
36203deb3ec6SMatthias Ringwald // GAP LE API
36213deb3ec6SMatthias Ringwald void gap_random_address_set_mode(gap_random_address_type_t random_address_type){
36223deb3ec6SMatthias Ringwald     gap_random_address_update_stop();
36233deb3ec6SMatthias Ringwald     gap_random_adress_type = random_address_type;
36243deb3ec6SMatthias Ringwald     if (random_address_type == GAP_RANDOM_ADDRESS_TYPE_OFF) return;
36253deb3ec6SMatthias Ringwald     gap_random_address_update_start();
36263deb3ec6SMatthias Ringwald     gap_random_address_trigger();
36273deb3ec6SMatthias Ringwald }
36283deb3ec6SMatthias Ringwald 
36293deb3ec6SMatthias Ringwald gap_random_address_type_t gap_random_address_get_mode(void){
36303deb3ec6SMatthias Ringwald     return gap_random_adress_type;
36313deb3ec6SMatthias Ringwald }
36323deb3ec6SMatthias Ringwald 
36333deb3ec6SMatthias Ringwald void gap_random_address_set_update_period(int period_ms){
36343deb3ec6SMatthias Ringwald     gap_random_adress_update_period = period_ms;
36353deb3ec6SMatthias Ringwald     if (gap_random_adress_type == GAP_RANDOM_ADDRESS_TYPE_OFF) return;
36363deb3ec6SMatthias Ringwald     gap_random_address_update_stop();
36373deb3ec6SMatthias Ringwald     gap_random_address_update_start();
36383deb3ec6SMatthias Ringwald }
36393deb3ec6SMatthias Ringwald 
36407e252622SMatthias Ringwald void gap_random_address_set(bd_addr_t addr){
36417e252622SMatthias Ringwald     gap_random_address_set_mode(GAP_RANDOM_ADDRESS_TYPE_OFF);
36427e252622SMatthias Ringwald     memcpy(sm_random_address, addr, 6);
36437e252622SMatthias Ringwald     rau_state = RAU_SET_ADDRESS;
36447e252622SMatthias Ringwald     sm_run();
36457e252622SMatthias Ringwald }
36467e252622SMatthias Ringwald 
36473deb3ec6SMatthias Ringwald /*
36483deb3ec6SMatthias Ringwald  * @brief Set Advertisement Paramters
36493deb3ec6SMatthias Ringwald  * @param adv_int_min
36503deb3ec6SMatthias Ringwald  * @param adv_int_max
36513deb3ec6SMatthias Ringwald  * @param adv_type
36523deb3ec6SMatthias Ringwald  * @param direct_address_type
36533deb3ec6SMatthias Ringwald  * @param direct_address
36543deb3ec6SMatthias Ringwald  * @param channel_map
36553deb3ec6SMatthias Ringwald  * @param filter_policy
36563deb3ec6SMatthias Ringwald  *
36573deb3ec6SMatthias Ringwald  * @note own_address_type is used from gap_random_address_set_mode
36583deb3ec6SMatthias Ringwald  */
36593deb3ec6SMatthias Ringwald void gap_advertisements_set_params(uint16_t adv_int_min, uint16_t adv_int_max, uint8_t adv_type,
36603deb3ec6SMatthias Ringwald     uint8_t direct_address_typ, bd_addr_t direct_address, uint8_t channel_map, uint8_t filter_policy){
36613deb3ec6SMatthias Ringwald     hci_le_advertisements_set_params(adv_int_min, adv_int_max, adv_type, gap_random_adress_type,
36623deb3ec6SMatthias Ringwald         direct_address_typ, direct_address, channel_map, filter_policy);
36633deb3ec6SMatthias Ringwald }
36643deb3ec6SMatthias Ringwald 
3665