util.cpp File Reference

#include "eve-common.h"

Include dependency graph for util.cpp:

Go to the source code of this file.

Namespaces
	Utils

Functions
int32	ascent_roundf (float v)
int32	ascent_roundd (double v)
vector< string >	StrSplit (const string &src, const string &sep)
time_t	convTimePeriod (uint32 dLength, char dType)
void	MakeIntString (char *buf, int num)
void	MakeIntStringNoZero (char *buf, int num)
string	ConvertTimeStampToString (uint32 timestamp)
string	ConvertTimeStampToStringNC (uint32 timestamp)
string	ConvertTimeStampToDataTime (uint32 timestamp)
bool	ParseCIDRBan (unsigned int IP, unsigned int Mask, unsigned int MaskBits)
unsigned int	MakeIP (const char *str)
int	get_tokens (const char *szInput, char **pOutput, int iMaxCount, char cSeperator)
bool	CheckIPs (const char *szIPList)

Variables
const char *	szDayNames [7]
const char *	szMonthNames [12]

Function Documentation

int32 ascent_roundd

(

double

v

)

Definition at line 34 of file util.cpp.

{
    double int_part = floor(v);
    double float_part = v - int_part;
    if( float_part > 0.5f )
        return int32(int_part + 1.0f);
    else
        return int32(int_part);
}

int32 ascent_roundf

(

float

v

)

Definition at line 24 of file util.cpp.

References float2int32().

{
    float int_part = floorf(v);
    float float_part = v - int_part;
    if( float_part > 0.5f )
        return float2int32(int_part + 1.0f);
    else
        return float2int32(int_part);
}

Here is the call graph for this function:

bool CheckIPs

(

const char *

szIPList

)

Definition at line 545 of file util.cpp.

References get_tokens(), ParseCIDRBan(), and strdup.

{
#ifdef HAVE_WINDOWS_H
    char* ip_string = strdup(szIPList);

    char* tokens[30];
    int count = get_tokens(ip_string, (char**)tokens, 30, ':');

    unsigned int ips[30];
    int masks[30];
    for(int i = 0; i < count; ++i)
    {
        char* t = strchr(tokens[i], '/');
        if( t != NULL )
        {
            *t = 0;
            masks[i] = atoi(t+1);
        }
        else
            masks[i] = 32;

        ips[i] = inet_addr(tokens[i]);
    }

    char computer_name[100] = {0};
    DWORD computer_name_len = 100;
    GetComputerNameA(computer_name, &computer_name_len);

    hostent* ent = gethostbyname(computer_name);
    if( ent == NULL )
    {
        free(ip_string);
        return false;
    }

    for( int i = 0; ent->h_addr_list[i] != NULL; ++i )
    {
        bool pass = false;

        for( int j = 0; j < count; ++j )
        {
            if( ParseCIDRBan(*(unsigned int*)ent->h_addr_list[i], ips[j], masks[j]) )
            {
                pass = true;
                break;
            }
        }

        if( !pass )
        {
            free(ip_string);
            return false;
        }
    }

    free(ip_string);
#endif /* HAVE_WINDOWS_H */

    return true;
}

Here is the call graph for this function:

string ConvertTimeStampToDataTime

(

uint32

timestamp

)

Definition at line 391 of file util.cpp.

References MakeIntString().

{
    char szTempBuf[100];
    time_t t = (time_t)timestamp;
    tm * pTM = localtime(&t);

    string szResult;
    szResult += szDayNames[pTM->tm_wday];
    szResult += ", ";

    MakeIntString(szTempBuf, pTM->tm_mday);
    szResult += szTempBuf;
    szResult += " ";

    szResult += szMonthNames[pTM->tm_mon];
    szResult += " ";

    MakeIntString(szTempBuf, pTM->tm_year+1900);
    szResult += szTempBuf;
    szResult += ", ";
    MakeIntString(szTempBuf, pTM->tm_hour);
    szResult += szTempBuf;
    szResult += ":";
    MakeIntString(szTempBuf, pTM->tm_min);
    szResult += szTempBuf;
    szResult += ":";
    MakeIntString(szTempBuf, pTM->tm_sec);
    szResult += szTempBuf;

    return szResult;
}

Here is the call graph for this function:

string ConvertTimeStampToString

(

uint32

timestamp

)

Definition at line 209 of file util.cpp.

References MakeIntStringNoZero().

