src/mangosd/CliRunnable.cpp

   1 /*
   2  * Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 /// \addtogroup mangosd
  20 /// @{
  21 /// \file
  22
  23 #include "Common.h"
  24 #include "Language.h"
  25 #include "Log.h"
  26 #include "World.h"
  27 #include "ScriptCalls.h"
  28 #include "ObjectMgr.h"
  29 #include "WorldSession.h"
  30 #include "Config/ConfigEnv.h"
  31 #include "Util.h"
  32 #include "AccountMgr.h"
  33 #include "CliRunnable.h"
  34 #include "MapManager.h"
  35 #include "Player.h"
  36 #include "Chat.h"
  37
  38 void utf8print(const char* str)
  39 {
  40 #if PLATFORM == PLATFORM_WINDOWS
  41     wchar_t wtemp_buf[6000];
  42     size_t wtemp_len = 6000-1;
  43     if(!Utf8toWStr(str,strlen(str),wtemp_buf,wtemp_len))
  44         return;
  45
  46     char temp_buf[6000];
  47     CharToOemBuffW(&wtemp_buf[0],&temp_buf[0],wtemp_len+1);
  48     printf(temp_buf);
  49 #else
  50     printf(str);
  51 #endif
  52 }
  53
  54 /// Delete a user account and all associated characters in this realm
  55 /// \todo This function has to be enhanced to respect the login/realm split (delete char, delete account chars in realm, delete account chars in realm then delete account
  56 bool ChatHandler::HandleAccountDeleteCommand(const char* args)
  57 {
  58     if(!*args)
  59         return false;
  60
  61     ///- Get the account name from the command line
  62     char *account_name_str=strtok ((char*)args," ");
  63     if (!account_name_str)
  64         return false;
  65
  66     std::string account_name = account_name_str;
  67     if(!AccountMgr::normilizeString(account_name))
  68     {
  69         PSendSysMessage(LANG_ACCOUNT_NOT_EXIST,account_name.c_str());
  70         SetSentErrorMessage(true);
  71         return false;
  72     }
  73
  74     uint32 account_id = accmgr.GetId(account_name);
  75     if(!account_id)
  76     {
  77         PSendSysMessage(LANG_ACCOUNT_NOT_EXIST,account_name.c_str());
  78         SetSentErrorMessage(true);
  79         return false;
  80     }
  81
  82     /// Commands not recommended call from chat, but support anyway
  83     /// can delete only for account with less security
  84     /// This is also reject self apply in fact
  85     if(HasLowerSecurityAccount (NULL,account_id,true))
  86         return false;
  87
  88     AccountOpResult result = accmgr.DeleteAccount(account_id);
  89     switch(result)
  90     {
  91         case AOR_OK:
  92             PSendSysMessage(LANG_ACCOUNT_DELETED,account_name.c_str());
  93             break;
  94         case AOR_NAME_NOT_EXIST:
  95             PSendSysMessage(LANG_ACCOUNT_NOT_EXIST,account_name.c_str());
  96             SetSentErrorMessage(true);
  97             return false;
  98         case AOR_DB_INTERNAL_ERROR:
  99             PSendSysMessage(LANG_ACCOUNT_NOT_DELETED_SQL_ERROR,account_name.c_str());
 100             SetSentErrorMessage(true);
 101             return false;
 102         default:
 103             PSendSysMessage(LANG_ACCOUNT_NOT_DELETED,account_name.c_str());
 104             SetSentErrorMessage(true);
 105             return false;
 106     }
 107
 108     return true;
 109 }
 110
 111 bool ChatHandler::HandleCharacterDeleteCommand(const char* args)
 112 {
 113     if(!*args)
 114         return false;
 115
 116     char *character_name_str = strtok((char*)args," ");
 117     if(!character_name_str)
 118         return false;
 119
 120     std::string character_name = character_name_str;
 121     if(!normalizePlayerName(character_name))
 122         return false;
 123
 124     uint64 character_guid;
 125     uint32 account_id;
 126
 127     Player *player = objmgr.GetPlayer(character_name.c_str());
 128     if(player)
 129     {
 130         character_guid = player->GetGUID();
 131         account_id = player->GetSession()->GetAccountId();
 132         player->GetSession()->KickPlayer();
 133     }
 134     else
 135     {
 136         character_guid = objmgr.GetPlayerGUIDByName(character_name);
 137         if(!character_guid)
 138         {
 139             PSendSysMessage(LANG_NO_PLAYER,character_name.c_str());
 140             SetSentErrorMessage(true);
 141             return false;
 142         }
 143
 144         account_id = objmgr.GetPlayerAccountIdByGUID(character_guid);
 145     }
 146
 147     std::string account_name;
 148     accmgr.GetName (account_id,account_name);
 149
 150     Player::DeleteFromDB(character_guid, account_id, true);
 151     PSendSysMessage(LANG_CHARACTER_DELETED,character_name.c_str(),GUID_LOPART(character_guid),account_name.c_str(), account_id);
 152     return true;
 153 }
 154
 155 /// Exit the realm
 156 bool ChatHandler::HandleServerExitCommand(const char* args)
 157 {
 158     SendSysMessage(LANG_COMMAND_EXIT);
 159     World::StopNow(SHUTDOWN_EXIT_CODE);
 160     return true;
 161 }
 162
 163 /// Display info on users currently in the realm
 164 bool ChatHandler::HandleAccountOnlineListCommand(const char* args)
 165 {
 166     ///- Get the list of accounts ID logged to the realm
 167     QueryResult *resultDB = CharacterDatabase.Query("SELECT name,account FROM characters WHERE online > 0");
 168     if (!resultDB)
 169         return true;
 170
 171     ///- Display the list of account/characters online
 172     SendSysMessage("=====================================================================");
 173     SendSysMessage(LANG_ACCOUNT_LIST_HEADER);
 174     SendSysMessage("=====================================================================");
 175
 176     ///- Circle through accounts
 177     do
 178     {
 179         Field *fieldsDB = resultDB->Fetch();
 180         std::string name = fieldsDB[0].GetCppString();
 181         uint32 account = fieldsDB[1].GetUInt32();
 182
 183         ///- Get the username, last IP and GM level of each account
 184         // No SQL injection. account is uint32.
 185         //                                                      0         1        2        3
 186         QueryResult *resultLogin = loginDatabase.PQuery("SELECT username, last_ip, gmlevel, expansion FROM account WHERE id = '%u'",account);
 187
 188         if(resultLogin)
 189         {
 190             Field *fieldsLogin = resultLogin->Fetch();
 191             PSendSysMessage("|%15s| %20s | %15s |%4d|%5d|",
 192                 fieldsLogin[0].GetString(),name.c_str(),fieldsLogin[1].GetString(),fieldsLogin[2].GetUInt32(),fieldsLogin[3].GetUInt32());
 193
 194             delete resultLogin;
 195         }
 196         else
 197             PSendSysMessage(LANG_ACCOUNT_LIST_ERROR,name.c_str());
 198
 199     }while(resultDB->NextRow());
 200
 201     delete resultDB;
 202
 203     SendSysMessage("=====================================================================");
 204     return true;
 205 }
 206
 207 /// Create an account
 208 bool ChatHandler::HandleAccountCreateCommand(const char* args)
 209 {
 210     if(!*args)
 211         return false;
 212
 213     ///- %Parse the command line arguments
 214     char *szAcc = strtok((char*)args, " ");
 215     char *szPassword = strtok(NULL, " ");
 216     if(!szAcc || !szPassword)
 217         return false;
 218
 219     // normilized in accmgr.CreateAccount
 220     std::string account_name = szAcc;
 221     std::string password = szPassword;
 222
 223     AccountOpResult result = accmgr.CreateAccount(account_name, password);
 224     switch(result)
 225     {
 226         case AOR_OK:
 227             PSendSysMessage(LANG_ACCOUNT_CREATED,account_name.c_str());
 228             break;
 229         case AOR_NAME_TOO_LONG:
 230             SendSysMessage(LANG_ACCOUNT_TOO_LONG);
 231             SetSentErrorMessage(true);
 232             return false;
 233         case AOR_NAME_ALREDY_EXIST:
 234             SendSysMessage(LANG_ACCOUNT_ALREADY_EXIST);
 235             SetSentErrorMessage(true);
 236             return false;
 237         case AOR_DB_INTERNAL_ERROR:
 238             PSendSysMessage(LANG_ACCOUNT_NOT_CREATED_SQL_ERROR,account_name.c_str());
 239             SetSentErrorMessage(true);
 240             return false;
 241         default:
 242             PSendSysMessage(LANG_ACCOUNT_NOT_CREATED,account_name.c_str());
 243             SetSentErrorMessage(true);
 244             return false;
 245     }
 246
 247     return true;
 248 }
 249
 250 /// Set the level of logging
 251 bool ChatHandler::HandleServerSetLogLevelCommand(const char *args)
 252 {
 253     if(!*args)
 254         return false;
 255
 256     char *NewLevel = strtok((char*)args, " ");
 257     if (!NewLevel)
 258         return false;
 259
 260     sLog.SetLogLevel(NewLevel);
 261     return true;
 262 }
 263
 264 /// @}
 265
 266 #ifdef linux
 267 // Non-blocking keypress detector, when return pressed, return 1, else always return 0
 268 int kb_hit_return()
 269 {
 270     struct timeval tv;
 271     fd_set fds;
 272     tv.tv_sec = 0;
 273     tv.tv_usec = 0;
 274     FD_ZERO(&fds);
 275     FD_SET(STDIN_FILENO, &fds);
 276     select(STDIN_FILENO+1, &fds, NULL, NULL, &tv);
 277     return FD_ISSET(STDIN_FILENO, &fds);
 278 }
 279 #endif
 280
 281 /// %Thread start
 282 void CliRunnable::run()
 283 {
 284     ///- Init new SQL thread for the world database (one connection call enough)
 285     WorldDatabase.ThreadStart();                                // let thread do safe mySQL requests
 286
 287     char commandbuf[256];
 288
 289     ///- Display the list of available CLI functions then beep
 290     sLog.outString();
 291
 292     if(sConfig.GetBoolDefault("BeepAtStart", true))
 293         printf("\a");                                       // \a = Alert
 294
 295     // print this here the first time
 296     // later it will be printed after command queue updates
 297     printf("mangos>");
 298
 299     ///- As long as the World is running (no World::m_stopEvent), get the command line and handle it
 300     while (!World::IsStopped())
 301     {
 302         fflush(stdout);
 303         #ifdef linux
 304         while (!kb_hit_return() && !World::IsStopped())
 305             // With this, we limit CLI to 10commands/second
 306             usleep(100);
 307         if (World::IsStopped())
 308             break;
 309         #endif
 310         char *command_str = fgets(commandbuf,sizeof(commandbuf),stdin);
 311         if (command_str != NULL)
 312         {
 313             for(int x=0;command_str[x];x++)
 314                 if(command_str[x]=='\r'||command_str[x]=='\n')
 315             {
 316                 command_str[x]=0;
 317                 break;
 318             }
 319
 320
 321             if(!*command_str)
 322             {
 323                 printf("mangos>");
 324                 continue;
 325             }
 326
 327             std::string command;
 328             if(!consoleToUtf8(command_str,command))         // convert from console encoding to utf8
 329             {
 330                 printf("mangos>");
 331                 continue;
 332             }
 333
 334             sWorld.QueueCliCommand(&utf8print,command.c_str());
 335         }
 336         else if (feof(stdin))
 337         {
 338             World::StopNow(SHUTDOWN_EXIT_CODE);
 339         }
 340     }
 341
 342     ///- End the database thread
 343     WorldDatabase.ThreadEnd();                                  // free mySQL thread resources
 344 }