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 methods 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-2014 Markus Grüneis, 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  *  In addition, as a special exception, the copyright holders give
  33  *  permission to link the code of portions of this program with the
  34  *  OpenSSL library under certain conditions as described in each
  35  *  individual source file, and distribute linked combinations
  36  *  including the two.
  37  *  You must obey the GNU General Public License in all respects
  38  *  for all of the code used other than OpenSSL.  If you modify
  39  *  file(s) with this exception, you may extend this exception to your
  40  *  version of the file(s), but you are not obligated to do so.  If you
  41  *  do not wish to do so, delete this exception statement from your
  42  *  version.  If you delete this exception statement from all source
  43  *  files in the program, then also delete it here.
  44  */
  45
  46 #include <string.h>
  47 #include <sstream>
  48 #include <windows.h>
  49
  50 #include "registryKey.h"
  51
  52 #include "../anytunError.h"
  53
  54 RegistryKey::RegistryKey() : opened_(false)
  55 {
  56 }
  57
  58 RegistryKey::RegistryKey(HKEY hkey, std::string subKey, REGSAM samDesired) : opened_(false)
  59 {
  60   open(hkey, subKey, samDesired);
  61 }
  62
  63 RegistryKey::~RegistryKey()
  64 {
  65   close();
  66 }
  67
  68 bool RegistryKey::isOpen() const
  69 {
  70   return opened_;
  71 }
  72
  73 std::string RegistryKey::getName() const
  74 {
  75   return name_;
  76 }
  77
  78 DWORD RegistryKey::open(HKEY hkey, std::string subKey, REGSAM samDesired)
  79 {
  80   if(opened_) {
  81     RegCloseKey(key_);
  82   }
  83
  84   opened_ = false;
  85   name_ = "";
  86   LONG err = RegOpenKeyExA(hkey, subKey.c_str(), 0, samDesired, &key_);
  87   if(err != ERROR_SUCCESS) {
  88     return err;
  89   }
  90
  91   name_ = subKey;
  92   opened_ = true;
  93   return ERROR_SUCCESS;
  94 }
  95
  96 void RegistryKey::close()
  97 {
  98   if(opened_) {
  99     RegCloseKey(key_);
 100   }
 101   opened_ = false;
 102 }
 103
 104 std::string RegistryKey::operator[](std::string const& name) const
 105 {
 106   if(!opened_) {
 107     throw AnytunErrno(ERROR_INVALID_HANDLE);
 108   }
 109
 110   char value[STRING_VALUE_LENGTH];
 111   DWORD len = sizeof(value);
 112   LONG err = RegQueryValueExA(key_, name.c_str(), NULL, NULL, (LPBYTE)value, &len);
 113   if(err != ERROR_SUCCESS) {
 114     throw AnytunErrno(err);
 115   }
 116
 117   if(value[len-1] != 0) {
 118     if(len < sizeof(value)) {
 119       value[len++] = 0;
 120     } else {
 121       throw AnytunErrno(ERROR_INSUFFICIENT_BUFFER);
 122     }
 123   }
 124   return std::string(value);
 125 }
 126
 127 DWORD RegistryKey::getSubKey(DWORD index, RegistryKey& subKey, REGSAM sam) const
 128 {
 129   char subkeyname[NAME_LENGTH];
 130   DWORD len = sizeof(subkeyname);
 131   DWORD err = RegEnumKeyExA(key_, index, subkeyname, &len, NULL, NULL, NULL, NULL);
 132   if(err != ERROR_SUCCESS) {
 133     return err;
 134   }
 135
 136   return subKey.open(key_, subkeyname, sam);
 137 }
 138
 139 DWORD RegistryKey::getSubKey(std::string name, RegistryKey& subKey, REGSAM sam) const
 140 {
 141   return subKey.open(key_, name.c_str(), sam);
 142 }