blob: fedd0d20bf679de3e447d58e826b794f7437ef33 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
package org.pacien.tincapp.commands
import android.os.AsyncTask
import java8.util.concurrent.CompletableFuture
import java8.util.function.Supplier
import java.io.BufferedReader
import java.io.IOException
import java.io.InputStream
import java.io.InputStreamReader
/**
* @author pacien
*/
internal object Executor {
private const val FAILED = -1
private const val SUCCESS = 0
class CommandExecutionException(msg: String) : Exception(msg)
init {
System.loadLibrary("exec")
}
/**
* @return FAILED (-1) on error, forked child PID otherwise
*/
private external fun forkExec(argcv: Array<String>): Int
/**
* @return FAILED (-1) on error, the exit status of the process otherwise
*/
private external fun wait(pid: Int): Int
private fun read(stream: InputStream) = BufferedReader(InputStreamReader(stream)).readLines()
fun forkExec(cmd: Command): CompletableFuture<Void> {
val pid = forkExec(cmd.asArray()).also {
if (it == FAILED) throw CommandExecutionException("Could not fork child process.")
}
return CompletableFuture.runAsync {
when (wait(pid)) {
SUCCESS -> Unit
FAILED -> throw CommandExecutionException("Process terminated abnormally.")
else -> throw CommandExecutionException("Non-zero exit status code.")
}
}
}
fun call(cmd: Command): CompletableFuture<List<String>> {
val proc = try {
ProcessBuilder(cmd.asList()).start()
} catch (e: IOException) {
throw CommandExecutionException(e.message ?: "Could not start process.")
}
return supplyAsyncTask<List<String>> {
if (proc.waitFor() == 0) read(proc.inputStream)
else throw CommandExecutionException(read(proc.errorStream).lastOrNull() ?: "Non-zero exit status.")
}
}
fun runAsyncTask(r: () -> Unit) = CompletableFuture.runAsync(Runnable(r), AsyncTask.THREAD_POOL_EXECUTOR)!!
fun <U> supplyAsyncTask(s: () -> U) = CompletableFuture.supplyAsync(Supplier(s), AsyncTask.THREAD_POOL_EXECUTOR)!!
}
|