/*      $NetBSD: am_glue.c,v 1.4 2010/06/23 18:07:59 yamt Exp $ */

/*-
 * Copyright (c) 1999 The NetBSD Foundation, Inc.
 * 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
 */

#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: am_glue.c,v 1.4 2010/06/23 18:07:59 yamt Exp $");
#endif /* not lint */

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif /* HAVE_CONFIG_H */
#include <am_defs.h>
#include <amu.h>
#include <rpc/pmap_prot.h>
#include <rpc/pmap_clnt.h>

#include <stdbool.h>

#include "am_glue.h"

static CLIENT *clnt;

static struct timeval tv = { 5, 0 };
/*
 * Appease lint: Properly typecast some numbers defined in
 * src/extern/bsd/am-utils/dist/include/amq_defs.h.
 */
#define xAMQ_PROGRAM            (rpcprog_t)AMQ_PROGRAM
#define xAMQ_VERSION            (rpcvers_t)AMQ_VERSION
#define xAMQPROC_SYNC_UMNT      (rpcproc_t)AMQPROC_SYNC_UMNT

static int
ping_pmap(void)
{
        u_short port = 0;
        CLIENT *cl;
        struct pmap parms;
        struct sockaddr_in si;
        int s = -1, rv;
        static struct timeval pingtv = { 1, 0 };

        (void)memset(&si, 0, sizeof(si));
        si.sin_family = AF_INET;
        si.sin_len = sizeof(si);
        si.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
        si.sin_port = htons(PMAPPORT);

        if ((cl = clntudp_bufcreate(&si, PMAPPROG, PMAPVERS, pingtv,
            &s, RPCSMALLMSGSIZE, RPCSMALLMSGSIZE)) == NULL)
                return -1;

        parms.pm_prog = PMAPPROG;
        parms.pm_vers = PMAPVERS;
        parms.pm_prot = IPPROTO_UDP;
        parms.pm_port = 0;  /* not needed or used */

        rv = CLNT_CALL(cl, (rpcproc_t)PMAPPROC_GETPORT,
            (xdrproc_t)xdr_pmap, &parms,
            (xdrproc_t)xdr_u_short, &port, pingtv) == RPC_SUCCESS ? 0 : -1;

        CLNT_DESTROY(cl);
        return rv;
}

static void
am_init(void)
{
        static const char *server = "localhost";
        static bool called;

        if (called)
                return;
        called = true;

        if (ping_pmap() == -1)
                return;
        /*
         * Create RPC endpoint
         */
        /* try tcp first */
        clnt = clnt_create(server, xAMQ_PROGRAM, xAMQ_VERSION, "tcp");
        if (clnt != NULL)
                return;

        /* try udp next */
        clnt = clnt_create(server, xAMQ_PROGRAM, xAMQ_VERSION, "udp");
        if (clnt != NULL)       /* set udp timeout */
                (void)clnt_control(clnt, CLSET_RETRY_TIMEOUT, (void *)&tv);
}

int
am_unmount(const char *dirname)
{
        static struct timeval timeout = { ALLOWED_MOUNT_TIME, 0 };
        amq_sync_umnt result;

        am_init();

        if (clnt == NULL)
                return AMQ_UMNT_FAILED;

        if (clnt_call(clnt, xAMQPROC_SYNC_UMNT, xdr_amq_string, &dirname,
            xdr_amq_sync_umnt, (void *)&result, timeout) != RPC_SUCCESS)
                return AMQ_UMNT_SERVER;

        return result.au_etype;
}