1
0
Fork 0
mirror of https://github.com/alacritty/alacritty.git synced 2024-11-25 14:05:41 -05:00

Clean up Windows PTY string handling

Removes widestring and dunce dependencies, reduces some code duplication
and corrects a few typos.
This commit is contained in:
David Hewitt 2020-01-02 11:49:27 +00:00 committed by Christian Duerr
parent 77acb26d73
commit f632919134
13 changed files with 65 additions and 93 deletions

15
Cargo.lock generated
View file

@ -58,7 +58,6 @@ dependencies = [
"base64 0.11.0 (registry+https://github.com/rust-lang/crates.io-index)", "base64 0.11.0 (registry+https://github.com/rust-lang/crates.io-index)",
"bitflags 1.2.1 (registry+https://github.com/rust-lang/crates.io-index)", "bitflags 1.2.1 (registry+https://github.com/rust-lang/crates.io-index)",
"copypasta 0.6.1", "copypasta 0.6.1",
"dunce 1.0.0 (registry+https://github.com/rust-lang/crates.io-index)",
"font 0.1.0", "font 0.1.0",
"libc 0.2.66 (registry+https://github.com/rust-lang/crates.io-index)", "libc 0.2.66 (registry+https://github.com/rust-lang/crates.io-index)",
"log 0.4.8 (registry+https://github.com/rust-lang/crates.io-index)", "log 0.4.8 (registry+https://github.com/rust-lang/crates.io-index)",
@ -79,7 +78,6 @@ dependencies = [
"unicode-width 0.1.7 (registry+https://github.com/rust-lang/crates.io-index)", "unicode-width 0.1.7 (registry+https://github.com/rust-lang/crates.io-index)",
"url 2.1.0 (registry+https://github.com/rust-lang/crates.io-index)", "url 2.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
"vte 0.3.3 (registry+https://github.com/rust-lang/crates.io-index)", "vte 0.3.3 (registry+https://github.com/rust-lang/crates.io-index)",
"widestring 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)",
"winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)", "winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)",
"winpty 0.1.0", "winpty 0.1.0",
] ]
@ -488,11 +486,6 @@ name = "dtoa"
version = "0.4.4" version = "0.4.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
[[package]]
name = "dunce"
version = "1.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
[[package]] [[package]]
name = "dwrote" name = "dwrote"
version = "0.9.0" version = "0.9.0"
@ -2242,11 +2235,6 @@ dependencies = [
"libc 0.2.66 (registry+https://github.com/rust-lang/crates.io-index)", "libc 0.2.66 (registry+https://github.com/rust-lang/crates.io-index)",
] ]
[[package]]
name = "widestring"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
[[package]] [[package]]
name = "winapi" name = "winapi"
version = "0.2.8" version = "0.2.8"
@ -2328,7 +2316,6 @@ dependencies = [
"http_req 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)", "http_req 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)",
"named_pipe 0.4.1 (registry+https://github.com/rust-lang/crates.io-index)", "named_pipe 0.4.1 (registry+https://github.com/rust-lang/crates.io-index)",
"tempfile 3.1.0 (registry+https://github.com/rust-lang/crates.io-index)", "tempfile 3.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
"widestring 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)",
"winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)", "winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)",
"winpty-sys 0.4.3 (registry+https://github.com/rust-lang/crates.io-index)", "winpty-sys 0.4.3 (registry+https://github.com/rust-lang/crates.io-index)",
"zip 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)", "zip 0.5.3 (registry+https://github.com/rust-lang/crates.io-index)",
@ -2469,7 +2456,6 @@ dependencies = [
"checksum dlib 0.4.1 (registry+https://github.com/rust-lang/crates.io-index)" = "77e51249a9d823a4cb79e3eca6dcd756153e8ed0157b6c04775d04bf1b13b76a" "checksum dlib 0.4.1 (registry+https://github.com/rust-lang/crates.io-index)" = "77e51249a9d823a4cb79e3eca6dcd756153e8ed0157b6c04775d04bf1b13b76a"
"checksum downcast-rs 1.1.1 (registry+https://github.com/rust-lang/crates.io-index)" = "52ba6eb47c2131e784a38b726eb54c1e1484904f013e576a25354d0124161af6" "checksum downcast-rs 1.1.1 (registry+https://github.com/rust-lang/crates.io-index)" = "52ba6eb47c2131e784a38b726eb54c1e1484904f013e576a25354d0124161af6"
"checksum dtoa 0.4.4 (registry+https://github.com/rust-lang/crates.io-index)" = "ea57b42383d091c85abcc2706240b94ab2a8fa1fc81c10ff23c4de06e2a90b5e" "checksum dtoa 0.4.4 (registry+https://github.com/rust-lang/crates.io-index)" = "ea57b42383d091c85abcc2706240b94ab2a8fa1fc81c10ff23c4de06e2a90b5e"
"checksum dunce 1.0.0 (registry+https://github.com/rust-lang/crates.io-index)" = "d0ad6bf6a88548d1126045c413548df1453d9be094a8ab9fd59bf1fdd338da4f"
"checksum dwrote 0.9.0 (registry+https://github.com/rust-lang/crates.io-index)" = "0bd1369e02db5e9b842a9b67bce8a2fcc043beafb2ae8a799dd482d46ea1ff0d" "checksum dwrote 0.9.0 (registry+https://github.com/rust-lang/crates.io-index)" = "0bd1369e02db5e9b842a9b67bce8a2fcc043beafb2ae8a799dd482d46ea1ff0d"
"checksum embed-resource 1.3.1 (registry+https://github.com/rust-lang/crates.io-index)" = "bbaba4684ab0af1cbb3ef0b1f540ddc4b57b31940c920ea594efe09ab86e2a6c" "checksum embed-resource 1.3.1 (registry+https://github.com/rust-lang/crates.io-index)" = "bbaba4684ab0af1cbb3ef0b1f540ddc4b57b31940c920ea594efe09ab86e2a6c"
"checksum env_logger 0.5.13 (registry+https://github.com/rust-lang/crates.io-index)" = "15b0a4d2e39f8420210be8b27eeda28029729e2fd4291019455016c348240c38" "checksum env_logger 0.5.13 (registry+https://github.com/rust-lang/crates.io-index)" = "15b0a4d2e39f8420210be8b27eeda28029729e2fd4291019455016c348240c38"
@ -2671,7 +2657,6 @@ dependencies = [
"checksum wayland-scanner 0.23.6 (registry+https://github.com/rust-lang/crates.io-index)" = "93b02247366f395b9258054f964fe293ddd019c3237afba9be2ccbe9e1651c3d" "checksum wayland-scanner 0.23.6 (registry+https://github.com/rust-lang/crates.io-index)" = "93b02247366f395b9258054f964fe293ddd019c3237afba9be2ccbe9e1651c3d"
"checksum wayland-sys 0.23.6 (registry+https://github.com/rust-lang/crates.io-index)" = "d94e89a86e6d6d7c7c9b19ebf48a03afaac4af6bc22ae570e9a24124b75358f4" "checksum wayland-sys 0.23.6 (registry+https://github.com/rust-lang/crates.io-index)" = "d94e89a86e6d6d7c7c9b19ebf48a03afaac4af6bc22ae570e9a24124b75358f4"
"checksum which 1.0.5 (registry+https://github.com/rust-lang/crates.io-index)" = "e84a603e7e0b1ce1aa1ee2b109c7be00155ce52df5081590d1ffb93f4f515cb2" "checksum which 1.0.5 (registry+https://github.com/rust-lang/crates.io-index)" = "e84a603e7e0b1ce1aa1ee2b109c7be00155ce52df5081590d1ffb93f4f515cb2"
"checksum widestring 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)" = "effc0e4ff8085673ea7b9b2e3c73f6bd4d118810c9009ed8f1e16bd96c331db6"
"checksum winapi 0.2.8 (registry+https://github.com/rust-lang/crates.io-index)" = "167dc9d6949a9b857f3451275e911c3f44255842c1f7a76f33c55103a909087a" "checksum winapi 0.2.8 (registry+https://github.com/rust-lang/crates.io-index)" = "167dc9d6949a9b857f3451275e911c3f44255842c1f7a76f33c55103a909087a"
"checksum winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)" = "8093091eeb260906a183e6ae1abdba2ef5ef2257a21801128899c3fc699229c6" "checksum winapi 0.3.8 (registry+https://github.com/rust-lang/crates.io-index)" = "8093091eeb260906a183e6ae1abdba2ef5ef2257a21801128899c3fc699229c6"
"checksum winapi-build 0.1.1 (registry+https://github.com/rust-lang/crates.io-index)" = "2d315eee3b34aca4797b2da6b13ed88266e6d612562a0c46390af8299fc699bc" "checksum winapi-build 0.1.1 (registry+https://github.com/rust-lang/crates.io-index)" = "2d315eee3b34aca4797b2da6b13ed88266e6d612562a0c46390af8299fc699bc"

View file

@ -254,12 +254,12 @@ impl Options {
} }
pub fn into_config(self, mut config: Config) -> Config { pub fn into_config(self, mut config: Config) -> Config {
config.set_live_config_reload( if let Some(lcr) = self.live_config_reload {
self.live_config_reload.unwrap_or_else(|| config.live_config_reload()), config.set_live_config_reload(lcr);
); }
config.set_working_directory( if let Some(wd) = self.working_dir {
self.working_dir.or_else(|| config.working_directory().to_owned()), config.set_working_directory(Some(wd));
); }
config.shell = self.command.or(config.shell); config.shell = self.command.or(config.shell);
config.hold = self.hold; config.hold = self.hold;

View file

@ -34,12 +34,10 @@ signal-hook = { version = "0.1", features = ["mio-support"] }
winpty = { path = "../winpty" } winpty = { path = "../winpty" }
mio-named-pipes = "0.1" mio-named-pipes = "0.1"
miow = "0.3" miow = "0.3"
dunce = "1.0"
winapi = { version = "0.3.7", features = [ winapi = { version = "0.3.7", features = [
"impl-default", "basetsd", "libloaderapi", "minwindef", "ntdef", "processthreadsapi", "winbase", "impl-default", "basetsd", "libloaderapi", "minwindef", "ntdef", "processthreadsapi", "winbase",
"wincon", "wincontypes", "winerror", "winnt", "winuser", "wincon", "wincontypes", "winerror", "winnt", "winuser",
]} ]}
widestring = "0.4"
mio-anonymous-pipes = "0.1" mio-anonymous-pipes = "0.1"
[target.'cfg(target_os = "macos")'.dependencies] [target.'cfg(target_os = "macos")'.dependencies]

View file

@ -15,7 +15,7 @@
use std::borrow::Cow; use std::borrow::Cow;
use std::collections::HashMap; use std::collections::HashMap;
use std::fmt::Display; use std::fmt::Display;
use std::path::PathBuf; use std::path::{Path, PathBuf};
use log::error; use log::error;
use serde::{Deserialize, Deserializer}; use serde::{Deserialize, Deserializer};
@ -209,8 +209,8 @@ impl<T> Config<T> {
} }
#[inline] #[inline]
pub fn working_directory(&self) -> &Option<PathBuf> { pub fn working_directory(&self) -> Option<&Path> {
&self.working_directory self.working_directory.as_ref().map(|path_buf| path_buf.as_path())
} }
#[inline] #[inline]

View file

@ -13,6 +13,8 @@
// limitations under the License. // limitations under the License.
// //
//! ANSI Terminal Stream Parsing //! ANSI Terminal Stream Parsing
#[cfg(windows)]
use crate::tty::windows::win32_string;
// Use the default behavior of the other platforms. // Use the default behavior of the other platforms.
#[cfg(not(windows))] #[cfg(not(windows))]
@ -41,13 +43,3 @@ pub fn attach_handler() {
} }
})); }));
} }
// Converts the string slice into a Windows-standard representation for "W"-
// suffixed function variants, which accept UTF-16 encoded string values.
#[cfg(windows)]
fn win32_string(value: &str) -> Vec<u16> {
use std::ffi::OsStr;
use std::iter::once;
use std::os::windows::ffi::OsStrExt;
OsStr::new(value).encode_wide().chain(once(0)).collect()
}

