NetBSD/usr.bin/eject/am_glue.c

/*	$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;
}