windows_registry.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:4k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: windows_registry.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 20:49:00  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.2
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: windows_registry.cpp,v 1000.1 2004/06/01 20:49:00 gouriano Exp $
  10.  * ===========================================================================
  11.  *
  12.  *                            PUBLIC DOMAIN NOTICE
  13.  *               National Center for Biotechnology Information
  14.  *
  15.  *  This software/database is a "United States Government Work" under the
  16.  *  terms of the United States Copyright Act.  It was written as part of
  17.  *  the author's official duties as a United States Government employee and
  18.  *  thus cannot be copyrighted.  This software/database is freely available
  19.  *  to the public for use. The National Library of Medicine and the U.S.
  20.  *  Government have not placed any restriction on its use or reproduction.
  21.  *
  22.  *  Although all reasonable efforts have been taken to ensure the accuracy
  23.  *  and reliability of the software and data, the NLM and the U.S.
  24.  *  Government do not and cannot warrant the performance or results that
  25.  *  may be obtained by using this software or data. The NLM and the U.S.
  26.  *  Government disclaim all warranties, express or implied, including
  27.  *  warranties of performance, merchantability or fitness for any particular
  28.  *  purpose.
  29.  *
  30.  *  Please cite the author in any work or product based on this material.
  31.  *
  32.  * ===========================================================================
  33.  *
  34.  * Authors:  Josh Cherry
  35.  *
  36.  * File Description:  Wrappers for interacting with Windows registry
  37.  *
  38.  */
  39. #include <ncbi_pch.hpp>
  40. #include "windows_registry.hpp"
  41. #include <windows.h>
  42. #include <winreg.h>
  43. BEGIN_NCBI_SCOPE
  44. // convert from enumeration for HKEY to Windows HKEY
  45. static HKEY s_GetRootKey(EHkey root_key)
  46. {
  47.     switch (root_key) {
  48.     case eHkey_classes_root:
  49.         return HKEY_CLASSES_ROOT;
  50.     case eHkey_current_user:
  51.         return HKEY_CURRENT_USER;
  52.     case eHkey_local_machine:
  53.         return HKEY_LOCAL_MACHINE;
  54.     case eHkey_users:
  55.         return HKEY_USERS;
  56.     case eHkey_performance_data:
  57.         return HKEY_PERFORMANCE_DATA;
  58.     default:
  59.         throw runtime_error("Invalid root key");
  60.     }
  61. }
  62. /// Set a Windows registry variable to a string value
  63. void SetWinRegVariable(EHkey root_key, const string& subkey,
  64.                        const string& varname, const string& value)
  65. {
  66.     HKEY open_key;
  67.     
  68.     if (RegCreateKey(s_GetRootKey(root_key), subkey.c_str(), &open_key)
  69.         != ERROR_SUCCESS) {
  70.         throw runtime_error("Error setting registry variable " + varname
  71.             + " in section " + subkey);
  72.     }
  73.     
  74.     if (RegSetValueEx(open_key, varname.c_str(), 0, REG_SZ,
  75.         (const unsigned char *) value.c_str(), value.size())
  76.         != ERROR_SUCCESS) {
  77.         throw runtime_error("Error setting registry variable " + varname
  78.             + " in section " + subkey);
  79.     }
  80.     
  81.     RegCloseKey(open_key);
  82. }
  83. END_NCBI_SCOPE
  84. /*
  85.  * ===========================================================================
  86.  * $Log: windows_registry.cpp,v $
  87.  * Revision 1000.1  2004/06/01 20:49:00  gouriano
  88.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.2
  89.  *
  90.  * Revision 1.2  2004/05/21 22:27:42  gorelenk
  91.  * Added PCH ncbi_pch.hpp
  92.  *
  93.  * Revision 1.1  2004/04/09 17:19:19  jcherry
  94.  * Initial version
  95.  *
  96.  * ===========================================================================
  97.  */