Added a QVH deactivation command that can be used in case device is not recognized by usbmuxd after recording (#87)

Co-authored-by: Ron Even <[email protected]>
Co-authored-by: danielpaulus <[email protected]>

Author: 3 people

main.go

@@ -27,6 +27,7 @@ func main() {
 Usage:
   qvh devices [-v]
   qvh activate [--udid=<udid>] [-v]
+  qvh deactivate [--udid=<udid>] [-v]
   qvh record <h264file> <wavfile> [--udid=<udid>] [-v]
   qvh audio <outfile> (--mp3 | --ogg | --wav) [--udid=<udid>] [-v]
   qvh gstreamer [--pipeline=<pipeline>] [--examples] [--udid=<udid>] [-v]
@@ -45,6 +46,8 @@ The commands work as following:
 	
 	activate	enables the video streaming config for the device specified by --udid
 
+	deactivate	disables the video streaming config for the device specified by --udid (in case it is stuck on streaming config)
+
 	record		will start video&audio recording. Video will be saved in a raw h264 file playable by VLC. 
 	            	Audio will be saved in a uncompressed wav file. Run like: "qvh record /home/yourname/out.h264 /home/yourname/out.wav"
 
@@ -95,6 +98,13 @@ The commands work as following:
 		activate(device)
 		return
 	}
+
+	deactivateCommand, _ := arguments.Bool("deactivate")
+	if deactivateCommand {
+		deactivate(device)
+		return
+	}
+
 	audioCommand, _ := arguments.Bool("audio")
 	if audioCommand {
 		outfile, err := arguments.String("<outfile>")
@@ -318,6 +328,20 @@ func activate(device screencapture.IosDevice) {
 	})
 }
 
+func deactivate(device screencapture.IosDevice) {
+        log.Debugf("Disabling device: %v", device)
+        var err error
+        device, err = screencapture.DisableQTConfig(device)
+        if err != nil {
+                printErrJSON(err, "Error disabling QT config")
+                return
+        }
+
+        printJSON(map[string]interface{}{
+                "device_activated": device.DetailsMap(),
+        })
+}
+
 func record(h264FilePath string, wavFilePath string, device screencapture.IosDevice) {
 	log.Debugf("Writing video output to:'%s' and audio to: %s", h264FilePath, wavFilePath)
 

screencapture/activator.go

@@ -53,6 +53,37 @@ func EnableQTConfig(device IosDevice) (IosDevice, error) {
 	return device, err
 }
 
+func DisableQTConfig(device IosDevice) (IosDevice, error) {
+        usbSerial := device.SerialNumber
+        ctx := gousb.NewContext()
+        usbDevice, err := OpenDevice(ctx, device)
+        if err != nil {
+                return IosDevice{}, err
+        }
+        if !isValidIosDeviceWithActiveQTConfig(usbDevice.Desc) {
+            log.Debugf("Skipping %s because it is already deactivated", usbSerial)
+            return device, nil
+        }
+
+        confignum, _ := usbDevice.ActiveConfigNum()
+        log.Debugf("Config is active: %d, QT config is: %d", confignum, device.QTConfigIndex)
+
+        for i := 0; i < 20; i++{
+            sendQTDisableConfigControlRequest(usbDevice)
+            log.Debugf("Resetting device config (#%d)", i + 1)
+            _, err := usbDevice.Config(device.UsbMuxConfigIndex)
+            if err != nil {
+                log.Warn(err)
+            }
+        }
+
+        confignum, _ = usbDevice.ActiveConfigNum()
+        log.Debugf("Config is active: %d, QT config is: %d", confignum, device.QTConfigIndex)
+
+
+        return device, err
+}
+
 func sendQTConfigControlRequest(device *gousb.Device) {
 	response := make([]byte, 0)
 	val, err := device.Control(0x40, 0x52, 0x00, 0x02, response)