From 27e1e910147465ad240a3e4393bbd4312937fed5 Mon Sep 17 00:00:00 2001
From: "Andrew J. Schorr" <aschorr@telemetry-investments.com>
Date: Mon, 1 Jul 2013 22:04:03 -0400
Subject: Add set_non_blocking function to the select extension.

---
 extension/ChangeLog |  6 ++++++
 extension/select.c  | 33 ++++++++++++++++++++++++++++++++-
 2 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/extension/ChangeLog b/extension/ChangeLog
index 8b3e4bc5..9218c744 100644
--- a/extension/ChangeLog
+++ b/extension/ChangeLog
@@ -1,3 +1,9 @@
+2013-07-01         Andrew J. Schorr     <aschorr@telemetry-investments.com>
+
+	* select.c (do_set_non_blocking): Implement new set_non_blocking
+	function.
+	(func_table): Add set_non_blocking.
+
 2013-07-01         Andrew J. Schorr     <aschorr@telemetry-investments.com>
 
 	* errlist.h: New file containing a list of all the errno values I could
diff --git a/extension/select.c b/extension/select.c
index 62116351..91898a20 100644
--- a/extension/select.c
+++ b/extension/select.c
@@ -33,6 +33,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
+#include <fcntl.h>
 
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -288,7 +289,7 @@ do_select(int nargs, awk_value_t *result)
 		maxwait.tv_sec = maxwait.tv_usec = 0;
 		timeout = &maxwait;
 	}
-        else if (get_argument(3, AWK_NUMBER, &timeout_arg)) {
+	else if (get_argument(3, AWK_NUMBER, &timeout_arg)) {
 		double secs = timeout_arg.num_value;
 		if (secs < 0) {
 			warning(ext_id, _("select: treating negative timeout as zero"));
@@ -363,9 +364,39 @@ do_select(int nargs, awk_value_t *result)
 	return make_number(rc, result);
 }
 
+/*  do_set_non_blocking --- Set a file to be non-blocking */
+
+static awk_value_t *
+do_set_non_blocking(int nargs, awk_value_t *result)
+{
+	awk_value_t cmd, cmdtype;
+
+	if (do_lint && nargs > 2)
+		lintwarn(ext_id, _("set_non_blocking: called with too many arguments"));
+	if (get_argument(0, AWK_STRING, & cmd) &&
+		get_argument(1, AWK_STRING, & cmdtype)) {
+		const awk_input_buf_t *buf;
+		if ((buf = get_file(cmd.str_value.str, cmd.str_value.len, cmdtype.str_value.str, cmdtype.str_value.len)) != NULL) {
+			int flags = fcntl(buf->fd, F_GETFL);
+			int rc = fcntl(buf->fd, F_SETFL, (flags|O_NONBLOCK));
+			if (rc < 0)
+				update_ERRNO_int(errno);
+			return make_number(rc, result);
+		} else
+			warning(ext_id, _("set_non_blocking: get_file(`%s', `%s') failed"), cmd.str_value.str, cmdtype.str_value.str);
+	} else if (do_lint) {
+		if (nargs < 2)
+			lintwarn(ext_id, _("set_non_blocking: called with too few arguments"));
+		else
+			lintwarn(ext_id, _("set_non_blocking: called with inappropriate argument(s)"));
+	}
+	return make_number(-1, result);
+}
+
 static awk_ext_func_t func_table[] = {
 	{ "select", do_select, 5 },
 	{ "select_signal", do_signal, 2 },
+	{ "set_non_blocking", do_set_non_blocking, 2 },
 };
 
 /* define the dl_load function using the boilerplate macro */
-- 
cgit v1.2.3