/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the
|
* Common Development and Distribution License, Version 1.0 only
|
* (the "License"). You may not use this file except in compliance
|
* with the License.
|
*
|
* You can obtain a copy of the license at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE
|
* or https://OpenDS.dev.java.net/OpenDS.LICENSE.
|
* See the License for the specific language governing permissions
|
* and limitations under the License.
|
*
|
* When distributing Covered Code, include this CDDL HEADER in each
|
* file and include the License file at
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE. If applicable,
|
* add the following below this CDDL HEADER, with the fields enclosed
|
* by brackets "[]" replaced with your own identifying information:
|
* Portions Copyright [yyyy] [name of copyright owner]
|
*
|
* CDDL HEADER END
|
*
|
*
|
* Copyright 2008 Sun Microsystems, Inc.
|
*/
|
|
import java.util.Hashtable;
|
import javax.naming.Context;
|
import javax.naming.NamingException;
|
import javax.naming.directory.Attributes;
|
import javax.naming.ldap.LdapContext;
|
import javax.naming.ldap.InitialLdapContext;
|
import javax.naming.CompositeName;
|
import javax.naming.directory.BasicAttribute;
|
import javax.naming.directory.BasicAttributes;
|
import javax.naming.CommunicationException;
|
import java.util.HashSet;
|
import java.util.Iterator;
|
|
/**
|
* Modify an entry with an attribute.
|
* The operation can be a replace, delete or a add new attribute
|
* The function returns the ldap error code
|
*/
|
public class modifyAnAttribute {
|
|
/**
|
* Main.
|
*
|
* @param args arguments
|
*/
|
public static void main(String[] args) {
|
|
String hostname=null;
|
String ldapPort=null;
|
String principal=null;
|
String credential=null;
|
String attributeToModify=null;
|
String dnToModify=null;
|
String newAttributeValue=null;
|
String changetype=null;
|
String errorCode=null;
|
String errorMessage=null;
|
String listAttributesToModify=null;
|
|
int ind1;
|
String attributeName;
|
String attributeValue;
|
Hashtable envLdap = new Hashtable();
|
LdapContext ctx;
|
|
Attributes attributes = new BasicAttributes();
|
HashSet attributeSet = new HashSet();
|
|
|
for (int k=0; k< args.length; k++) {
|
String opt1 = args[k];
|
String val1 = args[k+1];
|
|
if (opt1.equals("-h")) {
|
hostname = val1;
|
}
|
if (opt1.equals("-p")) {
|
ldapPort = val1;
|
}
|
if (opt1.equals("-D")) {
|
principal = val1;
|
}
|
if (opt1.equals("-w")) {
|
credential = val1;
|
}
|
if (opt1.equals("-d")) {
|
dnToModify = val1;
|
}
|
if (opt1.equals("-v")) {
|
newAttributeValue = val1;
|
}
|
if (opt1.equals("-a")) {
|
attributeToModify = val1;
|
}
|
if (opt1.equals("-t")) {
|
changetype = val1;
|
}
|
if (opt1.equals("-l")) {
|
listAttributesToModify = val1;
|
|
ind1= val1.indexOf(":");
|
|
attributeName=val1.substring(0,ind1);
|
if (ind1+1 < val1.length()) {
|
// assume empty strings == no specific value
|
attributeValue=val1.substring(ind1+1);
|
} else {
|
attributeValue = null;
|
}
|
|
BasicAttribute attrToComplete = null;
|
|
Iterator it = attributeSet.iterator();
|
while (attrToComplete == null && it.hasNext()) {
|
BasicAttribute attr = (BasicAttribute) it.next();
|
if ((attr.getID()).equalsIgnoreCase(attributeName)) {
|
attrToComplete = attr;
|
}
|
}
|
|
if (attrToComplete == null) {
|
attrToComplete = new BasicAttribute(attributeName);
|
attributeSet.add(attrToComplete);
|
}
|
if (attributeValue != null) {
|
// as opposed to (attributeValue == null), for example in some
|
// attribute delete operations
|
attributeValue=attributeValue.replaceAll("QUOT","\\\"");
|
attrToComplete.add(attributeValue);
|
}
|
}
|
k++;
|
}
|
|
if ( attributeToModify != null &&
|
( newAttributeValue != null || changetype.equals("delete") ) ) {
|
|
BasicAttribute attrToComplete = null;
|
|
attrToComplete = new BasicAttribute(attributeToModify);
|
attributeSet.add(attrToComplete);
|
if (newAttributeValue != null) {
|
// as opposed to (attributeValue == null), for example in some
|
// attribute delete operations
|
newAttributeValue=newAttributeValue.replaceAll("QUOT","\\\"");
|
attrToComplete.add(newAttributeValue);
|
}
|
}
|
|
Iterator it2 = attributeSet.iterator();
|
while (it2.hasNext()) {
|
BasicAttribute attr = (BasicAttribute)it2.next();
|
attributes.put(attr);
|
}
|
|
String provider = "ldap://" + hostname + ":" + ldapPort + "/";
|
|
envLdap.put("java.naming.factory.initial",
|
"com.sun.jndi.ldap.LdapCtxFactory");
|
envLdap.put(Context.SECURITY_AUTHENTICATION, "simple");
|
envLdap.put(Context.SECURITY_PRINCIPAL, principal);
|
envLdap.put(Context.SECURITY_CREDENTIALS, credential);
|
envLdap.put(Context.PROVIDER_URL, provider);
|
|
try {
|
CompositeName entryDN = new CompositeName(dnToModify);
|
System.out.println("Modify the entry " + dnToModify);
|
|
// connect to server
|
ctx = new InitialLdapContext(envLdap, null);
|
|
// replace attribute
|
if (changetype.equals("replace")) {
|
ctx.modifyAttributes(entryDN,
|
LdapContext.REPLACE_ATTRIBUTE,
|
attributes);
|
} else if (changetype.equals("add")) {
|
// add attribute
|
ctx.modifyAttributes(entryDN,
|
LdapContext.ADD_ATTRIBUTE,
|
attributes);
|
} else if (changetype.equals("delete")) {
|
// add attribute
|
ctx.modifyAttributes(entryDN,
|
LdapContext.REMOVE_ATTRIBUTE,
|
attributes);
|
}
|
|
ctx.close();
|
} catch (CommunicationException e1) {
|
errorMessage = e1.getMessage();
|
} catch (NamingException e2) {
|
errorMessage = e2.getMessage();
|
} catch (Exception e3) {
|
errorMessage= e3.getMessage();
|
}
|
|
// No error, the modify is success
|
if ( errorMessage == null ) {
|
errorCode="0";
|
} else {
|
System.out.println(errorMessage);
|
int ind=errorMessage.indexOf("-");
|
if ( ind > 0 ) {
|
errorCode=errorMessage.substring(18, ind-1);
|
} else errorCode="0";
|
}
|
|
int RC = Integer.parseInt(errorCode);
|
System.exit(RC);
|
}
|
|
}
|
