Search in sources :

Example 1 with ClientRequest

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;
}
Also used : ClientHandlerException(com.sun.jersey.api.client.ClientHandlerException) ClientResponse(com.sun.jersey.api.client.ClientResponse) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) LoggingFilter(com.sun.jersey.api.client.filter.LoggingFilter) ArrayList(java.util.ArrayList) WebResource(com.sun.jersey.api.client.WebResource) HTTPBasicAuthFilter(com.sun.jersey.api.client.filter.HTTPBasicAuthFilter) ClientFilter(com.sun.jersey.api.client.filter.ClientFilter) ClientConfig(com.sun.jersey.api.client.config.ClientConfig) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) Client(com.sun.jersey.api.client.Client) ClientRequest(com.sun.jersey.api.client.ClientRequest) NewCookie(javax.ws.rs.core.NewCookie)

Example 2 with ClientRequest

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;
}
Also used : ClientHandlerException(com.sun.jersey.api.client.ClientHandlerException) ClientResponse(com.sun.jersey.api.client.ClientResponse) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) LoggingFilter(com.sun.jersey.api.client.filter.LoggingFilter) ArrayList(java.util.ArrayList) WebResource(com.sun.jersey.api.client.WebResource) HTTPBasicAuthFilter(com.sun.jersey.api.client.filter.HTTPBasicAuthFilter) ClientFilter(com.sun.jersey.api.client.filter.ClientFilter) List(java.util.List) ArrayList(java.util.ArrayList) LinkedList(java.util.LinkedList) ClientConfig(com.sun.jersey.api.client.config.ClientConfig) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) Client(com.sun.jersey.api.client.Client) ClientRequest(com.sun.jersey.api.client.ClientRequest) NewCookie(javax.ws.rs.core.NewCookie)

Example 3 with ClientRequest

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;
        }
    });
}
Also used : ClientHandlerException(com.sun.jersey.api.client.ClientHandlerException) ClientResponse(com.sun.jersey.api.client.ClientResponse) LoggingFilter(com.sun.jersey.api.client.filter.LoggingFilter) ClientFilter(com.sun.jersey.api.client.filter.ClientFilter) WebResource(com.sun.jersey.api.client.WebResource) HTTPBasicAuthFilter(com.sun.jersey.api.client.filter.HTTPBasicAuthFilter) ClientRequest(com.sun.jersey.api.client.ClientRequest)

Example 4 with ClientRequest

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;
}
Also used : ClientResponse(com.sun.jersey.api.client.ClientResponse) ClientRequest(com.sun.jersey.api.client.ClientRequest)

Example 5 with ClientRequest

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;
}
Also used : ClientHandlerException(com.sun.jersey.api.client.ClientHandlerException) ClientResponse(com.sun.jersey.api.client.ClientResponse) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) ArrayList(java.util.ArrayList) ClientFilter(com.sun.jersey.api.client.filter.ClientFilter) ClientConfig(com.sun.jersey.api.client.config.ClientConfig) DefaultClientConfig(com.sun.jersey.api.client.config.DefaultClientConfig) Client(com.sun.jersey.api.client.Client) ClientRequest(com.sun.jersey.api.client.ClientRequest)

Aggregations

ClientRequest (com.sun.jersey.api.client.ClientRequest)5 ClientResponse (com.sun.jersey.api.client.ClientResponse)5 ClientHandlerException (com.sun.jersey.api.client.ClientHandlerException)4 ClientFilter (com.sun.jersey.api.client.filter.ClientFilter)4 Client (com.sun.jersey.api.client.Client)3 WebResource (com.sun.jersey.api.client.WebResource)3 ClientConfig (com.sun.jersey.api.client.config.ClientConfig)3 DefaultClientConfig (com.sun.jersey.api.client.config.DefaultClientConfig)3 HTTPBasicAuthFilter (com.sun.jersey.api.client.filter.HTTPBasicAuthFilter)3 LoggingFilter (com.sun.jersey.api.client.filter.LoggingFilter)3 ArrayList (java.util.ArrayList)3 NewCookie (javax.ws.rs.core.NewCookie)2 LinkedList (java.util.LinkedList)1 List (java.util.List)1