Add subscription to multiple topics

This commit is contained in:
Ondrej Babec 2022-03-04 15:32:11 +01:00
parent 57f0304d35
commit 3c00a7b81d
No known key found for this signature in database
GPG Key ID: 13E577E3769B2079
28 changed files with 399 additions and 274 deletions

View File

@ -22,7 +22,6 @@
* SOFTWARE. * SOFTWARE.
*/ */
use crate::packet::publish_packet::QualityOfService; use crate::packet::publish_packet::QualityOfService;
use crate::utils::types::{BinaryData, EncodedString}; use crate::utils::types::{BinaryData, EncodedString};
@ -31,7 +30,7 @@ pub struct ClientConfig<'a> {
pub username_flag: bool, pub username_flag: bool,
pub username: EncodedString<'a>, pub username: EncodedString<'a>,
pub password_flag: bool, pub password_flag: bool,
pub password: BinaryData<'a> pub password: BinaryData<'a>,
} }
impl ClientConfig<'a> { impl ClientConfig<'a> {
@ -45,11 +44,11 @@ impl ClientConfig<'a> {
} }
} }
pub fn add_qos(& mut self, qos: QualityOfService) { pub fn add_qos(&mut self, qos: QualityOfService) {
self.qos = qos; self.qos = qos;
} }
pub fn add_username(& mut self, username: &'a str) { pub fn add_username(&mut self, username: &'a str) {
let mut username_s: EncodedString = EncodedString::new(); let mut username_s: EncodedString = EncodedString::new();
username_s.string = username; username_s.string = username;
username_s.len = username.len() as u16; username_s.len = username.len() as u16;
@ -57,11 +56,11 @@ impl ClientConfig<'a> {
self.username = username_s; self.username = username_s;
} }
pub fn add_password(& mut self, password: &'a str) { pub fn add_password(&mut self, password: &'a str) {
let mut password_s: BinaryData = BinaryData::new(); let mut password_s: BinaryData = BinaryData::new();
password_s.bin = password.as_bytes(); password_s.bin = password.as_bytes();
password_s.len = password_s.bin.len() as u16; password_s.len = password_s.bin.len() as u16;
self.password = password_s; self.password = password_s;
self.password_flag = true; self.password_flag = true;
} }
} }

View File

@ -1,19 +1,22 @@
use rand_core::RngCore;
use crate::client::client_config::ClientConfig; use crate::client::client_config::ClientConfig;
use crate::network::network_trait::{Network}; use crate::network::network_trait::Network;
use crate::packet::connack_packet::ConnackPacket; use crate::packet::connack_packet::ConnackPacket;
use crate::packet::connect_packet::ConnectPacket; use crate::packet::connect_packet::ConnectPacket;
use crate::packet::disconnect_packet::DisconnectPacket; use crate::packet::disconnect_packet::DisconnectPacket;
use crate::packet::mqtt_packet::Packet; use crate::packet::mqtt_packet::Packet;
use crate::packet::pingreq_packet::PingreqPacket;
use crate::packet::pingresp_packet::PingrespPacket;
use crate::packet::puback_packet::PubackPacket; use crate::packet::puback_packet::PubackPacket;
use crate::packet::publish_packet::{PublishPacket, QualityOfService};
use crate::packet::publish_packet::QualityOfService::QoS1; use crate::packet::publish_packet::QualityOfService::QoS1;
use crate::packet::publish_packet::{PublishPacket, QualityOfService};
use crate::packet::reason_codes::ReasonCode; use crate::packet::reason_codes::ReasonCode;
use crate::packet::suback_packet::SubackPacket; use crate::packet::suback_packet::SubackPacket;
use crate::packet::subscription_packet::SubscriptionPacket; use crate::packet::subscription_packet::SubscriptionPacket;
use crate::utils::buffer_reader::BuffReader; use crate::utils::buffer_reader::BuffReader;
use crate::utils::rng_generator::CountingRng; use crate::utils::rng_generator::CountingRng;
use crate::utils::types::BufferError; use crate::utils::types::BufferError;
use heapless::Vec;
use rand_core::RngCore;
pub struct MqttClientV5<'a, T, const MAX_PROPERTIES: usize> { pub struct MqttClientV5<'a, T, const MAX_PROPERTIES: usize> {
network_driver: &'a mut T, network_driver: &'a mut T,
@ -29,7 +32,14 @@ impl<'a, T, const MAX_PROPERTIES: usize> MqttClientV5<'a, T, MAX_PROPERTIES>
where where
T: Network, T: Network,
{ {
pub fn new(network_driver: &'a mut T, buffer: &'a mut [u8], buffer_len: usize, recv_buffer: &'a mut [u8], recv_buffer_len: usize, config: ClientConfig<'a>) -> Self { pub fn new(
network_driver: &'a mut T,
buffer: &'a mut [u8],
buffer_len: usize,
recv_buffer: &'a mut [u8],
recv_buffer_len: usize,
config: ClientConfig<'a>,
) -> Self {
Self { Self {
network_driver, network_driver,
buffer, buffer,
@ -37,7 +47,7 @@ where
recv_buffer, recv_buffer,
recv_buffer_len, recv_buffer_len,
rng: CountingRng(50), rng: CountingRng(50),
config config,
} }
} }
@ -45,10 +55,10 @@ where
let len = { let len = {
let mut connect = ConnectPacket::<'b, 3, 0>::clean(); let mut connect = ConnectPacket::<'b, 3, 0>::clean();
if self.config.username_flag { if self.config.username_flag {
connect.add_username(& self.config.username); connect.add_username(&self.config.username);
} }
if self.config.password_flag { if self.config.password_flag {
connect.add_password(& self.config.password) connect.add_password(&self.config.password)
} }
connect.encode(self.buffer, self.buffer_len) connect.encode(self.buffer, self.buffer_len)
}; };
@ -57,7 +67,7 @@ where
log::error!("[DECODE ERR]: {}", err); log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
self.network_driver.send(self.buffer, len.unwrap()).await ?; self.network_driver.send(self.buffer, len.unwrap()).await?;
//connack //connack
let reason: Result<u8, BufferError> = { let reason: Result<u8, BufferError> = {
@ -80,7 +90,6 @@ where
} else { } else {
Ok(()) Ok(())
} }
} }
pub async fn disconnect<'b>(&'b mut self) -> Result<(), ReasonCode> { pub async fn disconnect<'b>(&'b mut self) -> Result<(), ReasonCode> {
@ -90,7 +99,7 @@ where
log::error!("[DECODE ERR]: {}", err); log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
self.network_driver.send(self.buffer, len.unwrap()).await ?; self.network_driver.send(self.buffer, len.unwrap()).await?;
Ok(()) Ok(())
} }
@ -99,7 +108,6 @@ where
topic_name: &'b str, topic_name: &'b str,
message: &'b str, message: &'b str,
) -> Result<(), ReasonCode> { ) -> Result<(), ReasonCode> {
let identifier: u16 = self.rng.next_u32() as u16; let identifier: u16 = self.rng.next_u32() as u16;
let len = { let len = {
let mut packet = PublishPacket::<'b, 5>::new(); let mut packet = PublishPacket::<'b, 5>::new();
@ -115,15 +123,17 @@ where
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
self.network_driver.send(self.buffer, len.unwrap()).await ?; self.network_driver.send(self.buffer, len.unwrap()).await?;
//QoS1 //QoS1
if <QualityOfService as Into<u8>>::into(self.config.qos ) == <QualityOfService as Into<u8>>::into(QoS1) { if <QualityOfService as Into<u8>>::into(self.config.qos)
== <QualityOfService as Into<u8>>::into(QoS1)
{
let reason: Result<[u16; 2], BufferError> = { let reason: Result<[u16; 2], BufferError> = {
self.network_driver.receive(self.buffer).await ?; self.network_driver.receive(self.buffer).await?;
let mut packet = PubackPacket::<'b, 5>::new(); let mut packet = PubackPacket::<'b, 5>::new();
if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len)) { if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len))
{
Err(err) Err(err)
} else { } else {
Ok([packet.packet_identifier, packet.reason_code as u16]) Ok([packet.packet_identifier, packet.reason_code as u16])
@ -147,9 +157,63 @@ where
Ok(()) Ok(())
} }
// TODO - multiple topic subscribe func pub async fn subscribe_to_topics<'b, const TOPICS: usize>(
&'b mut self,
topic_names: &'b Vec<&'b str, TOPICS>,
) -> Result<(), ReasonCode> {
let len = {
let mut subs = SubscriptionPacket::<'b, TOPICS, 1>::new();
let mut i = 0;
loop {
if i == TOPICS {
break;
}
subs.add_new_filter(topic_names.get(i).unwrap(), self.config.qos);
i = i + 1;
}
subs.encode(self.buffer, self.buffer_len)
};
pub async fn subscribe_to_topic<'b>(&'b mut self, topic_name: &'b str) -> Result<(), ReasonCode> { if let Err(err) = len {
log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError);
}
self.network_driver.send(self.buffer, len.unwrap()).await?;
let reason: Result<Vec<u8, TOPICS>, BufferError> = {
self.network_driver.receive(self.buffer).await?;
let mut packet = SubackPacket::<'b, TOPICS, 5>::new();
if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len)) {
Err(err)
} else {
Ok(packet.reason_codes)
}
};
if let Err(err) = reason {
log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError);
}
let reasons = reason.unwrap();
let mut i = 0;
loop {
if i == TOPICS {
break;
}
if *reasons.get(i).unwrap() != self.config.qos.into() {
return Err(ReasonCode::from(*reasons.get(i).unwrap()));
}
i = i + 1;
}
Ok(())
}
pub async fn subscribe_to_topic<'b>(
&'b mut self,
topic_name: &'b str,
) -> Result<(), ReasonCode> {
let len = { let len = {
let mut subs = SubscriptionPacket::<'b, 1, 1>::new(); let mut subs = SubscriptionPacket::<'b, 1, 1>::new();
subs.add_new_filter(topic_name, self.config.qos); subs.add_new_filter(topic_name, self.config.qos);
@ -161,10 +225,10 @@ where
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
self.network_driver.send(self.buffer, len.unwrap()).await ?; self.network_driver.send(self.buffer, len.unwrap()).await?;
let reason: Result<u8, BufferError> = { let reason: Result<u8, BufferError> = {
self.network_driver.receive(self.buffer).await ?; self.network_driver.receive(self.buffer).await?;
let mut packet = SubackPacket::<'b, 5, 5>::new(); let mut packet = SubackPacket::<'b, 5, 5>::new();
if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len)) { if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len)) {
@ -185,18 +249,21 @@ where
} else { } else {
Ok(()) Ok(())
} }
} }
pub async fn receive_message<'b>(&'b mut self) -> Result<&'b [u8], ReasonCode> { pub async fn receive_message<'b>(&'b mut self) -> Result<&'b [u8], ReasonCode> {
self.network_driver.receive(self.recv_buffer).await ?; self.network_driver.receive(self.recv_buffer).await?;
let mut packet = PublishPacket::<'b, 5>::new(); let mut packet = PublishPacket::<'b, 5>::new();
if let Err(err) = packet.decode(&mut BuffReader::new(self.recv_buffer, self.recv_buffer_len)) { if let Err(err) =
packet.decode(&mut BuffReader::new(self.recv_buffer, self.recv_buffer_len))
{
log::error!("[DECODE ERR]: {}", err); log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
if (packet.fixed_header & 0x06) == <QualityOfService as Into<u8>>::into(QualityOfService::QoS1) { if (packet.fixed_header & 0x06)
== <QualityOfService as Into<u8>>::into(QualityOfService::QoS1)
{
let mut puback = PubackPacket::<'b, 5>::new(); let mut puback = PubackPacket::<'b, 5>::new();
puback.packet_identifier = packet.packet_identifier; puback.packet_identifier = packet.packet_identifier;
puback.reason_code = 0x00; puback.reason_code = 0x00;
@ -206,10 +273,33 @@ where
log::error!("[DECODE ERR]: {}", err); log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError); return Err(ReasonCode::BuffError);
} }
self.network_driver.send(self.buffer, len.unwrap()).await ?; self.network_driver.send(self.buffer, len.unwrap()).await?;
} }
} }
return Ok(packet.message.unwrap()); return Ok(packet.message.unwrap());
} }
pub async fn send_ping<'b>(&'b mut self) -> Result<(), ReasonCode> {
let len = {
let mut packet = PingreqPacket::new();
packet.encode(self.buffer, self.buffer_len)
};
if let Err(err) = len {
log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError);
}
self.network_driver.send(self.buffer, len.unwrap()).await?;
self.network_driver.receive(self.buffer).await?;
let mut packet = PingrespPacket::new();
if let Err(err) = packet.decode(&mut BuffReader::new(self.buffer, self.buffer_len)) {
log::error!("[DECODE ERR]: {}", err);
return Err(ReasonCode::BuffError);
} else {
Ok(())
}
}
} }

