Skip to content

Commit

Permalink
Reduce pc_to_file_name time
Browse files Browse the repository at this point in the history
  • Loading branch information
@mooncat-greenpy
mooncat-greenpy committed May 18, 2023
1 parent d4b1a1e commit 8dea68b
Showing 1 changed file with 57 additions and 38 deletions.
95 changes: 57 additions & 38 deletions src/main/java/golanganalyzerextension/function/GolangFunction.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -404,6 +404,7 @@ private boolean init_file_line_map() {
int i=0;
boolean first=true;
int pc_offset=0;
List<PcFile> pcfile_list=get_pc_to_file_name_list(pc_offset);
while(true) {
int line_num_add;
int byte_size;
Expand All @@ -425,9 +426,12 @@ private boolean init_file_line_map() {
line_num_add=zig_zag_decode(line_num_add);
line_num+=line_num_add;
pc_offset+=byte_size*go_bin.get_quantum();
String file_name=pc_to_file_name(pc_offset);
if(file_name==null) {
file_name="not_found";
String file_name="not_found";
for(PcFile pcfile : pcfile_list) {
if(pcfile.offset<pc_offset && pc_offset<=pcfile.offset+pcfile.size && pcfile.name!=null) {
file_name=pcfile.name;
break;
}
}

file_line_comment_map.put(key, new FileLine(func_addr, key, pc_offset-key, file_name, line_num));
Expand Down Expand Up @@ -505,7 +509,19 @@ private boolean init_frame_map() {
return frame_size;
}*/

private String pc_to_file_name(int target_pc_offset) {
private class PcFile {
int offset;
int size;
String name;

public PcFile(int key, int size, String name) {
this.offset=key;
this.size=size;
this.name=name;
}
}

private List<PcFile> get_pc_to_file_name_list(int target_pc_offset) {
boolean is_go116=false;
boolean is_go118=false;
if(go_bin.ge_go_version("go1.16beta1")) {
Expand All @@ -515,6 +531,7 @@ private String pc_to_file_name(int target_pc_offset) {
is_go118=true;
}

List<PcFile> pcfile_list=new ArrayList<>();
int pointer_size=go_bin.get_pointer_size();
Address pcheader_base=go_bin.get_pcheader_base();
Address pcfile_base;
Expand All @@ -531,7 +548,7 @@ private String pc_to_file_name(int target_pc_offset) {
}
} catch (BinaryAccessException e) {
Logger.append_message(String.format("Failed to get pcfile base: info_addr=%s, message=%s", info_addr, e.getMessage()));
return null;
return pcfile_list;
}

long file_no=-1;
Expand All @@ -548,54 +565,56 @@ private String pc_to_file_name(int target_pc_offset) {
i+=Integer.toBinaryString(byte_size).length()/8+1;
} catch (BinaryAccessException e) {
Logger.append_message(String.format("Failed to get line num: info_addr=%s, message=%s", info_addr, e.getMessage()));
return null;
break;
}

if(file_no_add==0 && !first) {
break;
}
first=false;
int key=pc_offset;
file_no_add=zig_zag_decode(file_no_add);
file_no+=file_no_add;
pc_offset+=byte_size*go_bin.get_quantum();

if(target_pc_offset<=pc_offset) {
if(is_go116) {
Address file_name_addr;
try {
int cu_offset=(int)go_bin.get_address_value(info_addr, (is_go118?4:pointer_size)+4*7, 4);
Address cutab_base;
if(is_go118) {
cutab_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*4, pointer_size));
}else {
cutab_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*3, pointer_size));
}

long file_no_offset=go_bin.get_address_value(cutab_base, (cu_offset+file_no)*4, 4);
Address file_base;
if(is_go118) {
file_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*5, pointer_size));
}else {
file_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*4, pointer_size));
}
file_name_addr=go_bin.get_address(file_base, file_no_offset);
} catch (BinaryAccessException e) {
Logger.append_message(String.format("Failed to get file name addr: pcheader_addr=%s, pcfile_base=%s, file_no=%x, message=%s", pcheader_base, pcfile_base, file_no, e.getMessage()));
return null;
}

String file_name=go_bin.create_string_data(file_name_addr).orElse(String.format("not_found_%x", file_name_addr.getOffset()));
service.add_filename(file_name);
return file_name;
}
if(!is_go116) {
if((int)file_no-1<0 || file_name_list.size()<=(int)file_no-1) {
Logger.append_message(String.format("File name list index out of range: func_addr=%s, index=%x", func_addr, (int)file_no-1));
return null;
pcfile_list.add(new PcFile(key, pc_offset-key, null));
continue;
}
return file_name_list.get((int)file_no-1);
pcfile_list.add(new PcFile(key, pc_offset-key, file_name_list.get((int)file_no-1)));
continue;
}

Address file_name_addr;
try {
int cu_offset=(int)go_bin.get_address_value(info_addr, (is_go118?4:pointer_size)+4*7, 4);
Address cutab_base;
if(is_go118) {
cutab_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*4, pointer_size));
}else {
cutab_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*3, pointer_size));
}

long file_no_offset=go_bin.get_address_value(cutab_base, (cu_offset+file_no)*4, 4);
Address file_base;
if(is_go118) {
file_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*5, pointer_size));
}else {
file_base=go_bin.get_address(pcheader_base, go_bin.get_address_value(pcheader_base, 8+pointer_size*4, pointer_size));
}
file_name_addr=go_bin.get_address(file_base, file_no_offset);
} catch (BinaryAccessException e) {
Logger.append_message(String.format("Failed to get file name addr: pcheader_addr=%s, pcfile_base=%s, file_no=%x, message=%s", pcheader_base, pcfile_base, file_no, e.getMessage()));
break;
}

String file_name=go_bin.create_string_data(file_name_addr).orElse(String.format("not_found_%x", file_name_addr.getOffset()));
service.add_filename(file_name);
pcfile_list.add(new PcFile(key, pc_offset-key, file_name));
}
return null;
return pcfile_list;
}

private int zig_zag_decode(int value) {
Expand Down

0 comments on commit 8dea68b

Please sign in to comment.