vanitasvitae/Smack
1
0
Fork 0
mirror of https://github.com/vanitasvitae/Smack.git synced 2024-11-25 05:22:06 +01:00
Code Issues Releases Wiki Activity

Check callbackHandler/keystorePath for null

Browse source 
before using it.
This commit is contained in:
Florian Schmaus 2016-01-21 20:20:42 +01:00
parent d5c7eb7349
commit e0cbb95b5d
1 changed files with 14 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
smack-tcp/src/main/java/org/jivesoftware/smack/tcp/XMPPTCPConnection.java
View file

@ -91,6 +91,7 @@ import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.SSLContext; import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocket; import javax.net.ssl.SSLSocket;
import javax.security.auth.callback.Callback; import javax.security.auth.callback.Callback;
import javax.security.auth.callback.CallbackHandler;
import javax.security.auth.callback.PasswordCallback; import javax.security.auth.callback.PasswordCallback;
import java.io.BufferedReader; import java.io.BufferedReader;
@ -675,6 +676,8 @@ public class XMPPTCPConnection extends AbstractXMPPConnection {
if (context == null) { if (context == null) {
final String keyStoreType = config.getKeystoreType(); final String keyStoreType = config.getKeystoreType();
final CallbackHandler callbackHandler = config.getCallbackHandler();
final String keystorePath = config.getKeystorePath();
if ("PKCS11".equals(keyStoreType)) { if ("PKCS11".equals(keyStoreType)) {
try { try {
Constructor<?> c = Class.forName("sun.security.pkcs11.SunPKCS11").getConstructor(InputStream.class); Constructor<?> c = Class.forName("sun.security.pkcs11.SunPKCS11").getConstructor(InputStream.class);
@ -684,7 +687,7 @@ public class XMPPTCPConnection extends AbstractXMPPConnection {
Security.addProvider(p); Security.addProvider(p);
ks = KeyStore.getInstance("PKCS11",p); ks = KeyStore.getInstance("PKCS11",p);
pcb = new PasswordCallback("PKCS11 Password: ",false); pcb = new PasswordCallback("PKCS11 Password: ",false);
this.config.getCallbackHandler().handle(new Callback[]{pcb}); callbackHandler.handle(new Callback[]{pcb});
ks.load(null,pcb.getPassword()); ks.load(null,pcb.getPassword());
} }
catch (Exception e) { catch (Exception e) {
@ -700,14 +703,16 @@ public class XMPPTCPConnection extends AbstractXMPPConnection {
} }
else if (keyStoreType != null){ else if (keyStoreType != null){
ks = KeyStore.getInstance(keyStoreType); ks = KeyStore.getInstance(keyStoreType);
try { if (callbackHandler != null && StringUtils.isNotEmpty(keystorePath)) {
pcb = new PasswordCallback("Keystore Password: ",false); try {
config.getCallbackHandler().handle(new Callback[]{pcb}); pcb = new PasswordCallback("Keystore Password: ", false);
ks.load(new FileInputStream(config.getKeystorePath()), pcb.getPassword()); callbackHandler.handle(new Callback[] { pcb });
} ks.load(new FileInputStream(keystorePath), pcb.getPassword());
catch(Exception e) { }
LOGGER.log(Level.WARNING, "Exception", e); catch (Exception e) {
ks = null; LOGGER.log(Level.WARNING, "Exception", e);
ks = null;
}
} }
} }