Skip to content

Commit c980f0e

Browse files
thaliaarchithaliaarchi
committed
Unify owned Args types between platforms
1 parent e014fd6 commit c980f0e

File tree

8 files changed

+71
-225
lines changed

8 files changed

+71
-225
lines changed

‎library/std/src/sys/args/common.rs

+43
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
use crate::ffi::OsString;
2+
use crate::{fmt, vec};
3+
4+
pub struct Args {
5+
iter: vec::IntoIter<OsString>,
6+
}
7+
8+
impl !Send for Args {}
9+
impl !Sync for Args {}
10+
11+
impl Args {
12+
pub(super) fn new(args: Vec<OsString>) -> Self {
13+
Args { iter: args.into_iter() }
14+
}
15+
}
16+
17+
impl fmt::Debug for Args {
18+
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
19+
self.iter.as_slice().fmt(f)
20+
}
21+
}
22+
23+
impl Iterator for Args {
24+
type Item = OsString;
25+
fn next(&mut self) -> Option<OsString> {
26+
self.iter.next()
27+
}
28+
fn size_hint(&self) -> (usize, Option<usize>) {
29+
self.iter.size_hint()
30+
}
31+
}
32+
33+
impl ExactSizeIterator for Args {
34+
fn len(&self) -> usize {
35+
self.iter.len()
36+
}
37+
}
38+
39+
impl DoubleEndedIterator for Args {
40+
fn next_back(&mut self) -> Option<OsString> {
41+
self.iter.next_back()
42+
}
43+
}

‎library/std/src/sys/args/hermit.rs

+3-37
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
use crate::ffi::{CStr, OsString, c_char};
22
use crate::os::hermit::ffi::OsStringExt;
3+
use crate::ptr;
34
use crate::sync::atomic::Ordering::{Acquire, Relaxed, Release};
45
use crate::sync::atomic::{AtomicIsize, AtomicPtr};
5-
use crate::{fmt, ptr, vec};
6+
use crate::sys::args::Args;
67

78
static ARGC: AtomicIsize = AtomicIsize::new(0);
89
static ARGV: AtomicPtr<*const u8> = AtomicPtr::new(ptr::null_mut());
@@ -27,40 +28,5 @@ pub fn args() -> Args {
2728
})
2829
.collect();
2930

30-
Args { iter: args.into_iter() }
31-
}
32-
33-
pub struct Args {
34-
iter: vec::IntoIter<OsString>,
35-
}
36-
37-
impl fmt::Debug for Args {
38-
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
39-
self.iter.as_slice().fmt(f)
40-
}
41-
}
42-
43-
impl !Send for Args {}
44-
impl !Sync for Args {}
45-
46-
impl Iterator for Args {
47-
type Item = OsString;
48-
fn next(&mut self) -> Option<OsString> {
49-
self.iter.next()
50-
}
51-
fn size_hint(&self) -> (usize, Option<usize>) {
52-
self.iter.size_hint()
53-
}
54-
}
55-
56-
impl ExactSizeIterator for Args {
57-
fn len(&self) -> usize {
58-
self.iter.len()
59-
}
60-
}
61-
62-
impl DoubleEndedIterator for Args {
63-
fn next_back(&mut self) -> Option<OsString> {
64-
self.iter.next_back()
65-
}
31+
Args::new(args)
6632
}

‎library/std/src/sys/args/mod.rs

+9
Original file line numberDiff line numberDiff line change
@@ -2,27 +2,36 @@
22
33
#![forbid(unsafe_op_in_unsafe_fn)]
44

5+
#[allow(dead_code)]
6+
mod common;
7+
58
cfg_if::cfg_if! {
69
if #[cfg(all(target_family = "unix", not(any(target_os = "espidf", target_os = "vita"))))] {
710
mod unix;
11+
pub use common::Args;
812
pub use unix::*;
913
} else if #[cfg(target_family = "windows")] {
1014
mod windows;
15+
pub use common::Args;
1116
pub use windows::*;
1217
} else if #[cfg(target_os = "hermit")] {
1318
mod hermit;
19+
pub use common::Args;
1420
pub use hermit::*;
1521
} else if #[cfg(all(target_vendor = "fortanix", target_env = "sgx"))] {
1622
mod sgx;
1723
pub use sgx::*;
1824
} else if #[cfg(target_os = "uefi")] {
1925
mod uefi;
26+
pub use common::Args;
2027
pub use uefi::*;
2128
} else if #[cfg(target_os = "wasi")] {
2229
mod wasi;
30+
pub use common::Args;
2331
pub use wasi::*;
2432
} else if #[cfg(target_os = "xous")] {
2533
mod xous;
34+
pub use common::Args;
2635
pub use xous::*;
2736
} else if #[cfg(target_os = "zkvm")] {
2837
mod zkvm;

