/* kallsyms loader */
/* by goroh_kun    */
/* modified from H2enum Version 1.09 */

#include <idc.idc>

// returns -1 if symbol is NOT 'space'
static is_space(c)
{
	return strstr(" \t\r\n\b",c);
}

// strip leading blank characters from string.
static ltrim(str)
{
	auto pos,c,l;

	l = strlen(str);
	pos = 0;
	while (pos < l)
	{
		c = substr(str,pos,pos+1);
		if (is_space(c) == -1) break;
		pos++;
	}
	return substr(str,pos,-1);
}

// strip trailing blank characters from string.
static rtrim(str)
{
	auto pos,c;

	pos = strlen(str);
	while (pos > 0)
	{
		c = substr(str,pos-1,pos);
		if (is_space(c) == -1) break;
		pos--;
	}
	return substr(str,0,pos);
}

static trim(str)
{
	return rtrim( ltrim(str) );
}

// Find first delimiter position in string (SPACE or TAB).
static FindDelim(str)
{
	auto pos1,pos2;

	pos1 = strstr(str," ");
	pos2 = strstr(str,"\t");
	if (pos1 == -1)  return pos2;
	if (pos2 == -1)  return pos1;
	if (pos1 < pos2) return pos1;
	else		 return pos2;
}

// Main conversion routine
static load_kallsyms(fname)
{
        auto def_addr, def_name, def_type;
	auto hFile,in_str,pos,str_no,c;

        if ((hFile=fopen(fname,"r")) == 0)
	{
		Warning("Couldn't open file '%s'!",fname);
		return -1;
	}

	Message("Conversion started...\n");

	str_no = 0;
	while ((in_str=readstr(hFile)) != -1)
	{
		str_no++;
		pos = FindDelim(in_str);
		def_addr = xtol(substr(in_str, 0, pos));
		in_str = substr(in_str, pos + 1, -1);
		pos = FindDelim(in_str);
		def_type = substr(in_str, 0, pos);
		in_str = substr(in_str, pos + 1, -1);
		def_name = trim(in_str);
		MakeCode(def_addr);
		if(0 == MakeNameEx(def_addr, def_name, SN_NOCHECK | SN_NOWARN | SN_AUTO)){
			auto i;
			for(**1; i<100; i++){
				def_name = def_name + "_" + ltoa(i, 10);
				if(0 != MakeNameEx(def_addr, def_name, SN_NOCHECK | SN_NOWARN | SN_AUTO)){
					break;
				}
			}
		}
		Message("name %s = '%08x'(%s)\n", def_name, def_addr, def_type);
	}
	fclose(hFile);

	Message("Successful %d elements imported.\n", str_no);
	return 0;
}

static main()
{
	auto fname;

	fname = AskFile(0,"*.txt","Choose a kallsyms file to parse:");
	if (fname == "") return;
	load_kallsyms(fname);
}