|
|
package monkey
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"goqs/models"
|
|
|
"os/exec"
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
"github.com/spf13/cast"
|
|
|
)
|
|
|
|
|
|
// func runUninstallCmd(package_name string)error{
|
|
|
// cmd_content := "adb " +
|
|
|
// }
|
|
|
|
|
|
func runAndroidMonkeyCmd(task models.MonkeyTask) {
|
|
|
|
|
|
cmd_content := "cd /home/app/aimonkey && nohup python3 main_run.py" +
|
|
|
" -task_id " + cast.ToString(task.Id)
|
|
|
if task.PackageURL != "" {
|
|
|
pkg_path := "/home/tmp/pkg/" + cast.ToString(time.Now().Unix()) + ".apk"
|
|
|
log.Debug("正在下载apk：", task.PackageURL)
|
|
|
err := exec.Command("wget", task.PackageURL, "-O", pkg_path).Run()
|
|
|
if err != nil {
|
|
|
log.Error(err)
|
|
|
}
|
|
|
cmd_content += " -pkg_path " + pkg_path
|
|
|
log.Debug("下载完毕，安装包路径", pkg_path)
|
|
|
}
|
|
|
cmd_content += " -package " + task.PackageName +
|
|
|
" -activity " + task.LaunchActivity +
|
|
|
" -time " + cast.ToString(task.RunTime) +
|
|
|
" >> /home/app/logs/task_output/task_" + cast.ToString(task.Id) + ".log 2>&1"
|
|
|
|
|
|
log.Debug("开始调用monkey服务")
|
|
|
log.Debug(cmd_content)
|
|
|
cmd := exec.Command("bash", "-c", cmd_content)
|
|
|
err := cmd.Run()
|
|
|
if err != nil {
|
|
|
log.Error(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func GetAndroidDevices() map[string]string {
|
|
|
udid_map := make(map[string]string)
|
|
|
cmd := exec.Command("adb", "devices")
|
|
|
out, err := cmd.CombinedOutput()
|
|
|
if err != nil {
|
|
|
fmt.Println(err)
|
|
|
}
|
|
|
list := strings.Split(string(out), "\n")[1:]
|
|
|
for _, v := range list {
|
|
|
if len(v) < 1 {
|
|
|
continue
|
|
|
}
|
|
|
fmt.Println(v)
|
|
|
name := strings.Split(strings.Trim(strings.Trim(v, "\r"), "\t"), "\t")[0]
|
|
|
status := strings.Split(strings.Trim(strings.Trim(v, "\r"), "\t"), "\t")[1]
|
|
|
udid_map[name] = status
|
|
|
// devices = append(devices, getAndroidDeviceInfo(strings.Split(strings.Trim(strings.Trim(v, "\r"), "\t"), "\t")...))
|
|
|
}
|
|
|
return udid_map
|
|
|
}
|
|
|
|
|
|
type DeviceInfo struct {
|
|
|
Udid string `json:"string"`
|
|
|
Status string `json:"status"`
|
|
|
}
|
|
|
|
|
|
func getAndroidDeviceInfo(args ...string) models.MonkeyDevice {
|
|
|
var di models.MonkeyDevice
|
|
|
di.Platform = "adr"
|
|
|
di.Udid = args[0]
|
|
|
if args[1] == "device" {
|
|
|
di.Status = "online"
|
|
|
} else {
|
|
|
di.Status = args[1]
|
|
|
}
|
|
|
|
|
|
di.ProductName = getAndroidDeviceName(di.Udid)
|
|
|
return di
|
|
|
}
|
|
|
|
|
|
func getAndroidDeviceName(udid string) string {
|
|
|
command := `getprop` // | grep product.marketname'
|
|
|
cmd := exec.Command("adb", "-s", udid, "shell", command)
|
|
|
out, err := cmd.CombinedOutput()
|
|
|
if err != nil {
|
|
|
fmt.Println(err)
|
|
|
return udid
|
|
|
}
|
|
|
strs := strings.Split(string(out), "\n")
|
|
|
device_name := ""
|
|
|
for _, v := range strs {
|
|
|
if strings.Contains(v, "product.brand") {
|
|
|
str := strings.Split(strings.Trim(v, "\r"), ": ")[1]
|
|
|
device_name = str[1 : len(str)-1]
|
|
|
break
|
|
|
}
|
|
|
}
|
|
|
fmt.Println(device_name)
|
|
|
return device_name
|
|
|
}
|
|
|
|
|
|
func killPid(pid int) error {
|
|
|
cmd := exec.Command("kill", "-9", cast.ToString(pid))
|
|
|
err := cmd.Run()
|
|
|
return err
|
|
|
}
|
|
|
|
|
|
func checkTaskPids(pids []models.MonkeyPid) int {
|
|
|
if len(pids) == 0 {
|
|
|
return -1
|
|
|
}
|
|
|
var strs []string
|
|
|
for _, v := range pids {
|
|
|
strs = append(strs, cast.ToString(v.PId))
|
|
|
}
|
|
|
pids_str := strings.Join(strs, "|")
|
|
|
cmd_str := "ps -ef | grep -v -E 'bash|grep' | grep -E '" + pids_str + "'"
|
|
|
fmt.Println(cmd_str)
|
|
|
cmd := exec.Command("bash", "-c", cmd_str)
|
|
|
cmd.Run()
|
|
|
return cmd.ProcessState.ExitCode()
|
|
|
}
|