ansh/src/daemon2.c - diff

Return to daemon2.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / ansh / src

Diff for /ansh/src/daemon2.c between versions 1.1 and 1.5

version 1.1, 2011/10/04 22:37:46	version 1.5, 2015/05/19 23:25:30
Line 5	Line 5
* $Author$	* $Author$
* $Id$	* $Id$
*	*
*************************************************************************/	*************************************************************************
	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 <info@elwix.org>

	Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2013
	by Michael Pounov <misho@elwix.org>. 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 <misho@elwix.org>
	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 "global.h"
#include "anshd.h"	#include "anshd.h"

Line 22 pktTx(sched_task_t *task)	Line 59 pktTx(sched_task_t *task)
if (!(proc = TASK_ARG(task)))	if (!(proc = TASK_ARG(task)))
return (void*) -1;	return (void*) -1;

if ((wlen = pktSend(TASK_FD(task), proc->proc_id, proc->proc_flg, proc->proc_buf_[FD2NET],	if ((wlen = pktSend(TASK_FD(task), ++proc->proc_seq, proc->proc_flg, Crypted,
proc->proc_rlen_[FD2NET], &proc->proc_ea)) != ANSH_FLG_ERR) {	proc->proc_buf_[FD2NET], proc->proc_rlen_[FD2NET],
	&proc->proc_ea)) != ANSH_FLG_ERR) {
proc->proc_flg = ANSH_FLG_OK;	proc->proc_flg = ANSH_FLG_OK;
proc->proc_rlen_[FD2NET] = 0;	proc->proc_rlen_[FD2NET] = 0;
}	}
Line 41 pktRx(sched_task_t *task)	Line 79 pktRx(sched_task_t *task)
struct tagProc *proc = NULL;	struct tagProc *proc = NULL;
char ret;	char ret;
u_short *b;	u_short *b;
	u_int seq;

FTRACE(3);	FTRACE(3);

rlen = bpfLEN;	rlen = bpfLEN;
if (!(buf = malloc(rlen)))	if (!(buf = e_malloc(rlen)))
goto end;	goto end;
	else
	memset(buf, 0, rlen);

if ((ret = pktRecv(TASK_FD(task), buf, &rlen, &eth)) == ANSH_FLG_ERR)	if ((ret = pktRecv(TASK_FD(task), &seq, &Crypted, buf, &rlen, &eth)) == ANSH_FLG_ERR)
goto end;	goto end;
VERB(5) LOG("Received %d bytes", rlen);	VERB(5) LOG("Received %d bytes", rlen);
	if (!(ret & ANSH_FLG_CPOUT))
	goto end;

/* packet is ok find active session */	/* packet is ok find active session */
SLIST_FOREACH(proc, &pH, proc_next)	SLIST_FOREACH(proc, &pH, proc_next)
Line 59 pktRx(sched_task_t *task)	Line 102 pktRx(sched_task_t *task)
break;	break;
}	}
/* not found in sessions, drop packet */	/* not found in sessions, drop packet */
if (n != ANSH_CODE)	if (n != ANSH_CODE) {
	proc = NULL;
goto end;	goto end;
	}

switch (ret) {	switch (ret) {
case ANSH_FLG_EOF:	case ANSH_FLG_EOF:
case ANSH_FLG_CPOUT:	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;	break;
case ANSH_FLG_WINZ:	case ANSH_FLG_WINZ:
b = (u_short*) buf;	b = (u_short*) buf;
ioChgWinPTY(proc->proc_pty, ntohs(buf[0]), ntohs(buf[1]), ntohs(buf[2]), ntohs(buf[3]));	ioChgWinPTY(proc->proc_pty, ntohs(b[0]), ntohs(b[1]), ntohs(b[2]), ntohs(b[3]));
/* if not started login, lets start & go! */	/* if not started login, lets start & go! */
if (!proc->proc_pid) {	if (!proc->proc_pid) {
memcpy(&proc->proc_ea.octet, &eth.ether_shost, ETHER_ADDR_LEN);	memcpy(&proc->proc_ea, &eth.ether_shost, ETHER_ADDR_LEN);
spawnLogin(task, proc);	spawnLogin(task, proc);
}	}
default:	default:
Line 82 pktRx(sched_task_t *task)	Line 133 pktRx(sched_task_t *task)
proc->proc_rlen_[NET2FD] = rlen;	proc->proc_rlen_[NET2FD] = rlen;
memset(proc->proc_buf_[NET2FD], 0, proc->proc_blen);	memset(proc->proc_buf_[NET2FD], 0, proc->proc_blen);
memcpy(proc->proc_buf_[NET2FD], buf, proc->proc_rlen_[NET2FD]);	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:	end:
free(buf);	e_free(buf);
schedRead(TASK_ROOT(task), pktRx, NULL, proc ? proc->proc_sock : TASK_FD(task));	schedRead(TASK_ROOT(task), pktRx, NULL, proc ? proc->proc_sock : TASK_FD(task), NULL, 0);
return NULL;	return NULL;
}	}

