dddlZddlZddlmZddlmZddlmZddlmZddl Z ddl m Z m Z m Z ejdd ZGd d eZy) N) OrderedDict)Mapping) struct_parse) bisect_right)CStringStructIf NameLUTEntryzcu_ofs die_ofscNeZdZdZdZdZdZdZdZdZ dZ d d Z d Z d Z y )NameLUTaO A "Name LUT" holds any of the tables specified by .debug_pubtypes or .debug_pubnames sections. This is basically a dictionary where the key is the symbol name (either a public variable, function or a type), and the value is the tuple (cu_offset, die_offset) corresponding to the variable. The die_offset is an absolute offset (meaning, it can be used to search the CU by iterating until a match is obtained). An ordered dictionary is used to preserve the CU order (i.e, items are stored on a per-CU basis (as it was originally in the .debug_* section). Usage: The NameLUT walks and talks like a dictionary and hence it can be used as such. Some examples below: # get the pubnames (a NameLUT from DWARF info). pubnames = dwarf_info.get_pubnames() # lookup a variable. entry1 = pubnames["var_name1"] entry2 = pubnames.get("var_name2", default=) print(entry2.cu_ofs) ... # iterate over items. for (name, entry) in pubnames.items(): # do stuff with name, entry.cu_ofs, entry.die_ofs # iterate over items on a per-CU basis. import itertools for cu_ofs, item_list in itertools.groupby(pubnames.items(), key = lambda x: x[1].cu_ofs): # items are now grouped by cu_ofs. # item_list is an iterator yeilding NameLUTEntry'ies belonging # to cu_ofs. # We can parse the CU at cu_offset and use the parsed CU results # to parse the pubname DIEs in the CU listed by item_list. for item in item_list: # work with item which is part of the CU with cu_ofs. cJ||_||_||_d|_d|_yN)_stream_size_structs_entries _cu_headers)selfstreamsizestructss 8/usr/lib/python3/dist-packages/elftools/dwarf/namelut.py__init__zNameLUT.__init__@s&    cl|j|j\|_|_|jS)a Returns the parsed NameLUT entries. The returned object is a dictionary with the symbol name as the key and NameLUTEntry(cu_ofs, die_ofs) as the value. This is useful when dealing with very large ELF files with millions of entries. The returned entries can be pickled to a file and restored by calling set_entries on subsequent loads. )r _get_entriesrrs r get_entrieszNameLUT.get_entriesJs0 == .2.?.?.A +DM4+}}rc ||_||_y)a Set the NameLUT entries from an external source. The input is a dictionary with the symbol name as the key and NameLUTEntry(cu_ofs, die_ofs) as the value. This option is useful when dealing with very large ELF files with millions of entries. The entries can be parsed once and pickled to a file and can be restored via this function on subsequent loads. N)rr)rentries cu_headerss r set_entrieszNameLUT.set_entriesXs  %rc~|j|j\|_|_t|jS)z? Returns the number of entries in the NameLUT. )rrrlenrs r__len__zNameLUT.__len__es5 == .2.?.?.A +DM4+4==!!rc|j|j\|_|_|jj|S)z{ Returns a namedtuple - NameLUTEntry(cu_ofs, die_ofs) - that corresponds to the given symbol name. rrrget)rnames r __getitem__zNameLUT.__getitem__ms; == .2.?.?.A +DM4+}}  &&rc~|j|j\|_|_t|jS)z@ Returns an iterator to the NameLUT dictionary. )rrriterrs r__iter__zNameLUT.__iter__vs5 == .2.?.?.A +DM4+DMM""rc|j|j\|_|_|jjS)z7 Returns the NameLUT dictionary items. )rrritemsrs rr0z NameLUT.items~s9 == .2.?.?.A +DM4+}}""$$rNc|j|j\|_|_|jj||S)z Returns NameLUTEntry(cu_ofs, die_ofs) for the provided symbol name or None if the symbol does not exist in the corresponding section. r()rr*defaults rr)z NameLUT.gets= == .2.?.?.A +DM4+}}  w//rcl|j|j\|_|_|jS)zF Returns all CU headers. Mainly required for readelf. )rrrrs rget_cu_headerszNameLUT.get_cu_headerss4    #.2.?.?.A +DM4+rc |jjdt}g}d}td|jj dt dtd}||jkrt|jj|j|}|j|||jz|jjz}|j} t||j}|jdk(rn7t!|||jz||j"j%d<]||jkr||fS)zt Parse the (name, cu_ofs, die_ofs) information from this section and store as a dictionary. rDwarf_offset_name_pairdie_ofsc |dS)Nr7)ctxs rz&NameLUT._get_entries..s s9~rr*)cu_ofsr7zutf-8)rseekrr r Dwarf_offsetr rrrDwarf_nameLUT_headerappend unit_lengthinitial_length_field_sizedebug_info_offsetr7r r*decode)rr!r"offset entry_struct namelut_hdr hdr_cu_ofsentrys rrzNameLUT._get_entriess: !-  6 **95-wv?A tzz!'t}}'I'ILL&*K   k *{666]]<<>?F %66J$\4<<@==A%6B!+",u}}"<7> ))'23tzz!6$$rr)__name__ __module__ __qualname____doc__rrr#r&r+r.r0r)r4rr9rrr r s:)V   &"'#%0 .%rr )os collectionsrcollections.abcr common.utilsrbisectrmath constructrr r namedtupler r r9rrrVsE ##' ++%{%%n6FG r%gr%r