View File

@ -22,6 +22,5 @@
* SOFTWARE. * SOFTWARE.
*/ */
pub mod client_v5;
pub mod client_config; pub mod client_config;
pub mod client_v5;

View File

@ -22,5 +22,4 @@
* SOFTWARE. * SOFTWARE.
*/ */
pub mod variable_byte_integer; pub mod variable_byte_integer;

View File

@ -1,15 +1,17 @@
use std::time::Duration; use std::time::Duration;
use tokio::{join, task}; use heapless::Vec;
use tokio::time::sleep; use tokio::time::sleep;
use tokio::{join, task};
use rust_mqtt::client::client_config::ClientConfig; use rust_mqtt::client::client_config::ClientConfig;
use rust_mqtt::client::client_v5::MqttClientV5; use rust_mqtt::client::client_v5::MqttClientV5;
use rust_mqtt::network::network_trait::{Network, NetworkError}; use rust_mqtt::network::network_trait::{Network, NetworkError};
use rust_mqtt::packet::connect_packet::ConnectPacket; use rust_mqtt::packet::connect_packet::ConnectPacket;
use rust_mqtt::packet::mqtt_packet::Packet; use rust_mqtt::packet::mqtt_packet::Packet;
use rust_mqtt::packet::publish_packet::{PublishPacket, QualityOfService};
use rust_mqtt::packet::publish_packet::QualityOfService::QoS1; use rust_mqtt::packet::publish_packet::QualityOfService::QoS1;
use rust_mqtt::packet::publish_packet::{PublishPacket, QualityOfService};
use rust_mqtt::packet::subscription_packet::SubscriptionPacket; use rust_mqtt::packet::subscription_packet::SubscriptionPacket;
use rust_mqtt::tokio_network::TokioNetwork; use rust_mqtt::tokio_network::TokioNetwork;
@ -21,27 +23,49 @@ async fn receive() {
let mut config = ClientConfig::new(); let mut config = ClientConfig::new();
config.add_qos(QualityOfService::QoS1); config.add_qos(QualityOfService::QoS1);
config.add_username("test"); config.add_username("test");
config.add_password("testPass1"); config.add_password("testPass");
let mut res2 = vec![0; 260]; let mut res2 = vec![0; 260];
let mut res3 = vec![0; 260]; let mut res3 = vec![0; 260];
let mut client = MqttClientV5::<TokioNetwork, 5>::new(&mut tokio_network, &mut res2, 260, & mut res3, 260, config); let mut client = MqttClientV5::<TokioNetwork, 5>::new(
&mut tokio_network,
&mut res2,
260,
&mut res3,
260,
config,
);
let mut result = { let mut result = { client.connect_to_broker().await };
client.connect_to_broker().await
};
if let Err(r) = result { if let Err(r) = result {
log::error!("[ERROR]: {}", r); log::error!("[ERROR]: {}", r);
return; return;
} }
{ {
client.subscribe_to_topic("test/topic").await; const TOPICS: usize = 2;
let t1 = "test/topic1";
let t2 = "test/topic2";
let mut names = Vec::<&str, TOPICS>::new();
names.push(&t1);
names.push(&t2);
client.subscribe_to_topics::<TOPICS>(&names).await;
//client.subscribe_to_topic("test/topic").await;
}; };
{ {
sleep(Duration::from_secs(10));
client.send_ping().await;
}
let mut o = 0;
loop {
if o == 2 {
break;
}
log::info!("Waiting for new message!"); log::info!("Waiting for new message!");
let mes = client.receive_message().await.unwrap(); let mes = client.receive_message().await.unwrap();
let x = String::from_utf8_lossy(mes); let x = String::from_utf8_lossy(mes);
log::info!("Got new message: {}", x); log::info!("Got new message: {}", x);
o = o + 1;
} }
{ {
client.disconnect().await; client.disconnect().await;
@ -56,22 +80,30 @@ async fn publish(message: &str) {
let config = ClientConfig::new(); let config = ClientConfig::new();
let mut res2 = vec![0; 260]; let mut res2 = vec![0; 260];
let mut res3 = vec![0; 260]; let mut res3 = vec![0; 260];
let mut client = MqttClientV5::<TokioNetwork, 5>::new(&mut tokio_network, &mut res2, 260, & mut res3, 260, config); let mut client = MqttClientV5::<TokioNetwork, 5>::new(
&mut tokio_network,
&mut res2,
260,
&mut res3,
260,
config,
);
let mut result = { client.connect_to_broker().await }; let mut result = { client.connect_to_broker().await };
log::info!("Waiting until send!"); log::info!("Waiting until send!");
sleep(Duration::from_secs(15)); sleep(Duration::from_secs(15));
result= { result = {
log::info!("Sending new message!"); log::info!("Sending new message!");
client /*client
.send_message("test/topic", message) .send_message("test/topic", message)
.await .await*/
client.send_ping().await
}; };
if let Err(e) = result { if let Err(e) = result {
log::error!("Chyba!"); log::error!("Chyba!");
} }
result = { /*result = {
log::info!("Sending new message!"); log::info!("Sending new message!");
client client
.send_message("test/topic", "Dalsi zprava :)") .send_message("test/topic", "Dalsi zprava :)")
@ -79,7 +111,7 @@ async fn publish(message: &str) {
}; };
if let Err(err) = result { if let Err(err) = result {
log::error!("Chyba!"); log::error!("Chyba!");
} }*/
{ {
client.disconnect().await; client.disconnect().await;
@ -103,6 +135,7 @@ async fn main() {
join!(recv, publ);*/ join!(recv, publ);*/
receive().await; receive().await;
//publish("Ahoj 123").await; //publish("Ahoj 123").await;
log::info!("Done"); log::info!("Done");
} }

View File

@ -22,5 +22,4 @@
* SOFTWARE. * SOFTWARE.
*/ */
pub mod network_trait; pub mod network_trait;

View File

@ -22,7 +22,6 @@
* SOFTWARE. * SOFTWARE.
*/ */
use heapless::Vec; use heapless::Vec;
use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder; use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder;
@ -69,7 +68,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for AuthPacket<'a, MAX_PROPERTI
remain_len: 0, remain_len: 0,
auth_reason: 0x00, auth_reason: 0x00,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new() properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
} }
} }
@ -77,23 +76,22 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for AuthPacket<'a, MAX_PROPERTI
let mut buff_writer = BuffWriter::new(buffer, buff_len); let mut buff_writer = BuffWriter::new(buffer, buff_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32; rm_ln = rm_ln + property_len_len as u32;
rm_ln = rm_ln + 1; rm_ln = rm_ln + 1;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u8(self.auth_reason) ?; buff_writer.write_u8(self.auth_reason)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
self.decode_fixed_header(buff_reader) ?; self.decode_fixed_header(buff_reader)?;
self.auth_reason = buff_reader.read_u8() ?; self.auth_reason = buff_reader.read_u8()?;
return self.decode_properties(buff_reader); return self.decode_properties(buff_reader);
} }

View File

@ -43,13 +43,16 @@ pub struct ConnackPacket<'a, const MAX_PROPERTIES: usize> {
} }
impl<'a, const MAX_PROPERTIES: usize> ConnackPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> ConnackPacket<'a, MAX_PROPERTIES> {
pub fn decode_connack_packet(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { pub fn decode_connack_packet(
if self.decode_fixed_header(buff_reader) ? != (PacketType::Connack).into() { &mut self,
buff_reader: &mut BuffReader<'a>,
) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader)? != (PacketType::Connack).into() {
log::error!("Packet you are trying to decode is not CONNACK packet!"); log::error!("Packet you are trying to decode is not CONNACK packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.ack_flags = buff_reader.read_u8() ?; self.ack_flags = buff_reader.read_u8()?;
self.connect_reason_code = buff_reader.read_u8() ?; self.connect_reason_code = buff_reader.read_u8()?;
self.decode_properties(buff_reader) self.decode_properties(buff_reader)
} }
} }
@ -68,16 +71,16 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for ConnackPacket<'a, MAX_PROPE
fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> { fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> {
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
let property_len_enc = VariableByteIntegerEncoder::encode(self.property_len) ?; let property_len_enc = VariableByteIntegerEncoder::encode(self.property_len)?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
let rm_len: u32 = 2 + self.property_len + property_len_len as u32; let rm_len: u32 = 2 + self.property_len + property_len_len as u32;
buff_writer.write_variable_byte_int(rm_len) ?; buff_writer.write_variable_byte_int(rm_len)?;
buff_writer.write_u8(self.ack_flags) ?; buff_writer.write_u8(self.ack_flags)?;
buff_writer.write_u8(self.connect_reason_code) ?; buff_writer.write_u8(self.connect_reason_code)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties(&self.properties) ?; buff_writer.encode_properties(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }

View File

@ -128,15 +128,13 @@ impl<'a, const MAX_PROPERTIES: usize, const MAX_WILL_PROPERTIES: usize> Packet<'
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
// Number 12 => protocol_name_len + protocol_name + protocol_version + connect_flags + keep_alive + client_id_len // Number 12 => protocol_name_len + protocol_name + protocol_version + connect_flags + keep_alive + client_id_len
rm_ln = rm_ln + property_len_len as u32 + 12; rm_ln = rm_ln + property_len_len as u32 + 12;
if self.connect_flags & 0x04 != 0 { if self.connect_flags & 0x04 != 0 {
let wil_prop_len_enc = let wil_prop_len_enc = VariableByteIntegerEncoder::encode(self.will_property_len)?;
VariableByteIntegerEncoder::encode(self.will_property_len) ?;
let wil_prop_len_len = VariableByteIntegerEncoder::len(wil_prop_len_enc); let wil_prop_len_len = VariableByteIntegerEncoder::len(wil_prop_len_enc);
rm_ln = rm_ln rm_ln = rm_ln
+ wil_prop_len_len as u32 + wil_prop_len_len as u32
@ -152,31 +150,31 @@ impl<'a, const MAX_PROPERTIES: usize, const MAX_WILL_PROPERTIES: usize> Packet<'
rm_ln = rm_ln + self.password.len as u32 + 2; rm_ln = rm_ln + self.password.len as u32 + 2;
} }
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.protocol_name_len) ?; buff_writer.write_u16(self.protocol_name_len)?;
buff_writer.write_u32(self.protocol_name) ?; buff_writer.write_u32(self.protocol_name)?;
buff_writer.write_u8(self.protocol_version) ?; buff_writer.write_u8(self.protocol_version)?;
buff_writer.write_u8(self.connect_flags) ?; buff_writer.write_u8(self.connect_flags)?;
buff_writer.write_u16(self.keep_alive) ?; buff_writer.write_u16(self.keep_alive)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
buff_writer.write_string_ref(&self.client_id) ?; buff_writer.write_string_ref(&self.client_id)?;
if self.connect_flags & 0x04 != 0 { if self.connect_flags & 0x04 != 0 {
buff_writer.write_variable_byte_int(self.will_property_len) ?; buff_writer.write_variable_byte_int(self.will_property_len)?;
buff_writer.encode_properties(&self.will_properties) ?; buff_writer.encode_properties(&self.will_properties)?;
buff_writer.write_string_ref(&self.will_topic) ?; buff_writer.write_string_ref(&self.will_topic)?;
buff_writer.write_binary_ref(&self.will_payload) ?; buff_writer.write_binary_ref(&self.will_payload)?;
} }
if self.connect_flags & 0x80 != 0 { if self.connect_flags & 0x80 != 0 {
buff_writer.write_string_ref(&self.username) ?; buff_writer.write_string_ref(&self.username)?;
} }
if self.connect_flags & 0x40 != 0 { if self.connect_flags & 0x40 != 0 {
buff_writer.write_binary_ref(&self.password) ?; buff_writer.write_binary_ref(&self.password)?;
} }
Ok(buff_writer.position) Ok(buff_writer.position)

View File

@ -47,7 +47,6 @@ pub struct DisconnectPacket<'a, const MAX_PROPERTIES: usize> {
} }
impl<'a, const MAX_PROPERTIES: usize> DisconnectPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> DisconnectPacket<'a, MAX_PROPERTIES> {
fn add_reason(&mut self, reason: u8) { fn add_reason(&mut self, reason: u8) {
self.disconnect_reason = reason; self.disconnect_reason = reason;
} }
@ -66,24 +65,24 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for DisconnectPacket<'a, MAX_PR
fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> { fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> {
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
let property_len_enc = VariableByteIntegerEncoder::encode(self.property_len) ?; let property_len_enc = VariableByteIntegerEncoder::encode(self.property_len)?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
let rm_len: u32 = 1 + self.property_len + property_len_len as u32; let rm_len: u32 = 1 + self.property_len + property_len_len as u32;
buff_writer.write_variable_byte_int(rm_len) ?; buff_writer.write_variable_byte_int(rm_len)?;
buff_writer.write_u8(self.disconnect_reason) ?; buff_writer.write_u8(self.disconnect_reason)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties(&self.properties) ?; buff_writer.encode_properties(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Pingresp).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Pingresp).into() {
log::error!("Packet you are trying to decode is not PUBACK packet!"); log::error!("Packet you are trying to decode is not PUBACK packet!");
return Err(BufferError::WrongPacketToDecode); return Err(BufferError::WrongPacketToDecode);
} }
self.disconnect_reason = buff_reader.read_u8() ?; self.disconnect_reason = buff_reader.read_u8()?;
return self.decode_properties(buff_reader); return self.decode_properties(buff_reader);
} }

