| 64 |
static int stdin_conv_offset = 0; |
static int stdin_conv_offset = 0; |
| 65 |
static int stdout_conv_offset = 0; |
static int stdout_conv_offset = 0; |
| 66 |
|
|
| 67 |
static iconv_t stdin_cd = NULL; |
static iconv_t stdin_cd = (iconv_t)(-1); |
| 68 |
static iconv_t stdout_cd = NULL; |
static iconv_t stdout_cd = (iconv_t)(-1); |
| 69 |
|
|
| 70 |
int io_init(void) |
int io_init(void) |
| 71 |
{ |
{ |
| 246 |
#endif |
#endif |
| 247 |
|
|
| 248 |
int nfds; |
int nfds; |
|
int retry; |
|
| 249 |
int ret = 0; |
int ret = 0; |
| 250 |
|
|
| 251 |
// Retry wait / flush for at most 3 times |
// Retry wait / flush for at most 3 times |
| 252 |
retry = 3; |
for (int retry = 3; retry > 0 && !SYS_server_exit; retry--) |
|
while (retry > 0 && !SYS_server_exit) |
|
| 253 |
{ |
{ |
|
retry--; |
|
|
|
|
| 254 |
#ifdef HAVE_SYS_EPOLL_H |
#ifdef HAVE_SYS_EPOLL_H |
| 255 |
nfds = epoll_wait(stdout_epollfd, events, MAX_EVENTS, 100); // 0.1 second |
nfds = epoll_wait(stdout_epollfd, events, MAX_EVENTS, 100); // 0.1 second |
| 256 |
ret = nfds; |
ret = nfds; |
| 282 |
for (int i = 0; i < nfds; i++) |
for (int i = 0; i < nfds; i++) |
| 283 |
{ |
{ |
| 284 |
#ifdef HAVE_SYS_EPOLL_H |
#ifdef HAVE_SYS_EPOLL_H |
| 285 |
if (events[i].data.fd == STDOUT_FILENO) |
if (events[i].data.fd == STDOUT_FILENO && (events[i].events & (EPOLLRDHUP | EPOLLHUP | EPOLLERR))) |
| 286 |
|
#else |
| 287 |
|
if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & (POLLRDHUP | POLLHUP | POLLERR))) |
| 288 |
|
#endif |
| 289 |
|
{ |
| 290 |
|
#ifdef _DEBUG |
| 291 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 292 |
|
log_error("STDOUT error events (%d)\n", events[i].events); |
| 293 |
|
#else |
| 294 |
|
log_error("STDOUT error events (%d)\n", pfds[i].revents); |
| 295 |
|
#endif |
| 296 |
|
#endif |
| 297 |
|
retry = 0; |
| 298 |
|
break; |
| 299 |
|
} |
| 300 |
|
|
| 301 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 302 |
|
if (events[i].data.fd == STDOUT_FILENO && (events[i].events & EPOLLOUT)) |
| 303 |
#else |
#else |
| 304 |
if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT)) |
if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT)) |
| 305 |
#endif |
#endif |
| 321 |
ret = ssh_channel_write(SSH_channel, stdout_conv + stdout_conv_offset, (uint32_t)(stdout_conv_len - stdout_conv_offset)); |
ret = ssh_channel_write(SSH_channel, stdout_conv + stdout_conv_offset, (uint32_t)(stdout_conv_len - stdout_conv_offset)); |
| 322 |
if (ret == SSH_ERROR) |
if (ret == SSH_ERROR) |
| 323 |
{ |
{ |
| 324 |
|
#ifdef _DEBUG |
| 325 |
log_error("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session)); |
log_error("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session)); |
| 326 |
|
#endif |
| 327 |
retry = 0; |
retry = 0; |
| 328 |
break; |
break; |
| 329 |
} |
} |
| 407 |
{ |
{ |
| 408 |
if (SSH_v2 && ssh_channel_is_closed(SSH_channel)) |
if (SSH_v2 && ssh_channel_is_closed(SSH_channel)) |
| 409 |
{ |
{ |
| 410 |
|
#ifdef _DEBUG |
| 411 |
log_error("SSH channel is closed\n"); |
log_error("SSH channel is closed\n"); |
| 412 |
|
#endif |
| 413 |
loop = 0; |
loop = 0; |
| 414 |
break; |
break; |
| 415 |
} |
} |
| 448 |
for (int i = 0; i < nfds; i++) |
for (int i = 0; i < nfds; i++) |
| 449 |
{ |
{ |
| 450 |
#ifdef HAVE_SYS_EPOLL_H |
#ifdef HAVE_SYS_EPOLL_H |
| 451 |
if (events[i].data.fd == STDIN_FILENO) |
if (events[i].data.fd == STDIN_FILENO && (events[i].events & (EPOLLRDHUP | EPOLLHUP | EPOLLERR))) |
| 452 |
|
#else |
| 453 |
|
if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & (POLLRDHUP | POLLHUP | POLLERR))) |
| 454 |
|
#endif |
| 455 |
|
{ |
| 456 |
|
#ifdef _DEBUG |
| 457 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 458 |
|
log_error("STDIN error events (%d)\n", events[i].events); |
| 459 |
|
#else |
| 460 |
|
log_error("STDIN error events (%d)\n", pfds[i].revents); |
| 461 |
|
#endif |
| 462 |
|
#endif |
| 463 |
|
loop = 0; |
| 464 |
|
break; |
| 465 |
|
} |
| 466 |
|
|
| 467 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 468 |
|
if (events[i].data.fd == STDIN_FILENO && (events[i].events & EPOLLIN)) |
| 469 |
#else |
#else |
| 470 |
if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN)) |
if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN)) |
| 471 |
#endif |
#endif |
| 484 |
ret = ssh_channel_read_nonblocking(SSH_channel, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (uint32_t)stdin_buf_len, 0); |
ret = ssh_channel_read_nonblocking(SSH_channel, stdin_buf + stdin_buf_len, sizeof(stdin_buf) - (uint32_t)stdin_buf_len, 0); |
| 485 |
if (ret == SSH_ERROR) |
if (ret == SSH_ERROR) |
| 486 |
{ |
{ |
| 487 |
|
#ifdef _DEBUG |
| 488 |
log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session)); |
log_error("ssh_channel_read_nonblocking() error: %s\n", ssh_get_error(SSH_session)); |
| 489 |
|
#endif |
| 490 |
loop = 0; |
loop = 0; |
| 491 |
break; |
break; |
| 492 |
} |
} |
| 1225 |
{ |
{ |
| 1226 |
log_error("iconv_open(%s->%s) error: %d\n", BBS_default_charset, tocode, errno); |
log_error("iconv_open(%s->%s) error: %d\n", BBS_default_charset, tocode, errno); |
| 1227 |
iconv_close(stdin_cd); |
iconv_close(stdin_cd); |
| 1228 |
|
stdin_cd = (iconv_t)(-1); |
| 1229 |
return -2; |
return -2; |
| 1230 |
} |
} |
| 1231 |
|
|
| 1234 |
|
|
| 1235 |
int io_conv_cleanup(void) |
int io_conv_cleanup(void) |
| 1236 |
{ |
{ |
| 1237 |
if (stdin_cd != NULL) |
if (stdin_cd != (iconv_t)(-1)) |
| 1238 |
{ |
{ |
| 1239 |
iconv_close(stdin_cd); |
iconv_close(stdin_cd); |
| 1240 |
stdin_cd = NULL; |
stdin_cd = (iconv_t)(-1); |
| 1241 |
} |
} |
| 1242 |
if (stdout_cd != NULL) |
if (stdout_cd != (iconv_t)(-1)) |
| 1243 |
{ |
{ |
| 1244 |
iconv_close(stdout_cd); |
iconv_close(stdout_cd); |
| 1245 |
stdout_cd = NULL; |
stdout_cd = (iconv_t)(-1); |
| 1246 |
} |
} |
| 1247 |
|
|
| 1248 |
return 0; |
return 0; |