1 /* ldapmodify.c - generic program to modify or add entries using LDAP */
10 #include <ac/string.h>
11 #include <ac/unistd.h>
15 #ifdef HAVE_SYS_FILE_H
27 static char *binddn = NULL;
28 static char *passwd = NULL;
29 static char *ldaphost = NULL;
30 static int ldapport = 0;
31 static int new, replace, not, verbose, contoper, force, valsfromfiles;
34 #define safe_realloc( ptr, size ) ( ptr == NULL ? malloc( size ) : \
37 #define LDAPMOD_MAXLINE 4096
39 /* strings found in replog/LDIF entries (mostly lifted from slurpd/slurp.h) */
40 #define T_REPLICA_STR "replica"
42 #define T_CHANGETYPESTR "changetype"
43 #define T_ADDCTSTR "add"
44 #define T_MODIFYCTSTR "modify"
45 #define T_DELETECTSTR "delete"
46 #define T_MODRDNCTSTR "modrdn"
47 #define T_MODOPADDSTR "add"
48 #define T_MODOPREPLACESTR "replace"
49 #define T_MODOPDELETESTR "delete"
50 #define T_MODSEPSTR "-"
51 #define T_NEWRDNSTR "newrdn"
52 #define T_DELETEOLDRDNSTR "deleteoldrdn"
55 static int process_ldapmod_rec LDAP_P(( char *rbuf ));
56 static int process_ldif_rec LDAP_P(( char *rbuf ));
57 static void addmodifyop LDAP_P(( LDAPMod ***pmodsp, int modop, char *attr,
58 char *value, int vlen ));
59 static int domodify LDAP_P(( char *dn, LDAPMod **pmods, int newentry ));
60 static int dodelete LDAP_P(( char *dn ));
61 static int domodrdn LDAP_P(( char *dn, char *newrdn, int deleteoldrdn ));
62 static void freepmods LDAP_P(( LDAPMod **pmods ));
63 static int fromfile LDAP_P(( char *path, struct berval *bv ));
64 static char *read_one_record LDAP_P(( FILE *fp ));
68 main( int argc, char **argv )
70 char *infile, *rbuf, *start, *p, *q;
72 int rc, i, use_ldif, authmethod, version, want_bindpw, debug;
73 char *usage = "usage: %s [-abcknrvWF] [-d debug-level] [-h ldaphost] [-P version] [-p ldapport] [-D binddn] [-w passwd] [ -f file | < entryfile ]\n";
75 if (( prog = strrchr( argv[ 0 ], '/' )) == NULL ) {
80 new = ( strcmp( prog, "ldapadd" ) == 0 );
83 not = verbose = valsfromfiles = want_bindpw = debug = 0;
84 authmethod = LDAP_AUTH_SIMPLE;
87 while (( i = getopt( argc, argv, "WFabckKnrtvh:p:D:w:d:f:" )) != EOF ) {
92 case 'b': /* read values from files (for binary attributes) */
95 case 'c': /* continuous operation */
98 case 'r': /* default is to replace rather than add values */
101 case 'k': /* kerberos bind */
103 authmethod = LDAP_AUTH_KRBV4;
105 fprintf (stderr, "%s was not compiled with Kerberos support\n", argv[0]);
108 case 'K': /* kerberos bind, part 1 only */
110 authmethod = LDAP_AUTH_KRBV41;
112 fprintf (stderr, "%s was not compiled with Kerberos support\n", argv[0]);
115 case 'F': /* force all changes records to be used */
118 case 'h': /* ldap host */
119 ldaphost = strdup( optarg );
121 case 'D': /* bind DN */
122 binddn = strdup( optarg );
124 case 'w': /* password */
125 passwd = strdup( optarg );
128 debug |= atoi( optarg );
130 case 'f': /* read from file */
131 infile = strdup( optarg );
134 ldapport = atoi( optarg );
136 case 'n': /* print adds, don't actually do them */
139 case 'v': /* verbose mode */
149 version = LDAP_VERSION2;
152 version = LDAP_VERSION3;
157 fprintf( stderr, usage, prog );
162 if ( argc - optind != 0 ) {
163 fprintf( stderr, usage, prog );
167 if ( infile != NULL ) {
168 if (( fp = fopen( infile, "r" )) == NULL ) {
177 lber_set_option( NULL, LBER_OPT_DEBUG_LEVEL, &debug );
178 ldap_set_option( NULL, LDAP_OPT_DEBUG_LEVEL, &debug );
183 (void) SIGNAL( SIGPIPE, SIG_IGN );
187 if (( ld = ldap_init( ldaphost, ldapport )) == NULL ) {
188 perror( "ldap_init" );
192 /* this seems prudent */
194 int deref = LDAP_DEREF_NEVER;
195 ldap_set_option( ld, LDAP_OPT_DEREF, &deref);
199 passwd = getpass("Enter LDAP Password: ");
201 if( version != -1 ) {
202 ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, &version );
205 if ( ldap_bind_s( ld, binddn, passwd, authmethod ) != LDAP_SUCCESS ) {
206 ldap_perror( ld, "ldap_bind" );
213 while (( rc == 0 || contoper ) &&
214 ( rbuf = read_one_record( fp )) != NULL ) {
216 * we assume record is ldif/slapd.replog if the first line
217 * has a colon that appears to the left of any equal signs, OR
218 * if the first line consists entirely of digits (an entry id)
220 use_ldif = ( p = strchr( rbuf, ':' )) != NULL &&
221 ( q = strchr( rbuf, '\n' )) != NULL && p < q &&
222 (( q = strchr( rbuf, '=' )) == NULL || p < q );
226 if ( !use_ldif && ( q = strchr( rbuf, '\n' )) != NULL ) {
227 for ( p = rbuf; p < q; ++p ) {
228 if ( !isdigit( (unsigned char) *p )) {
239 rc = process_ldif_rec( start );
241 rc = process_ldapmod_rec( start );
259 process_ldif_rec( char *rbuf )
261 char *line, *dn, *type, *value, *newrdn, *p;
262 int rc, linenum, vlen, modop, replicaport;
263 int expect_modop, expect_sep, expect_ct, expect_newrdn;
264 int expect_deleteoldrdn, deleteoldrdn;
265 int saw_replica, use_record, new_entry, delete_entry, got_all;
270 rc = got_all = saw_replica = delete_entry = expect_modop = 0;
271 expect_deleteoldrdn = expect_newrdn = expect_sep = expect_ct = 0;
278 while ( rc == 0 && ( line = ldif_getline( &rbuf )) != NULL ) {
280 if ( expect_sep && strcasecmp( line, T_MODSEPSTR ) == 0 ) {
286 if ( ldif_parse_line( line, &type, &value, &vlen ) < 0 ) {
287 fprintf( stderr, "%s: invalid format (line %d of entry: %s\n",
288 prog, linenum, dn == NULL ? "" : dn );
289 rc = LDAP_PARAM_ERROR;
294 if ( !use_record && strcasecmp( type, T_REPLICA_STR ) == 0 ) {
296 if (( p = strchr( value, ':' )) == NULL ) {
300 replicaport = atoi( p );
302 if ( strcasecmp( value, ldaphost ) == 0 &&
303 replicaport == ldapport ) {
306 } else if ( strcasecmp( type, T_DN_STR ) == 0 ) {
307 if (( dn = strdup( value )) == NULL ) {
313 continue; /* skip all lines until we see "dn:" */
318 if ( !use_record && saw_replica ) {
319 printf( "%s: skipping change record for entry: %s\n\t(LDAP host/port does not match replica: lines)\n",
325 if ( strcasecmp( type, T_CHANGETYPESTR ) == 0 ) {
326 if ( strcasecmp( value, T_MODIFYCTSTR ) == 0 ) {
329 } else if ( strcasecmp( value, T_ADDCTSTR ) == 0 ) {
331 } else if ( strcasecmp( value, T_MODRDNCTSTR ) == 0 ) {
333 } else if ( strcasecmp( value, T_DELETECTSTR ) == 0 ) {
334 got_all = delete_entry = 1;
337 "%s: unknown %s \"%s\" (line %d of entry: %s)\n",
338 prog, T_CHANGETYPESTR, value, linenum, dn );
339 rc = LDAP_PARAM_ERROR;
342 } else if ( new ) { /* missing changetype => add */
344 modop = LDAP_MOD_ADD;
346 expect_modop = 1; /* missing changetype => modify */
350 if ( expect_modop ) {
353 if ( strcasecmp( type, T_MODOPADDSTR ) == 0 ) {
354 modop = LDAP_MOD_ADD;
356 } else if ( strcasecmp( type, T_MODOPREPLACESTR ) == 0 ) {
357 modop = LDAP_MOD_REPLACE;
359 } else if ( strcasecmp( type, T_MODOPDELETESTR ) == 0 ) {
360 modop = LDAP_MOD_DELETE;
361 addmodifyop( &pmods, modop, value, NULL, 0 );
363 } else { /* no modify op: use default */
364 modop = replace ? LDAP_MOD_REPLACE : LDAP_MOD_ADD;
368 if ( expect_newrdn ) {
369 if ( strcasecmp( type, T_NEWRDNSTR ) == 0 ) {
370 if (( newrdn = strdup( value )) == NULL ) {
374 expect_deleteoldrdn = 1;
377 fprintf( stderr, "%s: expecting \"%s:\" but saw \"%s:\" (line %d of entry %s)\n",
378 prog, T_NEWRDNSTR, type, linenum, dn );
379 rc = LDAP_PARAM_ERROR;
381 } else if ( expect_deleteoldrdn ) {
382 if ( strcasecmp( type, T_DELETEOLDRDNSTR ) == 0 ) {
383 deleteoldrdn = ( *value == '0' ) ? 0 : 1;
386 fprintf( stderr, "%s: expecting \"%s:\" but saw \"%s:\" (line %d of entry %s)\n",
387 prog, T_DELETEOLDRDNSTR, type, linenum, dn );
388 rc = LDAP_PARAM_ERROR;
390 } else if ( got_all ) {
392 "%s: extra lines at end (line %d of entry %s)\n",
394 rc = LDAP_PARAM_ERROR;
396 addmodifyop( &pmods, modop, type, value, vlen );
401 if ( delete_entry ) {
403 } else if ( newrdn != NULL ) {
404 rc = domodrdn( dn, newrdn, deleteoldrdn );
406 rc = domodify( dn, pmods, new_entry );
409 if ( rc == LDAP_SUCCESS ) {
417 if ( newrdn != NULL ) {
420 if ( pmods != NULL ) {
429 process_ldapmod_rec( char *rbuf )
431 char *line, *dn, *p, *q, *attr, *value;
432 int rc, linenum, modop;
441 while ( rc == 0 && rbuf != NULL && *rbuf != '\0' ) {
443 if (( p = strchr( rbuf, '\n' )) == NULL ) {
446 if ( *(p-1) == '\\' ) { /* lines ending in '\' are continued */
447 SAFEMEMCPY( p - 1, p, strlen( p ) + 1 );
455 if ( dn == NULL ) { /* first line contains DN */
456 if (( dn = strdup( line )) == NULL ) {
461 if (( p = strchr( line, '=' )) == NULL ) {
463 p = line + strlen( line );
470 *attr != '\0' && isspace( (unsigned char) *attr ); ++attr ) {
471 ; /* skip attribute leading white space */
474 for ( q = p - 1; q > attr && isspace( (unsigned char) *q ); --q ) {
475 *q = '\0'; /* remove attribute trailing white space */
478 if ( value != NULL ) {
479 while ( isspace( (unsigned char) *value )) {
480 ++value; /* skip value leading white space */
482 for ( q = value + strlen( value ) - 1; q > value &&
483 isspace( (unsigned char) *q ); --q ) {
484 *q = '\0'; /* remove value trailing white space */
486 if ( *value == '\0' ) {
492 if ( value == NULL && new ) {
493 fprintf( stderr, "%s: missing value on line %d (attr is %s)\n",
494 prog, linenum, attr );
495 rc = LDAP_PARAM_ERROR;
499 modop = LDAP_MOD_DELETE;
503 modop = LDAP_MOD_ADD;
507 modop = replace ? LDAP_MOD_REPLACE : LDAP_MOD_ADD;
510 addmodifyop( &pmods, modop, attr, value,
511 ( value == NULL ) ? 0 : strlen( value ));
520 rc = LDAP_PARAM_ERROR;
521 } else if (( rc = domodify( dn, pmods, new )) == LDAP_SUCCESS ) {
526 if ( pmods != NULL ) {
538 addmodifyop( LDAPMod ***pmodsp, int modop, char *attr, char *value, int vlen )
545 modop |= LDAP_MOD_BVALUES;
548 if ( pmods != NULL ) {
549 for ( ; pmods[ i ] != NULL; ++i ) {
550 if ( strcasecmp( pmods[ i ]->mod_type, attr ) == 0 &&
551 pmods[ i ]->mod_op == modop ) {
557 if ( pmods == NULL || pmods[ i ] == NULL ) {
558 if (( pmods = (LDAPMod **)safe_realloc( pmods, (i + 2) *
559 sizeof( LDAPMod * ))) == NULL ) {
560 perror( "safe_realloc" );
564 pmods[ i + 1 ] = NULL;
565 if (( pmods[ i ] = (LDAPMod *)calloc( 1, sizeof( LDAPMod )))
570 pmods[ i ]->mod_op = modop;
571 if (( pmods[ i ]->mod_type = strdup( attr )) == NULL ) {
577 if ( value != NULL ) {
579 if ( pmods[ i ]->mod_bvalues != NULL ) {
580 for ( ; pmods[ i ]->mod_bvalues[ j ] != NULL; ++j ) {
584 if (( pmods[ i ]->mod_bvalues =
585 (struct berval **)safe_realloc( pmods[ i ]->mod_bvalues,
586 (j + 2) * sizeof( struct berval * ))) == NULL ) {
587 perror( "safe_realloc" );
590 pmods[ i ]->mod_bvalues[ j + 1 ] = NULL;
591 if (( bvp = (struct berval *)malloc( sizeof( struct berval )))
596 pmods[ i ]->mod_bvalues[ j ] = bvp;
598 if ( valsfromfiles && *value == '/' ) { /* get value from file */
599 if ( fromfile( value, bvp ) < 0 ) {
604 if (( bvp->bv_val = (char *)malloc( vlen + 1 )) == NULL ) {
608 SAFEMEMCPY( bvp->bv_val, value, vlen );
609 bvp->bv_val[ vlen ] = '\0';
616 domodify( char *dn, LDAPMod **pmods, int newentry )
618 int i, j, k, notascii, op;
621 if ( pmods == NULL ) {
622 fprintf( stderr, "%s: no attributes to change or add (entry %s)\n",
624 return( LDAP_PARAM_ERROR );
628 for ( i = 0; pmods[ i ] != NULL; ++i ) {
629 op = pmods[ i ]->mod_op & ~LDAP_MOD_BVALUES;
630 printf( "%s %s:\n", op == LDAP_MOD_REPLACE ?
631 "replace" : op == LDAP_MOD_ADD ?
632 "add" : "delete", pmods[ i ]->mod_type );
633 if ( pmods[ i ]->mod_bvalues != NULL ) {
634 for ( j = 0; pmods[ i ]->mod_bvalues[ j ] != NULL; ++j ) {
635 bvp = pmods[ i ]->mod_bvalues[ j ];
637 for ( k = 0; (unsigned long) k < bvp->bv_len; ++k ) {
638 if ( !isascii( bvp->bv_val[ k ] )) {
644 printf( "\tNOT ASCII (%ld bytes)\n", bvp->bv_len );
646 printf( "\t%s\n", bvp->bv_val );
654 printf( "%sadding new entry %s\n", not ? "!" : "", dn );
656 printf( "%smodifying entry %s\n", not ? "!" : "", dn );
661 i = ldap_add_s( ld, dn, pmods );
663 i = ldap_modify_s( ld, dn, pmods );
665 if ( i != LDAP_SUCCESS ) {
666 ldap_perror( ld, newentry ? "ldap_add" : "ldap_modify" );
667 } else if ( verbose ) {
668 printf( "modify complete\n" );
685 printf( "%sdeleting entry %s\n", not ? "!" : "", dn );
687 if (( rc = ldap_delete_s( ld, dn )) != LDAP_SUCCESS ) {
688 ldap_perror( ld, "ldap_delete" );
689 } else if ( verbose ) {
690 printf( "delete complete" );
703 domodrdn( char *dn, char *newrdn, int deleteoldrdn )
708 printf( "new RDN: %s (%skeep existing values)\n",
709 newrdn, deleteoldrdn ? "do not " : "" );
712 printf( "%smodifying rdn of entry %s\n", not ? "!" : "", dn );
714 if (( rc = ldap_modrdn2_s( ld, dn, newrdn, deleteoldrdn ))
716 ldap_perror( ld, "ldap_modrdn" );
718 printf( "modrdn completed\n" );
732 freepmods( LDAPMod **pmods )
736 for ( i = 0; pmods[ i ] != NULL; ++i ) {
737 if ( pmods[ i ]->mod_bvalues != NULL ) {
738 ber_bvecfree( pmods[ i ]->mod_bvalues );
740 if ( pmods[ i ]->mod_type != NULL ) {
741 free( pmods[ i ]->mod_type );
750 fromfile( char *path, struct berval *bv )
756 if (( fp = fopen( path, "r" )) == NULL ) {
761 if ( fseek( fp, 0L, SEEK_END ) != 0 ) {
767 bv->bv_len = ftell( fp );
769 if (( bv->bv_val = (char *)malloc( bv->bv_len )) == NULL ) {
775 if ( fseek( fp, 0L, SEEK_SET ) != 0 ) {
781 rlen = fread( bv->bv_val, 1, bv->bv_len, fp );
785 if ( (unsigned long) rlen != bv->bv_len ) {
791 return( bv->bv_len );
796 read_one_record( FILE *fp )
799 char *buf, line[ LDAPMOD_MAXLINE ];
805 while (( fgets( line, sizeof(line), fp ) != NULL ) &&
806 (( len = strlen( line )) > 1 )) {
807 if ( lcur + len + 1 > lmax ) {
808 lmax = LDAPMOD_MAXLINE
809 * (( lcur + len + 1 ) / LDAPMOD_MAXLINE + 1 );
810 if (( buf = (char *)safe_realloc( buf, lmax )) == NULL ) {
811 perror( "safe_realloc" );
815 strcpy( buf + lcur, line );