Add: Nasl builtin function

rust/examples/tcp.nasl → rust/examples/socket/tcp.nasl

@@ -10,6 +10,8 @@ display("fd: ", sock);
 ret = send(socket: sock, data: 'foobar');
 display("num bytes sent: ", ret);
 rec = recv(socket: sock, length: 10, min: 3);
-display(rec);
+display("received: ", rec);
+port = get_source_port(sock);
+display("source port: ", port);
 close(sock);
-display("end");
+display("end");

rust/examples/socket/tcp_recv_line.nasl

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: 2024 Greenbone AG
+#
+# SPDX-License-Identifier: GPL-2.0-or-later WITH x11vnc-openssl-exception
+
+display("Start");
+display("is function defined: ", defined_func("open_sock_udp"));
+sock = open_sock_tcp(34254, transport: 1);
+if (isnull(sock)) {
+  display("Failed to open socket");
+  exit(0);
+}
+display("fd: ", sock);
+ret = send(socket: sock, data: '123');
+if (ret < 0) {
+  display("Failed to send data");
+  exit(0);
+}
+display("num bytes sent: ", ret);
+rec = recv_line(socket: sock, length: 10, timeout: 1);
+display("line1: ", rec);
+rec = recv_line(socket: sock, length: 10, timeout: 1);
+display("line2: ", rec);
+rec = recv_line(socket: sock, length: 10, timeout: 1);
+display("line3: ", rec);
+display("end");

rust/src/nasl/builtin/network/README.md

@@ -13,16 +13,15 @@
 - get_host_ip
 - scanner_add_port
 - recv_line
+- get_source_port
 
 ## Missing
 
-- end_denial
 - ftp_get_pasv_port
 - ftp_log_in
 - get_host_open_port
 - get_port_state
 - get_port_transport
-- get_source_port
 - get_tcp_port_state
 - get_udp_port_state
 - join_multicast_group
@@ -31,4 +30,5 @@
 - open_priv_sock_udp
 - scanner_get_port
 - start_denial
+- end_denial
 - telnet_init

rust/src/nasl/builtin/network/socket.rs

@@ -887,6 +887,28 @@ impl NaslSockets {
 
         Ok(NaslValue::Number(fd as i64))
     }
+
+    /// Get the source port of a open socket
+    #[nasl_function]
+    fn get_source_port(&self, socket: usize) -> Result<NaslValue, FunctionErrorKind> {
+        let handles = self.handles.read().unwrap();
+        let socket = handles
+            .handles
+            .get(socket)
+            .ok_or(FunctionErrorKind::WrongArgument(
+                "the given socket FD does not exist".to_string(),
+            ))?;
+        let port = match socket {
+            NaslSocket::Tcp(conn) => conn.socket.local_addr()?.port(),
+            NaslSocket::Udp(conn) => conn.socket.local_addr()?.port(),
+            NaslSocket::Close => {
+                return Err(FunctionErrorKind::WrongArgument(
+                    "the given socket FD is already closed".to_string(),
+                ))
+            }
+        };
+        Ok(NaslValue::Number(port as i64))
+    }
 }
 
 function_set! {
@@ -900,5 +922,6 @@ function_set! {
         (NaslSockets::send, "send"),
         (NaslSockets::recv, "recv"),
         (NaslSockets::recv_line, "recv_line"),
+        (NaslSockets::get_source_port, "get_source_port"),
     )
 }