| 1 |
/*************************************************************************** |
/* SPDX-License-Identifier: GPL-3.0-or-later */ |
| 2 |
io.c - description |
/* |
| 3 |
------------------- |
* io |
| 4 |
Copyright : (C) 2004-2025 by Leaflet |
* - basic terminal-based user input / output features |
| 5 |
Email : leaflet@leafok.com |
* |
| 6 |
***************************************************************************/ |
* Copyright (C) 2004-2025 by Leaflet <leaflet@leafok.com> |
| 7 |
|
*/ |
|
/*************************************************************************** |
|
|
* * |
|
|
* This program is free software; you can redistribute it and/or modify * |
|
|
* it under the terms of the GNU General Public License as published by * |
|
|
* the Free Software Foundation; either version 3 of the License, or * |
|
|
* (at your option) any later version. * |
|
|
* * |
|
|
***************************************************************************/ |
|
| 8 |
|
|
| 9 |
#include "common.h" |
#include "common.h" |
| 10 |
#include "io.h" |
#include "io.h" |
| 27 |
|
|
| 28 |
// static input / output buffer |
// static input / output buffer |
| 29 |
static char stdin_buf[LINE_BUFFER_LEN]; |
static char stdin_buf[LINE_BUFFER_LEN]; |
| 30 |
static char stdin_conv[LINE_BUFFER_LEN]; |
static char stdout_buf[BUFSIZ]; |
| 31 |
static int stdin_buf_len = 0; |
static int stdin_buf_len = 0; |
| 32 |
static int stdin_conv_len = 0; |
static int stdout_buf_len = 0; |
| 33 |
static int stdin_buf_offset = 0; |
static int stdin_buf_offset = 0; |
| 34 |
static int stdin_conv_offset = 0; |
static int stdout_buf_offset = 0; |
| 35 |
|
|
| 36 |
static char stdout_buf[BUFSIZ]; |
static char stdin_conv[LINE_BUFFER_LEN * 2]; |
| 37 |
static char stdout_conv[BUFSIZ]; |
static char stdout_conv[BUFSIZ * 2]; |
| 38 |
static int stdout_buf_len = 0; |
static int stdin_conv_len = 0; |
| 39 |
static int stdout_conv_len = 0; |
static int stdout_conv_len = 0; |
| 40 |
static int stdout_buf_offset = 0; |
static int stdin_conv_offset = 0; |
| 41 |
static int stdout_conv_offset = 0; |
static int stdout_conv_offset = 0; |
| 42 |
|
|
| 43 |
static iconv_t stdin_cd = NULL; |
static iconv_t stdin_cd = NULL; |
| 162 |
if (ret < 0) |
if (ret < 0) |
| 163 |
{ |
{ |
| 164 |
log_error("io_buf_conv(stdout, %d, %d, %d) error\n", stdout_buf_len, stdout_buf_offset, stdout_conv_len); |
log_error("io_buf_conv(stdout, %d, %d, %d) error\n", stdout_buf_len, stdout_buf_offset, stdout_conv_len); |
| 165 |
|
stdout_buf_len = stdout_buf_offset; // Discard invalid sequence |
| 166 |
} |
} |
| 167 |
} |
} |
| 168 |
|
|
| 237 |
|
|
| 238 |
int igetch(int timeout) |
int igetch(int timeout) |
| 239 |
{ |
{ |
| 240 |
|
static int stdin_read_wait = 0; |
| 241 |
|
|
| 242 |
struct epoll_event ev, events[MAX_EVENTS]; |
struct epoll_event ev, events[MAX_EVENTS]; |
| 243 |
int nfds, epollfd; |
int nfds, epollfd; |
| 244 |
int ret; |
int ret; |
| 280 |
flags = fcntl(STDIN_FILENO, F_GETFL, 0); |
flags = fcntl(STDIN_FILENO, F_GETFL, 0); |
| 281 |
fcntl(STDIN_FILENO, F_SETFL, flags | O_NONBLOCK); |
fcntl(STDIN_FILENO, F_SETFL, flags | O_NONBLOCK); |
| 282 |
|
|
| 283 |
for (loop = 1; loop && stdin_conv_offset >= stdin_conv_len && !SYS_server_exit;) |
for (loop = 1; loop && stdin_buf_len < sizeof(stdin_buf) && stdin_conv_offset >= stdin_conv_len && !SYS_server_exit;) |
| 284 |
{ |
{ |
| 285 |
if (SSH_v2 && ssh_channel_is_closed(SSH_channel)) |
if (SSH_v2 && ssh_channel_is_closed(SSH_channel)) |
| 286 |
{ |
{ |
| 289 |
break; |
break; |
| 290 |
} |
} |
| 291 |
|
|
| 292 |
nfds = epoll_wait(epollfd, events, MAX_EVENTS, timeout); |
if (!stdin_read_wait) |
|
|
|
|
if (nfds < 0) |
|
| 293 |
{ |
{ |
| 294 |
if (errno != EINTR) |
nfds = epoll_wait(epollfd, events, MAX_EVENTS, timeout); |
| 295 |
|
|
| 296 |
|
if (nfds < 0) |
| 297 |
{ |
{ |
| 298 |
log_error("epoll_wait() error (%d)\n", errno); |
if (errno != EINTR) |
| 299 |
|
{ |
| 300 |
|
log_error("epoll_wait() error (%d)\n", errno); |
| 301 |
|
break; |
| 302 |
|
} |
| 303 |
|
continue; |
| 304 |
|
} |
| 305 |
|
else if (nfds == 0) // timeout |
| 306 |
|
{ |
| 307 |
|
out = KEY_TIMEOUT; |
| 308 |
break; |
break; |
| 309 |
} |
} |
| 310 |
continue; |
|
| 311 |
} |
for (int i = 0; i < nfds; i++) |
| 312 |
else if (nfds == 0) // timeout |
{ |
| 313 |
{ |
if (events[i].data.fd == STDIN_FILENO) |
| 314 |
out = KEY_TIMEOUT; |
{ |
| 315 |
break; |
stdin_read_wait = 1; |
| 316 |
|
} |
| 317 |
|
} |
| 318 |
} |
} |
| 319 |
|
|
| 320 |
for (int i = 0; i < nfds; i++) |
if (stdin_read_wait) |
| 321 |
{ |
{ |
| 322 |
if (events[i].data.fd == STDIN_FILENO) |
while (stdin_buf_len < sizeof(stdin_buf) && !SYS_server_exit) // read until complete or error |
| 323 |
{ |
{ |
| 324 |
while (stdin_buf_len < sizeof(stdin_buf) && !SYS_server_exit) // read until complete or error |
if (SSH_v2) |
| 325 |
{ |
{ |
| 326 |
if (SSH_v2) |
ret = ssh_channel_read_nonblocking(SSH_channel, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (uint32_t)stdin_buf_len, 0); |
| 327 |
|
if (ret == SSH_ERROR) |
| 328 |
{ |
{ |
| 329 |
ret = ssh_channel_read_nonblocking(SSH_channel, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (uint32_t)stdin_buf_len, 0); |
log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session)); |
| 330 |
if (ret == SSH_ERROR) |
loop = 0; |
| 331 |
{ |
break; |
|
log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session)); |
|
|
loop = 0; |
|
|
break; |
|
|
} |
|
|
else if (ret == SSH_EOF) |
|
|
{ |
|
|
loop = 0; |
|
|
break; |
|
|
} |
|
|
else if (ret == 0) |
|
|
{ |
|
|
out = 0; |
|
|
loop = 0; |
|
|
break; |
|
|
} |
|
| 332 |
} |
} |
| 333 |
else |
else if (ret == SSH_EOF) |
| 334 |
{ |
{ |
| 335 |
ret = (int)read(STDIN_FILENO, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (size_t)stdin_buf_len); |
stdin_read_wait = 0; |
| 336 |
|
loop = 0; |
| 337 |
|
break; |
| 338 |
} |
} |
| 339 |
if (ret < 0) |
else if (ret == 0) |
| 340 |
{ |
{ |
| 341 |
if (errno == EAGAIN || errno == EWOULDBLOCK) |
out = 0; |
| 342 |
{ |
stdin_read_wait = 0; |
| 343 |
out = 0; |
loop = 0; |
| 344 |
loop = 0; |
break; |
|
break; |
|
|
} |
|
|
else if (errno == EINTR) |
|
|
{ |
|
|
continue; |
|
|
} |
|
|
else |
|
|
{ |
|
|
#ifdef _DEBUG |
|
|
log_error("read(STDIN) error (%d)\n", errno); |
|
|
#endif |
|
|
loop = 0; |
|
|
break; |
|
|
} |
|
| 345 |
} |
} |
| 346 |
else if (ret == 0) // broken pipe |
} |
| 347 |
|
else |
| 348 |
|
{ |
| 349 |
|
ret = (int)read(STDIN_FILENO, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (size_t)stdin_buf_len); |
| 350 |
|
} |
| 351 |
|
if (ret < 0) |
| 352 |
|
{ |
| 353 |
|
if (errno == EAGAIN || errno == EWOULDBLOCK) |
| 354 |
{ |
{ |
| 355 |
|
out = 0; |
| 356 |
|
stdin_read_wait = 0; |
| 357 |
loop = 0; |
loop = 0; |
| 358 |
break; |
break; |
| 359 |
} |
} |
| 360 |
else |
else if (errno == EINTR) |
| 361 |
{ |
{ |
|
stdin_buf_len += ret; |
|
| 362 |
continue; |
continue; |
| 363 |
} |
} |
| 364 |
|
else |
| 365 |
|
{ |
| 366 |
|
#ifdef _DEBUG |
| 367 |
|
log_error("read(STDIN) error (%d)\n", errno); |
| 368 |
|
#endif |
| 369 |
|
loop = 0; |
| 370 |
|
break; |
| 371 |
|
} |
| 372 |
|
} |
| 373 |
|
else if (ret == 0) // broken pipe |
| 374 |
|
{ |
| 375 |
|
stdin_read_wait = 0; |
| 376 |
|
loop = 0; |
| 377 |
|
break; |
| 378 |
|
} |
| 379 |
|
else |
| 380 |
|
{ |
| 381 |
|
stdin_buf_len += ret; |
| 382 |
|
continue; |
| 383 |
} |
} |
| 384 |
} |
} |
| 385 |
} |
} |
| 388 |
#ifdef _DEBUG |
#ifdef _DEBUG |
| 389 |
for (int j = stdin_buf_offset; j < stdin_buf_len; j++) |
for (int j = stdin_buf_offset; j < stdin_buf_len; j++) |
| 390 |
{ |
{ |
| 391 |
log_common("Debug: <--[%u]\n", (stdin_buf[j] + 256) % 256); |
log_error("Debug input: <--[%u]\n", (stdin_buf[j] + 256) % 256); |
| 392 |
} |
} |
| 393 |
#endif |
#endif |
| 394 |
} |
} |
| 406 |
if (ret < 0) |
if (ret < 0) |
| 407 |
{ |
{ |
| 408 |
log_error("io_buf_conv(stdin, %d, %d, %d) error\n", stdin_buf_len, stdin_buf_offset, stdin_conv_len); |
log_error("io_buf_conv(stdin, %d, %d, %d) error\n", stdin_buf_len, stdin_buf_offset, stdin_conv_len); |
| 409 |
|
stdin_buf_len = stdin_buf_offset; // Discard invalid sequence |
| 410 |
} |
} |
| 411 |
|
|
| 412 |
|
// For debug |
| 413 |
|
#ifdef _DEBUG |
| 414 |
|
for (int j = stdin_conv_offset; j < stdin_conv_len; j++) |
| 415 |
|
{ |
| 416 |
|
log_error("Debug input_conv: <--[%u]\n", (stdin_conv[j] + 256) % 256); |
| 417 |
|
} |
| 418 |
|
#endif |
| 419 |
} |
} |
| 420 |
} |
} |
| 421 |
|
|
| 879 |
#ifdef _DEBUG |
#ifdef _DEBUG |
| 880 |
if (out != KEY_TIMEOUT && out != KEY_NULL) |
if (out != KEY_TIMEOUT && out != KEY_NULL) |
| 881 |
{ |
{ |
| 882 |
log_common("Debug: -->[0x %x]\n", out); |
log_error("Debug: -->[0x %x]\n", out); |
| 883 |
} |
} |
| 884 |
#endif |
#endif |
| 885 |
|
|
| 915 |
size_t in_bytes; |
size_t in_bytes; |
| 916 |
size_t out_bytes; |
size_t out_bytes; |
| 917 |
int ret; |
int ret; |
| 918 |
|
int in_control = 0; |
| 919 |
|
size_t i = 0; |
| 920 |
|
|
| 921 |
if (cd == NULL || p_buf == NULL || p_buf_len == NULL || p_buf_offset == NULL || p_conv == NULL || p_conv_len == NULL) |
if (cd == NULL || p_buf == NULL || p_buf_len == NULL || p_buf_offset == NULL || p_conv == NULL || p_conv_len == NULL) |
| 922 |
{ |
{ |
| 931 |
|
|
| 932 |
while (in_bytes > 0) |
while (in_bytes > 0) |
| 933 |
{ |
{ |
| 934 |
|
if ((unsigned char)(*in_buf) == KEY_CONTROL) |
| 935 |
|
{ |
| 936 |
|
if (in_control == 0) |
| 937 |
|
{ |
| 938 |
|
in_control = 1; |
| 939 |
|
i = 0; |
| 940 |
|
} |
| 941 |
|
} |
| 942 |
|
|
| 943 |
|
if (in_control) |
| 944 |
|
{ |
| 945 |
|
if (out_bytes <= 0) |
| 946 |
|
{ |
| 947 |
|
log_error("No enough free space in p_conv, conv_len=%d, conv_size=%d\n", *p_conv_len, conv_size); |
| 948 |
|
return -2; |
| 949 |
|
} |
| 950 |
|
|
| 951 |
|
*out_buf = *in_buf; |
| 952 |
|
in_buf++; |
| 953 |
|
out_buf++; |
| 954 |
|
in_bytes--; |
| 955 |
|
out_bytes--; |
| 956 |
|
|
| 957 |
|
(*p_buf_offset)++; |
| 958 |
|
*p_conv_len = (int)(conv_size - out_bytes); |
| 959 |
|
|
| 960 |
|
i++; |
| 961 |
|
if (i >= 2) |
| 962 |
|
{ |
| 963 |
|
in_control = 0; |
| 964 |
|
} |
| 965 |
|
continue; |
| 966 |
|
} |
| 967 |
|
|
| 968 |
ret = (int)iconv(cd, &in_buf, &in_bytes, &out_buf, &out_bytes); |
ret = (int)iconv(cd, &in_buf, &in_bytes, &out_buf, &out_bytes); |
| 969 |
if (ret == -1) |
if (ret == -1) |
| 970 |
{ |
{ |
| 971 |
if (errno == EINVAL) // Incomplete |
if (errno == EINVAL) // Incomplete |
| 972 |
{ |
{ |
| 973 |
#ifdef _DEBUG |
#ifdef _DEBUG |
| 974 |
log_error("iconv(inbytes=%d, outbytes=%d) error: EINVAL\n", in_bytes, out_bytes); |
log_error("iconv(inbytes=%d, outbytes=%d) error: EINVAL, in_buf[0]=%d\n", in_bytes, out_bytes, in_buf[0]); |
| 975 |
#endif |
#endif |
| 976 |
*p_buf_len = (int)(p_buf + *p_buf_len - in_buf); |
if (p_buf != in_buf) |
| 977 |
*p_buf_offset = 0; |
{ |
| 978 |
*p_conv_len = (int)(conv_size - out_bytes); |
*p_buf_len = (int)(p_buf + *p_buf_len - in_buf); |
| 979 |
memmove(p_buf, in_buf, (size_t)(*p_buf_len)); |
*p_buf_offset = 0; |
| 980 |
|
*p_conv_len = (int)(conv_size - out_bytes); |
| 981 |
|
memmove(p_buf, in_buf, (size_t)(*p_buf_len)); |
| 982 |
|
} |
| 983 |
|
|
| 984 |
break; |
break; |
| 985 |
} |
} |
| 996 |
return -2; |
return -2; |
| 997 |
} |
} |
| 998 |
|
|
| 999 |
|
// reset in_bytes when "//IGNORE" is applied |
| 1000 |
|
if (in_bytes == 0) |
| 1001 |
|
{ |
| 1002 |
|
in_bytes = (size_t)(*p_buf_len - *p_buf_offset); |
| 1003 |
|
} |
| 1004 |
|
|
| 1005 |
*out_buf = *in_buf; |
*out_buf = *in_buf; |
| 1006 |
in_buf++; |
in_buf++; |
| 1007 |
out_buf++; |
out_buf++; |
| 1026 |
|
|
| 1027 |
int io_conv_init(const char *charset) |
int io_conv_init(const char *charset) |
| 1028 |
{ |
{ |
| 1029 |
|
char tocode[32]; |
| 1030 |
|
|
| 1031 |
if (charset == NULL) |
if (charset == NULL) |
| 1032 |
{ |
{ |
| 1033 |
log_error("NULL pointer error\n"); |
log_error("NULL pointer error\n"); |
| 1039 |
strncpy(stdio_charset, charset, sizeof(stdio_charset) - 1); |
strncpy(stdio_charset, charset, sizeof(stdio_charset) - 1); |
| 1040 |
stdio_charset[sizeof(stdio_charset) - 1] = '\0'; |
stdio_charset[sizeof(stdio_charset) - 1] = '\0'; |
| 1041 |
|
|
| 1042 |
stdin_cd = iconv_open(BBS_DEFAULT_CHARSET, stdio_charset); |
snprintf(tocode, sizeof(tocode), "%s%s", BBS_DEFAULT_CHARSET, |
| 1043 |
|
(strcasecmp(stdio_charset, BBS_DEFAULT_CHARSET) == 0 ? "" : "//IGNORE")); |
| 1044 |
|
stdin_cd = iconv_open(tocode, stdio_charset); |
| 1045 |
if (stdin_cd == (iconv_t)(-1)) |
if (stdin_cd == (iconv_t)(-1)) |
| 1046 |
{ |
{ |
| 1047 |
log_error("iconv_open(%s->%s) error: %d\n", stdio_charset, BBS_DEFAULT_CHARSET, errno); |
log_error("iconv_open(%s->%s) error: %d\n", stdio_charset, tocode, errno); |
| 1048 |
return -2; |
return -2; |
| 1049 |
} |
} |
| 1050 |
stdout_cd = iconv_open(stdio_charset, BBS_DEFAULT_CHARSET); |
|
| 1051 |
|
snprintf(tocode, sizeof(tocode), "%s%s", stdio_charset, |
| 1052 |
|
(strcasecmp(BBS_DEFAULT_CHARSET, stdio_charset) == 0 ? "" : "//TRANSLIT")); |
| 1053 |
|
stdout_cd = iconv_open(tocode, BBS_DEFAULT_CHARSET); |
| 1054 |
if (stdout_cd == (iconv_t)(-1)) |
if (stdout_cd == (iconv_t)(-1)) |
| 1055 |
{ |
{ |
| 1056 |
log_error("iconv_open(%s->%s) error: %d\n", BBS_DEFAULT_CHARSET, stdio_charset, errno); |
log_error("iconv_open(%s->%s) error: %d\n", BBS_DEFAULT_CHARSET, tocode, errno); |
| 1057 |
iconv_close(stdin_cd); |
iconv_close(stdin_cd); |
| 1058 |
return -2; |
return -2; |
| 1059 |
} |
} |