diff options
author | NingSun <ning.sun@intel.com> | 2018-02-08 08:34:03 -0800 |
---|---|---|
committer | NingSun <ning.sun@intel.com> | 2018-02-08 09:14:52 -0800 |
commit | 0c89b3ccba7c9b7332ab67ae1936aff51ca62367 (patch) | |
tree | 70c1b1d160d4c6d0a83395ca9a87c1264d0d3439 /SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp | |
parent | 945613b4db4e07f75d2bc7463db580ddfaa700fd (diff) |
Initial sshsm project structure
Issue-ID: AAF-94
Change-Id: I5e82fff418e7567b161acf9b98013a9b85ffc5b4
Signed-off-by: NingSun <ning.sun@intel.com>
Diffstat (limited to 'SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp')
-rw-r--r-- | SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp b/SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp new file mode 100644 index 0000000..3212d82 --- /dev/null +++ b/SoftHSMv2/src/lib/common/SimpleConfigLoader.cpp @@ -0,0 +1,352 @@ +/* + * Copyright (c) 2010 .SE, The Internet Infrastructure Foundation + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE + * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER + * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR + * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN + * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +/***************************************************************************** + SimpleConfigLoader.cpp + + Loads the configuration from the configuration file. + *****************************************************************************/ + +#include <stdio.h> +#include <string.h> +#include <stdlib.h> +#include <algorithm> +#include <limits.h> +#ifdef _WIN32 +# include <io.h> +#else +# include <unistd.h> +#endif +#include "config.h" +#if defined(HAVE_GETPWUID_R) +# include <sys/types.h> +# include <pwd.h> +#endif +#include "SimpleConfigLoader.h" +#include "log.h" +#include "Configuration.h" + +// Initialise the one-and-only instance +#ifdef HAVE_CXX11 +std::unique_ptr<SimpleConfigLoader> SimpleConfigLoader::instance(nullptr); +#else +std::auto_ptr<SimpleConfigLoader> SimpleConfigLoader::instance(NULL); +#endif + +// Return the one-and-only instance +SimpleConfigLoader* SimpleConfigLoader::i() +{ + if (instance.get() == NULL) + { + instance.reset(new SimpleConfigLoader()); + } + + return instance.get(); +} + +// Constructor +SimpleConfigLoader::SimpleConfigLoader() +{ +} + +// Load the configuration +bool SimpleConfigLoader::loadConfiguration() +{ + char* configPath = getConfigPath(); + + FILE* fp = fopen(configPath,"r"); + + if (fp == NULL) + { + ERROR_MSG("Could not open the config file: %s", configPath); + free(configPath); + return false; + } + free(configPath); + + char fileBuf[1024]; + + // Format in config file + // + // <name> = <value> + // # Line is ignored + + size_t line = 0; + while (fgets(fileBuf, sizeof(fileBuf), fp) != NULL) + { + line++; + + // End the string at the first comment or newline + fileBuf[strcspn(fileBuf, "#\n\r")] = '\0'; + + // Skip empty lines + if (fileBuf[0] == '\0') + { + continue; + } + + // Get the first part of the line + char* name = strtok(fileBuf, "="); + if (name == NULL) + { + WARNING_MSG("Bad format on line %lu, skip", (unsigned long)line); + continue; + } + + // Trim the name + char* trimmedName = trimString(name); + if (trimmedName == NULL) + { + WARNING_MSG("Bad format on line %lu, skip", (unsigned long)line); + continue; + } + + // Get the second part of the line + char* value = strtok(NULL, "="); + if(value == NULL) { + WARNING_MSG("Bad format on line %lu, skip", (unsigned long)line); + free(trimmedName); + continue; + } + + // Trim the value + char* trimmedValue = trimString(value); + if (trimmedValue == NULL) + { + WARNING_MSG("Bad format on line %lu, skip", (unsigned long)line); + free(trimmedName); + continue; + } + + // Save name,value + std::string stringName(trimmedName); + std::string stringValue(trimmedValue); + free(trimmedName); + free(trimmedValue); + + switch (Configuration::i()->getType(stringName)) + { + case CONFIG_TYPE_STRING: + Configuration::i()->setString(stringName, stringValue); + break; + case CONFIG_TYPE_INT: + Configuration::i()->setInt(stringName, atoi(stringValue.c_str())); + break; + case CONFIG_TYPE_BOOL: + bool boolValue; + if (string2bool(stringValue, &boolValue)) + { + Configuration::i()->setBool(stringName, boolValue); + } + else + { + WARNING_MSG("The value %s is not a boolean", stringValue.c_str()); + } + break; + case CONFIG_TYPE_UNSUPPORTED: + default: + WARNING_MSG("The following configuration is not supported: %s = %s", + stringName.c_str(), stringValue.c_str()); + break; + } + } + + fclose(fp); + + return true; +} + +// Get the boolean value from a string +bool SimpleConfigLoader::string2bool(std::string stringValue, bool* boolValue) +{ + // Convert to lowercase + std::transform(stringValue.begin(), stringValue.end(), stringValue.begin(), tolower); + + if (stringValue.compare("true") == 0) + { + *boolValue = true; + return true; + } + + if (stringValue.compare("false") == 0) + { + *boolValue = false; + return true; + } + + return false; +} + +#define CONFIG_FILE ".config/softhsm2/softhsm2.conf" + +/* Returns a user-specific path for configuration. + */ +static char *get_user_path(void) +{ +#ifdef _WIN32 + char path[512]; + const char *home_drive = getenv("HOMEDRIVE"); + const char *home_path = getenv("HOMEPATH"); + + if (home_drive && home_path) { + snprintf(path, sizeof(path), "%s%s\\softhsm2.conf", home_drive, home_path); + + if (_access(path, 0) == 0) + return strdup(path); + } + goto fail; +#else + char path[_POSIX_PATH_MAX]; + const char *home_dir = getenv("HOME"); + + if (home_dir != NULL && home_dir[0] != 0) { + snprintf(path, sizeof(path), "%s/" CONFIG_FILE, home_dir); + if (access(path, R_OK) == 0) + return strdup(path); + else + goto fail; + } + +# if defined(HAVE_GETPWUID_R) + if (home_dir == NULL || home_dir[0] == '\0') { + struct passwd *pwd; + struct passwd _pwd; + int ret; + char tmp[512]; + + ret = getpwuid_r(getuid(), &_pwd, tmp, sizeof(tmp), &pwd); + if (ret == 0 && pwd != NULL) { + snprintf(path, sizeof(path), "%s/" CONFIG_FILE, pwd->pw_dir); + if (access(path, R_OK) == 0) + return strdup(path); + else + goto fail; + } + } +# endif +#endif + + fail: + return NULL; +} + +static char *get_env_var_path(void) +{ +#ifdef _WIN32 + + LPSTR value = NULL; + DWORD size = 0; + + size = GetEnvironmentVariableA("SOFTHSM2_CONF", value, size); + if (size == 0) { + return NULL; + } + + value = (LPSTR) malloc(size); + if (NULL == value) { + return NULL; + } + + if (GetEnvironmentVariableA("SOFTHSM2_CONF", value, size) != (size - 1)) { + free(value); + return NULL; + } + + return value; + +#else + + char *value = getenv("SOFTHSM2_CONF"); + + if (value == NULL) { + return value; + } else { + return strdup(value); + } + +#endif +} + +char* SimpleConfigLoader::getConfigPath() +{ + char* configPath = get_env_var_path(); + char* tpath; + + if (configPath != NULL) + { + return configPath; + } + else + { + tpath = get_user_path(); + if (tpath != NULL) + { + return tpath; + } + } + + return strdup(DEFAULT_SOFTHSM2_CONF); +} + +char* SimpleConfigLoader::trimString(char* text) +{ + if (text == NULL) + { + return NULL; + } + + int startPos = 0; + int endPos = strlen(text) - 1; + + // Find the first position without a space + while (startPos <= endPos && isspace((int)*(text + startPos))) + { + startPos++; + } + // Find the last position without a space + while (startPos <= endPos && isspace((int)*(text + endPos))) + { + endPos--; + } + + // We must have a valid string + int length = endPos - startPos + 1; + if (length <= 0) + { + return NULL; + } + + // Create the trimmed text + char* trimmedText = (char*)malloc(length + 1); + if (trimmedText == NULL) + { + return NULL; + } + trimmedText[length] = '\0'; + memcpy(trimmedText, text + startPos, length); + + return trimmedText; +} |