use core::{ fmt::Debug, sync::atomic::{AtomicBool, AtomicUsize, Ordering}, }; use alloc::{ collections::LinkedList, string::String, sync::{Arc, Weak}, }; use system_error::SystemError; use crate::{ driver::{base::device::device_number::DeviceNumber, tty::pty::ptm_driver}, libs::{ rwlock::{RwLock, RwLockReadGuard, RwLockUpgradableGuard, RwLockWriteGuard}, spinlock::{SpinLock, SpinLockGuard}, wait_queue::EventWaitQueue, }, mm::VirtAddr, net::event_poll::{EPollEventType, EPollItem}, process::Pid, syscall::user_access::{UserBufferReader, UserBufferWriter}, }; use super::{ termios::{ControlMode, PosixTermios, Termios, TtySetTermiosOpt, WindowSize}, tty_driver::{TtyCorePrivateField, TtyDriver, TtyDriverSubType, TtyDriverType, TtyOperation}, tty_ldisc::{ ntty::{NTtyData, NTtyLinediscipline}, TtyLineDiscipline, }, tty_port::TtyPort, virtual_terminal::{vc_manager, virtual_console::VirtualConsoleData, DrawRegion}, }; #[derive(Debug)] pub struct TtyCore { core: TtyCoreData, /// 线路规程函数集 line_discipline: Arc, } impl Drop for TtyCore { fn drop(&mut self) { if self.core.driver().tty_driver_sub_type() == TtyDriverSubType::PtySlave { ptm_driver().ttys().remove(&self.core().index); } } } impl TtyCore { pub fn new(driver: Arc, index: usize) -> Arc { let name = driver.tty_line_name(index); let device_number = driver .device_number(index) .expect("Get tty device number failed."); let termios = driver.init_termios(); let core = TtyCoreData { tty_driver: driver, termios: RwLock::new(termios), name, flags: RwLock::new(TtyFlag::empty()), count: AtomicUsize::new(0), window_size: RwLock::new(WindowSize::default()), read_wq: EventWaitQueue::new(), write_wq: EventWaitQueue::new(), port: RwLock::new(None), index, vc_index: AtomicUsize::new(usize::MAX), ctrl: SpinLock::new(TtyContorlInfo::default()), closing: AtomicBool::new(false), flow: SpinLock::new(TtyFlowState::default()), link: RwLock::default(), epitems: SpinLock::new(LinkedList::new()), device_number, privete_fields: SpinLock::new(None), }; return Arc::new(Self { core, line_discipline: Arc::new(NTtyLinediscipline { data: SpinLock::new(NTtyData::new()), }), }); } #[inline] pub fn core(&self) -> &TtyCoreData { return &self.core; } pub fn private_fields(&self) -> Option> { self.core.privete_fields.lock().clone() } pub fn set_private_fields(&self, fields: Arc) { *self.core.privete_fields.lock() = Some(fields); } #[inline] pub fn ldisc(&self) -> Arc { self.line_discipline.clone() } pub fn write_to_core(&self, buf: &[u8], nr: usize) -> Result { self.core .driver() .driver_funcs() .write(self.core(), buf, nr) } pub fn reopen(&self) -> Result<(), SystemError> { let tty_core = self.core(); let driver = tty_core.driver(); if driver.tty_driver_type() == TtyDriverType::Pty && driver.tty_driver_sub_type() == TtyDriverSubType::PtyMaster { return Err(SystemError::EIO); } // if *tty_core.count.read() == 0 { // return Err(SystemError::EAGAIN_OR_EWOULDBLOCK); // } // TODO 判断flags tty_core.add_count(); Ok(()) } #[inline] pub fn set_port(&self, port: Arc) { *self.core.port.write() = Some(port); } pub fn tty_start(&self) { let mut flow = self.core.flow.lock_irqsave(); if !flow.stopped || flow.tco_stopped { return; } flow.stopped = false; let _ = self.start(self.core()); self.tty_wakeup(); } pub fn tty_stop(&self) { let mut flow = self.core.flow.lock_irqsave(); if flow.stopped { return; } flow.stopped = true; let _ = self.stop(self.core()); } pub fn tty_wakeup(&self) { if self.core.flags().contains(TtyFlag::DO_WRITE_WAKEUP) { let _ = self.ldisc().write_wakeup(self.core()); } self.core() .write_wq .wakeup_any(EPollEventType::EPOLLOUT.bits() as u64); } pub fn tty_mode_ioctl(tty: Arc, cmd: u32, arg: usize) -> Result { let core = tty.core(); let real_tty = if core.driver().tty_driver_type() == TtyDriverType::Pty && core.driver().tty_driver_sub_type() == TtyDriverSubType::PtyMaster { core.link().unwrap() } else { tty }; match cmd { TtyIoctlCmd::TCGETS => { let termios = PosixTermios::from_kernel_termios(*real_tty.core.termios()); let mut user_writer = UserBufferWriter::new( VirtAddr::new(arg).as_ptr::(), core::mem::size_of::(), true, )?; user_writer.copy_one_to_user(&termios, 0)?; return Ok(0); } TtyIoctlCmd::TCSETS => { return TtyCore::core_set_termios( real_tty, VirtAddr::new(arg), TtySetTermiosOpt::TERMIOS_OLD, ); } TtyIoctlCmd::TCSETSW => { return TtyCore::core_set_termios( real_tty, VirtAddr::new(arg), TtySetTermiosOpt::TERMIOS_WAIT | TtySetTermiosOpt::TERMIOS_OLD, ); } _ => { return Err(SystemError::ENOIOCTLCMD); } } } pub fn core_set_termios( tty: Arc, arg: VirtAddr, opt: TtySetTermiosOpt, ) -> Result { #[allow(unused_assignments)] // TERMIOS_TERMIO下会用到 let mut tmp_termios = *tty.core().termios(); if opt.contains(TtySetTermiosOpt::TERMIOS_TERMIO) { todo!() } else { let user_reader = UserBufferReader::new( arg.as_ptr::(), core::mem::size_of::(), true, )?; let mut term = PosixTermios::default(); user_reader.copy_one_from_user(&mut term, 0)?; tmp_termios = term.to_kernel_termios(); } if opt.contains(TtySetTermiosOpt::TERMIOS_FLUSH) { let ld = tty.ldisc(); let _ = ld.flush_buffer(tty.clone()); } if opt.contains(TtySetTermiosOpt::TERMIOS_WAIT) { // TODO } TtyCore::set_termios_next(tty, tmp_termios)?; Ok(0) } fn set_termios_next(tty: Arc, new_termios: Termios) -> Result<(), SystemError> { let mut termios = tty.core().termios_write(); let old_termios = *termios; *termios = new_termios; let tmp = termios.control_mode; termios.control_mode ^= (tmp ^ old_termios.control_mode) & ControlMode::ADDRB; drop(termios); let ret = tty.set_termios(tty.clone(), old_termios); let mut termios = tty.core().termios_write(); if ret.is_err() { termios.control_mode &= ControlMode::HUPCL | ControlMode::CREAD | ControlMode::CLOCAL; termios.control_mode |= old_termios.control_mode & !(ControlMode::HUPCL | ControlMode::CREAD | ControlMode::CLOCAL); termios.input_speed = old_termios.input_speed; termios.output_speed = old_termios.output_speed; } drop(termios); let ld = tty.ldisc(); ld.set_termios(tty, Some(old_termios)).ok(); Ok(()) } pub fn tty_do_resize(&self, windowsize: WindowSize) -> Result<(), SystemError> { // TODO: 向前台进程发送信号 *self.core.window_size_write() = windowsize; Ok(()) } } #[derive(Debug, Default)] pub struct TtyContorlInfo { /// 前台进程pid pub session: Option, /// 前台进程组id pub pgid: Option, /// packet模式下使用,目前未用到 pub pktstatus: TtyPacketStatus, pub packet: bool, } #[derive(Debug, Default)] pub struct TtyFlowState { /// 表示流控是否被停止 pub stopped: bool, /// 表示 TCO(Transmit Continuous Operation)流控是否被停止 pub tco_stopped: bool, } #[derive(Debug)] pub struct TtyCoreData { tty_driver: Arc, termios: RwLock, name: String, flags: RwLock, /// 在初始化时即确定不会更改,所以这里不用加锁 index: usize, vc_index: AtomicUsize, count: AtomicUsize, /// 窗口大小 window_size: RwLock, /// 读等待队列 read_wq: EventWaitQueue, /// 写等待队列 write_wq: EventWaitQueue, /// 端口 port: RwLock>>, /// 前台进程 ctrl: SpinLock, /// 是否正在关闭 closing: AtomicBool, /// 流控状态 flow: SpinLock, /// 链接tty link: RwLock>, /// epitems epitems: SpinLock>>, /// 设备号 device_number: DeviceNumber, privete_fields: SpinLock>>, } impl TtyCoreData { #[inline] pub fn driver(&self) -> &Arc { &self.tty_driver } #[inline] pub fn flow_irqsave(&self) -> SpinLockGuard { self.flow.lock_irqsave() } #[inline] pub fn port(&self) -> Option> { self.port.read().clone() } #[inline] pub fn index(&self) -> usize { self.index } #[inline] pub fn name(&self) -> &String { &self.name } pub fn device_number(&self) -> &DeviceNumber { &self.device_number } #[inline] pub fn flags(&self) -> TtyFlag { *self.flags.read_irqsave() } #[inline] pub fn flags_write(&self) -> RwLockWriteGuard<'_, TtyFlag> { self.flags.write_irqsave() } #[inline] pub fn termios(&self) -> RwLockReadGuard<'_, Termios> { self.termios.read_irqsave() } #[inline] pub fn termios_write(&self) -> RwLockWriteGuard { self.termios.write_irqsave() } #[inline] pub fn set_termios(&self, termios: Termios) { let mut termios_guard = self.termios_write(); *termios_guard = termios; } #[inline] pub fn count(&self) -> usize { self.count.load(Ordering::SeqCst) } #[inline] pub fn add_count(&self) { self.count .fetch_add(1, core::sync::atomic::Ordering::SeqCst); } #[inline] pub fn read_wq(&self) -> &EventWaitQueue { &self.read_wq } #[inline] pub fn write_wq(&self) -> &EventWaitQueue { &self.write_wq } #[inline] pub fn contorl_info_irqsave(&self) -> SpinLockGuard { self.ctrl.lock_irqsave() } #[inline] pub fn window_size_upgradeable(&self) -> RwLockUpgradableGuard { self.window_size.upgradeable_read() } #[inline] pub fn window_size(&self) -> RwLockReadGuard { self.window_size.read() } #[inline] pub fn window_size_write(&self) -> RwLockWriteGuard { self.window_size.write() } #[inline] pub fn is_closing(&self) -> bool { self.closing.load(core::sync::atomic::Ordering::SeqCst) } #[inline] pub fn vc_data(&self) -> Option>> { vc_manager().get(self.vc_index()?).unwrap().vc_data() } pub fn set_vc_index(&self, index: usize) { self.vc_index.store(index, Ordering::SeqCst); } pub fn vc_index(&self) -> Option { let x = self.vc_index.load(Ordering::SeqCst); if x == usize::MAX { return None; } return Some(x); } #[inline] pub fn link(&self) -> Option> { self.link.read().upgrade() } pub fn checked_link(&self) -> Result, SystemError> { if let Some(link) = self.link() { return Ok(link); } return Err(SystemError::ENODEV); } pub fn set_link(&self, link: Weak) { *self.link.write() = link; } pub fn init_termios(&self) { let tty_index = self.index(); let driver = self.driver(); // 初始化termios if !driver .flags() .contains(super::tty_driver::TtyDriverFlag::TTY_DRIVER_RESET_TERMIOS) { // 先查看是否有已经保存的termios if let Some(t) = driver.saved_termios().get(tty_index) { let mut termios = *t; termios.line = driver.init_termios().line; self.set_termios(termios); } } // TODO:设置termios波特率? } #[inline] pub fn add_epitem(&self, epitem: Arc) { self.epitems.lock().push_back(epitem) } pub fn eptiems(&self) -> &SpinLock>> { &self.epitems } pub fn do_write(&self, buf: &[u8], mut nr: usize) -> Result { // 关闭中断 if let Some(vc_data) = self.vc_data() { let mut vc_data_guard = vc_data.lock_irqsave(); let mut offset = 0; // 这个参数是用来扫描unicode字符的,但是这部分目前未完成,先写着 let mut rescan = false; let mut ch: u32 = 0; let mut draw = DrawRegion::default(); // 首先隐藏光标再写 vc_data_guard.hide_cursor(); while nr != 0 { if !rescan { ch = buf[offset] as u32; offset += 1; nr -= 1; } let (tc, rescan_last) = vc_data_guard.translate(&mut ch); if tc.is_none() { // 表示未转换完成 continue; } let tc = tc.unwrap(); rescan = rescan_last; if vc_data_guard.is_control(tc, ch) { vc_data_guard.flush(&mut draw); vc_data_guard.do_control(ch); continue; } if !vc_data_guard.console_write_normal(tc, ch, &mut draw) { continue; } } vc_data_guard.flush(&mut draw); // TODO: notify update return Ok(offset); } else { return Ok(0); } } } impl TtyOperation for TtyCore { #[inline] fn open(&self, tty: &TtyCoreData) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().open(tty); } #[inline] fn write_room(&self, tty: &TtyCoreData) -> usize { return self.core().tty_driver.driver_funcs().write_room(tty); } #[inline] fn write(&self, tty: &TtyCoreData, buf: &[u8], nr: usize) -> Result { return self.core().tty_driver.driver_funcs().write(tty, buf, nr); } #[inline] fn flush_chars(&self, tty: &TtyCoreData) { self.core().tty_driver.driver_funcs().flush_chars(tty); } #[inline] fn put_char(&self, tty: &TtyCoreData, ch: u8) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().put_char(tty, ch); } #[inline] fn install(&self, driver: Arc, tty: Arc) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().install(driver, tty); } #[inline] fn start(&self, tty: &TtyCoreData) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().start(tty); } #[inline] fn stop(&self, tty: &TtyCoreData) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().stop(tty); } #[inline] fn ioctl(&self, tty: Arc, cmd: u32, arg: usize) -> Result<(), SystemError> { return self.core().tty_driver.driver_funcs().ioctl(tty, cmd, arg); } #[inline] fn chars_in_buffer(&self) -> usize { return self.core().tty_driver.driver_funcs().chars_in_buffer(); } #[inline] fn set_termios(&self, tty: Arc, old_termios: Termios) -> Result<(), SystemError> { return self .core() .tty_driver .driver_funcs() .set_termios(tty, old_termios); } fn close(&self, tty: Arc) -> Result<(), SystemError> { self.core().tty_driver.driver_funcs().close(tty) } fn resize(&self, tty: Arc, winsize: WindowSize) -> Result<(), SystemError> { self.core.tty_driver.driver_funcs().resize(tty, winsize) } } bitflags! { pub struct TtyFlag: u32 { /// 终端被节流 const THROTTLED = 1 << 0; /// 终端输入输出错误状态 const IO_ERROR = 1 << 1; /// 终端的其他一方已关闭 const OTHER_CLOSED = 1 << 2; /// 终端处于独占状态 const EXCLUSIVE = 1 << 3; /// 终端执行写唤醒操作 const DO_WRITE_WAKEUP = 1 << 5; /// 终端线路驱动程序已打开 const LDISC_OPEN = 1 << 11; /// 终端伪终端设备已锁定 const PTY_LOCK = 1 << 16; /// 终端禁用写分裂操作 const NO_WRITE_SPLIT = 1 << 17; /// 终端挂断(挂起)状态 const HUPPED = 1 << 18; /// 终端正在挂断(挂起) const HUPPING = 1 << 19; /// 终端线路驱动程序正在更改 const LDISC_CHANGING = 1 << 20; /// 终端线路驱动程序已停止 const LDISC_HALTED = 1 << 22; } #[derive(Default)] pub struct TtyPacketStatus: u8 { /* Used for packet mode */ const TIOCPKT_DATA = 0; const TIOCPKT_FLUSHREAD = 1; const TIOCPKT_FLUSHWRITE = 2; const TIOCPKT_STOP = 4; const TIOCPKT_START = 8; const TIOCPKT_NOSTOP = 16; const TIOCPKT_DOSTOP = 32; const TIOCPKT_IOCTL = 64; } } #[derive(Debug, PartialEq)] pub enum EchoOperation { /// 开始特殊操作。 Start, /// 向后移动光标列。 MoveBackCol, /// 设置规范模式下的列位置。 SetCanonCol, /// 擦除制表符。 EraseTab, Undefined(u8), } impl EchoOperation { pub fn from_u8(num: u8) -> EchoOperation { match num { 0xff => Self::Start, 0x80 => Self::MoveBackCol, 0x81 => Self::SetCanonCol, 0x82 => Self::EraseTab, _ => Self::Undefined(num), } } pub fn to_u8(&self) -> u8 { match *self { EchoOperation::Start => 0xff, EchoOperation::MoveBackCol => 0x80, EchoOperation::SetCanonCol => 0x81, EchoOperation::EraseTab => 0x82, EchoOperation::Undefined(num) => num, } } } pub struct TtyIoctlCmd; #[allow(dead_code)] impl TtyIoctlCmd { /// 获取终端参数 pub const TCGETS: u32 = 0x5401; /// 设置终端参数 pub const TCSETS: u32 = 0x5402; /// 设置终端参数并等待所有输出完成 pub const TCSETSW: u32 = 0x5403; /// 设置终端参数并且等待所有输出完成,但在这之前将终端清空 pub const TCSETSF: u32 = 0x5404; /// 获取终端参数 pub const TCGETA: u32 = 0x5405; /// 设置终端参数 pub const TCSETA: u32 = 0x5406; /// 设置终端参数并等待所有输出完成 pub const TCSETAW: u32 = 0x5407; /// 设置终端参数并且等待所有输出完成,但在这之前将终端清空 pub const TCSETAF: u32 = 0x5408; /// 发送零字节,等待所有输出完成 pub const TCSBRK: u32 = 0x5409; /// 控制终端的流控 pub const TCXONC: u32 = 0x540A; /// 刷新输入/输出缓冲区或者丢弃输入缓冲区 pub const TCFLSH: u32 = 0x540B; /// 设置设备为独占模式 pub const TIOCEXCL: u32 = 0x540C; /// 设置设备为非独占模式 pub const TIOCNXCL: u32 = 0x540D; /// 设置当前进程的控制终端 pub const TIOCSCTTY: u32 = 0x540E; /// 获取前台进程组 pub const TIOCGPGRP: u32 = 0x540F; ///设置前台进程组 pub const TIOCSPGRP: u32 = 0x5410; /// 获取输出队列的字节数 pub const TIOCOUTQ: u32 = 0x5411; /// 模拟从终端输入字符 pub const TIOCSTI: u32 = 0x5412; /// 获取窗口大小 pub const TIOCGWINSZ: u32 = 0x5413; /// 设置窗口大小 pub const TIOCSWINSZ: u32 = 0x5414; /// 获取终端控制信号的状态 pub const TIOCMGET: u32 = 0x5415; /// 设置终端控制信号的位 pub const TIOCMBIS: u32 = 0x5416; /// 清除终端控制信号的位 pub const TIOCMBIC: u32 = 0x5417; /// 设置终端控制信号的状态 pub const TIOCMSET: u32 = 0x5418; /// 获取软件载波状态 pub const TIOCGSOFTCAR: u32 = 0x5419; /// 设置软件载波状态 pub const TIOCSSOFTCAR: u32 = 0x541A; /// 获取输入队列的字节数 pub const FIONREAD: u32 = 0x541B; /// Linux 特有命令 pub const TIOCLINUX: u32 = 0x541C; /// 获取控制台设备 pub const TIOCCONS: u32 = 0x541D; /// 获取串行设备参数 pub const TIOCGSERIAL: u32 = 0x541E; /// 设置串行设备参数 pub const TIOCSSERIAL: u32 = 0x541F; /// 设置套接字的报文模式 pub const TIOCPKT: u32 = 0x5420; /// 设置非阻塞 I/O pub const FIONBIO: u32 = 0x5421; /// 清除控制终端 pub const TIOCNOTTY: u32 = 0x5422; /// 设置终端线路驱动器 pub const TIOCSETD: u32 = 0x5423; /// 获取终端线路驱动器 pub const TIOCGETD: u32 = 0x5424; /// 发送终止条件 pub const TCSBRKP: u32 = 0x5425; /// 开始发送零比特 pub const TIOCSBRK: u32 = 0x5427; /// 停止发送零比特 pub const TIOCCBRK: u32 = 0x5428; /// Return the session ID of FD pub const TIOCGSID: u32 = 0x5429; /// 设置ptl锁标记 pub const TIOCSPTLCK: u32 = 0x40045431; /// 获取ptl锁标记 pub const TIOCGPTLCK: u32 = 0x80045439; /// 获取packet标记 pub const TIOCGPKT: u32 = 0x80045438; /// 获取pts index pub const TIOCGPTN: u32 = 0x80045430; }