summaryrefslogtreecommitdiff
path: root/src/win32/registryKey.cpp
blob: 4ce92b947e57e987d06538fe3922a9406ce50a4d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
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 = RegEnumKeyExA(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);
}