/[LeafOK_CVS]/lbbs/src/bbs_net.c
ViewVC logotype

Diff of /lbbs/src/bbs_net.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.86 by sysadm, Sun Nov 30 11:59:18 2025 UTC Revision 1.108 by sysadm, Sat Dec 20 10:46:42 2025 UTC
# Line 49  static const char MENU_CONF_DELIM[] = " Line 49  static const char MENU_CONF_DELIM[] = "
49    
50  enum _bbs_net_constant_t  enum _bbs_net_constant_t
51  {  {
         MAX_PROCESS_BAR_LEN = 30,  
52          MAXSTATION = 26 * 2,          MAXSTATION = 26 * 2,
53          STATION_PER_LINE = 4,          STATION_PER_LINE = 4,
54          USERNAME_MAX_LEN = 20,          USERNAME_MAX_LEN = 20,
55          PASSWORD_MAX_LEN = 20,          PASSWORD_MAX_LEN = 20,
56            REMOTE_CONNECT_TIMEOUT = 10, // seconds
57            SSH_CONNECT_TIMEOUT = 5,         // seconds
58            PROGRESS_BAR_LEN = 30,
59  };  };
60    
61  struct _bbsnet_conf  struct _bbsnet_conf
# Line 61  struct _bbsnet_conf Line 63  struct _bbsnet_conf
63          char org_name[40];          char org_name[40];
64          char site_name[40];          char site_name[40];
65          char host_name[IP_ADDR_LEN];          char host_name[IP_ADDR_LEN];
66          in_port_t port;          char port[6];
67          int8_t use_ssh;          int8_t use_ssh;
68          char charset[CHARSET_MAX_LEN + 1];          char charset[CHARSET_MAX_LEN + 1];
69  } bbsnet_conf[MAXSTATION];  } bbsnet_conf[MAXSTATION];
70    
71  static MENU_SET bbsnet_menu;  static MENU_SET bbsnet_menu;
72    
73    static void unload_bbsnet_conf(void);
74    
75  static int load_bbsnet_conf(const char *file_config)  static int load_bbsnet_conf(const char *file_config)
76  {  {
77          FILE *fp;          FILE *fp;
# Line 75  static int load_bbsnet_conf(const char * Line 79  static int load_bbsnet_conf(const char *
79          MENU_ITEM *p_menu_item;          MENU_ITEM *p_menu_item;
80          MENU_ITEM_ID menu_item_id;          MENU_ITEM_ID menu_item_id;
81          char line[LINE_BUFFER_LEN], *t1, *t2, *t3, *t4, *t5, *t6, *saveptr;          char line[LINE_BUFFER_LEN], *t1, *t2, *t3, *t4, *t5, *t6, *saveptr;
82            long port;
83            char *endptr;
84    
85          fp = fopen(file_config, "r");          unload_bbsnet_conf();
         if (fp == NULL)  
         {  
                 return -1;  
         }  
86    
87          bbsnet_menu.p_menu_pool = calloc(1, sizeof(MENU));          bbsnet_menu.p_menu_pool = calloc(1, sizeof(MENU));
88          if (bbsnet_menu.p_menu_pool == NULL)          if (bbsnet_menu.p_menu_pool == NULL)
89          {          {
90                  log_error("calloc(p_menu_pool) error\n");                  log_error("calloc(p_menu_pool) error");
91                  return -3;                  return -1;
92          }          }
93          bbsnet_menu.menu_count = 1;          bbsnet_menu.menu_count = 1;
94    
95          bbsnet_menu.p_menu_item_pool = calloc(MAXSTATION, sizeof(MENU_ITEM));          bbsnet_menu.p_menu_item_pool = calloc(MAXSTATION, sizeof(MENU_ITEM));
96          if (bbsnet_menu.p_menu_item_pool == NULL)          if (bbsnet_menu.p_menu_item_pool == NULL)
97          {          {
98                  log_error("calloc(p_menu_item_pool) error\n");                  log_error("calloc(p_menu_item_pool) error");
99                  return -3;                  unload_bbsnet_conf();
100                    return -1;
101          }          }
102          bbsnet_menu.menu_item_count = MAXSTATION;          bbsnet_menu.menu_item_count = MAXSTATION;
103    
# Line 105  static int load_bbsnet_conf(const char * Line 108  static int load_bbsnet_conf(const char *
108          p_menu->title.show = 0;          p_menu->title.show = 0;
109          p_menu->screen_show = 0;          p_menu->screen_show = 0;
110    
111            fp = fopen(file_config, "r");
112            if (fp == NULL)
113            {
114                    unload_bbsnet_conf();
115                    return -2;
116            }
117    
118          menu_item_id = 0;          menu_item_id = 0;
119          while (fgets(line, sizeof(line), fp) && menu_item_id < MAXSTATION)          while (fgets(line, sizeof(line), fp) && menu_item_id < MAXSTATION)
120          {          {
# Line 116  static int load_bbsnet_conf(const char * Line 126  static int load_bbsnet_conf(const char *
126                  t6 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);                  t6 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
127    
128                  if (t1 == NULL || t2 == NULL || t3 == NULL || t4 == NULL ||                  if (t1 == NULL || t2 == NULL || t3 == NULL || t4 == NULL ||
129                          t5 == NULL || t6 == NULL || line[0] == '#' || line[0] == '*')                          t5 == NULL || t6 == NULL || t1[0] == '#')
130                  {                  {
131                          continue;                          continue;
132                  }                  }
# Line 127  static int load_bbsnet_conf(const char * Line 137  static int load_bbsnet_conf(const char *
137                  bbsnet_conf[menu_item_id].org_name[sizeof(bbsnet_conf[menu_item_id].org_name) - 1] = '\0';                  bbsnet_conf[menu_item_id].org_name[sizeof(bbsnet_conf[menu_item_id].org_name) - 1] = '\0';
138                  strncpy(bbsnet_conf[menu_item_id].host_name, t3, sizeof(bbsnet_conf[menu_item_id].host_name) - 1);                  strncpy(bbsnet_conf[menu_item_id].host_name, t3, sizeof(bbsnet_conf[menu_item_id].host_name) - 1);
139                  bbsnet_conf[menu_item_id].host_name[sizeof(bbsnet_conf[menu_item_id].host_name) - 1] = '\0';                  bbsnet_conf[menu_item_id].host_name[sizeof(bbsnet_conf[menu_item_id].host_name) - 1] = '\0';
140                  bbsnet_conf[menu_item_id].port = (in_port_t)(t4 ? atoi(t4) : 23);                  port = strtol(t4, &endptr, 10);
141                    if (*endptr != '\0' || port <= 0 || port > 65535)
142                    {
143                            log_error("Invalid port value %ld of menu item %d", port, menu_item_id);
144                            fclose(fp);
145                            unload_bbsnet_conf();
146                            return -3;
147                    }
148                    strncpy(bbsnet_conf[menu_item_id].port, t4, sizeof(bbsnet_conf[menu_item_id].port) - 1);
149                    bbsnet_conf[menu_item_id].port[sizeof(bbsnet_conf[menu_item_id].port) - 1] = '\0';
150                  bbsnet_conf[menu_item_id].use_ssh = (toupper(t5[0]) == 'Y');                  bbsnet_conf[menu_item_id].use_ssh = (toupper(t5[0]) == 'Y');
151                  strncpy(bbsnet_conf[menu_item_id].charset, t6, sizeof(bbsnet_conf[menu_item_id].charset) - 1);                  strncpy(bbsnet_conf[menu_item_id].charset, t6, sizeof(bbsnet_conf[menu_item_id].charset) - 1);
152                  bbsnet_conf[menu_item_id].charset[sizeof(bbsnet_conf[menu_item_id].charset) - 1] = '\0';                  bbsnet_conf[menu_item_id].charset[sizeof(bbsnet_conf[menu_item_id].charset) - 1] = '\0';
# Line 135  static int load_bbsnet_conf(const char * Line 154  static int load_bbsnet_conf(const char *
154                  p_menu_item = get_menu_item_by_id(&bbsnet_menu, menu_item_id);                  p_menu_item = get_menu_item_by_id(&bbsnet_menu, menu_item_id);
155                  if (p_menu_item == NULL)                  if (p_menu_item == NULL)
156                  {                  {
157                          log_error("get_menu_item_by_id(%d) return NULL pointer\n", menu_item_id);                          log_error("get_menu_item_by_id(%d) return NULL pointer", menu_item_id);
158                          return -1;                          fclose(fp);
159                            unload_bbsnet_conf();
160                            return -3;
161                  }                  }
162    
163                  p_menu_item->row = (int16_t)(2 + menu_item_id / STATION_PER_LINE);                  p_menu_item->row = (int16_t)(2 + menu_item_id / STATION_PER_LINE);
# Line 146  static int load_bbsnet_conf(const char * Line 167  static int load_bbsnet_conf(const char *
167                  p_menu_item->priv = 0;                  p_menu_item->priv = 0;
168                  p_menu_item->level = 0;                  p_menu_item->level = 0;
169                  p_menu_item->name[0] =                  p_menu_item->name[0] =
170                          (char)(menu_item_id < MAXSTATION / 2 ? 'A' + menu_item_id : 'a' + menu_item_id);                          (char)(menu_item_id < MAXSTATION / 2 ? 'A' + menu_item_id : 'a' + menu_item_id - MAXSTATION / 2);
171                  p_menu_item->name[1] = '\0';                  p_menu_item->name[1] = '\0';
172                  snprintf(p_menu_item->text, sizeof(p_menu_item->text), "%c. %s",                  snprintf(p_menu_item->text, sizeof(p_menu_item->text), "\033[1;36m%c.\033[m %s",
173                                   p_menu_item->name[0], bbsnet_conf[menu_item_id].site_name);                                   p_menu_item->name[0], bbsnet_conf[menu_item_id].site_name);
174    
175                  p_menu->items[p_menu->item_count] = menu_item_id;                  p_menu->items[p_menu->item_count] = menu_item_id;
# Line 171  static void unload_bbsnet_conf(void) Line 192  static void unload_bbsnet_conf(void)
192          bbsnet_menu.menu_count = 0;          bbsnet_menu.menu_count = 0;
193          bbsnet_menu.menu_item_count = 0;          bbsnet_menu.menu_item_count = 0;
194    
195          free(bbsnet_menu.p_menu_pool);          if (bbsnet_menu.p_menu_pool)
196          bbsnet_menu.p_menu_pool = NULL;          {
197          free(bbsnet_menu.p_menu_item_pool);                  free(bbsnet_menu.p_menu_pool);
198          bbsnet_menu.p_menu_item_pool = NULL;                  bbsnet_menu.p_menu_pool = NULL;
199            }
200    
201            if (bbsnet_menu.p_menu_item_pool)
202            {
203                    free(bbsnet_menu.p_menu_item_pool);
204                    bbsnet_menu.p_menu_item_pool = NULL;
205            }
206  }  }
207    
208  static void process_bar(int n, int len)  static void progress_bar(int percent, int len)
209  {  {
210            char line[LINE_BUFFER_LEN];
211          char buf[LINE_BUFFER_LEN];          char buf[LINE_BUFFER_LEN];
212          char buf2[LINE_BUFFER_LEN];          int pos;
213    
214          if (len > LINE_BUFFER_LEN)          if (len < 4)
215            {
216                    len = 4;
217            }
218            else if (len + 2 > LINE_BUFFER_LEN)
219          {          {
220                  len = LINE_BUFFER_LEN - 1;                  len = LINE_BUFFER_LEN - 3;
221          }          }
222          if (n < 0)          if (percent < 0)
223          {          {
224                  n = 0;                  percent = 0;
225          }          }
226          else if (n > len)          else if (percent > 100)
227          {          {
228                  n = len;                  percent = 100;
229          }          }
230    
231            pos = len * percent / 100;
232    
233            line[0] = ' ';
234            for (int i = 1; i <= len; i++)
235            {
236                    line[i] = '-';
237            }
238            line[len + 1] = ' ';
239            line[len + 2] = '\0';
240    
241            snprintf(buf, sizeof(buf), "%*s%3d%%%*s",
242                             (len - 4) / 2, "", percent, (len - 4 + 1) / 2, "");
243    
244          moveto(4, 1);          moveto(4, 1);
245          prints(" ------------------------------ \r\n");          prints("%s\r\n", line);
246          snprintf(buf, sizeof(buf), "            %3d%%              ", n * 100 / len);          prints("|\033[46m%.*s\033[44m%s\033[m|\r\n", pos, buf, buf + pos);
247          memcpy(buf2, buf, (size_t)n);          prints("%s\r\n", line);
         buf2[n] = '\0';  
         prints("|\033[46m%s\033[44m%s\033[m|\r\n", buf2, buf + n);  
         prints(" ------------------------------ \r\n");  
248          iflush();          iflush();
249  }  }
250    
251    static int connect_timeout(struct timespec *p_ts_begin, struct timespec *p_ts_now,
252                                                       int total_time_ms, int *p_progress_last, int bar_len)
253    {
254            long duration_ms;
255            int progress;
256    
257            if (clock_gettime(CLOCK_REALTIME, p_ts_now) == -1)
258            {
259                    log_error("clock_gettime() error (%d)", errno);
260                    return -1;
261            }
262    
263            duration_ms = (p_ts_now->tv_sec - p_ts_begin->tv_sec) * 1000 +
264                                      (p_ts_now->tv_nsec - p_ts_begin->tv_nsec) / 1000 / 1000;
265    
266            if (duration_ms >= total_time_ms)
267            {
268                    progress_bar(100, bar_len);
269                    prints("\033[1;31m连接超时！\033[m\r\n");
270                    press_any_key();
271    
272                    return 1;
273            }
274    
275            progress = (int)(duration_ms * 100 / total_time_ms) + 1;
276    
277            if (progress != *p_progress_last)
278            {
279                    *p_progress_last = progress;
280                    progress_bar(progress, bar_len);
281            }
282    
283            return 0;
284    }
285    
286  static int bbsnet_connect(int n)  static int bbsnet_connect(int n)
287  {  {
288          int sock, ret, loop, error;          int sock = -1;
289            int ret = 0;
290            int loop;
291            int error;
292          int sock_connected = 0;          int sock_connected = 0;
293          int flags_sock;          int ssh_connected = 0;
294          int flags_stdin;          int flags_sock = -1;
295          int flags_stdout;          int flags_stdin = -1;
296          int len;          int flags_stdout = -1;
297          struct sockaddr_in sin;          struct sockaddr_in sin;
298          char input_buf[LINE_BUFFER_LEN];          char input_buf[LINE_BUFFER_LEN];
299          char output_buf[LINE_BUFFER_LEN];          char output_buf[LINE_BUFFER_LEN];
# Line 226  static int bbsnet_connect(int n) Line 307  static int bbsnet_connect(int n)
307          int output_conv_len = 0;          int output_conv_len = 0;
308          int input_conv_offset = 0;          int input_conv_offset = 0;
309          int output_conv_offset = 0;          int output_conv_offset = 0;
310          iconv_t input_cd = NULL;          iconv_t input_cd = (iconv_t)(-1);
311          iconv_t output_cd = NULL;          iconv_t output_cd = (iconv_t)(-1);
312          char tocode[32];          char tocode[CHARSET_MAX_LEN + 20];
313    
314  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
315          struct epoll_event ev, events[MAX_EVENTS];          struct epoll_event ev, events[MAX_EVENTS];
316          int epollfd;          int epollfd = -1;
317  #else  #else
318          struct pollfd pfds[3];          struct pollfd pfds[3];
319  #endif  #endif
# Line 242  static int bbsnet_connect(int n) Line 323  static int bbsnet_connect(int n)
323          int stdout_write_wait = 0;          int stdout_write_wait = 0;
324          int sock_read_wait = 0;          int sock_read_wait = 0;
325          int sock_write_wait = 0;          int sock_write_wait = 0;
326          struct hostent *p_host = NULL;          struct addrinfo hints, *res = NULL;
327          int tos;          int tos;
328          char remote_addr[IP_ADDR_LEN];          char remote_addr[INET_ADDRSTRLEN];
329          int remote_port;          int remote_port;
330          char local_addr[IP_ADDR_LEN];          char local_addr[INET_ADDRSTRLEN];
331          int local_port;          int local_port;
332          socklen_t sock_len;          socklen_t sock_len;
333          time_t t_used = time(NULL);          time_t t_begin, t_used;
334          struct tm *tm_used;          struct timespec ts_begin, ts_now;
335            int progress_last = 0;
336          int ch;          int ch;
337          char remote_user[USERNAME_MAX_LEN + 1];          char remote_user[USERNAME_MAX_LEN + 1];
338          char remote_pass[PASSWORD_MAX_LEN + 1];          char remote_pass[PASSWORD_MAX_LEN + 1];
339          ssh_session session = NULL;          ssh_session outbound_session = NULL;
340          ssh_channel channel = NULL;          ssh_channel outbound_channel = NULL;
341          int ssh_process_config = 0;          int ssh_process_config = 0;
342          int ssh_log_level = SSH_LOG_PROTOCOL;          int ssh_log_level = SSH_LOG_NOLOG;
343    
344          if (user_online_update("BBS_NET") < 0)          if (user_online_update("BBS_NET") < 0)
345          {          {
346                  log_error("user_online_update(BBS_NET) error\n");                  log_error("user_online_update(BBS_NET) error");
347          }          }
348    
349          if (bbsnet_conf[n].use_ssh)          if (bbsnet_conf[n].use_ssh)
# Line 308  static int bbsnet_connect(int n) Line 390  static int bbsnet_connect(int n)
390          moveto(1, 1);          moveto(1, 1);
391          prints("\033[1;32m正在测试往 %s (%s) 的连接，请稍候... \033[m\r\n",          prints("\033[1;32m正在测试往 %s (%s) 的连接，请稍候... \033[m\r\n",
392                     bbsnet_conf[n].site_name, bbsnet_conf[n].host_name);                     bbsnet_conf[n].site_name, bbsnet_conf[n].host_name);
393          iflush();          prints("\033[1;32m连接进行中，按\033[1;33mCtrl+C\033[1;32m中断。\033[m\r\n");
394            progress_bar(0, PROGRESS_BAR_LEN);
395    
396          p_host = gethostbyname(bbsnet_conf[n].host_name);          if (clock_gettime(CLOCK_REALTIME, &ts_begin) == -1)
397            {
398                    log_error("clock_gettime() error (%d)", errno);
399                    ret = -1;
400                    goto cleanup;
401            }
402            ts_now = ts_begin;
403    
404            memset(&hints, 0, sizeof(hints));
405            hints.ai_family = AF_INET;
406            hints.ai_socktype = SOCK_STREAM;
407            hints.ai_protocol = IPPROTO_TCP;
408    
409          if (p_host == NULL)          if ((ret = getaddrinfo(BBS_address, NULL, &hints, &res)) != 0)
410          {          {
411                  prints("\033[1;31m查找主机名失败！\033[m\r\n");                  log_error("getaddrinfo() error (%d): %s", ret, gai_strerror(ret));
412                  press_any_key();                  ret = -1;
413                  return -1;                  goto cleanup;
414          }          }
415    
416          sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);          if (inet_ntop(AF_INET, &(((struct sockaddr_in *)res->ai_addr)->sin_addr), local_addr, sizeof(local_addr)) == NULL)
417            {
418                    log_error("inet_ntop() error (%d)", errno);
419                    ret = -1;
420                    goto cleanup;
421            }
422            local_port = ntohs(((struct sockaddr_in *)res->ai_addr)->sin_port);
423    
424            sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
425          if (sock < 0)          if (sock < 0)
426          {          {
427                  prints("\033[1;31m无法创建socket！\033[m\r\n");                  log_error("socket() error (%d)", errno);
428                  press_any_key();                  ret = -1;
429                  return -1;                  goto cleanup;
430          }          }
431    
432          sin.sin_family = AF_INET;          if (bind(sock, res->ai_addr, res->ai_addrlen) < 0)
         sin.sin_addr.s_addr = (BBS_address[0] != '\0' ? inet_addr(BBS_address) : INADDR_ANY);  
         sin.sin_port = 0;  
   
         if (bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0)  
433          {          {
434                  log_error("Bind address %s:%u failed (%d)\n",                  log_error("bind(%s:%u) error (%d)", local_addr, local_port, errno);
435                                    inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), errno);                  ret = -1;
436                  return -2;                  goto cleanup;
437          }          }
438    
439          memset(&sin, 0, sizeof(sin));          freeaddrinfo(res);
440          sin.sin_family = AF_INET;          res = NULL;
441          sin.sin_addr = *(struct in_addr *)p_host->h_addr_list[0];  
442          sin.sin_port = htons(bbsnet_conf[n].port);          memset(&hints, 0, sizeof(hints));
443            hints.ai_family = AF_INET;
444          strncpy(remote_addr, inet_ntoa(sin.sin_addr), sizeof(remote_addr) - 1);          hints.ai_flags = AI_NUMERICSERV;
445          remote_addr[sizeof(remote_addr) - 1] = '\0';          hints.ai_socktype = SOCK_STREAM;
446          remote_port = ntohs(sin.sin_port);          hints.ai_protocol = IPPROTO_TCP;
447    
448            if ((ret = getaddrinfo(bbsnet_conf[n].host_name, bbsnet_conf[n].port, &hints, &res)) != 0)
449            {
450                    log_error("getaddrinfo() error (%d): %s", ret, gai_strerror(ret));
451                    prints("\033[1;31m查找主机名失败！\033[m\r\n");
452                    press_any_key();
453                    ret = -1;
454                    goto cleanup;
455            }
456    
457          prints("\033[1;32m穿梭进度条提示您当前已使用的时间，按\033[1;33mCtrl+C\033[1;32m中断。\033[m\r\n");          if (inet_ntop(AF_INET, &(((struct sockaddr_in *)res->ai_addr)->sin_addr), remote_addr, sizeof(remote_addr)) == NULL)
458          process_bar(0, MAX_PROCESS_BAR_LEN);          {
459                    log_error("inet_ntop() error (%d)", errno);
460                    ret = -1;
461                    goto cleanup;
462            }
463            remote_port = ntohs(((struct sockaddr_in *)res->ai_addr)->sin_port);
464    
465          // Set socket as non-blocking          // Set socket as non-blocking
466          flags_sock = fcntl(sock, F_GETFL, 0);          if ((flags_sock = fcntl(sock, F_GETFL, 0)) == -1)
467          fcntl(sock, F_SETFL, flags_sock | O_NONBLOCK);          {
468                    log_error("fcntl(F_GETFL) error (%d)", errno);
469                    ret = -1;
470                    goto cleanup;
471            }
472            if ((fcntl(sock, F_SETFL, flags_sock | O_NONBLOCK)) == -1)
473            {
474                    log_error("fcntl(F_SETFL) error (%d)", errno);
475                    ret = -1;
476                    goto cleanup;
477            }
478    
479          // Set STDIN/STDOUT as non-blocking          // Set STDIN/STDOUT as non-blocking
480          flags_stdin = fcntl(STDIN_FILENO, F_GETFL, 0);          if ((flags_stdin = fcntl(STDIN_FILENO, F_GETFL, 0)) == -1)
481          flags_stdout = fcntl(STDOUT_FILENO, F_GETFL, 0);          {
482          fcntl(STDIN_FILENO, F_SETFL, flags_stdin | O_NONBLOCK);                  log_error("fcntl(F_GETFL) error (%d)", errno);
483          fcntl(STDOUT_FILENO, F_SETFL, flags_stdout | O_NONBLOCK);                  ret = -1;
484                    goto cleanup;
485            }
486            if ((flags_stdout = fcntl(STDOUT_FILENO, F_GETFL, 0)) == -1)
487            {
488                    log_error("fcntl(F_GETFL) error (%d)", errno);
489                    ret = -1;
490                    goto cleanup;
491            }
492            if ((fcntl(STDIN_FILENO, F_SETFL, flags_stdin | O_NONBLOCK)) == -1)
493            {
494                    log_error("fcntl(F_SETFL) error (%d)", errno);
495                    ret = -1;
496                    goto cleanup;
497            }
498            if ((fcntl(STDOUT_FILENO, F_SETFL, flags_stdout | O_NONBLOCK)) == -1)
499            {
500                    log_error("fcntl(F_SETFL) error (%d)", errno);
501                    ret = -1;
502                    goto cleanup;
503            }
504    
505  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
506          epollfd = epoll_create1(0);          epollfd = epoll_create1(0);
507          if (epollfd < 0)          if (epollfd < 0)
508          {          {
509                  log_error("epoll_create1() error (%d)\n", errno);                  log_error("epoll_create1() error (%d)", errno);
510                  return -1;                  ret = -1;
511                    goto cleanup;
512          }          }
513    
514          ev.events = EPOLLOUT | EPOLLET;          ev.events = EPOLLOUT | EPOLLET;
515          ev.data.fd = sock;          ev.data.fd = sock;
516          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, sock, &ev) == -1)          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, sock, &ev) == -1)
517          {          {
518                  log_error("epoll_ctl(socket) error (%d)\n", errno);                  log_error("epoll_ctl(socket) error (%d)", errno);
519                    ret = -1;
520                  goto cleanup;                  goto cleanup;
521          }          }
522    
# Line 381  static int bbsnet_connect(int n) Line 524  static int bbsnet_connect(int n)
524          ev.data.fd = STDIN_FILENO;          ev.data.fd = STDIN_FILENO;
525          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDIN_FILENO, &ev) == -1)          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDIN_FILENO, &ev) == -1)
526          {          {
527                  log_error("epoll_ctl(STDIN_FILENO) error (%d)\n", errno);                  log_error("epoll_ctl(STDIN_FILENO) error (%d)", errno);
528                    ret = -1;
529                  goto cleanup;                  goto cleanup;
530          }          }
531  #endif  #endif
532    
533          while (!SYS_server_exit)          while (!SYS_server_exit &&
534                       !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
535          {          {
536                  if ((ret = connect(sock, (struct sockaddr *)&sin, sizeof(sin))) < 0)                  if ((ret = connect(sock, res->ai_addr, res->ai_addrlen)) == 0)
537                    {
538                            sock_connected = 1;
539                            break;
540                    }
541                    else if (ret < 0)
542                  {                  {
543                          if (errno == EAGAIN || errno == EALREADY || errno == EINPROGRESS)                          if (errno == EAGAIN || errno == EALREADY || errno == EINPROGRESS)
544                          {                          {
# Line 399  static int bbsnet_connect(int n) Line 549  static int bbsnet_connect(int n)
549                          }                          }
550                          else if (errno == EINTR)                          else if (errno == EINTR)
551                          {                          {
                                 continue;  
552                          }                          }
553                          else                          else
554                          {                          {
555                                  log_error("connect(socket) error (%d)\n", errno);                                  log_error("connect(socket) error (%d)", errno);
   
556                                  prints("\033[1;31m连接失败！\033[m\r\n");                                  prints("\033[1;31m连接失败！\033[m\r\n");
557                                  press_any_key();                                  press_any_key();
558                                    ret = -1;
559                                  goto cleanup;                                  goto cleanup;
560                          }                          }
561                  }                  }
562          }          }
563    
564          for (int j = 0; j < MAX_PROCESS_BAR_LEN && !sock_connected && !SYS_server_exit; j++)          while (!SYS_server_exit && !sock_connected &&
565                       !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
566          {          {
567  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
568                  nfds = epoll_wait(epollfd, events, MAX_EVENTS, 500); // 0.5 second                  nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second
569                  ret = nfds;                  ret = nfds;
570  #else  #else
571                  pfds[0].fd = sock;                  pfds[0].fd = sock;
# Line 424  static int bbsnet_connect(int n) Line 573  static int bbsnet_connect(int n)
573                  pfds[1].fd = STDIN_FILENO;                  pfds[1].fd = STDIN_FILENO;
574                  pfds[1].events = POLLIN;                  pfds[1].events = POLLIN;
575                  nfds = 2;                  nfds = 2;
576                  ret = poll(pfds, (nfds_t)nfds, 500); // 0.5 second                  ret = poll(pfds, (nfds_t)nfds, 100); // 0.1 second
577  #endif  #endif
578    
579                  if (ret < 0)                  if (ret < 0)
# Line 432  static int bbsnet_connect(int n) Line 581  static int bbsnet_connect(int n)
581                          if (errno != EINTR)                          if (errno != EINTR)
582                          {                          {
583  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
584                                  log_error("epoll_wait() error (%d)\n", errno);                                  log_error("epoll_wait() error (%d)", errno);
585  #else  #else
586                                  log_error("poll() error (%d)\n", errno);                                  log_error("poll() error (%d)", errno);
587  #endif  #endif
588                                  break;                                  break;
589                          }                          }
590                  }                  }
591                  else if (ret == 0) // timeout                  else if (ret == 0) // timeout
592                  {                  {
                         process_bar(j + 1, MAX_PROCESS_BAR_LEN);  
593                  }                  }
594                  else // ret > 0                  else // ret > 0
595                  {                  {
# Line 453  static int bbsnet_connect(int n) Line 601  static int bbsnet_connect(int n)
601                                  if (pfds[i].fd == sock && (pfds[i].revents & POLLOUT))                                  if (pfds[i].fd == sock && (pfds[i].revents & POLLOUT))
602  #endif  #endif
603                                  {                                  {
604                                          len = sizeof(error);                                          socklen_t len = sizeof(error);
605                                          if (getsockopt(sock, SOL_SOCKET, SO_ERROR, &error, (socklen_t *)&len) < 0)                                          if (getsockopt(sock, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
606                                          {                                          {
607                                                  log_error("getsockopt() error (%d) !\n", error);                                                  log_error("getsockopt() error (%d) !", errno);
608                                                    ret = -1;
609                                                  goto cleanup;                                                  goto cleanup;
610                                          }                                          }
611                                          if (error == 0)                                          if (error == 0)
612                                          {                                          {
613                                                  sock_connected = 1;                                                  sock_connected = 1;
614                                                    break;
615                                          }                                          }
616                                  }                                  }
617  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
# Line 476  static int bbsnet_connect(int n) Line 626  static int bbsnet_connect(int n)
626                                          } while (ch == 0);                                          } while (ch == 0);
627                                          if (ch == Ctrl('C') || ch == KEY_ESC)                                          if (ch == Ctrl('C') || ch == KEY_ESC)
628                                          {                                          {
629                                                    ret = 0;
630                                                  goto cleanup;                                                  goto cleanup;
631                                          }                                          }
632                                  }                                  }
633                          }                          }
634                  }                  }
635          }          }
         if (SYS_server_exit)  
         {  
                 goto cleanup;  
         }  
