Home | History | Annotate | Line # | Download | only in citrus
citrus_pivot_factory.c revision 1.2
      1 /*	$NetBSD: citrus_pivot_factory.c,v 1.2 2003/06/30 12:32:21 tshiozak Exp $	*/
      2 
      3 /*-
      4  * Copyright (c)2003 Citrus Project,
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     26  * SUCH DAMAGE.
     27  */
     28 
     29 #if HAVE_CONFIG_H
     30 #include "config.h"
     31 #endif
     32 
     33 #include <sys/cdefs.h>
     34 #if defined(LIBC_SCCS) && !defined(lint)
     35 __RCSID("$NetBSD: citrus_pivot_factory.c,v 1.2 2003/06/30 12:32:21 tshiozak Exp $");
     36 #endif /* LIBC_SCCS and not lint */
     37 
     38 #include <assert.h>
     39 #include <stdio.h>
     40 #include <stdlib.h>
     41 #include <string.h>
     42 #include <errno.h>
     43 #include <ctype.h>
     44 #include <limits.h>
     45 #include <sys/queue.h>
     46 
     47 #include "citrus_namespace.h"
     48 #include "citrus_region.h"
     49 #include "citrus_bcs.h"
     50 #include "citrus_db_factory.h"
     51 #include "citrus_db_hash.h"
     52 #include "citrus_pivot_file.h"
     53 #include "citrus_pivot_factory.h"
     54 
     55 struct src_entry {
     56 	char *se_name;
     57 	struct _citrus_db_factory *se_df;
     58 	SIMPLEQ_ENTRY(src_entry) se_entry;
     59 };
     60 SIMPLEQ_HEAD(src_head, src_entry);
     61 
     62 static int
     63 find_src(struct src_head *sh, struct src_entry **rse, const char *name)
     64 {
     65 	int ret;
     66 	struct src_entry *se;
     67 
     68 	SIMPLEQ_FOREACH(se, sh, se_entry) {
     69 		if (_bcs_strcasecmp(se->se_name, name) == 0) {
     70 			*rse = se;
     71 			return 0;
     72 		}
     73 	}
     74 	se = malloc(sizeof(*se));
     75 	if (se == NULL)
     76 		return errno;
     77 	se->se_name = strdup(name);
     78 	if (se->se_name == NULL) {
     79 		ret = errno;
     80 		free(se);
     81 		return ret;
     82 	}
     83 	ret = _db_factory_create(&se->se_df, &_db_hash_std, NULL);
     84 	if (ret) {
     85 		free(se->se_name);
     86 		free(se);
     87 		return ret;
     88 	}
     89 	SIMPLEQ_INSERT_TAIL(sh, se, se_entry);
     90 	*rse = se;
     91 
     92 	return 0;
     93 }
     94 
     95 static void
     96 free_src(struct src_head *sh)
     97 {
     98 	struct src_entry *se;
     99 
    100 	while((se = SIMPLEQ_FIRST(sh)) != NULL) {
    101 		SIMPLEQ_REMOVE_HEAD(sh, se_entry);
    102 		_db_factory_free(se->se_df);
    103 		free(se->se_name);
    104 		free(se);
    105 	}
    106 }
    107 
    108 
    109 #define T_COMM '#'
    110 static int
    111 convert_line(struct src_head *sh, const char *line, size_t len)
    112 {
    113 	int ret;
    114 	struct src_entry *se;
    115 	const char *p;
    116 	char key1[LINE_MAX], key2[LINE_MAX], data[LINE_MAX];
    117 	u_int32_t val;
    118 
    119 	/* cut off trailing comment */
    120 	p = memchr(line, T_COMM, len);
    121 	if (p)
    122 		len = p - line;
    123 
    124 	/* key1 */
    125 	line = _bcs_skip_ws_len(line, &len);
    126 	if (len == 0)
    127 		return 0;
    128 	p = _bcs_skip_nonws_len(line, &len);
    129 	if (p==line)
    130 		return 0;
    131 	snprintf(key1, sizeof(key1), "%.*s", (int)(p-line), line);
    132 
    133 	/* key2 */
    134 	line = _bcs_skip_ws_len(p, &len);
    135 	if (len == 0)
    136 		return 0;
    137 	p = _bcs_skip_nonws_len(line, &len);
    138 	if (p==line)
    139 		return 0;
    140 	snprintf(key2, sizeof(key2), "%.*s", (int)(p-line), line);
    141 
    142 	/* data */
    143 	line = _bcs_skip_ws_len(p, &len);
    144 	_bcs_trunc_rws_len(line, &len);
    145 	snprintf(data, sizeof(data), "%.*s", (int)len, line);
    146 	/* LINTED: discard const */
    147 	val = strtoul(data, (char **)&p, 0);
    148 	if (*p != '\0')
    149 		return EFTYPE;
    150 
    151 	/* insert to DB */
    152 	ret = find_src(sh, &se, key1);
    153 	if (ret)
    154 		return ret;
    155 
    156 	return _db_factory_add32_by_s(se->se_df, key2, val);
    157 }
    158 
    159 static int
    160 dump_db(struct src_head *sh, struct _region *r)
    161 {
    162 	int ret;
    163 	struct _db_factory *df;
    164 	struct src_entry *se;
    165 	size_t size;
    166 	void *ptr;
    167 	struct _region subr;
    168 
    169 	ret = _db_factory_create(&df, &_db_hash_std, NULL);
    170 	if (ret)
    171 		return ret;
    172 
    173 	SIMPLEQ_FOREACH(se, sh, se_entry) {
    174 		size = _db_factory_calc_size(se->se_df);
    175 		ptr = malloc(size);
    176 		if (ptr == NULL)
    177 			goto quit;
    178 		_region_init(&subr, ptr, size);
    179 		ret = _db_factory_serialize(se->se_df, _CITRUS_PIVOT_SUB_MAGIC,
    180 					    &subr);
    181 		if (ret)
    182 			goto quit;
    183 		ret = _db_factory_add_by_s(df, se->se_name, &subr, 1);
    184 		if (ret)
    185 			goto quit;
    186 	}
    187 
    188 	size = _db_factory_calc_size(df);
    189 	ptr = malloc(size);
    190 	if (ptr == NULL)
    191 		goto quit;
    192 	_region_init(r, ptr, size);
    193 
    194 	ret = _db_factory_serialize(df, _CITRUS_PIVOT_MAGIC, r);
    195 	ptr = NULL;
    196 
    197 quit:
    198 	free(ptr);
    199 	_db_factory_free(df);
    200 	return ret;
    201 }
    202 
    203 int
    204 _citrus_pivot_factory_convert(FILE *out, FILE *in)
    205 {
    206 	struct src_head sh;
    207 	struct _region r;
    208 	char *line;
    209 	size_t size;
    210 	int ret;
    211 
    212 	SIMPLEQ_INIT(&sh);
    213 
    214 	while ((line = fgetln(in, &size)) != NULL)
    215 		if ((ret = convert_line(&sh, line, size))) {
    216 			free_src(&sh);
    217 			return ret;
    218 		}
    219 
    220 	ret = dump_db(&sh, &r);
    221 	free_src(&sh);
    222 	if (ret)
    223 		return ret;
    224 
    225 	if (fwrite(_region_head(&r), _region_size(&r), 1, out) != 1)
    226 		return errno;
    227 
    228 	return 0;
    229 }
    230