Skip to content

Commit

Permalink
Add a 5 second timeout to server-side layout runs.
Browse files Browse the repository at this point in the history
Prevents configurations that produce infinite loops in ELK from eating
all the CPU.
  • Loading branch information
Eddykasp committed Apr 16, 2024
1 parent e63def6 commit cb7fca6
Showing 1 changed file with 46 additions and 18 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,16 @@
package de.cau.cs.kieler.elkgraph.web

import java.util.List
import java.util.concurrent.Callable
import java.util.concurrent.ExecutionException
import java.util.concurrent.Executors
import java.util.concurrent.TimeUnit
import java.util.concurrent.TimeoutException
import java.util.logging.Level
import java.util.logging.Logger
import org.eclipse.elk.core.IGraphLayoutEngine
import org.eclipse.elk.core.RecursiveGraphLayoutEngine
import org.eclipse.elk.core.UnsupportedConfigurationException
import org.eclipse.elk.core.options.CoreOptions
import org.eclipse.elk.core.util.BasicProgressMonitor
import org.eclipse.elk.graph.ElkEdge
Expand All @@ -29,8 +35,8 @@ import org.eclipse.sprotty.SModelElement
import org.eclipse.sprotty.SNode
import org.eclipse.sprotty.SPort
import org.eclipse.sprotty.xtext.IDiagramGenerator
import org.eclipse.sprotty.xtext.IDiagramGenerator.Context
import org.eclipse.xtend.lib.annotations.Accessors
import org.eclipse.elk.core.UnsupportedConfigurationException

/**
* Transforms ELK graphs into sprotty models to be transferred to clients.
Expand All @@ -57,25 +63,47 @@ class ElkGraphDiagramGenerator implements IDiagramGenerator {

val layoutVersion = context.state.options.get("layoutVersion")

val laidOutGraph = if (layoutVersion == "snapshot") {
layoutEngine.layout(elkGraph, new BasicProgressMonitor)
elkGraph
} else if (ElkLayoutVersionRegistry.versionToWrapper.containsKey(layoutVersion)) {
val wrapper = ElkLayoutVersionRegistry.versionToWrapper.get(layoutVersion)
val result = wrapper.layout(elkGraph)
if (!result.isPresent) {
throw new RuntimeException("Layout failed for version " + layoutVersion + ".")
}
result.get
} else {
throw new UnsupportedConfigurationException("Unknown layouter version: " + layoutVersion + ".")
val executor = Executors.newCachedThreadPool()
val layoutTask = new Callable<Object>() {
override call(){

val laidOutGraph = if (layoutVersion == "snapshot") {
layoutEngine.layout(elkGraph, new BasicProgressMonitor)
elkGraph
} else if (ElkLayoutVersionRegistry.versionToWrapper.containsKey(layoutVersion)) {
val wrapper = ElkLayoutVersionRegistry.versionToWrapper.get(layoutVersion)
val result = wrapper.layout(elkGraph)
if (!result.isPresent) {
throw new RuntimeException("Layout failed for version " + layoutVersion + ".")
}
result.get
} else {
throw new UnsupportedConfigurationException("Unknown layouter version: " + layoutVersion + ".")
}
return laidOutGraph
}
}
val timeoutInSeconds = 5

val sgraph = new SGraph
sgraph.type = 'graph'
sgraph.id = elkGraph.id
processContent(laidOutGraph, sgraph)
return sgraph
val future = executor.submit(layoutTask)
try {
val laidOutGraph = future.get(timeoutInSeconds, TimeUnit.SECONDS) as ElkNode
val sgraph = new SGraph
sgraph.type = 'graph'
sgraph.id = elkGraph.id
processContent(laidOutGraph, sgraph)
return sgraph
} catch (TimeoutException ex) {
throw new RuntimeException("Layout timed out after " + timeoutInSeconds + " seconds.")
} catch (InterruptedException ex) {
throw new RuntimeException(ex.message)
} catch (ExecutionException ex) {
throw new RuntimeException(ex.message)
} finally {
LOG.info("Layout timed out.")
future.cancel(true)
}

} catch (RuntimeException exc) {
LOG.log(Level.SEVERE, "Failed to generate ELK graph.", exc)
return showError(exc)
Expand Down

0 comments on commit cb7fca6

Please sign in to comment.