Line 93 void *	Line 144 void *
fdTx(sched_task_t *task)	fdTx(sched_task_t *task)
{	{
struct tagProc *proc;	struct tagProc *proc;
	struct timespec ts = { 0 };
int wlen;	int wlen;

FTRACE(3);	FTRACE(3);
Line 102 fdTx(sched_task_t *task)	Line 154 fdTx(sched_task_t *task)
return (void*) -1;	return (void*) -1;

/* if != ANSH_FLG_CPOUT isnt received from client */	/* if != ANSH_FLG_CPOUT isnt received from client */
if (proc->proc_flg != ANSH_FLG_CPOUT)	if (proc->proc_flg != ANSH_FLG_CPOUT \|\| !proc->proc_pid)
return NULL;	return NULL;

if (waitpid(proc->proc_pid, &wlen, WNOHANG)) {	/* if Timeout defined, disarm timer */
schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL);	if (Timeout)
ioFreePTY(TASK_FD(task), proc->proc_ttyname);	schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, TOfunc, NULL);

proc->proc_pid = 0;
proc->proc_flg = ANSH_FLG_EOF;
proc->proc_rlen_[FD2NET] = 0;

schedWrite(TASK_ROOT(task), pktTx, proc, proc->proc_sock);
return NULL;
}

wlen = write(TASK_FD(task), proc->proc_buf_[NET2FD], proc->proc_rlen_[NET2FD]);	wlen = write(TASK_FD(task), proc->proc_buf_[NET2FD], proc->proc_rlen_[NET2FD]);
switch (wlen) {	switch (wlen) {
case -1:	case -1:
ERR("write2tty #%d - %s", errno, strerror(errno));	ERR("write2tty #%d - %s", errno, strerror(errno));
/* exit from shell and release tty */	/* exit from shell and release tty */
schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL);
waitpid(proc->proc_pid, &wlen, 0);
ioFreePTY(TASK_FD(task), proc->proc_ttyname);

proc->proc_pid = 0;
proc->proc_flg = ANSH_FLG_EOF;
proc->proc_rlen_[FD2NET] = 0;

schedWrite(TASK_ROOT(task), pktTx, proc, proc->proc_sock);
return NULL;	return NULL;
default:	default:
proc->proc_flg = ANSH_FLG_OK;	proc->proc_flg = ANSH_FLG_OK;
Line 138 fdTx(sched_task_t *task)	Line 173 fdTx(sched_task_t *task)
}	}
VERB(3) LOG("Writed %d bytes - %s", wlen, proc->proc_buf_[NET2FD]);	VERB(3) LOG("Writed %d bytes - %s", wlen, proc->proc_buf_[NET2FD]);

	/* if Timeout defined, go arm timer */
	if (Timeout) {
	ts.tv_sec = Timeout;
	schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0);
	}
return NULL;	return NULL;
}	}

