/*
 * Copyright (c) 2014 Joyent, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 * See the IPFILTER.LICENCE file for details on licensing.
 */


#include <errno.h>
#include <net/if.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <zone.h>

#include "netinet/ip_fil.h"
#include "ipfzone.h"

static ipfzoneobj_t     ipzo;
static boolean_t        do_setzone = 0;
static int              num_setzones = 0;

extern int      errno;
extern int      opterr;
extern int      optind;
extern char     *optarg;

/*
 * Get the zonename if it's the last argument and set the zonename
 * in ipfzo to it. This is used by ipf(8) only - all of the other tools
 * specify the zone with the -z option, and therefore use getzoneopt() below.
 */
void
getzonearg(int argc, char *argv[], const char *optstr)
{
        int c;

        /*
         * Don't warn about unknown options - let subsequent calls to
         * getopt() handle this.
         */
        opterr = 0;

        /*
         * getopt is also used here to set optind so that we can
         * determine if the last argument belongs to a flag or is
         * actually a zonename.
         */
        while ((c = getopt(argc, argv, optstr)) != -1) {
                if (c == 'G')
                        ipzo.ipfz_gz = 1;
        }

        if (optind < argc)
                setzonename(argv[optind]);

        /*
         * Reset optind and opterr so the next getopt call will go through all
         * of argv again and warn about unknown options.
         */
        optind = 1;
        opterr = 1;
}

/*
 * Get a -z option from argv and set the zonename in ipfzo accordingly
 */
void
getzoneopt(int argc, char *argv[], const char *optstr)
{
        int c;

        /*
         * Don't warn about unknown options - let subsequent calls to
         * getopt() handle this.
         */
        opterr = 0;

        while ((c = getopt(argc, argv, optstr)) != -1) {
                if (c == 'G')
                        setzonename_global(optarg);

                if (c == 'z')
                        setzonename(optarg);
        }

        /*
         * Reset optind and opterr so the next getopt call will go through all
         * of argv again and warn about unknown options.
         */
        optind = 1;
        opterr = 1;
}

/*
 * Set the zonename in ipfzo to the given string: this is the zone all further
 * ioctls will act on.
 */
void
setzonename(const char *zonename)
{
        memcpy(ipzo.ipfz_zonename, zonename, sizeof (ipzo.ipfz_zonename));
        do_setzone = B_TRUE;
        num_setzones++;
}

/*
 * Set the zonename in ipfo, and the gz flag. This indicates that we want all
 * further ioctls to act on the GZ-controlled stack for that zone.
 */
void
setzonename_global(const char *zonename)
{
        setzonename(zonename);
        ipzo.ipfz_gz = 1;
}

/*
 * Set the zone that all further ioctls will operate on. See the "GZ-controlled
 * and per-zone stacks" note at the top of ip_fil_solaris.c for further
 * explanation.
 */
int
setzone(int fd)
{
        if (!do_setzone)
                return (0);

        if (num_setzones > 1) {
                (void) fprintf(stderr,
                    "Only one of -G and -z may be set\n");
                return (-1);
        }

        if (ipzo.ipfz_gz == 1 &&
            getzoneidbyname(ipzo.ipfz_zonename) == GLOBAL_ZONEID) {
                (void) fprintf(stderr,
                    "-G cannot be used with the global zone\n");
                return (-1);
        }

        if (ioctl(fd, SIOCIPFZONESET, &ipzo) == -1) {
                switch (errno) {
                case ENODEV:
                        (void) fprintf(stderr,
                            "Could not find running zone: %s\n",
                            ipzo.ipfz_zonename);
                        break;
                case EACCES:
                        (void) fprintf(stderr,
                            "Permission denied setting zone: %s\n",
                            ipzo.ipfz_zonename);
                        break;
                default:
                        perror("Error setting zone");
                }
                return (-1);
        }

        return (0);
}