Lustre Repository Browser

Viewing: gss_rawobj.c

// SPDX-License-Identifier: GPL-2.0

/*
 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
 * Use is subject to license terms.
 *
 * Copyright (c) 2011, Intel Corporation.
 */

/*
 * This file is part of Lustre, http://www.lustre.org/
 *
 * Author: Eric Mei <ericm@clusterfs.com>
 */

#define DEBUG_SUBSYSTEM S_SEC

#include <linux/mutex.h>

#include <obd.h>
#include <obd_class.h>
#include <obd_support.h>
#include <lustre_sec.h>

#include "gss_internal.h"

int rawobj_empty(rawobj_t *obj)
{
        LASSERT(equi(obj->len, obj->data));
        return (obj->len == 0);
}

int rawobj_alloc(rawobj_t *obj, char *buf, int len)
{
        LASSERT(obj);
        LASSERT(len >= 0);

        obj->len = len;
        if (len) {
                OBD_ALLOC_LARGE(obj->data, len);
                if (!obj->data) {
                        obj->len = 0;
                        RETURN(-ENOMEM);
                }
                memcpy(obj->data, buf, len);
        } else
                obj->data = NULL;
        return 0;
}

void rawobj_free(rawobj_t *obj)
{
        LASSERT(obj);

        if (obj->len) {
                LASSERT(obj->data);
                OBD_FREE_LARGE(obj->data, obj->len);
                obj->len = 0;
                obj->data = NULL;
        } else
                LASSERT(!obj->data);
}

int rawobj_equal(rawobj_t *a, rawobj_t *b)
{
        LASSERT(a && b);

        return (a->len == b->len &&
                (!a->len || !memcmp(a->data, b->data, a->len)));
}

int rawobj_dup(rawobj_t *dest, rawobj_t *src)
{
        LASSERT(src && dest);

        dest->len = src->len;
        if (dest->len) {
                OBD_ALLOC_LARGE(dest->data, dest->len);
                if (!dest->data) {
                        dest->len = 0;
                        return -ENOMEM;
                }
                memcpy(dest->data, src->data, dest->len);
        } else
                dest->data = NULL;
        return 0;
}

int rawobj_serialize(rawobj_t *obj, __u32 **buf, __u32 *buflen)
{
        __u32 len;

        LASSERT(obj);
        LASSERT(buf);
        LASSERT(buflen);

	len = round_up(obj->len, 4);

        if (*buflen < 4 + len) {
                CERROR("shorter buflen than needed: %u < %u\n",
                        *buflen, 4 + len);
                return -EINVAL;
        }

        *(*buf)++ = cpu_to_le32(obj->len);
        memcpy(*buf, obj->data, obj->len);
        *buf += (len >> 2);
        *buflen -= (4 + len);

        return 0;
}

static int __rawobj_extract(rawobj_t *obj, __u32 **buf, __u32 *buflen,
                            int alloc, int local)
{
        __u32 len;

        if (*buflen < sizeof(__u32)) {
                CERROR("too short buflen: %u\n", *buflen);
                return -EINVAL;
        }

        obj->len = *(*buf)++;
        if (!local)
                obj->len = le32_to_cpu(obj->len);
        *buflen -= sizeof(__u32);

        if (!obj->len) {
                obj->data = NULL;
                return 0;
        }

	len = local ? obj->len : round_up(obj->len, 4);
        if (*buflen < len) {
                CERROR("shorter buflen than object size: %u < %u\n",
                        *buflen, len);
                obj->len = 0;
                return -EINVAL;
        }

        if (!alloc)
                obj->data = (__u8 *) *buf;
        else {
                OBD_ALLOC_LARGE(obj->data, obj->len);
                if (!obj->data) {
                        CERROR("fail to alloc %u bytes\n", obj->len);
                        obj->len = 0;
                        return -ENOMEM;
                }
                memcpy(obj->data, *buf, obj->len);
        }

        *((char **)buf) += len;
        *buflen -= len;

        return 0;
}

int rawobj_extract(rawobj_t *obj, __u32 **buf, __u32 *buflen)
{
        return __rawobj_extract(obj, buf, buflen, 0, 0);
}

int rawobj_extract_alloc(rawobj_t *obj, __u32 **buf, __u32 *buflen)
{
        return __rawobj_extract(obj, buf, buflen, 1, 0);
}

int rawobj_extract_local(rawobj_t *obj, __u32 **buf, __u32 *buflen)
{
        return __rawobj_extract(obj, buf, buflen, 0, 1);
}

int rawobj_extract_local_alloc(rawobj_t *obj, __u32 **buf, __u32 *buflen)
{
        return __rawobj_extract(obj, buf, buflen, 1, 1);
}

int rawobj_from_netobj(rawobj_t *rawobj, netobj_t *netobj)
{
        rawobj->len = netobj->len;
        rawobj->data = netobj->data;
        return 0;
}

int rawobj_from_netobj_alloc(rawobj_t *rawobj, netobj_t *netobj)
{
        rawobj->len = 0;
        rawobj->data = NULL;

        if (netobj->len == 0)
                return 0;

        OBD_ALLOC_LARGE(rawobj->data, netobj->len);
        if (rawobj->data == NULL)
                return -ENOMEM;

        rawobj->len = netobj->len;
        memcpy(rawobj->data, netobj->data, netobj->len);
        return 0;
}

/****************************************
 * misc more                            *
 ****************************************/

int buffer_extract_bytes(const void **buf, __u32 *buflen,
                         void *res, __u32 reslen)
{
        if (*buflen < reslen) {
                CERROR("shorter buflen than expected: %u < %u\n",
                        *buflen, reslen);
                return -EINVAL;
        }

        memcpy(res, *buf, reslen);
        *buf += reslen;
        *buflen -= reslen;
        return 0;
}