view array.h @ 136:59680a727e9d

Improve previous. Just in case we ever crash and reach cleanup() while processing an -include foo option, take the array entry for it out of the array to make sure it doesn't get freed twice. This case shouldn't be reachable, but it's better to be safe.
author David A. Holland
date Tue, 09 Jul 2013 13:38:43 -0400
parents 0921c47b4f22
children c440247cbd69
line wrap: on
line source

/*-
 * Copyright (c) 2009 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by David A. Holland.
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 THE FOUNDATION 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.
 */

#ifndef ARRAY_H
#define ARRAY_H

#include "inlinedefs.h" // XXX
#include "utils.h"

#define ARRAYS_CHECKED

#ifdef ARRAYS_CHECKED
#include <assert.h>
#define arrayassert assert
#else
#define arrayassert(x) ((void)(x))
#endif

#ifndef ARRAYINLINE
#define ARRAYINLINE C99INLINE
#endif

////////////////////////////////////////////////////////////
// type and base operations

struct array {
	void **v;
	unsigned num, max;
};

struct array *array_create(void);
void array_destroy(struct array *);
void array_init(struct array *);
void array_cleanup(struct array *);
ARRAYINLINE unsigned array_num(const struct array *);
ARRAYINLINE void *array_get(const struct array *, unsigned index_);
ARRAYINLINE void array_set(const struct array *, unsigned index_, void *val);
void array_setsize(struct array *, unsigned num);
ARRAYINLINE void array_add(struct array *, void *val, unsigned *index_ret);
void array_insert(struct array *a, unsigned index_);
void array_remove(struct array *a, unsigned index_);

////////////////////////////////////////////////////////////
// inlining for base operations

ARRAYINLINE unsigned
array_num(const struct array *a)
{
	return a->num;
}

ARRAYINLINE void *
array_get(const struct array *a, unsigned index_)
{
	arrayassert(index_ < a->num);
	return a->v[index_];
}

ARRAYINLINE void
array_set(const struct array *a, unsigned index_, void *val)
{
	arrayassert(index_ < a->num);
	a->v[index_] = val;
}

ARRAYINLINE void
array_add(struct array *a, void *val, unsigned *index_ret)
{
	unsigned index_ = a->num;
	array_setsize(a, index_+1);
	a->v[index_] = val;
	if (index_ret != NULL) {
		*index_ret = index_;
	}
}

////////////////////////////////////////////////////////////
// bits for declaring and defining typed arrays

/*
 * Usage:
 *
 * DECLARRAY_BYTYPE(foo, bar, INLINE) declares "struct foo", which is
 * an array of pointers to "bar", plus the operations on it.
 *
 * DECLARRAY(foo, INLINE) is equivalent to
 * DECLARRAY_BYTYPE(fooarray, struct foo, INLINE).
 *
 * DEFARRAY_BYTYPE and DEFARRAY are the same as DECLARRAY except that
 * they define the operations.
 *
 * The argument INLINE can be used as follows:
 *
 * 1. For no inlining:
 *    In foo.h:
 *           DECLARRAY(foo, );
 *    In foo.c:
 *           DEFARRAY(foo, );
 *
 * 2. To be file-static:
 *    In foo.c:
 *           DECLARRAY(foo, static);
 *           DEFARRAY(foo, static);
 *
 * 3. To inline using C99:
 *    In foo.h:
 *           DECLARRAY(foo, inline);
 *           DEFARRAY(foo, inline);
 *
 * 4. To inline with old gcc:
 *    In foo.h:
 *           #ifndef FOO_INLINE
 *           #define FOO_INLINE extern inline
 *           #endif
 *           DECLARRAY(foo, );
 *           DEFARRAY(foo, FOO_INLINE);
 *    In foo.c:
 *           #define FOO_INLINE
 *           #include "foo.h"
 *
 * 5. To inline such that it works both with old gcc and C99:
 *    In foo.h:
 *           #ifndef FOO_INLINE
 *           #define FOO_INLINE extern inline
 *           #endif
 *           DECLARRAY(foo, FOO_INLINE);
 *           DEFARRAY(foo, FOO_INLINE);
 *    In foo.c:
 *           #define FOO_INLINE
 *           #include "foo.h"
 *
 * The mechanism in case (4) ensures that an externally linkable
 * definition exists.
 */

