colorscript.idc

//
//Set Color script
//by tomsons26
//Sets assembly color to a chosen color for example for making functions as decompiled

#include <idc.idc>

static mark_as_lib()
{
	auto Current,Selection_Start,Selection_End; 

	//Get the selection start and end address
	Selection_Start = ScreenEA(); 
	Selection_End	= Selection_Start + 4; 
	
	//Check if the start and end addresses are valid
	if(Selection_End == BADADDR || Selection_Start== BADADDR)
	{
		Message("**Set As Lib Script Range Error**\n") ;
		return -1; 
	}

	//Sets the color for the range from Selection start to the end
	for (Current = Selection_Start; Current!=BADADDR; Current = NextHead(Current, Selection_End))
	{
        SetFunctionFlags(Current,FUNC_LIB|GetFunctionFlags(Current));
	}
	//Refreshes all disassembly views
	Refresh();
    RefreshLists();
}

static mark_as_color()
{
	auto Current,Selection_Start,Selection_End; 

	//Get the selection start and end address
	Selection_Start = ScreenEA(); 
	Selection_End	= Selection_Start + 4; 
	
	//Check if the start and end addresses are valid
	if(Selection_End == BADADDR || Selection_Start== BADADDR)
	{
		Message("**Set Color Script Range Error**\n") ;
		return -1; 
	}

	//Sets the color for the range from Selection start to the end
	for (Current = Selection_Start; Current!=BADADDR; Current = NextHead(Current, Selection_End))
	{
		//Sets the color, format is address, type, and color in hex as 0xBBGGRR
		if (SetColor(Current,CIC_FUNC,0xEEFFF0)==0)
		{
			Message("**SetColor function Error**\n") ; 
		}
	}
	//Refreshes all disassembly views
	Refresh();
    RefreshLists();
}

static main(void)
{
	AddHotkey("6", "mark_as_lib");
	AddHotkey("7", "mark_as_color");
}