NASL builtin function for open privileged sockets with udp and tcp

doc/manual/nasl/built-in-functions/network-functions/open_priv_sock_tcp.md

@@ -6,11 +6,13 @@
 
 ## SYNOPSIS
 
-*any* **open_priv_sock_tcp**(dport: *int*, sport: *int*);
+*any* **open_priv_sock_tcp**(dport: *int*, sport: *int*, timeout: *int*);
 
-**open_priv_sock_tcp** takes two named integer arguments:
+**open_priv_sock_tcp** takes three named integer arguments:
 - dport is the destination port
-- sport is the source port, which may be inferior to 1024.
+- sport is the source port, which may be inferior to 1024. This argument is optional.
+  If it is not set, the function will try to open a socket on any port from 1 to 1023.
+- timeout: An integer with the timeout value in seconds.  The default timeout is controlled by a global value.
 
 ## DESCRIPTION
 

rust/Cargo.toml

@@ -6,14 +6,14 @@ license = "GPL-2.0-or-later"
 
 [dependencies]
 aes = "0.8.2"
-aes-gcm = { version = "0.10.1"}
+aes-gcm = { version = "0.10.1" }
 anyhow = "1.0.75"
 async-trait = "0.1.68"
 base64 = "0.21.2"
-cbc = { version = "0.1.2", features = ["alloc"]}
+cbc = { version = "0.1.2", features = ["alloc"] }
 ccm = "0.5.0"
 chacha20 = "0.9.1"
-chrono = { version = "0.4.23", default-features = false, features = ["clock"]}
+chrono = { version = "0.4.23", default-features = false, features = ["clock"] }
 clap = { version = "4.3.0", features = ["derive", "env"] }
 cmac = "0.7.2"
 configparser = "3"
@@ -57,11 +57,14 @@ rustls = "0.23.5"
 rustls-pemfile = "2.1.2"
 rustls-pemfile-old = { version = "1.0.2", package = "rustls-pemfile" }
 sequoia-ipc = "0.30.1"
-sequoia-openpgp = { version ="1.16.1", default-features = false, features = ["crypto-openssl"] }
-serde = { version = "1.0", features = ["derive"]}
+sequoia-openpgp = { version = "1.16.1", default-features = false, features = [
+  "crypto-openssl",
+] }
+serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0.96"
 sha1 = "0.10.5"
 sha2 = "0.10.7"
+socket2 = "0.5.7"
 sysinfo = "0.30.5"
 thiserror = "1.0.62"
 time = { version = "0", features = ["parsing"] }
@@ -78,11 +81,13 @@ rayon = { version = "1.8.0", optional = true }
 pcap = { version = "1.0.0", optional = true }
 pnet_base = { version = "0.33.0", optional = true }
 pnet = { version = "0.33.0", optional = true }
-socket2 = {version = "0.5.2", features = ["all"], optional = true}
 pnet_macros = { version = "0.33.0", optional = true }
 pnet_macros_support = { version = "0.33.0", optional = true }
 
-libssh-rs = {version = "~0.2", features = ["vendored-openssl", "vendored"], optional = true}
+libssh-rs = { version = "~0.2", features = [
+  "vendored-openssl",
+  "vendored",
+], optional = true }
 
 nasl-function-proc-macro = { path = "crates/nasl-function-proc-macro" }
 nasl-c-lib = { path = "crates/nasl-c-lib", optional = true }
@@ -92,10 +97,7 @@ rc4 = "0.1.0"
 
 [workspace]
 resolver = "2"
-members = [
-  "crates/smoketest",
-  "crates/nasl-function-proc-macro",
-]
+members = ["crates/smoketest", "crates/nasl-function-proc-macro"]
 
 [dev-dependencies]
 tracing-test = "0.2.5"
@@ -105,9 +107,20 @@ criterion = "0"
 dep-graph-parallel = ["rayon", "crossbeam-channel"]
 openvas_serde_support = []
 serde_support = []
