xref: /illumos-gate/usr/src/lib/libilb/common/ilb_nat.c (revision dbed73cb)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #include <stdlib.h>
28 #include <strings.h>
29 #include <unistd.h>
30 #include <stddef.h>
31 #include <sys/types.h>
32 #include <sys/socket.h>
33 #include "libilb.h"
34 #include "libilb_impl.h"
35 
36 enum which_tbl {
37 	show_nat = 1,
38 	show_persist
39 };
40 
41 /* The common function to show kernel info. */
42 static ilb_status_t ilb_show_info(ilb_handle_t, char *, size_t *, boolean_t *,
43     enum which_tbl);
44 
45 /*
46  * To get the ILB NAT table.
47  *
48  * buf: The buffer to return the NAT table entries.
49  * num: The caller sets it to the number of ilb_nat_info_t entries buf can
50  *      hold.  On return, it contains the actual number of entries put in buf.
51  * end: The caller sets it to B_TRUE if it only wants at most num entries to
52  *      be returned.  The transaction to ilbd will be termianted when this
53  *      call returns.
54  *      The caller sets it to B_FALSE if it intends to get the whole table.
55  *      If the whole table has more than num entries, the caller can call
56  *      this function again to retrieve the rest of the table.
57  *      On return, end is set to B_TRUE if end of table is reached; B_FALSE
58  *      if there are still remaining entries.
59  */
60 ilb_status_t
ilb_show_nat(ilb_handle_t h,ilb_nat_info_t buf[],size_t * num,boolean_t * end)61 ilb_show_nat(ilb_handle_t h, ilb_nat_info_t buf[], size_t *num,
62     boolean_t *end)
63 {
64 	return (ilb_show_info(h, (char *)buf, num, end, show_nat));
65 }
66 
67 /*
68  * To get the ILB persistent entry table.
69  *
70  * buf: The buffer to return the persistent table entries.
71  * num: The caller sets it to the number of ilb_persist_info_t entries buf can
72  *      hold.  On return, it contains the actual number of entries put in buf.
73  * end: The caller sets it to B_TRUE if it only wants at most num entries to
74  *      be returned.  The transaction to ilbd will be termianted when this
75  *      call returns.
76  *      The caller sets it to B_FALSE if it intends to get the whole table.
77  *      If the whole table has more than num entries, the caller can call
78  *      this function again to retrieve the rest of the table.
79  *      On return, end is set to B_TRUE if end of table is reached; B_FALSE
80  *      if there are still remaining entries.
81  */
82 ilb_status_t
ilb_show_persist(ilb_handle_t h,ilb_persist_info_t buf[],size_t * num,boolean_t * end)83 ilb_show_persist(ilb_handle_t h, ilb_persist_info_t buf[], size_t *num,
84     boolean_t *end)
85 {
86 	return (ilb_show_info(h, (char *)buf, num, end, show_persist));
87 }
88 
89 /*
90  * The function doing the work...  The tbl parameter determines whith table
91  * to show.
92  */
93 static ilb_status_t
ilb_show_info(ilb_handle_t h,char * buf,size_t * num,boolean_t * end,enum which_tbl tbl)94 ilb_show_info(ilb_handle_t h, char *buf, size_t *num, boolean_t *end,
95     enum which_tbl tbl)
96 {
97 	ilb_comm_t	*req, *rbuf;
98 	ilb_show_info_t	*req_si, *tmp_si;
99 	size_t		reqsz, rbufsz, tmp_rbufsz, cur_num;
100 	size_t		entry_sz;
101 	ilb_status_t	rc;
102 
103 	if (*num == 0)
104 		return (ILB_STATUS_EINVAL);
105 
106 	reqsz = sizeof (ilb_comm_t) + sizeof (ilb_show_info_t);
107 	if ((req = malloc(reqsz)) == NULL)
108 		return (ILB_STATUS_ENOMEM);
109 	req_si = (ilb_show_info_t *)&req->ic_data;
110 
111 	/*
112 	 * Need to allocate a receive buffer and then copy the buffer
113 	 * content to the passed in buf.  The reason is that the
114 	 * communication to ilbd is message based and the protocol
115 	 * includes a header in the reply.  We need to remove this header
116 	 * from the message, hence the copying...
117 	 */
118 	if (tbl == show_nat)
119 		entry_sz = sizeof (ilb_nat_info_t);
120 	else
121 		entry_sz = sizeof (ilb_persist_info_t);
122 	rbufsz = *num * entry_sz + sizeof (ilb_comm_t) +
123 	    sizeof (ilb_show_info_t);
124 	if ((rbuf = malloc(rbufsz)) == NULL) {
125 		free(req);
126 		return (ILB_STATUS_ENOMEM);
127 	}
128 
129 	if (tbl == show_nat)
130 		req->ic_cmd = ILBD_SHOW_NAT;
131 	else
132 		req->ic_cmd = ILBD_SHOW_PERSIST;
133 	req->ic_flags = 0;
134 	req_si->sn_num = *num;
135 	cur_num = 0;
136 
137 	do {
138 		tmp_rbufsz = rbufsz;
139 		rc = i_ilb_do_comm(h, req, reqsz, rbuf, &tmp_rbufsz);
140 		if (rc != ILB_STATUS_OK)
141 			goto out;
142 		if (rbuf->ic_cmd != ILBD_CMD_OK) {
143 			rc = *(ilb_status_t *)&rbuf->ic_data;
144 			goto out;
145 		}
146 
147 		tmp_si = (ilb_show_info_t *)&rbuf->ic_data;
148 
149 		cur_num += tmp_si->sn_num;
150 		bcopy(&tmp_si->sn_data, buf, tmp_si->sn_num * entry_sz);
151 		buf += tmp_si->sn_num * entry_sz;
152 
153 		/*
154 		 * Buffer is filled, regardless of this is the end of table or
155 		 * not, we need to stop.
156 		 */
157 		if (cur_num == *num)
158 			break;
159 		/* Try to fill in the rest. */
160 		req_si->sn_num = *num - cur_num;
161 	} while (!(rbuf->ic_flags & ILB_COMM_END));
162 
163 	*num = cur_num;
164 
165 	/* End of transaction, let the caller know. */
166 	if (rbuf->ic_flags & ILB_COMM_END) {
167 		*end = B_TRUE;
168 	} else {
169 		/* The user wants to terminate the transaction */
170 		if (*end) {
171 			req->ic_flags = ILB_COMM_END;
172 			tmp_rbufsz = rbufsz;
173 			rc = i_ilb_do_comm(h, req, reqsz, rbuf, &tmp_rbufsz);
174 		}
175 	}
176 out:
177 	free(req);
178 	free(rbuf);
179 	return (rc);
180 }
181