system.cpp

linux下的终端管理工具源码

/***************************************************************************/
/*                                                                         */
/* Copyright (c) 2005, 2006 2X Software Ltd, http://www.2X.com., NoMachine */
/*                                                                         */
/* NXCOMP, NX protocol compression and NX extensions to this software      */
/* are copyright of Nomachine. Redistribution and use of the present       */
/* software is allowed according to terms specified in the file LICENSE    */
/* which comes in the source distribution.                                 */
/*                                                                         */
/* NX and NoMachine are trademarks of Medialogic S.p.A.                    */
/*                                                                         */
/* 2X is a trademark of 2X Software Ltd.                                   */
/*                                                                         */
/* All rights reserved.                                                    */
/*                                                                         */
/***************************************************************************/

#include <stdlib.h>
#include <string.h>

#define PANIC
#define WARNING
#define TEST
#define DEBUG

#include "Logger.h"
#include "System.h"

#define NX_SYSTEM_NAME  "NX_SYSTEM"
#define NX_SYSTEM_SET   "NX_SYSTEM="
#define NX_SYSTEM_SIZE  (sizeof("NX_SYSTEM=") - 1)

#define NX_ROOT_NAME    "NX_ROOT"
#define NX_ROOT_SET     "NX_ROOT="
#define NX_ROOT_SIZE    (sizeof("NX_ROOT=") - 1)

#define NX_HOME_NAME    "NX_HOME"
#define NX_HOME_SET     "NX_HOME="
#define NX_HOME_SIZE    (sizeof("NX_HOME=") - 1)

static char _systemDir[1024];
static char _rootDir[1024];
static char _homeDir[1024];

int setSystemDir(const char *path)
{
  logTrace("System::setSystemDir");

  if (strlen(path) > 1023 - NX_SYSTEM_SIZE)
  {
    logError("System::setSystemDir", ESET(EINVAL));

    return -1;
  }

  strcpy(_systemDir, NX_SYSTEM_SET);
  strcat(_systemDir, path);

  putenv(_systemDir);

  logTest("System::setSystemDir", "Set @PRODUCT_NAME@ Client system "
              "directory to '%s'", _systemDir +
                  NX_SYSTEM_SIZE);

  return 1;
}

int setRootDir(const char *path)
{
  logTrace("System::setRootDir");

  if (strlen(path) > 1023 - NX_ROOT_SIZE)
  {
    logError("System::setRootDir", ESET(EINVAL));

    return -1;
  }

  strcpy(_rootDir, NX_ROOT_SET);
  strcat(_rootDir, path);

  putenv(_rootDir);

  logTest("System::setRootDir", "Set @PRODUCT_NAME@ Client root "
              "directory to '%s'", _rootDir +
                  NX_ROOT_SIZE);

  return 1;
}

extern int setHomeDir(const char *path)
{
  logTrace("System::setHomeDir");

  if (strlen(path) > 1023 - NX_HOME_SIZE)
  {
    logError("System::setHomeDir", ESET(EINVAL));

    return -1;
  }

  strcpy(_homeDir, NX_HOME_SET);
  strcat(_homeDir, path);

  putenv(_homeDir);

  logTest("System::setHomeDir", "Set @PRODUCT_NAME@ Client home "
              "directory to '%s'", _homeDir +
                  NX_HOME_SIZE);

  return 1;
}

const char *getSystemDir()
{
  logTrace("System::getSystemDir");

  if (*_systemDir != '\0')
  {
    return _systemDir + NX_SYSTEM_SIZE;
  }

  const char *path = getenv(NX_SYSTEM_NAME);

  if (path == NULL)
  {
    return NULL;
  }

  setSystemDir(path);

  logTest("System::getSystemDir", "Got @PRODUCT_NAME@ Client system "
              "directory '%s'", _systemDir +
                  NX_SYSTEM_SIZE);

  return _systemDir + NX_SYSTEM_SIZE;
}

const char *getRootDir()
{
  logTrace("System::getRootDir");

  if (*_rootDir != '\0')
  {
    return _rootDir + NX_ROOT_SIZE;
  }

  const char *path = getenv(NX_ROOT_NAME);

  if (path == NULL)
  {
    return NULL;
  }

  setRootDir(path);

  logTest("System::getRootDir", "Got @PRODUCT_NAME@ Client root "
              "directory '%s'", _rootDir +
                  NX_ROOT_SIZE);

  return _rootDir + NX_ROOT_SIZE;
}

const char *getHomeDir()
{
  logTrace("System::getHomeDir");

  if (*_homeDir != '\0')
  {
    return _homeDir + NX_HOME_SIZE;
  }

  const char *path = getenv(NX_HOME_NAME);

  if (path == NULL)
  {
    return NULL;
  }

  setHomeDir(path);

  logTest("System::getHomeDir", "Got @PRODUCT_NAME@ Client home "
              "directory '%s'", _homeDir +
                  NX_HOME_SIZE);

  return _homeDir + NX_HOME_SIZE;
}