summaryrefslogtreecommitdiff
path: root/source4/scripting/swig/torture/winreg.py
blob: b7a42042a8a9aca102fe1472b59f07cfded1dd20 (plain)
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#!/usr/bin/python

import sys, dcerpc

def test_OpenHKLM(pipe):

    r = {}
    r['unknown'] = {}
    r['unknown']['unknown0'] = 0x9038
    r['unknown']['unknown1'] = 0x0000
    r['access_required'] = 0x02000000

    result = dcerpc.winreg_OpenHKLM(pipe, r)

    return result['handle']

def test_QueryInfoKey(pipe, handle):

    r = {}
    r['handle'] = handle
    r['class'] = {}
    r['class']['name'] = None

    return dcerpc.winreg_QueryInfoKey(pipe, r)

def test_CloseKey(pipe, handle):

    r = {}
    r['handle'] = handle

    dcerpc.winreg_CloseKey(pipe, r)

def test_Enum(pipe, handle, name, depth = 0):

    if depth > 2:
        return

    try:
        keyinfo = test_QueryInfoKey(pipe, handle)
    except dcerpc.WERROR, arg:
        if arg[0] == dcerpc.WERR_ACCESS_DENIED:
            return

    # Enumerate keys

    r = {}
    r['handle'] = handle
    r['key_name_len'] = 0
    r['unknown'] = 0x0414
    r['in_name'] = {}
    r['in_name']['unknown'] = 0x20a
    r['in_name']['key_name'] = {}
    r['in_name']['key_name']['name'] = None
    r['class'] = {}
    r['class']['name'] = None
    r['last_changed_time'] = {}
    r['last_changed_time']['low'] = 0
    r['last_changed_time']['high'] = 0

    for i in range(0, keyinfo['num_subkeys']):

        r['enum_index'] = i

        subkey = dcerpc.winreg_EnumKey(pipe, r)

        s = {}
        s['handle'] = handle
        s['keyname'] = {}
        s['keyname']['name'] = subkey['out_name']['name']
        s['unknown'] = 0
        s['access_mask'] = 0x02000000

        result = dcerpc.winreg_OpenKey(pipe, s)

        test_Enum(pipe, result['handle'], name + '/' + s['keyname']['name'],
                  depth + 1)

        test_CloseKey(pipe, result['handle'])

    # Enumerate values

    r = {}
    r['handle'] = handle

    r['name_in'] = {}
    r['name_in']['len'] = 0
    r['name_in']['max_len'] = (keyinfo['max_valnamelen'] + 1) * 2
    r['name_in']['buffer'] = {}
    r['name_in']['buffer']['max_len'] = keyinfo['max_valnamelen']  + 1
    r['name_in']['buffer']['offset'] = 0
    r['name_in']['buffer']['len'] = 0
    r['type'] = 0
    r['value_in'] = {}
    r['value_in']['max_len'] = keyinfo['max_valbufsize']
    r['value_in']['offset'] = 0
    r['value_in']['len'] = 0
    r['value_len1'] = keyinfo['max_valbufsize']
    r['value_len2'] = 0
    
    for i in range(0, keyinfo['num_values']):

        r['enum_index'] = i

        dcerpc.winreg_EnumValue(pipe, r)

def test_Key(pipe, handle, name):

    test_Enum(pipe, handle, name)

def runtests(binding, domain, username, password):
    
    print 'Testing WINREG pipe'

    pipe = dcerpc.pipe_connect(binding,
            dcerpc.DCERPC_WINREG_UUID, dcerpc.DCERPC_WINREG_VERSION,
            domain, username, password)

    handle = test_OpenHKLM(pipe)

    test_Key(pipe, handle, 'HKLM')