setup.c 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. #include <signal.h>
  2. #include <stdbool.h>
  3. #include "../../util/cache.h"
  4. #include "../../util/debug.h"
  5. #include "../browser.h"
  6. #include "../helpline.h"
  7. #include "../ui.h"
  8. #include "../util.h"
  9. #include "../libslang.h"
  10. #include "../keysyms.h"
  11. static volatile int ui__need_resize;
  12. extern struct perf_error_ops perf_tui_eops;
  13. extern void hist_browser__init_hpp(void);
  14. void ui__refresh_dimensions(bool force)
  15. {
  16. if (force || ui__need_resize) {
  17. ui__need_resize = 0;
  18. pthread_mutex_lock(&ui__lock);
  19. SLtt_get_screen_size();
  20. SLsmg_reinit_smg();
  21. pthread_mutex_unlock(&ui__lock);
  22. }
  23. }
  24. static void ui__sigwinch(int sig __maybe_unused)
  25. {
  26. ui__need_resize = 1;
  27. }
  28. static void ui__setup_sigwinch(void)
  29. {
  30. static bool done;
  31. if (done)
  32. return;
  33. done = true;
  34. pthread__unblock_sigwinch();
  35. signal(SIGWINCH, ui__sigwinch);
  36. }
  37. int ui__getch(int delay_secs)
  38. {
  39. struct timeval timeout, *ptimeout = delay_secs ? &timeout : NULL;
  40. fd_set read_set;
  41. int err, key;
  42. ui__setup_sigwinch();
  43. FD_ZERO(&read_set);
  44. FD_SET(0, &read_set);
  45. if (delay_secs) {
  46. timeout.tv_sec = delay_secs;
  47. timeout.tv_usec = 0;
  48. }
  49. err = select(1, &read_set, NULL, NULL, ptimeout);
  50. if (err == 0)
  51. return K_TIMER;
  52. if (err == -1) {
  53. if (errno == EINTR)
  54. return K_RESIZE;
  55. return K_ERROR;
  56. }
  57. key = SLang_getkey();
  58. if (key != K_ESC)
  59. return key;
  60. FD_ZERO(&read_set);
  61. FD_SET(0, &read_set);
  62. timeout.tv_sec = 0;
  63. timeout.tv_usec = 20;
  64. err = select(1, &read_set, NULL, NULL, &timeout);
  65. if (err == 0)
  66. return K_ESC;
  67. SLang_ungetkey(key);
  68. return SLkp_getkey();
  69. }
  70. static void ui__signal(int sig)
  71. {
  72. ui__exit(false);
  73. psignal(sig, "perf");
  74. exit(0);
  75. }
  76. int ui__init(void)
  77. {
  78. int err;
  79. SLutf8_enable(-1);
  80. SLtt_get_terminfo();
  81. SLtt_get_screen_size();
  82. err = SLsmg_init_smg();
  83. if (err < 0)
  84. goto out;
  85. err = SLang_init_tty(0, 0, 0);
  86. if (err < 0)
  87. goto out;
  88. err = SLkp_init();
  89. if (err < 0) {
  90. pr_err("TUI initialization failed.\n");
  91. goto out;
  92. }
  93. SLkp_define_keysym((char *)"^(kB)", SL_KEY_UNTAB);
  94. ui_helpline__init();
  95. ui_browser__init();
  96. ui_progress__init();
  97. signal(SIGSEGV, ui__signal);
  98. signal(SIGFPE, ui__signal);
  99. signal(SIGINT, ui__signal);
  100. signal(SIGQUIT, ui__signal);
  101. signal(SIGTERM, ui__signal);
  102. perf_error__register(&perf_tui_eops);
  103. hist_browser__init_hpp();
  104. out:
  105. return err;
  106. }
  107. void ui__exit(bool wait_for_ok)
  108. {
  109. if (wait_for_ok)
  110. ui__question_window("Fatal Error",
  111. ui_helpline__last_msg,
  112. "Press any key...", 0);
  113. SLtt_set_cursor_visibility(1);
  114. SLsmg_refresh();
  115. SLsmg_reset_smg();
  116. SLang_reset_tty();
  117. perf_error__unregister(&perf_tui_eops);
  118. }