636          if (!sock_connected)          if (!sock_connected)
637          {          {
638                  prints("\033[1;31m连接失败！\033[m\r\n");                  ret = -1;
                 press_any_key();  
   
639                  goto cleanup;                  goto cleanup;
640          }          }
641    
642          tos = IPTOS_LOWDELAY;          tos = IPTOS_LOWDELAY;
643          if (setsockopt(sock, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)) < 0)          if (setsockopt(sock, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)) < 0)
644          {          {
645                  log_error("setsockopt IP_TOS=%d error (%d)\n", tos, errno);                  log_error("setsockopt IP_TOS=%d error (%d)", tos, errno);
646          }          }
647    
648          sock_len = sizeof(sin);          sock_len = sizeof(sin);
649          if (getsockname(sock, (struct sockaddr *)&sin, &sock_len) < 0)          if (getsockname(sock, (struct sockaddr *)&sin, &sock_len) < 0)
650          {          {
651                  log_error("getsockname() error: %d", errno);                  log_error("getsockname() error: %d", errno);
652                    ret = -1;
653                  goto cleanup;                  goto cleanup;
654          }          }
655    
656          strncpy(local_addr, inet_ntoa(sin.sin_addr), sizeof(local_addr) - 1);          if (inet_ntop(AF_INET, &(sin.sin_addr), local_addr, sizeof(local_addr)) == NULL)
657          local_addr[sizeof(local_addr) - 1] = '\0';          {
658                    log_error("inet_ntop() error (%d)", errno);
659                    ret = -1;
660                    goto cleanup;
661            }
662          local_port = ntohs(sin.sin_port);          local_port = ntohs(sin.sin_port);
663    
664          if (bbsnet_conf[n].use_ssh)          if (bbsnet_conf[n].use_ssh)
665          {          {
666                  session = ssh_new();                  outbound_session = ssh_new();
667                  if (session == NULL)                  if (outbound_session == NULL)
668                  {                  {
669                          log_error("ssh_new() error\n");                          log_error("ssh_new() error");
670                            ret = -1;
671                          goto cleanup;                          goto cleanup;
672                  }                  }
673    
674                  if (ssh_options_set(session, SSH_OPTIONS_FD, &sock) < 0 ||                  if (ssh_options_set(outbound_session, SSH_OPTIONS_FD, &sock) < 0 ||
675                          ssh_options_set(session, SSH_OPTIONS_PROCESS_CONFIG, &ssh_process_config) < 0 ||                          ssh_options_set(outbound_session, SSH_OPTIONS_PROCESS_CONFIG, &ssh_process_config) < 0 ||
676                          ssh_options_set(session, SSH_OPTIONS_KNOWNHOSTS, SSH_KNOWN_HOSTS_FILE) < 0 ||                          ssh_options_set(outbound_session, SSH_OPTIONS_KNOWNHOSTS, SSH_KNOWN_HOSTS_FILE) < 0 ||
677                          ssh_options_set(session, SSH_OPTIONS_HOST, bbsnet_conf[n].host_name) < 0 ||                          ssh_options_set(outbound_session, SSH_OPTIONS_HOST, bbsnet_conf[n].host_name) < 0 ||
678                          ssh_options_set(session, SSH_OPTIONS_USER, remote_user) < 0 ||                          ssh_options_set(outbound_session, SSH_OPTIONS_USER, remote_user) < 0 ||
679                          ssh_options_set(session, SSH_OPTIONS_HOSTKEYS, "+ssh-rsa") < 0 ||                          ssh_options_set(outbound_session, SSH_OPTIONS_HOSTKEYS, "+ssh-ed25519,ecdsa-sha2-nistp256,ssh-rsa") < 0 ||
680                          ssh_options_set(session, SSH_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0)                          ssh_options_set(outbound_session, SSH_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0)
681                  {                  {
682                          log_error("Error setting SSH options: %s\n", ssh_get_error(session));                          log_error("Error setting SSH options: %s", ssh_get_error(outbound_session));
683                            ret = -1;
684                          goto cleanup;                          goto cleanup;
685                  }                  }
686    
687                  ssh_set_blocking(session, 0);                  ssh_set_blocking(outbound_session, 0);
688    
689                  while (!SYS_server_exit)                  ssh_connected = 0;
690                    while (!SYS_server_exit &&
691                               !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
692                  {                  {
693                          ret = ssh_connect(session);                          ret = ssh_connect(outbound_session);
694                          if (ret == SSH_OK)                          if (ret == SSH_OK)
695                          {                          {
696                                    ssh_connected = 1;
697                                  break;                                  break;
698                          }                          }
699                          else if (ret == SSH_ERROR)                          else if (ret == SSH_AGAIN)
700                          {                          {
701                                  log_error("ssh_connect() error\n");                                  // log_debug("ssh_connect() error: SSH_AGAIN");
702                            }
703                            else // if (ret == SSH_ERROR)
704                            {
705                                    log_error("ssh_connect() error: SSH_ERROR");
706                                    ret = -1;
707                                  goto cleanup;                                  goto cleanup;
708                          }                          }
709                  }                  }
710                    if (!ssh_connected)
711                    {
712                            ret = -1;
713                            goto cleanup;
714                    }
715    
716                  ret = ssh_session_is_known_server(session);                  ret = ssh_session_is_known_server(outbound_session);
717                  switch (ret)                  switch (ret)
718                  {                  {
719                  case SSH_KNOWN_HOSTS_NOT_FOUND:                  case SSH_KNOWN_HOSTS_NOT_FOUND:
720                  case SSH_KNOWN_HOSTS_UNKNOWN:                  case SSH_KNOWN_HOSTS_UNKNOWN:
721                          if (ssh_session_update_known_hosts(session) != SSH_OK)                          if (ssh_session_update_known_hosts(outbound_session) != SSH_OK)
722                          {                          {
723                                  log_error("ssh_session_update_known_hosts(%s) error\n", bbsnet_conf[n].host_name);                                  log_error("ssh_session_update_known_hosts(%s) error", bbsnet_conf[n].host_name);
724                                  prints("\033[1;31m无法添加服务器证书\033[m");                                  prints("\033[1;31m无法添加服务器证书\033[m\r\n");
725                                  press_any_key();                                  press_any_key();
726                                    ret = -1;
727                                  goto cleanup;                                  goto cleanup;
728                          }                          }
729                          log_common("SSH key of (%s) is added into %s\n", bbsnet_conf[n].host_name, SSH_KNOWN_HOSTS_FILE);                          log_common("SSH key of (%s) is added into %s", bbsnet_conf[n].host_name, SSH_KNOWN_HOSTS_FILE);
730                  case SSH_KNOWN_HOSTS_OK:                  case SSH_KNOWN_HOSTS_OK:
731                          break;                          break;
732                  case SSH_KNOWN_HOSTS_CHANGED:                  case SSH_KNOWN_HOSTS_CHANGED:
733                  case SSH_KNOWN_HOSTS_OTHER:                  case SSH_KNOWN_HOSTS_OTHER:
734                          log_error("ssh_session_is_known_server(%s) error: %d\n", bbsnet_conf[n].host_name, ret);                          log_error("ssh_session_is_known_server(%s) error: %d", bbsnet_conf[n].host_name, ret);
735                          prints("\033[1;31m服务器证书已变更\033[m");                          prints("\033[1;31m服务器证书已变更\033[m\r\n");
736                          press_any_key();                          press_any_key();
737                            ret = -1;
738                          goto cleanup;                          goto cleanup;
739                  }                  }
740    
741                  for (int i = 0; !SYS_server_exit;)                  ssh_connected = 0;
742                    while (!SYS_server_exit &&
743                               !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
744                  {                  {
745                          ret = ssh_userauth_password(session, NULL, remote_pass);                          ret = ssh_userauth_password(outbound_session, NULL, remote_pass);
746                          if (ret == SSH_AUTH_SUCCESS)                          if (ret == SSH_AUTH_SUCCESS)
747                          {                          {
748                                    ssh_connected = 1;
749                                  break;                                  break;
750                          }                          }
751                          else if (ret == SSH_AUTH_AGAIN)                          else if (ret == SSH_AUTH_AGAIN)
752                          {                          {
753  #ifdef _DEBUG                                  // log_debug("ssh_userauth_password() error: SSH_AUTH_AGAIN");
                                 log_error("ssh_userauth_password() error: SSH_AUTH_AGAIN\n");  
 #endif  
754                          }                          }
755                          else if (ret == SSH_AUTH_ERROR)                          else if (ret == SSH_AUTH_ERROR)
756                          {                          {
757                                  log_error("ssh_userauth_password() error: %d\n", ret);                                  log_error("ssh_userauth_password() error: SSH_AUTH_ERROR");
758                                    ret = -1;
759                                  goto cleanup;                                  goto cleanup;
760                          }                          }
761                          else // if (ret == SSH_AUTH_DENIED)                          else // if (ret == SSH_AUTH_DENIED)
762                          {                          {
763                                    log_debug("ssh_userauth_password() error: SSH_AUTH_DENIED");
764                                  prints("\033[1;31m身份验证失败！\033[m\r\n");                                  prints("\033[1;31m身份验证失败！\033[m\r\n");
765                                  i++;                                  press_any_key();
766                                  if (i < BBS_login_retry_times)                                  ret = 0;
767                                  {                                  goto cleanup;
                                         prints("请输入密码: ");  
                                         iflush();  
                                         if (str_input(remote_pass, sizeof(remote_pass), NOECHO) < 0)  
                                         {  
                                                 goto cleanup;  
                                         }  
                                         if (remote_pass[0] == '\0')  
                                         {  
                                                 goto cleanup;  
                                         }  
                                 }  
                                 else  
                                 {  
                                         goto cleanup;  
                                 }  
768                          }                          }
769                  }                  }
770                    if (!ssh_connected)
771                    {
772                            ret = -1;
773                            goto cleanup;
774                    }
775    
776                  channel = ssh_channel_new(session);                  outbound_channel = ssh_channel_new(outbound_session);
777                  if (channel == NULL)                  if (outbound_channel == NULL)
778                  {                  {
779                          log_error("ssh_channel_new() error\n");                          log_error("ssh_channel_new() error");
780                            ret = -1;
781                          goto cleanup;                          goto cleanup;
782                  }                  }
783    
784                  while (!SYS_server_exit)                  ssh_connected = 0;
785                    while (!SYS_server_exit &&
786                               !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
787                  {                  {
788                          ret = ssh_channel_open_session(channel);                          ret = ssh_channel_open_session(outbound_channel);
789                          if (ret == SSH_OK)                          if (ret == SSH_OK)
790                          {                          {
791                                    ssh_connected = 1;
792                                  break;                                  break;
793                          }                          }
794                          else if (ret == SSH_ERROR)                          else if (ret == SSH_AGAIN)
795                          {                          {
796                                  log_error("ssh_channel_open_session() error\n");                                  // log_debug("ssh_channel_open_session() error: SSH_AGAIN");
797                            }
798                            else // if (ret == SSH_ERROR)
799                            {
800                                    log_error("ssh_channel_open_session() error: SSH_ERROR");
801                                    ret = -1;
802                                  goto cleanup;                                  goto cleanup;
803                          }                          }
804                  }                  }
805                    if (!ssh_connected)
806                    {
807                            ret = -1;
808                            goto cleanup;
809                    }
810    
811                  while (!SYS_server_exit)                  ssh_connected = 0;
812                    while (!SYS_server_exit &&
813                               !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
814                  {                  {
815                          ret = ssh_channel_request_pty(channel);                          ret = ssh_channel_request_pty(outbound_channel);
816                          if (ret == SSH_OK)                          if (ret == SSH_OK)
817                          {                          {
818                                    ssh_connected = 1;
819                                  break;                                  break;
820                          }                          }
821                          else if (ret == SSH_ERROR)                          else if (ret == SSH_AGAIN)
822                            {
823                                    // log_debug("ssh_channel_request_pty() error: SSH_AGAIN");
824                            }
825                            else // if (ret == SSH_ERROR)
826                          {                          {
827                                  log_error("ssh_channel_request_pty() error\n");                                  log_error("ssh_channel_request_pty() error: SSH_ERROR");
828                                    ret = -1;
829                                  goto cleanup;                                  goto cleanup;
830                          }                          }
831                  }                  }
832                    if (!ssh_connected)
833                    {
834                            ret = -1;
835                            goto cleanup;
836                    }
837    
838                  while (!SYS_server_exit)                  ssh_connected = 0;
839                    while (!SYS_server_exit &&
840                               !connect_timeout(&ts_begin, &ts_now, REMOTE_CONNECT_TIMEOUT * 1000, &progress_last, PROGRESS_BAR_LEN))
841                  {                  {
842                          ret = ssh_channel_request_shell(channel);                          ret = ssh_channel_request_shell(outbound_channel);
843                          if (ret == SSH_OK)                          if (ret == SSH_OK)
844                          {                          {
845                                    ssh_connected = 1;
846                                  break;                                  break;
847                          }                          }
848                          else if (ret == SSH_ERROR)                          else if (ret == SSH_AGAIN)
849                            {
850                                    // log_debug("ssh_channel_request_shell() error: SSH_AGAIN");
851                            }
852                            else // if (ret == SSH_ERROR)
853                          {                          {
854                                  log_error("ssh_channel_request_shell() error\n");                                  log_error("ssh_channel_request_shell() error: SSH_ERROR");
855                                    ret = -1;
856                                  goto cleanup;                                  goto cleanup;
857                          }                          }
858                  }                  }
859                    if (!ssh_connected)
860                    {
861                            ret = -1;
862                            goto cleanup;
863                    }
864          }          }
865    
866          prints("\033[1;31m连接成功！\033[m\r\n");          prints("\033[1;31m连接成功！\033[m\r\n");
867          iflush();          iflush();
868          log_common("BBSNET connect to %s:%d from %s:%d by [%s]\n",          log_common("BBSNET connect to %s:%d from %s:%d by [%s]",
869                             remote_addr, remote_port, local_addr, local_port, BBS_username);                             remote_addr, remote_port, local_addr, local_port, BBS_username);
870    
871          snprintf(tocode, sizeof(tocode), "%s%s", bbsnet_conf[n].charset,          snprintf(tocode, sizeof(tocode), "%s%s", bbsnet_conf[n].charset,
# Line 673  static int bbsnet_connect(int n) Line 873  static int bbsnet_connect(int n)
873          input_cd = iconv_open(tocode, stdio_charset);          input_cd = iconv_open(tocode, stdio_charset);
874          if (input_cd == (iconv_t)(-1))          if (input_cd == (iconv_t)(-1))
875          {          {
876                  log_error("iconv_open(%s->%s) error: %d\n", stdio_charset, tocode, errno);                  log_error("iconv_open(%s->%s) error: %d", stdio_charset, tocode, errno);
877                    ret = -1;
878                  goto cleanup;                  goto cleanup;
879          }          }
880    
# Line 682  static int bbsnet_connect(int n) Line 883  static int bbsnet_connect(int n)
883          output_cd = iconv_open(tocode, bbsnet_conf[n].charset);          output_cd = iconv_open(tocode, bbsnet_conf[n].charset);
884          if (output_cd == (iconv_t)(-1))          if (output_cd == (iconv_t)(-1))
885          {          {
886                  log_error("iconv_open(%s->%s) error: %d\n", bbsnet_conf[n].charset, tocode, errno);                  log_error("iconv_open(%s->%s) error: %d", bbsnet_conf[n].charset, tocode, errno);
887                  iconv_close(input_cd);                  ret = -1;
888                  goto cleanup;                  goto cleanup;
889          }          }
890    
# Line 692  static int bbsnet_connect(int n) Line 893  static int bbsnet_connect(int n)
893          ev.data.fd = sock;          ev.data.fd = sock;
894          if (epoll_ctl(epollfd, EPOLL_CTL_MOD, sock, &ev) == -1)          if (epoll_ctl(epollfd, EPOLL_CTL_MOD, sock, &ev) == -1)
895          {          {
896                  log_error("epoll_ctl(socket) error (%d)\n", errno);                  log_error("epoll_ctl(socket) error (%d)", errno);
897                    ret = -1;
898                  goto cleanup;                  goto cleanup;
899          }          }
900    
# Line 700  static int bbsnet_connect(int n) Line 902  static int bbsnet_connect(int n)
902          ev.data.fd = STDOUT_FILENO;          ev.data.fd = STDOUT_FILENO;
903          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDOUT_FILENO, &ev) == -1)          if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDOUT_FILENO, &ev) == -1)
904          {          {
905                  log_error("epoll_ctl(STDOUT_FILENO) error (%d)\n", errno);                  log_error("epoll_ctl(STDOUT_FILENO) error (%d)", errno);
906                    ret = -1;
907                  goto cleanup;                  goto cleanup;
908          }          }
909  #endif  #endif
910    
911          BBS_last_access_tm = t_used = time(NULL);          BBS_last_access_tm = t_begin = time(NULL);
912          loop = 1;          loop = 1;
913    
914          while (loop && !SYS_server_exit)          while (loop && !SYS_server_exit)
915          {          {
916                  if (SSH_v2 && ssh_channel_is_closed(SSH_channel))                  if (SSH_v2 && ssh_channel_is_closed(SSH_channel))
917                  {                  {
918                          log_error("SSH channel is closed\n");                          log_debug("SSH channel is closed");
919                          loop = 0;                          loop = 0;
920                          break;                          break;
921                  }                  }
922    
923                  if (bbsnet_conf[n].use_ssh && ssh_channel_is_closed(channel))                  if (bbsnet_conf[n].use_ssh && ssh_channel_is_closed(outbound_channel))
924                  {                  {
925                          log_error("Remote SSH channel is closed\n");                          log_debug("Outbound channel is closed");
926                          loop = 0;                          loop = 0;
927                          break;                          break;
928                  }                  }
# Line 743  static int bbsnet_connect(int n) Line 946  static int bbsnet_connect(int n)
946                          if (errno != EINTR)                          if (errno != EINTR)
947                          {                          {
948  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
949                                  log_error("epoll_wait() error (%d)\n", errno);                                  log_error("epoll_wait() error (%d)", errno);
950  #else  #else
951                                  log_error("poll() error (%d)\n", errno);                                  log_error("poll() error (%d)", errno);
952  #endif  #endif
953                                  break;                                  break;
954                          }                          }
# Line 755  static int bbsnet_connect(int n) Line 958  static int bbsnet_connect(int n)
958                  {                  {
959                          if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)                          if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)
960                          {                          {
961                                    log_debug("User input timeout");
962                                  break;                                  break;
963                          }                          }
964                  }                  }
# Line 762  static int bbsnet_connect(int n) Line 966  static int bbsnet_connect(int n)
966                  for (int i = 0; i < nfds; i++)                  for (int i = 0; i < nfds; i++)
967                  {                  {
968  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
969                          if (events[i].data.fd == STDIN_FILENO)                          if (events[i].events & (EPOLLHUP | EPOLLERR))
970    #else
971                            if (pfds[i].revents & (POLLHUP | POLLERR))
972    #endif
973                            {
974    #ifdef HAVE_SYS_EPOLL_H
975                                    log_debug("FD (%d) error events (%d)", events[i].data.fd, events[i].events);
976    #else
977                                    log_debug("FD (%d) error events (%d)", pfds[i].fd, pfds[i].revents);
978    #endif
979                                    loop = 0;
980                                    break;
981                            }
982    
983    #ifdef HAVE_SYS_EPOLL_H
984                            if (events[i].data.fd == STDIN_FILENO && (events[i].events & EPOLLIN))
985  #else  #else
986                          if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))                          if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))
987  #endif  #endif
# Line 796  static int bbsnet_connect(int n) Line 1015  static int bbsnet_connect(int n)
1015                          }                          }
1016    
1017  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
1018                          if (events[i].data.fd == STDOUT_FILENO)                          if (events[i].data.fd == STDOUT_FILENO && (events[i].events & EPOLLOUT))
1019  #else  #else
1020                          if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT))                          if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT))
1021  #endif  #endif
# Line 814  static int bbsnet_connect(int n) Line 1033  static int bbsnet_connect(int n)
1033                                          ret = ssh_channel_read_nonblocking(SSH_channel, input_buf + input_buf_len, sizeof(input_buf) - (uint32_t)input_buf_len, 0);                                          ret = ssh_channel_read_nonblocking(SSH_channel, input_buf + input_buf_len, sizeof(input_buf) - (uint32_t)input_buf_len, 0);
1034                                          if (ret == SSH_ERROR)                                          if (ret == SSH_ERROR)
1035                                          {                                          {
1036                                                  log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session));                                                  log_debug("ssh_channel_read_nonblocking() error: %s", ssh_get_error(SSH_session));
1037                                                  loop = 0;                                                  loop = 0;
1038                                                  break;                                                  break;
1039                                          }                                          }
# Line 829  static int bbsnet_connect(int n) Line 1048  static int bbsnet_connect(int n)
1048                                                  // Send NO-OP to remote server                                                  // Send NO-OP to remote server
1049                                                  input_buf[input_buf_len] = '\0';                                                  input_buf[input_buf_len] = '\0';
1050                                                  input_buf_len++;                                                  input_buf_len++;
                                                 BBS_last_access_tm = time(NULL);  
