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 }