001/* 002 * Copyright (C) 2012 eXo Platform SAS. 003 * 004 * This is free software; you can redistribute it and/or modify it 005 * under the terms of the GNU Lesser General Public License as 006 * published by the Free Software Foundation; either version 2.1 of 007 * the License, or (at your option) any later version. 008 * 009 * This software is distributed in the hope that it will be useful, 010 * but WITHOUT ANY WARRANTY; without even the implied warranty of 011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 012 * Lesser General Public License for more details. 013 * 014 * You should have received a copy of the GNU Lesser General Public 015 * License along with this software; if not, write to the Free 016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 018 */ 019package org.crsh.shell.impl.command; 020 021import org.crsh.console.KeyHandler; 022import org.crsh.shell.ShellProcess; 023import org.crsh.shell.ShellProcessContext; 024import org.crsh.shell.ShellResponse; 025 026import java.util.logging.Level; 027 028abstract class CRaSHProcess implements ShellProcess { 029 030 /** . */ 031 protected final CRaSHSession crash; 032 033 /** . */ 034 protected final String request; 035 036 /** . */ 037 private volatile Thread thread; 038 039 /** . */ 040 private volatile boolean cancelled; 041 042 protected CRaSHProcess(CRaSHSession crash, String request) { 043 this.crash = crash; 044 this.request = request; 045 } 046 047 public void execute(ShellProcessContext processContext) { 048 ClassLoader previous = crash.setCRaSHLoader(); 049 try { 050 ShellResponse resp; 051 thread = Thread.currentThread(); 052 053 // 054 String userName = crash.user != null ? crash.user.getName() : "unauthenticated"; 055 CRaSHSession.accessLog.log(Level.FINE, "User " + userName + " executes " + request); 056 057 // 058 try { 059 try { 060 resp = doInvoke(processContext); 061 if (Thread.interrupted() || cancelled) { 062 throw new InterruptedException("like a goto"); 063 } 064 } 065 catch (InterruptedException e) { 066 Thread.currentThread().interrupt(); 067 resp = ShellResponse.cancelled(); 068 } 069 } catch (Throwable t) { 070 resp = ShellResponse.internalError("Unexpected error when executing process", t); 071 } finally { 072 thread = null; 073 } 074 075 // 076 processContext.end(resp); 077 078 // 079 if (resp instanceof ShellResponse.Error) { 080 ShellResponse.Error error = (ShellResponse.Error)resp; 081 Throwable t = error.getThrowable(); 082 if (t != null) { 083 CRaSHSession.log.log(Level.SEVERE, "Error while evaluating request '" + request + "' " + error.getMessage(), t); 084 } else { 085 CRaSHSession.log.log(Level.SEVERE, "Error while evaluating request '" + request + "' " + error.getMessage()); 086 } 087 } 088 } 089 finally { 090 crash.setPreviousLoader(previous); 091 } 092 } 093 094 @Override 095 public KeyHandler getKeyHandler() { 096 return null; 097 } 098 099 abstract ShellResponse doInvoke(ShellProcessContext context) throws InterruptedException; 100 101 public void cancel() { 102 ClassLoader previous = crash.setCRaSHLoader(); 103 try { 104 Thread t = thread; 105 if (t != null) { 106 t.interrupt(); 107 } 108 cancelled = true; 109 } 110 finally { 111 crash.setPreviousLoader(previous); 112 } 113 } 114}