#define DECLARRAY_BYTYPE(ARRAY, T, INLINE) \
	struct ARRAY {							\
		struct array arr;					\
	};								\
									\
	INLINE struct ARRAY *ARRAY##_create(void);			\
	INLINE void ARRAY##_destroy(struct ARRAY *a);			\
	INLINE void ARRAY##_init(struct ARRAY *a);			\
	INLINE void ARRAY##_cleanup(struct ARRAY *a);			\
	INLINE unsigned ARRAY##_num(const struct ARRAY *a);		\
	INLINE T *ARRAY##_get(const struct ARRAY *a, unsigned index_);	\
	INLINE void ARRAY##_set(struct ARRAY *a, unsigned index_, T *val); \
	INLINE void ARRAY##_setsize(struct ARRAY *a, unsigned num);	\
	INLINE void ARRAY##_add(struct ARRAY *a, T *val, unsigned *index_ret);\
	INLINE void ARRAY##_insert(struct ARRAY *a, unsigned index_);	\
	INLINE void ARRAY##_remove(struct ARRAY *a, unsigned index_)


#define DEFARRAY_BYTYPE(ARRAY, T, INLINE) \
	INLINE void						\
	ARRAY##_init(struct ARRAY *a)				\
	{							\
		array_init(&a->arr);				\
	}							\
								\
	INLINE void						\
	ARRAY##_cleanup(struct ARRAY *a)			\
	{							\
		array_cleanup(&a->arr);				\
	}							\
								\
	INLINE struct						\
	ARRAY *ARRAY##_create(void)				\
	{							\
		struct ARRAY *a;				\
								\
		a = domalloc(sizeof(*a));			\
		ARRAY##_init(a);				\
		return a;					\
	}							\
								\
	INLINE void						\
	ARRAY##_destroy(struct ARRAY *a)			\
	{							\
		ARRAY##_cleanup(a);				\
		dofree(a, sizeof(*a));				\
	}							\
								\
	INLINE unsigned						\
	ARRAY##_num(const struct ARRAY *a)			\
	{							\
		return array_num(&a->arr);			\
	}							\
								\
	INLINE T *						\
	ARRAY##_get(const struct ARRAY *a, unsigned index_)	\
	{				 			\
		return (T *)array_get(&a->arr, index_);		\
	}							\
								\
	INLINE void						\
	ARRAY##_set(struct ARRAY *a, unsigned index_, T *val)	\
	{				 			\
		array_set(&a->arr, index_, (void *)val);	\
	}							\
								\
	INLINE void						\
	ARRAY##_setsize(struct ARRAY *a, unsigned num)		\
	{				 			\
		array_setsize(&a->arr, num);			\
	}							\
								\
	INLINE void						\
	ARRAY##_add(struct ARRAY *a, T *val, unsigned *ret)	\
	{				 			\
		array_add(&a->arr, (void *)val, ret);		\
	}							\
								\
	INLINE void						\
	ARRAY##_insert(struct ARRAY *a, unsigned index_)	\
	{				 			\
		array_insert(&a->arr, index_);			\
	}							\
								\
	INLINE void						\
	ARRAY##_remove(struct ARRAY *a, unsigned index_)	\
	{				 			\
		array_remove(&a->arr, index_);			\
	}

#define DECLARRAY(T, INLINE) DECLARRAY_BYTYPE(T##array, struct T, INLINE)
#define DEFARRAY(T, INLINE) DEFARRAY_BYTYPE(T##array, struct T, INLINE)

#define DESTROYALL_ARRAY(T, INLINE) \
	void T##array_destroyall(struct T##array *arr);	\
							\
	INLINE void					\
	T##array_destroyall(struct T##array *arr)	\
	{						\
		unsigned i, num;			\
		struct T *t;				\
							\
		num = T##array_num(arr);		\
		for (i=0; i<num; i++) {			\
			t = T##array_get(arr, i);	\
			T##_destroy(t);			\
		}					\
		T##array_setsize(arr, 0);		\
	}


////////////////////////////////////////////////////////////
// basic array types

DECLARRAY_BYTYPE(stringarray, char, ARRAYINLINE);
DEFARRAY_BYTYPE(stringarray, char, ARRAYINLINE);

#endif /* ARRAY_H */