summaryrefslogtreecommitdiff
path: root/src/third_party/wiredtiger/tools/gdb/wt_debug_script_update.py
blob: 13a04c18ef94d1b400878b696b775fc763f63494 (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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
import gdb
import bson
from pprint import pprint

# Usage examples:
# coll_dh = get_data_handle(conn, 'file:collection-5550--7194480883124807592.wt')
# dump_handle(coll_dh)
# index_dh = get_data_handle(conn, 'file:index-5558--7194480883124807592.wt')
# dump_handle(index_dh)

def dbg(ident, var):
    print('----------')
    if type(var) == gdb.Value:
        print('{}: ({}*){}'.format(ident, var.type, var.address))
    else:
        print(ident)
    print('  ' + str(type(var)))
    methods = dir(var)
    out = [name for name in methods if not name.startswith("__")]
    for item in out:
        print('    ' + item)

    if type(var) == gdb.Value:
        print('\n  Fields:')
        print('\t' + '\n\t'.join(str(var).split('\n')))

conn_impl_ptr = gdb.lookup_type("WT_CONNECTION_IMPL").pointer()
dbg('impl', conn_impl_ptr)

conn = gdb.parse_and_eval("session->iface->connection").reinterpret_cast(conn_impl_ptr).dereference()
dbg('conn', conn)

def walk_wt_list(lst):
    ret = []
    node = conn['dhqh']['tqh_first']
    #dbg('node', node)
    while True:
        if not node:
            break
        ret.append(node.dereference())
        node = node['q']['tqe_next']

    return ret

def get_data_handle(conn, handle_name):
    #dbg('datahandles', conn['dhqh'])
    ret = None
    for handle in walk_wt_list(conn['dhqh']):
        #print("Handle: " + str(handle['name']))
        if handle['name'].string() == handle_name:
            ret = handle

    return ret

def get_btree_handle(dhandle):
    btree = gdb.lookup_type('WT_BTREE').pointer()
    return dhandle['handle'].reinterpret_cast(btree).dereference()

def dump_update_chain(update_chain):
    while True:
        if not update_chain:
            print('  λ')
            break
        #dbg('update', update_chain)
        wt_val = update_chain.dereference()
        obj = None
        #dbg('wt_val', wt_val)
        val_bytes = gdb.selected_inferior().read_memory(wt_val['data'], wt_val['size'])
        can_bson = wt_val['type'] == 3
        if can_bson:
            try:
                obj = bson.decode_all(val_bytes)[0]
            except:
                pass
        print('  ' + '\n  '.join(str(wt_val).split('\n')) + " " + str(obj) + " =>")

        update_chain = update_chain['next']

def dump_insert_list(wt_insert):
    key_struct = wt_insert['u']['key']
    key = gdb.selected_inferior().read_memory(int(wt_insert.address) + key_struct['offset'], key_struct['size']).tobytes()
    print('Key: ' + str(key))
    print('Value:')
    update_chain = wt_insert['upd']
    dump_update_chain(update_chain)

def dump_skip_list(wt_insert_head):
    wt_insert = wt_insert_head['head'][0]
    idx = 0
    while True:
        if not wt_insert:
            break
        dump_insert_list(wt_insert.dereference())
        #dbg('insert' + str(idx), wt_insert.dereference())
        idx+=1
        wt_insert = wt_insert['next'][0]

def dump_modified(leaf_page):
    print("Modify:")
    if not leaf_page['modify']:
        print("No modifies")
        return

    leaf_modify = leaf_page['modify'].dereference()
    #dbg('modify', leaf_modify)
    row_leaf_insert = leaf_modify['u2']['row_leaf']['insert']
    #dbg('row store', row_leaf_modify)
    if not row_leaf_insert:
        print("No insert list")
    else:
        print("Insert list:")
        dump_skip_list(row_leaf_insert.dereference().dereference())
    
    row_leaf_update = leaf_modify['u2']['row_leaf']['update']
    if not row_leaf_update:
        print("No update list")
    else:
        print("Update list:")
        leaf_num_entries = int(leaf_page['entries'])
        for i in range(0, leaf_num_entries):
            dump_update_chain(row_leaf_update[i])

def dump_disk(leaf_page):
    leaf_num_entries = int(leaf_page['entries'])
    dbg('in-memory page:', leaf_page)
    dsk = leaf_page['dsk'].dereference()
    if int(dsk.address) == 0:
        print("No page loaded from disk.")
        return
    dbg('on-disk page:', dsk)
    wt_page_header_size = 28
    wt_block_header_size = 12
    page_bytes = gdb.selected_inferior().read_memory(int(dsk.address) + wt_page_header_size + wt_block_header_size, int(dsk['mem_size'])).tobytes()
    print("Dsk:\n" + str(page_bytes))

def dump_handle(dhandle):
    print("Dumping: " + dhandle['name'].string())
    btree = get_btree_handle(dhandle)
    root = btree['root']
    root_page = root['page'].dereference()
    dbg('btree', get_btree_handle(user))
    dbg('root', btree['root'])
    dbg('root page', root_page)
    rpindex = root_page['u']['intl']['__index'].dereference()
    dbg('rpindex', rpindex)
    dbg('rp-pre-index', rpindex['index'].dereference().dereference())
    leaf_page = rpindex['index'].dereference().dereference()['page'].dereference()
    dbg('leaf', leaf_page)
    dump_disk(leaf_page)
    dump_modified(leaf_page)