View file

@ -26,7 +26,7 @@ mod unix;
pub use self::unix::*; pub use self::unix::*;
#[cfg(windows)] #[cfg(windows)]
mod windows; pub mod windows;
#[cfg(windows)] #[cfg(windows)]
pub use self::windows::*; pub use self::windows::*;

View file

@ -209,7 +209,7 @@ pub fn new<C>(config: &Config<C>, size: &SizeInfo, window_id: Option<usize>) ->
// Handle set working directory option // Handle set working directory option
if let Some(ref dir) = config.working_directory() { if let Some(ref dir) = config.working_directory() {
builder.current_dir(dir.as_path()); builder.current_dir(dir);
} }
// Prepare signal handling before spawning child // Prepare signal handling before spawning child

View file

@ -18,10 +18,8 @@ use std::mem;
use std::os::windows::io::IntoRawHandle; use std::os::windows::io::IntoRawHandle;
use std::ptr; use std::ptr;
use dunce::canonicalize;
use mio_anonymous_pipes::{EventedAnonRead, EventedAnonWrite}; use mio_anonymous_pipes::{EventedAnonRead, EventedAnonWrite};
use miow; use miow;
use widestring::U16CString;
use winapi::shared::basetsd::{PSIZE_T, SIZE_T}; use winapi::shared::basetsd::{PSIZE_T, SIZE_T};
use winapi::shared::minwindef::{BYTE, DWORD}; use winapi::shared::minwindef::{BYTE, DWORD};
use winapi::shared::ntdef::{HANDLE, HRESULT, LPWSTR}; use winapi::shared::ntdef::{HANDLE, HRESULT, LPWSTR};
@ -34,11 +32,11 @@ use winapi::um::processthreadsapi::{
use winapi::um::winbase::{EXTENDED_STARTUPINFO_PRESENT, STARTF_USESTDHANDLES, STARTUPINFOEXW}; use winapi::um::winbase::{EXTENDED_STARTUPINFO_PRESENT, STARTF_USESTDHANDLES, STARTUPINFOEXW};
use winapi::um::wincontypes::{COORD, HPCON}; use winapi::um::wincontypes::{COORD, HPCON};
use crate::config::{Config, Shell}; use crate::config::Config;
use crate::event::OnResize; use crate::event::OnResize;
use crate::term::SizeInfo; use crate::term::SizeInfo;
use crate::tty::windows::child::ChildExitWatcher; use crate::tty::windows::child::ChildExitWatcher;
use crate::tty::windows::Pty; use crate::tty::windows::{cmdline, win32_string, Pty};
// TODO: Replace with winapi's implementation. This cannot be // TODO: Replace with winapi's implementation. This cannot be
// done until a safety net is in place for versions of Windows // done until a safety net is in place for versions of Windows
@ -141,8 +139,7 @@ pub fn new<C>(config: &Config<C>, size: &SizeInfo, _window_id: Option<usize>) ->
let mut startup_info_ex: STARTUPINFOEXW = Default::default(); let mut startup_info_ex: STARTUPINFOEXW = Default::default();
let title = config.window.title.clone(); let title = win32_string(&config.window.title);
let title = U16CString::from_str(title).unwrap();
startup_info_ex.StartupInfo.lpTitle = title.as_ptr() as LPWSTR; startup_info_ex.StartupInfo.lpTitle = title.as_ptr() as LPWSTR;
startup_info_ex.StartupInfo.cb = mem::size_of::<STARTUPINFOEXW>() as u32; startup_info_ex.StartupInfo.cb = mem::size_of::<STARTUPINFOEXW>() as u32;
@ -205,19 +202,11 @@ pub fn new<C>(config: &Config<C>, size: &SizeInfo, _window_id: Option<usize>) ->
} }
} }
// Get process commandline let cmdline = win32_string(&cmdline(&config));
let default_shell = &Shell::new("powershell"); let cwd = config
let shell = config.shell.as_ref().unwrap_or(default_shell); .working_directory()
let mut cmdline = shell.args.clone(); .map(|dir| dir.canonicalize().unwrap())
cmdline.insert(0, shell.program.to_string()); .map(|path| win32_string(&path));
// Warning, here be borrow hell
let cwd = config.working_directory().as_ref().map(|dir| canonicalize(dir).unwrap());
let cwd = cwd.as_ref().map(|dir| dir.to_str().unwrap());
// Create the client application, using startup info containing ConPTY info
let cmdline = U16CString::from_str(&cmdline.join(" ")).unwrap();
let cwd = cwd.map(|s| U16CString::from_str(&s).unwrap());
let mut proc_info: PROCESS_INFORMATION = Default::default(); let mut proc_info: PROCESS_INFORMATION = Default::default();
unsafe { unsafe {

View file

@ -12,7 +12,10 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
use std::ffi::OsStr;
use std::io::{self, Read, Write}; use std::io::{self, Read, Write};
use std::iter::once;
use std::os::windows::ffi::OsStrExt;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::mpsc::TryRecvError; use std::sync::mpsc::TryRecvError;
@ -22,7 +25,7 @@ use mio_named_pipes::NamedPipe;
use log::info; use log::info;
use crate::config::Config; use crate::config::{Config, Shell};
use crate::event::OnResize; use crate::event::OnResize;
use crate::term::SizeInfo; use crate::term::SizeInfo;
use crate::tty::windows::child::ChildExitWatcher; use crate::tty::windows::child::ChildExitWatcher;
@ -298,3 +301,19 @@ impl OnResize for Pty {
} }
} }
} }
fn cmdline<C>(config: &Config<C>) -> String {
let default_shell = Shell::new("powershell");
let shell = config.shell.as_ref().unwrap_or(&default_shell);
once(shell.program.as_ref())
.chain(shell.args.iter().map(|a| a.as_ref()))
.collect::<Vec<_>>()
.join(" ")
}
/// Converts the string slice into a Windows-standard representation for "W"-
/// suffixed function variants, which accept UTF-16 encoded string values.
pub fn win32_string<S: AsRef<OsStr> + ?Sized>(value: &S) -> Vec<u16> {
OsStr::new(value).encode_wide().chain(once(0)).collect()
}

