/*
 * Copyright (C) 1998  Internet Software Consortium.
 * 
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
 * SOFTWARE.
 */

 /* $Id: gen.c,v 1.7 1999/01/20 05:20:18 marka Exp $ */

#include <sys/types.h>

#include <ctype.h>
#include <dirent.h>
#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>


#define FROMTEXTDECL "dns_rdataclass_t class, dns_rdatatype_t type, isc_lex_t *lexer, dns_name_t *origin, isc_boolean_t downcase, isc_buffer_t *target"
#define FROMTEXTARGS "class, type, lexer, origin, downcase, target"
#define FROMTEXTCLASS "class"
#define FROMTEXTTYPE "type"
#define FROMTEXTDEF "use_default = ISC_TRUE"

#define TOTEXTDECL "dns_rdata_t *rdata, dns_name_t *origin, isc_buffer_t *target"
#define TOTEXTARGS "rdata, origin, target"
#define TOTEXTCLASS "rdata->class"
#define TOTEXTTYPE "rdata->type"
#define TOTEXTDEF "use_default = ISC_TRUE"

#define FROMWIREDECL "dns_rdataclass_t class, dns_rdatatype_t type, isc_buffer_t *source, dns_decompress_t *dctx, isc_boolean_t downcase, isc_buffer_t *target"
#define FROMWIREARGS "class, type, source, dctx, downcase, target"
#define FROMWIRECLASS "class"
#define FROMWIRETYPE "type"
#define FROMWIREDEF "use_default = ISC_TRUE"

#define TOWIREDECL "dns_rdata_t *rdata, dns_compress_t *cctx, isc_buffer_t *target"
#define TOWIREARGS "rdata, cctx, target"
#define TOWIRECLASS "rdata->class"
#define TOWIRETYPE "rdata->type"
#define TOWIREDEF "use_default = ISC_TRUE"

#define FROMSTRUCTDECL "dns_rdataclass_t class, dns_rdatatype_t type, void *source, isc_buffer_t *target"
#define FROMSTRUCTARGS "class, type, source, target"
#define FROMSTRUCTCLASS "class"
#define FROMSTRUCTTYPE "type"
#define FROMSTRUCTDEF "use_default = ISC_TRUE"

#define TOSTRUCTDECL "dns_rdata_t *rdata, void *target"
#define TOSTRUCTARGS "rdata, target"
#define TOSTRUCTCLASS "rdata->class"
#define TOSTRUCTTYPE "rdata->type"
#define TOSTRUCTDEF "use_default = ISC_TRUE"

#define COMPAREDECL "dns_rdata_t *rdata1, dns_rdata_t *rdata2"
#define COMPAREARGS "rdata1, rdata2"
#define COMPARECLASS "rdata1->class"
#define COMPARETYPE "rdata1->type"
#define COMPAREDEF "use_default = ISC_TRUE"

char copyright[] =
"/*\n\
 * Copyright (C) 1998%s Internet Software Consortium.\n\
 *\n\
 * Permission to use, copy, modify, and distribute this software for any\n\
 * purpose with or without fee is hereby granted, provided that the above\n\
 * copyright notice and this permission notice appear in all copies.\n\
 *\n\
 * THE SOFTWARE IS PROVIDED \"AS IS\" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS\n\
 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES\n\
 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE\n\
 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL\n\
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR\n\
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS\n\
 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS\n\
 * SOFTWARE.\n\
 */\n\
\n\
 /* THIS FILE IS AUTOMATICALLY GENERATED: DO NOT EDIT */\n\
\n";

struct cc {
	struct cc *next;
	int type;
} *classes;

struct tt {
	struct tt *next;
	int class;
	int type;
	char classname[11];
	char typename[11];
	char dirname[sizeof "rdata/0123456789_65535" ];
} *types;

char *	upper(char *);
void	doswitch(char *, char *, char *, char *, char *, char *);
void	dodecl(char *, char *, char *);
void	add(int, char *, int, char *, char *);
void	sd(int, char *, char *);

char *
upper(char *s) {
	static char buf[11];
	char *b = buf;
	char c;

	while ((c = *s++)) {
		
		*b++ = islower(c) ? toupper(c) : c;
	}
	*b = '\0';
	return (buf);
}

