- All component locks moved to memory

[public/netxms.git] / src / server / core / config.cpp

/* 
** Project X - Network Management System
** Copyright (C) 2003 Victor Kirhenshtein
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
**
** $module: config.cpp
**
**/

#include "nms_core.h"


//
// Help text
//

static char help_text[]="NetXMS Server Version " NETXMS_VERSION_STRING "\n"
                        "Copyright (c) 2003 SecurityProjects.org\n\n"
                        "Usage: netxmsd [<options>] <command>\n\n"
                        "Valid options are:\n"
                        "   --config <file>     : Set non-default configuration file\n"
                        "                       : Default is " DEFAULT_CONFIG_FILE "\n"
                        "   --debug-all         : Turn on all possible debug output\n"
                        "   --debug-actions     : Print debug information for event actions.\n"
                        "   --debug-cscp        : Print client-server communication protocol debug\n"
                        "                       : information to console.\n"
                        "   --debug-dc          : Print data collection debug information to console.\n"
                        "   --debug-discovery   : Print network discovery debug information to console.\n"
                        "   --debug-events      : Print events to console.\n"
                        "   --debug-housekeeper : Print debug information for housekeeping thread.\n"
                        "   --debug-locks       : Print debug information about component locking.\n"
                        "   --debug-snmp        : Print SNMP debug information.\n"
                        "   --dump-sql          : Dump all SQL queries to log.\n"
#ifndef _WIN32
                        "   --pid-file <file>   : Specify pid file.\n"
#endif
                        "\n"
                        "Valid commands are:\n"
                        "   check-config        : Check configuration file syntax\n"
#ifdef _WIN32
                        "   install             : Install Win32 service\n"
                        "   install-events      : Install Win32 event source\n"
#endif
                        "   help                : Display help and exit\n"
#ifdef _WIN32
                        "   remove              : Remove Win32 service\n"
                        "   remove-events       : Remove Win32 event source\n"
#endif
                        "   standalone          : Run in standalone mode (not as service)\n"
                        "   version             : Display version and exit\n"
                        "\n"
                        "NOTE: All debug options will work only in standalone mode.\n\n";


//
// Load and parse configuration file
// Returns TRUE on success and FALSE on failure
//

static NX_CFG_TEMPLATE m_cfgTemplate[] =
{
   { "DBDriver", CT_STRING, 0, 0, MAX_PATH, 0, g_szDbDriver },
   { "DBDrvParams", CT_STRING, 0, 0, MAX_PATH, 0, g_szDbDrvParams },
   { "DBLogin", CT_STRING, 0, 0, MAX_DB_LOGIN, 0, g_szDbLogin },
   { "DBName", CT_STRING, 0, 0, MAX_DB_NAME, 0, g_szDbName },
   { "DBPassword", CT_STRING, 0, 0, MAX_DB_PASSWORD, 0, g_szDbPassword },
   { "DBServer", CT_STRING, 0, 0, MAX_PATH, 0, g_szDbServer },
   { "LogFailedSQLQueries", CT_BOOLEAN, 0, 0, AF_LOG_SQL_ERRORS, 0, &g_dwFlags },
   { "LogFile", CT_STRING, 0, 0, MAX_PATH, 0, g_szLogFile },
   { "", CT_END_OF_LIST, 0, 0, 0, 0, NULL }
};

BOOL LoadConfig(void)
{
   BOOL bSuccess = FALSE;

   if (IsStandalone())
      printf("Using configuration file \"%s\"\n", g_szConfigFile);

   if (NxLoadConfig(g_szConfigFile, "", m_cfgTemplate, IsStandalone()) == NXCFG_ERR_OK)
   {
      if ((!stricmp(g_szLogFile,"{EventLog}")) ||
          (!stricmp(g_szLogFile,"{syslog}")))
      {
         g_dwFlags |= AF_USE_EVENT_LOG;
      }
      else
      {
         g_dwFlags &= ~AF_USE_EVENT_LOG;
      }
      bSuccess = TRUE;
   }
   return bSuccess;
}


