Skip to content

Commit

Permalink
ISSUE-576: Allow complex ADB server commands execution (#597)
Browse files Browse the repository at this point in the history
* ISSUE-576: Allow complex ADB server commands execution

* ISSUE-576: Introduce new constructors instead of separate types; change complex commands syntax

* ISSUE-576: Update docs

* ISSUE-576: Update adbserver-desktop.jar

* ISSUE-576: Use single method

* ISSUE-576: Deprecate methods

* ISSUE-576: Update docs

* ISSUE-576: Add test

* ISSUE-576: Update docs; fix perform adb

* ISSUE-576: minor test refactoring

* ISSUE-576: lint

* ISSUE-576: remove test log from adb-server binary; fix test on android 21
  • Loading branch information
Nikitae57 authored Jan 22, 2024
1 parent 02074cd commit de6c16a
Show file tree
Hide file tree
Showing 17 changed files with 349 additions and 92 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,7 @@ package com.kaspersky.adbserver.commandtypes

import com.kaspersky.adbserver.common.api.Command

data class AdbCommand(override val body: String) : Command(body)
data class AdbCommand(
override val command: String,
override val arguments: List<String> = emptyList()
) : Command(command, arguments)
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,7 @@ package com.kaspersky.adbserver.commandtypes

import com.kaspersky.adbserver.common.api.Command

data class CmdCommand(override val body: String) : Command(body)
data class CmdCommand(
override val command: String,
override val arguments: List<String> = emptyList()
) : Command(command, arguments)
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,4 @@ import java.io.Serializable
/**
* Command to execute by AdbServer
*/
abstract class Command(open val body: String) : Serializable
abstract class Command(open val command: String, open val arguments: List<String> = emptyList()) : Serializable
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
package com.kaspersky.adbserver.device

import com.kaspersky.adbserver.common.api.CommandResult
import com.kaspersky.adbserver.commandtypes.AdbCommand
import com.kaspersky.adbserver.commandtypes.CmdCommand
import com.kaspersky.adbserver.common.api.CommandResult
import com.kaspersky.adbserver.common.log.LoggerFactory
import com.kaspersky.adbserver.common.log.logger.LogLevel
import com.kaspersky.adbserver.common.log.logger.Logger
Expand All @@ -27,10 +27,32 @@ object AdbTerminal {
AdbCommand(command)
) ?: throw IllegalStateException("Please first of all call [connect] method to establish a connection")

/**
* Allows more control over how arguments are parsed. Each element in the [arguments] list
* is used as is without tokenizing.
* Refer to the https://docs.oracle.com/javase/7/docs/api/java/lang/Runtime.html#exec(java.lang.String[])
*
* Please first of all call [connect] method to establish a connection
*/
fun executeAdb(command: String, arguments: List<String>): CommandResult = device?.fulfill(
AdbCommand(command, arguments)
) ?: throw IllegalStateException("Please first of all call [connect] method to establish a connection")

/**
* Please first of all call [connect] method to establish a connection
*/
fun executeCmd(command: String): CommandResult = device?.fulfill(
CmdCommand(command)
) ?: throw IllegalStateException("Please first of all call [connect] method to establish a connection")

/**
* Allows more control over how arguments are parsed. Each element in the [arguments] list
* is used as is without tokenizing.
* Refer to the https://docs.oracle.com/javase/7/docs/api/java/lang/Runtime.html#exec(java.lang.String[])
*
* Please first of all call [connect] method to establish a connection
*/
fun executeCmd(command: String, arguments: List<String>): CommandResult = device?.fulfill(
CmdCommand(command, arguments)
) ?: throw IllegalStateException("Please first of all call [connect] method to establish a connection")
}
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,12 @@ package com.kaspersky.adbserver.desktop

import com.kaspersky.adbserver.common.api.CommandResult
import com.kaspersky.adbserver.common.api.ExecutorResultStatus
import com.kaspersky.adbserver.common.log.logger.Logger
import java.util.concurrent.TimeUnit

