static void
check_trigger(packet_info_t *packet, iprecord_t *rec)
{
trigger_t *i;
flowdata_t *flow;
flow = ipstate_lookup_flowdata(rec, packet->ip_type);
if (flow == NULL)
return;
for (i = t_list[packet->ip_type]; i != NULL; i = i->next)
{
int pps, mbps;
int do_trigger = 0, nullable_trigger = 0, threshold_trigger = 0;
DPRINTF("check trigger packet %p record %p protocol %d pktproto %d\n", packet, rec, i->protocol, packet->ip_type);
mbps = (int) floor((flow->flow / 1000000.));
pps = flow->pps;
DPRINTF("... pps %d mbps %d target_pps %d target_mbps %d\n", pps, mbps, i->target_pps, i->target_mbps);
if (i->target_pps && (pps > i->target_pps))
do_trigger = 1;
if (i->target_mbps && (mbps > i->target_mbps))
do_trigger = 1;
if (i->target_mbps && (mbps < i->target_mbps))
do_trigger = 0;
if (i->below_mbps && (mbps > i->below_mbps))
do_trigger = 0;
if (i->tcp_synonly && packet->tcp_flags != TCP_SYN)
do_trigger = 0;
if (i->target_flowcount && flow->count < i->target_flowcount)
do_trigger = 0;
DPRINTF("trigger %p conditions %s for flow %p\n", i, do_trigger == 1 ? "met" : "not met", rec);
if (do_trigger)
HOOK_CALL(HOOK_CHECK_EXEMPT, packet, rec, &do_trigger);
DPRINTF("HOOK_CHECK_EXEMPT result %d\n", do_trigger);
HOOK_CALL(HOOK_CHECK_NULLABLE, packet, rec, &nullable_trigger);
DPRINTF("HOOK_CHECK_NULLABLE result %d\n", nullable_trigger);
HOOK_CALL(HOOK_CHECK_THRES, packet, rec, &threshold_trigger, pps, mbps, i->target_pps, i->target_mbps);
DPRINTF("HOOK_CHECK_THRES result %d\n", threshold_trigger);
if (do_trigger && nullable_trigger && !threshold_trigger)
trigger_ban(i, packet, rec);
}
}
static void
check_trigger(packet_info_t *packet, iprecord_t *rec)
{
trigger_t *i;
flowdata_t *flow;
flow = ipstate_lookup_flowdata(rec, packet->ip_type);
if (flow == NULL)
return;
for (i = t_list[packet->ip_type]; i != NULL; i = i->next)
{
int pps, mbps;
int do_trigger = 0;
float mbps_ratio, pps_ratio;
DPRINTF("check trigger packet %p record %p protocol %d pktproto %d\n", packet, rec, i->protocol, packet->ip_type);
mbps = (int) floor((flow->flow / 1000000.));
pps = flow->pps;
mbps_ratio = (float) ((flow->count + 1) / (mbps + 1));
pps_ratio = (float) ((flow->count + 1) / (pps + 1));
if (i->flow_mbps_ratio > 0.0 && i->flow_mbps_ratio < mbps_ratio)
do_trigger = 1;
if (i->flow_pps_ratio > 0.0 && i->flow_pps_ratio < pps_ratio)
do_trigger = 1;
if (i->minimum_flows && i->minimum_flows > flow->count)
do_trigger = 0;
if (i->minimum_mbps && i->minimum_mbps > mbps)
do_trigger = 0;
if (i->minimum_pps && i->minimum_pps > pps)
do_trigger = 0;
DPRINTF("trigger %p conditions %s for flow %p\n", i, do_trigger == 1 ? "met" : "not met", rec);
if (do_trigger)
HOOK_CALL(HOOK_CHECK_EXEMPT, packet, rec, &do_trigger);
DPRINTF("HOOK_CHECK_EXEMPT result %d\n", do_trigger);
if (do_trigger)
trigger_ban(i, packet, rec);
}
}
void Unload(){
HOOK_CALL( 0x004018A0, 0x00401C8A ); // ReferencedPortraits
HOOK_CALL( 0x004018A0, 0x00401E21 ); // ReferencedPortraits
HOOK_CALL( 0x0041EC00, 0x0042A3C6 ); // PortraitShutdown
HOOK_CALL( 0x0041EB40, 0x00428A1F ); // PortraitInit
*((BYTE*)0x0041EE08) = 0x3F; // max head_db entries
*((BYTE*)0x0041EC9B) = 0x3F; // max head_db entries
memcpy( (void*)0x004184B0, "\x8B\x44\x24\x04\x8D", 5 );
memcpy( (void*)0x004184D0, "\x8B\x44\x24\x04\x8D", 5 );
memcpy( (void*)0x004184F0, "\x8B\x44\x24\x04\x8D", 5 );
memcpy( (void*)0x00418510, "\x56\x57\x66\x33\xF6", 5 );
memcpy( (void*)0x00418540, "\x8B\x44\x24\x04\x8D", 5 );
memcpy( (void*)0x0042B510, "\x81\xEC\x00\x01\x00", 5 );
memcpy( (void*)0x0042B600, "\xB8\x78\xEB\x4C\x00", 5 );
memcpy( (void*)0x0042B9E0, "\x81\xEC\x00\x01\x00", 5 );
*((DWORD*)0x004018DF) = 0x004CFBD0; // Portrait_First
*((DWORD*)0x00401910) = 0x004D09E8; // Portrait_Last
*((DWORD*)0x0041D3A5) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0042459C) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0042BD81) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0042BEDD) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0042E238) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0042B705) = 0x004CEB78; // &(SpriteTable[0].loaded_index)
*((DWORD*)0x0047FE01) = 0x004EBC88; // offset_table
*((DWORD*)0x00526D24) = 0; // Loaded_Sprite_Array
*((DWORD*)0x00526D2C) = 0; // Loaded_Banner_Array
*((DWORD*)0x00409F11) = 0x004C0940; // troop_book_unit_pic
*((DWORD*)0x00409F52) = 0x004C0940; // troop_book_unit_pic
*((DWORD*)0x0040C0CC) = 0x004C0940; // troop_book_unit_pic
*((DWORD*)0x0040C108) = 0x004C0940; // troop_book_unit_pic
*((DWORD*)0x00409DE3) = 0x004C0A80; // troop_book_leader_pic
*((DWORD*)0x0040BD24) = 0x004C0A80; // troop_book_leader_pic
*((DWORD*)0x00409EC2) = 0x004E2640; // unit name
*((DWORD*)0x0040B876) = 0x004E2640; // unit name
*((DWORD*)0x0040BE4A) = 0x004E2640; // unit name
*((DWORD*)0x0040C20F) = 0x004E2640; // unit name
*((DWORD*)0x0040C461) = 0x004E2640; // unit name
*((DWORD*)0x0040D213) = 0x004E2640; // unit name
*((DWORD*)0x0040D9C3) = 0x004E2640; // unit name
*((DWORD*)0x0041985C) = 0x004E2640; // unit name
*((DWORD*)0x0041A5E4) = 0x004E2640; // unit name
*((DWORD*)0x0041B1B0) = 0x004E2640; // unit name
*((DWORD*)0x0041B4CA) = 0x004E2640; // unit name
*((DWORD*)0x0042671A) = 0x004E2640; // unit name
*((DWORD*)0x004286E1) = 0x004E2640; // unit name
*((DWORD*)0x00409E63) = 0x004E2780; // unit_desc
*((DWORD*)0x0040BDE0) = 0x004E2780; // unit_desc
*((DWORD*)0x004099B5) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x00409DAD) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040B83D) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040BCE0) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040C1D5) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040C413) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040D97C) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0040E1E9) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x004192A4) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0041981E) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0041A39E) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0041A5A6) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0041B172) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x0041B48C) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x00425B26) = 0x004C0BC0; // small_banner_index
*((DWORD*)0x00426DEB) = 0x004C0BC0; // small_banner_index
for( int i = 80; i < 180; i++ ){
if( troop_book_leader_pic[i] != NULL ){
HeapFree( GetProcessHeap(), 0, troop_book_leader_pic[i] );
troop_book_leader_pic[i] = NULL;
}
if( troop_book_unit_pic[i] != NULL ){
HeapFree( GetProcessHeap(), 0, troop_book_unit_pic[i] );
troop_book_unit_pic[i] = NULL;
}
if( troop_book_unit_desc[i] != NULL ){
HeapFree( GetProcessHeap(), 0, troop_book_unit_desc[i] );
troop_book_unit_desc[i] = NULL;
}
}
}
void Load()
{
// parse text file
std::string path = darkomen::modmenu::getCurrentModPath();
if (path.empty())
{
return;
}
path += "\\PRG_ENG\\xslots.txt";
if( 0xFFFFFFFF == getFileAttributes_orig( path.c_str() ) ) return;
char* readbuf = ReadInFile( path.c_str() );
if( readbuf == NULL ) return;
char* text = readbuf;
int line = 1;
while( *text != '\0' )
{
// temp vars
DWORD slot_id;
char buf[ 12 ];
DWORD len;
DWORD required_len;
// scan till some content is found
SkipWhiteSpaceAndComments( text );
if( *text == '\r' ) text++;
if( *text == '\n' ){ text++; line++; continue; }
//
required_len = ReadOutWord( buf, sizeof(buf), text, &len );
if( ( required_len > sizeof(buf) ) || ( required_len <= 1 ) ){
Log( "xslot: Error ( line %d )", line );
goto the_end;
}
text += len;
SkipWhiteSpaceAndComments( text );
//
slot_id = ReadOutInteger( text, &len );
if( len == 0 ){
Log( "xslot: Error expected slot_id ( line %d )", line );
goto the_end;
}
text += len;
SkipWhiteSpaceAndComments( text );
//
if( 0 == strcmp( buf, "SPRITE" ) ){
if( slot_id > ( X_SPRITE_LAST - X_SPRITE_FIRST ) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "slot_id", ( X_SPRITE_LAST - X_SPRITE_FIRST ), line );
goto the_end;
}
SPRITE_SLOT* p = &SpriteTable[ slot_id + X_SPRITE_FIRST ];
//
required_len = ReadOutWord( p->name, sizeof(p->name), text, &len );
if( required_len > sizeof(p->name) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "sprite_file_name", sizeof(p->name), line );
goto the_end;
}
if( len == 0 ){ continue; }
text += len;
SkipWhiteSpaceAndComments( text );
//
DWORD off = ReadOutInteger( text, &len );
text += len;
if( len == 0 ){
Log( "xslot: Error missing sprite_type_offset ( line %d )", line );
goto the_end;
}
if( ( off > 0x0000023E ) || ( ( off % 0x000000E ) != 0 ) ){
Log( "xslot: Error invalid value (0x%08X) for sprite_type_offset ( line %d )", off, line );
goto the_end;
}
p->type_id = 8; // troop sprite
offset_table[ slot_id + X_SPRITE_FIRST ] = (WORD)off;
}
else {
if( 0 == strcmp( buf, "NAME" ) ){
if( slot_id > ( X_NAME_LAST - X_NAME_FIRST ) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "slot_id", ( X_NAME_LAST - X_NAME_FIRST ), line );
goto the_end;
}
required_len = ReadOutWord( &x_names[slot_id][0], 64, text, &len );
if( len == 0 ){ continue; }
text += len;
if( required_len > 64 ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "name", 64, line );
goto the_end;
}
unit_name[ slot_id + X_NAME_FIRST ] = &x_names[slot_id][0];
}
else {
if( 0 == strcmp( buf, "BANNER" ) ){
if( slot_id > ( X_BANNER_LAST - X_BANNER_FIRST ) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "slot_id", ( X_BANNER_LAST - X_BANNER_FIRST ), line );
goto the_end;
}
SPRITE_SLOT* p = &SpriteTable[ slot_id + X_BANNER_FIRST ];
//
required_len = ReadOutWord( p->name, sizeof(p->name), text, &len );
if( required_len > sizeof(p->name) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "sprite_file_name", sizeof(p->name), line );
goto the_end;
}
if( len == 0 ){ continue; }
text += len;
SkipWhiteSpaceAndComments( text );
//
DWORD shape = ReadOutInteger( text, &len );
if( len == 0 ){
Log( "xslot: Error missing banner_shape_id ( line %d )", line );
goto the_end;
}
text += len;
p->banner_bkgrnd_id = shape;
}
else {
if( 0 == strcmp( buf, "BOOK" ) ){
if( slot_id > ( X_BOOK_LAST - X_BOOK_FIRST ) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "slot_id", ( X_BOOK_LAST - X_BOOK_FIRST ), line );
goto the_end;
}
//
troop_book_banner_index[ slot_id + X_BOOK_FIRST ] = (char) ReadOutInteger( text, &len );
if( len == 0 ){ continue; }
text += len;
SkipWhiteSpaceAndComments( text );
//
required_len = ReadOutWord( 0, 0, text, &len );
if( len == 0 ){
Log( "xslot: Error missing leader_pic ( line %d )", line );
}
troop_book_leader_pic[ slot_id + X_BOOK_FIRST ] = (char*) HeapAlloc( GetProcessHeap(), 0, required_len );
if( troop_book_leader_pic[ slot_id + X_BOOK_FIRST ] == NULL ){
Log( "xslot: Alloc Failed ( line %d )", line );
goto the_end;
}
ReadOutWord( troop_book_leader_pic[ slot_id + X_BOOK_FIRST ], required_len, text, &len );
text += len;
SkipWhiteSpaceAndComments( text );
//
required_len = ReadOutWord( 0, 0, text, &len );
if( len == 0 ){
Log( "xslot: Error missing troop_pic ( line %d )", line );
goto the_end;
}
troop_book_unit_pic[ slot_id + X_BOOK_FIRST ] = (char*) HeapAlloc( GetProcessHeap(), 0, required_len );
if( troop_book_unit_pic[ slot_id + X_BOOK_FIRST ] == NULL ){
Log( "xslot: Alloc Failed ( line %d )", line );
goto the_end;
}
ReadOutWord( troop_book_unit_pic[ slot_id + X_BOOK_FIRST ], required_len, text, &len );
text += len;
SkipWhiteSpaceAndComments( text );
//
required_len = ReadOutWord( 0, 0, text, &len );
if( len == 0 ){
Log( "xslot: Error missing troop_desc ( line %d )", line );
goto the_end;
}
troop_book_unit_desc[ slot_id + X_BOOK_FIRST ] = (char*) HeapAlloc( GetProcessHeap(), 0, required_len );
if( troop_book_unit_desc[ slot_id + X_BOOK_FIRST ] == NULL ){
Log( "xslot: Alloc Failed ( line %d )", line );
goto the_end;
}
ReadOutWord( troop_book_unit_desc[ slot_id + X_BOOK_FIRST ], required_len, text, &len );
text += len;
}
else{
if( 0 == strcmp( buf, "PORTRAIT" ) ){
if( slot_id > ( X_PORTRAIT_LAST - X_PORTRAIT_FIRST ) ){
Log( "xslot: Error %s exceeds max of %d ( line %d )", "slot_id", ( X_PORTRAIT_LAST - X_PORTRAIT_FIRST ), line );
goto the_end;
}
DWORD m3d_id = ReadOutInteger( text, &len );
if( len == 0 ){
Log( "xslot: Error expected m3d_id ( line %d )", line );
goto the_end;
}
text += len;
SPRITE_SLOT* p = &SpriteTable[ slot_id + X_PORTRAIT_FIRST ];
p->loaded_index = m3d_id;
}
else {
Log( "xslot: Error unrecognized command \"%s\" ( line %d )", buf, line );
goto the_end;
}
}
}
}
}
} // while
// copy original data in from the game executable
memcpy( SpriteTable, (void*)0x004CEB50, 257 * sizeof( SPRITE_SLOT ) );
memcpy( offset_table, (void*)0x004EBC88, 96 * sizeof( WORD ) );
memcpy( unit_name, (void*)0x004E2640, 0x50 * sizeof( char* ) );
memcpy( troop_book_unit_pic, (void*)0x004C0940, 0x50 * sizeof( char* ) );
memcpy( troop_book_leader_pic, (void*)0x004C0A80, 0x50 * sizeof( char* ) );
memcpy( troop_book_unit_desc, (void*)0x004E2780, 0x50 * sizeof( char* ) );
memcpy( troop_book_banner_index, (void*)0x004C0BC0, 0x50 * sizeof( char ) );
// install hooks
HOOK_CALL( &ReferencedPortraits, 0x00401C8A );
HOOK_CALL( &ReferencedPortraits, 0x00401E21 );
HOOK_CALL( PortraitShutdown, 0x0042A3C6 );
HOOK_CALL( PortraitInit, 0x00428A1F );
*((BYTE*)0x0041EE08) = 0x7F; // override max head_db entries (0x3F)
*((BYTE*)0x0041EC9B) = 0x7F; // override max head_db entries (0x3F)
WRITE_JMP( GetType, 0x004184B0 );
WRITE_JMP( AddRef, 0x004184D0 );
WRITE_JMP( Release, 0x004184F0 );
WRITE_JMP( ZeroRef, 0x00418510 );
WRITE_JMP( GetLoadedIndex, 0x00418540 );
WRITE_JMP( LoadReferencedSprites, 0x0042B510 );
WRITE_JMP( LoadedSpriteArrayUnloaded, 0x0042B600 );
WRITE_JMP( LoadReferencedBanners, 0x0042B9E0 );
*((SPRITE_SLOT**)0x004018DF) = &SpriteTable[ PORTRAIT_FIRST ]; // sub_4018A0 ( PortraitsInit )
*((SPRITE_SLOT**)0x00401910) = &SpriteTable[ PORTRAIT_LAST ]; // sub_4018A0 ( PortraitsInit )
*((DWORD**)0x0041D3A5) = &(SpriteTable[0].loaded_index); // sub_41D360 ( WHMTG_SpeakNoWait )
*((DWORD**)0x0042459C) = &(SpriteTable[0].loaded_index); // sub_424530 ( ?? )
*((DWORD**)0x0042BD81) = &(SpriteTable[0].loaded_index); // sub_42BCC0 ( ?? )
*((DWORD**)0x0042BEDD) = &(SpriteTable[0].loaded_index); // sub_42BCC0 ( ?? )
*((DWORD**)0x0042E238) = &(SpriteTable[0].loaded_index); // sub_42E1E0 ( ?? )
*((DWORD**)0x0042B705) = &(SpriteTable[0].loaded_index); // sub_42B6A0 ( ?? )
*((WORD**) 0x0047FE01) = offset_table; // sub_47FDF0 ( get sprite type info by id...? )
*((DWORD*)0x00526D24) = (DWORD)&Loaded_Sprite_Array;
*((DWORD*)0x00526D2C) = (DWORD)&Loaded_Banner_Array;
*((DWORD*)0x00409F11) = (DWORD)troop_book_unit_pic;
*((DWORD*)0x00409F52) = (DWORD)troop_book_unit_pic;
*((DWORD*)0x0040C0CC) = (DWORD)troop_book_unit_pic;
*((DWORD*)0x0040C108) = (DWORD)troop_book_unit_pic;
*((DWORD*)0x00409DE3) = (DWORD)troop_book_leader_pic;
*((DWORD*)0x0040BD24) = (DWORD)troop_book_leader_pic;
*((DWORD*)0x00409EC2) = (DWORD)unit_name;
*((DWORD*)0x0040B876) = (DWORD)unit_name;
*((DWORD*)0x0040BE4A) = (DWORD)unit_name;
*((DWORD*)0x0040C20F) = (DWORD)unit_name;
*((DWORD*)0x0040C461) = (DWORD)unit_name;
*((DWORD*)0x0040D213) = (DWORD)unit_name;
*((DWORD*)0x0040D9C3) = (DWORD)unit_name;
*((DWORD*)0x0041985C) = (DWORD)unit_name;
*((DWORD*)0x0041A5E4) = (DWORD)unit_name;
*((DWORD*)0x0041B1B0) = (DWORD)unit_name;
*((DWORD*)0x0041B4CA) = (DWORD)unit_name;
*((DWORD*)0x0042671A) = (DWORD)unit_name;
*((DWORD*)0x004286E1) = (DWORD)unit_name;
*((DWORD*)0x00409E63) = (DWORD)troop_book_unit_desc;
*((DWORD*)0x0040BDE0) = (DWORD)troop_book_unit_desc;
*((DWORD*)0x004099B5) = (DWORD)troop_book_banner_index;
*((DWORD*)0x00409DAD) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040B83D) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040BCE0) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040C1D5) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040C413) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040D97C) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0040E1E9) = (DWORD)troop_book_banner_index;
*((DWORD*)0x004192A4) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0041981E) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0041A39E) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0041A5A6) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0041B172) = (DWORD)troop_book_banner_index;
*((DWORD*)0x0041B48C) = (DWORD)troop_book_banner_index;
*((DWORD*)0x00425B26) = (DWORD)troop_book_banner_index;
*((DWORD*)0x00426DEB) = (DWORD)troop_book_banner_index;
the_end:
VirtualFree( readbuf, 0, MEM_RELEASE );
return;
}
