/*
 * Copyright (C) 2012 by Darren Reed.
 *
 * See the IPFILTER.LICENCE file for details on licencing.
 *
 * $Id$
 */

#include "ipf.h"


int
genmask(int family, char *msk, i6addr_t *mskp)
{
        char *endptr = 0L;
        u_32_t addr;
        int bits;

        if (strchr(msk, '.') || strchr(msk, 'x') || strchr(msk, ':')) {
                /* possibly of the form xxx.xxx.xxx.xxx
                 * or 0xYYYYYYYY */
                switch (family)
                {
#ifdef USE_INET6
                case AF_INET6 :
                        if (inet_pton(AF_INET6, msk, &mskp->in4) != 1)
                                return (-1);
                        break;
#endif
                case AF_INET :
                        if (inet_aton(msk, &mskp->in4) == 0)
                                return (-1);
                        break;
                default :
                        return (-1);
                        /*NOTREACHED*/
                }
        } else {
                /*
                 * set x most significant bits
                 */
                bits = (int)strtol(msk, &endptr, 0);

                switch (family)
                {
                case AF_INET6 :
                        if ((*endptr != '\0') || (bits < 0) || (bits > 128))
                                return (-1);
                        fill6bits(bits, mskp->i6);
                        break;
                case AF_INET :
                        if (*endptr != '\0' || bits > 32 || bits < 0)
                                return (-1);
                        if (bits == 0)
                                addr = 0;
                        else
                                addr = htonl(0xffffffff << (32 - bits));
                        mskp->in4.s_addr = addr;
                        break;
                default :
                        return (-1);
                        /*NOTREACHED*/
                }
        }
        return (0);
}