mirror of
				https://github.com/python/cpython.git
				synced 2025-10-30 21:21:22 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			1495 lines
		
	
	
	
		
			42 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1495 lines
		
	
	
	
		
			42 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|   _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 _Py_HashPointer(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;
 | |
| 	}
 | |
| 	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);
 | |
| 	}
 | |
| #if SIZEOF_LONG >= SIZEOF_HKEY
 | |
| 	else if (PyInt_Check(obHandle)) {
 | |
| 		long rc = RegCloseKey((HKEY)PyInt_AsLong(obHandle));
 | |
| 		ok = (rc == ERROR_SUCCESS);
 | |
| 		if (!ok)
 | |
| 			PyErr_SetFromWindowsErrWithFunction(rc, "RegCloseKey");
 | |
| 	}
 | |
| #else
 | |
| 	else if (PyLong_Check(obHandle)) {
 | |
| 		long rc = RegCloseKey((HKEY)PyLong_AsVoidPtr(obHandle));
 | |
| 		ok = (rc == ERROR_SUCCESS);
 | |
| 		if (!ok)
 | |
| 			PyErr_SetFromWindowsErrWithFunction(rc, "RegCloseKey");
 | |
| 	}
 | |
| #endif
 | |
| 	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 which 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 {
 | |
| 				void *src_buf;
 | |
| 				PyBufferProcs *pb = value->ob_type->tp_as_buffer;
 | |
| 				if (pb==NULL) {
 | |
| 					PyErr_Format(PyExc_TypeError, 
 | |
| 						"Objects of type '%s' can not "
 | |
| 						"be used as binary registry values", 
 | |
| 						value->ob_type->tp_name);
 | |
| 					return FALSE;
 | |
| 				}
 | |
| 				*retDataSize = (*pb->bf_getreadbuffer)(value, 0, &src_buf);
 | |
| 				*retDataBuf = (BYTE *)PyMem_NEW(char,
 | |
| 								*retDataSize);
 | |
| 				if (*retDataBuf==NULL){
 | |
| 					PyErr_NoMemory();
 | |
| 					return FALSE;
 | |
| 				}
 | |
| 				memcpy(*retDataBuf, src_buf, *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);
 | |
| 				if (obData == NULL)
 | |
| 					return NULL;
 | |
| 				for (index = 0; index < s; index++)
 | |
| 				{
 | |
| 					size_t len = _mbstrlen(str[index]);
 | |
| 					if (len > INT_MAX) {
 | |
| 						PyErr_SetString(PyExc_OverflowError,
 | |
| 							"registry string is too long for a Python string");
 | |
| 						Py_DECREF(obData);
 | |
| 						return NULL;
 | |
| 					}
 | |
| 					PyList_SetItem(obData,
 | |
| 						       index,
 | |
| 						       PyUnicode_DecodeMBCS(
 | |
| 						            (const char *)str[index],
 | |
| 							   (int)len,
 | |
| 							    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 init_winreg(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);
 | |
| }
 | |
| 
 | 
