D:/pwsrc2.0/docs/gauthd/gmysqlclient.cpp

Go to the documentation of this file.

 
 
#include "gmysqlclient.hpp"
#include "state.hxx"
#include "timertask.h"
 
using namespace GNET;
//----------------------------------------------------------------------------------------------------------
GMysqlClient* GMysqlClient::instance = NULL; 
 
//----------------------------------------------------------------------------------------------------------
void GMysqlClient::Init(const char * ip, const int port, const char * user, const char * passwd, const char * db, const int h)
{
    int size = 0;
    
    pthread_mutex_init(&cash_mutex,NULL);
    pthread_mutex_lock(&mysql_mutex);
    this->active = false;
    this->DBMysql = NULL;
    
    memset(this->ip,0x00,20u); size = strlen(ip); if(size < 20u) memcpy(this->ip, ip, size);
    this->port = port;
    memset(this->user,0x00,128u); size = strlen(user); if(size < 128u) memcpy(this->user, user, size);
    memset(this->passwd,0x00,128u); size = strlen(passwd); if(size < 128u) memcpy(this->passwd, passwd, size);
    memset(this->db,0x00,128u); size = strlen(db); if(size < 128u) memcpy(this->db, db, size);
    this->hash = h;
    
    this->mysql_state = 0;
    this->zoneid = 0;
    this->aid = 0;
    
    this->lkmod = false;
    this->loginlog = false;
    memset(this->logoutshell,0x00,1024);
    
    this->online_user.clear();
    this->online_user.reserve(5000);
 
    this->cash_user.clear();
    this->cash_user.reserve(5000);
 
    this->antibrut_user.clear();
    this->antibrut_user.reserve(5000);
    
    pthread_mutex_init(&mysql_mutex,NULL);
    pthread_mutex_init(&online_mutex,NULL);
    pthread_mutex_init(&antibrut_mutex,NULL);
    
    pthread_mutex_unlock(&mysql_mutex);
}
//----------------------------------------------------------------------------------------------------------
void GMysqlClient::AlexPatch(const bool lkmod, const bool log, const char * shell)
{
    if(!lkmod) return;
    this->lkmod = lkmod;
    int size = strlen(shell);
    this->loginlog = log;
    if(size > 1000) return;
    memcpy(this->logoutshell,shell,size);
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::Connect()
{
    if(active) 
    return true;
    active = false;
    pthread_mutex_lock(&mysql_mutex);
 
    DBMysql = mysql_init(NULL);
    if(DBMysql != NULL)
    {
        if( !mysql_real_connect(DBMysql,ip,user,passwd,db,port,0,0) )
        {
            printf("GMysqlClient::OnAbortSession \n");
        } else {
            active = true;
            mysql_state = 1;
            printf("GMysqlClient::OnAddSession \n");
        }
    }
    pthread_mutex_unlock(&mysql_mutex);
    return active;
}
//----------------------------------------------------------------------------------------------------------
void GMysqlClient::Disconnect()
{
    pthread_mutex_lock(&mysql_mutex);
    mysql_close(DBMysql);
    active = false;
    mysql_state = 0;
    printf("GMysqlClient::OnDelSession \n");
    pthread_mutex_unlock(&mysql_mutex);
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::MysqlQuery(char * ptr, size_t len)
{
    if(mysql_state < 2 || !ptr || !ptr[0] || !len || len > 1024 )
    return false;
 
    bool res = true;
    pthread_mutex_lock(&mysql_mutex);
    while ( mysql_next_result(DBMysql) == 0 );
    query_state = mysql_real_query(DBMysql,ptr,len);
    if (query_state !=0)
    {
        std::cout << mysql_error(DBMysql) << std::endl;
        res = false;
    }
    pthread_mutex_unlock(&mysql_mutex);
    return res;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::MatrixPasswd(int &uid, char * login, char * passwd, int ipaddr)
{
    uid = 0;
    char QUERY[1024];
    sprintf(QUERY,"CALL acquireuserpasswd('%s', @userid, @passwd)",login);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    sprintf(QUERY,"SELECT @userid, @passwd");
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    
    char AUTH[64];
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    if( row && row[0] && row[1] )
    {
        uid = ATOI(row[0]);
        memcpy(AUTH,row[1],64);      
    }
    mysql_free_result(result);
    AddUser(uid,zoneid,login,ipaddr);
    pthread_mutex_unlock(&mysql_mutex);
    if(uid <= 0 || !AuthPasswd(passwd,AUTH))
    return false;
    return true;
}
//----------------------------------------------------------------------------------------------------------    
bool GMysqlClient::ClearOnlineRecord(char zoneid, char aid)
{
    mysql_state = 2;
    char QUERY[1024];
    sprintf(QUERY,"CALL clearonlinerecords(%d, %d)",zoneid, aid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    this->zoneid = zoneid;
    this->aid = aid;
    return true;
}   
//----------------------------------------------------------------------------------------------------------    
bool GMysqlClient::OnlineRecord(int uid, char aid, char &zoneid, int &zonelocalid, int &overwrite)
{
    char QUERY[1024];
    sprintf(QUERY,"SET @zoneid = %d, @aid = %d, @zonelocalid = %d",zoneid,aid,zonelocalid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    sprintf(QUERY,"CALL recordonline(%d, %d, @zoneid, @zonelocalid, @overwrite)",uid, aid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    sprintf(QUERY,"SELECT @zoneid, @zonelocalid, @overwrite");
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
 
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    if( row && row[0] && row[1] && row[2] )
    {
        zoneid = ATOI(row[0]);
        zonelocalid = ATOI(row[1]);
        overwrite = ATOI(row[2]);
    }
    mysql_free_result(result);
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}   
//----------------------------------------------------------------------------------------------------------    
bool GMysqlClient::OnfflineRecord(int uid, char aid, char &zoneid, int &zonelocalid, int &overwrite)
{
    char QUERY[1024];
    sprintf(QUERY,"SET @zoneid = %d, @aid = %d, @zonelocalid = %d",zoneid,aid,zonelocalid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    sprintf(QUERY,"CALL recordoffline(%d, %d, @zoneid, @zonelocalid, @overwrite)",uid, aid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    sprintf(QUERY,"SELECT @zoneid, @zonelocalid, @overwrite");
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
 
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    if( row && row[0] && row[1] && row[2] )
    {
        zoneid = ATOI(row[0]);
        zonelocalid = ATOI(row[1]);
        overwrite = ATOI(row[2]);
    }
    DelUser(uid);
    mysql_free_result(result);
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}   
//----------------------------------------------------------------------------------------------------------    
bool GMysqlClient::UserCreatime(int uid, int &timestamp)
{
    char QUERY[1024];
    sprintf(QUERY,"SELECT UNIX_TIMESTAMP(creatime) FROM users WHERE ID=%d",uid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
 
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    if( row && row[0] )
    {
        timestamp = ATOI(row[0]);
    }
    mysql_free_result(result);
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::UserGMPrivilege(int uid, char zoneid, bool &IsGM)
{
    IsGM = false;
    char QUERY[1024];
    sprintf(QUERY,"SELECT rid FROM auth WHERE userid=%d AND zoneid=%d",uid,zoneid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    if( row ) IsGM = true;
    mysql_free_result(result);
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::QueryGMPrivilege(int uid, char zoneid, unsigned char * gm_auth, int &count)
{
    count = 0;
    char QUERY[1024];
    sprintf(QUERY,"SELECT rid FROM auth WHERE userid=%d AND zoneid=%d",uid,zoneid);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    while (( row = mysql_fetch_row(result)) != NULL) 
    {
        gm_auth[count++] = ATOI(row[0]);
    }
    mysql_free_result(result);
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::DeleteErrorRow(int status, int uid, char * creatime)
{
    char QUERY[1024];
    sprintf(QUERY,"DELETE FROM usecashnow WHERE status=%d AND userid=%d AND creatime='%s'",status, uid, creatime);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) )
    return false;
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::GetUseCashNow(int status, int &count)
{
    char QUERY[1024];
    sprintf(QUERY,"SELECT * FROM usecashnow WHERE status=%d",status);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    
    pthread_mutex_lock(&mysql_mutex);
    ClearCash();
    result = mysql_store_result(DBMysql);
    while (( row = mysql_fetch_row(result)) != NULL)
    {
        AddCashUser( ATOI(row[0]), ATOI(row[1]), ATOI(row[2]), ATOI(row[4]), ATOI(row[5]), row[7] );
    }
    count = cash_user.size();
    pthread_mutex_unlock(&mysql_mutex);
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::SetUseCashNow(int status, int sn, int uid, char * creatime)
{
    char QUERY[1024];
    sprintf(QUERY,"SELECT * FROM usecashnow WHERE status=%d AND userid=%d AND creatime='%s'",status, uid, creatime);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    pthread_mutex_lock(&mysql_mutex);
    result = mysql_store_result(DBMysql);
    row = mysql_fetch_row(result);
    pthread_mutex_unlock(&mysql_mutex);
    if( !row )
    {
        sprintf(QUERY,"UPDATE usecashnow SET status=%d , sn=%d WHERE userid=%d AND creatime='%s'",status, sn, uid, creatime);
        if ( !MysqlQuery(QUERY,strlen(QUERY)) ) return false;
    }
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::DelUseCashNow(int status, int uid, int sn)
{
    char QUERY[1024];
    sprintf(QUERY,"DELETE FROM usecashnow WHERE userid=%d AND sn=%d AND status=%d",uid, sn, status);    
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) 
    return false;
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::AddCashLog(int uid, int zoneid, int sn, int aid, int point, int cash, int status, char * creatime)
{
    char QUERY[1024];
    sprintf(QUERY,"REPLACE INTO usecashlog VALUES (%d, %d, %d, %d, %d, %d, %d, '%s', NOW())",uid,zoneid,sn,aid,point,cash,status,creatime);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) ) 
    return false;
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::AlexAddLogin(int uid, int action)
{
    if(!lkmod || !loginlog)
    return false;
    char QUERY[1024];
    char IP[20];
    ONLINE_USER * user = GetUserData(uid);
    if(!user) return false;
    AddrToStr(user->ipaddr,IP);
    sprintf(QUERY,"REPLACE INTO login_log (data, ip, userid, login, action) VALUES (NOW(),'%s',%d,'%s',%d)", IP, uid , user->login, action);
    if ( !MysqlQuery(QUERY,strlen(QUERY)) )
    return false;
    return true;
}
//----------------------------------------------------------------------------------------------------------
bool GMysqlClient::AlexShell(int uid)
{
    if(!lkmod || !logoutshell || !logoutshell[0]) 
    return false;
    char QUERY[1024];
    sprintf(QUERY,logoutshell,uid);
    system(QUERY);
    return true;
}
//----------------------------------------------------------------------------------------------------------