nslookup.c

Go to the documentation of this file.

/*
 * Copyright (C) 2012 Patrick Plenefisch <phplenefisch@wpi.edu>.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

FILE_LICENCE ( GPL2_OR_LATER );
FILE_SECBOOT ( PERMITTED );

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <ipxe/resolv.h>
#include <ipxe/tcpip.h>
#include <ipxe/monojob.h>
#include <ipxe/settings.h>
#include <usr/nslookup.h>

/** @file
 *
 * Standalone name resolution
 *
 */

/** A name resolution request */
struct nslookup {
        /** Reference count for this object */
        struct refcnt refcnt;

        /** Job control interface */
        struct interface job;
        /** Data transfer interface */
        struct interface resolver;

        /** Setting name */
        char *setting_name;
};

/**
 * Terminate name resolution
 *
 * @v nslookup          Name resolution request
 * @v rc                Reason for termination
 */
static void nslookup_close ( struct nslookup *nslookup, int rc ) {

        /* Shut down interfaces */
        intf_shutdown ( &nslookup->resolver, rc );
        intf_shutdown ( &nslookup->job, rc );
}

/**
 * Handle resolved name
 *
 * @v nslookup          Name resolution request
 * @v sa                Completed socket address
 */
static void nslookup_resolv_done ( struct nslookup *nslookup,
                                   struct sockaddr *sa ) {
        struct sockaddr_in *sin;
        struct sockaddr_in6 *sin6;
        const struct setting_type *default_type;
        struct settings *settings;
        struct setting setting;
        void *data;
        size_t len;
        int rc;

        /* Extract address */
        switch ( sa->sa_family ) {
        case AF_INET:
                sin = ( ( struct sockaddr_in * ) sa );
                data = &sin->sin_addr;
                len = sizeof ( sin->sin_addr );
                default_type = &setting_type_ipv4;
                break;
        case AF_INET6:
                sin6 = ( ( struct sockaddr_in6 * ) sa );
                data = &sin6->sin6_addr;
                len = sizeof ( sin6->sin6_addr );
                default_type = &setting_type_ipv6;
                break;
        default:
                rc = -ENOTSUP;
                goto err;
        }

        /* Parse specified setting name */
        if ( ( rc = parse_setting_name ( nslookup->setting_name,
                                         autovivify_child_settings, &settings,
                                         &setting ) ) != 0 )
                goto err;

        /* Apply default type if necessary */
        if ( ! setting.type )
                setting.type = default_type;

        /* Store in specified setting */
        if ( ( rc = store_setting ( settings, &setting, data, len ) ) != 0 )
                goto err;

 err:
        /* Terminate name resolution */
        nslookup_close ( nslookup, rc );
}

/** Name resolution resolver interface operations */
static struct interface_operation nslookup_resolver_operations[] = {
        INTF_OP ( resolv_done, struct nslookup *, nslookup_resolv_done ),
        INTF_OP ( intf_close, struct nslookup *, nslookup_close ),
};

/** Name resolution resolver interface descriptor */
static struct interface_descriptor nslookup_resolver_desc =
        INTF_DESC_PASSTHRU ( struct nslookup, resolver,
                             nslookup_resolver_operations, job );

/** Name resolution job control interface operations */
static struct interface_operation nslookup_job_operations[] = {
        INTF_OP ( intf_close, struct nslookup *, nslookup_close ),
};

/** Name resolution job control interface descriptor */
static struct interface_descriptor nslookup_job_desc =
        INTF_DESC_PASSTHRU ( struct nslookup, job,
                             nslookup_job_operations, resolver );

/**
 * Initiate standalone name resolution
 *
 * @v job               Parent interface
 * @v name              Name to resolve
 * @v setting_name      Setting name
 * @ret rc              Return status code
 */
static int resolv_setting ( struct interface *job, const char *name,
                            const char *setting_name ) {
        struct nslookup *nslookup;
        struct sockaddr sa;
        char *setting_name_copy;
        int rc;

        /* Allocate and initialise structure */
        nslookup = zalloc ( sizeof ( *nslookup ) + strlen ( setting_name )
                            + 1 /* NUL */ );
        if ( ! nslookup )
                return -ENOMEM;
        ref_init ( &nslookup->refcnt, NULL );
        intf_init ( &nslookup->job, &nslookup_job_desc, &nslookup->refcnt );
        intf_init ( &nslookup->resolver, &nslookup_resolver_desc,
                    &nslookup->refcnt );
        setting_name_copy = ( ( void * ) ( nslookup + 1 ) );
        strcpy ( setting_name_copy, setting_name );
        nslookup->setting_name = setting_name_copy;

        /* Start name resolution */
        memset ( &sa, 0, sizeof ( sa ) );
        if ( ( rc = resolv ( &nslookup->resolver, name, &sa ) ) != 0 )
                goto err_resolv;

        /* Attach parent interface, mortalise self, and return */
        intf_plug_plug ( &nslookup->job, job );
        ref_put ( &nslookup->refcnt );
        return 0;

 err_resolv:
        ref_put ( &nslookup->refcnt );
        return rc;
}

/**
 * Perform (blocking) standalone name resolution
 *
 * @v name              Name to resolve
 * @v setting_name      Setting name
 * @ret rc              Return status code
 */
int nslookup ( const char *name, const char *setting_name ) {
        int rc;

        /* Perform name resolution */
        if ( ( rc = resolv_setting ( &monojob, name, setting_name ) ) == 0 )
                rc = monojob_wait ( NULL, 0 );
        if ( rc != 0 ) {
                printf ( "Could not resolve %s: %s\n", name, strerror ( rc ) );
                return rc;
        }

        return 0;
}