12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123 |
- use core::intrinsics::likely;
- use core::ops::BitXor;
- use bitmap::{traits::BitMapOps, StaticBitmap};
- use alloc::sync::{Arc, Weak};
- use system_error::SystemError;
- use crate::{
- arch::ipc::signal::Signal,
- driver::tty::{
- termios::{ControlCharIndex, InputMode, LocalMode, OutputMode, Termios},
- tty_core::{EchoOperation, TtyCore, TtyCoreData, TtyFlag, TtyIoctlCmd, TtyPacketStatus},
- tty_driver::{TtyDriverFlag, TtyOperation},
- tty_job_control::TtyJobCtrlManager,
- },
- filesystem::vfs::file::FileMode,
- libs::{
- rwlock::RwLockReadGuard,
- spinlock::{SpinLock, SpinLockGuard},
- },
- mm::VirtAddr,
- net::event_poll::EPollEventType,
- process::ProcessManager,
- syscall::{user_access::UserBufferWriter, Syscall},
- };
- use super::TtyLineDiscipline;
- pub const NTTY_BUFSIZE: usize = 4096;
- pub const ECHO_COMMIT_WATERMARK: usize = 256;
- pub const ECHO_BLOCK: usize = 256;
- pub const ECHO_DISCARD_WATERMARK: usize = NTTY_BUFSIZE - (ECHO_BLOCK + 32);
- fn ntty_buf_mask(idx: usize) -> usize {
- return idx & (NTTY_BUFSIZE - 1);
- }
- #[derive(Debug)]
- pub struct NTtyLinediscipline {
- pub data: SpinLock<NTtyData>,
- }
- impl NTtyLinediscipline {
- #[inline]
- pub fn disc_data(&self) -> SpinLockGuard<NTtyData> {
- self.data.lock_irqsave()
- }
- #[inline]
- pub fn disc_data_try_lock(&self) -> Result<SpinLockGuard<NTtyData>, SystemError> {
- self.data.try_lock_irqsave()
- }
- fn ioctl_helper(&self, tty: Arc<TtyCore>, cmd: u32, arg: usize) -> Result<usize, SystemError> {
- match cmd {
- TtyIoctlCmd::TCXONC => {
- todo!()
- }
- TtyIoctlCmd::TCFLSH => {
- todo!()
- }
- _ => {
- return TtyCore::tty_mode_ioctl(tty.clone(), cmd, arg);
- }
- }
- }
- }
- #[derive(Debug)]
- pub struct NTtyData {
- /// 写者管理,tty只有一个写者,即ttydevice,所以不需要加锁
- /// 读取缓冲区的头指针,表示下一个将要接受进buf的字符的位置
- read_head: usize,
- /// 提交缓冲区的头指针,用于行规程处理
- commit_head: usize,
- /// 规范缓冲区的头指针,用于规范模式的处理
- canon_head: usize,
- /// 回显缓冲区的头指针,用于存储需要回显的字符
- echo_head: usize,
- /// 回显过程中用于提交的头指针
- echo_commit: usize,
- /// 标记回显字符的起始位置
- echo_mark: usize,
- /// 读者管理
- /// 读取字符的尾指针,即当前读取位置
- read_tail: usize,
- /// 行的起始位置
- line_start: usize,
- /// 预读字符数,用于处理控制字符
- lookahead_count: usize,
- // 更改以下六个标记时必须持有termios的锁
- /// Line-next 标志,表示下一个输入字符应当按字面处理
- lnext: bool,
- /// 擦除状态的标志
- erasing: bool,
- /// Raw 模式的标志
- raw: bool,
- /// Real raw 模式的标志
- real_raw: bool,
- /// 规范模式的标志
- icanon: bool,
- /// 是否开启echo
- echo: bool,
- /// 标志是否正在进行推送
- pushing: bool,
- /// 是否没有空间可写
- no_room: bool,
- /// 光标所在列
- cursor_column: u32,
- /// 规范模式下光标所在列
- canon_cursor_column: u32,
- /// 回显缓冲区的尾指针
- echo_tail: usize,
- /// 写者与读者共享
- read_buf: [u8; NTTY_BUFSIZE],
- echo_buf: [u8; NTTY_BUFSIZE],
- read_flags: StaticBitmap<NTTY_BUFSIZE>,
- char_map: StaticBitmap<256>,
- tty: Weak<TtyCore>,
- }
- impl NTtyData {
- pub fn new() -> Self {
- Self {
- read_head: 0,
- commit_head: 0,
- canon_head: 0,
- echo_head: 0,
- echo_commit: 0,
- echo_mark: 0,
- read_tail: 0,
- line_start: 0,
- lookahead_count: 0,
- lnext: false,
- erasing: false,
- raw: false,
- real_raw: false,
- icanon: false,
- pushing: false,
- echo: false,
- cursor_column: 0,
- canon_cursor_column: 0,
- echo_tail: 0,
- read_buf: [0; NTTY_BUFSIZE],
- echo_buf: [0; NTTY_BUFSIZE],
- read_flags: StaticBitmap::new(),
- char_map: StaticBitmap::new(),
- tty: Weak::default(),
- no_room: false,
- }
- }
- #[inline]
- pub fn read_cnt(&self) -> usize {
- self.read_head - self.read_tail
- }
- #[inline]
- pub fn read_at(&self, i: usize) -> u8 {
- let i = i & (NTTY_BUFSIZE - 1);
- self.read_buf[i]
- }
- /// ### 接收数据到NTTY
- pub fn receive_buf_common(
- &mut self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- flags: Option<&[u8]>,
- mut count: usize,
- flow: bool,
- ) -> Result<usize, SystemError> {
- // 获取termios读锁
- let termios = tty.core().termios();
- let mut overflow;
- let mut n;
- let mut offset = 0;
- let mut recved = 0;
- loop {
- let tail = self.read_tail;
- let mut room = NTTY_BUFSIZE - (self.read_head - tail);
- if termios.input_mode.contains(InputMode::PARMRK) {
- room = (room + 2) / 3;
- }
- room -= 1;
- if room == 0 || room > NTTY_BUFSIZE {
- // 可能溢出
- overflow = self.icanon && self.canon_head == tail;
- if room > NTTY_BUFSIZE && overflow {
- self.read_head -= 1;
- }
- self.no_room = flow && !overflow;
- room = if overflow { !0 } else { 0 }
- } else {
- overflow = false;
- }
- n = count.min(room);
- if n == 0 {
- break;
- }
- if !overflow {
- if let Some(flags) = flags {
- self.receive_buf(tty.clone(), &buf[offset..], Some(&flags[offset..]), n);
- } else {
- self.receive_buf(tty.clone(), &buf[offset..], flags, n);
- }
- }
- offset += n;
- count -= n;
- recved += n;
- if tty.core().flags().contains(TtyFlag::LDISC_CHANGING) {
- break;
- }
- }
- // TODO: throttle
- Ok(recved)
- }
- pub fn receive_buf(
- &mut self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- flags: Option<&[u8]>,
- count: usize,
- ) {
- let termios = tty.core().termios();
- let preops = termios.input_mode.contains(InputMode::ISTRIP)
- || termios.input_mode.contains(InputMode::IUCLC)
- || termios.local_mode.contains(LocalMode::IEXTEN);
- let look_ahead = self.lookahead_count.min(count);
- if self.real_raw {
- self.receive_buf_real_raw(buf, count);
- } else if self.raw || (termios.local_mode.contains(LocalMode::EXTPROC) && !preops) {
- self.receive_buf_raw(buf, flags, count);
- } else if tty.core().is_closing() && !termios.local_mode.contains(LocalMode::EXTPROC) {
- todo!()
- } else {
- if look_ahead > 0 {
- self.receive_buf_standard(tty.clone(), buf, flags, look_ahead, true);
- }
- if count > look_ahead {
- self.receive_buf_standard(tty.clone(), buf, flags, count - look_ahead, false);
- }
- // 刷新echo
- self.flush_echoes(tty.clone());
- tty.flush_chars(tty.core());
- }
- self.lookahead_count -= look_ahead;
- if self.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
- return;
- }
- self.commit_head = self.read_head;
- if self.read_cnt() > 0 {
- tty.core()
- .read_wq()
- .wakeup_any((EPollEventType::EPOLLIN | EPollEventType::EPOLLRDBAND).bits() as u64);
- }
- }
- fn receive_buf_real_raw(&mut self, buf: &[u8], mut count: usize) {
- let mut head = ntty_buf_mask(self.read_head);
- let mut n = count.min(NTTY_BUFSIZE - head);
- // 假如有一部分在队列头部,则这部分是拷贝尾部部分
- self.read_buf[head..(head + n)].copy_from_slice(&buf[0..n]);
- self.read_head += n;
- count -= n;
- let offset = n;
- // 假如有一部分在队列头部,则这部分是拷贝头部部分
- head = ntty_buf_mask(self.read_head);
- n = count.min(NTTY_BUFSIZE - head);
- self.read_buf[head..(head + n)].copy_from_slice(&buf[offset..(offset + n)]);
- self.read_head += n;
- }
- fn receive_buf_raw(&mut self, buf: &[u8], flags: Option<&[u8]>, mut count: usize) {
- // TTY_NORMAL 目前这部分未做,所以先占位置而不做抽象
- let mut flag = 1;
- let mut f_offset = 0;
- let mut c_offset = 0;
- while count != 0 {
- if flags.is_some() {
- flag = flags.as_ref().unwrap()[f_offset];
- f_offset += 1;
- }
- if likely(flag == 1) {
- self.read_buf[self.read_head] = buf[c_offset];
- c_offset += 1;
- self.read_head += 1;
- } else {
- todo!()
- }
- count -= 1;
- }
- }
- pub fn flush_echoes(&mut self, tty: Arc<TtyCore>) {
- let termios = tty.core().termios();
- if !termios.local_mode.contains(LocalMode::ECHO)
- && !termios.local_mode.contains(LocalMode::ECHONL)
- || self.echo_commit == self.echo_head
- {
- return;
- }
- self.echo_commit = self.echo_head;
- drop(termios);
- let _ = self.echoes(tty);
- }
- pub fn receive_buf_standard(
- &mut self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- flags: Option<&[u8]>,
- mut count: usize,
- lookahead_done: bool,
- ) {
- let termios = tty.core().termios();
- if flags.is_some() {
- todo!("ntty recv buf flags todo");
- }
- let mut offset = 0;
- while count > 0 {
- if offset >= buf.len() {
- break;
- }
- let mut c = buf[offset];
- offset += 1;
- if self.lnext {
- // 将下一个字符当做字面值处理
- self.lnext = false;
- if termios.input_mode.contains(InputMode::ISTRIP) {
- c &= 0x7f;
- }
- if termios.input_mode.contains(InputMode::IUCLC)
- && termios.local_mode.contains(LocalMode::IEXTEN)
- {
- c = (c as char).to_ascii_lowercase() as u8;
- self.receive_char(c, tty.clone())
- }
- continue;
- }
- if termios.input_mode.contains(InputMode::ISTRIP) {
- c &= 0x7f;
- }
- if termios.input_mode.contains(InputMode::IUCLC)
- && termios.local_mode.contains(LocalMode::IEXTEN)
- {
- c = (c as char).to_ascii_lowercase() as u8;
- }
- if termios.local_mode.contains(LocalMode::EXTPROC) {
- self.add_read_byte(c);
- continue;
- }
- if self.char_map.get(c as usize).unwrap() {
- // 特殊字符
- self.receive_special_char(c, tty.clone(), lookahead_done)
- } else {
- self.receive_char(c, tty.clone());
- }
- count -= 1;
- }
- }
- #[inline(never)]
- pub fn receive_special_char(&mut self, mut c: u8, tty: Arc<TtyCore>, lookahead_done: bool) {
- let is_flow_ctrl = self.is_flow_ctrl_char(tty.clone(), c, lookahead_done);
- let termios = tty.core().termios();
- // 启用软件流控,并且该字符已经当做软件流控字符处理
- if termios.input_mode.contains(InputMode::IXON) && is_flow_ctrl {
- return;
- }
- if termios.local_mode.contains(LocalMode::ISIG) {
- if c == termios.control_characters[ControlCharIndex::VINTR] {
- self.recv_sig_char(tty.clone(), &termios, Signal::SIGINT, c);
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VQUIT] {
- self.recv_sig_char(tty.clone(), &termios, Signal::SIGQUIT, c);
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VSUSP] {
- self.recv_sig_char(tty.clone(), &termios, Signal::SIGTSTP, c);
- return;
- }
- }
- let flow = tty.core().flow_irqsave();
- if flow.stopped
- && !flow.tco_stopped
- && termios.input_mode.contains(InputMode::IXON)
- && termios.input_mode.contains(InputMode::IXANY)
- {
- tty.tty_start();
- self.process_echoes(tty.clone());
- }
- drop(flow);
- if c == b'\r' {
- if termios.input_mode.contains(InputMode::IGNCR) {
- // 忽略
- return;
- }
- if termios.input_mode.contains(InputMode::ICRNL) {
- // 映射为换行
- c = b'\n';
- }
- } else if c == b'\n' && termios.input_mode.contains(InputMode::INLCR) {
- // 映射为回车
- c = b'\r';
- }
- if self.icanon {
- if c == termios.control_characters[ControlCharIndex::VERASE]
- || c == termios.control_characters[ControlCharIndex::VKILL]
- || (c == termios.control_characters[ControlCharIndex::VWERASE]
- && termios.local_mode.contains(LocalMode::IEXTEN))
- {
- self.eraser(c, &termios);
- self.commit_echoes(tty.clone());
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VLNEXT]
- && termios.local_mode.contains(LocalMode::IEXTEN)
- {
- self.lnext = true;
- if termios.local_mode.contains(LocalMode::ECHO) {
- self.finish_erasing();
- if termios.local_mode.contains(LocalMode::ECHOCTL) {
- self.echo_char_raw(b'^');
- self.echo_char_raw(8);
- self.commit_echoes(tty.clone());
- }
- }
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VREPRINT]
- && termios.local_mode.contains(LocalMode::ECHO)
- && termios.local_mode.contains(LocalMode::IEXTEN)
- {
- let mut tail = self.canon_head;
- self.finish_erasing();
- self.echo_char(c, &termios);
- self.echo_char_raw(b'\n');
- while ntty_buf_mask(tail) != ntty_buf_mask(self.read_head) {
- self.echo_char(self.read_buf[ntty_buf_mask(tail)], &termios);
- tail += 1;
- }
- self.commit_echoes(tty.clone());
- return;
- }
- if c == b'\n' {
- if termios.local_mode.contains(LocalMode::ECHO)
- || termios.local_mode.contains(LocalMode::ECHONL)
- {
- self.echo_char_raw(b'\n');
- self.commit_echoes(tty.clone());
- }
- self.read_flags.set(ntty_buf_mask(self.read_head), true);
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- self.canon_head = self.read_head;
- tty.core().read_wq().wakeup_any(
- (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
- );
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VEOF] {
- c = ControlCharIndex::DISABLE_CHAR;
- self.read_flags.set(ntty_buf_mask(self.read_head), true);
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- self.canon_head = self.read_head;
- tty.core().read_wq().wakeup_any(
- (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
- );
- return;
- }
- if c == termios.control_characters[ControlCharIndex::VEOL]
- || (c == termios.control_characters[ControlCharIndex::VEOL2]
- && termios.local_mode.contains(LocalMode::IEXTEN))
- {
- if termios.local_mode.contains(LocalMode::ECHO) {
- if self.canon_head == self.read_head {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
- }
- self.echo_char(c, &termios);
- self.commit_echoes(tty.clone());
- }
- if c == 0o377 && termios.input_mode.contains(InputMode::PARMRK) {
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- }
- self.read_flags.set(ntty_buf_mask(self.read_head), true);
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- self.canon_head = self.read_head;
- tty.core().read_wq().wakeup_any(
- (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
- );
- return;
- }
- }
- if termios.local_mode.contains(LocalMode::ECHO) {
- self.finish_erasing();
- if c == b'\n' {
- self.echo_char_raw(b'\n');
- } else {
- if self.canon_head == self.read_head {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
- }
- self.echo_char(c, &termios);
- }
- self.commit_echoes(tty.clone());
- }
- if c == 0o377 && termios.input_mode.contains(InputMode::PARMRK) {
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- }
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- }
- /// ## ntty默认eraser function
- #[inline(never)]
- fn eraser(&mut self, mut c: u8, termios: &RwLockReadGuard<Termios>) {
- if self.read_head == self.canon_head {
- return;
- }
- let erase = c == termios.control_characters[ControlCharIndex::VERASE];
- let werase = c == termios.control_characters[ControlCharIndex::VWERASE];
- let kill = !erase && !werase;
- if kill {
- if !termios.local_mode.contains(LocalMode::ECHO) {
- self.read_head = self.canon_head;
- return;
- }
- if !termios.local_mode.contains(LocalMode::ECHOK)
- || !termios.local_mode.contains(LocalMode::ECHOKE)
- || !termios.local_mode.contains(LocalMode::ECHOE)
- {
- self.read_head = self.canon_head;
- if self.erasing {
- self.echo_char_raw(c);
- self.erasing = false;
- }
- self.echo_char(c, termios);
- if termios.local_mode.contains(LocalMode::ECHOK) {
- // 添加新行
- self.echo_char_raw(b'\n');
- }
- return;
- }
- }
- let mut head;
- let mut cnt;
- while ntty_buf_mask(self.read_head) != ntty_buf_mask(self.canon_head) {
- head = self.read_head;
- loop {
- // 消除多字节字符
- head -= 1;
- c = self.read_buf[ntty_buf_mask(head)];
- if !(Self::is_continuation(c, termios)
- && ntty_buf_mask(head) != ntty_buf_mask(self.canon_head))
- {
- break;
- }
- }
- if Self::is_continuation(c, termios) {
- break;
- }
- if werase {
- todo!()
- }
- cnt = self.read_head - head;
- self.read_head = head;
- if termios.local_mode.contains(LocalMode::ECHO) {
- if termios.local_mode.contains(LocalMode::ECHOPRT) {
- if !self.erasing {
- self.echo_char_raw(b'\\');
- self.erasing = true;
- }
- self.echo_char(c, termios);
- cnt -= 1;
- while cnt > 0 {
- cnt -= 1;
- head += 1;
- self.echo_char_raw(self.read_buf[ntty_buf_mask(head)]);
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::MoveBackCol.to_u8());
- }
- } else if erase && !termios.local_mode.contains(LocalMode::ECHOE) {
- self.echo_char(
- termios.control_characters[ControlCharIndex::VERASE],
- termios,
- );
- } else if c == b'\t' {
- let mut num_chars = 0;
- let mut after_tab = false;
- let mut tail = self.read_head;
- while ntty_buf_mask(tail) != ntty_buf_mask(self.canon_head) {
- tail -= 1;
- c = self.read_buf[ntty_buf_mask(tail)];
- if c == b'\t' {
- after_tab = true;
- break;
- } else if (c as char).is_control() {
- if termios.local_mode.contains(LocalMode::ECHOCTL) {
- num_chars += 2;
- }
- } else if !Self::is_continuation(c, termios) {
- num_chars += 1;
- }
- }
- self.echo_erase_tab(num_chars, after_tab);
- } else {
- if (c as char).is_control() && termios.local_mode.contains(LocalMode::ECHOCTL) {
- // 8 => '\b'
- self.echo_char_raw(8);
- self.echo_char_raw(b' ');
- self.echo_char_raw(8);
- }
- if !(c as char).is_control() || termios.local_mode.contains(LocalMode::ECHOCTL)
- {
- // 8 => '\b'
- self.echo_char_raw(8);
- self.echo_char_raw(b' ');
- self.echo_char_raw(8);
- }
- }
- }
- if erase {
- break;
- }
- }
- if self.read_head == self.canon_head && termios.local_mode.contains(LocalMode::ECHO) {
- self.finish_erasing();
- }
- }
- fn finish_erasing(&mut self) {
- if self.erasing {
- self.echo_char_raw(b'/');
- self.erasing = false;
- }
- }
- fn echo_erase_tab(&mut self, mut num: u8, after_tab: bool) {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::EraseTab.to_u8());
- num &= 7;
- if after_tab {
- num |= 0x80;
- }
- self.add_echo_byte(num);
- }
- /// ## 多字节字符检测
- /// 检测是否为多字节字符的后续字节
- fn is_continuation(c: u8, termios: &RwLockReadGuard<Termios>) -> bool {
- return termios.input_mode.contains(InputMode::IUTF8) && (c & 0xc0) == 0x80;
- }
- /// ## 该字符是否已经当做流控字符处理
- pub fn is_flow_ctrl_char(&mut self, tty: Arc<TtyCore>, c: u8, lookahead_done: bool) -> bool {
- let termios = tty.core().termios();
- if !(termios.control_characters[ControlCharIndex::VSTART] == c
- || termios.control_characters[ControlCharIndex::VSTOP] == c)
- {
- return false;
- }
- if lookahead_done {
- return true;
- }
- if termios.control_characters[ControlCharIndex::VSTART] == c {
- tty.tty_start();
- self.process_echoes(tty.clone());
- return true;
- } else {
- tty.tty_stop();
- return true;
- }
- }
- /// ## 接收到信号字符时的处理
- fn recv_sig_char(
- &mut self,
- tty: Arc<TtyCore>,
- termios: &RwLockReadGuard<Termios>,
- signal: Signal,
- c: u8,
- ) {
- self.input_signal(tty.clone(), termios, signal);
- if termios.input_mode.contains(InputMode::IXON) {
- tty.tty_start();
- }
- if termios.local_mode.contains(LocalMode::ECHO) {
- self.echo_char(c, termios);
- self.commit_echoes(tty);
- } else {
- self.process_echoes(tty);
- }
- }
- /// ## 处理输入信号
- pub fn input_signal(
- &mut self,
- tty: Arc<TtyCore>,
- termios: &RwLockReadGuard<Termios>,
- signal: Signal,
- ) {
- // 先处理信号
- let mut ctrl_info = tty.core().contorl_info_irqsave();
- let pg = ctrl_info.pgid;
- if let Some(pg) = pg {
- let _ = Syscall::kill(pg, signal as i32);
- }
- ctrl_info.pgid = None;
- ctrl_info.session = None;
- if !termios.local_mode.contains(LocalMode::NOFLSH) {
- // 重置
- self.echo_head = 0;
- self.echo_tail = 0;
- self.echo_mark = 0;
- self.echo_commit = 0;
- let _ = tty.flush_buffer(tty.core());
- self.read_head = 0;
- self.canon_head = 0;
- self.read_tail = 0;
- self.line_start = 0;
- self.erasing = false;
- self.read_flags.set_all(false);
- self.pushing = false;
- self.lookahead_count = 0;
- if tty.core().link().is_some() {
- self.packet_mode_flush(tty.core());
- }
- }
- }
- pub fn receive_char(&mut self, c: u8, tty: Arc<TtyCore>) {
- let termios = tty.core().termios();
- if termios.local_mode.contains(LocalMode::ECHO) {
- if self.erasing {
- self.add_echo_byte(b'/');
- self.erasing = false;
- }
- if self.canon_head == self.read_head {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
- }
- self.echo_char(c, &termios);
- self.commit_echoes(tty.clone());
- }
- if c == 0o377 && tty.core().termios().input_mode.contains(InputMode::PARMRK) {
- self.add_read_byte(c);
- }
- self.add_read_byte(c);
- }
- pub fn echo_char(&mut self, c: u8, termios: &RwLockReadGuard<Termios>) {
- if c == EchoOperation::Start.to_u8() {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::Start.to_u8());
- } else {
- if termios.local_mode.contains(LocalMode::ECHOCTL)
- && (c as char).is_control()
- && c != b'\t'
- {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- }
- self.add_echo_byte(c);
- }
- }
- pub fn echo_char_raw(&mut self, c: u8) {
- if c == EchoOperation::Start.to_u8() {
- self.add_echo_byte(EchoOperation::Start.to_u8());
- self.add_echo_byte(EchoOperation::Start.to_u8());
- } else {
- self.add_echo_byte(c);
- }
- }
- /// ## 提交echobuf里的数据显示
- pub fn commit_echoes(&mut self, tty: Arc<TtyCore>) {
- let head = self.echo_head;
- self.echo_mark = head;
- let old = self.echo_commit - self.echo_tail;
- // 需要echo的字符个数
- let nr = head - self.echo_tail;
- if nr < ECHO_COMMIT_WATERMARK || nr % ECHO_BLOCK > old % ECHO_BLOCK {
- return;
- }
- self.echo_commit = head;
- let echoed = self.echoes(tty.clone());
- if echoed.is_ok() && echoed.unwrap() > 0 {
- tty.flush_chars(tty.core());
- }
- }
- pub fn add_echo_byte(&mut self, c: u8) {
- self.echo_buf[ntty_buf_mask(self.echo_head)] = c;
- self.echo_head += 1;
- }
- pub fn add_read_byte(&mut self, c: u8) {
- self.read_buf[ntty_buf_mask(self.read_head)] = c;
- self.read_head += 1;
- }
- /// ### 将read_buffer的部分值置0
- ///
- /// 只会在规范模式和禁用echo下执行
- #[inline]
- pub fn zero_buffer(&mut self, offset: usize, size: usize) {
- let offset = offset & (NTTY_BUFSIZE - 1);
- if self.icanon && !self.echo {
- let n = offset + size;
- if n > NTTY_BUFSIZE {
- for c in &mut self.read_buf[offset..NTTY_BUFSIZE] {
- *c = 0
- }
- for c in &mut self.read_buf[0..(n - NTTY_BUFSIZE)] {
- *c = 0
- }
- } else {
- for c in &mut self.read_buf[offset..n] {
- *c = 0
- }
- };
- }
- }
- /// ## 从ntty中拷贝数据
- ///
- /// ### 参数
- ///
- /// ### to: 存储数据
- /// ### tail: 读取尾
- pub fn ntty_copy(
- &mut self,
- to: &mut [u8],
- tail: usize,
- n: &mut usize,
- ) -> Result<(), SystemError> {
- if to.len() < *n {
- *n = to.len();
- // return Err(SystemError::EINVAL);
- }
- if tail > NTTY_BUFSIZE {
- return Err(SystemError::EINVAL);
- }
- let size = NTTY_BUFSIZE - tail;
- if size < *n {
- // 有一部分数据在头部,则先拷贝后面部分,再拷贝头部
- // TODO: tty审计?
- to[0..size].copy_from_slice(&self.read_buf[tail..(tail + size)]);
- to[size..(*n)].copy_from_slice(&self.read_buf[0..(*n - size)]);
- } else {
- to[..*n].copy_from_slice(&self.read_buf[tail..(tail + *n)])
- }
- self.zero_buffer(tail, *n);
- Ok(())
- }
- /// ## 规范模式下跳过EOF
- pub fn canon_skip_eof(&mut self) {
- // 没有数据
- if self.read_tail == self.canon_head {
- return;
- }
- let tail = self.read_tail & (NTTY_BUFSIZE - 1);
- // 查看read_flags是否读取位置为特殊字符
- if !self.read_flags.get(tail).unwrap() {
- return;
- }
- // 确保读取位置是'\0'字符
- if self.read_buf[tail] != ControlCharIndex::DISABLE_CHAR {
- return;
- }
- // 处理该字符,将read_flagsw该位清除
- self.read_flags.set(tail, false);
- // 读取位置+1,即跳过该字符不做处理
- self.read_tail += 1;
- }
- /// ## 在规范模式(canonical mode)下从读缓冲中复制一行
- ///
- /// 一次只拷贝一行
- ///
- /// ## 参数
- /// ### dst: 存放数据
- /// ### nr: 需要拷贝的数据大小
- ///
- /// ## 返回值
- /// ### true: 表示一行未结束并且还有数据可读
- /// ### false: 一行已结束或者没有数据可读
- pub fn canon_copy_from_read_buf(
- &mut self,
- dst: &mut [u8],
- nr: &mut usize,
- offset: &mut usize,
- ) -> Result<bool, SystemError> {
- if *nr == 0 {
- return Ok(false);
- }
- let canon_head = self.canon_head;
- // 取得能够读到的字符数,即canon_head - self.read_tail和nr最小值
- let mut n = (*nr).min(canon_head - self.read_tail);
- // 获得读尾index
- let tail = self.read_tail & (NTTY_BUFSIZE - 1);
- // 避免越界,这个size才是实际读取大小
- let size = if tail + n > NTTY_BUFSIZE {
- NTTY_BUFSIZE
- } else {
- tail + n
- };
- // 找到eol的坐标
- let tmp = self.read_flags.next_index(tail);
- // 找到的话即为坐标,未找到的话即为NTTY_BUFSIZE
- let mut eol = if let Some(tmp) = tmp { tmp } else { size };
- if eol > size {
- eol = size
- }
- // 是否需要绕回缓冲区头部
- let more = n - (size - tail);
- // 是否找到eol
- let found = if eol == NTTY_BUFSIZE && more > 0 {
- // 需要返回头部
- let ret = self.read_flags.first_index();
- if let Some(tmp) = ret {
- // 在头部范围内找到eol
- if tmp < more {
- eol = tmp;
- }
- } else {
- eol = more;
- }
- eol != more
- } else {
- // 不需要返回头部
- eol != size
- };
- n = eol - tail;
- if n > NTTY_BUFSIZE {
- // 减法溢出则加上BUFSIZE即可限制在0-NTTY_BUFSIZE内
- n += NTTY_BUFSIZE;
- }
- // 规范模式下实际扫描过的字符数,需要将eol计算在内
- let count = if found { n + 1 } else { n };
- // 表示这一行未结束
- if !found || self.read_at(eol) != ControlCharIndex::DISABLE_CHAR {
- n = count;
- }
- self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
- *nr -= n;
- *offset += n;
- if found {
- self.read_flags.set(eol, false);
- }
- self.read_tail += count;
- if found {
- if !self.pushing {
- self.line_start = self.read_tail;
- } else {
- self.pushing = false;
- }
- // todo: 审计?
- return Ok(false);
- }
- // 这里是表示没有找到eol,根据是否还有数据可读返回
- Ok(self.read_tail != canon_head)
- }
- /// ## 根据终端的模式和输入缓冲区中的数据量,判断是否可读字符
- pub fn input_available(&self, termios: RwLockReadGuard<Termios>, poll: bool) -> bool {
- // 计算最小字符数
- let amt = if poll
- && termios.control_characters[ControlCharIndex::VTIME] as u32 == 0
- && termios.control_characters[ControlCharIndex::VMIN] as u32 != 0
- {
- termios.control_characters[ControlCharIndex::VMIN] as usize
- } else {
- 1
- };
- // 规范模式且非拓展
- if self.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
- return self.canon_head != self.read_tail;
- } else {
- return (self.commit_head - self.read_tail) >= amt;
- }
- }
- /// ## 非规范模式下从read_buf读取数据
- ///
- /// ## 参数
- /// ### termios: tty对应的termioss读锁守卫
- /// ### dst: 存储读取数据
- /// ### nr: 读取长度
- ///
- /// ## 返回值
- /// ### true: 还有更多数据可读
- /// ### false: 无更多数据可读
- pub fn copy_from_read_buf(
- &mut self,
- termios: RwLockReadGuard<Termios>,
- dst: &mut [u8],
- nr: &mut usize,
- offset: &mut usize,
- ) -> Result<bool, SystemError> {
- let head = self.commit_head;
- let tail = self.read_tail & (NTTY_BUFSIZE - 1);
- // 计算出可读的字符数
- let mut n = (NTTY_BUFSIZE - tail).min(head - self.read_tail);
- n = n.min(*nr);
- if n > 0 {
- // 拷贝数据
- self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
- // todo:审计?
- self.read_tail += n;
- // 是否只读取了eof
- let eof =
- n == 1 && self.read_buf[tail] == termios.control_characters[ControlCharIndex::VEOF];
- if termios.local_mode.contains(LocalMode::EXTPROC)
- && self.icanon
- && eof
- && head == self.read_tail
- {
- return Ok(false);
- }
- *nr -= n;
- *offset += n;
- return Ok(head != self.read_tail);
- }
- Ok(false)
- }
- /// ## 用于处理带有 OPOST(Output Post-processing)标志的输出块的函数
- /// OPOST 是 POSIX 终端驱动器标志之一,用于指定在写入终端设备之前对输出数据进行一些后期处理。
- pub fn process_output_block(
- &mut self,
- core: &TtyCoreData,
- termios: RwLockReadGuard<Termios>,
- buf: &[u8],
- nr: usize,
- ) -> Result<usize, SystemError> {
- let mut nr = nr;
- let tty = self.tty.upgrade().unwrap();
- let space = tty.write_room(tty.core());
- // 如果读取数量大于了可用空间,则取最小的为真正的写入数量
- if nr > space {
- nr = space
- }
- let mut cnt = 0;
- for (i, c) in buf.iter().enumerate().take(nr) {
- cnt = i;
- let c = *c;
- if c as usize == 8 {
- // 表示退格
- if self.cursor_column > 0 {
- self.cursor_column -= 1;
- }
- continue;
- }
- match c as char {
- '\n' => {
- if termios.output_mode.contains(OutputMode::ONLRET) {
- // 将回车映射为\n,即将\n换为回车
- self.cursor_column = 0;
- }
- if termios.output_mode.contains(OutputMode::ONLCR) {
- // 输出时将\n换为\r\n
- break;
- }
- self.canon_cursor_column = self.cursor_column;
- }
- '\r' => {
- if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
- // 光标已经在第0列,则不输出回车符
- break;
- }
- if termios.output_mode.contains(OutputMode::OCRNL) {
- break;
- }
- self.canon_cursor_column = self.cursor_column;
- }
- '\t' => {
- break;
- }
- _ => {
- // 判断是否为控制字符
- if !(c as char).is_control() {
- if termios.output_mode.contains(OutputMode::OLCUC) {
- break;
- }
- // 判断是否为utf8模式下的连续字符
- if !(termios.input_mode.contains(InputMode::IUTF8)
- && (c as usize) & 0xc0 == 0x80)
- {
- self.cursor_column += 1;
- }
- }
- }
- }
- }
- drop(termios);
- return tty.write(core, buf, cnt);
- }
- /// ## 处理回显
- pub fn process_echoes(&mut self, tty: Arc<TtyCore>) {
- if self.echo_mark == self.echo_tail {
- return;
- }
- self.echo_commit = self.echo_mark;
- let echoed = self.echoes(tty.clone());
- if echoed.is_ok() && echoed.unwrap() > 0 {
- tty.flush_chars(tty.core());
- }
- }
- #[inline(never)]
- pub fn echoes(&mut self, tty: Arc<TtyCore>) -> Result<usize, SystemError> {
- let mut space = tty.write_room(tty.core());
- let ospace = space;
- let termios = tty.core().termios();
- let core = tty.core();
- let mut tail = self.echo_tail;
- while ntty_buf_mask(self.echo_commit) != ntty_buf_mask(tail) {
- let c = self.echo_buf[ntty_buf_mask(tail)];
- if EchoOperation::from_u8(c) == EchoOperation::Start {
- if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 1) {
- self.echo_tail = tail;
- return Ok(ospace - space);
- }
- // 获取到start,之后取第一个作为op
- let op = EchoOperation::from_u8(self.echo_buf[ntty_buf_mask(tail + 1)]);
- match op {
- EchoOperation::Start => {
- if space == 0 {
- break;
- }
- if tty
- .put_char(tty.core(), EchoOperation::Start.to_u8())
- .is_err()
- {
- tty.write(core, &[EchoOperation::Start.to_u8()], 1)?;
- }
- self.cursor_column += 1;
- space -= 1;
- tail += 2;
- }
- EchoOperation::MoveBackCol => {
- if self.cursor_column > 0 {
- self.cursor_column -= 1;
- }
- tail += 2;
- }
- EchoOperation::SetCanonCol => {
- self.canon_cursor_column = self.cursor_column;
- tail += 2;
- }
- EchoOperation::EraseTab => {
- if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 2) {
- self.echo_tail = tail;
- return Ok(ospace - space);
- }
- // 要擦除的制表符所占用的列数
- let mut char_num = self.echo_buf[ntty_buf_mask(tail + 2)] as usize;
- /*
- 如果 num_chars 的最高位(0x80)未设置,
- 表示这是从输入的起始位置而不是从先前的制表符开始计算的列数。
- 在这种情况下,将 num_chars 与 ldata->canon_column 相加,否则,列数就是正常的制表符列数。
- */
- if char_num & 0x80 == 0 {
- char_num += self.canon_cursor_column as usize;
- }
- // 计算要回退的列数,即制表符宽度减去实际占用的列数
- let mut num_bs = 8 - (char_num & 7);
- if num_bs > space {
- // 表示左边没有足够空间回退
- break;
- }
- space -= num_bs;
- while num_bs != 0 {
- num_bs -= 1;
- // 8 => '\b'
- if tty.put_char(tty.core(), 8).is_err() {
- tty.write(core, &[8], 1)?;
- }
- if self.cursor_column > 0 {
- self.cursor_column -= 1;
- }
- }
- // 已经读取了 tail tail+1 tail+2,所以这里偏移加3
- tail += 3;
- }
- EchoOperation::Undefined(ch) => {
- match ch {
- 8 => {
- if tty.put_char(tty.core(), 8).is_err() {
- tty.write(core, &[8], 1)?;
- }
- if tty.put_char(tty.core(), b' ').is_err() {
- tty.write(core, b" ", 1)?;
- }
- self.cursor_column -= 1;
- space -= 1;
- tail += 1;
- }
- _ => {
- // 不是特殊字节码,则表示控制字符 例如 ^C
- if space < 2 {
- break;
- }
- if tty.put_char(tty.core(), b'^').is_err() {
- tty.write(core, b"^", 1)?;
- }
- if tty.put_char(tty.core(), ch ^ 0o100).is_err() {
- tty.write(core, &[ch ^ 0o100], 1)?;
- }
- self.cursor_column += 2;
- space -= 2;
- tail += 2;
- }
- }
- }
- }
- } else {
- if termios.output_mode.contains(OutputMode::OPOST) {
- let ret = self.do_output_char(tty.clone(), c, space);
- if ret.is_err() {
- break;
- }
- space -= ret.unwrap();
- } else {
- if space == 0 {
- break;
- }
- if tty.put_char(tty.core(), c).is_err() {
- tty.write(core, &[c], 1)?;
- }
- space -= 1;
- }
- tail += 1;
- }
- }
- // 如果回显缓冲区接近满(在下一次提交之前可能会发生回显溢出的情况),则丢弃足够的尾部数据以防止随后的溢出。
- while self.echo_commit > tail && self.echo_commit - tail >= ECHO_DISCARD_WATERMARK {
- if self.echo_buf[ntty_buf_mask(tail)] == EchoOperation::Start.to_u8() {
- if self.echo_buf[ntty_buf_mask(tail + 1)] == EchoOperation::EraseTab.to_u8() {
- tail += 3;
- } else {
- tail += 2;
- }
- } else {
- tail += 1;
- }
- }
- self.echo_tail = tail;
- return Ok(ospace - space);
- }
- /// ## 处理输出字符(带有 OPOST 处理)
- pub fn process_output(&mut self, tty: Arc<TtyCore>, c: u8) -> bool {
- let space = tty.write_room(tty.core());
- if self.do_output_char(tty, c, space).is_err() {
- return false;
- }
- true
- }
- // ## 设置带有 OPOST 处理的tty输出一个字符
- pub fn do_output_char(
- &mut self,
- tty: Arc<TtyCore>,
- c: u8,
- space: usize,
- ) -> Result<usize, SystemError> {
- if space == 0 {
- return Err(SystemError::ENOBUFS);
- }
- let termios = tty.core().termios();
- let core = tty.core();
- let mut c = c;
- if c as usize == 8 {
- // 表示退格
- if self.cursor_column > 0 {
- self.cursor_column -= 1;
- }
- if tty.put_char(tty.core(), c).is_err() {
- tty.write(core, &[c], 1)?;
- }
- return Ok(1);
- }
- match c as char {
- '\n' => {
- if termios.output_mode.contains(OutputMode::ONLRET) {
- // 回车符
- self.cursor_column = 0;
- }
- if termios.output_mode.contains(OutputMode::ONLCR) {
- // 映射为“\r\n”
- if space < 2 {
- return Err(SystemError::ENOBUFS);
- }
- self.cursor_column = 0;
- self.canon_cursor_column = 0;
- // 通过驱动写入
- tty.write(core, "\r\n".as_bytes(), 2)?;
- return Ok(2);
- }
- self.canon_cursor_column = self.cursor_column;
- }
- '\r' => {
- if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
- // 光标已经在第0列,则不输出回车符
- return Ok(0);
- }
- if termios.output_mode.contains(OutputMode::OCRNL) {
- // 输出的\r映射为\n
- c = b'\n';
- if termios.output_mode.contains(OutputMode::ONLRET) {
- // \r映射为\n,但是保留\r特性
- self.cursor_column = 0;
- self.canon_cursor_column = 0;
- }
- } else {
- self.cursor_column = 0;
- self.canon_cursor_column = 0;
- }
- }
- '\t' => {
- // 计算输出一个\t需要的空间
- let spaces = 8 - (self.cursor_column & 7) as usize;
- if termios.output_mode.contains(OutputMode::TABDLY)
- && OutputMode::TABDLY.bits() == OutputMode::XTABS.bits()
- {
- // 配置的tab选项是真正输出空格到驱动
- if space < spaces {
- // 空间不够
- return Err(SystemError::ENOBUFS);
- }
- self.cursor_column += spaces as u32;
- // 写入sapces个空格
- tty.write(core, " ".as_bytes(), spaces)?;
- return Ok(spaces);
- }
- self.cursor_column += spaces as u32;
- }
- _ => {
- // 判断是否为控制字符
- if !(c as char).is_control() {
- if termios.output_mode.contains(OutputMode::OLCUC) {
- c = c.to_ascii_uppercase();
- }
- // 判断是否为utf8模式下的连续字符
- if !(termios.input_mode.contains(InputMode::IUTF8)
- && (c as usize) & 0xc0 == 0x80)
- {
- self.cursor_column += 1;
- }
- }
- }
- }
- if tty.put_char(tty.core(), c).is_err() {
- tty.write(core, &[c], 1)?;
- }
- Ok(1)
- }
- fn packet_mode_flush(&self, tty: &TtyCoreData) {
- let link = tty.link().unwrap();
- if link.core().contorl_info_irqsave().packet {
- tty.contorl_info_irqsave()
- .pktstatus
- .insert(TtyPacketStatus::TIOCPKT_FLUSHREAD);
- link.core().read_wq().wakeup_all();
- }
- }
- }
- impl TtyLineDiscipline for NTtyLinediscipline {
- fn open(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
- // 反向绑定tty到disc
- self.disc_data().tty = Arc::downgrade(&tty);
- // 特定的tty设备在这里可能需要取消端口节流
- return self.set_termios(tty, None);
- }
- fn close(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
- todo!()
- }
- /// ## 重置缓冲区的基本信息
- fn flush_buffer(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
- let core = tty.core();
- let _ = core.termios();
- let mut ldata = self.disc_data();
- ldata.read_head = 0;
- ldata.canon_head = 0;
- ldata.read_tail = 0;
- ldata.commit_head = 0;
- ldata.line_start = 0;
- ldata.erasing = false;
- ldata.read_flags.set_all(false);
- ldata.pushing = false;
- ldata.lookahead_count = 0;
- // todo: kick worker?
- // packet mode?
- if core.link().is_some() {
- ldata.packet_mode_flush(core);
- }
- Ok(())
- }
- #[inline(never)]
- fn read(
- &self,
- tty: Arc<TtyCore>,
- buf: &mut [u8],
- len: usize,
- cookie: &mut bool,
- _offset: usize,
- mode: FileMode,
- ) -> Result<usize, system_error::SystemError> {
- let mut ldata;
- if mode.contains(FileMode::O_NONBLOCK) {
- let ret = self.disc_data_try_lock();
- if ret.is_err() {
- return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
- }
- ldata = ret.unwrap();
- } else {
- ldata = self.disc_data();
- }
- let core = tty.core();
- let termios = core.termios();
- let mut nr = len;
- let mut offset = 0;
- // 表示接着读
- if *cookie {
- // 规范且非拓展模式
- if ldata.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
- // 跳过EOF字符
- if len == 0 {
- ldata.canon_skip_eof();
- } else if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
- return Ok(len - nr);
- }
- } else if ldata.copy_from_read_buf(termios, buf, &mut nr, &mut offset)? {
- return Ok(len - nr);
- }
- // 没有数据可读
- // todo: kick worker? or 关闭节流?
- *cookie = false;
- return Ok(len - nr);
- }
- drop(termios);
- TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTIN)?;
- let mut minimum: usize = 0;
- if !ldata.icanon {
- let core = tty.core();
- let termios = core.termios();
- minimum = termios.control_characters[ControlCharIndex::VMIN] as usize;
- if minimum == 0 {
- minimum = 1;
- }
- }
- let packet = core.contorl_info_irqsave().packet;
- let mut ret: Result<usize, SystemError> = Ok(0);
- // 记录读取前 的tail
- let tail = ldata.read_tail;
- drop(ldata);
- while nr != 0 {
- // todo: 处理packet模式
- if packet {
- let link = core.link().unwrap();
- let link = link.core();
- let mut ctrl = link.contorl_info_irqsave();
- if !ctrl.pktstatus.is_empty() {
- if offset != 0 {
- break;
- }
- let cs = ctrl.pktstatus;
- ctrl.pktstatus = TtyPacketStatus::empty();
- buf[offset] = cs.bits();
- offset += 1;
- // nr -= 1;
- break;
- }
- }
- let mut ldata = self.disc_data();
- let core = tty.core();
- if !ldata.input_available(core.termios(), false) {
- if core.flags().contains(TtyFlag::OTHER_CLOSED) {
- ret = Err(SystemError::EIO);
- break;
- }
- if core.flags().contains(TtyFlag::HUPPED) || core.flags().contains(TtyFlag::HUPPING)
- {
- break;
- }
- if mode.contains(FileMode::O_NONBLOCK)
- || core.flags().contains(TtyFlag::LDISC_CHANGING)
- {
- ret = Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
- break;
- }
- if ProcessManager::current_pcb()
- .sig_info_irqsave()
- .sig_pending()
- .has_pending()
- {
- ret = Err(SystemError::ERESTARTSYS);
- break;
- }
- // 休眠一段时间
- // 获取到termios读锁,避免termios被更改导致行为异常
- // let termios = core.termios_preempt_enable();
- // let helper = WakeUpHelper::new(ProcessManager::current_pcb());
- // let wakeup_helper = Timer::new(helper, timeout);
- // wakeup_helper.activate();
- // drop(termios);
- drop(ldata);
- core.read_wq()
- .sleep((EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64);
- continue;
- }
- if ldata.icanon && !core.termios().local_mode.contains(LocalMode::EXTPROC) {
- if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
- *cookie = true;
- offset += len - nr;
- return Ok(offset);
- }
- } else {
- // 非标准模式
- // todo: 处理packet模式
- if packet && offset == 0 {
- buf[offset] = TtyPacketStatus::TIOCPKT_DATA.bits();
- offset += 1;
- nr -= 1;
- }
- // 拷贝数据
- if ldata.copy_from_read_buf(core.termios(), buf, &mut nr, &mut offset)?
- && offset >= minimum
- {
- *cookie = true;
- return Ok(offset);
- }
- }
- if offset >= minimum {
- break;
- }
- }
- let ldata = self.disc_data();
- if tail != ldata.read_tail {
- // todo: kick worker?
- }
- if offset > 0 {
- return Ok(offset);
- }
- ret
- }
- #[inline(never)]
- fn write(
- &self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- len: usize,
- mode: FileMode,
- ) -> Result<usize, system_error::SystemError> {
- let mut nr = len;
- let mut ldata = self.disc_data();
- let pcb = ProcessManager::current_pcb();
- let binding = tty.clone();
- let core = binding.core();
- let termios = *core.termios();
- if termios.local_mode.contains(LocalMode::TOSTOP) {
- TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTOU)?;
- }
- ldata.process_echoes(tty.clone());
- // drop(ldata);
- let mut offset = 0;
- loop {
- if pcb.sig_info_irqsave().sig_pending().has_pending() {
- return Err(SystemError::ERESTARTSYS);
- }
- if core.flags().contains(TtyFlag::HUPPED) {
- return Err(SystemError::EIO);
- }
- if termios.output_mode.contains(OutputMode::OPOST) {
- while nr > 0 {
- // let mut ldata = self.disc_data();
- // 获得一次处理后的数量
- let ret = ldata.process_output_block(core, core.termios(), &buf[offset..], nr);
- let num = match ret {
- Ok(num) => num,
- Err(e) => {
- if e == SystemError::EAGAIN_OR_EWOULDBLOCK {
- break;
- } else {
- return Err(e);
- }
- }
- };
- offset += num;
- nr -= num;
- if nr == 0 {
- break;
- }
- let c = buf[offset];
- if !ldata.process_output(tty.clone(), c) {
- break;
- }
- offset += 1;
- nr -= 1;
- }
- tty.flush_chars(core);
- } else {
- while nr > 0 {
- let write = tty.write(core, &buf[offset..], nr)?;
- if write == 0 {
- break;
- }
- offset += write;
- nr -= write;
- }
- }
- if nr == 0 {
- break;
- }
- if mode.contains(FileMode::O_NONBLOCK) || core.flags().contains(TtyFlag::LDISC_CHANGING)
- {
- return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
- }
- // 到这里表明没位置可写了
- // 休眠一段时间
- // 获取到termios读锁,避免termios被更改导致行为异常
- core.write_wq()
- .sleep(EPollEventType::EPOLLOUT.bits() as u64);
- }
- Ok(offset)
- }
- fn ioctl(
- &self,
- tty: Arc<TtyCore>,
- cmd: u32,
- arg: usize,
- ) -> Result<usize, system_error::SystemError> {
- match cmd {
- TtyIoctlCmd::TIOCOUTQ => {
- let mut user_writer = UserBufferWriter::new(
- VirtAddr::new(arg).as_ptr::<i32>(),
- core::mem::size_of::<i32>(),
- true,
- )?;
- let count = tty.chars_in_buffer();
- user_writer.copy_one_to_user::<i32>(&(count as i32), 0)?;
- return Ok(0);
- }
- TtyIoctlCmd::FIONREAD => {
- let ldata = self.disc_data();
- let termios = tty.core().termios();
- let retval;
- if termios.local_mode.contains(LocalMode::ICANON)
- && !termios.local_mode.contains(LocalMode::EXTPROC)
- {
- if ldata.canon_head == ldata.read_tail {
- retval = 0;
- } else {
- let head = ldata.canon_head;
- let mut tail = ldata.read_tail;
- let mut nr = head - tail;
- while ntty_buf_mask(head) != ntty_buf_mask(tail) {
- if ldata.read_flags.get(ntty_buf_mask(tail)).unwrap()
- && ldata.read_buf[ntty_buf_mask(tail)]
- == ControlCharIndex::DISABLE_CHAR
- {
- nr -= 1;
- }
- tail += 1;
- }
- retval = nr;
- }
- } else {
- retval = ldata.read_cnt();
- }
- let mut user_writer = UserBufferWriter::new(
- VirtAddr::new(arg).as_ptr::<i32>(),
- core::mem::size_of::<i32>(),
- true,
- )?;
- user_writer.copy_one_to_user::<i32>(&(retval as i32), 0)?;
- return Ok(0);
- }
- _ => {
- return self.ioctl_helper(tty, cmd, arg);
- }
- }
- }
- #[inline(never)]
- fn set_termios(
- &self,
- tty: Arc<TtyCore>,
- old: Option<crate::driver::tty::termios::Termios>,
- ) -> Result<(), system_error::SystemError> {
- let core = tty.core();
- let termios = core.termios();
- let mut ldata = self.disc_data();
- let contorl_chars = termios.control_characters;
- // 第一次设置或者规范模式 (ICANON) 或者扩展处理 (EXTPROC) 标志发生变化
- let mut spec_mode_changed = false;
- if let Some(old) = old {
- let local_mode = old.local_mode.bitxor(termios.local_mode);
- spec_mode_changed =
- local_mode.contains(LocalMode::ICANON) || local_mode.contains(LocalMode::EXTPROC);
- }
- if old.is_none() || spec_mode_changed {
- // 重置read_flags
- ldata.read_flags.set_all(false);
- ldata.line_start = ldata.read_tail;
- // 不是规范模式或者有可读数据
- if !termios.local_mode.contains(LocalMode::ICANON) || ldata.read_cnt() != 0 {
- ldata.canon_head = ldata.read_tail;
- ldata.pushing = false;
- } else {
- let read_head = ldata.read_head;
- ldata
- .read_flags
- .set((read_head - 1) & (NTTY_BUFSIZE - 1), true);
- ldata.canon_head = ldata.read_head;
- ldata.pushing = true;
- }
- ldata.commit_head = ldata.read_head;
- ldata.erasing = false;
- ldata.lnext = false;
- }
- // 设置模式
- ldata.icanon = termios.local_mode.contains(LocalMode::ICANON);
- // 设置回显
- if termios.local_mode.contains(LocalMode::ECHO) {
- ldata.echo = true;
- }
- if termios.input_mode.contains(InputMode::ISTRIP)
- || termios.input_mode.contains(InputMode::IUCLC)
- || termios.input_mode.contains(InputMode::IGNCR)
- || termios.input_mode.contains(InputMode::IXON)
- || termios.local_mode.contains(LocalMode::ISIG)
- || termios.local_mode.contains(LocalMode::ECHO)
- || termios.input_mode.contains(InputMode::PARMRK)
- {
- // 非原模式
- ldata.char_map.set_all(false);
- // 忽略回车符或者将回车映射为换行符
- if termios.input_mode.contains(InputMode::IGNCR)
- || termios.input_mode.contains(InputMode::ICRNL)
- {
- ldata.char_map.set('\r' as usize, true);
- }
- // 将换行映射为回车
- if termios.input_mode.contains(InputMode::INLCR) {
- ldata.char_map.set('\n' as usize, true);
- }
- // 规范模式
- if termios.local_mode.contains(LocalMode::ICANON) {
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VERASE] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VKILL] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VEOF] as usize, true);
- ldata.char_map.set('\n' as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VEOL] as usize, true);
- if termios.local_mode.contains(LocalMode::IEXTEN) {
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VWERASE] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VLNEXT] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VEOL2] as usize, true);
- if termios.local_mode.contains(LocalMode::ECHO) {
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VREPRINT] as usize, true);
- }
- }
- }
- // 软件流控制
- if termios.input_mode.contains(InputMode::IXON) {
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VSTART] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VSTOP] as usize, true);
- }
- if termios.local_mode.contains(LocalMode::ISIG) {
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VINTR] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VQUIT] as usize, true);
- ldata
- .char_map
- .set(contorl_chars[ControlCharIndex::VSUSP] as usize, true);
- }
- ldata
- .char_map
- .set(ControlCharIndex::DISABLE_CHAR as usize, true);
- ldata.raw = false;
- ldata.real_raw = false;
- } else {
- // 原模式或real_raw
- ldata.raw = true;
- ldata.real_raw = termios.input_mode.contains(InputMode::IGNBRK)
- || (!termios.input_mode.contains(InputMode::BRKINT)
- && !termios.input_mode.contains(InputMode::PARMRK))
- && (termios.input_mode.contains(InputMode::IGNPAR)
- || !termios.input_mode.contains(InputMode::INPCK))
- && (core
- .driver()
- .flags()
- .contains(TtyDriverFlag::TTY_DRIVER_REAL_RAW));
- }
- // if !termios.input_mode.contains(InputMode::IXON)
- // && old.is_some()
- // && old.unwrap().input_mode.contains(InputMode::IXON) && !
- // {}
- core.read_wq().wakeup_all();
- core.write_wq().wakeup_all();
- Ok(())
- }
- fn poll(&self, tty: Arc<TtyCore>) -> Result<usize, system_error::SystemError> {
- let core = tty.core();
- let ldata = self.disc_data();
- let mut event = EPollEventType::empty();
- if ldata.input_available(core.termios(), true) {
- event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM)
- }
- if core.contorl_info_irqsave().packet {
- let link = core.link();
- if link.is_some()
- && !link
- .unwrap()
- .core()
- .contorl_info_irqsave()
- .pktstatus
- .is_empty()
- {
- event.insert(
- EPollEventType::EPOLLPRI
- | EPollEventType::EPOLLIN
- | EPollEventType::EPOLLRDNORM,
- );
- }
- }
- if core.flags().contains(TtyFlag::OTHER_CLOSED) {
- event.insert(EPollEventType::EPOLLHUP);
- }
- if core.driver().driver_funcs().chars_in_buffer() < 256
- && core.driver().driver_funcs().write_room(core) > 0
- {
- event.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM);
- }
- Ok(event.bits() as usize)
- }
- fn hangup(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
- todo!()
- }
- fn receive_buf(
- &self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- flags: Option<&[u8]>,
- count: usize,
- ) -> Result<usize, SystemError> {
- let mut ldata = self.disc_data();
- ldata.receive_buf_common(tty, buf, flags, count, false)
- }
- fn receive_buf2(
- &self,
- tty: Arc<TtyCore>,
- buf: &[u8],
- flags: Option<&[u8]>,
- count: usize,
- ) -> Result<usize, SystemError> {
- let mut ldata = self.disc_data();
- ldata.receive_buf_common(tty, buf, flags, count, true)
- }
- }
|