159fdb447SLoGin use core::{ 259fdb447SLoGin fmt::Debug, 3dfe53cf0SGnoCiYeH sync::atomic::{AtomicBool, AtomicUsize, Ordering}, 459fdb447SLoGin }; 552da9a59SGnoCiYeH 6dfe53cf0SGnoCiYeH use alloc::{ 7dfe53cf0SGnoCiYeH collections::LinkedList, 8dfe53cf0SGnoCiYeH string::String, 9dfe53cf0SGnoCiYeH sync::{Arc, Weak}, 10dfe53cf0SGnoCiYeH vec::Vec, 11dfe53cf0SGnoCiYeH }; 1252da9a59SGnoCiYeH use system_error::SystemError; 1352da9a59SGnoCiYeH 1452da9a59SGnoCiYeH use crate::{ 15*9365e801SGnoCiYeH driver::{serial::serial8250::send_to_default_serial8250_port, tty::pty::ptm_driver}, 1652da9a59SGnoCiYeH libs::{ 1752da9a59SGnoCiYeH rwlock::{RwLock, RwLockReadGuard, RwLockUpgradableGuard, RwLockWriteGuard}, 1852da9a59SGnoCiYeH spinlock::{SpinLock, SpinLockGuard}, 1952da9a59SGnoCiYeH wait_queue::EventWaitQueue, 2052da9a59SGnoCiYeH }, 2152da9a59SGnoCiYeH mm::VirtAddr, 2252bcb59eSGnoCiYeH net::event_poll::{EPollEventType, EPollItem}, 2352da9a59SGnoCiYeH process::Pid, 24be60c929SGnoCiYeH syscall::user_access::{UserBufferReader, UserBufferWriter}, 2552da9a59SGnoCiYeH }; 2652da9a59SGnoCiYeH 2752da9a59SGnoCiYeH use super::{ 2852da9a59SGnoCiYeH termios::{ControlMode, PosixTermios, Termios, TtySetTermiosOpt, WindowSize}, 2952da9a59SGnoCiYeH tty_driver::{TtyDriver, TtyDriverSubType, TtyDriverType, TtyOperation}, 3052da9a59SGnoCiYeH tty_ldisc::{ 3152da9a59SGnoCiYeH ntty::{NTtyData, NTtyLinediscipline}, 3252da9a59SGnoCiYeH TtyLineDiscipline, 3352da9a59SGnoCiYeH }, 3452da9a59SGnoCiYeH tty_port::TtyPort, 3552da9a59SGnoCiYeH virtual_terminal::{virtual_console::VirtualConsoleData, VIRT_CONSOLES}, 3652da9a59SGnoCiYeH }; 3752da9a59SGnoCiYeH 3852da9a59SGnoCiYeH #[derive(Debug)] 3952da9a59SGnoCiYeH pub struct TtyCore { 4052da9a59SGnoCiYeH core: TtyCoreData, 4152da9a59SGnoCiYeH /// 线路规程函数集 4252da9a59SGnoCiYeH line_discipline: Arc<dyn TtyLineDiscipline>, 4352da9a59SGnoCiYeH } 4452da9a59SGnoCiYeH 45*9365e801SGnoCiYeH impl Drop for TtyCore { 46*9365e801SGnoCiYeH fn drop(&mut self) { 47*9365e801SGnoCiYeH if self.core.driver().tty_driver_sub_type() == TtyDriverSubType::PtySlave { 48*9365e801SGnoCiYeH ptm_driver().ttys().remove(&self.core().index); 49*9365e801SGnoCiYeH } 50*9365e801SGnoCiYeH } 51*9365e801SGnoCiYeH } 52*9365e801SGnoCiYeH 5352da9a59SGnoCiYeH impl TtyCore { 5452da9a59SGnoCiYeH pub fn new(driver: Arc<TtyDriver>, index: usize) -> Arc<Self> { 5552da9a59SGnoCiYeH let name = driver.tty_line_name(index); 5652da9a59SGnoCiYeH let termios = driver.init_termios(); 5752da9a59SGnoCiYeH let core = TtyCoreData { 5852da9a59SGnoCiYeH tty_driver: driver, 5952da9a59SGnoCiYeH termios: RwLock::new(termios), 6052da9a59SGnoCiYeH name, 6152da9a59SGnoCiYeH flags: RwLock::new(TtyFlag::empty()), 6259fdb447SLoGin count: AtomicUsize::new(0), 6352da9a59SGnoCiYeH window_size: RwLock::new(WindowSize::default()), 6452da9a59SGnoCiYeH read_wq: EventWaitQueue::new(), 6552da9a59SGnoCiYeH write_wq: EventWaitQueue::new(), 6652da9a59SGnoCiYeH port: RwLock::new(None), 6752da9a59SGnoCiYeH index, 6852da9a59SGnoCiYeH ctrl: SpinLock::new(TtyContorlInfo::default()), 6952da9a59SGnoCiYeH closing: AtomicBool::new(false), 7052da9a59SGnoCiYeH flow: SpinLock::new(TtyFlowState::default()), 71dfe53cf0SGnoCiYeH link: RwLock::default(), 7252bcb59eSGnoCiYeH epitems: SpinLock::new(LinkedList::new()), 7352da9a59SGnoCiYeH }; 7452da9a59SGnoCiYeH 7552da9a59SGnoCiYeH return Arc::new(Self { 7652da9a59SGnoCiYeH core, 7752da9a59SGnoCiYeH line_discipline: Arc::new(NTtyLinediscipline { 7852da9a59SGnoCiYeH data: SpinLock::new(NTtyData::new()), 7952da9a59SGnoCiYeH }), 8052da9a59SGnoCiYeH }); 8152da9a59SGnoCiYeH } 8252da9a59SGnoCiYeH 8352da9a59SGnoCiYeH #[inline] 8452da9a59SGnoCiYeH pub fn core(&self) -> &TtyCoreData { 8552da9a59SGnoCiYeH return &self.core; 8652da9a59SGnoCiYeH } 8752da9a59SGnoCiYeH 8852da9a59SGnoCiYeH #[inline] 8952da9a59SGnoCiYeH pub fn ldisc(&self) -> Arc<dyn TtyLineDiscipline> { 9052da9a59SGnoCiYeH self.line_discipline.clone() 9152da9a59SGnoCiYeH } 9252da9a59SGnoCiYeH 93f3b05a97SGnoCiYeH pub fn write_without_serial(&self, buf: &[u8], nr: usize) -> Result<usize, SystemError> { 94f3b05a97SGnoCiYeH self.core 95f3b05a97SGnoCiYeH .driver() 96f3b05a97SGnoCiYeH .driver_funcs() 97f3b05a97SGnoCiYeH .write(self.core(), buf, nr) 98f3b05a97SGnoCiYeH } 99f3b05a97SGnoCiYeH 10052da9a59SGnoCiYeH pub fn reopen(&self) -> Result<(), SystemError> { 10152da9a59SGnoCiYeH let tty_core = self.core(); 10252da9a59SGnoCiYeH let driver = tty_core.driver(); 10352da9a59SGnoCiYeH 10452da9a59SGnoCiYeH if driver.tty_driver_type() == TtyDriverType::Pty 10552da9a59SGnoCiYeH && driver.tty_driver_sub_type() == TtyDriverSubType::PtyMaster 10652da9a59SGnoCiYeH { 10752da9a59SGnoCiYeH return Err(SystemError::EIO); 10852da9a59SGnoCiYeH } 10952da9a59SGnoCiYeH 11052da9a59SGnoCiYeH // if *tty_core.count.read() == 0 { 11152da9a59SGnoCiYeH // return Err(SystemError::EAGAIN_OR_EWOULDBLOCK); 11252da9a59SGnoCiYeH // } 11352da9a59SGnoCiYeH 11452da9a59SGnoCiYeH // TODO 判断flags 11552da9a59SGnoCiYeH 11652da9a59SGnoCiYeH tty_core.add_count(); 11752da9a59SGnoCiYeH 11852da9a59SGnoCiYeH Ok(()) 11952da9a59SGnoCiYeH } 12052da9a59SGnoCiYeH 12152da9a59SGnoCiYeH #[inline] 12252da9a59SGnoCiYeH pub fn set_port(&self, port: Arc<dyn TtyPort>) { 12352da9a59SGnoCiYeH *self.core.port.write() = Some(port); 12452da9a59SGnoCiYeH } 12552da9a59SGnoCiYeH 12652da9a59SGnoCiYeH pub fn tty_start(&self) { 12752da9a59SGnoCiYeH let mut flow = self.core.flow.lock_irqsave(); 12852da9a59SGnoCiYeH if !flow.stopped || flow.tco_stopped { 12952da9a59SGnoCiYeH return; 13052da9a59SGnoCiYeH } 13152da9a59SGnoCiYeH 13252da9a59SGnoCiYeH flow.stopped = false; 13352da9a59SGnoCiYeH let _ = self.start(self.core()); 13452da9a59SGnoCiYeH self.tty_wakeup(); 13552da9a59SGnoCiYeH } 13652da9a59SGnoCiYeH 13752da9a59SGnoCiYeH pub fn tty_stop(&self) { 13852da9a59SGnoCiYeH let mut flow = self.core.flow.lock_irqsave(); 13952da9a59SGnoCiYeH if flow.stopped { 14052da9a59SGnoCiYeH return; 14152da9a59SGnoCiYeH } 14252da9a59SGnoCiYeH flow.stopped = true; 14352da9a59SGnoCiYeH 14452da9a59SGnoCiYeH let _ = self.stop(self.core()); 14552da9a59SGnoCiYeH } 14652da9a59SGnoCiYeH 14752da9a59SGnoCiYeH pub fn tty_wakeup(&self) { 14859fdb447SLoGin if self.core.flags().contains(TtyFlag::DO_WRITE_WAKEUP) { 14952da9a59SGnoCiYeH let _ = self.ldisc().write_wakeup(self.core()); 15052da9a59SGnoCiYeH } 15152da9a59SGnoCiYeH 15252da9a59SGnoCiYeH self.core() 15352da9a59SGnoCiYeH .write_wq 154dfe53cf0SGnoCiYeH .wakeup_any(EPollEventType::EPOLLOUT.bits() as u64); 15552da9a59SGnoCiYeH } 15652da9a59SGnoCiYeH 157be60c929SGnoCiYeH pub fn tty_mode_ioctl(tty: Arc<TtyCore>, cmd: u32, arg: usize) -> Result<usize, SystemError> { 158be60c929SGnoCiYeH let core = tty.core(); 159b5b571e0SLoGin let real_tty = if core.driver().tty_driver_type() == TtyDriverType::Pty 160be60c929SGnoCiYeH && core.driver().tty_driver_sub_type() == TtyDriverSubType::PtyMaster 161be60c929SGnoCiYeH { 162b5b571e0SLoGin core.link().unwrap() 163be60c929SGnoCiYeH } else { 164b5b571e0SLoGin tty 165b5b571e0SLoGin }; 16652da9a59SGnoCiYeH match cmd { 16752da9a59SGnoCiYeH TtyIoctlCmd::TCGETS => { 168b5b571e0SLoGin let termios = PosixTermios::from_kernel_termios(*real_tty.core.termios()); 16952da9a59SGnoCiYeH let mut user_writer = UserBufferWriter::new( 17052da9a59SGnoCiYeH VirtAddr::new(arg).as_ptr::<PosixTermios>(), 17152da9a59SGnoCiYeH core::mem::size_of::<PosixTermios>(), 17252da9a59SGnoCiYeH true, 17352da9a59SGnoCiYeH )?; 17452da9a59SGnoCiYeH 17552da9a59SGnoCiYeH user_writer.copy_one_to_user(&termios, 0)?; 17652da9a59SGnoCiYeH return Ok(0); 17752da9a59SGnoCiYeH } 17852bcb59eSGnoCiYeH TtyIoctlCmd::TCSETS => { 17952bcb59eSGnoCiYeH return TtyCore::core_set_termios( 18052bcb59eSGnoCiYeH real_tty, 18152bcb59eSGnoCiYeH VirtAddr::new(arg), 18252bcb59eSGnoCiYeH TtySetTermiosOpt::TERMIOS_OLD, 18352bcb59eSGnoCiYeH ); 18452bcb59eSGnoCiYeH } 18552da9a59SGnoCiYeH TtyIoctlCmd::TCSETSW => { 186be60c929SGnoCiYeH return TtyCore::core_set_termios( 187be60c929SGnoCiYeH real_tty, 18852da9a59SGnoCiYeH VirtAddr::new(arg), 18952da9a59SGnoCiYeH TtySetTermiosOpt::TERMIOS_WAIT | TtySetTermiosOpt::TERMIOS_OLD, 19052da9a59SGnoCiYeH ); 19152da9a59SGnoCiYeH } 19252da9a59SGnoCiYeH _ => { 19352da9a59SGnoCiYeH return Err(SystemError::ENOIOCTLCMD); 19452da9a59SGnoCiYeH } 19552da9a59SGnoCiYeH } 19652da9a59SGnoCiYeH } 19752da9a59SGnoCiYeH 19852da9a59SGnoCiYeH pub fn core_set_termios( 19952da9a59SGnoCiYeH tty: Arc<TtyCore>, 20052da9a59SGnoCiYeH arg: VirtAddr, 20152da9a59SGnoCiYeH opt: TtySetTermiosOpt, 20252da9a59SGnoCiYeH ) -> Result<usize, SystemError> { 203be60c929SGnoCiYeH #[allow(unused_assignments)] 204be60c929SGnoCiYeH // TERMIOS_TERMIO下会用到 205b5b571e0SLoGin let mut tmp_termios = *tty.core().termios(); 20652da9a59SGnoCiYeH 20752da9a59SGnoCiYeH if opt.contains(TtySetTermiosOpt::TERMIOS_TERMIO) { 20852da9a59SGnoCiYeH todo!() 20952da9a59SGnoCiYeH } else { 210be60c929SGnoCiYeH let user_reader = UserBufferReader::new( 21152da9a59SGnoCiYeH arg.as_ptr::<PosixTermios>(), 21252da9a59SGnoCiYeH core::mem::size_of::<PosixTermios>(), 21352da9a59SGnoCiYeH true, 21452da9a59SGnoCiYeH )?; 21552da9a59SGnoCiYeH 216be60c929SGnoCiYeH let mut term = PosixTermios::default(); 217be60c929SGnoCiYeH user_reader.copy_one_from_user(&mut term, 0)?; 218be60c929SGnoCiYeH 219be60c929SGnoCiYeH tmp_termios = term.to_kernel_termios(); 22052da9a59SGnoCiYeH } 22152da9a59SGnoCiYeH 22252da9a59SGnoCiYeH if opt.contains(TtySetTermiosOpt::TERMIOS_FLUSH) { 223be60c929SGnoCiYeH let ld = tty.ldisc(); 22452da9a59SGnoCiYeH let _ = ld.flush_buffer(tty.clone()); 22552da9a59SGnoCiYeH } 22652da9a59SGnoCiYeH 22752da9a59SGnoCiYeH if opt.contains(TtySetTermiosOpt::TERMIOS_WAIT) { 22852da9a59SGnoCiYeH // TODO 22952da9a59SGnoCiYeH } 23052da9a59SGnoCiYeH 231be60c929SGnoCiYeH TtyCore::set_termios_next(tty, tmp_termios)?; 23252da9a59SGnoCiYeH Ok(0) 23352da9a59SGnoCiYeH } 23452da9a59SGnoCiYeH 235be60c929SGnoCiYeH pub fn set_termios_next(tty: Arc<TtyCore>, new_termios: Termios) -> Result<(), SystemError> { 236be60c929SGnoCiYeH let mut termios = tty.core().termios_write(); 23752da9a59SGnoCiYeH 238b5b571e0SLoGin let old_termios = *termios; 23952da9a59SGnoCiYeH *termios = new_termios; 24052da9a59SGnoCiYeH let tmp = termios.control_mode; 24152da9a59SGnoCiYeH termios.control_mode ^= (tmp ^ old_termios.control_mode) & ControlMode::ADDRB; 24252da9a59SGnoCiYeH 243*9365e801SGnoCiYeH drop(termios); 244be60c929SGnoCiYeH let ret = tty.set_termios(tty.clone(), old_termios); 245*9365e801SGnoCiYeH let mut termios = tty.core().termios_write(); 24652da9a59SGnoCiYeH if ret.is_err() { 24752da9a59SGnoCiYeH termios.control_mode &= ControlMode::HUPCL | ControlMode::CREAD | ControlMode::CLOCAL; 24852da9a59SGnoCiYeH termios.control_mode |= old_termios.control_mode 24952da9a59SGnoCiYeH & !(ControlMode::HUPCL | ControlMode::CREAD | ControlMode::CLOCAL); 25052da9a59SGnoCiYeH termios.input_speed = old_termios.input_speed; 25152da9a59SGnoCiYeH termios.output_speed = old_termios.output_speed; 25252da9a59SGnoCiYeH } 25352da9a59SGnoCiYeH 25452da9a59SGnoCiYeH drop(termios); 255be60c929SGnoCiYeH let ld = tty.ldisc(); 25652da9a59SGnoCiYeH ld.set_termios(tty, Some(old_termios))?; 25752da9a59SGnoCiYeH 25852da9a59SGnoCiYeH Ok(()) 25952da9a59SGnoCiYeH } 260*9365e801SGnoCiYeH 261*9365e801SGnoCiYeH pub fn tty_do_resize(&self, windowsize: WindowSize) -> Result<(), SystemError> { 262*9365e801SGnoCiYeH // TODO: 向前台进程发送信号 263*9365e801SGnoCiYeH *self.core.window_size_write() = windowsize; 264*9365e801SGnoCiYeH Ok(()) 265*9365e801SGnoCiYeH } 26652da9a59SGnoCiYeH } 26752da9a59SGnoCiYeH 268b5b571e0SLoGin #[derive(Debug, Default)] 26952da9a59SGnoCiYeH pub struct TtyContorlInfo { 27052da9a59SGnoCiYeH /// 前台进程pid 27152da9a59SGnoCiYeH pub session: Option<Pid>, 27252da9a59SGnoCiYeH /// 前台进程组id 27352da9a59SGnoCiYeH pub pgid: Option<Pid>, 27452da9a59SGnoCiYeH 27552da9a59SGnoCiYeH /// packet模式下使用,目前未用到 276dfe53cf0SGnoCiYeH pub pktstatus: TtyPacketStatus, 27752da9a59SGnoCiYeH pub packet: bool, 27852da9a59SGnoCiYeH } 27952da9a59SGnoCiYeH 28052da9a59SGnoCiYeH #[derive(Debug, Default)] 28152da9a59SGnoCiYeH pub struct TtyCoreWriteData { 28252da9a59SGnoCiYeH /// 写缓冲区 28352da9a59SGnoCiYeH pub write_buf: Vec<u8>, 28452da9a59SGnoCiYeH /// 写入数量 28552da9a59SGnoCiYeH pub write_cnt: usize, 28652da9a59SGnoCiYeH } 28752da9a59SGnoCiYeH 28852da9a59SGnoCiYeH #[derive(Debug, Default)] 28952da9a59SGnoCiYeH pub struct TtyFlowState { 29052da9a59SGnoCiYeH /// 表示流控是否被停止 29152da9a59SGnoCiYeH pub stopped: bool, 29252da9a59SGnoCiYeH /// 表示 TCO(Transmit Continuous Operation)流控是否被停止 29352da9a59SGnoCiYeH pub tco_stopped: bool, 29452da9a59SGnoCiYeH } 29552da9a59SGnoCiYeH 29652da9a59SGnoCiYeH #[derive(Debug)] 29752da9a59SGnoCiYeH pub struct TtyCoreData { 29852da9a59SGnoCiYeH tty_driver: Arc<TtyDriver>, 29952da9a59SGnoCiYeH termios: RwLock<Termios>, 30052da9a59SGnoCiYeH name: String, 30152da9a59SGnoCiYeH flags: RwLock<TtyFlag>, 30252da9a59SGnoCiYeH /// 在初始化时即确定不会更改,所以这里不用加锁 30352da9a59SGnoCiYeH index: usize, 30459fdb447SLoGin count: AtomicUsize, 30552da9a59SGnoCiYeH /// 窗口大小 30652da9a59SGnoCiYeH window_size: RwLock<WindowSize>, 30752da9a59SGnoCiYeH /// 读等待队列 30852da9a59SGnoCiYeH read_wq: EventWaitQueue, 30952da9a59SGnoCiYeH /// 写等待队列 31052da9a59SGnoCiYeH write_wq: EventWaitQueue, 31152da9a59SGnoCiYeH /// 端口 31252da9a59SGnoCiYeH port: RwLock<Option<Arc<dyn TtyPort>>>, 31352da9a59SGnoCiYeH /// 前台进程 31452da9a59SGnoCiYeH ctrl: SpinLock<TtyContorlInfo>, 31552da9a59SGnoCiYeH /// 是否正在关闭 31652da9a59SGnoCiYeH closing: AtomicBool, 31752da9a59SGnoCiYeH /// 流控状态 31852da9a59SGnoCiYeH flow: SpinLock<TtyFlowState>, 319be60c929SGnoCiYeH /// 链接tty 320dfe53cf0SGnoCiYeH link: RwLock<Weak<TtyCore>>, 32152bcb59eSGnoCiYeH /// epitems 32252bcb59eSGnoCiYeH epitems: SpinLock<LinkedList<Arc<EPollItem>>>, 32352da9a59SGnoCiYeH } 32452da9a59SGnoCiYeH 32552da9a59SGnoCiYeH impl TtyCoreData { 32652da9a59SGnoCiYeH #[inline] 32752da9a59SGnoCiYeH pub fn driver(&self) -> Arc<TtyDriver> { 32852da9a59SGnoCiYeH self.tty_driver.clone() 32952da9a59SGnoCiYeH } 33052da9a59SGnoCiYeH 33152da9a59SGnoCiYeH #[inline] 33252da9a59SGnoCiYeH pub fn flow_irqsave(&self) -> SpinLockGuard<TtyFlowState> { 33352da9a59SGnoCiYeH self.flow.lock_irqsave() 33452da9a59SGnoCiYeH } 33552da9a59SGnoCiYeH 33652da9a59SGnoCiYeH #[inline] 33752da9a59SGnoCiYeH pub fn port(&self) -> Option<Arc<dyn TtyPort>> { 33852da9a59SGnoCiYeH self.port.read().clone() 33952da9a59SGnoCiYeH } 34052da9a59SGnoCiYeH 34152da9a59SGnoCiYeH #[inline] 34252da9a59SGnoCiYeH pub fn index(&self) -> usize { 34352da9a59SGnoCiYeH self.index 34452da9a59SGnoCiYeH } 34552da9a59SGnoCiYeH 34652da9a59SGnoCiYeH #[inline] 34752da9a59SGnoCiYeH pub fn name(&self) -> String { 34852da9a59SGnoCiYeH self.name.clone() 34952da9a59SGnoCiYeH } 35052da9a59SGnoCiYeH 35152da9a59SGnoCiYeH #[inline] 35252da9a59SGnoCiYeH pub fn flags(&self) -> TtyFlag { 353b5b571e0SLoGin *self.flags.read_irqsave() 35452da9a59SGnoCiYeH } 35552da9a59SGnoCiYeH 35652da9a59SGnoCiYeH #[inline] 357dfe53cf0SGnoCiYeH pub fn flags_write(&self) -> RwLockWriteGuard<'_, TtyFlag> { 358dfe53cf0SGnoCiYeH self.flags.write_irqsave() 359dfe53cf0SGnoCiYeH } 360dfe53cf0SGnoCiYeH 361dfe53cf0SGnoCiYeH #[inline] 36252da9a59SGnoCiYeH pub fn termios(&self) -> RwLockReadGuard<'_, Termios> { 36352bcb59eSGnoCiYeH self.termios.read_irqsave() 36452da9a59SGnoCiYeH } 36552da9a59SGnoCiYeH 36652da9a59SGnoCiYeH #[inline] 36752da9a59SGnoCiYeH pub fn termios_write(&self) -> RwLockWriteGuard<Termios> { 36852bcb59eSGnoCiYeH self.termios.write_irqsave() 36952da9a59SGnoCiYeH } 37052da9a59SGnoCiYeH 37152da9a59SGnoCiYeH #[inline] 37252da9a59SGnoCiYeH pub fn set_termios(&self, termios: Termios) { 37359fdb447SLoGin let mut termios_guard = self.termios_write(); 37452da9a59SGnoCiYeH *termios_guard = termios; 37552da9a59SGnoCiYeH } 37652da9a59SGnoCiYeH 37752da9a59SGnoCiYeH #[inline] 378dfe53cf0SGnoCiYeH pub fn count(&self) -> usize { 379dfe53cf0SGnoCiYeH self.count.load(Ordering::SeqCst) 380dfe53cf0SGnoCiYeH } 381dfe53cf0SGnoCiYeH 382dfe53cf0SGnoCiYeH #[inline] 38352da9a59SGnoCiYeH pub fn add_count(&self) { 38459fdb447SLoGin self.count 38559fdb447SLoGin .fetch_add(1, core::sync::atomic::Ordering::SeqCst); 38652da9a59SGnoCiYeH } 38752da9a59SGnoCiYeH 38852da9a59SGnoCiYeH #[inline] 38952da9a59SGnoCiYeH pub fn read_wq(&self) -> &EventWaitQueue { 39052da9a59SGnoCiYeH &self.read_wq 39152da9a59SGnoCiYeH } 39252da9a59SGnoCiYeH 39352da9a59SGnoCiYeH #[inline] 39452da9a59SGnoCiYeH pub fn write_wq(&self) -> &EventWaitQueue { 39552da9a59SGnoCiYeH &self.write_wq 39652da9a59SGnoCiYeH } 39752da9a59SGnoCiYeH 39852da9a59SGnoCiYeH #[inline] 39952da9a59SGnoCiYeH pub fn contorl_info_irqsave(&self) -> SpinLockGuard<TtyContorlInfo> { 40052da9a59SGnoCiYeH self.ctrl.lock_irqsave() 40152da9a59SGnoCiYeH } 40252da9a59SGnoCiYeH 40352da9a59SGnoCiYeH #[inline] 40452da9a59SGnoCiYeH pub fn window_size_upgradeable(&self) -> RwLockUpgradableGuard<WindowSize> { 40552da9a59SGnoCiYeH self.window_size.upgradeable_read() 40652da9a59SGnoCiYeH } 40752da9a59SGnoCiYeH 40852da9a59SGnoCiYeH #[inline] 40952da9a59SGnoCiYeH pub fn window_size(&self) -> RwLockReadGuard<WindowSize> { 41052da9a59SGnoCiYeH self.window_size.read() 41152da9a59SGnoCiYeH } 41252da9a59SGnoCiYeH 41352da9a59SGnoCiYeH #[inline] 414*9365e801SGnoCiYeH pub fn window_size_write(&self) -> RwLockWriteGuard<WindowSize> { 415*9365e801SGnoCiYeH self.window_size.write() 416*9365e801SGnoCiYeH } 417*9365e801SGnoCiYeH 418*9365e801SGnoCiYeH #[inline] 41952da9a59SGnoCiYeH pub fn is_closing(&self) -> bool { 42052da9a59SGnoCiYeH self.closing.load(core::sync::atomic::Ordering::SeqCst) 42152da9a59SGnoCiYeH } 42252da9a59SGnoCiYeH 42352da9a59SGnoCiYeH #[inline] 42452da9a59SGnoCiYeH pub fn vc_data_irqsave(&self) -> SpinLockGuard<VirtualConsoleData> { 42552da9a59SGnoCiYeH VIRT_CONSOLES[self.index].lock_irqsave() 42652da9a59SGnoCiYeH } 427be60c929SGnoCiYeH 428be60c929SGnoCiYeH #[inline] 429be60c929SGnoCiYeH pub fn link(&self) -> Option<Arc<TtyCore>> { 430dfe53cf0SGnoCiYeH self.link.read().upgrade() 431dfe53cf0SGnoCiYeH } 432dfe53cf0SGnoCiYeH 433dfe53cf0SGnoCiYeH pub fn checked_link(&self) -> Result<Arc<TtyCore>, SystemError> { 434dfe53cf0SGnoCiYeH if let Some(link) = self.link() { 435dfe53cf0SGnoCiYeH return Ok(link); 436dfe53cf0SGnoCiYeH } 437dfe53cf0SGnoCiYeH return Err(SystemError::ENODEV); 438dfe53cf0SGnoCiYeH } 439dfe53cf0SGnoCiYeH 440dfe53cf0SGnoCiYeH pub fn set_link(&self, link: Weak<TtyCore>) { 441dfe53cf0SGnoCiYeH *self.link.write() = link; 442dfe53cf0SGnoCiYeH } 443dfe53cf0SGnoCiYeH 444dfe53cf0SGnoCiYeH pub fn init_termios(&self) { 445dfe53cf0SGnoCiYeH let tty_index = self.index(); 446dfe53cf0SGnoCiYeH let driver = self.driver(); 447dfe53cf0SGnoCiYeH // 初始化termios 448dfe53cf0SGnoCiYeH if !driver 449dfe53cf0SGnoCiYeH .flags() 450dfe53cf0SGnoCiYeH .contains(super::tty_driver::TtyDriverFlag::TTY_DRIVER_RESET_TERMIOS) 451dfe53cf0SGnoCiYeH { 452dfe53cf0SGnoCiYeH // 先查看是否有已经保存的termios 453dfe53cf0SGnoCiYeH if let Some(t) = driver.saved_termios().get(tty_index) { 454dfe53cf0SGnoCiYeH let mut termios = *t; 455dfe53cf0SGnoCiYeH termios.line = driver.init_termios().line; 456dfe53cf0SGnoCiYeH self.set_termios(termios); 457dfe53cf0SGnoCiYeH } 458dfe53cf0SGnoCiYeH } 459dfe53cf0SGnoCiYeH // TODO:设置termios波特率? 460be60c929SGnoCiYeH } 46152bcb59eSGnoCiYeH 46252bcb59eSGnoCiYeH #[inline] 46352bcb59eSGnoCiYeH pub fn add_epitem(&self, epitem: Arc<EPollItem>) { 46452bcb59eSGnoCiYeH self.epitems.lock().push_back(epitem) 46552bcb59eSGnoCiYeH } 46652da9a59SGnoCiYeH } 46752da9a59SGnoCiYeH 46852da9a59SGnoCiYeH /// TTY 核心接口,不同的tty需要各自实现这个trait 46952da9a59SGnoCiYeH pub trait TtyCoreFuncs: Debug + Send + Sync {} 47052da9a59SGnoCiYeH 47152da9a59SGnoCiYeH impl TtyOperation for TtyCore { 47252da9a59SGnoCiYeH #[inline] 47352da9a59SGnoCiYeH fn open(&self, tty: &TtyCoreData) -> Result<(), SystemError> { 47452da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().open(tty); 47552da9a59SGnoCiYeH } 47652da9a59SGnoCiYeH 47752da9a59SGnoCiYeH #[inline] 47852da9a59SGnoCiYeH fn write_room(&self, tty: &TtyCoreData) -> usize { 47952da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().write_room(tty); 48052da9a59SGnoCiYeH } 48152da9a59SGnoCiYeH 48252da9a59SGnoCiYeH #[inline] 48352da9a59SGnoCiYeH fn write(&self, tty: &TtyCoreData, buf: &[u8], nr: usize) -> Result<usize, SystemError> { 484f3b05a97SGnoCiYeH send_to_default_serial8250_port(buf); 48552da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().write(tty, buf, nr); 48652da9a59SGnoCiYeH } 48752da9a59SGnoCiYeH 48852da9a59SGnoCiYeH #[inline] 48952da9a59SGnoCiYeH fn flush_chars(&self, tty: &TtyCoreData) { 49052da9a59SGnoCiYeH self.core().tty_driver.driver_funcs().flush_chars(tty); 49152da9a59SGnoCiYeH } 49252da9a59SGnoCiYeH 49352da9a59SGnoCiYeH #[inline] 49452da9a59SGnoCiYeH fn put_char(&self, tty: &TtyCoreData, ch: u8) -> Result<(), SystemError> { 49552da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().put_char(tty, ch); 49652da9a59SGnoCiYeH } 49752da9a59SGnoCiYeH 49852da9a59SGnoCiYeH #[inline] 49952da9a59SGnoCiYeH fn install(&self, driver: Arc<TtyDriver>, tty: Arc<TtyCore>) -> Result<(), SystemError> { 50052da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().install(driver, tty); 50152da9a59SGnoCiYeH } 50252da9a59SGnoCiYeH 50352da9a59SGnoCiYeH #[inline] 50452da9a59SGnoCiYeH fn start(&self, tty: &TtyCoreData) -> Result<(), SystemError> { 50552da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().start(tty); 50652da9a59SGnoCiYeH } 50752da9a59SGnoCiYeH 50852da9a59SGnoCiYeH #[inline] 50952da9a59SGnoCiYeH fn stop(&self, tty: &TtyCoreData) -> Result<(), SystemError> { 51052da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().stop(tty); 51152da9a59SGnoCiYeH } 51252da9a59SGnoCiYeH 51352da9a59SGnoCiYeH #[inline] 51452da9a59SGnoCiYeH fn ioctl(&self, tty: Arc<TtyCore>, cmd: u32, arg: usize) -> Result<(), SystemError> { 51552da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().ioctl(tty, cmd, arg); 51652da9a59SGnoCiYeH } 51752da9a59SGnoCiYeH 51852da9a59SGnoCiYeH #[inline] 51952da9a59SGnoCiYeH fn chars_in_buffer(&self) -> usize { 52052da9a59SGnoCiYeH return self.core().tty_driver.driver_funcs().chars_in_buffer(); 52152da9a59SGnoCiYeH } 52252da9a59SGnoCiYeH 52352da9a59SGnoCiYeH #[inline] 52452da9a59SGnoCiYeH fn set_termios(&self, tty: Arc<TtyCore>, old_termios: Termios) -> Result<(), SystemError> { 52552da9a59SGnoCiYeH return self 52652da9a59SGnoCiYeH .core() 52752da9a59SGnoCiYeH .tty_driver 52852da9a59SGnoCiYeH .driver_funcs() 52952da9a59SGnoCiYeH .set_termios(tty, old_termios); 53052da9a59SGnoCiYeH } 531dfe53cf0SGnoCiYeH 532dfe53cf0SGnoCiYeH fn close(&self, tty: Arc<TtyCore>) -> Result<(), SystemError> { 533dfe53cf0SGnoCiYeH self.core().tty_driver.driver_funcs().close(tty) 534dfe53cf0SGnoCiYeH } 535*9365e801SGnoCiYeH 536*9365e801SGnoCiYeH fn resize(&self, tty: Arc<TtyCore>, winsize: WindowSize) -> Result<(), SystemError> { 537*9365e801SGnoCiYeH self.core.tty_driver.driver_funcs().resize(tty, winsize) 538*9365e801SGnoCiYeH } 53952da9a59SGnoCiYeH } 54052da9a59SGnoCiYeH 54152da9a59SGnoCiYeH bitflags! { 54252da9a59SGnoCiYeH pub struct TtyFlag: u32 { 54352da9a59SGnoCiYeH /// 终端被节流 54452da9a59SGnoCiYeH const THROTTLED = 1 << 0; 54552da9a59SGnoCiYeH /// 终端输入输出错误状态 54652da9a59SGnoCiYeH const IO_ERROR = 1 << 1; 54752da9a59SGnoCiYeH /// 终端的其他一方已关闭 54852da9a59SGnoCiYeH const OTHER_CLOSED = 1 << 2; 54952da9a59SGnoCiYeH /// 终端处于独占状态 55052da9a59SGnoCiYeH const EXCLUSIVE = 1 << 3; 55152da9a59SGnoCiYeH /// 终端执行写唤醒操作 55252da9a59SGnoCiYeH const DO_WRITE_WAKEUP = 1 << 5; 55352da9a59SGnoCiYeH /// 终端线路驱动程序已打开 55452da9a59SGnoCiYeH const LDISC_OPEN = 1 << 11; 55552da9a59SGnoCiYeH /// 终端伪终端设备已锁定 55652da9a59SGnoCiYeH const PTY_LOCK = 1 << 16; 55752da9a59SGnoCiYeH /// 终端禁用写分裂操作 55852da9a59SGnoCiYeH const NO_WRITE_SPLIT = 1 << 17; 55952da9a59SGnoCiYeH /// 终端挂断(挂起)状态 56052da9a59SGnoCiYeH const HUPPED = 1 << 18; 56152da9a59SGnoCiYeH /// 终端正在挂断(挂起) 56252da9a59SGnoCiYeH const HUPPING = 1 << 19; 56352da9a59SGnoCiYeH /// 终端线路驱动程序正在更改 56452da9a59SGnoCiYeH const LDISC_CHANGING = 1 << 20; 56552da9a59SGnoCiYeH /// 终端线路驱动程序已停止 56652da9a59SGnoCiYeH const LDISC_HALTED = 1 << 22; 56752da9a59SGnoCiYeH } 568dfe53cf0SGnoCiYeH 569dfe53cf0SGnoCiYeH #[derive(Default)] 570dfe53cf0SGnoCiYeH pub struct TtyPacketStatus: u8 { 571dfe53cf0SGnoCiYeH /* Used for packet mode */ 572dfe53cf0SGnoCiYeH const TIOCPKT_DATA = 0; 573dfe53cf0SGnoCiYeH const TIOCPKT_FLUSHREAD = 1; 574dfe53cf0SGnoCiYeH const TIOCPKT_FLUSHWRITE = 2; 575dfe53cf0SGnoCiYeH const TIOCPKT_STOP = 4; 576dfe53cf0SGnoCiYeH const TIOCPKT_START = 8; 577dfe53cf0SGnoCiYeH const TIOCPKT_NOSTOP = 16; 578dfe53cf0SGnoCiYeH const TIOCPKT_DOSTOP = 32; 579dfe53cf0SGnoCiYeH const TIOCPKT_IOCTL = 64; 580dfe53cf0SGnoCiYeH } 58152da9a59SGnoCiYeH } 58252da9a59SGnoCiYeH 58352da9a59SGnoCiYeH #[derive(Debug, PartialEq)] 58452da9a59SGnoCiYeH pub enum EchoOperation { 58552da9a59SGnoCiYeH /// 开始特殊操作。 58652da9a59SGnoCiYeH Start, 58752da9a59SGnoCiYeH /// 向后移动光标列。 58852da9a59SGnoCiYeH MoveBackCol, 58952da9a59SGnoCiYeH /// 设置规范模式下的列位置。 59052da9a59SGnoCiYeH SetCanonCol, 59152da9a59SGnoCiYeH /// 擦除制表符。 59252da9a59SGnoCiYeH EraseTab, 59352da9a59SGnoCiYeH 59452da9a59SGnoCiYeH Undefined(u8), 59552da9a59SGnoCiYeH } 59652da9a59SGnoCiYeH 59752da9a59SGnoCiYeH impl EchoOperation { 59852da9a59SGnoCiYeH pub fn from_u8(num: u8) -> EchoOperation { 59952da9a59SGnoCiYeH match num { 60052da9a59SGnoCiYeH 0xff => Self::Start, 60152da9a59SGnoCiYeH 0x80 => Self::MoveBackCol, 60252da9a59SGnoCiYeH 0x81 => Self::SetCanonCol, 60352da9a59SGnoCiYeH 0x82 => Self::EraseTab, 60452da9a59SGnoCiYeH _ => Self::Undefined(num), 60552da9a59SGnoCiYeH } 60652da9a59SGnoCiYeH } 60752da9a59SGnoCiYeH 60852da9a59SGnoCiYeH pub fn to_u8(&self) -> u8 { 60952da9a59SGnoCiYeH match *self { 61052da9a59SGnoCiYeH EchoOperation::Start => 0xff, 61152da9a59SGnoCiYeH EchoOperation::MoveBackCol => 0x80, 61252da9a59SGnoCiYeH EchoOperation::SetCanonCol => 0x81, 61352da9a59SGnoCiYeH EchoOperation::EraseTab => 0x82, 61452da9a59SGnoCiYeH EchoOperation::Undefined(num) => num, 61552da9a59SGnoCiYeH } 61652da9a59SGnoCiYeH } 61752da9a59SGnoCiYeH } 61852da9a59SGnoCiYeH 61952da9a59SGnoCiYeH pub struct TtyIoctlCmd; 62052da9a59SGnoCiYeH 62152da9a59SGnoCiYeH #[allow(dead_code)] 62252da9a59SGnoCiYeH impl TtyIoctlCmd { 62352da9a59SGnoCiYeH /// 获取终端参数 62452da9a59SGnoCiYeH pub const TCGETS: u32 = 0x5401; 62552da9a59SGnoCiYeH /// 设置终端参数 62652da9a59SGnoCiYeH pub const TCSETS: u32 = 0x5402; 62752da9a59SGnoCiYeH /// 设置终端参数并等待所有输出完成 62852da9a59SGnoCiYeH pub const TCSETSW: u32 = 0x5403; 62952da9a59SGnoCiYeH /// 设置终端参数并且等待所有输出完成,但在这之前将终端清空 63052da9a59SGnoCiYeH pub const TCSETSF: u32 = 0x5404; 63152da9a59SGnoCiYeH /// 获取终端参数 63252da9a59SGnoCiYeH pub const TCGETA: u32 = 0x5405; 63352da9a59SGnoCiYeH /// 设置终端参数 63452da9a59SGnoCiYeH pub const TCSETA: u32 = 0x5406; 63552da9a59SGnoCiYeH /// 设置终端参数并等待所有输出完成 63652da9a59SGnoCiYeH pub const TCSETAW: u32 = 0x5407; 63752da9a59SGnoCiYeH /// 设置终端参数并且等待所有输出完成,但在这之前将终端清空 63852da9a59SGnoCiYeH pub const TCSETAF: u32 = 0x5408; 63952da9a59SGnoCiYeH /// 发送零字节,等待所有输出完成 64052da9a59SGnoCiYeH pub const TCSBRK: u32 = 0x5409; 64152da9a59SGnoCiYeH /// 控制终端的流控 64252da9a59SGnoCiYeH pub const TCXONC: u32 = 0x540A; 64352da9a59SGnoCiYeH /// 刷新输入/输出缓冲区或者丢弃输入缓冲区 64452da9a59SGnoCiYeH pub const TCFLSH: u32 = 0x540B; 64552da9a59SGnoCiYeH /// 设置设备为独占模式 64652da9a59SGnoCiYeH pub const TIOCEXCL: u32 = 0x540C; 64752da9a59SGnoCiYeH /// 设置设备为非独占模式 64852da9a59SGnoCiYeH pub const TIOCNXCL: u32 = 0x540D; 64952da9a59SGnoCiYeH /// 设置当前进程的控制终端 65052da9a59SGnoCiYeH pub const TIOCSCTTY: u32 = 0x540E; 65152da9a59SGnoCiYeH /// 获取前台进程组 65252da9a59SGnoCiYeH pub const TIOCGPGRP: u32 = 0x540F; 65352da9a59SGnoCiYeH ///设置前台进程组 65452da9a59SGnoCiYeH pub const TIOCSPGRP: u32 = 0x5410; 65552da9a59SGnoCiYeH /// 获取输出队列的字节数 65652da9a59SGnoCiYeH pub const TIOCOUTQ: u32 = 0x5411; 65752da9a59SGnoCiYeH /// 模拟从终端输入字符 65852da9a59SGnoCiYeH pub const TIOCSTI: u32 = 0x5412; 65952da9a59SGnoCiYeH /// 获取窗口大小 66052da9a59SGnoCiYeH pub const TIOCGWINSZ: u32 = 0x5413; 66152da9a59SGnoCiYeH /// 设置窗口大小 66252da9a59SGnoCiYeH pub const TIOCSWINSZ: u32 = 0x5414; 66352da9a59SGnoCiYeH /// 获取终端控制信号的状态 66452da9a59SGnoCiYeH pub const TIOCMGET: u32 = 0x5415; 66552da9a59SGnoCiYeH /// 设置终端控制信号的位 66652da9a59SGnoCiYeH pub const TIOCMBIS: u32 = 0x5416; 66752da9a59SGnoCiYeH /// 清除终端控制信号的位 66852da9a59SGnoCiYeH pub const TIOCMBIC: u32 = 0x5417; 66952da9a59SGnoCiYeH /// 设置终端控制信号的状态 67052da9a59SGnoCiYeH pub const TIOCMSET: u32 = 0x5418; 67152da9a59SGnoCiYeH /// 获取软件载波状态 67252da9a59SGnoCiYeH pub const TIOCGSOFTCAR: u32 = 0x5419; 67352da9a59SGnoCiYeH /// 设置软件载波状态 67452da9a59SGnoCiYeH pub const TIOCSSOFTCAR: u32 = 0x541A; 67552da9a59SGnoCiYeH /// 获取输入队列的字节数 67652da9a59SGnoCiYeH pub const FIONREAD: u32 = 0x541B; 67752da9a59SGnoCiYeH /// Linux 特有命令 67852da9a59SGnoCiYeH pub const TIOCLINUX: u32 = 0x541C; 67952da9a59SGnoCiYeH /// 获取控制台设备 68052da9a59SGnoCiYeH pub const TIOCCONS: u32 = 0x541D; 68152da9a59SGnoCiYeH /// 获取串行设备参数 68252da9a59SGnoCiYeH pub const TIOCGSERIAL: u32 = 0x541E; 68352da9a59SGnoCiYeH /// 设置串行设备参数 68452da9a59SGnoCiYeH pub const TIOCSSERIAL: u32 = 0x541F; 68552da9a59SGnoCiYeH /// 设置套接字的报文模式 68652da9a59SGnoCiYeH pub const TIOCPKT: u32 = 0x5420; 68752da9a59SGnoCiYeH /// 设置非阻塞 I/O 68852da9a59SGnoCiYeH pub const FIONBIO: u32 = 0x5421; 68952da9a59SGnoCiYeH /// 清除控制终端 69052da9a59SGnoCiYeH pub const TIOCNOTTY: u32 = 0x5422; 69152da9a59SGnoCiYeH /// 设置终端线路驱动器 69252da9a59SGnoCiYeH pub const TIOCSETD: u32 = 0x5423; 69352da9a59SGnoCiYeH /// 获取终端线路驱动器 69452da9a59SGnoCiYeH pub const TIOCGETD: u32 = 0x5424; 69552da9a59SGnoCiYeH /// 发送终止条件 69652da9a59SGnoCiYeH pub const TCSBRKP: u32 = 0x5425; 69752da9a59SGnoCiYeH /// 开始发送零比特 69852da9a59SGnoCiYeH pub const TIOCSBRK: u32 = 0x5427; 69952da9a59SGnoCiYeH /// 停止发送零比特 70052da9a59SGnoCiYeH pub const TIOCCBRK: u32 = 0x5428; 70152da9a59SGnoCiYeH /// Return the session ID of FD 70252da9a59SGnoCiYeH pub const TIOCGSID: u32 = 0x5429; 703dfe53cf0SGnoCiYeH /// 设置ptl锁标记 704dfe53cf0SGnoCiYeH pub const TIOCSPTLCK: u32 = 0x40045431; 705dfe53cf0SGnoCiYeH /// 获取ptl锁标记 706dfe53cf0SGnoCiYeH pub const TIOCGPTLCK: u32 = 0x80045439; 707dfe53cf0SGnoCiYeH /// 获取packet标记 708dfe53cf0SGnoCiYeH pub const TIOCGPKT: u32 = 0x80045438; 709dfe53cf0SGnoCiYeH /// 获取pts index 710dfe53cf0SGnoCiYeH pub const TIOCGPTN: u32 = 0x80045430; 71152da9a59SGnoCiYeH } 712