bluetooth/adapter_nrf528xx.go at 5ca5416572d34ee84d8047e448d84ee04c6525b1 · funkycode/bluetooth · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
//go:build (softdevice && s113v7) || (softdevice && s132v6) || (softdevice && s140v6) || (softdevice && s140v7)

package bluetooth

// This file defines the SoftDevice adapter for all nrf52-series chips.

/*
#include "nrf_sdm.h"
#include "nrf_nvic.h"
#include "ble.h"
#include "ble_gap.h"

void assertHandler(void);
*/
import "C"

import (
	"machine"
	"unsafe"
)

// TODO: Probably it should be in adapter_sd, but as it's usage is added only for nrf528xx-full.go
// as well as i do not have other machines to test, adding it here for now

type GapIOCapability uint8

const (
	DisplayOnlyGapIOCapability     = C.BLE_GAP_IO_CAPS_DISPLAY_ONLY
	DisplayYesNoGapIOCapability    = C.BLE_GAP_IO_CAPS_DISPLAY_YESNO
	KeyboardOnlyGapIOCapability    = C.BLE_GAP_IO_CAPS_KEYBOARD_ONLY
	NoneGapIOCapability            = C.BLE_GAP_IO_CAPS_NONE
	KeyboardDisplayGapIOCapability = C.BLE_GAP_IO_CAPS_KEYBOARD_DISPLAY
)

var (
	secParams = C.ble_gap_sec_params_t{
		min_key_size: 7, // not sure if those are the best default length
		max_key_size: 16,
	}

	secKeySet C.ble_gap_sec_keyset_t
)

// are those should be methods for adapter as they are relevant for sd only
func SetSecParamsBonding() {
	secParams.set_bitfield_bond(1)
}

func SetSecParamsLesc() {
	secParams.set_bitfield_bond(0) //1)
	secParams.set_bitfield_lesc(1)
	//secParams.set_bitfield_mitm(1)
	//secParams.set_bitfield_io_caps(uint8(DisplayOnlyGapIOCapability))
}

func SetLesPublicKey(key []uint8) {
	secKeySet.keys_peer = C.ble_gap_sec_keys_t{
		p_pk:       &C.ble_gap_lesc_p256_pk_t{},
		p_enc_key:  &C.ble_gap_enc_key_t{},
		p_id_key:   &C.ble_gap_id_key_t{},
		p_sign_key: &C.ble_gap_sign_info_t{},
	}
	secKeySet.keys_own = C.ble_gap_sec_keys_t{
		p_enc_key:  &C.ble_gap_enc_key_t{},
		p_id_key:   &C.ble_gap_id_key_t{},
		p_sign_key: &C.ble_gap_sign_info_t{},
		p_pk: &C.ble_gap_lesc_p256_pk_t{
			pk: [C.BLE_GAP_LESC_P256_PK_LEN]uint8(key),
		},
	}
}

func ReplyLesc(key []byte) error {
	lescKey := C.ble_gap_lesc_dhkey_t{
		key: [C.BLE_GAP_LESC_DHKEY_LEN]uint8(key),
	}
	errCode := C.sd_ble_gap_lesc_dhkey_reply(currentConnection.Reg, &lescKey)
	if errCode != 0 {
		return Error(errCode)

	}
	return nil
}

func SetSecCapabilities(cap GapIOCapability) {
	secParams.set_bitfield_io_caps(uint8(cap))
}

//export assertHandler
func assertHandler() {
	println("SoftDevice assert")
}

var clockConfigXtal C.nrf_clock_lf_cfg_t = C.nrf_clock_lf_cfg_t{
	source:       C.NRF_CLOCK_LF_SRC_XTAL,
	rc_ctiv:      0,
	rc_temp_ctiv: 0,
	accuracy:     C.NRF_CLOCK_LF_ACCURACY_250_PPM,
}

//go:extern __app_ram_base
var appRAMBase [0]uint32

func (a *Adapter) enable() error {
	// Enable the SoftDevice.
	var clockConfig *C.nrf_clock_lf_cfg_t
	if machine.HasLowFrequencyCrystal {
		clockConfig = &clockConfigXtal
	}
	errCode := C.sd_softdevice_enable(clockConfig, C.nrf_fault_handler_t(C.assertHandler))
	if errCode != 0 {
		return Error(errCode)
	}

	// Enable the BLE stack.
	appRAMBase := uint32(uintptr(unsafe.Pointer(&appRAMBase)))
	errCode = C.sd_ble_enable(&appRAMBase)
	return makeError(errCode)
}

func (a *Adapter) Address() (MACAddress, error) {
	var addr C.ble_gap_addr_t
	errCode := C.sd_ble_gap_addr_get(&addr)
	if errCode != 0 {
		return MACAddress{}, Error(errCode)
	}
	return MACAddress{MAC: addr.addr}, nil
}

func (a *Adapter) SetLescRequestHandler(c func(pubKey []uint8)) {
	a.lescRequestHandler = c
}