当前位置：首页 > news >正文

Rust 中的 Pin 和 Unpin：内存安全与异步编程的守护者

news 来源：原创 2025/5/10 7:14:28

在 Rust 的世界里，Pin 和 Unpin 是两个看似不起眼、实则至关重要的概念。它们在内存安全和异步编程中扮演着关键角色，是 Rust 开发者必须掌握的知识。今天，就让我们深入探讨这两个概念，看看它们是如何在 Rust 的生态系统中发挥作用的。

一、Pin：固定值的“魔法”

想象一下，你正在处理一个复杂的程序，其中包含了许多动态分配的内存和指针操作。在这种情况下，确保内存中的值不会被意外移动是非常重要的，因为这可能会导致指针失效，进而引发各种难以调试的错误。而 Pin，就像是一个神奇的“钉子”，能够将值固定在内存中的某个位置，防止它们被移动。

1. Pin 的作用

Pin<P> 是一个智能指针，它包装了任意的指针类型 P。它的主要功能是确保包装的值在内存中的位置保持不变。这听起来可能有点抽象，但其实它的应用场景非常广泛，尤其是在处理自引用类型时。

自引用类型是指一个类型内部包含指向其自身字段的指针。例如，考虑以下结构体：

struct SelfRef {value: String,pointer_to_value: *mut String,
}

在这个结构体中，pointer_to_value 是一个裸指针，指向 value 字段。如果 value 被移动，pointer_to_value 将会指向一个无效的地址，从而导致内存安全问题。而 Pin 可以防止这种情况发生。通过将 SelfRef 固定在内存中的某个位置，Pin 确保 value 的地址不会改变，从而保证 pointer_to_value 始终有效。

2. Pin 的使用

Pin 的使用非常简单。你可以通过 Pin::new 或 Pin::new_unchecked 来创建一个 Pin。例如：

use std::pin::Pin;fn main() {let mut value = String::from("Hello, world!");let pinned_value = Pin::new(&mut value);
}

在这个例子中，pinned_value 是一个 Pin<&mut String>，它将 value 固定在内存中的某个位置。需要注意的是，Pin::new 只能用于那些已经实现了 Unpin 的类型。如果类型没有实现 Unpin，你需要使用 Pin::new_unchecked，但这需要你非常小心，因为如果使用不当，可能会导致内存安全问题。

二、Unpin：自由移动的“通行证”

与 Pin 相对的是 Unpin。Unpin 是一个标记 trait，表示对象可以安全地被移动。默认情况下，Rust 为大多数类型自动实现了 Unpin，这意味着这些类型的值可以在内存中自由移动。

1. Unpin 的作用

Unpin 的存在主要是为了与 Pin 配合使用。如果一个类型实现了 Unpin，那么它就可以被自由地移动，即使它被包装在 Pin 中。这听起来可能有点矛盾，但其实这是非常合理的。因为如果一个类型不需要固定在内存中的某个位置，那么就没有必要限制它的移动。

例如，以下代码展示了 Unpin 的自动实现：

struct MyStruct {data: String,
}fn move_struct<T>(val: T) -> T {val
}fn main() {let mut s = MyStruct { data: String::from("Rust") };let pinned_s = Pin::new(&mut s);let s = move_struct(s); // 因为实现了 `Unpin`，可以自由移动println!("{}", s.data);
}

在这个例子中，MyStruct 自动实现了 Unpin，因此即使我们使用 Pin 将其固定，它仍然可以被移动。

2. 阻止类型被移动

如果你希望某个类型在被 Pin 固定后不能被移动，可以通过引入 PhantomPinned 来阻止 Rust 自动为该类型实现 Unpin。例如：

use std::pin::PhantomPinned;struct NoMove {data: String,_pin: PhantomPinned,
}fn move_struct<T>(val: T) -> T {val
}fn main() {let mut s = NoMove { data: String::from("No Move"), _pin: PhantomPinned };let pinned_s = Pin::new(&mut s);// let s = move_struct(s); // 编译错误，因为 `NoMove` 没有实现 `Unpin`println!("{}", pinned_s.data);
}

在这个例子中，NoMove 类型没有实现 Unpin，因此它在被固定后不能被移动。

三、异步编程中的 Pin 和 Unpin

在 Rust 的异步编程模型中，Pin 和 Unpin 尤其重要。Future trait 的 poll 方法要求 self 是 Pin<&mut Self>，确保在轮询期间对象不会被移动。

1. 异步编程中的 Pin

在异步编程中，Pin 的作用是确保 Future 在被轮询时不会被移动。这是因为 Future 可能会包含一些需要固定在内存中的状态。例如：

use std::pin::Pin;
use std::future::Future;
use std::task::{Context, Poll};struct MyFuture {// 可以包含一些状态
}impl Future for MyFuture {type Output = u32;fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {// 这里可以安全地访问固定的内存Poll::Ready(42)}
}fn main() {let mut my_future = MyFuture {};let mut pinned_future = Box::pin(my_future);let waker = /* 创建或获取一个 waker */;let mut cx = Context::from_waker(&waker);let output = pinned_future.as_mut().poll(&mut cx);println!("Output: {:?}", output);
}

在这个例子中，我们创建了一个 MyFuture 结构体，并将其固定在堆上，确保它在异步任务执行期间不会被移动。

2. 异步编程中的 Unpin

在异步编程中，Unpin 的作用是允许某些 Future 被自由地移动。这对于那些不需要固定在内存中的 Future 来说是非常有用的。例如：

struct MyUnpinFuture {data: String,
}impl Future for MyUnpinFuture {type Output = u32;fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {// 这里可以安全地访问固定的内存Poll::Ready(42)}
}impl Unpin for MyUnpinFuture {}fn main() {let mut my_future = MyUnpinFuture { data: String::from("Unpin") };let mut pinned_future = Box::pin(my_future);let waker = /* 创建或获取一个 waker */;let mut cx = Context::from_waker(&waker);let output = pinned_future.as_mut().poll(&mut cx);println!("Output: {:?}", output);
}