src/win32/registryKey.cpp

   1 /*
   2  *  anytun
   3  *
   4  *  The secure anycast tunneling protocol (satp) defines a protocol used
   5  *  for communication between any combination of unicast and anycast
   6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
   7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
   8  *  ethernet, ip, arp ...). satp directly includes cryptography and
   9  *  message authentication based on the methodes used by SRTP.  It is
  10  *  intended to deliver a generic, scaleable and secure solution for
  11  *  tunneling and relaying of packets of any protocol.
  12  *
  13  *
  14  *  Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl,
  15  *                          Christian Pointner <satp@wirdorange.org>
  16  *
  17  *  This file is part of Anytun.
  18  *
  19  *  Anytun is free software: you can redistribute it and/or modify
  20  *  it under the terms of the GNU General Public License as published by
  21  *  the Free Software Foundation, either version 3 of the License, or
  22  *  any later version.
  23  *
  24  *  Anytun is distributed in the hope that it will be useful,
  25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27  *  GNU General Public License for more details.
  28  *
  29  *  You should have received a copy of the GNU General Public License
  30  *  along with anytun.  If not, see <http://www.gnu.org/licenses/>.
  31  */
  32
  33 #include <string.h>
  34 #include <sstream>
  35 #include <windows.h>
  36
  37 #include "registryKey.h"
  38
  39 #include "../anytunError.h"
  40
  41 RegistryKey::RegistryKey() : opened_(false)
  42 {
  43 }
  44
  45 RegistryKey::RegistryKey(HKEY hkey, std::string subKey, REGSAM samDesired) : opened_(false)
  46 {
  47   open(hkey, subKey, samDesired);
  48 }
  49
  50 RegistryKey::~RegistryKey()
  51 {
  52   close();
  53 }
  54
  55 bool RegistryKey::isOpen() const
  56 {
  57   return opened_;
  58 }
  59
  60 std::string RegistryKey::getName() const
  61 {
  62   return name_;
  63 }
  64
  65 DWORD RegistryKey::open(HKEY hkey, std::string subKey, REGSAM samDesired)
  66 {
  67   if(opened_) {
  68     RegCloseKey(key_);
  69   }
  70
  71   opened_ = false;
  72   name_ = "";
  73   LONG err = RegOpenKeyExA(hkey, subKey.c_str(), 0, samDesired, &key_);
  74   if(err != ERROR_SUCCESS) {
  75     return err;
  76   }
  77
  78   name_ = subKey;
  79   opened_ = true;
  80   return ERROR_SUCCESS;
  81 }
  82
  83 void RegistryKey::close()
  84 {
  85   if(opened_) {
  86     RegCloseKey(key_);
  87   }
  88   opened_ = false;
  89 }
  90
  91 std::string RegistryKey::operator[](std::string const& name) const
  92 {
  93   if(!opened_) {
  94     throw AnytunErrno(ERROR_INVALID_HANDLE);
  95   }
  96
  97   char value[STRING_VALUE_LENGTH];
  98   DWORD len = sizeof(value);
  99   LONG err = RegQueryValueExA(key_, name.c_str(), NULL, NULL, (LPBYTE)value, &len);
 100   if(err != ERROR_SUCCESS) {
 101     throw AnytunErrno(err);
 102   }
 103
 104   if(value[len-1] != 0) {
 105     if(len < sizeof(value)) {
 106       value[len++] = 0;
 107     } else {
 108       throw AnytunErrno(ERROR_INSUFFICIENT_BUFFER);
 109     }
 110   }
 111   return std::string(value);
 112 }
 113
 114 DWORD RegistryKey::getSubKey(DWORD index, RegistryKey& subKey, REGSAM sam) const
 115 {
 116   char subkeyname[NAME_LENGTH];
 117   DWORD len = sizeof(subkeyname);
 118   DWORD err = RegEnumKeyExA(key_, index, subkeyname, &len, NULL, NULL, NULL, NULL);
 119   if(err != ERROR_SUCCESS) {
 120     return err;
 121   }
 122
 123   return subKey.open(key_, subkeyname, sam);
 124 }
 125
 126 DWORD RegistryKey::getSubKey(std::string name, RegistryKey& subKey, REGSAM sam) const
 127 {
 128   return subKey.open(key_, name.c_str(), sam);
 129 }