// 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;
}