Refactor sockets

greenbone · Oct 30, 2024 · feea120 · feea120
1 parent 1ce91d8
commit feea120
Show file tree

Hide file tree

Showing 4 changed files with 205 additions and 182 deletions.
diff --git a/rust/examples/tcp.nasl b/rust/examples/tcp.nasl
@@ -11,4 +11,5 @@ ret = send(socket: sock, data: 'foobar');
 display("num bytes sent: ", ret);
 rec = recv(socket: sock, length: 10, min: 3);
 display(rec);
-display("end");
+close(sock);
+display("end");
diff --git a/rust/examples/udp.nasl b/rust/examples/udp.nasl
@@ -11,4 +11,5 @@ ret = send(socket: sock, data: '123');
 display("num bytes sent: ", ret);
 rec = recv(socket: sock, length: 10);
 display(rec);
+close(sock);
 display("end");
diff --git a/rust/src/nasl/builtin/network/mod.rs b/rust/src/nasl/builtin/network/mod.rs
@@ -75,6 +75,24 @@ impl Display for OpenvasEncaps {
     }
 }
 
+pub fn get_retry(context: &Context) -> u8 {
+    if let Ok(Some(val)) = get_kb_item(context, "timeout_retry") {
+        match val {
+            NaslValue::String(val) => val.parse::<u8>().unwrap_or(2),
+            NaslValue::Number(val) => {
+                if val < 1 || val > 255 {
+                    2
+                } else {
+                    val as u8
+                }
+            }
+            _ => 2,
+        }
+    } else {
+        2
+    }
+}
+
 pub fn get_kb_item(context: &Context, name: &str) -> Result<Option<NaslValue>, FunctionErrorKind> {
     context
         .retriever()