1051    
1052                                                    BBS_last_access_tm = time(NULL);
1053                                                  stdin_read_wait = 0;                                                  stdin_read_wait = 0;
1054                                                  break; // Check whether channel is still open                                                  break; // Check whether channel is still open
1055                                          }                                          }
# Line 852  static int bbsnet_connect(int n) Line 1071  static int bbsnet_connect(int n)
1071                                          }                                          }
1072                                          else                                          else
1073                                          {                                          {
1074                                                  log_error("read(STDIN) error (%d)\n", errno);                                                  log_error("read(STDIN) error (%d)", errno);
1075                                                  loop = 0;                                                  loop = 0;
1076                                                  break;                                                  break;
1077                                          }                                          }
1078                                  }                                  }
1079                                  else if (ret == 0) // broken pipe                                  else if (ret == 0) // broken pipe
1080                                  {                                  {
1081  #ifdef _DEBUG                                          log_debug("read(STDIN) EOF");
                                         log_error("read(STDIN) EOF\n");  
 #endif  
1082                                          stdin_read_wait = 0;                                          stdin_read_wait = 0;
1083                                          loop = 0;                                          loop = 0;
1084                                          break;                                          break;
# Line 874  static int bbsnet_connect(int n) Line 1091  static int bbsnet_connect(int n)
1091                                          // Refresh current action while user input                                          // Refresh current action while user input
1092                                          if (user_online_update("BBS_NET") < 0)                                          if (user_online_update("BBS_NET") < 0)
1093                                          {                                          {
1094                                                  log_error("user_online_update(BBS_NET) error\n");                                                  log_error("user_online_update(BBS_NET) error");
1095                                          }                                          }
1096    
1097                                          continue;                                          continue;
# Line 890  static int bbsnet_connect(int n) Line 1107  static int bbsnet_connect(int n)
1107  #ifdef _DEBUG  #ifdef _DEBUG
1108                                  for (int j = input_buf_offset; j < input_buf_len; j++)                                  for (int j = input_buf_offset; j < input_buf_len; j++)
1109                                  {                                  {
1110                                          log_error("Debug input: <--[%u]\n", (input_buf[j] + 256) % 256);                                          log_debug("input: <--[%u]", (unsigned char)(input_buf[j]));
1111                                  }                                  }
1112  #endif  #endif
1113    
1114                                  ret = io_buf_conv(input_cd, input_buf, &input_buf_len, &input_buf_offset, input_conv, sizeof(input_conv), &input_conv_len);                                  ret = io_buf_conv(input_cd, input_buf, &input_buf_len, &input_buf_offset, input_conv, sizeof(input_conv), &input_conv_len);
1115                                  if (ret < 0)                                  if (ret < 0)
1116                                  {                                  {
1117                                          log_error("io_buf_conv(input, %d, %d, %d) error\n", input_buf_len, input_buf_offset, input_conv_len);                                          log_error("io_buf_conv(input, %d, %d, %d) error", input_buf_len, input_buf_offset, input_conv_len);
1118                                          input_buf_len = input_buf_offset; // Discard invalid sequence                                          input_buf_len = input_buf_offset; // Discard invalid sequence
1119                                  }                                  }
1120    
# Line 905  static int bbsnet_connect(int n) Line 1122  static int bbsnet_connect(int n)
1122  #ifdef _DEBUG  #ifdef _DEBUG
1123                                  for (int j = input_conv_offset; j < input_conv_len; j++)                                  for (int j = input_conv_offset; j < input_conv_len; j++)
1124                                  {                                  {
1125                                          log_error("Debug input_conv: <--[%u]\n", (input_conv[j] + 256) % 256);                                          log_debug("input_conv: <--[%u]", (unsigned char)(input_conv[j]));
1126                                  }                                  }
1127  #endif  #endif
1128                          }                          }
# Line 914  static int bbsnet_connect(int n) Line 1131  static int bbsnet_connect(int n)
1131                          {                          {
1132                                  if (bbsnet_conf[n].use_ssh)                                  if (bbsnet_conf[n].use_ssh)
1133                                  {                                  {
1134                                          ret = ssh_channel_write(channel, input_conv + input_conv_offset, (uint32_t)(input_conv_len - input_conv_offset));                                          ret = ssh_channel_write(outbound_channel, input_conv + input_conv_offset, (uint32_t)(input_conv_len - input_conv_offset));
1135                                          if (ret == SSH_ERROR)                                          if (ret == SSH_ERROR)
1136                                          {                                          {
1137                                                  log_error("ssh_channel_write() error: %s\n", ssh_get_error(session));                                                  log_debug("ssh_channel_write() error: %s", ssh_get_error(outbound_session));
1138                                                  loop = 0;                                                  loop = 0;
1139                                                  break;                                                  break;
1140                                          }                                          }
# Line 939  static int bbsnet_connect(int n) Line 1156  static int bbsnet_connect(int n)
1156                                          }                                          }
1157                                          else                                          else
1158                                          {                                          {
1159                                                  log_error("write(socket) error (%d)\n", errno);                                                  log_debug("write(socket) error (%d)", errno);
1160                                                  loop = 0;                                                  loop = 0;
1161                                                  break;                                                  break;
1162                                          }                                          }
1163                                  }                                  }
1164                                  else if (ret == 0) // broken pipe                                  else if (ret == 0) // broken pipe
1165                                  {                                  {
1166  #ifdef _DEBUG                                          log_debug("write(socket) EOF");
                                         log_error("write(socket) EOF\n");  
 #endif  
1167                                          sock_write_wait = 0;                                          sock_write_wait = 0;
1168                                          loop = 0;                                          loop = 0;
1169                                          break;                                          break;
# Line 973  static int bbsnet_connect(int n) Line 1188  static int bbsnet_connect(int n)
1188                          {                          {
1189                                  if (bbsnet_conf[n].use_ssh)                                  if (bbsnet_conf[n].use_ssh)
1190                                  {                                  {
1191                                          ret = ssh_channel_read_nonblocking(channel, output_buf + output_buf_len,                                          ret = ssh_channel_read_nonblocking(outbound_channel, output_buf + output_buf_len,
1192                                                                                                             (uint32_t)(sizeof(output_buf) - (size_t)output_buf_len), 0);                                                                                                             (uint32_t)(sizeof(output_buf) - (size_t)output_buf_len), 0);
1193                                          if (ret == SSH_ERROR)                                          if (ret == SSH_ERROR)
1194                                          {                                          {
1195                                                  log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(session));                                                  log_debug("ssh_channel_read_nonblocking() error: %s", ssh_get_error(outbound_session));
1196                                                  loop = 0;                                                  loop = 0;
1197                                                  break;                                                  break;
1198                                          }                                          }
# Line 1010  static int bbsnet_connect(int n) Line 1225  static int bbsnet_connect(int n)
1225                                          }                                          }
1226                                          else                                          else
1227                                          {                                          {
1228                                                  log_error("read(socket) error (%d)\n", errno);                                                  log_debug("read(socket) error (%d)", errno);
1229                                                  loop = 0;                                                  loop = 0;
1230                                                  break;                                                  break;
1231                                          }                                          }
1232                                  }                                  }
1233                                  else if (ret == 0) // broken pipe                                  else if (ret == 0) // broken pipe
1234                                  {                                  {
1235  #ifdef _DEBUG                                          log_debug("read(socket) EOF");
                                         log_error("read(socket) EOF\n");  
 #endif  
1236                                          sock_read_wait = 0;                                          sock_read_wait = 0;
1237                                          loop = 0;                                          loop = 0;
1238                                          break;                                          break;
# Line 1039  static int bbsnet_connect(int n) Line 1252  static int bbsnet_connect(int n)
1252                                  ret = io_buf_conv(output_cd, output_buf, &output_buf_len, &output_buf_offset, output_conv, sizeof(output_conv), &output_conv_len);                                  ret = io_buf_conv(output_cd, output_buf, &output_buf_len, &output_buf_offset, output_conv, sizeof(output_conv), &output_conv_len);
1253                                  if (ret < 0)                                  if (ret < 0)
1254                                  {                                  {
1255                                          log_error("io_buf_conv(output, %d, %d, %d) error\n", output_buf_len, output_buf_offset, output_conv_len);                                          log_error("io_buf_conv(output, %d, %d, %d) error", output_buf_len, output_buf_offset, output_conv_len);
1256                                          output_buf_len = output_buf_offset; // Discard invalid sequence                                          output_buf_len = output_buf_offset; // Discard invalid sequence
1257                                  }                                  }
1258                          }                          }
# Line 1051  static int bbsnet_connect(int n) Line 1264  static int bbsnet_connect(int n)
1264                                          ret = ssh_channel_write(SSH_channel, output_conv + output_conv_offset, (uint32_t)(output_conv_len - output_conv_offset));                                          ret = ssh_channel_write(SSH_channel, output_conv + output_conv_offset, (uint32_t)(output_conv_len - output_conv_offset));
1265                                          if (ret == SSH_ERROR)                                          if (ret == SSH_ERROR)
1266                                          {                                          {
1267                                                  log_error("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session));                                                  log_debug("ssh_channel_write() error: %s", ssh_get_error(SSH_session));
1268                                                  loop = 0;                                                  loop = 0;
1269                                                  break;                                                  break;
1270                                          }                                          }
# Line 1073  static int bbsnet_connect(int n) Line 1286  static int bbsnet_connect(int n)
1286                                          }                                          }
1287                                          else                                          else
1288                                          {                                          {
1289                                                  log_error("write(STDOUT) error (%d)\n", errno);                                                  log_debug("write(STDOUT) error (%d)", errno);
1290                                                  loop = 0;                                                  loop = 0;
1291                                                  break;                                                  break;
1292                                          }                                          }
1293                                  }                                  }
1294                                  else if (ret == 0) // broken pipe                                  else if (ret == 0) // broken pipe
1295                                  {                                  {
1296  #ifdef _DEBUG                                          log_debug("write(STDOUT) EOF");
                                         log_error("write(STDOUT) EOF\n");  
 #endif  
1297                                          stdout_write_wait = 0;                                          stdout_write_wait = 0;
1298                                          loop = 0;                                          loop = 0;
1299                                          break;                                          break;
# Line 1102  static int bbsnet_connect(int n) Line 1313  static int bbsnet_connect(int n)
1313                  }                  }
1314          }          }
1315    
1316          iconv_close(input_cd);          ret = 1; // Normal disconnect
1317          iconv_close(output_cd);          BBS_last_access_tm = time(NULL);
1318            t_used = BBS_last_access_tm - t_begin;
1319            log_common("BBSNET disconnect, %ld days %ld hours %ld minutes %ld seconds used",
1320                               t_used / 86400, t_used % 86400 / 3600, t_used % 3600 / 60, t_used % 60);
1321    
1322  cleanup:  cleanup:
1323            // Clear sensitive data
1324            memset(remote_pass, 0, sizeof(remote_pass));
1325            memset(remote_user, 0, sizeof(remote_user));
1326    
1327            if (input_cd != (iconv_t)(-1) && iconv_close(input_cd) < 0)
1328            {
1329                    log_error("iconv_close(input) error (%d)", errno);
1330            }
1331            if (output_cd != (iconv_t)(-1) && iconv_close(output_cd) < 0)
1332            {
1333                    log_error("iconv_close(output) error (%d)", errno);
1334            }
1335    
1336  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
1337          if (close(epollfd) < 0)          if (epollfd != -1 && close(epollfd) < 0)
1338          {          {
1339                  log_error("close(epoll) error (%d)\n");                  log_error("close(epoll) error (%d)");
1340          }          }
1341  #endif  #endif
1342    
1343          if (bbsnet_conf[n].use_ssh)          if (bbsnet_conf[n].use_ssh)
1344          {          {
1345                  ssh_channel_free(channel);                  if (outbound_channel != NULL)
1346                  ssh_disconnect(session);                  {
1347                  ssh_free(session);                          ssh_channel_send_eof(outbound_channel);
1348                            ssh_channel_close(outbound_channel);
1349                            ssh_channel_free(outbound_channel);
1350                    }
1351                    if (outbound_session != NULL)
1352                    {
1353                            ssh_disconnect(outbound_session);
1354                            ssh_free(outbound_session);
1355                    }
1356          }          }
1357    
1358          // Restore STDIN/STDOUT flags          // Restore STDIN/STDOUT flags
1359          fcntl(STDIN_FILENO, F_SETFL, flags_stdin);          if (flags_stdin != -1 && fcntl(STDIN_FILENO, F_SETFL, flags_stdin) == -1)
         fcntl(STDOUT_FILENO, F_SETFL, flags_stdout);  
   
         if (close(sock) == -1)  