View File

@ -39,6 +39,6 @@ pub mod connect_packet;
pub mod disconnect_packet; pub mod disconnect_packet;
pub mod pingreq_packet; pub mod pingreq_packet;
pub mod pingresp_packet; pub mod pingresp_packet;
pub mod reason_codes;
pub mod suback_packet; pub mod suback_packet;
pub mod unsuback_packet; pub mod unsuback_packet;
pub mod reason_codes;

View File

@ -57,7 +57,7 @@ pub trait Packet<'a> {
let mut prop: Property; let mut prop: Property;
if self.get_property_len() != 0 { if self.get_property_len() != 0 {
loop { loop {
prop = Property::decode(buff_reader) ?; prop = Property::decode(buff_reader)?;
log::debug!("Parsed property {:?}", prop); log::debug!("Parsed property {:?}", prop);
x = x + prop.len() as u32 + 1; x = x + prop.len() as u32 + 1;
self.push_to_properties(prop); self.push_to_properties(prop);
@ -71,10 +71,13 @@ pub trait Packet<'a> {
} }
/// Method is decoding packet header into fixed header part and remaining length /// Method is decoding packet header into fixed header part and remaining length
fn decode_fixed_header(&mut self, buff_reader: &mut BuffReader) -> Result<PacketType, BufferError> { fn decode_fixed_header(
&mut self,
buff_reader: &mut BuffReader,
) -> Result<PacketType, BufferError> {
let first_byte: u8 = buff_reader.read_u8()?; let first_byte: u8 = buff_reader.read_u8()?;
self.set_fixed_header(first_byte); self.set_fixed_header(first_byte);
self.set_remaining_len(buff_reader.read_variable_byte_int() ?); self.set_remaining_len(buff_reader.read_variable_byte_int()?);
return Ok(PacketType::from(first_byte)); return Ok(PacketType::from(first_byte));
} }
} }

