feat: Implement ReadVolatile and WriteVolatile for TcpStream

Jinrong Liang · roypat · commit c2cdad5d2143 · 2024-11-04T07:34:18.000Z
Add implementations of the ReadVolatile and WriteVolatile traits for
std::net::TcpStream by extending the impl_read_write_volatile_for_raw_fd!
macro.

Suggested-by: Patrick Roy &lt;roypat@amazon.co.uk&gt;
Signed-off-by: Yi Wang &lt;foxywang@tencent.com&gt;
Signed-off-by: Jinrong Liang &lt;cloudliang@tencent.com&gt;
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## Upcoming version
 
+### Added
+
+- \[[#304](https://github.com/rust-vmm/vm-memory/pull/304)\] Implement ReadVolatile and WriteVolatile for TcpStream
+
 ## \[v0.16.0\]
 
 ### Added
diff --git a/src/io.rs b/src/io.rs
@@ -144,6 +144,7 @@ impl WriteVolatile for Stdout {
 }
 
 impl_read_write_volatile_for_raw_fd!(std::fs::File);
+impl_read_write_volatile_for_raw_fd!(std::net::TcpStream);
 impl_read_write_volatile_for_raw_fd!(std::os::unix::net::UnixStream);
 impl_read_write_volatile_for_raw_fd!(std::os::fd::OwnedFd);
 impl_read_write_volatile_for_raw_fd!(std::os::fd::BorrowedFd<'_>);

Original file line number	Diff line number	Diff line change
`@@ -144,6 +144,7 @@ impl WriteVolatile for Stdout {`
`144`	`144`	`}`
`145`	`145`
`146`	`146`	`impl_read_write_volatile_for_raw_fd!(std::fs::File);`
	`147`	`+impl_read_write_volatile_for_raw_fd!(std::net::TcpStream);`
`147`	`148`	`impl_read_write_volatile_for_raw_fd!(std::os::unix::net::UnixStream);`
`148`	`149`	`impl_read_write_volatile_for_raw_fd!(std::os::fd::OwnedFd);`
`149`	`150`	`impl_read_write_volatile_for_raw_fd!(std::os::fd::BorrowedFd<'_>);`