ntty.rs 69 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121
  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 ((c as usize) < self.char_map.size()) && 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 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. if !termios.local_mode.contains(LocalMode::NOFLSH) {
  690. // 重置
  691. self.echo_head = 0;
  692. self.echo_tail = 0;
  693. self.echo_mark = 0;
  694. self.echo_commit = 0;
  695. let _ = tty.flush_buffer(tty.core());
  696. self.read_head = 0;
  697. self.canon_head = 0;
  698. self.read_tail = 0;
  699. self.line_start = 0;
  700. self.erasing = false;
  701. self.read_flags.set_all(false);
  702. self.pushing = false;
  703. self.lookahead_count = 0;
  704. if tty.core().link().is_some() {
  705. self.packet_mode_flush(tty.core());
  706. }
  707. }
  708. }
  709. pub fn receive_char(&mut self, c: u8, tty: Arc<TtyCore>) {
  710. let termios = tty.core().termios();
  711. if termios.local_mode.contains(LocalMode::ECHO) {
  712. if self.erasing {
  713. self.add_echo_byte(b'/');
  714. self.erasing = false;
  715. }
  716. if self.canon_head == self.read_head {
  717. self.add_echo_byte(EchoOperation::Start.to_u8());
  718. self.add_echo_byte(EchoOperation::SetCanonCol.to_u8());
  719. }
  720. self.echo_char(c, &termios);
  721. self.commit_echoes(tty.clone());
  722. }
  723. if c == 0o377 && tty.core().termios().input_mode.contains(InputMode::PARMRK) {
  724. self.add_read_byte(c);
  725. }
  726. self.add_read_byte(c);
  727. }
  728. pub fn echo_char(&mut self, c: u8, termios: &RwLockReadGuard<Termios>) {
  729. if c == EchoOperation::Start.to_u8() {
  730. self.add_echo_byte(EchoOperation::Start.to_u8());
  731. self.add_echo_byte(EchoOperation::Start.to_u8());
  732. } else {
  733. if termios.local_mode.contains(LocalMode::ECHOCTL)
  734. && (c as char).is_control()
  735. && c != b'\t'
  736. {
  737. self.add_echo_byte(EchoOperation::Start.to_u8());
  738. }
  739. self.add_echo_byte(c);
  740. }
  741. }
  742. pub fn echo_char_raw(&mut self, c: u8) {
  743. if c == EchoOperation::Start.to_u8() {
  744. self.add_echo_byte(EchoOperation::Start.to_u8());
  745. self.add_echo_byte(EchoOperation::Start.to_u8());
  746. } else {
  747. self.add_echo_byte(c);
  748. }
  749. }
  750. /// ## 提交echobuf里的数据显示
  751. pub fn commit_echoes(&mut self, tty: Arc<TtyCore>) {
  752. let head = self.echo_head;
  753. self.echo_mark = head;
  754. let old = self.echo_commit - self.echo_tail;
  755. // 需要echo的字符个数
  756. let nr = head - self.echo_tail;
  757. if nr < ECHO_COMMIT_WATERMARK || nr % ECHO_BLOCK > old % ECHO_BLOCK {
  758. return;
  759. }
  760. self.echo_commit = head;
  761. let echoed = self.echoes(tty.clone());
  762. if echoed.is_ok() && echoed.unwrap() > 0 {
  763. tty.flush_chars(tty.core());
  764. }
  765. }
  766. pub fn add_echo_byte(&mut self, c: u8) {
  767. self.echo_buf[ntty_buf_mask(self.echo_head)] = c;
  768. self.echo_head += 1;
  769. }
  770. pub fn add_read_byte(&mut self, c: u8) {
  771. self.read_buf[ntty_buf_mask(self.read_head)] = c;
  772. self.read_head += 1;
  773. }
  774. /// ### 将read_buffer的部分值置0
  775. ///
  776. /// 只会在规范模式和禁用echo下执行
  777. #[inline]
  778. pub fn zero_buffer(&mut self, offset: usize, size: usize) {
  779. let offset = offset & (NTTY_BUFSIZE - 1);
  780. if self.icanon && !self.echo {
  781. let n = offset + size;
  782. if n > NTTY_BUFSIZE {
  783. for c in &mut self.read_buf[offset..NTTY_BUFSIZE] {
  784. *c = 0
  785. }
  786. for c in &mut self.read_buf[0..(n - NTTY_BUFSIZE)] {
  787. *c = 0
  788. }
  789. } else {
  790. for c in &mut self.read_buf[offset..n] {
  791. *c = 0
  792. }
  793. };
  794. }
  795. }
  796. /// ## 从ntty中拷贝数据
  797. ///
  798. /// ### 参数
  799. ///
  800. /// ### to: 存储数据
  801. /// ### tail: 读取尾
  802. pub fn ntty_copy(
  803. &mut self,
  804. to: &mut [u8],
  805. tail: usize,
  806. n: &mut usize,
  807. ) -> Result<(), SystemError> {
  808. if to.len() < *n {
  809. *n = to.len();
  810. // return Err(SystemError::EINVAL);
  811. }
  812. if tail > NTTY_BUFSIZE {
  813. return Err(SystemError::EINVAL);
  814. }
  815. let size = NTTY_BUFSIZE - tail;
  816. if size < *n {
  817. // 有一部分数据在头部,则先拷贝后面部分,再拷贝头部
  818. // TODO: tty审计?
  819. to[0..size].copy_from_slice(&self.read_buf[tail..(tail + size)]);
  820. to[size..(*n)].copy_from_slice(&self.read_buf[0..(*n - size)]);
  821. } else {
  822. to[..*n].copy_from_slice(&self.read_buf[tail..(tail + *n)])
  823. }
  824. self.zero_buffer(tail, *n);
  825. Ok(())
  826. }
  827. /// ## 规范模式下跳过EOF
  828. pub fn canon_skip_eof(&mut self) {
  829. // 没有数据
  830. if self.read_tail == self.canon_head {
  831. return;
  832. }
  833. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  834. // 查看read_flags是否读取位置为特殊字符
  835. if !self.read_flags.get(tail).unwrap() {
  836. return;
  837. }
  838. // 确保读取位置是'\0'字符
  839. if self.read_buf[tail] != ControlCharIndex::DISABLE_CHAR {
  840. return;
  841. }
  842. // 处理该字符,将read_flagsw该位清除
  843. self.read_flags.set(tail, false);
  844. // 读取位置+1,即跳过该字符不做处理
  845. self.read_tail += 1;
  846. }
  847. /// ## 在规范模式(canonical mode)下从读缓冲中复制一行
  848. ///
  849. /// 一次只拷贝一行
  850. ///
  851. /// ## 参数
  852. /// ### dst: 存放数据
  853. /// ### nr: 需要拷贝的数据大小
  854. ///
  855. /// ## 返回值
  856. /// ### true: 表示一行未结束并且还有数据可读
  857. /// ### false: 一行已结束或者没有数据可读
  858. pub fn canon_copy_from_read_buf(
  859. &mut self,
  860. dst: &mut [u8],
  861. nr: &mut usize,
  862. offset: &mut usize,
  863. ) -> Result<bool, SystemError> {
  864. if *nr == 0 {
  865. return Ok(false);
  866. }
  867. let canon_head = self.canon_head;
  868. // 取得能够读到的字符数,即canon_head - self.read_tail和nr最小值
  869. let mut n = (*nr).min(canon_head - self.read_tail);
  870. // 获得读尾index
  871. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  872. // 避免越界,这个size才是实际读取大小
  873. let size = if tail + n > NTTY_BUFSIZE {
  874. NTTY_BUFSIZE
  875. } else {
  876. tail + n
  877. };
  878. // 找到eol的坐标
  879. let tmp = self.read_flags.next_index(tail);
  880. // 找到的话即为坐标,未找到的话即为NTTY_BUFSIZE
  881. let mut eol = if let Some(tmp) = tmp { tmp } else { size };
  882. if eol > size {
  883. eol = size
  884. }
  885. // 是否需要绕回缓冲区头部
  886. let more = n - (size - tail);
  887. // 是否找到eol
  888. let found = if eol == NTTY_BUFSIZE && more > 0 {
  889. // 需要返回头部
  890. let ret = self.read_flags.first_index();
  891. if let Some(tmp) = ret {
  892. // 在头部范围内找到eol
  893. if tmp < more {
  894. eol = tmp;
  895. }
  896. } else {
  897. eol = more;
  898. }
  899. eol != more
  900. } else {
  901. // 不需要返回头部
  902. eol != size
  903. };
  904. n = eol - tail;
  905. if n > NTTY_BUFSIZE {
  906. // 减法溢出则加上BUFSIZE即可限制在0-NTTY_BUFSIZE内
  907. n += NTTY_BUFSIZE;
  908. }
  909. // 规范模式下实际扫描过的字符数,需要将eol计算在内
  910. let count = if found { n + 1 } else { n };
  911. // 表示这一行未结束
  912. if !found || self.read_at(eol) != ControlCharIndex::DISABLE_CHAR {
  913. n = count;
  914. }
  915. self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
  916. *nr -= n;
  917. *offset += n;
  918. if found {
  919. self.read_flags.set(eol, false);
  920. }
  921. self.read_tail += count;
  922. if found {
  923. if !self.pushing {
  924. self.line_start = self.read_tail;
  925. } else {
  926. self.pushing = false;
  927. }
  928. // todo: 审计?
  929. return Ok(false);
  930. }
  931. // 这里是表示没有找到eol,根据是否还有数据可读返回
  932. Ok(self.read_tail != canon_head)
  933. }
  934. /// ## 根据终端的模式和输入缓冲区中的数据量,判断是否可读字符
  935. pub fn input_available(&self, termios: RwLockReadGuard<Termios>, poll: bool) -> bool {
  936. // 计算最小字符数
  937. let amt = if poll
  938. && termios.control_characters[ControlCharIndex::VTIME] as u32 == 0
  939. && termios.control_characters[ControlCharIndex::VMIN] as u32 != 0
  940. {
  941. termios.control_characters[ControlCharIndex::VMIN] as usize
  942. } else {
  943. 1
  944. };
  945. // 规范模式且非拓展
  946. if self.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
  947. return self.canon_head != self.read_tail;
  948. } else {
  949. return (self.commit_head - self.read_tail) >= amt;
  950. }
  951. }
  952. /// ## 非规范模式下从read_buf读取数据
  953. ///
  954. /// ## 参数
  955. /// ### termios: tty对应的termioss读锁守卫
  956. /// ### dst: 存储读取数据
  957. /// ### nr: 读取长度
  958. ///
  959. /// ## 返回值
  960. /// ### true: 还有更多数据可读
  961. /// ### false: 无更多数据可读
  962. pub fn copy_from_read_buf(
  963. &mut self,
  964. termios: RwLockReadGuard<Termios>,
  965. dst: &mut [u8],
  966. nr: &mut usize,
  967. offset: &mut usize,
  968. ) -> Result<bool, SystemError> {
  969. let head = self.commit_head;
  970. let tail = self.read_tail & (NTTY_BUFSIZE - 1);
  971. // 计算出可读的字符数
  972. let mut n = (NTTY_BUFSIZE - tail).min(head - self.read_tail);
  973. n = n.min(*nr);
  974. if n > 0 {
  975. // 拷贝数据
  976. self.ntty_copy(&mut dst[*offset..], tail, &mut n)?;
  977. // todo:审计?
  978. self.read_tail += n;
  979. // 是否只读取了eof
  980. let eof =
  981. n == 1 && self.read_buf[tail] == termios.control_characters[ControlCharIndex::VEOF];
  982. if termios.local_mode.contains(LocalMode::EXTPROC)
  983. && self.icanon
  984. && eof
  985. && head == self.read_tail
  986. {
  987. return Ok(false);
  988. }
  989. *nr -= n;
  990. *offset += n;
  991. return Ok(head != self.read_tail);
  992. }
  993. Ok(false)
  994. }
  995. /// ## 用于处理带有 OPOST(Output Post-processing)标志的输出块的函数
  996. /// OPOST 是 POSIX 终端驱动器标志之一,用于指定在写入终端设备之前对输出数据进行一些后期处理。
  997. pub fn process_output_block(
  998. &mut self,
  999. core: &TtyCoreData,
  1000. termios: RwLockReadGuard<Termios>,
  1001. buf: &[u8],
  1002. nr: usize,
  1003. ) -> Result<usize, SystemError> {
  1004. let mut nr = nr;
  1005. let tty = self.tty.upgrade().unwrap();
  1006. let space = tty.write_room(tty.core());
  1007. // 如果读取数量大于了可用空间,则取最小的为真正的写入数量
  1008. if nr > space {
  1009. nr = space
  1010. }
  1011. let mut cnt = 0;
  1012. for (i, c) in buf.iter().enumerate().take(nr) {
  1013. cnt = i;
  1014. let c = *c;
  1015. if c as usize == 8 {
  1016. // 表示退格
  1017. if self.cursor_column > 0 {
  1018. self.cursor_column -= 1;
  1019. }
  1020. continue;
  1021. }
  1022. match c as char {
  1023. '\n' => {
  1024. if termios.output_mode.contains(OutputMode::ONLRET) {
  1025. // 将回车映射为\n,即将\n换为回车
  1026. self.cursor_column = 0;
  1027. }
  1028. if termios.output_mode.contains(OutputMode::ONLCR) {
  1029. // 输出时将\n换为\r\n
  1030. break;
  1031. }
  1032. self.canon_cursor_column = self.cursor_column;
  1033. }
  1034. '\r' => {
  1035. if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
  1036. // 光标已经在第0列,则不输出回车符
  1037. break;
  1038. }
  1039. if termios.output_mode.contains(OutputMode::OCRNL) {
  1040. break;
  1041. }
  1042. self.cursor_column = 0;
  1043. self.canon_cursor_column = 0;
  1044. }
  1045. '\t' => {
  1046. break;
  1047. }
  1048. _ => {
  1049. // 判断是否为控制字符
  1050. if !(c as char).is_control() {
  1051. if termios.output_mode.contains(OutputMode::OLCUC) {
  1052. break;
  1053. }
  1054. // 判断是否为utf8模式下的连续字符
  1055. if !(termios.input_mode.contains(InputMode::IUTF8)
  1056. && (c as usize) & 0xc0 == 0x80)
  1057. {
  1058. self.cursor_column += 1;
  1059. }
  1060. }
  1061. }
  1062. }
  1063. }
  1064. drop(termios);
  1065. return tty.write(core, buf, cnt);
  1066. }
  1067. /// ## 处理回显
  1068. pub fn process_echoes(&mut self, tty: Arc<TtyCore>) {
  1069. if self.echo_mark == self.echo_tail {
  1070. return;
  1071. }
  1072. self.echo_commit = self.echo_mark;
  1073. let echoed = self.echoes(tty.clone());
  1074. if echoed.is_ok() && echoed.unwrap() > 0 {
  1075. tty.flush_chars(tty.core());
  1076. }
  1077. }
  1078. #[inline(never)]
  1079. pub fn echoes(&mut self, tty: Arc<TtyCore>) -> Result<usize, SystemError> {
  1080. let mut space = tty.write_room(tty.core());
  1081. let ospace = space;
  1082. let termios = tty.core().termios();
  1083. let core = tty.core();
  1084. let mut tail = self.echo_tail;
  1085. while ntty_buf_mask(self.echo_commit) != ntty_buf_mask(tail) {
  1086. let c = self.echo_buf[ntty_buf_mask(tail)];
  1087. if EchoOperation::from_u8(c) == EchoOperation::Start {
  1088. if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 1) {
  1089. self.echo_tail = tail;
  1090. return Ok(ospace - space);
  1091. }
  1092. // 获取到start,之后取第一个作为op
  1093. let op = EchoOperation::from_u8(self.echo_buf[ntty_buf_mask(tail + 1)]);
  1094. match op {
  1095. EchoOperation::Start => {
  1096. if space == 0 {
  1097. break;
  1098. }
  1099. if tty
  1100. .put_char(tty.core(), EchoOperation::Start.to_u8())
  1101. .is_err()
  1102. {
  1103. tty.write(core, &[EchoOperation::Start.to_u8()], 1)?;
  1104. }
  1105. self.cursor_column += 1;
  1106. space -= 1;
  1107. tail += 2;
  1108. }
  1109. EchoOperation::MoveBackCol => {
  1110. if self.cursor_column > 0 {
  1111. self.cursor_column -= 1;
  1112. }
  1113. tail += 2;
  1114. }
  1115. EchoOperation::SetCanonCol => {
  1116. self.canon_cursor_column = self.cursor_column;
  1117. tail += 2;
  1118. }
  1119. EchoOperation::EraseTab => {
  1120. if ntty_buf_mask(self.echo_commit) == ntty_buf_mask(tail + 2) {
  1121. self.echo_tail = tail;
  1122. return Ok(ospace - space);
  1123. }
  1124. // 要擦除的制表符所占用的列数
  1125. let mut char_num = self.echo_buf[ntty_buf_mask(tail + 2)] as usize;
  1126. /*
  1127. 如果 num_chars 的最高位(0x80)未设置,
  1128. 表示这是从输入的起始位置而不是从先前的制表符开始计算的列数。
  1129. 在这种情况下,将 num_chars 与 ldata->canon_column 相加,否则,列数就是正常的制表符列数。
  1130. */
  1131. if char_num & 0x80 == 0 {
  1132. char_num += self.canon_cursor_column as usize;
  1133. }
  1134. // 计算要回退的列数,即制表符宽度减去实际占用的列数
  1135. let mut num_bs = 8 - (char_num & 7);
  1136. if num_bs > space {
  1137. // 表示左边没有足够空间回退
  1138. break;
  1139. }
  1140. space -= num_bs;
  1141. while num_bs != 0 {
  1142. num_bs -= 1;
  1143. // 8 => '\b'
  1144. if tty.put_char(tty.core(), 8).is_err() {
  1145. tty.write(core, &[8], 1)?;
  1146. }
  1147. if self.cursor_column > 0 {
  1148. self.cursor_column -= 1;
  1149. }
  1150. }
  1151. // 已经读取了 tail tail+1 tail+2,所以这里偏移加3
  1152. tail += 3;
  1153. }
  1154. EchoOperation::Undefined(ch) => {
  1155. match ch {
  1156. 8 => {
  1157. if tty.put_char(tty.core(), 8).is_err() {
  1158. tty.write(core, &[8], 1)?;
  1159. }
  1160. if tty.put_char(tty.core(), b' ').is_err() {
  1161. tty.write(core, b" ", 1)?;
  1162. }
  1163. self.cursor_column -= 1;
  1164. space -= 1;
  1165. tail += 1;
  1166. }
  1167. _ => {
  1168. // 不是特殊字节码,则表示控制字符 例如 ^C
  1169. if space < 2 {
  1170. break;
  1171. }
  1172. if tty.put_char(tty.core(), b'^').is_err() {
  1173. tty.write(core, b"^", 1)?;
  1174. }
  1175. if tty.put_char(tty.core(), ch ^ 0o100).is_err() {
  1176. tty.write(core, &[ch ^ 0o100], 1)?;
  1177. }
  1178. self.cursor_column += 2;
  1179. space -= 2;
  1180. tail += 2;
  1181. }
  1182. }
  1183. }
  1184. }
  1185. } else {
  1186. if termios.output_mode.contains(OutputMode::OPOST) {
  1187. let ret = self.do_output_char(tty.clone(), c, space);
  1188. if ret.is_err() {
  1189. break;
  1190. }
  1191. space -= ret.unwrap();
  1192. } else {
  1193. if space == 0 {
  1194. break;
  1195. }
  1196. if tty.put_char(tty.core(), c).is_err() {
  1197. tty.write(core, &[c], 1)?;
  1198. }
  1199. space -= 1;
  1200. }
  1201. tail += 1;
  1202. }
  1203. }
  1204. // 如果回显缓冲区接近满(在下一次提交之前可能会发生回显溢出的情况),则丢弃足够的尾部数据以防止随后的溢出。
  1205. while self.echo_commit > tail && self.echo_commit - tail >= ECHO_DISCARD_WATERMARK {
  1206. if self.echo_buf[ntty_buf_mask(tail)] == EchoOperation::Start.to_u8() {
  1207. if self.echo_buf[ntty_buf_mask(tail + 1)] == EchoOperation::EraseTab.to_u8() {
  1208. tail += 3;
  1209. } else {
  1210. tail += 2;
  1211. }
  1212. } else {
  1213. tail += 1;
  1214. }
  1215. }
  1216. self.echo_tail = tail;
  1217. return Ok(ospace - space);
  1218. }
  1219. /// ## 处理输出字符(带有 OPOST 处理)
  1220. pub fn process_output(&mut self, tty: Arc<TtyCore>, c: u8) -> bool {
  1221. let space = tty.write_room(tty.core());
  1222. if self.do_output_char(tty, c, space).is_err() {
  1223. return false;
  1224. }
  1225. true
  1226. }
  1227. // ## 设置带有 OPOST 处理的tty输出一个字符
  1228. pub fn do_output_char(
  1229. &mut self,
  1230. tty: Arc<TtyCore>,
  1231. c: u8,
  1232. space: usize,
  1233. ) -> Result<usize, SystemError> {
  1234. if space == 0 {
  1235. return Err(SystemError::ENOBUFS);
  1236. }
  1237. let termios = tty.core().termios();
  1238. let core = tty.core();
  1239. let mut c = c;
  1240. if c as usize == 8 {
  1241. // 表示退格
  1242. if self.cursor_column > 0 {
  1243. self.cursor_column -= 1;
  1244. }
  1245. if tty.put_char(tty.core(), c).is_err() {
  1246. tty.write(core, &[c], 1)?;
  1247. }
  1248. return Ok(1);
  1249. }
  1250. match c as char {
  1251. '\n' => {
  1252. if termios.output_mode.contains(OutputMode::ONLRET) {
  1253. // 回车符
  1254. self.cursor_column = 0;
  1255. }
  1256. if termios.output_mode.contains(OutputMode::ONLCR) {
  1257. // 映射为“\r\n”
  1258. if space < 2 {
  1259. return Err(SystemError::ENOBUFS);
  1260. }
  1261. self.cursor_column = 0;
  1262. self.canon_cursor_column = 0;
  1263. // 通过驱动写入
  1264. tty.write(core, "\r\n".as_bytes(), 2)?;
  1265. return Ok(2);
  1266. }
  1267. self.canon_cursor_column = self.cursor_column;
  1268. }
  1269. '\r' => {
  1270. if termios.output_mode.contains(OutputMode::ONOCR) && self.cursor_column == 0 {
  1271. // 光标已经在第0列,则不输出回车符
  1272. return Ok(0);
  1273. }
  1274. if termios.output_mode.contains(OutputMode::OCRNL) {
  1275. // 输出的\r映射为\n
  1276. c = b'\n';
  1277. if termios.output_mode.contains(OutputMode::ONLRET) {
  1278. // \r映射为\n,但是保留\r特性
  1279. self.cursor_column = 0;
  1280. self.canon_cursor_column = 0;
  1281. }
  1282. } else {
  1283. self.cursor_column = 0;
  1284. self.canon_cursor_column = 0;
  1285. }
  1286. }
  1287. '\t' => {
  1288. // 计算输出一个\t需要的空间
  1289. let spaces = 8 - (self.cursor_column & 7) as usize;
  1290. if termios.output_mode.contains(OutputMode::TABDLY)
  1291. && OutputMode::TABDLY.bits() == OutputMode::XTABS.bits()
  1292. {
  1293. // 配置的tab选项是真正输出空格到驱动
  1294. if space < spaces {
  1295. // 空间不够
  1296. return Err(SystemError::ENOBUFS);
  1297. }
  1298. self.cursor_column += spaces as u32;
  1299. // 写入sapces个空格
  1300. tty.write(core, " ".as_bytes(), spaces)?;
  1301. return Ok(spaces);
  1302. }
  1303. self.cursor_column += spaces as u32;
  1304. }
  1305. _ => {
  1306. // 判断是否为控制字符
  1307. if !(c as char).is_control() {
  1308. if termios.output_mode.contains(OutputMode::OLCUC) {
  1309. c = c.to_ascii_uppercase();
  1310. }
  1311. // 判断是否为utf8模式下的连续字符
  1312. if !(termios.input_mode.contains(InputMode::IUTF8)
  1313. && (c as usize) & 0xc0 == 0x80)
  1314. {
  1315. self.cursor_column += 1;
  1316. }
  1317. }
  1318. }
  1319. }
  1320. if tty.put_char(tty.core(), c).is_err() {
  1321. tty.write(core, &[c], 1)?;
  1322. }
  1323. Ok(1)
  1324. }
  1325. fn packet_mode_flush(&self, tty: &TtyCoreData) {
  1326. let link = tty.link().unwrap();
  1327. if link.core().contorl_info_irqsave().packet {
  1328. tty.contorl_info_irqsave()
  1329. .pktstatus
  1330. .insert(TtyPacketStatus::TIOCPKT_FLUSHREAD);
  1331. link.core().read_wq().wakeup_all();
  1332. }
  1333. }
  1334. }
  1335. impl TtyLineDiscipline for NTtyLinediscipline {
  1336. fn open(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1337. // 反向绑定tty到disc
  1338. self.disc_data().tty = Arc::downgrade(&tty);
  1339. // 特定的tty设备在这里可能需要取消端口节流
  1340. return self.set_termios(tty, None);
  1341. }
  1342. fn close(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1343. todo!()
  1344. }
  1345. /// ## 重置缓冲区的基本信息
  1346. fn flush_buffer(&self, tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1347. let core = tty.core();
  1348. let _ = core.termios();
  1349. let mut ldata = self.disc_data();
  1350. ldata.read_head = 0;
  1351. ldata.canon_head = 0;
  1352. ldata.read_tail = 0;
  1353. ldata.commit_head = 0;
  1354. ldata.line_start = 0;
  1355. ldata.erasing = false;
  1356. ldata.read_flags.set_all(false);
  1357. ldata.pushing = false;
  1358. ldata.lookahead_count = 0;
  1359. // todo: kick worker?
  1360. // packet mode?
  1361. if core.link().is_some() {
  1362. ldata.packet_mode_flush(core);
  1363. }
  1364. Ok(())
  1365. }
  1366. #[inline(never)]
  1367. fn read(
  1368. &self,
  1369. tty: Arc<TtyCore>,
  1370. buf: &mut [u8],
  1371. len: usize,
  1372. cookie: &mut bool,
  1373. _offset: usize,
  1374. mode: FileMode,
  1375. ) -> Result<usize, system_error::SystemError> {
  1376. let mut ldata;
  1377. if mode.contains(FileMode::O_NONBLOCK) {
  1378. let ret = self.disc_data_try_lock();
  1379. if ret.is_err() {
  1380. return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1381. }
  1382. ldata = ret.unwrap();
  1383. } else {
  1384. ldata = self.disc_data();
  1385. }
  1386. let core = tty.core();
  1387. let termios = core.termios();
  1388. let mut nr = len;
  1389. let mut offset = 0;
  1390. // 表示接着读
  1391. if *cookie {
  1392. // 规范且非拓展模式
  1393. if ldata.icanon && !termios.local_mode.contains(LocalMode::EXTPROC) {
  1394. // 跳过EOF字符
  1395. if len == 0 {
  1396. ldata.canon_skip_eof();
  1397. } else if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
  1398. return Ok(len - nr);
  1399. }
  1400. } else if ldata.copy_from_read_buf(termios, buf, &mut nr, &mut offset)? {
  1401. return Ok(len - nr);
  1402. }
  1403. // 没有数据可读
  1404. // todo: kick worker? or 关闭节流?
  1405. *cookie = false;
  1406. return Ok(len - nr);
  1407. }
  1408. drop(termios);
  1409. TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTIN)?;
  1410. let mut minimum: usize = 0;
  1411. if !ldata.icanon {
  1412. let core = tty.core();
  1413. let termios = core.termios();
  1414. minimum = termios.control_characters[ControlCharIndex::VMIN] as usize;
  1415. if minimum == 0 {
  1416. minimum = 1;
  1417. }
  1418. }
  1419. let packet = core.contorl_info_irqsave().packet;
  1420. let mut ret: Result<usize, SystemError> = Ok(0);
  1421. // 记录读取前 的tail
  1422. let tail = ldata.read_tail;
  1423. drop(ldata);
  1424. while nr != 0 {
  1425. // todo: 处理packet模式
  1426. if packet {
  1427. let link = core.link().unwrap();
  1428. let link = link.core();
  1429. let mut ctrl = link.contorl_info_irqsave();
  1430. if !ctrl.pktstatus.is_empty() {
  1431. if offset != 0 {
  1432. break;
  1433. }
  1434. let cs = ctrl.pktstatus;
  1435. ctrl.pktstatus = TtyPacketStatus::empty();
  1436. buf[offset] = cs.bits();
  1437. offset += 1;
  1438. // nr -= 1;
  1439. break;
  1440. }
  1441. }
  1442. let mut ldata = self.disc_data();
  1443. let core = tty.core();
  1444. if !ldata.input_available(core.termios(), false) {
  1445. if core.flags().contains(TtyFlag::OTHER_CLOSED) {
  1446. ret = Err(SystemError::EIO);
  1447. break;
  1448. }
  1449. if core.flags().contains(TtyFlag::HUPPED) || core.flags().contains(TtyFlag::HUPPING)
  1450. {
  1451. break;
  1452. }
  1453. if mode.contains(FileMode::O_NONBLOCK)
  1454. || core.flags().contains(TtyFlag::LDISC_CHANGING)
  1455. {
  1456. ret = Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1457. break;
  1458. }
  1459. if ProcessManager::current_pcb()
  1460. .sig_info_irqsave()
  1461. .sig_pending()
  1462. .has_pending()
  1463. {
  1464. ret = Err(SystemError::ERESTARTSYS);
  1465. break;
  1466. }
  1467. // 休眠一段时间
  1468. // 获取到termios读锁,避免termios被更改导致行为异常
  1469. // let termios = core.termios_preempt_enable();
  1470. // let helper = WakeUpHelper::new(ProcessManager::current_pcb());
  1471. // let wakeup_helper = Timer::new(helper, timeout);
  1472. // wakeup_helper.activate();
  1473. // drop(termios);
  1474. drop(ldata);
  1475. core.read_wq()
  1476. .sleep((EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM).bits() as u64);
  1477. continue;
  1478. }
  1479. if ldata.icanon && !core.termios().local_mode.contains(LocalMode::EXTPROC) {
  1480. if ldata.canon_copy_from_read_buf(buf, &mut nr, &mut offset)? {
  1481. *cookie = true;
  1482. offset += len - nr;
  1483. return Ok(offset);
  1484. }
  1485. } else {
  1486. // 非标准模式
  1487. // todo: 处理packet模式
  1488. if packet && offset == 0 {
  1489. buf[offset] = TtyPacketStatus::TIOCPKT_DATA.bits();
  1490. offset += 1;
  1491. nr -= 1;
  1492. }
  1493. // 拷贝数据
  1494. if ldata.copy_from_read_buf(core.termios(), buf, &mut nr, &mut offset)?
  1495. && offset >= minimum
  1496. {
  1497. *cookie = true;
  1498. return Ok(offset);
  1499. }
  1500. }
  1501. if offset >= minimum {
  1502. break;
  1503. }
  1504. }
  1505. let ldata = self.disc_data();
  1506. if tail != ldata.read_tail {
  1507. // todo: kick worker?
  1508. }
  1509. if offset > 0 {
  1510. return Ok(offset);
  1511. }
  1512. ret
  1513. }
  1514. #[inline(never)]
  1515. fn write(
  1516. &self,
  1517. tty: Arc<TtyCore>,
  1518. buf: &[u8],
  1519. len: usize,
  1520. mode: FileMode,
  1521. ) -> Result<usize, system_error::SystemError> {
  1522. let mut nr = len;
  1523. let mut ldata = self.disc_data();
  1524. let pcb = ProcessManager::current_pcb();
  1525. let binding = tty.clone();
  1526. let core = binding.core();
  1527. let termios = *core.termios();
  1528. if termios.local_mode.contains(LocalMode::TOSTOP) {
  1529. TtyJobCtrlManager::tty_check_change(tty.clone(), Signal::SIGTTOU)?;
  1530. }
  1531. ldata.process_echoes(tty.clone());
  1532. // drop(ldata);
  1533. let mut offset = 0;
  1534. loop {
  1535. if pcb.sig_info_irqsave().sig_pending().has_pending() {
  1536. return Err(SystemError::ERESTARTSYS);
  1537. }
  1538. if core.flags().contains(TtyFlag::HUPPED) {
  1539. return Err(SystemError::EIO);
  1540. }
  1541. if termios.output_mode.contains(OutputMode::OPOST) {
  1542. while nr > 0 {
  1543. // let mut ldata = self.disc_data();
  1544. // 获得一次处理后的数量
  1545. let ret = ldata.process_output_block(core, core.termios(), &buf[offset..], nr);
  1546. let num = match ret {
  1547. Ok(num) => num,
  1548. Err(e) => {
  1549. if e == SystemError::EAGAIN_OR_EWOULDBLOCK {
  1550. break;
  1551. } else {
  1552. return Err(e);
  1553. }
  1554. }
  1555. };
  1556. offset += num;
  1557. nr -= num;
  1558. if nr == 0 {
  1559. break;
  1560. }
  1561. let c = buf[offset];
  1562. if !ldata.process_output(tty.clone(), c) {
  1563. break;
  1564. }
  1565. offset += 1;
  1566. nr -= 1;
  1567. }
  1568. tty.flush_chars(core);
  1569. } else {
  1570. while nr > 0 {
  1571. let write = tty.write(core, &buf[offset..], nr)?;
  1572. if write == 0 {
  1573. break;
  1574. }
  1575. offset += write;
  1576. nr -= write;
  1577. }
  1578. }
  1579. if nr == 0 {
  1580. break;
  1581. }
  1582. if mode.contains(FileMode::O_NONBLOCK) || core.flags().contains(TtyFlag::LDISC_CHANGING)
  1583. {
  1584. return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
  1585. }
  1586. // 到这里表明没位置可写了
  1587. // 休眠一段时间
  1588. // 获取到termios读锁,避免termios被更改导致行为异常
  1589. core.write_wq()
  1590. .sleep(EPollEventType::EPOLLOUT.bits() as u64);
  1591. }
  1592. Ok(offset)
  1593. }
  1594. fn ioctl(
  1595. &self,
  1596. tty: Arc<TtyCore>,
  1597. cmd: u32,
  1598. arg: usize,
  1599. ) -> Result<usize, system_error::SystemError> {
  1600. match cmd {
  1601. TtyIoctlCmd::TIOCOUTQ => {
  1602. let mut user_writer = UserBufferWriter::new(
  1603. VirtAddr::new(arg).as_ptr::<i32>(),
  1604. core::mem::size_of::<i32>(),
  1605. true,
  1606. )?;
  1607. let count = tty.chars_in_buffer();
  1608. user_writer.copy_one_to_user::<i32>(&(count as i32), 0)?;
  1609. return Ok(0);
  1610. }
  1611. TtyIoctlCmd::FIONREAD => {
  1612. let ldata = self.disc_data();
  1613. let termios = tty.core().termios();
  1614. let retval;
  1615. if termios.local_mode.contains(LocalMode::ICANON)
  1616. && !termios.local_mode.contains(LocalMode::EXTPROC)
  1617. {
  1618. if ldata.canon_head == ldata.read_tail {
  1619. retval = 0;
  1620. } else {
  1621. let head = ldata.canon_head;
  1622. let mut tail = ldata.read_tail;
  1623. let mut nr = head - tail;
  1624. while ntty_buf_mask(head) != ntty_buf_mask(tail) {
  1625. if ldata.read_flags.get(ntty_buf_mask(tail)).unwrap()
  1626. && ldata.read_buf[ntty_buf_mask(tail)]
  1627. == ControlCharIndex::DISABLE_CHAR
  1628. {
  1629. nr -= 1;
  1630. }
  1631. tail += 1;
  1632. }
  1633. retval = nr;
  1634. }
  1635. } else {
  1636. retval = ldata.read_cnt();
  1637. }
  1638. let mut user_writer = UserBufferWriter::new(
  1639. VirtAddr::new(arg).as_ptr::<i32>(),
  1640. core::mem::size_of::<i32>(),
  1641. true,
  1642. )?;
  1643. user_writer.copy_one_to_user::<i32>(&(retval as i32), 0)?;
  1644. return Ok(0);
  1645. }
  1646. _ => {
  1647. return self.ioctl_helper(tty, cmd, arg);
  1648. }
  1649. }
  1650. }
  1651. #[inline(never)]
  1652. fn set_termios(
  1653. &self,
  1654. tty: Arc<TtyCore>,
  1655. old: Option<crate::driver::tty::termios::Termios>,
  1656. ) -> Result<(), system_error::SystemError> {
  1657. let core = tty.core();
  1658. let termios = core.termios();
  1659. let mut ldata = self.disc_data();
  1660. let contorl_chars = termios.control_characters;
  1661. // 第一次设置或者规范模式 (ICANON) 或者扩展处理 (EXTPROC) 标志发生变化
  1662. let mut spec_mode_changed = false;
  1663. if let Some(old) = old {
  1664. let local_mode = old.local_mode.bitxor(termios.local_mode);
  1665. spec_mode_changed =
  1666. local_mode.contains(LocalMode::ICANON) || local_mode.contains(LocalMode::EXTPROC);
  1667. }
  1668. if old.is_none() || spec_mode_changed {
  1669. // 重置read_flags
  1670. ldata.read_flags.set_all(false);
  1671. ldata.line_start = ldata.read_tail;
  1672. // 不是规范模式或者有可读数据
  1673. if !termios.local_mode.contains(LocalMode::ICANON) || ldata.read_cnt() != 0 {
  1674. ldata.canon_head = ldata.read_tail;
  1675. ldata.pushing = false;
  1676. } else {
  1677. let read_head = ldata.read_head;
  1678. ldata
  1679. .read_flags
  1680. .set((read_head - 1) & (NTTY_BUFSIZE - 1), true);
  1681. ldata.canon_head = ldata.read_head;
  1682. ldata.pushing = true;
  1683. }
  1684. ldata.commit_head = ldata.read_head;
  1685. ldata.erasing = false;
  1686. ldata.lnext = false;
  1687. }
  1688. // 设置模式
  1689. ldata.icanon = termios.local_mode.contains(LocalMode::ICANON);
  1690. // 设置回显
  1691. if termios.local_mode.contains(LocalMode::ECHO) {
  1692. ldata.echo = true;
  1693. }
  1694. if termios.input_mode.contains(InputMode::ISTRIP)
  1695. || termios.input_mode.contains(InputMode::IUCLC)
  1696. || termios.input_mode.contains(InputMode::IGNCR)
  1697. || termios.input_mode.contains(InputMode::IXON)
  1698. || termios.local_mode.contains(LocalMode::ISIG)
  1699. || termios.local_mode.contains(LocalMode::ECHO)
  1700. || termios.input_mode.contains(InputMode::PARMRK)
  1701. {
  1702. // 非原模式
  1703. ldata.char_map.set_all(false);
  1704. // 忽略回车符或者将回车映射为换行符
  1705. if termios.input_mode.contains(InputMode::IGNCR)
  1706. || termios.input_mode.contains(InputMode::ICRNL)
  1707. {
  1708. ldata.char_map.set('\r' as usize, true);
  1709. }
  1710. // 将换行映射为回车
  1711. if termios.input_mode.contains(InputMode::INLCR) {
  1712. ldata.char_map.set('\n' as usize, true);
  1713. }
  1714. // 规范模式
  1715. if termios.local_mode.contains(LocalMode::ICANON) {
  1716. ldata
  1717. .char_map
  1718. .set(contorl_chars[ControlCharIndex::VERASE] as usize, true);
  1719. ldata
  1720. .char_map
  1721. .set(contorl_chars[ControlCharIndex::VKILL] as usize, true);
  1722. ldata
  1723. .char_map
  1724. .set(contorl_chars[ControlCharIndex::VEOF] as usize, true);
  1725. ldata.char_map.set('\n' as usize, true);
  1726. ldata
  1727. .char_map
  1728. .set(contorl_chars[ControlCharIndex::VEOL] as usize, true);
  1729. if termios.local_mode.contains(LocalMode::IEXTEN) {
  1730. ldata
  1731. .char_map
  1732. .set(contorl_chars[ControlCharIndex::VWERASE] as usize, true);
  1733. ldata
  1734. .char_map
  1735. .set(contorl_chars[ControlCharIndex::VLNEXT] as usize, true);
  1736. ldata
  1737. .char_map
  1738. .set(contorl_chars[ControlCharIndex::VEOL2] as usize, true);
  1739. if termios.local_mode.contains(LocalMode::ECHO) {
  1740. ldata
  1741. .char_map
  1742. .set(contorl_chars[ControlCharIndex::VREPRINT] as usize, true);
  1743. }
  1744. }
  1745. }
  1746. // 软件流控制
  1747. if termios.input_mode.contains(InputMode::IXON) {
  1748. ldata
  1749. .char_map
  1750. .set(contorl_chars[ControlCharIndex::VSTART] as usize, true);
  1751. ldata
  1752. .char_map
  1753. .set(contorl_chars[ControlCharIndex::VSTOP] as usize, true);
  1754. }
  1755. if termios.local_mode.contains(LocalMode::ISIG) {
  1756. ldata
  1757. .char_map
  1758. .set(contorl_chars[ControlCharIndex::VINTR] as usize, true);
  1759. ldata
  1760. .char_map
  1761. .set(contorl_chars[ControlCharIndex::VQUIT] as usize, true);
  1762. ldata
  1763. .char_map
  1764. .set(contorl_chars[ControlCharIndex::VSUSP] as usize, true);
  1765. }
  1766. ldata
  1767. .char_map
  1768. .set(ControlCharIndex::DISABLE_CHAR as usize, true);
  1769. ldata.raw = false;
  1770. ldata.real_raw = false;
  1771. } else {
  1772. // 原模式或real_raw
  1773. ldata.raw = true;
  1774. ldata.real_raw = termios.input_mode.contains(InputMode::IGNBRK)
  1775. || (!termios.input_mode.contains(InputMode::BRKINT)
  1776. && !termios.input_mode.contains(InputMode::PARMRK))
  1777. && (termios.input_mode.contains(InputMode::IGNPAR)
  1778. || !termios.input_mode.contains(InputMode::INPCK))
  1779. && (core
  1780. .driver()
  1781. .flags()
  1782. .contains(TtyDriverFlag::TTY_DRIVER_REAL_RAW));
  1783. }
  1784. // if !termios.input_mode.contains(InputMode::IXON)
  1785. // && old.is_some()
  1786. // && old.unwrap().input_mode.contains(InputMode::IXON) && !
  1787. // {}
  1788. core.read_wq().wakeup_all();
  1789. core.write_wq().wakeup_all();
  1790. Ok(())
  1791. }
  1792. fn poll(&self, tty: Arc<TtyCore>) -> Result<usize, system_error::SystemError> {
  1793. let core = tty.core();
  1794. let ldata = self.disc_data();
  1795. let mut event = EPollEventType::empty();
  1796. if ldata.input_available(core.termios(), true) {
  1797. event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM)
  1798. }
  1799. if core.contorl_info_irqsave().packet {
  1800. let link = core.link();
  1801. if link.is_some()
  1802. && !link
  1803. .unwrap()
  1804. .core()
  1805. .contorl_info_irqsave()
  1806. .pktstatus
  1807. .is_empty()
  1808. {
  1809. event.insert(
  1810. EPollEventType::EPOLLPRI
  1811. | EPollEventType::EPOLLIN
  1812. | EPollEventType::EPOLLRDNORM,
  1813. );
  1814. }
  1815. }
  1816. if core.flags().contains(TtyFlag::OTHER_CLOSED) {
  1817. event.insert(EPollEventType::EPOLLHUP);
  1818. }
  1819. if core.driver().driver_funcs().chars_in_buffer() < 256
  1820. && core.driver().driver_funcs().write_room(core) > 0
  1821. {
  1822. event.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM);
  1823. }
  1824. Ok(event.bits() as usize)
  1825. }
  1826. fn hangup(&self, _tty: Arc<TtyCore>) -> Result<(), system_error::SystemError> {
  1827. todo!()
  1828. }
  1829. fn receive_buf(
  1830. &self,
  1831. tty: Arc<TtyCore>,
  1832. buf: &[u8],
  1833. flags: Option<&[u8]>,
  1834. count: usize,
  1835. ) -> Result<usize, SystemError> {
  1836. let mut ldata = self.disc_data();
  1837. ldata.receive_buf_common(tty, buf, flags, count, false)
  1838. }
  1839. fn receive_buf2(
  1840. &self,
  1841. tty: Arc<TtyCore>,
  1842. buf: &[u8],
  1843. flags: Option<&[u8]>,
  1844. count: usize,
  1845. ) -> Result<usize, SystemError> {
  1846. let mut ldata = self.disc_data();
  1847. ldata.receive_buf_common(tty, buf, flags, count, true)
  1848. }
  1849. }