This commit is contained in:
Marcel 2023-01-11 22:12:04 +01:00
parent 9288503a1e
commit 38b758398d
5 changed files with 270 additions and 145 deletions

View file

@ -25,7 +25,7 @@ use crate::{
fabric::FabricMgr,
interaction_model::InteractionModel,
mdns::Mdns,
pairing::print_pairing_code_and_qr,
pairing::{print_pairing_code_and_qr, DiscoveryCapabilities},
secure_channel::core::SecureChannel,
transport,
};
@ -70,7 +70,7 @@ impl Matter {
&dev_det.device_name,
);
print_pairing_code_and_qr(dev_det, dev_comm);
print_pairing_code_and_qr(dev_det, dev_comm, DiscoveryCapabilities::default());
let fabric_mgr = Arc::new(FabricMgr::new()?);
let acl_mgr = Arc::new(AclMgr::new()?);

View file

@ -0,0 +1,60 @@
use super::*;
pub(super) fn compute_pairing_code(comm_data: &CommissioningData) -> String {
// 0: no Vendor ID and Product ID present in Manual Pairing Code
const VID_PID_PRESENT: u8 = 0;
let CommissioningData {
discriminator,
passwd,
..
} = comm_data;
let mut digits = String::new();
digits.push_str(&((VID_PID_PRESENT << 2) | (discriminator >> 10) as u8).to_string());
digits.push_str(&format!(
"{:0>5}",
((discriminator & 0x300) << 6) | (passwd & 0x3FFF) as u16
));
digits.push_str(&format!("{:0>4}", passwd >> 14));
let check_digit = digits.calculate_verhoeff_check_digit();
digits.push_str(&check_digit.to_string());
digits
}
pub(super) fn pretty_print_pairing_code(pairing_code: &str) {
assert!(pairing_code.len() == 11);
let mut pretty = String::new();
pretty.push_str(&pairing_code[..4]);
pretty.push('-');
pretty.push_str(&pairing_code[4..8]);
pretty.push('-');
pretty.push_str(&pairing_code[8..]);
info!("Pairing Code: {}", pretty);
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn can_compute_pairing_code() {
let comm_data = CommissioningData {
passwd: 123456,
discriminator: 250,
..Default::default()
};
let pairing_code = compute_pairing_code(&comm_data);
assert_eq!(pairing_code, "00876800071");
let comm_data = CommissioningData {
passwd: 34567890,
discriminator: 2976,
..Default::default()
};
let pairing_code = compute_pairing_code(&comm_data);
assert_eq!(pairing_code, "26318621095");
}
}

96
matter/src/pairing/mod.rs Normal file
View file

