tinyrl.c 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473
  1. /*
  2. * tinyrl.c
  3. */
  4. /* make sure we can get fileno() */
  5. #undef __STRICT_ANSI__
  6. /* LIBC HEADERS */
  7. #include <assert.h>
  8. #include <stdio.h>
  9. #include <stdlib.h>
  10. #include <string.h>
  11. #include <ctype.h>
  12. /* POSIX HEADERS */
  13. #include <unistd.h>
  14. #include "lub/string.h"
  15. #include "private.h"
  16. /*----------------------------------------------------------------------- */
  17. static void
  18. tty_set_raw_mode(tinyrl_t *this)
  19. {
  20. struct termios new_termios;
  21. int fd = fileno(tinyrl_vt100__get_istream(this->term));
  22. int status;
  23. status = tcgetattr(fd,&this->default_termios);
  24. if(-1 != status)
  25. {
  26. status = tcgetattr(fd,&new_termios);
  27. assert(-1 != status);
  28. new_termios.c_iflag = 0;
  29. new_termios.c_oflag = OPOST | ONLCR;
  30. new_termios.c_lflag = 0;
  31. new_termios.c_cc[VMIN] = 1;
  32. new_termios.c_cc[VTIME] = 0;
  33. /* Do the mode switch */
  34. status = tcsetattr(fd,TCSAFLUSH,&new_termios);
  35. assert(-1 != status);
  36. }
  37. }
  38. /*----------------------------------------------------------------------- */
  39. static void
  40. tty_restore_mode(const tinyrl_t *this)
  41. {
  42. int fd = fileno(tinyrl_vt100__get_istream(this->term));
  43. /* Do the mode switch */
  44. (void)tcsetattr(fd,TCSAFLUSH,&this->default_termios);
  45. }
  46. /*----------------------------------------------------------------------- */
  47. /*
  48. This is called whenever a line is edited in any way.
  49. It signals that if we are currently viewing a history line we should transfer it
  50. to the current buffer
  51. */
  52. static void
  53. changed_line(tinyrl_t *this)
  54. {
  55. /* if the current line is not our buffer then make it so */
  56. if(this->line != this->buffer)
  57. {
  58. /* replace the current buffer with the new details */
  59. free(this->buffer);
  60. this->line = this->buffer = lub_string_dup(this->line);
  61. this->buffer_size = strlen(this->buffer);
  62. assert(this->line);
  63. }
  64. }
  65. /*----------------------------------------------------------------------- */
  66. static bool_t
  67. tinyrl_key_default(tinyrl_t *this,
  68. int key)
  69. {
  70. bool_t result = BOOL_FALSE;
  71. if(key > 31)
  72. {
  73. char tmp[2];
  74. tmp[0] = (key & 0xFF),
  75. tmp[1] = '\0';
  76. /* inject this text into the buffer */
  77. result = tinyrl_insert_text(this,tmp);
  78. }
  79. else
  80. {
  81. char tmp[10];
  82. sprintf(tmp,"~%d",key);
  83. /* inject control characters as ~N where N is the ASCII code */
  84. result = tinyrl_insert_text(this,tmp);
  85. }
  86. return result;
  87. }
  88. /*-------------------------------------------------------- */
  89. static bool_t
  90. tinyrl_key_interrupt(tinyrl_t *this,
  91. int key)
  92. {
  93. tinyrl_delete_text(this,0,this->end);
  94. this->done = BOOL_TRUE;
  95. /* keep the compiler happy */
  96. key = key;
  97. return BOOL_TRUE;
  98. }
  99. /*-------------------------------------------------------- */
  100. static bool_t
  101. tinyrl_key_start_of_line(tinyrl_t *this,
  102. int key)
  103. {
  104. /* set the insertion point to the start of the line */
  105. this->point = 0;
  106. /* keep the compiler happy */
  107. key = key;
  108. return BOOL_TRUE;
  109. }
  110. /*-------------------------------------------------------- */
  111. static bool_t
  112. tinyrl_key_end_of_line(tinyrl_t *this,
  113. int key)
  114. {
  115. /* set the insertion point to the end of the line */
  116. this->point = this->end;
  117. /* keep the compiler happy */
  118. key = key;
  119. return BOOL_TRUE;
  120. }
  121. /*-------------------------------------------------------- */
  122. static bool_t
  123. tinyrl_key_kill(tinyrl_t *this,
  124. int key)
  125. {
  126. /* release any old kill string */
  127. lub_string_free(this->kill_string);
  128. /* store the killed string */
  129. this->kill_string = lub_string_dup(&this->buffer[this->point]);
  130. /* delete the text to the end of the line */
  131. tinyrl_delete_text(this,this->point,this->end);
  132. /* keep the compiler happy */
  133. key = key;
  134. return BOOL_TRUE;
  135. }
  136. /*-------------------------------------------------------- */
  137. static bool_t
  138. tinyrl_key_yank(tinyrl_t *this,
  139. int key)
  140. {
  141. bool_t result = BOOL_FALSE;
  142. if(this->kill_string)
  143. {
  144. /* insert the kill string at the current insertion point */
  145. result = tinyrl_insert_text(this,this->kill_string);
  146. }
  147. /* keep the compiler happy */
  148. key = key;
  149. return result;
  150. }
  151. /*-------------------------------------------------------- */
  152. static bool_t
  153. tinyrl_key_crlf(tinyrl_t *this,
  154. int key)
  155. {
  156. tinyrl_crlf(this);
  157. this->done = BOOL_TRUE;
  158. /* keep the compiler happy */
  159. key = key;
  160. return BOOL_TRUE;
  161. }
  162. /*-------------------------------------------------------- */
  163. static bool_t
  164. tinyrl_key_up(tinyrl_t *this,
  165. int key)
  166. {
  167. bool_t result = BOOL_FALSE;
  168. tinyrl_history_entry_t *entry = NULL;
  169. if(this->line == this->buffer)
  170. {
  171. /* go to the last history entry */
  172. entry = tinyrl_history_getlast(this->history,
  173. &this->hist_iter);
  174. }
  175. else
  176. {
  177. /* already traversing the history list so get previous */
  178. entry = tinyrl_history_getprevious(&this->hist_iter);
  179. }
  180. if(NULL != entry)
  181. {
  182. /* display the entry moving the insertion point
  183. * to the end of the line
  184. */
  185. this->line = tinyrl_history_entry__get_line(entry);
  186. this->point = this->end = strlen(this->line);
  187. result = BOOL_TRUE;
  188. }
  189. /* keep the compiler happy */
  190. key = key;
  191. return result;
  192. }
  193. /*-------------------------------------------------------- */
  194. static bool_t
  195. tinyrl_key_down(tinyrl_t *this,
  196. int key)
  197. {
  198. bool_t result = BOOL_FALSE;
  199. if(this->line != this->buffer)
  200. {
  201. /* we are not already at the bottom */
  202. /* the iterator will have been set up by the key_up() function */
  203. tinyrl_history_entry_t *entry = tinyrl_history_getnext(&this->hist_iter);
  204. if(NULL == entry)
  205. {
  206. /* nothing more in the history list */
  207. this->line = this->buffer;
  208. }
  209. else
  210. {
  211. this->line = tinyrl_history_entry__get_line(entry);
  212. }
  213. /* display the entry moving the insertion point
  214. * to the end of the line
  215. */
  216. this->point = this->end = strlen(this->line);
  217. result = BOOL_TRUE;
  218. }
  219. /* keep the compiler happy */
  220. key = key;
  221. return result;
  222. }
  223. /*-------------------------------------------------------- */
  224. static bool_t
  225. tinyrl_key_left(tinyrl_t *this,
  226. int key)
  227. {
  228. bool_t result = BOOL_FALSE;
  229. if(this->point > 0)
  230. {
  231. this->point--;
  232. result = BOOL_TRUE;
  233. }
  234. /* keep the compiler happy */
  235. key = key;
  236. return result;
  237. }
  238. /*-------------------------------------------------------- */
  239. static bool_t
  240. tinyrl_key_right(tinyrl_t *this,
  241. int key)
  242. {
  243. bool_t result = BOOL_FALSE;
  244. if(this->point < this->end)
  245. {
  246. this->point++;
  247. result = BOOL_TRUE;
  248. }
  249. /* keep the compiler happy */
  250. key = key;
  251. return result;
  252. }
  253. /*-------------------------------------------------------- */
  254. static bool_t
  255. tinyrl_key_backspace(tinyrl_t *this,
  256. int key)
  257. {
  258. bool_t result = BOOL_FALSE;
  259. if(this->point)
  260. {
  261. this->point--;
  262. tinyrl_delete_text(this,this->point,this->point);
  263. result = BOOL_TRUE;
  264. }
  265. /* keep the compiler happy */
  266. key = key;
  267. return result;
  268. }
  269. /*-------------------------------------------------------- */
  270. static bool_t
  271. tinyrl_key_delete(tinyrl_t *this,
  272. int key)
  273. {
  274. bool_t result = BOOL_FALSE;
  275. if(this->point < this->end)
  276. {
  277. tinyrl_delete_text(this,this->point,this->point);
  278. result = BOOL_TRUE;
  279. }
  280. /* keep the compiler happy */
  281. key = key;
  282. return result;
  283. }
  284. /*-------------------------------------------------------- */
  285. static bool_t
  286. tinyrl_key_clear_screen(tinyrl_t *this,
  287. int key)
  288. {
  289. tinyrl_vt100_clear_screen(this->term);
  290. tinyrl_vt100_cursor_home(this->term);
  291. tinyrl_reset_line_state(this);
  292. /* keep the compiler happy */
  293. key = key;
  294. this = this;
  295. return BOOL_TRUE;
  296. }
  297. /*-------------------------------------------------------- */
  298. static bool_t
  299. tinyrl_key_erase_line(tinyrl_t *this, int key)
  300. {
  301. tinyrl_delete_text(this,0,this->point);
  302. this->point = 0;
  303. /* keep the compiler happy */
  304. key = key;
  305. this = this;
  306. return BOOL_TRUE;
  307. }/*-------------------------------------------------------- */
  308. static bool_t
  309. tinyrl_key_escape(tinyrl_t *this,
  310. int key)
  311. {
  312. bool_t result= BOOL_FALSE;
  313. switch(tinyrl_vt100_escape_decode(this->term))
  314. {
  315. case tinyrl_vt100_CURSOR_UP:
  316. result = tinyrl_key_up(this,key);
  317. break;
  318. case tinyrl_vt100_CURSOR_DOWN:
  319. result = tinyrl_key_down(this,key);
  320. break;
  321. case tinyrl_vt100_CURSOR_LEFT:
  322. result = tinyrl_key_left(this,key);
  323. break;
  324. case tinyrl_vt100_CURSOR_RIGHT:
  325. result = tinyrl_key_right(this,key);
  326. break;
  327. case tinyrl_vt100_UNKNOWN:
  328. break;
  329. }
  330. return result;
  331. }
  332. /*-------------------------------------------------------- */
  333. static bool_t
  334. tinyrl_key_tab(tinyrl_t *this,
  335. int key)
  336. {
  337. bool_t result = BOOL_FALSE;
  338. tinyrl_match_e status = tinyrl_complete_with_extensions(this);
  339. switch(status)
  340. {
  341. case TINYRL_COMPLETED_MATCH:
  342. case TINYRL_MATCH:
  343. {
  344. /* everything is OK with the world... */
  345. result = tinyrl_insert_text(this," ");
  346. break;
  347. }
  348. case TINYRL_NO_MATCH:
  349. case TINYRL_MATCH_WITH_EXTENSIONS:
  350. case TINYRL_AMBIGUOUS:
  351. case TINYRL_COMPLETED_AMBIGUOUS:
  352. {
  353. /* oops don't change the result and let the bell ring */
  354. break;
  355. }
  356. }
  357. /* keep the compiler happy */
  358. key = key;
  359. return result;
  360. }
  361. /*-------------------------------------------------------- */
  362. static void
  363. tinyrl_fini(tinyrl_t *this)
  364. {
  365. /* delete the history session */
  366. tinyrl_history_delete(this->history);
  367. /* delete the terminal session */
  368. tinyrl_vt100_delete(this->term);
  369. /* free up any dynamic strings */
  370. lub_string_free(this->buffer);
  371. this->buffer = NULL;
  372. lub_string_free(this->kill_string);
  373. this->kill_string = NULL;
  374. lub_string_free(this->last_buffer);
  375. this->last_buffer = NULL;
  376. }
  377. /*-------------------------------------------------------- */
  378. static void
  379. tinyrl_init(tinyrl_t *this,
  380. FILE *instream,
  381. FILE *outstream,
  382. unsigned stifle,
  383. tinyrl_completion_func_t *complete_fn)
  384. {
  385. int i;
  386. for(i = 0;i < NUM_HANDLERS; i++)
  387. {
  388. this->handlers[i] = tinyrl_key_default;
  389. }
  390. /* default handlers */
  391. this->handlers[KEY_CR] = tinyrl_key_crlf;
  392. this->handlers[KEY_LF] = tinyrl_key_crlf;
  393. this->handlers[KEY_ETX] = tinyrl_key_interrupt;
  394. this->handlers[KEY_DEL] = tinyrl_key_backspace;
  395. this->handlers[KEY_BS] = tinyrl_key_backspace;
  396. this->handlers[KEY_EOT] = tinyrl_key_delete;
  397. this->handlers[KEY_ESC] = tinyrl_key_escape;
  398. this->handlers[KEY_FF] = tinyrl_key_clear_screen;
  399. this->handlers[KEY_NAK] = tinyrl_key_erase_line;
  400. this->handlers[KEY_SOH] = tinyrl_key_start_of_line;
  401. this->handlers[KEY_ENQ] = tinyrl_key_end_of_line;
  402. this->handlers[KEY_VT] = tinyrl_key_kill;
  403. this->handlers[KEY_EM] = tinyrl_key_yank;
  404. this->handlers[KEY_HT] = tinyrl_key_tab;
  405. this->line = NULL;
  406. this->max_line_length = 0;
  407. this->prompt = NULL;
  408. this->prompt_size = 0;
  409. this->buffer = NULL;
  410. this->buffer_size = 0;
  411. this->done = BOOL_FALSE;
  412. this->completion_over = BOOL_FALSE;
  413. this->point = 0;
  414. this->end = 0;
  415. this->attempted_completion_function = complete_fn;
  416. this->state = 0;
  417. this->kill_string = NULL;
  418. this->echo_char = '\0';
  419. this->echo_enabled = BOOL_TRUE;
  420. if (instream)
  421. this->isatty = isatty(fileno(instream)) ? BOOL_TRUE : BOOL_FALSE;
  422. else
  423. this->isatty = BOOL_FALSE;
  424. this->last_buffer = NULL;
  425. this->last_point = 0;
  426. this->utf8 = BOOL_FALSE;
  427. /* create the vt100 terminal */
  428. this->term = tinyrl_vt100_new(instream,outstream);
  429. /* create the history */
  430. this->history = tinyrl_history_new(stifle);
  431. }
  432. /*-------------------------------------------------------- */
  433. int
  434. tinyrl_printf(const tinyrl_t *this,
  435. const char *fmt,
  436. ...)
  437. {
  438. va_list args;
  439. int len;
  440. va_start(args, fmt);
  441. len = tinyrl_vt100_vprintf(this->term, fmt, args);
  442. va_end(args);
  443. return len;
  444. }
  445. /*-------------------------------------------------------- */
  446. void
  447. tinyrl_delete(tinyrl_t *this)
  448. {
  449. assert(this);
  450. if(this)
  451. {
  452. /* let the object tidy itself up */
  453. tinyrl_fini(this);
  454. /* release the memory associate with this instance */
  455. free(this);
  456. }
  457. }
  458. /*-------------------------------------------------------- */
  459. /*#####################################
  460. * EXPORTED INTERFACE
  461. *##################################### */
  462. /*----------------------------------------------------------------------- */
  463. int
  464. tinyrl_getchar(const tinyrl_t *this)
  465. {
  466. return tinyrl_vt100_getchar(this->term);
  467. }
  468. /*----------------------------------------------------------------------- */
  469. static void
  470. tinyrl_internal_print(const tinyrl_t *this,
  471. const char *text)
  472. {
  473. if(BOOL_TRUE == this->echo_enabled)
  474. {
  475. /* simply echo the line */
  476. tinyrl_vt100_printf(this->term,"%s",text);
  477. }
  478. else
  479. {
  480. /* replace the line with echo char if defined */
  481. if(this->echo_char)
  482. {
  483. unsigned i = strlen(text);
  484. while(i--)
  485. {
  486. tinyrl_vt100_printf(this->term,"%c",this->echo_char);
  487. }
  488. }
  489. }
  490. }
  491. /*----------------------------------------------------------------------- */
  492. void
  493. tinyrl_redisplay(tinyrl_t *this)
  494. {
  495. int delta;
  496. unsigned line_len, last_line_len,count;
  497. line_len = strlen(this->line);
  498. last_line_len = (this->last_buffer ? strlen(this->last_buffer) : 0);
  499. do
  500. {
  501. if(this->last_buffer)
  502. {
  503. delta = (int)(line_len - last_line_len);
  504. if(delta > 0)
  505. {
  506. count = (unsigned)delta;
  507. /* is the current line simply an extension of the previous one? */
  508. if(0 == strncmp(this->line,this->last_buffer,last_line_len))
  509. {
  510. /* output the line accounting for the echo behaviour */
  511. tinyrl_internal_print(this,&this->line[line_len - count]);
  512. break;
  513. }
  514. }
  515. else if(delta < 0)
  516. {
  517. /* is the current line simply a deletion of some characters from the end? */
  518. if(0 == strncmp(this->line,this->last_buffer,line_len))
  519. {
  520. if(this->echo_enabled || this->echo_char)
  521. {
  522. int shift = (int)(this->last_point - this->point);
  523. /* just get the terminal to delete the characters */
  524. if(shift > 0)
  525. {
  526. count = (unsigned)shift;
  527. /* we've moved the cursor backwards */
  528. tinyrl_vt100_cursor_back(this->term,count);
  529. }
  530. else if(shift < 0)
  531. {
  532. count = (unsigned)-shift;
  533. /* we've moved the cursor forwards */
  534. tinyrl_vt100_cursor_forward(this->term,count);
  535. }
  536. /* now delete the characters */
  537. count = (unsigned)-delta;
  538. tinyrl_vt100_erase(this->term,count);
  539. }
  540. break;
  541. }
  542. }
  543. else
  544. {
  545. /* are the lines are the same content? */
  546. if(0 == strcmp(this->line,this->last_buffer))
  547. {
  548. if(this->echo_enabled || this->echo_char)
  549. {
  550. delta = (int)(this->point - this->last_point);
  551. if(delta > 0)
  552. {
  553. count = (unsigned)delta;
  554. /* move the point forwards */
  555. tinyrl_vt100_cursor_forward(this->term,count);
  556. }
  557. else if(delta < 0)
  558. {
  559. count = (unsigned)-delta;
  560. /* move the cursor backwards */
  561. tinyrl_vt100_cursor_back(this->term,count);
  562. }
  563. }
  564. /* done for now */
  565. break;
  566. }
  567. }
  568. }
  569. else
  570. {
  571. /* simply display the prompt and the line */
  572. tinyrl_vt100_printf(this->term,"%s",this->prompt);
  573. tinyrl_internal_print(this,this->line);
  574. if(this->point < line_len)
  575. {
  576. /* move the cursor to the insertion point */
  577. tinyrl_vt100_cursor_back(this->term,line_len-this->point);
  578. }
  579. break;
  580. }
  581. /*
  582. * to have got this far we must have edited the middle of a line.
  583. */
  584. if(this->last_point)
  585. {
  586. /* move to just after the prompt of the line */
  587. tinyrl_vt100_cursor_back(this->term,this->last_point);
  588. }
  589. /* erase the previous line */
  590. tinyrl_vt100_erase(this->term,strlen(this->last_buffer));
  591. /* output the line accounting for the echo behaviour */
  592. tinyrl_internal_print(this,this->line);
  593. delta = (int)(line_len - this->point);
  594. if(delta)
  595. {
  596. /* move the cursor back to the insertion point */
  597. tinyrl_vt100_cursor_back(this->term,
  598. (strlen(this->line) - this->point));
  599. }
  600. } /*lint -e717 */ while(0) /*lint +e717 */;
  601. /* update the display */
  602. (void)tinyrl_vt100_oflush(this->term);
  603. /* set up the last line buffer */
  604. lub_string_free(this->last_buffer);
  605. this->last_buffer = lub_string_dup(this->line);
  606. this->last_point = this->point;
  607. }
  608. /*----------------------------------------------------------------------- */
  609. tinyrl_t *
  610. tinyrl_new(FILE *instream,
  611. FILE *outstream,
  612. unsigned stifle,
  613. tinyrl_completion_func_t *complete_fn)
  614. {
  615. tinyrl_t *this=NULL;
  616. this = malloc(sizeof(tinyrl_t));
  617. if(NULL != this)
  618. {
  619. tinyrl_init(this,instream,outstream,stifle,complete_fn);
  620. }
  621. return this;
  622. }
  623. /*----------------------------------------------------------------------- */
  624. static char *
  625. internal_insertline(tinyrl_t *this,
  626. char *buffer)
  627. {
  628. char *p;
  629. char *s = buffer;
  630. /* strip any spurious '\r' or '\n' */
  631. p = strchr(buffer,'\r');
  632. if(NULL == p)
  633. {
  634. p = strchr(buffer,'\n');
  635. }
  636. if (NULL != p)
  637. {
  638. *p = '\0';
  639. }
  640. /* skip any whitespace at the beginning of the line */
  641. if(0 == this->point)
  642. {
  643. while(*s && isspace(*s))
  644. {
  645. s++;
  646. }
  647. }
  648. if(*s)
  649. {
  650. /* append this string to the input buffer */
  651. (void) tinyrl_insert_text(this,s);
  652. /* echo the command to the output stream */
  653. tinyrl_redisplay(this);
  654. }
  655. return s;
  656. }
  657. /*----------------------------------------------------------------------- */
  658. static char *
  659. internal_readline(tinyrl_t *this,
  660. const char *prompt,
  661. void *context,
  662. const char *str)
  663. {
  664. FILE *istream = tinyrl_vt100__get_istream(this->term);
  665. int crlf = 1; /* Enable crlf if result is NULL */
  666. /* initialise for reading a line */
  667. this->done = BOOL_FALSE;
  668. this->point = 0;
  669. this->end = 0;
  670. this->buffer = lub_string_dup("");
  671. this->buffer_size = strlen(this->buffer);
  672. this->line = this->buffer;
  673. this->prompt = prompt;
  674. this->prompt_size = strlen(prompt);
  675. this->context = context;
  676. if((BOOL_TRUE == this->isatty) && (!str))
  677. {
  678. /* set the terminal into raw input mode */
  679. tty_set_raw_mode(this);
  680. tinyrl_reset_line_state(this);
  681. while(!this->done)
  682. {
  683. int key;
  684. /* update the display */
  685. tinyrl_redisplay(this);
  686. /* get a key */
  687. key = tinyrl_getchar(this);
  688. /* has the input stream terminated? */
  689. if(EOF != key)
  690. {
  691. /* call the handler for this key */
  692. if(BOOL_FALSE == this->handlers[key](this,key))
  693. {
  694. /* an issue has occured */
  695. tinyrl_ding(this);
  696. }
  697. if (BOOL_TRUE == this->done)
  698. {
  699. /*
  700. * If the last character in the line (other than
  701. * the null) is a space remove it.
  702. */
  703. if (this->end && isspace(this->line[this->end-1]))
  704. {
  705. tinyrl_delete_text(this,this->end-1,this->end);
  706. }
  707. }
  708. }
  709. else
  710. {
  711. /* time to finish the session */
  712. this->done = BOOL_TRUE;
  713. this->line = NULL;
  714. }
  715. }
  716. /* restores the terminal mode */
  717. tty_restore_mode(this);
  718. }
  719. else
  720. {
  721. /* This is a non-interactive set of commands */
  722. char *s = 0, buffer[80];
  723. size_t len = sizeof(buffer);
  724. char *tmp = NULL;
  725. /* manually reset the line state without redisplaying */
  726. lub_string_free(this->last_buffer);
  727. this->last_buffer = NULL;
  728. if (str) {
  729. tmp = lub_string_dup(str);
  730. s = internal_insertline(this, tmp);
  731. } else {
  732. while((sizeof(buffer) == len) &&
  733. (s = fgets(buffer,sizeof(buffer),istream)))
  734. {
  735. s = internal_insertline(this, buffer);
  736. len = strlen(buffer) + 1; /* account for the '\0' */
  737. }
  738. if (s == NULL || (this->line[0] == '\0' && feof(istream)))
  739. {
  740. /* time to finish the session */
  741. this->line = NULL;
  742. crlf = 0;
  743. }
  744. }
  745. /*
  746. * check against fgets returning null as either error or end of file.
  747. * This is a measure to stop potential task spin on encountering an
  748. * error from fgets.
  749. */
  750. if (this->line)
  751. {
  752. if (this->line[0] == '\0')
  753. {
  754. tinyrl_reset_line_state(this);
  755. } /* call the handler for the newline key */
  756. else if(BOOL_FALSE == this->handlers[KEY_LF](this,KEY_LF))
  757. {
  758. /* an issue has occured */
  759. tinyrl_ding(this);
  760. this->line = NULL;
  761. }
  762. }
  763. if (str)
  764. lub_string_free(tmp);
  765. }
  766. /*
  767. * duplicate the string for return to the client
  768. * we have to duplicate as we may be referencing a
  769. * history entry or our internal buffer
  770. */
  771. {
  772. char *result = this->line ? lub_string_dup(this->line) : NULL;
  773. /* free our internal buffer */
  774. free(this->buffer);
  775. this->buffer = NULL;
  776. if(crlf && ((NULL == result) || ('\0' == *result)))
  777. {
  778. /* make sure we're not left on a prompt line */
  779. tinyrl_crlf(this);
  780. }
  781. return result;
  782. }
  783. }
  784. /*----------------------------------------------------------------------- */
  785. char *
  786. tinyrl_readline(tinyrl_t *this,
  787. const char *prompt,
  788. void *context)
  789. {
  790. return internal_readline(this, prompt, context, NULL);
  791. }
  792. /*----------------------------------------------------------------------- */
  793. char *
  794. tinyrl_forceline(tinyrl_t *this,
  795. const char *prompt,
  796. void *context,
  797. const char *line)
  798. {
  799. return internal_readline(this, prompt, context, line);
  800. }
  801. /*----------------------------------------------------------------------- */
  802. /*
  803. * Ensure that buffer has enough space to hold len characters,
  804. * possibly reallocating it if necessary. The function returns BOOL_TRUE
  805. * if the line is successfully extended, BOOL_FALSE if not.
  806. */
  807. bool_t
  808. tinyrl_extend_line_buffer(tinyrl_t *this,
  809. unsigned len)
  810. {
  811. bool_t result = BOOL_TRUE;
  812. if(this->buffer_size < len)
  813. {
  814. char * new_buffer;
  815. size_t new_len = len;
  816. /*
  817. * What we do depends on whether we are limited by
  818. * memory or a user imposed limit.
  819. */
  820. if (this->max_line_length == 0)
  821. {
  822. if(new_len < this->buffer_size + 10)
  823. {
  824. /* make sure we don't realloc too often */
  825. new_len = this->buffer_size + 10;
  826. }
  827. /* leave space for terminator */
  828. new_buffer = realloc(this->buffer,new_len+1);
  829. if(NULL == new_buffer)
  830. {
  831. tinyrl_ding(this);
  832. result = BOOL_FALSE;
  833. }
  834. else
  835. {
  836. this->buffer_size = new_len;
  837. this->line = this->buffer = new_buffer;
  838. }
  839. }
  840. else
  841. {
  842. if(new_len < this->max_line_length)
  843. {
  844. /* Just reallocate once to the max size */
  845. new_buffer = realloc(this->buffer,this->max_line_length);
  846. if(NULL == new_buffer)
  847. {
  848. tinyrl_ding(this);
  849. result = BOOL_FALSE;
  850. }
  851. else
  852. {
  853. this->buffer_size = this->max_line_length - 1;
  854. this->line = this->buffer = new_buffer;
  855. }
  856. }
  857. else
  858. {
  859. tinyrl_ding(this);
  860. result = BOOL_FALSE;
  861. }
  862. }
  863. }
  864. return result;
  865. }
  866. /*----------------------------------------------------------------------- */
  867. /*
  868. * Insert text into the line at the current cursor position.
  869. */
  870. bool_t
  871. tinyrl_insert_text(tinyrl_t *this,
  872. const char *text)
  873. {
  874. unsigned delta = strlen(text);
  875. /*
  876. * If the client wants to change the line ensure that the line and buffer
  877. * references are in sync
  878. */
  879. changed_line(this);
  880. if((delta + this->end) > (this->buffer_size))
  881. {
  882. /* extend the current buffer */
  883. if (BOOL_FALSE == tinyrl_extend_line_buffer(this,this->end + delta))
  884. {
  885. return BOOL_FALSE;
  886. }
  887. }
  888. if(this->point < this->end)
  889. {
  890. /* move the current text to the right (including the terminator) */
  891. memmove(&this->buffer[this->point+delta],
  892. &this->buffer[this->point],
  893. (this->end - this->point) + 1);
  894. }
  895. else
  896. {
  897. /* terminate the string */
  898. this->buffer[this->end+delta] = '\0';
  899. }
  900. /* insert the new text */
  901. strncpy(&this->buffer[this->point],text,delta);
  902. /* now update the indexes */
  903. this->point += delta;
  904. this->end += delta;
  905. return BOOL_TRUE;
  906. }
  907. /*----------------------------------------------------------------------- */
  908. /*
  909. * A convenience function for displaying a list of strings in columnar
  910. * format on Readline's output stream. matches is the list of strings,
  911. * in argv format, such as a list of completion matches. len is the number
  912. * of strings in matches, and max is the length of the longest string in matches.
  913. * This function uses the setting of print-completions-horizontally to select
  914. * how the matches are displayed
  915. */
  916. void
  917. tinyrl_display_matches(const tinyrl_t *this,
  918. char *const *matches,
  919. unsigned len,
  920. size_t max)
  921. {
  922. unsigned r,c;
  923. unsigned width = tinyrl_vt100__get_width(this->term);
  924. unsigned cols = width/(max+1); /* allow for a space between words */
  925. unsigned rows = len/cols + 1;
  926. assert(matches);
  927. if(matches)
  928. {
  929. len--,matches++; /* skip the subtitution string */
  930. /* print out a table of completions */
  931. for(r=0; r<rows && len; r++)
  932. {
  933. for(c=0; c<cols && len; c++)
  934. {
  935. const char *match = *matches++;
  936. len--;
  937. tinyrl_vt100_printf(this->term,"%-*s ",max,match);
  938. }
  939. tinyrl_crlf(this);
  940. }
  941. }
  942. }
  943. /*----------------------------------------------------------------------- */
  944. /*
  945. * Delete the text between start and end in the current line. (inclusive)
  946. * This adjusts the rl_point and rl_end indexes appropriately.
  947. */
  948. void
  949. tinyrl_delete_text(tinyrl_t *this,
  950. unsigned start,
  951. unsigned end)
  952. {
  953. unsigned delta;
  954. /*
  955. * If the client wants to change the line ensure that the line and buffer
  956. * references are in sync
  957. */
  958. changed_line(this);
  959. /* make sure we play it safe */
  960. if(start > end)
  961. {
  962. unsigned tmp = end;
  963. start = end;
  964. end = tmp;
  965. }
  966. if(end > this->end)
  967. {
  968. end = this->end;
  969. }
  970. delta = (end - start) + 1;
  971. /* move any text which is left */
  972. memmove(&this->buffer[start],
  973. &this->buffer[start+delta],
  974. this->end - end);
  975. /* now adjust the indexs */
  976. if(this->point >= start)
  977. {
  978. if(this->point > end)
  979. {
  980. /* move the insertion point back appropriately */
  981. this->point -= delta;
  982. }
  983. else
  984. {
  985. /* move the insertion point to the start */
  986. this->point = start;
  987. }
  988. }
  989. if(this->end > end)
  990. {
  991. this->end -= delta;
  992. }
  993. else
  994. {
  995. this->end = start;
  996. }
  997. /* put a terminator at the end of the buffer */
  998. this->buffer[this->end] ='\0';
  999. }
  1000. /*----------------------------------------------------------------------- */
  1001. bool_t
  1002. tinyrl_bind_key(tinyrl_t *this,
  1003. int key,
  1004. tinyrl_key_func_t *fn)
  1005. {
  1006. bool_t result = BOOL_FALSE;
  1007. if((key >= 0) && (key < 256))
  1008. {
  1009. /* set the key handling function */
  1010. this->handlers[key] = fn;
  1011. result = BOOL_TRUE;
  1012. }
  1013. return result;
  1014. }
  1015. /*-------------------------------------------------------- */
  1016. /*
  1017. * Returns an array of strings which is a list of completions for text.
  1018. * If there are no completions, returns NULL. The first entry in the
  1019. * returned array is the substitution for text. The remaining entries
  1020. * are the possible completions. The array is terminated with a NULL pointer.
  1021. *
  1022. * entry_func is a function of two args, and returns a char *.
  1023. * The first argument is text. The second is a state argument;
  1024. * it is zero on the first call, and non-zero on subsequent calls.
  1025. * entry_func returns a NULL pointer to the caller when there are no
  1026. * more matches.
  1027. */
  1028. char **
  1029. tinyrl_completion(tinyrl_t *this,
  1030. const char *line,
  1031. unsigned start,
  1032. unsigned end,
  1033. tinyrl_compentry_func_t *entry_func)
  1034. {
  1035. unsigned state = 0;
  1036. size_t size = 1;
  1037. unsigned offset = 1; /* need at least one entry for the substitution */
  1038. char **matches = NULL;
  1039. char *match;
  1040. /* duplicate the string upto the insertion point */
  1041. char *text = lub_string_dupn(line,end);
  1042. /* now try and find possible completions */
  1043. while( (match = entry_func(this,text,start,state++)) )
  1044. {
  1045. if(size == offset)
  1046. {
  1047. /* resize the buffer if needed - the +1 is for the NULL terminator */
  1048. size += 10;
  1049. matches = realloc(matches,(sizeof(char *) * (size + 1)));
  1050. }
  1051. if (NULL == matches)
  1052. {
  1053. /* not much we can do... */
  1054. break;
  1055. }
  1056. matches[offset] = match;
  1057. /*
  1058. * augment the substitute string with this entry
  1059. */
  1060. if(1 == offset)
  1061. {
  1062. /* let's be optimistic */
  1063. matches[0] = lub_string_dup(match);
  1064. }
  1065. else
  1066. {
  1067. char *p = matches[0];
  1068. size_t match_len = strlen(p);
  1069. /* identify the common prefix */
  1070. while((tolower(*p) == tolower(*match)) && match_len--)
  1071. {
  1072. p++,match++;
  1073. }
  1074. /* terminate the prefix string */
  1075. *p = '\0';
  1076. }
  1077. offset++;
  1078. }
  1079. /* be a good memory citizen */
  1080. lub_string_free(text);
  1081. if(matches)
  1082. {
  1083. matches[offset] = NULL;
  1084. }
  1085. return matches;
  1086. }
  1087. /*-------------------------------------------------------- */
  1088. void
  1089. tinyrl_delete_matches(char **this)
  1090. {
  1091. char **matches = this;
  1092. while(*matches)
  1093. {
  1094. /* release the memory for each contained string */
  1095. free(*matches++);
  1096. }
  1097. /* release the memory for the array */
  1098. free(this);
  1099. }
  1100. /*-------------------------------------------------------- */
  1101. void
  1102. tinyrl_crlf(const tinyrl_t *this)
  1103. {
  1104. tinyrl_vt100_printf(this->term,"\n");
  1105. }
  1106. /*-------------------------------------------------------- */
  1107. /*
  1108. * Ring the terminal bell, obeying the setting of bell-style.
  1109. */
  1110. void
  1111. tinyrl_ding(const tinyrl_t *this)
  1112. {
  1113. tinyrl_vt100_ding(this->term);
  1114. }
  1115. /*-------------------------------------------------------- */
  1116. void
  1117. tinyrl_reset_line_state(tinyrl_t *this)
  1118. {
  1119. /* start from scratch */
  1120. lub_string_free(this->last_buffer);
  1121. this->last_buffer = NULL;
  1122. tinyrl_redisplay(this);
  1123. }
  1124. /*-------------------------------------------------------- */
  1125. void
  1126. tinyrl_replace_line(tinyrl_t *this,
  1127. const char *text,
  1128. int clear_undo)
  1129. {
  1130. size_t new_len = strlen(text);
  1131. /* ignored for now */
  1132. clear_undo = clear_undo;
  1133. /* ensure there is sufficient space */
  1134. if (BOOL_TRUE == tinyrl_extend_line_buffer(this,new_len))
  1135. {
  1136. /* overwrite the current contents of the buffer */
  1137. strcpy(this->buffer,text);
  1138. /* set the insert point and end point */
  1139. this->point = this->end = new_len;
  1140. }
  1141. tinyrl_redisplay(this);
  1142. }
  1143. /*-------------------------------------------------------- */
  1144. static tinyrl_match_e
  1145. tinyrl_do_complete(tinyrl_t *this,
  1146. bool_t with_extensions)
  1147. {
  1148. tinyrl_match_e result = TINYRL_NO_MATCH;
  1149. char **matches = NULL;
  1150. unsigned start,end;
  1151. bool_t completion = BOOL_FALSE;
  1152. bool_t prefix = BOOL_FALSE;
  1153. /* find the start and end of the current word */
  1154. start = end = this->point;
  1155. while(start && !isspace(this->line[start-1]))
  1156. {
  1157. start--;
  1158. }
  1159. if(this->attempted_completion_function)
  1160. {
  1161. this->completion_over = BOOL_FALSE;
  1162. this->completion_error_over = BOOL_FALSE;
  1163. /* try and complete the current line buffer */
  1164. matches = this->attempted_completion_function(this,
  1165. this->line,
  1166. start,
  1167. end);
  1168. }
  1169. if((NULL == matches)
  1170. && (BOOL_FALSE == this->completion_over))
  1171. {
  1172. /* insert default completion call here... */
  1173. }
  1174. if(matches)
  1175. {
  1176. /* identify and insert a common prefix if there is one */
  1177. if(0 != strncmp(matches[0],&this->line[start],strlen(matches[0])))
  1178. {
  1179. /*
  1180. * delete the original text not including
  1181. * the current insertion point character
  1182. */
  1183. if(this->end != end)
  1184. {
  1185. end--;
  1186. }
  1187. tinyrl_delete_text(this,start,end);
  1188. if (BOOL_FALSE == tinyrl_insert_text(this,matches[0]))
  1189. {
  1190. return TINYRL_NO_MATCH;
  1191. }
  1192. completion = BOOL_TRUE;
  1193. }
  1194. if(0 == lub_string_nocasecmp(matches[0],matches[1]))
  1195. {
  1196. /* this is just a prefix string */
  1197. prefix = BOOL_TRUE;
  1198. }
  1199. /* is there more than one completion? */
  1200. if(matches[2] != NULL)
  1201. {
  1202. char **tmp = matches;
  1203. unsigned max,len;
  1204. max = len = 0;
  1205. while(*tmp)
  1206. {
  1207. size_t size = strlen(*tmp++);
  1208. len++;
  1209. if(size > max)
  1210. {
  1211. max = size;
  1212. }
  1213. }
  1214. if(BOOL_TRUE == completion)
  1215. {
  1216. result = TINYRL_COMPLETED_AMBIGUOUS;
  1217. }
  1218. else if(BOOL_TRUE == prefix)
  1219. {
  1220. result = TINYRL_MATCH_WITH_EXTENSIONS;
  1221. }
  1222. else
  1223. {
  1224. result = TINYRL_AMBIGUOUS;
  1225. }
  1226. if((BOOL_TRUE == with_extensions)
  1227. || (BOOL_FALSE == prefix))
  1228. {
  1229. /* Either we always want to show extensions or
  1230. * we haven't been able to complete the current line
  1231. * and there is just a prefix, so let the user see the options
  1232. */
  1233. tinyrl_crlf(this);
  1234. tinyrl_display_matches(this,matches,len,max);
  1235. tinyrl_reset_line_state(this);
  1236. }
  1237. }
  1238. else
  1239. {
  1240. result = completion ? TINYRL_COMPLETED_MATCH : TINYRL_MATCH;
  1241. }
  1242. /* free the memory */
  1243. tinyrl_delete_matches(matches);
  1244. /* redisplay the line */
  1245. tinyrl_redisplay(this);
  1246. }
  1247. return result;
  1248. }
  1249. /*-------------------------------------------------------- */
  1250. tinyrl_match_e
  1251. tinyrl_complete_with_extensions(tinyrl_t *this)
  1252. {
  1253. return tinyrl_do_complete(this,BOOL_TRUE);
  1254. }
  1255. /*-------------------------------------------------------- */
  1256. tinyrl_match_e
  1257. tinyrl_complete(tinyrl_t *this)
  1258. {
  1259. return tinyrl_do_complete(this,BOOL_FALSE);
  1260. }
  1261. /*-------------------------------------------------------- */
  1262. void *
  1263. tinyrl__get_context(const tinyrl_t *this)
  1264. {
  1265. return this->context;
  1266. }
  1267. /*--------------------------------------------------------- */
  1268. const char *
  1269. tinyrl__get_line(const tinyrl_t *this)
  1270. {
  1271. return this->line;
  1272. }
  1273. /*--------------------------------------------------------- */
  1274. tinyrl_history_t *
  1275. tinyrl__get_history(const tinyrl_t *this)
  1276. {
  1277. return this->history;
  1278. }
  1279. /*--------------------------------------------------------- */
  1280. void
  1281. tinyrl_completion_over(tinyrl_t *this)
  1282. {
  1283. this->completion_over = BOOL_TRUE;
  1284. }
  1285. /*--------------------------------------------------------- */
  1286. void
  1287. tinyrl_completion_error_over(tinyrl_t *this)
  1288. {
  1289. this->completion_error_over = BOOL_TRUE;
  1290. }
  1291. /*--------------------------------------------------------- */
  1292. bool_t
  1293. tinyrl_is_completion_error_over(const tinyrl_t *this)
  1294. {
  1295. return this->completion_error_over;
  1296. }
  1297. /*--------------------------------------------------------- */
  1298. void
  1299. tinyrl_done(tinyrl_t *this)
  1300. {
  1301. this->done = BOOL_TRUE;
  1302. }
  1303. /*--------------------------------------------------------- */
  1304. void
  1305. tinyrl_enable_echo(tinyrl_t *this)
  1306. {
  1307. this->echo_enabled = BOOL_TRUE;
  1308. }
  1309. /*--------------------------------------------------------- */
  1310. void
  1311. tinyrl_disable_echo(tinyrl_t *this,
  1312. char echo_char)
  1313. {
  1314. this->echo_enabled = BOOL_FALSE;
  1315. this->echo_char = echo_char;
  1316. }
  1317. /*--------------------------------------------------------- */
  1318. void
  1319. tinyrl__set_istream(tinyrl_t *this,
  1320. FILE *istream)
  1321. {
  1322. tinyrl_vt100__set_istream(this->term,istream);
  1323. this->isatty = isatty(fileno(istream)) ? BOOL_TRUE : BOOL_FALSE;
  1324. }
  1325. /*-------------------------------------------------------- */
  1326. bool_t
  1327. tinyrl__get_isatty(const tinyrl_t *this)
  1328. {
  1329. return this->isatty;
  1330. }
  1331. /*-------------------------------------------------------- */
  1332. FILE *
  1333. tinyrl__get_istream(const tinyrl_t *this)
  1334. {
  1335. return tinyrl_vt100__get_istream(this->term);
  1336. }
  1337. /*-------------------------------------------------------- */
  1338. FILE *
  1339. tinyrl__get_ostream(const tinyrl_t *this)
  1340. {
  1341. return tinyrl_vt100__get_ostream(this->term);
  1342. }
  1343. /*-------------------------------------------------------- */
  1344. const char *
  1345. tinyrl__get_prompt(const tinyrl_t *this)
  1346. {
  1347. return this->prompt;
  1348. }
  1349. /*-------------------------------------------------------- */
  1350. bool_t
  1351. tinyrl__get_utf8(const tinyrl_t *this)
  1352. {
  1353. return this->utf8;
  1354. }
  1355. /*-------------------------------------------------------- */
  1356. void
  1357. tinyrl__set_utf8(tinyrl_t *this, bool_t utf8)
  1358. {
  1359. this->utf8 = utf8;
  1360. }
  1361. /*-------------------------------------------------------- */
  1362. bool_t
  1363. tinyrl_is_quoting(const tinyrl_t *this)
  1364. {
  1365. bool_t result = BOOL_FALSE;
  1366. /* count the quotes upto the current insertion point */
  1367. unsigned i = 0;
  1368. while(i < this->point)
  1369. {
  1370. if(this->line[i++] == '"')
  1371. {
  1372. result = result ? BOOL_FALSE : BOOL_TRUE;
  1373. }
  1374. }
  1375. return result;
  1376. }
  1377. /*--------------------------------------------------------- */
  1378. void
  1379. tinyrl_limit_line_length(tinyrl_t *this, unsigned length)
  1380. {
  1381. this->max_line_length = length;
  1382. }
  1383. /*--------------------------------------------------------- */