def extract_item(viewer, stripped=False): cnt = ida_kernwin.get_custom_viewer_place(viewer, True)[1] line = ida_kernwin.get_custom_viewer_curline(viewer, True) tags = [ida_lines.SCOLOR_ON, ida_lines.SCOLOR_OFF] expr = re.compile("[A-Z0-9_]") def find_pos(line, pos, inc): while 0 <= pos < len(line): if not stripped and line[pos] in tags: break if stripped and not expr.match(line[pos]): break pos += inc return pos pos = ida_lines.tag_advance(line, cnt) if pos < 0 or pos >= len(line): return while line[pos] in [ida_lines.SCOLOR_ON, ida_lines.SCOLOR_OFF]: pos += 2 if pos < 0 or pos >= len(line): return prev_pos, next_pos = find_pos(line, pos, -1), find_pos(line, pos, +1) if stripped: return None, line[prev_pos + 1:next_pos].strip() return line[prev_pos + 1], line[prev_pos + 2:next_pos].strip()
def _process(self): cf = self.vu.cfunc ci = ida_hexrays.ctree_item_t() ccode = cf.get_pseudocode() for ypos in range(cf.hdrlines, len(ccode)): tline = ccode.at(ypos).line # TODO: optimize the following loop idx = 0 while idx < len(tline): citem_len = 0 # get all items on a line if cf.get_line_item(tline, idx, True, None, ci, None): iea = ci.it.ea if iea != BADADDR: # generate color-tagged/addr-tagged text of current item citem = ci.it.print1(None) citem_len = len(ida_lines.tag_remove(citem)) # find (tagged) item text in current line pos = tline.find(citem) while pos != -1: # calculate x position of item text in line # by subtracting the number of color tag # characters up to position "pos" xpos = len(ida_lines.tag_remove(tline[:pos])) self._add_item(iea, xpos, ypos, citem_len) pos = tline.find(citem, pos + citem_len) idx += ida_lines.tag_advance(tline[idx], 1) return
def remove_spaces(sl): dbg("*" * 80) l = sl.line out = [] def push(c): dbg("Appending '%s'" % c) out.append(c) # skip initial spaces, do not compress them while True: l = my_tag_skipcodes(l, out) if not l: break c = l[0] if not c.isspace(): break push(c) l = l[1:] # remove all spaces except in string and char constants delim = None # if not None, then we are skipping until 'delim' last = None # last seen character while True: # go until comments dbg("-" * 60) nchars = ida_lines.tag_advance(l, 1) push(l[0:nchars]) l = l[nchars:] l = my_tag_skipcodes(l, out) if not l: break if l.startswith("//"): push(l) break c = l[0] dbg("c: '%s', last: '%s', l: '%s'" % (c, last, l)) if delim: # we're inside a literal. if c == delim: delim = None # literal ended elif c == '"' or c == "'": delim = c # string/char literal started elif c.isspace(): end = l.lstrip() nptr = my_tag_skipcodes(end, out) dbg("end: '%s', nptr: '%s'" % (end, nptr)) # do not concatenate idents if not is_cident_char(last) or not is_cident_char(nptr[0]): l = end c = l[0] if l else '' dbg("new l: '%s'" % l) last = l[0] if l else '' sl.line = "".join(out)