Rust 网络编程

在本章中，我们将探讨如何使用 Rust 语言进行网络编程。Rust 是一种系统级编程语言，它提供了内存安全和并发性方面的强大支持，使其成为网络编程的理想选择。我们将介绍一些基本概念，并通过实际示例来演示如何构建网络应用程序。

TCP 客户端与服务器

TCP (传输控制协议) 是一种面向连接的、可靠的、基于字节流的传输层通信协议。在这一部分，我们将学习如何使用 Rust 创建 TCP 客户端和服务器。

创建 TCP 服务器

首先，我们来看一下如何创建一个简单的 TCP 服务器。这里我们将使用 std::net 模块中的 TcpListener 类型。

// www.javascriptcn.com code example
use std::io::{self, Write};
use std::net::TcpListener;

fn main() -> io::Result<()> {
    // 创建一个 TcpListener，绑定到本地的 7878 端口
    let listener = TcpListener::bind("127.0.0.1:7878")?;

    println!("服务器正在监听 127.0.0.1:7878...");

    for stream in listener.incoming() {
        let stream = stream?;
        
        handle_connection(stream);
    }

    Ok(())
}

fn handle_connection(mut stream: impl std::io::Read + std::io::Write) {
    let mut buffer = [0; 512];
    
    stream.read(&mut buffer).unwrap();
    
    println!("接收到请求：{}", String::from_utf8_lossy(&buffer[..]));
    
    let response = "HTTP/1.1 200 OK\r\n\r\nHello, world!";
    
    stream.write(response.as_bytes()).unwrap();
    stream.flush().unwrap();
}

上述代码创建了一个简单的 TCP 服务器，它监听 7878 端口上的所有连接，并为每个传入的连接调用 handle_connection 函数。该函数读取客户端发送的数据，然后发送一条简单的 HTTP 响应消息。

创建 TCP 客户端

接下来，我们来实现一个 TCP 客户端，该客户端向我们的服务器发送请求并接收响应。

// www.javascriptcn.com code example
use std::io::{self, Read, Write};
use std::net::TcpStream;

fn main() -> io::Result<()> {
    let mut stream = TcpStream::connect("127.0.0.1:7878")?;

    println!("已连接到服务器");

    let request = b"GET / HTTP/1.1\r\nHost: 127.0.0.1\r\n\r\n";
    
    stream.write(request)?;

    let mut response = vec![];
    stream.read_to_end(&mut response)?;
    
    println!("收到响应：{}", String::from_utf8_lossy(&response));

    Ok(())
}

在这个例子中，客户端尝试连接到 127.0.0.1 上的 7878 端口，发送一个简单的 HTTP GET 请求，并打印出服务器返回的响应。

UDP 编程

UDP (用户数据报协议) 是一种无连接的协议，它提供了一种轻量级的数据交换方式。在这一部分，我们将探讨如何使用 Rust 实现 UDP 客户端和服务器。

UDP 服务器

Rust 的标准库提供了 UdpSocket 类型，可以用来处理 UDP 数据报。下面是一个简单的 UDP 服务器的例子：

// www.javascriptcn.com code example
use std::io;
use std::net::UdpSocket;

fn main() -> io::Result<()> {
    let socket = UdpSocket::bind("127.0.0.1:7878")?;
    
    println!("UDP 服务器正在监听 127.0.0.1:7878...");

    let mut buf = [0; 10];

    loop {
        let (amt, src) = socket.recv_from(&mut buf)?;
        
        println!("从 {} 接收到 {} 字节：{:?}", src, amt, &buf[..amt]);
        
        socket.send_to(&buf[..amt], src)?;
    }
}

UDP 客户端

下面是 UDP 客户端的一个简单例子，它将向服务器发送数据，并打印出服务器的响应：

// www.javascriptcn.com code example
use std::io;
use std::net::UdpSocket;

fn main() -> io::Result<()> {
    let socket = UdpSocket::bind("127.0.0.1:34254")?;
    
    socket.connect("127.0.0.1:7878")?;
    
    let message = b"Hello from UDP client!";
    
    socket.send(message)?;
    
    let mut buf = [0; 10];
    
    let amt = socket.recv(&mut buf)?;
    
    println!("收到响应：{:?}", &buf[..amt]);

    Ok(())
}

以上是 Rust 中进行网络编程的基本示例。通过这些示例，我们可以看到 Rust 提供了简洁而强大的工具来处理网络通信。随着进一步的学习，你可以探索更多高级功能，如异步 I/O 和更复杂的错误处理策略。