void
doswitch(char *name, char *function, char *args,
	 char *tsw, char *csw, char *res)
{
	struct tt *tt;
	int first = 1;
	int lasttype = 0;
	int subswitch = 0;

	for (tt = types; tt != NULL ; tt = tt->next) {
		if (first) {
			fprintf(stdout, "\n#define %s \\\n", name);
			fprintf(stdout, "\tswitch (%s) { \\\n" /*}*/, tsw);
			first = 0;
		}
		if (tt->type != lasttype && subswitch) {
			fprintf(stdout, "\t\tdefault: %s; break; \\\n", res);
			fputs(/*{*/ "\t\t} \\\n", stdout);
			fputs("\t\tbreak; \\\n", stdout);
			subswitch = 0;
		}
		if (tt->class && tt->type != lasttype) {
			fprintf(stdout, "\tcase %d: switch (%s) { \\\n" /*}*/,
				tt->type, csw);
			subswitch = 1;
		}
		if (tt->class == 0)
			fprintf(stdout,
				"\tcase %d: result = %s_%s(%s); break;",
				tt->type, function, tt->typename, args);
		else
			fprintf(stdout,
			        "\t\tcase %d: result = %s_%s_%s(%s); break;",
				tt->class, function, tt->classname,
				tt->typename, args);
		fputs(" \\\n", stdout);
		lasttype = tt->type;
	}
	if (subswitch) {
		fprintf(stdout, "\t\tdefault: %s; break; \\\n", res);
		fputs(/*{*/ "\t\t}\n", stdout);
		fputs("\t\tbreak; \\\n", stdout);
	}
	if (first)
		fprintf(stdout, "\n#define %s %s;\n", name, res);
	else {
		fprintf(stdout, "\tdefault: %s; break; \\\n", res);
		fputs(/*{*/ "\t}\n", stdout);
	}
}

void
dodecl(char *type, char *function, char *args) {
	struct tt *tt;

	fputs("\n", stdout);
	for (tt = types; tt ; tt = tt->next)
		if (tt->class)
			fprintf(stdout,
				"static %s %s_%s_%s(%s);\n",
				type, function, tt->classname,
				tt->typename, args);
		else
			fprintf(stdout,
				"static %s %s_%s(%s);\n",
				type, function, tt->typename, args);
}

void
add(int class, char *classname, int type, char *typename, char *dirname) {
	struct tt *newtt = (struct tt *)malloc(sizeof *newtt);
	struct tt *tt, *oldtt;
	struct cc *newcc;
	struct cc *cc, *oldcc;


	if (newtt == NULL)
		exit(1);

	newtt->next = NULL;
	newtt->class = class;
	newtt->type = type;
	strcpy(newtt->classname, classname);
	strcpy(newtt->typename, typename);
	strcpy(newtt->dirname, dirname);

	tt = types;
	oldtt = NULL;

	while ((tt != NULL) && (tt->type < type)) {
		oldtt = tt;
		tt = tt->next;
	}

	while ((tt != NULL) && (tt->type == type) && (tt->class < class)) {
		if (strcmp(tt->typename, typename) != 0)
			exit(1);
		oldtt = tt;
		tt = tt->next;
	}

	if ((tt != NULL) && (tt->type == type) && (tt->class == class))
		exit(1);

	newtt->next = tt;
	if (oldtt != NULL)
		oldtt->next = newtt;
	else
		types = newtt;

	/* do a class switch for this type */
	 
	if (class == 0)
		return;

	newcc = (struct cc *)malloc(sizeof *newcc);
	newcc->type = type;
	cc = classes;
	oldcc = NULL;
	
	while ((cc != NULL) && (cc->type < type)) {
		oldcc = cc;
		cc = cc->next;
	}

	if ((cc != NULL) && cc->type == type) {
		free((char *)newcc);
		return;
	}

	newcc->next = cc;
	if (oldcc != NULL)
		oldcc->next = newcc;
	else
		classes = newcc;
}

