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