--- ansh/src/daemon3.c 2011/10/04 22:37:46 1.1 +++ ansh/src/daemon3.c 2012/07/22 22:41:33 1.4 @@ -3,9 +3,46 @@ * by Michael Pounov * * $Author: misho $ - * $Id: daemon3.c,v 1.1 2011/10/04 22:37:46 misho Exp $ + * $Id: daemon3.c,v 1.4 2012/07/22 22:41:33 misho Exp $ * - *************************************************************************/ + ************************************************************************* +The ELWIX and AITNET software is distributed under the following +terms: + +All of the documentation and software included in the ELWIX and AITNET +Releases is copyrighted by ELWIX - Sofia/Bulgaria + +Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 + by Michael Pounov . All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. All advertising materials mentioning features or use of this software + must display the following acknowledgement: +This product includes software developed by Michael Pounov +ELWIX - Embedded LightWeight unIX and its contributors. +4. Neither the name of AITNET nor the names of its contributors + may be used to endorse or promote products derived from this software + without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS +OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) +HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT +LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY +OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF +SUCH DAMAGE. +*/ #include "global.h" #include "anshd.h" @@ -15,6 +52,7 @@ icmpTx(sched_task_t *task) { struct tagProc *proc; int wlen; + u_char *str; FTRACE(3); @@ -22,8 +60,17 @@ icmpTx(sched_task_t *task) if (!(proc = TASK_ARG(task))) return (void*) -1; - if ((wlen = icmpSend(TASK_FD(task), proc->proc_id, proc->proc_flg, proc->proc_buf_[FD2NET], - proc->proc_rlen_[FD2NET], &proc->proc_cli, sizeof proc->proc_cli)) != ANSH_FLG_ERR) { + if (Crypted) { + str = cryptBuffer(proc->proc_buf_[FD2NET], proc->proc_rlen_[FD2NET], Crypted); + if (str) { + memcpy(proc->proc_buf_[FD2NET], str, proc->proc_rlen_[FD2NET]); + io_free(str); + } + } + + if ((wlen = icmpSend(TASK_FD(task), ++proc->proc_seq, proc->proc_id, proc->proc_flg, Crypted, + proc->proc_buf_[FD2NET], proc->proc_rlen_[FD2NET], &proc->proc_cli, + sizeof proc->proc_cli)) != ANSH_FLG_ERR) { proc->proc_flg = ANSH_FLG_OK; proc->proc_rlen_[FD2NET] = 0; } @@ -35,22 +82,26 @@ icmpTx(sched_task_t *task) void * icmpRx(sched_task_t *task) { - u_char *buf; - struct sockaddr sa; + u_char *buf, *str; + io_sockaddr_t sa; int rlen, n = 0, salen = sizeof sa; struct tagProc *proc = NULL; char ret; u_short id, *b; + u_int seq; FTRACE(3); rlen = bpfLEN; - if (!(buf = malloc(rlen))) + if (!(buf = io_malloc(rlen))) goto end; + else + memset(buf, 0, rlen); - if ((ret = icmpRecv(TASK_FD(task), &id, buf, &rlen, &sa, (socklen_t *) &salen)) == ANSH_FLG_ERR) + if ((ret = icmpRecv(TASK_FD(task), &seq, &id, &Crypted, buf, &rlen, &sa, + (socklen_t *) &salen)) == ANSH_FLG_ERR) goto end; - VERB(5) LOG("Received %d bytes", rlen); + VERB(5) LOG("Received %d bytes %d", rlen, seq); if (!(ret & ANSH_FLG_CPOUT)) goto end; @@ -66,9 +117,23 @@ icmpRx(sched_task_t *task) goto end; } + if (Crypted) { + str = cryptBuffer(buf, rlen, Crypted); + if (str) { + memcpy(buf, str, rlen); + io_free(str); + } + } + switch (ret) { case ANSH_FLG_EOF: case ANSH_FLG_CPOUT: + if (seq <= proc->proc_seq) + goto end; + else if (seq > (proc->proc_seq + 1)) + LOG("LOST PACKET(s) detect: %d; received seq=%d - %d", + seq - proc->proc_seq + 1, seq, proc->proc_seq); + proc->proc_seq = seq; break; case ANSH_FLG_WINZ: b = (u_short*) buf; @@ -86,10 +151,10 @@ icmpRx(sched_task_t *task) proc->proc_rlen_[NET2FD] = rlen; memset(proc->proc_buf_[NET2FD], 0, proc->proc_blen); memcpy(proc->proc_buf_[NET2FD], buf, proc->proc_rlen_[NET2FD]); - schedWrite(TASK_ROOT(task), fdTx, proc, proc->proc_pty); + schedWrite(TASK_ROOT(task), fdTx, proc, proc->proc_pty, NULL, 0); end: - free(buf); - schedRead(TASK_ROOT(task), icmpRx, NULL, proc ? proc->proc_sock : TASK_FD(task)); + io_free(buf); + schedRead(TASK_ROOT(task), icmpRx, NULL, proc ? proc->proc_sock : TASK_FD(task), NULL, 0); return NULL; } @@ -97,7 +162,7 @@ void * fdTx(sched_task_t *task) { struct tagProc *proc; - struct timeval tv = { 0 }; + struct timespec ts = { 0 }; int wlen; FTRACE(3); @@ -110,36 +175,15 @@ fdTx(sched_task_t *task) if (proc->proc_flg != ANSH_FLG_CPOUT || !proc->proc_pid) return NULL; - if (waitpid(proc->proc_pid, &wlen, WNOHANG)) { - ioFreePTY(TASK_FD(task), proc->proc_ttyname); - schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL); - - proc->proc_pid = 0; - proc->proc_flg = ANSH_FLG_EOF; - proc->proc_rlen_[FD2NET] = 0; - - schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock); - return NULL; - } - /* if Timeout defined, disarm timer */ if (Timeout) - schedCancelby(TASK_ROOT(task), &TASK_ROOT(task)->root_timer, CRITERIA_CALL, TOfunc, NULL); + schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, TOfunc, NULL); wlen = write(TASK_FD(task), proc->proc_buf_[NET2FD], proc->proc_rlen_[NET2FD]); switch (wlen) { case -1: ERR("write2tty #%d - %s", errno, strerror(errno)); /* exit from shell and release tty */ - waitpid(proc->proc_pid, &wlen, 0); - ioFreePTY(TASK_FD(task), proc->proc_ttyname); - schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL); - - proc->proc_pid = 0; - proc->proc_flg = ANSH_FLG_EOF; - proc->proc_rlen_[FD2NET] = 0; - - schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock); return NULL; default: proc->proc_flg = ANSH_FLG_OK; @@ -149,8 +193,8 @@ fdTx(sched_task_t *task) /* if Timeout defined, go arm timer */ if (Timeout) { - tv.tv_sec = Timeout; - schedTimer(TASK_ROOT(task), TOfunc, proc, tv); + ts.tv_sec = Timeout; + schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0); } return NULL; } @@ -159,7 +203,7 @@ void * fdRx(sched_task_t *task) { struct tagProc *proc; - struct timeval tv = { 0 }; + struct timespec ts = { 0 }; int rlen; FTRACE(3); @@ -170,39 +214,18 @@ fdRx(sched_task_t *task) if (!proc->proc_pid) return NULL; - if (waitpid(proc->proc_pid, &rlen, WNOHANG)) { - ioFreePTY(TASK_FD(task), proc->proc_ttyname); - schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL); - - proc->proc_pid = 0; - proc->proc_flg = ANSH_FLG_EOF; - proc->proc_rlen_[FD2NET] = 0; - - schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock); - return NULL; - } - /* if Timeout defined, disarm timer */ if (Timeout) - schedCancelby(TASK_ROOT(task), &TASK_ROOT(task)->root_timer, CRITERIA_CALL, TOfunc, NULL); + schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, TOfunc, NULL); memset(proc->proc_buf_[FD2NET], 0, proc->proc_blen); - rlen = read(TASK_FD(task), proc->proc_buf_[FD2NET], proc->proc_blen); + rlen = read(TASK_FD(task), proc->proc_buf_[FD2NET], + proc->proc_blen - sizeof(struct icmp) + sizeof(struct ansh_hdr)); switch (rlen) { case -1: ERR("readtty #%d - %s", errno, strerror(errno)); case 0: /* exit from shell and release tty */ - waitpid(proc->proc_pid, &rlen, 0); - ioFreePTY(TASK_FD(task), proc->proc_ttyname); - schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL); - VERB(3) LOG("EOF process status %d", rlen); - - proc->proc_pid = 0; - proc->proc_flg = ANSH_FLG_EOF; - proc->proc_rlen_[FD2NET] = 0; - - schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock); return NULL; default: proc->proc_flg = ANSH_FLG_OK; @@ -210,13 +233,13 @@ fdRx(sched_task_t *task) } VERB(3) LOG("Readed %d bytes - %s", rlen, proc->proc_buf_[FD2NET]); - schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock); - schedRead(TASK_ROOT(task), fdRx, proc, proc->proc_pty); + schedWrite(TASK_ROOT(task), icmpTx, proc, proc->proc_sock, NULL, 0); + schedRead(TASK_ROOT(task), fdRx, proc, proc->proc_pty, NULL, 0); /* if Timeout defined, go arm timer */ if (Timeout) { - tv.tv_sec = Timeout; - schedTimer(TASK_ROOT(task), TOfunc, proc, tv); + ts.tv_sec = Timeout; + schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0); } return NULL; } @@ -225,10 +248,8 @@ int spawnLogin(sched_task_t *task, struct tagProc *proc) { int flg; - struct timeval tv = { 0 }; + struct timespec ts = { 0 }; char str[STRSIZ] = { 0 }; - struct sockaddr_in *sin; - struct sockaddr_in6 *sin6; FTRACE(3); @@ -243,13 +264,10 @@ spawnLogin(sched_task_t *task, struct tagProc *proc) printf("ansh3d ELWIX remote management system over ICMP (%s)\n\n", proc->proc_ttyname); strlcpy(str, "-hansh3@", sizeof str); - if (proc->proc_cli.sa_family == AF_INET) { - sin = (struct sockaddr_in*) &proc->proc_cli; - inet_ntop(AF_INET, &sin->sin_addr, str + 8, INET_ADDRSTRLEN); - } else if (proc->proc_cli.sa_family == AF_INET) { - sin6 = (struct sockaddr_in6*) &proc->proc_cli; - inet_ntop(AF_INET6, &sin6->sin6_addr, str + 8, INET6_ADDRSTRLEN); - } + if (proc->proc_cli.sa.sa_family == AF_INET) + inet_ntop(AF_INET, &proc->proc_cli.sin.sin_addr, str + 8, INET_ADDRSTRLEN); + else if (proc->proc_cli.sa.sa_family == AF_INET6) + inet_ntop(AF_INET6, &proc->proc_cli.sin6.sin6_addr, str + 8, INET6_ADDRSTRLEN); execl("/usr/bin/login", "login", str, NULL); /* never reached */ return -1; @@ -258,12 +276,12 @@ spawnLogin(sched_task_t *task, struct tagProc *proc) fcntl(proc->proc_pty, F_SETFL, flg | O_NONBLOCK); VERB(3) LOG("Parent know child pid %d", proc->proc_pid); - schedRead(TASK_ROOT(task), fdRx, proc, proc->proc_pty); + schedRead(TASK_ROOT(task), fdRx, proc, proc->proc_pty, NULL, 0); /* if Timeout defined, go arm timer */ if (Timeout) { - tv.tv_sec = Timeout; - schedTimer(TASK_ROOT(task), TOfunc, proc, tv); + ts.tv_sec = Timeout; + schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0); } break; }