Examples with HttpServerCookie org.wildfly.security.http.HttpServerCookie used on opensource projects

Example 1 with HttpServerCookie

use of org.wildfly.security.http.HttpServerCookie in project keycloak by keycloak.

the class ElytronHttpFacade method getResponse.

@Override
public Response getResponse() {
    return new Response() {

        @Override
        public void setStatus(final int status) {
            if (status < 200 || status > 300) {
                responseConsumer = responseConsumer.andThen(response -> response.setStatusCode(status));
            }
        }

        @Override
        public void addHeader(final String name, final String value) {
            headers.put(name, value);
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse response) {
                    String latestValue = headers.get(name);
                    if (latestValue.equals(value)) {
                        response.addResponseHeader(name, latestValue);
                    }
                }
            });
        }

        @Override
        public void setHeader(String name, String value) {
            addHeader(name, value);
        }

        @Override
        public void resetCookie(final String name, final String path) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, "", path, null, 0, false, false, response));
            HttpScope exchangeScope = getScope(Scope.EXCHANGE);
            ProtectedHttpServerExchange undertowExchange = ProtectedHttpServerExchange.class.cast(exchangeScope.getAttachment(UNDERTOW_EXCHANGE));
            if (undertowExchange != null) {
                CookieImpl cookie = new CookieImpl(name, "");
                cookie.setMaxAge(0);
                cookie.setPath(path);
                undertowExchange.getExchange().setResponseCookie(cookie);
            }
        }

        @Override
        public void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, value, path, domain, maxAge, secure, httpOnly, response));
        }

        private void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly, HttpServerResponse response) {
            response.setResponseCookie(new HttpServerCookie() {

                @Override
                public String getName() {
                    return name;
                }

                @Override
                public String getValue() {
                    return value;
                }

                @Override
                public String getDomain() {
                    return domain;
                }

                @Override
                public int getMaxAge() {
                    return maxAge;
                }

                @Override
                public String getPath() {
                    return path;
                }

                @Override
                public boolean isSecure() {
                    return secure;
                }

                @Override
                public int getVersion() {
                    return 0;
                }

                @Override
                public boolean isHttpOnly() {
                    return httpOnly;
                }
            });
        }

        @Override
        public OutputStream getOutputStream() {
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse httpServerResponse) {
                    try {
                        httpServerResponse.getOutputStream().write(stream.toByteArray());
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to write to response output stream", e);
                    }
                }
            });
            return stream;
        }

        @Override
        public void sendError(int code) {
            setStatus(code);
        }

        @Override
        public void sendError(final int code, final String message) {
            responseConsumer = responseConsumer.andThen(response -> {
                response.setStatusCode(code);
                response.addResponseHeader("Content-Type", "text/html");
                try {
                    response.getOutputStream().write(message.getBytes());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            });
        }

        @Override
        public void end() {
        }
    };
}

Example 2 with HttpServerCookie

use of org.wildfly.security.http.HttpServerCookie in project keycloak by keycloak.

the class ElytronHttpFacade method getRequest.

