ntty.rs 70 KB

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