From: Joseph Sutton <josephsutton@catalyst.net.nz>
Date: Wed, 28 Jul 2021 07:27:02 +0000 (+1200)
Subject: tests/krb5: Add generate_ap_req() method
X-Git-Tag: ldb-2.5.0~931
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4824dd4e9f40abcbd4134b79e2b2b8fb960f47e7;p=thirdparty%2Fsamba.git

tests/krb5: Add generate_ap_req() method

This method will be useful to generate an AP-REQ for use as FAST armor.

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>
---

diff --git a/python/samba/tests/krb5/raw_testcase.py b/python/samba/tests/krb5/raw_testcase.py
index 60e589464f3..67b359f07d8 100644
--- a/python/samba/tests/krb5/raw_testcase.py
+++ b/python/samba/tests/krb5/raw_testcase.py
@@ -1971,10 +1971,10 @@ class RawKerberosTest(TestCaseInTempDir):
         kdc_exchange_dict['preauth_etype_info2'] = etype_info2
         return
 
-    def generate_simple_tgs_padata(self,
-                                   kdc_exchange_dict,
-                                   callback_dict,
-                                   req_body):
+    def generate_ap_req(self,
+                        kdc_exchange_dict,
+                        _callback_dict,
+                        req_body):
         tgt = kdc_exchange_dict['tgt']
         authenticator_subkey = kdc_exchange_dict['authenticator_subkey']
         body_checksum_type = kdc_exchange_dict['body_checksum_type']
@@ -2014,6 +2014,16 @@ class RawKerberosTest(TestCaseInTempDir):
                                         ticket=tgt.ticket,
                                         authenticator=authenticator)
         ap_req = self.der_encode(ap_req_obj, asn1Spec=krb5_asn1.AP_REQ())
+
+        return ap_req
+
+    def generate_simple_tgs_padata(self,
+                                   kdc_exchange_dict,
+                                   callback_dict,
+                                   req_body):
+        ap_req = self.generate_ap_req(kdc_exchange_dict,
+                                      callback_dict,
+                                      req_body)
         pa_tgs_req = self.PA_DATA_create(PADATA_KDC_REQ, ap_req)
         padata = [pa_tgs_req]