Skip to content
Projeler
Gruplar
Parçacıklar
Yardım
Yükleniyor...
Oturum aç / Kaydol
Gezinmeyi değiştir
C
cpython
Proje
Proje
Ayrıntılar
Etkinlik
Cycle Analytics
Depo (repository)
Depo (repository)
Dosyalar
Kayıtlar (commit)
Dallar (branch)
Etiketler
Katkıda bulunanlar
Grafik
Karşılaştır
Grafikler
Konular (issue)
0
Konular (issue)
0
Liste
Pano
Etiketler
Kilometre Taşları
Birleştirme (merge) Talepleri
0
Birleştirme (merge) Talepleri
0
CI / CD
CI / CD
İş akışları (pipeline)
İşler
Zamanlamalar
Grafikler
Paketler
Paketler
Wiki
Wiki
Parçacıklar
Parçacıklar
Üyeler
Üyeler
Collapse sidebar
Close sidebar
Etkinlik
Grafik
Grafikler
Yeni bir konu (issue) oluştur
İşler
Kayıtlar (commit)
Konu (issue) Panoları
Kenar çubuğunu aç
Batuhan Osman TASKAYA
cpython
Commits
ee8c4184
Kaydet (Commit)
ee8c4184
authored
Haz 29, 2000
tarafından
Fred Drake
Dosyalara gözat
Seçenekler
Dosyalara Gözat
İndir
Eposta Yamaları
Sade Fark
Finish converting the winreg extension to _winreg.
üst
5917ecb0
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
1477 deletions
+3
-1477
winreg.c
PC/winreg.c
+0
-1474
winreg.dsp
PCbuild/winreg.dsp
+3
-3
No files found.
PC/winreg.c
deleted
100644 → 0
Dosyayı görüntüle @
5917ecb0
/*
winreg.c
Windows Registry access module for Python.
* Simple registry access written by Mark Hammond in win32api
module circa 1995.
* Bill Tutt expanded the support significantly not long after.
* Numerous other people have submitted patches since then.
* Ripped from win32api module 03-Feb-2000 by Mark Hammond, and
basic Unicode support added.
*/
#include "windows.h"
#include "Python.h"
#include "structmember.h"
#include "malloc.h"
/* for alloca */
static
BOOL
PyHKEY_AsHKEY
(
PyObject
*
ob
,
HKEY
*
pRes
,
BOOL
bNoneOK
);
static
PyObject
*
PyHKEY_FromHKEY
(
HKEY
h
);
static
BOOL
PyHKEY_Close
(
PyObject
*
obHandle
);
static
char
errNotAHandle
[]
=
"Object is not a handle"
;
/* The win32api module reports the function name that failed,
but this concept is not in the Python core.
Hopefully it will one day, and in the meantime I dont
want to lose this info...
*/
#define PyErr_SetFromWindowsErrWithFunction(rc, fnname) \
PyErr_SetFromWindowsErr(rc)
/* Forward declares */
/* Doc strings */
static
char
module_doc
[]
=
"This module provides access to the Windows registry API.
\n
"
"
\n
"
"Functions:
\n
"
"
\n
"
"CloseKey() - Closes a registry key.
\n
"
"ConnectRegistry() - Establishes a connection to a predefined registry handle
\n
"
" on another computer.
\n
"
"CreateKey() - Creates the specified key, or opens it if it already exists.
\n
"
"DeleteKey() - Deletes the specified key.
\n
"
"DeleteValue() - Removes a named value from the specified registry key.
\n
"
"EnumKey() - Enumerates subkeys of the specified open registry key.
\n
"
"EnumValue() - Enumerates values of the specified open registry key.
\n
"
"FlushKey() - Writes all the attributes of the specified key to the registry.
\n
"
"LoadKey() - Creates a subkey under HKEY_USER or HKEY_LOCAL_MACHINE and stores
\n
"
" registration information from a specified file into that subkey.
\n
"
"OpenKey() - Alias for <om win32api.RegOpenKeyEx>
\n
"
"OpenKeyEx() - Opens the specified key.
\n
"
"QueryValue() - Retrieves the value associated with the unnamed value for a
\n
"
" specified key in the registry.
\n
"
"QueryValueEx() - Retrieves the type and data for a specified value name
\n
"
" associated with an open registry key.
\n
"
"QueryInfoKey() - Returns information about the specified key.
\n
"
"SaveKey() - Saves the specified key, and all its subkeys a file.
\n
"
"SetValue() - Associates a value with a specified key.
\n
"
"SetValueEx() - Stores data in the value field of an open registry key.
\n
"
"
\n
"
"Special objects:
\n
"
"
\n
"
"HKEYType -- type object for HKEY objects
\n
"
"error -- exception raised for Win32 errors
\n
"
"
\n
"
"Integer constants:
\n
"
"Many constants are defined - see the documentation for each function
\n
"
"to see what constants are used, and where."
;
static
char
CloseKey_doc
[]
=
"CloseKey(hkey) - Closes a previously opened registry key.
\n
"
"
\n
"
"The hkey argument specifies a previously opened key.
\n
"
"
\n
"
"Note that if the key is not closed using this method, it will be
\n
"
"closed when the hkey object is destroyed by Python."
;
static
char
ConnectRegistry_doc
[]
=
"key = ConnectRegistry(computer_name, key) - "
"Establishes a connection to a predefined registry handle on another computer.
\n
"
"
\n
"
"computer_name is the name of the remote computer, of the form
\\\\
computername.
\n
"
" If None, the local computer is used.
\n
"
"key is the predefined handle to connect to.
\n
"
"
\n
"
"The return value is the handle of the opened key.
\n
"
"If the function fails, an EnvironmentError exception is raised."
;
static
char
CreateKey_doc
[]
=
"key = CreateKey(key, sub_key) - Creates or opens the specified key.
\n
"
"
\n
"
"key is an already open key, or one of the predefined HKEY_* constants
\n
"
"sub_key is a string that names the key this method opens or creates.
\n
"
" If key is one of the predefined keys, sub_key may be None. In that case,
\n
"
" the handle returned is the same key handle passed in to the function.
\n
"
"
\n
"
"If the key already exists, this function opens the existing key
\n
"
"
\n
"
"The return value is the handle of the opened key.
\n
"
"If the function fails, an exception is raised."
;
static
char
DeleteKey_doc
[]
=
"DeleteKey(key, sub_key) - Deletes the specified key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that must be a subkey of the key identified by the key parameter.
\n
"
" This value must not be None, and the key may not have subkeys.
\n
"
"
\n
"
"This method can not delete keys with subkeys.
\n
"
"
\n
"
"If the method succeeds, the entire key, including all of its values,
\n
"
"is removed. If the method fails, an EnvironmentError exception is raised."
;
static
char
DeleteValue_doc
[]
=
"DeleteValue(key, value) - Removes a named value from a registry key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"value is a string that identifies the value to remove."
;
static
char
EnumKey_doc
[]
=
"string = EnumKey(key, index) - Enumerates subkeys of an open registry key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"index is an integer that identifies the index of the key to retrieve.
\n
"
"
\n
"
"The function retrieves the name of one subkey each time it is called.
\n
"
"It is typically called repeatedly until an EnvironmentError exception is
\n
"
"raised, indicating no more values are available.
\n
"
;
static
char
EnumValue_doc
[]
=
"tuple = EnumValue(key, index) - Enumerates values of an open registry key.
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"index is an integer that identifies the index of the value to retrieve.
\n
"
"
\n
"
"The function retrieves the name of one subkey each time it is called.
\n
"
"It is typically called repeatedly, until an EnvironmentError exception
\n
"
"is raised, indicating no more values.
\n
"
"
\n
"
"The result is a tuple of 3 items:
\n
"
"value_name is a string that identifies the value.
\n
"
"value_data is an object that holds the value data, and whose type depends
\n
"
" on the underlying registry type.
\n
"
"data_type is an integer that identifies the type of the value data."
;
static
char
FlushKey_doc
[]
=
"FlushKey(key) - Writes all the attributes of a key to the registry.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"
\n
"
"It is not necessary to call RegFlushKey to change a key.
\n
"
"Registry changes are flushed to disk by the registry using its lazy flusher.
\n
"
"Registry changes are also flushed to disk at system shutdown.
\n
"
"Unlike CloseKey(), the FlushKey() method returns only when all the data has
\n
"
"been written to the registry.
\n
"
"An application should only call FlushKey() if it requires absolute certainty that registry changes are on disk.
\n
"
"If you don't know whether a FlushKey() call is required, it probably isn't.
\n
"
;
static
char
LoadKey_doc
[]
=
"LoadKey(key, sub_key, file_name) - Creates a subkey under the specified key
\n
"
"and stores registration information from a specified file into that subkey.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that identifies the sub_key to load
\n
"
"file_name is the name of the file to load registry data from.
\n
"
" This file must have been created with the SaveKey() function.
\n
"
" Under the file allocation table (FAT) file system, the filename may not
\n
"
"have an extension.
\n
"
"
\n
"
"A call to LoadKey() fails if the calling process does not have the
\n
"
"SE_RESTORE_PRIVILEGE privilege.
\n
"
"
\n
"
"If key is a handle returned by ConnectRegistry(), then the path specified
\n
"
"in fileName is relative to the remote computer.
\n
"
"
\n
"
"The docs imply key must be in the HKEY_USER or HKEY_LOCAL_MACHINE tree"
;
static
char
OpenKey_doc
[]
=
"key = OpenKey(key, sub_key, res = 0, sam = KEY_READ) - Opens the specified key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that identifies the sub_key to open
\n
"
"res is a reserved integer, and must be zero. Default is zero.
\n
"
"sam is an integer that specifies an access mask that describes the desired
\n
"
" security access for the key. Default is KEY_READ
\n
"
"
\n
"
"The result is a new handle to the specified key
\n
"
"If the function fails, an EnvironmentError exception is raised.
\n
"
;
static
char
OpenKeyEx_doc
[]
=
"See OpenKey()"
;
static
char
QueryInfoKey_doc
[]
=
"tuple = QueryInfoKey(key) - Returns information about a key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"
\n
"
"The result is a tuple of 3 items:"
"An integer that identifies the number of sub keys this key has.
\n
"
"An integer that identifies the number of values this key has.
\n
"
"A long integer that identifies when the key was last modified (if available)
\n
"
" as 100's of nanoseconds since Jan 1, 1600.
\n
"
;
static
char
QueryValue_doc
[]
=
"string = QueryValue(key, sub_key) - retrieves the unnamed value for a key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that holds the name of the subkey with which the value
\n
"
" is associated. If this parameter is None or empty, the function retrieves
\n
"
" the value set by the SetValue() method for the key identified by key."
"
\n
"
"Values in the registry have name, type, and data components. This method
\n
"
"retrieves the data for a key's first value that has a NULL name.
\n
"
"But the underlying API call doesn't return the type, Lame Lame Lame, DONT USE THIS!!!"
;
static
char
QueryValueEx_doc
[]
=
"value,type_id = QueryValueEx(key, value_name) - Retrieves the type and data for a specified value name associated with an open registry key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"value_name is a string indicating the value to query"
;
static
char
SaveKey_doc
[]
=
"SaveKey(key, file_name) - Saves the specified key, and all its subkeys to the specified file.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"file_name is the name of the file to save registry data to.
\n
"
" This file cannot already exist. If this filename includes an extension,
\n
"
" it cannot be used on file allocation table (FAT) file systems by the
\n
"
" LoadKey(), ReplaceKey() or RestoreKey() methods.
\n
"
"
\n
"
"If key represents a key on a remote computer, the path described by
\n
"
"file_name is relative to the remote computer.
\n
"
"The caller of this method must possess the SeBackupPrivilege security privilege.
\n
"
"This function passes NULL for security_attributes to the API."
;
static
char
SetValue_doc
[]
=
"SetValue(key, sub_key, type, value) - Associates a value with a specified key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that names the subkey with which the value is associated.
\n
"
"type is an integer that specifies the type of the data. Currently this
\n
"
" must be REG_SZ, meaning only strings are supported.
\n
"
"value is a string that specifies the new value.
\n
"
"
\n
"
"If the key specified by the sub_key parameter does not exist, the SetValue
\n
"
"function creates it.
\n
"
"
\n
"
"Value lengths are limited by available memory. Long values (more than
\n
"
"2048 bytes) should be stored as files with the filenames stored in
\n
"
"the configuration registry. This helps the registry perform efficiently.
\n
"
"
\n
"
"The key identified by the key parameter must have been opened with
\n
"
"KEY_SET_VALUE access."
;
static
char
SetValueEx_doc
[]
=
"SetValueEx(key, value_name, reserved, type, value) - Stores data in the value field of an open registry key.
\n
"
"
\n
"
"key is an already open key, or any one of the predefined HKEY_* constants.
\n
"
"sub_key is a string that names the subkey with which the value is associated.
\n
"
"type is an integer that specifies the type of the data. This should be one of:
\n
"
" REG_BINARY -- Binary data in any form.
\n
"
" REG_DWORD -- A 32-bit number.
\n
"
" REG_DWORD_LITTLE_ENDIAN -- A 32-bit number in little-endian format.
\n
"
" REG_DWORD_BIG_ENDIAN -- A 32-bit number in big-endian format.
\n
"
" REG_EXPAND_SZ -- A null-terminated string that contains unexpanded references
\n
"
" to environment variables (for example, %PATH%).
\n
"
" REG_LINK -- A Unicode symbolic link.
\n
"
" REG_MULTI_SZ -- An sequence of null-terminated strings, terminated by
\n
"
" two null characters. Note that Python handles this
\n
"
" termination automatically.
\n
"
" REG_NONE -- No defined value type.
\n
"
" REG_RESOURCE_LIST -- A device-driver resource list.
\n
"
" REG_SZ -- A null-terminated string.
\n
"
"reserved can be anything - zero is always passed to the API.
\n
"
"value is a string that specifies the new value.
\n
"
"
\n
"
"This method can also set additional value and type information for the
\n
"
"specified key. The key identified by the key parameter must have been
\n
"
"opened with KEY_SET_VALUE access.
\n
"
"
\n
"
"To open the key, use the CreateKeyEx() or OpenKeyEx() methods.
\n
"
"
\n
"
"Value lengths are limited by available memory. Long values (more than
\n
"
"2048 bytes) should be stored as files with the filenames stored in
\n
"
"the configuration registry. This helps the registry perform efficiently.
\n
"
;
/* PyHKEY docstrings */
static
char
PyHKEY_doc
[]
=
"PyHKEY Object - A Python object, representing a win32 registry key.
\n
"
"
\n
"
"This object wraps a Windows HKEY object, automatically closing it when
\n
"
"the object is destroyed. To guarantee cleanup, you can call either
\n
"
"the Close() method on the PyHKEY, or the CloseKey() method.
\n
"
"
\n
"
"All functions which accept a handle object also accept an integer -
\n
"
"however, use of the handle object is encouraged.
\n
"
"
\n
"
"Functions:
\n
"
"Close() - Closes the underlying handle.
\n
"
"Detach() - Returns the integer Win32 handle, detaching it from the object
\n
"
"
\n
"
"Properties:
\n
"
"handle - The integer Win32 handle.
\n
"
"
\n
"
"Operations:
\n
"
"__nonzero__ - Handles with an open object return true, otherwise false.
\n
"
"__int__ - Converting a handle to an integer returns the Win32 handle.
\n
"
"__cmp__ - Handle objects are compared using the handle value.
\n
"
;
static
char
PyHKEY_Close_doc
[]
=
"key.Close() - Closes the underlying Windows handle.
\n
"
"
\n
"
"If the handle is already closed, no error is raised."
;
static
char
PyHKEY_Detach_doc
[]
=
"int = key.Detach() - Detaches the Windows handle from the handle object.
\n
"
"
\n
"
"The result is the value of the handle before it is detached. If the
\n
"
"handle is already detached, this will return zero.
\n
"
"
\n
"
"After calling this function, the handle is effectively invalidated,
\n
"
"but the handle is not closed. You would call this function when you
\n
"
"need the underlying win32 handle to exist beyond the lifetime of the
\n
"
"handle object.
\n
"
"On 64 bit windows, the result of this function is a long integer
\n
"
;
/************************************************************************
The PyHKEY object definition
************************************************************************/
typedef
struct
{
PyObject_VAR_HEAD
HKEY
hkey
;
}
PyHKEYObject
;
#define PyHKEY_Check(op) ((op)->ob_type == &PyHKEY_Type)
static
char
*
failMsg
=
"bad operand type"
;
static
PyObject
*
PyHKEY_unaryFailureFunc
(
PyObject
*
ob
)
{
PyErr_SetString
(
PyExc_TypeError
,
failMsg
);
return
NULL
;
}
static
PyObject
*
PyHKEY_binaryFailureFunc
(
PyObject
*
ob1
,
PyObject
*
ob2
)
{
PyErr_SetString
(
PyExc_TypeError
,
failMsg
);
return
NULL
;
}
static
PyObject
*
PyHKEY_ternaryFailureFunc
(
PyObject
*
ob1
,
PyObject
*
ob2
,
PyObject
*
ob3
)
{
PyErr_SetString
(
PyExc_TypeError
,
failMsg
);
return
NULL
;
}
static
void
PyHKEY_deallocFunc
(
PyObject
*
ob
)
{
/* Can not call PyHKEY_Close, as the ob->tp_type
has already been cleared, thus causing the type
check to fail!
*/
PyHKEYObject
*
obkey
=
(
PyHKEYObject
*
)
ob
;
if
(
obkey
->
hkey
)
RegCloseKey
((
HKEY
)
obkey
->
hkey
);
PyObject_DEL
(
ob
);
}
static
int
PyHKEY_nonzeroFunc
(
PyObject
*
ob
)
{
return
((
PyHKEYObject
*
)
ob
)
->
hkey
!=
0
;
}
static
PyObject
*
PyHKEY_intFunc
(
PyObject
*
ob
)
{
PyHKEYObject
*
pyhkey
=
(
PyHKEYObject
*
)
ob
;
return
PyLong_FromVoidPtr
(
pyhkey
->
hkey
);
}
static
int
PyHKEY_printFunc
(
PyObject
*
ob
,
FILE
*
fp
,
int
flags
)
{
PyHKEYObject
*
pyhkey
=
(
PyHKEYObject
*
)
ob
;
char
resBuf
[
160
];
wsprintf
(
resBuf
,
"<PyHKEY at %p (%p)>"
,
ob
,
pyhkey
->
hkey
);
fputs
(
resBuf
,
fp
);
return
0
;
}
static
PyObject
*
PyHKEY_strFunc
(
PyObject
*
ob
)
{
PyHKEYObject
*
pyhkey
=
(
PyHKEYObject
*
)
ob
;
char
resBuf
[
160
];
wsprintf
(
resBuf
,
"<PyHKEY:%p>"
,
pyhkey
->
hkey
);
return
PyString_FromString
(
resBuf
);
}
static
int
PyHKEY_compareFunc
(
PyObject
*
ob1
,
PyObject
*
ob2
)
{
PyHKEYObject
*
pyhkey1
=
(
PyHKEYObject
*
)
ob1
;
PyHKEYObject
*
pyhkey2
=
(
PyHKEYObject
*
)
ob2
;
return
pyhkey1
==
pyhkey2
?
0
:
(
pyhkey1
<
pyhkey2
?
-
1
:
1
);
}
static
long
PyHKEY_hashFunc
(
PyObject
*
ob
)
{
/* Just use the address.
XXX - should we use the handle value?
*/
return
(
long
)
ob
;
}
static
PyNumberMethods
PyHKEY_NumberMethods
=
{
PyHKEY_binaryFailureFunc
,
/* nb_add */
PyHKEY_binaryFailureFunc
,
/* nb_subtract */
PyHKEY_binaryFailureFunc
,
/* nb_multiply */
PyHKEY_binaryFailureFunc
,
/* nb_divide */
PyHKEY_binaryFailureFunc
,
/* nb_remainder */
PyHKEY_binaryFailureFunc
,
/* nb_divmod */
PyHKEY_ternaryFailureFunc
,
/* nb_power */
PyHKEY_unaryFailureFunc
,
/* nb_negative */
PyHKEY_unaryFailureFunc
,
/* nb_positive */
PyHKEY_unaryFailureFunc
,
/* nb_absolute */
PyHKEY_nonzeroFunc
,
/* nb_nonzero */
PyHKEY_unaryFailureFunc
,
/* nb_invert */
PyHKEY_binaryFailureFunc
,
/* nb_lshift */
PyHKEY_binaryFailureFunc
,
/* nb_rshift */
PyHKEY_binaryFailureFunc
,
/* nb_and */
PyHKEY_binaryFailureFunc
,
/* nb_xor */
PyHKEY_binaryFailureFunc
,
/* nb_or */
0
,
/* nb_coerce (allowed to be zero) */
PyHKEY_intFunc
,
/* nb_int */
PyHKEY_unaryFailureFunc
,
/* nb_long */
PyHKEY_unaryFailureFunc
,
/* nb_float */
PyHKEY_unaryFailureFunc
,
/* nb_oct */
PyHKEY_unaryFailureFunc
,
/* nb_hex */
};
/* fwd declare __getattr__ */
static
PyObject
*
PyHKEY_getattr
(
PyObject
*
self
,
char
*
name
);
/* The type itself */
PyTypeObject
PyHKEY_Type
=
{
PyObject_HEAD_INIT
(
0
)
/* fill in type at module init */
0
,
"PyHKEY"
,
sizeof
(
PyHKEYObject
),
0
,
PyHKEY_deallocFunc
,
/* tp_dealloc */
PyHKEY_printFunc
,
/* tp_print */
PyHKEY_getattr
,
/* tp_getattr */
0
,
/* tp_setattr */
PyHKEY_compareFunc
,
/* tp_compare */
0
,
/* tp_repr */
&
PyHKEY_NumberMethods
,
/* tp_as_number */
0
,
/* tp_as_sequence */
0
,
/* tp_as_mapping */
PyHKEY_hashFunc
,
/* tp_hash */
0
,
/* tp_call */
PyHKEY_strFunc
,
/* tp_str */
0
,
/* tp_getattro */
0
,
/* tp_setattro */
0
,
/* tp_as_buffer */
0
,
/* tp_flags */
PyHKEY_doc
,
/* tp_doc */
};
#define OFF(e) offsetof(PyHKEYObject, e)
static
struct
memberlist
PyHKEY_memberlist
[]
=
{
{
"handle"
,
T_INT
,
OFF
(
hkey
)},
{
NULL
}
/* Sentinel */
};
/************************************************************************
The PyHKEY object methods
************************************************************************/
static
PyObject
*
PyHKEY_CloseMethod
(
PyObject
*
self
,
PyObject
*
args
)
{
if
(
!
PyArg_ParseTuple
(
args
,
":Close"
))
return
NULL
;
if
(
!
PyHKEY_Close
(
self
))
return
NULL
;
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyHKEY_DetachMethod
(
PyObject
*
self
,
PyObject
*
args
)
{
void
*
ret
;
PyHKEYObject
*
pThis
=
(
PyHKEYObject
*
)
self
;
if
(
!
PyArg_ParseTuple
(
args
,
":Detach"
))
return
NULL
;
ret
=
(
void
*
)
pThis
->
hkey
;
pThis
->
hkey
=
0
;
return
PyLong_FromVoidPtr
(
ret
);
}
static
struct
PyMethodDef
PyHKEY_methods
[]
=
{
{
"Close"
,
PyHKEY_CloseMethod
,
1
,
PyHKEY_Close_doc
},
{
"Detach"
,
PyHKEY_DetachMethod
,
1
,
PyHKEY_Detach_doc
},
{
NULL
}
};
/*static*/
PyObject
*
PyHKEY_getattr
(
PyObject
*
self
,
char
*
name
)
{
PyObject
*
res
;
res
=
Py_FindMethod
(
PyHKEY_methods
,
self
,
name
);
if
(
res
!=
NULL
)
return
res
;
PyErr_Clear
();
if
(
strcmp
(
name
,
"handle"
)
==
0
)
return
PyLong_FromVoidPtr
(((
PyHKEYObject
*
)
self
)
->
hkey
);
return
PyMember_Get
((
char
*
)
self
,
PyHKEY_memberlist
,
name
);
}
/************************************************************************
The public PyHKEY API (well, not public yet :-)
************************************************************************/
PyObject
*
PyHKEY_New
(
HKEY
hInit
)
{
PyHKEYObject
*
key
=
PyObject_NEW
(
PyHKEYObject
,
&
PyHKEY_Type
);
if
(
key
)
key
->
hkey
=
hInit
;
return
(
PyObject
*
)
key
;
}
BOOL
PyHKEY_Close
(
PyObject
*
ob_handle
)
{
LONG
rc
;
PyHKEYObject
*
key
;
if
(
!
PyHKEY_Check
(
ob_handle
))
{
PyErr_SetString
(
PyExc_TypeError
,
"bad operand type"
);
return
FALSE
;
}
key
=
(
PyHKEYObject
*
)
ob_handle
;
rc
=
key
->
hkey
?
RegCloseKey
((
HKEY
)
key
->
hkey
)
:
ERROR_SUCCESS
;
key
->
hkey
=
0
;
if
(
rc
!=
ERROR_SUCCESS
)
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegCloseKey"
);
return
rc
==
ERROR_SUCCESS
;
}
BOOL
PyHKEY_AsHKEY
(
PyObject
*
ob
,
HKEY
*
pHANDLE
,
BOOL
bNoneOK
)
{
if
(
ob
==
Py_None
)
{
if
(
!
bNoneOK
)
{
PyErr_SetString
(
PyExc_TypeError
,
"None is not a valid HKEY in this context"
);
return
FALSE
;
}
*
pHANDLE
=
(
HKEY
)
0
;
}
else
if
(
PyHKEY_Check
(
ob
))
{
PyHKEYObject
*
pH
=
(
PyHKEYObject
*
)
ob
;
*
pHANDLE
=
pH
->
hkey
;
}
else
if
(
PyInt_Check
(
ob
)
||
PyLong_Check
(
ob
))
{
/* We also support integers */
PyErr_Clear
();
*
pHANDLE
=
(
HKEY
)
PyLong_AsVoidPtr
(
ob
);
if
(
PyErr_Occurred
())
return
FALSE
;
*
pHANDLE
=
(
HKEY
)
PyInt_AsLong
(
ob
);
}
else
{
PyErr_SetString
(
PyExc_TypeError
,
"The object is not a PyHKEY object"
);
return
FALSE
;
}
return
TRUE
;
}
PyObject
*
PyHKEY_FromHKEY
(
HKEY
h
)
{
PyHKEYObject
*
op
;
/* PyObject_New is inlined */
op
=
(
PyHKEYObject
*
)
PyObject_MALLOC
(
sizeof
(
PyHKEYObject
));
if
(
op
==
NULL
)
return
PyErr_NoMemory
();
PyObject_INIT
(
op
,
&
PyHKEY_Type
);
op
->
hkey
=
h
;
return
(
PyObject
*
)
op
;
}
/************************************************************************
The module methods
************************************************************************/
BOOL
PyWinObject_CloseHKEY
(
PyObject
*
obHandle
)
{
BOOL
ok
;
if
(
PyHKEY_Check
(
obHandle
))
{
ok
=
PyHKEY_Close
(
obHandle
);
}
else
if
(
PyInt_Check
(
obHandle
))
{
long
rc
=
RegCloseKey
((
HKEY
)
PyInt_AsLong
(
obHandle
));
ok
=
(
rc
==
ERROR_SUCCESS
);
if
(
!
ok
)
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegCloseKey"
);
}
else
{
PyErr_SetString
(
PyExc_TypeError
,
"A handle must be a HKEY object or an integer"
);
return
FALSE
;
}
return
ok
;
}
/*
Private Helper functions for the registry interfaces
** Note that fixupMultiSZ and countString have both had changes
** made to support "incorrect strings". The registry specification
** calls for strings to be terminated with 2 null bytes. It seems
** some commercial packages install strings whcich dont conform,
** causing this code to fail - however, "regedit" etc still work
** with these strings (ie only we dont!).
*/
static
void
fixupMultiSZ
(
char
**
str
,
char
*
data
,
int
len
)
{
char
*
P
;
int
i
;
char
*
Q
;
Q
=
data
+
len
;
for
(
P
=
data
,
i
=
0
;
P
<
Q
&&
*
P
!=
'\0'
;
P
++
,
i
++
)
{
str
[
i
]
=
P
;
for
(;
*
P
!=
'\0'
;
P
++
)
;
}
}
static
int
countStrings
(
char
*
data
,
int
len
)
{
int
strings
;
char
*
P
;
char
*
Q
=
data
+
len
;
for
(
P
=
data
,
strings
=
0
;
P
<
Q
&&
*
P
!=
'\0'
;
P
++
,
strings
++
)
for
(;
P
<
Q
&&
*
P
!=
'\0'
;
P
++
)
;
return
strings
;
}
/* Convert PyObject into Registry data.
Allocates space as needed. */
static
BOOL
Py2Reg
(
PyObject
*
value
,
DWORD
typ
,
BYTE
**
retDataBuf
,
DWORD
*
retDataSize
)
{
int
i
,
j
;
switch
(
typ
)
{
case
REG_DWORD
:
if
(
value
!=
Py_None
&&
!
PyInt_Check
(
value
))
return
FALSE
;
*
retDataBuf
=
(
BYTE
*
)
PyMem_NEW
(
DWORD
,
sizeof
(
DWORD
));
if
(
*
retDataBuf
==
NULL
){
PyErr_NoMemory
();
return
FALSE
;
}
*
retDataSize
=
sizeof
(
DWORD
);
if
(
value
==
Py_None
)
{
DWORD
zero
=
0
;
memcpy
(
*
retDataBuf
,
&
zero
,
sizeof
(
DWORD
));
}
else
memcpy
(
*
retDataBuf
,
&
PyInt_AS_LONG
((
PyIntObject
*
)
value
),
sizeof
(
DWORD
));
break
;
case
REG_SZ
:
case
REG_EXPAND_SZ
:
{
int
need_decref
=
0
;
if
(
value
==
Py_None
)
*
retDataSize
=
1
;
else
{
if
(
PyUnicode_Check
(
value
))
{
value
=
PyUnicode_AsEncodedString
(
value
,
"mbcs"
,
NULL
);
if
(
value
==
NULL
)
return
FALSE
;
need_decref
=
1
;
}
if
(
!
PyString_Check
(
value
))
return
FALSE
;
*
retDataSize
=
1
+
strlen
(
PyString_AS_STRING
(
(
PyStringObject
*
)
value
));
}
*
retDataBuf
=
(
BYTE
*
)
PyMem_NEW
(
DWORD
,
*
retDataSize
);
if
(
*
retDataBuf
==
NULL
){
PyErr_NoMemory
();
return
FALSE
;
}
if
(
value
==
Py_None
)
strcpy
((
char
*
)
*
retDataBuf
,
""
);
else
strcpy
((
char
*
)
*
retDataBuf
,
PyString_AS_STRING
(
(
PyStringObject
*
)
value
));
if
(
need_decref
)
Py_DECREF
(
value
);
break
;
}
case
REG_MULTI_SZ
:
{
DWORD
size
=
0
;
char
*
P
;
PyObject
**
obs
=
NULL
;
if
(
value
==
Py_None
)
i
=
0
;
else
{
if
(
!
PyList_Check
(
value
))
return
FALSE
;
i
=
PyList_Size
(
value
);
}
obs
=
malloc
(
sizeof
(
PyObject
*
)
*
i
);
memset
(
obs
,
0
,
sizeof
(
PyObject
*
)
*
i
);
for
(
j
=
0
;
j
<
i
;
j
++
)
{
PyObject
*
t
;
t
=
PyList_GET_ITEM
(
(
PyListObject
*
)
value
,
j
);
if
(
PyString_Check
(
t
))
{
obs
[
j
]
=
t
;
Py_INCREF
(
t
);
}
else
if
(
PyUnicode_Check
(
t
))
{
obs
[
j
]
=
PyUnicode_AsEncodedString
(
t
,
"mbcs"
,
NULL
);
if
(
obs
[
j
]
==
NULL
)
goto
reg_multi_fail
;
}
else
goto
reg_multi_fail
;
size
+=
1
+
strlen
(
PyString_AS_STRING
(
(
PyStringObject
*
)
obs
[
j
]));
}
*
retDataSize
=
size
+
1
;
*
retDataBuf
=
(
BYTE
*
)
PyMem_NEW
(
char
,
*
retDataSize
);
if
(
*
retDataBuf
==
NULL
){
PyErr_NoMemory
();
goto
reg_multi_fail
;
}
P
=
(
char
*
)
*
retDataBuf
;
for
(
j
=
0
;
j
<
i
;
j
++
)
{
PyObject
*
t
;
t
=
obs
[
j
];
strcpy
(
P
,
PyString_AS_STRING
(
(
PyStringObject
*
)
t
));
P
+=
1
+
strlen
(
PyString_AS_STRING
(
(
PyStringObject
*
)
t
));
Py_DECREF
(
obs
[
j
]);
}
/* And doubly-terminate the list... */
*
P
=
'\0'
;
free
(
obs
);
break
;
reg_multi_fail:
if
(
obs
)
{
for
(
j
=
0
;
j
<
i
;
j
++
)
Py_XDECREF
(
obs
[
j
]);
free
(
obs
);
}
return
FALSE
;
}
case
REG_BINARY
:
/* ALSO handle ALL unknown data types here. Even if we can't
support it natively, we should handle the bits. */
default:
if
(
value
==
Py_None
)
*
retDataSize
=
0
;
else
{
if
(
!
PyString_Check
(
value
))
return
0
;
*
retDataSize
=
PyString_Size
(
value
);
*
retDataBuf
=
(
BYTE
*
)
PyMem_NEW
(
char
,
*
retDataSize
);
if
(
*
retDataBuf
==
NULL
){
PyErr_NoMemory
();
return
FALSE
;
}
memcpy
(
*
retDataBuf
,
PyString_AS_STRING
(
(
PyStringObject
*
)
value
),
*
retDataSize
);
}
break
;
}
return
TRUE
;
}
/* Convert Registry data into PyObject*/
static
PyObject
*
Reg2Py
(
char
*
retDataBuf
,
DWORD
retDataSize
,
DWORD
typ
)
{
PyObject
*
obData
;
switch
(
typ
)
{
case
REG_DWORD
:
if
(
retDataSize
==
0
)
obData
=
Py_BuildValue
(
"i"
,
0
);
else
obData
=
Py_BuildValue
(
"i"
,
*
(
int
*
)
retDataBuf
);
break
;
case
REG_SZ
:
case
REG_EXPAND_SZ
:
/* retDataBuf may or may not have a trailing NULL in
the buffer. */
if
(
retDataSize
&&
retDataBuf
[
retDataSize
-
1
]
==
'\0'
)
--
retDataSize
;
if
(
retDataSize
==
0
)
retDataBuf
=
""
;
obData
=
PyUnicode_DecodeMBCS
(
retDataBuf
,
retDataSize
,
NULL
);
break
;
case
REG_MULTI_SZ
:
if
(
retDataSize
==
0
)
obData
=
PyList_New
(
0
);
else
{
int
index
=
0
;
int
s
=
countStrings
(
retDataBuf
,
retDataSize
);
char
**
str
=
(
char
**
)
malloc
(
sizeof
(
char
*
)
*
s
);
if
(
str
==
NULL
)
return
PyErr_NoMemory
();
fixupMultiSZ
(
str
,
retDataBuf
,
retDataSize
);
obData
=
PyList_New
(
s
);
for
(
index
=
0
;
index
<
s
;
index
++
)
{
PyList_SetItem
(
obData
,
index
,
PyUnicode_DecodeMBCS
(
(
const
char
*
)
str
[
index
],
_mbstrlen
(
str
[
index
]),
NULL
)
);
}
free
(
str
);
break
;
}
case
REG_BINARY
:
/* ALSO handle ALL unknown data types here. Even if we can't
support it natively, we should handle the bits. */
default:
if
(
retDataSize
==
0
)
{
Py_INCREF
(
Py_None
);
obData
=
Py_None
;
}
else
obData
=
Py_BuildValue
(
"s#"
,
(
char
*
)
retDataBuf
,
retDataSize
);
break
;
}
if
(
obData
==
NULL
)
return
NULL
;
else
return
obData
;
}
/* The Python methods */
static
PyObject
*
PyCloseKey
(
PyObject
*
self
,
PyObject
*
args
)
{
PyObject
*
obKey
;
if
(
!
PyArg_ParseTuple
(
args
,
"O:CloseKey"
,
&
obKey
))
return
NULL
;
if
(
!
PyHKEY_Close
(
obKey
))
return
NULL
;
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyConnectRegistry
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
szCompName
=
NULL
;
HKEY
retKey
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"zO:ConnectRegistry"
,
&
szCompName
,
&
obKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
rc
=
RegConnectRegistry
(
szCompName
,
hKey
,
&
retKey
);
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"ConnectRegistry"
);
return
PyHKEY_FromHKEY
(
retKey
);
}
static
PyObject
*
PyCreateKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
HKEY
retKey
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oz:CreateKey"
,
&
obKey
,
&
subKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
rc
=
RegCreateKey
(
hKey
,
subKey
,
&
retKey
);
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"CreateKey"
);
return
PyHKEY_FromHKEY
(
retKey
);
}
static
PyObject
*
PyDeleteKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"Os:DeleteKey"
,
&
obKey
,
&
subKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
rc
=
RegDeleteKey
(
hKey
,
subKey
);
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegDeleteKey"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyDeleteValue
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oz:DeleteValue"
,
&
obKey
,
&
subKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
Py_BEGIN_ALLOW_THREADS
rc
=
RegDeleteValue
(
hKey
,
subKey
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegDeleteValue"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyEnumKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
int
index
;
long
rc
;
char
*
retBuf
;
DWORD
len
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oi:EnumKey"
,
&
obKey
,
&
index
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
((
rc
=
RegQueryInfoKey
(
hKey
,
NULL
,
NULL
,
NULL
,
NULL
,
&
len
,
NULL
,
NULL
,
NULL
,
NULL
,
NULL
,
NULL
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryInfoKey"
);
++
len
;
/* include null terminator */
retBuf
=
(
char
*
)
alloca
(
len
);
if
((
rc
=
RegEnumKey
(
hKey
,
index
,
retBuf
,
len
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegEnumKey"
);
return
Py_BuildValue
(
"s"
,
retBuf
);
}
static
PyObject
*
PyEnumValue
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
int
index
;
long
rc
;
char
*
retValueBuf
;
char
*
retDataBuf
;
DWORD
retValueSize
;
DWORD
retDataSize
;
DWORD
typ
;
PyObject
*
obData
;
PyObject
*
retVal
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oi:EnumValue"
,
&
obKey
,
&
index
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
((
rc
=
RegQueryInfoKey
(
hKey
,
NULL
,
NULL
,
NULL
,
NULL
,
NULL
,
NULL
,
NULL
,
&
retValueSize
,
&
retDataSize
,
NULL
,
NULL
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryInfoKey"
);
++
retValueSize
;
/* include null terminators */
++
retDataSize
;
retValueBuf
=
(
char
*
)
alloca
(
retValueSize
);
retDataBuf
=
(
char
*
)
alloca
(
retDataSize
);
Py_BEGIN_ALLOW_THREADS
rc
=
RegEnumValue
(
hKey
,
index
,
retValueBuf
,
&
retValueSize
,
NULL
,
&
typ
,
(
BYTE
*
)
retDataBuf
,
&
retDataSize
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"PyRegEnumValue"
);
obData
=
Reg2Py
(
retDataBuf
,
retDataSize
,
typ
);
if
(
obData
==
NULL
)
return
NULL
;
retVal
=
Py_BuildValue
(
"sOi"
,
retValueBuf
,
obData
,
typ
);
Py_DECREF
(
obData
);
return
retVal
;
}
static
PyObject
*
PyFlushKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"O:FlushKey"
,
&
obKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
Py_BEGIN_ALLOW_THREADS
rc
=
RegFlushKey
(
hKey
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegFlushKey"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyLoadKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
char
*
fileName
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oss:LoadKey"
,
&
obKey
,
&
subKey
,
&
fileName
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
Py_BEGIN_ALLOW_THREADS
rc
=
RegLoadKey
(
hKey
,
subKey
,
fileName
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegLoadKey"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PyOpenKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
int
res
=
0
;
HKEY
retKey
;
long
rc
;
REGSAM
sam
=
KEY_READ
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oz|ii:OpenKey"
,
&
obKey
,
&
subKey
,
&
res
,
&
sam
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
Py_BEGIN_ALLOW_THREADS
rc
=
RegOpenKeyEx
(
hKey
,
subKey
,
res
,
sam
,
&
retKey
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegOpenKeyEx"
);
return
PyHKEY_FromHKEY
(
retKey
);
}
static
PyObject
*
PyQueryInfoKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
long
rc
;
DWORD
nSubKeys
,
nValues
;
FILETIME
ft
;
LARGE_INTEGER
li
;
PyObject
*
l
;
PyObject
*
ret
;
if
(
!
PyArg_ParseTuple
(
args
,
"O:QueryInfoKey"
,
&
obKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
((
rc
=
RegQueryInfoKey
(
hKey
,
NULL
,
NULL
,
0
,
&
nSubKeys
,
NULL
,
NULL
,
&
nValues
,
NULL
,
NULL
,
NULL
,
&
ft
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryInfoKey"
);
li
.
LowPart
=
ft
.
dwLowDateTime
;
li
.
HighPart
=
ft
.
dwHighDateTime
;
l
=
PyLong_FromLongLong
(
li
.
QuadPart
);
if
(
l
==
NULL
)
return
NULL
;
ret
=
Py_BuildValue
(
"iiO"
,
nSubKeys
,
nValues
,
l
);
Py_DECREF
(
l
);
return
ret
;
}
static
PyObject
*
PyQueryValue
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
long
rc
;
char
*
retBuf
;
long
bufSize
=
0
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oz:QueryValue"
,
&
obKey
,
&
subKey
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
((
rc
=
RegQueryValue
(
hKey
,
subKey
,
NULL
,
&
bufSize
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryValue"
);
retBuf
=
(
char
*
)
alloca
(
bufSize
);
if
((
rc
=
RegQueryValue
(
hKey
,
subKey
,
retBuf
,
&
bufSize
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryValue"
);
return
Py_BuildValue
(
"s"
,
retBuf
);
}
static
PyObject
*
PyQueryValueEx
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
valueName
;
long
rc
;
char
*
retBuf
;
DWORD
bufSize
=
0
;
DWORD
typ
;
PyObject
*
obData
;
PyObject
*
result
;
if
(
!
PyArg_ParseTuple
(
args
,
"Oz:QueryValueEx"
,
&
obKey
,
&
valueName
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
((
rc
=
RegQueryValueEx
(
hKey
,
valueName
,
NULL
,
NULL
,
NULL
,
&
bufSize
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryValueEx"
);
retBuf
=
(
char
*
)
alloca
(
bufSize
);
if
((
rc
=
RegQueryValueEx
(
hKey
,
valueName
,
NULL
,
&
typ
,
(
BYTE
*
)
retBuf
,
&
bufSize
))
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegQueryValueEx"
);
obData
=
Reg2Py
(
retBuf
,
bufSize
,
typ
);
if
(
obData
==
NULL
)
return
NULL
;
result
=
Py_BuildValue
(
"Oi"
,
obData
,
typ
);
Py_DECREF
(
obData
);
return
result
;
}
static
PyObject
*
PySaveKey
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
fileName
;
LPSECURITY_ATTRIBUTES
pSA
=
NULL
;
long
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"Os:SaveKey"
,
&
obKey
,
&
fileName
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
/* One day we may get security into the core?
if (!PyWinObject_AsSECURITY_ATTRIBUTES(obSA, &pSA, TRUE))
return NULL;
*/
Py_BEGIN_ALLOW_THREADS
rc
=
RegSaveKey
(
hKey
,
fileName
,
pSA
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegSaveKey"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PySetValue
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
subKey
;
char
*
str
;
DWORD
typ
;
DWORD
len
;
long
rc
;
PyObject
*
obStrVal
;
PyObject
*
obSubKey
;
if
(
!
PyArg_ParseTuple
(
args
,
"OOiO:SetValue"
,
&
obKey
,
&
obSubKey
,
&
typ
,
&
obStrVal
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
(
typ
!=
REG_SZ
)
{
PyErr_SetString
(
PyExc_TypeError
,
"Type must be win32con.REG_SZ"
);
return
NULL
;
}
/* XXX - need Unicode support */
str
=
PyString_AsString
(
obStrVal
);
if
(
str
==
NULL
)
return
NULL
;
len
=
PyString_Size
(
obStrVal
);
if
(
obSubKey
==
Py_None
)
subKey
=
NULL
;
else
{
subKey
=
PyString_AsString
(
obSubKey
);
if
(
subKey
==
NULL
)
return
NULL
;
}
Py_BEGIN_ALLOW_THREADS
rc
=
RegSetValue
(
hKey
,
subKey
,
REG_SZ
,
str
,
len
+
1
);
Py_END_ALLOW_THREADS
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegSetValue"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
PyObject
*
PySetValueEx
(
PyObject
*
self
,
PyObject
*
args
)
{
HKEY
hKey
;
PyObject
*
obKey
;
char
*
valueName
;
PyObject
*
obRes
;
PyObject
*
value
;
BYTE
*
data
;
DWORD
len
;
DWORD
typ
;
LONG
rc
;
if
(
!
PyArg_ParseTuple
(
args
,
"OzOiO:SetValueEx"
,
&
obKey
,
&
valueName
,
&
obRes
,
&
typ
,
&
value
))
return
NULL
;
if
(
!
PyHKEY_AsHKEY
(
obKey
,
&
hKey
,
FALSE
))
return
NULL
;
if
(
!
Py2Reg
(
value
,
typ
,
&
data
,
&
len
))
{
if
(
!
PyErr_Occurred
())
PyErr_SetString
(
PyExc_ValueError
,
"Could not convert the data to the specified type."
);
return
NULL
;
}
Py_BEGIN_ALLOW_THREADS
rc
=
RegSetValueEx
(
hKey
,
valueName
,
0
,
typ
,
data
,
len
);
Py_END_ALLOW_THREADS
PyMem_DEL
(
data
);
if
(
rc
!=
ERROR_SUCCESS
)
return
PyErr_SetFromWindowsErrWithFunction
(
rc
,
"RegSetValueEx"
);
Py_INCREF
(
Py_None
);
return
Py_None
;
}
static
struct
PyMethodDef
winreg_methods
[]
=
{
{
"CloseKey"
,
PyCloseKey
,
1
,
CloseKey_doc
},
{
"ConnectRegistry"
,
PyConnectRegistry
,
1
,
ConnectRegistry_doc
},
{
"CreateKey"
,
PyCreateKey
,
1
,
CreateKey_doc
},
{
"DeleteKey"
,
PyDeleteKey
,
1
,
DeleteKey_doc
},
{
"DeleteValue"
,
PyDeleteValue
,
1
,
DeleteValue_doc
},
{
"EnumKey"
,
PyEnumKey
,
1
,
EnumKey_doc
},
{
"EnumValue"
,
PyEnumValue
,
1
,
EnumValue_doc
},
{
"FlushKey"
,
PyFlushKey
,
1
,
FlushKey_doc
},
{
"LoadKey"
,
PyLoadKey
,
1
,
LoadKey_doc
},
{
"OpenKey"
,
PyOpenKey
,
1
,
OpenKey_doc
},
{
"OpenKeyEx"
,
PyOpenKey
,
1
,
OpenKeyEx_doc
},
{
"QueryValue"
,
PyQueryValue
,
1
,
QueryValue_doc
},
{
"QueryValueEx"
,
PyQueryValueEx
,
1
,
QueryValueEx_doc
},
{
"QueryInfoKey"
,
PyQueryInfoKey
,
1
,
QueryInfoKey_doc
},
{
"SaveKey"
,
PySaveKey
,
1
,
SaveKey_doc
},
{
"SetValue"
,
PySetValue
,
1
,
SetValue_doc
},
{
"SetValueEx"
,
PySetValueEx
,
1
,
SetValueEx_doc
},
NULL
,
};
static
void
insint
(
PyObject
*
d
,
char
*
name
,
long
value
)
{
PyObject
*
v
=
PyInt_FromLong
(
value
);
if
(
!
v
||
PyDict_SetItemString
(
d
,
name
,
v
))
PyErr_Clear
();
Py_XDECREF
(
v
);
}
#define ADD_INT(val) insint(d, #val, val)
static
void
inskey
(
PyObject
*
d
,
char
*
name
,
HKEY
key
)
{
PyObject
*
v
=
PyLong_FromVoidPtr
(
key
);
if
(
!
v
||
PyDict_SetItemString
(
d
,
name
,
v
))
PyErr_Clear
();
Py_XDECREF
(
v
);
}
#define ADD_KEY(val) inskey(d, #val, val)
__declspec
(
dllexport
)
void
initwinreg
(
void
)
{
PyObject
*
m
,
*
d
;
m
=
Py_InitModule3
(
"winreg"
,
winreg_methods
,
module_doc
);
d
=
PyModule_GetDict
(
m
);
PyHKEY_Type
.
ob_type
=
&
PyType_Type
;
PyHKEY_Type
.
tp_doc
=
PyHKEY_doc
;
Py_INCREF
(
&
PyHKEY_Type
);
if
(
PyDict_SetItemString
(
d
,
"HKEYType"
,
(
PyObject
*
)
&
PyHKEY_Type
)
!=
0
)
return
;
Py_INCREF
(
PyExc_WindowsError
);
if
(
PyDict_SetItemString
(
d
,
"error"
,
PyExc_WindowsError
)
!=
0
)
return
;
/* Add the relevant constants */
ADD_KEY
(
HKEY_CLASSES_ROOT
);
ADD_KEY
(
HKEY_CURRENT_USER
);
ADD_KEY
(
HKEY_LOCAL_MACHINE
);
ADD_KEY
(
HKEY_USERS
);
ADD_KEY
(
HKEY_PERFORMANCE_DATA
);
#ifdef HKEY_CURRENT_CONFIG
ADD_KEY
(
HKEY_CURRENT_CONFIG
);
#endif
#ifdef HKEY_DYN_DATA
ADD_KEY
(
HKEY_DYN_DATA
);
#endif
ADD_INT
(
KEY_QUERY_VALUE
);
ADD_INT
(
KEY_SET_VALUE
);
ADD_INT
(
KEY_CREATE_SUB_KEY
);
ADD_INT
(
KEY_ENUMERATE_SUB_KEYS
);
ADD_INT
(
KEY_NOTIFY
);
ADD_INT
(
KEY_CREATE_LINK
);
ADD_INT
(
KEY_READ
);
ADD_INT
(
KEY_WRITE
);
ADD_INT
(
KEY_EXECUTE
);
ADD_INT
(
KEY_ALL_ACCESS
);
ADD_INT
(
REG_OPTION_RESERVED
);
ADD_INT
(
REG_OPTION_NON_VOLATILE
);
ADD_INT
(
REG_OPTION_VOLATILE
);
ADD_INT
(
REG_OPTION_CREATE_LINK
);
ADD_INT
(
REG_OPTION_BACKUP_RESTORE
);
ADD_INT
(
REG_OPTION_OPEN_LINK
);
ADD_INT
(
REG_LEGAL_OPTION
);
ADD_INT
(
REG_CREATED_NEW_KEY
);
ADD_INT
(
REG_OPENED_EXISTING_KEY
);
ADD_INT
(
REG_WHOLE_HIVE_VOLATILE
);
ADD_INT
(
REG_REFRESH_HIVE
);
ADD_INT
(
REG_NO_LAZY_FLUSH
);
ADD_INT
(
REG_NOTIFY_CHANGE_NAME
);
ADD_INT
(
REG_NOTIFY_CHANGE_ATTRIBUTES
);
ADD_INT
(
REG_NOTIFY_CHANGE_LAST_SET
);
ADD_INT
(
REG_NOTIFY_CHANGE_SECURITY
);
ADD_INT
(
REG_LEGAL_CHANGE_FILTER
);
ADD_INT
(
REG_NONE
);
ADD_INT
(
REG_SZ
);
ADD_INT
(
REG_EXPAND_SZ
);
ADD_INT
(
REG_BINARY
);
ADD_INT
(
REG_DWORD
);
ADD_INT
(
REG_DWORD_LITTLE_ENDIAN
);
ADD_INT
(
REG_DWORD_BIG_ENDIAN
);
ADD_INT
(
REG_LINK
);
ADD_INT
(
REG_MULTI_SZ
);
ADD_INT
(
REG_RESOURCE_LIST
);
ADD_INT
(
REG_FULL_RESOURCE_DESCRIPTOR
);
ADD_INT
(
REG_RESOURCE_REQUIREMENTS_LIST
);
}
PCbuild/winreg.dsp
Dosyayı görüntüle @
ee8c4184
...
@@ -54,7 +54,7 @@ BSC32=bscmake.exe
...
@@ -54,7 +54,7 @@ BSC32=bscmake.exe
# ADD BSC32 /nologo
# ADD BSC32 /nologo
LINK32=link.exe
LINK32=link.exe
# ADD BASE LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /dll /machine:I386
# ADD BASE LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /dll /machine:I386
# ADD LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /base:"0x1e1C0000" /dll /machine:I386 /out:"winreg.pyd"
# ADD LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /base:"0x1e1C0000" /dll /machine:I386 /out:"
_
winreg.pyd"
!ELSEIF "$(CFG)" == "winreg - Win32 Debug"
!ELSEIF "$(CFG)" == "winreg - Win32 Debug"
...
@@ -81,7 +81,7 @@ BSC32=bscmake.exe
...
@@ -81,7 +81,7 @@ BSC32=bscmake.exe
# ADD BSC32 /nologo
# ADD BSC32 /nologo
LINK32=link.exe
LINK32=link.exe
# ADD BASE LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /dll /debug /machine:I386 /pdbtype:sept
# ADD BASE LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /dll /debug /machine:I386 /pdbtype:sept
# ADD LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /base:"0x1e1C0000" /dll /debug /machine:I386 /out:"winreg_d.pyd" /pdbtype:sept
# ADD LINK32 kernel32.lib user32.lib gdi32.lib winspool.lib comdlg32.lib advapi32.lib shell32.lib ole32.lib oleaut32.lib uuid.lib odbc32.lib odbccp32.lib /nologo /base:"0x1e1C0000" /dll /debug /machine:I386 /out:"
_
winreg_d.pyd" /pdbtype:sept
!ENDIF
!ENDIF
...
@@ -91,7 +91,7 @@ LINK32=link.exe
...
@@ -91,7 +91,7 @@ LINK32=link.exe
# Name "winreg - Win32 Debug"
# Name "winreg - Win32 Debug"
# Begin Source File
# Begin Source File
SOURCE=..\PC\winreg.c
SOURCE=..\PC\
_
winreg.c
!IF "$(CFG)" == "winreg - Win32 Release"
!IF "$(CFG)" == "winreg - Win32 Release"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment