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