View file

@ -17,17 +17,16 @@ use std::os::windows::fs::OpenOptionsExt;
use std::os::windows::io::{FromRawHandle, IntoRawHandle}; use std::os::windows::io::{FromRawHandle, IntoRawHandle};
use std::u16; use std::u16;
use dunce::canonicalize;
use log::info; use log::info;
use mio_named_pipes::NamedPipe; use mio_named_pipes::NamedPipe;
use winapi::um::winbase::FILE_FLAG_OVERLAPPED; use winapi::um::winbase::FILE_FLAG_OVERLAPPED;
use winpty::{Config as WinptyConfig, ConfigFlags, MouseMode, SpawnConfig, SpawnFlags, Winpty}; use winpty::{Config as WinptyConfig, ConfigFlags, MouseMode, SpawnConfig, SpawnFlags, Winpty};
use crate::config::{Config, Shell}; use crate::config::Config;
use crate::event::OnResize; use crate::event::OnResize;
use crate::term::SizeInfo; use crate::term::SizeInfo;
use crate::tty::windows::child::ChildExitWatcher; use crate::tty::windows::child::ChildExitWatcher;
use crate::tty::windows::Pty; use crate::tty::windows::{cmdline, Pty};
pub use winpty::Winpty as Agent; pub use winpty::Winpty as Agent;
@ -42,22 +41,15 @@ pub fn new<C>(config: &Config<C>, size: &SizeInfo, _window_id: Option<usize>) ->
let mut agent = Winpty::open(&wconfig).unwrap(); let mut agent = Winpty::open(&wconfig).unwrap();
let (conin, conout) = (agent.conin_name(), agent.conout_name()); let (conin, conout) = (agent.conin_name(), agent.conout_name());
// Get process commandline let cmdline = cmdline(&config);
let default_shell = &Shell::new("powershell"); let cwd = config.working_directory().map(|dir| dir.canonicalize().unwrap());
let shell = config.shell.as_ref().unwrap_or(default_shell);
let mut cmdline = shell.args.clone();
cmdline.insert(0, shell.program.to_string());
// Warning, here be borrow hell
let cwd = config.working_directory().as_ref().map(|dir| canonicalize(dir).unwrap());
let cwd = cwd.as_ref().map(|dir| dir.to_str().unwrap());
// Spawn process // Spawn process
let spawnconfig = SpawnConfig::new( let spawnconfig = SpawnConfig::new(
SpawnFlags::AUTO_SHUTDOWN | SpawnFlags::EXIT_AFTER_SHUTDOWN, SpawnFlags::AUTO_SHUTDOWN | SpawnFlags::EXIT_AFTER_SHUTDOWN,
None, // appname None, // appname
Some(&cmdline.join(" ")), Some(&cmdline),
cwd, cwd.as_ref().map(|p| p.as_ref()),
None, // Env None, // Env
) )
.unwrap(); .unwrap();

