/*
* 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 legal-notices/CDDLv1_0.txt
* or http://forgerock.org/license/CDDLv1.0.html.
* 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 legal-notices/CDDLv1_0.txt.
* 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-2010 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;
/**
* Add a new entry.
* Input :
* -D : principal
* -w : credential
* -p : ldapport
* -h : ldaphost
* -d : dn to add
* -v : attribut name:attribut value
* : this option is multi valued
*/
public class addAnEntry {
/**
* Main.
*
* @param args arguments
*/
public static void main(String[] args) {
String hostname=null;
String ldapPort=null;
String principal=null;
String credential=null;
String dnToAdd=null;
String attributeToAdd=null;
String errorCode=null;
String errorMessage=null;
int ind1;
String attributeName;
String attributeValue;
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")) {
dnToAdd = val1;
}
if (opt1.equals("-v")) {
attributeToAdd = val1;
ind1= val1.indexOf(":");
attributeName=val1.substring(0,ind1);
attributeValue=val1.substring(ind1+1);
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) {
attributeValue = attributeValue.replaceAll("QUOT","\\\"");
attrToComplete.add(attributeValue);
}
}
k++;
}
Iterator it2 = attributeSet.iterator();
while (it2.hasNext()) {
BasicAttribute attr = (BasicAttribute)it2.next();
attributes.put(attr);
}
String provider = "ldap://" + hostname + ":" + ldapPort + "/";
Hashtable envLdap = new Hashtable();
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);
LdapContext ctx;
try {
CompositeName entryDN = new CompositeName(dnToAdd);
System.out.println("Add a new entry " + entryDN);
// connect to server
ctx = new InitialLdapContext(envLdap, null);
ctx.createSubcontext(entryDN, 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);
}
}