@ -0,0 +1,96 @@
/*
*
* Copyright (c) 2020-2022 Project CHIP Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//! This module contains the logic for generating the pairing code and the QR code for easy pairing.
pub mod code;
pub mod qr;
pub mod vendor_identifiers;
use log::info;
use qrcode::{render::unicode, QrCode, Version};
use verhoeff::Verhoeff;
use crate::{
codec::base38, data_model::cluster_basic_information::BasicInfoConfig, error::Error,
CommissioningData,
};
use self::{
code::{compute_pairing_code, pretty_print_pairing_code},
qr::{payload_base38_representation, print_qr_code, QrCodeData},
};
pub struct DiscoveryCapabilities {
on_ip_network: bool,
ble: bool,
soft_access_point: bool,
}
impl DiscoveryCapabilities {
pub fn new(on_ip_network: bool, ble: bool, soft_access_point: bool) -> Self {
DiscoveryCapabilities {
on_ip_network,
ble,
soft_access_point,
}
}
pub fn has_value(&self) -> bool {
self.on_ip_network || self.ble || self.soft_access_point
}
}
impl Default for DiscoveryCapabilities {
fn default() -> Self {
DiscoveryCapabilities {
on_ip_network: true,
ble: false,
soft_access_point: false,
}
}
}
impl DiscoveryCapabilities {
fn as_bits(&self) -> u8 {
let mut bits = 0;
if self.soft_access_point {
bits |= 1 << 0;
}
if self.ble {
bits |= 1 << 1;
}
if self.on_ip_network {
bits |= 1 << 2;
}
bits
}
}
/// Prepares and prints the pairing code and the QR code for easy pairing.
pub fn print_pairing_code_and_qr(
dev_det: &BasicInfoConfig,
comm_data: &CommissioningData,
discovery_capabilities: DiscoveryCapabilities,
) {
let pairing_code = compute_pairing_code(comm_data);
let qr_code_data = QrCodeData::new(dev_det, comm_data, discovery_capabilities);
let data_str = payload_base38_representation(&qr_code_data).expect("Failed to encode");
pretty_print_pairing_code(&pairing_code);
print_qr_code(&data_str);
}

View file

@ -1,29 +1,6 @@
/*
*
* Copyright (c) 2020-2022 Project CHIP Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//! This module contains the logic for generating the pairing code and the QR code for easy pairing.
use log::info;
use qrcode::{render::unicode, QrCode, Version};
use verhoeff::Verhoeff;
use crate::{
codec::base38, data_model::cluster_basic_information::BasicInfoConfig, error::Error,
CommissioningData,
use super::{
vendor_identifiers::{is_vendor_id_valid_operationally, VendorId},
*,
};
const LONG_BITS: usize = 12;
@ -45,49 +22,10 @@ const TOTAL_PAYLOAD_DATA_SIZE_IN_BITS: usize = VERSION_FIELD_LENGTH_IN_BITS
+ PADDING_FIELD_LENGTH_IN_BITS;
const TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES: usize = TOTAL_PAYLOAD_DATA_SIZE_IN_BITS / 8;
#[repr(u8)]
#[derive(Clone, Copy)]
pub enum CommissionningFlowType {
Standard = 0,
UserIntent = 1,
Custom = 2,
}
pub struct DiscoveryCapabilitiesSchema {
on_ip_network: bool,
ble: bool,
soft_access_point: bool,
}
impl DiscoveryCapabilitiesSchema {
pub fn new(on_ip_network: bool, ble: bool, soft_access_point: bool) -> Self {
DiscoveryCapabilitiesSchema {
on_ip_network,
ble,
soft_access_point,
}
}
}
impl DiscoveryCapabilitiesSchema {
fn as_bits(&self) -> u8 {
let mut bits = 0;
if self.soft_access_point {
bits |= 1 << 0;
}
if self.ble {
bits |= 1 << 1;
}
if self.on_ip_network {
bits |= 1 << 2;
}
bits
}
}
pub struct QrCodeData<'data> {
version: u8,
flow_type: CommissionningFlowType,
discovery_capabilities: DiscoveryCapabilitiesSchema,
discovery_capabilities: DiscoveryCapabilities,
dev_det: &'data BasicInfoConfig,
comm_data: &'data CommissioningData,
}
@ -96,70 +34,113 @@ impl<'data> QrCodeData<'data> {
pub fn new(
dev_det: &'data BasicInfoConfig,
comm_data: &'data CommissioningData,
discovery_capabilities: DiscoveryCapabilitiesSchema,
discovery_capabilities: DiscoveryCapabilities,
) -> Self {
const DEFAULT_VERSION: u8 = 0;
QrCodeData {
version: DEFAULT_VERSION,
flow_type: CommissionningFlowType::Standard,
discovery_capabilities,
dev_det,
comm_data,
}
}
fn is_valid(&self) -> bool {
// 3-bit value specifying the QR code payload version.
if self.version >= 1 << VERSION_FIELD_LENGTH_IN_BITS {
return false;
}
if !self.discovery_capabilities.has_value() {
return false;
}
if self.comm_data.passwd >= 1 << SETUP_PINCODE_FIELD_LENGTH_IN_BITS {
return false;
}
self.check_payload_common_constraints()
}
fn check_payload_common_constraints(&self) -> bool {
// A version not equal to 0 would be invalid for v1 and would indicate new format (e.g. version 2)
if self.version != 0 {
return false;
}
if !Self::is_valid_setup_pin(self.comm_data.passwd) {
return false;
}
// VendorID must be unspecified (0) or in valid range expected.
if !is_vendor_id_valid_operationally(self.dev_det.vid)
&& (self.dev_det.vid != VendorId::CommonOrUnspecified as u16)
{
return false;
}
// A value of 0x0000 SHALL NOT be assigned to a product since Product ID = 0x0000 is used for these specific cases:
// * To announce an anonymized Product ID as part of device discovery
// * To indicate an OTA software update file applies to multiple Product IDs equally.
// * To avoid confusion when presenting the Onboarding Payload for ECM with multiple nodes
if self.dev_det.pid == 0 && self.dev_det.vid != VendorId::CommonOrUnspecified as u16 {
return false;
}
true
}
fn is_valid_setup_pin(setup_pin: u32) -> bool {
const SETUP_PINCODE_MAXIMUM_VALUE: u32 = 99999998;
const SETUP_PINCODE_UNDEFINED_VALUE: u32 = 0;
// SHALL be restricted to the values 0x0000001 to 0x5F5E0FE (00000001 to 99999998 in decimal), excluding the invalid Passcode
// values.
if setup_pin == SETUP_PINCODE_UNDEFINED_VALUE
|| setup_pin > SETUP_PINCODE_MAXIMUM_VALUE
|| setup_pin == 11111111
|| setup_pin == 22222222
|| setup_pin == 33333333
|| setup_pin == 44444444
|| setup_pin == 55555555
|| setup_pin == 66666666
|| setup_pin == 77777777
|| setup_pin == 88888888
|| setup_pin == 12345678
|| setup_pin == 87654321
{
return false;
}
true
}
}
#[repr(u8)]
#[derive(Clone, Copy)]
pub enum CommissionningFlowType {
Standard = 0,
UserIntent = 1,
Custom = 2,
}
struct TlvData {
data_length_in_bytes: u32,
}
/// Prepares and prints the pairing code and the QR code for easy pairing.
pub fn print_pairing_code_and_qr(dev_det: &BasicInfoConfig, comm_data: &CommissioningData) {
let pairing_code = compute_pairing_code(comm_data);
pub(super) fn payload_base38_representation(payload: &QrCodeData) -> Result<String, Error> {
let mut bits: [u8; TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES] = [0; TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES];
// todo: allow the discovery capabilities to be passed in
let disc_cap = DiscoveryCapabilitiesSchema::new(true, false, false);
let qr_code_data = QrCodeData::new(dev_det, comm_data, disc_cap);
let data_str = payload_base38_representation(&qr_code_data).expect("Failed to encode");
if !payload.is_valid() {
return Err(Error::InvalidArgument);
}
pretty_print_pairing_code(&pairing_code);
print_qr_code(&data_str);
payload_base38_representation_with_tlv(payload, &mut bits, None)
}
fn compute_pairing_code(comm_data: &CommissioningData) -> String {
// 0: no Vendor ID and Product ID present in Manual Pairing Code
const VID_PID_PRESENT: u8 = 0;
let CommissioningData {
discriminator,
passwd,
..
} = comm_data;
let mut digits = String::new();
digits.push_str(&((VID_PID_PRESENT << 2) | (discriminator >> 10) as u8).to_string());
digits.push_str(&format!(
"{:0>5}",
((discriminator & 0x300) << 6) | (passwd & 0x3FFF) as u16
));
digits.push_str(&format!("{:0>4}", passwd >> 14));
let check_digit = digits.calculate_verhoeff_check_digit();
digits.push_str(&check_digit.to_string());
digits
}
fn pretty_print_pairing_code(pairing_code: &str) {
assert!(pairing_code.len() == 11);
let mut pretty = String::new();
pretty.push_str(&pairing_code[..4]);
pretty.push('-');
pretty.push_str(&pairing_code[4..8]);
pretty.push('-');
pretty.push_str(&pairing_code[8..]);
info!("Pairing Code: {}", pretty);
}
fn print_qr_code(qr_data: &str) {
pub(super) fn print_qr_code(qr_data: &str) {
let code = QrCode::with_version(qr_data, Version::Normal(2), qrcode::EcLevel::M).unwrap();
let image = code
.render::<unicode::Dense1x2>()
@ -209,14 +190,6 @@ fn payload_base38_representation_with_tlv(
Ok(format!("MT:{}", base38_encoded))
}
fn payload_base38_representation(payload: &QrCodeData) -> Result<String, Error> {
let mut bits: [u8; TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES] = [0; TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES];
// VerifyOrReturnError(mPayload.isValidQRCodePayload(), CHIP_ERROR_INVALID_ARGUMENT);
payload_base38_representation_with_tlv(payload, &mut bits, None)
}
fn generate_bit_set(
payload: &QrCodeData,
bits: &mut [u8; TOTAL_PAYLOAD_DATA_SIZE_IN_BYTES],
@ -233,11 +206,10 @@ fn generate_bit_set(
return Err(Error::BufferTooSmall);
};
const VERSION: u64 = 0;
populate_bits(
bits,
&mut offset,
VERSION,
payload.version as u64,
VERSION_FIELD_LENGTH_IN_BITS,
total_payload_size_in_bits,
)?;
@ -308,25 +280,6 @@ fn generate_bit_set(
mod tests {
use super::*;
#[test]
fn can_compute_pairing_code() {
let comm_data = CommissioningData {
passwd: 123456,
discriminator: 250,
..Default::default()
};
let pairing_code = compute_pairing_code(&comm_data);
assert_eq!(pairing_code, "00876800071");
let comm_data = CommissioningData {
passwd: 34567890,
discriminator: 2976,
..Default::default()
};
let pairing_code = compute_pairing_code(&comm_data);
assert_eq!(pairing_code, "26318621095");
}
#[test]
fn can_base38_encode() {
const QR_CODE: &str = "MT:YNJV7VSC00CMVH7SR00";
@ -342,7 +295,7 @@ mod tests {
..Default::default()
};
let disc_cap = DiscoveryCapabilitiesSchema::new(false, true, false);
let disc_cap = DiscoveryCapabilities::new(false, true, false);
let qr_code_data = QrCodeData::new(&dev_det, &comm_data, disc_cap);
let data_str = payload_base38_representation(&qr_code_data).expect("Failed to encode");
assert_eq!(data_str, QR_CODE)

View file

@ -0,0 +1,16 @@
#[repr(u16)]
pub enum VendorId {
CommonOrUnspecified = 0x0000,
Apple = 0x1349,
Google = 0x6006,
TestVendor1 = 0xFFF1,
TestVendor2 = 0xFFF2,
TestVendor3 = 0xFFF3,
TestVendor4 = 0xFFF4,
NotSpecified = 0xFFFF,
}
pub fn is_vendor_id_valid_operationally(vendor_id: u16) -> bool {
(vendor_id != VendorId::CommonOrUnspecified as u16)
&& (vendor_id <= VendorId::TestVendor4 as u16)
}