@Override
public Request getRequest() {
    return new Request() {

        private InputStream inputStream;

        @Override
        public String getMethod() {
            return request.getRequestMethod();
        }

        @Override
        public String getURI() {
            try {
                return URLDecoder.decode(request.getRequestURI().toString(), "UTF-8");
            } catch (UnsupportedEncodingException e) {
                throw new RuntimeException("Failed to decode request URI", e);
            }
        }

        @Override
        public String getRelativePath() {
            return request.getRequestPath();
        }

        @Override
        public boolean isSecure() {
            return request.getRequestURI().getScheme().equals("https");
        }

        @Override
        public String getFirstParam(String param) {
            return request.getFirstParameterValue(param);
        }

        @Override
        public String getQueryParamValue(String param) {
            URI requestURI = request.getRequestURI();
            String query = requestURI.getQuery();
            if (query != null) {
                String[] parameters = query.split("&");
                for (String parameter : parameters) {
                    String[] keyValue = parameter.split("=", 2);
                    if (keyValue[0].equals(param)) {
                        try {
                            return URLDecoder.decode(keyValue[1], "UTF-8");
                        } catch (IOException e) {
                            throw new RuntimeException("Failed to decode request URI", e);
                        }
                    }
                }
            }
            return null;
        }

        @Override
        public Cookie getCookie(final String cookieName) {
            List<HttpServerCookie> cookies = request.getCookies();
            if (cookies != null) {
                for (HttpServerCookie cookie : cookies) {
                    if (cookie.getName().equals(cookieName)) {
                        return new Cookie(cookie.getName(), cookie.getValue(), cookie.getVersion(), cookie.getDomain(), cookie.getPath());
                    }
                }
            }
            return null;
        }

        @Override
        public String getHeader(String name) {
            return request.getFirstRequestHeaderValue(name);
        }

        @Override
        public List<String> getHeaders(String name) {
            return request.getRequestHeaderValues(name);
        }

        @Override
        public InputStream getInputStream() {
            return getInputStream(false);
        }

        @Override
        public InputStream getInputStream(boolean buffered) {
            if (inputStream != null) {
                return inputStream;
            }
            if (buffered) {
                return inputStream = new BufferedInputStream(request.getInputStream());
            }
            return request.getInputStream();
        }

        @Override
        public String getRemoteAddr() {
            InetSocketAddress sourceAddress = request.getSourceAddress();
            if (sourceAddress == null) {
                return "";
            }
            InetAddress address = sourceAddress.getAddress();
            if (address == null) {
                // returning null
                return sourceAddress.getHostString();
            }
            return address.getHostAddress();
        }

        @Override
        public void setError(AuthenticationError error) {
            request.getScope(Scope.EXCHANGE).setAttachment(AuthenticationError.class.getName(), error);
        }

        @Override
        public void setError(LogoutError error) {
            request.getScope(Scope.EXCHANGE).setAttachment(LogoutError.class.getName(), error);
        }
    };
}

Example 3 with HttpServerCookie

use of org.wildfly.security.http.HttpServerCookie in project keycloak by keycloak.

the class ElytronHttpFacade method getRequest.

@Override
public Request getRequest() {
    return new Request() {

        private InputStream inputStream;

        @Override
        public String getMethod() {
            return request.getRequestMethod();
        }

        @Override
        public String getURI() {
            try {
                return URLDecoder.decode(request.getRequestURI().toString(), "UTF-8");
            } catch (UnsupportedEncodingException e) {
                throw new RuntimeException("Failed to decode request URI", e);
            }
        }

        @Override
        public String getRelativePath() {
            return request.getRequestPath();
        }

        @Override
        public boolean isSecure() {
            return request.getRequestURI().getScheme().equals("https");
        }

        @Override
        public String getFirstParam(String param) {
            return request.getFirstParameterValue(param);
        }

        @Override
        public String getQueryParamValue(String param) {
            URI requestURI = request.getRequestURI();
            String query = requestURI.getQuery();
            if (query != null) {
                String[] parameters = query.split("&");
                for (String parameter : parameters) {
                    String[] keyValue = parameter.split("=", 2);
                    if (keyValue[0].equals(param)) {
                        try {
                            return URLDecoder.decode(keyValue[1], "UTF-8");
                        } catch (IOException e) {
                            throw new RuntimeException("Failed to decode request URI", e);
                        }
                    }
                }
            }
            return null;
        }

        @Override
        public Cookie getCookie(final String cookieName) {
            List<HttpServerCookie> cookies = request.getCookies();
            if (cookies != null) {
                for (HttpServerCookie cookie : cookies) {
                    if (cookie.getName().equals(cookieName)) {
                        return new Cookie(cookie.getName(), cookie.getValue(), cookie.getVersion(), cookie.getDomain(), cookie.getPath());
                    }
                }
            }
            return null;
        }

        @Override
        public String getHeader(String name) {
            return request.getFirstRequestHeaderValue(name);
        }

        @Override
        public List<String> getHeaders(String name) {
            return request.getRequestHeaderValues(name);
        }

        @Override
        public InputStream getInputStream() {
            return getInputStream(false);
        }

        @Override
        public InputStream getInputStream(boolean buffered) {
            if (inputStream != null) {
                return inputStream;
            }
            if (buffered) {
                HttpScope exchangeScope = getScope(Scope.EXCHANGE);
                HttpServerExchange exchange = ProtectedHttpServerExchange.class.cast(exchangeScope.getAttachment(UNDERTOW_EXCHANGE)).getExchange();
                ServletRequestContext context = exchange.getAttachment(ServletRequestContext.ATTACHMENT_KEY);
                ServletRequest servletRequest = context.getServletRequest();
                inputStream = new BufferedInputStream(exchange.getInputStream());
                context.setServletRequest(new HttpServletRequestWrapper((HttpServletRequest) servletRequest) {

                    @Override
                    public ServletInputStream getInputStream() {
                        inputStream.mark(0);
                        return new ServletInputStream() {

                            @Override
                            public int read() throws IOException {
                                return inputStream.read();
                            }
                        };
                    }
                });
                return inputStream;
            }
            return request.getInputStream();
        }

        @Override
        public String getRemoteAddr() {
            InetSocketAddress sourceAddress = request.getSourceAddress();
            if (sourceAddress == null) {
                return "";
            }
            InetAddress address = sourceAddress.getAddress();
            if (address == null) {
                // returning null
                return sourceAddress.getHostString();
            }
            return address.getHostAddress();
        }

        @Override
        public void setError(AuthenticationError error) {
            request.getScope(Scope.EXCHANGE).setAttachment(AuthenticationError.class.getName(), error);
        }

        @Override
        public void setError(LogoutError error) {
            request.getScope(Scope.EXCHANGE).setAttachment(LogoutError.class.getName(), error);
        }
    };
}

