| 1 | //! Rust friendly bindings to the various *nix system functions. |
| 2 | //! |
| 3 | //! Modules are structured according to the C header file that they would be |
| 4 | //! defined in. |
| 5 | //! |
| 6 | //! # Features |
| 7 | //! |
| 8 | //! Nix uses the following Cargo features to enable optional functionality. |
| 9 | //! They may be enabled in any combination. |
| 10 | //! * `acct` - Process accounting |
| 11 | //! * `aio` - POSIX AIO |
| 12 | //! * `dir` - Stuff relating to directory iteration |
| 13 | //! * `env` - Manipulate environment variables |
| 14 | //! * `event` - Event-driven APIs, like `kqueue` and `epoll` |
| 15 | //! * `fanotify` - Linux's `fanotify` filesystem events monitoring API |
| 16 | //! * `feature` - Query characteristics of the OS at runtime |
| 17 | //! * `fs` - File system functionality |
| 18 | //! * `hostname` - Get and set the system's hostname |
| 19 | //! * `inotify` - Linux's `inotify` file system notification API |
| 20 | //! * `ioctl` - The `ioctl` syscall, and wrappers for many specific instances |
| 21 | //! * `kmod` - Load and unload kernel modules |
| 22 | //! * `mman` - Stuff relating to memory management |
| 23 | //! * `mount` - Mount and unmount file systems |
| 24 | //! * `mqueue` - POSIX message queues |
| 25 | //! * `net` - Networking-related functionality |
| 26 | //! * `personality` - Set the process execution domain |
| 27 | //! * `poll` - APIs like `poll` and `select` |
| 28 | //! * `process` - Stuff relating to running processes |
| 29 | //! * `pthread` - POSIX threads |
| 30 | //! * `ptrace` - Process tracing and debugging |
| 31 | //! * `quota` - File system quotas |
| 32 | //! * `reboot` - Reboot the system |
| 33 | //! * `resource` - Process resource limits |
| 34 | //! * `sched` - Manipulate process's scheduling |
| 35 | //! * `socket` - Sockets, whether for networking or local use |
| 36 | //! * `signal` - Send and receive signals to processes |
| 37 | //! * `term` - Terminal control APIs |
| 38 | //! * `time` - Query the operating system's clocks |
| 39 | //! * `ucontext` - User thread context |
| 40 | //! * `uio` - Vectored I/O |
| 41 | //! * `user` - Stuff relating to users and groups |
| 42 | //! * `zerocopy` - APIs like `sendfile` and `copy_file_range` |
| 43 | #![crate_name = "nix" ] |
| 44 | #![cfg (unix)] |
| 45 | #![allow (non_camel_case_types)] |
| 46 | #![cfg_attr (test, deny(warnings))] |
| 47 | #![recursion_limit = "500" ] |
| 48 | #![deny (unused)] |
| 49 | #![allow (unused_macros)] |
| 50 | #![cfg_attr ( |
| 51 | not(all( |
| 52 | feature = "acct" , |
| 53 | feature = "aio" , |
| 54 | feature = "dir" , |
| 55 | feature = "env" , |
| 56 | feature = "event" , |
| 57 | feature = "fanotify" , |
| 58 | feature = "feature" , |
| 59 | feature = "fs" , |
| 60 | feature = "hostname" , |
| 61 | feature = "inotify" , |
| 62 | feature = "ioctl" , |
| 63 | feature = "kmod" , |
| 64 | feature = "mman" , |
| 65 | feature = "mount" , |
| 66 | feature = "mqueue" , |
| 67 | feature = "net" , |
| 68 | feature = "personality" , |
| 69 | feature = "poll" , |
| 70 | feature = "process" , |
| 71 | feature = "pthread" , |
| 72 | feature = "ptrace" , |
| 73 | feature = "quota" , |
| 74 | feature = "reboot" , |
| 75 | feature = "resource" , |
| 76 | feature = "sched" , |
| 77 | feature = "socket" , |
| 78 | feature = "signal" , |
| 79 | feature = "term" , |
| 80 | feature = "time" , |
| 81 | feature = "ucontext" , |
| 82 | feature = "uio" , |
| 83 | feature = "user" , |
| 84 | feature = "zerocopy" , |
| 85 | )), |
| 86 | allow(unused_imports) |
| 87 | )] |
| 88 | #![deny (unstable_features)] |
| 89 | #![deny (missing_copy_implementations)] |
| 90 | #![deny (missing_debug_implementations)] |
| 91 | #![warn (missing_docs)] |
| 92 | #![cfg_attr (docsrs, feature(doc_cfg))] |
| 93 | #![deny (clippy::cast_ptr_alignment)] |
| 94 | #![deny (unsafe_op_in_unsafe_fn)] |
| 95 | |
| 96 | // Re-exported external crates |
| 97 | pub use libc; |
| 98 | |
| 99 | // Private internal modules |
| 100 | #[macro_use ] |
| 101 | mod macros; |
| 102 | |
| 103 | // Public crates |
| 104 | #[cfg (not(target_os = "redox" ))] |
| 105 | feature! { |
| 106 | #![feature = "dir" ] |
| 107 | pub mod dir; |
| 108 | } |
| 109 | feature! { |
| 110 | #![feature = "env" ] |
| 111 | pub mod env; |
| 112 | } |
| 113 | #[allow (missing_docs)] |
| 114 | pub mod errno; |
| 115 | feature! { |
| 116 | #![feature = "feature" ] |
| 117 | |
| 118 | #[deny (missing_docs)] |
| 119 | pub mod features; |
| 120 | } |
| 121 | pub mod fcntl; |
| 122 | feature! { |
| 123 | #![feature = "net" ] |
| 124 | |
| 125 | #[cfg (any(linux_android, |
| 126 | bsd, |
| 127 | solarish))] |
| 128 | #[deny (missing_docs)] |
| 129 | pub mod ifaddrs; |
| 130 | #[cfg (not(target_os = "redox" ))] |
| 131 | #[deny (missing_docs)] |
| 132 | pub mod net; |
| 133 | } |
| 134 | #[cfg (linux_android)] |
| 135 | feature! { |
| 136 | #![feature = "kmod" ] |
| 137 | pub mod kmod; |
| 138 | } |
| 139 | feature! { |
| 140 | #![feature = "mount" ] |
| 141 | pub mod mount; |
| 142 | } |
| 143 | #[cfg (any(freebsdlike, target_os = "linux" , target_os = "netbsd" ))] |
| 144 | feature! { |
| 145 | #![feature = "mqueue" ] |
| 146 | pub mod mqueue; |
| 147 | } |
| 148 | feature! { |
| 149 | #![feature = "poll" ] |
| 150 | pub mod poll; |
| 151 | } |
| 152 | #[cfg (not(any(target_os = "redox" , target_os = "fuchsia" )))] |
| 153 | feature! { |
| 154 | #![feature = "term" ] |
| 155 | #[deny (missing_docs)] |
| 156 | pub mod pty; |
| 157 | } |
| 158 | feature! { |
| 159 | #![feature = "sched" ] |
| 160 | pub mod sched; |
| 161 | } |
| 162 | pub mod sys; |
| 163 | feature! { |
| 164 | #![feature = "time" ] |
| 165 | pub mod time; |
| 166 | } |
| 167 | // This can be implemented for other platforms as soon as libc |
| 168 | // provides bindings for them. |
| 169 | #[cfg (all( |
| 170 | target_os = "linux" , |
| 171 | any( |
| 172 | target_arch = "aarch64" , |
| 173 | target_arch = "s390x" , |
| 174 | target_arch = "x86" , |
| 175 | target_arch = "x86_64" |
| 176 | ) |
| 177 | ))] |
| 178 | feature! { |
| 179 | #![feature = "ucontext" ] |
| 180 | #[allow (missing_docs)] |
| 181 | pub mod ucontext; |
| 182 | } |
| 183 | pub mod unistd; |
| 184 | |
| 185 | #[cfg (any(feature = "poll" , feature = "event" ))] |
| 186 | mod poll_timeout; |
| 187 | |
| 188 | use std::ffi::{CStr, CString, OsStr}; |
| 189 | use std::mem::MaybeUninit; |
| 190 | use std::os::unix::ffi::OsStrExt; |
| 191 | use std::path::{Path, PathBuf}; |
| 192 | use std::{ptr, result, slice}; |
| 193 | |
| 194 | use errno::Errno; |
| 195 | |
| 196 | /// Nix Result Type |
| 197 | pub type Result<T> = result::Result<T, Errno>; |
| 198 | |
| 199 | /// Nix's main error type. |
| 200 | /// |
| 201 | /// It's a wrapper around Errno. As such, it's very interoperable with |
| 202 | /// [`std::io::Error`], but it has the advantages of: |
| 203 | /// * `Clone` |
| 204 | /// * `Copy` |
| 205 | /// * `Eq` |
| 206 | /// * Small size |
| 207 | /// * Represents all of the system's errnos, instead of just the most common |
| 208 | /// ones. |
| 209 | pub type Error = Errno; |
| 210 | |
| 211 | /// Common trait used to represent file system paths by many Nix functions. |
| 212 | pub trait NixPath { |
| 213 | /// Is the path empty? |
| 214 | fn is_empty(&self) -> bool; |
| 215 | |
| 216 | /// Length of the path in bytes |
| 217 | fn len(&self) -> usize; |
| 218 | |
| 219 | /// Execute a function with this path as a `CStr`. |
| 220 | /// |
| 221 | /// Mostly used internally by Nix. |
| 222 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 223 | where |
| 224 | F: FnOnce(&CStr) -> T; |
| 225 | } |
| 226 | |
| 227 | impl NixPath for str { |
| 228 | fn is_empty(&self) -> bool { |
| 229 | NixPath::is_empty(self:OsStr::new(self)) |
| 230 | } |
| 231 | |
| 232 | fn len(&self) -> usize { |
| 233 | NixPath::len(self:OsStr::new(self)) |
| 234 | } |
| 235 | |
| 236 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 237 | where |
| 238 | F: FnOnce(&CStr) -> T, |
| 239 | { |
| 240 | OsStr::new(self).with_nix_path(f) |
| 241 | } |
| 242 | } |
| 243 | |
| 244 | impl NixPath for OsStr { |
| 245 | fn is_empty(&self) -> bool { |
| 246 | self.as_bytes().is_empty() |
| 247 | } |
| 248 | |
| 249 | fn len(&self) -> usize { |
| 250 | self.as_bytes().len() |
| 251 | } |
| 252 | |
| 253 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 254 | where |
| 255 | F: FnOnce(&CStr) -> T, |
| 256 | { |
| 257 | self.as_bytes().with_nix_path(f) |
| 258 | } |
| 259 | } |
| 260 | |
| 261 | impl NixPath for CStr { |
| 262 | fn is_empty(&self) -> bool { |
| 263 | self.to_bytes().is_empty() |
| 264 | } |
| 265 | |
| 266 | fn len(&self) -> usize { |
| 267 | self.to_bytes().len() |
| 268 | } |
| 269 | |
| 270 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 271 | where |
| 272 | F: FnOnce(&CStr) -> T, |
| 273 | { |
| 274 | Ok(f(self)) |
| 275 | } |
| 276 | } |
| 277 | |
| 278 | impl NixPath for [u8] { |
| 279 | fn is_empty(&self) -> bool { |
| 280 | self.is_empty() |
| 281 | } |
| 282 | |
| 283 | fn len(&self) -> usize { |
| 284 | self.len() |
| 285 | } |
| 286 | |
| 287 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 288 | where |
| 289 | F: FnOnce(&CStr) -> T, |
| 290 | { |
| 291 | // The real PATH_MAX is typically 4096, but it's statistically unlikely to have a path |
| 292 | // longer than ~300 bytes. See the the PR description to get stats for your own machine. |
| 293 | // https://github.com/nix-rust/nix/pull/1656 |
| 294 | // |
| 295 | // By being smaller than a memory page, we also avoid the compiler inserting a probe frame: |
| 296 | // https://docs.rs/compiler_builtins/latest/compiler_builtins/probestack/index.html |
| 297 | const MAX_STACK_ALLOCATION: usize = 1024; |
| 298 | |
| 299 | if self.len() >= MAX_STACK_ALLOCATION { |
| 300 | return with_nix_path_allocating(self, f); |
| 301 | } |
| 302 | |
| 303 | let mut buf = MaybeUninit::<[u8; MAX_STACK_ALLOCATION]>::uninit(); |
| 304 | let buf_ptr = buf.as_mut_ptr().cast(); |
| 305 | |
| 306 | unsafe { |
| 307 | ptr::copy_nonoverlapping(self.as_ptr(), buf_ptr, self.len()); |
| 308 | buf_ptr.add(self.len()).write(0); |
| 309 | } |
| 310 | |
| 311 | match CStr::from_bytes_with_nul(unsafe { |
| 312 | slice::from_raw_parts(buf_ptr, self.len() + 1) |
| 313 | }) { |
| 314 | Ok(s) => Ok(f(s)), |
| 315 | Err(_) => Err(Errno::EINVAL), |
| 316 | } |
| 317 | } |
| 318 | } |
| 319 | |
| 320 | #[cold ] |
| 321 | #[inline (never)] |
| 322 | fn with_nix_path_allocating<T, F>(from: &[u8], f: F) -> Result<T> |
| 323 | where |
| 324 | F: FnOnce(&CStr) -> T, |
| 325 | { |
| 326 | match CString::new(from) { |
| 327 | Ok(s: CString) => Ok(f(&s)), |
| 328 | Err(_) => Err(Errno::EINVAL), |
| 329 | } |
| 330 | } |
| 331 | |
| 332 | impl NixPath for Path { |
| 333 | fn is_empty(&self) -> bool { |
| 334 | NixPath::is_empty(self.as_os_str()) |
| 335 | } |
| 336 | |
| 337 | fn len(&self) -> usize { |
| 338 | NixPath::len(self.as_os_str()) |
| 339 | } |
| 340 | |
| 341 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 342 | where |
| 343 | F: FnOnce(&CStr) -> T, |
| 344 | { |
| 345 | self.as_os_str().with_nix_path(f) |
| 346 | } |
| 347 | } |
| 348 | |
| 349 | impl NixPath for PathBuf { |
| 350 | fn is_empty(&self) -> bool { |
| 351 | NixPath::is_empty(self.as_os_str()) |
| 352 | } |
| 353 | |
| 354 | fn len(&self) -> usize { |
| 355 | NixPath::len(self.as_os_str()) |
| 356 | } |
| 357 | |
| 358 | fn with_nix_path<T, F>(&self, f: F) -> Result<T> |
| 359 | where |
| 360 | F: FnOnce(&CStr) -> T, |
| 361 | { |
| 362 | self.as_os_str().with_nix_path(f) |
| 363 | } |
| 364 | } |
| 365 | |
| 366 | /// Like `NixPath::with_nix_path()`, but allow the `path` argument to be optional. |
| 367 | /// |
| 368 | /// A NULL pointer will be provided if `path.is_none()`. |
| 369 | #[cfg (any( |
| 370 | all(apple_targets, feature = "mount" ), |
| 371 | all(linux_android, any(feature = "mount" , feature = "fanotify" )) |
| 372 | ))] |
| 373 | pub(crate) fn with_opt_nix_path<P, T, F>(path: Option<&P>, f: F) -> Result<T> |
| 374 | where |
| 375 | P: ?Sized + NixPath, |
| 376 | F: FnOnce(*const libc::c_char) -> T, |
| 377 | { |
| 378 | match path { |
| 379 | Some(path) => path.with_nix_path(|p_str| f(p_str.as_ptr())), |
| 380 | None => Ok(f(ptr::null())), |
| 381 | } |
| 382 | } |
| 383 | |