use of com.sun.jersey.api.client.ClientRequest in project coprhd-controller by CoprHD.
the class ApiTestBase method createCookieHttpsClient.
/**
* Use this client if you want to use cookies instead of the http headers for holding the
* auth token
*/
protected Client createCookieHttpsClient(final String username, final String password) throws NoSuchAlgorithmException {
// Disable server certificate validation as we are using
// self-signed certificate
disableCertificateValidation();
final ClientConfig config = new DefaultClientConfig();
final Client c = Client.create(config);
c.addFilter(new LoggingFilter());
c.setFollowRedirects(false);
c.addFilter(new HTTPBasicAuthFilter(username, password));
c.addFilter(new ClientFilter() {
private ArrayList<Object> cookies;
private ArrayList<Object> getCookiesToSet() {
if (cookies != null && !cookies.isEmpty()) {
ArrayList<Object> cookiesToSet = new ArrayList<Object>();
StringBuilder cookieToAdd = new StringBuilder();
for (Object cookieRaw : cookies) {
NewCookie cookie = (NewCookie) cookieRaw;
cookieToAdd.append(cookie.getName());
cookieToAdd.append("=");
cookieToAdd.append(cookie.getValue());
cookieToAdd.append("; ");
}
cookiesToSet.add(cookieToAdd);
return cookiesToSet;
}
return null;
}
@Override
public ClientResponse handle(ClientRequest request) throws ClientHandlerException {
ArrayList<Object> cookiesToSet = getCookiesToSet();
if (cookiesToSet != null) {
request.getHeaders().put("Cookie", cookiesToSet);
}
ClientResponse response = getNext().handle(request);
if (response.getCookies() != null) {
if (cookies == null) {
cookies = new ArrayList<Object>();
}
// simple addAll just for illustration (should probably check for duplicates and expired cookies)
cookies.addAll(response.getCookies());
}
if (response.getStatus() == 302) {
WebResource wb = c.resource(response.getLocation());
cookiesToSet = getCookiesToSet();
if (cookiesToSet != null) {
response = wb.header("Cookie", cookiesToSet).get(ClientResponse.class);
} else {
response = wb.get(ClientResponse.class);
}
}
return response;
}
});
return c;
}
use of com.sun.jersey.api.client.ClientRequest in project coprhd-controller by CoprHD.
the class ApiTestBase method createHttpsClient.
protected BalancedWebResource createHttpsClient(final String username, final String password, List<String> hosts, boolean addAuthFilters) throws NoSuchAlgorithmException {
String verb = System.getenv("API_TEST_VERBOSE");
boolean verbose = false;
if (verb != null && verb.equalsIgnoreCase("true")) {
verbose = true;
}
// Disable server certificate validation as we are using
// self-signed certificate
disableCertificateValidation();
BalancedWebResource lbw = new BalancedWebResource();
for (String h : hosts) {
final ClientConfig config = new DefaultClientConfig();
final Client c = Client.create(config);
if (verbose) {
c.addFilter(new LoggingFilter());
}
if (addAuthFilters) {
c.setFollowRedirects(false);
c.addFilter(new HTTPBasicAuthFilter(username, password));
c.addFilter(new ClientFilter() {
@Override
public ClientResponse handle(ClientRequest request) throws ClientHandlerException {
if (_savedTokens.containsKey(username)) {
ArrayList<Object> token = new ArrayList<Object>();
token.add(_savedTokens.get(username));
request.getHeaders().put(AUTH_TOKEN_HEADER, token);
}
ClientResponse response = getNext().handle(request);
// this only used to inspect cookies after the fact.
if (null == _lastUsedAuthTokenCookie) {
System.out.println("Cookie was null");
List<NewCookie> allCookies = response.getCookies();
System.out.println("Cookie list size:" + allCookies.size());
for (NewCookie ck : allCookies) {
System.out.print("Cookie name " + ck.getName());
if (ck.getName().equals(AUTH_TOKEN_HEADER)) {
_lastUsedAuthTokenCookie = ck.getValue();
break;
}
}
}
if (response.getHeaders() != null && response.getHeaders().get(AUTH_TOKEN_HEADER) != null) {
_savedTokens.put(username, response.getHeaders().getFirst(AUTH_TOKEN_HEADER));
}
if (response.getHeaders() != null && response.getHeaders().get(AUTH_PROXY_TOKEN_HEADER) != null) {
_savedProxyTokens.put(username, response.getHeaders().getFirst(AUTH_PROXY_TOKEN_HEADER));
}
if (response.getStatus() == 302) {
WebResource wb = c.resource(response.getLocation());
response = wb.header(AUTH_TOKEN_HEADER, _savedTokens.get(username)).get(ClientResponse.class);
}
return response;
}
});
} else {
c.setFollowRedirects(true);
}
lbw.addWebResource(c.resource(h));
}
return lbw;
}
use of com.sun.jersey.api.client.ClientRequest in project coprhd-controller by CoprHD.
the class RESTClientUtil method authenticate.
private void authenticate(final Client c, final int authenticateCallCount) throws NoSuchAlgorithmException {
final String methodName = "authenticate(): ";
log.debug(methodName + "Called");
disableCertificateValidation();
c.setFollowRedirects(false);
if (log.isTraceEnabled()) {
c.addFilter(new LoggingFilter());
}
c.addFilter(new HTTPBasicAuthFilter(this._username, this._password));
c.addFilter(new ClientFilter() {
private Object authToken;
// private int callCount = authenticateCallCount;
@Override
public ClientResponse handle(ClientRequest request) throws ClientHandlerException {
if (authToken != null) {
request.getHeaders().put(AUTH_TOKEN_HEADER, Collections.singletonList(authToken));
}
ClientResponse response = getNext().handle(request);
if (response.getHeaders() != null && response.getHeaders().containsKey(AUTH_TOKEN_HEADER)) {
authToken = (response.getHeaders().getFirst(AUTH_TOKEN_HEADER));
}
if (response.getStatus() == 302) {
WebResource wb = c.resource(response.getLocation());
response = wb.header(AUTH_TOKEN_HEADER, authToken).get(ClientResponse.class);
}
return response;
}
});
}
use of com.sun.jersey.api.client.ClientRequest in project coprhd-controller by CoprHD.
the class AuthTokenFilter method handle.
@Override
public ClientResponse handle(ClientRequest request) throws ClientHandlerException {
addTokenToRequest(request);
ClientResponse response = getNext().handle(request);
// Handle a redirect
if (response.getClientResponseStatus() == ClientResponse.Status.FOUND) {
if (response.getHeaders().containsKey(HttpHeaders.LOCATION)) {
String location = response.getHeaders().getFirst(HttpHeaders.LOCATION);
final ClientRequest newRequest = ClientRequest.create().build(URI.create(location), request.getMethod());
// Handle the token from the existing response, add to this new request
checkResponseForToken(response);
addTokenToRequest(newRequest);
// Call handler to perform redirect to new page
response = handle(newRequest);
}
}
checkResponseForToken(response);
return response;
}
use of com.sun.jersey.api.client.ClientRequest in project coprhd-controller by CoprHD.
the class ApiTestBase method createHttpClient.
/**
* create the httpclient, returns a BalancedWebResource that can be used the same
* way a WebResource is.
*/
protected BalancedWebResource createHttpClient(final String username, final String attribute, final String groups, List<String> hostNames) {
BalancedWebResource lbw = new BalancedWebResource();
for (String h : hostNames) {
final ClientConfig config = new DefaultClientConfig();
final Client c = Client.create(config);
c.addFilter(new ClientFilter() {
@Override
public ClientResponse handle(ClientRequest request) throws ClientHandlerException {
ArrayList<Object> headerValue = new ArrayList<Object>();
String userAndAttribute = username;
if (!attribute.isEmpty()) {
userAndAttribute += "," + attribute;
}
headerValue.add(userAndAttribute + ";" + groups);
request.getHeaders().put(NoAuthHeaderUserFilter.USER_INFO_HEADER_TAG, headerValue);
return getNext().handle(request);
}
});
lbw.addWebResource(c.resource(h));
}
return lbw;
}
Aggregations