/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.qpid.client;
import org.apache.qpid.jms.BrokerDetails;
import org.apache.qpid.transport.ConnectionSettings;
import org.apache.qpid.url.URLHelper;
import org.apache.qpid.url.URLSyntaxException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;
import java.util.Map;
public class AMQBrokerDetails implements BrokerDetails
{
private String _host;
private int _port;
private String _transport;
private Map<String, String> _options = new HashMap<String, String>();
public AMQBrokerDetails(){}
public AMQBrokerDetails(String url) throws URLSyntaxException
{
// URL should be of format tcp://host:port?option='value',option='value'
try
{
URI connection = new URI(url);
String transport = connection.getScheme();
// Handles some defaults to minimise changes to existing broker URLS e.g. localhost
if (transport != null)
{
//todo this list of valid transports should be enumerated somewhere
if (!(transport.equalsIgnoreCase(BrokerDetails.TCP)))
{
if (transport.equalsIgnoreCase("localhost"))
{
connection = new URI(DEFAULT_TRANSPORT + "://" + url);
transport = connection.getScheme();
}
else
{
if (url.charAt(transport.length()) == ':' && url.charAt(transport.length() + 1) != '/')
{
//Then most likely we have a host:port value
connection = new URI(DEFAULT_TRANSPORT + "://" + url);
transport = connection.getScheme();
}
else
{
throw URLHelper.parseError(0, transport.length(), "Unknown transport", url);
}
}
}
else if (url.indexOf("//") == -1)
{
throw new URLSyntaxException(url, "Missing '//' after the transport In broker URL",transport.length()+1,1);
}
}
else
{
//Default the transport
connection = new URI(DEFAULT_TRANSPORT + "://" + url);
transport = connection.getScheme();
}
if (transport == null)
{
throw URLHelper.parseError(-1, "Unknown transport in broker URL:'"
+ url + "' Format: " + URL_FORMAT_EXAMPLE, "");
}
setTransport(transport);
String host = connection.getHost();
// Fix for Java 1.5
if (host == null)
{
host = "";
String auth = connection.getAuthority();
if (auth != null)
{
// contains both host & port myhost:5672
if (auth.contains(":"))
{
host = auth.substring(0,auth.indexOf(":"));
}
else
{
host = auth;
}
}
}
setHost(host);
int port = connection.getPort();
if (port == -1)
{
// Fix for when there is port data but it is not automatically parseable by getPort().
String auth = connection.getAuthority();
if (auth != null && auth.contains(":"))
{
int start = auth.indexOf(":") + 1;
int end = start;
boolean looking = true;
boolean found = false;
// Throw an URL exception if the port number is not specified
if (start == auth.length())
{
throw URLHelper.parseError(connection.toString().indexOf(auth) + end - 1,
connection.toString().indexOf(auth) + end, "Port number must be specified",
connection.toString());
}
//Walk the authority looking for a port value.
while (looking)
{
try
{
end++;
Integer.parseInt(auth.substring(start, end));
if (end >= auth.length())
{
looking = false;
found = true;
}
}
catch (NumberFormatException nfe)
{
looking = false;
}
}
if (found)
{
setPort(Integer.parseInt(auth.substring(start, end)));
}
else
{
throw URLHelper.parseError(connection.toString().indexOf(connection.getAuthority()) + end - 1,
"Illegal character in port number", connection.toString());
}
}
else
{
setPort(DEFAULT_PORT);
}
}
else
{
setPort(port);
}
String queryString = connection.getQuery();
URLHelper.parseOptions(_options, queryString);
//Fragment is #string (not used)
}
catch (URISyntaxException uris)
{
if (uris instanceof URLSyntaxException)
{
throw(URLSyntaxException) uris;
}
throw URLHelper.parseError(uris.getIndex(), uris.getReason(), uris.getInput());
}
}
public AMQBrokerDetails(String host, int port)
{
_host = host;
_port = port;
}
public String getHost()
{
return _host;
}
public void setHost(String _host)
{
this._host = _host;
}
public int getPort()
{
return _port;
}
public void setPort(int _port)
{
this._port = _port;
}
public String getTransport()
{
return _transport;
}
public void setTransport(String _transport)
{
this._transport = _transport;
}
public String getProperty(String key)
{
return _options.get(key);
}
public void setProperty(String key, String value)
{
_options.put(key, value);
}
private int lookupConnectTimeout()
{
if (_options.containsKey(OPTIONS_CONNECT_TIMEOUT))
{
try
{
return Integer.parseInt(_options.get(OPTIONS_CONNECT_TIMEOUT));
}
catch (NumberFormatException nfe)
{
//Do nothing as we will use the default below.
}
}
return BrokerDetails.DEFAULT_CONNECT_TIMEOUT;
}
public boolean getBooleanProperty(String propName)
{
return getBooleanProperty(propName, false);
}
public boolean getBooleanProperty(String propName, boolean defaultValue)
{
if (_options.containsKey(propName))
{
if (_options.get(propName).equalsIgnoreCase("false"))
{
return false;
}
else if (_options.get(propName).equalsIgnoreCase("true"))
{
return true;
}
else
{
return defaultValue;
}
}
else
{
return defaultValue;
}
}
private int getIntegerProperty(String key)
{
String stringValue = getProperty(key);
try
{
return Integer.parseInt(stringValue);
}
catch (NumberFormatException e)
{
throw new IllegalArgumentException("Cannot parse key " + key + " with value '" + stringValue + "' as integer.", e);
}
}
public String toString()
{
StringBuffer sb = new StringBuffer();
sb.append(_transport);
sb.append("://");
sb.append(_host);
sb.append(':');
sb.append(_port);
sb.append(printOptionsURL());
return sb.toString();
}
public boolean equals(Object o)
{
if (!(o instanceof BrokerDetails))
{
return false;
}
BrokerDetails bd = (BrokerDetails) o;
return _host.toLowerCase().equals(bd.getHost() == null ? null : bd.getHost().toLowerCase()) &&
(_port == bd.getPort()) &&
_transport.toLowerCase().equals(bd.getTransport() == null ? null : bd.getTransport().toLowerCase());
//TODO do we need to compare all the options as well?
}
@Override
public int hashCode()
{
int result = _host != null ? _host.toLowerCase().hashCode() : 0;
result = 31 * result + _port;
result = 31 * result + (_transport != null ? _transport.toLowerCase().hashCode() : 0);
return result;
}
private String printOptionsURL()
{
StringBuffer optionsURL = new StringBuffer();
optionsURL.append('?');
if (!(_options.isEmpty()))
{
for (String key : _options.keySet())
{
optionsURL.append(key);
optionsURL.append("='");
if (OPTIONS_TRUST_STORE_PASSWORD.equals(key) || OPTIONS_KEY_STORE_PASSWORD.equals(key))
{
optionsURL.append("********");
}
else
{
optionsURL.append(_options.get(key));
}
optionsURL.append("'");
optionsURL.append(URLHelper.DEFAULT_OPTION_SEPERATOR);
}
}
//removeKey the extra DEFAULT_OPTION_SEPERATOR or the '?' if there are no options
optionsURL.deleteCharAt(optionsURL.length() - 1);
return optionsURL.toString();
}
public static String checkTransport(String broker)
{
if ((!broker.contains("://")))
{
return "tcp://" + broker;
}
else
{
return broker;
}
}
public Map<String, String> getProperties()
{
return _options;
}
public void setProperties(Map<String, String> props)
{
_options = props;
}
public ConnectionSettings buildConnectionSettings()
{
ConnectionSettings conSettings = new ConnectionSettings();
conSettings.setHost(getHost());
conSettings.setPort(getPort());
// ------------ sasl options ---------------
if (getProperty(BrokerDetails.OPTIONS_SASL_MECHS) != null)
{
conSettings.setSaslMechs(
getProperty(BrokerDetails.OPTIONS_SASL_MECHS));
}
// Sun SASL Kerberos client uses the
// protocol + servername as the service key.
if (getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME) != null)
{
conSettings.setSaslProtocol(
getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME));
}
if (getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME) != null)
{
conSettings.setSaslServerName(
getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME));
}
conSettings.setUseSASLEncryption(
getBooleanProperty(BrokerDetails.OPTIONS_SASL_ENCRYPTION));
// ------------- ssl options ---------------------
conSettings.setUseSSL(getBooleanProperty(BrokerDetails.OPTIONS_SSL));
if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE) != null)
{
conSettings.setTrustStorePath(
getProperty(BrokerDetails.OPTIONS_TRUST_STORE));
}
if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD) != null)
{
conSettings.setTrustStorePassword(
getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD));
}
if (getProperty(BrokerDetails.OPTIONS_KEY_STORE) != null)
{
conSettings.setKeyStorePath(
getProperty(BrokerDetails.OPTIONS_KEY_STORE));
}
if (getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD) != null)
{
conSettings.setKeyStorePassword(
getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD));
}
if (getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS) != null)
{
conSettings.setCertAlias(
getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS));
}
// ----------------------------
conSettings.setVerifyHostname(getBooleanProperty(BrokerDetails.OPTIONS_SSL_VERIFY_HOSTNAME));
if (getProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY) != null)
{
conSettings.setTcpNodelay(
getBooleanProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY,true));
}
conSettings.setConnectTimeout(lookupConnectTimeout());
if (getProperty(BrokerDetails.OPTIONS_HEARTBEAT) != null)
{
conSettings.setHeartbeatInterval(getIntegerProperty(BrokerDetails.OPTIONS_HEARTBEAT));
}
else if (getProperty(BrokerDetails.OPTIONS_IDLE_TIMEOUT) != null)
{
conSettings.setHeartbeatInterval(getIntegerProperty(BrokerDetails.OPTIONS_IDLE_TIMEOUT) / 1000);
}
return conSettings;
}
}