[Rust] src for task 3 & unit tests for task 1

Author: TrifanBogdan24

rust-method/Cargo.lock

@@ -1,10 +1,12 @@
 [package]
-name = "rust-method"
+name = "ac-sql"
 version = "0.1.0"
 edition = "2024"
 
 [[bin]]
 name = "ac-sql"
 path = "src/task2.rs"
 
+
+
 [dependencies]

rust-method/Cargo.toml

@@ -1,8 +1,10 @@
 use crate::task1::calculeaza_medii_generale;
-use crate::Secretariat;
-use crate::Student;
-use crate::Materie;
-use crate::Inrolare;
+use crate::structuri::Secretariat;
+use crate::structuri::Student;
+use crate::structuri::Materie;
+use crate::structuri::Inrolare;
+
+
 
 use crate::queries::where_clause::*;
 

rust-method/src/queries/delete.rs

@@ -1,7 +1,7 @@
-use crate::Secretariat;
-use crate::Student;
-use crate::Materie;
-use crate::Inrolare;
+use crate::structuri::Secretariat;
+use crate::structuri::Student;
+use crate::structuri::Materie;
+use crate::structuri::Inrolare;
 
 use crate::queries::where_clause::*;
 

rust-method/src/queries/select.rs

@@ -1,8 +1,7 @@
-use crate::Secretariat;
-use crate::Student;
-use crate::Materie;
-use crate::Inrolare;
-
+use crate::structuri::Secretariat;
+use crate::structuri::Student;
+use crate::structuri::Materie;
+use crate::structuri::Inrolare;
 
 use crate::queries::where_clause::*;
 use crate::queries::where_clause::parseaza_conditiile_where;

rust-method/src/queries/update.rs

@@ -1,6 +1,6 @@
-use crate::Student;
-use crate::Materie;
-use crate::Inrolare;
+use crate::structuri::Student;
+use crate::structuri::Materie;
+use crate::structuri::Inrolare;
 
 use std::str::FromStr;
 

rust-method/src/queries/where_clause.rs

@@ -101,10 +101,10 @@ fn parseaza_intrare_inrolare(linie: &str) -> Inrolare {
 }
 
 