void
sd(int class, char *classname, char *dir) {
	char buf[sizeof "0123456789_65535.h"];
	DIR *d;
	int type;
	char typename[11];
	struct dirent *dp;

	if ((d = opendir(dir)) == NULL)
		return;

	while ((dp = readdir(d)) != NULL) {
		if (sscanf(dp->d_name, "%10[0-9a-z]_%d.h",
			   typename, &type) != 2)
			continue;
		if ((type > 65535) || (type < 0))
			continue;

		sprintf(buf, "%s_%d.h", typename, type);
		if (strcmp(buf, dp->d_name) != 0)
			continue;
		add(class, classname, type, typename, dir);
	}
	closedir(d);
}

int
main(int argc, char **argv) {
	DIR *d;
	char buf[sizeof "rdata/0123456789_65535" ];
	int class;
	char classname[11];
	struct dirent *dp;
	struct tt *tt;
	struct tm *tm;
	time_t now;
	char year[11];
	int lasttype;

	argc = argc;
	argv = argv;

	if ((d = opendir("rdata")) == NULL)
		exit(1);

	while ((dp = readdir(d)) != NULL) {
		if (sscanf(dp->d_name, "%10[0-9a-z]_%d",
			   classname, &class) != 2)
			continue;
		if ((class > 65535) || (class < 0))
			continue;

		sprintf(buf, "rdata/%s_%d", classname, class);
		if (strcmp(buf + 6, dp->d_name) != 0)
			continue;
		sd(class, classname, buf);
	}
	sd(0, "", "rdata/generic");
	closedir(d);

	if (time(&now) != -1) {
		if ((tm = localtime(&now)) != NULL && tm->tm_year > 98)
			sprintf(year, "-%d", tm->tm_year + 1900);
		else
			year[0] = 0;
	} else
		year[0] = 0;

	fprintf(stdout, copyright, year);

	dodecl("dns_result_t", "fromtext", FROMTEXTDECL);
	dodecl("dns_result_t", "totext", TOTEXTDECL);
	dodecl("dns_result_t", "fromwire", FROMWIREDECL);
	dodecl("dns_result_t", "towire", TOWIREDECL);
	dodecl("int", "compare", COMPAREDECL);
	dodecl("dns_result_t", "fromstruct", FROMSTRUCTDECL);
	dodecl("dns_result_t", "tostruct", TOSTRUCTDECL);

	doswitch("FROMTEXTSWITCH", "fromtext", FROMTEXTARGS,
		 FROMTEXTTYPE, FROMTEXTCLASS, FROMTEXTDEF);
	doswitch("TOTEXTSWITCH", "totext", TOTEXTARGS,
		 TOTEXTTYPE, TOTEXTCLASS, TOTEXTDEF);
	doswitch("FROMWIRESWITCH", "fromwire", FROMWIREARGS,
		 FROMWIRETYPE, FROMWIRECLASS, FROMWIREDEF);
	doswitch("TOWIRESWITCH", "towire", TOWIREARGS,
		 TOWIRETYPE, TOWIRECLASS, TOWIREDEF);
	doswitch("COMPARESWITCH", "compare", COMPAREARGS,
		  COMPARETYPE, COMPARECLASS, COMPAREDEF);
	doswitch("FROMSTRUCTSWITCH", "fromstruct", FROMSTRUCTARGS,
		  FROMSTRUCTTYPE, FROMSTRUCTCLASS, FROMSTRUCTDEF);
	doswitch("TOSTRUCTSWITCH", "tostruct", TOSTRUCTARGS,
		  TOSTRUCTTYPE, TOSTRUCTCLASS, TOSTRUCTDEF);

	fprintf(stdout, "\n#define TYPENAMES%s\n",
		types != NULL ? " \\" : "");

	lasttype = 0;
	for (tt = types; tt != NULL ; tt = tt->next)
		if (tt->type != lasttype)
			fprintf(stdout,
				"\t{ %d, \"%s\" },%s\n",
				lasttype = tt->type, tt->typename,
				tt->next != NULL ? " \\" : "");

	fputs("\n", stdout);
	for (tt = types; tt ; tt = tt->next)
		fprintf(stdout, "#include \"%s/%s_%d.h\"\n",
			tt->dirname, tt->typename, tt->type);

	if (ferror(stdout) != 0)
		exit(1);

	exit(0);
}