‎library/std/src/sys/args/uefi.rs

+4-42
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,8 @@ use r_efi::protocols::loaded_image;
33
use crate::env::current_exe;
44
use crate::ffi::OsString;
55
use crate::iter::Iterator;
6+
use crate::sys::args::Args;
67
use crate::sys::pal::helpers;
7-
use crate::{fmt, vec};
8-
9-
pub struct Args {
10-
parsed_args_list: vec::IntoIter<OsString>,
11-
}
128

139
pub fn args() -> Args {
1410
let lazy_current_exe = || Vec::from([current_exe().map(Into::into).unwrap_or_default()]);
@@ -22,51 +18,17 @@ pub fn args() -> Args {
2218
let lp_size = unsafe { (*protocol.as_ptr()).load_options_size } as usize;
2319
// Break if we are sure that it cannot be UTF-16
2420
if lp_size < size_of::<u16>() || lp_size % size_of::<u16>() != 0 {
25-
return Args { parsed_args_list: lazy_current_exe().into_iter() };
21+
return Args::new(lazy_current_exe());
2622
}
2723
let lp_size = lp_size / size_of::<u16>();
2824

2925
let lp_cmd_line = unsafe { (*protocol.as_ptr()).load_options as *const u16 };
3026
if !lp_cmd_line.is_aligned() {
31-
return Args { parsed_args_list: lazy_current_exe().into_iter() };
27+
return Args::new(lazy_current_exe());
3228
}
3329
let lp_cmd_line = unsafe { crate::slice::from_raw_parts(lp_cmd_line, lp_size) };
3430

35-
Args {
36-
parsed_args_list: parse_lp_cmd_line(lp_cmd_line)
37-
.unwrap_or_else(lazy_current_exe)
38-
.into_iter(),
39-
}
40-
}
41-
42-
impl fmt::Debug for Args {
43-
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
44-
self.parsed_args_list.as_slice().fmt(f)
45-
}
46-
}
47-
48-
impl Iterator for Args {
49-
type Item = OsString;
50-
51-
fn next(&mut self) -> Option<OsString> {
52-
self.parsed_args_list.next()
53-
}
54-
55-
fn size_hint(&self) -> (usize, Option<usize>) {
56-
self.parsed_args_list.size_hint()
57-
}
58-
}
59-
60-
impl ExactSizeIterator for Args {
61-
fn len(&self) -> usize {
62-
self.parsed_args_list.len()
63-
}
64-
}
65-
66-
impl DoubleEndedIterator for Args {
67-
fn next_back(&mut self) -> Option<OsString> {
68-
self.parsed_args_list.next_back()
69-
}
31+
Args::new(parse_lp_cmd_line(lp_cmd_line).unwrap_or_else(lazy_current_exe))
7032
}
7133

7234
/// Implements the UEFI command-line argument parsing algorithm.

‎library/std/src/sys/args/unix.rs

+3-38
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55
66
#![allow(dead_code)] // runtime init functions not used during testing
77

8-
use crate::ffi::{CStr, OsString};
8+
use crate::ffi::CStr;
99
use crate::os::unix::ffi::OsStringExt;
10-
use crate::{fmt, vec};
10+
use crate::sys::args::Args;
1111

1212
/// One-time global initialization.
1313
pub unsafe fn init(argc: isize, argv: *const *const u8) {
@@ -55,42 +55,7 @@ pub fn args() -> Args {
5555
vec.push(OsStringExt::from_vec(cstr.to_bytes().to_vec()));
5656
}
5757

58-
Args { iter: vec.into_iter() }
59-
}
60-
61-
pub struct Args {
62-
iter: vec::IntoIter<OsString>,
63-
}
64-
65-
impl !Send for Args {}
66-
impl !Sync for Args {}
67-
68-
impl fmt::Debug for Args {
69-
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
70-
self.iter.as_slice().fmt(f)
71-
}
72-
}
73-
74-
impl Iterator for Args {
75-
type Item = OsString;
76-
fn next(&mut self) -> Option<OsString> {
77-
self.iter.next()
78-
}
79-
fn size_hint(&self) -> (usize, Option<usize>) {
80-
self.iter.size_hint()
81-
}
82-
}
83-
84-
impl ExactSizeIterator for Args {
85-
fn len(&self) -> usize {
86-
self.iter.len()
87-
}
88-
}
89-
90-
impl DoubleEndedIterator for Args {
91-
fn next_back(&mut self) -> Option<OsString> {
92-
self.iter.next_back()
93-
}
58+
Args::new(vec)
9459
}
9560

