--- libaitcli/src/aitcli.c	2013/11/20 16:04:42	1.9.2.1
+++ libaitcli/src/aitcli.c	2013/11/20 16:18:34	1.9.2.2
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: aitcli.c,v 1.9.2.1 2013/11/20 16:04:42 misho Exp $
+* $Id: aitcli.c,v 1.9.2.2 2013/11/20 16:18:34 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -1267,10 +1267,11 @@ cliInitLine(linebuffer_t * __restrict cli_buffer)
  *
  * @cli_buffer = CLI buffer
  * @timeout = Session timeout (-1 infinit)
+ * @cmd_name = If timeout reached, we should call with this cmd_name (default name is "exit")
  * return: NULL if error or !=NULL readed line, must be e_free after use!
 */
 char *
-cliReadLine(linebuffer_t * __restrict cli_buffer, int timeout)
+cliReadLine(linebuffer_t * __restrict cli_buffer, int timeout, const char *cmd_name)
 {
 	int code, readLen, ret;
 	register int i;
@@ -1289,8 +1290,13 @@ cliReadLine(linebuffer_t * __restrict cli_buffer, int 
 	printfCR(cli_buffer, 1);
 	while (42) {
 		if ((ret = poll(&fds, 1, timeout)) < 1) {
-			LOGERR;
-			return ret ? str : "";
+			if (!ret) {
+				if (str)
+					e_free(str);
+				str = e_strdup(cmd_name ? cmd_name : "exit");
+			} else
+				LOGERR;
+			return str;
 		}
 
 		memset(buf, 0, sizeof buf);
@@ -1345,11 +1351,12 @@ recheck:
  * @csHistFile = History file name
  * @sock = client socket
  * @timeout = Session timeout (-1 infinit)
+ * @cmd_name = If timeout reached, we should call with this cmd_name (default name is "exit")
  * return: RETCODE_ERR error, RETCODE_OK ok
 */
 int
 cliNetLoop(linebuffer_t * __restrict cli_buffer, const char *csHistFile, 
-		int sock, int timeout)
+		int sock, int timeout, const char *cmd_name)
 {
 	u_char buf[BUFSIZ];
 	int pid, stat, pty, r, s, alen, flg, attrlen = 0, ret = 0;
@@ -1368,7 +1375,7 @@ cliNetLoop(linebuffer_t * __restrict cli_buffer, const
 			} else
 				close(sock);
 
-			ret = cliLoop(cli_buffer, csHistFile, timeout) < 0 ? 1 : 0;
+			ret = cliLoop(cli_buffer, csHistFile, timeout, cmd_name) < 0 ? 1 : 0;
 			cliEnd(cli_buffer);
 
 			_exit(ret);
@@ -1462,10 +1469,12 @@ cliNetLoop(linebuffer_t * __restrict cli_buffer, const
  * @cli_buffer = CLI buffer
  * @csHistFile = History file name
  * @timeout = Session timeout (-1 infinit)
+ * @cmd_name = If timeout reached, we should call with this cmd_name (default name is "exit")
  * return: RETCODE_ERR error, RETCODE_OK ok
 */
 int
-cliLoop(linebuffer_t * __restrict cli_buffer, const char *csHistFile, int timeout)
+cliLoop(linebuffer_t * __restrict cli_buffer, const char *csHistFile, 
+		int timeout, const char *cmd_name)
 {
 	char *line, *s, *t, **app, *items[MAX_PROMPT_ITEMS];
 	register int i;
@@ -1479,7 +1488,7 @@ cliLoop(linebuffer_t * __restrict cli_buffer, const ch
 		return RETCODE_ERR;
 
 	do {
-		line = cliReadLine(cli_buffer, timeout);
+		line = cliReadLine(cli_buffer, timeout, cmd_name);
 		if (!line) {
 			printfNL(cli_buffer, 0);
 			break;