View File

@ -41,14 +41,14 @@ impl<'a> Packet<'a> for PingreqPacket {
fn new() -> Self { fn new() -> Self {
Self { Self {
fixed_header: PacketType::Pingreq.into(), fixed_header: PacketType::Pingreq.into(),
remain_len: 0 remain_len: 0,
} }
} }
fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> { fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> {
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(0 as u32) ?; buff_writer.write_variable_byte_int(0 as u32)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }

View File

@ -35,26 +35,26 @@ pub struct PingrespPacket {
pub remain_len: u32, pub remain_len: u32,
} }
impl<'a> PingrespPacket { impl<'a> PingrespPacket {}
}
impl<'a> Packet<'a> for PingrespPacket { impl<'a> Packet<'a> for PingrespPacket {
fn new() -> Self { fn new() -> Self {
Self { Self {
fixed_header: PacketType::Pingresp.into(), fixed_header: PacketType::Pingresp.into(),
remain_len: 0 remain_len: 0,
} }
} }
fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> { fn encode(&mut self, buffer: &mut [u8], buffer_len: usize) -> Result<usize, BufferError> {
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(0 as u32) ?; buff_writer.write_variable_byte_int(0 as u32)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Pingresp).into() { let x = self.decode_fixed_header(buff_reader)?;
if x != (PacketType::Pingresp).into() {
log::error!("Packet you are trying to decode is not PINGRESP packet!"); log::error!("Packet you are trying to decode is not PINGRESP packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }

View File

@ -42,9 +42,7 @@ pub struct PubackPacket<'a, const MAX_PROPERTIES: usize> {
pub properties: Vec<Property<'a>, MAX_PROPERTIES>, pub properties: Vec<Property<'a>, MAX_PROPERTIES>,
} }
impl<'a, const MAX_PROPERTIES: usize> PubackPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> PubackPacket<'a, MAX_PROPERTIES> {}
}
impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubackPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubackPacket<'a, MAX_PROPERTIES> {
fn new() -> Self { fn new() -> Self {
@ -54,7 +52,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubackPacket<'a, MAX_PROPER
packet_identifier: 0, packet_identifier: 0,
reason_code: 0, reason_code: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new() properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
} }
} }
@ -62,33 +60,32 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubackPacket<'a, MAX_PROPER
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32 + 3; rm_ln = rm_ln + property_len_len as u32 + 3;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_u8(self.reason_code) ?; buff_writer.write_u8(self.reason_code)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Puback).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Puback).into() {
log::error!("Packet you are trying to decode is not PUBACK packet!"); log::error!("Packet you are trying to decode is not PUBACK packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
if self.remain_len != 2 { if self.remain_len != 2 {
self.reason_code = buff_reader.read_u8() ?; self.reason_code = buff_reader.read_u8()?;
} }
if self.remain_len < 4 { if self.remain_len < 4 {
self.property_len = 0; self.property_len = 0;
} else { } else {
self.decode_properties(buff_reader) ?; self.decode_properties(buff_reader)?;
} }
Ok(()) Ok(())
} }

View File

@ -42,8 +42,7 @@ pub struct PubcompPacket<'a, const MAX_PROPERTIES: usize> {
pub properties: Vec<Property<'a>, MAX_PROPERTIES>, pub properties: Vec<Property<'a>, MAX_PROPERTIES>,
} }
impl<'a, const MAX_PROPERTIES: usize> PubcompPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> PubcompPacket<'a, MAX_PROPERTIES> {}
}
impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubcompPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubcompPacket<'a, MAX_PROPERTIES> {
fn new() -> Self { fn new() -> Self {
@ -53,7 +52,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubcompPacket<'a, MAX_PROPE
packet_identifier: 0, packet_identifier: 0,
reason_code: 0, reason_code: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new() properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
} }
} }
@ -61,28 +60,27 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubcompPacket<'a, MAX_PROPE
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32 + 3; rm_ln = rm_ln + property_len_len as u32 + 3;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_u8(self.reason_code) ?; buff_writer.write_u8(self.reason_code)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Pubcomp).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Pubcomp).into() {
log::error!("Packet you are trying to decode is not PUBCOMP packet!"); log::error!("Packet you are trying to decode is not PUBCOMP packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
self.reason_code = buff_reader.read_u8() ?; self.reason_code = buff_reader.read_u8()?;
self.decode_properties(buff_reader) ?; self.decode_properties(buff_reader)?;
Ok(()) Ok(())
} }

