ntty.rs 69 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123
  1. use core::intrinsics::likely;
  2. use core::ops::BitXor;
  3. use bitmap::{traits::BitMapOps, StaticBitmap};
  4. use alloc::sync::{Arc, Weak};
  5. use system_error::SystemError;
  6. use crate::{
  7. arch::ipc::signal::Signal,
  8. driver::tty::{
  9. termios::{ControlCharIndex, InputMode, LocalMode, OutputMode, Termios},
  10. tty_core::{EchoOperation, TtyCore, TtyCoreData, TtyFlag, TtyIoctlCmd, TtyPacketStatus},
  11. tty_driver::{TtyDriverFlag, TtyOperation},
  12. tty_job_control::TtyJobCtrlManager,
  13. },
  14. filesystem::vfs::file::FileMode,
  15. libs::{
  16. rwlock::RwLockReadGuard,
  17. spinlock::{SpinLock, SpinLockGuard},
  18. },
  19. mm::VirtAddr,
  20. net::event_poll::EPollEventType,
  21. process::ProcessManager,
  22. syscall::{user_access::UserBufferWriter, Syscall},
  23. };
  24. use super::TtyLineDiscipline;
  25. pub const NTTY_BUFSIZE: usize = 4096;
  26. pub const ECHO_COMMIT_WATERMARK: usize = 256;
  27. pub const ECHO_BLOCK: usize = 256;
  28. pub const ECHO_DISCARD_WATERMARK: usize = NTTY_BUFSIZE - (ECHO_BLOCK + 32);
  29. fn ntty_buf_mask(idx: usize) -> usize {
  30. return idx & (NTTY_BUFSIZE - 1);
  31. }
  32. #[derive(Debug)]
  33. pub struct NTtyLinediscipline {
  34. pub data: SpinLock<NTtyData>,
  35. }
  36. impl NTtyLinediscipline {
  37. #[inline]
  38. pub fn disc_data(&self) -> SpinLockGuard<NTtyData> {
  39. self.data.lock_irqsave()
  40. }
  41. #[inline]
  42. pub fn disc_data_try_lock(&self) -> Result<SpinLockGuard<NTtyData>, SystemError> {
  43. self.data.try_lock_irqsave()
  44. }
  45. fn ioctl_helper(&self, tty: Arc<TtyCore>, cmd: u32, arg: usize) -> Result<usize, SystemError> {
  46. match cmd {
  47. TtyIoctlCmd::TCXONC => {
  48. todo!()
  49. }
  50. TtyIoctlCmd::TCFLSH => {
  51. todo!()
  52. }
  53. _ => {
  54. return TtyCore::tty_mode_ioctl(tty.clone(), cmd, arg);
  55. }
  56. }
  57. }
  58. }
  59. #[derive(Debug)]
  60. pub struct NTtyData {
  61. /// 写者管理,tty只有一个写者,即ttydevice,所以不需要加锁
  62. /// 读取缓冲区的头指针,表示下一个将要接受进buf的字符的位置
  63. read_head: usize,
  64. /// 提交缓冲区的头指针,用于行规程处理
  65. commit_head: usize,
  66. /// 规范缓冲区的头指针,用于规范模式的处理
  67. canon_head: usize,
  68. /// 回显缓冲区的头指针,用于存储需要回显的字符
  69. echo_head: usize,
  70. /// 回显过程中用于提交的头指针
  71. echo_commit: usize,
  72. /// 标记回显字符的起始位置
  73. echo_mark: usize,
  74. /// 读者管理
  75. /// 读取字符的尾指针,即当前读取位置
  76. read_tail: usize,
  77. /// 行的起始位置
  78. line_start: usize,
  79. /// 预读字符数,用于处理控制字符
  80. lookahead_count: usize,
  81. // 更改以下六个标记时必须持有termios的锁
  82. /// Line-next 标志,表示下一个输入字符应当按字面处理
  83. lnext: bool,
  84. /// 擦除状态的标志
  85. erasing: bool,
  86. /// Raw 模式的标志
  87. raw: bool,
  88. /// Real raw 模式的标志
  89. real_raw: bool,
  90. /// 规范模式的标志
  91. icanon: bool,
  92. /// 是否开启echo
  93. echo: bool,
  94. /// 标志是否正在进行推送
  95. pushing: bool,
  96. /// 是否没有空间可写
  97. no_room: bool,
  98. /// 光标所在列
  99. cursor_column: u32,
  100. /// 规范模式下光标所在列
  101. canon_cursor_column: u32,
  102. /// 回显缓冲区的尾指针
  103. echo_tail: usize,
  104. /// 写者与读者共享
  105. read_buf: [u8; NTTY_BUFSIZE],
  106. echo_buf: [u8; NTTY_BUFSIZE],
  107. read_flags: StaticBitmap<NTTY_BUFSIZE>,
  108. char_map: StaticBitmap<256>,
  109. tty: Weak<TtyCore>,
  110. }
  111. impl NTtyData {
  112. pub fn new() -> Self {
  113. Self {
  114. read_head: 0,
  115. commit_head: 0,
  116. canon_head: 0,
  117. echo_head: 0,
  118. echo_commit: 0,
  119. echo_mark: 0,
  120. read_tail: 0,
  121. line_start: 0,
  122. lookahead_count: 0,
  123. lnext: false,
  124. erasing: false,
  125. raw: false,
  126. real_raw: false,
  127. icanon: false,
  128. pushing: false,
  129. echo: false,
  130. cursor_column: 0,
  131. canon_cursor_column: 0,
  132. echo_tail: 0,
  133. read_buf: [0; NTTY_BUFSIZE],
  134. echo_buf: [0; NTTY_BUFSIZE],
  135. read_flags: StaticBitmap::new(),
  136. char_map: StaticBitmap::new(),
  137. tty: Weak::default(),
  138. no_room: false,
  139. }
  140. }
  141. #[inline]
  142. pub fn read_cnt(&self) -> usize {
  143. self.read_head - self.read_tail
  144. }
  145. #[inline]
  146. pub fn read_at(&self, i: usize) -> u8 {
  147. let i = i & (NTTY_BUFSIZE - 1);
  148. self.read_buf[i]
  149. }
  150. /// ### 接收数据到NTTY
  151. pub fn receive_buf_common(
  152. &mut self,
  153. tty: Arc<TtyCore>,
  154. buf: &[u8],
  155. flags: Option<&[u8]>,
  156. mut count: usize,
  157. flow: bool,
  158. ) -> Result<usize, SystemError> {
  159. // 获取termios读锁
  160. let termios = tty.core().termios();
  161. let mut overflow;
  162. let mut n;
  163. let mut offset = 0;
  164. let mut recved = 0;
  165. loop {
  166. let tail = self.read_tail;
  167. let mut room = NTTY_BUFSIZE - (self.read_head - tail);
  168. if termios.input_mode.contains(InputMode::PARMRK) {
  169. room = (room + 2) / 3;
  170. }
  171. room -= 1;
  172. if room == 0 || room > NTTY_BUFSIZE {
  173. // 可能溢出
  174. overflow = self.icanon && self.canon_head == tail;
  175. if room > NTTY_BUFSIZE && overflow {
  176. self.read_head -= 1;
  177. }
  178. self.no_room = flow && !overflow;
  179. room = if overflow { !0 } else { 0 }
  180. } else {
  181. overflow = false;
  182. }
  183. n = count.min(room);
  184. if n == 0 {
  185. break;
  186. }
  187. if !overflow {
  188. if let Some(flags) = flags {
  189. self.receive_buf(tty.clone(), &buf[offset..], Some(&flags[offset..]), n);
  190. } else {
  191. self.receive_buf(tty.clone(), &buf[offset..], flags, n);
  192. }
  193. }
  194. offset += n;
  195. count -= n;
  196. recved += n;
  197. if tty.core().flags().contains(TtyFlag::LDISC_CHANGING) {
  198. break;
  199. }
  200. }
  201. // TODO: throttle
  202. Ok(recved)
  203. }
  204. pub fn receive_buf(
  205. &mut self,
  206. tty: Arc<TtyCore>,
  207. buf: &[u8],
  208. flags: Option<&[u8]>,
  209. count: usize,
  210. ) {
  211. let termios = tty.core().termios();
  212. let preops = termios.input_mode.contains(InputMode::ISTRIP)
  213. || termios.input_mode.contains(InputMode::IUCLC)
  214. || termios.local_mode.contains(LocalMode::IEXTEN);
  215. let look_ahead = self.lookahead_count.min(count);
  216. if self.real_raw {
  217. self.receive_buf_real_raw(buf, count);
  218. } else if self.raw || (termios.local_mode.contains(LocalMode::EXTPROC) && !preops) {
  219. self.receive_buf_raw(buf, flags, count);
  220. } else if tty.core().is_closing() && !termios.local_mode.contains(LocalMode::EXTPROC) {
  221. todo!()
  222. } else {
  223. if look_ahead > 0 {
  224. self.receive_buf_standard(tty.clone(), buf, flags, look_ahead, true);
  225. }
  226. if count > look_ahead {
  227. self.receive_buf_standard(tty.clone(), buf, flags, count - look_ahead, false);
  228. }
  229. // 刷新echo
  230. self.flush_echoes(tty.clone());
  231. tty.flush_chars(tty.core());
  232. }
  233. self.lookahead_count -= look_ahead;
  234. if self.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
  235. return;
  236. }
  237. self.commit_head = self.read_head;
  238. if self.read_cnt() > 0 {
  239. tty.core()
  240. .read_wq()
  241. .wakeup_any((EPollEventType::EPOLLIN | EPollEventType::EPOLLRDBAND).bits() as u64);
  242. }
  243. }
  244. fn receive_buf_real_raw(&mut self, buf: &[u8], mut count: usize) {
  245. let mut head = ntty_buf_mask(self.read_head);
  246. let mut n = count.min(NTTY_BUFSIZE - head);
  247. // 假如有一部分在队列头部,则这部分是拷贝尾部部分
  248. self.read_buf[head..(head + n)].copy_from_slice(&buf[0..n]);
  249. self.read_head += n;
  250. count -= n;
  251. let offset = n;
  252. // 假如有一部分在队列头部,则这部分是拷贝头部部分
  253. head = ntty_buf_mask(self.read_head);
  254. n = count.min(NTTY_BUFSIZE - head);
  255. self.read_buf[head..(head + n)].copy_from_slice(&buf[offset..(offset + n)]);
  256. self.read_head += n;
  257. }
  258. fn receive_buf_raw(&mut self, buf: &[u8], flags: Option<&[u8]>, mut count: usize) {
  259. // TTY_NORMAL 目前这部分未做,所以先占位置而不做抽象
  260. let mut flag = 1;
  261. let mut f_offset = 0;
  262. let mut c_offset = 0;
  263. while count != 0 {
  264. if flags.is_some() {
  265. flag = flags.as_ref().unwrap()[f_offset];
  266. f_offset += 1;
  267. }
  268. if likely(flag == 1) {
  269. self.read_buf[self.read_head] = buf[c_offset];
  270. c_offset += 1;
  271. self.read_head += 1;
  272. } else {
  273. todo!()
  274. }
  275. count -= 1;
  276. }
  277. }
  278. pub fn flush_echoes(&mut self, tty: Arc<TtyCore>) {
  279. let termios = tty.core().termios();
  280. if !termios.local_mode.contains(LocalMode::ECHO)
  281. && !termios.local_mode.contains(LocalMode::ECHONL)
  282. || self.echo_commit == self.echo_head
  283. {
  284. return;
  285. }
  286. self.echo_commit = self.echo_head;
  287. drop(termios);
  288. let _ = self.echoes(tty);
  289. }
  290. pub fn receive_buf_standard(
  291. &mut self,
  292. tty: Arc<TtyCore>,
  293. buf: &[u8],
  294. flags: Option<&[u8]>,
  295. mut count: usize,
  296. lookahead_done: bool,
  297. ) {
  298. let termios = tty.core().termios();
  299. if flags.is_some() {
  300. todo!("ntty recv buf flags todo");
  301. }
  302. let mut offset = 0;
  303. while count > 0 {
  304. if offset >= buf.len() {
  305. break;
  306. }
  307. let mut c = buf[offset];
  308. offset += 1;
  309. if self.lnext {
  310. // 将下一个字符当做字面值处理
  311. self.lnext = false;
  312. if termios.input_mode.contains(InputMode::ISTRIP) {
  313. c &= 0x7f;
  314. }
  315. if termios.input_mode.contains(InputMode::IUCLC)
  316. && termios.local_mode.contains(LocalMode::IEXTEN)
  317. {
  318. c = (c as char).to_ascii_lowercase() as u8;
  319. self.receive_char(c, tty.clone())
  320. }
  321. continue;
  322. }
  323. if termios.input_mode.contains(InputMode::ISTRIP) {
  324. c &= 0x7f;
  325. }
  326. if termios.input_mode.contains(InputMode::IUCLC)
  327. && termios.local_mode.contains(LocalMode::IEXTEN)
  328. {
  329. c = (c as char).to_ascii_lowercase() as u8;
  330. }
  331. if termios.local_mode.contains(LocalMode::EXTPROC) {
  332. self.add_read_byte(c);
  333. continue;
  334. }
  335. if self.char_map.get(c as usize).unwrap() {
  336. // 特殊字符
  337. self.receive_special_char(c, tty.clone(), lookahead_done)
  338. } else {
  339. self.receive_char(c, tty.clone());
  340. }
  341. count -= 1;
  342. }
  343. }
  344. #[inline(never)]
  345. pub fn receive_special_char(&mut self, mut c: u8, tty: Arc<TtyCore>, lookahead_done: bool) {
  346. let is_flow_ctrl = self.is_flow_ctrl_char(tty.clone(), c, lookahead_done);
  347. let termios = tty.core().termios();
  348. // 启用软件流控,并且该字符已经当做软件流控字符处理
  349. if termios.input_mode.contains(InputMode::IXON) && is_flow_ctrl {
  350. return;
  351. }
  352. if termios.local_mode.contains(LocalMode::ISIG) {
  353. if c == termios.control_characters[ControlCharIndex::VINTR] {
  354. self.recv_sig_char(tty.clone(), &termios, Signal::SIGINT, c);
  355. return;
  356. }
  357. if c == termios.control_characters[ControlCharIndex::VQUIT] {
  358. self.recv_sig_char(tty.clone(), &termios, Signal::SIGQUIT, c);
  359. return;
  360. }
  361. if c == termios.control_characters[ControlCharIndex::VSUSP] {
  362. self.recv_sig_char(tty.clone(), &termios, Signal::SIGTSTP, c);
  363. return;
  364. }
  365. }
  366. let flow = tty.core().flow_irqsave();
  367. if flow.stopped
  368. && !flow.tco_stopped
  369. && termios.input_mode.contains(InputMode::IXON)
  370. && termios.input_mode.contains(InputMode::IXANY)
  371. {
  372. tty.tty_start();
  373. self.process_echoes(tty.clone());
  374. }
  375. drop(flow);
  376. if c == b'\r' {
  377. if termios.input_mode.contains(InputMode::IGNCR) {
  378. // 忽略
  379. return;
  380. }
  381. if termios.input_mode.contains(InputMode::ICRNL) {
  382. // 映射为换行
  383. c = b'\n';
  384. }
  385. } else if c == b'\n' && termios.input_mode.contains(InputMode::INLCR) {
  386. // 映射为回车
  387. c = b'\r';
  388. }
  389. if self.icanon {
  390. if c == termios.control_characters[ControlCharIndex::VERASE]
  391. || c == termios.control_characters[ControlCharIndex::VKILL]
  392. || (c == termios.control_characters[ControlCharIndex::VWERASE]
  393. && termios.local_mode.contains(LocalMode::IEXTEN))
  394. {
  395. self.eraser(c, &termios);
  396. self.commit_echoes(tty.clone());
  397. return;
  398. }
  399. if c == termios.control_characters[ControlCharIndex::VLNEXT]
  400. && termios.local_mode.contains(LocalMode::IEXTEN)
  401. {
  402. self.lnext = true;
  403. if termios.local_mode.contains(LocalMode::ECHO) {
  404. self.finish_erasing();
  405. if termios.local_mode.contains(LocalMode::ECHOCTL) {
  406. self.echo_char_raw(b'^');
  407. self.echo_char_raw(8);
  408. self.commit_echoes(tty.clone());
  409. }
  410. }
  411. return;
  412. }
  413. if c == termios.control_characters[ControlCharIndex::VREPRINT]
  414. && termios.local_mode.contains(LocalMode::ECHO)
  415. && termios.local_mode.contains(LocalMode::IEXTEN)
  416. {
  417. let mut tail = self.canon_head;
  418. self.finish_erasing();
  419. self.echo_char(c, &termios);
  420. self.echo_char_raw(b'\n');
  421. while ntty_buf_mask(tail) != ntty_buf_mask(self.read_head) {
  422. self.echo_char(self.read_buf[ntty_buf_mask(tail)], &termios);
  423. tail += 1;
  424. }
  425. self.commit_echoes(tty.clone());
  426. return;
  427. }
  428. if c == b'\n' {
  429. if termios.local_mode.contains(LocalMode::ECHO)
  430. || termios.local_mode.contains(LocalMode::ECHONL)
  431. {
  432. self.echo_char_raw(b'\n');
  433. self.commit_echoes(tty.clone());
  434. }
  435. self.read_flags.set(ntty_buf_mask(self.read_head), true);
  436. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  437. self.read_head += 1;
  438. self.canon_head = self.read_head;
  439. tty.core().read_wq().wakeup_any(
  440. (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
  441. );
  442. return;
  443. }
  444. if c == termios.control_characters[ControlCharIndex::VEOF] {
  445. c = ControlCharIndex::DISABLE_CHAR;
  446. self.read_flags.set(ntty_buf_mask(self.read_head), true);
  447. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  448. self.read_head += 1;
  449. self.canon_head = self.read_head;
  450. tty.core().read_wq().wakeup_any(
  451. (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
  452. );
  453. return;
  454. }
  455. if c == termios.control_characters[ControlCharIndex::VEOL]
  456. || (c == termios.control_characters[ControlCharIndex::VEOL2]
  457. && termios.local_mode.contains(LocalMode::IEXTEN))
  458. {
  459. if termios.local_mode.contains(LocalMode::ECHO) {
  460. if self.canon_head == self.read_head {
  461. self.add_echo_byte(EchoOperation::Start.to_u8());
  462. self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
  463. }
  464. self.echo_char(c, &termios);
  465. self.commit_echoes(tty.clone());
  466. }
  467. if c == 0o377 && termios.input_mode.contains(InputMode::PARMRK) {
  468. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  469. self.read_head += 1;
  470. }
  471. self.read_flags.set(ntty_buf_mask(self.read_head), true);
  472. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  473. self.read_head += 1;
  474. self.canon_head = self.read_head;
  475. tty.core().read_wq().wakeup_any(
  476. (EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64,
  477. );
  478. return;
  479. }
  480. }
  481. if termios.local_mode.contains(LocalMode::ECHO) {
  482. self.finish_erasing();
  483. if c == b'\n' {
  484. self.echo_char_raw(b'\n');
  485. } else {
  486. if self.canon_head == self.read_head {
  487. self.add_echo_byte(EchoOperation::Start.to_u8());
  488. self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
  489. }
  490. self.echo_char(c, &termios);
  491. }
  492. self.commit_echoes(tty.clone());
  493. }
  494. if c == 0o377 && termios.input_mode.contains(InputMode::PARMRK) {
  495. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  496. self.read_head += 1;
  497. }
  498. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  499. self.read_head += 1;
  500. }
  501. /// ## ntty默认eraser function
  502. #[inline(never)]
  503. fn eraser(&mut self, mut c: u8, termios: &RwLockReadGuard<Termios>) {
  504. if self.read_head == self.canon_head {
  505. return;
  506. }
  507. let erase = c == termios.control_characters[ControlCharIndex::VERASE];
  508. let werase = c == termios.control_characters[ControlCharIndex::VWERASE];
  509. let kill = !erase && !werase;
  510. if kill {
  511. if !termios.local_mode.contains(LocalMode::ECHO) {
  512. self.read_head = self.canon_head;
  513. return;
  514. }
  515. if !termios.local_mode.contains(LocalMode::ECHOK)
  516. || !termios.local_mode.contains(LocalMode::ECHOKE)
  517. || !termios.local_mode.contains(LocalMode::ECHOE)
  518. {
  519. self.read_head = self.canon_head;
  520. if self.erasing {
  521. self.echo_char_raw(c);
  522. self.erasing = false;
  523. }
  524. self.echo_char(c, termios);
  525. if termios.local_mode.contains(LocalMode::ECHOK) {
  526. // 添加新行
  527. self.echo_char_raw(b'\n');
  528. }
  529. return;
  530. }
  531. }
  532. let mut head;
  533. let mut cnt;
  534. while ntty_buf_mask(self.read_head) != ntty_buf_mask(self.canon_head) {
  535. head = self.read_head;
  536. loop {
  537. // 消除多字节字符
  538. head -= 1;
  539. c = self.read_buf[ntty_buf_mask(head)];
  540. if !(Self::is_continuation(c, termios)
  541. && ntty_buf_mask(head) != ntty_buf_mask(self.canon_head))
  542. {
  543. break;
  544. }
  545. }
  546. if Self::is_continuation(c, termios) {
  547. break;
  548. }
  549. if werase {
  550. todo!()
  551. }
  552. cnt = self.read_head - head;
  553. self.read_head = head;
  554. if termios.local_mode.contains(LocalMode::ECHO) {
  555. if termios.local_mode.contains(LocalMode::ECHOPRT) {
  556. if !self.erasing {
  557. self.echo_char_raw(b'\\');
  558. self.erasing = true;
  559. }
  560. self.echo_char(c, termios);
  561. cnt -= 1;
  562. while cnt > 0 {
  563. cnt -= 1;
  564. head += 1;
  565. self.echo_char_raw(self.read_buf[ntty_buf_mask(head)]);
  566. self.add_echo_byte(EchoOperation::Start.to_u8());
  567. self.add_echo_byte(EchoOperation::MoveBackCol.to_u8());
  568. }
  569. } else if erase && !termios.local_mode.contains(LocalMode::ECHOE) {
  570. self.echo_char(
  571. termios.control_characters[ControlCharIndex::VERASE],
  572. termios,
  573. );
  574. } else if c == b'\t' {
  575. let mut num_chars = 0;
  576. let mut after_tab = false;
  577. let mut tail = self.read_head;
  578. while ntty_buf_mask(tail) != ntty_buf_mask(self.canon_head) {
  579. tail -= 1;
  580. c = self.read_buf[ntty_buf_mask(tail)];
  581. if c == b'\t' {
  582. after_tab = true;
  583. break;
  584. } else if (c as char).is_control() {
  585. if termios.local_mode.contains(LocalMode::ECHOCTL) {
  586. num_chars += 2;
  587. }
  588. } else if !Self::is_continuation(c, termios) {
  589. num_chars += 1;
  590. }
  591. }
  592. self.echo_erase_tab(num_chars, after_tab);
  593. } else {
  594. if (c as char).is_control() && termios.local_mode.contains(LocalMode::ECHOCTL) {
  595. // 8 => '\b'
  596. self.echo_char_raw(8);
  597. self.echo_char_raw(b' ');
  598. self.echo_char_raw(8);
  599. }
  600. if !(c as char).is_control() || termios.local_mode.contains(LocalMode::ECHOCTL)
  601. {
  602. // 8 => '\b'
  603. self.echo_char_raw(8);
  604. self.echo_char_raw(b' ');
  605. self.echo_char_raw(8);
  606. }
  607. }
  608. }
  609. if erase {
  610. break;
  611. }
  612. }
  613. if self.read_head == self.canon_head && termios.local_mode.contains(LocalMode::ECHO) {
  614. self.finish_erasing();
  615. }
  616. }
  617. fn finish_erasing(&mut self) {
  618. if self.erasing {
  619. self.echo_char_raw(b'/');
  620. self.erasing = false;
  621. }
  622. }
  623. fn echo_erase_tab(&mut self, mut num: u8, after_tab: bool) {
  624. self.add_echo_byte(EchoOperation::Start.to_u8());
  625. self.add_echo_byte(EchoOperation::EraseTab.to_u8());
  626. num &= 7;
  627. if after_tab {
  628. num |= 0x80;
  629. }
  630. self.add_echo_byte(num);
  631. }
  632. /// ## 多字节字符检测
  633. /// 检测是否为多字节字符的后续字节
  634. fn is_continuation(c: u8, termios: &RwLockReadGuard<Termios>) -> bool {
  635. return termios.input_mode.contains(InputMode::IUTF8) && (c & 0xc0) == 0x80;
  636. }
  637. /// ## 该字符是否已经当做流控字符处理
  638. pub fn is_flow_ctrl_char(&mut self, tty: Arc<TtyCore>, c: u8, lookahead_done: bool) -> bool {
  639. let termios = tty.core().termios();
  640. if !(termios.control_characters[ControlCharIndex::VSTART] == c
  641. || termios.control_characters[ControlCharIndex::VSTOP] == c)
  642. {
  643. return false;
  644. }
  645. if lookahead_done {
  646. return true;
  647. }
  648. if termios.control_characters[ControlCharIndex::VSTART] == c {
  649. tty.tty_start();
  650. self.process_echoes(tty.clone());
  651. return true;
  652. } else {
  653. tty.tty_stop();
  654. return true;
  655. }
  656. }
  657. /// ## 接收到信号字符时的处理
  658. fn recv_sig_char(
  659. &mut self,
  660. tty: Arc<TtyCore>,
  661. termios: &RwLockReadGuard<Termios>,
  662. signal: Signal,
  663. c: u8,
  664. ) {
  665. self.input_signal(tty.clone(), termios, signal);
  666. if termios.input_mode.contains(InputMode::IXON) {
  667. tty.tty_start();
  668. }
  669. if termios.local_mode.contains(LocalMode::ECHO) {
  670. self.echo_char(c, termios);
  671. self.commit_echoes(tty);
  672. } else {
  673. self.process_echoes(tty);
  674. }
  675. }
  676. /// ## 处理输入信号
  677. pub fn input_signal(
  678. &mut self,
  679. tty: Arc<TtyCore>,
  680. termios: &RwLockReadGuard<Termios>,
  681. signal: Signal,
  682. ) {
  683. // 先处理信号
  684. let mut ctrl_info = tty.core().contorl_info_irqsave();
  685. let pg = ctrl_info.pgid;
  686. if let Some(pg) = pg {
  687. let _ = Syscall::kill(pg, signal as i32);
  688. }
  689. ctrl_info.pgid = None;
  690. ctrl_info.session = None;
  691. if !termios.local_mode.contains(LocalMode::NOFLSH) {
  692. // 重置
  693. self.echo_head = 0;
  694. self.echo_tail = 0;
  695. self.echo_mark = 0;
  696. self.echo_commit = 0;
  697. let _ = tty.flush_buffer(tty.core());
  698. self.read_head = 0;
  699. self.canon_head = 0;
  700. self.read_tail = 0;
  701. self.line_start = 0;
  702. self.erasing = false;
  703. self.read_flags.set_all(false);
  704. self.pushing = false;
  705. self.lookahead_count = 0;
  706. if tty.core().link().is_some() {
  707. self.packet_mode_flush(tty.core());
  708. }
  709. }
  710. }
  711. pub fn receive_char(&mut self, c: u8, tty: Arc<TtyCore>) {
  712. let termios = tty.core().termios();
  713. if termios.local_mode.contains(LocalMode::ECHO) {
  714. if self.erasing {
  715. self.add_echo_byte(b'/');
  716. self.erasing = false;
  717. }
  718. if self.canon_head == self.read_head {
  719. self.add_echo_byte(EchoOperation::Start.to_u8());
  720. self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
  721. }
  722. self.echo_char(c, &termios);
  723. self.commit_echoes(tty.clone());
  724. }
  725. if c == 0o377 && tty.core().termios().input_mode.contains(InputMode::PARMRK) {
  726. self.add_read_byte(c);
  727. }
  728. self.add_read_byte(c);
  729. }
  730. pub fn echo_char(&mut self, c: u8, termios: &RwLockReadGuard<Termios>) {
  731. if c == EchoOperation::Start.to_u8() {
  732. self.add_echo_byte(EchoOperation::Start.to_u8());
  733. self.add_echo_byte(EchoOperation::Start.to_u8());
  734. } else {
  735. if termios.local_mode.contains(LocalMode::ECHOCTL)
  736. && (c as char).is_control()
  737. && c != b'\t'
  738. {
  739. self.add_echo_byte(EchoOperation::Start.to_u8());
  740. }
  741. self.add_echo_byte(c);
  742. }
  743. }
  744. pub fn echo_char_raw(&mut self, c: u8) {
  745. if c == EchoOperation::Start.to_u8() {
  746. self.add_echo_byte(EchoOperation::Start.to_u8());
  747. self.add_echo_byte(EchoOperation::Start.to_u8());
  748. } else {
  749. self.add_echo_byte(c);
  750. }
  751. }
  752. /// ## 提交echobuf里的数据显示
  753. pub fn commit_echoes(&mut self, tty: Arc<TtyCore>) {
  754. let head = self.echo_head;
  755. self.echo_mark = head;
  756. let old = self.echo_commit - self.echo_tail;
  757. // 需要echo的字符个数
  758. let nr = head - self.echo_tail;
  759. if nr < ECHO_COMMIT_WATERMARK || nr % ECHO_BLOCK > old % ECHO_BLOCK {
  760. return;
  761. }
  762. self.echo_commit = head;
  763. let echoed = self.echoes(tty.clone());
  764. if echoed.is_ok() && echoed.unwrap() > 0 {
  765. tty.flush_chars(tty.core());
  766. }
  767. }
  768. pub fn add_echo_byte(&mut self, c: u8) {
  769. self.echo_buf[ntty_buf_mask(self.echo_head)] = c;
  770. self.echo_head += 1;
  771. }
  772. pub fn add_read_byte(&mut self, c: u8) {
  773. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  774. self.read_head += 1;
  775. }
  776. /// ### 将read_buffer的部分值置0
  777. ///
  778. /// 只会在规范模式和禁用echo下执行
  779. #[inline]
  780. pub fn zero_buffer(&mut self, offset: usize, size: usize) {
  781. let offset = offset & (NTTY_BUFSIZE - 1);
  782. if self.icanon && !self.echo {
  783. let n = offset + size;
  784. if n > NTTY_BUFSIZE {
  785. for c in &mut self.read_buf[offset..NTTY_BUFSIZE] {
  786. *c = 0
  787. }
  788. for c in &mut self.read_buf[0..(n - NTTY_BUFSIZE)] {
  789. *c = 0
  790. }
  791. } else {
  792. for c in &mut self.read_buf[offset..n] {
  793. *c = 0
  794. }
  795. };
  796. }
  797. }
  798. /// ## 从ntty中拷贝数据
  799. ///
  800. /// ### 参数
  801. ///
  802. /// ### to: 存储数据
  803. /// ### tail: 读取尾
  804. pub fn ntty_copy(
  805. &mut self,
  806. to: &mut [u8],
  807. tail: usize,
  808. n: &mut usize,
  809. ) -> Result<(), SystemError> {
  810. if to.len() < *n {
  811. *n = to.len();
  812. // return Err(SystemError::EINVAL);
  813. }
  814. if tail > NTTY_BUFSIZE {
  815. return Err(SystemError::EINVAL);
  816. }
  817. let size = NTTY_BUFSIZE - tail;
  818. if size < *n {
  819. // 有一部分数据在头部,则先拷贝后面部分,再拷贝头部
  820. // TODO: tty审计?
  821. to[0..size].copy_from_slice(&self.read_buf[tail..(tail + size)]);
  822. to[size..(*n)].copy_from_slice(&self.read_buf[0..(*n - size)]);
  823. } else {
  824. to[..*n].copy_from_slice(&self.read_buf[tail..(tail + *n)])
  825. }
  826. self.zero_buffer(tail, *n);
  827. Ok(())
  828. }
  829. /// ## 规范模式下跳过EOF
  830. pub fn canon_skip_eof(&mut self) {
  831. // 没有数据
  832. if self.read_tail == self.canon_head {
  833. return;
  834. }
  835. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  836. // 查看read_flags是否读取位置为特殊字符
  837. if !self.read_flags.get(tail).unwrap() {
  838. return;
  839. }
  840. // 确保读取位置是'\0'字符
  841. if self.read_buf[tail] != ControlCharIndex::DISABLE_CHAR {
  842. return;
  843. }
  844. // 处理该字符,将read_flagsw该位清除
  845. self.read_flags.set(tail, false);
  846. // 读取位置+1,即跳过该字符不做处理
  847. self.read_tail += 1;
  848. }
  849. /// ## 在规范模式(canonical mode)下从读缓冲中复制一行
  850. ///
  851. /// 一次只拷贝一行
  852. ///
  853. /// ## 参数
  854. /// ### dst: 存放数据
  855. /// ### nr: 需要拷贝的数据大小
  856. ///
  857. /// ## 返回值
  858. /// ### true: 表示一行未结束并且还有数据可读
  859. /// ### false: 一行已结束或者没有数据可读
  860. pub fn canon_copy_from_read_buf(
  861. &mut self,
  862. dst: &mut [u8],
  863. nr: &mut usize,
  864. offset: &mut usize,
  865. ) -> Result<bool, SystemError> {
  866. if *nr == 0 {
  867. return Ok(false);
  868. }
  869. let canon_head = self.canon_head;
  870. // 取得能够读到的字符数,即canon_head - self.read_tail和nr最小值
  871. let mut n = (*nr).min(canon_head - self.read_tail);
  872. // 获得读尾index
  873. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  874. // 避免越界,这个size才是实际读取大小
  875. let size = if tail + n > NTTY_BUFSIZE {
  876. NTTY_BUFSIZE
  877. } else {
  878. tail + n
  879. };
  880. // 找到eol的坐标
  881. let tmp = self.read_flags.next_index(tail);
  882. // 找到的话即为坐标,未找到的话即为NTTY_BUFSIZE
  883. let mut eol = if let Some(tmp) = tmp { tmp } else { size };
  884. if eol > size {
  885. eol = size
  886. }
  887. // 是否需要绕回缓冲区头部
  888. let more = n - (size - tail);
  889. // 是否找到eol
  890. let found = if eol == NTTY_BUFSIZE && more > 0 {
  891. // 需要返回头部
  892. let ret = self.read_flags.first_index();
  893. if let Some(tmp) = ret {
  894. // 在头部范围内找到eol
  895. if tmp < more {
  896. eol = tmp;
  897. }
  898. } else {
  899. eol = more;
  900. }
  901. eol != more
  902. } else {
  903. // 不需要返回头部
  904. eol != size
  905. };
  906. n = eol - tail;
  907. if n > NTTY_BUFSIZE {
  908. // 减法溢出则加上BUFSIZE即可限制在0-NTTY_BUFSIZE内
  909. n += NTTY_BUFSIZE;
  910. }
  911. // 规范模式下实际扫描过的字符数,需要将eol计算在内
  912. let count = if found { n + 1 } else { n };
  913. // 表示这一行未结束
  914. if !found || self.read_at(eol) != ControlCharIndex::DISABLE_CHAR {
  915. n = count;
  916. }
  917. self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
  918. *nr -= n;
  919. *offset += n;
  920. if found {
  921. self.read_flags.set(eol, false);
  922. }
  923. self.read_tail += count;
  924. if found {
  925. if !self.pushing {
  926. self.line_start = self.read_tail;
  927. } else {
  928. self.pushing = false;
  929. }
  930. // todo: 审计?
  931. return Ok(false);
  932. }
  933. // 这里是表示没有找到eol,根据是否还有数据可读返回
  934. Ok(self.read_tail != canon_head)
  935. }
  936. /// ## 根据终端的模式和输入缓冲区中的数据量,判断是否可读字符
  937. pub fn input_available(&self, termios: RwLockReadGuard<Termios>, poll: bool) -> bool {
  938. // 计算最小字符数
  939. let amt = if poll
  940. && termios.control_characters[ControlCharIndex::VTIME] as u32 == 0
  941. && termios.control_characters[ControlCharIndex::VMIN] as u32 != 0
  942. {
  943. termios.control_characters[ControlCharIndex::VMIN] as usize
  944. } else {
  945. 1
  946. };
  947. // 规范模式且非拓展
  948. if self.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
  949. return self.canon_head != self.read_tail;
  950. } else {
  951. return (self.commit_head - self.read_tail) >= amt;
  952. }
  953. }
  954. /// ## 非规范模式下从read_buf读取数据
  955. ///
  956. /// ## 参数
  957. /// ### termios: tty对应的termioss读锁守卫
  958. /// ### dst: 存储读取数据
  959. /// ### nr: 读取长度
  960. ///
  961. /// ## 返回值
  962. /// ### true: 还有更多数据可读
  963. /// ### false: 无更多数据可读
  964. pub fn copy_from_read_buf(
  965. &mut self,
  966. termios: RwLockReadGuard<Termios>,
  967. dst: &mut [u8],
  968. nr: &mut usize,
  969. offset: &mut usize,
  970. ) -> Result<bool, SystemError> {
  971. let head = self.commit_head;
  972. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  973. // 计算出可读的字符数
  974. let mut n = (NTTY_BUFSIZE - tail).min(head - self.read_tail);
  975. n = n.min(*nr);
  976. if n > 0 {
  977. // 拷贝数据
  978. self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
  979. // todo:审计?
  980. self.read_tail += n;
  981. // 是否只读取了eof
  982. let eof =
  983. n == 1 && self.read_buf[tail] == termios.control_characters[ControlCharIndex::VEOF];
  984. if termios.local_mode.contains(LocalMode::EXTPROC)
  985. && self.icanon
  986. && eof
  987. && head == self.read_tail
  988. {
  989. return Ok(false);
  990. }
  991. *nr -= n;
  992. *offset += n;
  993. return Ok(head != self.read_tail);
  994. }
  995. Ok(false)
  996. }
  997. /// ## 用于处理带有 OPOST(Output Post-processing)标志的输出块的函数
  998. /// OPOST 是 POSIX 终端驱动器标志之一,用于指定在写入终端设备之前对输出数据进行一些后期处理。
  999. pub fn process_output_block(
  1000. &mut self,
  1001. core: &TtyCoreData,
  1002. termios: RwLockReadGuard<Termios>,
  1003. buf: &[u8],
  1004. nr: usize,
  1005. ) -> Result<usize, SystemError> {
  1006. let mut nr = nr;
  1007. let tty = self.tty.upgrade().unwrap();
  1008. let space = tty.write_room(tty.core());
  1009. // 如果读取数量大于了可用空间,则取最小的为真正的写入数量
  1010. if nr > space {
  1011. nr = space
  1012. }
  1013. let mut cnt = 0;
  1014. for (i, c) in buf.iter().enumerate().take(nr) {
  1015. cnt = i;
  1016. let c = *c;
  1017. if c as usize == 8 {
  1018. // 表示退格
  1019. if self.cursor_column > 0 {
  1020. self.cursor_column -= 1;
  1021. }
  1022. continue;
  1023. }
  1024. match c as char {
  1025. '\n' => {
  1026. if termios.output_mode.contains(OutputMode::ONLRET) {
  1027. // 将回车映射为\n,即将\n换为回车
  1028. self.cursor_column = 0;
  1029. }
  1030. if termios.output_mode.contains(OutputMode::ONLCR) {
  1031. // 输出时将\n换为\r\n
  1032. break;
  1033. }
  1034. self.canon_cursor_column = self.cursor_column;
  1035. }
  1036. '\r' => {
  1037. if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
  1038. // 光标已经在第0列,则不输出回车符
  1039. break;
  1040. }
  1041. if termios.output_mode.contains(OutputMode::OCRNL) {
  1042. break;
  1043. }
  1044. self.canon_cursor_column = self.cursor_column;
  1045. }
  1046. '\t' => {
  1047. break;
  1048. }
  1049. _ => {
  1050. // 判断是否为控制字符
  1051. if !(c as char).is_control() {
  1052. if termios.output_mode.contains(OutputMode::OLCUC) {
  1053. break;
  1054. }
  1055. // 判断是否为utf8模式下的连续字符
  1056. if !(termios.input_mode.contains(InputMode::IUTF8)
  1057. && (c as usize) & 0xc0 == 0x80)
  1058. {
  1059. self.cursor_column += 1;
  1060. }
  1061. }
  1062. }
  1063. }
  1064. }
  1065. drop(termios);
  1066. return tty.write(core, buf, cnt);
  1067. }
  1068. /// ## 处理回显
  1069. pub fn process_echoes(&mut self, tty: Arc<TtyCore>) {
  1070. if self.echo_mark == self.echo_tail {
  1071. return;
  1072. }
  1073. self.echo_commit = self.echo_mark;
  1074. let echoed = self.echoes(tty.clone());
  1075. if echoed.is_ok() && echoed.unwrap() > 0 {
  1076. tty.flush_chars(tty.core());
  1077. }
  1078. }
  1079. #[inline(never)]
  1080. pub fn echoes(&mut self, tty: Arc<TtyCore>) -> Result<usize, SystemError> {
  1081. let mut space = tty.write_room(tty.core());
  1082. let ospace = space;
  1083. let termios = tty.core().termios();
  1084. let core = tty.core();
  1085. let mut tail = self.echo_tail;
  1086. while ntty_buf_mask(self.echo_commit) != ntty_buf_mask(tail) {
  1087. let c = self.echo_buf[ntty_buf_mask(tail)];
  1088. if EchoOperation::from_u8(c) == EchoOperation::Start {
  1089. if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 1) {
  1090. self.echo_tail = tail;
  1091. return Ok(ospace - space);
  1092. }
  1093. // 获取到start,之后取第一个作为op
  1094. let op = EchoOperation::from_u8(self.echo_buf[ntty_buf_mask(tail + 1)]);
  1095. match op {
  1096. EchoOperation::Start => {
  1097. if space == 0 {
  1098. break;
  1099. }
  1100. if tty
  1101. .put_char(tty.core(), EchoOperation::Start.to_u8())
  1102. .is_err()
  1103. {
  1104. tty.write(core, &[EchoOperation::Start.to_u8()], 1)?;
  1105. }
  1106. self.cursor_column += 1;
  1107. space -= 1;
  1108. tail += 2;
  1109. }
  1110. EchoOperation::MoveBackCol => {
  1111. if self.cursor_column > 0 {
  1112. self.cursor_column -= 1;
  1113. }
  1114. tail += 2;
  1115. }
  1116. EchoOperation::SetCanonCol => {
  1117. self.canon_cursor_column = self.cursor_column;
  1118. tail += 2;
  1119. }
  1120. EchoOperation::EraseTab => {
  1121. if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 2) {
  1122. self.echo_tail = tail;
  1123. return Ok(ospace - space);
  1124. }
  1125. // 要擦除的制表符所占用的列数
  1126. let mut char_num = self.echo_buf[ntty_buf_mask(tail + 2)] as usize;
  1127. /*
  1128. 如果 num_chars 的最高位(0x80)未设置,
  1129. 表示这是从输入的起始位置而不是从先前的制表符开始计算的列数。
  1130. 在这种情况下,将 num_chars 与 ldata->canon_column 相加,否则,列数就是正常的制表符列数。
  1131. */
  1132. if char_num & 0x80 == 0 {
  1133. char_num += self.canon_cursor_column as usize;
  1134. }
  1135. // 计算要回退的列数,即制表符宽度减去实际占用的列数
  1136. let mut num_bs = 8 - (char_num & 7);
  1137. if num_bs > space {
  1138. // 表示左边没有足够空间回退
  1139. break;
  1140. }
  1141. space -= num_bs;
  1142. while num_bs != 0 {
  1143. num_bs -= 1;
  1144. // 8 => '\b'
  1145. if tty.put_char(tty.core(), 8).is_err() {
  1146. tty.write(core, &[8], 1)?;
  1147. }
  1148. if self.cursor_column > 0 {
  1149. self.cursor_column -= 1;
  1150. }
  1151. }
  1152. // 已经读取了 tail tail+1 tail+2,所以这里偏移加3
  1153. tail += 3;
  1154. }
  1155. EchoOperation::Undefined(ch) => {
  1156. match ch {
  1157. 8 => {
  1158. if tty.put_char(tty.core(), 8).is_err() {
  1159. tty.write(core, &[8], 1)?;
  1160. }
  1161. if tty.put_char(tty.core(), b' ').is_err() {
  1162. tty.write(core, b" ", 1)?;
  1163. }
  1164. self.cursor_column -= 1;
  1165. space -= 1;
  1166. tail += 1;
  1167. }
  1168. _ => {
  1169. // 不是特殊字节码,则表示控制字符 例如 ^C
  1170. if space < 2 {
  1171. break;
  1172. }
  1173. if tty.put_char(tty.core(), b'^').is_err() {
  1174. tty.write(core, b"^", 1)?;
  1175. }
  1176. if tty.put_char(tty.core(), ch ^ 0o100).is_err() {
  1177. tty.write(core, &[ch ^ 0o100], 1)?;
  1178. }
  1179. self.cursor_column += 2;
  1180. space -= 2;
  1181. tail += 2;
  1182. }
  1183. }
  1184. }
  1185. }
  1186. } else {
  1187. if termios.output_mode.contains(OutputMode::OPOST) {
  1188. let ret = self.do_output_char(tty.clone(), c, space);
  1189. if ret.is_err() {
  1190. break;
  1191. }
  1192. space -= ret.unwrap();
  1193. } else {
  1194. if space == 0 {
  1195. break;
  1196. }
  1197. if tty.put_char(tty.core(), c).is_err() {
  1198. tty.write(core, &[c], 1)?;
  1199. }
  1200. space -= 1;
  1201. }
  1202. tail += 1;
  1203. }
  1204. }
  1205. // 如果回显缓冲区接近满(在下一次提交之前可能会发生回显溢出的情况),则丢弃足够的尾部数据以防止随后的溢出。
  1206. while self.echo_commit > tail && self.echo_commit - tail >= ECHO_DISCARD_WATERMARK {
  1207. if self.echo_buf[ntty_buf_mask(tail)] == EchoOperation::Start.to_u8() {
  1208. if self.echo_buf[ntty_buf_mask(tail + 1)] == EchoOperation::EraseTab.to_u8() {
  1209. tail += 3;
  1210. } else {
  1211. tail += 2;
  1212. }
  1213. } else {
  1214. tail += 1;
  1215. }
  1216. }
  1217. self.echo_tail = tail;
  1218. return Ok(ospace - space);
  1219. }
  1220. /// ## 处理输出字符(带有 OPOST 处理)
  1221. pub fn process_output(&mut self, tty: Arc<TtyCore>, c: u8) -> bool {
  1222. let space = tty.write_room(tty.core());
  1223. if self.do_output_char(tty, c, space).is_err() {
  1224. return false;
  1225. }
  1226. true
  1227. }
  1228. // ## 设置带有 OPOST 处理的tty输出一个字符
  1229. pub fn do_output_char(
  1230. &mut self,
  1231. tty: Arc<TtyCore>,
  1232. c: u8,
  1233. space: usize,
  1234. ) -> Result<usize, SystemError> {
  1235. if space == 0 {
  1236. return Err(SystemError::ENOBUFS);
  1237. }
  1238. let termios = tty.core().termios();
  1239. let core = tty.core();
  1240. let mut c = c;
  1241. if c as usize == 8 {
  1242. // 表示退格
  1243. if self.cursor_column > 0 {
  1244. self.cursor_column -= 1;
  1245. }
  1246. if tty.put_char(tty.core(), c).is_err() {
  1247. tty.write(core, &[c], 1)?;
  1248. }
  1249. return Ok(1);
  1250. }
  1251. match c as char {
  1252. '\n' => {
  1253. if termios.output_mode.contains(OutputMode::ONLRET) {
  1254. // 回车符
  1255. self.cursor_column = 0;
  1256. }
  1257. if termios.output_mode.contains(OutputMode::ONLCR) {
  1258. // 映射为“\r\n”
  1259. if space < 2 {
  1260. return Err(SystemError::ENOBUFS);
  1261. }
  1262. self.cursor_column = 0;
  1263. self.canon_cursor_column = 0;
  1264. // 通过驱动写入
  1265. tty.write(core, "\r\n".as_bytes(), 2)?;
  1266. return Ok(2);
  1267. }
  1268. self.canon_cursor_column = self.cursor_column;
  1269. }
  1270. '\r' => {
  1271. if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
  1272. // 光标已经在第0列,则不输出回车符
  1273. return Ok(0);
  1274. }
  1275. if termios.output_mode.contains(OutputMode::OCRNL) {
  1276. // 输出的\r映射为\n
  1277. c = b'\n';
  1278. if termios.output_mode.contains(OutputMode::ONLRET) {
  1279. // \r映射为\n,但是保留\r特性
  1280. self.cursor_column = 0;
  1281. self.canon_cursor_column = 0;
  1282. }
  1283. } else {
  1284. self.cursor_column = 0;
  1285. self.canon_cursor_column = 0;
  1286. }
  1287. }
  1288. '\t' => {
  1289. // 计算输出一个\t需要的空间
  1290. let spaces = 8 - (self.cursor_column & 7) as usize;
  1291. if termios.output_mode.contains(OutputMode::TABDLY)
  1292. && OutputMode::TABDLY.bits() == OutputMode::XTABS.bits()
  1293. {
  1294. // 配置的tab选项是真正输出空格到驱动
  1295. if space < spaces {
  1296. // 空间不够
  1297. return Err(SystemError::ENOBUFS);
  1298. }
  1299. self.cursor_column += spaces as u32;
  1300. // 写入sapces个空格
  1301. tty.write(core, " ".as_bytes(), spaces)?;
  1302. return Ok(spaces);
  1303. }
  1304. self.cursor_column += spaces as u32;
  1305. }
  1306. _ => {
  1307. // 判断是否为控制字符
  1308. if !(c as char).is_control() {
  1309. if termios.output_mode.contains(OutputMode::OLCUC) {
  1310. c = c.to_ascii_uppercase();
  1311. }
  1312. // 判断是否为utf8模式下的连续字符
  1313. if !(termios.input_mode.contains(InputMode::IUTF8)
  1314. && (c as usize) & 0xc0 == 0x80)
  1315. {
  1316. self.cursor_column += 1;
  1317. }
  1318. }
  1319. }
  1320. }
  1321. if tty.put_char(tty.core(), c).is_err() {
  1322. tty.write(core, &[c], 1)?;
  1323. }
  1324. Ok(1)
  1325. }
  1326. fn packet_mode_flush(&self, tty: &TtyCoreData) {
  1327. let link = tty.link().unwrap();
  1328. if link.core().contorl_info_irqsave().packet {
  1329. tty.contorl_info_irqsave()
  1330. .pktstatus
  1331. .insert(TtyPacketStatus::TIOCPKT_FLUSHREAD);
  1332. link.core().read_wq().wakeup_all();
  1333. }
  1334. }
  1335. }
  1336. impl TtyLineDiscipline for NTtyLinediscipline {
  1337. fn open(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1338. // 反向绑定tty到disc
  1339. self.disc_data().tty = Arc::downgrade(&tty);
  1340. // 特定的tty设备在这里可能需要取消端口节流
  1341. return self.set_termios(tty, None);
  1342. }
  1343. fn close(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1344. todo!()
  1345. }
  1346. /// ## 重置缓冲区的基本信息
  1347. fn flush_buffer(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1348. let core = tty.core();
  1349. let _ = core.termios();
  1350. let mut ldata = self.disc_data();
  1351. ldata.read_head = 0;
  1352. ldata.canon_head = 0;
  1353. ldata.read_tail = 0;
  1354. ldata.commit_head = 0;
  1355. ldata.line_start = 0;
  1356. ldata.erasing = false;
  1357. ldata.read_flags.set_all(false);
  1358. ldata.pushing = false;
  1359. ldata.lookahead_count = 0;
  1360. // todo: kick worker?
  1361. // packet mode?
  1362. if core.link().is_some() {
  1363. ldata.packet_mode_flush(core);
  1364. }
  1365. Ok(())
  1366. }
  1367. #[inline(never)]
  1368. fn read(
  1369. &self,
  1370. tty: Arc<TtyCore>,
  1371. buf: &mut [u8],
  1372. len: usize,
  1373. cookie: &mut bool,
  1374. _offset: usize,
  1375. mode: FileMode,
  1376. ) -> Result<usize, system_error::SystemError> {
  1377. let mut ldata;
  1378. if mode.contains(FileMode::O_NONBLOCK) {
  1379. let ret = self.disc_data_try_lock();
  1380. if ret.is_err() {
  1381. return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1382. }
  1383. ldata = ret.unwrap();
  1384. } else {
  1385. ldata = self.disc_data();
  1386. }
  1387. let core = tty.core();
  1388. let termios = core.termios();
  1389. let mut nr = len;
  1390. let mut offset = 0;
  1391. // 表示接着读
  1392. if *cookie {
  1393. // 规范且非拓展模式
  1394. if ldata.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
  1395. // 跳过EOF字符
  1396. if len == 0 {
  1397. ldata.canon_skip_eof();
  1398. } else if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
  1399. return Ok(len - nr);
  1400. }
  1401. } else if ldata.copy_from_read_buf(termios, buf, &mut nr, &mut offset)? {
  1402. return Ok(len - nr);
  1403. }
  1404. // 没有数据可读
  1405. // todo: kick worker? or 关闭节流?
  1406. *cookie = false;
  1407. return Ok(len - nr);
  1408. }
  1409. drop(termios);
  1410. TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTIN)?;
  1411. let mut minimum: usize = 0;
  1412. if !ldata.icanon {
  1413. let core = tty.core();
  1414. let termios = core.termios();
  1415. minimum = termios.control_characters[ControlCharIndex::VMIN] as usize;
  1416. if minimum == 0 {
  1417. minimum = 1;
  1418. }
  1419. }
  1420. let packet = core.contorl_info_irqsave().packet;
  1421. let mut ret: Result<usize, SystemError> = Ok(0);
  1422. // 记录读取前 的tail
  1423. let tail = ldata.read_tail;
  1424. drop(ldata);
  1425. while nr != 0 {
  1426. // todo: 处理packet模式
  1427. if packet {
  1428. let link = core.link().unwrap();
  1429. let link = link.core();
  1430. let mut ctrl = link.contorl_info_irqsave();
  1431. if !ctrl.pktstatus.is_empty() {
  1432. if offset != 0 {
  1433. break;
  1434. }
  1435. let cs = ctrl.pktstatus;
  1436. ctrl.pktstatus = TtyPacketStatus::empty();
  1437. buf[offset] = cs.bits();
  1438. offset += 1;
  1439. // nr -= 1;
  1440. break;
  1441. }
  1442. }
  1443. let mut ldata = self.disc_data();
  1444. let core = tty.core();
  1445. if !ldata.input_available(core.termios(), false) {
  1446. if core.flags().contains(TtyFlag::OTHER_CLOSED) {
  1447. ret = Err(SystemError::EIO);
  1448. break;
  1449. }
  1450. if core.flags().contains(TtyFlag::HUPPED) || core.flags().contains(TtyFlag::HUPPING)
  1451. {
  1452. break;
  1453. }
  1454. if mode.contains(FileMode::O_NONBLOCK)
  1455. || core.flags().contains(TtyFlag::LDISC_CHANGING)
  1456. {
  1457. ret = Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1458. break;
  1459. }
  1460. if ProcessManager::current_pcb()
  1461. .sig_info_irqsave()
  1462. .sig_pending()
  1463. .has_pending()
  1464. {
  1465. ret = Err(SystemError::ERESTARTSYS);
  1466. break;
  1467. }
  1468. // 休眠一段时间
  1469. // 获取到termios读锁,避免termios被更改导致行为异常
  1470. // let termios = core.termios_preempt_enable();
  1471. // let helper = WakeUpHelper::new(ProcessManager::current_pcb());
  1472. // let wakeup_helper = Timer::new(helper, timeout);
  1473. // wakeup_helper.activate();
  1474. // drop(termios);
  1475. drop(ldata);
  1476. core.read_wq()
  1477. .sleep((EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64);
  1478. continue;
  1479. }
  1480. if ldata.icanon && !core.termios().local_mode.contains(LocalMode::EXTPROC) {
  1481. if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
  1482. *cookie = true;
  1483. offset += len - nr;
  1484. return Ok(offset);
  1485. }
  1486. } else {
  1487. // 非标准模式
  1488. // todo: 处理packet模式
  1489. if packet && offset == 0 {
  1490. buf[offset] = TtyPacketStatus::TIOCPKT_DATA.bits();
  1491. offset += 1;
  1492. nr -= 1;
  1493. }
  1494. // 拷贝数据
  1495. if ldata.copy_from_read_buf(core.termios(), buf, &mut nr, &mut offset)?
  1496. && offset >= minimum
  1497. {
  1498. *cookie = true;
  1499. return Ok(offset);
  1500. }
  1501. }
  1502. if offset >= minimum {
  1503. break;
  1504. }
  1505. }
  1506. let ldata = self.disc_data();
  1507. if tail != ldata.read_tail {
  1508. // todo: kick worker?
  1509. }
  1510. if offset > 0 {
  1511. return Ok(offset);
  1512. }
  1513. ret
  1514. }
  1515. #[inline(never)]
  1516. fn write(
  1517. &self,
  1518. tty: Arc<TtyCore>,
  1519. buf: &[u8],
  1520. len: usize,
  1521. mode: FileMode,
  1522. ) -> Result<usize, system_error::SystemError> {
  1523. let mut nr = len;
  1524. let mut ldata = self.disc_data();
  1525. let pcb = ProcessManager::current_pcb();
  1526. let binding = tty.clone();
  1527. let core = binding.core();
  1528. let termios = *core.termios();
  1529. if termios.local_mode.contains(LocalMode::TOSTOP) {
  1530. TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTOU)?;
  1531. }
  1532. ldata.process_echoes(tty.clone());
  1533. // drop(ldata);
  1534. let mut offset = 0;
  1535. loop {
  1536. if pcb.sig_info_irqsave().sig_pending().has_pending() {
  1537. return Err(SystemError::ERESTARTSYS);
  1538. }
  1539. if core.flags().contains(TtyFlag::HUPPED) {
  1540. return Err(SystemError::EIO);
  1541. }
  1542. if termios.output_mode.contains(OutputMode::OPOST) {
  1543. while nr > 0 {
  1544. // let mut ldata = self.disc_data();
  1545. // 获得一次处理后的数量
  1546. let ret = ldata.process_output_block(core, core.termios(), &buf[offset..], nr);
  1547. let num = match ret {
  1548. Ok(num) => num,
  1549. Err(e) => {
  1550. if e == SystemError::EAGAIN_OR_EWOULDBLOCK {
  1551. break;
  1552. } else {
  1553. return Err(e);
  1554. }
  1555. }
  1556. };
  1557. offset += num;
  1558. nr -= num;
  1559. if nr == 0 {
  1560. break;
  1561. }
  1562. let c = buf[offset];
  1563. if !ldata.process_output(tty.clone(), c) {
  1564. break;
  1565. }
  1566. offset += 1;
  1567. nr -= 1;
  1568. }
  1569. tty.flush_chars(core);
  1570. } else {
  1571. while nr > 0 {
  1572. let write = tty.write(core, &buf[offset..], nr)?;
  1573. if write == 0 {
  1574. break;
  1575. }
  1576. offset += write;
  1577. nr -= write;
  1578. }
  1579. }
  1580. if nr == 0 {
  1581. break;
  1582. }
  1583. if mode.contains(FileMode::O_NONBLOCK) || core.flags().contains(TtyFlag::LDISC_CHANGING)
  1584. {
  1585. return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1586. }
  1587. // 到这里表明没位置可写了
  1588. // 休眠一段时间
  1589. // 获取到termios读锁,避免termios被更改导致行为异常
  1590. core.write_wq()
  1591. .sleep(EPollEventType::EPOLLOUT.bits() as u64);
  1592. }
  1593. Ok(offset)
  1594. }
  1595. fn ioctl(
  1596. &self,
  1597. tty: Arc<TtyCore>,
  1598. cmd: u32,
  1599. arg: usize,
  1600. ) -> Result<usize, system_error::SystemError> {
  1601. match cmd {
  1602. TtyIoctlCmd::TIOCOUTQ => {
  1603. let mut user_writer = UserBufferWriter::new(
  1604. VirtAddr::new(arg).as_ptr::<i32>(),
  1605. core::mem::size_of::<i32>(),
  1606. true,
  1607. )?;
  1608. let count = tty.chars_in_buffer();
  1609. user_writer.copy_one_to_user::<i32>(&(count as i32), 0)?;
  1610. return Ok(0);
  1611. }
  1612. TtyIoctlCmd::FIONREAD => {
  1613. let ldata = self.disc_data();
  1614. let termios = tty.core().termios();
  1615. let retval;
  1616. if termios.local_mode.contains(LocalMode::ICANON)
  1617. && !termios.local_mode.contains(LocalMode::EXTPROC)
  1618. {
  1619. if ldata.canon_head == ldata.read_tail {
  1620. retval = 0;
  1621. } else {
  1622. let head = ldata.canon_head;
  1623. let mut tail = ldata.read_tail;
  1624. let mut nr = head - tail;
  1625. while ntty_buf_mask(head) != ntty_buf_mask(tail) {
  1626. if ldata.read_flags.get(ntty_buf_mask(tail)).unwrap()
  1627. && ldata.read_buf[ntty_buf_mask(tail)]
  1628. == ControlCharIndex::DISABLE_CHAR
  1629. {
  1630. nr -= 1;
  1631. }
  1632. tail += 1;
  1633. }
  1634. retval = nr;
  1635. }
  1636. } else {
  1637. retval = ldata.read_cnt();
  1638. }
  1639. let mut user_writer = UserBufferWriter::new(
  1640. VirtAddr::new(arg).as_ptr::<i32>(),
  1641. core::mem::size_of::<i32>(),
  1642. true,
  1643. )?;
  1644. user_writer.copy_one_to_user::<i32>(&(retval as i32), 0)?;
  1645. return Ok(0);
  1646. }
  1647. _ => {
  1648. return self.ioctl_helper(tty, cmd, arg);
  1649. }
  1650. }
  1651. }
  1652. #[inline(never)]
  1653. fn set_termios(
  1654. &self,
  1655. tty: Arc<TtyCore>,
  1656. old: Option<crate::driver::tty::termios::Termios>,
  1657. ) -> Result<(), system_error::SystemError> {
  1658. let core = tty.core();
  1659. let termios = core.termios();
  1660. let mut ldata = self.disc_data();
  1661. let contorl_chars = termios.control_characters;
  1662. // 第一次设置或者规范模式 (ICANON) 或者扩展处理 (EXTPROC) 标志发生变化
  1663. let mut spec_mode_changed = false;
  1664. if let Some(old) = old {
  1665. let local_mode = old.local_mode.bitxor(termios.local_mode);
  1666. spec_mode_changed =
  1667. local_mode.contains(LocalMode::ICANON) || local_mode.contains(LocalMode::EXTPROC);
  1668. }
  1669. if old.is_none() || spec_mode_changed {
  1670. // 重置read_flags
  1671. ldata.read_flags.set_all(false);
  1672. ldata.line_start = ldata.read_tail;
  1673. // 不是规范模式或者有可读数据
  1674. if !termios.local_mode.contains(LocalMode::ICANON) || ldata.read_cnt() != 0 {
  1675. ldata.canon_head = ldata.read_tail;
  1676. ldata.pushing = false;
  1677. } else {
  1678. let read_head = ldata.read_head;
  1679. ldata
  1680. .read_flags
  1681. .set((read_head - 1) & (NTTY_BUFSIZE - 1), true);
  1682. ldata.canon_head = ldata.read_head;
  1683. ldata.pushing = true;
  1684. }
  1685. ldata.commit_head = ldata.read_head;
  1686. ldata.erasing = false;
  1687. ldata.lnext = false;
  1688. }
  1689. // 设置模式
  1690. ldata.icanon = termios.local_mode.contains(LocalMode::ICANON);
  1691. // 设置回显
  1692. if termios.local_mode.contains(LocalMode::ECHO) {
  1693. ldata.echo = true;
  1694. }
  1695. if termios.input_mode.contains(InputMode::ISTRIP)
  1696. || termios.input_mode.contains(InputMode::IUCLC)
  1697. || termios.input_mode.contains(InputMode::IGNCR)
  1698. || termios.input_mode.contains(InputMode::IXON)
  1699. || termios.local_mode.contains(LocalMode::ISIG)
  1700. || termios.local_mode.contains(LocalMode::ECHO)
  1701. || termios.input_mode.contains(InputMode::PARMRK)
  1702. {
  1703. // 非原模式
  1704. ldata.char_map.set_all(false);
  1705. // 忽略回车符或者将回车映射为换行符
  1706. if termios.input_mode.contains(InputMode::IGNCR)
  1707. || termios.input_mode.contains(InputMode::ICRNL)
  1708. {
  1709. ldata.char_map.set('\r' as usize, true);
  1710. }
  1711. // 将换行映射为回车
  1712. if termios.input_mode.contains(InputMode::INLCR) {
  1713. ldata.char_map.set('\n' as usize, true);
  1714. }
  1715. // 规范模式
  1716. if termios.local_mode.contains(LocalMode::ICANON) {
  1717. ldata
  1718. .char_map
  1719. .set(contorl_chars[ControlCharIndex::VERASE] as usize, true);
  1720. ldata
  1721. .char_map
  1722. .set(contorl_chars[ControlCharIndex::VKILL] as usize, true);
  1723. ldata
  1724. .char_map
  1725. .set(contorl_chars[ControlCharIndex::VEOF] as usize, true);
  1726. ldata.char_map.set('\n' as usize, true);
  1727. ldata
  1728. .char_map
  1729. .set(contorl_chars[ControlCharIndex::VEOL] as usize, true);
  1730. if termios.local_mode.contains(LocalMode::IEXTEN) {
  1731. ldata
  1732. .char_map
  1733. .set(contorl_chars[ControlCharIndex::VWERASE] as usize, true);
  1734. ldata
  1735. .char_map
  1736. .set(contorl_chars[ControlCharIndex::VLNEXT] as usize, true);
  1737. ldata
  1738. .char_map
  1739. .set(contorl_chars[ControlCharIndex::VEOL2] as usize, true);
  1740. if termios.local_mode.contains(LocalMode::ECHO) {
  1741. ldata
  1742. .char_map
  1743. .set(contorl_chars[ControlCharIndex::VREPRINT] as usize, true);
  1744. }
  1745. }
  1746. }
  1747. // 软件流控制
  1748. if termios.input_mode.contains(InputMode::IXON) {
  1749. ldata
  1750. .char_map
  1751. .set(contorl_chars[ControlCharIndex::VSTART] as usize, true);
  1752. ldata
  1753. .char_map
  1754. .set(contorl_chars[ControlCharIndex::VSTOP] as usize, true);
  1755. }
  1756. if termios.local_mode.contains(LocalMode::ISIG) {
  1757. ldata
  1758. .char_map
  1759. .set(contorl_chars[ControlCharIndex::VINTR] as usize, true);
  1760. ldata
  1761. .char_map
  1762. .set(contorl_chars[ControlCharIndex::VQUIT] as usize, true);
  1763. ldata
  1764. .char_map
  1765. .set(contorl_chars[ControlCharIndex::VSUSP] as usize, true);
  1766. }
  1767. ldata
  1768. .char_map
  1769. .set(ControlCharIndex::DISABLE_CHAR as usize, true);
  1770. ldata.raw = false;
  1771. ldata.real_raw = false;
  1772. } else {
  1773. // 原模式或real_raw
  1774. ldata.raw = true;
  1775. ldata.real_raw = termios.input_mode.contains(InputMode::IGNBRK)
  1776. || (!termios.input_mode.contains(InputMode::BRKINT)
  1777. && !termios.input_mode.contains(InputMode::PARMRK))
  1778. && (termios.input_mode.contains(InputMode::IGNPAR)
  1779. || !termios.input_mode.contains(InputMode::INPCK))
  1780. && (core
  1781. .driver()
  1782. .flags()
  1783. .contains(TtyDriverFlag::TTY_DRIVER_REAL_RAW));
  1784. }
  1785. // if !termios.input_mode.contains(InputMode::IXON)
  1786. // && old.is_some()
  1787. // && old.unwrap().input_mode.contains(InputMode::IXON) && !
  1788. // {}
  1789. core.read_wq().wakeup_all();
  1790. core.write_wq().wakeup_all();
  1791. Ok(())
  1792. }
  1793. fn poll(&self, tty: Arc<TtyCore>) -> Result<usize, system_error::SystemError> {
  1794. let core = tty.core();
  1795. let ldata = self.disc_data();
  1796. let mut event = EPollEventType::empty();
  1797. if ldata.input_available(core.termios(), true) {
  1798. event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM)
  1799. }
  1800. if core.contorl_info_irqsave().packet {
  1801. let link = core.link();
  1802. if link.is_some()
  1803. && !link
  1804. .unwrap()
  1805. .core()
  1806. .contorl_info_irqsave()
  1807. .pktstatus
  1808. .is_empty()
  1809. {
  1810. event.insert(
  1811. EPollEventType::EPOLLPRI
  1812. | EPollEventType::EPOLLIN
  1813. | EPollEventType::EPOLLRDNORM,
  1814. );
  1815. }
  1816. }
  1817. if core.flags().contains(TtyFlag::OTHER_CLOSED) {
  1818. event.insert(EPollEventType::EPOLLHUP);
  1819. }
  1820. if core.driver().driver_funcs().chars_in_buffer() < 256
  1821. && core.driver().driver_funcs().write_room(core) > 0
  1822. {
  1823. event.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM);
  1824. }
  1825. Ok(event.bits() as usize)
  1826. }
  1827. fn hangup(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1828. todo!()
  1829. }
  1830. fn receive_buf(
  1831. &self,
  1832. tty: Arc<TtyCore>,
  1833. buf: &[u8],
  1834. flags: Option<&[u8]>,
  1835. count: usize,
  1836. ) -> Result<usize, SystemError> {
  1837. let mut ldata = self.disc_data();
  1838. ldata.receive_buf_common(tty, buf, flags, count, false)
  1839. }
  1840. fn receive_buf2(
  1841. &self,
  1842. tty: Arc<TtyCore>,
  1843. buf: &[u8],
  1844. flags: Option<&[u8]>,
  1845. count: usize,
  1846. ) -> Result<usize, SystemError> {
  1847. let mut ldata = self.disc_data();
  1848. ldata.receive_buf_common(tty, buf, flags, count, true)
  1849. }
  1850. }