Examples with CustomResult com.tremolosecurity.proxy.results.CustomResult used on opensource projects

Example 1 with CustomResult

use of com.tremolosecurity.proxy.results.CustomResult in project OpenUnison by TremoloSecurity.

the class AzSys method proccessResponseResult.

public void proccessResponseResult(ServletRequest request, ServletResponse response, ResultGroupType resGrouping, boolean forceError, AuthInfo authData, CookieConfigType cookieCfg) throws IOException, InstantiationException, IllegalAccessException, ClassNotFoundException, ServletException {
    String redir = null;
    if (resGrouping == null) {
        return;
    }
    Iterator<ResultType> it = resGrouping.getResult().iterator();
    while (it.hasNext()) {
        ResultType rt = it.next();
        if (rt.getType().equals("redirect")) {
            boolean isCustom = rt.getSource().equalsIgnoreCase("custom");
            if (!isCustom) {
                redir = rt.getValue();
            } else {
                if (isCustom) {
                    CustomResult cr = (CustomResult) Class.forName(rt.getValue()).newInstance();
                    redir = cr.getResultValue((HttpServletRequest) request, (HttpServletResponse) response);
                }
            }
        } else if (rt.getType().equalsIgnoreCase("cookie")) {
            String val = rt.getValue();
            String name, value;
            boolean isCustom = rt.getSource().equalsIgnoreCase("custom");
            // failure cookie, so can not be based on the user
            if (rt.getSource().equalsIgnoreCase("static")) {
                name = val.substring(0, val.indexOf('='));
                value = val.substring(val.indexOf('=') + 1);
            } else if (rt.getSource().equalsIgnoreCase("user") || isCustom) {
                name = val.substring(0, val.indexOf('='));
                value = val.substring(val.indexOf('=') + 1);
                if (authData.getAttribs().get(value) != null) {
                    value = authData.getAttribs().get(value).getValues().get(0);
                }
            // attrib.getValues().addAll(authData.getAttribs().get(value).getValues());
            } else {
                name = "";
                value = "";
            }
            Cookie cookie = new Cookie(name, value);
            // cookie.setDomain(((HttpServletRequest) request).getServerName() );
            String domain = ProxyTools.getInstance().getCookieDomain(cookieCfg, (HttpServletRequest) request);
            if (domain != null) {
                cookie.setDomain(domain);
            }
            cookie.setPath("/");
            cookie.setSecure(false);
            if (isCustom) {
                CustomResult cr = (CustomResult) Class.forName(cookie.getValue()).newInstance();
                cr.createResultCookie(cookie, (HttpServletRequest) request, (HttpServletResponse) response);
            }
            ((HttpServletResponse) response).addCookie(cookie);
        }
    }
    if (redir != null) {
        ((HttpServletResponse) response).sendRedirect(redir);
    } else {
        if (forceError) {
            ((HttpServletResponse) response).sendError(401);
        }
    }
}

Example 2 with CustomResult

use of com.tremolosecurity.proxy.results.CustomResult in project OpenUnison by TremoloSecurity.

the class AzSys method processRequestResult.

public void processRequestResult(ServletRequest request, ServletResponse response, ResultGroupType resGrouping, AuthInfo authData) throws ServletException, InstantiationException, IllegalAccessException, ClassNotFoundException {
    String redir = null;
    if (resGrouping == null) {
        return;
    }
    Iterator<ResultType> it = resGrouping.getResult().iterator();
    while (it.hasNext()) {
        ResultType rt = it.next();
        if (rt.getType().equals("header")) {
            String val = rt.getValue();
            String name, value;
            name = val.substring(0, val.indexOf('='));
            value = val.substring(val.indexOf('=') + 1);
            HashMap<String, Attribute> headers = (HashMap<String, Attribute>) request.getAttribute(AzSys.AUTO_IDM_HTTP_HEADERS);
            if (headers == null) {
                headers = new HashMap<String, Attribute>();
                request.setAttribute(AzSys.AUTO_IDM_HTTP_HEADERS, headers);
            }
            Attribute attrib = headers.get(name);
            if (attrib == null) {
                attrib = new Attribute(name);
                headers.put(attrib.getName(), attrib);
            }
            if (rt.getSource().equalsIgnoreCase("static")) {
                attrib.getValues().add(value);
            } else if (rt.getSource().equalsIgnoreCase("user")) {
                if (authData.getAttribs().get(value) != null) {
                    attrib.getValues().addAll(authData.getAttribs().get(value).getValues());
                }
            } else if (rt.getSource().equalsIgnoreCase("custom")) {
                CustomResult cr = (CustomResult) Class.forName(value).newInstance();
                attrib.getValues().add(cr.getResultValue((HttpServletRequest) request, (HttpServletResponse) response));
            } else {
                attrib.getValues().add("");
            }
        }
    }
}

Example 3 with CustomResult

use of com.tremolosecurity.proxy.results.CustomResult in project OpenUnison by TremoloSecurity.

the class AuthMgrSys method proccessResponseResult.

private void proccessResponseResult(ServletRequest request, ServletResponse response, ResultGroupType resGrouping, boolean forceError, UrlHolder holder) throws IOException, InstantiationException, IllegalAccessException, ClassNotFoundException, ServletException {
    String redir = null;
    if (resGrouping == null) {
        return;
    }
    Iterator<ResultType> it = resGrouping.getResult().iterator();
    while (it.hasNext()) {
        ResultType rt = it.next();
        if (rt.getType().equals("redirect")) {
            redir = rt.getValue();
        } else if (rt.getType().equalsIgnoreCase("cookie")) {
            String val = rt.getValue();
            String name, value;
            boolean isCustom = rt.getSource().equalsIgnoreCase("custom");
            // failure cookie, so can not be based on the user
            if (rt.getSource().equalsIgnoreCase("static") || isCustom) {
                name = val.substring(0, val.indexOf('='));
                value = val.substring(val.indexOf('=') + 1);
            } else {
                name = "";
                value = "";
            }
            Cookie cookie = new Cookie(name, value);
            String domain = getCookieDomain(holder, (HttpServletRequest) request);
            if (domain != null) {
                cookie.setDomain(domain);
            }
            // cookie.setDomain(((HttpServletRequest) request).getServerName());
            cookie.setPath("/");
            if (isCustom) {
                CustomResult cr = (CustomResult) Class.forName(cookie.getValue()).newInstance();
                cr.createResultCookie(cookie, (HttpServletRequest) request, (HttpServletResponse) response);
            }
            ((HttpServletResponse) response).addCookie(cookie);
        }
    }
    if (redir != null) {
        ((ProxyResponse) response).removeHeader("Location");
        ((HttpServletResponse) response).sendRedirect(redir);
    } else {
        if (forceError) {
            ((HttpServletResponse) response).sendError(401);
        }
    }
}