X-Git-Url: https://git.argeo.org/?a=blobdiff_plain;f=org.argeo.cms%2Fsrc%2Forg%2Fargeo%2Fcms%2Finternal%2Fkernel%2FKernel.java;h=1002304007ff3e64d42a86834a8de8a64fa95f6a;hb=070656b6431e699a0f6ac8dc1839bce3fc2509ce;hp=c6a1630ae51c88db1032c4af72923e029346d7ed;hpb=63446804f4954bfedd50d8c692bde0fab13aa1ec;p=lgpl%2Fargeo-commons.git diff --git a/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java b/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java index c6a1630ae..100230400 100644 --- a/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java +++ b/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java @@ -1,14 +1,57 @@ package org.argeo.cms.internal.kernel; +import static bitronix.tm.TransactionManagerServices.getTransactionManager; +import static bitronix.tm.TransactionManagerServices.getTransactionSynchronizationRegistry; +import static java.util.Locale.ENGLISH; +import static org.argeo.cms.internal.kernel.KernelUtils.getFrameworkProp; +import static org.argeo.cms.internal.kernel.KernelUtils.getOsgiInstanceDir; +import static org.argeo.jcr.ArgeoJcrConstants.ALIAS_NODE; +import static org.argeo.jcr.ArgeoJcrConstants.JCR_REPOSITORY_ALIAS; +import static org.argeo.util.LocaleChoice.asLocaleList; +import static org.osgi.framework.Constants.FRAMEWORK_UUID; + +import java.io.File; +import java.io.FileFilter; +import java.io.IOException; +import java.lang.management.ManagementFactory; +import java.security.PrivilegedAction; +import java.util.HashMap; +import java.util.Hashtable; +import java.util.List; +import java.util.Locale; +import java.util.Map; + +import javax.jcr.Repository; import javax.jcr.RepositoryFactory; +import javax.security.auth.Subject; +import javax.transaction.TransactionManager; +import javax.transaction.TransactionSynchronizationRegistry; +import javax.transaction.UserTransaction; +import org.apache.commons.io.FileUtils; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; +import org.apache.jackrabbit.util.TransientFileFactory; import org.argeo.ArgeoException; +import org.argeo.ArgeoLogger; +import org.argeo.cms.CmsException; import org.argeo.jackrabbit.OsgiJackrabbitRepositoryFactory; -import org.argeo.security.core.InternalAuthentication; +import org.argeo.jcr.ArgeoJcrConstants; +import org.eclipse.equinox.http.jetty.JettyConfigurator; +import org.eclipse.equinox.http.jetty.JettyConstants; +import org.eclipse.equinox.http.servlet.ExtendedHttpService; import org.osgi.framework.BundleContext; -import org.springframework.security.core.context.SecurityContextHolder; +import org.osgi.framework.ServiceEvent; +import org.osgi.framework.ServiceListener; +import org.osgi.framework.ServiceReference; +import org.osgi.framework.ServiceRegistration; +import org.osgi.service.cm.Configuration; +import org.osgi.service.cm.ConfigurationAdmin; +import org.osgi.service.useradmin.UserAdmin; + +import bitronix.tm.BitronixTransactionManager; +import bitronix.tm.BitronixTransactionSynchronizationRegistry; +import bitronix.tm.TransactionManagerServices; /** * Argeo CMS Kernel. Responsible for : @@ -21,58 +64,415 @@ import org.springframework.security.core.context.SecurityContextHolder; *
  • OS access
  • * */ -final class Kernel { +final class Kernel implements KernelHeader, KernelConstants, ServiceListener { + /* + * SERVICE REFERENCES + */ + private ServiceReference configurationAdmin; + /* + * REGISTERED SERVICES + */ + private ServiceRegistration loggerReg; + private ServiceRegistration tmReg; + private ServiceRegistration utReg; + private ServiceRegistration tsrReg; + private ServiceRegistration repositoryReg; + private ServiceRegistration repositoryFactoryReg; + private ServiceRegistration userAdminReg; + + /* + * SERVICES IMPLEMENTATIONS + */ + private NodeLogger logger; + private BitronixTransactionManager transactionManager; + private BitronixTransactionSynchronizationRegistry transactionSynchronizationRegistry; + private OsgiJackrabbitRepositoryFactory repositoryFactory; + NodeRepository repository; + private NodeUserAdmin userAdmin; + + // Members private final static Log log = LogFactory.getLog(Kernel.class); + ThreadGroup threadGroup = new ThreadGroup(Kernel.class.getSimpleName()); + private final BundleContext bc = Activator.getBundleContext(); + private final NodeSecurity nodeSecurity; + private DataHttp dataHttp; + private KernelThread kernelThread; - private final BundleContext bundleContext; + private Locale defaultLocale = null; + private List locales = null; - private JackrabbitNode node; - private OsgiJackrabbitRepositoryFactory repositoryFactory; - private NodeSecurity nodeSecurity; - private NodeHttpFilter httpFilter; + public Kernel() { + nodeSecurity = new NodeSecurity(); + // log.debug(bc.getDataFile("")); + // log.debug(bc.getDataFile("test")); + } - Kernel(BundleContext bundleContext) { - this.bundleContext = bundleContext; + final void init() { + Subject.doAs(nodeSecurity.getKernelSubject(), + new PrivilegedAction() { + @Override + public Void run() { + doInit(); + return null; + } + }); } - void init() { + private void doInit() { long begin = System.currentTimeMillis(); - InternalAuthentication initAuth = new InternalAuthentication( - KernelConstants.DEFAULT_SECURITY_KEY); - SecurityContextHolder.getContext().setAuthentication(initAuth); - + ConfigurationAdmin conf = findConfigurationAdmin(); + // Use CMS bundle classloader + ClassLoader currentContextCl = Thread.currentThread() + .getContextClassLoader(); + Thread.currentThread().setContextClassLoader( + Kernel.class.getClassLoader()); try { - node = new JackrabbitNode(bundleContext); + if (nodeSecurity.isFirstInit()) + firstInit(); + + defaultLocale = new Locale(getFrameworkProp(I18N_DEFAULT_LOCALE, + ENGLISH.getLanguage())); + locales = asLocaleList(getFrameworkProp(I18N_LOCALES)); + logger = new NodeLogger(); + + // Initialise services + initTransactionManager(); + repository = new NodeRepository(); repositoryFactory = new OsgiJackrabbitRepositoryFactory(); - nodeSecurity = new NodeSecurity(bundleContext, node); - httpFilter = new NodeHttpFilter(bundleContext, nodeSecurity); - - // Publish services to OSGi register - nodeSecurity.publish(); - node.publish(); - bundleContext.registerService(RepositoryFactory.class, - repositoryFactory, null); - httpFilter.publish(); + userAdmin = new NodeUserAdmin(transactionManager, repository); + + // HTTP + initWebServer(conf); + ServiceReference sr = bc + .getServiceReference(ExtendedHttpService.class); + if (sr != null) + addHttpService(sr); + + // Kernel thread + kernelThread = new KernelThread(this); + kernelThread.setContextClassLoader(Kernel.class.getClassLoader()); + kernelThread.start(); + + // Publish services to OSGi + publish(); } catch (Exception e) { log.error("Cannot initialize Argeo CMS", e); throw new ArgeoException("Cannot initialize", e); + } finally { + Thread.currentThread().setContextClassLoader(currentContextCl); } - long duration = System.currentTimeMillis() - begin; - log.info("## ARGEO CMS UP in " + (duration / 1000) + "." - + (duration % 1000) + "s ##"); + long jvmUptime = ManagementFactory.getRuntimeMXBean().getUptime(); + log.info("## ARGEO CMS UP in " + (jvmUptime / 1000) + "." + + (jvmUptime % 1000) + "s ##"); + long initDuration = System.currentTimeMillis() - begin; + if (log.isTraceEnabled()) + log.trace("Kernel initialization took " + initDuration + "ms"); + directorsCut(initDuration); + } + + private void firstInit() { + log.info("## FIRST INIT ##"); + String nodeInit = getFrameworkProp(NODE_INIT); + if (nodeInit == null) + nodeInit = "../../init"; + File initDir; + if (nodeInit.startsWith(".")) + initDir = KernelUtils.getExecutionDir(nodeInit); + else + initDir = new File(nodeInit); + // TODO also uncompress archives + if (initDir.exists()) + try { + FileUtils.copyDirectory(initDir, getOsgiInstanceDir(), + new FileFilter() { + + @Override + public boolean accept(File pathname) { + if (pathname.getName().equals(".svn") + || pathname.getName().equals(".git")) + return false; + return true; + } + }); + log.info("CMS initialized from " + initDir.getCanonicalPath()); + } catch (IOException e) { + throw new CmsException("Cannot initialize from " + initDir, e); + } + } + + /** Can be null */ + private ConfigurationAdmin findConfigurationAdmin() { + configurationAdmin = bc.getServiceReference(ConfigurationAdmin.class); + if (configurationAdmin == null) { + return null; + } + return bc.getService(configurationAdmin); + } + + private void initTransactionManager() { + bitronix.tm.Configuration tmConf = TransactionManagerServices + .getConfiguration(); + tmConf.setServerId(getFrameworkProp(FRAMEWORK_UUID)); + + // File tmBaseDir = new File(getFrameworkProp(TRANSACTIONS_HOME, + // getOsgiInstancePath(DIR_TRANSACTIONS))); + File tmBaseDir = bc.getDataFile(DIR_TRANSACTIONS); + File tmDir1 = new File(tmBaseDir, "btm1"); + tmDir1.mkdirs(); + tmConf.setLogPart1Filename(new File(tmDir1, tmDir1.getName() + ".tlog") + .getAbsolutePath()); + File tmDir2 = new File(tmBaseDir, "btm2"); + tmDir2.mkdirs(); + tmConf.setLogPart2Filename(new File(tmDir2, tmDir2.getName() + ".tlog") + .getAbsolutePath()); + transactionManager = getTransactionManager(); + transactionSynchronizationRegistry = getTransactionSynchronizationRegistry(); + } + + private void initWebServer(ConfigurationAdmin conf) { + String httpPort = getFrameworkProp("org.osgi.service.http.port"); + String httpsPort = getFrameworkProp("org.osgi.service.http.port.secure"); + try { + if (httpPort != null || httpsPort != null) { + Hashtable jettyProps = new Hashtable(); + if (httpPort != null) { + jettyProps.put(JettyConstants.HTTP_PORT, httpPort); + jettyProps.put(JettyConstants.HTTP_ENABLED, true); + } + if (httpsPort != null) { + jettyProps.put(JettyConstants.HTTPS_PORT, httpsPort); + jettyProps.put(JettyConstants.HTTPS_ENABLED, true); + jettyProps.put(JettyConstants.SSL_KEYSTORETYPE, "PKCS12"); + jettyProps.put(JettyConstants.SSL_KEYSTORE, nodeSecurity + .getHttpServerKeyStore().getCanonicalPath()); + jettyProps.put(JettyConstants.SSL_PASSWORD, "changeit"); + jettyProps.put(JettyConstants.SSL_WANTCLIENTAUTH, true); + } + if (conf != null) { + // TODO make filter more generic + String filter = "(" + JettyConstants.HTTP_PORT + "=" + + httpPort + ")"; + if (conf.listConfigurations(filter) != null) + return; + Configuration jettyConf = conf.createFactoryConfiguration( + JETTY_FACTORY_PID, null); + jettyConf.update(jettyProps); + } else { + JettyConfigurator.startServer("default", jettyProps); + } + } + } catch (Exception e) { + throw new CmsException("Cannot initialize web server on " + + httpPortsMsg(httpPort, httpsPort), e); + } + } + + private void publish() { + // Listen to service publication (also ours) + bc.addServiceListener(Kernel.this); + + // Logging + loggerReg = bc.registerService(ArgeoLogger.class, logger, null); + // Transaction + tmReg = bc.registerService(TransactionManager.class, + transactionManager, null); + utReg = bc.registerService(UserTransaction.class, transactionManager, + null); + tsrReg = bc.registerService(TransactionSynchronizationRegistry.class, + transactionSynchronizationRegistry, null); + // User admin + userAdminReg = bc.registerService(UserAdmin.class, userAdmin, + userAdmin.currentState()); + // JCR + Hashtable regProps = new Hashtable(); + regProps.put(JCR_REPOSITORY_ALIAS, ALIAS_NODE); + repositoryReg = bc.registerService(Repository.class, repository, + regProps); + repositoryFactoryReg = bc.registerService(RepositoryFactory.class, + repositoryFactory, null); } void destroy() { long begin = System.currentTimeMillis(); + unpublish(); - httpFilter = null; - nodeSecurity.destroy(); - node.destroy(); + kernelThread.destroyAndJoin(); + + if (dataHttp != null) + dataHttp.destroy(); + if (userAdmin != null) + userAdmin.destroy(); + if (repository != null) + repository.destroy(); + if (transactionManager != null) + transactionManager.shutdown(); + bc.removeServiceListener(this); + + // Clean hanging threads from Jackrabbit + TransientFileFactory.shutdown(); + + // Clean hanging Gogo shell thread + new GogoShellKiller().start(); + + nodeSecurity.destroy(); long duration = System.currentTimeMillis() - begin; log.info("## ARGEO CMS DOWN in " + (duration / 1000) + "." + (duration % 1000) + "s ##"); } -} + private void unpublish() { + userAdminReg.unregister(); + repositoryFactoryReg.unregister(); + repositoryReg.unregister(); + tmReg.unregister(); + utReg.unregister(); + tsrReg.unregister(); + loggerReg.unregister(); + } + + @Override + public void serviceChanged(ServiceEvent event) { + ServiceReference sr = event.getServiceReference(); + Object service = bc.getService(sr); + if (service instanceof Repository) { + Object jcrRepoAlias = sr + .getProperty(ArgeoJcrConstants.JCR_REPOSITORY_ALIAS); + if (jcrRepoAlias != null) {// JCR repository + String alias = jcrRepoAlias.toString(); + Repository repository = (Repository) bc.getService(sr); + Map props = new HashMap(); + for (String key : sr.getPropertyKeys()) + props.put(key, sr.getProperty(key)); + if (ServiceEvent.REGISTERED == event.getType()) { + try { + repositoryFactory.register(repository, props); + dataHttp.registerRepositoryServlets(alias, repository); + } catch (Exception e) { + throw new CmsException( + "Could not publish JCR repository " + alias, e); + } + } else if (ServiceEvent.UNREGISTERING == event.getType()) { + repositoryFactory.unregister(repository, props); + dataHttp.unregisterRepositoryServlets(alias); + } + } + } else if (service instanceof ExtendedHttpService) { + if (ServiceEvent.REGISTERED == event.getType()) { + addHttpService(sr); + } else if (ServiceEvent.UNREGISTERING == event.getType()) { + dataHttp.destroy(); + dataHttp = null; + } + } + } + + private void addHttpService(ServiceReference sr) { + // for (String key : sr.getPropertyKeys()) + // log.debug(key + "=" + sr.getProperty(key)); + ExtendedHttpService httpService = (ExtendedHttpService) bc + .getService(sr); + // TODO find constants + Object httpPort = sr.getProperty("http.port"); + Object httpsPort = sr.getProperty("https.port"); + dataHttp = new DataHttp(httpService, repository); + if (log.isDebugEnabled()) + log.debug(httpPortsMsg(httpPort, httpsPort)); + } + + private String httpPortsMsg(Object httpPort, Object httpsPort) { + return "HTTP " + httpPort + + (httpsPort != null ? " - HTTPS " + httpsPort : ""); + } + + @Override + public Locale getDefaultLocale() { + return defaultLocale; + } + + /** Can be null. */ + @Override + public List getLocales() { + return locales; + } + + final private static void directorsCut(long initDuration) { + // final long ms = 128l + (long) (Math.random() * 128d); + long ms = initDuration / 100; + log.info("Spend " + ms + "ms" + + " reflecting on the progress brought to mankind" + + " by Free Software..."); + long beginNano = System.nanoTime(); + try { + Thread.sleep(ms, 0); + } catch (InterruptedException e) { + // silent + } + long durationNano = System.nanoTime() - beginNano; + final double M = 1000d * 1000d; + double sleepAccuracy = ((double) durationNano) / (ms * M); + if (log.isDebugEnabled()) + log.debug("Sleep accuracy: " + + String.format("%.2f", 100 - (sleepAccuracy * 100 - 100)) + + " %"); + } + + /** Workaround for blocking Gogo shell by system shutdown. */ + private class GogoShellKiller extends Thread { + + public GogoShellKiller() { + super("Gogo shell killer"); + setDaemon(true); + } + + @Override + public void run() { + ThreadGroup rootTg = getRootThreadGroup(null); + Thread gogoShellThread = findGogoShellThread(rootTg); + if (gogoShellThread == null) + return; + while (getNonDaemonCount(rootTg) > 2) { + try { + Thread.sleep(100); + } catch (InterruptedException e) { + // silent + } + } + gogoShellThread = findGogoShellThread(rootTg); + if (gogoShellThread == null) + return; + System.exit(0); + } + } + + private static ThreadGroup getRootThreadGroup(ThreadGroup tg) { + if (tg == null) + tg = Thread.currentThread().getThreadGroup(); + if (tg.getParent() == null) + return tg; + else + return getRootThreadGroup(tg.getParent()); + } + + private static int getNonDaemonCount(ThreadGroup rootThreadGroup) { + Thread[] threads = new Thread[rootThreadGroup.activeCount()]; + rootThreadGroup.enumerate(threads); + int nonDameonCount = 0; + for (Thread t : threads) + if (!t.isDaemon()) + nonDameonCount++; + return nonDameonCount; + } + + private static Thread findGogoShellThread(ThreadGroup rootThreadGroup) { + Thread[] threads = new Thread[rootThreadGroup.activeCount()]; + rootThreadGroup.enumerate(threads, true); + for (Thread thread : threads) { + if (thread.getName().equals("Gogo shell")) + return thread; + } + return null; + } + +} \ No newline at end of file