View file

@ -9,7 +9,6 @@ edition = "2018"
[target.'cfg(windows)'.dependencies] [target.'cfg(windows)'.dependencies]
winpty-sys = "0.4.3" winpty-sys = "0.4.3"
bitflags = "1.0" bitflags = "1.0"
widestring = "0.4.0"
[target.'cfg(windows)'.dev-dependencies] [target.'cfg(windows)'.dev-dependencies]
named_pipe = "0.4.1" named_pipe = "0.4.1"

View file

@ -1,13 +1,5 @@
#![deny(clippy::all, clippy::if_not_else, clippy::enum_glob_use, clippy::wrong_pub_self_convention)] #![deny(clippy::all, clippy::if_not_else, clippy::enum_glob_use, clippy::wrong_pub_self_convention)]
#[macro_use]
#[cfg(windows)]
extern crate bitflags;
#[cfg(windows)]
extern crate widestring;
#[cfg(windows)]
extern crate winpty_sys;
#[cfg(windows)] #[cfg(windows)]
pub mod windows; pub mod windows;

View file

@ -1,12 +1,15 @@
use std::ffi::{OsStr, OsString};
use std::fmt::{self, Display, Formatter}; use std::fmt::{self, Display, Formatter};
use std::iter::once;
use std::os::windows::ffi::{OsStrExt, OsStringExt};
use std::os::windows::io::RawHandle; use std::os::windows::io::RawHandle;
use std::path::PathBuf; use std::path::{Path, PathBuf};
use std::ptr::{null, null_mut}; use std::ptr::{null, null_mut};
use std::result::Result; use std::result::Result;
use winpty_sys::*; use bitflags::bitflags;
use widestring::WideCString; use winpty_sys::*;
#[derive(Copy, Clone, Debug)] #[derive(Copy, Clone, Debug)]
pub enum ErrorCode { pub enum ErrorCode {
@ -167,7 +170,7 @@ impl Winpty {
pub fn conin_name(&mut self) -> PathBuf { pub fn conin_name(&mut self) -> PathBuf {
unsafe { unsafe {
let raw = winpty_conin_name(self.0); let raw = winpty_conin_name(self.0);
PathBuf::from(&String::from_utf16_lossy(std::slice::from_raw_parts(raw, wcslen(raw)))) OsString::from_wide(std::slice::from_raw_parts(raw, wcslen(raw))).into()
} }
} }
@ -176,7 +179,7 @@ impl Winpty {
pub fn conout_name(&mut self) -> PathBuf { pub fn conout_name(&mut self) -> PathBuf {
unsafe { unsafe {
let raw = winpty_conout_name(self.0); let raw = winpty_conout_name(self.0);
PathBuf::from(&String::from_utf16_lossy(std::slice::from_raw_parts(raw, wcslen(raw)))) OsString::from_wide(std::slice::from_raw_parts(raw, wcslen(raw))).into()
} }
} }
@ -186,7 +189,7 @@ impl Winpty {
pub fn conerr_name(&mut self) -> PathBuf { pub fn conerr_name(&mut self) -> PathBuf {
unsafe { unsafe {
let raw = winpty_conerr_name(self.0); let raw = winpty_conerr_name(self.0);
PathBuf::from(&String::from_utf16_lossy(std::slice::from_raw_parts(raw, wcslen(raw)))) OsString::from_wide(std::slice::from_raw_parts(raw, wcslen(raw))).into()
} }
} }
@ -283,25 +286,28 @@ impl SpawnConfig {
spawnflags: SpawnFlags, spawnflags: SpawnFlags,
appname: Option<&str>, appname: Option<&str>,
cmdline: Option<&str>, cmdline: Option<&str>,
cwd: Option<&str>, cwd: Option<&Path>,
end: Option<&str>, env: Option<&str>,
) -> Result<Self, Error> { ) -> Result<Self, Error> {
let mut err = null_mut() as *mut winpty_error_t; let mut err = null_mut() as *mut winpty_error_t;
let to_wstring = |s| WideCString::from_str(s).unwrap(); fn to_wstring<S: AsRef<OsStr> + ?Sized>(s: &S) -> Vec<u16> {
OsStr::new(s).encode_wide().chain(once(0)).collect()
}
let appname = appname.map(to_wstring); let appname = appname.map(to_wstring);
let cmdline = cmdline.map(to_wstring); let cmdline = cmdline.map(to_wstring);
let cwd = cwd.map(to_wstring); let cwd = cwd.map(to_wstring);
let end = end.map(to_wstring); let env = env.map(to_wstring);
let wstring_ptr = |opt: &Option<WideCString>| opt.as_ref().map_or(null(), |ws| ws.as_ptr()); let wstring_ptr = |opt: &Option<Vec<u16>>| opt.as_ref().map_or(null(), |ws| ws.as_ptr());
let spawn_config = unsafe { let spawn_config = unsafe {
winpty_spawn_config_new( winpty_spawn_config_new(
spawnflags.bits(), spawnflags.bits(),
wstring_ptr(&appname), wstring_ptr(&appname),
wstring_ptr(&cmdline), wstring_ptr(&cmdline),
wstring_ptr(&cwd), wstring_ptr(&cwd),
wstring_ptr(&end), wstring_ptr(&env),
&mut err, &mut err,
) )
}; };