unit_info.cpp

#include "unit_info.h"
#include "srvmgr.h"
#include "lib\utils.hpp"

std::vector<Unit> Units;

void UI_Clear(Unit& unit, byte* cunit, uint32_t type)
{
    unit.Type = type;
    unit.Class = cunit;
    unit.Invalid = (unit.Class && *(byte**)(unit.Class+0x14));
    unit.VisibleFlags = 0;
    unit.VisibleFlagsLast = 0;
}

void UI_Create(byte* cunit, uint32_t type)
{
    bool already_exists = false;
    for(std::vector<Unit>::iterator it = Units.begin();
        it != Units.end(); ++it)
    {
        Unit& unit = (*it);
        if(unit.Class == cunit)
            return;
    }
    
    if(already_exists) return;

    Unit unit;
    UI_Clear(unit, cunit, type);
    Units.push_back(unit);
}

void UI_Delete(byte* cunit)
{
    for(std::vector<Unit>::iterator it = Units.begin();
        it != Units.end(); ++it)
    {
        Unit& unit = (*it);
        if(unit.Class == cunit)
        {
            Units.erase(it);
            return;
        }
    }
}

void UI_Destruct(byte* cunit)
{
    if(cunit)
    {
        for(std::vector<Unit>::iterator it = Units.begin();
            it != Units.end(); ++it)
        {
            Unit& unit = (*it);
            if(unit.Class == cunit)
            {
                uint32_t size = 0x254;
                if(unit.Type == UnitType_Monster) size = 0x208;
                if(!IsBadReadPtr(unit.Class, size))
                {
                    __asm
                    {
                        push    1
                        mov        ecx, [cunit]
                        mov        edx, [ecx]
                        call    dword ptr [edx+4]
                    }
                }
            }
        }
    }
}

void UI_Tick()
{
    //...
}

Unit* UI_Get(byte* cunit)
{
    if(!cunit) return NULL;
    for(std::vector<Unit>::iterator it = Units.begin();
        it != Units.end(); ++it)
    {
        Unit& unit = (*it);
        if(unit.Class == cunit)
            return &unit;
    }

    return NULL;
}

void UI_Reset()
{
    Units.clear();
}