2 * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
3 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5 /* ldapmodify.c - generic program to modify or add entries using LDAP */
11 #include <ac/stdlib.h>
14 #include <ac/signal.h>
15 #include <ac/string.h>
16 #include <ac/unistd.h>
20 #ifdef HAVE_SYS_FILE_H
31 #include <ldap_defaults.h>
34 static char *binddn = NULL;
35 static char *passwd = NULL;
36 static char *ldaphost = NULL;
37 static int ldapport = 0;
38 static int new, replace, not, verbose, contoper, force, valsfromfiles;
41 #define LDAPMOD_MAXLINE 4096
43 /* strings found in replog/LDIF entries (mostly lifted from slurpd/slurp.h) */
44 #define T_REPLICA_STR "replica"
46 #define T_CHANGETYPESTR "changetype"
47 #define T_ADDCTSTR "add"
48 #define T_MODIFYCTSTR "modify"
49 #define T_DELETECTSTR "delete"
50 #define T_MODRDNCTSTR "modrdn"
51 #define T_MODOPADDSTR "add"
52 #define T_MODOPREPLACESTR "replace"
53 #define T_MODOPDELETESTR "delete"
54 #define T_MODSEPSTR "-"
55 #define T_NEWRDNSTR "newrdn"
56 #define T_DELETEOLDRDNSTR "deleteoldrdn"
59 static void usage LDAP_P(( const char *prog ));
60 static int process_ldapmod_rec LDAP_P(( char *rbuf ));
61 static int process_ldif_rec LDAP_P(( char *rbuf ));
62 static void addmodifyop LDAP_P(( LDAPMod ***pmodsp, int modop, char *attr,
63 char *value, int vlen ));
64 static int domodify LDAP_P(( char *dn, LDAPMod **pmods, int newentry ));
65 static int dodelete LDAP_P(( char *dn ));
66 static int domodrdn LDAP_P(( char *dn, char *newrdn, int deleteoldrdn ));
67 static int fromfile LDAP_P(( char *path, struct berval *bv ));
68 static char *read_one_record LDAP_P(( FILE *fp ));
71 usage( const char *prog )
74 "Add or modify entries from an LDAP server\n\n"
75 "usage: %s [-abcknrvF] [-d debug-level] [-P version] [-h ldaphost]\n"
76 " [-p ldapport] [-D binddn] [-w passwd] [ -f file | < entryfile ]\n"
77 " a - add values (default%s)\n"
78 " b - read values from files (for binary attributes)\n"
79 " c - continuous operation\n"
82 " f - read from file\n"
83 " F - force all changes records to be used\n"
85 " n - print adds, don't actually do them\n"
87 " r - replace values\n"
90 , prog, (strcmp( prog, "ldapadd" ) ? " is to replace" : "") );
96 main( int argc, char **argv )
98 char *infile, *rbuf, *start, *p, *q;
100 int rc, i, use_ldif, authmethod, version, want_bindpw, debug;
102 if (( prog = strrchr( argv[ 0 ], *LDAP_DIRSEP )) == NULL ) {
108 /* Print usage when no parameters */
112 new = ( strcmp( prog, "ldapadd" ) == 0 );
115 not = verbose = valsfromfiles = want_bindpw = debug = 0;
116 authmethod = LDAP_AUTH_SIMPLE;
119 while (( i = getopt( argc, argv, "WFabckKnrtvh:p:D:w:d:f:" )) != EOF ) {
124 case 'b': /* read values from files (for binary attributes) */
127 case 'c': /* continuous operation */
130 case 'r': /* default is to replace rather than add values */
133 case 'k': /* kerberos bind */
135 authmethod = LDAP_AUTH_KRBV4;
137 fprintf (stderr, "%s was not compiled with Kerberos support\n", argv[0]);
139 return( EXIT_FAILURE );
142 case 'K': /* kerberos bind, part 1 only */
144 authmethod = LDAP_AUTH_KRBV41;
146 fprintf (stderr, "%s was not compiled with Kerberos support\n", argv[0]);
148 return( EXIT_FAILURE );
151 case 'F': /* force all changes records to be used */
154 case 'h': /* ldap host */
155 ldaphost = strdup( optarg );
157 case 'D': /* bind DN */
158 binddn = strdup( optarg );
160 case 'w': /* password */
161 passwd = strdup( optarg );
164 debug |= atoi( optarg );
166 case 'f': /* read from file */
167 infile = strdup( optarg );
170 ldapport = atoi( optarg );
172 case 'n': /* print adds, don't actually do them */
175 case 'v': /* verbose mode */
182 switch( atoi(optarg) )
185 version = LDAP_VERSION2;
188 version = LDAP_VERSION3;
191 fprintf( stderr, "protocol version should be 2 or 3\n" );
200 if ( argc != optind )
203 if ( infile != NULL ) {
204 if (( fp = fopen( infile, "r" )) == NULL ) {
206 return( EXIT_FAILURE );
213 if( ber_set_option( NULL, LBER_OPT_DEBUG_LEVEL, &debug ) != LBER_OPT_ERROR ) {
214 fprintf( stderr, "Could not set LBER_OPT_DEBUG_LEVEL %d\n", debug );
216 if( ldap_set_option( NULL, LDAP_OPT_DEBUG_LEVEL, &debug ) != LDAP_OPT_ERROR ) {
217 fprintf( stderr, "Could not set LDAP_OPT_DEBUG_LEVEL %d\n", debug );
223 (void) SIGNAL( SIGPIPE, SIG_IGN );
227 if (( ld = ldap_init( ldaphost, ldapport )) == NULL ) {
228 perror( "ldap_init" );
229 return( EXIT_FAILURE );
232 /* this seems prudent */
234 int deref = LDAP_DEREF_NEVER;
235 ldap_set_option( ld, LDAP_OPT_DEREF, &deref);
239 passwd = getpass("Enter LDAP Password: ");
242 ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, &version ) == LDAP_OPT_ERROR)
244 fprintf( stderr, "Could not set LDAP_OPT_PROTOCOL_VERSION %d\n", version );
247 if ( ldap_bind_s( ld, binddn, passwd, authmethod ) != LDAP_SUCCESS ) {
248 ldap_perror( ld, "ldap_bind" );
249 return( EXIT_FAILURE );
255 while (( rc == 0 || contoper ) &&
256 ( rbuf = read_one_record( fp )) != NULL ) {
258 * we assume record is ldif/slapd.replog if the first line
259 * has a colon that appears to the left of any equal signs, OR
260 * if the first line consists entirely of digits (an entry id)
262 use_ldif = ( p = strchr( rbuf, ':' )) != NULL &&
263 ( q = strchr( rbuf, '\n' )) != NULL && p < q &&
264 (( q = strchr( rbuf, '=' )) == NULL || p < q );
268 if ( !use_ldif && ( q = strchr( rbuf, '\n' )) != NULL ) {
269 for ( p = rbuf; p < q; ++p ) {
270 if ( !isdigit( (unsigned char) *p )) {
281 rc = process_ldif_rec( start );
283 rc = process_ldapmod_rec( start );
287 fprintf( stderr, "%s() = %d\n",
288 use_ldif ? "ldif_rec" : "ldapmod_rec" , rc );
302 process_ldif_rec( char *rbuf )
304 char *line, *dn, *type, *value, *newrdn, *p;
305 int rc, linenum, vlen, modop, replicaport;
306 int expect_modop, expect_sep, expect_ct, expect_newrdn;
307 int expect_deleteoldrdn, deleteoldrdn;
308 int saw_replica, use_record, new_entry, delete_entry, got_all;
313 rc = got_all = saw_replica = delete_entry = expect_modop = 0;
314 expect_deleteoldrdn = expect_newrdn = expect_sep = expect_ct = 0;
321 while ( rc == 0 && ( line = ldif_getline( &rbuf )) != NULL ) {
323 if ( expect_sep && strcasecmp( line, T_MODSEPSTR ) == 0 ) {
329 if ( ldif_parse_line( line, &type, &value, &vlen ) < 0 ) {
330 fprintf( stderr, "%s: invalid format (line %d of entry: %s\n",
331 prog, linenum, dn == NULL ? "" : dn );
332 rc = LDAP_PARAM_ERROR;
337 if ( !use_record && strcasecmp( type, T_REPLICA_STR ) == 0 ) {
339 if (( p = strchr( value, ':' )) == NULL ) {
343 replicaport = atoi( p );
345 if ( strcasecmp( value, ldaphost ) == 0 &&
346 replicaport == ldapport ) {
349 } else if ( strcasecmp( type, T_DN_STR ) == 0 ) {
350 if (( dn = strdup( value )) == NULL ) {
356 continue; /* skip all lines until we see "dn:" */
361 if ( !use_record && saw_replica ) {
362 printf( "%s: skipping change record for entry: %s\n\t(LDAP host/port does not match replica: lines)\n",
368 if ( strcasecmp( type, T_CHANGETYPESTR ) == 0 ) {
369 if ( strcasecmp( value, T_MODIFYCTSTR ) == 0 ) {
372 } else if ( strcasecmp( value, T_ADDCTSTR ) == 0 ) {
374 } else if ( strcasecmp( value, T_MODRDNCTSTR ) == 0 ) {
376 } else if ( strcasecmp( value, T_DELETECTSTR ) == 0 ) {
377 got_all = delete_entry = 1;
380 "%s: unknown %s \"%s\" (line %d of entry: %s)\n",
381 prog, T_CHANGETYPESTR, value, linenum, dn );
382 rc = LDAP_PARAM_ERROR;
385 } else if ( new ) { /* missing changetype => add */
387 modop = LDAP_MOD_ADD;
389 expect_modop = 1; /* missing changetype => modify */
393 if ( expect_modop ) {
396 if ( strcasecmp( type, T_MODOPADDSTR ) == 0 ) {
397 modop = LDAP_MOD_ADD;
399 } else if ( strcasecmp( type, T_MODOPREPLACESTR ) == 0 ) {
400 modop = LDAP_MOD_REPLACE;
402 } else if ( strcasecmp( type, T_MODOPDELETESTR ) == 0 ) {
403 modop = LDAP_MOD_DELETE;
404 addmodifyop( &pmods, modop, value, NULL, 0 );
406 } else { /* no modify op: use default */
407 modop = replace ? LDAP_MOD_REPLACE : LDAP_MOD_ADD;
411 if ( expect_newrdn ) {
412 if ( strcasecmp( type, T_NEWRDNSTR ) == 0 ) {
413 if (( newrdn = strdup( value )) == NULL ) {
417 expect_deleteoldrdn = 1;
420 fprintf( stderr, "%s: expecting \"%s:\" but saw \"%s:\" (line %d of entry %s)\n",
421 prog, T_NEWRDNSTR, type, linenum, dn );
422 rc = LDAP_PARAM_ERROR;
424 } else if ( expect_deleteoldrdn ) {
425 if ( strcasecmp( type, T_DELETEOLDRDNSTR ) == 0 ) {
426 deleteoldrdn = ( *value == '0' ) ? 0 : 1;
429 fprintf( stderr, "%s: expecting \"%s:\" but saw \"%s:\" (line %d of entry %s)\n",
430 prog, T_DELETEOLDRDNSTR, type, linenum, dn );
431 rc = LDAP_PARAM_ERROR;
433 } else if ( got_all ) {
435 "%s: extra lines at end (line %d of entry %s)\n",
437 rc = LDAP_PARAM_ERROR;
439 addmodifyop( &pmods, modop, type, value, vlen );
444 if ( delete_entry ) {
446 } else if ( newrdn != NULL ) {
447 rc = domodrdn( dn, newrdn, deleteoldrdn );
449 rc = domodify( dn, pmods, new_entry );
452 if ( rc == LDAP_SUCCESS ) {
460 if ( newrdn != NULL ) {
463 if ( pmods != NULL ) {
464 ldap_mods_free( pmods, 1 );
472 process_ldapmod_rec( char *rbuf )
474 char *line, *dn, *p, *q, *attr, *value;
475 int rc, linenum, modop;
484 while ( rc == 0 && rbuf != NULL && *rbuf != '\0' ) {
486 if (( p = strchr( rbuf, '\n' )) == NULL ) {
489 if ( *(p-1) == '\\' ) { /* lines ending in '\' are continued */
490 SAFEMEMCPY( p - 1, p, strlen( p ) + 1 );
498 if ( dn == NULL ) { /* first line contains DN */
499 if (( dn = strdup( line )) == NULL ) {
504 if (( p = strchr( line, '=' )) == NULL ) {
506 p = line + strlen( line );
513 *attr != '\0' && isspace( (unsigned char) *attr ); ++attr ) {
514 ; /* skip attribute leading white space */
517 for ( q = p - 1; q > attr && isspace( (unsigned char) *q ); --q ) {
518 *q = '\0'; /* remove attribute trailing white space */
521 if ( value != NULL ) {
522 while ( isspace( (unsigned char) *value )) {
523 ++value; /* skip value leading white space */
525 for ( q = value + strlen( value ) - 1; q > value &&
526 isspace( (unsigned char) *q ); --q ) {
527 *q = '\0'; /* remove value trailing white space */
529 if ( *value == '\0' ) {
535 if ( value == NULL && new ) {
536 fprintf( stderr, "%s: missing value on line %d (attr is %s)\n",
537 prog, linenum, attr );
538 rc = LDAP_PARAM_ERROR;
542 modop = LDAP_MOD_DELETE;
546 modop = LDAP_MOD_ADD;
550 modop = replace ? LDAP_MOD_REPLACE : LDAP_MOD_ADD;
553 addmodifyop( &pmods, modop, attr, value,
554 ( value == NULL ) ? 0 : strlen( value ));
563 rc = LDAP_PARAM_ERROR;
564 } else if (( rc = domodify( dn, pmods, new )) == LDAP_SUCCESS ) {
569 if ( pmods != NULL ) {
570 ldap_mods_free( pmods, 1 );
581 addmodifyop( LDAPMod ***pmodsp, int modop, char *attr, char *value, int vlen )
588 modop |= LDAP_MOD_BVALUES;
591 if ( pmods != NULL ) {
592 for ( ; pmods[ i ] != NULL; ++i ) {
593 if ( strcasecmp( pmods[ i ]->mod_type, attr ) == 0 &&
594 pmods[ i ]->mod_op == modop ) {
600 if ( pmods == NULL || pmods[ i ] == NULL ) {
601 if (( pmods = (LDAPMod **)ber_memrealloc( pmods, (i + 2) *
602 sizeof( LDAPMod * ))) == NULL ) {
607 pmods[ i + 1 ] = NULL;
608 if (( pmods[ i ] = (LDAPMod *)ber_memcalloc( 1, sizeof( LDAPMod )))
613 pmods[ i ]->mod_op = modop;
614 if (( pmods[ i ]->mod_type = ber_strdup( attr )) == NULL ) {
620 if ( value != NULL ) {
622 if ( pmods[ i ]->mod_bvalues != NULL ) {
623 for ( ; pmods[ i ]->mod_bvalues[ j ] != NULL; ++j ) {
627 if (( pmods[ i ]->mod_bvalues =
628 (struct berval **)ber_memrealloc( pmods[ i ]->mod_bvalues,
629 (j + 2) * sizeof( struct berval * ))) == NULL ) {
630 perror( "ber_realloc" );
633 pmods[ i ]->mod_bvalues[ j + 1 ] = NULL;
634 if (( bvp = (struct berval *)ber_memalloc( sizeof( struct berval )))
636 perror( "ber_memalloc" );
639 pmods[ i ]->mod_bvalues[ j ] = bvp;
641 if ( valsfromfiles && *value == '/' ) { /* get value from file */
642 if ( fromfile( value, bvp ) < 0 ) {
647 if (( bvp->bv_val = (char *)ber_memalloc( vlen + 1 )) == NULL ) {
651 SAFEMEMCPY( bvp->bv_val, value, vlen );
652 bvp->bv_val[ vlen ] = '\0';
659 domodify( char *dn, LDAPMod **pmods, int newentry )
661 int i, j, k, notascii, op;
664 if ( pmods == NULL ) {
665 fprintf( stderr, "%s: no attributes to change or add (entry %s)\n",
667 return( LDAP_PARAM_ERROR );
671 for ( i = 0; pmods[ i ] != NULL; ++i ) {
672 op = pmods[ i ]->mod_op & ~LDAP_MOD_BVALUES;
673 printf( "%s %s:\n", op == LDAP_MOD_REPLACE ?
674 "replace" : op == LDAP_MOD_ADD ?
675 "add" : "delete", pmods[ i ]->mod_type );
676 if ( pmods[ i ]->mod_bvalues != NULL ) {
677 for ( j = 0; pmods[ i ]->mod_bvalues[ j ] != NULL; ++j ) {
678 bvp = pmods[ i ]->mod_bvalues[ j ];
680 for ( k = 0; (unsigned long) k < bvp->bv_len; ++k ) {
681 if ( !isascii( bvp->bv_val[ k ] )) {
687 printf( "\tNOT ASCII (%ld bytes)\n", bvp->bv_len );
689 printf( "\t%s\n", bvp->bv_val );
697 printf( "%sadding new entry %s\n", not ? "!" : "", dn );
699 printf( "%smodifying entry %s\n", not ? "!" : "", dn );
704 i = ldap_add_s( ld, dn, pmods );
706 i = ldap_modify_s( ld, dn, pmods );
708 if ( i != LDAP_SUCCESS ) {
709 ldap_perror( ld, newentry ? "ldap_add" : "ldap_modify" );
710 } else if ( verbose ) {
711 printf( "modify complete\n" );
728 printf( "%sdeleting entry %s\n", not ? "!" : "", dn );
730 if (( rc = ldap_delete_s( ld, dn )) != LDAP_SUCCESS ) {
731 ldap_perror( ld, "ldap_delete" );
732 } else if ( verbose ) {
733 printf( "delete complete" );
746 domodrdn( char *dn, char *newrdn, int deleteoldrdn )
751 printf( "new RDN: %s (%skeep existing values)\n",
752 newrdn, deleteoldrdn ? "do not " : "" );
755 printf( "%smodifying rdn of entry %s\n", not ? "!" : "", dn );
757 if (( rc = ldap_modrdn2_s( ld, dn, newrdn, deleteoldrdn ))
759 ldap_perror( ld, "ldap_modrdn" );
761 printf( "modrdn completed\n" );
774 fromfile( char *path, struct berval *bv )
780 if (( fp = fopen( path, "r" )) == NULL ) {
785 if ( fseek( fp, 0L, SEEK_END ) != 0 ) {
791 bv->bv_len = ftell( fp );
793 if (( bv->bv_val = (char *)ber_memalloc( bv->bv_len )) == NULL ) {
799 if ( fseek( fp, 0L, SEEK_SET ) != 0 ) {
802 ber_memfree( bv->bv_val );
807 rlen = fread( bv->bv_val, 1, bv->bv_len, fp );
811 if ( (unsigned long) rlen != bv->bv_len ) {
813 ber_memfree( bv->bv_val );
818 return( bv->bv_len );
823 read_one_record( FILE *fp )
826 char *buf, line[ LDAPMOD_MAXLINE ];
832 while (( fgets( line, sizeof(line), fp ) != NULL ) &&
833 (( len = strlen( line )) > 1 )) {
834 if ( lcur + len + 1 > lmax ) {
835 lmax = LDAPMOD_MAXLINE
836 * (( lcur + len + 1 ) / LDAPMOD_MAXLINE + 1 );
837 if (( buf = (char *)realloc( buf, lmax )) == NULL ) {
842 strcpy( buf + lcur, line );