mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-09-27 18:19:33 +02:00
665e7914f2
on package layer instead of Declarative Service (DS) approach. Restructuring and cleanup of initialization process to ensure that all internal config files are found by the corresponding bundle classloaders. SMACK-343
105 lines
3.4 KiB
Java
105 lines
3.4 KiB
Java
/**
|
|
*
|
|
* Copyright 2013-2014 Florian Schmaus
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package org.jivesoftware.smack.util.dns.javax;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Hashtable;
|
|
import java.util.List;
|
|
|
|
import javax.naming.NamingEnumeration;
|
|
import javax.naming.NamingException;
|
|
import javax.naming.directory.Attribute;
|
|
import javax.naming.directory.Attributes;
|
|
import javax.naming.directory.DirContext;
|
|
import javax.naming.directory.InitialDirContext;
|
|
|
|
import org.jivesoftware.smack.initializer.SmackInitializer;
|
|
import org.jivesoftware.smack.util.DNSUtil;
|
|
import org.jivesoftware.smack.util.dns.DNSResolver;
|
|
import org.jivesoftware.smack.util.dns.SRVRecord;
|
|
|
|
/**
|
|
* A DNS resolver (mostly for SRV records), which makes use of the API provided in the javax.* namespace.
|
|
*
|
|
* @author Florian Schmaus
|
|
*
|
|
*/
|
|
public class JavaxResolver implements SmackInitializer, DNSResolver {
|
|
|
|
private static JavaxResolver instance;
|
|
private static DirContext dirContext;
|
|
|
|
static {
|
|
try {
|
|
Hashtable<String, String> env = new Hashtable<String, String>();
|
|
env.put("java.naming.factory.initial", "com.sun.jndi.dns.DnsContextFactory");
|
|
dirContext = new InitialDirContext(env);
|
|
} catch (Exception e) {
|
|
// Ignore.
|
|
}
|
|
|
|
// Try to set this DNS resolver as primary one
|
|
setup();
|
|
}
|
|
|
|
public static synchronized DNSResolver getInstance() {
|
|
if (instance == null && isSupported()) {
|
|
instance = new JavaxResolver();
|
|
}
|
|
return instance;
|
|
}
|
|
|
|
public static boolean isSupported() {
|
|
return dirContext != null;
|
|
}
|
|
|
|
public static void setup() {
|
|
DNSUtil.setDNSResolver(getInstance());
|
|
}
|
|
|
|
@Override
|
|
public List<SRVRecord> lookupSRVRecords(String name) throws NamingException {
|
|
List<SRVRecord> res = new ArrayList<SRVRecord>();
|
|
|
|
Attributes dnsLookup = dirContext.getAttributes(name, new String[] { "SRV" });
|
|
Attribute srvAttribute = dnsLookup.get("SRV");
|
|
if (srvAttribute == null)
|
|
return res;
|
|
@SuppressWarnings("unchecked")
|
|
NamingEnumeration<String> srvRecords = (NamingEnumeration<String>) srvAttribute.getAll();
|
|
while (srvRecords.hasMore()) {
|
|
String srvRecordString = srvRecords.next();
|
|
String[] srvRecordEntries = srvRecordString.split(" ");
|
|
int priority = Integer.parseInt(srvRecordEntries[srvRecordEntries.length - 4]);
|
|
int port = Integer.parseInt(srvRecordEntries[srvRecordEntries.length - 2]);
|
|
int weight = Integer.parseInt(srvRecordEntries[srvRecordEntries.length - 3]);
|
|
String host = srvRecordEntries[srvRecordEntries.length - 1];
|
|
|
|
SRVRecord srvRecord = new SRVRecord(host, port, priority, weight);
|
|
res.add(srvRecord);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
@Override
|
|
public List<Exception> initialize() {
|
|
setup();
|
|
return null;
|
|
}
|
|
|
|
}
|