Jak zablokować dostęp do niektórych klas podczas wykonywania groovy skryptów z java?

Question

Jestem całkiem nowy dla groovy i skryptów w java ogólnie, i naprawdę mam nadzieję, że istnieje proste rozwiązanie mojego problemu. W naszej aplikacji użytkownicy mogą uruchamiać groovy skrypty, które piszą samodzielnie, i musimy kontrolować, co te skrypty mogą i nie mogą zrobić. Dużo czytam na temat piaskownicy, ale albo patrzę na niewłaściwych miejsc, albo przeoczyłem oczywistość. Aby było prostsze, mam mały przykład demonstrujący problem. To moja klasa ładowarka, która powinna zapobiec java.lang.System od bycia załadowany i dostępne dla skryptów:

public class MyClassLoader extends ClassLoader { 

    @Override 
    public Class<?> loadClass(String name) throws ClassNotFoundException { 
     if (name.startsWith("java.lang.System")) { 
      throw new ClassNotFoundException("Class not found: " + name); 
     } 
     return super.loadClass(name); 
    } 
} 

A to prosty program, który próbuje wywołać System.currentTimeMillis():

public static void main(String[] args) { 
    String code = "java.lang.System.currentTimeMillis();"; 
    ClassLoader classLoader = new MyClassLoader(); 
    Thread.currentThread().setContextClassLoader(classLoader); 

    GroovyShell shell = new GroovyShell(); 
    Script script = shell.parse(code); 
    Object result = script.run(); 
    log.debug(result); 
} 

MyClassLoader generuje wyjątki dla java.lang.SystemBeanInfo i java.lang.SystemCustomizer, ale kod zostanie wykonany. samo dzieje się, jeśli mogę użyć javax.script klasy:

ScriptEngineManager factory = new ScriptEngineManager(); 
ScriptEngine engine = factory.getEngineByName("Groovy"); 
Object o = engine.eval(code); 
log.debug(o); 

A jeśli próbuję go z silnikiem JavaScriptu, to działa zgodnie z oczekiwaniami (wystarczy wymienić „Groovy” z „Javascript” w powyższym przykładzie).

Czy ktoś może mi w tym pomóc? BTW, używam groovy-all-1.8.8.jar, z jdk1.7.0_55.

Dzięki

Answer 1

Mogę w tym celu polecić Groovy Sandbox. W przeciwieństwie do SecureASTCustomizer sprawdza, czy wykonanie jest dozwolone dynamicznie w czasie wykonywania. Przechwytuje wszystkie wywołania metod, przydziały obiektów, dostęp do właściwości/atrybutów, dostęp do macierzy itd. - a zatem masz bardzo drobną kontrolę nad tym, co pozwalasz (białe listy).

Oczywiście konfiguracja dozwolona jest bardzo ważna. Na przykład możesz zezwolić na używanie String s i użyć metod takich jak substring, ale prawdopodobnie nie na execute na String, które mogą być wykorzystane z czymś takim jak 'rm -R ~/*'.execute(). Tworzenie konfiguracji, która jest naprawdę bezpieczna, jest wyzwaniem i jest trudniejsze, im więcej pozwalasz.

Wadą Groovy Sandbox jest to, że kod musi działać z zarejestrowanym przechwytywaczem, a podczas wykonywania zostanie nałożona kara za wydajność.

Ten obraz [1] pokazuje przykład z projektu, w którym wykorzystaliśmy Groovy Sandbox dla kodu Groovy wprowadzonego przez użytkownika.Kod jest uruchamiany na valide skrypt - więc jeśli oświadczenie nie będzie faktycznie zrealizowanej w ramach niego, wniosek byłby wyszedł zanim mogłem zrobić zrzut ekranu;)

Answer 2

Może byłbyś zainteresowany innymi SecureASTCustomizer w połączeniu z CompilerConfiguration. Jeśli chodzi o bezpieczeństwo, wyraźna biała lista może być lepsza niż czarna.

def s = new SecureASTCustomizer() 
s.importsWhiteList = [ 'a.legal.Klass', 'other.legal.Klass' ] 

def c = new CompilerConfiguration() 
c.addCompilationCustomizers(s) 

def sh = new GroovyShell(c) 

Spójrz na tę klasę, zawiera wiele opcji, które są gotowe do użycia.

Answer 3

import groovy.lang.GroovyClassLoader; 
import groovy.lang.GroovyShell; 
import groovy.lang.Script; 

public class SandboxGroovyClassLoader extends ClassLoader { 

public SandboxGroovyClassLoader(ClassLoader parent) { 
    super(parent); 
} 

@Override 
public Class<?> loadClass(String name) throws ClassNotFoundException { 
    if (name.startsWith("java.lang.System")) 
     return null; 
    return super.loadClass(name); 
} 

@Override 
protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException { 
    if (name.startsWith("java.lang.System")) 
     return null; 
    return super.loadClass(name, resolve); 
} 

static void runWithGroovyClassLoader() throws Exception { 
    System.out.println("Begin runWithGroovyClassLoader"); 

    String code = "def hello_world() { java.lang.System.currentTimeMillis(); };"; 

    GroovyClassLoader groovyClassLoader = new GroovyClassLoader(); 
    Class<?> scriptClass = groovyClassLoader.parseClass(code); 
    Object scriptInstance = scriptClass.newInstance(); 
    Object result = scriptClass.getDeclaredMethod("hello_world", new Class[] {}).invoke(scriptInstance, new Object[] {}); 
    System.out.println(result); 
    groovyClassLoader.close(); 
    System.out.println("End runWithGroovyClassLoader"); 
} 

static void runWithSandboxGroovyClassLoader() throws Exception { 
    System.out.println("Begin runWithSandboxGroovyClassLoader"); 
    ClassLoader parentClassLoader = SandboxGroovyClassLoader.class.getClassLoader(); 
    SandboxGroovyClassLoader classLoader = new SandboxGroovyClassLoader(parentClassLoader); 

    String code = "def hello_world() { java.lang.System.currentTimeMillis(); };"; 

    GroovyClassLoader groovyClassLoader = new GroovyClassLoader(classLoader); 
    Class<?> scriptClass = groovyClassLoader.parseClass(code); 
    Object scriptInstance = scriptClass.newInstance(); 
    Object result = scriptClass.getDeclaredMethod("hello_world", new Class[] {}).invoke(scriptInstance, new Object[] {}); 
    System.out.println(result); 
    groovyClassLoader.close(); 
    System.out.println("End runWithSandboxGroovyClassLoader"); 
} 

static void runWithSandboxGroovyShellClassLoader() throws Exception { 
    System.out.println("Begin runWithSandboxGroovyShellClassLoader"); 

    String code = "java.lang.System.currentTimeMillis();"; 

    ClassLoader parentClassLoader = SandboxGroovyClassLoader.class.getClassLoader(); 
    SandboxGroovyClassLoader classLoader = new SandboxGroovyClassLoader(parentClassLoader); 
    Thread.currentThread().setContextClassLoader(classLoader); 

    GroovyShell shell = new GroovyShell(); 
    Script script = shell.parse(code); 
    Object result = script.run(); 
    System.out.println(result); 
    System.out.println("End runWithSandboxGroovyShellClassLoader"); 
} 

public static void main(String[] args) throws Exception { 

    runWithGroovyClassLoader(); 
    runWithSandboxGroovyClassLoader(); 
    runWithSandboxGroovyShellClassLoader(); 

} 
} 

Czy go czego chcesz ?