From febb518a0e2401299970ad031842db3f90ab112d Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Wed, 21 Jan 2009 13:22:52 +0000 Subject: added minimalisstic registry wrapper --- src/win32/registryKey.cpp | 121 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 121 insertions(+) create mode 100644 src/win32/registryKey.cpp (limited to 'src/win32/registryKey.cpp') 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 + * + * 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 . + */ + +#include +#include +#include + +#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); +} -- cgit v1.2.3