//
// Parse command line
// Returns TRUE on success and FALSE on failure
//

BOOL ParseCommandLine(int argc, char *argv[])
{
   int i;

   for(i = 1; i < argc; i++)
   {
      if (!strcmp(argv[i], "help"))    // Display help and exit
      {
         printf(help_text);
         return FALSE;
      }
      else if (!strcmp(argv[i], "version"))    // Display version and exit
      {
         printf("NMS Version " NETXMS_VERSION_STRING " Build of " __DATE__ "\n");
         return FALSE;
      }
      else if (!strcmp(argv[i], "--config"))  // Config file
      {
         i++;
         strcpy(g_szConfigFile, argv[i]);     // Next word should contain name of the config file
      }
#ifndef _WIN32
      else if (!strcmp(argv[i], "--pid-file"))  // PID file
      {
         i++;
         strcpy(g_szPIDFile, argv[i]);     // Next word should contain name of the PID file
      }
#endif
      else if (!strcmp(argv[i], "--debug-all"))
      {
         g_dwFlags |= AF_DEBUG_ALL;
      }
      else if (!strcmp(argv[i], "--debug-events"))
      {
         g_dwFlags |= AF_DEBUG_EVENTS;
      }
      else if (!strcmp(argv[i], "--debug-cscp"))
      {
         g_dwFlags |= AF_DEBUG_CSCP;
      }
      else if (!strcmp(argv[i], "--debug-discovery"))
      {
         g_dwFlags |= AF_DEBUG_DISCOVERY;
      }
      else if (!strcmp(argv[i], "--debug-dc"))
      {
         g_dwFlags |= AF_DEBUG_DC;
      }
      else if (!strcmp(argv[i], "--debug-locks"))
      {
         g_dwFlags |= AF_DEBUG_LOCKS;
      }
      else if (!strcmp(argv[i], "--debug-housekeeper"))
      {
         g_dwFlags |= AF_DEBUG_HOUSEKEEPER;
      }
      else if (!strcmp(argv[i], "--debug-actions"))
      {
         g_dwFlags |= AF_DEBUG_ACTIONS;
      }
      else if (!strcmp(argv[i], "--debug-snmp"))
      {
         g_dwFlags |= AF_DEBUG_SNMP;
      }
      else if (!strcmp(argv[i], "--dump-sql"))
      {
         g_dwFlags |= AF_DEBUG_SQL;
      }
      else if (!strcmp(argv[i], "check-config"))
      {
         g_dwFlags |= AF_STANDALONE;
         printf("Checking configuration file (%s):\n\n", g_szConfigFile);
         LoadConfig();
         return FALSE;
      }
      else if (!strcmp(argv[i], "standalone"))  // Run in standalone mode
      {
         g_dwFlags |= AF_STANDALONE;
         return TRUE;
      }
#ifdef _WIN32
      else if ((!strcmp(argv[i], "install"))||
               (!strcmp(argv[i], "install-events")))
      {
         char exePath[MAX_PATH], dllPath[MAX_PATH], *ptr;

         ptr = strrchr(argv[0], '\\');
         if (ptr != NULL)
            ptr++;
         else
            ptr = argv[0];

         _fullpath(exePath, ptr, 255);

         if (stricmp(&exePath[strlen(exePath)-4], ".exe"))
            strcat(exePath, ".exe");
         strcpy(dllPath, exePath);
         ptr = strrchr(dllPath, '\\');
         if (ptr != NULL)  // Shouldn't be NULL
         {
            ptr++;
            strcpy(ptr, "libnxsrv.dll");
         }

         if (!strcmp(argv[i], "install"))
            InstallService(exePath, dllPath);
         else
            InstallEventSource(dllPath);
         return FALSE;
      }
      else if (!strcmp(argv[i], "remove"))
      {
         RemoveService();
         return FALSE;
      }
      else if (!strcmp(argv[i], "remove-events"))
      {
         RemoveEventSource();
         return FALSE;
      }
      else if (!strcmp(argv[i], "start"))
      {
         StartCoreService();
         return FALSE;
      }
      else if (!strcmp(argv[i], "stop"))
      {
         StopCoreService();
         return FALSE;
      }
#endif   /* _WIN32 */
      else
      {
         printf("ERROR: Invalid command line argument\n\n");
         return FALSE;
      }
   }

   return TRUE;
}


