import java.io.DataOutputStream
import java.lang.Exception
import java.lang.StringBuilder
object ShellUtils {
fun sh(cmd: String) = execCmd("sh", cmd)
fun su(cmd: String) = execCmd("su", cmd)
fun isRootAccess() = execCmd("su", "echo test", false).status == 0
@JvmOverloads
fun execCmd(
program: String,
input: String,
needResult: Boolean = true
): ExecResult {
var process: Process? = null
try {
process = Runtime.getRuntime().exec(program)
DataOutputStream(process.outputStream)
.use {
it.write(input.toByteArray())
it.writeBytes("\n")
it.flush()
it.writeBytes("exit\n")
it.flush()
if (!needResult) {
return NO_RESULT
}
val resultCode = process.waitFor()
val output = process.inputStream
.bufferedReader()
.useLines { lines ->
val results = StringBuilder()
lines.forEach { results.append(it) }
results.toString()
}
val error = process.errorStream
.bufferedReader()
.useLines { lines ->
val results = StringBuilder()
lines.forEach { results.append(it) }
results.toString()
}
return ExecResult(output, error, resultCode)
}
} catch (e: Exception) {
e.printStackTrace()
return ExecResult("", e.toString(), -1)
} finally {
process?.destroy()
}
}
val NO_RESULT=ExecResult("","",-1)
class ExecResult(val output: String?, val error: String?, val status: Int) {
override fun toString(): String {
val stringBuilder = StringBuilder()
stringBuilder.append("ret:").append(status).append('\n')
stringBuilder.append("out{\n").append(output).append("}\n")
stringBuilder.append("err{\n").append(error).append("}\n")
return stringBuilder.toString()
}
}
}
网友评论