use std::fs::OpenOptions; use std::os::windows::fs::OpenOptionsExt; use std::os::windows::io::{FromRawHandle, IntoRawHandle}; use std::u16; use log::info; use mio_named_pipes::NamedPipe; use winapi::um::winbase::FILE_FLAG_OVERLAPPED; use winpty::{Config as WinptyConfig, ConfigFlags, MouseMode, SpawnConfig, SpawnFlags, Winpty}; use crate::config::Config; use crate::event::OnResize; use crate::term::SizeInfo; use crate::tty::windows::child::ChildExitWatcher; use crate::tty::windows::{cmdline, Pty}; pub use winpty::Winpty as Agent; pub fn new(config: &Config, size: &SizeInfo, _window_id: Option) -> Pty { // Create config. let mut wconfig = WinptyConfig::new(ConfigFlags::empty()).unwrap(); wconfig.set_initial_size(size.cols().0 as i32, size.screen_lines().0 as i32); wconfig.set_mouse_mode(&MouseMode::Auto); // Start agent. let mut agent = Winpty::open(&wconfig).unwrap(); let (conin, conout) = (agent.conin_name(), agent.conout_name()); let cmdline = cmdline(&config); // Spawn process. let spawnconfig = SpawnConfig::new( SpawnFlags::AUTO_SHUTDOWN | SpawnFlags::EXIT_AFTER_SHUTDOWN, None, // appname. Some(&cmdline), config.working_directory.as_deref(), None, // Env. ) .unwrap(); let default_opts = &mut OpenOptions::new(); default_opts.share_mode(0).custom_flags(FILE_FLAG_OVERLAPPED); let (conout_pipe, conin_pipe); unsafe { conout_pipe = NamedPipe::from_raw_handle( default_opts.clone().read(true).open(conout).unwrap().into_raw_handle(), ); conin_pipe = NamedPipe::from_raw_handle( default_opts.clone().write(true).open(conin).unwrap().into_raw_handle(), ); }; agent.spawn(&spawnconfig).unwrap(); let child_watcher = ChildExitWatcher::new(agent.raw_handle()).unwrap(); Pty::new(agent, conout_pipe, conin_pipe, child_watcher) } impl OnResize for Agent { fn on_resize(&mut self, size: &SizeInfo) { let (cols, lines) = (size.cols().0, size.screen_lines().0); if cols > 0 && cols <= u16::MAX as usize && lines > 0 && lines <= u16::MAX as usize { self.set_size(cols as u16, lines as u16) .unwrap_or_else(|_| info!("Unable to set WinPTY size, did it die?")); } } }