1 /*
2 * xdr_sizeof.c
3 *
4 * Copyright (c) 2010, Oracle America, Inc.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are
8 * met:
9 *
10 * * Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * * Redistributions in binary form must reproduce the above
13 * copyright notice, this list of conditions and the following
14 * disclaimer in the documentation and/or other materials
15 * provided with the distribution.
16 * * Neither the name of the "Oracle America, Inc." nor the names of its
17 * contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
27 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
30 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32 *
33 * General purpose routine to see how much space something will use
34 * when serialized using XDR.
35 */
36
37 #include <rpc/types.h>
38 #include <rpc/xdr.h>
39 #include <sys/types.h>
40 #include <stdlib.h>
41 #include <shlib-compat.h>
42
43 /* ARGSUSED */
44 static bool_t
45 x_putlong (XDR *xdrs, const long *longp)
46 {
47 xdrs->x_handy += BYTES_PER_XDR_UNIT;
48 return TRUE;
49 }
50
51 /* ARGSUSED */
52 static bool_t
53 x_putbytes (XDR *xdrs, const char *bp, u_int len)
54 {
55 xdrs->x_handy += len;
56 return TRUE;
57 }
58
59 static u_int
60 x_getpostn (const XDR *xdrs)
61 {
62 return xdrs->x_handy;
63 }
64
65 /* ARGSUSED */
66 static bool_t
67 x_setpostn (XDR *xdrs, u_int len)
68 {
69 /* This is not allowed */
70 return FALSE;
71 }
72
73 static int32_t *
74 x_inline (XDR *xdrs, u_int len)
75 {
76 if (len == 0)
77 return NULL;
78 if (xdrs->x_op != XDR_ENCODE)
79 return NULL;
80 if (len < (u_int) (long int) xdrs->x_base)
81 {
82 /* x_private was already allocated */
83 xdrs->x_handy += len;
84 return (int32_t *) xdrs->x_private;
85 }
86 else
87 {
88 /* Free the earlier space and allocate new area */
89 free (xdrs->x_private);
90 if ((xdrs->x_private = (caddr_t) malloc (len)) == NULL)
91 {
92 xdrs->x_base = 0;
93 return NULL;
94 }
95 xdrs->x_base = (void *) (long) len;
96 xdrs->x_handy += len;
97 return (int32_t *) xdrs->x_private;
98 }
99 }
100
101 static int
102 harmless (void)
103 {
104 /* Always return FALSE/NULL, as the case may be */
105 return 0;
106 }
107
108 static void
109 x_destroy (XDR *xdrs)
110 {
111 xdrs->x_handy = 0;
112 xdrs->x_base = 0;
113 if (xdrs->x_private)
114 {
115 free (xdrs->x_private);
116 xdrs->x_private = NULL;
117 }
118 return;
119 }
120
121 static bool_t
122 x_putint32 (XDR *xdrs, const int32_t *int32p)
123 {
124 xdrs->x_handy += BYTES_PER_XDR_UNIT;
125 return TRUE;
126 }
127
128 unsigned long
129 xdr_sizeof (xdrproc_t func, void *data)
130 {
131 XDR x;
132 struct xdr_ops ops;
133 bool_t stat;
134 /* to stop ANSI-C compiler from complaining */
135 typedef bool_t (*dummyfunc1) (XDR *, long *);
136 typedef bool_t (*dummyfunc2) (XDR *, caddr_t, u_int);
137 typedef bool_t (*dummyfunc3) (XDR *, int32_t *);
138
139 ops.x_putlong = x_putlong;
140 ops.x_putbytes = x_putbytes;
141 ops.x_inline = x_inline;
142 ops.x_getpostn = x_getpostn;
143 ops.x_setpostn = x_setpostn;
144 ops.x_destroy = x_destroy;
145 ops.x_putint32 = x_putint32;
146
147 /* the other harmless ones */
148 ops.x_getlong = (dummyfunc1) harmless;
149 ops.x_getbytes = (dummyfunc2) harmless;
150 ops.x_getint32 = (dummyfunc3) harmless;
151
152 x.x_op = XDR_ENCODE;
153 x.x_ops = &ops;
154 x.x_handy = 0;
155 x.x_private = (caddr_t) NULL;
156 x.x_base = (caddr_t) 0;
157
158 stat = func (&x, data);
159 free (x.x_private);
160 return stat == TRUE ? x.x_handy : 0;
161 }
162 #ifdef EXPORT_RPC_SYMBOLS
163 libc_hidden_def (xdr_sizeof)
164 #else
165 libc_hidden_nolink_sunrpc (xdr_sizeof, GLIBC_2_1)
166 #endif