Skip to content

Commit e4fd1f5

Browse files
committed
Compute qr version
1 parent 9ba0e01 commit e4fd1f5

File tree

1 file changed

+26
-15
lines changed

1 file changed

+26
-15
lines changed

matter/src/pairing/qr.rs

Lines changed: 26 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -92,14 +92,20 @@ impl<'data> QrSetupPayload<'data> {
9292
) -> Self {
9393
const DEFAULT_VERSION: u8 = 0;
9494

95-
QrSetupPayload {
95+
let mut result = QrSetupPayload {
9696
version: DEFAULT_VERSION,
9797
flow_type: CommissionningFlowType::Standard,
9898
discovery_capabilities,
9999
dev_det,
100100
comm_data,
101101
optional_data: BTreeMap::new(),
102+
};
103+
104+
if !dev_det.serial_no.is_empty() {
105+
result.add_serial_number(SerialNumber::String(dev_det.serial_no.clone()));
102106
}
107+
108+
result
103109
}
104110

105111
fn is_valid(&self) -> bool {
@@ -157,7 +163,7 @@ impl<'data> QrSetupPayload<'data> {
157163
&self.optional_data
158164
}
159165

160-
pub fn add_serial_number(&mut self, serial_number: SerialNumber) -> Result<(), Error> {
166+
pub fn add_serial_number(&mut self, serial_number: SerialNumber) {
161167
match serial_number {
162168
SerialNumber::String(serial_number) => self.add_optional_extension_data(
163169
SERIAL_NUMBER_TAG,
@@ -168,6 +174,7 @@ impl<'data> QrSetupPayload<'data> {
168174
QRCodeInfoType::UInt32(serial_number),
169175
),
170176
}
177+
.expect("can not add serial number");
171178
}
172179

173180
fn check_payload_common_constraints(&self) -> bool {
@@ -311,7 +318,9 @@ fn estimate_struct_overhead(first_field_size: usize) -> usize {
311318
}
312319

313320
pub(super) fn print_qr_code(qr_data: &str) {
314-
let code = QrCode::with_version(qr_data, Version::Normal(2), qrcode::EcLevel::M).unwrap();
321+
let needed_version = compute_qr_version(qr_data);
322+
let code =
323+
QrCode::with_version(qr_data, Version::Normal(needed_version), qrcode::EcLevel::M).unwrap();
315324
let image = code
316325
.render::<unicode::Dense1x2>()
317326
.dark_color(unicode::Dense1x2::Light)
@@ -320,6 +329,15 @@ pub(super) fn print_qr_code(qr_data: &str) {
320329
info!("\n{}", image);
321330
}
322331

332+
fn compute_qr_version(qr_data: &str) -> i16 {
333+
match qr_data.len() {
334+
0..=38 => 2,
335+
39..=61 => 3,
336+
62..=90 => 4,
337+
_ => 5,
338+
}
339+
}
340+
323341
fn populate_bits(
324342
bits: &mut [u8],
325343
offset: &mut usize,
@@ -543,22 +561,18 @@ mod tests {
543561
const QR_CODE: &str = "MT:-24J0AFN00KA064IJ3P0IXZB0DK5N1K8SQ1RYCU1-A40";
544562

545563
let comm_data = CommissioningData {
546-
passwd: 20202021,
564+
verifier: VerifierData::new_with_pw(20202021),
547565
discriminator: 3840,
548-
..Default::default()
549566
};
550567
let dev_det = BasicInfoConfig {
551568
vid: 65521,
552569
pid: 32769,
570+
serial_no: "1234567890".to_string(),
553571
..Default::default()
554572
};
555573

556574
let disc_cap = DiscoveryCapabilities::new(true, false, false);
557-
let mut qr_code_data = QrSetupPayload::new(&dev_det, &comm_data, disc_cap);
558-
qr_code_data
559-
.add_serial_number(SerialNumber::String("1234567890".to_string()))
560-
.expect("Failed to add serial number");
561-
575+
let qr_code_data = QrSetupPayload::new(&dev_det, &comm_data, disc_cap);
562576
let data_str = payload_base38_representation(&qr_code_data).expect("Failed to encode");
563577
assert_eq!(data_str, QR_CODE)
564578
}
@@ -574,21 +588,18 @@ mod tests {
574588
const OPTIONAL_DEFAULT_INT_VALUE: i32 = 65550;
575589

576590
let comm_data = CommissioningData {
577-
passwd: 20202021,
591+
verifier: VerifierData::new_with_pw(20202021),
578592
discriminator: 3840,
579-
..Default::default()
580593
};
581594
let dev_det = BasicInfoConfig {
582595
vid: 65521,
583596
pid: 32769,
597+
serial_no: "1234567890".to_string(),
584598
..Default::default()
585599
};
586600

587601
let disc_cap = DiscoveryCapabilities::new(true, false, false);
588602
let mut qr_code_data = QrSetupPayload::new(&dev_det, &comm_data, disc_cap);
589-
qr_code_data
590-
.add_serial_number(SerialNumber::String("1234567890".to_string()))
591-
.expect("Failed to add serial number");
592603

593604
qr_code_data
594605
.add_optional_vendor_data(

0 commit comments

Comments
 (0)