python/examples/samr.py

   1 #!/usr/bin/env python3
   2 # -*- coding: utf-8 -*-
   3
   4 # Unix SMB/CIFS implementation.
   5 # Copyright © Jelmer Vernooij <jelmer@samba.org> 2008
   6 #
   7 # Based on samr.js © Andrew Tridgell <tridge@samba.org>
   8 #
   9 # This program is free software; you can redistribute it and/or modify
  10 # it under the terms of the GNU General Public License as published by
  11 # the Free Software Foundation; either version 3 of the License, or
  12 # (at your option) any later version.
  13 #
  14 # This program is distributed in the hope that it will be useful,
  15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 # GNU General Public License for more details.
  18 #
  19 # You should have received a copy of the GNU General Public License
  20 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 #
  22
  23 import sys
  24
  25 sys.path.insert(0, "bin/python")
  26
  27 from samba.dcerpc import samr, security
  28
  29
  30 def display_lsa_string(str):
  31     return str.string
  32
  33
  34 def FillUserInfo(samr, dom_handle, users, level):
  35     """fill a user array with user information from samrQueryUserInfo"""
  36     for i in range(len(users)):
  37         user_handle = samr.OpenUser(handle, security.SEC_FLAG_MAXIMUM_ALLOWED, users[i].idx)
  38         info = samr.QueryUserInfo(user_handle, level)
  39         info.name = users[i].name
  40         info.idx  = users[i].idx
  41         users[i] = info
  42         samr.Close(user_handle)
  43
  44
  45 def toArray(handle, array, num_entries):
  46     ret = []
  47     for x in range(num_entries):
  48         ret.append((array.entries[x].idx, array.entries[x].name))
  49     return ret
  50
  51
  52 def test_Connect(samr):
  53     """test the samr_Connect interface"""
  54     print("Testing samr_Connect")
  55     return samr.Connect2(None, security.SEC_FLAG_MAXIMUM_ALLOWED)
  56
  57
  58 def test_LookupDomain(samr, handle, domain):
  59     """test the samr_LookupDomain interface"""
  60     print("Testing samr_LookupDomain")
  61     return samr.LookupDomain(handle, domain)
  62
  63
  64 def test_OpenDomain(samr, handle, sid):
  65     """test the samr_OpenDomain interface"""
  66     print("Testing samr_OpenDomain")
  67     return samr.OpenDomain(handle, security.SEC_FLAG_MAXIMUM_ALLOWED, sid)
  68
  69
  70 def test_EnumDomainUsers(samr, dom_handle):
  71     """test the samr_EnumDomainUsers interface"""
  72     print("Testing samr_EnumDomainUsers")
  73     users = toArray(*samr.EnumDomainUsers(dom_handle, 0, 0, 0xffffffff))
  74     print("Found %d users" % len(users))
  75     for idx, user in users:
  76         print("\t%s\t(%d)" % (user.string, idx))
  77
  78
  79 def test_EnumDomainGroups(samr, dom_handle):
  80     """test the samr_EnumDomainGroups interface"""
  81     print("Testing samr_EnumDomainGroups")
  82     groups = toArray(*samr.EnumDomainGroups(dom_handle, 0, 0))
  83     print("Found %d groups" % len(groups))
  84     for idx, group in groups:
  85         print("\t%s\t(%d)" % (group.string, idx))
  86
  87
  88 def test_domain_ops(samr, dom_handle):
  89     """test domain specific ops"""
  90     test_EnumDomainUsers(samr, dom_handle)
  91     test_EnumDomainGroups(samr, dom_handle)
  92
  93
  94 def test_EnumDomains(samr, handle):
  95     """test the samr_EnumDomains interface"""
  96     print("Testing samr_EnumDomains")
  97
  98     domains = toArray(*samr.EnumDomains(handle, 0, 0xffffffff))
  99     print("Found %d domains" % len(domains))
 100     for idx, domain in domains:
 101         print("\t%s (%d)" % (display_lsa_string(domain), idx))
 102     for idx, domain in domains:
 103         print("Testing domain %s" % display_lsa_string(domain))
 104         sid = samr.LookupDomain(handle, domain)
 105         dom_handle = test_OpenDomain(samr, handle, sid)
 106         test_domain_ops(samr, dom_handle)
 107         samr.Close(dom_handle)
 108
 109
 110 if len(sys.argv) != 2:
 111     print("Usage: samr.js <BINDING>")
 112     sys.exit(1)
 113
 114 binding = sys.argv[1]
 115
 116 print("Connecting to %s" % binding)
 117 try:
 118     samr = samr.samr(binding)
 119 except Exception as e:
 120     print("Failed to connect to %s: %s" % (binding, e.message))
 121     sys.exit(1)
 122
 123 handle = test_Connect(samr)
 124 test_EnumDomains(samr, handle)
 125 samr.Close(handle)
 126
 127 print("All OK")