libipg_lookup.c

Go to the documentation of this file.

/*
 *  $Id: libipg_lookup.c,v 1.2 2003/12/29 08:10:10 mike Exp $
 *
 *  libipg
 *  libipb_lookup.c - lookup routines
 *
 *  Copyright (c) 2003 - 2004 Mike D. Schiffman <mike@infonexus.com>
 *  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 */

#if (HAVE_CONFIG_H)
#include "../include/config.h"
#endif

#include "./libipg.h"

u_int32_t
ipgeo_ipa2ipn(char *ip)
{
    char *p = ip;
    u_int32_t o1, o2, o3, o4;

    if (!isdigit(ip[0]))
    {
        return (0);
    }

    o1 = 16777216 * atol(strsep(&p, "."));

    o2 = 65536 * atol(strsep(&p, "."));

    o3 = 256 * atol(strsep(&p, "."));

    o4 = atol(strsep(&p, "."));

    return (o1 + o2 + o3 + o4);
}

int
ipgeo_lookup(u_int32_t ipn, u_int8_t flags, ipgeo_t *ipg)
{
    int n;
    char *p, *q;
    u_int32_t min, max;
    char buf[IPGEO_BUF_SIZE];

    if (ipg == NULL)
    {
        return (-1);
    }

#if (LIBIPG_LIL_ENDIAN)
    ipn = htonl(ipn);
#endif

    /* perform a linear search through the CSV file */
    for (; fgets(buf, IPGEO_BUF_SIZE - 1, ipg->db); )
    {
        if (buf[0] == '#')
        {
            /* ignore comments */
            continue;
        }

        p = buf;

        /* step over quote */
        p += 1;
        min = strtoul(strsep(&p, ","), (char **)NULL, 10);

        /* step over quote */
        p += 1;
        max = strtoul(strsep(&p, ","), (char **)NULL, 10);

        if (ipn >= min && ipn <= max)
        {
            p += 1;
            q = strsep(&p, ",");
            for (n = 0; n < sizeof(ipg->cc) - 1; n++)
            {
                if (q[n] == '"')
                {
                    ipg->cc[n] = NULL;
                    break;
                }
                ipg->cc[n] = q[n];
            }

            /* get cc */
            p += 1;
            q = strsep(&p, ",");
            for (n = 0; n < sizeof(ipg->country) - 1; n++)
            {
                if (q[n] == '"')
                {
                    ipg->country[n] = NULL;
                    break;
                }
                ipg->country[n] = q[n];
            }

            /* get country */
            p += 1;
            q = strsep(&p, ",");
            for (n = 0; n < sizeof(ipg->region) - 1; n++)
            {
                if (q[n] == '"')
                {
                    ipg->region[n] = NULL;
                    break;
                }
                ipg->region[n] = q[n];
            }

            /* get city */
            p += 1;
            q = strsep(&p, ",");
            for (n = 0; n < sizeof(ipg->city) - 1; n++)
            {
                if (q[n] == '"')
                {
                    ipg->city[n] = NULL;
                    break;
                }
                ipg->city[n] = q[n];
            }

            /* get latitude */
            p += 1;
            ipg->latitude = strtod(strsep(&p, ","), (char **)NULL);

            /* get longitude */
            p += 1;
            ipg->longitude = strtod(strsep(&p, ","), (char **)NULL);

            /* get isp */
            p += 1;
            q = strsep(&p, ",");
            for (n = 0; n < sizeof(ipg->isp) - 1; n++)
            {
                if (q[n] == '"')
                {
                    ipg->isp[n] = NULL;
                    break;
                }
                ipg->isp[n] = q[n];
            }

            rewind(ipg->db);
            return (1);
        }
    }
    /* no match */
    return (0);
}

char *
ipgeo_get_cc(ipgeo_t *ipg)
{
    if (ipg->cc[0] == '-')
    {
        return("UNKNOWN");
    }
    return (ipg->cc);
}

char *
ipgeo_get_country(ipgeo_t *ipg)
{
    if (ipg->country[0] == '-')
    {
        return("UNKNOWN");
    }
    return (ipg->country);
}

char *
ipgeo_get_region(ipgeo_t *ipg)
{
    if (ipg->region[0] == '-')
    {
        return("UNKNOWN");
    }
    return (ipg->region);
}

char *
ipgeo_get_city(ipgeo_t *ipg)
{
    if (ipg->city[0] == '-')
    {
        return("UNKNOWN");
    }
    return (ipg->city);
}

char *
ipgeo_get_isp(ipgeo_t *ipg)
{
    if (ipg->isp[0] == '-')
    {
        return("UNKNOWN");
    }
    return (ipg->isp);
}

double
ipgeo_get_lat(ipgeo_t *ipg)
{
    return (ipg->latitude);
}

double
ipgeo_get_long(ipgeo_t *ipg)
{
    return (ipg->longitude);
}

/* EOF */

---