Add pb-event helper utility

Add a new helper utility pb_event.  pb_event reads
data on stdin and writes it to the petitboot event
socket.

Signed-off-by: Geoff Levand <geoffrey.levand@am.sony.com>
Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

3 files changed, 109 insertions, 3 deletions

diff --git a/configure.ac b/configure.ac
index c67f0ed..524cbc5 100644
--- a/configure.ac
+++ b/configure.ac
@@ -51,6 +51,6 @@ AS_IF([test "x$with_twin" != xno],
 		[${twin_LIBS}])])
 
 mkdir -p discover lib/list lib/log lib/pb-protocol lib/system lib/talloc \
-	lib/waiter test ui/common ui/ncurses ui/test ui/twin
+	lib/waiter test ui/common ui/ncurses ui/test ui/twin utils
 
 AC_OUTPUT
diff --git a/rules.mk b/rules.mk
index ff807e4..3251b04 100644
--- a/rules.mk
+++ b/rules.mk
@@ -13,6 +13,7 @@ pb_discover = discover/pb-discover
 pb_cui = ui/ncurses/pb-cui
 pb_test = ui/test/pb-test
 pb_twin = ui/twin/pb-twin
+pb_event = utils/pb-event
 parser_test = test/parser-test
 
 # install targets and components
@@ -20,6 +21,7 @@ daemons = $(pb_discover)
 parsers = event kboot yaboot
 uis = $(pb_test)
 tests = $(parser_test)
+utils = $(pb_event)
 
 ifeq ($(PBTWIN),y)
 	uis += $(pb_twin)
@@ -61,7 +63,7 @@ daemon_objs = $(lib_objs) $(parser_objs) $(discover_objs)
 
 client_objs = $(lib_objs) $(ui_common_objs)
 
-all: $(uis) $(daemons)
+all: $(uis) $(daemons) $(utils)
 
 # ncurses cui
 pb_cui_objs = $(client_objs) $(ncurses_objs) ui/ncurses/ps3-cui.o \
@@ -97,6 +99,13 @@ $(pb_discover_objs): $(makefiles)
 $(pb_discover): $(pb_discover_objs)
 	$(LINK.o) -o $@ $^
 
+# utils
+pb_event_objs = utils/pb-event.o
+$(pb_event_objs): $(makefiles)
+
+$(pb_event): $(pb_event_objs)
+	$(LINK.o) -o $@ $^
+
 # parser-test
 parser_test_objs = $(lib_objs) $(parser_objs) test/parser-test.o
 $(parser_test_objs): $(makefiles)
@@ -108,7 +117,7 @@ parser-test: $(parser_test)
 
 install: all $(rules)
 	$(INSTALL) -d $(DESTDIR)$(sbindir)/
-	$(INSTALL) $(daemons) $(uis) $(DESTDIR)$(sbindir)/
+	$(INSTALL) $(daemons) $(uis) $(utils) $(DESTDIR)$(sbindir)/
 	$(INSTALL) -d $(DESTDIR)$(pkgdatadir)/artwork/
 	$(INSTALL) $(addprefix $(top_srcdir)/ui/twin/artwork/,$(artwork)) \
 		$(DESTDIR)$(pkgdatadir)/artwork/
@@ -142,6 +151,8 @@ clean:
 	rm -f $(pb_twin_objs)
 	rm -f $(daemons)
 	rm -f $(pb_discover_objs)
+	rm -f $(utils)
+	rm -f $(pb_event_objs)
 	rm -f $(tests)
 	rm -f $(parser_test_objs)
 
diff --git a/utils/pb-event.c b/utils/pb-event.c
new file mode 100644
index 0000000..5ef8115
--- /dev/null
+++ b/utils/pb-event.c
@@ -0,0 +1,95 @@
+/*
+ *  Copyright (C) 2009 Sony Computer Entertainment Inc.
+ *  Copyright 2009 Sony Corp.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; version 2 of the License.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#if defined(HAVE_CONFIG_H)
+#include "config.h"
+#endif
+
+#define _GNU_SOURCE
+#include <assert.h>
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <sys/un.h>
+
+#include "discover/user-event.h"
+
+#if defined(DEBUG)
+#define DBG(_args...) do {fprintf(stderr, _args); fflush(stderr); } while (0)
+#else
+static inline int __attribute__ ((format (printf, 1, 2))) DBG(
+	__attribute__((unused)) const char *fmt, ...) {return 0; }
+#endif
+
+int main(void)
+{
+	int result;
+	struct sockaddr_un addr;
+	char buf[PBOOT_USER_EVENT_SIZE];
+	ssize_t len;
+	int s;
+	int i;
+
+	s = socket(PF_UNIX, SOCK_DGRAM, 0);
+
+	if (s < 0) {
+		fprintf(stderr, "pb-event: socket: %s\n", strerror(errno));
+		return EXIT_FAILURE;
+	}
+
+	result = EXIT_SUCCESS;
+
+	len = fread(buf, 1, sizeof(buf), stdin);
+
+	if (!feof(stdin)) {
+		fprintf(stderr, "pb-event: msg too big (%u byte max)\n",
+			sizeof(buf));
+		result = EXIT_FAILURE;
+		/* continue on and try to write msg */
+	}
+
+	if (!len)
+		return result;
+
+	memset(&addr, 0, sizeof(addr));
+	addr.sun_family = AF_UNIX;
+	strcpy(addr.sun_path, PBOOT_USER_EVENT_SOCKET);
+
+	for (i = 10; i; i--) {
+		ssize_t sent = sendto(s, buf, len, 0, (struct sockaddr *)&addr,
+			SUN_LEN(&addr));
+
+		if (sent == len)
+			break;
+
+		DBG("pb-event: waiting for server %d\n", i);
+		sleep(1);
+	}
+
+	if (!i) {
+		fprintf(stderr, "pb-event: send: %s\n", strerror(errno));
+		return EXIT_FAILURE;
+	}
+
+	DBG("pb-event: wrote %u bytes\n", len);
+	return result;
+}