X-Git-Url: http://git.droids-corp.org/?a=blobdiff_plain;f=src%2Flib%2Fcmdline.c;h=e6662bd339105c2d72cd142a446fa3a21832027e;hb=2e949478bc20b52358b9093a8d3ae88d186fb8fc;hp=481507153379a9ad250715a4b6b967d9cdb5fc0e;hpb=369246d596ca24a9193b730c5aba2ec8837fb134;p=libcmdline.git

diff --git a/src/lib/cmdline.c b/src/lib/cmdline.c
index 4815071..e6662bd 100644
--- a/src/lib/cmdline.c
+++ b/src/lib/cmdline.c
@@ -1,3 +1,37 @@
+/*-
+ * Copyright (c) <2010>, Intel Corporation
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ *
+ * - 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.
+ *
+ * - Neither the name of Intel Corporation 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 THE COPYRIGHT HOLDERS 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
+ * COPYRIGHT OWNER 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.
+ */
+
 /*
  * Copyright (c) 2009, Olivier MATZ <zer0@droids-corp.org>
  * All rights reserved.
@@ -32,23 +66,14 @@
 #include <stdarg.h>
 #include <inttypes.h>
 #include <fcntl.h>
-
-#include <sys/socket.h>
+#include <termios.h>
 #include <netinet/in.h>
-#include <sys/socket.h>
-#include <sys/un.h>
 
 #include "cmdline_parse.h"
 #include "cmdline_rdline.h"
 #include "cmdline.h"
 
-
-#define PROMPT_SUFFIX "> "
-
-
-/**********************/
-
-void
+static void
 cmdline_valid_buffer(struct rdline *rdl, const char *buf,
 		     __attribute__((unused)) unsigned int size)
 {
@@ -63,7 +88,7 @@ cmdline_valid_buffer(struct rdline *rdl, const char *buf,
 		cmdline_printf(cl, "Bad arguments\n");
 }
 
-int
+static int
 cmdline_complete_buffer(struct rdline *rdl, const char *buf,
 			char *dstbuf, unsigned int dstsize,
 			int *state)
@@ -72,13 +97,16 @@ cmdline_complete_buffer(struct rdline *rdl, const char *buf,
 	return cmdline_complete(cl, buf, state, dstbuf, dstsize);
 }
 
-void
+int
 cmdline_write_char(struct rdline *rdl, char c)
 {
+	int ret = -1;
 	struct cmdline *cl = rdl->opaque;
-	if (cl->s_out >= 0) {
-		write(cl->s_out, &c, 1);
-	}
+
+	if (cl->s_out >= 0)
+		ret = write(cl->s_out, &c, 1);
+
+	return ret;
 }
 
 
@@ -124,14 +152,33 @@ void
 cmdline_printf(const struct cmdline *cl, const char *fmt, ...)
 {
 	va_list ap;
-	char buffer[BUFSIZ];
 
+#ifdef _GNU_SOURCE
+	if (cl->s_out < 0)
+		return;
 	va_start(ap, fmt);
-	vsnprintf(buffer, 512, fmt, ap);
+	vdprintf(cl->s_out, fmt, ap);
 	va_end(ap);
-	if (cl->s_out >= 0) {
-		write(cl->s_out, buffer, strlen(buffer));
-	}
+#else
+	int ret;
+	char *buf;
+
+	if (cl->s_out < 0)
+		return;
+
+	buf = malloc(BUFSIZ);
+	if (buf == NULL)
+		return;
+	va_start(ap, fmt);
+	ret = vsnprintf(buf, BUFSIZ, fmt, ap);
+	va_end(ap);
+	if (ret < 0)
+		return;
+	if (ret >= BUFSIZ)
+		ret = BUFSIZ - 1;
+	write(cl->s_out, buf, ret);
+	free(buf);
+#endif
 }
 
 int
@@ -141,11 +188,10 @@ cmdline_in(struct cmdline *cl, const char *buf, int size)
 	int ret = 0;
 	int i, same;
 
-	/* XXX use defines instead of hardcoded values */
 	for (i=0; i<size; i++) {
 		ret = rdline_char_in(&cl->rdl, buf[i]);
 
-		if (ret == 1) {
+		if (ret == RDLINE_RES_VALIDATED) {
 			buffer = rdline_get_buffer(&cl->rdl);
 			history = rdline_get_history_item(&cl->rdl, 0);
 			if (history) {
@@ -158,8 +204,31 @@ cmdline_in(struct cmdline *cl, const char *buf, int size)
 				rdline_add_history(&cl->rdl, buffer);
 			rdline_newline(&cl->rdl, cl->prompt);
 		}
-		else if (ret == -2)
+		else if (ret == RDLINE_RES_EOF)
+			return -1;
+		else if (ret == RDLINE_RES_EXITED)
 			return -1;
 	}
 	return i;
 }
+
+void
+cmdline_quit(struct cmdline *cl)
+{
+	rdline_quit(&cl->rdl);
+}
+
+void
+cmdline_interact(struct cmdline *cl)
+{
+	char c;
+
+	c = -1;
+	while (1) {
+		if (read(cl->s_in, &c, 1) < 0)
+			break;
+		if (cmdline_in(cl, &c, 1) < 0)
+			break;
+	}
+	cmdline_free(cl);
+}