View File

@ -26,7 +26,7 @@ use heapless::Vec;
use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder; use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder;
use crate::packet::mqtt_packet::Packet; use crate::packet::mqtt_packet::Packet;
use crate::packet::publish_packet::QualityOfService::{INVALID, QoS0, QoS1, QoS2}; use crate::packet::publish_packet::QualityOfService::{QoS0, QoS1, QoS2, INVALID};
use crate::utils::buffer_reader::BuffReader; use crate::utils::buffer_reader::BuffReader;
use crate::utils::buffer_writer::BuffWriter; use crate::utils::buffer_writer::BuffWriter;
use crate::utils::types::{BufferError, EncodedString}; use crate::utils::types::{BufferError, EncodedString};
@ -80,15 +80,15 @@ impl<'a, const MAX_PROPERTIES: usize> PublishPacket<'a, MAX_PROPERTIES> {
self.topic_name.len = topic_name.len() as u16; self.topic_name.len = topic_name.len() as u16;
} }
pub fn add_message(& mut self, message: &'a [u8]) { pub fn add_message(&mut self, message: &'a [u8]) {
self.message = Some(message); self.message = Some(message);
} }
pub fn add_qos(& mut self, qos: QualityOfService) { pub fn add_qos(&mut self, qos: QualityOfService) {
self.fixed_header = self.fixed_header | <QualityOfService as Into<u8>>::into(qos); self.fixed_header = self.fixed_header | <QualityOfService as Into<u8>>::into(qos);
} }
pub fn add_identifier(& mut self, identifier: u16) { pub fn add_identifier(&mut self, identifier: u16) {
self.packet_identifier = identifier; self.packet_identifier = identifier;
} }
} }
@ -102,7 +102,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PublishPacket<'a, MAX_PROPE
packet_identifier: 1, packet_identifier: 1,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(), properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
message: None message: None,
} }
} }
@ -110,45 +110,44 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PublishPacket<'a, MAX_PROPE
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
let msg_len = self.message.unwrap().len() as u32; let msg_len = self.message.unwrap().len() as u32;
rm_ln = rm_ln + property_len_len as u32 + msg_len + self.topic_name.len as u32 + 2; rm_ln = rm_ln + property_len_len as u32 + msg_len + self.topic_name.len as u32 + 2;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
let qos = self.fixed_header & 0x03; let qos = self.fixed_header & 0x03;
if qos != 0 { if qos != 0 {
rm_ln = rm_ln + 2; rm_ln = rm_ln + 2;
} }
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_string_ref(&self.topic_name) ?; buff_writer.write_string_ref(&self.topic_name)?;
if qos != 0 { if qos != 0 {
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
} }
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
buff_writer.insert_ref(msg_len as usize, self.message.unwrap()) ?; buff_writer.insert_ref(msg_len as usize, self.message.unwrap())?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Publish).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Publish).into() {
log::error!("Packet you are trying to decode is not PUBLISH packet!"); log::error!("Packet you are trying to decode is not PUBLISH packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.topic_name = buff_reader.read_string() ?; self.topic_name = buff_reader.read_string()?;
let qos = self.fixed_header & 0x03; let qos = self.fixed_header & 0x03;
if qos != 0 { if qos != 0 {
// Decode only for QoS 1 / 2 // Decode only for QoS 1 / 2
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
} }
self.decode_properties(buff_reader) ?; self.decode_properties(buff_reader)?;
let mut total_len = VariableByteIntegerEncoder::len( let mut total_len =
VariableByteIntegerEncoder::encode(self.remain_len) ?); VariableByteIntegerEncoder::len(VariableByteIntegerEncoder::encode(self.remain_len)?);
total_len = total_len + 1 + self.remain_len as usize; total_len = total_len + 1 + self.remain_len as usize;
self.message = Some(buff_reader.read_message(total_len)); self.message = Some(buff_reader.read_message(total_len));
Ok(()) Ok(())

View File

@ -42,8 +42,7 @@ pub struct PubrecPacket<'a, const MAX_PROPERTIES: usize> {
pub properties: Vec<Property<'a>, MAX_PROPERTIES>, pub properties: Vec<Property<'a>, MAX_PROPERTIES>,
} }
impl<'a, const MAX_PROPERTIES: usize> PubrecPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> PubrecPacket<'a, MAX_PROPERTIES> {}
}
impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrecPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrecPacket<'a, MAX_PROPERTIES> {
fn new() -> Self { fn new() -> Self {
@ -53,7 +52,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrecPacket<'a, MAX_PROPER
packet_identifier: 0, packet_identifier: 0,
reason_code: 0, reason_code: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new() properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
} }
} }
@ -61,27 +60,26 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrecPacket<'a, MAX_PROPER
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32 + 3; rm_ln = rm_ln + property_len_len as u32 + 3;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_u8(self.reason_code) ?; buff_writer.write_u8(self.reason_code)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Pubrec).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Pubrec).into() {
log::error!("Packet you are trying to decode is not PUBREC packet!"); log::error!("Packet you are trying to decode is not PUBREC packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
self.reason_code = buff_reader.read_u8() ?; self.reason_code = buff_reader.read_u8()?;
return self.decode_properties(buff_reader); return self.decode_properties(buff_reader);
} }

View File

@ -42,8 +42,7 @@ pub struct PubrelPacket<'a, const MAX_PROPERTIES: usize> {
pub properties: Vec<Property<'a>, MAX_PROPERTIES>, pub properties: Vec<Property<'a>, MAX_PROPERTIES>,
} }
impl<'a, const MAX_PROPERTIES: usize> PubrelPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> PubrelPacket<'a, MAX_PROPERTIES> {}
}
impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrelPacket<'a, MAX_PROPERTIES> { impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrelPacket<'a, MAX_PROPERTIES> {
fn new() -> Self { fn new() -> Self {
@ -53,7 +52,7 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrelPacket<'a, MAX_PROPER
packet_identifier: 0, packet_identifier: 0,
reason_code: 0, reason_code: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new() properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
} }
} }
@ -61,27 +60,26 @@ impl<'a, const MAX_PROPERTIES: usize> Packet<'a> for PubrelPacket<'a, MAX_PROPER
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32 + 3; rm_ln = rm_ln + property_len_len as u32 + 3;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_u8(self.reason_code) ?; buff_writer.write_u8(self.reason_code)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Pubrel).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Pubrel).into() {
log::error!("Packet you are trying to decode is not PUBREL packet!"); log::error!("Packet you are trying to decode is not PUBREL packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
self.reason_code = buff_reader.read_u8() ?; self.reason_code = buff_reader.read_u8()?;
return self.decode_properties(buff_reader); return self.decode_properties(buff_reader);
} }

View File

