Whamcloud - gitweb
fd4e390f5d081617aa4fad7fb0e1ddb55a9328ae
[fs/lustre-release.git] / lustre / ptlrpc / gss / gss_rawobj.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 only,
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License version 2 for more details (a copy is included
14  * in the LICENSE file that accompanied this code).
15  *
16  * You should have received a copy of the GNU General Public License
17  * version 2 along with this program; If not, see
18  * http://www.gnu.org/licenses/gpl-2.0.html
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Use is subject to license terms.
25  *
26  * Copyright (c) 2011, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  *
31  * lustre/ptlrpc/gss/gss_rawobj.c
32  *
33  * Author: Eric Mei <ericm@clusterfs.com>
34  */
35
36 #define DEBUG_SUBSYSTEM S_SEC
37
38 #include <linux/mutex.h>
39
40 #include <obd.h>
41 #include <obd_class.h>
42 #include <obd_support.h>
43 #include <lustre_sec.h>
44
45 #include "gss_internal.h"
46
47 int rawobj_empty(rawobj_t *obj)
48 {
49         LASSERT(equi(obj->len, obj->data));
50         return (obj->len == 0);
51 }
52
53 int rawobj_alloc(rawobj_t *obj, char *buf, int len)
54 {
55         LASSERT(obj);
56         LASSERT(len >= 0);
57
58         obj->len = len;
59         if (len) {
60                 OBD_ALLOC_LARGE(obj->data, len);
61                 if (!obj->data) {
62                         obj->len = 0;
63                         RETURN(-ENOMEM);
64                 }
65                 memcpy(obj->data, buf, len);
66         } else
67                 obj->data = NULL;
68         return 0;
69 }
70
71 void rawobj_free(rawobj_t *obj)
72 {
73         LASSERT(obj);
74
75         if (obj->len) {
76                 LASSERT(obj->data);
77                 OBD_FREE_LARGE(obj->data, obj->len);
78                 obj->len = 0;
79                 obj->data = NULL;
80         } else
81                 LASSERT(!obj->data);
82 }
83
84 int rawobj_equal(rawobj_t *a, rawobj_t *b)
85 {
86         LASSERT(a && b);
87
88         return (a->len == b->len &&
89                 (!a->len || !memcmp(a->data, b->data, a->len)));
90 }
91
92 int rawobj_dup(rawobj_t *dest, rawobj_t *src)
93 {
94         LASSERT(src && dest);
95
96         dest->len = src->len;
97         if (dest->len) {
98                 OBD_ALLOC_LARGE(dest->data, dest->len);
99                 if (!dest->data) {
100                         dest->len = 0;
101                         return -ENOMEM;
102                 }
103                 memcpy(dest->data, src->data, dest->len);
104         } else
105                 dest->data = NULL;
106         return 0;
107 }
108
109 int rawobj_serialize(rawobj_t *obj, __u32 **buf, __u32 *buflen)
110 {
111         __u32 len;
112
113         LASSERT(obj);
114         LASSERT(buf);
115         LASSERT(buflen);
116
117         len = cfs_size_round4(obj->len);
118
119         if (*buflen < 4 + len) {
120                 CERROR("buflen %u <  %u\n", *buflen, 4 + len);
121                 return -EINVAL;
122         }
123
124         *(*buf)++ = cpu_to_le32(obj->len);
125         memcpy(*buf, obj->data, obj->len);
126         *buf += (len >> 2);
127         *buflen -= (4 + len);
128
129         return 0;
130 }
131
132 static int __rawobj_extract(rawobj_t *obj, __u32 **buf, __u32 *buflen,
133                             int alloc, int local)
134 {
135         __u32 len;
136
137         if (*buflen < sizeof(__u32)) {
138                 CERROR("buflen %u\n", *buflen);
139                 return -EINVAL;
140         }
141
142         obj->len = *(*buf)++;
143         if (!local)
144                 obj->len = le32_to_cpu(obj->len);
145         *buflen -= sizeof(__u32);
146
147         if (!obj->len) {
148                 obj->data = NULL;
149                 return 0;
150         }
151
152         len = local ? obj->len : cfs_size_round4(obj->len);
153         if (*buflen < len) {
154                 CERROR("buflen %u < %u\n", *buflen, len);
155                 obj->len = 0;
156                 return -EINVAL;
157         }
158
159         if (!alloc)
160                 obj->data = (__u8 *) *buf;
161         else {
162                 OBD_ALLOC_LARGE(obj->data, obj->len);
163                 if (!obj->data) {
164                         CERROR("fail to alloc %u bytes\n", obj->len);
165                         obj->len = 0;
166                         return -ENOMEM;
167                 }
168                 memcpy(obj->data, *buf, obj->len);
169         }
170
171         *((char **)buf) += len;
172         *buflen -= len;
173
174         return 0;
175 }
176
177 int rawobj_extract(rawobj_t *obj, __u32 **buf, __u32 *buflen)
178 {
179         return __rawobj_extract(obj, buf, buflen, 0, 0);
180 }
181
182 int rawobj_extract_alloc(rawobj_t *obj, __u32 **buf, __u32 *buflen)
183 {
184         return __rawobj_extract(obj, buf, buflen, 1, 0);
185 }
186
187 int rawobj_extract_local(rawobj_t *obj, __u32 **buf, __u32 *buflen)
188 {
189         return __rawobj_extract(obj, buf, buflen, 0, 1);
190 }
191
192 int rawobj_extract_local_alloc(rawobj_t *obj, __u32 **buf, __u32 *buflen)
193 {
194         return __rawobj_extract(obj, buf, buflen, 1, 1);
195 }
196
197 int rawobj_from_netobj(rawobj_t *rawobj, netobj_t *netobj)
198 {
199         rawobj->len = netobj->len;
200         rawobj->data = netobj->data;
201         return 0;
202 }
203
204 int rawobj_from_netobj_alloc(rawobj_t *rawobj, netobj_t *netobj)
205 {
206         rawobj->len = 0;
207         rawobj->data = NULL;
208
209         if (netobj->len == 0)
210                 return 0;
211                 
212         OBD_ALLOC_LARGE(rawobj->data, netobj->len);
213         if (rawobj->data == NULL)
214                 return -ENOMEM;
215
216         rawobj->len = netobj->len;
217         memcpy(rawobj->data, netobj->data, netobj->len);
218         return 0;
219 }
220
221 /****************************************
222  * misc more                            *
223  ****************************************/
224
225 int buffer_extract_bytes(const void **buf, __u32 *buflen,
226                          void *res, __u32 reslen)
227 {
228         if (*buflen < reslen) {
229                 CERROR("buflen %u < %u\n", *buflen, reslen);
230                 return -EINVAL;
231         }
232
233         memcpy(res, *buf, reslen);
234         *buf += reslen;
235         *buflen -= reslen;
236         return 0;
237 }