1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#!/usr/bin/env python3
# Unix SMB/CIFS implementation.
# Copyright (C) Stefan Metzmacher 2020
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import sys
import os
sys.path.insert(0, 'bin/python')
os.environ['PYTHONUNBUFFERED'] = '1'
from samba.dcerpc import krb5pac
from samba.ndr import ndr_unpack
from samba.tests import DynamicTestCase
from samba.tests.krb5.kdc_base_test import KDCBaseTest
global_asn1_print = False
global_hexdump = False
@DynamicTestCase
class PacAlignTests(KDCBaseTest):
base_name = 'krbpac'
@classmethod
def setUpDynamicTestCases(cls):
for length in range(len(cls.base_name), 21):
cls.generate_dynamic_test('test_pac_align',
f'{length}_chars',
length)
def setUp(self):
super().setUp()
self.do_asn1_print = global_asn1_print
self.do_hexdump = global_hexdump
def _test_pac_align_with_args(self, length):
samdb = self.get_samdb()
account_name = self.base_name + 'a' * (length - len(self.base_name))
creds, _ = self.create_account(samdb, account_name)
tgt = self.get_tgt(creds, expect_pac=True)
pac_data = self.get_ticket_pac(tgt)
self.assertIsNotNone(pac_data)
self.assertEqual(0, len(pac_data) & 7)
pac = ndr_unpack(krb5pac.PAC_DATA_RAW, pac_data)
for pac_buffer in pac.buffers:
buffer_type = pac_buffer.type
buffer_size = pac_buffer.ndr_size
with self.subTest(buffer_type=buffer_type):
if buffer_type == krb5pac.PAC_TYPE_LOGON_NAME:
self.assertEqual(length * 2 + 10, buffer_size)
elif buffer_type == krb5pac.PAC_TYPE_REQUESTER_SID:
self.assertEqual(28, buffer_size)
elif buffer_type in {krb5pac.PAC_TYPE_SRV_CHECKSUM,
krb5pac.PAC_TYPE_KDC_CHECKSUM,
krb5pac.PAC_TYPE_TICKET_CHECKSUM}:
self.assertEqual(0, buffer_size & 3,
f'buffer type was: {buffer_type}, '
f'buffer size was: {buffer_size}')
else:
self.assertEqual(0, buffer_size & 7,
f'buffer type was: {buffer_type}, '
f'buffer size was: {buffer_size}')
rounded_len = (buffer_size + 7) & ~7
self.assertEqual(rounded_len, len(pac_buffer.info.remaining))
if __name__ == '__main__':
global_asn1_print = False
global_hexdump = False
import unittest
unittest.main()
|
