/*
* JBoss, Home of Professional Open Source.
* Copyright (c) 2011, 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.test.integration.security.auditing;
import static org.junit.Assert.assertTrue;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.ADD;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP_ADDR;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.REMOVE;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUBSYSTEM;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.as.arquillian.api.ServerSetup;
import org.jboss.as.arquillian.container.ManagementClient;
import org.jboss.as.test.integration.ejb.security.AnnSBTest;
import org.jboss.as.test.integration.ejb.security.authorization.SingleMethodsAnnOnlyCheckSFSB;
import org.jboss.as.test.integration.management.base.AbstractMgmtServerSetupTask;
import org.jboss.as.test.integration.security.common.AbstractSecurityDomainsServerSetupTask;
import org.jboss.as.test.integration.security.common.Utils;
import org.jboss.as.test.integration.security.common.config.SecurityDomain;
import org.jboss.as.test.integration.security.common.config.SecurityModule;
import org.jboss.dmr.ModelNode;
import org.jboss.logging.Logger;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Test;
import org.junit.runner.RunWith;
/**
* This class tests Security auditing functionality
*
* @author <a href="mailto:jlanik@redhat.com">Jan Lanik</a>.
*/
@RunWith(Arquillian.class)
@RunAsClient
@ServerSetup({SecurityAuditingTestCase.SecurityDomainsSetup.class,
SecurityAuditingTestCase.SecurityAuditingTestCaseSetup.class})
public class SecurityAuditingTestCase extends AnnSBTest {
private static final Logger log = Logger.getLogger(testClass());
private static File auditLog = new File(System.getProperty("jboss.home", null), "standalone" + File.separator + "log" + File.separator + "audit.log");
static class SecurityAuditingTestCaseSetup extends AbstractMgmtServerSetupTask {
/**
* The LOGGING
*/
private static final String LOGGING = "logging";
@Override
protected void doSetup(final ManagementClient managementClient) throws Exception {
final List<ModelNode> updates = new ArrayList<ModelNode>();
ModelNode op = new ModelNode();
op.get(OP).set(ADD);
op.get(OP_ADDR).add(SUBSYSTEM, LOGGING);
op.get(OP_ADDR).add("periodic-rotating-file-handler", "AUDIT");
op.get("level").set("TRACE");
op.get("append").set("true");
op.get("suffix").set(".yyyy-MM-dd");
ModelNode file = new ModelNode();
file.get("relative-to").set("jboss.server.log.dir");
file.get("path").set("audit.log");
op.get("file").set(file);
op.get("formatter").set("%d{HH:mm:ss,SSS} %-5p [%c] (%t) %s%E%n");
updates.add(op);
op = new ModelNode();
op.get(OP).set(ADD);
op.get(OP_ADDR).add(SUBSYSTEM, LOGGING);
op.get(OP_ADDR).add("logger", "org.jboss.security.audit");
op.get("level").set("TRACE");
ModelNode list = op.get("handlers");
list.add("AUDIT");
updates.add(op);
executeOperations(updates);
}
@Override
public void tearDown(final ManagementClient managementClient, final String containerId) throws Exception {
final List<ModelNode> updates = new ArrayList<ModelNode>();
// /subsystem=logging/logger=org.jboss.security.audit:remove
ModelNode op = new ModelNode();
op.get(OP).set(REMOVE);
op.get(OP_ADDR).add(SUBSYSTEM, LOGGING);
op.get(OP_ADDR).add("logger", "org.jboss.security.audit");
updates.add(op);
// /subsystem=logging/periodic-rotating-file-handler=AUDIT:remove
op = new ModelNode();
op.get(OP).set(REMOVE);
op.get(OP_ADDR).add(SUBSYSTEM, LOGGING);
op.get(OP_ADDR).add("periodic-rotating-file-handler", "AUDIT");
updates.add(op);
executeOperations(updates);
}
}
@Deployment
public static WebArchive deploy() {
WebArchive war = ShrinkWrap.create(WebArchive.class, "form-auth.war");
war.setWebXML("org/jboss/as/test/integration/security/auditing/form-auth/web.xml");
war.addAsWebInfResource("org/jboss/as/test/integration/security/auditing/form-auth/jboss-web.xml");
war.addAsWebResource("org/jboss/as/test/integration/security/auditing/form-auth/index.jsp");
war.addAsWebResource("org/jboss/as/test/integration/security/auditing/form-auth/login.jsp");
war.addAsWebResource("org/jboss/as/test/integration/security/auditing/form-auth/loginerror.jsp");
war.addAsWebResource("org/jboss/as/test/integration/security/auditing/form-auth/logout.jsp");
war.addAsResource("org/jboss/as/test/integration/security/auditing/form-auth/users.properties", "/users.properties");
war.addAsResource("org/jboss/as/test/integration/security/auditing/form-auth/roles.properties", "/roles.properties");
return war;
}
private static final String MODULE = "singleMethodsAnnOnlySFSB";
private static Class<?> testClass() {
return SecurityAuditingTestCase.class;
}
private static Class<?> beanClass() {
return SingleMethodsAnnOnlyCheckSFSB.class;
}
@Deployment(name = MODULE + ".jar", order = 1, testable = false)
public static Archive<JavaArchive> deployment() {
return testAppDeployment(Logger.getLogger(testClass()), MODULE, beanClass());
}
/**
* Basic test if auditing works in EJB module.
*
* @throws Exception
*/
@Test
public void testSingleMethodAnnotationsUser1Template() throws Exception {
assertTrue("Audit log file has not been created (" + auditLog.getAbsolutePath() + ")", auditLog.exists());
assertTrue("Audit log file is closed for reading (" + auditLog.getAbsolutePath() + ")", auditLog.canRead());
BufferedReader reader = new BufferedReader(new FileReader(auditLog));
while (reader.readLine() != null) {
// we need to get trough all old records (if any)
}
testSingleMethodAnnotationsUser1Template(MODULE, log, beanClass());
checkAuditLog(reader, "TRACE.+org.jboss.security.audit.+Success.+user1");
}
/**
* Basic test if auditing works for web module.
*
* @param url
* @throws Exception
*/
@Test
public void test(@ArquillianResource URL url) throws Exception {
assertTrue("Audit log file has not been created (" + auditLog.getAbsolutePath() + ")", auditLog.exists());
assertTrue("Audit log file is closed for reading (" + auditLog.getAbsolutePath() + ")", auditLog.canRead());
BufferedReader reader = new BufferedReader(new FileReader(auditLog));
while (reader.readLine() != null) {
// we need to get trough all old records (if any)
}
Utils.makeCall(url.toString(), "anil", "anil", 200);
checkAuditLog(reader, "TRACE.+org.jboss.security.audit.+Success.+anil");
}
private void checkAuditLog(BufferedReader reader, String regex) throws Exception {
Pattern successPattern = Pattern.compile(regex);
// we'll be actively waiting for a given INTERVAL for the record to appear
final long INTERVAL = 5000;
long startTime = System.currentTimeMillis();
String line;
search_for_log:
while (true) {
// some new lines were added -> go trough those and check whether our record is present
while (null != (line = reader.readLine())) {
if (successPattern.matcher(line).find()) {
break search_for_log;
}
}
// record not written to log yet -> continue checking if the time has not yet expired
if (System.currentTimeMillis() > startTime + INTERVAL) {
// time expired
throw new AssertionError("Login record has not been written into audit log! (time expired). Checked regex="
+ regex);
}
}
}
/*
* A {@link ServerSetupTask} instance which creates security domains for this test case.
*
* @author Josef Cacek
*/
static class SecurityDomainsSetup extends AbstractSecurityDomainsServerSetupTask {
/*
* Returns SecurityDomains configuration for this testcase.
*
* @see org.jboss.as.test.integration.security.common.AbstractSecurityDomainsServerSetupTask#getSecurityDomains()
*/
@Override
protected SecurityDomain[] getSecurityDomains() {
final SecurityDomain sd = new SecurityDomain.Builder().name("form-auth")
.loginModules(new SecurityModule.Builder().name("UsersRoles").build()).build();
return new SecurityDomain[]{sd};
}
}
}