1/*	$NetBSD$	*/
2
3/* OpenLDAP: pkg/ldap/libraries/librewrite/rewrite.c,v 1.16.2.5 2010/04/13 20:23:09 kurt Exp */
4/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
5 *
6 * Copyright 2000-2010 The OpenLDAP Foundation.
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted only as authorized by the OpenLDAP
11 * Public License.
12 *
13 * A copy of this license is available in the file LICENSE in the
14 * top-level directory of the distribution or, alternatively, at
15 * <http://www.OpenLDAP.org/license.html>.
16 */
17/* ACKNOWLEDGEMENT:
18 * This work was initially developed by Pierangelo Masarati for
19 * inclusion in OpenLDAP Software.
20 */
21
22#include <portable.h>
23
24#include <ac/stdlib.h>
25#include <ac/string.h>
26#include <ac/syslog.h>
27#include <ac/regex.h>
28#include <ac/socket.h>
29#include <ac/unistd.h>
30#include <ac/ctype.h>
31#include <ac/string.h>
32#include <stdio.h>
33
34#include <rewrite.h>
35#include <lutil.h>
36#include <ldap.h>
37
38int ldap_debug;
39int ldap_syslog;
40int ldap_syslog_level;
41
42static void
43apply(
44		FILE *fin,
45		const char *rewriteContext,
46		const char *arg
47)
48{
49	struct rewrite_info *info;
50	char *string, *sep, *result = NULL;
51	int rc;
52	void *cookie = &info;
53
54	info = rewrite_info_init( REWRITE_MODE_ERR );
55
56	if ( rewrite_read( fin, info ) != 0 ) {
57		exit( EXIT_FAILURE );
58	}
59
60	rewrite_param_set( info, "prog", "rewrite" );
61
62	rewrite_session_init( info, cookie );
63
64	string = (char *)arg;
65	for ( sep = strchr( rewriteContext, ',' );
66			rewriteContext != NULL;
67			rewriteContext = sep,
68			sep ? sep = strchr( rewriteContext, ',' ) : NULL )
69	{
70		char	*errmsg = "";
71
72		if ( sep != NULL ) {
73			sep[ 0 ] = '\0';
74			sep++;
75		}
76		/* rc = rewrite( info, rewriteContext, string, &result ); */
77		rc = rewrite_session( info, rewriteContext, string,
78				cookie, &result );
79
80		switch ( rc ) {
81		case REWRITE_REGEXEC_OK:
82			errmsg = "ok";
83			break;
84
85		case REWRITE_REGEXEC_ERR:
86			errmsg = "error";
87			break;
88
89		case REWRITE_REGEXEC_STOP:
90			errmsg = "stop";
91			break;
92
93		case REWRITE_REGEXEC_UNWILLING:
94			errmsg = "unwilling to perform";
95			break;
96
97		default:
98			if (rc >= REWRITE_REGEXEC_USER) {
99				errmsg = "user-defined";
100			} else {
101				errmsg = "unknown";
102			}
103			break;
104		}
105
106		fprintf( stdout, "%s -> %s [%d:%s]\n", string,
107				( result ? result : "(null)" ),
108				rc, errmsg );
109		if ( result == NULL ) {
110			break;
111		}
112		if ( string != arg && string != result ) {
113			free( string );
114		}
115		string = result;
116	}
117
118	if ( result && result != arg ) {
119		free( result );
120	}
121
122	rewrite_session_delete( info, cookie );
123
124	rewrite_info_delete( &info );
125}
126
127int
128main( int argc, char *argv[] )
129{
130	FILE	*fin = NULL;
131	char	*rewriteContext = REWRITE_DEFAULT_CONTEXT;
132	int	debug = 0;
133
134	while ( 1 ) {
135		int opt = getopt( argc, argv, "d:f:hr:" );
136
137		if ( opt == EOF ) {
138			break;
139		}
140
141		switch ( opt ) {
142		case 'd':
143			if ( lutil_atoi( &debug, optarg ) != 0 ) {
144				fprintf( stderr, "illegal log level '%s'\n",
145						optarg );
146				exit( EXIT_FAILURE );
147			}
148			break;
149
150		case 'f':
151			fin = fopen( optarg, "r" );
152			if ( fin == NULL ) {
153				fprintf( stderr, "unable to open file '%s'\n",
154						optarg );
155				exit( EXIT_FAILURE );
156			}
157			break;
158
159		case 'h':
160			fprintf( stderr,
161	"usage: rewrite [options] string\n"
162	"\n"
163	"\t\t-f file\t\tconfiguration file\n"
164	"\t\t-r rule[s]\tlist of comma-separated rules\n"
165	"\n"
166	"\tsyntax:\n"
167	"\t\trewriteEngine\t{on|off}\n"
168	"\t\trewriteContext\tcontextName [alias aliasedContextName]\n"
169	"\t\trewriteRule\tpattern subst [flags]\n"
170	"\n"
171				);
172			exit( EXIT_SUCCESS );
173
174		case 'r':
175			rewriteContext = optarg;
176			break;
177		}
178	}
179
180	if ( debug != 0 ) {
181		ber_set_option(NULL, LBER_OPT_DEBUG_LEVEL, &debug);
182		ldap_set_option(NULL, LDAP_OPT_DEBUG_LEVEL, &debug);
183	}
184
185	if ( optind >= argc ) {
186		return -1;
187	}
188
189	apply( ( fin ? fin : stdin ), rewriteContext, argv[ optind ] );
190
191	if ( fin ) {
192		fclose( fin );
193	}
194
195	return 0;
196}
197
198