/jboss-as-7.1.1.Final/web/src/main/java/org/jboss/as/web/WebVirtualHostService.java
Java | 218 lines | 165 code | 24 blank | 29 comment | 28 complexity | 36eb75deb243332356d4954a0038b354 MD5 | raw file
Possible License(s): LGPL-2.1, Apache-2.0
- /*
- * JBoss, Home of Professional Open Source.
- * Copyright 2010, Red Hat, Inc., and individual contributors
- * as indicated by the @author tags. See the copyright.txt file in the
- * distribution for a full listing of individual contributors.
- *
- * This is free software; you can redistribute it and/or modify it
- * under the terms of the GNU Lesser General Public License as
- * published by the Free Software Foundation; either version 2.1 of
- * the License, or (at your option) any later version.
- *
- * This software is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this software; if not, write to the Free
- * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
- * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
- */
- package org.jboss.as.web;
- import org.apache.catalina.Container;
- import org.apache.catalina.Valve;
- import org.apache.catalina.authenticator.SingleSignOn;
- import org.apache.catalina.core.StandardHost;
- import org.apache.catalina.valves.AccessLogValve;
- import org.apache.catalina.valves.ExtendedAccessLogValve;
- import org.jboss.as.clustering.web.sso.SSOClusterManager;
- import org.jboss.as.web.sso.ClusteredSingleSignOn;
- import org.jboss.dmr.ModelNode;
- import org.jboss.msc.service.Service;
- import org.jboss.msc.service.StartContext;
- import org.jboss.msc.service.StartException;
- import org.jboss.msc.service.StopContext;
- import org.jboss.msc.value.InjectedValue;
- import org.jboss.web.rewrite.RewriteValve;
- /**
- * Service creating and registering a virtual host.
- *
- * @author Emanuel Muckenhuber
- */
- public class WebVirtualHostService implements Service<VirtualHost> {
- private final String name;
- private final String[] aliases;
- private String defaultWebModule;
- private boolean hasWelcomeRoot;
- private ModelNode accessLog;
- private ModelNode rewrite;
- private ModelNode sso;
- private final InjectedValue<String> tempPathInjector = new InjectedValue<String>();
- private final InjectedValue<String> accessLogPathInjector = new InjectedValue<String>();
- private final InjectedValue<WebServer> webServer = new InjectedValue<WebServer>();
- private final InjectedValue<SSOClusterManager> ssoManager = new InjectedValue<SSOClusterManager>();
- private VirtualHost host;
- public WebVirtualHostService(String name, String[] aliases, boolean hasWelcomeRoot) {
- this.name = name;
- this.aliases = aliases;
- this.hasWelcomeRoot = hasWelcomeRoot;
- }
- /** {@inheritDoc} */
- public synchronized void start(StartContext context) throws StartException {
- final StandardHost host = new StandardHost();
- host.setAppBase(tempPathInjector.getValue());
- host.setName(name);
- for(final String alias : aliases) {
- host.addAlias(alias);
- }
- if(accessLog != null) {
- host.addValve(createAccessLogValve(host, accessLogPathInjector.getValue(), accessLog));
- }
- if(rewrite != null) {
- host.addValve(createRewriteValve(host, rewrite));
- }
- if(sso != null) {
- host.addValve(createSsoValve(host, sso));
- }
- if (defaultWebModule != null) {
- host.setDefaultWebapp(defaultWebModule);
- }
- try {
- final WebServer server = webServer.getValue();
- server.addHost(host);
- } catch(Exception e) {
- throw new StartException(e);
- }
- this.host = new VirtualHost(host, hasWelcomeRoot);
- }
- /** {@inheritDoc} */
- public synchronized void stop(StopContext context) {
- final VirtualHost host = this.host;
- this.host = null;
- final WebServer server = webServer.getValue();
- server.removeHost(host.getHost());
- }
- /** {@inheritDoc} */
- public synchronized VirtualHost getValue() throws IllegalStateException {
- final VirtualHost host = this.host;
- if(host == null) {
- throw new IllegalStateException();
- }
- return host;
- }
- void setAccessLog(final ModelNode accessLog) {
- this.accessLog = accessLog;
- }
- void setRewrite(ModelNode rewrite) {
- this.rewrite = rewrite;
- }
- void setSso(final ModelNode sso) {
- this.sso = sso;
- }
- protected String getDefaultWebModule() {
- return defaultWebModule;
- }
- protected void setDefaultWebModule(String defaultWebModule) {
- this.defaultWebModule = defaultWebModule;
- }
- public InjectedValue<String> getAccessLogPathInjector() {
- return accessLogPathInjector;
- }
- public InjectedValue<String> getTempPathInjector() {
- return tempPathInjector;
- }
- public InjectedValue<WebServer> getWebServer() {
- return webServer;
- }
- public InjectedValue<SSOClusterManager> getSSOClusterManager() {
- return ssoManager;
- }
- static Valve createAccessLogValve(final Container container, final String logDirectory, final ModelNode element) {
- boolean extended = false;
- if (element.hasDefined(Constants.EXTENDED)) {
- extended = element.get(Constants.EXTENDED).asBoolean();
- }
- final AccessLogValve log;
- if (extended) {
- log = new ExtendedAccessLogValve();
- } else {
- log = new AccessLogValve();
- }
- log.setDirectory(logDirectory);
- if (element.hasDefined(Constants.RESOLVE_HOSTS)) log.setResolveHosts(element.get(Constants.RESOLVE_HOSTS).asBoolean());
- if (element.hasDefined(Constants.ROTATE)) log.setRotatable(element.get(Constants.ROTATE).asBoolean());
- if (element.hasDefined(Constants.PATTERN)) {
- log.setPattern(element.get(Constants.PATTERN).asString());
- } else {
- log.setPattern("common");
- }
- if (element.hasDefined(Constants.PREFIX)) log.setPrefix(element.get(Constants.PREFIX).asString());
- return log;
- }
- static Valve createRewriteValve(final Container container, final ModelNode element) throws StartException {
- final RewriteValve rewriteValve = new RewriteValve();
- rewriteValve.setContainer(container);
- StringBuffer configuration = new StringBuffer();
- for (final ModelNode rewriteElement : element.asList()) {
- final ModelNode rewrite = rewriteElement.asProperty().getValue();
- if (rewrite.has(Constants.CONDITION)) {
- for (final ModelNode conditionElement : rewrite.get(Constants.CONDITION).asList()) {
- final ModelNode condition = conditionElement.asProperty().getValue();
- configuration.append("RewriteCond ")
- .append(condition.get(Constants.TEST).asString())
- .append(" ").append(condition.get(Constants.PATTERN).asString());
- if (condition.hasDefined(Constants.FLAGS)) {
- configuration.append(" [").append(condition.get(Constants.FLAGS).asString()).append("]\r\n");
- } else {
- configuration.append("\r\n");
- }
- }
- }
- configuration.append("RewriteRule ")
- .append(rewrite.get(Constants.PATTERN).asString())
- .append(" ").append(rewrite.get(Constants.SUBSTITUTION).asString());
- if (rewrite.hasDefined(Constants.FLAGS)) {
- configuration.append(" [").append(rewrite.get(Constants.FLAGS).asString()).append("]\r\n");
- } else {
- configuration.append("\r\n");
- }
- }
- try {
- rewriteValve.setConfiguration(configuration.toString());
- } catch(Exception e) {
- throw new StartException(e);
- }
- return rewriteValve;
- }
- Valve createSsoValve(final Container container, final ModelNode element) throws StartException {
- final SingleSignOn ssoValve = element.hasDefined(Constants.CACHE_CONTAINER) ? new ClusteredSingleSignOn(this.ssoManager.getValue()) : new SingleSignOn();
- if (element.hasDefined(Constants.DOMAIN)) ssoValve.setCookieDomain(element.get(Constants.DOMAIN).asString());
- if (element.hasDefined(Constants.REAUTHENTICATE)) ssoValve.setRequireReauthentication(element.get(Constants.REAUTHENTICATE).asBoolean());
- return ssoValve;
- }
- }