xref: /btstack/src/mesh/mesh_upper_transport.c (revision 16fd08f92830ca12e481ef4da53f0fe5a9c71221)
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 
38d8b26eceSMatthias Ringwald #include <stdarg.h>
39d8b26eceSMatthias Ringwald #include "btstack_tlv.h"
40d8b26eceSMatthias Ringwald #include "mesh/mesh_foundation.h"
41d8b26eceSMatthias Ringwald #include "mesh_upper_transport.h"
42d8b26eceSMatthias Ringwald #include "mesh/mesh.h"
43d8b26eceSMatthias Ringwald #include "mesh/mesh_proxy.h"
44d8b26eceSMatthias Ringwald #include "mesh/mesh_node.h"
45d8b26eceSMatthias Ringwald 
465beab91eSMatthias Ringwald #define BTSTACK_FILE__ "mesh_upper_transport.c"
47f4854a5eSMatthias Ringwald 
48f4854a5eSMatthias Ringwald #include "mesh/mesh_upper_transport.h"
4977ba3d3fSMatthias Ringwald 
5077ba3d3fSMatthias Ringwald #include <stdio.h>
5177ba3d3fSMatthias Ringwald #include <stdlib.h>
5277ba3d3fSMatthias Ringwald #include <string.h>
53f4854a5eSMatthias Ringwald 
5477ba3d3fSMatthias Ringwald #include "btstack_util.h"
5577ba3d3fSMatthias Ringwald #include "btstack_memory.h"
565beab91eSMatthias Ringwald #include "btstack_debug.h"
575368239bSMatthias Ringwald #include "btstack_bool.h"
58f4854a5eSMatthias Ringwald 
59f4854a5eSMatthias Ringwald #include "mesh/beacon.h"
60f4854a5eSMatthias Ringwald #include "mesh/mesh_iv_index_seq_number.h"
61f4854a5eSMatthias Ringwald #include "mesh/mesh_keys.h"
62f4854a5eSMatthias Ringwald #include "mesh/mesh_lower_transport.h"
63f4854a5eSMatthias Ringwald #include "mesh/mesh_peer.h"
64f4854a5eSMatthias Ringwald #include "mesh/mesh_virtual_addresses.h"
6577ba3d3fSMatthias Ringwald 
6610cd83b4SMatthias Ringwald // TODO: extract mesh_pdu functions into lower transport or network
6710cd83b4SMatthias Ringwald #include "mesh/mesh_access.h"
6810cd83b4SMatthias Ringwald 
6977ba3d3fSMatthias Ringwald // combined key x address iterator for upper transport decryption
7077ba3d3fSMatthias Ringwald 
7177ba3d3fSMatthias Ringwald typedef struct {
7277ba3d3fSMatthias Ringwald     // state
7377ba3d3fSMatthias Ringwald     mesh_transport_key_iterator_t  key_it;
7477ba3d3fSMatthias Ringwald     mesh_virtual_address_iterator_t address_it;
7577ba3d3fSMatthias Ringwald     // elements
7677ba3d3fSMatthias Ringwald     const mesh_transport_key_t *   key;
7777ba3d3fSMatthias Ringwald     const mesh_virtual_address_t * address;
7877ba3d3fSMatthias Ringwald     // address - might be virtual
7977ba3d3fSMatthias Ringwald     uint16_t dst;
8077ba3d3fSMatthias Ringwald     // key info
8177ba3d3fSMatthias Ringwald } mesh_transport_key_and_virtual_address_iterator_t;
8277ba3d3fSMatthias Ringwald 
83274c04d6SMatthias Ringwald static void mesh_upper_transport_validate_access_message(void);
84cae44b3bSMatthias Ringwald static void mesh_upper_transport_run(void);
85cae44b3bSMatthias Ringwald 
8643d07e7fSMatthias Ringwald // upper transport callbacks - in access layer
8743d07e7fSMatthias Ringwald static void (*mesh_access_message_handler)( mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu);
8843d07e7fSMatthias Ringwald static void (*mesh_control_message_handler)( mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu);
8943d07e7fSMatthias Ringwald 
9043d07e7fSMatthias Ringwald //
91cae44b3bSMatthias Ringwald static int crypto_active;
9243d07e7fSMatthias Ringwald static uint8_t application_nonce[13];
9343d07e7fSMatthias Ringwald static btstack_crypto_ccm_t ccm;
9443d07e7fSMatthias Ringwald static mesh_transport_key_and_virtual_address_iterator_t mesh_transport_key_it;
95160d2809SMatthias Ringwald 
96274c04d6SMatthias Ringwald // incoming segmented (mesh_segmented_pdu_t) or unsegmented (network_pdu_t)
97da491525SMatthias Ringwald static mesh_pdu_t *          incoming_access_encrypted;
98da491525SMatthias Ringwald 
99274c04d6SMatthias Ringwald // multi-purpose union: segmented control reassembly, decrypted access pdu
1008fb906aaSMatthias Ringwald static union {
1018fb906aaSMatthias Ringwald     mesh_control_pdu_t    control;
1028fb906aaSMatthias Ringwald     mesh_access_pdu_t     access;
1038fb906aaSMatthias Ringwald } incoming_pdu_singleton;
1048fb906aaSMatthias Ringwald 
105274c04d6SMatthias Ringwald // pointer to incoming_pdu_singleton.access
106274c04d6SMatthias Ringwald static mesh_access_pdu_t *   incoming_access_decrypted;
107274c04d6SMatthias Ringwald 
108274c04d6SMatthias Ringwald // pointer to incoming_pdu_singleton.access
109274c04d6SMatthias Ringwald static mesh_control_pdu_t *  incoming_control_pdu;
110274c04d6SMatthias Ringwald 
111cae44b3bSMatthias Ringwald // incoming unsegmented (network) and segmented (transport) control and access messages
112cae44b3bSMatthias Ringwald static btstack_linked_list_t upper_transport_incoming;
113cae44b3bSMatthias Ringwald 
114ee7327abSMatthias Ringwald // outgoing unsegmented and segmented control and access messages
115cae44b3bSMatthias Ringwald static btstack_linked_list_t upper_transport_outgoing;
116cae44b3bSMatthias Ringwald 
117ee7327abSMatthias Ringwald // outgoing upper transport messages that have been sent to lower transport and wait for sent event
118ee7327abSMatthias Ringwald static btstack_linked_list_t upper_transport_outgoing_active;
119cae44b3bSMatthias Ringwald 
120cae44b3bSMatthias Ringwald // TODO: higher layer define used for assert
121cae44b3bSMatthias Ringwald #define MESH_ACCESS_OPCODE_NOT_SET 0xFFFFFFFEu
122cae44b3bSMatthias Ringwald 
123cae44b3bSMatthias Ringwald static void mesh_print_hex(const char * name, const uint8_t * data, uint16_t len){
124cae44b3bSMatthias Ringwald     printf("%-20s ", name);
125cae44b3bSMatthias Ringwald     printf_hexdump(data, len);
126cae44b3bSMatthias Ringwald }
127cae44b3bSMatthias Ringwald // static void mesh_print_x(const char * name, uint32_t value){
128cae44b3bSMatthias Ringwald //     printf("%20s: 0x%x", name, (int) value);
129cae44b3bSMatthias Ringwald // }
130cae44b3bSMatthias Ringwald 
13177ba3d3fSMatthias Ringwald static void mesh_transport_key_and_virtual_address_iterator_init(mesh_transport_key_and_virtual_address_iterator_t *it,
13277ba3d3fSMatthias Ringwald                                                                  uint16_t dst, uint16_t netkey_index, uint8_t akf,
13377ba3d3fSMatthias Ringwald                                                                  uint8_t aid) {
13477ba3d3fSMatthias Ringwald     printf("KEY_INIT: dst %04x, akf %x, aid %x\n", dst, akf, aid);
13577ba3d3fSMatthias Ringwald     // config
13677ba3d3fSMatthias Ringwald     it->dst   = dst;
13777ba3d3fSMatthias Ringwald     // init elements
13877ba3d3fSMatthias Ringwald     it->key     = NULL;
13977ba3d3fSMatthias Ringwald     it->address = NULL;
14077ba3d3fSMatthias Ringwald     // init element iterators
14177ba3d3fSMatthias Ringwald     mesh_transport_key_aid_iterator_init(&it->key_it, netkey_index, akf, aid);
14277ba3d3fSMatthias Ringwald     // init address iterator
14377ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)){
14477ba3d3fSMatthias Ringwald         mesh_virtual_address_iterator_init(&it->address_it, dst);
14577ba3d3fSMatthias Ringwald         // get first key
14677ba3d3fSMatthias Ringwald         if (mesh_transport_key_aid_iterator_has_more(&it->key_it)) {
14777ba3d3fSMatthias Ringwald             it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
14877ba3d3fSMatthias Ringwald         }
14977ba3d3fSMatthias Ringwald     }
15077ba3d3fSMatthias Ringwald }
15177ba3d3fSMatthias Ringwald 
15277ba3d3fSMatthias Ringwald // cartesian product: keys x addressses
15377ba3d3fSMatthias Ringwald static int mesh_transport_key_and_virtual_address_iterator_has_more(mesh_transport_key_and_virtual_address_iterator_t * it){
15477ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)) {
15577ba3d3fSMatthias Ringwald         // find next valid entry
156ff3cc4a5SMatthias Ringwald         while (true){
15777ba3d3fSMatthias Ringwald             if (mesh_virtual_address_iterator_has_more(&it->address_it)) return 1;
15877ba3d3fSMatthias Ringwald             if (!mesh_transport_key_aid_iterator_has_more(&it->key_it)) return 0;
15977ba3d3fSMatthias Ringwald             // get next key
16077ba3d3fSMatthias Ringwald             it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
16177ba3d3fSMatthias Ringwald             mesh_virtual_address_iterator_init(&it->address_it, it->dst);
16277ba3d3fSMatthias Ringwald         }
16377ba3d3fSMatthias Ringwald     } else {
16477ba3d3fSMatthias Ringwald         return mesh_transport_key_aid_iterator_has_more(&it->key_it);
16577ba3d3fSMatthias Ringwald     }
16677ba3d3fSMatthias Ringwald }
16777ba3d3fSMatthias Ringwald 
16877ba3d3fSMatthias Ringwald static void mesh_transport_key_and_virtual_address_iterator_next(mesh_transport_key_and_virtual_address_iterator_t * it){
16977ba3d3fSMatthias Ringwald     if (mesh_network_address_virtual(it->dst)) {
17077ba3d3fSMatthias Ringwald         it->address = mesh_virtual_address_iterator_get_next(&it->address_it);
17177ba3d3fSMatthias Ringwald     } else {
17277ba3d3fSMatthias Ringwald         it->key = mesh_transport_key_aid_iterator_get_next(&it->key_it);
17377ba3d3fSMatthias Ringwald     }
17477ba3d3fSMatthias Ringwald }
17577ba3d3fSMatthias Ringwald 
17677ba3d3fSMatthias Ringwald // UPPER TRANSPORT
17777ba3d3fSMatthias Ringwald 
178242605faSMatthias Ringwald static void mesh_segmented_pdu_flatten(btstack_linked_list_t * segments, uint8_t segment_len, uint8_t * buffer) {
179242605faSMatthias Ringwald     // assemble payload
180242605faSMatthias Ringwald     btstack_linked_list_iterator_t it;
181242605faSMatthias Ringwald     btstack_linked_list_iterator_init(&it, segments);
182242605faSMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)) {
183242605faSMatthias Ringwald         mesh_network_pdu_t *segment = (mesh_network_pdu_t *) btstack_linked_list_iterator_next(&it);
184242605faSMatthias Ringwald         btstack_assert(segment->pdu_header.pdu_type == MESH_PDU_TYPE_NETWORK);
185*16fd08f9SMatthias Ringwald         uint8_t offset = 0;
186*16fd08f9SMatthias Ringwald         while (offset < segment->len){
187*16fd08f9SMatthias Ringwald             uint8_t seg_o = segment->data[offset++];
188*16fd08f9SMatthias Ringwald             (void) memcpy(&buffer[seg_o * segment_len], &segment->data[offset], segment_len);
189*16fd08f9SMatthias Ringwald             offset += segment_len;
190*16fd08f9SMatthias Ringwald         }
191242605faSMatthias Ringwald     }
192242605faSMatthias Ringwald }
193242605faSMatthias Ringwald 
194242605faSMatthias Ringwald static uint16_t mesh_upper_pdu_flatten(mesh_upper_transport_pdu_t * upper_pdu, uint8_t * buffer, uint16_t buffer_len) {
195242605faSMatthias Ringwald     // assemble payload
196242605faSMatthias Ringwald     btstack_linked_list_iterator_t it;
1971b51af65SMatthias Ringwald     btstack_linked_list_iterator_init(&it, &upper_pdu->segments);
198242605faSMatthias Ringwald     uint16_t offset = 0;
199242605faSMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)) {
200242605faSMatthias Ringwald         mesh_network_pdu_t *segment = (mesh_network_pdu_t *) btstack_linked_list_iterator_next(&it);
201242605faSMatthias Ringwald         btstack_assert(segment->pdu_header.pdu_type == MESH_PDU_TYPE_NETWORK);
202242605faSMatthias Ringwald         btstack_assert((offset + segment->len) <= buffer_len);
203242605faSMatthias Ringwald         (void) memcpy(&buffer[offset], segment->data, segment->len);
204242605faSMatthias Ringwald         offset += segment->len;
205242605faSMatthias Ringwald     }
206242605faSMatthias Ringwald     return offset;
207242605faSMatthias Ringwald }
208242605faSMatthias Ringwald 
2095368239bSMatthias Ringwald // store payload in provided list of network pdus
2105368239bSMatthias Ringwald static void mesh_segmented_store_payload(const uint8_t * payload, uint16_t payload_len, btstack_linked_list_t * in_segments, btstack_linked_list_t * out_segments){
211242605faSMatthias Ringwald     uint16_t payload_offset = 0;
212242605faSMatthias Ringwald     uint16_t bytes_current_segment = 0;
2135368239bSMatthias Ringwald     mesh_network_pdu_t * network_pdu = NULL;
214242605faSMatthias Ringwald     while (payload_offset < payload_len){
215242605faSMatthias Ringwald         if (bytes_current_segment == 0){
2165368239bSMatthias Ringwald             network_pdu = (mesh_network_pdu_t *) btstack_linked_list_pop(in_segments);
217242605faSMatthias Ringwald             btstack_assert(network_pdu != NULL);
2185368239bSMatthias Ringwald             btstack_linked_list_add_tail(out_segments, (btstack_linked_item_t *) network_pdu);
219242605faSMatthias Ringwald             bytes_current_segment = MESH_NETWORK_PAYLOAD_MAX;
220242605faSMatthias Ringwald         }
221242605faSMatthias Ringwald         uint16_t bytes_to_copy = btstack_min(bytes_current_segment, payload_len - payload_offset);
222242605faSMatthias Ringwald         (void) memcpy(&network_pdu->data[network_pdu->len], &payload[payload_offset], bytes_to_copy);
223242605faSMatthias Ringwald         bytes_current_segment -= bytes_to_copy;
224242605faSMatthias Ringwald         network_pdu->len += bytes_to_copy;
225242605faSMatthias Ringwald         payload_offset += bytes_to_copy;
226242605faSMatthias Ringwald     }
227242605faSMatthias Ringwald }
228242605faSMatthias Ringwald 
2295368239bSMatthias Ringwald // tries allocate and add enough segments to store payload of given size
2305368239bSMatthias Ringwald static bool mesh_segmented_allocate_segments(btstack_linked_list_t * segments, uint16_t payload_len){
2315368239bSMatthias Ringwald     uint16_t storage_size = btstack_linked_list_count(segments) * MESH_NETWORK_PAYLOAD_MAX;
2325368239bSMatthias Ringwald     while (storage_size < payload_len){
2335368239bSMatthias Ringwald         mesh_network_pdu_t * network_pdu = mesh_network_pdu_get();
2345368239bSMatthias Ringwald         if (network_pdu == NULL) break;
2355368239bSMatthias Ringwald         storage_size += MESH_NETWORK_PAYLOAD_MAX;
2365368239bSMatthias Ringwald         btstack_linked_list_add(segments, (btstack_linked_item_t *) network_pdu);
2375368239bSMatthias Ringwald     }
2385368239bSMatthias Ringwald     return (storage_size >= payload_len);
2395368239bSMatthias Ringwald }
2405368239bSMatthias Ringwald 
24177ba3d3fSMatthias Ringwald // stub lower transport
24277ba3d3fSMatthias Ringwald 
243dd729594SMatthias Ringwald static void mesh_upper_transport_dump_pdus(const char *name, btstack_linked_list_t *list){
244dd729594SMatthias Ringwald     printf("List: %s:\n", name);
245dd729594SMatthias Ringwald     btstack_linked_list_iterator_t it;
246dd729594SMatthias Ringwald     btstack_linked_list_iterator_init(&it, list);
247dd729594SMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)){
248dd729594SMatthias Ringwald         mesh_pdu_t * pdu = (mesh_pdu_t*) btstack_linked_list_iterator_next(&it);
249dd729594SMatthias Ringwald         printf("- %p\n", pdu);
250dd729594SMatthias Ringwald         // printf_hexdump( mesh_pdu_data(pdu), mesh_pdu_len(pdu));
251dd729594SMatthias Ringwald     }
252dd729594SMatthias Ringwald }
253dd729594SMatthias Ringwald 
254dd729594SMatthias Ringwald static void mesh_upper_transport_reset_pdus(btstack_linked_list_t *list){
255dd729594SMatthias Ringwald     while (!btstack_linked_list_empty(list)){
25659613427SMatthias Ringwald         mesh_upper_transport_pdu_free((mesh_pdu_t *) btstack_linked_list_pop(list));
257dd729594SMatthias Ringwald     }
258dd729594SMatthias Ringwald }
259dd729594SMatthias Ringwald 
260dd729594SMatthias Ringwald void mesh_upper_transport_dump(void){
261dd729594SMatthias Ringwald     mesh_upper_transport_dump_pdus("upper_transport_incoming", &upper_transport_incoming);
262dd729594SMatthias Ringwald }
263dd729594SMatthias Ringwald 
264dd729594SMatthias Ringwald void mesh_upper_transport_reset(void){
265dd729594SMatthias Ringwald     crypto_active = 0;
266b86620d9SMatthias Ringwald     mesh_upper_transport_reset_pdus(&upper_transport_incoming);
267dd729594SMatthias Ringwald }
26877ba3d3fSMatthias Ringwald 
269e9c16304SMatthias Ringwald static mesh_transport_key_t * mesh_upper_transport_get_outgoing_appkey(uint16_t netkey_index, uint16_t appkey_index){
270e9c16304SMatthias Ringwald     // Device Key is fixed
271e9c16304SMatthias Ringwald     if (appkey_index == MESH_DEVICE_KEY_INDEX) {
272e9c16304SMatthias Ringwald         return mesh_transport_key_get(appkey_index);
273e9c16304SMatthias Ringwald     }
274e9c16304SMatthias Ringwald 
275e9c16304SMatthias Ringwald     // Get key refresh state from subnet
276e9c16304SMatthias Ringwald     mesh_subnet_t * subnet = mesh_subnet_get_by_netkey_index(netkey_index);
277e9c16304SMatthias Ringwald     if (subnet == NULL) return NULL;
278e9c16304SMatthias Ringwald 
279e9c16304SMatthias Ringwald     // identify old and new app keys for given appkey_index
280e9c16304SMatthias Ringwald     mesh_transport_key_t * old_key = NULL;
281e9c16304SMatthias Ringwald     mesh_transport_key_t * new_key = NULL;
282e9c16304SMatthias Ringwald     mesh_transport_key_iterator_t it;
283e9c16304SMatthias Ringwald     mesh_transport_key_iterator_init(&it, netkey_index);
284e9c16304SMatthias Ringwald     while (mesh_transport_key_iterator_has_more(&it)){
285e9c16304SMatthias Ringwald         mesh_transport_key_t * transport_key = mesh_transport_key_iterator_get_next(&it);
286e9c16304SMatthias Ringwald         if (transport_key->appkey_index != appkey_index) continue;
287e9c16304SMatthias Ringwald         if (transport_key->old_key == 0) {
288e9c16304SMatthias Ringwald             new_key = transport_key;
289e9c16304SMatthias Ringwald         } else {
290e9c16304SMatthias Ringwald             old_key = transport_key;
291e9c16304SMatthias Ringwald         }
292e9c16304SMatthias Ringwald     }
293e9c16304SMatthias Ringwald 
294e9c16304SMatthias Ringwald     // if no key is marked as old, just use the current one
295e9c16304SMatthias Ringwald     if (old_key == NULL) return new_key;
296e9c16304SMatthias Ringwald 
297e9c16304SMatthias Ringwald     // use new key if it exists in phase two
298e9c16304SMatthias Ringwald     if ((subnet->key_refresh == MESH_KEY_REFRESH_SECOND_PHASE) && (new_key != NULL)){
299e9c16304SMatthias Ringwald         return new_key;
300e9c16304SMatthias Ringwald     } else {
301e9c16304SMatthias Ringwald         return old_key;
302e9c16304SMatthias Ringwald     }
303e9c16304SMatthias Ringwald }
304e9c16304SMatthias Ringwald 
305cae44b3bSMatthias Ringwald static uint32_t iv_index_for_ivi_nid(uint8_t ivi_nid){
306cae44b3bSMatthias Ringwald     // get IV Index and IVI
307cae44b3bSMatthias Ringwald     uint32_t iv_index = mesh_get_iv_index();
308cae44b3bSMatthias Ringwald     int ivi = ivi_nid >> 7;
309cae44b3bSMatthias Ringwald 
310cae44b3bSMatthias Ringwald     // if least significant bit differs, use previous IV Index
311cae44b3bSMatthias Ringwald     if ((iv_index & 1 ) ^ ivi){
312cae44b3bSMatthias Ringwald         iv_index--;
313cae44b3bSMatthias Ringwald     }
314cae44b3bSMatthias Ringwald     return iv_index;
315cae44b3bSMatthias Ringwald }
316cae44b3bSMatthias Ringwald 
317160d2809SMatthias Ringwald static void transport_segmented_setup_nonce(uint8_t * nonce, const mesh_pdu_t * pdu){
318160d2809SMatthias Ringwald     mesh_access_pdu_t * access_pdu;
319242605faSMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu;
320160d2809SMatthias Ringwald     switch (pdu->pdu_type){
321160d2809SMatthias Ringwald         case MESH_PDU_TYPE_ACCESS:
322160d2809SMatthias Ringwald             access_pdu = (mesh_access_pdu_t *) pdu;
323160d2809SMatthias Ringwald             nonce[1] = access_pdu->transmic_len == 8 ? 0x80 : 0x00;
324c3fc3eeaSMatthias Ringwald             big_endian_store_24(nonce, 2, access_pdu->seq);
325c3fc3eeaSMatthias Ringwald             big_endian_store_16(nonce, 5, access_pdu->src);
326c3fc3eeaSMatthias Ringwald             big_endian_store_16(nonce, 7, access_pdu->dst);
327c3fc3eeaSMatthias Ringwald             big_endian_store_32(nonce, 9, iv_index_for_ivi_nid(access_pdu->ivi_nid));
328160d2809SMatthias Ringwald             break;
329242605faSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
3301b51af65SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
331242605faSMatthias Ringwald             upper_pdu = (mesh_upper_transport_pdu_t *) pdu;
332242605faSMatthias Ringwald             nonce[1] = upper_pdu->transmic_len == 8 ? 0x80 : 0x00;
333242605faSMatthias Ringwald             // 'network header'
334242605faSMatthias Ringwald             big_endian_store_24(nonce, 2, upper_pdu->seq);
335242605faSMatthias Ringwald             big_endian_store_16(nonce, 5, upper_pdu->src);
336242605faSMatthias Ringwald             big_endian_store_16(nonce, 7, upper_pdu->dst);
337242605faSMatthias Ringwald             big_endian_store_32(nonce, 9, iv_index_for_ivi_nid(upper_pdu->ivi_nid));
338242605faSMatthias Ringwald             break;
339160d2809SMatthias Ringwald         default:
340160d2809SMatthias Ringwald             btstack_assert(0);
341160d2809SMatthias Ringwald             break;
342160d2809SMatthias Ringwald     }
343cae44b3bSMatthias Ringwald }
344cae44b3bSMatthias Ringwald 
345160d2809SMatthias Ringwald static void transport_segmented_setup_application_nonce(uint8_t * nonce, const mesh_pdu_t * pdu){
346cae44b3bSMatthias Ringwald     nonce[0] = 0x01;
347160d2809SMatthias Ringwald     transport_segmented_setup_nonce(nonce, pdu);
348cae44b3bSMatthias Ringwald     mesh_print_hex("AppNonce", nonce, 13);
349cae44b3bSMatthias Ringwald }
350cae44b3bSMatthias Ringwald 
351160d2809SMatthias Ringwald static void transport_segmented_setup_device_nonce(uint8_t * nonce, const mesh_pdu_t * pdu){
352cae44b3bSMatthias Ringwald     nonce[0] = 0x02;
353160d2809SMatthias Ringwald     transport_segmented_setup_nonce(nonce, pdu);
354cae44b3bSMatthias Ringwald     mesh_print_hex("DeviceNonce", nonce, 13);
355cae44b3bSMatthias Ringwald }
356cae44b3bSMatthias Ringwald 
3575131e195SMatthias Ringwald static void mesh_upper_transport_process_access_message_done(mesh_access_pdu_t *access_pdu){
35877ba3d3fSMatthias Ringwald     crypto_active = 0;
3592ddfd9e2SMatthias Ringwald     btstack_assert((access_pdu->ctl_ttl & 0x80) == 0);
360da491525SMatthias Ringwald     mesh_lower_transport_message_processed_by_higher_layer(incoming_access_encrypted);
361da491525SMatthias Ringwald     incoming_access_encrypted = NULL;
362274c04d6SMatthias Ringwald     incoming_access_decrypted = NULL;
3635beab91eSMatthias Ringwald     mesh_upper_transport_run();
36477ba3d3fSMatthias Ringwald }
36577ba3d3fSMatthias Ringwald 
3665131e195SMatthias Ringwald static void mesh_upper_transport_process_control_message_done(mesh_control_pdu_t * control_pdu){
3675131e195SMatthias Ringwald     crypto_active = 0;
3685131e195SMatthias Ringwald     incoming_control_pdu = NULL;
3695131e195SMatthias Ringwald     mesh_upper_transport_run();
3705131e195SMatthias Ringwald }
3715131e195SMatthias Ringwald 
372274c04d6SMatthias Ringwald static void mesh_upper_transport_validate_access_message_ccm(void * arg){
373b86620d9SMatthias Ringwald     UNUSED(arg);
37477ba3d3fSMatthias Ringwald 
375274c04d6SMatthias Ringwald     uint8_t * upper_transport_pdu     = incoming_access_decrypted->data;
376274c04d6SMatthias Ringwald     uint8_t   upper_transport_pdu_len = incoming_access_decrypted->len - incoming_access_decrypted->transmic_len;
37777ba3d3fSMatthias Ringwald 
37877ba3d3fSMatthias Ringwald     mesh_print_hex("Decrypted PDU", upper_transport_pdu, upper_transport_pdu_len);
37977ba3d3fSMatthias Ringwald 
38077ba3d3fSMatthias Ringwald     // store TransMIC
38177ba3d3fSMatthias Ringwald     uint8_t trans_mic[8];
38277ba3d3fSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, trans_mic);
383274c04d6SMatthias Ringwald     mesh_print_hex("TransMIC", trans_mic, incoming_access_decrypted->transmic_len);
38477ba3d3fSMatthias Ringwald 
385274c04d6SMatthias Ringwald     if (memcmp(trans_mic, &upper_transport_pdu[upper_transport_pdu_len], incoming_access_decrypted->transmic_len) == 0){
38677ba3d3fSMatthias Ringwald         printf("TransMIC matches\n");
38777ba3d3fSMatthias Ringwald 
38877ba3d3fSMatthias Ringwald         // remove TransMIC from payload
389274c04d6SMatthias Ringwald         incoming_access_decrypted->len -= incoming_access_decrypted->transmic_len;
39077ba3d3fSMatthias Ringwald 
39177ba3d3fSMatthias Ringwald         // if virtual address, update dst to pseudo_dst
3922ddfd9e2SMatthias Ringwald         if (mesh_network_address_virtual(incoming_access_decrypted->dst)){
393c3fc3eeaSMatthias Ringwald             incoming_access_decrypted->dst = mesh_transport_key_it.address->pseudo_dst;
39477ba3d3fSMatthias Ringwald         }
39577ba3d3fSMatthias Ringwald 
39677ba3d3fSMatthias Ringwald         // pass to upper layer
3977667b5d1SMatthias Ringwald         btstack_assert(mesh_access_message_handler != NULL);
398274c04d6SMatthias Ringwald         mesh_pdu_t * pdu = (mesh_pdu_t*) incoming_access_decrypted;
399404d2482SMatthias Ringwald         mesh_access_message_handler(MESH_TRANSPORT_PDU_RECEIVED, MESH_TRANSPORT_STATUS_SUCCESS, pdu);
40077ba3d3fSMatthias Ringwald 
40177ba3d3fSMatthias Ringwald         printf("\n");
40277ba3d3fSMatthias Ringwald 
40377ba3d3fSMatthias Ringwald     } else {
404274c04d6SMatthias Ringwald         uint8_t akf = incoming_access_decrypted->akf_aid_control & 0x40;
40577ba3d3fSMatthias Ringwald         if (akf){
40677ba3d3fSMatthias Ringwald             printf("TransMIC does not match, try next key\n");
407274c04d6SMatthias Ringwald             mesh_upper_transport_validate_access_message();
40877ba3d3fSMatthias Ringwald         } else {
40977ba3d3fSMatthias Ringwald             printf("TransMIC does not match device key, done\n");
41077ba3d3fSMatthias Ringwald             // done
411274c04d6SMatthias Ringwald             mesh_upper_transport_process_access_message_done(incoming_access_decrypted);
41277ba3d3fSMatthias Ringwald         }
41377ba3d3fSMatthias Ringwald     }
41477ba3d3fSMatthias Ringwald }
41577ba3d3fSMatthias Ringwald 
416274c04d6SMatthias Ringwald static void mesh_upper_transport_validate_access_message_digest(void * arg){
417b86620d9SMatthias Ringwald     UNUSED(arg);
418274c04d6SMatthias Ringwald     uint8_t   upper_transport_pdu_len      = incoming_access_decrypted->len - incoming_access_decrypted->transmic_len;
419274c04d6SMatthias Ringwald     uint8_t * upper_transport_pdu_data_in  = incoming_access_decrypted->data;
420274c04d6SMatthias Ringwald     uint8_t * upper_transport_pdu_data_out = incoming_access_decrypted->data;
421274c04d6SMatthias Ringwald 
422274c04d6SMatthias Ringwald     mesh_network_pdu_t * unsegmented_pdu = NULL;
423274c04d6SMatthias Ringwald     mesh_segmented_pdu_t * segmented_pdu = NULL;
424274c04d6SMatthias Ringwald     switch (incoming_access_encrypted->pdu_type){
425274c04d6SMatthias Ringwald         case MESH_PDU_TYPE_SEGMENTED:
426274c04d6SMatthias Ringwald             segmented_pdu = (mesh_segmented_pdu_t *) incoming_access_encrypted;
427274c04d6SMatthias Ringwald             mesh_segmented_pdu_flatten(&segmented_pdu->segments, 12, upper_transport_pdu_data_out);
428*16fd08f9SMatthias Ringwald             mesh_print_hex("Encrypted Payload:", upper_transport_pdu_data_out, upper_transport_pdu_len);
429274c04d6SMatthias Ringwald             btstack_crypto_ccm_decrypt_block(&ccm, upper_transport_pdu_len, upper_transport_pdu_data_out, upper_transport_pdu_data_out,
430274c04d6SMatthias Ringwald                                              &mesh_upper_transport_validate_access_message_ccm, NULL);
431274c04d6SMatthias Ringwald             break;
432274c04d6SMatthias Ringwald         case MESH_PDU_TYPE_UNSEGMENTED:
433274c04d6SMatthias Ringwald             unsegmented_pdu = (mesh_network_pdu_t *) incoming_access_encrypted;
434274c04d6SMatthias Ringwald             (void)memcpy(upper_transport_pdu_data_out, &unsegmented_pdu->data[10], incoming_access_decrypted->len);
435274c04d6SMatthias Ringwald             btstack_crypto_ccm_decrypt_block(&ccm, upper_transport_pdu_len, upper_transport_pdu_data_out, upper_transport_pdu_data_out,
436274c04d6SMatthias Ringwald                                              &mesh_upper_transport_validate_access_message_ccm, NULL);
437274c04d6SMatthias Ringwald             break;
438274c04d6SMatthias Ringwald         default:
439274c04d6SMatthias Ringwald             btstack_assert(false);
440274c04d6SMatthias Ringwald             break;
44177ba3d3fSMatthias Ringwald     }
44277ba3d3fSMatthias Ringwald 
443274c04d6SMatthias Ringwald }
444274c04d6SMatthias Ringwald 
445274c04d6SMatthias Ringwald static void mesh_upper_transport_validate_access_message(void){
446274c04d6SMatthias Ringwald     uint8_t * upper_transport_pdu_data =  incoming_access_decrypted->data;
447274c04d6SMatthias Ringwald     uint8_t   upper_transport_pdu_len  = incoming_access_decrypted->len - incoming_access_decrypted->transmic_len;
44877ba3d3fSMatthias Ringwald 
44977ba3d3fSMatthias Ringwald     if (!mesh_transport_key_and_virtual_address_iterator_has_more(&mesh_transport_key_it)){
45077ba3d3fSMatthias Ringwald         printf("No valid transport key found\n");
451274c04d6SMatthias Ringwald         mesh_upper_transport_process_access_message_done(incoming_access_decrypted);
45277ba3d3fSMatthias Ringwald         return;
45377ba3d3fSMatthias Ringwald     }
45477ba3d3fSMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_next(&mesh_transport_key_it);
45577ba3d3fSMatthias Ringwald     const mesh_transport_key_t * message_key = mesh_transport_key_it.key;
45677ba3d3fSMatthias Ringwald 
45777ba3d3fSMatthias Ringwald     if (message_key->akf){
458274c04d6SMatthias Ringwald         transport_segmented_setup_application_nonce(application_nonce, (mesh_pdu_t *) incoming_access_decrypted);
45977ba3d3fSMatthias Ringwald     } else {
460274c04d6SMatthias Ringwald         transport_segmented_setup_device_nonce(application_nonce, (mesh_pdu_t *) incoming_access_decrypted);
46177ba3d3fSMatthias Ringwald     }
46277ba3d3fSMatthias Ringwald 
46377ba3d3fSMatthias Ringwald     // store application / device key index
46477ba3d3fSMatthias Ringwald     mesh_print_hex("AppOrDevKey", message_key->key, 16);
465274c04d6SMatthias Ringwald     incoming_access_decrypted->appkey_index = message_key->appkey_index;
46677ba3d3fSMatthias Ringwald 
46777ba3d3fSMatthias Ringwald     mesh_print_hex("EncAccessPayload", upper_transport_pdu_data, upper_transport_pdu_len);
46877ba3d3fSMatthias Ringwald 
46977ba3d3fSMatthias Ringwald     // decrypt ccm
47077ba3d3fSMatthias Ringwald     crypto_active = 1;
47177ba3d3fSMatthias Ringwald     uint16_t aad_len  = 0;
4722ddfd9e2SMatthias Ringwald     if (mesh_network_address_virtual(incoming_access_decrypted->dst)){
47377ba3d3fSMatthias Ringwald         aad_len  = 16;
47477ba3d3fSMatthias Ringwald     }
475274c04d6SMatthias Ringwald     btstack_crypto_ccm_init(&ccm, message_key->key, application_nonce, upper_transport_pdu_len, aad_len, incoming_access_decrypted->transmic_len);
47677ba3d3fSMatthias Ringwald 
47777ba3d3fSMatthias Ringwald     if (aad_len){
478274c04d6SMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, (uint8_t *) mesh_transport_key_it.address->label_uuid, aad_len,
479274c04d6SMatthias Ringwald                                   &mesh_upper_transport_validate_access_message_digest, NULL);
48077ba3d3fSMatthias Ringwald     } else {
481274c04d6SMatthias Ringwald         mesh_upper_transport_validate_access_message_digest(NULL);
48277ba3d3fSMatthias Ringwald     }
48377ba3d3fSMatthias Ringwald }
48477ba3d3fSMatthias Ringwald 
485274c04d6SMatthias Ringwald static void mesh_upper_transport_process_access_message(void){
486274c04d6SMatthias Ringwald     uint8_t * upper_transport_pdu     =  incoming_access_decrypted->data;
487274c04d6SMatthias Ringwald     uint8_t   upper_transport_pdu_len = incoming_access_decrypted->len - incoming_access_decrypted->transmic_len;
48877ba3d3fSMatthias Ringwald     mesh_print_hex("Upper Transport pdu", upper_transport_pdu, upper_transport_pdu_len);
48977ba3d3fSMatthias Ringwald 
490274c04d6SMatthias Ringwald     uint8_t aid = incoming_access_decrypted->akf_aid_control & 0x3f;
491274c04d6SMatthias Ringwald     uint8_t akf = (incoming_access_decrypted->akf_aid_control & 0x40) >> 6;
49277ba3d3fSMatthias Ringwald 
49377ba3d3fSMatthias Ringwald     printf("AKF: %u\n",   akf);
49477ba3d3fSMatthias Ringwald     printf("AID: %02x\n", aid);
49577ba3d3fSMatthias Ringwald 
4962ddfd9e2SMatthias Ringwald     mesh_transport_key_and_virtual_address_iterator_init(&mesh_transport_key_it, incoming_access_decrypted->dst,
497274c04d6SMatthias Ringwald                                                          incoming_access_decrypted->netkey_index, akf, aid);
498274c04d6SMatthias Ringwald     mesh_upper_transport_validate_access_message();
49977ba3d3fSMatthias Ringwald }
50077ba3d3fSMatthias Ringwald 
5012983fbcbSMatthias Ringwald static void mesh_upper_transport_message_received(mesh_pdu_t * pdu){
50277ba3d3fSMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_incoming, (btstack_linked_item_t*) pdu);
5035beab91eSMatthias Ringwald     mesh_upper_transport_run();
50477ba3d3fSMatthias Ringwald }
50577ba3d3fSMatthias Ringwald 
506e9c16304SMatthias Ringwald static void mesh_upper_transport_send_access_segmented(mesh_upper_transport_pdu_t * upper_pdu){
507e9c16304SMatthias Ringwald 
508370463bfSMatthias Ringwald     mesh_segmented_pdu_t * segmented_pdu   = (mesh_segmented_pdu_t *) upper_pdu->lower_pdu;
509370463bfSMatthias Ringwald     segmented_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_SEGMENTED;
5108fa6125aSMatthias Ringwald 
511e9c16304SMatthias Ringwald     // convert mesh_access_pdu_t into mesh_segmented_pdu_t
5125368239bSMatthias Ringwald     btstack_linked_list_t free_segments = segmented_pdu->segments;
5135368239bSMatthias Ringwald     segmented_pdu->segments = NULL;
5148fb906aaSMatthias Ringwald     mesh_segmented_store_payload(incoming_pdu_singleton.access.data, upper_pdu->len, &free_segments, &segmented_pdu->segments);
515242605faSMatthias Ringwald 
5168fa6125aSMatthias Ringwald     // copy meta
517370463bfSMatthias Ringwald     segmented_pdu->len = upper_pdu->len;
518370463bfSMatthias Ringwald     segmented_pdu->netkey_index = upper_pdu->netkey_index;
519370463bfSMatthias Ringwald     segmented_pdu->transmic_len = upper_pdu->transmic_len;
520370463bfSMatthias Ringwald     segmented_pdu->akf_aid_control = upper_pdu->akf_aid_control;
521370463bfSMatthias Ringwald     segmented_pdu->flags = upper_pdu->flags;
522a97cf70aSMatthias Ringwald 
523370463bfSMatthias Ringwald     // setup segmented_pdu header
524370463bfSMatthias Ringwald     // (void)memcpy(segmented_pdu->network_header, upper_pdu->network_header, 9);
525a97cf70aSMatthias Ringwald     // TODO: use fields in mesh_segmented_pdu_t and setup network header in lower transport
5267cb45abfSMatthias Ringwald     segmented_pdu->ivi_nid = upper_pdu->ivi_nid;
5277cb45abfSMatthias Ringwald     segmented_pdu->ctl_ttl = upper_pdu->ctl_ttl;
5287cb45abfSMatthias Ringwald     segmented_pdu->seq = upper_pdu->seq;
5297cb45abfSMatthias Ringwald     segmented_pdu->src = upper_pdu->src;
5307cb45abfSMatthias Ringwald     segmented_pdu->dst = upper_pdu->dst;
5318fa6125aSMatthias Ringwald 
532ee7327abSMatthias Ringwald     // queue up
533370463bfSMatthias Ringwald     upper_pdu->lower_pdu = (mesh_pdu_t *) segmented_pdu;
534ee7327abSMatthias Ringwald     btstack_linked_list_add(&upper_transport_outgoing_active, (btstack_linked_item_t *) upper_pdu);
535ee7327abSMatthias Ringwald 
536370463bfSMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t*) segmented_pdu);
53712ba52abSMatthias Ringwald }
53812ba52abSMatthias Ringwald 
539e9c16304SMatthias Ringwald static void mesh_upper_transport_send_access_unsegmented(mesh_upper_transport_pdu_t * upper_pdu){
540e9c16304SMatthias Ringwald 
541e9c16304SMatthias Ringwald     // provide segment
54243d07e7fSMatthias Ringwald     mesh_network_pdu_t * network_pdu = (mesh_network_pdu_t *) upper_pdu->lower_pdu;
543e9c16304SMatthias Ringwald 
544e9c16304SMatthias Ringwald     // setup network pdu
545e9c16304SMatthias Ringwald     network_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS;
546e9c16304SMatthias Ringwald     network_pdu->data[0] = upper_pdu->ivi_nid;
547e9c16304SMatthias Ringwald     network_pdu->data[1] = upper_pdu->ctl_ttl;
548e9c16304SMatthias Ringwald     big_endian_store_24(network_pdu->data, 2, upper_pdu->seq);
549e9c16304SMatthias Ringwald     big_endian_store_16(network_pdu->data, 5, upper_pdu->src);
550e9c16304SMatthias Ringwald     big_endian_store_16(network_pdu->data, 7, upper_pdu->dst);
551e9c16304SMatthias Ringwald     network_pdu->netkey_index = upper_pdu->netkey_index;
552e9c16304SMatthias Ringwald 
553e9c16304SMatthias Ringwald     // setup access message
554e9c16304SMatthias Ringwald     network_pdu->data[9] = upper_pdu->akf_aid_control;
555e9c16304SMatthias Ringwald     btstack_assert(upper_pdu->len < 15);
5568fb906aaSMatthias Ringwald     (void)memcpy(&network_pdu->data[10], &incoming_pdu_singleton.access.data, upper_pdu->len);
557e9c16304SMatthias Ringwald     network_pdu->len = 10 + upper_pdu->len;
558e9c16304SMatthias Ringwald     network_pdu->flags = 0;
559e9c16304SMatthias Ringwald 
560ee7327abSMatthias Ringwald     // queue up
561ee7327abSMatthias Ringwald     btstack_linked_list_add(&upper_transport_outgoing_active, (btstack_linked_item_t *) upper_pdu);
562ee7327abSMatthias Ringwald 
563e9c16304SMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t*) network_pdu);
564e9c16304SMatthias Ringwald }
565e9c16304SMatthias Ringwald 
566e9c16304SMatthias Ringwald static void mesh_upper_transport_send_access_ccm(void * arg){
567e9c16304SMatthias Ringwald     crypto_active = 0;
568e9c16304SMatthias Ringwald 
569e9c16304SMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu = (mesh_upper_transport_pdu_t *) arg;
5708fb906aaSMatthias Ringwald     mesh_print_hex("EncAccessPayload", incoming_pdu_singleton.access.data, upper_pdu->len);
571e9c16304SMatthias Ringwald     // store TransMIC
5728fb906aaSMatthias Ringwald     btstack_crypto_ccm_get_authentication_value(&ccm, &incoming_pdu_singleton.access.data[upper_pdu->len]);
5738fb906aaSMatthias Ringwald     mesh_print_hex("TransMIC", &incoming_pdu_singleton.access.data[upper_pdu->len], upper_pdu->transmic_len);
574e9c16304SMatthias Ringwald     upper_pdu->len += upper_pdu->transmic_len;
5758fb906aaSMatthias Ringwald     mesh_print_hex("UpperTransportPDU", incoming_pdu_singleton.access.data, upper_pdu->len);
576e9c16304SMatthias Ringwald     switch (upper_pdu->pdu_header.pdu_type){
577e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
578e9c16304SMatthias Ringwald             mesh_upper_transport_send_access_unsegmented(upper_pdu);
579e9c16304SMatthias Ringwald             break;
580e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
581e9c16304SMatthias Ringwald             mesh_upper_transport_send_access_segmented(upper_pdu);
582e9c16304SMatthias Ringwald             break;
583e9c16304SMatthias Ringwald         default:
584e9c16304SMatthias Ringwald             btstack_assert(false);
585e9c16304SMatthias Ringwald     }
586e9c16304SMatthias Ringwald }
587e9c16304SMatthias Ringwald 
588e9c16304SMatthias Ringwald static void mesh_upper_transport_send_access_digest(void *arg){
589e9c16304SMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu = (mesh_upper_transport_pdu_t *) arg;
590e9c16304SMatthias Ringwald     uint16_t  access_pdu_len  = upper_pdu->len;
5918fb906aaSMatthias Ringwald     btstack_crypto_ccm_encrypt_block(&ccm, access_pdu_len, incoming_pdu_singleton.access.data, incoming_pdu_singleton.access.data,
592e9c16304SMatthias Ringwald                                      &mesh_upper_transport_send_access_ccm, upper_pdu);
593e9c16304SMatthias Ringwald }
594e9c16304SMatthias Ringwald 
595e9c16304SMatthias Ringwald static void mesh_upper_transport_send_access(mesh_upper_transport_pdu_t * upper_pdu){
596e9c16304SMatthias Ringwald 
597e9c16304SMatthias Ringwald     // if dst is virtual address, lookup label uuid and hash
598e9c16304SMatthias Ringwald     uint16_t aad_len = 0;
599e9c16304SMatthias Ringwald     mesh_virtual_address_t * virtual_address = NULL;
600e9c16304SMatthias Ringwald     if (mesh_network_address_virtual(upper_pdu->dst)){
601e9c16304SMatthias Ringwald         virtual_address = mesh_virtual_address_for_pseudo_dst(upper_pdu->dst);
602e9c16304SMatthias Ringwald         if (!virtual_address){
603e9c16304SMatthias Ringwald             printf("No virtual address register for pseudo dst %4x\n", upper_pdu->dst);
604e9c16304SMatthias Ringwald             mesh_access_message_handler(MESH_TRANSPORT_PDU_SENT, MESH_TRANSPORT_STATUS_SEND_FAILED, (mesh_pdu_t *) upper_pdu);
605e9c16304SMatthias Ringwald             return;
606e9c16304SMatthias Ringwald         }
607e9c16304SMatthias Ringwald         // printf("Using hash %4x with LabelUUID: ", virtual_address->hash);
608e9c16304SMatthias Ringwald         // printf_hexdump(virtual_address->label_uuid, 16);
609e9c16304SMatthias Ringwald         aad_len = 16;
610e9c16304SMatthias Ringwald         upper_pdu->dst = virtual_address->hash;
611e9c16304SMatthias Ringwald     }
612e9c16304SMatthias Ringwald 
613e9c16304SMatthias Ringwald     // get app or device key
614e9c16304SMatthias Ringwald     uint16_t appkey_index = upper_pdu->appkey_index;
615e9c16304SMatthias Ringwald     const mesh_transport_key_t * appkey = mesh_upper_transport_get_outgoing_appkey(upper_pdu->netkey_index, appkey_index);
616e9c16304SMatthias Ringwald     if (appkey == NULL){
617e9c16304SMatthias Ringwald         printf("AppKey %04x not found, drop message\n", appkey_index);
618e9c16304SMatthias Ringwald         mesh_access_message_handler(MESH_TRANSPORT_PDU_SENT, MESH_TRANSPORT_STATUS_SEND_FAILED, (mesh_pdu_t *) upper_pdu);
619e9c16304SMatthias Ringwald         return;
620e9c16304SMatthias Ringwald     }
621e9c16304SMatthias Ringwald 
622e9c16304SMatthias Ringwald     // reserve slot
623e9c16304SMatthias Ringwald     mesh_lower_transport_reserve_slot();
624e9c16304SMatthias Ringwald 
625e9c16304SMatthias Ringwald     // reserve one sequence number, which is also used to encrypt access payload
626e9c16304SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
627e9c16304SMatthias Ringwald     upper_pdu->flags |= MESH_TRANSPORT_FLAG_SEQ_RESERVED;
628e9c16304SMatthias Ringwald     upper_pdu->seq = seq;
629e9c16304SMatthias Ringwald 
630e9c16304SMatthias Ringwald     // also reserves crypto_buffer
631e9c16304SMatthias Ringwald     crypto_active = 1;
632e9c16304SMatthias Ringwald 
633e9c16304SMatthias Ringwald     // flatten segmented pdu into crypto buffer
6348fb906aaSMatthias Ringwald     uint16_t payload_len = mesh_upper_pdu_flatten(upper_pdu, incoming_pdu_singleton.access.data, sizeof(incoming_pdu_singleton.access.data));
635e9c16304SMatthias Ringwald     btstack_assert(payload_len == upper_pdu->len);
636e9c16304SMatthias Ringwald 
637e9c16304SMatthias Ringwald     // Dump PDU
638e9c16304SMatthias Ringwald     printf("[+] Upper transport, send upper (un)segmented Access PDU - dest %04x, seq %06x\n", upper_pdu->dst, upper_pdu->seq);
6398fb906aaSMatthias Ringwald     mesh_print_hex("Access Payload", incoming_pdu_singleton.access.data, upper_pdu->len);
640e9c16304SMatthias Ringwald 
641e9c16304SMatthias Ringwald     // setup nonce - uses dst, so after pseudo address translation
642e9c16304SMatthias Ringwald     if (appkey_index == MESH_DEVICE_KEY_INDEX){
643e9c16304SMatthias Ringwald         transport_segmented_setup_device_nonce(application_nonce, (mesh_pdu_t *) upper_pdu);
644e9c16304SMatthias Ringwald     } else {
645e9c16304SMatthias Ringwald         transport_segmented_setup_application_nonce(application_nonce, (mesh_pdu_t *) upper_pdu);
646e9c16304SMatthias Ringwald     }
647e9c16304SMatthias Ringwald 
648e9c16304SMatthias Ringwald     // Dump key
649e9c16304SMatthias Ringwald     mesh_print_hex("AppOrDevKey", appkey->key, 16);
650e9c16304SMatthias Ringwald 
651e9c16304SMatthias Ringwald     // encrypt ccm
652e9c16304SMatthias Ringwald     uint8_t   transmic_len    = upper_pdu->transmic_len;
653e9c16304SMatthias Ringwald     uint16_t  access_pdu_len  = upper_pdu->len;
654e9c16304SMatthias Ringwald     btstack_crypto_ccm_init(&ccm, appkey->key, application_nonce, access_pdu_len, aad_len, transmic_len);
655e9c16304SMatthias Ringwald     if (virtual_address){
656e9c16304SMatthias Ringwald         mesh_print_hex("LabelUUID", virtual_address->label_uuid, 16);
657e9c16304SMatthias Ringwald         btstack_crypto_ccm_digest(&ccm, virtual_address->label_uuid, 16,
658e9c16304SMatthias Ringwald                                   &mesh_upper_transport_send_access_digest, upper_pdu);
659e9c16304SMatthias Ringwald     } else {
660e9c16304SMatthias Ringwald         mesh_upper_transport_send_access_digest(upper_pdu);
661e9c16304SMatthias Ringwald     }
662e9c16304SMatthias Ringwald }
663e9c16304SMatthias Ringwald 
664e9c16304SMatthias Ringwald static void mesh_upper_transport_send_unsegmented_control_pdu(mesh_network_pdu_t * network_pdu){
665e9c16304SMatthias Ringwald     // reserve slot
666e9c16304SMatthias Ringwald     mesh_lower_transport_reserve_slot();
667e9c16304SMatthias Ringwald     // reserve sequence number
668e9c16304SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
669e9c16304SMatthias Ringwald     mesh_network_pdu_set_seq(network_pdu, seq);
670e9c16304SMatthias Ringwald     // Dump PDU
671e9c16304SMatthias Ringwald     uint8_t opcode = network_pdu->data[9];
672e9c16304SMatthias Ringwald     printf("[+] Upper transport, send unsegmented Control PDU %p - seq %06x opcode %02x\n", network_pdu, seq, opcode);
673e9c16304SMatthias Ringwald     mesh_print_hex("Access Payload", &network_pdu->data[10], network_pdu->len - 10);
674e9c16304SMatthias Ringwald 
675e9c16304SMatthias Ringwald     // send
676e9c16304SMatthias Ringwald      mesh_lower_transport_send_pdu((mesh_pdu_t *) network_pdu);
677e9c16304SMatthias Ringwald }
678e9c16304SMatthias Ringwald 
679e9c16304SMatthias Ringwald static void mesh_upper_transport_send_segmented_control_pdu(mesh_upper_transport_pdu_t * upper_pdu){
680e9c16304SMatthias Ringwald     // reserve slot
681e9c16304SMatthias Ringwald     mesh_lower_transport_reserve_slot();
682e9c16304SMatthias Ringwald     // reserve sequence number
683e9c16304SMatthias Ringwald     uint32_t seq = mesh_sequence_number_next();
684e9c16304SMatthias Ringwald     upper_pdu->flags |= MESH_TRANSPORT_FLAG_SEQ_RESERVED;
685e9c16304SMatthias Ringwald     upper_pdu->seq = seq;
686e9c16304SMatthias Ringwald     // Dump PDU
687e9c16304SMatthias Ringwald     // uint8_t opcode = upper_pdu->data[0];
688e9c16304SMatthias Ringwald     // printf("[+] Upper transport, send segmented Control PDU %p - seq %06x opcode %02x\n", upper_pdu, seq, opcode);
689e9c16304SMatthias Ringwald     // mesh_print_hex("Access Payload", &upper_pdu->data[1], upper_pdu->len - 1);
690e9c16304SMatthias Ringwald     // send
691370463bfSMatthias Ringwald     mesh_segmented_pdu_t * segmented_pdu   = (mesh_segmented_pdu_t *) upper_pdu->lower_pdu;
692370463bfSMatthias Ringwald     segmented_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_SEGMENTED;
693e9c16304SMatthias Ringwald 
694e9c16304SMatthias Ringwald     // lend segments to lower transport pdu
695370463bfSMatthias Ringwald     segmented_pdu->segments = upper_pdu->segments;
696e9c16304SMatthias Ringwald     upper_pdu->segments = NULL;
697e9c16304SMatthias Ringwald 
698e9c16304SMatthias Ringwald     // copy meta
699370463bfSMatthias Ringwald     segmented_pdu->len = upper_pdu->len;
700370463bfSMatthias Ringwald     segmented_pdu->netkey_index = upper_pdu->netkey_index;
701370463bfSMatthias Ringwald     segmented_pdu->transmic_len = 0;   // no TransMIC for control
702370463bfSMatthias Ringwald     segmented_pdu->akf_aid_control = upper_pdu->akf_aid_control;
703370463bfSMatthias Ringwald     segmented_pdu->flags = upper_pdu->flags;
704e9c16304SMatthias Ringwald 
705370463bfSMatthias Ringwald     // setup segmented_pdu header
706e9c16304SMatthias Ringwald     // TODO: use fields in mesh_segmented_pdu_t and setup network header in lower transport
7077cb45abfSMatthias Ringwald     segmented_pdu->ivi_nid = upper_pdu->ivi_nid;
7087cb45abfSMatthias Ringwald     segmented_pdu->ctl_ttl = upper_pdu->ctl_ttl;
7097cb45abfSMatthias Ringwald     segmented_pdu->seq = upper_pdu->seq;
7107cb45abfSMatthias Ringwald     segmented_pdu->src = upper_pdu->src;
7117cb45abfSMatthias Ringwald     segmented_pdu->dst = upper_pdu->dst;
712e9c16304SMatthias Ringwald 
713ee7327abSMatthias Ringwald     // queue up
714370463bfSMatthias Ringwald     upper_pdu->lower_pdu = (mesh_pdu_t *) segmented_pdu;
715ee7327abSMatthias Ringwald     btstack_linked_list_add(&upper_transport_outgoing_active, (btstack_linked_item_t *) upper_pdu);
716ee7327abSMatthias Ringwald 
717370463bfSMatthias Ringwald     mesh_lower_transport_send_pdu((mesh_pdu_t *) segmented_pdu);
718e9c16304SMatthias Ringwald }
719e9c16304SMatthias Ringwald 
720e9c16304SMatthias Ringwald static void mesh_upper_transport_run(void){
721e9c16304SMatthias Ringwald 
722e9c16304SMatthias Ringwald     while(!btstack_linked_list_empty(&upper_transport_incoming)){
723e9c16304SMatthias Ringwald 
724e9c16304SMatthias Ringwald         if (crypto_active) return;
725e9c16304SMatthias Ringwald 
726e9c16304SMatthias Ringwald         // get next message
727e9c16304SMatthias Ringwald         mesh_pdu_t * pdu =  (mesh_pdu_t *) btstack_linked_list_pop(&upper_transport_incoming);
728e9c16304SMatthias Ringwald         mesh_network_pdu_t   * network_pdu;
729e9c16304SMatthias Ringwald         mesh_segmented_pdu_t   * message_pdu;
730e9c16304SMatthias Ringwald         switch (pdu->pdu_type){
731e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_UNSEGMENTED:
732e4121a34SMatthias Ringwald                 network_pdu = (mesh_network_pdu_t *) pdu;
733e9c16304SMatthias Ringwald                 // control?
734e9c16304SMatthias Ringwald                 if (mesh_network_control(network_pdu)) {
735e9c16304SMatthias Ringwald 
7368fb906aaSMatthias Ringwald                     incoming_control_pdu =  &incoming_pdu_singleton.control;
737e9c16304SMatthias Ringwald                     incoming_control_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_CONTROL;
738e9c16304SMatthias Ringwald                     incoming_control_pdu->len =  network_pdu->len;
739e9c16304SMatthias Ringwald                     incoming_control_pdu->netkey_index =  network_pdu->netkey_index;
740e9c16304SMatthias Ringwald 
741e9c16304SMatthias Ringwald                     uint8_t * lower_transport_pdu = mesh_network_pdu_data(network_pdu);
742e9c16304SMatthias Ringwald 
743e9c16304SMatthias Ringwald                     incoming_control_pdu->akf_aid_control = lower_transport_pdu[0];
744e9c16304SMatthias Ringwald                     incoming_control_pdu->len = network_pdu->len - 10; // 9 header + 1 opcode
745e9c16304SMatthias Ringwald                     (void)memcpy(incoming_control_pdu->data, &lower_transport_pdu[1], incoming_control_pdu->len);
746e9c16304SMatthias Ringwald 
747e9c16304SMatthias Ringwald                     // copy meta data into encrypted pdu buffer
748a2495898SMatthias Ringwald                     incoming_control_pdu->ivi_nid = network_pdu->data[0];
749a2495898SMatthias Ringwald                     incoming_control_pdu->ctl_ttl = network_pdu->data[1];
750a2495898SMatthias Ringwald                     incoming_control_pdu->seq = big_endian_read_24(network_pdu->data, 2);
751a2495898SMatthias Ringwald                     incoming_control_pdu->src = big_endian_read_16(network_pdu->data, 5);
752a2495898SMatthias Ringwald                     incoming_control_pdu->dst = big_endian_read_16(network_pdu->data, 7);
753e9c16304SMatthias Ringwald 
754e9c16304SMatthias Ringwald                     mesh_print_hex("Assembled payload", incoming_control_pdu->data, incoming_control_pdu->len);
755e9c16304SMatthias Ringwald 
756e9c16304SMatthias Ringwald                     // free mesh message
757e9c16304SMatthias Ringwald                     mesh_lower_transport_message_processed_by_higher_layer(pdu);
758e9c16304SMatthias Ringwald 
759e9c16304SMatthias Ringwald                     btstack_assert(mesh_control_message_handler != NULL);
760e9c16304SMatthias Ringwald                     mesh_pdu_t * pdu = (mesh_pdu_t*) incoming_control_pdu;
761e9c16304SMatthias Ringwald                     mesh_control_message_handler(MESH_TRANSPORT_PDU_RECEIVED, MESH_TRANSPORT_STATUS_SUCCESS, pdu);
762e9c16304SMatthias Ringwald 
763e9c16304SMatthias Ringwald                 } else {
764e9c16304SMatthias Ringwald 
765da491525SMatthias Ringwald                     incoming_access_encrypted = (mesh_pdu_t *) network_pdu;
766da491525SMatthias Ringwald 
767274c04d6SMatthias Ringwald                     incoming_access_decrypted = &incoming_pdu_singleton.access;
768274c04d6SMatthias Ringwald                     incoming_access_decrypted->pdu_header.pdu_type = MESH_PDU_TYPE_ACCESS;
769274c04d6SMatthias Ringwald                     incoming_access_decrypted->netkey_index = network_pdu->netkey_index;
770274c04d6SMatthias Ringwald                     incoming_access_decrypted->transmic_len = 4;
771274c04d6SMatthias Ringwald                     incoming_access_decrypted->akf_aid_control = network_pdu->data[9];
772274c04d6SMatthias Ringwald                     incoming_access_decrypted->len = network_pdu->len - 10; // 9 header + 1 AID
773c3fc3eeaSMatthias Ringwald                     incoming_access_decrypted->ivi_nid = network_pdu->data[0];
774c3fc3eeaSMatthias Ringwald                     incoming_access_decrypted->ctl_ttl = network_pdu->data[1];
775c3fc3eeaSMatthias Ringwald                     incoming_access_decrypted->seq = big_endian_read_24(network_pdu->data, 2);
776c3fc3eeaSMatthias Ringwald                     incoming_access_decrypted->src = big_endian_read_16(network_pdu->data, 5);
777c3fc3eeaSMatthias Ringwald                     incoming_access_decrypted->dst = big_endian_read_16(network_pdu->data, 7);
778c3fc3eeaSMatthias Ringwald                     //  (void)memcpy(incoming_access_decrypted->network_header, network_pdu->data, 9);
779e9c16304SMatthias Ringwald 
780274c04d6SMatthias Ringwald                     mesh_upper_transport_process_access_message();
781e9c16304SMatthias Ringwald                 }
782e9c16304SMatthias Ringwald                 break;
783e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_SEGMENTED:
784e9c16304SMatthias Ringwald                 message_pdu = (mesh_segmented_pdu_t *) pdu;
7853945bdb6SMatthias Ringwald                 uint8_t ctl = message_pdu->ctl_ttl >> 7;
786e9c16304SMatthias Ringwald                 if (ctl){
7878fb906aaSMatthias Ringwald                     incoming_control_pdu=  &incoming_pdu_singleton.control;
788e9c16304SMatthias Ringwald                     incoming_control_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_CONTROL;
789e9c16304SMatthias Ringwald 
790e9c16304SMatthias Ringwald                     // flatten
791e9c16304SMatthias Ringwald                     mesh_segmented_pdu_flatten(&message_pdu->segments, 8, incoming_control_pdu->data);
792e9c16304SMatthias Ringwald 
793e9c16304SMatthias Ringwald                     // copy meta data into encrypted pdu buffer
794e9c16304SMatthias Ringwald                     incoming_control_pdu->flags = 0;
795a2495898SMatthias Ringwald                     incoming_control_pdu->len =  message_pdu->len;
796a2495898SMatthias Ringwald                     incoming_control_pdu->netkey_index =  message_pdu->netkey_index;
797a2495898SMatthias Ringwald                     incoming_control_pdu->akf_aid_control = message_pdu->akf_aid_control;
7987cb45abfSMatthias Ringwald                     incoming_access_decrypted->ivi_nid = message_pdu->ivi_nid;
7997cb45abfSMatthias Ringwald                     incoming_access_decrypted->ctl_ttl = message_pdu->ctl_ttl;
8007cb45abfSMatthias Ringwald                     incoming_access_decrypted->seq = message_pdu->seq;
8017cb45abfSMatthias Ringwald                     incoming_access_decrypted->src = message_pdu->src;
8027cb45abfSMatthias Ringwald                     incoming_access_decrypted->dst = message_pdu->dst;
803e9c16304SMatthias Ringwald 
804e9c16304SMatthias Ringwald                     mesh_print_hex("Assembled payload", incoming_control_pdu->data, incoming_control_pdu->len);
805e9c16304SMatthias Ringwald 
806e9c16304SMatthias Ringwald                     // free mesh message
807e9c16304SMatthias Ringwald                     mesh_lower_transport_message_processed_by_higher_layer((mesh_pdu_t *)message_pdu);
808e9c16304SMatthias Ringwald 
809e9c16304SMatthias Ringwald                     btstack_assert(mesh_control_message_handler != NULL);
810e9c16304SMatthias Ringwald                     mesh_pdu_t * pdu = (mesh_pdu_t*) incoming_control_pdu;
811e9c16304SMatthias Ringwald                     mesh_access_message_handler(MESH_TRANSPORT_PDU_RECEIVED, MESH_TRANSPORT_STATUS_SUCCESS, pdu);
812e9c16304SMatthias Ringwald 
813e9c16304SMatthias Ringwald                 } else {
814e9c16304SMatthias Ringwald 
815da491525SMatthias Ringwald                     incoming_access_encrypted = (mesh_pdu_t *) message_pdu;
816da491525SMatthias Ringwald 
817274c04d6SMatthias Ringwald                     incoming_access_decrypted = &incoming_pdu_singleton.access;
818274c04d6SMatthias Ringwald                     incoming_access_decrypted->pdu_header.pdu_type = MESH_PDU_TYPE_ACCESS;
819274c04d6SMatthias Ringwald                     incoming_access_decrypted->len =  message_pdu->len;
820274c04d6SMatthias Ringwald                     incoming_access_decrypted->netkey_index =  message_pdu->netkey_index;
821274c04d6SMatthias Ringwald                     incoming_access_decrypted->transmic_len =  message_pdu->transmic_len;
822274c04d6SMatthias Ringwald                     incoming_access_decrypted->akf_aid_control =  message_pdu->akf_aid_control;
8237cb45abfSMatthias Ringwald                     incoming_access_decrypted->ivi_nid = message_pdu->ivi_nid;
8247cb45abfSMatthias Ringwald                     incoming_access_decrypted->ctl_ttl = message_pdu->ctl_ttl;
8257cb45abfSMatthias Ringwald                     incoming_access_decrypted->seq = message_pdu->seq;
8267cb45abfSMatthias Ringwald                     incoming_access_decrypted->src = message_pdu->src;
8277cb45abfSMatthias Ringwald                     incoming_access_decrypted->dst = message_pdu->dst;
828e9c16304SMatthias Ringwald 
829274c04d6SMatthias Ringwald                     mesh_upper_transport_process_access_message();
830e9c16304SMatthias Ringwald                 }
831e9c16304SMatthias Ringwald                 break;
832e9c16304SMatthias Ringwald             default:
833e9c16304SMatthias Ringwald                 btstack_assert(0);
834e9c16304SMatthias Ringwald                 break;
835e9c16304SMatthias Ringwald         }
836e9c16304SMatthias Ringwald     }
837e9c16304SMatthias Ringwald 
838e9c16304SMatthias Ringwald     while (!btstack_linked_list_empty(&upper_transport_outgoing)){
839e9c16304SMatthias Ringwald 
840e9c16304SMatthias Ringwald         if (crypto_active) break;
841e9c16304SMatthias Ringwald 
842e9c16304SMatthias Ringwald         mesh_pdu_t * pdu =  (mesh_pdu_t *) btstack_linked_list_get_first_item(&upper_transport_outgoing);
843e9c16304SMatthias Ringwald         if (mesh_lower_transport_can_send_to_dest(mesh_pdu_dst(pdu)) == 0) break;
844e9c16304SMatthias Ringwald 
84543d07e7fSMatthias Ringwald         mesh_upper_transport_pdu_t * upper_pdu;
8465368239bSMatthias Ringwald         mesh_segmented_pdu_t * segmented_pdu;
8475368239bSMatthias Ringwald         bool ok;
848e9c16304SMatthias Ringwald 
849e9c16304SMatthias Ringwald         switch (pdu->pdu_type){
850e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_UPPER_UNSEGMENTED_CONTROL:
85143d07e7fSMatthias Ringwald                 // control pdus can go through directly
852e9c16304SMatthias Ringwald                 btstack_assert(mesh_pdu_ctl(pdu) != 0);
85343d07e7fSMatthias Ringwald                 (void) btstack_linked_list_pop(&upper_transport_outgoing);
854e9c16304SMatthias Ringwald                 mesh_upper_transport_send_unsegmented_control_pdu((mesh_network_pdu_t *) pdu);
855e9c16304SMatthias Ringwald                 break;
856e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
85743d07e7fSMatthias Ringwald                 // control pdus can go through directly
85843d07e7fSMatthias Ringwald                 btstack_assert(mesh_pdu_ctl(pdu) != 0);
85943d07e7fSMatthias Ringwald                 (void) btstack_linked_list_pop(&upper_transport_outgoing);
860e9c16304SMatthias Ringwald                 mesh_upper_transport_send_segmented_control_pdu((mesh_upper_transport_pdu_t *) pdu);
861e9c16304SMatthias Ringwald                 break;
862e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
86343d07e7fSMatthias Ringwald                 // segmented access pdus required a mesh-segmented-pdu
864370463bfSMatthias Ringwald                 upper_pdu = (mesh_upper_transport_pdu_t *) pdu;
865370463bfSMatthias Ringwald                 if (upper_pdu->lower_pdu == NULL){
8665368239bSMatthias Ringwald                     segmented_pdu = btstack_memory_mesh_segmented_pdu_get();
86743d07e7fSMatthias Ringwald                 }
8685368239bSMatthias Ringwald                 if (segmented_pdu == NULL) break;
8695368239bSMatthias Ringwald                 upper_pdu->lower_pdu = (mesh_pdu_t *) segmented_pdu;
8705368239bSMatthias Ringwald                 segmented_pdu->pdu_header.pdu_type = MESH_PDU_TYPE_SEGMENTED;
8715368239bSMatthias Ringwald                 // and a mesh-network-pdu for each segment in upper pdu
8725368239bSMatthias Ringwald                 ok = mesh_segmented_allocate_segments(&segmented_pdu->segments, upper_pdu->len + upper_pdu->transmic_len);
8735368239bSMatthias Ringwald                 if (!ok) break;
8745368239bSMatthias Ringwald                 // all buffers available, get started
87543d07e7fSMatthias Ringwald                 (void) btstack_linked_list_pop(&upper_transport_outgoing);
876370463bfSMatthias Ringwald                 mesh_upper_transport_send_access(upper_pdu);
87743d07e7fSMatthias Ringwald                 break;
878e9c16304SMatthias Ringwald             case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
87943d07e7fSMatthias Ringwald                 // unsegmented access pdus require a single mesh-network-dpu
88043d07e7fSMatthias Ringwald                 upper_pdu = (mesh_upper_transport_pdu_t *) pdu;
88143d07e7fSMatthias Ringwald                 if (upper_pdu->lower_pdu == NULL){
88243d07e7fSMatthias Ringwald                     upper_pdu->lower_pdu = (mesh_pdu_t *) mesh_network_pdu_get();
88343d07e7fSMatthias Ringwald                 }
88443d07e7fSMatthias Ringwald                 if (upper_pdu->lower_pdu == NULL) break;
88543d07e7fSMatthias Ringwald                 (void) btstack_linked_list_pop(&upper_transport_outgoing);
886e9c16304SMatthias Ringwald                 mesh_upper_transport_send_access((mesh_upper_transport_pdu_t *) pdu);
887e9c16304SMatthias Ringwald                 break;
888e9c16304SMatthias Ringwald             default:
889e9c16304SMatthias Ringwald                 btstack_assert(false);
890e9c16304SMatthias Ringwald                 break;
891e9c16304SMatthias Ringwald         }
892e9c16304SMatthias Ringwald     }
893e9c16304SMatthias Ringwald }
894e9c16304SMatthias Ringwald 
895ee7327abSMatthias Ringwald static mesh_upper_transport_pdu_t * mesh_upper_transport_find_pdu_for_lower(mesh_pdu_t * pdu_to_find){
896ee7327abSMatthias Ringwald     btstack_linked_list_iterator_t it;
897ee7327abSMatthias Ringwald     btstack_linked_list_iterator_init(&it, &upper_transport_outgoing_active);
898e9c16304SMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu;
899ee7327abSMatthias Ringwald     while (btstack_linked_list_iterator_has_next(&it)){
900ee7327abSMatthias Ringwald         mesh_pdu_t * mesh_pdu = (mesh_pdu_t *) btstack_linked_list_iterator_next(&it);
901ee7327abSMatthias Ringwald         switch (mesh_pdu->pdu_type){
902ee7327abSMatthias Ringwald             case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
903ee7327abSMatthias Ringwald             case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
904ee7327abSMatthias Ringwald             case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
905ee7327abSMatthias Ringwald                 upper_pdu = (mesh_upper_transport_pdu_t *) mesh_pdu;
906ee7327abSMatthias Ringwald                 if (upper_pdu->lower_pdu == pdu_to_find){
907ee7327abSMatthias Ringwald                     btstack_linked_list_iterator_remove(&it);
908ee7327abSMatthias Ringwald                     return upper_pdu;
909ee7327abSMatthias Ringwald                 }
910ee7327abSMatthias Ringwald                 break;
911ee7327abSMatthias Ringwald             default:
912ee7327abSMatthias Ringwald                 break;
913ee7327abSMatthias Ringwald         }
914ee7327abSMatthias Ringwald     }
915ee7327abSMatthias Ringwald     return NULL;
916ee7327abSMatthias Ringwald }
917ee7327abSMatthias Ringwald 
918ee7327abSMatthias Ringwald static void mesh_upper_transport_pdu_handler(mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu){
919ee7327abSMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu;
920ee7327abSMatthias Ringwald     mesh_network_pdu_t * network_pdu;
921ee7327abSMatthias Ringwald     mesh_segmented_pdu_t * segmented_pdu;
922e9c16304SMatthias Ringwald     switch (callback_type){
923e9c16304SMatthias Ringwald         case MESH_TRANSPORT_PDU_RECEIVED:
924e9c16304SMatthias Ringwald             mesh_upper_transport_message_received(pdu);
925e9c16304SMatthias Ringwald             break;
926e9c16304SMatthias Ringwald         case MESH_TRANSPORT_PDU_SENT:
927e9c16304SMatthias Ringwald             switch (pdu->pdu_type){
928e9c16304SMatthias Ringwald                 case MESH_PDU_TYPE_SEGMENTED:
929ee7327abSMatthias Ringwald                     // try to find in outgoing active
930ee7327abSMatthias Ringwald                     upper_pdu = mesh_upper_transport_find_pdu_for_lower(pdu);
931ee7327abSMatthias Ringwald                     btstack_assert(upper_pdu != NULL);
932ee7327abSMatthias Ringwald                     segmented_pdu = (mesh_segmented_pdu_t *) pdu;
933e9c16304SMatthias Ringwald                     // free chunks
934ee7327abSMatthias Ringwald                     while (!btstack_linked_list_empty(&segmented_pdu->segments)){
935ee7327abSMatthias Ringwald                         mesh_network_pdu_t * network_pdu = (mesh_network_pdu_t *) btstack_linked_list_pop(&segmented_pdu->segments);
936e9c16304SMatthias Ringwald                         mesh_network_pdu_free(network_pdu);
937e9c16304SMatthias Ringwald                     }
938370463bfSMatthias Ringwald                     // free segmented pdu
939370463bfSMatthias Ringwald                     btstack_memory_mesh_segmented_pdu_free(segmented_pdu);
940ee7327abSMatthias Ringwald                     // TODO: free segmented_pdu
94143d07e7fSMatthias Ringwald                     upper_pdu->lower_pdu = NULL;
942ee7327abSMatthias Ringwald                     switch (upper_pdu->pdu_header.pdu_type){
943ee7327abSMatthias Ringwald                         case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
944ee7327abSMatthias Ringwald                             mesh_control_message_handler(callback_type, status, (mesh_pdu_t *) upper_pdu);
945ee7327abSMatthias Ringwald                             break;
946ee7327abSMatthias Ringwald                         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
947ee7327abSMatthias Ringwald                             mesh_access_message_handler(callback_type, status, (mesh_pdu_t *) upper_pdu);
948ee7327abSMatthias Ringwald                             break;
949ee7327abSMatthias Ringwald                         default:
950ee7327abSMatthias Ringwald                             btstack_assert(false);
951ee7327abSMatthias Ringwald                             break;
952e9c16304SMatthias Ringwald                     }
953e9c16304SMatthias Ringwald                     break;
954e9c16304SMatthias Ringwald                 case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
95543d07e7fSMatthias Ringwald                     // find corresponding upper transport pdu and free single segment
956ee7327abSMatthias Ringwald                     upper_pdu = mesh_upper_transport_find_pdu_for_lower(pdu);
957ee7327abSMatthias Ringwald                     btstack_assert(upper_pdu != NULL);
95843d07e7fSMatthias Ringwald                     btstack_assert(upper_pdu->lower_pdu == (mesh_pdu_t *) pdu);
95943d07e7fSMatthias Ringwald                     mesh_network_pdu_free((mesh_network_pdu_t *) pdu);
96043d07e7fSMatthias Ringwald                     upper_pdu->lower_pdu = NULL;
961e9c16304SMatthias Ringwald                     mesh_access_message_handler(callback_type, status, (mesh_pdu_t*) upper_pdu);
962e9c16304SMatthias Ringwald                     break;
963e9c16304SMatthias Ringwald                 case MESH_PDU_TYPE_UPPER_UNSEGMENTED_CONTROL:
964e9c16304SMatthias Ringwald                     mesh_access_message_handler(callback_type, status, pdu);
965e9c16304SMatthias Ringwald                     break;
966e9c16304SMatthias Ringwald                 default:
967e9c16304SMatthias Ringwald                     btstack_assert(false);
968e9c16304SMatthias Ringwald                     break;
969e9c16304SMatthias Ringwald             }
970e9c16304SMatthias Ringwald             mesh_upper_transport_run();
971e9c16304SMatthias Ringwald             break;
972e9c16304SMatthias Ringwald         default:
973e9c16304SMatthias Ringwald             break;
974e9c16304SMatthias Ringwald     }
975e9c16304SMatthias Ringwald }
976e9c16304SMatthias Ringwald 
977e9c16304SMatthias Ringwald void mesh_upper_transport_pdu_free(mesh_pdu_t * pdu){
9789efa7acaSMatthias Ringwald     btstack_assert(pdu != NULL);
979e9c16304SMatthias Ringwald     mesh_network_pdu_t   * network_pdu;
980e9c16304SMatthias Ringwald     mesh_segmented_pdu_t   * message_pdu;
9819efa7acaSMatthias Ringwald     mesh_upper_transport_pdu_t * upper_pdu;
982e9c16304SMatthias Ringwald     switch (pdu->pdu_type) {
9839efa7acaSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_CONTROL:
984e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_NETWORK:
985e9c16304SMatthias Ringwald             network_pdu = (mesh_network_pdu_t *) pdu;
986e9c16304SMatthias Ringwald             mesh_network_pdu_free(network_pdu);
987e9c16304SMatthias Ringwald             break;
988e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_SEGMENTED:
989e9c16304SMatthias Ringwald             message_pdu = (mesh_segmented_pdu_t *) pdu;
990e9c16304SMatthias Ringwald             mesh_message_pdu_free(message_pdu);
9919efa7acaSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
9929efa7acaSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
9939efa7acaSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
9949efa7acaSMatthias Ringwald             upper_pdu = (mesh_upper_transport_pdu_t *) pdu;
9959efa7acaSMatthias Ringwald             while (upper_pdu->segments) {
9969efa7acaSMatthias Ringwald                 mesh_network_pdu_t *segment = (mesh_network_pdu_t *) btstack_linked_list_pop(&upper_pdu->segments);
9979efa7acaSMatthias Ringwald                 mesh_network_pdu_free(segment);
9989efa7acaSMatthias Ringwald             }
9999efa7acaSMatthias Ringwald             btstack_memory_mesh_upper_transport_pdu_free(upper_pdu);
10009efa7acaSMatthias Ringwald             break;
1001e9c16304SMatthias Ringwald         default:
1002e9c16304SMatthias Ringwald             btstack_assert(false);
1003e9c16304SMatthias Ringwald             break;
1004e9c16304SMatthias Ringwald     }
1005e9c16304SMatthias Ringwald }
1006e9c16304SMatthias Ringwald 
1007e9c16304SMatthias Ringwald void mesh_upper_transport_message_processed_by_higher_layer(mesh_pdu_t * pdu){
1008e9c16304SMatthias Ringwald     crypto_active = 0;
1009e9c16304SMatthias Ringwald     switch (pdu->pdu_type){
1010e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_ACCESS:
1011e9c16304SMatthias Ringwald             mesh_upper_transport_process_access_message_done((mesh_access_pdu_t *) pdu);
1012e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_CONTROL:
1013e9c16304SMatthias Ringwald             mesh_upper_transport_process_control_message_done((mesh_control_pdu_t *) pdu);
1014e9c16304SMatthias Ringwald             break;
1015e9c16304SMatthias Ringwald         default:
1016e9c16304SMatthias Ringwald             btstack_assert(0);
1017e9c16304SMatthias Ringwald             break;
1018e9c16304SMatthias Ringwald     }
1019e9c16304SMatthias Ringwald }
1020e9c16304SMatthias Ringwald 
1021e9c16304SMatthias Ringwald void mesh_upper_transport_send_access_pdu(mesh_pdu_t *pdu){
1022e9c16304SMatthias Ringwald     switch (pdu->pdu_type){
1023e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
1024e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
1025e9c16304SMatthias Ringwald             break;
1026e9c16304SMatthias Ringwald         default:
1027e9c16304SMatthias Ringwald             btstack_assert(false);
1028e9c16304SMatthias Ringwald             break;
1029e9c16304SMatthias Ringwald     }
1030e9c16304SMatthias Ringwald 
103143d07e7fSMatthias Ringwald     btstack_assert(((mesh_upper_transport_pdu_t *) pdu)->lower_pdu == NULL);
103243d07e7fSMatthias Ringwald 
1033e9c16304SMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_outgoing, (btstack_linked_item_t*) pdu);
1034e9c16304SMatthias Ringwald     mesh_upper_transport_run();
1035e9c16304SMatthias Ringwald }
1036e9c16304SMatthias Ringwald 
1037e9c16304SMatthias Ringwald void mesh_upper_transport_send_control_pdu(mesh_pdu_t * pdu){
1038e9c16304SMatthias Ringwald     switch (pdu->pdu_type){
1039e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
1040e9c16304SMatthias Ringwald             break;
1041e9c16304SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_CONTROL:
1042e9c16304SMatthias Ringwald             btstack_assert( ((mesh_network_pdu_t *) pdu)->len >= 9);
1043e9c16304SMatthias Ringwald             break;
1044e9c16304SMatthias Ringwald         default:
1045e9c16304SMatthias Ringwald             btstack_assert(false);
1046e9c16304SMatthias Ringwald             break;
1047e9c16304SMatthias Ringwald     }
1048e9c16304SMatthias Ringwald 
1049e9c16304SMatthias Ringwald     btstack_linked_list_add_tail(&upper_transport_outgoing, (btstack_linked_item_t*) pdu);
1050e9c16304SMatthias Ringwald     mesh_upper_transport_run();
1051e9c16304SMatthias Ringwald }
1052e9c16304SMatthias Ringwald 
105377ba3d3fSMatthias 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,
105477ba3d3fSMatthias Ringwald                                                                   const uint8_t * control_pdu_data, uint16_t control_pdu_len){
105577ba3d3fSMatthias Ringwald 
105677ba3d3fSMatthias Ringwald     if (control_pdu_len > 11) return 1;
105777ba3d3fSMatthias Ringwald 
105877ba3d3fSMatthias Ringwald     const mesh_network_key_t * network_key = mesh_network_key_list_get(netkey_index);
105977ba3d3fSMatthias Ringwald     if (!network_key) return 1;
106077ba3d3fSMatthias Ringwald 
106177ba3d3fSMatthias Ringwald     uint8_t transport_pdu_data[12];
106277ba3d3fSMatthias Ringwald     transport_pdu_data[0] = opcode;
10636535961aSMatthias Ringwald     (void)memcpy(&transport_pdu_data[1], control_pdu_data, control_pdu_len);
106477ba3d3fSMatthias Ringwald     uint16_t transport_pdu_len = control_pdu_len + 1;
106577ba3d3fSMatthias Ringwald 
106677ba3d3fSMatthias Ringwald     // setup network_pdu
106787e3599cSMatthias Ringwald     mesh_network_setup_pdu(network_pdu, netkey_index, network_key->nid, 1, ttl, 0, src, dest, transport_pdu_data, transport_pdu_len);
106877ba3d3fSMatthias Ringwald 
106977ba3d3fSMatthias Ringwald     return 0;
107077ba3d3fSMatthias Ringwald }
107177ba3d3fSMatthias Ringwald 
1072a97cf70aSMatthias Ringwald static uint8_t mesh_upper_transport_setup_segmented_control_pdu(mesh_upper_transport_pdu_t * upper_pdu, uint16_t netkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t opcode,
107377ba3d3fSMatthias Ringwald                                                                 const uint8_t * control_pdu_data, uint16_t control_pdu_len){
107477ba3d3fSMatthias Ringwald 
107577ba3d3fSMatthias Ringwald     if (control_pdu_len > 256) return 1;
107677ba3d3fSMatthias Ringwald 
107777ba3d3fSMatthias Ringwald     const mesh_network_key_t * network_key = mesh_network_key_list_get(netkey_index);
107877ba3d3fSMatthias Ringwald     if (!network_key) return 1;
107977ba3d3fSMatthias Ringwald 
1080a97cf70aSMatthias Ringwald     upper_pdu->ivi_nid = network_key->nid | ((mesh_get_iv_index_for_tx() & 1) << 7);
1081a97cf70aSMatthias Ringwald     upper_pdu->ctl_ttl = ttl;
1082a97cf70aSMatthias Ringwald     upper_pdu->src = src;
1083a97cf70aSMatthias Ringwald     upper_pdu->dst = dest;
1084a97cf70aSMatthias Ringwald     upper_pdu->transmic_len = 0;    // no TransMIC for control
1085a97cf70aSMatthias Ringwald     upper_pdu->netkey_index = netkey_index;
1086a97cf70aSMatthias Ringwald     upper_pdu->akf_aid_control = opcode;
108777ba3d3fSMatthias Ringwald 
10885368239bSMatthias Ringwald     // allocate segments
10895368239bSMatthias Ringwald     btstack_linked_list_t free_segments = NULL;
10905368239bSMatthias Ringwald     bool ok = mesh_segmented_allocate_segments( &free_segments, control_pdu_len);
10915368239bSMatthias Ringwald     if (!ok) return 1;
10925368239bSMatthias Ringwald     // store control pdu
10935368239bSMatthias Ringwald     mesh_segmented_store_payload(control_pdu_data, control_pdu_len, &free_segments, &upper_pdu->segments);
1094a97cf70aSMatthias Ringwald     upper_pdu->len = control_pdu_len;
109577ba3d3fSMatthias Ringwald     return 0;
109677ba3d3fSMatthias Ringwald }
109777ba3d3fSMatthias Ringwald 
109877ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_control_pdu(mesh_pdu_t * pdu, uint16_t netkey_index,
109977ba3d3fSMatthias Ringwald                                                uint8_t ttl, uint16_t src, uint16_t dest, uint8_t opcode, const uint8_t * control_pdu_data, uint16_t control_pdu_len){
110077ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
11018ec600ffSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_CONTROL:
110277ba3d3fSMatthias 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);
1103a97cf70aSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_CONTROL:
1104a97cf70aSMatthias Ringwald             return mesh_upper_transport_setup_segmented_control_pdu((mesh_upper_transport_pdu_t *) pdu,  netkey_index, ttl, src, dest, opcode, control_pdu_data, control_pdu_len);
110577ba3d3fSMatthias Ringwald         default:
1106ba660d42SMatthias Ringwald             btstack_assert(0);
110777ba3d3fSMatthias Ringwald             return 1;
110877ba3d3fSMatthias Ringwald     }
110977ba3d3fSMatthias Ringwald }
111077ba3d3fSMatthias Ringwald 
1111b1a7ae8fSMatthias Ringwald static uint8_t mesh_upper_transport_setup_segmented_access_pdu_header(mesh_access_pdu_t * access_pdu, uint16_t netkey_index,
1112a332c5a2SMatthias Ringwald                                                                       uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic){
111377ba3d3fSMatthias Ringwald 
111477ba3d3fSMatthias Ringwald     // get app or device key
111577ba3d3fSMatthias Ringwald     const mesh_transport_key_t *appkey;
111677ba3d3fSMatthias Ringwald     appkey = mesh_transport_key_get(appkey_index);
111777ba3d3fSMatthias Ringwald     if (appkey == NULL) {
1118a332c5a2SMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - appkey_index %x unknown\n", appkey_index);
111977ba3d3fSMatthias Ringwald         return 1;
112077ba3d3fSMatthias Ringwald     }
112177ba3d3fSMatthias Ringwald     uint8_t akf_aid = (appkey->akf << 6) | appkey->aid;
112277ba3d3fSMatthias Ringwald 
112377ba3d3fSMatthias Ringwald     // lookup network by netkey_index
112477ba3d3fSMatthias Ringwald     const mesh_network_key_t *network_key = mesh_network_key_list_get(netkey_index);
112577ba3d3fSMatthias Ringwald     if (!network_key) return 1;
1126a332c5a2SMatthias Ringwald     if (network_key == NULL) {
1127a332c5a2SMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - netkey_index %x unknown\n", appkey_index);
1128a332c5a2SMatthias Ringwald         return 1;
1129a332c5a2SMatthias Ringwald     }
113077ba3d3fSMatthias Ringwald 
113177ba3d3fSMatthias Ringwald     const uint8_t trans_mic_len = szmic ? 8 : 4;
113277ba3d3fSMatthias Ringwald 
113377ba3d3fSMatthias Ringwald     // store in transport pdu
1134b1a7ae8fSMatthias Ringwald     access_pdu->transmic_len = trans_mic_len;
1135b1a7ae8fSMatthias Ringwald     access_pdu->netkey_index = netkey_index;
1136b1a7ae8fSMatthias Ringwald     access_pdu->appkey_index = appkey_index;
1137b1a7ae8fSMatthias Ringwald     access_pdu->akf_aid_control = akf_aid;
11382ddfd9e2SMatthias Ringwald     uint8_t iviNid = network_key->nid | ((mesh_get_iv_index_for_tx() & 1) << 7);
11392ddfd9e2SMatthias Ringwald     access_pdu->ivi_nid = iviNid;
11402ddfd9e2SMatthias Ringwald     access_pdu->src = src;
11412ddfd9e2SMatthias Ringwald     access_pdu->dst = dest;
11422ddfd9e2SMatthias Ringwald     access_pdu->ctl_ttl = ttl;
114377ba3d3fSMatthias Ringwald     return 0;
114477ba3d3fSMatthias Ringwald }
114577ba3d3fSMatthias Ringwald 
11461b51af65SMatthias Ringwald static uint8_t mesh_upper_transport_setup_upper_access_pdu_header(mesh_upper_transport_pdu_t * upper_pdu, uint16_t netkey_index,
1147242605faSMatthias Ringwald                                                                   uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic){
1148242605faSMatthias Ringwald 
1149242605faSMatthias Ringwald     // get app or device key
1150242605faSMatthias Ringwald     const mesh_transport_key_t *appkey;
1151242605faSMatthias Ringwald     appkey = mesh_transport_key_get(appkey_index);
1152242605faSMatthias Ringwald     if (appkey == NULL) {
1153242605faSMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - appkey_index %x unknown\n", appkey_index);
1154242605faSMatthias Ringwald         return 1;
1155242605faSMatthias Ringwald     }
1156242605faSMatthias Ringwald     uint8_t akf_aid = (appkey->akf << 6) | appkey->aid;
1157242605faSMatthias Ringwald 
1158242605faSMatthias Ringwald     // lookup network by netkey_index
1159242605faSMatthias Ringwald     const mesh_network_key_t *network_key = mesh_network_key_list_get(netkey_index);
1160242605faSMatthias Ringwald     if (!network_key) return 1;
1161242605faSMatthias Ringwald     if (network_key == NULL) {
1162242605faSMatthias Ringwald         printf("[!] Upper transport, setup segmented Access PDU - netkey_index %x unknown\n", appkey_index);
1163242605faSMatthias Ringwald         return 1;
1164242605faSMatthias Ringwald     }
1165242605faSMatthias Ringwald 
1166242605faSMatthias Ringwald     const uint8_t trans_mic_len = szmic ? 8 : 4;
1167242605faSMatthias Ringwald 
1168242605faSMatthias Ringwald     // store in transport pdu
1169242605faSMatthias Ringwald     upper_pdu->ivi_nid = network_key->nid | ((mesh_get_iv_index_for_tx() & 1) << 7);
1170242605faSMatthias Ringwald     upper_pdu->ctl_ttl = ttl;
1171242605faSMatthias Ringwald     upper_pdu->src = src;
1172242605faSMatthias Ringwald     upper_pdu->dst = dest;
1173242605faSMatthias Ringwald     upper_pdu->transmic_len = trans_mic_len;
1174242605faSMatthias Ringwald     upper_pdu->netkey_index = netkey_index;
1175242605faSMatthias Ringwald     upper_pdu->appkey_index = appkey_index;
1176242605faSMatthias Ringwald     upper_pdu->akf_aid_control = akf_aid;
1177242605faSMatthias Ringwald     return 0;
1178242605faSMatthias Ringwald }
117977ba3d3fSMatthias Ringwald 
11801b51af65SMatthias Ringwald static uint8_t mesh_upper_transport_setup_upper_access_pdu(mesh_upper_transport_pdu_t * upper_pdu, uint16_t netkey_index, uint16_t appkey_index, uint8_t ttl, uint16_t src, uint16_t dest,
1181242605faSMatthias Ringwald                                                            uint8_t szmic, const uint8_t * access_pdu_data, uint8_t access_pdu_len){
11821b51af65SMatthias Ringwald     int status = mesh_upper_transport_setup_upper_access_pdu_header(upper_pdu, netkey_index, appkey_index, ttl, src,
11831b51af65SMatthias Ringwald                                                                     dest, szmic);
1184242605faSMatthias Ringwald     if (status) return status;
1185242605faSMatthias Ringwald 
11865368239bSMatthias Ringwald     // allocate segments
11875368239bSMatthias Ringwald     btstack_linked_list_t free_segments = NULL;
11885368239bSMatthias Ringwald     bool ok = mesh_segmented_allocate_segments( &free_segments, access_pdu_len);
11895368239bSMatthias Ringwald     if (!ok) return 1;
11905368239bSMatthias Ringwald     // store control pdu
11915368239bSMatthias Ringwald     mesh_segmented_store_payload(access_pdu_data, access_pdu_len, &free_segments, &upper_pdu->segments);
1192242605faSMatthias Ringwald     upper_pdu->len = access_pdu_len;
1193242605faSMatthias Ringwald     return 0;
1194242605faSMatthias Ringwald }
1195242605faSMatthias Ringwald 
1196242605faSMatthias Ringwald 
119777ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_access_pdu_header(mesh_pdu_t * pdu, uint16_t netkey_index, uint16_t appkey_index,
119877ba3d3fSMatthias Ringwald                                                      uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic){
119977ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
1200b1a7ae8fSMatthias Ringwald         case MESH_PDU_TYPE_ACCESS:
1201b1a7ae8fSMatthias Ringwald             return mesh_upper_transport_setup_segmented_access_pdu_header((mesh_access_pdu_t *) pdu, netkey_index, appkey_index, ttl, src, dest, szmic);
120277ba3d3fSMatthias Ringwald         default:
1203d38a7664SMatthias Ringwald             btstack_assert(false);
120477ba3d3fSMatthias Ringwald             return 1;
120577ba3d3fSMatthias Ringwald     }
120677ba3d3fSMatthias Ringwald }
120777ba3d3fSMatthias Ringwald 
120877ba3d3fSMatthias Ringwald uint8_t mesh_upper_transport_setup_access_pdu(mesh_pdu_t * pdu, uint16_t netkey_index, uint16_t appkey_index,
120977ba3d3fSMatthias Ringwald                                               uint8_t ttl, uint16_t src, uint16_t dest, uint8_t szmic,
121077ba3d3fSMatthias Ringwald                                               const uint8_t * access_pdu_data, uint8_t access_pdu_len){
121177ba3d3fSMatthias Ringwald     switch (pdu->pdu_type){
1212242605faSMatthias Ringwald         case MESH_PDU_TYPE_UPPER_SEGMENTED_ACCESS:
12131b51af65SMatthias Ringwald         case MESH_PDU_TYPE_UPPER_UNSEGMENTED_ACCESS:
12141b51af65SMatthias Ringwald             return mesh_upper_transport_setup_upper_access_pdu((mesh_upper_transport_pdu_t *) pdu, netkey_index,
12151b51af65SMatthias Ringwald                                                                appkey_index, ttl, src, dest, szmic, access_pdu_data,
12161b51af65SMatthias Ringwald                                                                access_pdu_len);
121777ba3d3fSMatthias Ringwald         default:
1218d38a7664SMatthias Ringwald             btstack_assert(false);
121977ba3d3fSMatthias Ringwald             return 1;
122077ba3d3fSMatthias Ringwald     }
122177ba3d3fSMatthias Ringwald }
122277ba3d3fSMatthias Ringwald 
1223404d2482SMatthias Ringwald void mesh_upper_transport_register_access_message_handler(void (*callback)(mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu)) {
12245beab91eSMatthias Ringwald     mesh_access_message_handler = callback;
12255beab91eSMatthias Ringwald }
12265beab91eSMatthias Ringwald 
1227404d2482SMatthias Ringwald void mesh_upper_transport_register_control_message_handler(void (*callback)(mesh_transport_callback_type_t callback_type, mesh_transport_status_t status, mesh_pdu_t * pdu)){
12285beab91eSMatthias Ringwald     mesh_control_message_handler = callback;
12295beab91eSMatthias Ringwald }
12305beab91eSMatthias Ringwald 
12315beab91eSMatthias Ringwald void mesh_upper_transport_init(){
12325beab91eSMatthias Ringwald     mesh_lower_transport_set_higher_layer_handler(&mesh_upper_transport_pdu_handler);
12335beab91eSMatthias Ringwald }
1234d8b26eceSMatthias Ringwald 
1235d8b26eceSMatthias Ringwald 
1236f1219446SMatthias Ringwald void mesh_upper_transport_message_init(mesh_upper_transport_builder_t * builder, mesh_pdu_type_t pdu_type) {
1237f1219446SMatthias Ringwald     btstack_assert(builder != NULL);
1238d8b26eceSMatthias Ringwald 
1239f1219446SMatthias Ringwald     builder->pdu = btstack_memory_mesh_upper_transport_pdu_get();
1240f1219446SMatthias Ringwald     if (!builder->pdu) return;
1241d8b26eceSMatthias Ringwald 
1242ce16a729SMatthias Ringwald     builder->segment = NULL;
1243f1219446SMatthias Ringwald     builder->pdu->pdu_header.pdu_type = pdu_type;
1244f1219446SMatthias Ringwald     builder->pdu->transmic_len = 4;
1245f1219446SMatthias Ringwald     builder->pdu->ack_opcode = MESH_ACCESS_OPCODE_NOT_SET;
1246d8b26eceSMatthias Ringwald }
1247d8b26eceSMatthias Ringwald 
1248d8b26eceSMatthias Ringwald 
1249f1219446SMatthias Ringwald void mesh_upper_transport_message_add_data(mesh_upper_transport_builder_t * builder, const uint8_t * data, uint16_t data_len){
1250f1219446SMatthias Ringwald     btstack_assert(builder != NULL);
1251f1219446SMatthias Ringwald 
1252f1219446SMatthias Ringwald     if (builder->pdu == NULL) return;
1253923523a3SMatthias Ringwald 
1254d8b26eceSMatthias Ringwald     uint16_t bytes_current_segment = 0;
1255ce16a729SMatthias Ringwald     if (builder->segment){
1256ce16a729SMatthias Ringwald         bytes_current_segment = MESH_NETWORK_PAYLOAD_MAX - builder->segment->len;
1257d8b26eceSMatthias Ringwald     }
1258d8b26eceSMatthias Ringwald     while (data_len > 0){
1259d8b26eceSMatthias Ringwald         if (bytes_current_segment == 0){
1260ce16a729SMatthias Ringwald             builder->segment = (mesh_network_pdu_t *) mesh_network_pdu_get();
1261ce16a729SMatthias Ringwald             if (builder->segment == NULL) {
1262f1219446SMatthias Ringwald                 mesh_upper_transport_pdu_free((mesh_pdu_t *) builder->pdu);
1263f1219446SMatthias Ringwald                 builder->pdu = NULL;
1264923523a3SMatthias Ringwald                 return;
1265923523a3SMatthias Ringwald             }
1266ce16a729SMatthias Ringwald             btstack_linked_list_add_tail(&builder->pdu->segments, (btstack_linked_item_t *) builder->segment);
1267d8b26eceSMatthias Ringwald             bytes_current_segment = MESH_NETWORK_PAYLOAD_MAX;
1268d8b26eceSMatthias Ringwald         }
1269d8b26eceSMatthias Ringwald         uint16_t bytes_to_copy = btstack_min(bytes_current_segment, data_len);
1270ce16a729SMatthias Ringwald         (void) memcpy(&builder->segment->data[builder->segment->len], data, bytes_to_copy);
1271ce16a729SMatthias Ringwald         builder->segment->len += bytes_to_copy;
1272d8b26eceSMatthias Ringwald         bytes_current_segment -= bytes_to_copy;
1273d8b26eceSMatthias Ringwald         data                  += bytes_to_copy;
1274d8b26eceSMatthias Ringwald         data_len              -= bytes_to_copy;
1275d8b26eceSMatthias Ringwald     }
1276d8b26eceSMatthias Ringwald }
1277d8b26eceSMatthias Ringwald 
1278f1219446SMatthias Ringwald void mesh_upper_transport_message_add_uint8(mesh_upper_transport_builder_t * builder, uint8_t value){
1279f1219446SMatthias Ringwald     mesh_upper_transport_message_add_data(builder, &value, 1);
1280d8b26eceSMatthias Ringwald }
1281d8b26eceSMatthias Ringwald 
1282f1219446SMatthias Ringwald void mesh_upper_transport_message_add_uint16(mesh_upper_transport_builder_t * builder, uint16_t value){
1283d8b26eceSMatthias Ringwald     uint8_t buffer[2];
1284d8b26eceSMatthias Ringwald     little_endian_store_16(buffer, 0, value);
1285f1219446SMatthias Ringwald     mesh_upper_transport_message_add_data(builder, buffer, sizeof(buffer));
1286d8b26eceSMatthias Ringwald }
1287d8b26eceSMatthias Ringwald 
1288f1219446SMatthias Ringwald void mesh_upper_transport_message_add_uint24(mesh_upper_transport_builder_t * builder, uint16_t value){
1289d8b26eceSMatthias Ringwald     uint8_t buffer[3];
1290d8b26eceSMatthias Ringwald     little_endian_store_24(buffer, 0, value);
1291f1219446SMatthias Ringwald     mesh_upper_transport_message_add_data(builder, buffer, sizeof(buffer));
1292d8b26eceSMatthias Ringwald }
1293d8b26eceSMatthias Ringwald 
1294f1219446SMatthias Ringwald void mesh_upper_transport_message_add_uint32(mesh_upper_transport_builder_t * builder, uint16_t value){
1295d8b26eceSMatthias Ringwald     uint8_t buffer[4];
1296d8b26eceSMatthias Ringwald     little_endian_store_32(buffer, 0, value);
1297f1219446SMatthias Ringwald     mesh_upper_transport_message_add_data(builder, buffer, sizeof(buffer));
1298d8b26eceSMatthias Ringwald }
1299eeaf8ec1SMatthias Ringwald 
1300f1219446SMatthias Ringwald mesh_upper_transport_pdu_t * mesh_upper_transport_message_finalize(mesh_upper_transport_builder_t * builder){
1301f1219446SMatthias Ringwald     return builder->pdu;
1302eeaf8ec1SMatthias Ringwald }
1303