src/config.cpp

   1 #include <iostream>
   2 #include <vector>
   3 #include <fstream>
   4 #include <dirent.h>
   5 #include <unordered_map>
   6
   7 #include "crypto/sslUtil.h"
   8
   9 std::string keyDir;
  10 std::unordered_map<std::string, Profile> profiles;
  11 std::unordered_map<std::string, std::shared_ptr<CAConfig>> CAs;
  12 std::string sqlHost, sqlUser, sqlPass, sqlDB;
  13 std::string serialPath;
  14
  15 std::shared_ptr<std::unordered_map<std::string, std::string>> parseConf( std::string path ) {
  16     std::shared_ptr<std::unordered_map<std::string, std::string>> map( new std::unordered_map<std::string, std::string>() );
  17     std::ifstream config;
  18     config.open( path );
  19
  20     if( !config.is_open() ) {
  21         std::cout << "Where is " << path << "?" << std::endl;
  22         throw "Config missing";
  23     }
  24
  25     std::string line1;
  26
  27     while( std::getline( config, line1 ) ) {
  28         if( line1[0] == '#' || line1.size() == 0 ) {
  29             continue;
  30         }
  31
  32         int splitter = line1.find( "=" );
  33
  34         if( splitter == -1 ) {
  35             std::cerr << "Ignoring malformed config line: " << line1 << std::endl;
  36             continue;
  37         }
  38
  39         std::string key = line1.substr( 0, splitter );
  40         std::string value = line1.substr( splitter + 1 );
  41         map->emplace( key, value );
  42     }
  43
  44     config.close();
  45
  46     return map;
  47 }
  48
  49 int parseProfiles() {
  50     CAs = std::unordered_map<std::string, std::shared_ptr<CAConfig>>();
  51
  52     DIR* dp;
  53     struct dirent* ep;
  54     dp = opendir( "profiles" );
  55
  56     if( dp == NULL ) {
  57         std::cerr << "Profiles not found " << std::endl;
  58         return -1;
  59     }
  60
  61     while( ( ep = readdir( dp ) ) ) {
  62         if( ep->d_name[0] == '.' ) {
  63             continue;
  64         }
  65
  66         std::string profileName( ep->d_name );
  67
  68         int splitter = profileName.find( "-" );
  69
  70         if( splitter == -1 ) {
  71             std::cerr << "Ignoring malformed profile: " << profileName << std::endl;
  72             continue;
  73         }
  74
  75         std::string id = profileName.substr( 0, splitter );
  76
  77         if( profileName.substr( profileName.size() - 4 ) != ".cfg" ) {
  78             std::cerr << "Ignoring malformed profile: " << profileName << std::endl;
  79             continue;
  80         }
  81
  82         auto map = parseConf( std::string( "profiles/" ) + profileName );
  83
  84         profileName = profileName.substr( 0, profileName.size() - 4 );
  85
  86         Profile prof;
  87         prof.id = std::stoi( id );
  88         prof.eku = map->at( "eku" );
  89         prof.ku = map->at( "ku" );
  90
  91         if( CAs.find( map->at( "ca" ) ) == CAs.end() ) {
  92             std::shared_ptr<CAConfig> ca( new CAConfig( map->at( "ca" ) ) );
  93             CAs.emplace( map->at( "ca" ), ca );
  94         }
  95
  96         prof.ca = CAs.at( map->at( "ca" ) );
  97
  98         profiles.emplace( profileName, prof );
  99         std::cout << "Profile: " << profileName << " up and running." << std::endl;
 100     }
 101
 102     ( void ) closedir( dp );
 103
 104
 105     std::cout << profiles.size() << " profiles loaded." << std::endl;
 106
 107     return 0;
 108 }
 109
 110 int parseConfig( std::string path ) {
 111
 112     auto masterConf = parseConf( path );
 113
 114     keyDir = masterConf->at( "key.directory" );
 115     sqlHost = masterConf->at( "sql.host" );
 116     sqlUser = masterConf->at( "sql.user" );
 117     sqlPass = masterConf->at( "sql.password" );
 118     sqlDB = masterConf->at( "sql.database" );
 119     serialPath = masterConf->at( "serialPath" );
 120
 121     if( keyDir == "" ) {
 122         std::cerr << "Missing config property key.directory" << std::endl;
 123         return -1;
 124     }
 125
 126     if( parseProfiles() != 0 ) {
 127         return -1;
 128     }
 129
 130     return 0;
 131 }