internal class CmdCommandPerformer(
private val desktopName: String
private val desktopName: String,
private val logger: Logger,
) {

companion object {
Expand All @@ -15,9 +17,24 @@ internal class CmdCommandPerformer(
/**
* Be aware it's a synchronous method
*/
fun perform(command: String): CommandResult {
fun perform(command: String, arguments: List<String> = emptyList()): CommandResult {
val serviceInfo = "The command was executed on desktop=$desktopName"
val process = Runtime.getRuntime().exec(command)

val process = try {
if (arguments.isEmpty()) {
Runtime.getRuntime().exec(command)
} else {
val fullCommand = buildList {
add(command)
addAll(arguments)
}.toTypedArray()
Runtime.getRuntime().exec(fullCommand)
}
} catch (ex: Throwable) {
logger.e(ex.stackTraceToString())
return CommandResult(ExecutorResultStatus.FAILURE, "failed to start process. See exception in AdbServer logs")
}

try {
if (process.waitFor(EXECUTION_TIMEOUT_SECONDS, TimeUnit.SECONDS)) {
val exitCode = process.exitValue()
Expand All @@ -43,7 +60,7 @@ internal class CmdCommandPerformer(
serviceInfo = serviceInfo
)
} finally {
process.destroy()
process?.destroy()
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,32 @@ internal class CommandExecutorImpl(

override fun execute(command: Command): CommandResult {
return when (command) {
is CmdCommand -> cmdCommandPerformer.perform(command.body)
is CmdCommand -> cmdCommandPerformer.perform(command.command, command.arguments)

is AdbCommand -> {
val adbCommand = "$adbPath ${ adbServerPort?.let { "-P $adbServerPort " } ?: "" }-s $deviceName ${command.body}"
logger.d("The created adbCommand=$adbCommand")
cmdCommandPerformer.perform(adbCommand)
val adbCommand: String
val adbArguments: List<String>

if (command.arguments.isEmpty()) {
adbCommand = "$adbPath ${adbServerPort?.let { "-P $adbServerPort " } ?: ""}-s $deviceName ${command.command}"
adbArguments = emptyList()
} else {
adbCommand = adbPath
adbArguments = buildList {
adbServerPort?.let {
add("-P")
add(adbServerPort)
}
add("-s")
add(deviceName)
add(command.command)
addAll(command.arguments)
}
}
logger.d("The created adbCommand=$adbCommand, arguments=$adbArguments")
cmdCommandPerformer.perform(adbCommand, adbArguments)
}

else -> throw UnsupportedOperationException("The command=$command is unsupported command")
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ internal fun main(args: Array<String>) {

desktopLogger.i("Desktop started with arguments: emulators=$emulators, adbServerPort=$port, adbPath=$adbPath")

val cmdCommandPerformer = CmdCommandPerformer(desktopName)
val cmdCommandPerformer = CmdCommandPerformer(desktopName, desktopLogger)
val desktop = Desktop(
cmdCommandPerformer = cmdCommandPerformer,
presetEmulators = emulators,
Expand Down
Binary file modified artifacts/adbserver-desktop.jar
Binary file not shown.
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import org.jetbrains.kotlin.gradle.tasks.KotlinCompile

plugins {
id("com.android.application")
id("convention.kotlin-base")
Expand All @@ -6,6 +8,11 @@ plugins {
}

android {
compileOptions {
targetCompatibility = JavaVersion.VERSION_17
sourceCompatibility = JavaVersion.VERSION_17
}

testBuildType = "debug"

defaultConfig {
Expand Down Expand Up @@ -33,3 +40,9 @@ android {
jvmToolchain(JavaVersion.VERSION_17.majorVersion.toInt())
}
}

tasks.withType<KotlinCompile>().configureEach {
kotlinOptions {
jvmTarget = JavaVersion.VERSION_17.toString()
}
}
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import org.jetbrains.kotlin.gradle.tasks.KotlinCompile

plugins {
id("com.android.library")
id("convention.kotlin-base")
Expand All @@ -17,4 +19,15 @@ android {
ignore = true
}
}

compileOptions {
targetCompatibility = JavaVersion.VERSION_17
sourceCompatibility = JavaVersion.VERSION_17
}
}

tasks.withType<KotlinCompile>().configureEach {
kotlinOptions {
jvmTarget = JavaVersion.VERSION_17.toString()
}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
import org.gradle.kotlin.dsl.application

plugins {
id("kotlin")
id("convention.kotlin-base")
Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@

import org.jetbrains.kotlin.gradle.tasks.KotlinCompile

val kotlinLanguageVersion = "1.9"

tasks.withType<KotlinCompile>().configureEach {
kotlinOptions {

jvmTarget = JavaVersion.VERSION_17.toString()
jvmTarget = JavaVersion.VERSION_11.toString()

// TODO: fix all warnings before
// allWarningsAsErrors = true
Expand All @@ -18,3 +19,8 @@ tasks.withType<KotlinCompile>().configureEach {
"-progressive"
}
}

tasks.withType<JavaCompile>().configureEach {
sourceCompatibility = JavaVersion.VERSION_11.toString()
targetCompatibility = JavaVersion.VERSION_11.toString()
}
21 changes: 21 additions & 0 deletions docs/Wiki/Executing_adb_commands.en.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,10 @@ You can notice that the app uses `AdbTerminal` class to execute ADB commands.
## Usage in Kaspresso
In Kaspresso, we wrap `AdbTerminal` into a special interface `AdbServer`.
`AdbServer`'s instance is available in `BaseTestContext` scope and `BaseTestCase` with `adbServer` property: <br>
There're two types of AbdServer methods signatures:

`fun perform(vararg commands: String): List<String>` and `perform(command: String, arguments: List<String>): String` with an important difference.
First signature accept one or more commands and execute them one by one. Example below:
```kotlin
@Test
fun test() =
Expand All @@ -50,6 +54,23 @@ fun test() =
// ....
}
```
This method passes each command to the java runtime which tokenizes it by whitespaces. It could be not ideal. It can't be used for the commands with
the whitespaces in their arguments (e.g. `adb pull "/sdcard/Documents/path_with whitespace to/file.txt"`) and it doesn't allow piping (e.g. `cat $BUILD_DIR/report.txt | grep filte > filtered.txt`).
That's why there's a second AdbServer methods signature type.

It executes a single command and allows you to pass a list of arguments. It doesn't tokenize your command or arguments, they are used "as is".
This allows you to use piping and complex arguments
See example below (yes, a command could be an argument for another command):
```kotlin
@Test
fun test() = before{
adbServer.performCmd("bash", listOf("-c", "adb shell dumpsys deviceidle | grep mForceIdle"))
}.after {
}.run {
// ...
}
```

Also, don't forget to grant necessary permission:
```
<uses-permission android:name="android.permission.INTERNET" />
Expand Down
21 changes: 21 additions & 0 deletions docs/Wiki/Executing_adb_commands.ru.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,10 @@
## Использование в Kaspresso
В Kaspresso мы оборачиваем AdbTerminal в специальный интерфейс AdbServer.
Экземпляр `AdbServer` доступен в области `BaseTestContext` и `BaseTestCase` со свойством `adbServer`: <br>
У AdbServer два типа сигнатур методов:

`fun perform(vararg commands: String): List<String>` и `perform(command: String, arguments: List<String>): String` с важными различиями между ними.
Первая сигнатура принимает одну или более комманду и исполняет их одну за одной. Пример далее:
```kotlin
@Test
fun test() =
Expand All @@ -50,6 +54,23 @@ fun test() =
// ....
}
```
Этот метод передает каждую команду среде выполнения java, которая разбивает ее по пробелам. Это может быть не самым лучшим вариантом. Он не может использоваться для команд с
пробелами в их аргументах (например, `adb pull "/sdcard/Documents/path_with whitespace to/file.txt"`) и не допускает пайпинга комманд (например, `cat $BUILD_DIR/report.txt | grep filte > filtered.txt`).
Вот почему существует второй тип сигнатур методов AdbServer.

Он выполняет одну команду и позволяет вам передавать список аргументов. Он не разбивает вашу команду или аргументы, они используются в том виде, в
котором вы их передали. Это позволяет вам использовать пайпинг и сложные аргументы.
Смотрите пример ниже (да, команда может быть аргументом для другой команды):
```kotlin
@Test
fun test() = before{
adbServer.performCmd("bash", listOf("-c", "adb shell dumpsys deviceidle | grep mForceIdle"))
}.after {
}.run {
// ...
}
```

Также не забудьте предоставить необходимое разрешение:
```
<uses-permission android:name="android.permission.INTERNET" />
Expand Down
Loading

0 comments on commit de6c16a

Please sign in to comment.