/**
* Copyright (c) 2009-2011 VMware, Inc. All Rights Reserved.
*
* 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 com.springsource.insight.plugin.apache.http.hc3;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
import com.springsource.insight.intercept.endpoint.EndPointAnalyzersRegistry;
import com.springsource.insight.intercept.operation.Operation;
import com.springsource.insight.intercept.operation.OperationFields;
import com.springsource.insight.intercept.operation.OperationList;
import com.springsource.insight.intercept.operation.OperationMap;
import com.springsource.insight.intercept.operation.OperationUtils;
import com.springsource.insight.intercept.topology.AbstractExternalResourceAnalyzer;
import com.springsource.insight.intercept.topology.ExternalResourceDescriptor;
import com.springsource.insight.intercept.topology.ExternalResourceType;
import com.springsource.insight.intercept.topology.MD5NameGenerator;
import com.springsource.insight.intercept.trace.Frame;
import com.springsource.insight.intercept.trace.FrameUtil;
import com.springsource.insight.intercept.trace.Trace;
import com.springsource.insight.util.ListUtil;
import com.springsource.insight.util.StringUtil;
/**
*
*/
public class HttpExternalResourceAnalyzer extends AbstractExternalResourceAnalyzer {
private static final HttpExternalResourceAnalyzer INSTANCE = new HttpExternalResourceAnalyzer();
private HttpExternalResourceAnalyzer() {
super(HttpClientDefinitions.TYPE);
}
public static final HttpExternalResourceAnalyzer getInstance() {
return INSTANCE;
}
public Collection<ExternalResourceDescriptor> locateExternalResourceName(Trace trace, Collection<Frame> framesList) {
if (ListUtil.size(framesList) <= 0) {
return Collections.emptyList();
}
Set<ExternalResourceDescriptor> resSet = new HashSet<ExternalResourceDescriptor>(framesList.size());
for (Frame frame : framesList) {
ExternalResourceDescriptor res = extractExternalResourceDescriptor(frame);
if (res == null) { // can happen if failed to parse the URI somehow
continue;
}
if (!resSet.add(res))
continue; // debug breakpoint
}
return resSet;
}
ExternalResourceDescriptor extractExternalResourceDescriptor(Frame frame) {
Operation op = (frame == null) ? null : frame.getOperation();
OperationMap requestDetails = (op == null) ? null : op.get("request", OperationMap.class);
String uriValue = (requestDetails == null) ? null : requestDetails.get(OperationFields.URI, String.class);
if (StringUtil.getSafeLength(uriValue) <= 0) {
return null;
}
try {
URI uri = new URI(uriValue);
String host = uri.getHost();
int port = resolvePort(uri);
String color = colorManager.getColor(op);
String app = null;
String ser = null;
String ep = null;
OperationMap responseDetails = (op == null) ? null : op.get("response", OperationMap.class);
if (responseDetails != null) {
OperationList headersList = responseDetails.get("headers", OperationList.class);
if (headersList != null) {
for (int i = 0; i < headersList.size(); i++) {
OperationMap map = headersList.get(i, OperationMap.class);
String headerName = map.get(OperationUtils.NAME_KEY, String.class);
String headerValue = map.get(OperationUtils.VALUE_KEY, String.class);
if ((app != null) && EndPointAnalyzersRegistry.APP_TOKEN_NAME.equals(headerName)) {
app = headerValue;
} else if ((ser != null) && EndPointAnalyzersRegistry.SERVER_TOKEN_NAME.equals(headerName)) {
ser = headerValue;
} else if ((ep != null) && EndPointAnalyzersRegistry.TOKEN_NAME.equals(headerName)) {
ep = headerValue;
}
if ((app != null) && (ser != null) && (ep != null)) {
break;
}
}
}
}
String hostPort = host + ":" + port;
String name = createName(hostPort);
String lbl = hostPort;
Operation rootFrameOperation = getRootFrameOperation(frame);
if (rootFrameOperation != null) {
String unresolvedURI = findUnresolvedURI(rootFrameOperation, uriValue);
if (!StringUtil.isEmpty(unresolvedURI)) {
try {
URI origuri = new URI(sanitizeURI(unresolvedURI));
lbl = origuri.getHost();
} catch (URISyntaxException e) {
// Ignore, use other label
}
}
}
return new ExternalResourceDescriptor(frame, name,
lbl, // label
ExternalResourceType.WEB_SERVER.name(),
null, // vendor
host,
port,
color, false,
app, ser, ep, null);
} catch (URISyntaxException e) {
_logger.warning("Failed to parse " + uriValue + ": " + e.getMessage());
return null;
}
}
static String createName(String lbl) {
return MD5NameGenerator.getName(lbl);
}
static int resolvePort(URI uri) {
if (uri == null) {
return (-1);
}
int port = uri.getPort();
if (port <= 0) {
if ("http".equals(uri.getScheme()))
port = 80;
else
port = 443;
}
return port;
}
private static Operation getRootFrameOperation(Frame start) {
Frame rootFrame = null;
if (start.isRoot())
rootFrame = start;
else
rootFrame = FrameUtil.getRoot(start);
if (rootFrame != null)
return rootFrame.getOperation();
return null;
}
private static String findUnresolvedURI(Operation rootFrameOperation, String resolvedURL) {
OperationMap resolvedMap = rootFrameOperation.get(OperationFields.UNRESOLVED_URI, OperationMap.class);
if (resolvedMap != null) {
return resolvedMap.get(resolvedURL, String.class);
}
return null;
}
private static String sanitizeURI(String unresolvedURI) {
return deleteAny(unresolvedURI, "{}");
}
private static String deleteAny(String inString, String charsToDelete) {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < inString.length(); i++) {
char c = inString.charAt(i);
if (charsToDelete.indexOf(c) == -1) {
sb.append(c);
}
}
return sb.toString();
}
}