//
// Read string value from configuration table
//

BOOL ConfigReadStr(char *szVar, char *szBuffer, int iBufSize, const char *szDefault)
{
   DB_RESULT hResult;
   char szQuery[256];
   BOOL bSuccess = FALSE;

   strncpy(szBuffer, szDefault, iBufSize);
   if (strlen(szVar) > 127)
      return FALSE;

   sprintf(szQuery, "SELECT var_value FROM config WHERE var_name='%s'", szVar);
   hResult = DBSelect(g_hCoreDB, szQuery);
   if (hResult == 0)
      return FALSE;

   if (DBGetNumRows(hResult) > 0)
   {
      strncpy(szBuffer, DBGetField(hResult, 0, 0), iBufSize - 1);
      bSuccess = TRUE;
   }

   DBFreeResult(hResult);
   return bSuccess;
}


//
// Read integer value from configuration table
//

int ConfigReadInt(char *szVar, int iDefault)
{
   char szBuffer[64];

   if (ConfigReadStr(szVar, szBuffer, 64, ""))
      return strtol(szBuffer, NULL, 0);
   else
      return iDefault;
}


//
// Read unsigned long value from configuration table
//

DWORD ConfigReadULong(char *szVar, DWORD dwDefault)
{
   char szBuffer[64];

   if (ConfigReadStr(szVar, szBuffer, 64, ""))
      return strtoul(szBuffer, NULL, 0);
   else
      return dwDefault;
}


//
// Write string value to configuration table
//

BOOL ConfigWriteStr(char *szVar, char *szValue, BOOL bCreate)
{
   DB_RESULT hResult;
   char szQuery[1024];
   BOOL bVarExist = FALSE;

   if (strlen(szVar) > 127)
      return FALSE;

   // Check for variable existence
   sprintf(szQuery, "SELECT var_value FROM config WHERE var_name='%s'", szVar);
   hResult = DBSelect(g_hCoreDB, szQuery);
   if (hResult != 0)
   {
      if (DBGetNumRows(hResult) > 0)
         bVarExist = TRUE;
      DBFreeResult(hResult);
   }

   // Don't create non-existing variable if creation flag not set
   if (!bCreate && !bVarExist)
      return FALSE;

   // Create or update variable value
   if (bVarExist)
      sprintf(szQuery, "UPDATE config SET var_value='%s' WHERE var_name='%s'", szValue, szVar);
   else
      sprintf(szQuery, "INSERT INTO config (var_name,var_value,is_visible,"
                       "need_server_restart) VALUES ('%s','%s',0,0)", szVar, szValue);
   return DBQuery(g_hCoreDB, szQuery);
}


//
// Write integer value to configuration table
//

BOOL ConfigWriteInt(char *szVar, int iValue, BOOL bCreate)
{
   char szBuffer[64];

   sprintf(szBuffer, "%ld", iValue);
   return ConfigWriteStr(szVar, szBuffer, bCreate);
}


//
// Write unsigned long value to configuration table
//

BOOL ConfigWriteULong(char *szVar, DWORD dwValue, BOOL bCreate)
{
   char szBuffer[64];

   sprintf(szBuffer, "%lu", dwValue);
   return ConfigWriteStr(szVar, szBuffer, bCreate);
}