--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/openenvutils/commandshell/shell/src/modules/socket.c Tue Feb 02 10:12:00 2010 +0200
@@ -0,0 +1,303 @@
+// socket.c - Unix domain socket module
+//
+// © Portions Copyright (c) Symbian Software Ltd 2007. All rights reserved.
+//
+/*
+ * This file is part of zsh, the Z shell.
+ *
+ * Copyright (c) 2002 Peter Stephenson
+ * All rights reserved.
+ *
+ * Permission is hereby granted, without written agreement and without
+ * license or royalty fees, to use, copy, modify, and distribute this
+ * software and to distribute modified versions of this software for any
+ * purpose, provided that the above copyright notice and the following
+ * two paragraphs appear in all copies of this software.
+ *
+ * In no event shall Peter Stephenson or the Zsh Development
+ * Group be liable to any party for direct, indirect, special, incidental,
+ * or consequential damages arising out of the use of this software and
+ * its documentation, even if Peter Stephenson, and the Zsh
+ * Development Group have been advised of the possibility of such damage.
+ *
+ * Peter Stephenson and the Zsh Development Group specifically
+ * disclaim any warranties, including, but not limited to, the implied
+ * warranties of merchantability and fitness for a particular purpose. The
+ * software provided hereunder is on an "as is" basis, and Peter Stephenson
+ * and the Zsh Development Group have no obligation to provide maintenance,
+ * support, updates, enhancements, or modifications.
+ *
+ */
+#include "socket.mdh"
+#include "socket.pro"
+
+#ifdef __SYMBIAN32__
+#include "dummy.h" //poll
+#endif //__SYMBIAN32__
+
+
+#include <sys/socket.h>
+#ifndef __SYMBIAN32__
+#include <sys/un.h>
+#else
+#include "un.h"
+#endif
+
+#ifndef UNIX_PATH_MAX
+# define UNIX_PATH_MAX 108
+#endif
+
+/*
+ * We need to include the zsh headers later to avoid clashes with
+ * the definitions on some systems, however we need the configuration
+ * file to decide whether we can include netinet/in_systm.h, which
+ * doesn't exist on cygwin.
+ */
+
+/*
+ * We use poll() in preference to select because some subset of manuals says
+ * that's the thing to do, plus it's a bit less fiddly. I don't actually
+ * have access to a system with poll but not select, however, though
+ * both bits of the code have been tested on a machine with both.
+ */
+#ifdef HAVE_POLL_H
+#ifndef __SYMBIAN32__
+# include <sys/poll.h>
+#else
+# include "poll.h"
+#endif
+#endif
+#if defined(HAVE_POLL) && !defined(POLLIN) && !defined(POLLNORM)
+# undef HAVE_POLL
+#endif
+
+#ifdef __SYMBIAN32__
+#ifdef __WINSCW__
+#pragma warn_unusedarg off
+#endif//__WINSCW__
+#endif//__SYMBIAN32__
+
+static int
+bin_zsocket(char *nam, char **args, Options ops, UNUSED(int func))
+{
+ int err=1, verbose=0, test=0, targetfd=0;
+ ZSOCKLEN_T len;
+ struct sockaddr_un soun;
+ int sfd;
+
+ if (OPT_ISSET(ops,'v'))
+ verbose = 1;
+
+ if (OPT_ISSET(ops,'t'))
+ test = 1;
+
+ if (OPT_ISSET(ops,'d')) {
+ targetfd = atoi(OPT_ARG(ops,'d'));
+ if (!targetfd) {
+ zwarnnam(nam, "%s is an invalid argument to -d",
+ OPT_ARG(ops, 'd'), 0);
+ return 1;
+ }
+ }
+
+ if (OPT_ISSET(ops,'l')) {
+ char *localfn;
+
+ if (!args[0]) {
+ zwarnnam(nam, "-l requires an argument", NULL, 0);
+ return 1;
+ }
+
+ localfn = args[0];
+
+ sfd = socket(PF_UNIX, SOCK_STREAM, 0);
+
+ if (sfd == -1) {
+ zwarnnam(nam, "socket error: %e ", NULL, errno);
+ return 1;
+ }
+
+ soun.sun_family = AF_UNIX;
+ strncpy(soun.sun_path, localfn, UNIX_PATH_MAX);
+
+ if (bind(sfd, (struct sockaddr *)&soun, sizeof(struct sockaddr_un)))
+ {
+ zwarnnam(nam, "could not bind to %s: %e", soun.sun_path, errno);
+ close(sfd);
+ return 1;
+ }
+
+ if (listen(sfd, 1))
+ {
+ zwarnnam(nam, "could not listen on socket: %e", NULL, errno);
+ close(sfd);
+ return 1;
+ }
+
+ if (targetfd) {
+ redup(sfd, targetfd);
+ sfd = targetfd;
+ }
+ else {
+ /* move the fd since no one will want to read from it */
+ sfd = movefd(sfd);
+ }
+
+ setiparam("REPLY", sfd);
+
+ if (verbose)
+ printf("%s listener is on fd %d\n", soun.sun_path, sfd);
+
+ return 0;
+
+ }
+ else if (OPT_ISSET(ops,'a'))
+ {
+ int lfd, rfd;
+
+ if (!args[0]) {
+ zwarnnam(nam, "-a requires an argument", NULL, 0);
+ return 1;
+ }
+
+ lfd = atoi(args[0]);
+
+ if (!lfd) {
+ zwarnnam(nam, "invalid numerical argument", NULL, 0);
+ return 1;
+ }
+
+ if (test) {
+#if defined(HAVE_POLL) || defined(HAVE_SELECT)
+# ifdef HAVE_POLL
+ struct pollfd pfd;
+ int ret;
+
+ pfd.fd = lfd;
+ pfd.events = POLLIN;
+ if ((ret = poll(&pfd, 1, 0)) == 0) return 1;
+ else if (ret == -1)
+ {
+ zwarnnam(nam, "poll error: %e", NULL, errno);
+ return 1;
+ }
+# else
+ fd_set rfds;
+ struct timeval tv;
+ int ret;
+
+ FD_ZERO(&rfds);
+ FD_SET(lfd, &rfds);
+ tv.tv_sec = 0;
+ tv.tv_usec = 0;
+
+ if ((ret = select(lfd+1, &rfds, NULL, NULL, &tv))) return 1;
+ else if (ret == -1)
+ {
+ zwarnnam(nam, "select error: %e", NULL, errno);
+ return 1;
+ }
+
+# endif
+
+#else
+ zwarnnam(nam, "not currently supported", NULL, 0);
+ return 1;
+#endif
+ }
+
+ len = sizeof(soun);
+ if ((rfd = accept(lfd, (struct sockaddr *)&soun, &len)) == -1)
+ {
+ zwarnnam(nam, "could not accept connection: %e", NULL, errno);
+ return 1;
+ }
+
+ if (targetfd) {
+ redup(rfd, targetfd);
+ sfd = targetfd;
+ }
+ else {
+ sfd = rfd;
+ }
+
+ setiparam("REPLY", sfd);
+
+ if (verbose)
+ printf("new connection from %s is on fd %d\n", soun.sun_path, sfd);
+ }
+ else
+ {
+ if (!args[0]) {
+ zwarnnam(nam, "zsocket requires an argument", NULL, 0);
+ return 1;
+ }
+
+ sfd = socket(PF_UNIX, SOCK_STREAM, 0);
+
+ if (sfd == -1) {
+ zwarnnam(nam, "socket creation failed: %e", NULL, errno);
+ return 1;
+ }
+
+ soun.sun_family = AF_UNIX;
+ strncpy(soun.sun_path, args[0], UNIX_PATH_MAX);
+
+ if ((err = connect(sfd, (struct sockaddr *)&soun, sizeof(struct sockaddr_un)))) {
+ zwarnnam(nam, "connection failed: %e", NULL, errno);
+ close(sfd);
+ return 1;
+ }
+ else
+ {
+ if (targetfd) {
+ redup(sfd, targetfd);
+ sfd = targetfd;
+ }
+
+ setiparam("REPLY", sfd);
+
+ if (verbose)
+ printf("%s is now on fd %d\n", soun.sun_path, sfd);
+ }
+
+ }
+
+ return 0;
+}
+
+static struct builtin bintab[] = {
+ BUILTIN("zsocket", 0, bin_zsocket, 0, 3, 0, "ad:ltv", NULL),
+};
+
+/* The load/unload routines required by the zsh library interface */
+
+/**/
+int
+setup_(UNUSED(Module m))
+{
+ return 0;
+}
+
+/**/
+int
+boot_(Module m)
+{
+ return !addbuiltins(m->nam, bintab, sizeof(bintab)/sizeof(*bintab));
+}
+
+
+/**/
+int
+cleanup_(Module m)
+{
+ deletebuiltins(m->nam, bintab, sizeof(bintab)/sizeof(*bintab));
+ return 0;
+}
+
+/**/
+int
+finish_(UNUSED(Module m))
+{
+ return 0;
+}