移植

Author: LIN-Matrix

docs/asserts/0313/socket.rs

@@ -0,0 +1,92 @@
+// socket.rs
+
+use crate::net::{SocketSetWrapper, TcpSocket, UdpSocket, dns_query};
+use axnet::smoltcp::socket::{self, AnySocket};
+use axnet::smoltcp::iface::SocketHandle;
+use axnet::smoltcp::wire::{IpAddress, IpCidr};
+
+// A wrapper struct to represent a socket in our application
+pub struct Socket {
+    handle: SocketHandle,
+    socket_type: SocketType,
+}
+
+pub enum SocketType {
+    Tcp(TcpSocket),
+    Udp(UdpSocket),
+    Dns(socket::dns::Socket),
+}
+
+impl Socket {
+    pub fn new_tcp(socket_set: &SocketSetWrapper) -> Self {
+        let tcp_socket = socket_set.new_tcp_socket();
+        let handle = socket_set.add(tcp_socket);
+        Socket {
+            handle,
+            socket_type: SocketType::Tcp(tcp_socket),
+        }
+    }
+
+    pub fn new_udp(socket_set: &SocketSetWrapper) -> Self {
+        let udp_socket = socket_set.new_udp_socket();
+        let handle = socket_set.add(udp_socket);
+        Socket {
+            handle,
+            socket_type: SocketType::Udp(udp_socket),
+        }
+    }
+
+    pub fn new_dns(socket_set: &SocketSetWrapper) -> Self {
+        let dns_socket = socket_set.new_dns_socket();
+        let handle = socket_set.add(dns_socket);
+        Socket {
+            handle,
+            socket_type: SocketType::Dns(dns_socket),
+        }
+    }
+
+    pub fn send(&self, data: &[u8]) -> Result<(), String> {
+        match &self.socket_type {
+            SocketType::Tcp(socket) => {
+                // Here you could implement logic for sending data over a TCP socket
+                unimplemented!("TCP send not implemented")
+            }
+            SocketType::Udp(socket) => {
+                // Here you could implement logic for sending data over a UDP socket
+                unimplemented!("UDP send not implemented")
+            }
+            SocketType::Dns(socket) => {
+                // For DNS socket, implement logic to query DNS
+                dns_query(socket, data).map_err(|e| format!("DNS query failed: {}", e))
+            }
+        }
+    }
+
+    pub fn receive(&self) -> Result<Vec<u8>, String> {
+        match &self.socket_type {
+            SocketType::Tcp(socket) => {
+                // Here you could implement logic for receiving data from a TCP socket
+                unimplemented!("TCP receive not implemented")
+            }
+            SocketType::Udp(socket) => {
+                // Here you could implement logic for receiving data from a UDP socket
+                unimplemented!("UDP receive not implemented")
+            }
+            SocketType::Dns(socket) => {
+                // For DNS socket, implement logic to receive DNS responses
+                unimplemented!("DNS receive not implemented")
+            }
+        }
+    }
+
+    pub fn close(self) {
+        // Implement logic to clean up and close the socket
+        unimplemented!("Socket close not implemented")
+    }
+}
+
+// This function initializes the socket set wrapper and adds TCP/UDP/DNS sockets to it
+pub fn create_socket_set() -> SocketSetWrapper<'static> {
+    let socket_set = SocketSetWrapper::new();
+    socket_set
+}

docs/docs/day/0313.md

@@ -3,4 +3,6 @@
 1. 编译运行asterinas、DragonOs，比较二者的net部分的实现和starry-old的不同
 2. 编写syscall_imp下面的net部分，如socket.rs
 
+[filename](../../asserts/0313/socket.rs ':include :type=code')
+
 ![](../../asserts/0313/1.jpg ':class=myImageClass')