Diff for /libaitio/src/sock.c between versions 1.5 and 1.9

version 1.5, 2013/11/22 13:49:14 version 1.9, 2013/11/25 11:46:17
Line 51  io_closeClient(sched_task_t *task) Line 51  io_closeClient(sched_task_t *task)
 {  {
         sock_cli_t *cli = (sock_cli_t*) TASK_ARG(task);          sock_cli_t *cli = (sock_cli_t*) TASK_ARG(task);
         sock_t *s = (sock_t*) cli->cli_parent;          sock_t *s = (sock_t*) cli->cli_parent;
        int stat, sock = (int) TASK_DATLEN(task);        int stat;
   
         pthread_mutex_lock(&s->sock_mtx);          pthread_mutex_lock(&s->sock_mtx);
         TAILQ_REMOVE(&s->sock_cli, cli, cli_node);          TAILQ_REMOVE(&s->sock_cli, cli, cli_node);
Line 59  io_closeClient(sched_task_t *task) Line 59  io_closeClient(sched_task_t *task)
   
         schedCancelby(s->sock_root, taskMAX, CRITERIA_ARG, cli, NULL);          schedCancelby(s->sock_root, taskMAX, CRITERIA_ARG, cli, NULL);
   
           if (*cli->cli_name)
                   ioFreePTY(cli->cli_pty, cli->cli_name);
   
         if (s->sock_type == SOCK_STREAM) {          if (s->sock_type == SOCK_STREAM) {
                shutdown(sock, SHUT_RDWR);                shutdown(cli->cli_fd, SHUT_RDWR);
                close(sock);                close(cli->cli_fd);
         }          }
         AIT_FREE_VAL(&cli->cli_buf[1]);          AIT_FREE_VAL(&cli->cli_buf[1]);
         AIT_FREE_VAL(&cli->cli_buf[0]);          AIT_FREE_VAL(&cli->cli_buf[0]);
   
         if (cli->cli_pid > 0) {          if (cli->cli_pid > 0) {
                kill(cli->cli_pid, SIGTERM);                kill(cli->cli_pid, SIGKILL);
                 while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {                  while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {
                         usleep(1000);                          usleep(1000);
                        kill(cli->cli_pid, SIGTERM);                        kill(cli->cli_pid, SIGKILL);
                 }                  }
         }          }
   
Line 123  io_acceptClient(sched_task_t *task) Line 126  io_acceptClient(sched_task_t *task)
         AIT_SET_BUFSIZ(&cli->cli_buf[1], 0, AIT_LEN(&s->sock_buf));          AIT_SET_BUFSIZ(&cli->cli_buf[1], 0, AIT_LEN(&s->sock_buf));
   
         schedRead(TASK_ROOT(task), cli->cli_func, cli, cli->cli_fd, TASK_ARG(task), 0);          schedRead(TASK_ROOT(task), cli->cli_func, cli, cli->cli_fd, TASK_ARG(task), 0);
        ioUpdTimerSocket(cli, NULL);        ioUpdTimerSocket(cli);
 end:  end:
         schedReadSelf(task);          schedReadSelf(task);
         taskExit(task, NULL);          taskExit(task, NULL);
Line 136  io_txNet(sched_task_t *task) Line 139  io_txNet(sched_task_t *task)
         sock_cli_t *cli = TASK_ARG(task);          sock_cli_t *cli = TASK_ARG(task);
         sock_t *s = (sock_t*) cli->cli_parent;          sock_t *s = (sock_t*) cli->cli_parent;
   
        ioUpdTimerSocket(cli, NULL);        ioUpdTimerSocket(cli);
   
         if (s->sock_type == SOCK_STREAM)          if (s->sock_type == SOCK_STREAM)
                 wlen = send(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task), 0);                  wlen = send(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task), 0);
Line 144  io_txNet(sched_task_t *task) Line 147  io_txNet(sched_task_t *task)
                 wlen = sendto(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task), 0,                   wlen = sendto(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task), 0, 
                                 &cli->cli_addr.sa, cli->cli_addr.sa.sa_len);                                  &cli->cli_addr.sa, cli->cli_addr.sa.sa_len);
         if (wlen < 1)          if (wlen < 1)
                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0,                 schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                                (void*) cli->cli_func, cli->cli_fd); 
   
         taskExit(task, NULL);          taskExit(task, NULL);
 }  }
Line 156  io_txPty(sched_task_t *task) Line 158  io_txPty(sched_task_t *task)
         int wlen;          int wlen;
         sock_cli_t *cli = TASK_ARG(task);          sock_cli_t *cli = TASK_ARG(task);
   
        ioUpdTimerSocket(cli, NULL);        ioUpdTimerSocket(cli);
   
         wlen = write(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task));          wlen = write(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task));
         if (wlen < 1)          if (wlen < 1)
                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0,                 schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                                (void*) cli->cli_func, cli->cli_fd); 
   
         taskExit(task, NULL);          taskExit(task, NULL);
 }  }
Line 175  io_rxNet(sched_task_t *task) Line 176  io_rxNet(sched_task_t *task)
         sockaddr_t sa;          sockaddr_t sa;
         socklen_t salen = sizeof sa.ss;          socklen_t salen = sizeof sa.ss;
   
        ioUpdTimerSocket(cli, NULL);        ioUpdTimerSocket(cli);
   
         if (s->sock_type == SOCK_STREAM)          if (s->sock_type == SOCK_STREAM)
                 rlen = recv(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[0]),                   rlen = recv(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[0]), 
