kinit/nfsmount/portmap.c

#include <sys/types.h>
#include <netinet/in.h>
#include <asm/byteorder.h>	/* __constant_hton* */
#include <stdio.h>
#include <stdlib.h>

#include "nfsmount.h"
#include "sunrpc.h"

struct portmap_call {
	struct rpc_call rpc;
	uint32_t program;
	uint32_t version;
	uint32_t proto;
	uint32_t port;
};

struct portmap_reply {
	struct rpc_reply rpc;
	uint32_t port;
};

static struct portmap_call call = {
	.rpc = {
		.program	= __constant_htonl(RPC_PMAP_PROGRAM),
		.prog_vers	= __constant_htonl(RPC_PMAP_VERSION),
		.proc		= __constant_htonl(PMAP_PROC_GETPORT),
	}
};

uint32_t portmap(uint32_t server, uint32_t program, uint32_t version, uint32_t proto)
{
	struct portmap_reply reply;
	struct client *clnt;
	struct rpc rpc;
	uint32_t port = 0;

	if ((clnt = tcp_client(server, RPC_PMAP_PORT, 0)) == NULL) {
		if ((clnt = udp_client(server, RPC_PMAP_PORT, 0)) == NULL) {
			goto bail;
		}
	}

	call.program = htonl(program);
	call.version = htonl(version);
	call.proto = htonl(proto);

	rpc.call = (struct rpc_call *)&call;
	rpc.call_len = sizeof(call);
	rpc.reply = (struct rpc_reply *)&reply;
	rpc.reply_len = sizeof(reply);

	if (rpc_call(clnt, &rpc) < 0)
		goto bail;

	if (rpc.reply_len < sizeof(reply)) {
		fprintf(stderr, "incomplete reply: %zu < %zu\n",
			rpc.reply_len, sizeof(reply));
		goto bail;
	}

	port = ntohl(reply.port);

bail:
	dprintf("Port for %d/%d[%s]: %d\n", program, version,
		proto == IPPROTO_TCP ? "tcp" : "udp", port);

	if (clnt) {
		client_free(clnt);
	}

	return port;
}
1	#include <sys/types.h>
2	#include <netinet/in.h>
3	#include <asm/byteorder.h> /* __constant_hton* */
4	#include <stdio.h>
5	#include <stdlib.h>
6
7	#include "nfsmount.h"
8	#include "sunrpc.h"
9
10	struct portmap_call {
11	struct rpc_call rpc;
12	uint32_t program;
13	uint32_t version;
14	uint32_t proto;
15	uint32_t port;
16	};
17
18	struct portmap_reply {
19	struct rpc_reply rpc;
20	uint32_t port;
21	};
22
23	static struct portmap_call call = {
24	.rpc = {
25	.program = __constant_htonl(RPC_PMAP_PROGRAM),
26	.prog_vers = __constant_htonl(RPC_PMAP_VERSION),
27	.proc = __constant_htonl(PMAP_PROC_GETPORT),
28	}
29	};
30
31	uint32_t portmap(uint32_t server, uint32_t program, uint32_t version, uint32_t proto)
32	{
33	struct portmap_reply reply;
34	struct client *clnt;
35	struct rpc rpc;
36	uint32_t port = 0;
37
38	if ((clnt = tcp_client(server, RPC_PMAP_PORT, 0)) == NULL) {
39	if ((clnt = udp_client(server, RPC_PMAP_PORT, 0)) == NULL) {
40	goto bail;
41	}
42	}
43
44	call.program = htonl(program);
45	call.version = htonl(version);
46	call.proto = htonl(proto);
47
48	rpc.call = (struct rpc_call *)&call;
49	rpc.call_len = sizeof(call);
50	rpc.reply = (struct rpc_reply *)&reply;
51	rpc.reply_len = sizeof(reply);
52
53	if (rpc_call(clnt, &rpc) < 0)
54	goto bail;
55
56	if (rpc.reply_len < sizeof(reply)) {
57	fprintf(stderr, "incomplete reply: %zu < %zu\n",
58	rpc.reply_len, sizeof(reply));
59	goto bail;
60	}
61
62	port = ntohl(reply.port);
63
64	bail:
65	dprintf("Port for %d/%d[%s]: %d\n", program, version,
66	proto == IPPROTO_TCP ? "tcp" : "udp", port);
67
68	if (clnt) {
69	client_free(clnt);
70	}
71
72	return port;
73	}