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

Diff of /lbbs/src/io.c

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

Revision 1.73 by sysadm, Wed Dec 17 03:56:39 2025 UTC Revision 1.78 by sysadm, Thu Dec 18 11:37:08 2025 UTC
# Line 88  int io_init(void) Line 88  int io_init(void)
88                          log_error("epoll_ctl(STDIN_FILENO) error (%d)\n", errno);                          log_error("epoll_ctl(STDIN_FILENO) error (%d)\n", errno);
89                          if (close(stdin_epollfd) < 0)                          if (close(stdin_epollfd) < 0)
90                          {                          {
91                                  log_error("close(stdin_epollfd) error (%d)\n");                                  log_error("close(stdin_epollfd) error (%d)\n", errno);
92                          }                          }
93                          stdin_epollfd = -1;                          stdin_epollfd = -1;
94                          return -1;                          return -1;
95                  }                  }
96    
97                  stdin_flags = fcntl(STDIN_FILENO, F_GETFL, 0);                  if ((stdin_flags = fcntl(STDIN_FILENO, F_GETFL, 0)) == -1)
98                  fcntl(STDIN_FILENO, F_SETFL, stdin_flags | O_NONBLOCK);                  {
99                            log_error("fcntl(F_GETFL) error (%d)\n", errno);
100                            if (close(stdin_epollfd) < 0)
101                            {
102                                    log_error("close(stdin_epollfd) error (%d)\n", errno);
103                            }
104                            stdin_epollfd = -1;
105                            return -1;
106                    }
107                    if ((fcntl(STDIN_FILENO, F_SETFL, stdin_flags | O_NONBLOCK)) == -1)
108                    {
109                            log_error("fcntl(F_SETFL) error (%d)\n", errno);
110                            if (close(stdin_epollfd) < 0)
111                            {
112                                    log_error("close(stdin_epollfd) error (%d)\n", errno);
113                            }
114                            stdin_epollfd = -1;
115                            return -1;
116                    }
117          }          }
118    
119          if (stdout_epollfd == -1)          if (stdout_epollfd == -1)
# Line 114  int io_init(void) Line 132  int io_init(void)
132                          log_error("epoll_ctl(STDOUT_FILENO) error (%d)\n", errno);                          log_error("epoll_ctl(STDOUT_FILENO) error (%d)\n", errno);
133                          if (close(stdout_epollfd) < 0)                          if (close(stdout_epollfd) < 0)
134                          {                          {
135                                  log_error("close(stdout_epollfd) error (%d)\n");                                  log_error("close(stdout_epollfd) error (%d)\n", errno);
136                          }                          }
137                          stdout_epollfd = -1;                          stdout_epollfd = -1;
138                          return -1;                          return -1;
139                  }                  }
140    
141                  stdout_flags = fcntl(STDOUT_FILENO, F_GETFL, 0);                  if ((stdout_flags = fcntl(STDOUT_FILENO, F_GETFL, 0)) == -1)
142                  fcntl(STDOUT_FILENO, F_SETFL, stdout_flags | O_NONBLOCK);                  {
143                            log_error("fcntl(F_GETFL) error (%d)\n", errno);
144                            if (close(stdout_epollfd) < 0)
145                            {
146                                    log_error("close(stdout_epollfd) error (%d)\n", errno);
147                            }
148                            stdout_epollfd = -1;
149                            return -1;
150                    }
151                    if ((fcntl(STDOUT_FILENO, F_SETFL, stdout_flags | O_NONBLOCK)) == -1)
152                    {
153                            log_error("fcntl(F_SETFL) error (%d)\n", errno);
154                            if (close(stdout_epollfd) < 0)
155                            {
156                                    log_error("close(stdout_epollfd) error (%d)\n", errno);
157                            }
158                            stdout_epollfd = -1;
159                            return -1;
160                    }
161          }          }
162  #else  #else
163          if (stdin_flags == 0)          if (stdin_flags == 0)
164          {          {
165                  stdin_flags = fcntl(STDIN_FILENO, F_GETFL, 0);                  if ((stdin_flags = fcntl(STDIN_FILENO, F_GETFL, 0)) == -1)
166                  fcntl(STDIN_FILENO, F_SETFL, stdin_flags | O_NONBLOCK);                  {
167                            log_error("fcntl(F_GETFL) error (%d)\n", errno);
168                            return -1;
169                    }
170                    if ((fcntl(STDIN_FILENO, F_SETFL, stdin_flags | O_NONBLOCK)) == -1)
171                    {
172                            log_error("fcntl(F_SETFL) error (%d)\n", errno);
173                            return -1;
174                    }
175          }          }
176    
177          if (stdout_flags == 0)          if (stdout_flags == 0)
178          {          {
179                  stdout_flags = fcntl(STDOUT_FILENO, F_GETFL, 0);                  if ((stdout_flags = fcntl(STDOUT_FILENO, F_GETFL, 0)) == -1)
180                  fcntl(STDOUT_FILENO, F_SETFL, stdout_flags | O_NONBLOCK);                  {
181                            log_error("fcntl(F_GETFL) error (%d)\n", errno);
182                            return -1;
183                    }
184                    if ((fcntl(STDOUT_FILENO, F_SETFL, stdout_flags | O_NONBLOCK)) == -1)
185                    {
186                            log_error("fcntl(F_SETFL) error (%d)\n", errno);
187                            return -1;
188                    }
189          }          }
190  #endif  #endif
191    
# Line 150  void io_cleanup(void) Line 202  void io_cleanup(void)
202    
203                  if (close(stdin_epollfd) < 0)                  if (close(stdin_epollfd) < 0)
204                  {                  {
205                          log_error("close(stdin_epollfd) error (%d)\n");                          log_error("close(stdin_epollfd) error (%d)\n", errno);
206                  }                  }
207                  stdin_epollfd = -1;                  stdin_epollfd = -1;
208          }          }
# Line 162  void io_cleanup(void) Line 214  void io_cleanup(void)
214    
215                  if (close(stdout_epollfd) < 0)                  if (close(stdout_epollfd) < 0)
216                  {                  {
217                          log_error("close(stdout_epollfd) error (%d)\n");                          log_error("close(stdout_epollfd) error (%d)\n", errno);
218                  }                  }
219                  stdout_epollfd = -1;                  stdout_epollfd = -1;
220          }          }
# Line 193  int prints(const char *format, ...) Line 245  int prints(const char *format, ...)
245    
246          if (ret > 0)          if (ret > 0)
247          {          {
248                  if (stdout_buf_len + ret > OUTPUT_BUF_SIZE)                  int written = (ret >= sizeof(buf) ? (int)sizeof(buf) - 1 : ret);
249    
250                    if (stdout_buf_len + written > OUTPUT_BUF_SIZE)
251                  {                  {
252                          iflush();                          iflush();
253                  }                  }
254    
255                  if (stdout_buf_len + ret <= OUTPUT_BUF_SIZE)                  if (stdout_buf_len + written <= OUTPUT_BUF_SIZE)
256                  {                  {
257                          memcpy(stdout_buf + stdout_buf_len, buf, (size_t)ret);                          memcpy(stdout_buf + stdout_buf_len, buf, (size_t)written);
258                          stdout_buf_len += ret;                          stdout_buf_len += written;
259                  }                  }
260                  else                  else
261                  {                  {
262                          errno = EAGAIN;                          errno = EAGAIN;
263                          ret = (OUTPUT_BUF_SIZE - stdout_buf_len - ret);                          int need = stdout_buf_len + ret - OUTPUT_BUF_SIZE;
264                          log_error("Output buffer is full, additional %d is required\n", ret);                          log_error("Output buffer is full, additional %d is required\n", need);
265                            ret = -1;
266                  }                  }
267          }          }
268    
# Line 216  int prints(const char *format, ...) Line 271  int prints(const char *format, ...)
271    
272  int outc(char c)  int outc(char c)
273  {  {
274          int ret;          int ret = 0;
275    
276          if (stdout_buf_len + 1 > OUTPUT_BUF_SIZE)          if (stdout_buf_len + 1 > OUTPUT_BUF_SIZE)
277          {          {
# Line 246  int iflush(void) Line 301  int iflush(void)
301  #endif  #endif
302    
303          int nfds;          int nfds;
         int retry;  
304          int ret = 0;          int ret = 0;
305    
306          // Retry wait / flush for at most 3 times          // Retry wait / flush for at most 3 times
307          retry = 3;          for (int retry = 3; retry > 0 && !SYS_server_exit; retry--)
         while (retry > 0 && !SYS_server_exit)  
308          {          {
                 retry--;  
   
309  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
310                  nfds = epoll_wait(stdout_epollfd, events, MAX_EVENTS, 100); // 0.1 second                  nfds = epoll_wait(stdout_epollfd, events, MAX_EVENTS, 100); // 0.1 second
311                  ret = nfds;                  ret = nfds;
# Line 286  int iflush(void) Line 337  int iflush(void)
337                  for (int i = 0; i < nfds; i++)                  for (int i = 0; i < nfds; i++)
338                  {                  {
339  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
340                          if (events[i].data.fd == STDOUT_FILENO)                          if (events[i].data.fd == STDOUT_FILENO && (events[i].events & (EPOLLHUP | EPOLLERR)))
341    #else
342                            if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & (POLLHUP | POLLERR)))
343    #endif
344                            {
345    #ifdef HAVE_SYS_EPOLL_H
346                                    log_debug("STDOUT error events (%d)\n", events[i].events);
347    #else
348                                    log_debug("STDOUT error events (%d)\n", pfds[i].revents);
349    #endif
350                                    retry = 0;
351                                    break;
352                            }
353    
354    #ifdef HAVE_SYS_EPOLL_H
355                            if (events[i].data.fd == STDOUT_FILENO && (events[i].events & EPOLLOUT))
356  #else  #else
357                          if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT))                          if (pfds[i].fd == STDOUT_FILENO && (pfds[i].revents & POLLOUT))
358  #endif  #endif
# Line 308  int iflush(void) Line 374  int iflush(void)
374                                                  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));
375                                                  if (ret == SSH_ERROR)                                                  if (ret == SSH_ERROR)
376                                                  {                                                  {
377  #ifdef _DEBUG                                                          log_debug("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session));
                                                         log_error("ssh_channel_write() error: %s\n", ssh_get_error(SSH_session));  
 #endif  
378                                                          retry = 0;                                                          retry = 0;
379                                                          break;                                                          break;
380                                                  }                                                  }
# Line 331  int iflush(void) Line 395  int iflush(void)
395                                                  }                                                  }
396                                                  else                                                  else
397                                                  {                                                  {
398  #ifdef _DEBUG                                                          log_debug("write(STDOUT) error (%d)\n", errno);
                                                         log_error("write(STDOUT) error (%d)\n", errno);  
 #endif  
399                                                          retry = 0;                                                          retry = 0;
400                                                          break;                                                          break;
401                                                  }                                                  }
# Line 394  int igetch(int timeout) Line 456  int igetch(int timeout)
456                  {                  {
457                          if (SSH_v2 && ssh_channel_is_closed(SSH_channel))                          if (SSH_v2 && ssh_channel_is_closed(SSH_channel))
458                          {                          {
459  #ifdef _DEBUG                                  log_debug("SSH channel is closed\n");
                                 log_error("SSH channel is closed\n");  
 #endif  
460                                  loop = 0;                                  loop = 0;
461                                  break;                                  break;
462                          }                          }
# Line 435  int igetch(int timeout) Line 495  int igetch(int timeout)
495                                  for (int i = 0; i < nfds; i++)                                  for (int i = 0; i < nfds; i++)
496                                  {                                  {
497  #ifdef HAVE_SYS_EPOLL_H  #ifdef HAVE_SYS_EPOLL_H
498                                          if (events[i].data.fd == STDIN_FILENO)                                          if (events[i].data.fd == STDIN_FILENO && (events[i].events & (EPOLLHUP | EPOLLERR)))
499    #else
500                                            if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & (POLLHUP | POLLERR)))
501    #endif
502                                            {
503    #ifdef HAVE_SYS_EPOLL_H
504                                                    log_debug("STDIN error events (%d)\n", events[i].events);
505    #else
506                                                    log_debug("STDIN error events (%d)\n", pfds[i].revents);
507    #endif
508                                                    loop = 0;
509                                                    break;
510                                            }
511    
512    #ifdef HAVE_SYS_EPOLL_H
513                                            if (events[i].data.fd == STDIN_FILENO && (events[i].events & EPOLLIN))
514  #else  #else
515                                          if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))                                          if (pfds[i].fd == STDIN_FILENO && (pfds[i].revents & POLLIN))
516  #endif  #endif
# Line 454  int igetch(int timeout) Line 529  int igetch(int timeout)
529                                                  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);
530                                                  if (ret == SSH_ERROR)                                                  if (ret == SSH_ERROR)
531                                                  {                                                  {
532  #ifdef _DEBUG                                                          log_debug("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));  
 #endif  
533                                                          loop = 0;                                                          loop = 0;
534                                                          break;                                                          break;
535                                                  }                                                  }
# Line 493  int igetch(int timeout) Line 566  int igetch(int timeout)
566                                                  }                                                  }
567                                                  else                                                  else
568                                                  {                                                  {
569  #ifdef _DEBUG                                                          log_debug("read(STDIN) error (%d)\n", errno);
                                                         log_error("read(STDIN) error (%d)\n", errno);  
 #endif  
570                                                          loop = 0;                                                          loop = 0;
571                                                          break;                                                          break;
572                                                  }                                                  }
# Line 518  int igetch(int timeout) Line 589  int igetch(int timeout)
589  #ifdef _DEBUG  #ifdef _DEBUG
590                          for (int j = stdin_buf_offset; j < stdin_buf_len; j++)                          for (int j = stdin_buf_offset; j < stdin_buf_len; j++)
591                          {                          {
592                                  log_error("Debug input: <--[%u]\n", (stdin_buf[j] + 256) % 256);                                  log_debug("input: <--[%u]\n", (stdin_buf[j] + 256) % 256);
593                          }                          }
594  #endif  #endif
595                  }                  }
# Line 536  int igetch(int timeout) Line 607  int igetch(int timeout)
607  #ifdef _DEBUG  #ifdef _DEBUG
608                          for (int j = stdin_conv_offset; j < stdin_conv_len; j++)                          for (int j = stdin_conv_offset; j < stdin_conv_len; j++)
609                          {                          {
610                                  log_error("Debug input_conv: <--[%u]\n", (stdin_conv[j] + 256) % 256);                                  log_debug("input_conv: <--[%u]\n", (stdin_conv[j] + 256) % 256);
611                          }                          }
612  #endif  #endif
613                  }                  }
# Line 1002  int igetch(int timeout) Line 1073  int igetch(int timeout)
1073  #ifdef _DEBUG  #ifdef _DEBUG
1074          if (out != KEY_TIMEOUT && out != KEY_NULL)          if (out != KEY_TIMEOUT && out != KEY_NULL)
1075          {          {
1076                  log_error("Debug: -->[0x %x]\n", out);                  log_debug("output: -->[0x %x]\n", out);
1077          }          }
1078  #endif  #endif
1079    
# Line 1042  int io_buf_conv(iconv_t cd, char *p_buf, Line 1113  int io_buf_conv(iconv_t cd, char *p_buf,
1113          size_t i = 0;          size_t i = 0;
1114          int skip_current = 0;          int skip_current = 0;
1115    
1116          if (cd == NULL || p_buf == NULL || p_buf_len == NULL || p_buf_offset == NULL || p_conv == NULL || p_conv_len == NULL)          if (cd == (iconv_t)(-1) || p_buf == NULL || p_buf_len == NULL || p_buf_offset == NULL || p_conv == NULL || p_conv_len == NULL)
1117          {          {
1118                  log_error("NULL pointer error\n");                  log_error("NULL pointer error\n");
1119                  return -1;                  return -1;
# Line 1096  int io_buf_conv(iconv_t cd, char *p_buf, Line 1167  int io_buf_conv(iconv_t cd, char *p_buf,
1167                  {                  {
1168                          if (errno == EINVAL) // Incomplete                          if (errno == EINVAL) // Incomplete
1169                          {                          {
1170  #ifdef _DEBUG                                  log_debug("iconv(inbytes=%zu, outbytes=%zu) error: EINVAL, in_buf[0]=%d\n",
1171                                  log_error("iconv(inbytes=%d, outbytes=%d) error: EINVAL, in_buf[0]=%d\n", in_bytes, out_bytes, in_buf[0]);                                                    in_bytes, out_bytes, (unsigned char)in_buf[0]);
 #endif  
1172                                  if (p_buf != in_buf)                                  if (p_buf != in_buf)
1173                                  {                                  {
1174                                          *p_buf_len -= (int)(in_buf - p_buf);                                          *p_buf_len -= (int)(in_buf - p_buf);
# Line 1111  int io_buf_conv(iconv_t cd, char *p_buf, Line 1181  int io_buf_conv(iconv_t cd, char *p_buf,
1181                          }                          }
1182                          else if (errno == E2BIG)                          else if (errno == E2BIG)
1183                          {                          {
1184                                  log_error("iconv(inbytes=%d, outbytes=%d) error: E2BIG\n", in_bytes, out_bytes);                                  log_error("iconv(inbytes=%zu, outbytes=%zu) error: E2BIG\n", in_bytes, out_bytes);
1185                                  return -1;                                  return -1;
1186                          }                          }
1187                          else if (errno == EILSEQ)                          else if (errno == EILSEQ)
1188                          {                          {
1189                                  if (in_bytes > out_bytes || out_bytes <= 0)                                  if (in_bytes > out_bytes || out_bytes <= 0)
1190                                  {                                  {
1191                                          log_error("iconv(inbytes=%d, outbytes=%d) error: EILSEQ and E2BIG\n", in_bytes, out_bytes);                                          log_error("iconv(inbytes=%zu, outbytes=%zu) error: EILSEQ\n", in_bytes, out_bytes);
1192                                          return -2;                                          return -2;
1193                                  }                                  }
1194    
# Line 1126  int io_buf_conv(iconv_t cd, char *p_buf, Line 1196  int io_buf_conv(iconv_t cd, char *p_buf,
1196                                  if (in_bytes == 0)                                  if (in_bytes == 0)
1197                                  {                                  {
1198                                          in_bytes = (size_t)(*p_buf_len - *p_buf_offset);                                          in_bytes = (size_t)(*p_buf_len - *p_buf_offset);
1199  #ifdef _DEBUG                                          log_debug("Reset in_bytes from 0 to %zu\n", in_bytes);
                                         log_error("Reset in_bytes from 0 to %d\n", in_bytes);  
 #endif  
1200                                  }                                  }
1201    
1202  #ifdef _DEBUG                                  log_debug("iconv(in_bytes=%zu, out_bytes=%zu) error: EILSEQ, in_buf[0]=%d\n",
1203                                  log_error("iconv(in_bytes=%d, out_bytes=%d) error: EILSEQ, in_buf[0]=%d\n",                                                    in_bytes, out_bytes, (unsigned char)in_buf[0]);
                                                   in_bytes, out_bytes, in_buf[0]);  
 #endif  
1204                                  skip_current = 1;                                  skip_current = 1;
1205                          }                          }
1206                          else // something strange                          else // something strange
1207                          {                          {
1208  #ifdef _DEBUG                                  log_debug("iconv(in_bytes=%zu, out_bytes=%zu) error: %d, in_buf[0]=%d\n",
1209                                  log_error("iconv(in_bytes=%d, out_bytes=%d) error: %d, in_buf[0]=%d\n",                                                    in_bytes, out_bytes, errno, (unsigned char)in_buf[0]);
                                                   in_bytes, out_bytes, errno, in_buf[0]);  
 #endif  
1210                                  *p_buf_offset = (int)(in_buf - p_buf);                                  *p_buf_offset = (int)(in_buf - p_buf);
1211                                  *p_conv_len = (int)(conv_size - out_bytes);                                  *p_conv_len = (int)(conv_size - out_bytes);
1212                                  skip_current = 1;                                  skip_current = 1;

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