@ -96,9 +96,8 @@ impl Into<u8> for ReasonCode {
ReasonCode::SubscriptionIdentifiersNotSupported => 0xA1, ReasonCode::SubscriptionIdentifiersNotSupported => 0xA1,
ReasonCode::WildcardSubscriptionNotSupported => 0xA2, ReasonCode::WildcardSubscriptionNotSupported => 0xA2,
ReasonCode::BuffError => 0xFE, ReasonCode::BuffError => 0xFE,
ReasonCode::NetworkError => 0xFF ReasonCode::NetworkError => 0xFF,
} };
} }
} }
@ -148,8 +147,8 @@ impl From<u8> for ReasonCode {
0xA1 => ReasonCode::SubscriptionIdentifiersNotSupported, 0xA1 => ReasonCode::SubscriptionIdentifiersNotSupported,
0xA2 => ReasonCode::WildcardSubscriptionNotSupported, 0xA2 => ReasonCode::WildcardSubscriptionNotSupported,
0xFE => ReasonCode::BuffError, 0xFE => ReasonCode::BuffError,
_ => ReasonCode::NetworkError _ => ReasonCode::NetworkError,
} };
} }
} }
@ -170,7 +169,9 @@ impl Display for ReasonCode {
ReasonCode::ImplementationSpecificError => write!(f, "Implementation specific error!"), ReasonCode::ImplementationSpecificError => write!(f, "Implementation specific error!"),
ReasonCode::UnsupportedProtocolVersion => write!(f, "Unsupported protocol version!"), ReasonCode::UnsupportedProtocolVersion => write!(f, "Unsupported protocol version!"),
ReasonCode::ClientIdNotValid => write!(f, "Client sent not valid identification"), ReasonCode::ClientIdNotValid => write!(f, "Client sent not valid identification"),
ReasonCode::BadUserNameOrPassword => write!(f, "Authentication error, username of password not valid!"), ReasonCode::BadUserNameOrPassword => {
write!(f, "Authentication error, username of password not valid!")
}
ReasonCode::NotAuthorized => write!(f, "Client not authorized!"), ReasonCode::NotAuthorized => write!(f, "Client not authorized!"),
ReasonCode::ServerUnavailable => write!(f, "Server unavailable!"), ReasonCode::ServerUnavailable => write!(f, "Server unavailable!"),
ReasonCode::ServerBusy => write!(f, "Server is busy!"), ReasonCode::ServerBusy => write!(f, "Server is busy!"),
@ -194,13 +195,19 @@ impl Display for ReasonCode {
ReasonCode::QoSNotSupported => write!(f, "Used QoS is not supported!"), ReasonCode::QoSNotSupported => write!(f, "Used QoS is not supported!"),
ReasonCode::UseAnotherServer => write!(f, "Use another server!"), ReasonCode::UseAnotherServer => write!(f, "Use another server!"),
ReasonCode::ServerMoved => write!(f, "Server moved!"), ReasonCode::ServerMoved => write!(f, "Server moved!"),
ReasonCode::SharedSubscriptionNotSupported => write!(f, "Shared subscription is not supported"), ReasonCode::SharedSubscriptionNotSupported => {
write!(f, "Shared subscription is not supported")
}
ReasonCode::ConnectionRateExceeded => write!(f, "Connection rate exceeded!"), ReasonCode::ConnectionRateExceeded => write!(f, "Connection rate exceeded!"),
ReasonCode::MaximumConnectTime => write!(f, "Maximum connect time exceeded!"), ReasonCode::MaximumConnectTime => write!(f, "Maximum connect time exceeded!"),
ReasonCode::SubscriptionIdentifiersNotSupported => write!(f, "Subscription identifier not supported!"), ReasonCode::SubscriptionIdentifiersNotSupported => {
ReasonCode::WildcardSubscriptionNotSupported => write!(f, "Wildcard subscription not supported!"), write!(f, "Subscription identifier not supported!")
ReasonCode::BuffError => write!(f , "Error encountered during write / read from packet"), }
ReasonCode::WildcardSubscriptionNotSupported => {
write!(f, "Wildcard subscription not supported!")
}
ReasonCode::BuffError => write!(f, "Error encountered during write / read from packet"),
ReasonCode::NetworkError => write!(f, "Unknown error!"), ReasonCode::NetworkError => write!(f, "Unknown error!"),
} }
} }
} }

View File

