Add size restrictions

zns-cli/src/main.rs

@@ -239,7 +239,6 @@ impl KeyTransformer for OpenSSHKey {
         }?;
 
         let length = reader.read_u32()?;
-        println!("{}", length);
         reader.read(length as usize)?;
 
         Ok(Self {

zns-daemon/src/db/models.rs

@@ -1,3 +1,5 @@
+use std::fmt::format;
+
 use diesel::prelude::*;
 use diesel::sql_types::Text;
 use zns::{
@@ -90,7 +92,18 @@ impl Record {
     }
 }
 
+const MAX_RDATA_SIZE: usize = 1000;
+
 pub fn insert_into_database(rr: &RR, connection: &mut PgConnection) -> Result<(), ZNSError> {
+    if rr.rdata.len() > MAX_RDATA_SIZE {
+        return Err(ZNSError::Refused {
+            message: format!(
+                "RDATA size of record is bigger then maximum limit: {}",
+                MAX_RDATA_SIZE
+            ),
+        });
+    }
+
     let record = Record {
         name: rr.name.join("."),
         _type: rr._type.clone().into(),

zns/src/parser.rs

@@ -131,6 +131,13 @@ impl FromBytes for LabelString {
                 })?,
             );
             code = reader.read_u8()?;
+
+            // Set maximum number of labels.
+            if out.len() > 255 {
+                return Err(ZNSError::Refused {
+                    message: String::from("Exceeded maximum number of labels"),
+                });
+            }
         }
 
         if code & 0b11000000 != 0 {