source4/scripting/swig/torture/winreg.py

   1 #!/usr/bin/python
   2
   3 import sys, dcerpc
   4
   5 def test_OpenHKLM(pipe):
   6
   7     r = {}
   8     r['unknown'] = {}
   9     r['unknown']['unknown0'] = 0x9038
  10     r['unknown']['unknown1'] = 0x0000
  11     r['access_required'] = 0x02000000
  12
  13     result = dcerpc.winreg_OpenHKLM(pipe, r)
  14
  15     return result['handle']
  16
  17 def test_QueryInfoKey(pipe, handle):
  18
  19     r = {}
  20     r['handle'] = handle
  21     r['class'] = {}
  22     r['class']['name'] = None
  23
  24     return dcerpc.winreg_QueryInfoKey(pipe, r)
  25
  26 def test_CloseKey(pipe, handle):
  27
  28     r = {}
  29     r['handle'] = handle
  30
  31     dcerpc.winreg_CloseKey(pipe, r)
  32
  33 def test_Enum(pipe, handle, depth = 0):
  34
  35     if depth > 2:
  36         return
  37
  38     keyinfo = test_QueryInfoKey(pipe, handle)
  39
  40     # Enumerate keys
  41
  42     r = {}
  43     r['handle'] = handle
  44     r['key_name_len'] = 0
  45     r['unknown'] = 0x0414
  46     r['in_name'] = {}
  47     r['in_name']['unknown'] = 0x20a
  48     r['in_name']['key_name'] = {}
  49     r['in_name']['key_name']['name'] = None
  50     r['class'] = {}
  51     r['class']['name'] = None
  52     r['last_changed_time'] = {}
  53     r['last_changed_time']['low'] = 0
  54     r['last_changed_time']['high'] = 0
  55
  56     for i in range(0, keyinfo['num_subkeys']):
  57
  58         r['enum_index'] = i
  59
  60         subkey = dcerpc.winreg_EnumKey(pipe, r)
  61
  62         s = {}
  63         s['handle'] = handle
  64         s['keyname'] = {}
  65         s['keyname']['name'] = subkey['out_name']['name']
  66         s['unknown'] = 0
  67         s['access_mask'] = 0x02000000
  68
  69         result = dcerpc.winreg_OpenKey(pipe, s)
  70
  71         test_Enum(pipe, result['handle'], depth + 1)
  72
  73         test_CloseKey(pipe, result['handle'])
  74
  75     # Enumerate values
  76
  77     r = {}
  78     r['handle'] = handle
  79
  80     keyinfo['max_valnamelen'] = 18
  81     keyinfo['max_valbufsize'] = 0x31f5
  82
  83     r['foo'] = {}
  84     r['foo']['len'] = 0
  85     r['foo']['max_len'] = keyinfo['max_valnamelen'] * 2
  86     r['foo']['buffer'] = {}
  87     r['foo']['buffer']['max_len'] = keyinfo['max_valnamelen']
  88     r['foo']['buffer']['offset'] = 0
  89     r['foo']['buffer']['len'] = 0
  90     r['foo']['buffer']['buffer'] = ''
  91     r['type'] = 0
  92     r['value'] = {}
  93     r['value']['max_len'] = keyinfo['max_valbufsize']
  94     r['value']['offset'] = 0
  95     r['value']['len'] = 0
  96     r['value']['buffer'] = []
  97     r['returned_len'] = 0
  98     r['foo2'] = {}
  99     r['foo2']['max_len'] = keyinfo['max_valbufsize']
 100     r['foo2']['offset'] = 0
 101     r['foo2']['len'] = 0
 102     r['foo2']['buffer'] = ''
 103     r['value1'] = keyinfo['max_valbufsize']
 104     r['value2'] = 0
 105
 106     for i in range(0, keyinfo['num_values']):
 107
 108         r['enum_index'] = i
 109
 110         print keyinfo
 111         print dcerpc.winreg_EnumValue(pipe, r)
 112
 113         sys.exit(1)
 114
 115 def test_Key(pipe, handle):
 116
 117     test_Enum(pipe, handle)
 118
 119 def runtests(binding, domain, username, password):
 120
 121     print 'Testing WINREG pipe'
 122
 123     pipe = dcerpc.pipe_connect(binding,
 124             dcerpc.DCERPC_WINREG_UUID, dcerpc.DCERPC_WINREG_VERSION,
 125             domain, username, password)
 126
 127     handle = test_OpenHKLM(pipe)
 128
 129     test_Key(pipe, handle)
 130