unit-tests/regex/parse_t.c

/*
 * Copyright (C) 2001-2004 Sistina Software, Inc. All rights reserved.
 * Copyright (C) 2004-2010 Red Hat, Inc. All rights reserved.
 *
 * This file is part of LVM2.
 *
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU General Public License v.2.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/* hack - using unexported internal function */
#define DEBUG
#include "regex/parse_rx.c"

#include <stdio.h>
#include <ctype.h>

static void _pretty_print(struct rx_node *rx, int depth)
{
	int i;
	for (i = 0; i < depth; i++)
		printf(" ");

	/* display info about the node */
	switch (rx->type) {
	case CAT:
		printf("Cat");
		break;

	case OR:
		printf("Or");
		break;

	case STAR:
		printf("Star");
		break;

	case PLUS:
		printf("Plus");
		break;

	case QUEST:
		printf("Quest");
		break;

	case CHARSET:
		printf("Charset : ");
		for (i = 0; i < 256; i++) {
			if (dm_bit(rx->charset, i) && isprint(i))
				printf("%c", (char) i);
		}
		break;

	default:
		printf("Unknown type");
	}
	printf("\n");

	if (rx->left)
		_pretty_print(rx->left, depth + 1);

	if (rx->right)
		_pretty_print(rx->right, depth + 1);
}

int main(int argc, char **argv)
{
	struct dm_pool *mem;
	struct rx_node *rx;
	int regex_print = 0;
	int show_nodes = 0;
	int regex_arg = 1;

	if (argc == 3 && !strcmp(argv[1], "-r")) {
		regex_print++;
		regex_arg++;
		argc--;
	}

	if (argc == 3 && !strcmp(argv[1], "-R")) {
		regex_print++;
		show_nodes++;
		regex_arg++;
		argc--;
	}

	if (argc != 2) {
		fprintf(stderr, "Usage : %s [-r] <regex>\n", argv[0]);
		exit(0);
	}

	dm_log_init_verbose(_LOG_DEBUG);

	if (!(mem = dm_pool_create("parse_regex", 1024))) {
		fprintf(stderr, "Couldn't create pool\n");
		exit(1);
	}

	if (!(rx = rx_parse_str(mem, argv[regex_arg]))) {
		dm_pool_destroy(mem);
		fprintf(stderr, "Couldn't parse regex\n");
		exit(1);
	}

	if (regex_print)
		_regex_print(rx, 0, show_nodes);
	else
		_pretty_print(rx, 0);

	dm_pool_destroy(mem);

	return 0;
}