tss: Update AP keys

doronz88 · Feb 25, 2025 · 1baa12e · 1baa12e
1 parent ecd4e5b
commit 1baa12e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pymobiledevice3/restore/base_restore.py b/pymobiledevice3/restore/base_restore.py
@@ -94,6 +94,7 @@ def populate_tss_request_from_manifest(self, parameters: dict, additional_keys:
                          'Baobab,ChipID', 'Baobab,ManifestEpoch', 'Baobab,SecurityDomain', 'eUICC,ChipID',
                          'PearlCertificationRootPub', 'Timer,BoardID,1', 'Timer,BoardID,2', 'Timer,ChipID,1',
                          'Timer,ChipID,2', 'Timer,SecurityDomain,1', 'Timer,SecurityDomain,2', 'Manifest',
+                         'NeRDEpoch',
                          ]
         else:
             key_list += additional_keys

diff --git a/pymobiledevice3/restore/tss.py b/pymobiledevice3/restore/tss.py
@@ -286,7 +286,7 @@ def add_ap_tags(self, parameters: dict, overrides=None):
         manifest_node = parameters['Manifest']
 
         # add components to request
-        skipped_keys = ('BasebandFirmware', 'SE,UpdatePayload', 'BaseSystem', 'Diags',)
+        skipped_keys = ('BasebandFirmware', 'SE,UpdatePayload', 'BaseSystem', 'Diags', 'Ap,ExclaveOS')
         for key, manifest_entry in manifest_node.items():
             if key in skipped_keys:
                 continue
@@ -348,7 +348,8 @@ def add_ap_img4_tags(self, parameters):
         keys_to_copy = (
             'ApNonce', 'ApProductionMode', 'ApSecurityMode', 'Ap,OSLongVersion', 'ApSecurityMode', 'ApSepNonce',
             'Ap,SDKPlatform', 'PearlCertificationRootPub', 'NeRDEpoch', 'ApSikaFuse', 'Ap,SikaFuse', 'Ap,OSReleaseType',
-            'Ap,ProductType', 'Ap,Target', 'Ap,TargetType'
+            'Ap,ProductType', 'Ap,Target', 'Ap,TargetType', 'AllowNeRDBoot', 'Ap,ProductMarketingVersion',
+            'Ap,Timestamp',
         )
         for k in keys_to_copy:
             if k in parameters:
@@ -361,6 +362,9 @@ def add_ap_img4_tags(self, parameters):
 
         uid_mode = parameters.get('UID_MODE', False)
 
+        if 'NeRDEpoch' in parameters:
+            self._request['PermitNeRDPivot'] = b''
+
         self._request['UID_MODE'] = uid_mode
         self._request['@ApImg4Ticket'] = True
         self._request['@BBTicket'] = True