废话少说，上代码。
import javax.naming.*;
import javax.naming.directory.*;
import java.io.File;
import java.io.FileInputStream;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class LdapInsert {
private static String url = "ldap://192.168.88.146:1389";// ldap服务器地址
private static String user = "cn=Directory Manager";// ldap用户信息
private static String passwd = "123456";
static LdapInsert insertEr = new LdapInsert();
static LDAPConnection connection = null;
public static void main(String[] args) {
try {
connection = insertEr.getConnection(url, user, passwd);
Boolean result = insertEr.updateCert(insertEr.initCertEntity("", new Random().nextInt(100)), connection);
System.out.println("插入结果：" + result);
connection.close();
System.out.println("关闭连接成功");
} catch (Exception e) {
e.printStackTrace();
}
}
private CertEntity initCertEntity(String cerPath, int j) throws Exception {
CertEntity c = new CertEntity();
String CN = "CN=注入测试";
if (cerPath != null && cerPath.trim().length() > 0) {
CN += "-带证书-";
byte[] cerFile = readCer(cerPath);
c.setContent(cerFile);
} else {
CN += "-无证书-";
}
String i = new Random().nextInt(100000) + "_" + new Random().nextInt(100000) + "_" + j;
Date d = new Date();
c.setDn(CN + i + ",OU=OU-" + formatDate4(d) /*+ "-" + new Random().nextInt(16)*/ + ",O=ZYYDD-" + formatDate3(d) + ",C=CN");
c.setSn(i + "_" + formatDate4(d) + "_" + formatDate3(d));
c.setTemlateId("1000");
c.setStatus("5");
c.setBaseDn("O=ZYYDD Operation CA2,C=CN");
System.out.println("初始化DN：" + c.getDn());
return c;
}
public String formatDate3(Date date) {
SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmm");
return format.format(date.getTime());
}
public String formatDate4(Date date) {
SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss");
return format.format(date.getTime());
}
public byte[] readCer(String cerPath) throws Exception {
File readfile = new File(cerPath);
FileInputStream fis = new FileInputStream(readfile);
int filelength = (int) readfile.length();
byte[] bytesFile = new byte[filelength];
fis.read(bytesFile);
fis.close();
return bytesFile;
}
public LDAPConnection getConnection(String url, String user, String passwd) throws NamingException {
this.initEnv(url, user, passwd);
DirContext ctx = new InitialDirContext(this.env);
LDAPConnection conn = new LDAPConnection(ctx);
System.out.println("获取连接成功");
return conn;
}
public void initEnv(String url, String user, String passwd) {
System.setProperty("com.sun.jndi.ldap.connect.pool.initsize", String.valueOf(10));
System.setProperty("com.sun.jndi.ldap.connect.pool.maxsize", String.valueOf(20));
System.setProperty("com.sun.jndi.ldap.connect.pool.prefsize", String.valueOf(15));
System.setProperty("com.sun.jndi.ldap.connect.pool.timeout", String.valueOf(300));
System.setProperty("com.sun.jndi.ldap.connect.pool.debug", "false");
this.env = new Hashtable();
this.env.put("java.naming.factory.initial", "com.sun.jndi.ldap.LdapCtxFactory");
this.env.put("java.naming.provider.url", url);
this.env.put("com.sun.jndi.ldap.connect.pool", "true");
this.env.put("java.naming.security.authentication", "simple");
this.env.put("java.naming.security.principal", user);
this.env.put("java.naming.security.credentials", passwd);
}
public Boolean updateCert(CertEntity cert, LDAPConnection conn) throws Exception {
try {
CertEntity entity = new CertEntity();
entity.setDn(cert.getDn());
entity.setBaseDn(cert.getBaseDn());
entity.setContent(cert.getContent());
entity.setSn(cert.getSn());
entity.setTemlateId(cert.getTemlateId());
addDistributionDN(conn, entity);
return true;
} catch (Exception ex) {
System.out.println(cert.getDn() + " 插入异常 " + ex.getMessage());
return false;
}
}
private void addDistributionDN(LDAPConnection conn, CertEntity entity)
throws NamingException {
String baseDN = entity.getBaseDn();
String distributionDN = entity.getDn();
Pattern pattern = Pattern.compile("\\s*,\\s*", 2);
Matcher matcher = pattern.matcher(distributionDN.trim());
distributionDN = matcher.replaceAll(",");
int baseDNLen = baseDN.length();
int distributionDNLen = distributionDN.length();
String tempEntryDn = "," + distributionDN;
String tempBaseDn = "," + baseDN;
if (tempEntryDn.toUpperCase().indexOf(tempBaseDn.toUpperCase()) != -1) {
String subDN = "";
if (distributionDNLen > baseDNLen) {
subDN =
distributionDN.substring(0, distributionDN.length() - baseDN.length() - 1).trim();
}
boolean baseDNExist = conn.checkEntryExist(baseDN);
if (!baseDNExist) {
List dnList = new ArrayList();
String tDN = baseDN;
while (tDN.indexOf(",") != -1) {
dnList.add(tDN.substring(0, tDN.indexOf(",")));
tDN = tDN.substring(tDN.indexOf(",") + 1, tDN.length());
if (conn.checkEntryExist(tDN)) {
break;
}
}
dnList.add(tDN);
String entryDN = null;
StringBuffer entryDNBF = null;
int tCount = 0;
int tSize = dnList.size();
for (int i = tSize - 1; i >= 0; i--) {
entryDNBF = new StringBuffer();
if (tCount == 0) {
entryDNBF.append((String) dnList.get(i));
} else {
entryDNBF.append((String) dnList.get(i));
entryDNBF.append(",");
entryDNBF.append(entryDN);
}
tCount++;
entryDN = entryDNBF.toString().trim();
boolean e = conn.checkEntryExist(entryDN);
if (!e) {
addEntry(conn, entryDN, entity);
}
}
}
if (subDN.equals("")) {
return;
}
List dnList = new ArrayList();
String tDN = subDN;
while (tDN.indexOf(",") != -1) {
dnList.add(tDN.substring(0, tDN.indexOf(",")));
tDN = tDN.substring(tDN.indexOf(",") + 1, tDN.length());
if (conn.checkEntryExist(tDN)) {
break;
}
}
dnList.add(tDN);
String entryDN = null;
StringBuffer entryDNBF = null;
int tCount = 0;
int tSize = dnList.size();
for (int i = tSize - 1; i >= 0; i--) {
entryDNBF = new StringBuffer();
if (tCount == 0) {
entryDNBF.append((String) dnList.get(i));
} else {
entryDNBF.append((String) dnList.get(i));
entryDNBF.append(",");
entryDNBF.append(entryDN);
}
tCount++;
entryDN = entryDNBF.toString().trim();
boolean e = conn.checkEntryExist(entryDN + "," + baseDN);
if (!e) {
addEntry(conn, entryDN + "," + baseDN, entity);
}
}
} else {
List dnList = new ArrayList();
String tDN = distributionDN;
while (tDN.indexOf(",") != -1) {
dnList.add(tDN.substring(0, tDN.indexOf(",")));
tDN = tDN.substring(tDN.indexOf(",") + 1, tDN.length());
if (conn.checkEntryExist(tDN)) {
break;
}
}
dnList.add(tDN);
String entryDN = null;
StringBuffer entryDNBF = null;
int tCount = 0;
int tSize = dnList.size();
for (int i = tSize - 1; i >= 0; i--) {
entryDNBF = new StringBuffer();
if (tCount == 0) {
entryDNBF.append((String) dnList.get(i));
} else {
entryDNBF.append((String) dnList.get(i));
entryDNBF.append(",");
entryDNBF.append(entryDN);
}
tCount++;
entryDN = entryDNBF.toString().trim();
boolean e = conn.checkEntryExist(entryDN);
if (!e) {
addEntry(conn, entryDN, entity);
}
}
}
}
private void addEntry(LDAPConnection conn, String dn, CertEntity entity) throws NamingException {
String headInfo = dn.substring(0, dn.indexOf("=")).toLowerCase();
String value = "";
if (dn.indexOf(",") == -1) {
value = dn.substring(dn.indexOf("=") + 1);
} else {
value = dn.substring(dn.indexOf("=") + 1, dn.indexOf(","));
}
if (headInfo.trim().equals("c")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("country");
atts.put(att);
conn.add(dn, atts);
} else if ((headInfo.trim().equals("l")) || (headInfo.trim().equals("st")) || (headInfo.trim().equals("street"))) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("locality");
atts.put(att);
conn.add(dn, atts);
} else if (headInfo.trim().equals("o")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass", true);
att.add("top");
att.add("organization");
atts.put(att);
conn.add(dn, atts);
} else if (headInfo.trim().equals("ou")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("organizationalUnit");
atts.put(att);
conn.add(dn, atts);
} else if (headInfo.trim().equals("cn")) {
Attributes atts = new BasicAttributes(true);
atts.put(new BasicAttribute("objectClass", "jitUser"));
atts.put(new BasicAttribute("cn", value));
atts.put(new BasicAttribute("sn", value));
atts.put(new BasicAttribute("uid", entity.getSn(), true));
atts.put(new BasicAttribute("jitcertpos", "0"));
//Attributes atts = new BasicAttributes(true);
atts.put(new BasicAttribute("objectClass", "jitUser"));
atts.put(new BasicAttribute("jitDn", entity.getDn()));
atts.put(new BasicAttribute("cn", getCnFromDn(entity.getDn())));
atts.put(new BasicAttribute("sn", getCnFromDn(entity.getDn())));
atts.put(new BasicAttribute("jituserstatus", "5"));
atts.put(new BasicAttribute("jitusertype", entity.getTemlateId()));
if (entity.getContent() != null) {
atts.put(new BasicAttribute("userCertificate;binary", entity.getContent()));
}
conn.add(dn, atts);
} else if (headInfo.trim().equals("sn")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("inetOrgPerson");
att.add("idaPerson");
atts.put(att);
atts.put(new BasicAttribute("cn", value));
conn.add(dn, atts);
} else if (headInfo.trim().equals("dc")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("domain");
atts.put(att);
conn.add(dn, atts);
} else if (headInfo.trim().equals("uid")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("inetOrgPerson");
att.add("idaPerson");
atts.put(att);
atts.put(new BasicAttribute("sn", value));
atts.put(new BasicAttribute("cn", value));
conn.add(dn, atts);
} else if (headInfo.trim().equals("e")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("email");
atts.put(att);
conn.add(dn, atts);
} else if (headInfo.trim().equals("t")) {
Attributes atts = new BasicAttributes(true);
Attribute att = new BasicAttribute("objectClass");
att.add("top");
att.add("titleObject");
atts.put(att);
conn.add(dn, atts);
}
}
public String getCnFromDn(String dn) {
String cn = null;
StringTokenizer strToken = new StringTokenizer(dn, ",");
while (strToken.hasMoreTokens()) {
String tempCN = strToken.nextToken();
if (tempCN.trim().toUpperCase().indexOf("CN=") == 0) {
cn = tempCN.substring(tempCN.indexOf("=") + 1);
}
}
if (cn == null) {
cn = dn.substring(dn.indexOf("=") + 1, dn.indexOf(","));
}
return cn;
}
private Hashtable env = null;
public class LDAPConnection {
private DirContext ctx;
public Set ENTRY_CACHE = new HashSet();
public LDAPConnection(DirContext ctx) {
this.ctx = ctx;
}
public void add(String entryDN, Attributes attrs)
throws NamingException {
Name compositeName = new CompositeName().add(entryDN);
this.ctx.createSubcontext(compositeName, attrs);
}
public boolean checkEntryExist(String entryDN)
throws NamingException {
if (ENTRY_CACHE.contains(entryDN)) {
return true;
}
NamingEnumeration tResults = null;
if (entryDN == null) {
return false;
}
SearchControls tConstraints = new SearchControls();
tConstraints.setSearchScope(0);
try {
Name searchedName = new CompositeName().add(entryDN);
tResults = this.ctx.search(searchedName, "(objectclass=*)", tConstraints);
} catch (NameNotFoundException ex) {
return false;
}
if (tResults != null) {
tResults.close();
try {
String tmp = entryDN.toUpperCase();
if ((tmp.startsWith("C=")) || (tmp.startsWith("O=")) || (tmp.startsWith("OU="))) {
ENTRY_CACHE.add(entryDN);
}
} catch (Exception localException) {
}
return true;
}
return false;
}
public void close() {
try {
if (this.ctx != null) {
this.ctx.close();
System.gc();
}
} catch (NamingException e) {
System.out.println("关闭LDAP连接失败：" + e.toString());
}
}
}
public class CertEntity {
protected String baseDn = null;
protected String dn = null;
private String sn = null;
private String status = "3";
private String temlateId = "0";
private byte[] content = null;
public String getBaseDn() {
return baseDn;
}
public void setBaseDn(String baseDn) {
this.baseDn = baseDn;
}
public String getDn() {
return dn;
}
public void setDn(String dn) {
this.dn = dn;
}
public byte[] getContent() {
return this.content;
}
public void setContent(byte[] content) {
this.content = content;
}
public String getSn() {
return this.sn;
}
public void setSn(String sn) {
this.sn = sn;
}
public String getStatus() {
return this.status;
}
public void setStatus(String status) {
this.status = status;
}
public String getTemlateId() {
return this.temlateId;
}
public void setTemlateId(String temlateId) {
this.temlateId = temlateId;
}
}
}
欢迎分享，转载请注明来源：内存溢出
微信扫一扫
支付宝扫一扫
评论列表（0条）