added minimalisstic registry wrapper

1 files changed, 121 insertions, 0 deletions

diff --git a/src/win32/registryKey.cpp b/src/win32/registryKey.cpp
new file mode 100644
index 0000000..f0ab73d
--- /dev/null
+++ b/src/win32/registryKey.cpp
@@ -0,0 +1,121 @@
+/*
+ *  anytun
+ *
+ *  The secure anycast tunneling protocol (satp) defines a protocol used
+ *  for communication between any combination of unicast and anycast
+ *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
+ *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
+ *  ethernet, ip, arp ...). satp directly includes cryptography and
+ *  message authentication based on the methodes used by SRTP.  It is
+ *  intended to deliver a generic, scaleable and secure solution for
+ *  tunneling and relaying of packets of any protocol.
+ *
+ *
+ *  Copyright (C) 2007-2008 Othmar Gsenger, Erwin Nindl, 
+ *                          Christian Pointner <satp@wirdorange.org>
+ *
+ *  This file is part of Anytun.
+ *
+ *  Anytun is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License version 3 as
+ *  published by the Free Software Foundation.
+ *
+ *  Anytun is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with anytun.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <string.h>
+#include <sstream>
+#include <windows.h>
+
+#include "registryKey.h"
+
+#include "../log.h"
+
+RegistryKey::RegistryKey() : opened_(false)
+{
+}
+
+RegistryKey::RegistryKey(HKEY hkey, std::string subKey, REGSAM samDesired) : opened_(false)
+{
+  open(hkey, subKey, samDesired);
+}
+
+RegistryKey::~RegistryKey()
+{
+  close();
+}
+
+bool RegistryKey::isOpen() const
+{
+  return opened_;
+}
+
+std::string RegistryKey::getName() const
+{
+  return name_;
+}
+
+DWORD RegistryKey::open(HKEY hkey, std::string subKey, REGSAM samDesired)
+{
+  if(opened_)
+    RegCloseKey(key_);
+
+  opened_ = false;
+  name_ = "";
+  LONG err = RegOpenKeyExA(hkey, subKey.c_str(), 0, samDesired, &key_);
+  if(err != ERROR_SUCCESS)
+    return err;
+
+  name_ = subKey;
+  opened_ = true;
+  return ERROR_SUCCESS;
+}
+
+void RegistryKey::close()
+{
+  if(opened_)
+    RegCloseKey(key_);
+  opened_ = false;
+}
+
+std::string RegistryKey::operator[](std::string const& name) const
+{
+  if(!opened_)
+    throw LogErrno(ERROR_INVALID_HANDLE);
+
+  char value[STRING_VALUE_LENGTH];
+  DWORD len = sizeof(value);
+  LONG err = RegQueryValueExA(key_, name.c_str(), NULL, NULL, (LPBYTE)value, &len);
+  if(err != ERROR_SUCCESS)
+    throw LogErrno(err);
+
+  if(value[len-1] != 0) {
+    if(len < sizeof(value))
+      value[len++] = 0;
+    else
+      throw LogErrno(ERROR_INSUFFICIENT_BUFFER);
+  }  
+  return std::string(value);
+}
+
+DWORD RegistryKey::getSubKey(DWORD index, RegistryKey& subKey, REGSAM sam) const
+{
+  char subkeyname[NAME_LENGTH];
+  DWORD len = sizeof(subkeyname);
+	DWORD err = RegEnumKeyEx(key_, index, subkeyname, &len, NULL, NULL, NULL, NULL);
+  if(err != ERROR_SUCCESS)
+    return err;
+
+  return subKey.open(key_, subkeyname, sam);
+}
+
+DWORD RegistryKey::getSubKey(std::string name, RegistryKey& subKey, REGSAM sam) const
+{
+  return subKey.open(key_, name.c_str(), sam);
+}