{
    int seconds = (int)timestamp;
    int mins=0;
    int hours=0;
    int days=0;
    int months=0;
    int years=0;
    if(seconds >= 60)
    {
        mins = seconds / 60;
        if(mins)
        {
            seconds -= mins*60;
            if(mins >= 60)
            {
                hours = mins / 60;
                if(hours)
                {
                    mins -= hours*60;
                    if(hours >= 24)
                    {
                        days = hours/24;
                        if(days)
                        {
                            hours -= days*24;
                            if(days >= 30)
                            {
                                months = days / 30;
                                if(months)
                                {
                                    days -= months*30;
                                    if(months >= 12)
                                    {
                                        years = months / 12;
                                        if(years)
                                        {
                                            months -= years*12;
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    char szTempBuf[100];
    string szResult;

    if(years) {
        MakeIntStringNoZero(szTempBuf, years);
        szResult += szTempBuf;
        szResult += " years, ";
    }

    if(months) {
        MakeIntStringNoZero(szTempBuf, months);
        szResult += szTempBuf;
        szResult += " months, ";
    }

    if(days) {
        MakeIntStringNoZero(szTempBuf, days);
        szResult += szTempBuf;
        szResult += " days, ";
    }

    if(hours) {
        MakeIntStringNoZero(szTempBuf, hours);
        szResult += szTempBuf;
        szResult += " hours, ";
    }

    if(mins) {
        MakeIntStringNoZero(szTempBuf, mins);
        szResult += szTempBuf;
        szResult += " minutes, ";
    }

    if(seconds) {
        MakeIntStringNoZero(szTempBuf, seconds);
        szResult += szTempBuf;
        szResult += " seconds";
    }

    return szResult;
}

Here is the call graph for this function:

string ConvertTimeStampToStringNC

(

uint32

timestamp

)

Definition at line 300 of file util.cpp.

References MakeIntStringNoZero().

{
    int seconds = (int)timestamp;
    int mins=0;
    int hours=0;
    int days=0;
    int months=0;
    int years=0;
    if(seconds >= 60)
    {
        mins = seconds / 60;
        if(mins)
        {
            seconds -= mins*60;
            if(mins >= 60)
            {
                hours = mins / 60;
                if(hours)
                {
                    mins -= hours*60;
                    if(hours >= 24)
                    {
                        days = hours/24;
                        if(days)
                        {
                            hours -= days*24;
                            if(days >= 30)
                            {
                                months = days / 30;
                                if(months)
                                {
                                    days -= months*30;
                                    if(months >= 12)
                                    {
                                        years = months / 12;
                                        if(years)
                                        {
                                            months -= years*12;
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    char szTempBuf[100];
    string szResult;

    if(years) {
        MakeIntStringNoZero(szTempBuf, years);
        szResult += szTempBuf;
        szResult += " yrs ";
    }

    if(months) {
        MakeIntStringNoZero(szTempBuf, months);
        szResult += szTempBuf;
        szResult += " months ";
    }

    if(days) {
        MakeIntStringNoZero(szTempBuf, days);
        szResult += szTempBuf;
        szResult += " days ";
    }

    if(hours) {
        MakeIntStringNoZero(szTempBuf, hours);
        szResult += szTempBuf;
        szResult += " hrs ";
    }

    if(mins) {
        MakeIntStringNoZero(szTempBuf, mins);
        szResult += szTempBuf;
        szResult += " mins ";
    }

    if(seconds) {
        MakeIntStringNoZero(szTempBuf, seconds);
        szResult += szTempBuf;
        szResult += " secs";
    }

    return szResult;
}

Here is the call graph for this function:

time_t convTimePeriod	(	uint32	dLength,
		char	dType
	)

Definition at line 96 of file util.cpp.

{
    time_t rawtime = 0;
    if (dLength == 0)
        return rawtime;
    struct tm * ti = localtime( &rawtime );
    switch(dType)
    {
        case 'h':        // hours
            ti->tm_hour += dLength;
            break;
        case 'd':        // days
            ti->tm_mday += dLength;
            break;
        case 'w':        // weeks
            ti->tm_mday += 7 * dLength;
            break;
        case 'm':        // months
            ti->tm_mon += dLength;
            break;
        case 'y':        // years
            // are leap years considered ? do we care ?
            ti->tm_year += dLength;
            break;
        default:        // minutes
            ti->tm_min += dLength;
            break;
    }
    return mktime(ti);
}

int get_tokens	(	const char *	szInput,
		char **	pOutput,
		int	iMaxCount,
		char	cSeperator
	)

Definition at line 491 of file util.cpp.

Referenced by CheckIPs().

{
    char* p;
    char* q;
    int count = 0;

    p = (char*)szInput;
    q = p;

    for(;;)
    {
        // get some tokenz!
        while(*p != '\0' && *p != cSeperator)
        {
            ++p;
        }

        // add token
        if( *p == 0 )
        {
            // end of string.
            // do we have data?
            if( p != q )
                pOutput[count++] = q;

            // bai.
            break;
        }

        *p = 0;
        ++p;

        // set output pointer
        pOutput[count++] = q;

        // skip while bytes are the same as the seperator
        while(*p == cSeperator)
            ++p;

        // reset q pointer
        q = p;

        // make sure we're not at EOL
        if( *p == '\0' )
            break;

        // make sure we don't overflow
        if( count == iMaxCount )
            break;
    }

    return count;
}

Here is the caller graph for this function:

void MakeIntString	(	char *	buf,
		int	num
	)

Definition at line 194 of file util.cpp.

References sprintf().

Referenced by ConvertTimeStampToDataTime().

{
    if(num<10) {
        buf[0] = '0';
        sprintf(&buf[1], "%u", num);
    } else {
        sprintf(buf,"%u",num);
    }
}

Here is the call graph for this function:

Here is the caller graph for this function:

void MakeIntStringNoZero	(	char *	buf,
		int	num
	)

Definition at line 204 of file util.cpp.

References sprintf().

Referenced by ConvertTimeStampToString(), and ConvertTimeStampToStringNC().

{
    sprintf(buf,"%u",num);
}

Here is the call graph for this function:

Here is the caller graph for this function:

unsigned int MakeIP

(

const char *

str

)

Definition at line 478 of file util.cpp.

{
    unsigned int bytes[4];
    unsigned int res;
    if( sscanf(str, "%u.%u.%u.%u", &bytes[0], &bytes[1], &bytes[2], &bytes[3]) != 4 )
    {
        return 0;
    }

    res = bytes[0] | (bytes[1] << 8) | (bytes[2] << 16) | (bytes[3] << 24);
    return res;
}

bool ParseCIDRBan	(	unsigned int	IP,
		unsigned int	Mask,
		unsigned int	MaskBits
	)

Definition at line 424 of file util.cpp.

Referenced by CheckIPs().

{
    // CIDR bans are a compacted form of IP / Submask
    // So 192.168.1.0/255.255.255.0 would be 192.168.1.0/24
    // IP's in the 192.168l.1.x range would be hit, others not.
    unsigned char * source_ip = (unsigned char*)&IP;
    unsigned char * mask = (unsigned char*)&Mask;
    int full_bytes = MaskBits / 8;
    int leftover_bits = MaskBits % 8;
    //int byte;

    // sanity checks for the data first
    if( MaskBits > 32 )
    {
        return false;
    }

    // this is the table for comparing leftover bits
    static const unsigned char leftover_bits_compare[9] = {
        0x00,            // 00000000
        0x80,            // 10000000
        0xC0,            // 11000000
        0xE0,            // 11100000
        0xF0,            // 11110000
        0xF8,            // 11111000
        0xFC,            // 11111100
        0xFE,            // 11111110
        0xFF,            // 11111111 - This one isn't used
    };

    // if we have any full bytes, compare them with memcpy
    if( full_bytes > 0 )
    {
        if( memcmp( source_ip, mask, full_bytes ) != 0 )
        {
            return false;
        }
    }

    // compare the left over bits
    if( leftover_bits > 0 )
    {
        if( ( source_ip[full_bytes] & leftover_bits_compare[leftover_bits] ) !=
            ( mask[full_bytes] & leftover_bits_compare[leftover_bits] ) )
        {
            // one of the bits does not match
            return false;
        }
    }

    // all of the bits match that were testable
    return true;
}

Here is the caller graph for this function:

vector<string> StrSplit	(	const string &	src,
		const string &	sep
	)

Definition at line 44 of file util.cpp.

{
    vector<string> r;
    string s;
    for (string::const_iterator i = src.begin(); i != src.end(); i++) {
        if (sep.find(*i) != string::npos) {
            if (s.length()) r.push_back(s);
            s = "";
        } else {
            s += *i;
        }
    }
    if (s.length()) r.push_back(s);
    return r;
}

Variable Documentation

const char* szDayNames[7]

Initial value:

= {
    "Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"
}

Definition at line 186 of file util.cpp.

const char* szMonthNames[12]

Initial value:

= {
    "January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"
}

Definition at line 190 of file util.cpp.