-default = ["dep-graph-parallel", "openvas_serde_support", "enforce-no-trailing-arguments", "serde_support"]
+default = [
+  "dep-graph-parallel",
+  "openvas_serde_support",
+  "enforce-no-trailing-arguments",
+  "serde_support",
+]
 
-nasl-builtin-raw-ip = ["pcap", "pnet_base", "pnet", "socket2", "pnet_macros", "pnet_macros_support",]
+nasl-builtin-raw-ip = [
+  "pcap",
+  "pnet_base",
+  "pnet",
+  "pnet_macros",
+  "pnet_macros_support",
+]
 nasl-builtin-ssh = ["libssh-rs"]
 experimental = ["nasl-builtin-raw-ip", "nasl-builtin-ssh", "nasl-c-lib"]
 

rust/examples/socket/tcp_priv.nasl

@@ -0,0 +1,17 @@
+# 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_priv_sock_tcp"));
+sock = open_priv_sock_tcp(dport: 34254);
+display("was socket created: ", !isnull(sock));
+display("fd: ", sock);
+ret = send(socket: sock, data: 'foobar');
+display("num bytes sent: ", ret);
+rec = recv(socket: sock, length: 10, min: 3);
+display("received: ", rec);
+port = get_source_port(sock);
+display("source port: ", port);
+close(sock);
+display("end");

rust/examples/socket/udp_priv.nasl

@@ -0,0 +1,15 @@
+# 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_priv_sock_udp"));
+sock = open_priv_sock_udp(dport: 34254);
+display("was socket created: ", !isnull(sock));
+display("fd: ", sock);
+ret = send(socket: sock, data: '123');
+display("num bytes sent: ", ret);
+rec = recv(socket: sock, length: 10);
+display(rec);
+close(sock);
+display("end");

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

@@ -1,7 +1,9 @@
 ## Implements
 - open_sock_kdc
 - open_sock_tcp
+- open_priv_sock_tcp
 - open_sock_udp
+- open_priv_sock_udp
 - close
 - send
 - recv
@@ -26,8 +28,6 @@
 - get_udp_port_state
 - join_multicast_group
 - leave_multicast_group
-- open_priv_sock_tcp
-- open_priv_sock_udp
 - scanner_get_port
 - start_denial
 - end_denial

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

@@ -27,7 +27,9 @@ fn this_host(context: &Context) -> Result<String, FunctionErrorKind> {
 
     let port: u16 = DEFAULT_PORT;
 
-    get_source_ip(dst, port).map(|ip| ip.to_string())
+    get_source_ip(dst, port)
+        .map(|ip| ip.to_string())
+        .map_err(|_| FunctionErrorKind::Diagnostic("No route to destination".to_string(), None))
 }
 
 /// Get the host name of the current (attacking) machine

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

@@ -40,16 +40,12 @@ pub fn bind_local_socket(dst: &SocketAddr) -> io::Result<UdpSocket> {
 }
 
 /// Return the source IP address given the destination IP address
-pub fn get_source_ip(dst: IpAddr, port: u16) -> Result<IpAddr, FunctionErrorKind> {
+pub fn get_source_ip(dst: IpAddr, port: u16) -> io::Result<IpAddr> {
     let socket = SocketAddr::new(dst, port);
     let sd = format!("{}:{}", dst, port);
     let local_socket = bind_local_socket(&socket)?;
-    local_socket
-        .connect(sd)
-        .ok()
-        .and_then(|_| local_socket.local_addr().ok())
-        .and_then(|l_addr| IpAddr::from_str(&l_addr.ip().to_string()).ok())
-        .ok_or_else(|| FunctionErrorKind::Diagnostic("No route to destination".to_string(), None))
+    local_socket.connect(sd)?;
+    Ok(local_socket.local_addr()?.ip())
 }
 
 /// Tests whether a packet sent to IP is LIKELY to route through the