Line 187  io_rxNet(sched_task_t *task) Line 188  io_rxNet(sched_task_t *task)
                         goto end;                          goto end;
         }          }
         if (rlen < 1)          if (rlen < 1)
                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0,                 schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                                (void*) cli->cli_func, cli->cli_fd); 
         else          else
                 schedEvent(TASK_ROOT(task), io_txPty, cli, cli->cli_pty,                   schedEvent(TASK_ROOT(task), io_txPty, cli, cli->cli_pty, 
                                 AIT_GET_BUF(&cli->cli_buf[0]), rlen);                                  AIT_GET_BUF(&cli->cli_buf[0]), rlen);
Line 203  io_rxPty(sched_task_t *task) Line 203  io_rxPty(sched_task_t *task)
         int rlen;          int rlen;
         sock_cli_t *cli = TASK_ARG(task);          sock_cli_t *cli = TASK_ARG(task);
   
        ioUpdTimerSocket(cli, NULL);        ioUpdTimerSocket(cli);
   
         rlen = read(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[1]), AIT_LEN(&cli->cli_buf[1]));          rlen = read(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[1]), AIT_LEN(&cli->cli_buf[1]));
         if (rlen < 1)          if (rlen < 1)
                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0,                 schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                                (void*) cli->cli_func, cli->cli_fd); 
         else          else
                 schedEvent(TASK_ROOT(task), io_txNet, cli, cli->cli_fd,                   schedEvent(TASK_ROOT(task), io_txNet, cli, cli->cli_fd, 
                                 AIT_GET_BUF(&cli->cli_buf[1]), rlen);                                  AIT_GET_BUF(&cli->cli_buf[1]), rlen);
Line 275  io_bridgeClient(sched_task_t *task) Line 274  io_bridgeClient(sched_task_t *task)
                         array_Destroy(&args);                          array_Destroy(&args);
   
                         printf("Console %s\n", cli->cli_name);                          printf("Console %s\n", cli->cli_name);
                        execv(*argv, argv);                        rlen = execv(*argv, argv);
                         _exit(rlen);
                         break;                          break;
                 default:                  default:
                         cli->cli_pid = pid;                          cli->cli_pid = pid;
Line 284  io_bridgeClient(sched_task_t *task) Line 284  io_bridgeClient(sched_task_t *task)
                                         TASK_ARG(task), 0);                                          TASK_ARG(task), 0);
                         schedRead(TASK_ROOT(task), io_rxNet, cli, cli->cli_fd,                           schedRead(TASK_ROOT(task), io_rxNet, cli, cli->cli_fd, 
                                         TASK_ARG(task), 0);                                          TASK_ARG(task), 0);
                        ioUpdTimerSocket(cli, NULL);                        ioUpdTimerSocket(cli);
                         break;                          break;
         }          }
 end:  end:
Line 406  ioCloseSocket(sock_t ** __restrict s) Line 406  ioCloseSocket(sock_t ** __restrict s)
                         AIT_FREE_VAL(&cli->cli_buf[0]);                          AIT_FREE_VAL(&cli->cli_buf[0]);
   
                         if (cli->cli_pid > 0) {                          if (cli->cli_pid > 0) {
                                kill(cli->cli_pid, SIGTERM);                                kill(cli->cli_pid, SIGKILL);
                                 while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {                                  while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {
                                         usleep(1000);                                          usleep(1000);
                                        kill(cli->cli_pid, SIGTERM);                                        kill(cli->cli_pid, SIGKILL);
                                 }                                  }
                         }                          }
   
Line 486  ioUpSocket(sock_t * __restrict s, void *arg, int timeo Line 486  ioUpSocket(sock_t * __restrict s, void *arg, int timeo
  * ioUpdTimerSocket() - Update timeout of socket   * ioUpdTimerSocket() - Update timeout of socket
  *   *
  * @c = Client socket   * @c = Client socket
  * @arg = Optional data argument  
  * return:  none   * return:  none
  */   */
 void  void
ioUpdTimerSocket(sock_cli_t * __restrict c, void *arg)ioUpdTimerSocket(sock_cli_t * __restrict c)
 {  {
         sock_t *s;          sock_t *s;
   
Line 499  ioUpdTimerSocket(sock_cli_t * __restrict c, void *arg) Line 498  ioUpdTimerSocket(sock_cli_t * __restrict c, void *arg)
         else          else
                 s = c->cli_parent;                  s = c->cli_parent;
   
        schedCancelby(s->sock_root, taskTIMER, CRITERIA_DATLEN, (void*) c->cli_fd, NULL);        schedCancelby(s->sock_root, taskTIMER, CRITERIA_ARG, c, NULL);
        schedTimer(s->sock_root, io_closeClient, c, s->sock_timeout, arg, c->cli_fd);        schedTimer(s->sock_root, io_closeClient, c, s->sock_timeout, NULL, 0);
 }  }
   
 /*  /*
Line 519  ioCloseClient(sock_cli_t * __restrict c) Line 518  ioCloseClient(sock_cli_t * __restrict c)
         else          else
                 s = c->cli_parent;                  s = c->cli_parent;
   
        return !schedEvent(s->sock_root, io_closeClient, c, 0, NULL, c->cli_fd);        return !schedEvent(s->sock_root, io_closeClient, c, 0, NULL, 0);
 }  }
   
 /*  /*

Removed from v.1.5  
changed lines
  Added in v.1.9


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>