1360          {          {
1361                  log_error("Close socket failed\n");                  log_error("fcntl(F_SETFL) error (%d)", errno);
1362            }
1363            if (flags_stdout != -1 && fcntl(STDOUT_FILENO, F_SETFL, flags_stdout) == -1)
1364            {
1365                    log_error("fcntl(F_SETFL) error (%d)", errno);
1366          }          }
1367    
1368          t_used = time(NULL) - t_used;          if (sock != -1)
1369          tm_used = gmtime(&t_used);          {
1370                    close(sock);
1371          log_common("BBSNET disconnect, %d days %d hours %d minutes %d seconds used\n",          }
                            tm_used->tm_mday - 1, tm_used->tm_hour, tm_used->tm_min,  
                            tm_used->tm_sec);  
1372    
1373          BBS_last_access_tm = time(NULL);          if (res)
1374            {
1375                    freeaddrinfo(res);
1376            }
1377    
1378          return 0;          return ret;
1379  }  }
1380    
1381  static int bbsnet_refresh()  static int bbsnet_refresh()
# Line 1172  static int bbsnet_selchange() Line 1409  static int bbsnet_selchange()
1409    
1410          moveto(20, 1);          moveto(20, 1);
1411          clrtoeol();          clrtoeol();
1412          prints("|\033[1m单位: \033[1;33m%s\033[m%*s  站名: \033[1;33m%s\033[m",          prints("|\033[1m单位: \033[1;33m%s\033[m%*s  站名: \033[1;33m%s\033[m%*s  类型: \033[1;33m%s\033[m",
1413                     bbsnet_conf[i].org_name, 20 - str_length(bbsnet_conf[i].org_name, 1), "", bbsnet_conf[i].site_name);                     bbsnet_conf[i].org_name, 20 - str_length(bbsnet_conf[i].org_name, 1), "",
1414                       bbsnet_conf[i].site_name, 20 - str_length(bbsnet_conf[i].site_name, 1), "",
1415                       (bbsnet_conf[i].use_ssh ? "SSH" : "Telnet"));
1416          moveto(20, 80);          moveto(20, 80);
1417          prints("|");          prints("|");
1418          moveto(21, 1);          moveto(21, 1);
1419          clrtoeol();          clrtoeol();
1420          prints("|\033[1m连往: \033[1;33m%-20s\033[m  端口: \033[1;33m%-5d\033[m            类型: \033[1;33m%s\033[m",          prints("|\033[1m连往: \033[1;33m%-20s\033[m  端口: \033[1;33m%-5s\033[m                 编码: \033[1;33m%s\033[m",
1421                     bbsnet_conf[i].host_name, bbsnet_conf[i].port, (bbsnet_conf[i].use_ssh ? "SSH" : "Telnet"));                     bbsnet_conf[i].host_name, bbsnet_conf[i].port, bbsnet_conf[i].charset);
1422          moveto(21, 80);          moveto(21, 80);
1423          prints("|");          prints("|");
1424          iflush();          iflush();
# Line 1189  static int bbsnet_selchange() Line 1428  static int bbsnet_selchange()
1428    
1429  int bbs_net()  int bbs_net()
1430  {  {
1431          int ch, i;          int ch;
1432    
1433          load_bbsnet_conf(CONF_BBSNET);          if (load_bbsnet_conf(CONF_BBSNET) < 0)
1434            {
1435                    clearscr();
1436                    moveto(1, 1);
1437                    prints("加载穿梭配置失败！");
1438                    press_any_key();
1439                    return -1;
1440            }
1441    
1442          bbsnet_refresh();          bbsnet_refresh();
1443          display_menu(&bbsnet_menu);          display_menu(&bbsnet_menu);
# Line 1209  int bbs_net() Line 1455  int bbs_net()
1455                  switch (ch)                  switch (ch)
1456                  {                  {
1457                  case KEY_NULL: // broken pipe                  case KEY_NULL: // broken pipe
1458                          log_error("KEY_NULL\n");                          log_debug("KEY_NULL");
1459                          goto cleanup;                          goto cleanup;
1460                  case KEY_TIMEOUT:                  case KEY_TIMEOUT:
1461                          if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)                          if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)
1462                          {                          {
1463                                  log_error("User input timeout\n");                                  log_debug("User input timeout");
1464                                  goto cleanup;                                  goto cleanup;
1465                          }                          }
1466                          continue;                          continue;
# Line 1222  int bbs_net() Line 1468  int bbs_net()
1468                  case Ctrl('C'): // user cancel                  case Ctrl('C'): // user cancel
1469                          goto cleanup;                          goto cleanup;
1470                  case CR:                  case CR:
1471                          bbsnet_connect(bbsnet_menu.menu_item_pos[0]);                          if (bbsnet_connect(bbsnet_menu.menu_item_pos[0]) < 0)
1472                            {
1473                                    log_debug("bbsnet_connect() error");
1474                            }
1475                          // Force cleanup anything remaining in the output buffer                          // Force cleanup anything remaining in the output buffer
1476                          clearscr();                          clearscr();
1477                          iflush();                          iflush();
# Line 1232  int bbs_net() Line 1481  int bbs_net()
1481                          bbsnet_selchange();                          bbsnet_selchange();
1482                          break;                          break;
1483                  case KEY_UP:                  case KEY_UP:
1484                          for (i = 0; i < STATION_PER_LINE; i++)                          for (int i = 0; i < STATION_PER_LINE; i++)
1485                          {                          {
1486                                  menu_control(&bbsnet_menu, KEY_UP);                                  menu_control(&bbsnet_menu, KEY_UP);
1487                          }                          }
1488                          bbsnet_selchange();                          bbsnet_selchange();
1489                          break;                          break;
1490                  case KEY_DOWN:                  case KEY_DOWN:
1491                          for (i = 0; i < STATION_PER_LINE; i++)                          for (int i = 0; i < STATION_PER_LINE; i++)
1492                          {                          {
1493                                  menu_control(&bbsnet_menu, KEY_DOWN);                                  menu_control(&bbsnet_menu, KEY_DOWN);
1494                          }                          }

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1