Example 4 with HttpServerCookie

use of org.wildfly.security.http.HttpServerCookie in project keycloak by keycloak.

the class ElytronHttpFacade method getResponse.

@Override
public Response getResponse() {
    return new Response() {

        @Override
        public void setStatus(final int status) {
            responseConsumer = responseConsumer.andThen(response -> response.setStatusCode(status));
        }

        @Override
        public void addHeader(final String name, final String value) {
            responseConsumer = responseConsumer.andThen(response -> response.addResponseHeader(name, value));
        }

        @Override
        public void setHeader(String name, String value) {
            addHeader(name, value);
        }

        @Override
        public void resetCookie(final String name, final String path) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, "", path, null, 0, false, false, response));
        }

        @Override
        public void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly) {
            responseConsumer = responseConsumer.andThen(response -> setCookie(name, value, path, domain, maxAge, secure, httpOnly, response));
        }

        private void setCookie(final String name, final String value, final String path, final String domain, final int maxAge, final boolean secure, final boolean httpOnly, HttpServerResponse response) {
            response.setResponseCookie(new HttpServerCookie() {

                @Override
                public String getName() {
                    return name;
                }

                @Override
                public String getValue() {
                    return value;
                }

                @Override
                public String getDomain() {
                    return domain;
                }

                @Override
                public int getMaxAge() {
                    return maxAge;
                }

                @Override
                public String getPath() {
                    return path;
                }

                @Override
                public boolean isSecure() {
                    return secure;
                }

                @Override
                public int getVersion() {
                    return 0;
                }

                @Override
                public boolean isHttpOnly() {
                    return httpOnly;
                }
            });
        }

        @Override
        public OutputStream getOutputStream() {
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            responseConsumer = responseConsumer.andThen(new Consumer<HttpServerResponse>() {

                @Override
                public void accept(HttpServerResponse httpServerResponse) {
                    try {
                        httpServerResponse.getOutputStream().write(stream.toByteArray());
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to write to response output stream", e);
                    }
                }
            });
            return stream;
        }

        @Override
        public void sendError(int code) {
            setStatus(code);
        }

        @Override
        public void sendError(final int code, final String message) {
            responseConsumer = responseConsumer.andThen(response -> {
                response.setStatusCode(code);
                response.addResponseHeader("Content-Type", "text/html");
                try {
                    response.getOutputStream().write(message.getBytes());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            });
        }

        @Override
        public void end() {
        }
    };
}