Line 145 void *	Line 185 void *
fdRx(sched_task_t *task)	fdRx(sched_task_t *task)
{	{
struct tagProc *proc;	struct tagProc *proc;
	struct timespec ts = { 0 };
int rlen;	int rlen;

FTRACE(3);	FTRACE(3);
Line 152 fdRx(sched_task_t *task)	Line 193 fdRx(sched_task_t *task)
/* not found argument, drop data */	/* not found argument, drop data */
if (!(proc = TASK_ARG(task)))	if (!(proc = TASK_ARG(task)))
return (void*) -1;	return (void*) -1;
	if (!proc->proc_pid)
if (waitpid(proc->proc_pid, &rlen, WNOHANG)) {
schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL);
ioFreePTY(TASK_FD(task), proc->proc_ttyname);

proc->proc_pid = 0;
proc->proc_flg = ANSH_FLG_EOF;
proc->proc_rlen_[FD2NET] = 0;

schedWrite(TASK_ROOT(task), pktTx, proc, proc->proc_sock);
return NULL;	return NULL;
}

	/* if Timeout defined, disarm timer */
	if (Timeout)
	schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, TOfunc, NULL);

memset(proc->proc_buf_[FD2NET], 0, proc->proc_blen);	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 - ETHER_HDR_LEN + sizeof(struct ansh_hdr));
switch (rlen) {	switch (rlen) {
case -1:	case -1:
ERR("readtty #%d - %s", errno, strerror(errno));	ERR("readtty #%d - %s", errno, strerror(errno));
case 0:	case 0:
/* exit from shell and release tty */	/* exit from shell and release tty */
schedCancelby(TASK_ROOT(task), NULL, CRITERIA_FD, (void*) TASK_FD(task), NULL);
waitpid(proc->proc_pid, &rlen, 0);
VERB(3) LOG("EOF process status %d", rlen);
ioFreePTY(TASK_FD(task), proc->proc_ttyname);

proc->proc_pid = 0;
proc->proc_flg = ANSH_FLG_EOF;
proc->proc_rlen_[FD2NET] = 0;

schedWrite(TASK_ROOT(task), pktTx, proc, proc->proc_sock);
return NULL;	return NULL;
default:	default:
proc->proc_flg = ANSH_FLG_OK;	proc->proc_flg = ANSH_FLG_OK;
Line 189 fdRx(sched_task_t *task)	Line 215 fdRx(sched_task_t *task)
}	}
VERB(3) LOG("Readed %d bytes - %s", rlen, proc->proc_buf_[FD2NET]);	VERB(3) LOG("Readed %d bytes - %s", rlen, proc->proc_buf_[FD2NET]);

schedWrite(TASK_ROOT(task), pktTx, proc, proc->proc_sock);	schedCallOnce(TASK_ROOT(task), pktTx, proc, proc->proc_sock, NULL, 0);
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) {
	ts.tv_sec = Timeout;
	schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0);
	}
return NULL;	return NULL;
}	}

int	int
spawnLogin(sched_task_t task, struct tagProc proc)	spawnLogin(sched_task_t task, struct tagProc proc)
{	{
	int flg;
	struct timespec ts = { 0 };
	char str[STRSIZ] = { 0 };

FTRACE(3);	FTRACE(3);

assert(proc);	assert(proc);
Line 207 spawnLogin(sched_task_t task, struct tagProc proc)	Line 243 spawnLogin(sched_task_t task, struct tagProc proc)
ERR("ioForkPTY() #%d - %s", io_GetErrno(), io_GetError());	ERR("ioForkPTY() #%d - %s", io_GetErrno(), io_GetError());
return -1;	return -1;
case 0:	case 0:
execl("/usr/bin/login", "login", NULL);	printf("anshd ELWIX remote management system (%s)\n\n", proc->proc_ttyname);
// execl("/bin/echo", "echo", "MUUUUUUUUUUUUUUUUU", NULL);	strlcpy(str, "-hansh@", sizeof str);
	e_ether_ntoa((const ether_addr_t*) &proc->proc_ea, str + 7, 18);

	execl("/usr/bin/login", "login", str, NULL);
/* never reached */	/* never reached */
return -1;	return -1;
default:	default:
VERB(3) LOG("Parent know child pid %d - ptyfd=%d", proc->proc_pid, proc->proc_pty);	flg = fcntl(proc->proc_pty, F_GETFL);
schedRead(TASK_ROOT(task), fdRx, proc, proc->proc_pty);	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, NULL, 0);

	/* if Timeout defined, go arm timer */
	if (Timeout) {
	ts.tv_sec = Timeout;
	schedTimer(TASK_ROOT(task), TOfunc, proc, ts, NULL, 0);
	}
break;	break;
}	}

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

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