python/samba/tests/ndr.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Unix SMB/CIFS implementation.
   4 # Copyright © Andrew Bartlett <abartlet@samba.org> 2021
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19
  20
  21 """Test of Network Data Representation (NDR) marshalling and unmarshalling."""
  22 import samba
  23 import samba.ndr as ndr
  24 from samba.dcerpc import winbind, security, lsa
  25
  26 class NdrTestCase(samba.tests.TestCase):
  27     def test_wbint_Principal(self):
  28         x = winbind.wbint_Principal()
  29
  30         x.sid = security.dom_sid(security.SID_NT_SCHANNEL_AUTHENTICATION)
  31
  32         x.type = lsa.SID_NAME_USER
  33
  34         x.name = "fred"
  35
  36         b = ndr.ndr_pack(x)
  37
  38         y = ndr.ndr_unpack(winbind.wbint_Principal, b)
  39
  40         self.assertEqual(x.sid, y.sid)
  41         self.assertEqual(x.type, y.type)
  42         self.assertEqual(x.name, y.name)
  43
  44     def test_wbint_Principal_null_name(self):
  45         x = winbind.wbint_Principal()
  46
  47         x.sid = security.dom_sid(security.SID_NT_SCHANNEL_AUTHENTICATION)
  48
  49         x.type = lsa.SID_NAME_USER
  50
  51         x.name = None
  52
  53         b = ndr.ndr_pack(x)
  54
  55         y = ndr.ndr_unpack(winbind.wbint_Principal, b)
  56
  57         self.assertEqual(x.sid, y.sid)
  58         self.assertEqual(x.type, y.type)
  59         self.assertEqual(x.name, y.name)
  60
  61     def test_wbint_Principals(self):
  62
  63         principals = []
  64
  65         for i in range(0, 10):
  66             x = winbind.wbint_Principal()
  67
  68             x.sid = security.dom_sid(security.SID_NT_SCHANNEL_AUTHENTICATION)
  69
  70             x.type = lsa.SID_NAME_USER
  71
  72             x.name = None
  73
  74             principals.append(x)
  75
  76         wb_principals = winbind.wbint_Principals()
  77         wb_principals.num_principals = 10
  78         wb_principals.principals = principals
  79
  80         b = ndr.ndr_pack(wb_principals)
  81
  82         unpacked_principals = ndr.ndr_unpack(winbind.wbint_Principals,
  83                                              b)
  84
  85         self.assertEqual(wb_principals.num_principals,
  86                          unpacked_principals.num_principals)
  87
  88         for i in range(0, 10):
  89             x = principals[i]
  90             y = unpacked_principals.principals[i]
  91             self.assertEqual(x.sid, y.sid)
  92             self.assertEqual(x.type, y.type)
  93             self.assertEqual(x.name, y.name)
  94
  95     def test_wbint_10_Principals(self):
  96         num = 10
  97         (principals, unpacked_principals) = self._test_wbint_Principals(num)
  98
  99         for i in range(0, num):
 100             x = principals[i]
 101             y = unpacked_principals.principals[i]
 102             self.assertEqual(x.sid, y.sid)
 103             self.assertEqual(x.type, y.type)
 104             self.assertEqual(x.name, y.name)
 105
 106     def test_wbint_max_token_Principals(self):
 107         self._test_wbint_Principals(samba._glue.ndr_token_max_list_size()+1)
 108
 109     def _test_wbint_Principals(self, num):
 110
 111         principals = []
 112         for i in range(0, num):
 113             x = winbind.wbint_Principal()
 114
 115             x.sid = security.dom_sid(security.SID_NT_SCHANNEL_AUTHENTICATION + "-%d" % num)
 116
 117             x.type = lsa.SID_NAME_USER
 118
 119             x.name = "fred%d" % num
 120
 121             principals.append(x)
 122
 123         wb_principals = winbind.wbint_Principals()
 124         wb_principals.num_principals = num
 125         wb_principals.principals = principals
 126
 127         b = ndr.ndr_pack(wb_principals)
 128
 129         try:
 130             unpacked_principals = ndr.ndr_unpack(winbind.wbint_Principals,
 131                                                  b)
 132         except RuntimeError as e:
 133             self.fail(e)
 134
 135         self.assertEqual(wb_principals.num_principals,
 136                          unpacked_principals.num_principals)
 137
 138         return (principals, unpacked_principals)