9661
#[cfg(any(

‎library/std/src/sys/args/wasi.rs

+2-37
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,11 @@
22

33
use crate::ffi::{CStr, OsStr, OsString};
44
use crate::os::wasi::ffi::OsStrExt;
5-
use crate::{fmt, vec};
6-
7-
pub struct Args {
8-
iter: vec::IntoIter<OsString>,
9-
}
10-
11-
impl !Send for Args {}
12-
impl !Sync for Args {}
5+
use crate::sys::args::Args;
136

147
/// Returns the command line arguments
158
pub fn args() -> Args {
16-
Args { iter: maybe_args().unwrap_or(Vec::new()).into_iter() }
9+
Args::new(maybe_args().unwrap_or(Vec::new()))
1710
}
1811

1912
fn maybe_args() -> Option<Vec<OsString>> {
@@ -31,31 +24,3 @@ fn maybe_args() -> Option<Vec<OsString>> {
3124
Some(ret)
3225
}
3326
}
34-
35-
impl fmt::Debug for Args {
36-
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
37-
self.iter.as_slice().fmt(f)
38-
}
39-
}
40-
41-
impl Iterator for Args {
42-
type Item = OsString;
43-
fn next(&mut self) -> Option<OsString> {
44-
self.iter.next()
45-
}
46-
fn size_hint(&self) -> (usize, Option<usize>) {
47-
self.iter.size_hint()
48-
}
49-
}
50-
51-
impl ExactSizeIterator for Args {
52-
fn len(&self) -> usize {
53-
self.iter.len()
54-
}
55-
}
56-
57-
impl DoubleEndedIterator for Args {
58-
fn next_back(&mut self) -> Option<OsString> {
59-
self.iter.next_back()
60-
}
61-
}

‎library/std/src/sys/args/windows.rs

+3-34
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,14 @@ use crate::ffi::{OsStr, OsString};
1010
use crate::num::NonZero;
1111
use crate::os::windows::prelude::*;
1212
use crate::path::{Path, PathBuf};
13+
use crate::sys::args::Args;
1314
use crate::sys::pal::os::current_exe;
1415
use crate::sys::pal::{ensure_no_nuls, fill_utf16_buf};
1516
use crate::sys::path::get_long_path;
1617
use crate::sys::{c, to_u16s};
1718
use crate::sys_common::AsInner;
1819
use crate::sys_common::wstr::WStrUnits;
19-
use crate::{fmt, io, iter, ptr, vec};
20+
use crate::{io, iter, ptr};
2021

2122
pub fn args() -> Args {
2223
// SAFETY: `GetCommandLineW` returns a pointer to a null terminated UTF-16
@@ -27,7 +28,7 @@ pub fn args() -> Args {
2728
current_exe().map(PathBuf::into_os_string).unwrap_or_else(|_| OsString::new())
2829
});
2930

30-
Args { parsed_args_list: parsed_args_list.into_iter() }
31+
Args::new(parsed_args_list)
3132
}
3233
}
3334

@@ -153,38 +154,6 @@ fn parse_lp_cmd_line<'a, F: Fn() -> OsString>(
153154
ret_val
154155
}
155156

156-
pub struct Args {
157-
parsed_args_list: vec::IntoIter<OsString>,
158-
}
159-
160-
impl fmt::Debug for Args {
161-
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
162-
self.parsed_args_list.as_slice().fmt(f)
163-
}
164-
}
165-
166-
impl Iterator for Args {
167-
type Item = OsString;
168-
fn next(&mut self) -> Option<OsString> {
169-
self.parsed_args_list.next()
170-
}
171-
fn size_hint(&self) -> (usize, Option<usize>) {
172-
self.parsed_args_list.size_hint()
173-
}
174-
}
175-
176-
impl DoubleEndedIterator for Args {
177-
fn next_back(&mut self) -> Option<OsString> {
178-
self.parsed_args_list.next_back()
179-
}
180-
}
181-
182-
impl ExactSizeIterator for Args {
183-
fn len(&self) -> usize {
184-
self.parsed_args_list.len()
185-
}
186-
}
187-
188157
#[derive(Debug)]
189158
pub(crate) enum Arg {
190159
/// Add quotes (if needed)

0 commit comments

Comments
 (0)