This commit is contained in:
Max Lv
2016-03-07 17:59:15 +08:00
parent 9e09e0329b
commit 9965a71c93
4 changed files with 38 additions and 42 deletions
@@ -1,147 +0,0 @@
/*
* Shadowsocks - A shadowsocks client for Android
* Copyright (C) 2015 <max.c.lv@gmail.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*
* ___====-_ _-====___
* _--^^^#####// \\#####^^^--_
* _-^##########// ( ) \\##########^-_
* -############// |\^^/| \\############-
* _/############// (@::@) \\############\_
* /#############(( \\// ))#############\
* -###############\\ (oo) //###############-
* -#################\\ / VV \ //#################-
* -###################\\/ \//###################-
* _#/|##########/\######( /\ )######/\##########|\#_
* |/ |#/\#/\#/\/ \#/\##\ | | /##/\#/ \/\#/\#/\#| \|
* ` |/ V V ` V \#\| | | |/#/ V ' V V \| '
* ` ` ` ` / | | | | \ ' ' ' '
* ( | | | | )
* __\ | | | | /__
* (vvv(VVV)(VVV)vvv)
*
* HERE BE DRAGONS
*
*/
package com.github.shadowsocks
import android.util.Log
import java.io.IOException
import java.io.InputStream
import java.io.OutputStream
import java.util.concurrent.CountDownLatch
import java.util.concurrent.atomic.AtomicReference
import collection.JavaConversions._
object GuardedProcess {
private val TAG: String = classOf[GuardedProcess].getSimpleName
@throws(classOf[IOException])
private def startProcess(cmd: Seq[String]): Process = {
new ProcessBuilder(cmd).redirectErrorStream(true).start
}
}
class GuardedProcess extends Process {
private var guardThread: Thread = null
@volatile
private var isDestroyed: Boolean = false
@volatile
private var process: Process = null
@throws(classOf[InterruptedException])
@throws(classOf[IOException])
def this(cmd: Seq[String], onRestartCallback: Runnable = null) {
this()
initThread(cmd, onRestartCallback)
}
def initThread(cmd: Seq[String], onRestartCallback: Runnable): Unit = {
val atomicIoException: AtomicReference[IOException] = new AtomicReference[IOException](null)
val countDownLatch: CountDownLatch = new CountDownLatch(1)
guardThread = new Thread(new Runnable() {
override def run(): Unit = {
try {
while (!isDestroyed) {
Log.i(GuardedProcess.TAG, "start process: " + cmd)
val startTime: Long = java.lang.System.currentTimeMillis
process = GuardedProcess.startProcess(cmd)
if (onRestartCallback != null && countDownLatch.getCount <= 0) {
onRestartCallback.run()
}
countDownLatch.countDown()
process.waitFor
if (java.lang.System.currentTimeMillis - startTime < 1000) {
Log.w(GuardedProcess.TAG, "process exit too fast, stop guard: " + cmd)
return
}
}
}
catch {
case ignored: InterruptedException =>
Log.i(GuardedProcess.TAG, "thread interrupt, destroy process: " + cmd)
process.destroy()
case e: IOException =>
atomicIoException.compareAndSet(null, e)
} finally {
countDownLatch.countDown()
}
}
}, "GuardThread-" + cmd)
guardThread.start()
countDownLatch.await()
val ioException: IOException = atomicIoException.get
if (ioException != null) {
throw ioException
}
}
def destroy() {
isDestroyed = true
guardThread.interrupt()
process.destroy()
try {
guardThread.join()
}
catch {
case ignored: InterruptedException =>
}
}
def exitValue: Int = {
throw new UnsupportedOperationException
}
def getErrorStream: InputStream = {
throw new UnsupportedOperationException
}
def getInputStream: InputStream = {
throw new UnsupportedOperationException
}
def getOutputStream: OutputStream = {
throw new UnsupportedOperationException
}
@throws(classOf[InterruptedException])
def waitFor: Int = {
guardThread.join()
0
}
}