xref: /btstack/src/mesh/mesh_upper_transport.c (revision 6535961a1f105e850ecdf294c128b9fc982ee05a)
177ba3d3fSMatthias Ringwald /*
277ba3d3fSMatthias Ringwald  * Copyright (C) 2014 BlueKitchen GmbH
377ba3d3fSMatthias Ringwald  *
477ba3d3fSMatthias Ringwald  * Redistribution and use in source and binary forms, with or without
577ba3d3fSMatthias Ringwald  * modification, are permitted provided that the following conditions
677ba3d3fSMatthias Ringwald  * are met:
777ba3d3fSMatthias Ringwald  *
877ba3d3fSMatthias Ringwald  * 1. Redistributions of source code must retain the above copyright
977ba3d3fSMatthias Ringwald  *    notice, this list of conditions and the following disclaimer.
1077ba3d3fSMatthias Ringwald  * 2. Redistributions in binary form must reproduce the above copyright
1177ba3d3fSMatthias Ringwald  *    notice, this list of conditions and the following disclaimer in the
1277ba3d3fSMatthias Ringwald  *    documentation and/or other materials provided with the distribution.
1377ba3d3fSMatthias Ringwald  * 3. Neither the name of the copyright holders nor the names of
1477ba3d3fSMatthias Ringwald  *    contributors may be used to endorse or promote products derived
1577ba3d3fSMatthias Ringwald  *    from this software without specific prior written permission.
1677ba3d3fSMatthias Ringwald  * 4. Any redistribution, use, or modification is done solely for
1777ba3d3fSMatthias Ringwald  *    personal benefit and not for any commercial purpose or for
1877ba3d3fSMatthias Ringwald  *    monetary gain.
1977ba3d3fSMatthias Ringwald  *
2077ba3d3fSMatthias Ringwald  * THIS SOFTWARE IS PROVIDED BY BLUEKITCHEN GMBH AND CONTRIBUTORS
2177ba3d3fSMatthias Ringwald  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
2277ba3d3fSMatthias Ringwald  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
2377ba3d3fSMatthias Ringwald  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL MATTHIAS
2477ba3d3fSMatthias Ringwald  * RINGWALD OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
2577ba3d3fSMatthias Ringwald  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
2677ba3d3fSMatthias Ringwald  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
2777ba3d3fSMatthias Ringwald  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
2877ba3d3fSMatthias Ringwald  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
2977ba3d3fSMatthias Ringwald  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
3077ba3d3fSMatthias Ringwald  * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3177ba3d3fSMatthias Ringwald  * SUCH DAMAGE.
3277ba3d3fSMatthias Ringwald  *
3377ba3d3fSMatthias Ringwald  * Please inquire about commercial licensing options at
3477ba3d3fSMatthias Ringwald  * [email protected]
3577ba3d3fSMatthias Ringwald  *
3677ba3d3fSMatthias Ringwald  */
3777ba3d3fSMatthias Ringwald 
385beab91eSMatthias Ringwald #define BTSTACK_FILE__ "mesh_upper_transport.c"
39f4854a5eSMatthias Ringwald 
40f4854a5eSMatthias Ringwald #include "mesh/mesh_upper_transport.h"
4177ba3d3fSMatthias Ringwald 
4277ba3d3fSMatthias Ringwald #include <stdio.h>
4377ba3d3fSMatthias Ringwald #include <stdlib.h>
4477ba3d3fSMatthias Ringwald #include <string.h>
45f4854a5eSMatthias Ringwald 
4677ba3d3fSMatthias Ringwald #include "btstack_util.h"
4777ba3d3fSMatthias Ringwald #include "btstack_memory.h"
485beab91eSMatthias Ringwald #include "btstack_debug.h"
49f4854a5eSMatthias Ringwald 
50f4854a5eSMatthias Ringwald #include "mesh/beacon.h"
51f4854a5eSMatthias Ringwald #include "mesh/mesh_iv_index_seq_number.h"
52f4854a5eSMatthias Ringwald #include "mesh/mesh_keys.h"
53f4854a5eSMatthias Ringwald #include "mesh/mesh_lower_transport.h"
54f4854a5eSMatthias Ringwald #include "mesh/mesh_peer.h"
55f4854a5eSMatthias Ringwald #include "mesh/mesh_virtual_addresses.h"
5677ba3d3fSMatthias Ringwald 
5710cd83b4SMatthias Ringwald // TODO: extract mesh_pdu functions into lower transport or network
5810cd83b4SMatthias Ringwald #include "mesh/mesh_access.h"
5910cd83b4SMatthias Ringwald 
6077ba3d3fSMatthias Ringwald static void (*higher_layer_handler)( mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu);
6177ba3d3fSMatthias Ringwald 
6277ba3d3fSMatthias Ringwald static void mesh_print_hex(const char * name, const uint8_t * data, uint16_t len){
6377ba3d3fSMatthias Ringwald     printf("%-20s ", name);
6477ba3d3fSMatthias Ringwald     printf_hexdump(data, len);
6577ba3d3fSMatthias Ringwald }
6677ba3d3fSMatthias Ringwald // static void mesh_print_x(const char * name, uint32_t value){
6777ba3d3fSMatthias Ringwald //     printf("%20s: 0x%x", name, (int) value);
6877ba3d3fSMatthias Ringwald // }
6977ba3d3fSMatthias Ringwald 
7077ba3d3fSMatthias Ringwald 
7177ba3d3fSMatthias Ringwald // combined key x address iterator for upper transport decryption
7277ba3d3fSMatthias Ringwald 
7377ba3d3fSMatthias Ringwald typedef struct {
7477ba3d3fSMatthias Ringwald     // state
7577ba3d3fSMatthias Ringwald     mesh_transport_key_iterator_t  key_it;
7677ba3d3fSMatthias Ringwald     mesh_virtual_address_iterator_t address_it;
7777ba3d3fSMatthias Ringwald     // elements
7877ba3d3fSMatthias Ringwald     const mesh_transport_key_t *   key;
7977ba3d3fSMatthias Ringwald     const mesh_virtual_address_t * address;
8077ba3d3fSMatthias Ringwald     // address - might be virtual
8177ba3d3fSMatthias Ringwald     uint16_t dst;
8277ba3d3fSMatthias Ringwald     // key info
8377ba3d3fSMatthias Ringwald } mesh_transport_key_and_virtual_address_iterator_t;
8477ba3d3fSMatthias Ringwald 
8577ba3d3fSMatthias Ringwald static void mesh_transport_key_and_virtual_address_iterator_init(mesh_transport_key_and_virtual_address_iterator_t *it,
8677ba3d3fSMatthias Ringwald                                                                  uint16_t dst, uint16_t netkey_index, uint8_t akf,
8777ba3d3fSMatthias Ringwald                                                                  uint8_t aid) {
8877ba3d3fSMatthias Ringwald     printf("KEY_INIT: dst %04x, akf %x, aid %x\n", dst, akf, aid);
8977ba3d3fSMatthias Ringwald     // config
9077ba3d3fSMatthias Ringwald     it->dst   = dst;
9177ba3d3fSMatthias Ringwald     // init elements
9277ba3d3fSMatthias Ringwald     it->key     = NULL;
9377ba3d3fSMatthias Ringwald     it->address = NULL;
9477ba3d3fSMatthias Ringwald     // init element iterators
9577ba3d3fSMatthias Ringwald     mesh_transport_key_aid_iterator_init(&it->key_it, netkey_index, akf, aid);
9677ba3d3fSMatthias Ringwald     // init address iterator
9777ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)){
9877ba3d3fSMatthias Ringwald         mesh_virtual_address_iterator_init(&it->address_it, dst);
9977ba3d3fSMatthias Ringwald         // get first key
10077ba3d3fSMatthias Ringwald         if (mesh_transport_key_aid_iterator_has_more(&it->key_it)) {
10177ba3d3fSMatthias Ringwald             it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
10277ba3d3fSMatthias Ringwald         }
10377ba3d3fSMatthias Ringwald     }
10477ba3d3fSMatthias Ringwald }
10577ba3d3fSMatthias Ringwald 
10677ba3d3fSMatthias Ringwald // cartesian product: keys x addressses
10777ba3d3fSMatthias Ringwald static int mesh_transport_key_and_virtual_address_iterator_has_more(mesh_transport_key_and_virtual_address_iterator_t * it){
10877ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)) {
10977ba3d3fSMatthias Ringwald         // find next valid entry
11077ba3d3fSMatthias Ringwald         while (1){
11177ba3d3fSMatthias Ringwald             if (mesh_virtual_address_iterator_has_more(&it->address_it)) return 1;
11277ba3d3fSMatthias Ringwald             if (!mesh_transport_key_aid_iterator_has_more(&it->key_it)) return 0;
11377ba3d3fSMatthias Ringwald             // get next key
11477ba3d3fSMatthias Ringwald             it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
11577ba3d3fSMatthias Ringwald             mesh_virtual_address_iterator_init(&it->address_it, it->dst);
11677ba3d3fSMatthias Ringwald         }
11777ba3d3fSMatthias Ringwald     } else {
11877ba3d3fSMatthias Ringwald         return mesh_transport_key_aid_iterator_has_more(&it->key_it);
11977ba3d3fSMatthias Ringwald     }
12077ba3d3fSMatthias Ringwald }
12177ba3d3fSMatthias Ringwald 
12277ba3d3fSMatthias Ringwald static void mesh_transport_key_and_virtual_address_iterator_next(mesh_transport_key_and_virtual_address_iterator_t * it){
12377ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)) {
12477ba3d3fSMatthias Ringwald         it->address = mesh_virtual_address_iterator_get_next(&it->address_it);
12577ba3d3fSMatthias Ringwald     } else {
12677ba3d3fSMatthias Ringwald         it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
12777ba3d3fSMatthias Ringwald     }
12877ba3d3fSMatthias Ringwald }
12977ba3d3fSMatthias Ringwald 
13077ba3d3fSMatthias Ringwald // UPPER TRANSPORT
13177ba3d3fSMatthias Ringwald 
13277ba3d3fSMatthias Ringwald // stub lower transport
13377ba3d3fSMatthias Ringwald 
134dd729594SMatthias Ringwald static void mesh_upper_transport_validate_unsegmented_message(void);
135b86620d9SMatthias Ringwald static void mesh_upper_transport_validate_segmented_message(void);
13677ba3d3fSMatthias Ringwald 
1375beab91eSMatthias Ringwald static void mesh_upper_transport_run(void);
13877ba3d3fSMatthias Ringwald 
13977ba3d3fSMatthias Ringwald static int crypto_active;
140dd729594SMatthias Ringwald static mesh_network_pdu_t   * incoming_network_pdu_raw;
141dd729594SMatthias Ringwald static mesh_network_pdu_t   * incoming_network_pdu_decoded;
142dd729594SMatthias Ringwald static mesh_transport_pdu_t * incoming_transport_pdu_raw;
143dd729594SMatthias Ringwald static mesh_transport_pdu_t * incoming_transport_pdu_decoded;
14477ba3d3fSMatthias Ringwald static uint8_t application_nonce[13];
14577ba3d3fSMatthias Ringwald static btstack_crypto_ccm_t ccm;
14677ba3d3fSMatthias Ringwald static mesh_transport_key_and_virtual_address_iterator_t mesh_transport_key_it;
14777ba3d3fSMatthias Ringwald 
14877ba3d3fSMatthias Ringwald // upper transport callbacks - in access layer
14977ba3d3fSMatthias Ringwald static void (*mesh_access_message_handler)(mesh_pdu_t * pdu);
15077ba3d3fSMatthias Ringwald static void (*mesh_control_message_handler)(mesh_pdu_t * pdu);
15177ba3d3fSMatthias Ringwald 
1525beab91eSMatthias Ringwald // incoming unsegmented (network) and segmented (transport) control and access messages
15377ba3d3fSMatthias Ringwald static btstack_linked_list_t upper_transport_incoming;
15477ba3d3fSMatthias Ringwald 
1555beab91eSMatthias Ringwald // outgoing unsegmented (network) and segmented (uppert_transport_outgoing) control and access messages
1565beab91eSMatthias Ringwald static btstack_linked_list_t upper_transport_outgoing;
1575beab91eSMatthias Ringwald 
158dd729594SMatthias Ringwald static void mesh_upper_transport_dump_pdus(const char *name, btstack_linked_list_t *list){
159dd729594SMatthias Ringwald     printf("List: %s:\n", name);
160dd729594SMatthias Ringwald     btstack_linked_list_iterator_t it;
161dd729594SMatthias Ringwald     btstack_linked_list_iterator_init(&it, list);
162dd729594SMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)){
163dd729594SMatthias Ringwald         mesh_pdu_t * pdu = (mesh_pdu_t*) btstack_linked_list_iterator_next(&it);
164dd729594SMatthias Ringwald         printf("- %p\n", pdu);
165dd729594SMatthias Ringwald         // printf_hexdump( mesh_pdu_data(pdu), mesh_pdu_len(pdu));
166dd729594SMatthias Ringwald     }
167dd729594SMatthias Ringwald }
168dd729594SMatthias Ringwald 
169dd729594SMatthias Ringwald static void mesh_upper_transport_reset_pdus(btstack_linked_list_t *list){
170dd729594SMatthias Ringwald     while (!btstack_linked_list_empty(list)){
171dd729594SMatthias Ringwald         mesh_pdu_t * pdu = (mesh_pdu_t *) btstack_linked_list_pop(list);
172dd729594SMatthias Ringwald         switch (pdu->pdu_type){
173dd729594SMatthias Ringwald             case MESH_PDU_TYPE_NETWORK:
174dd729594SMatthias Ringwald                 btstack_memory_mesh_network_pdu_free((mesh_network_pdu_t *) pdu);
175dd729594SMatthias Ringwald                 break;
176dd729594SMatthias Ringwald             case MESH_PDU_TYPE_TRANSPORT:
177dd729594SMatthias Ringwald                 btstack_memory_mesh_transport_pdu_free((mesh_transport_pdu_t *) pdu);
178dd729594SMatthias Ringwald                 break;
179dd729594SMatthias Ringwald             default:
180dd729594SMatthias Ringwald                 break;
181dd729594SMatthias Ringwald         }
182dd729594SMatthias Ringwald     }
183dd729594SMatthias Ringwald }
184dd729594SMatthias Ringwald 
185dd729594SMatthias Ringwald void mesh_upper_transport_dump(void){
186dd729594SMatthias Ringwald     printf("incoming_network_pdu_raw: %p\n", incoming_network_pdu_raw);
187dd729594SMatthias Ringwald     printf("incoming_network_pdu_decoded: %p\n", incoming_network_pdu_decoded);
188dd729594SMatthias Ringwald     mesh_upper_transport_dump_pdus("upper_transport_incoming", &upper_transport_incoming);
189dd729594SMatthias Ringwald }
190dd729594SMatthias Ringwald 
191dd729594SMatthias Ringwald void mesh_upper_transport_reset(void){
192dd729594SMatthias Ringwald     crypto_active = 0;
193dd729594SMatthias Ringwald     if (incoming_network_pdu_raw){
194dd729594SMatthias Ringwald         mesh_network_pdu_free(incoming_network_pdu_raw);
195dd729594SMatthias Ringwald         incoming_network_pdu_raw = NULL;
196dd729594SMatthias Ringwald     }
197dd729594SMatthias Ringwald     // if (incoming_network_pdu_decoded){
198dd729594SMatthias Ringwald     //     mesh_network_pdu_free(incoming_network_pdu_decoded);
199dd729594SMatthias Ringwald     //     incoming_network_pdu_decoded = NULL;
200dd729594SMatthias Ringwald     // }
201b86620d9SMatthias Ringwald     mesh_upper_transport_reset_pdus(&upper_transport_incoming);
202dd729594SMatthias Ringwald }
20377ba3d3fSMatthias Ringwald 
2042983fbcbSMatthias Ringwald static void mesh_upper_unsegmented_control_message_received(mesh_network_pdu_t * network_pdu){
20577ba3d3fSMatthias Ringwald     uint8_t * lower_transport_pdu     = mesh_network_pdu_data(network_pdu);
20677ba3d3fSMatthias Ringwald     uint8_t  opcode = lower_transport_pdu[0];
20777ba3d3fSMatthias Ringwald     if (mesh_control_message_handler){
20877ba3d3fSMatthias Ringwald         mesh_control_message_handler((mesh_pdu_t*) network_pdu);
20977ba3d3fSMatthias Ringwald     } else {
21077ba3d3fSMatthias Ringwald         printf("[!] Unhandled Control message with opcode %02x\n", opcode);
21177ba3d3fSMatthias Ringwald         // done
21277ba3d3fSMatthias Ringwald         mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *) network_pdu);
21377ba3d3fSMatthias Ringwald     }
21477ba3d3fSMatthias Ringwald }
21577ba3d3fSMatthias Ringwald 
21677ba3d3fSMatthias Ringwald static void mesh_upper_transport_process_unsegmented_message_done(mesh_network_pdu_t *network_pdu){
21777ba3d3fSMatthias Ringwald     crypto_active = 0;
21877ba3d3fSMatthias Ringwald     if (mesh_network_control(network_pdu)) {
21977ba3d3fSMatthias Ringwald         mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *) network_pdu);
22077ba3d3fSMatthias Ringwald     } else {
22177ba3d3fSMatthias Ringwald         mesh_network_pdu_free(network_pdu);
222dd729594SMatthias Ringwald         mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *) incoming_network_pdu_raw);
223dd729594SMatthias Ringwald         incoming_network_pdu_raw = NULL;
22477ba3d3fSMatthias Ringwald     }
2255beab91eSMatthias Ringwald     mesh_upper_transport_run();
22677ba3d3fSMatthias Ringwald }
22777ba3d3fSMatthias Ringwald 
22877ba3d3fSMatthias Ringwald static void mesh_upper_transport_process_segmented_message_done(mesh_transport_pdu_t *transport_pdu){
22977ba3d3fSMatthias Ringwald     crypto_active = 0;
23077ba3d3fSMatthias Ringwald     if (mesh_transport_ctl(transport_pdu)) {
23177ba3d3fSMatthias Ringwald         mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *)transport_pdu);
23277ba3d3fSMatthias Ringwald     } else {
23377ba3d3fSMatthias Ringwald         mesh_transport_pdu_free(transport_pdu);
234dd729594SMatthias Ringwald         mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *)incoming_transport_pdu_raw);
235dd729594SMatthias Ringwald         incoming_transport_pdu_raw = NULL;
23677ba3d3fSMatthias Ringwald     }
2375beab91eSMatthias Ringwald     mesh_upper_transport_run();
23877ba3d3fSMatthias Ringwald }
23977ba3d3fSMatthias Ringwald 
24077ba3d3fSMatthias Ringwald static uint32_t iv_index_for_ivi_nid(uint8_t ivi_nid){
24177ba3d3fSMatthias Ringwald     // get IV Index and IVI
24277ba3d3fSMatthias Ringwald     uint32_t iv_index = mesh_get_iv_index();
24377ba3d3fSMatthias Ringwald     int ivi = ivi_nid >> 7;
24477ba3d3fSMatthias Ringwald 
24577ba3d3fSMatthias Ringwald     // if least significant bit differs, use previous IV Index
24677ba3d3fSMatthias Ringwald     if ((iv_index & 1 ) ^ ivi){
24777ba3d3fSMatthias Ringwald         iv_index--;
24877ba3d3fSMatthias Ringwald     }
24977ba3d3fSMatthias Ringwald     return iv_index;
25077ba3d3fSMatthias Ringwald }
25177ba3d3fSMatthias Ringwald 
25277ba3d3fSMatthias Ringwald static void transport_unsegmented_setup_nonce(uint8_t * nonce, const mesh_network_pdu_t * network_pdu){
25377ba3d3fSMatthias Ringwald     nonce[1] = 0x00;    // SZMIC if a Segmented Access message or 0 for all other message formats
254*6535961aSMatthias Ringwald     (void)memcpy(&nonce[2], &network_pdu->data[2], 7);
25577ba3d3fSMatthias Ringwald     big_endian_store_32(nonce, 9, iv_index_for_ivi_nid(network_pdu->data[0]));
25677ba3d3fSMatthias Ringwald }
25777ba3d3fSMatthias Ringwald 
25877ba3d3fSMatthias Ringwald static void transport_segmented_setup_nonce(uint8_t * nonce, const mesh_transport_pdu_t * transport_pdu){
25977ba3d3fSMatthias Ringwald     nonce[1] = transport_pdu->transmic_len == 8 ? 0x80 : 0x00;
260*6535961aSMatthias Ringwald     (void)memcpy(&nonce[2], &transport_pdu->network_header[2], 7);
26177ba3d3fSMatthias Ringwald     big_endian_store_32(nonce, 9, iv_index_for_ivi_nid(transport_pdu->network_header[0]));
26277ba3d3fSMatthias Ringwald }
26377ba3d3fSMatthias Ringwald 
26477ba3d3fSMatthias Ringwald static void transport_unsegmented_setup_application_nonce(uint8_t * nonce, const mesh_network_pdu_t * network_pdu){
26577ba3d3fSMatthias Ringwald     nonce[0] = 0x01;
26677ba3d3fSMatthias Ringwald     transport_unsegmented_setup_nonce(nonce, network_pdu);
26777ba3d3fSMatthias Ringwald     mesh_print_hex("AppNonce", nonce, 13);
26877ba3d3fSMatthias Ringwald }
26977ba3d3fSMatthias Ringwald 
27077ba3d3fSMatthias Ringwald static void transport_unsegmented_setup_device_nonce(uint8_t * nonce, const mesh_network_pdu_t * network_pdu){
27177ba3d3fSMatthias Ringwald     nonce[0] = 0x02;
27277ba3d3fSMatthias Ringwald     transport_unsegmented_setup_nonce(nonce, network_pdu);
27377ba3d3fSMatthias Ringwald     mesh_print_hex("DeviceNonce", nonce, 13);
27477ba3d3fSMatthias Ringwald }
27577ba3d3fSMatthias Ringwald 
27677ba3d3fSMatthias Ringwald static void transport_segmented_setup_application_nonce(uint8_t * nonce, const mesh_transport_pdu_t * transport_pdu){
27777ba3d3fSMatthias Ringwald     nonce[0] = 0x01;
27877ba3d3fSMatthias Ringwald     transport_segmented_setup_nonce(nonce, transport_pdu);
27977ba3d3fSMatthias Ringwald     mesh_print_hex("AppNonce", nonce, 13);
28077ba3d3fSMatthias Ringwald }
28177ba3d3fSMatthias Ringwald 
28277ba3d3fSMatthias Ringwald static void transport_segmented_setup_device_nonce(uint8_t * nonce, const mesh_transport_pdu_t * transport_pdu){
28377ba3d3fSMatthias Ringwald     nonce[0] = 0x02;
28477ba3d3fSMatthias Ringwald     transport_segmented_setup_nonce(nonce, transport_pdu);
28577ba3d3fSMatthias Ringwald     mesh_print_hex("DeviceNonce", nonce, 13);
28677ba3d3fSMatthias Ringwald }
28777ba3d3fSMatthias Ringwald 
28877ba3d3fSMatthias Ringwald static void mesh_upper_transport_validate_unsegmented_message_ccm(void * arg){
289dd729594SMatthias Ringwald     UNUSED(arg);
29077ba3d3fSMatthias Ringwald 
291dd729594SMatthias Ringwald     uint8_t * lower_transport_pdu = mesh_network_pdu_data(incoming_network_pdu_decoded);
29277ba3d3fSMatthias Ringwald     uint8_t trans_mic_len = 4;
29377ba3d3fSMatthias Ringwald 
29477ba3d3fSMatthias Ringwald     // store TransMIC
29577ba3d3fSMatthias Ringwald     uint8_t trans_mic[8];
29677ba3d3fSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, trans_mic);
29777ba3d3fSMatthias Ringwald     mesh_print_hex("TransMIC", trans_mic, trans_mic_len);
29877ba3d3fSMatthias Ringwald 
299dd729594SMatthias Ringwald     uint8_t * upper_transport_pdu     = mesh_network_pdu_data(incoming_network_pdu_decoded) + 1;
300dd729594SMatthias Ringwald     uint8_t   upper_transport_pdu_len = mesh_network_pdu_len(incoming_network_pdu_decoded)  - 1;
30177ba3d3fSMatthias Ringwald 
30277ba3d3fSMatthias Ringwald     mesh_print_hex("Decryted PDU", upper_transport_pdu, upper_transport_pdu_len - trans_mic_len);
30377ba3d3fSMatthias Ringwald 
30477ba3d3fSMatthias Ringwald     if (memcmp(trans_mic, &upper_transport_pdu[upper_transport_pdu_len - trans_mic_len], trans_mic_len) == 0){
30577ba3d3fSMatthias Ringwald         printf("TransMIC matches\n");
30677ba3d3fSMatthias Ringwald 
30777ba3d3fSMatthias Ringwald         // remove TransMIC from payload
308dd729594SMatthias Ringwald         incoming_network_pdu_decoded->len -= trans_mic_len;
30977ba3d3fSMatthias Ringwald 
31077ba3d3fSMatthias Ringwald         // if virtual address, update dst to pseudo_dst
311dd729594SMatthias Ringwald         if (mesh_network_address_virtual(mesh_network_dst(incoming_network_pdu_decoded))){
312dd729594SMatthias Ringwald             big_endian_store_16(incoming_network_pdu_decoded->data, 7, mesh_transport_key_it.address->pseudo_dst);
31377ba3d3fSMatthias Ringwald         }
31477ba3d3fSMatthias Ringwald 
31577ba3d3fSMatthias Ringwald         // pass to upper layer
31677ba3d3fSMatthias Ringwald         if (mesh_access_message_handler){
317dd729594SMatthias Ringwald             mesh_pdu_t * pdu = (mesh_pdu_t*) incoming_network_pdu_decoded;
318dd729594SMatthias Ringwald             incoming_network_pdu_decoded = NULL;
319dd729594SMatthias Ringwald             mesh_access_message_handler(pdu);
32077ba3d3fSMatthias Ringwald         } else {
32177ba3d3fSMatthias Ringwald             printf("[!] Unhandled Unsegmented Access message\n");
32277ba3d3fSMatthias Ringwald             // done
323dd729594SMatthias Ringwald             mesh_upper_transport_process_unsegmented_message_done(incoming_network_pdu_decoded);
32477ba3d3fSMatthias Ringwald         }
32577ba3d3fSMatthias Ringwald 
32677ba3d3fSMatthias Ringwald         printf("\n");
32777ba3d3fSMatthias Ringwald     } else {
32877ba3d3fSMatthias Ringwald         uint8_t afk = lower_transport_pdu[0] & 0x40;
32977ba3d3fSMatthias Ringwald         if (afk){
33077ba3d3fSMatthias Ringwald             printf("TransMIC does not match, try next key\n");
331dd729594SMatthias Ringwald             mesh_upper_transport_validate_unsegmented_message();
33277ba3d3fSMatthias Ringwald         } else {
33377ba3d3fSMatthias Ringwald             printf("TransMIC does not match device key, done\n");
33477ba3d3fSMatthias Ringwald             // done
335dd729594SMatthias Ringwald             mesh_upper_transport_process_unsegmented_message_done(incoming_network_pdu_decoded);
33677ba3d3fSMatthias Ringwald         }
33777ba3d3fSMatthias Ringwald     }
33877ba3d3fSMatthias Ringwald }
33977ba3d3fSMatthias Ringwald 
34077ba3d3fSMatthias Ringwald static void mesh_upper_transport_validate_segmented_message_ccm(void * arg){
341b86620d9SMatthias Ringwald     UNUSED(arg);
34277ba3d3fSMatthias Ringwald 
343b86620d9SMatthias Ringwald     uint8_t * upper_transport_pdu     =  incoming_transport_pdu_decoded->data;
344b86620d9SMatthias Ringwald     uint8_t   upper_transport_pdu_len =  incoming_transport_pdu_decoded->len - incoming_transport_pdu_decoded->transmic_len;
34577ba3d3fSMatthias Ringwald 
34677ba3d3fSMatthias Ringwald     mesh_print_hex("Decrypted PDU", upper_transport_pdu, upper_transport_pdu_len);
34777ba3d3fSMatthias Ringwald 
34877ba3d3fSMatthias Ringwald     // store TransMIC
34977ba3d3fSMatthias Ringwald     uint8_t trans_mic[8];
35077ba3d3fSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, trans_mic);
351b86620d9SMatthias Ringwald     mesh_print_hex("TransMIC", trans_mic, incoming_transport_pdu_decoded->transmic_len);
35277ba3d3fSMatthias Ringwald 
353b86620d9SMatthias Ringwald     if (memcmp(trans_mic, &upper_transport_pdu[upper_transport_pdu_len], incoming_transport_pdu_decoded->transmic_len) == 0){
35477ba3d3fSMatthias Ringwald         printf("TransMIC matches\n");
35577ba3d3fSMatthias Ringwald 
35677ba3d3fSMatthias Ringwald         // remove TransMIC from payload
357b86620d9SMatthias Ringwald         incoming_transport_pdu_decoded->len -= incoming_transport_pdu_decoded->transmic_len;
35877ba3d3fSMatthias Ringwald 
35977ba3d3fSMatthias Ringwald         // if virtual address, update dst to pseudo_dst
360b86620d9SMatthias Ringwald         if (mesh_network_address_virtual(mesh_transport_dst(incoming_transport_pdu_decoded))){
361b86620d9SMatthias Ringwald             big_endian_store_16(incoming_transport_pdu_decoded->network_header, 7, mesh_transport_key_it.address->pseudo_dst);
36277ba3d3fSMatthias Ringwald         }
36377ba3d3fSMatthias Ringwald 
36477ba3d3fSMatthias Ringwald         // pass to upper layer
36577ba3d3fSMatthias Ringwald         if (mesh_access_message_handler){
366b86620d9SMatthias Ringwald             mesh_pdu_t * pdu = (mesh_pdu_t*) incoming_transport_pdu_decoded;
367b86620d9SMatthias Ringwald             incoming_network_pdu_decoded = NULL;
368b86620d9SMatthias Ringwald             mesh_access_message_handler(pdu);
36977ba3d3fSMatthias Ringwald         } else {
37077ba3d3fSMatthias Ringwald             printf("[!] Unhandled Segmented Access/Control message\n");
37177ba3d3fSMatthias Ringwald             // done
372b86620d9SMatthias Ringwald             mesh_upper_transport_process_segmented_message_done(incoming_transport_pdu_decoded);
37377ba3d3fSMatthias Ringwald         }
37477ba3d3fSMatthias Ringwald 
37577ba3d3fSMatthias Ringwald         printf("\n");
37677ba3d3fSMatthias Ringwald 
37777ba3d3fSMatthias Ringwald     } else {
378b86620d9SMatthias Ringwald         uint8_t akf = incoming_transport_pdu_decoded->akf_aid_control & 0x40;
37977ba3d3fSMatthias Ringwald         if (akf){
38077ba3d3fSMatthias Ringwald             printf("TransMIC does not match, try next key\n");
381b86620d9SMatthias Ringwald             mesh_upper_transport_validate_segmented_message();
38277ba3d3fSMatthias Ringwald         } else {
38377ba3d3fSMatthias Ringwald             printf("TransMIC does not match device key, done\n");
38477ba3d3fSMatthias Ringwald             // done
385b86620d9SMatthias Ringwald             mesh_upper_transport_process_segmented_message_done(incoming_transport_pdu_decoded);
38677ba3d3fSMatthias Ringwald         }
38777ba3d3fSMatthias Ringwald     }
38877ba3d3fSMatthias Ringwald }
38977ba3d3fSMatthias Ringwald 
39077ba3d3fSMatthias Ringwald void mesh_upper_transport_message_processed_by_higher_layer(mesh_pdu_t * pdu){
39177ba3d3fSMatthias Ringwald     crypto_active = 0;
39277ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
39377ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
39477ba3d3fSMatthias Ringwald             mesh_upper_transport_process_unsegmented_message_done((mesh_network_pdu_t *) pdu);
39577ba3d3fSMatthias Ringwald             break;
39677ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_TRANSPORT:
39777ba3d3fSMatthias Ringwald             mesh_upper_transport_process_segmented_message_done((mesh_transport_pdu_t *) pdu);
39877ba3d3fSMatthias Ringwald             break;
39977ba3d3fSMatthias Ringwald         default:
40077ba3d3fSMatthias Ringwald             break;
40177ba3d3fSMatthias Ringwald     }
40277ba3d3fSMatthias Ringwald }
40377ba3d3fSMatthias Ringwald 
40477ba3d3fSMatthias Ringwald static void mesh_upper_transport_validate_segmented_message_digest(void * arg){
405b86620d9SMatthias Ringwald     UNUSED(arg);
406dd729594SMatthias Ringwald     uint8_t   upper_transport_pdu_len      = incoming_transport_pdu_raw->len - incoming_transport_pdu_raw->transmic_len;
407dd729594SMatthias Ringwald     uint8_t * upper_transport_pdu_data_in  = incoming_transport_pdu_raw->data;
408b86620d9SMatthias Ringwald     uint8_t * upper_transport_pdu_data_out = incoming_transport_pdu_decoded->data;
409b86620d9SMatthias Ringwald     btstack_crypto_ccm_decrypt_block(&ccm, upper_transport_pdu_len, upper_transport_pdu_data_in, upper_transport_pdu_data_out, &mesh_upper_transport_validate_segmented_message_ccm, NULL);
41077ba3d3fSMatthias Ringwald }
41177ba3d3fSMatthias Ringwald 
41277ba3d3fSMatthias Ringwald static void mesh_upper_transport_validate_unsegmented_message_digest(void * arg){
413dd729594SMatthias Ringwald     UNUSED(arg);
41477ba3d3fSMatthias Ringwald     uint8_t   trans_mic_len = 4;
415dd729594SMatthias Ringwald     uint8_t   lower_transport_pdu_len      = incoming_network_pdu_raw->len - 9;
416dd729594SMatthias Ringwald     uint8_t * upper_transport_pdu_data_in  = &incoming_network_pdu_raw->data[10];
417dd729594SMatthias Ringwald     uint8_t * upper_transport_pdu_data_out = &incoming_network_pdu_decoded->data[10];
41877ba3d3fSMatthias Ringwald     uint8_t   upper_transport_pdu_len      = lower_transport_pdu_len - 1 - trans_mic_len;
419dd729594SMatthias Ringwald     btstack_crypto_ccm_decrypt_block(&ccm, upper_transport_pdu_len, upper_transport_pdu_data_in, upper_transport_pdu_data_out, &mesh_upper_transport_validate_unsegmented_message_ccm, NULL);
42077ba3d3fSMatthias Ringwald }
42177ba3d3fSMatthias Ringwald 
422dd729594SMatthias Ringwald static void mesh_upper_transport_validate_unsegmented_message(void){
42377ba3d3fSMatthias Ringwald 
42477ba3d3fSMatthias Ringwald     if (!mesh_transport_key_and_virtual_address_iterator_has_more(&mesh_transport_key_it)){
42577ba3d3fSMatthias Ringwald         printf("No valid transport key found\n");
426dd729594SMatthias Ringwald         mesh_upper_transport_process_unsegmented_message_done(incoming_network_pdu_decoded);
42777ba3d3fSMatthias Ringwald         return;
42877ba3d3fSMatthias Ringwald     }
42977ba3d3fSMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_next(&mesh_transport_key_it);
43077ba3d3fSMatthias Ringwald     const mesh_transport_key_t * message_key = mesh_transport_key_it.key;
43177ba3d3fSMatthias Ringwald 
43277ba3d3fSMatthias Ringwald     if (message_key->akf){
433dd729594SMatthias Ringwald         transport_unsegmented_setup_application_nonce(application_nonce, incoming_network_pdu_raw);
43477ba3d3fSMatthias Ringwald     } else {
435dd729594SMatthias Ringwald         transport_unsegmented_setup_device_nonce(application_nonce, incoming_network_pdu_raw);
43677ba3d3fSMatthias Ringwald     }
43777ba3d3fSMatthias Ringwald 
43877ba3d3fSMatthias Ringwald     // store application / device key index
43977ba3d3fSMatthias Ringwald     mesh_print_hex("AppOrDevKey", message_key->key, 16);
440dd729594SMatthias Ringwald     incoming_network_pdu_decoded->appkey_index = message_key->appkey_index;
44177ba3d3fSMatthias Ringwald 
44277ba3d3fSMatthias Ringwald     // unsegmented message have TransMIC of 32 bit
44377ba3d3fSMatthias Ringwald     uint8_t trans_mic_len = 4;
44477ba3d3fSMatthias Ringwald     printf("Unsegmented Access message with TransMIC len 4\n");
44577ba3d3fSMatthias Ringwald 
446dd729594SMatthias Ringwald     uint8_t   lower_transport_pdu_len = incoming_network_pdu_raw->len - 9;
447dd729594SMatthias Ringwald     uint8_t * upper_transport_pdu_data = &incoming_network_pdu_raw->data[10];
44877ba3d3fSMatthias Ringwald     uint8_t   upper_transport_pdu_len  = lower_transport_pdu_len - 1 - trans_mic_len;
44977ba3d3fSMatthias Ringwald 
45077ba3d3fSMatthias Ringwald     mesh_print_hex("EncAccessPayload", upper_transport_pdu_data, upper_transport_pdu_len);
45177ba3d3fSMatthias Ringwald 
45277ba3d3fSMatthias Ringwald     // decrypt ccm
45377ba3d3fSMatthias Ringwald     crypto_active = 1;
45477ba3d3fSMatthias Ringwald     uint16_t aad_len  = 0;
455dd729594SMatthias Ringwald     if (mesh_network_address_virtual(mesh_network_dst(incoming_network_pdu_decoded))){
45677ba3d3fSMatthias Ringwald         aad_len  = 16;
45777ba3d3fSMatthias Ringwald     }
45877ba3d3fSMatthias Ringwald     btstack_crypto_ccm_init(&ccm, message_key->key, application_nonce, upper_transport_pdu_len, aad_len, trans_mic_len);
45977ba3d3fSMatthias Ringwald     if (aad_len){
460dd729594SMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, (uint8_t*) mesh_transport_key_it.address->label_uuid, aad_len, &mesh_upper_transport_validate_unsegmented_message_digest, NULL);
46177ba3d3fSMatthias Ringwald     } else {
462dd729594SMatthias Ringwald         mesh_upper_transport_validate_unsegmented_message_digest(incoming_network_pdu_decoded);
46377ba3d3fSMatthias Ringwald     }
46477ba3d3fSMatthias Ringwald }
46577ba3d3fSMatthias Ringwald 
466b86620d9SMatthias Ringwald static void mesh_upper_transport_validate_segmented_message(void){
467b86620d9SMatthias Ringwald     uint8_t * upper_transport_pdu_data =  incoming_transport_pdu_decoded->data;
468b86620d9SMatthias Ringwald     uint8_t   upper_transport_pdu_len  =  incoming_transport_pdu_decoded->len - incoming_transport_pdu_decoded->transmic_len;
46977ba3d3fSMatthias Ringwald 
47077ba3d3fSMatthias Ringwald     if (!mesh_transport_key_and_virtual_address_iterator_has_more(&mesh_transport_key_it)){
47177ba3d3fSMatthias Ringwald         printf("No valid transport key found\n");
472b86620d9SMatthias Ringwald         mesh_upper_transport_process_segmented_message_done(incoming_transport_pdu_decoded);
47377ba3d3fSMatthias Ringwald         return;
47477ba3d3fSMatthias Ringwald     }
47577ba3d3fSMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_next(&mesh_transport_key_it);
47677ba3d3fSMatthias Ringwald     const mesh_transport_key_t * message_key = mesh_transport_key_it.key;
47777ba3d3fSMatthias Ringwald 
47877ba3d3fSMatthias Ringwald     if (message_key->akf){
479dd729594SMatthias Ringwald         transport_segmented_setup_application_nonce(application_nonce, incoming_transport_pdu_raw);
48077ba3d3fSMatthias Ringwald     } else {
481dd729594SMatthias Ringwald         transport_segmented_setup_device_nonce(application_nonce, incoming_transport_pdu_raw);
48277ba3d3fSMatthias Ringwald     }
48377ba3d3fSMatthias Ringwald 
48477ba3d3fSMatthias Ringwald     // store application / device key index
48577ba3d3fSMatthias Ringwald     mesh_print_hex("AppOrDevKey", message_key->key, 16);
486b86620d9SMatthias Ringwald     incoming_transport_pdu_decoded->appkey_index = message_key->appkey_index;
48777ba3d3fSMatthias Ringwald 
48877ba3d3fSMatthias Ringwald     mesh_print_hex("EncAccessPayload", upper_transport_pdu_data, upper_transport_pdu_len);
48977ba3d3fSMatthias Ringwald 
49077ba3d3fSMatthias Ringwald     // decrypt ccm
49177ba3d3fSMatthias Ringwald     crypto_active = 1;
49277ba3d3fSMatthias Ringwald     uint16_t aad_len  = 0;
493b86620d9SMatthias Ringwald     if (mesh_network_address_virtual(mesh_transport_dst(incoming_transport_pdu_decoded))){
49477ba3d3fSMatthias Ringwald         aad_len  = 16;
49577ba3d3fSMatthias Ringwald     }
496b86620d9SMatthias Ringwald     btstack_crypto_ccm_init(&ccm, message_key->key, application_nonce, upper_transport_pdu_len, aad_len, incoming_transport_pdu_decoded->transmic_len);
49777ba3d3fSMatthias Ringwald 
49877ba3d3fSMatthias Ringwald     if (aad_len){
499b86620d9SMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, (uint8_t *) mesh_transport_key_it.address->label_uuid, aad_len, &mesh_upper_transport_validate_segmented_message_digest, NULL);
50077ba3d3fSMatthias Ringwald     } else {
501b86620d9SMatthias Ringwald         mesh_upper_transport_validate_segmented_message_digest(NULL);
50277ba3d3fSMatthias Ringwald     }
50377ba3d3fSMatthias Ringwald }
50477ba3d3fSMatthias Ringwald 
505dd729594SMatthias Ringwald static void mesh_upper_transport_process_unsegmented_access_message(void){
50677ba3d3fSMatthias Ringwald     // copy original pdu
507dd729594SMatthias Ringwald     incoming_network_pdu_decoded->len = incoming_network_pdu_raw->len;
508*6535961aSMatthias Ringwald     (void)memcpy(incoming_network_pdu_decoded->data,
509*6535961aSMatthias Ringwald                  incoming_network_pdu_raw->data,
510*6535961aSMatthias Ringwald                  incoming_network_pdu_decoded->len);
51177ba3d3fSMatthias Ringwald 
51277ba3d3fSMatthias Ringwald     //
513dd729594SMatthias Ringwald     uint8_t * lower_transport_pdu     = &incoming_network_pdu_raw->data[9];
514dd729594SMatthias Ringwald     uint8_t   lower_transport_pdu_len = incoming_network_pdu_raw->len - 9;
51577ba3d3fSMatthias Ringwald 
516dd729594SMatthias Ringwald     mesh_print_hex("Lower Transport network pdu", &incoming_network_pdu_raw->data[9], lower_transport_pdu_len);
51777ba3d3fSMatthias Ringwald 
51877ba3d3fSMatthias Ringwald     uint8_t aid =  lower_transport_pdu[0] & 0x3f;
51977ba3d3fSMatthias Ringwald     uint8_t akf = (lower_transport_pdu[0] & 0x40) >> 6;
52077ba3d3fSMatthias Ringwald     printf("AKF: %u\n",   akf);
52177ba3d3fSMatthias Ringwald     printf("AID: %02x\n", aid);
52277ba3d3fSMatthias Ringwald 
523dd729594SMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_init(&mesh_transport_key_it, mesh_network_dst(incoming_network_pdu_decoded),
524dd729594SMatthias Ringwald             incoming_network_pdu_decoded->netkey_index, akf, aid);
525dd729594SMatthias Ringwald     mesh_upper_transport_validate_unsegmented_message();
52677ba3d3fSMatthias Ringwald }
52777ba3d3fSMatthias Ringwald 
528b86620d9SMatthias Ringwald static void mesh_upper_transport_process_message(void){
52977ba3d3fSMatthias Ringwald     // copy original pdu
530*6535961aSMatthias Ringwald     (void)memcpy(incoming_transport_pdu_decoded, incoming_transport_pdu_raw,
531*6535961aSMatthias Ringwald                  sizeof(mesh_transport_pdu_t));
53277ba3d3fSMatthias Ringwald 
53377ba3d3fSMatthias Ringwald     //
534b86620d9SMatthias Ringwald     uint8_t * upper_transport_pdu     =  incoming_transport_pdu_decoded->data;
535b86620d9SMatthias Ringwald     uint8_t   upper_transport_pdu_len =  incoming_transport_pdu_decoded->len - incoming_transport_pdu_decoded->transmic_len;
53677ba3d3fSMatthias Ringwald     mesh_print_hex("Upper Transport pdu", upper_transport_pdu, upper_transport_pdu_len);
53777ba3d3fSMatthias Ringwald 
538b86620d9SMatthias Ringwald     uint8_t aid =  incoming_transport_pdu_decoded->akf_aid_control & 0x3f;
539b86620d9SMatthias Ringwald     uint8_t akf = (incoming_transport_pdu_decoded->akf_aid_control & 0x40) >> 6;
54077ba3d3fSMatthias Ringwald 
54177ba3d3fSMatthias Ringwald     printf("AKF: %u\n",   akf);
54277ba3d3fSMatthias Ringwald     printf("AID: %02x\n", aid);
54377ba3d3fSMatthias Ringwald 
544b86620d9SMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_init(&mesh_transport_key_it, mesh_transport_dst(incoming_transport_pdu_decoded),
545b86620d9SMatthias Ringwald             incoming_transport_pdu_decoded->netkey_index, akf, aid);
546b86620d9SMatthias Ringwald     mesh_upper_transport_validate_segmented_message();
54777ba3d3fSMatthias Ringwald }
54877ba3d3fSMatthias Ringwald 
5492983fbcbSMatthias Ringwald static void mesh_upper_transport_message_received(mesh_pdu_t * pdu){
55077ba3d3fSMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_incoming, (btstack_linked_item_t*) pdu);
5515beab91eSMatthias Ringwald     mesh_upper_transport_run();
55277ba3d3fSMatthias Ringwald }
55377ba3d3fSMatthias Ringwald 
554b707feffSMatthias Ringwald void mesh_upper_transport_pdu_free(mesh_pdu_t * pdu){
55577ba3d3fSMatthias Ringwald     mesh_network_pdu_t * network_pdu;
55677ba3d3fSMatthias Ringwald     mesh_transport_pdu_t * transport_pdu;
55777ba3d3fSMatthias Ringwald     switch (pdu->pdu_type) {
55877ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
55977ba3d3fSMatthias Ringwald             network_pdu = (mesh_network_pdu_t *) pdu;
56077ba3d3fSMatthias Ringwald             mesh_network_pdu_free(network_pdu);
56177ba3d3fSMatthias Ringwald             break;
56277ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_TRANSPORT:
56377ba3d3fSMatthias Ringwald             transport_pdu = (mesh_transport_pdu_t *) pdu;
56477ba3d3fSMatthias Ringwald             mesh_transport_pdu_free(transport_pdu);
56577ba3d3fSMatthias Ringwald             break;
56677ba3d3fSMatthias Ringwald         default:
56777ba3d3fSMatthias Ringwald             break;
56877ba3d3fSMatthias Ringwald     }
56977ba3d3fSMatthias Ringwald }
570b707feffSMatthias Ringwald 
5712983fbcbSMatthias Ringwald static void mesh_upper_transport_pdu_handler(mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu){
572b707feffSMatthias Ringwald     switch (callback_type){
573b707feffSMatthias Ringwald         case MESH_TRANSPORT_PDU_RECEIVED:
574b707feffSMatthias Ringwald             mesh_upper_transport_message_received(pdu);
575b707feffSMatthias Ringwald             break;
576b707feffSMatthias Ringwald         case MESH_TRANSPORT_PDU_SENT:
577b707feffSMatthias Ringwald             // notify upper layer (or just free pdu)
578b707feffSMatthias Ringwald             if (higher_layer_handler){
579b707feffSMatthias Ringwald                 higher_layer_handler(callback_type, status, pdu);
580b707feffSMatthias Ringwald             } else {
581b707feffSMatthias Ringwald                 mesh_upper_transport_pdu_free(pdu);
582b707feffSMatthias Ringwald             }
583d19dd9d7SMatthias Ringwald             mesh_upper_transport_run();
58477ba3d3fSMatthias Ringwald             break;
58577ba3d3fSMatthias Ringwald         default:
58677ba3d3fSMatthias Ringwald             break;
58777ba3d3fSMatthias Ringwald     }
58877ba3d3fSMatthias Ringwald }
58977ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_unsegmented_access_pdu_ccm(void * arg){
59077ba3d3fSMatthias Ringwald     crypto_active = 0;
59177ba3d3fSMatthias Ringwald 
59277ba3d3fSMatthias Ringwald     mesh_network_pdu_t * network_pdu = (mesh_network_pdu_t *) arg;
59377ba3d3fSMatthias Ringwald     uint8_t * upper_transport_pdu     = mesh_network_pdu_data(network_pdu) + 1;
59477ba3d3fSMatthias Ringwald     uint8_t   upper_transport_pdu_len = mesh_network_pdu_len(network_pdu)  - 1;
59577ba3d3fSMatthias Ringwald     mesh_print_hex("EncAccessPayload", upper_transport_pdu, upper_transport_pdu_len);
59677ba3d3fSMatthias Ringwald     // store TransMIC
59777ba3d3fSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, &upper_transport_pdu[upper_transport_pdu_len]);
59877ba3d3fSMatthias Ringwald     mesh_print_hex("TransMIC", &upper_transport_pdu[upper_transport_pdu_len], 4);
59977ba3d3fSMatthias Ringwald     network_pdu->len        += 4;
600b16fda24SMatthias Ringwald     upper_transport_pdu_len += 4;
601b16fda24SMatthias Ringwald     mesh_print_hex("UpperTransportPDU", upper_transport_pdu, upper_transport_pdu_len);
60277ba3d3fSMatthias Ringwald     // send network pdu
60377ba3d3fSMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t*) network_pdu);
60477ba3d3fSMatthias Ringwald }
60577ba3d3fSMatthias Ringwald 
60677ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_segmented_access_pdu_ccm(void * arg){
60777ba3d3fSMatthias Ringwald     crypto_active = 0;
60877ba3d3fSMatthias Ringwald 
60977ba3d3fSMatthias Ringwald     mesh_transport_pdu_t * transport_pdu = (mesh_transport_pdu_t *) arg;
61077ba3d3fSMatthias Ringwald     mesh_print_hex("EncAccessPayload", transport_pdu->data, transport_pdu->len);
61177ba3d3fSMatthias Ringwald     // store TransMIC
61277ba3d3fSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, &transport_pdu->data[transport_pdu->len]);
61377ba3d3fSMatthias Ringwald     mesh_print_hex("TransMIC", &transport_pdu->data[transport_pdu->len], transport_pdu->transmic_len);
61477ba3d3fSMatthias Ringwald     transport_pdu->len += transport_pdu->transmic_len;
6157a1ad92dSMatthias Ringwald     mesh_print_hex("UpperTransportPDU", transport_pdu->data, transport_pdu->len);
61677ba3d3fSMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t*) transport_pdu);
61777ba3d3fSMatthias Ringwald }
61877ba3d3fSMatthias Ringwald 
61977ba3d3fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_unsegmented_control_pdu(mesh_network_pdu_t * network_pdu, uint16_t netkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t opcode,
62077ba3d3fSMatthias Ringwald                           const uint8_t * control_pdu_data, uint16_t control_pdu_len){
62177ba3d3fSMatthias Ringwald 
62277ba3d3fSMatthias Ringwald     if (control_pdu_len > 11) return 1;
62377ba3d3fSMatthias Ringwald 
62477ba3d3fSMatthias Ringwald     const mesh_network_key_t * network_key = mesh_network_key_list_get(netkey_index);
62577ba3d3fSMatthias Ringwald     if (!network_key) return 1;
62677ba3d3fSMatthias Ringwald 
62777ba3d3fSMatthias Ringwald     uint8_t transport_pdu_data[12];
62877ba3d3fSMatthias Ringwald     transport_pdu_data[0] = opcode;
629*6535961aSMatthias Ringwald     (void)memcpy(&transport_pdu_data[1], control_pdu_data, control_pdu_len);
63077ba3d3fSMatthias Ringwald     uint16_t transport_pdu_len = control_pdu_len + 1;
63177ba3d3fSMatthias Ringwald 
63277ba3d3fSMatthias Ringwald     // setup network_pdu
63387e3599cSMatthias Ringwald     mesh_network_setup_pdu(network_pdu, netkey_index, network_key->nid, 1, ttl, 0, src, dest, transport_pdu_data, transport_pdu_len);
63477ba3d3fSMatthias Ringwald 
63577ba3d3fSMatthias Ringwald     return 0;
63677ba3d3fSMatthias Ringwald }
63777ba3d3fSMatthias Ringwald 
63877ba3d3fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_segmented_control_pdu(mesh_transport_pdu_t * transport_pdu, uint16_t netkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t opcode,
63977ba3d3fSMatthias Ringwald                           const uint8_t * control_pdu_data, uint16_t control_pdu_len){
64077ba3d3fSMatthias Ringwald 
64177ba3d3fSMatthias Ringwald     if (control_pdu_len > 256) return 1;
64277ba3d3fSMatthias Ringwald 
64377ba3d3fSMatthias Ringwald     const mesh_network_key_t * network_key = mesh_network_key_list_get(netkey_index);
64477ba3d3fSMatthias Ringwald     if (!network_key) return 1;
64577ba3d3fSMatthias Ringwald 
646*6535961aSMatthias Ringwald     (void)memcpy(transport_pdu->data, control_pdu_data, control_pdu_len);
64777ba3d3fSMatthias Ringwald     transport_pdu->len = control_pdu_len;
64877ba3d3fSMatthias Ringwald     transport_pdu->netkey_index = netkey_index;
649e9292fe8SMatthias Ringwald     transport_pdu->akf_aid_control = opcode;
65077ba3d3fSMatthias Ringwald     transport_pdu->transmic_len = 0;    // no TransMIC for control
65177ba3d3fSMatthias Ringwald     mesh_transport_set_nid_ivi(transport_pdu, network_key->nid);
65277ba3d3fSMatthias Ringwald     mesh_transport_set_src(transport_pdu, src);
65377ba3d3fSMatthias Ringwald     mesh_transport_set_dest(transport_pdu, dest);
65477ba3d3fSMatthias Ringwald     mesh_transport_set_ctl_ttl(transport_pdu, 0x80 | ttl);
65577ba3d3fSMatthias Ringwald 
65677ba3d3fSMatthias Ringwald     return 0;
65777ba3d3fSMatthias Ringwald }
65877ba3d3fSMatthias Ringwald 
65977ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_control_pdu(mesh_pdu_t * pdu, uint16_t netkey_index,
66077ba3d3fSMatthias Ringwald                                                uint8_t ttl, uint16_t src, uint16_t dest, uint8_t opcode, const uint8_t * control_pdu_data, uint16_t control_pdu_len){
66177ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
66277ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
66377ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_unsegmented_control_pdu((mesh_network_pdu_t *) pdu, netkey_index, ttl, src, dest, opcode, control_pdu_data, control_pdu_len);
66477ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_TRANSPORT:
66577ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_segmented_control_pdu((mesh_transport_pdu_t *) pdu, netkey_index, ttl, src, dest, opcode, control_pdu_data, control_pdu_len);
66677ba3d3fSMatthias Ringwald         default:
66777ba3d3fSMatthias Ringwald             return 1;
66877ba3d3fSMatthias Ringwald     }
66977ba3d3fSMatthias Ringwald }
67077ba3d3fSMatthias Ringwald 
67177ba3d3fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_unsegmented_access_pdu_header(mesh_network_pdu_t * network_pdu, uint16_t netkey_index,
67277ba3d3fSMatthias Ringwald         uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest){
67377ba3d3fSMatthias Ringwald 
67477ba3d3fSMatthias Ringwald     // get app or device key
67577ba3d3fSMatthias Ringwald     const mesh_transport_key_t * appkey;
67677ba3d3fSMatthias Ringwald     appkey = mesh_transport_key_get(appkey_index);
67777ba3d3fSMatthias Ringwald     if (appkey == NULL){
67877ba3d3fSMatthias Ringwald         printf("appkey_index %x unknown\n", appkey_index);
67977ba3d3fSMatthias Ringwald         return 1;
68077ba3d3fSMatthias Ringwald     }
68177ba3d3fSMatthias Ringwald     uint8_t akf_aid = (appkey->akf << 6) | appkey->aid;
68277ba3d3fSMatthias Ringwald 
68377ba3d3fSMatthias Ringwald     // lookup network by netkey_index
68477ba3d3fSMatthias Ringwald     const mesh_network_key_t * network_key = mesh_network_key_list_get(netkey_index);
68577ba3d3fSMatthias Ringwald     if (!network_key) return 1;
68677ba3d3fSMatthias Ringwald 
68777ba3d3fSMatthias Ringwald     network_pdu->data[9] = akf_aid;
68877ba3d3fSMatthias Ringwald     // setup network_pdu
689e41c0017SMatthias Ringwald     mesh_network_setup_pdu_header(network_pdu, netkey_index, network_key->nid, 0, ttl, 0, src, dest);
69077ba3d3fSMatthias Ringwald     network_pdu->appkey_index = appkey_index;
69177ba3d3fSMatthias Ringwald     return 0;
69277ba3d3fSMatthias Ringwald }
69377ba3d3fSMatthias Ringwald 
69477ba3d3fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_unsegmented_access_pdu(mesh_network_pdu_t * network_pdu, uint16_t netkey_index, uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest,
69577ba3d3fSMatthias Ringwald                                                        const uint8_t * access_pdu_data, uint8_t access_pdu_len){
69677ba3d3fSMatthias Ringwald 
69777ba3d3fSMatthias Ringwald     int status = mesh_upper_transport_setup_unsegmented_access_pdu_header(network_pdu, netkey_index, appkey_index, ttl, src, dest);
69877ba3d3fSMatthias Ringwald     if (status) return status;
69977ba3d3fSMatthias Ringwald 
70077ba3d3fSMatthias Ringwald     // store in transport pdu
701*6535961aSMatthias Ringwald     (void)memcpy(&network_pdu->data[10], access_pdu_data, access_pdu_len);
70277ba3d3fSMatthias Ringwald     network_pdu->len = 10 + access_pdu_len;
70377ba3d3fSMatthias Ringwald     return 0;
70477ba3d3fSMatthias Ringwald }
70577ba3d3fSMatthias Ringwald 
706a332c5a2SMatthias Ringwald static uint8_t mesh_upper_transport_setup_segmented_access_pdu_header(mesh_transport_pdu_t * transport_pdu, uint16_t netkey_index,
707a332c5a2SMatthias Ringwald     uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic){
70877ba3d3fSMatthias Ringwald 
70977ba3d3fSMatthias Ringwald     // get app or device key
71077ba3d3fSMatthias Ringwald     const mesh_transport_key_t *appkey;
71177ba3d3fSMatthias Ringwald     appkey = mesh_transport_key_get(appkey_index);
71277ba3d3fSMatthias Ringwald     if (appkey == NULL) {
713a332c5a2SMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - appkey_index %x unknown\n", appkey_index);
71477ba3d3fSMatthias Ringwald         return 1;
71577ba3d3fSMatthias Ringwald     }
71677ba3d3fSMatthias Ringwald     uint8_t akf_aid = (appkey->akf << 6) | appkey->aid;
71777ba3d3fSMatthias Ringwald 
71877ba3d3fSMatthias Ringwald     // lookup network by netkey_index
71977ba3d3fSMatthias Ringwald     const mesh_network_key_t *network_key = mesh_network_key_list_get(netkey_index);
72077ba3d3fSMatthias Ringwald     if (!network_key) return 1;
721a332c5a2SMatthias Ringwald     if (network_key == NULL) {
722a332c5a2SMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - netkey_index %x unknown\n", appkey_index);
723a332c5a2SMatthias Ringwald         return 1;
724a332c5a2SMatthias Ringwald     }
72577ba3d3fSMatthias Ringwald 
72677ba3d3fSMatthias Ringwald     const uint8_t trans_mic_len = szmic ? 8 : 4;
72777ba3d3fSMatthias Ringwald 
72877ba3d3fSMatthias Ringwald     // store in transport pdu
72977ba3d3fSMatthias Ringwald     transport_pdu->transmic_len = trans_mic_len;
73077ba3d3fSMatthias Ringwald     transport_pdu->netkey_index = netkey_index;
73177ba3d3fSMatthias Ringwald     transport_pdu->appkey_index = appkey_index;
732e9292fe8SMatthias Ringwald     transport_pdu->akf_aid_control = akf_aid;
733d075c322SMatthias Ringwald     mesh_transport_set_nid_ivi(transport_pdu, network_key->nid | ((mesh_get_iv_index_for_tx() & 1) << 7));
73477ba3d3fSMatthias Ringwald     mesh_transport_set_src(transport_pdu, src);
73577ba3d3fSMatthias Ringwald     mesh_transport_set_dest(transport_pdu, dest);
73677ba3d3fSMatthias Ringwald     mesh_transport_set_ctl_ttl(transport_pdu, ttl);
73777ba3d3fSMatthias Ringwald     return 0;
73877ba3d3fSMatthias Ringwald }
73977ba3d3fSMatthias Ringwald 
74077ba3d3fSMatthias Ringwald 
74177ba3d3fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_segmented_access_pdu(mesh_transport_pdu_t * transport_pdu, uint16_t netkey_index, uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest,
74277ba3d3fSMatthias Ringwald                           uint8_t szmic, const uint8_t * access_pdu_data, uint8_t access_pdu_len){
74377ba3d3fSMatthias Ringwald     int status = mesh_upper_transport_setup_segmented_access_pdu_header(transport_pdu, netkey_index, appkey_index, ttl, src, dest, szmic);
74477ba3d3fSMatthias Ringwald     if (status) return status;
74577ba3d3fSMatthias Ringwald 
74677ba3d3fSMatthias Ringwald     // store in transport pdu
747*6535961aSMatthias Ringwald     (void)memcpy(transport_pdu->data, access_pdu_data, access_pdu_len);
74877ba3d3fSMatthias Ringwald     transport_pdu->len = access_pdu_len;
74977ba3d3fSMatthias Ringwald     return 0;
75077ba3d3fSMatthias Ringwald }
75177ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_access_pdu_header(mesh_pdu_t * pdu, uint16_t netkey_index, uint16_t appkey_index,
75277ba3d3fSMatthias Ringwald                                               uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic){
75377ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
75477ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
75577ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_unsegmented_access_pdu_header((mesh_network_pdu_t *) pdu, netkey_index, appkey_index, ttl, src, dest);
75677ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_TRANSPORT:
75777ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_segmented_access_pdu_header((mesh_transport_pdu_t *) pdu, netkey_index, appkey_index, ttl, src, dest, szmic);
75877ba3d3fSMatthias Ringwald         default:
75977ba3d3fSMatthias Ringwald             return 1;
76077ba3d3fSMatthias Ringwald     }
76177ba3d3fSMatthias Ringwald }
76277ba3d3fSMatthias Ringwald 
76377ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_access_pdu(mesh_pdu_t * pdu, uint16_t netkey_index, uint16_t appkey_index,
76477ba3d3fSMatthias Ringwald                                               uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic,
76577ba3d3fSMatthias Ringwald                                               const uint8_t * access_pdu_data, uint8_t access_pdu_len){
76677ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
76777ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
76877ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_unsegmented_access_pdu((mesh_network_pdu_t *) pdu, netkey_index, appkey_index, ttl, src, dest, access_pdu_data, access_pdu_len);
76977ba3d3fSMatthias Ringwald         case MESH_PDU_TYPE_TRANSPORT:
77077ba3d3fSMatthias Ringwald             return mesh_upper_transport_setup_segmented_access_pdu((mesh_transport_pdu_t *) pdu, netkey_index, appkey_index, ttl, src, dest, szmic, access_pdu_data, access_pdu_len);
77177ba3d3fSMatthias Ringwald         default:
77277ba3d3fSMatthias Ringwald             return 1;
77377ba3d3fSMatthias Ringwald     }
77477ba3d3fSMatthias Ringwald }
77577ba3d3fSMatthias Ringwald 
77677ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_unsegmented_access_pdu_digest(void * arg){
77777ba3d3fSMatthias Ringwald     mesh_network_pdu_t * network_pdu = (mesh_network_pdu_t *) arg;
77877ba3d3fSMatthias Ringwald     uint8_t * access_pdu_data = mesh_network_pdu_data(network_pdu) + 1;
77977ba3d3fSMatthias Ringwald     uint16_t  access_pdu_len  = mesh_network_pdu_len(network_pdu)  - 1;
78077ba3d3fSMatthias Ringwald     btstack_crypto_ccm_encrypt_block(&ccm, access_pdu_len, access_pdu_data, access_pdu_data, &mesh_upper_transport_send_unsegmented_access_pdu_ccm, network_pdu);
78177ba3d3fSMatthias Ringwald }
78277ba3d3fSMatthias Ringwald 
7834296b056SMatthias Ringwald static mesh_transport_key_t * mesh_upper_transport_get_outgoing_appkey(uint16_t netkey_index, uint16_t appkey_index){
7844296b056SMatthias Ringwald     // Device Key is fixed
7854296b056SMatthias Ringwald     if (appkey_index == MESH_DEVICE_KEY_INDEX) {
7864296b056SMatthias Ringwald         return mesh_transport_key_get(appkey_index);
7874296b056SMatthias Ringwald     }
7884296b056SMatthias Ringwald 
7894296b056SMatthias Ringwald     // Get key refresh state from subnet
7904296b056SMatthias Ringwald     mesh_subnet_t * subnet = mesh_subnet_get_by_netkey_index(netkey_index);
7914296b056SMatthias Ringwald     if (subnet == NULL) return NULL;
7924296b056SMatthias Ringwald 
7934296b056SMatthias Ringwald     // identify old and new app keys for given appkey_index
7944296b056SMatthias Ringwald     mesh_transport_key_t * old_key = NULL;
7954296b056SMatthias Ringwald     mesh_transport_key_t * new_key = NULL;
7964296b056SMatthias Ringwald     mesh_transport_key_iterator_t it;
7974296b056SMatthias Ringwald     mesh_transport_key_iterator_init(&it, netkey_index);
7984296b056SMatthias Ringwald     while (mesh_transport_key_iterator_has_more(&it)){
7994296b056SMatthias Ringwald         mesh_transport_key_t * transport_key = mesh_transport_key_iterator_get_next(&it);
8004296b056SMatthias Ringwald         if (transport_key->appkey_index != appkey_index) continue;
8014296b056SMatthias Ringwald         if (transport_key->old_key == 0) {
8024296b056SMatthias Ringwald             new_key = transport_key;
8034296b056SMatthias Ringwald         } else {
8044296b056SMatthias Ringwald             old_key = transport_key;
8054296b056SMatthias Ringwald         }
8064296b056SMatthias Ringwald     }
8074296b056SMatthias Ringwald 
8084296b056SMatthias Ringwald     // if no key is marked as old, just use the current one
8094296b056SMatthias Ringwald     if (old_key == NULL) return new_key;
8104296b056SMatthias Ringwald 
8114296b056SMatthias Ringwald     // use new key if it exists in phase two
8124296b056SMatthias Ringwald     if ((subnet->key_refresh == MESH_KEY_REFRESH_SECOND_PHASE) && (new_key != NULL)){
8134296b056SMatthias Ringwald         return new_key;
8144296b056SMatthias Ringwald     } else {
8154296b056SMatthias Ringwald         return old_key;
8164296b056SMatthias Ringwald     }
8174296b056SMatthias Ringwald }
8184296b056SMatthias Ringwald 
81977ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_unsegmented_access_pdu(mesh_network_pdu_t * network_pdu){
82077ba3d3fSMatthias Ringwald 
82177ba3d3fSMatthias Ringwald     // if dst is virtual address, lookup label uuid and hash
82277ba3d3fSMatthias Ringwald     uint16_t aad_len = 0;
82377ba3d3fSMatthias Ringwald     mesh_virtual_address_t * virtual_address = NULL;
82477ba3d3fSMatthias Ringwald     uint16_t dst = mesh_network_dst(network_pdu);
82577ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(dst)){
82677ba3d3fSMatthias Ringwald         virtual_address = mesh_virtual_address_for_pseudo_dst(dst);
82777ba3d3fSMatthias Ringwald         if (!virtual_address){
82877ba3d3fSMatthias Ringwald             printf("No virtual address register for pseudo dst %4x\n", dst);
82977ba3d3fSMatthias Ringwald             btstack_memory_mesh_network_pdu_free(network_pdu);
83077ba3d3fSMatthias Ringwald             return;
83177ba3d3fSMatthias Ringwald         }
83277ba3d3fSMatthias Ringwald         aad_len = 16;
83377ba3d3fSMatthias Ringwald         big_endian_store_16(network_pdu->data, 7, virtual_address->hash);
83477ba3d3fSMatthias Ringwald     }
83577ba3d3fSMatthias Ringwald 
836e41c0017SMatthias Ringwald     // reserve slot
837e41c0017SMatthias Ringwald     mesh_lower_transport_reserve_slot();
838e41c0017SMatthias Ringwald 
839e41c0017SMatthias Ringwald     // Nonce for Access Payload based on Network Sequence number: needs to be fixed now and lower layers need to send packet in right order
840e41c0017SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
841e41c0017SMatthias Ringwald     mesh_network_pdu_set_seq(network_pdu, seq);
842e41c0017SMatthias Ringwald 
843e8df83a5SMatthias Ringwald     // Dump PDU
844d703e88fSMatthias Ringwald     printf("[+] Upper transport, send unsegmented Access PDU - dest %04x, seq %06x\n", dst, mesh_network_seq(network_pdu));
845e8df83a5SMatthias Ringwald     mesh_print_hex("Access Payload", &network_pdu->data[10], network_pdu->len - 10);
846e8df83a5SMatthias Ringwald 
84777ba3d3fSMatthias Ringwald     // setup nonce
84877ba3d3fSMatthias Ringwald     uint16_t appkey_index = network_pdu->appkey_index;
84977ba3d3fSMatthias Ringwald     if (appkey_index == MESH_DEVICE_KEY_INDEX){
85077ba3d3fSMatthias Ringwald         transport_unsegmented_setup_device_nonce(application_nonce, network_pdu);
85177ba3d3fSMatthias Ringwald     } else {
85277ba3d3fSMatthias Ringwald         transport_unsegmented_setup_application_nonce(application_nonce, network_pdu);
85377ba3d3fSMatthias Ringwald     }
85477ba3d3fSMatthias Ringwald 
85577ba3d3fSMatthias Ringwald     // get app or device key
8564296b056SMatthias Ringwald     const mesh_transport_key_t * appkey = mesh_upper_transport_get_outgoing_appkey(network_pdu->netkey_index, appkey_index);
85777ba3d3fSMatthias Ringwald     mesh_print_hex("AppOrDevKey", appkey->key, 16);
85877ba3d3fSMatthias Ringwald 
85977ba3d3fSMatthias Ringwald     // encrypt ccm
86077ba3d3fSMatthias Ringwald     uint8_t   trans_mic_len = 4;
86177ba3d3fSMatthias Ringwald     uint16_t  access_pdu_len  = mesh_network_pdu_len(network_pdu)  - 1;
86277ba3d3fSMatthias Ringwald     crypto_active = 1;
86377ba3d3fSMatthias Ringwald 
86477ba3d3fSMatthias Ringwald     btstack_crypto_ccm_init(&ccm, appkey->key, application_nonce, access_pdu_len, aad_len, trans_mic_len);
86577ba3d3fSMatthias Ringwald     if (virtual_address){
86677ba3d3fSMatthias Ringwald         mesh_print_hex("LabelUUID", virtual_address->label_uuid, 16);
86777ba3d3fSMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, virtual_address->label_uuid, 16, &mesh_upper_transport_send_unsegmented_access_pdu_digest, network_pdu);
86877ba3d3fSMatthias Ringwald     } else {
86977ba3d3fSMatthias Ringwald         mesh_upper_transport_send_unsegmented_access_pdu_digest(network_pdu);
87077ba3d3fSMatthias Ringwald     }
87177ba3d3fSMatthias Ringwald }
87277ba3d3fSMatthias Ringwald 
87377ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_segmented_access_pdu_digest(void *arg){
87477ba3d3fSMatthias Ringwald     mesh_transport_pdu_t * transport_pdu = (mesh_transport_pdu_t *) arg;
87577ba3d3fSMatthias Ringwald     uint16_t  access_pdu_len  = transport_pdu->len;
87677ba3d3fSMatthias Ringwald     uint8_t * access_pdu_data = transport_pdu->data;
87777ba3d3fSMatthias Ringwald     btstack_crypto_ccm_encrypt_block(&ccm, access_pdu_len,access_pdu_data, access_pdu_data, &mesh_upper_transport_send_segmented_access_pdu_ccm, transport_pdu);
87877ba3d3fSMatthias Ringwald }
87977ba3d3fSMatthias Ringwald 
88077ba3d3fSMatthias Ringwald static void mesh_upper_transport_send_segmented_access_pdu(mesh_transport_pdu_t * transport_pdu){
88177ba3d3fSMatthias Ringwald 
88277ba3d3fSMatthias Ringwald     // if dst is virtual address, lookup label uuid and hash
88377ba3d3fSMatthias Ringwald     uint16_t aad_len = 0;
88477ba3d3fSMatthias Ringwald     mesh_virtual_address_t * virtual_address = NULL;
88577ba3d3fSMatthias Ringwald     uint16_t dst = mesh_transport_dst(transport_pdu);
88677ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(dst)){
88777ba3d3fSMatthias Ringwald         virtual_address = mesh_virtual_address_for_pseudo_dst(dst);
88877ba3d3fSMatthias Ringwald         if (!virtual_address){
88977ba3d3fSMatthias Ringwald             printf("No virtual address register for pseudo dst %4x\n", dst);
89077ba3d3fSMatthias Ringwald             btstack_memory_mesh_transport_pdu_free(transport_pdu);
89177ba3d3fSMatthias Ringwald             return;
89277ba3d3fSMatthias Ringwald         }
89377ba3d3fSMatthias Ringwald         // printf("Using hash %4x with LabelUUID: ", virtual_address->hash);
89477ba3d3fSMatthias Ringwald         // printf_hexdump(virtual_address->label_uuid, 16);
89577ba3d3fSMatthias Ringwald         aad_len = 16;
89677ba3d3fSMatthias Ringwald         big_endian_store_16(transport_pdu->network_header, 7, virtual_address->hash);
89777ba3d3fSMatthias Ringwald     }
89877ba3d3fSMatthias Ringwald 
89934b15033SMatthias Ringwald     // get app or device key
90034b15033SMatthias Ringwald     uint16_t appkey_index = transport_pdu->appkey_index;
90134b15033SMatthias Ringwald     const mesh_transport_key_t * appkey = mesh_upper_transport_get_outgoing_appkey(transport_pdu->netkey_index, appkey_index);
90234b15033SMatthias Ringwald     if (appkey == NULL){
90334b15033SMatthias Ringwald         printf("AppKey %04x not found, drop message\n", appkey_index);
90434b15033SMatthias Ringwald         btstack_memory_mesh_transport_pdu_free(transport_pdu);
90534b15033SMatthias Ringwald         return;
90634b15033SMatthias Ringwald     }
90734b15033SMatthias Ringwald 
90810cd83b4SMatthias Ringwald     // reserve slot
90910cd83b4SMatthias Ringwald     mesh_lower_transport_reserve_slot();
91010cd83b4SMatthias Ringwald 
911e41c0017SMatthias Ringwald     // reserve one sequence number, which is also used to encrypt access payload
912e41c0017SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
913e41c0017SMatthias Ringwald     transport_pdu->flags |= MESH_TRANSPORT_FLAG_SEQ_RESERVED;
914e41c0017SMatthias Ringwald     mesh_transport_set_seq(transport_pdu, seq);
915e41c0017SMatthias Ringwald 
916e8df83a5SMatthias Ringwald     // Dump PDU
917d703e88fSMatthias Ringwald     printf("[+] Upper transport, send segmented Access PDU - dest %04x, seq %06x\n", dst, mesh_transport_seq(transport_pdu));
918e8df83a5SMatthias Ringwald     mesh_print_hex("Access Payload", transport_pdu->data, transport_pdu->len);
919e8df83a5SMatthias Ringwald 
92077ba3d3fSMatthias Ringwald     // setup nonce - uses dst, so after pseudo address translation
92177ba3d3fSMatthias Ringwald     if (appkey_index == MESH_DEVICE_KEY_INDEX){
92277ba3d3fSMatthias Ringwald         transport_segmented_setup_device_nonce(application_nonce, transport_pdu);
92377ba3d3fSMatthias Ringwald     } else {
92477ba3d3fSMatthias Ringwald         transport_segmented_setup_application_nonce(application_nonce, transport_pdu);
92577ba3d3fSMatthias Ringwald     }
92677ba3d3fSMatthias Ringwald 
92734b15033SMatthias Ringwald     // Dump key
92877ba3d3fSMatthias Ringwald     mesh_print_hex("AppOrDevKey", appkey->key, 16);
92977ba3d3fSMatthias Ringwald 
93077ba3d3fSMatthias Ringwald     // encrypt ccm
93177ba3d3fSMatthias Ringwald     uint8_t   transmic_len    = transport_pdu->transmic_len;
93277ba3d3fSMatthias Ringwald     uint16_t  access_pdu_len  = transport_pdu->len;
93377ba3d3fSMatthias Ringwald     crypto_active = 1;
93477ba3d3fSMatthias Ringwald     btstack_crypto_ccm_init(&ccm, appkey->key, application_nonce, access_pdu_len, aad_len, transmic_len);
93577ba3d3fSMatthias Ringwald     if (virtual_address){
93677ba3d3fSMatthias Ringwald         mesh_print_hex("LabelUUID", virtual_address->label_uuid, 16);
93777ba3d3fSMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, virtual_address->label_uuid, 16, &mesh_upper_transport_send_segmented_access_pdu_digest, transport_pdu);
93877ba3d3fSMatthias Ringwald     } else {
93977ba3d3fSMatthias Ringwald         mesh_upper_transport_send_segmented_access_pdu_digest(transport_pdu);
94077ba3d3fSMatthias Ringwald     }
94177ba3d3fSMatthias Ringwald }
94277ba3d3fSMatthias Ringwald 
943e41c0017SMatthias Ringwald static void mesh_upper_transport_send_unsegmented_control_pdu(mesh_network_pdu_t * network_pdu){
944e41c0017SMatthias Ringwald     // reserve slot
945e41c0017SMatthias Ringwald     mesh_lower_transport_reserve_slot();
946e41c0017SMatthias Ringwald     // reserve sequence number
947e41c0017SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
948e41c0017SMatthias Ringwald     mesh_network_pdu_set_seq(network_pdu, seq);
949e8df83a5SMatthias Ringwald     // Dump PDU
950e8df83a5SMatthias Ringwald     uint8_t opcode = network_pdu->data[9];
951e8df83a5SMatthias Ringwald     printf("[+] Upper transport, send unsegmented Control PDU %p - seq %06x opcode %02x\n", network_pdu, seq, opcode);
952e8df83a5SMatthias Ringwald     mesh_print_hex("Access Payload", &network_pdu->data[10], network_pdu->len - 10);
953e41c0017SMatthias Ringwald     // send
954e41c0017SMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t *) network_pdu);
955e41c0017SMatthias Ringwald }
956e41c0017SMatthias Ringwald 
957e41c0017SMatthias Ringwald static void mesh_upper_transport_send_segmented_control_pdu(mesh_transport_pdu_t * transport_pdu){
958e41c0017SMatthias Ringwald     // reserve slot
959e41c0017SMatthias Ringwald     mesh_lower_transport_reserve_slot();
960e41c0017SMatthias Ringwald     // reserve sequence number
961e41c0017SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
962e41c0017SMatthias Ringwald     transport_pdu->flags |= MESH_TRANSPORT_FLAG_SEQ_RESERVED;
963e41c0017SMatthias Ringwald     mesh_transport_set_seq(transport_pdu, seq);
964e8df83a5SMatthias Ringwald     // Dump PDU
965e8df83a5SMatthias Ringwald     uint8_t opcode = transport_pdu->data[0];
966e8df83a5SMatthias Ringwald     printf("[+] Upper transport, send segmented Control PDU %p - seq %06x opcode %02x\n", transport_pdu, seq, opcode);
967e8df83a5SMatthias Ringwald     mesh_print_hex("Access Payload", &transport_pdu->data[1], transport_pdu->len - 1);
968e41c0017SMatthias Ringwald     // send
969e41c0017SMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t *) transport_pdu);
970e41c0017SMatthias Ringwald }
971e41c0017SMatthias Ringwald 
9725beab91eSMatthias Ringwald static void mesh_upper_transport_run(void){
973d19dd9d7SMatthias Ringwald 
97477ba3d3fSMatthias Ringwald     while(!btstack_linked_list_empty(&upper_transport_incoming)){
97577ba3d3fSMatthias Ringwald 
97677ba3d3fSMatthias Ringwald         if (crypto_active) return;
97777ba3d3fSMatthias Ringwald 
97877ba3d3fSMatthias Ringwald         // peek at next message
97977ba3d3fSMatthias Ringwald         mesh_pdu_t * pdu =  (mesh_pdu_t *) btstack_linked_list_get_first_item(&upper_transport_incoming);
98077ba3d3fSMatthias Ringwald         mesh_transport_pdu_t * transport_pdu;
98177ba3d3fSMatthias Ringwald         mesh_network_pdu_t   * network_pdu;
98277ba3d3fSMatthias Ringwald         switch (pdu->pdu_type){
98377ba3d3fSMatthias Ringwald             case MESH_PDU_TYPE_NETWORK:
98477ba3d3fSMatthias Ringwald                 network_pdu = (mesh_network_pdu_t *) pdu;
98577ba3d3fSMatthias Ringwald                 // control?
98677ba3d3fSMatthias Ringwald                 if (mesh_network_control(network_pdu)) {
98777ba3d3fSMatthias Ringwald                     (void) btstack_linked_list_pop(&upper_transport_incoming);
98877ba3d3fSMatthias Ringwald                     mesh_upper_unsegmented_control_message_received(network_pdu);
98977ba3d3fSMatthias Ringwald                 } else {
990dd729594SMatthias Ringwald                     incoming_network_pdu_decoded = mesh_network_pdu_get();
991dd729594SMatthias Ringwald                     if (!incoming_network_pdu_decoded) return;
99277ba3d3fSMatthias Ringwald                     // get encoded network pdu and start processing
993dd729594SMatthias Ringwald                     incoming_network_pdu_raw = network_pdu;
99477ba3d3fSMatthias Ringwald                     (void) btstack_linked_list_pop(&upper_transport_incoming);
995dd729594SMatthias Ringwald                     mesh_upper_transport_process_unsegmented_access_message();
99677ba3d3fSMatthias Ringwald                 }
99777ba3d3fSMatthias Ringwald                 break;
99877ba3d3fSMatthias Ringwald             case MESH_PDU_TYPE_TRANSPORT:
99977ba3d3fSMatthias Ringwald                 transport_pdu = (mesh_transport_pdu_t *) pdu;
100077ba3d3fSMatthias Ringwald                 uint8_t ctl = mesh_transport_ctl(transport_pdu);
100177ba3d3fSMatthias Ringwald                 if (ctl){
100277ba3d3fSMatthias Ringwald                     printf("Ignoring Segmented Control Message\n");
100377ba3d3fSMatthias Ringwald                     (void) btstack_linked_list_pop(&upper_transport_incoming);
100477ba3d3fSMatthias Ringwald                     mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *) transport_pdu);
100577ba3d3fSMatthias Ringwald                 } else {
1006b86620d9SMatthias Ringwald                     incoming_transport_pdu_decoded = mesh_transport_pdu_get();
1007b86620d9SMatthias Ringwald                     if (!incoming_transport_pdu_decoded) return;
100877ba3d3fSMatthias Ringwald                     // get encoded transport pdu and start processing
1009dd729594SMatthias Ringwald                     incoming_transport_pdu_raw = transport_pdu;
101077ba3d3fSMatthias Ringwald                     (void) btstack_linked_list_pop(&upper_transport_incoming);
1011b86620d9SMatthias Ringwald                     mesh_upper_transport_process_message();
101277ba3d3fSMatthias Ringwald                 }
101377ba3d3fSMatthias Ringwald                 break;
101477ba3d3fSMatthias Ringwald             default:
101577ba3d3fSMatthias Ringwald                 break;
101677ba3d3fSMatthias Ringwald         }
101777ba3d3fSMatthias Ringwald     }
10185beab91eSMatthias Ringwald 
10195beab91eSMatthias Ringwald     while (!btstack_linked_list_empty(&upper_transport_outgoing)){
10205beab91eSMatthias Ringwald 
102110cd83b4SMatthias Ringwald         if (crypto_active) break;
10225beab91eSMatthias Ringwald 
1023d19dd9d7SMatthias Ringwald         mesh_pdu_t * pdu =  (mesh_pdu_t *) btstack_linked_list_get_first_item(&upper_transport_outgoing);
102410cd83b4SMatthias Ringwald         if (mesh_lower_transport_can_send_to_dest(mesh_pdu_dst(pdu)) == 0) break;
102510cd83b4SMatthias Ringwald 
1026d19dd9d7SMatthias Ringwald         (void) btstack_linked_list_pop(&upper_transport_outgoing);
1027d19dd9d7SMatthias Ringwald 
102810cd83b4SMatthias Ringwald         if (mesh_pdu_ctl(pdu)){
1029e41c0017SMatthias Ringwald             switch (pdu->pdu_type){
1030e41c0017SMatthias Ringwald                 case MESH_PDU_TYPE_NETWORK:
1031e41c0017SMatthias Ringwald                     mesh_upper_transport_send_unsegmented_control_pdu((mesh_network_pdu_t *) pdu);
1032e41c0017SMatthias Ringwald                     break;
1033e41c0017SMatthias Ringwald                 case MESH_PDU_TYPE_TRANSPORT:
1034e41c0017SMatthias Ringwald                     mesh_upper_transport_send_segmented_control_pdu((mesh_transport_pdu_t *) pdu);
1035e41c0017SMatthias Ringwald                     break;
1036e41c0017SMatthias Ringwald                 default:
1037e41c0017SMatthias Ringwald                     break;
103810cd83b4SMatthias Ringwald             }
1039e41c0017SMatthias Ringwald         } else {
10405beab91eSMatthias Ringwald             switch (pdu->pdu_type){
10415beab91eSMatthias Ringwald                 case MESH_PDU_TYPE_NETWORK:
10425beab91eSMatthias Ringwald                     mesh_upper_transport_send_unsegmented_access_pdu((mesh_network_pdu_t *) pdu);
10435beab91eSMatthias Ringwald                     break;
10445beab91eSMatthias Ringwald                 case MESH_PDU_TYPE_TRANSPORT:
10455beab91eSMatthias Ringwald                     mesh_upper_transport_send_segmented_access_pdu((mesh_transport_pdu_t *) pdu);
10465beab91eSMatthias Ringwald                     break;
10475beab91eSMatthias Ringwald                 default:
10485beab91eSMatthias Ringwald                     break;
10495beab91eSMatthias Ringwald             }
10505beab91eSMatthias Ringwald         }
10515beab91eSMatthias Ringwald     }
1052e41c0017SMatthias Ringwald }
10535beab91eSMatthias Ringwald 
10545beab91eSMatthias Ringwald void mesh_upper_transport_register_access_message_handler(void (*callback)(mesh_pdu_t *pdu)){
10555beab91eSMatthias Ringwald     mesh_access_message_handler = callback;
10565beab91eSMatthias Ringwald }
10575beab91eSMatthias Ringwald 
10585beab91eSMatthias Ringwald void mesh_upper_transport_register_control_message_handler(void (*callback)(mesh_pdu_t *pdu)){
10595beab91eSMatthias Ringwald     mesh_control_message_handler = callback;
10605beab91eSMatthias Ringwald }
10615beab91eSMatthias Ringwald 
10625beab91eSMatthias Ringwald void mesh_upper_transport_set_higher_layer_handler(void (*pdu_handler)( mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu)){
10635beab91eSMatthias Ringwald     higher_layer_handler = pdu_handler;
10645beab91eSMatthias Ringwald }
10655beab91eSMatthias Ringwald 
10665beab91eSMatthias Ringwald void mesh_upper_transport_init(){
10675beab91eSMatthias Ringwald     mesh_lower_transport_set_higher_layer_handler(&mesh_upper_transport_pdu_handler);
10685beab91eSMatthias Ringwald }
10695beab91eSMatthias Ringwald 
1070e6a669a7SMatthias Ringwald // TODO: higher layer define used for assert
1071e6a669a7SMatthias Ringwald #define MESH_ACCESS_OPCODE_NOT_SET 0xFFFFFFFEu
1072e6a669a7SMatthias Ringwald 
10735beab91eSMatthias Ringwald void mesh_upper_transport_send_access_pdu(mesh_pdu_t *pdu){
1074e6a669a7SMatthias Ringwald     btstack_assert(pdu->ack_opcode != MESH_ACCESS_OPCODE_NOT_SET);
1075e6a669a7SMatthias Ringwald 
10765beab91eSMatthias Ringwald     if (pdu->pdu_type == MESH_PDU_TYPE_NETWORK){
107751b20e01SMatthias Ringwald         btstack_assert( ((mesh_network_pdu_t *) pdu)->len >= 9);
10785beab91eSMatthias Ringwald     }
1079d4e23df0SMatthias Ringwald 
10805beab91eSMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_outgoing, (btstack_linked_item_t*) pdu);
10815beab91eSMatthias Ringwald     mesh_upper_transport_run();
108277ba3d3fSMatthias Ringwald }
108310cd83b4SMatthias Ringwald 
108410cd83b4SMatthias Ringwald void mesh_upper_transport_send_control_pdu(mesh_pdu_t * pdu){
108510cd83b4SMatthias Ringwald     if (pdu->pdu_type == MESH_PDU_TYPE_NETWORK){
108610cd83b4SMatthias Ringwald         btstack_assert( ((mesh_network_pdu_t *) pdu)->len >= 9);
108710cd83b4SMatthias Ringwald     }
108810cd83b4SMatthias Ringwald 
108910cd83b4SMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_outgoing, (btstack_linked_item_t*) pdu);
109010cd83b4SMatthias Ringwald     mesh_upper_transport_run();
109110cd83b4SMatthias Ringwald }
1092