Home | History | Annotate | Line # | Download | only in eject
am_glue.c revision 1.2
      1 /*	$NetBSD: am_glue.c,v 1.2 2010/02/26 20:18:37 christos Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1999 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     26  * POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 #include <sys/cdefs.h>
     30 #ifndef lint
     31 __RCSID("$NetBSD: am_glue.c,v 1.2 2010/02/26 20:18:37 christos Exp $");
     32 #endif /* not lint */
     33 
     34 #ifdef HAVE_CONFIG_H
     35 # include <config.h>
     36 #endif /* HAVE_CONFIG_H */
     37 #include <am_defs.h>
     38 #include <amu.h>
     39 #include <rpc/pmap_prot.h>
     40 #include <rpc/pmap_clnt.h>
     41 
     42 #include "am_glue.h"
     43 
     44 static CLIENT *clnt;
     45 
     46 static struct timeval tv = { 5, 0 };
     47 /*
     48  * Appease lint: Properly typecast some numbers defined in
     49  * src/extern/bsd/am-utils/dist/include/amq_defs.h.
     50  */
     51 #define	xAMQ_PROGRAM		(rpcprog_t)AMQ_PROGRAM
     52 #define xAMQ_VERSION		(rpcvers_t)AMQ_VERSION
     53 #define xAMQPROC_SYNC_UMNT	(rpcproc_t)AMQPROC_SYNC_UMNT
     54 
     55 static int
     56 ping_pmap(void)
     57 {
     58 	u_short port = 0;
     59 	CLIENT *cl;
     60 	struct pmap parms;
     61 	struct sockaddr_in si;
     62 	int s = -1, rv;
     63 
     64 	(void)memset(&si, 0, sizeof(si));
     65 	si.sin_family = AF_INET;
     66 	si.sin_len = sizeof(si);
     67 	si.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
     68 	si.sin_port = htons(PMAPPORT);
     69 
     70 	if ((cl = clntudp_bufcreate(&si, PMAPPROG, PMAPVERS, tv,
     71 	    &s, RPCSMALLMSGSIZE, RPCSMALLMSGSIZE)) == NULL)
     72 		return -1;
     73 
     74 	parms.pm_prog = PMAPPROG;
     75 	parms.pm_vers = PMAPVERS;
     76 	parms.pm_prot = IPPROTO_UDP;
     77 	parms.pm_port = 0;  /* not needed or used */
     78 
     79 	rv = CLNT_CALL(cl, (rpcproc_t)PMAPPROC_GETPORT,
     80 	    (xdrproc_t)xdr_pmap, &parms,
     81 	    (xdrproc_t)xdr_u_short, &port, tv) == RPC_SUCCESS ? 0 : -1;
     82 
     83 	CLNT_DESTROY(cl);
     84 	return rv;
     85 }
     86 
     87 void
     88 am_init(void)
     89 {
     90 	static const char *server = "localhost";
     91 
     92 	if (ping_pmap() == -1)
     93 		return;
     94 	/*
     95 	 * Create RPC endpoint
     96 	 */
     97 	/* try tcp first */
     98 	clnt = clnt_create(server, xAMQ_PROGRAM, xAMQ_VERSION, "tcp");
     99 	if (clnt != NULL)
    100 		return;
    101 
    102 	/* try udp next */
    103 	clnt = clnt_create(server, xAMQ_PROGRAM, xAMQ_VERSION, "udp");
    104 	if (clnt != NULL)	/* set udp timeout */
    105 		(void)clnt_control(clnt, CLSET_RETRY_TIMEOUT, (void *)&tv);
    106 }
    107 
    108 int
    109 am_unmount(const char *dirname)
    110 {
    111 	static struct timeval timeout = { ALLOWED_MOUNT_TIME, 0 };
    112 	amq_sync_umnt result;
    113 
    114 	if (clnt == NULL)
    115 		return AMQ_UMNT_FAILED;
    116 
    117 	if (clnt_call(clnt, xAMQPROC_SYNC_UMNT, xdr_amq_string, &dirname,
    118 	    xdr_amq_sync_umnt, (void *)&result, timeout) != RPC_SUCCESS)
    119 		return AMQ_UMNT_SERVER;
    120 
    121 	return result.au_etype;
    122 }
    123