refactor: move gidevice to hrp pkg

2026-09-04 23:16:57 +08:00 · 2022-10-23 22:59:34 +08:00
parent 896f39c3f2
commit 91626bcdd4
85 changed files with 10476 additions and 65 deletions
@@ -0,0 +1,142 @@
+package gidevice
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/httprunner/httprunner/v4/hrp/pkg/gidevice/pkg/libimobiledevice"
+)
+
+var _ ImageMounter = (*imageMounter)(nil)
+
+func newImageMounter(client *libimobiledevice.ImageMounterClient) *imageMounter {
+	return &imageMounter{
+		client: client,
+	}
+}
+
+type imageMounter struct {
+	client *libimobiledevice.ImageMounterClient
+}
+
+func (m *imageMounter) Images(imgType string) (imageSignatures [][]byte, err error) {
+	var pkt libimobiledevice.Packet
+	if pkt, err = m.client.NewXmlPacket(
+		m.client.NewBasicRequest(libimobiledevice.CommandTypeLookupImage, imgType),
+	); err != nil {
+		return nil, err
+	}
+
+	if err = m.client.SendPacket(pkt); err != nil {
+		return nil, err
+	}
+
+	var respPkt libimobiledevice.Packet
+	if respPkt, err = m.client.ReceivePacket(); err != nil {
+		return nil, err
+	}
+
+	var reply libimobiledevice.ImageMounterLookupImageResponse
+	if err = respPkt.Unmarshal(&reply); err != nil {
+		return nil, err
+	}
+
+	imageSignatures = reply.ImageSignature
+	return
+}
+
+func (m *imageMounter) UploadImage(imgType, dmgPath string, signatureData []byte) (err error) {
+	var dmgFileInfo os.FileInfo
+	if dmgFileInfo, err = os.Stat(dmgPath); err != nil {
+		return err
+	}
+
+	var pkt libimobiledevice.Packet
+	if pkt, err = m.client.NewXmlPacket(
+		m.client.NewReceiveBytesRequest(imgType, uint32(dmgFileInfo.Size()), signatureData),
+	); err != nil {
+		return err
+	}
+
+	if err = m.client.SendPacket(pkt); err != nil {
+		return err
+	}
+
+	var respPkt libimobiledevice.Packet
+	if respPkt, err = m.client.ReceivePacket(); err != nil {
+		return err
+	}
+
+	var reply libimobiledevice.ImageMounterBasicResponse
+	if err = respPkt.Unmarshal(&reply); err != nil {
+		return err
+	}
+
+	if reply.Status != "ReceiveBytesAck" {
+		return fmt.Errorf("image mounter 'ReceiveBytes' status: %s", reply.Status)
+	}
+
+	var dmgData []byte
+	if dmgData, err = os.ReadFile(dmgPath); err != nil {
+		return err
+	}
+
+	if err = m.client.SendDmg(dmgData); err != nil {
+		return err
+	}
+	if respPkt, err = m.client.ReceivePacket(); err != nil {
+		return err
+	}
+
+	if err = respPkt.Unmarshal(&reply); err != nil {
+		return err
+	}
+
+	if reply.Status != "Complete" {
+		return fmt.Errorf("image mounter 'SendDmg' status: %s", reply.Status)
+	}
+
+	return
+}
+
+func (m *imageMounter) Mount(imgType, devImgPath string, signatureData []byte) (err error) {
+	var pkt libimobiledevice.Packet
+	if pkt, err = m.client.NewXmlPacket(
+		m.client.NewMountImageRequest(imgType, devImgPath, signatureData),
+	); err != nil {
+		return err
+	}
+
+	if err = m.client.SendPacket(pkt); err != nil {
+		return err
+	}
+
+	var respPkt libimobiledevice.Packet
+	if respPkt, err = m.client.ReceivePacket(); err != nil {
+		return err
+	}
+
+	var reply libimobiledevice.ImageMounterBasicResponse
+	if err = respPkt.Unmarshal(&reply); err != nil {
+		return err
+	}
+
+	if reply.Status != "Complete" {
+		return fmt.Errorf("image mounter 'MountImage' status: %s", reply.Status)
+	}
+	return
+}
+
+func (m *imageMounter) UploadImageAndMount(imgType, devImgPath, dmgPath, signaturePath string) (err error) {
+	var signatureData []byte
+	if signatureData, err = os.ReadFile(signaturePath); err != nil {
+		return err
+	}
+	if err = m.UploadImage(imgType, dmgPath, signatureData); err != nil {
+		return err
+	}
+	if err = m.Mount(imgType, devImgPath, signatureData); err != nil {
+		return err
+	}
+	return
+}