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

Contents of /lbbs/src/bbs_net.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.84 - (show annotations)
Sun Nov 30 10:05:16 2025 UTC (3 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.83: +268 -32 lines
Content type: text/x-csrc
Support SSH connect in bbs_net

1 /* SPDX-License-Identifier: GPL-3.0-or-later */
2 /*
3 * bbs_net
4 * - user interactive feature of site shuttle
5 *
6 * Copyright (C) 2004-2025 Leaflet <leaflet@leafok.com>
7 */
8
9 #ifdef HAVE_CONFIG_H
10 #include "config.h"
11 #endif
12
13 #include "bbs.h"
14 #include "bbs_net.h"
15 #include "common.h"
16 #include "io.h"
17 #include "log.h"
18 #include "login.h"
19 #include "menu.h"
20 #include "screen.h"
21 #include "str_process.h"
22 #include <errno.h>
23 #include <fcntl.h>
24 #include <netdb.h>
25 #include <stdarg.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <time.h>
30 #include <unistd.h>
31 #include <arpa/inet.h>
32 #include <ctype.h>
33 #include <libssh/libssh.h>
34 #include <libssh/server.h>
35 #include <libssh/callbacks.h>
36 #include <netinet/in.h>
37 #include <netinet/ip.h>
38 #include <sys/select.h>
39 #include <sys/ioctl.h>
40 #include <sys/socket.h>
41
42 #ifdef HAVE_SYS_EPOLL_H
43 #include <sys/epoll.h>
44 #else
45 #include <poll.h>
46 #endif
47
48 static const char MENU_CONF_DELIM[] = " \t\r\n";
49
50 enum _bbs_net_constant_t
51 {
52 MAX_PROCESS_BAR_LEN = 30,
53 MAXSTATION = 26 * 2,
54 STATION_PER_LINE = 4,
55 USERNAME_MAX_LEN = 20,
56 PASSWORD_MAX_LEN = 20,
57 };
58
59 struct _bbsnet_conf
60 {
61 char host1[20];
62 char host2[40];
63 char ip[40];
64 in_port_t port;
65 int8_t use_ssh;
66 char charset[CHARSET_MAX_LEN + 1];
67 } bbsnet_conf[MAXSTATION];
68
69 static MENU_SET bbsnet_menu;
70
71 static int load_bbsnet_conf(const char *file_config)
72 {
73 FILE *fp;
74 MENU *p_menu;
75 MENU_ITEM *p_menu_item;
76 MENU_ITEM_ID menu_item_id;
77 char t[256], *t1, *t2, *t3, *t4, *t5, *t6, *saveptr;
78
79 fp = fopen(file_config, "r");
80 if (fp == NULL)
81 {
82 return -1;
83 }
84
85 bbsnet_menu.p_menu_pool = calloc(1, sizeof(MENU));
86 if (bbsnet_menu.p_menu_pool == NULL)
87 {
88 log_error("calloc(p_menu_pool) error\n");
89 return -3;
90 }
91 bbsnet_menu.menu_count = 1;
92
93 bbsnet_menu.p_menu_item_pool = calloc(MAXSTATION, sizeof(MENU_ITEM));
94 if (bbsnet_menu.p_menu_item_pool == NULL)
95 {
96 log_error("calloc(p_menu_item_pool) error\n");
97 return -3;
98 }
99 bbsnet_menu.menu_item_count = MAXSTATION;
100
101 p_menu = (MENU *)get_menu_by_id(&bbsnet_menu, 0);
102
103 strncpy(p_menu->name, "BBSNET", sizeof(p_menu->name) - 1);
104 p_menu->name[sizeof(p_menu->name) - 1] = '\0';
105 p_menu->title.show = 0;
106 p_menu->screen_show = 0;
107
108 menu_item_id = 0;
109 while (fgets(t, 255, fp) && menu_item_id < MAXSTATION)
110 {
111 t1 = strtok_r(t, MENU_CONF_DELIM, &saveptr);
112 t2 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
113 t3 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
114 t4 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
115 t5 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
116 t6 = strtok_r(NULL, MENU_CONF_DELIM, &saveptr);
117
118 if (t1 == NULL || t2 == NULL || t3 == NULL || t4 == NULL ||
119 t5 == NULL || t6 == NULL || t[0] == '#' || t[0] == '*')
120 {
121 continue;
122 }
123
124 strncpy(bbsnet_conf[menu_item_id].host1, t2, sizeof(bbsnet_conf[menu_item_id].host1) - 1);
125 bbsnet_conf[menu_item_id].host1[sizeof(bbsnet_conf[menu_item_id].host1) - 1] = '\0';
126 strncpy(bbsnet_conf[menu_item_id].host2, t1, sizeof(bbsnet_conf[menu_item_id].host2) - 1);
127 bbsnet_conf[menu_item_id].host2[sizeof(bbsnet_conf[menu_item_id].host2) - 1] = '\0';
128 strncpy(bbsnet_conf[menu_item_id].ip, t3, sizeof(bbsnet_conf[menu_item_id].ip) - 1);
129 bbsnet_conf[menu_item_id].ip[sizeof(bbsnet_conf[menu_item_id].ip) - 1] = '\0';
130 bbsnet_conf[menu_item_id].port = (in_port_t)(t4 ? atoi(t4) : 23);
131 bbsnet_conf[menu_item_id].use_ssh = (toupper(t5[0]) == 'Y');
132 strncpy(bbsnet_conf[menu_item_id].charset, t6, sizeof(bbsnet_conf[menu_item_id].charset) - 1);
133 bbsnet_conf[menu_item_id].charset[sizeof(bbsnet_conf[menu_item_id].charset) - 1] = '\0';
134
135 p_menu_item = get_menu_item_by_id(&bbsnet_menu, menu_item_id);
136 if (p_menu_item == NULL)
137 {
138 log_error("get_menu_item_by_id(%d) return NULL pointer\n", menu_item_id);
139 return -1;
140 }
141
142 p_menu_item->row = (int16_t)(2 + menu_item_id / STATION_PER_LINE);
143 p_menu_item->col = (int16_t)(5 + menu_item_id % STATION_PER_LINE * 20);
144 snprintf(p_menu_item->action, sizeof(p_menu_item->action), "%d", (int16_t)menu_item_id);
145 p_menu_item->submenu = 0;
146 p_menu_item->priv = 0;
147 p_menu_item->level = 0;
148 p_menu_item->name[0] =
149 (char)(menu_item_id < MAXSTATION / 2 ? 'A' + menu_item_id : 'a' + menu_item_id);
150 p_menu_item->name[1] = '\0';
151 snprintf(p_menu_item->text, sizeof(p_menu_item->text), "%c. %s",
152 p_menu_item->name[0], bbsnet_conf[menu_item_id].host1);
153
154 p_menu->items[p_menu->item_count] = menu_item_id;
155 p_menu->item_count++;
156 menu_item_id++;
157 }
158
159 bbsnet_menu.menu_item_count = (int16_t)menu_item_id;
160 bbsnet_menu.menu_id_path[0] = 0;
161 bbsnet_menu.menu_item_pos[0] = 0;
162 bbsnet_menu.choose_step = 0;
163
164 fclose(fp);
165
166 return 0;
167 }
168
169 static void unload_bbsnet_conf(void)
170 {
171 bbsnet_menu.menu_count = 0;
172 bbsnet_menu.menu_item_count = 0;
173
174 free(bbsnet_menu.p_menu_pool);
175 bbsnet_menu.p_menu_pool = NULL;
176 free(bbsnet_menu.p_menu_item_pool);
177 bbsnet_menu.p_menu_item_pool = NULL;
178 }
179
180 static void process_bar(int n, int len)
181 {
182 char buf[LINE_BUFFER_LEN];
183 char buf2[LINE_BUFFER_LEN];
184
185 if (len > LINE_BUFFER_LEN)
186 {
187 len = LINE_BUFFER_LEN - 1;
188 }
189 if (n < 0)
190 {
191 n = 0;
192 }
193 else if (n > len)
194 {
195 n = len;
196 }
197
198 moveto(4, 1);
199 prints(" ------------------------------ \r\n");
200 snprintf(buf, sizeof(buf), " %3d%% ", n * 100 / len);
201 memcpy(buf2, buf, (size_t)n);
202 buf2[n] = '\0';
203 prints("|\033[46m%s\033[44m%s\033[m|\r\n", buf2, buf + n);
204 prints(" ------------------------------ \r\n");
205 iflush();
206 }
207
208 static int bbsnet_connect(int n)
209 {
210 int sock, ret, loop, error;
211 int sock_connected = 0;
212 int flags_sock;
213 int flags_stdin;
214 int flags_stdout;
215 int len;
216 struct sockaddr_in sin;
217 char input_buf[LINE_BUFFER_LEN];
218 char output_buf[LINE_BUFFER_LEN];
219 int input_buf_len = 0;
220 int output_buf_len = 0;
221 int input_buf_offset = 0;
222 int output_buf_offset = 0;
223 char input_conv[LINE_BUFFER_LEN * 2];
224 char output_conv[LINE_BUFFER_LEN * 2];
225 int input_conv_len = 0;
226 int output_conv_len = 0;
227 int input_conv_offset = 0;
228 int output_conv_offset = 0;
229 iconv_t input_cd = NULL;
230 iconv_t output_cd = NULL;
231 char tocode[32];
232
233 #ifdef HAVE_SYS_EPOLL_H
234 struct epoll_event ev, events[MAX_EVENTS];
235 int epollfd;
236 #else
237 struct pollfd pfds[3];
238 #endif
239
240 int nfds;
241 int stdin_read_wait = 0;
242 int stdout_write_wait = 0;
243 int sock_read_wait = 0;
244 int sock_write_wait = 0;
245 struct hostent *p_host = NULL;
246 int tos;
247 char remote_addr[IP_ADDR_LEN];
248 int remote_port;
249 char local_addr[IP_ADDR_LEN];
250 int local_port;
251 socklen_t sock_len;
252 time_t t_used = time(NULL);
253 struct tm *tm_used;
254 int ch;
255 char remote_user[USERNAME_MAX_LEN + 1];
256 char remote_pass[PASSWORD_MAX_LEN + 1];
257 ssh_session session = NULL;
258 ssh_channel channel = NULL;
259 int ssh_process_config = 0;
260 int ssh_log_level = SSH_LOG_PROTOCOL;
261
262 if (user_online_update("BBS_NET") < 0)
263 {
264 log_error("user_online_update(BBS_NET) error\n");
265 }
266
267 if (bbsnet_conf[n].use_ssh)
268 {
269 clearscr();
270
271 if (!SSH_v2)
272 {
273 moveto(1, 1);
274 prints("只有在以SSH方式登陆本站时,才能使用SSH站点穿梭。");
275 press_any_key();
276 return 0;
277 }
278
279 moveto(1, 1);
280 prints("通过SSH方式连接[%s]...", bbsnet_conf[n].host1);
281 moveto(2, 1);
282 prints("请输入用户名: ");
283 iflush();
284 if (str_input(remote_user, sizeof(remote_user), DOECHO) < 0)
285 {
286 return -1;
287 }
288 if (remote_user[0] == '\0')
289 {
290 return 0;
291 }
292
293 moveto(3, 1);
294 prints("请输入密码: ");
295 iflush();
296 if (str_input(remote_pass, sizeof(remote_pass), NOECHO) < 0)
297 {
298 return -1;
299 }
300 if (remote_pass[0] == '\0')
301 {
302 return 0;
303 }
304 }
305
306 clearscr();
307
308 moveto(1, 1);
309 prints("\033[1;32m正在测试往 %s (%s) 的连接,请稍候... \033[m\r\n",
310 bbsnet_conf[n].host1, bbsnet_conf[n].ip);
311 iflush();
312
313 p_host = gethostbyname(bbsnet_conf[n].ip);
314
315 if (p_host == NULL)
316 {
317 prints("\033[1;31m查找主机名失败!\033[m\r\n");
318 press_any_key();
319 return -1;
320 }
321
322 sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
323
324 if (sock < 0)
325 {
326 prints("\033[1;31m无法创建socket!\033[m\r\n");
327 press_any_key();
328 return -1;
329 }
330
331 sin.sin_family = AF_INET;
332 sin.sin_addr.s_addr = (BBS_address[0] != '\0' ? inet_addr(BBS_address) : INADDR_ANY);
333 sin.sin_port = 0;
334
335 if (bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0)
336 {
337 log_error("Bind address %s:%u failed (%d)\n",
338 inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), errno);
339 return -2;
340 }
341
342 memset(&sin, 0, sizeof(sin));
343 sin.sin_family = AF_INET;
344 sin.sin_addr = *(struct in_addr *)p_host->h_addr_list[0];
345 sin.sin_port = htons(bbsnet_conf[n].port);
346
347 strncpy(remote_addr, inet_ntoa(sin.sin_addr), sizeof(remote_addr) - 1);
348 remote_addr[sizeof(remote_addr) - 1] = '\0';
349 remote_port = ntohs(sin.sin_port);
350
351 prints("\033[1;32m穿梭进度条提示您当前已使用的时间,按\033[1;33mCtrl+C\033[1;32m中断。\033[m\r\n");
352 process_bar(0, MAX_PROCESS_BAR_LEN);
353
354 // Set socket as non-blocking
355 flags_sock = fcntl(sock, F_GETFL, 0);
356 fcntl(sock, F_SETFL, flags_sock | O_NONBLOCK);
357
358 // Set STDIN/STDOUT as non-blocking
359 flags_stdin = fcntl(STDIN_FILENO, F_GETFL, 0);
360 flags_stdout = fcntl(STDOUT_FILENO, F_GETFL, 0);
361 fcntl(STDIN_FILENO, F_SETFL, flags_stdin | O_NONBLOCK);
362 fcntl(STDOUT_FILENO, F_SETFL, flags_stdout | O_NONBLOCK);
363
364 #ifdef HAVE_SYS_EPOLL_H
365 epollfd = epoll_create1(0);
366 if (epollfd < 0)
367 {
368 log_error("epoll_create1() error (%d)\n", errno);
369 return -1;
370 }
371
372 ev.events = EPOLLOUT | EPOLLET;
373 ev.data.fd = sock;
374 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, sock, &ev) == -1)
375 {
376 log_error("epoll_ctl(socket) error (%d)\n", errno);
377 goto cleanup;
378 }
379
380 ev.events = EPOLLIN | EPOLLET;
381 ev.data.fd = STDIN_FILENO;
382 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDIN_FILENO, &ev) == -1)
383 {
384 log_error("epoll_ctl(STDIN_FILENO) error (%d)\n", errno);
385 goto cleanup;
386 }
387 #endif
388
389 while (!SYS_server_exit)
390 {
391 if ((ret = connect(sock, (struct sockaddr *)&sin, sizeof(sin))) < 0)
392 {
393 if (errno == EAGAIN || errno == EALREADY || errno == EINPROGRESS)
394 {
395 // Use select / epoll to check writability of the socket,
396 // then use getsockopt to check the status of the socket.
397 // See man connect(2)
398 break;
399 }
400 else if (errno == EINTR)
401 {
402 continue;
403 }
404 else
405 {
406 log_error("connect(socket) error (%d)\n", errno);
407
408 prints("\033[1;31m连接失败!\033[m\r\n");
409 press_any_key();
410
411 goto cleanup;
412 }
413 }
414 }
415
416 for (int j = 0; j < MAX_PROCESS_BAR_LEN && !sock_connected && !SYS_server_exit; j++)
417 {
418 #ifdef HAVE_SYS_EPOLL_H
419 nfds = epoll_wait(epollfd, events, MAX_EVENTS, 500); // 0.5 second
420 ret = nfds;
421 #else
422 pfds[0].fd = sock;
423 pfds[0].events = POLLOUT;
424 pfds[1].fd = STDIN_FILENO;
425 pfds[1].events = POLLIN;
426 nfds = 2;
427 ret = poll(pfds, (nfds_t)nfds, 500); // 0.5 second
428 #endif
429
430 if (ret < 0)
431 {
432 if (errno != EINTR)
433 {
434 #ifdef HAVE_SYS_EPOLL_H
435 log_error("epoll_wait() error (%d)\n", errno);
436 #else
437 log_error("poll() error (%d)\n", errno);
438 #endif
439 break;
440 }
441 }
442 else if (ret == 0) // timeout
443 {
444 process_bar(j + 1, MAX_PROCESS_BAR_LEN);
445 }
446 else // ret > 0
447 {
448 for (int i = 0; i < nfds; i++)
449 {
450 #ifdef HAVE_SYS_EPOLL_H
451 if (events[i].data.fd == sock)
452 #else
453 if (pfds[i].fd == sock && (pfds[i].revents & POLLOUT))
454 #endif
455 {
456 len = sizeof(error);
457 if (getsockopt(sock, SOL_SOCKET, SO_ERROR, &error, (socklen_t *)&len) < 0)
458 {
459 log_error("getsockopt() error (%d) !\n", error);
460 goto cleanup;
461 }
462 if (error == 0)
463 {
464 sock_connected = 1;
465 }
466 }
467 #ifdef HAVE_SYS_EPOLL_H
468 else if (events[i].data.fd == STDIN_FILENO)
469 #else
470 else if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))
471 #endif
472 {
473 do
474 {
475 ch = igetch(0);
476 } while (ch == 0);
477 if (ch == Ctrl('C') || ch == KEY_ESC)
478 {
479 goto cleanup;
480 }
481 }
482 }
483 }
484 }
485 if (SYS_server_exit)
486 {
487 goto cleanup;
488 }
489 if (!sock_connected)
490 {
491 prints("\033[1;31m连接失败!\033[m\r\n");
492 press_any_key();
493
494 goto cleanup;
495 }
496
497 tos = IPTOS_LOWDELAY;
498 if (setsockopt(sock, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)) < 0)
499 {
500 log_error("setsockopt IP_TOS=%d error (%d)\n", tos, errno);
501 }
502
503 sock_len = sizeof(sin);
504 if (getsockname(sock, (struct sockaddr *)&sin, &sock_len) < 0)
505 {
506 log_error("getsockname() error: %d", errno);
507 goto cleanup;
508 }
509
510 strncpy(local_addr, inet_ntoa(sin.sin_addr), sizeof(local_addr) - 1);
511 local_addr[sizeof(local_addr) - 1] = '\0';
512 local_port = ntohs(sin.sin_port);
513
514 if (bbsnet_conf[n].use_ssh)
515 {
516 session = ssh_new();
517 if (session == NULL)
518 {
519 log_error("ssh_new() error\n");
520 goto cleanup;
521 }
522
523 if (ssh_options_set(session, SSH_OPTIONS_FD, &sock) < 0 ||
524 ssh_options_set(session, SSH_OPTIONS_PROCESS_CONFIG, &ssh_process_config) < 0 ||
525 ssh_options_set(session, SSH_OPTIONS_HOST, bbsnet_conf[n].ip) < 0 ||
526 ssh_options_set(session, SSH_OPTIONS_USER, remote_user) < 0 ||
527 ssh_options_set(session, SSH_OPTIONS_HOSTKEYS, "+ssh-rsa") < 0 ||
528 ssh_options_set(session, SSH_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0)
529 {
530 log_error("Error setting SSH options: %s\n", ssh_get_error(session));
531 goto cleanup;
532 }
533
534 while (!SYS_server_exit)
535 {
536 ret = ssh_connect(session);
537 if (ret == SSH_OK)
538 {
539 break;
540 }
541 else if (ret == SSH_ERROR)
542 {
543 log_error("ssh_connect() error\n");
544 goto cleanup;
545 }
546 }
547
548 ssh_set_blocking(session, 0);
549
550 for (int i = 0; !SYS_server_exit;)
551 {
552 ret = ssh_userauth_password(session, NULL, remote_pass);
553 if (ret == SSH_AUTH_SUCCESS)
554 {
555 break;
556 }
557 else if (ret == SSH_AUTH_AGAIN)
558 {
559 #ifdef _DEBUG
560 log_error("ssh_userauth_password() error: SSH_AUTH_AGAIN\n");
561 #endif
562 }
563 else if (ret == SSH_AUTH_ERROR)
564 {
565 log_error("ssh_userauth_password() error: %d\n", ret);
566 goto cleanup;
567 }
568 else // if (ret == SSH_AUTH_DENIED)
569 {
570 prints("\033[1;31m身份验证失败!\033[m\r\n");
571 i++;
572 if (i < BBS_login_retry_times)
573 {
574 prints("请输入密码: ");
575 iflush();
576 if (str_input(remote_pass, sizeof(remote_pass), NOECHO) < 0)
577 {
578 goto cleanup;
579 }
580 if (remote_pass[0] == '\0')
581 {
582 goto cleanup;
583 }
584 }
585 else
586 {
587 goto cleanup;
588 }
589 }
590 }
591
592 channel = ssh_channel_new(session);
593 if (channel == NULL)
594 {
595 log_error("ssh_channel_new() error\n");
596 goto cleanup;
597 }
598
599 while (!SYS_server_exit)
600 {
601 ret = ssh_channel_open_session(channel);
602 if (ret == SSH_OK)
603 {
604 break;
605 }
606 else if (ret == SSH_ERROR)
607 {
608 log_error("ssh_channel_open_session() error\n");
609 goto cleanup;
610 }
611 }
612
613 while (!SYS_server_exit)
614 {
615 ret = ssh_channel_request_pty(channel);
616 if (ret == SSH_OK)
617 {
618 break;
619 }
620 else if (ret == SSH_ERROR)
621 {
622 log_error("ssh_channel_request_pty() error\n");
623 goto cleanup;
624 }
625 }
626
627 while (!SYS_server_exit)
628 {
629 ret = ssh_channel_request_shell(channel);
630 if (ret == SSH_OK)
631 {
632 break;
633 }
634 else if (ret == SSH_ERROR)
635 {
636 log_error("ssh_channel_request_shell() error\n");
637 goto cleanup;
638 }
639 }
640 }
641
642 prints("\033[1;31m连接成功!\033[m\r\n");
643 iflush();
644 log_common("BBSNET connect to %s:%d from %s:%d by [%s]\n",
645 remote_addr, remote_port, local_addr, local_port, BBS_username);
646
647 snprintf(tocode, sizeof(tocode), "%s%s", bbsnet_conf[n].charset,
648 (strcasecmp(stdio_charset, bbsnet_conf[n].charset) == 0 ? "" : "//IGNORE"));
649 input_cd = iconv_open(tocode, stdio_charset);
650 if (input_cd == (iconv_t)(-1))
651 {
652 log_error("iconv_open(%s->%s) error: %d\n", stdio_charset, tocode, errno);
653 goto cleanup;
654 }
655
656 snprintf(tocode, sizeof(tocode), "%s%s", stdio_charset,
657 (strcasecmp(bbsnet_conf[n].charset, stdio_charset) == 0 ? "" : "//TRANSLIT"));
658 output_cd = iconv_open(tocode, bbsnet_conf[n].charset);
659 if (output_cd == (iconv_t)(-1))
660 {
661 log_error("iconv_open(%s->%s) error: %d\n", bbsnet_conf[n].charset, tocode, errno);
662 iconv_close(input_cd);
663 goto cleanup;
664 }
665
666 #ifdef HAVE_SYS_EPOLL_H
667 ev.events = EPOLLIN | EPOLLOUT | EPOLLET;
668 ev.data.fd = sock;
669 if (epoll_ctl(epollfd, EPOLL_CTL_MOD, sock, &ev) == -1)
670 {
671 log_error("epoll_ctl(socket) error (%d)\n", errno);
672 goto cleanup;
673 }
674
675 ev.events = EPOLLOUT | EPOLLET;
676 ev.data.fd = STDOUT_FILENO;
677 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, STDOUT_FILENO, &ev) == -1)
678 {
679 log_error("epoll_ctl(STDOUT_FILENO) error (%d)\n", errno);
680 goto cleanup;
681 }
682 #endif
683
684 BBS_last_access_tm = t_used = time(NULL);
685 loop = 1;
686
687 while (loop && !SYS_server_exit)
688 {
689 if (SSH_v2 && ssh_channel_is_closed(SSH_channel))
690 {
691 log_error("SSH channel is closed\n");
692 loop = 0;
693 break;
694 }
695
696 if (bbsnet_conf[n].use_ssh && ssh_channel_is_closed(channel))
697 {
698 log_error("Remote SSH channel is closed\n");
699 loop = 0;
700 break;
701 }
702
703 #ifdef HAVE_SYS_EPOLL_H
704 nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second
705 ret = nfds;
706 #else
707 pfds[0].fd = STDIN_FILENO;
708 pfds[0].events = POLLIN;
709 pfds[1].fd = sock;
710 pfds[1].events = POLLIN | POLLOUT;
711 pfds[2].fd = STDOUT_FILENO;
712 pfds[2].events = POLLOUT;
713 nfds = 3;
714 ret = poll(pfds, (nfds_t)nfds, 100); // 0.1 second
715 #endif
716
717 if (ret < 0)
718 {
719 if (errno != EINTR)
720 {
721 #ifdef HAVE_SYS_EPOLL_H
722 log_error("epoll_wait() error (%d)\n", errno);
723 #else
724 log_error("poll() error (%d)\n", errno);
725 #endif
726 break;
727 }
728 continue;
729 }
730 else if (ret == 0) // timeout
731 {
732 if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)
733 {
734 break;
735 }
736 }
737
738 for (int i = 0; i < nfds; i++)
739 {
740 #ifdef HAVE_SYS_EPOLL_H
741 if (events[i].data.fd == STDIN_FILENO)
742 #else
743 if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))
744 #endif
745 {
746 stdin_read_wait = 1;
747 }
748
749 #ifdef HAVE_SYS_EPOLL_H
750 if (events[i].data.fd == sock)
751 #else
752 if (pfds[i].fd == sock)
753 #endif
754 {
755 #ifdef HAVE_SYS_EPOLL_H
756 if (events[i].events & EPOLLIN)
757 #else
758 if (pfds[i].revents & POLLIN)
759 #endif
760 {
761 sock_read_wait = 1;
762 }
763
764 #ifdef HAVE_SYS_EPOLL_H
765 if (events[i].events & EPOLLOUT)
766 #else
767 if (pfds[i].revents & POLLOUT)
768 #endif
769 {
770 sock_write_wait = 1;
771 }
772 }
773
774 #ifdef HAVE_SYS_EPOLL_H
775 if (events[i].data.fd == STDOUT_FILENO)
776 #else
777 if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT))
778 #endif
779 {
780 stdout_write_wait = 1;
781 }
782 }
783
784 if (stdin_read_wait)
785 {
786 while (input_buf_len < sizeof(input_buf) && !SYS_server_exit)
787 {
788 if (SSH_v2)
789 {
790 ret = ssh_channel_read_nonblocking(SSH_channel, input_buf + input_buf_len, sizeof(input_buf) - (uint32_t)input_buf_len, 0);
791 if (ret == SSH_ERROR)
792 {
793 log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session));
794 loop = 0;
795 break;
796 }
797 else if (ret == SSH_EOF)
798 {
799 stdin_read_wait = 0;
800 loop = 0;
801 break;
802 }
803 else if (ret == 0)
804 {
805 // Send NO-OP to remote server
806 input_buf[input_buf_len] = '\0';
807 input_buf_len++;
808 BBS_last_access_tm = time(NULL);
809
810 stdin_read_wait = 0;
811 break; // Check whether channel is still open
812 }
813 }
814 else
815 {
816 ret = (int)read(STDIN_FILENO, input_buf + input_buf_len, sizeof(input_buf) - (size_t)input_buf_len);
817 }
818 if (ret < 0)
819 {
820 if (errno == EAGAIN || errno == EWOULDBLOCK)
821 {
822 stdin_read_wait = 0;
823 break;
824 }
825 else if (errno == EINTR)
826 {
827 continue;
828 }
829 else
830 {
831 log_error("read(STDIN) error (%d)\n", errno);
832 loop = 0;
833 break;
834 }
835 }
836 else if (ret == 0) // broken pipe
837 {
838 #ifdef _DEBUG
839 log_error("read(STDIN) EOF\n");
840 #endif
841 stdin_read_wait = 0;
842 loop = 0;
843 break;
844 }
845 else
846 {
847 input_buf_len += ret;
848 BBS_last_access_tm = time(NULL);
849
850 // Refresh current action while user input
851 if (user_online_update("BBS_NET") < 0)
852 {
853 log_error("user_online_update(BBS_NET) error\n");
854 }
855
856 continue;
857 }
858 }
859 }
860
861 if (sock_write_wait)
862 {
863 if (input_buf_offset < input_buf_len)
864 {
865 // For debug
866 #ifdef _DEBUG
867 for (int j = input_buf_offset; j < input_buf_len; j++)
868 {
869 log_error("Debug input: <--[%u]\n", (input_buf[j] + 256) % 256);
870 }
871 #endif
872
873 ret = io_buf_conv(input_cd, input_buf, &input_buf_len, &input_buf_offset, input_conv, sizeof(input_conv), &input_conv_len);
874 if (ret < 0)
875 {
876 log_error("io_buf_conv(input, %d, %d, %d) error\n", input_buf_len, input_buf_offset, input_conv_len);
877 input_buf_len = input_buf_offset; // Discard invalid sequence
878 }
879
880 // For debug
881 #ifdef _DEBUG
882 for (int j = input_conv_offset; j < input_conv_len; j++)
883 {
884 log_error("Debug input_conv: <--[%u]\n", (input_conv[j] + 256) % 256);
885 }
886 #endif
887 }
888
889 while (input_conv_offset < input_conv_len && !SYS_server_exit)
890 {
891 if (bbsnet_conf[n].use_ssh)
892 {
893 ret = ssh_channel_write(channel, input_conv + input_conv_offset, (uint32_t)(input_conv_len - input_conv_offset));
894 if (ret == SSH_ERROR)
895 {
896 log_error("ssh_channel_write() error: %s\n", ssh_get_error(session));
897 loop = 0;
898 break;
899 }
900 }
901 else
902 {
903 ret = (int)write(sock, input_conv + input_conv_offset, (size_t)(input_conv_len - input_conv_offset));
904 }
905 if (ret < 0)
906 {
907 if (errno == EAGAIN || errno == EWOULDBLOCK)
908 {
909 sock_write_wait = 0;
910 break;
911 }
912 else if (errno == EINTR)
913 {
914 continue;
915 }
916 else
917 {
918 log_error("write(socket) error (%d)\n", errno);
919 loop = 0;
920 break;
921 }
922 }
923 else if (ret == 0) // broken pipe
924 {
925 #ifdef _DEBUG
926 log_error("write(socket) EOF\n");
927 #endif
928 sock_write_wait = 0;
929 loop = 0;
930 break;
931 }
932 else
933 {
934 input_conv_offset += ret;
935 if (input_conv_offset >= input_conv_len) // Output buffer complete
936 {
937 input_conv_offset = 0;
938 input_conv_len = 0;
939 break;
940 }
941 continue;
942 }
943 }
944 }
945
946 if (sock_read_wait)
947 {
948 while (output_buf_len < sizeof(output_buf) && !SYS_server_exit)
949 {
950 if (bbsnet_conf[n].use_ssh)
951 {
952 ret = ssh_channel_read_nonblocking(channel, output_buf + output_buf_len,
953 (uint32_t)(sizeof(output_buf) - (size_t)output_buf_len), 0);
954 if (ret == SSH_ERROR)
955 {
956 log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(session));
957 loop = 0;
958 break;
959 }
960 else if (ret == SSH_EOF)
961 {
962 sock_read_wait = 0;
963 loop = 0;
964 break;
965 }
966 else if (ret == 0)
967 {
968 sock_read_wait = 0;
969 break;
970 }
971 }
972 else
973 {
974 ret = (int)read(sock, output_buf + output_buf_len, sizeof(output_buf) - (size_t)output_buf_len);
975 }
976 if (ret < 0)
977 {
978 if (errno == EAGAIN || errno == EWOULDBLOCK)
979 {
980 sock_read_wait = 0;
981 break;
982 }
983 else if (errno == EINTR)
984 {
985 continue;
986 }
987 else
988 {
989 log_error("read(socket) error (%d)\n", errno);
990 loop = 0;
991 break;
992 }
993 }
994 else if (ret == 0) // broken pipe
995 {
996 #ifdef _DEBUG
997 log_error("read(socket) EOF\n");
998 #endif
999 sock_read_wait = 0;
1000 loop = 0;
1001 break;
1002 }
1003 else
1004 {
1005 output_buf_len += ret;
1006 continue;
1007 }
1008 }
1009 }
1010
1011 if (stdout_write_wait)
1012 {
1013 if (output_buf_offset < output_buf_len)
1014 {
1015 ret = io_buf_conv(output_cd, output_buf, &output_buf_len, &output_buf_offset, output_conv, sizeof(output_conv), &output_conv_len);
1016 if (ret < 0)
1017 {
1018 log_error("io_buf_conv(output, %d, %d, %d) error\n", output_buf_len, output_buf_offset, output_conv_len);
1019 output_buf_len = output_buf_offset; // Discard invalid sequence
1020 }
1021 }
1022
1023 while (output_conv_offset < output_conv_len && !SYS_server_exit)
1024 {
1025 if (SSH_v2)
1026 {
1027 ret = ssh_channel_write(SSH_channel, output_conv + output_conv_offset, (uint32_t)(output_conv_len - output_conv_offset));
1028 if (ret == SSH_ERROR)
1029 {
1030 log_error("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session));
1031 loop = 0;
1032 break;
1033 }
1034 }
1035 else
1036 {
1037 ret = (int)write(STDOUT_FILENO, output_conv + output_conv_offset, (size_t)(output_conv_len - output_conv_offset));
1038 }
1039 if (ret < 0)
1040 {
1041 if (errno == EAGAIN || errno == EWOULDBLOCK)
1042 {
1043 stdout_write_wait = 0;
1044 break;
1045 }
1046 else if (errno == EINTR)
1047 {
1048 continue;
1049 }
1050 else
1051 {
1052 log_error("write(STDOUT) error (%d)\n", errno);
1053 loop = 0;
1054 break;
1055 }
1056 }
1057 else if (ret == 0) // broken pipe
1058 {
1059 #ifdef _DEBUG
1060 log_error("write(STDOUT) EOF\n");
1061 #endif
1062 stdout_write_wait = 0;
1063 loop = 0;
1064 break;
1065 }
1066 else
1067 {
1068 output_conv_offset += ret;
1069 if (output_conv_offset >= output_conv_len) // Output buffer complete
1070 {
1071 output_conv_offset = 0;
1072 output_conv_len = 0;
1073 break;
1074 }
1075 continue;
1076 }
1077 }
1078 }
1079 }
1080
1081 iconv_close(input_cd);
1082 iconv_close(output_cd);
1083
1084 cleanup:
1085 #ifdef HAVE_SYS_EPOLL_H
1086 if (close(epollfd) < 0)
1087 {
1088 log_error("close(epoll) error (%d)\n");
1089 }
1090 #endif
1091
1092 if (bbsnet_conf[n].use_ssh)
1093 {
1094 ssh_channel_free(channel);
1095 ssh_disconnect(session);
1096 ssh_free(session);
1097 }
1098
1099 // Restore STDIN/STDOUT flags
1100 fcntl(STDIN_FILENO, F_SETFL, flags_stdin);
1101 fcntl(STDOUT_FILENO, F_SETFL, flags_stdout);
1102
1103 if (close(sock) == -1)
1104 {
1105 log_error("Close socket failed\n");
1106 }
1107
1108 t_used = time(NULL) - t_used;
1109 tm_used = gmtime(&t_used);
1110
1111 log_common("BBSNET disconnect, %d days %d hours %d minutes %d seconds used\n",
1112 tm_used->tm_mday - 1, tm_used->tm_hour, tm_used->tm_min,
1113 tm_used->tm_sec);
1114
1115 BBS_last_access_tm = time(NULL);
1116
1117 return 0;
1118 }
1119
1120 static int bbsnet_refresh()
1121 {
1122 clearscr();
1123
1124 moveto(1, 1);
1125 prints(" ------------------------------------------------------------------------------ ");
1126 for (int i = 2; i < 19; i++)
1127 {
1128 moveto(i, 1);
1129 prints("|");
1130 moveto(i, 80);
1131 prints("|");
1132 }
1133 moveto(19, 1);
1134 prints("|------------------------------------------------------------------------------|");
1135 moveto(22, 1);
1136 prints(" ------------------------------------------------------------------------------ ");
1137 moveto(23, 1);
1138 prints(" [\033[1;32mCtrl+C\033[m]退出");
1139
1140 iflush();
1141
1142 return 0;
1143 }
1144
1145 static int bbsnet_selchange()
1146 {
1147 int i = bbsnet_menu.menu_item_pos[0];
1148
1149 moveto(20, 1);
1150 clrtoeol();
1151 prints("|\033[1m单位: \033[1;33m%s\033[m%*s 站名: \033[1;33m%s\033[m",
1152 bbsnet_conf[i].host2, 20 - str_length(bbsnet_conf[i].host2, 1), "", bbsnet_conf[i].host1);
1153 moveto(20, 80);
1154 prints("|");
1155 moveto(21, 1);
1156 clrtoeol();
1157 prints("|\033[1m连往: \033[1;33m%-20s\033[m 端口: \033[1;33m%-5d\033[m 类型: \033[1;33m%s\033[m",
1158 bbsnet_conf[i].ip, bbsnet_conf[i].port, (bbsnet_conf[i].use_ssh ? "SSH" : "Telnet"));
1159 moveto(21, 80);
1160 prints("|");
1161 iflush();
1162
1163 return 0;
1164 }
1165
1166 int bbs_net()
1167 {
1168 int ch, i;
1169
1170 load_bbsnet_conf(CONF_BBSNET);
1171
1172 bbsnet_refresh();
1173 display_menu(&bbsnet_menu);
1174 bbsnet_selchange();
1175
1176 while (!SYS_server_exit)
1177 {
1178 ch = igetch(100);
1179
1180 if (ch != KEY_NULL && ch != KEY_TIMEOUT)
1181 {
1182 BBS_last_access_tm = time(NULL);
1183 }
1184
1185 switch (ch)
1186 {
1187 case KEY_NULL: // broken pipe
1188 log_error("KEY_NULL\n");
1189 goto cleanup;
1190 case KEY_TIMEOUT:
1191 if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)
1192 {
1193 log_error("User input timeout\n");
1194 goto cleanup;
1195 }
1196 continue;
1197 case KEY_ESC:
1198 case Ctrl('C'): // user cancel
1199 goto cleanup;
1200 case CR:
1201 bbsnet_connect(bbsnet_menu.menu_item_pos[0]);
1202 // Force cleanup anything remaining in the output buffer
1203 clearscr();
1204 iflush();
1205 // Clear screen and redraw menu
1206 bbsnet_refresh();
1207 display_menu(&bbsnet_menu);
1208 bbsnet_selchange();
1209 break;
1210 case KEY_UP:
1211 for (i = 0; i < STATION_PER_LINE; i++)
1212 {
1213 menu_control(&bbsnet_menu, KEY_UP);
1214 }
1215 bbsnet_selchange();
1216 break;
1217 case KEY_DOWN:
1218 for (i = 0; i < STATION_PER_LINE; i++)
1219 {
1220 menu_control(&bbsnet_menu, KEY_DOWN);
1221 }
1222 bbsnet_selchange();
1223 break;
1224 case KEY_LEFT:
1225 menu_control(&bbsnet_menu, KEY_UP);
1226 bbsnet_selchange();
1227 break;
1228 case KEY_RIGHT:
1229 menu_control(&bbsnet_menu, KEY_DOWN);
1230 bbsnet_selchange();
1231 break;
1232 case KEY_HOME:
1233 case KEY_PGUP:
1234 menu_control(&bbsnet_menu, KEY_PGUP);
1235 bbsnet_selchange();
1236 break;
1237 case KEY_END:
1238 case KEY_PGDN:
1239 menu_control(&bbsnet_menu, KEY_PGDN);
1240 bbsnet_selchange();
1241 break;
1242 default:
1243 menu_control(&bbsnet_menu, ch);
1244 bbsnet_selchange();
1245 break;
1246 }
1247 }
1248
1249 cleanup:
1250 unload_bbsnet_conf();
1251
1252 return 0;
1253 }

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1