/*-
 * SPDX-License-Identifier: BSD-3-Clause
 *
 * Copyright (c) 2004, PADL Software Pty Ltd.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * 3. Neither the name of PADL Software nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY PADL SOFTWARE AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL PADL SOFTWARE OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <gssapi/gssapi.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>

/* RCSID("$Id: gss_buffer_set.c 18885 2006-10-24 21:53:02Z lha $"); */

OM_uint32
gss_create_empty_buffer_set(OM_uint32 * minor_status,
    gss_buffer_set_t *buffer_set)
{
	gss_buffer_set_t set;

	set = (gss_buffer_set_desc *) malloc(sizeof(*set));
	if (set == GSS_C_NO_BUFFER_SET) {
		*minor_status = ENOMEM;
		return (GSS_S_FAILURE);
	}

	set->count = 0;
	set->elements = NULL;

	*buffer_set = set;

	*minor_status = 0;
	return (GSS_S_COMPLETE);
}

OM_uint32
gss_add_buffer_set_member(OM_uint32 * minor_status,
    const gss_buffer_t member_buffer, gss_buffer_set_t *buffer_set)
{
	gss_buffer_set_t set;
	gss_buffer_t p;
	OM_uint32 ret;

	if (*buffer_set == GSS_C_NO_BUFFER_SET) {
		ret = gss_create_empty_buffer_set(minor_status,
		    buffer_set);
		if (ret) {
			return (ret);
		}
	}

	set = *buffer_set;
	set->elements = reallocarray(set->elements, set->count + 1,
	    sizeof(set->elements[0]));
	if (set->elements == NULL) {
		*minor_status = ENOMEM;
		return (GSS_S_FAILURE);
	}

	p = &set->elements[set->count];

	p->value = malloc(member_buffer->length);
	if (p->value == NULL) {
		*minor_status = ENOMEM;
		return (GSS_S_FAILURE);
	}
	memcpy(p->value, member_buffer->value, member_buffer->length);
	p->length = member_buffer->length;

	set->count++;

	*minor_status = 0;
	return (GSS_S_COMPLETE);
}

OM_uint32
gss_release_buffer_set(OM_uint32 * minor_status, gss_buffer_set_t *buffer_set)
{
	size_t i;
	OM_uint32 minor;

	*minor_status = 0;

	if (*buffer_set == GSS_C_NO_BUFFER_SET)
		return (GSS_S_COMPLETE);

	for (i = 0; i < (*buffer_set)->count; i++)
		gss_release_buffer(&minor, &((*buffer_set)->elements[i]));

	free((*buffer_set)->elements);

	(*buffer_set)->elements = NULL;
	(*buffer_set)->count = 0;

	free(*buffer_set);
	*buffer_set = GSS_C_NO_BUFFER_SET;

	return (GSS_S_COMPLETE);
}