Add feature flag to set max payload size

Author: KennethKnudsen97

mqttrust_core/Cargo.toml

@@ -43,8 +43,17 @@ dns-lookup = "1.0.3"
 env_logger = "0.9.0"
 
 [features]
-default = []
+default = ["max_payload_size_4096"]
+max_payload_size_2048 = []
+max_payload_size_4096 = []
+max_payload_size_8192 = []
+
 
 std = []
 
-defmt-impl = ["defmt", "mqttrust/defmt-impl", "heapless/defmt-impl", "fugit/defmt"]
+defmt-impl = [
+    "defmt",
+    "mqttrust/defmt-impl",
+    "heapless/defmt-impl",
+    "fugit/defmt",
+]

mqttrust_core/src/eventloop.rs

@@ -1,3 +1,4 @@
+use crate::max_payload::MAX_PAYLOAD_SIZE;
 use crate::options::Broker;
 use crate::packet::SerializedPacket;
 use crate::state::{MqttConnectionStatus, MqttState};
@@ -417,7 +418,7 @@ impl<S> NetworkHandle<S> {
 #[derive(Debug)]
 struct PacketBuffer {
     range: RangeTo<usize>,
-    buffer: Vec<u8, 4096>,
+    buffer: Vec<u8, { MAX_PAYLOAD_SIZE }>,
 }
 
 impl PacketBuffer {

mqttrust_core/src/lib.rs

@@ -8,6 +8,7 @@ pub(crate) mod fmt;
 
 mod client;
 mod eventloop;
+mod max_payload;
 mod options;
 mod packet;
 mod state;
@@ -18,6 +19,7 @@ pub use client::Client;
 use core::convert::TryFrom;
 pub use eventloop::EventLoop;
 use heapless::{String, Vec};
+use max_payload::MAX_PAYLOAD_SIZE;
 pub use mqttrust::encoding::v4::{Pid, Publish, QoS, QosPid, Suback};
 pub use mqttrust::*;
 pub use options::{Broker, MqttOptions};
@@ -30,7 +32,7 @@ pub struct PublishNotification {
     pub qospid: QoS,
     pub retain: bool,
     pub topic_name: String<256>,
-    pub payload: Vec<u8, 4096>,
+    pub payload: Vec<u8, MAX_PAYLOAD_SIZE>,
 }
 
 /// Includes incoming packets from the network and other interesting events

mqttrust_core/src/max_payload.rs

@@ -0,0 +1,15 @@
+#[cfg(not(any(
+    feature = "max_payload_size_2048",
+    feature = "max_payload_size_4096",
+    feature = "max_payload_size_8192"
+)))]
+pub const MAX_PAYLOAD_SIZE: usize = 4096;
+
+#[cfg(feature = "max_payload_size_2048")]
+pub const MAX_PAYLOAD_SIZE: usize = 2048;
+
+#[cfg(feature = "max_payload_size_4096")]
+pub const MAX_PAYLOAD_SIZE: usize = 4096;
+
+#[cfg(feature = "max_payload_size_8192")]
+pub const MAX_PAYLOAD_SIZE: usize = 8192;