-pub fn adauga_student(s: &mut Secretariat, id: usize, nume: &String, an_studiu: u8, statut: char, medie_generala: f32) {
+pub fn adauga_student(s: &mut Secretariat, id: usize, nume: &str, an_studiu: u8, statut: char, medie_generala: f32) {
     let student = Student {
         id: id,
-        nume: nume.clone(),
+        nume: nume.to_string(),
         an_studiu: an_studiu,
         statut: statut,
         medie_generala: medie_generala

rust-method/src/task1.rs

@@ -1,28 +1,26 @@
-mod structuri;
-mod task1;
-mod task3;
+use crate::queries::select::select;
+use crate::queries::delete::delete;
+use crate::queries::update::update;
+use crate::task1::citeste_secretariat;
 
-mod queries {
+pub mod structuri;
+pub mod task1;
+pub mod task3;
+
+pub mod queries {
     pub mod where_clause;
     pub mod select;
     pub mod delete;
     pub mod update;
 }
 
-use crate::structuri::*;
-use crate::queries::select::select;
-use crate::queries::delete::delete;
-use crate::queries::update::update;
-
+pub mod tests {
+    pub mod test_task1;
+}
 
 use std::{env, process::exit};
-use crate::task1::citeste_secretariat;
 use std::io;
 
-
-
-
-
 fn main() {
     let args: Vec<String> = env::args().collect();
 

rust-method/src/task2.rs

@@ -1,6 +1,84 @@
 use crate::structuri::*;
 
 
-pub fn cripteaza_studenti(_secretariat: &mut Secretariat, _key: &str, _iv: &str, _cale_output: &str) {
-    todo!("");
+const NR_BLOCKS: usize = 4;
+
+fn xor(block: &[u8], key: &[u8]) -> Vec<u8> {
+    block
+        .iter()
+        .zip(key.iter().cycle()) // repeta cheia daca e mai scurta
+        .map(|(&b, &k)| b ^ k)
+        .collect()
+
+}
+
+fn p_box(block: &[u8]) -> Vec<u8> {
+    let block_len: usize = block.len();
+    let mut array: Vec<u8> = vec![0; block_len];
+
+
+    for i in 0..block.len() {
+        let j: usize =
+            (i * (block_len - 1) + 2) % block_len;
+        array[j] = block[i];
+    }
+
+    array
+}
+
+
+fn split_into_blocks(secretariat: &Secretariat) -> Vec<Vec<u8>> {
+    let mut bytes_studenti: Vec<u8> = Vec::new();
+
+    for student in secretariat.studenti.iter() {
+        // id
+        bytes_studenti.extend_from_slice(&student.id.to_be_bytes());
+
+        // an_studiu
+        bytes_studenti.extend_from_slice(&student.an_studiu.to_be_bytes());
+
+        // statut (char -> u32 -> bytes)
+        bytes_studenti.extend_from_slice(&(student.statut as u32).to_be_bytes());
+
+        // medie_generala
+        bytes_studenti.extend_from_slice(&student.medie_generala.to_be_bytes());
+
+        // nume (trebuie serializat separat ca string)
+        bytes_studenti.extend_from_slice(student.nume.as_bytes());
+    }
+
+    // Adaugare padding (daca este cazul):
+    while bytes_studenti.len() % 4 != 0 {
+        bytes_studenti.push(0 as u8);
+    }
+
+
+    // Impartire in blocuri de aceasi dimensiune
+    let mut blocks: Vec<Vec<u8>> = Vec::new();
+    let block_length: usize = bytes_studenti.len() / NR_BLOCKS;
+    for i in 0..NR_BLOCKS {
+        let start = i * block_length;
+        let end = start + block_length;
+        blocks.push(bytes_studenti[start..end].to_vec());
+    }
+
+    blocks
+}
+
+
+pub fn cripteaza_studenti(secretariat: &Secretariat, key: &str, iv: &str, _cale_output: &str) {
+    let mut blocks: Vec<Vec<u8>> = split_into_blocks(secretariat);
+
+    let bytes_key: Vec<u8> = key.as_bytes().to_vec();
+    let bytes_iv: Vec<u8> = iv.as_bytes().to_vec();
+
+    blocks[0] = xor(&blocks[0], &bytes_iv);
+    blocks[0] = xor(&blocks[0], &bytes_key);
+    blocks[0] = p_box(&blocks[0]);
+
+    for i in 1..NR_BLOCKS {
+        blocks[i] = xor(&blocks[i], &blocks[i - 1]);
+        blocks[i] = xor(&blocks[i], &bytes_key);
+        blocks[i] = p_box(&blocks[i]);
+    }
 }

rust-method/src/task3.rs

@@ -0,0 +1,57 @@
+#![cfg(test)]
+
+use crate::structuri::Secretariat;
+use crate::task1::*;
+
+const TEST_DB: &str = "../tests/db/small.db";
+const EPSILONE: f32 = 1e-3;
+
+#[test]
+fn test_citire_secretariat() {
+    let s: Secretariat = citeste_secretariat(TEST_DB);
+
+    assert!(s.studenti.len() == 10, "Numar stdenti gresit!");
+    assert!(s.materii.len() == 9, "Numar materii gresit!");
+    assert!(s.inrolari.len() == 25, "Numar inrolari gresit!");
+
+    assert!(
+        s.studenti[4].nume == "Popescu Adrian".to_string()
+        && s.studenti[4].id == 4
+        && s.studenti[4].an_studiu == 3
+        && s.studenti[4].statut == 'b'
+        && (s.studenti[4].medie_generala - 7.74).abs() < EPSILONE,
+        "Studentii au fost salvati gresit!"
+    );
+    
+    assert!(
+        s.materii[3].id == 3
+        && s.materii[3].nume == "Fizica".to_string()
+        && s.materii[3].nume_titular == "Petrescu Cristina".to_string(),
+        "Materiile au fost salvate gresit!"
+    );
+
+    assert!(
+        s.inrolari[5].id_student == 2 && s.inrolari[5].id_materie == 0
+        && (s.inrolari[5].note[0] - 2.22).abs() < EPSILONE
+        && (s.inrolari[5].note[1] - 1.50).abs() < EPSILONE
+        && (s.inrolari[5].note[2] - 3.08).abs() < EPSILONE,
+        "Inrolarile au fost salvate gresit!"
+    );
+}
+
+
+#[test]
+fn test_adauga_student() {
+    let mut s: Secretariat = citeste_secretariat(TEST_DB);
+    
+    adauga_student(&mut s, 11, "Popescu Adrian", 3, 'b', 9.99 as f32);
+
+    assert!(
+        s.studenti[10].nume == "Popescu Adrian".to_string()
+        && s.studenti[10].id == 11
+        && s.studenti[10].an_studiu == 3
+        && s.studenti[10].statut == 'b',
+        "Studentul nu a fost adaguat corect!"
+    )
+
+}