asn1_gen.c

/*
 * Copyright (c) 2005 Kungliga Tekniska Högskolan
 * (Royal Institute of Technology, Stockholm, Sweden).
 * 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 the Institute 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 THE INSTITUTE 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 INSTITUTE 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 "der_locl.h"
#include <com_err.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <ctype.h>
#include <getarg.h>
#include <hex.h>
#include <err.h>

RCSID("$Id$");

static int
doit(const char *fn)
{
    char buf[2048];
    char *fnout = NULL;
    const char *bname;
    unsigned long line = 0;
    FILE *f, *fout;
    size_t offset = 0;

    f = fopen(fn, "r");
    if (f == NULL)
	err(1, "fopen");

    bname = strrchr(fn, '/');
    if (bname)
	bname++;
    else
	bname = fn;

    if (asprintf(&fnout, "%s.out", bname) < 0 || fnout == NULL)
	errx(1, "malloc");

    fout = fopen(fnout, "w");
    if (fout == NULL)
	err(1, "fopen: output file");

    while (fgets(buf, sizeof(buf), f) != NULL) {
	char *ptr, *class, *type, *tag, *length, *data, *foo;
	int ret, l, c, ty, ta;
	unsigned char p[6], *pdata;
	size_t sz;

	line++;

	buf[strcspn(buf, "\r\n")] = '\0';
	if (buf[0] == '#' || buf[0] == '\0')
	    continue;

	ptr = buf;
	while (isspace((unsigned char)*ptr))
	       ptr++;

	class = strtok_r(ptr, " \t\n", &foo);
	if (class == NULL) errx(1, "class missing on line %lu", line);
	type = strtok_r(NULL, " \t\n", &foo);
	if (type == NULL) errx(1, "type missing on line %lu", line);
	tag = strtok_r(NULL, " \t\n", &foo);
	if (tag == NULL) errx(1, "tag missing on line %lu", line);
	length = strtok_r(NULL, " \t\n", &foo);
	if (length == NULL) errx(1, "length missing on line %lu", line);
	data = strtok_r(NULL, " \t\n", &foo);

	c = der_get_class_num(class);
	if (c == -1) errx(1, "no valid class on line %lu", line);
	ty = der_get_type_num(type);
	if (ty == -1) errx(1, "no valid type on line %lu", line);
	ta = der_get_tag_num(tag);
	if (ta == -1)
	    ta = atoi(tag);

	l = atoi(length);

	printf("line: %3lu offset: %3lu class: %d type: %d "
	       "tag: %3d length: %3d %s\n",
	       line, (unsigned long)offset, c, ty, ta, l,
	       data ? "<have data>" : "<no data>");

	ret = der_put_length_and_tag(p + sizeof(p) - 1, sizeof(p),
				     l,
				     c,
				     ty,
				     ta,
				     &sz);
	if (ret)
	    errx(1, "der_put_length_and_tag: %d", ret);

	if (fwrite(p + sizeof(p) - sz , sz, 1, fout) != 1)
	    err(1, "fwrite length/tag failed");
	offset += sz;

	if (data) {
	    size_t datalen;

	    datalen = strlen(data) / 2;
	    pdata = emalloc(sz);

	    if (hex_decode(data, pdata, datalen) != datalen)
		errx(1, "failed to decode data");

	    if (fwrite(pdata, datalen, 1, fout) != 1)
		err(1, "fwrite data failed");
	    offset += datalen;

	    free(pdata);
	}
    }
    printf("line: eof offset: %lu\n", (unsigned long)offset);

    fclose(fout);
    fclose(f);
    return 0;
}


static int version_flag;
static int help_flag;
struct getargs args[] = {
    { "version", 0, arg_flag, &version_flag },
    { "help", 0, arg_flag, &help_flag }
};
int num_args = sizeof(args) / sizeof(args[0]);

static void
usage(int code)
{
    arg_printusage(args, num_args, NULL, "parse-file");
    exit(code);
}

int
main(int argc, char **argv)
{
    int optidx = 0;

    setprogname (argv[0]);

    if(getarg(args, num_args, argc, argv, &optidx))
	usage(1);
    if(help_flag)
	usage(0);
    if(version_flag) {
	print_version(NULL);
	exit(0);
    }
    argv += optidx;
    argc -= optidx;
    if (argc != 1)
	usage (1);

    return doit (argv[0]);
}