NetBSD/usr.bin/eject/am_glue.c

133 lines
3.7 KiB
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;
}