@ -43,16 +43,19 @@ pub struct SubackPacket<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usiz
impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize> impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize>
SubackPacket<'a, MAX_REASONS, MAX_PROPERTIES> SubackPacket<'a, MAX_REASONS, MAX_PROPERTIES>
{ {
pub fn read_reason_codes(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { pub fn read_reason_codes(
&mut self,
buff_reader: &mut BuffReader<'a>,
) -> Result<(), BufferError> {
let mut i = 0; let mut i = 0;
loop { loop {
self.reason_codes.push(buff_reader.read_u8() ?); self.reason_codes.push(buff_reader.read_u8()?);
i = i + 1; i = i + 1;
if i == MAX_REASONS { if i == MAX_REASONS {
break; break;
} }
} }
return Ok(()) return Ok(());
} }
} }
@ -66,22 +69,22 @@ impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize> Packet<'a>
packet_identifier: 0, packet_identifier: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(), properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
reason_codes: Vec::<u8, MAX_REASONS>::new() reason_codes: Vec::<u8, MAX_REASONS>::new(),
} }
} }
fn encode(&mut self, _buffer: &mut [u8], _buffer_len: usize) -> Result<usize, BufferError> { fn encode(&mut self, _buffer: &mut [u8], _buffer_len: usize) -> Result<usize, BufferError> {
log::error!("SUBACK packet does not support encoding!"); log::error!("SUBACK packet does not support encoding!");
return Err(BufferError::WrongPacketToEncode) return Err(BufferError::WrongPacketToEncode);
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError>{ fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Suback).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Suback).into() {
log::error!("Packet you are trying to decode is not SUBACK packet!"); log::error!("Packet you are trying to decode is not SUBACK packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
self.decode_properties(buff_reader) ?; self.decode_properties(buff_reader)?;
return self.read_reason_codes(buff_reader); return self.read_reason_codes(buff_reader);
} }

View File

@ -24,14 +24,14 @@
use heapless::Vec; use heapless::Vec;
use super::packet_type::PacketType;
use super::property::Property;
use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder; use crate::encoding::variable_byte_integer::VariableByteIntegerEncoder;
use crate::packet::mqtt_packet::Packet; use crate::packet::mqtt_packet::Packet;
use crate::packet::publish_packet::QualityOfService; use crate::packet::publish_packet::QualityOfService;
use crate::utils::buffer_reader::BuffReader; use crate::utils::buffer_reader::BuffReader;
use crate::utils::buffer_writer::BuffWriter; use crate::utils::buffer_writer::BuffWriter;
use crate::utils::types::{BufferError, TopicFilter}; use crate::utils::types::{BufferError, TopicFilter};
use super::packet_type::PacketType;
use super::property::Property;
pub struct SubscriptionPacket<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> { pub struct SubscriptionPacket<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> {
pub fixed_header: u8, pub fixed_header: u8,
@ -46,12 +46,13 @@ pub struct SubscriptionPacket<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES
impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize>
SubscriptionPacket<'a, MAX_FILTERS, MAX_PROPERTIES> SubscriptionPacket<'a, MAX_FILTERS, MAX_PROPERTIES>
{ {
pub fn add_new_filter(& mut self, topic_name: &'a str, qos: QualityOfService) { pub fn add_new_filter(&mut self, topic_name: &'a str, qos: QualityOfService) {
let len = topic_name.len(); let len = topic_name.len();
let mut new_filter = TopicFilter::new(); let mut new_filter = TopicFilter::new();
new_filter.filter.string = topic_name; new_filter.filter.string = topic_name;
new_filter.filter.len = len as u16; new_filter.filter.len = len as u16;
new_filter.sub_options = new_filter.sub_options | (<QualityOfService as Into<u8>>::into(qos) >> 1); new_filter.sub_options =
new_filter.sub_options | (<QualityOfService as Into<u8>>::into(qos) >> 1);
self.topic_filters.push(new_filter); self.topic_filters.push(new_filter);
self.topic_filter_len = self.topic_filter_len + 1; self.topic_filter_len = self.topic_filter_len + 1;
} }
@ -60,7 +61,7 @@ impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize>
impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a> impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a>
for SubscriptionPacket<'a, MAX_FILTERS, MAX_PROPERTIES> for SubscriptionPacket<'a, MAX_FILTERS, MAX_PROPERTIES>
{ {
fn new() -> Self { fn new() -> Self {
let x = Self { let x = Self {
fixed_header: PacketType::Subscribe.into(), fixed_header: PacketType::Subscribe.into(),
remain_len: 0, remain_len: 0,
@ -77,8 +78,7 @@ impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a>
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
let mut lt = 0; let mut lt = 0;
@ -92,16 +92,16 @@ impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a>
} }
rm_ln = rm_ln + property_len_len as u32 + 2 + filters_len as u32; rm_ln = rm_ln + property_len_len as u32 + 2 + filters_len as u32;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
buff_writer.encode_topic_filters_ref( buff_writer.encode_topic_filters_ref(
true, true,
self.topic_filter_len as usize, self.topic_filter_len as usize,
&self.topic_filters, &self.topic_filters,
) ?; )?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }

View File

@ -43,10 +43,13 @@ pub struct UnsubackPacket<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: us
impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize> impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize>
UnsubackPacket<'a, MAX_REASONS, MAX_PROPERTIES> UnsubackPacket<'a, MAX_REASONS, MAX_PROPERTIES>
{ {
pub fn read_reason_codes(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { pub fn read_reason_codes(
&mut self,
buff_reader: &mut BuffReader<'a>,
) -> Result<(), BufferError> {
let mut i = 0; let mut i = 0;
loop { loop {
self.reason_codes.push(buff_reader.read_u8() ?); self.reason_codes.push(buff_reader.read_u8()?);
i = i + 1; i = i + 1;
if i == MAX_REASONS { if i == MAX_REASONS {
break; break;
@ -66,7 +69,7 @@ impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize> Packet<'a>
packet_identifier: 0, packet_identifier: 0,
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(), properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
reason_codes: Vec::<u8, MAX_REASONS>::new() reason_codes: Vec::<u8, MAX_REASONS>::new(),
} }
} }
@ -76,12 +79,12 @@ impl<'a, const MAX_REASONS: usize, const MAX_PROPERTIES: usize> Packet<'a>
} }
fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> { fn decode(&mut self, buff_reader: &mut BuffReader<'a>) -> Result<(), BufferError> {
if self.decode_fixed_header(buff_reader) ? != (PacketType::Unsuback).into() { if self.decode_fixed_header(buff_reader)? != (PacketType::Unsuback).into() {
log::error!("Packet you are trying to decode is not UNSUBACK packet!"); log::error!("Packet you are trying to decode is not UNSUBACK packet!");
return Err(BufferError::PacketTypeMismatch); return Err(BufferError::PacketTypeMismatch);
} }
self.packet_identifier = buff_reader.read_u16() ?; self.packet_identifier = buff_reader.read_u16()?;
self.decode_properties(buff_reader) ?; self.decode_properties(buff_reader)?;
return self.read_reason_codes(buff_reader); return self.read_reason_codes(buff_reader);
} }

View File

@ -58,7 +58,7 @@ impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a>
property_len: 0, property_len: 0,
properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(), properties: Vec::<Property<'a>, MAX_PROPERTIES>::new(),
topic_filter_len: 0, topic_filter_len: 0,
topic_filters: Vec::<TopicFilter<'a>, MAX_FILTERS>::new() topic_filters: Vec::<TopicFilter<'a>, MAX_FILTERS>::new(),
} }
} }
@ -66,22 +66,21 @@ impl<'a, const MAX_FILTERS: usize, const MAX_PROPERTIES: usize> Packet<'a>
let mut buff_writer = BuffWriter::new(buffer, buffer_len); let mut buff_writer = BuffWriter::new(buffer, buffer_len);
let mut rm_ln = self.property_len; let mut rm_ln = self.property_len;
let property_len_enc: [u8; 4] = let property_len_enc: [u8; 4] = VariableByteIntegerEncoder::encode(self.property_len)?;
VariableByteIntegerEncoder::encode(self.property_len) ?;
let property_len_len = VariableByteIntegerEncoder::len(property_len_enc); let property_len_len = VariableByteIntegerEncoder::len(property_len_enc);
rm_ln = rm_ln + property_len_len as u32 + 4 + self.topic_filter_len as u32; rm_ln = rm_ln + property_len_len as u32 + 4 + self.topic_filter_len as u32;
buff_writer.write_u8(self.fixed_header) ?; buff_writer.write_u8(self.fixed_header)?;
buff_writer.write_variable_byte_int(rm_ln) ?; buff_writer.write_variable_byte_int(rm_ln)?;
buff_writer.write_u16(self.packet_identifier) ?; buff_writer.write_u16(self.packet_identifier)?;
buff_writer.write_variable_byte_int(self.property_len) ?; buff_writer.write_variable_byte_int(self.property_len)?;
buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties) ?; buff_writer.encode_properties::<MAX_PROPERTIES>(&self.properties)?;
buff_writer.write_u16(self.topic_filter_len) ?; buff_writer.write_u16(self.topic_filter_len)?;
buff_writer.encode_topic_filters_ref( buff_writer.encode_topic_filters_ref(
false, false,
self.topic_filter_len as usize, self.topic_filter_len as usize,
&self.topic_filters, &self.topic_filters,
) ?; )?;
Ok(buff_writer.position) Ok(buff_writer.position)
} }

View File

@ -45,7 +45,7 @@ impl<'a> BuffReader<'a> {
return BuffReader { return BuffReader {
buffer, buffer,
position: 0, position: 0,
len: buff_len len: buff_len,
}; };
} }
@ -101,7 +101,7 @@ impl<'a> BuffReader<'a> {
let len = self.read_u16(); let len = self.read_u16();
match len { match len {
Err(err) => return Err(err), Err(err) => return Err(err),
_ => {}, _ => {}
} }
let len_res = len.unwrap(); let len_res = len.unwrap();
let res_str = let res_str =
@ -119,7 +119,7 @@ impl<'a> BuffReader<'a> {
/// Read Binary data from buffer /// Read Binary data from buffer
pub fn read_binary(&mut self) -> Result<BinaryData<'a>, BufferError> { pub fn read_binary(&mut self) -> Result<BinaryData<'a>, BufferError> {
let len = self.read_u16() ?; let len = self.read_u16()?;
let res_bin = &(self.buffer[self.position..(self.position + len as usize)]); let res_bin = &(self.buffer[self.position..(self.position + len as usize)]);
return Ok(BinaryData { return Ok(BinaryData {
bin: res_bin, bin: res_bin,
@ -147,7 +147,6 @@ impl<'a> BuffReader<'a> {
/// Read payload message from buffer /// Read payload message from buffer
pub fn read_message(&mut self, total_len: usize) -> &'a [u8] { pub fn read_message(&mut self, total_len: usize) -> &'a [u8] {
if total_len > self.len { if total_len > self.len {
return &self.buffer[self.position..self.len]; return &self.buffer[self.position..self.len];
} }

View File

@ -26,12 +26,12 @@ use heapless::Vec;
use crate::encoding::variable_byte_integer::{VariableByteInteger, VariableByteIntegerEncoder}; use crate::encoding::variable_byte_integer::{VariableByteInteger, VariableByteIntegerEncoder};
use crate::packet::property::Property; use crate::packet::property::Property;
use crate::utils::types::{BinaryData, EncodedString, StringPair, BufferError, TopicFilter}; use crate::utils::types::{BinaryData, BufferError, EncodedString, StringPair, TopicFilter};
pub struct BuffWriter<'a> { pub struct BuffWriter<'a> {
buffer: &'a mut [u8], buffer: &'a mut [u8],
pub position: usize, pub position: usize,
len: usize len: usize,
} }
impl<'a> BuffWriter<'a> { impl<'a> BuffWriter<'a> {
@ -39,7 +39,7 @@ impl<'a> BuffWriter<'a> {
return BuffWriter { return BuffWriter {
buffer, buffer,
position: 0, position: 0,
len: buff_len len: buff_len,
}; };
} }
@ -72,8 +72,7 @@ impl<'a> BuffWriter<'a> {
self.buffer[self.position] = byte; self.buffer[self.position] = byte;
self.increment_position(1); self.increment_position(1);
Ok(()) Ok(())
} };
} }
pub fn write_u16(&mut self, two_bytes: u16) -> Result<(), BufferError> { pub fn write_u16(&mut self, two_bytes: u16) -> Result<(), BufferError> {
@ -87,56 +86,59 @@ impl<'a> BuffWriter<'a> {
} }
pub fn write_string_ref(&mut self, str: &EncodedString<'a>) -> Result<(), BufferError> { pub fn write_string_ref(&mut self, str: &EncodedString<'a>) -> Result<(), BufferError> {
self.write_u16(str.len) ?; self.write_u16(str.len)?;
let bytes = str.string.as_bytes(); let bytes = str.string.as_bytes();
return self.insert_ref(str.len as usize, bytes); return self.insert_ref(str.len as usize, bytes);
} }
pub fn write_string(&mut self, str: EncodedString<'a>) -> Result<(), BufferError> { pub fn write_string(&mut self, str: EncodedString<'a>) -> Result<(), BufferError> {
self.write_u16(str.len) ?; self.write_u16(str.len)?;
let bytes = str.string.as_bytes(); let bytes = str.string.as_bytes();
return self.insert_ref(str.len as usize, bytes); return self.insert_ref(str.len as usize, bytes);
} }
pub fn write_binary_ref(&mut self, bin: &BinaryData<'a>) -> Result<(), BufferError> { pub fn write_binary_ref(&mut self, bin: &BinaryData<'a>) -> Result<(), BufferError> {
self.write_u16(bin.len) ?; self.write_u16(bin.len)?;
return self.insert_ref(bin.len as usize, bin.bin); return self.insert_ref(bin.len as usize, bin.bin);
} }
pub fn write_binary(&mut self, bin: BinaryData<'a>) -> Result<(), BufferError> { pub fn write_binary(&mut self, bin: BinaryData<'a>) -> Result<(), BufferError> {
self.write_u16(bin.len) ?; self.write_u16(bin.len)?;
return self.insert_ref(bin.len as usize, bin.bin); return self.insert_ref(bin.len as usize, bin.bin);
} }
pub fn write_string_pair_ref(&mut self, str_pair: &StringPair<'a>) -> Result<(), BufferError> { pub fn write_string_pair_ref(&mut self, str_pair: &StringPair<'a>) -> Result<(), BufferError> {
self.write_string_ref(&str_pair.name) ?; self.write_string_ref(&str_pair.name)?;
return self.write_string_ref(&str_pair.value); return self.write_string_ref(&str_pair.value);
} }
pub fn write_string_pair(&mut self, str_pair: StringPair<'a>) -> Result<(), BufferError> { pub fn write_string_pair(&mut self, str_pair: StringPair<'a>) -> Result<(), BufferError> {
self.write_string(str_pair.name) ?; self.write_string(str_pair.name)?;
return self.write_string(str_pair.value); return self.write_string(str_pair.value);
} }
pub fn write_variable_byte_int(&mut self, int: u32) -> Result<(), BufferError> { pub fn write_variable_byte_int(&mut self, int: u32) -> Result<(), BufferError> {
let x: VariableByteInteger = VariableByteIntegerEncoder::encode(int) ?; let x: VariableByteInteger = VariableByteIntegerEncoder::encode(int)?;
let len = VariableByteIntegerEncoder::len(x); let len = VariableByteIntegerEncoder::len(x);
return self.insert_ref(len, &x); return self.insert_ref(len, &x);
} }
pub fn encode_property(&mut self, property: &Property<'a>) -> Result<(), BufferError> { pub fn encode_property(&mut self, property: &Property<'a>) -> Result<(), BufferError> {
let x: u8 = property.into(); let x: u8 = property.into();
self.write_u8(x) ?; self.write_u8(x)?;
return property.encode(self); return property.encode(self);
} }
pub fn encode_properties<const LEN: usize>(&mut self, properties: &Vec<Property<'a>, LEN>) -> Result<(), BufferError> { pub fn encode_properties<const LEN: usize>(
&mut self,
properties: &Vec<Property<'a>, LEN>,
) -> Result<(), BufferError> {
let mut i = 0; let mut i = 0;
let len = properties.len(); let len = properties.len();
if len != 0 { if len != 0 {
loop { loop {
let prop: &Property = properties.get(i).unwrap_or(&Property::Reserved()); let prop: &Property = properties.get(i).unwrap_or(&Property::Reserved());
self.encode_property(prop) ?; self.encode_property(prop)?;
i = i + 1; i = i + 1;
if i == len { if i == len {
break; break;
@ -146,12 +148,16 @@ impl<'a> BuffWriter<'a> {
Ok(()) Ok(())
} }
fn encode_topic_filter_ref(&mut self, sub: bool, topic_filter: &TopicFilter<'a>) -> Result<(), BufferError> { fn encode_topic_filter_ref(
self.write_string_ref(&topic_filter.filter) ?; &mut self,
sub: bool,
topic_filter: &TopicFilter<'a>,
) -> Result<(), BufferError> {
self.write_string_ref(&topic_filter.filter)?;
if sub { if sub {
self.write_u8(topic_filter.sub_options) ?; self.write_u8(topic_filter.sub_options)?;
} }
return Ok(()) return Ok(());
} }
pub fn encode_topic_filters_ref<const MAX: usize>( pub fn encode_topic_filters_ref<const MAX: usize>(
@ -163,7 +169,7 @@ impl<'a> BuffWriter<'a> {
let mut i = 0; let mut i = 0;
loop { loop {
let topic_filter: &TopicFilter<'a> = filters.get(i).unwrap(); let topic_filter: &TopicFilter<'a> = filters.get(i).unwrap();
self.encode_topic_filter_ref(sub, topic_filter) ?; self.encode_topic_filter_ref(sub, topic_filter)?;
i = i + 1; i = i + 1;
if i == len { if i == len {
break; break;

View File

@ -1,7 +1,7 @@
// This code is handed from Embedded Rust documentation and // This code is handed from Embedded Rust documentation and
// is accessible from https://docs.rust-embedded.org/cortex-m-rt/0.6.0/rand/trait.RngCore.html // is accessible from https://docs.rust-embedded.org/cortex-m-rt/0.6.0/rand/trait.RngCore.html
use rand_core::{Error, impls, RngCore}; use rand_core::{impls, Error, RngCore};
pub struct CountingRng(pub u64); pub struct CountingRng(pub u64);
@ -25,4 +25,4 @@ impl RngCore for CountingRng {
fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), Error> { fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), Error> {
Ok(self.fill_bytes(dest)) Ok(self.fill_bytes(dest))
} }
} }

View File

@ -35,7 +35,7 @@ pub enum BufferError {
PacketTypeMismatch, PacketTypeMismatch,
WrongPacketToDecode, WrongPacketToDecode,
WrongPacketToEncode, WrongPacketToEncode,
PropertyNotFound PropertyNotFound,
} }
impl Display for BufferError { impl Display for BufferError {
@ -55,8 +55,7 @@ impl Display for BufferError {
} }
} }
/// Encoded string provides structure representing UTF-8 encoded string in MQTTv5 packets /// Encoded string provides structure representing UTF-8 encoded string in MQTTv5 packets
#[derive(Debug)] #[derive(Debug, Clone)]
#[derive(Clone)]
pub struct EncodedString<'a> { pub struct EncodedString<'a> {
pub string: &'a str, pub string: &'a str,
pub len: u16, pub len: u16,
@ -73,8 +72,7 @@ impl EncodedString<'_> {
} }
/// Binary data represents `Binary data` in MQTTv5 protocol /// Binary data represents `Binary data` in MQTTv5 protocol
#[derive(Debug)] #[derive(Debug, Clone)]
#[derive(Clone)]
pub struct BinaryData<'a> { pub struct BinaryData<'a> {
pub bin: &'a [u8], pub bin: &'a [u8],
pub len: u16, pub len: u16,
@ -123,4 +121,4 @@ impl TopicFilter<'_> {
pub fn len(&self) -> u16 { pub fn len(&self) -> u16 {
return self.filter.len + 3; return self.filter.len + 3;
} }
} }