开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
setup.c
资源名称：myodbc-2_50_17a-src.zip　[点击查看]
上传用户：jmzj888
上传日期：2007-01-02
资源大小：220k
文件大小：24k
源码类别：

MySQL数据库

开发平台：

Visual C++

setup.c：源码内容
  1. /* Copyright Abandoned 1996 TCX DataKonsult AB & Monty Program KB & Detron HB
  2.    This file is public domain and comes with NO WARRANTY of any kind */
  4. /*
  5. ** SETUP.C - This is the ODBC sample driver code for
  6. ** setup.
  7. **
  8. ** This code is furnished on an as-is basis as part of the ODBC SDK and is
  9. ** intended for example purposes only.
  10. **
  11. */
  12. /*--------------------------------------------------------------------------
  13.   setup.c -- Sample ODBC setup
  15.   This code demonstrates how to interact with the ODBC Installer.  These
  16.   functions may be part of your ODBC driver or in a separate DLL.
  18.   The ODBC Installer allows a driver to control the management of
  19.   data sources by calling the ConfigDSN entry point in the appropriate
  20.   DLL.  When called, ConfigDSN receives four parameters:
  22.     hwndParent ---- Handle of the parent window for any dialogs which
  23.                     may need to be created.  If this handle is NULL,
  24.                     then no dialogs should be displayed (that is, the
  25.                     request should be processed silently).
  27.     fRequest ------ Flag indicating the type of request (add, configure
  28.                     (edit), or remove).
  30.     lpszDriver ---- Far pointer to a null-terminated string containing
  31.                     the name of your driver.  This is the same string you
  32.                     supply in the ODBC.INF file as your section header
  33.                     and which ODBC Setup displays to the user in lieu
  34.                     of the actual driver filename.  This string needs to
  35.                     be passed back to the ODBC Installer when adding a
  36.                     new data source name.
  38.     lpszAttributes- Far pointer to a list of null-terminated attribute
  39.                     keywords.  This list is similar to the list passed
  40.                     to SQLDriverConnect, except that each key-value
  41.                     pair is separated by a null-byte rather than a
  42.                     semicolon.  The entire list is then terminated with
  43.                     a null-byte (that is, two consecutive null-bytes
  44.                     mark the end of the list).  The keywords accepted
  45.                     should be those for SQLDriverConnect which are
  46.                     applicable, any new keywords you define for ODBC.INI,
  47.                     and any additional keywords you decide to document.
  49.   ConfigDSN should return TRUE if the requested operation succeeds and
  50.   FALSE otherwise.  The complete prototype for ConfigDSN is:
  52.   BOOL FAR PASCAL ConfigDSN(HWND    hwndParent,
  53.                             WORD    fRequest,
  54.                             LPSTR   lpszDriver,
  55.                             LPCSTR  lpszAttributes)
  57.   Your setup code should not write to ODBC.INI directly to add or remove
  58.   data source names.  Instead, link with ODBCINST.LIB (the ODBC Installer
  59.   library) and call SQLWriteDSNToIni and SQLRemoveDSNFromIni.
  60.   Use SQLWriteDSNToIni to add data source names.  If the data source name
  61.   already exists, SQLWriteDSNToIni will delete it (removing all of its
  62.   associated keys) and rewrite it.  SQLRemoveDSNToIni removes a data
  63.   source name and all of its associated keys.
  65.   For NT compatibility, the driver code should not use the
  66.   Get/WritePrivateProfileString windows functions for ODBC.INI, but instead,
  67.   use SQLGet/SQLWritePrivateProfileString functions that are macros (16 bit) or
  68.   calls to the odbcinst.dll (32 bit).
  70. --------------------------------------------------------------------------*/
  73. // Includes ----------------------------------------------------------------
  74. #include  "myodbc.h" // Local include files
  75. #undef VOID                                             // Because of ctl3d.h
  76. #define VOID void
  77. #include <ctl3d.h>
  78. #include "odbcinst.h" // ODBC installer prototypes
  79. #include <string.h> // C include files
  80. #include "dialogs.h"
  82. // Constants ---------------------------------------------------------------
  84. const char EMPTYSTR  []= "";
  85. const char OPTIONON  []= "Yes";
  86. const char OPTIONOFF []= "No";
  88. // ODBC.INI keywords
  89. const char ODBC_INI    []="ODBC.INI";     // ODBC initialization file
  90. const char INI_KDESC   []="Description";  // Data source description
  91. const char INI_KDB     []="Database";
  92. const char INI_KSERVER []="Server";
  93. const char INI_KUSER   []="User";   // Second option
  94. const char INI_KPASSWORD[] = "Password";
  95. const char INI_KPORT   []  = "Port";
  96. const char INI_KFLAG   []  = "Flag";
  97. const char INI_SDEFAULT[]= "Default";
  98. const char szTranslateName[] = "TranslationName";
  99. const char szTranslateDLL[] = "TranslationDLL";
  100. const char szTranslateOption[] = "TranslationOption";
  101. const char szUnkTrans[] = "Unknown Translator";
  103. // Attribute string look-up table (maps keys to associated indexes)
  105. static struct {
  106.   char  szKey[MAXKEYLEN];
  107.   int    iKey;
  108. } s_aLookup[] = { "DSN", KEY_DSN,
  109.   "DESC", KEY_DESC,
  110.   "Description",KEY_DESC,
  111.   "Database", KEY_DB,
  112.   "DB", KEY_DB,
  113.   "Server", KEY_SERVER,
  114.   "UID", KEY_USER,
  115.   "User", KEY_USER,
  116.   "Password", KEY_PASSWORD,
  117.   "PWD", KEY_PASSWORD,
  118.   "Port", KEY_PORT,
  119.   "Flag", KEY_FLAG,
  120.                   "",0
  121. };
  124. static void INTFUNC CenterDialog (HWND   hdlg);
  125. int  CALLBACK ConfigDlgProc (HWND hdlg,WORD wMsg,WPARAM wParam, LPARAM lParam);
  126. BOOL CALLBACK SetDSNAttributes(HWND hwnd, LPSETUPDLG lpsetupdlg);
  128. /* ConfigDSN ---------------------------------------------------------------
  129.   Description:  ODBC Setup entry point
  130.                 This entry point is called by the ODBC Installer
  131.                 (see file header for more details)
  132.   Input      :  hwnd ----------- Parent window handle
  133.                 fRequest ------- Request type (i.e., add, config, or remove)
  134.                 lpszDriver ----- Driver name
  135.                 lpszAttributes - data source attribute string
  136.   Output     :  TRUE success, FALSE otherwise
  137. --------------------------------------------------------------------------*/
  138. BOOL EXPFUNC ConfigDSN (HWND  hwnd,
  139.  WORD  fRequest,
  140.  LPCSTR  lpszDriver,
  141.  LPCSTR  lpszAttributes)
  142. {
  143.   BOOL  fSuccess; // Success/fail flag
  144.   GLOBALHANDLE hglbAttr;
  145.   LPSETUPDLG lpsetupdlg;
  147.   // Allocate attribute array
  148.   hglbAttr = GlobalAlloc(GMEM_MOVEABLE | GMEM_ZEROINIT, sizeof(SETUPDLG));
  149.   if (!hglbAttr)
  150.     return FALSE;
  151.   lpsetupdlg = (LPSETUPDLG)GlobalLock(hglbAttr);
  153.   // Parse attribute string
  154.   if (lpszAttributes)
  155.     ParseAttributes(lpszAttributes, lpsetupdlg);
  157.   // Save original data source name
  158.   if (lpsetupdlg->aAttr[KEY_DSN].fSupplied)
  159.     lstrcpy(lpsetupdlg->szDSN, lpsetupdlg->aAttr[KEY_DSN].szAttr);
  160.   else
  161.     lpsetupdlg->szDSN[0] = '';
  163.   // Remove data source
  164.   if (ODBC_REMOVE_DSN == fRequest) {
  165.     // Fail if no data source name was supplied
  166.     if (!lpsetupdlg->aAttr[KEY_DSN].fSupplied)
  167.       fSuccess = FALSE;
  169.     // Otherwise remove data source from ODBC.INI
  170.     else
  171.       fSuccess = SQLRemoveDSNFromIni(lpsetupdlg->aAttr[KEY_DSN].szAttr);
  172.   }
  174.   // Add or Configure data source
  175.   else {
  176.     // Save passed variables for global access (e.g., dialog access)
  177.     lpsetupdlg->hwndParent = hwnd;
  178.     lpsetupdlg->lpszDrvr   = lpszDriver;
  179.     lpsetupdlg->fNewDSN    = (ODBC_ADD_DSN == fRequest);
  180.     lpsetupdlg->fDefault   = !lstrcmpi(lpsetupdlg->aAttr[KEY_DSN].szAttr,
  181.        INI_SDEFAULT);
  183.     // Display the appropriate dialog (if parent window handle supplied)
  184.     if (hwnd) {
  185.       // Display dialog(s)
  186.       fSuccess = (IDOK == DialogBoxParam(s_hModule,
  187.  MAKEINTRESOURCE(CONFIGDSN),
  188.  hwnd,
  189.  ConfigDlgProc,
  190.  (LONG)(LPSTR)lpsetupdlg));
  191.     }
  193.     else if (lpsetupdlg->aAttr[KEY_DSN].fSupplied)
  194.       fSuccess = SetDSNAttributes(hwnd, lpsetupdlg);
  195.     else
  196.       fSuccess = FALSE;
  197.   }
  199.   GlobalUnlock(hglbAttr);
  200.   GlobalFree(hglbAttr);
  201.   return fSuccess;
  202. }
  205. /* CenterDialog ------------------------------------------------------------
  206. Description:  Center the dialog over the frame window
  207. Input    :  hdlg -- Dialog window handle
  208. Output    :  None
  209. --------------------------------------------------------------------------*/
  210. static void INTFUNC CenterDialog(HWND hdlg)
  211. {
  212.   HWND hwndFrame;
  213.   RECT rcDlg, rcScr, rcFrame;
  214.   int cx, cy;
  216.   hwndFrame = GetParent(hdlg);
  218.   GetWindowRect(hdlg, &rcDlg);
  219.   cx = rcDlg.right  - rcDlg.left;
  220.   cy = rcDlg.bottom - rcDlg.top;
  222.   GetClientRect(hwndFrame, &rcFrame);
  223.   ClientToScreen(hwndFrame, (LPPOINT)(&rcFrame.left));
  224.   ClientToScreen(hwndFrame, (LPPOINT)(&rcFrame.right));
  225.   rcDlg.top    = rcFrame.top  + (((rcFrame.bottom - rcFrame.top) - cy) >> 1);
  226.   rcDlg.left   = rcFrame.left + (((rcFrame.right - rcFrame.left) - cx) >> 1);
  227.   rcDlg.bottom = rcDlg.top  + cy;
  228.   rcDlg.right  = rcDlg.left + cx;
  230.   GetWindowRect(GetDesktopWindow(), &rcScr);
  231.   if (rcDlg.bottom > rcScr.bottom)
  232.   {
  233.     rcDlg.bottom = rcScr.bottom;
  234.     rcDlg.top    = rcDlg.bottom - cy;
  235.   }
  236.   if (rcDlg.right  > rcScr.right)
  237.   {
  238.     rcDlg.right = rcScr.right;
  239.     rcDlg.left  = rcDlg.right - cx;
  240.   }
  242.   if (rcDlg.left < 0) rcDlg.left = 0;
  243.   if (rcDlg.top  < 0) rcDlg.top  = 0;
  245.   MoveWindow(hdlg, rcDlg.left, rcDlg.top, cx, cy, TRUE);
  246.   return;
  247. }
  250. /* ConfigDlgProc -----------------------------------------------------------
  251.   Description:  Manage add data source name dialog
  252.   Input      :  hdlg --- Dialog window handle
  253.                 wMsg --- Message
  254.                 wParam - Message parameter
  255.                 lParam - Message parameter
  256.   Output     :  TRUE if message processed, FALSE otherwise
  257. --------------------------------------------------------------------------*/
  258. int CALLBACK ConfigDlgProc (HWND hdlg,WORD wMsg,WPARAM wParam,LPARAM lParam)
  259. {
  260.   switch (wMsg) {
  261.     // Initialize the dialog
  262.   case WM_INITDIALOG:
  263.   {
  264.     LPSETUPDLG lpsetupdlg;
  266.     Ctl3dRegister (s_hModule);
  267.     SetWindowLong(hdlg, DWL_USER, lParam);
  268. #ifdef WIN32
  269.     Ctl3dSubclassDlg(hdlg, CTL3D_ALL);
  270. #else
  271.     Ctl3dSubclassDlgEx(hdlg, CTL3D_ALL);
  272. #endif
  273.     CenterDialog(hdlg);  // Center dialog
  275.     lpsetupdlg = (LPSETUPDLG) lParam;
  276.     set_attributes(lpsetupdlg);
  278.     // Initialize dialog fields
  279.     // NOTE: Values supplied in the attribute string will always
  280.     //  override settings in ODBC.INI
  281.     SetDlgItemText(hdlg, IDC_DSNAME, lpsetupdlg->aAttr[KEY_DSN].szAttr);
  282.     SetDlgItemText(hdlg, IDC_DESC,   lpsetupdlg->aAttr[KEY_DESC].szAttr);
  283.     SetDlgItemText(hdlg, IDC_DB,     lpsetupdlg->aAttr[KEY_DB].szAttr);
  284.     SetDlgItemText(hdlg, IDC_SERVER, lpsetupdlg->aAttr[KEY_SERVER].szAttr);
  285.     SetDlgItemText(hdlg, IDC_USER,   lpsetupdlg->aAttr[KEY_USER].szAttr);
  286.     SetDlgItemText(hdlg, IDC_PASSWORD, lpsetupdlg->aAttr[KEY_PASSWORD].szAttr);
  287.     SetDlgItemText(hdlg, IDC_PORT,   lpsetupdlg->aAttr[KEY_PORT].szAttr);
  288.     SetDlgItemText(hdlg, IDC_FLAG,   lpsetupdlg->aAttr[KEY_FLAG].szAttr);
  290.     if (lpsetupdlg->fDefault)
  291.     {
  292.       EnableWindow(GetDlgItem(hdlg, IDC_DSNAME), FALSE);
  293.       EnableWindow(GetDlgItem(hdlg, IDC_DSNAMETEXT), FALSE);
  294.     }
  295.     else
  296.       SendDlgItemMessage(hdlg, IDC_DSNAME,
  297.  EM_LIMITTEXT, (WPARAM)(MAXDSNAME-1), 0L);
  298.     SendDlgItemMessage(hdlg, IDC_DESC,
  299.        EM_LIMITTEXT, (WPARAM)(MAXDESC-1), 0L);
  300.     SendDlgItemMessage(hdlg, IDC_DB,
  301.        EM_LIMITTEXT, (WORD)(NAME_LEN), 0L);
  302.     SendDlgItemMessage(hdlg, IDC_SERVER,
  303.        EM_LIMITTEXT, (WORD)(MAXSERVER-1), 0L);
  304.     SendDlgItemMessage(hdlg, IDC_USER,
  305.        EM_LIMITTEXT, (WORD)(MAXUSER-1), 0L);
  306.     SendDlgItemMessage(hdlg, IDC_PASSWORD,
  307.        EM_LIMITTEXT, (WORD)(MAXPASSWORD-1), 0L);
  308.     SendDlgItemMessage(hdlg, IDC_PORT,
  309.        EM_LIMITTEXT, (WORD) 5, 0L);
  310.     SendDlgItemMessage(hdlg, IDC_FLAG,
  311.        EM_LIMITTEXT, (WORD) 5, 0L);
  312.     return TRUE; // Focus was not set
  313.   }
  316. #ifdef WIN32
  317.   case WM_CTLCOLORBTN:
  318.   case WM_CTLCOLORDLG:
  319.   case WM_CTLCOLOREDIT:
  320.   case WM_CTLCOLORLISTBOX:
  321.   case WM_CTLCOLORMSGBOX:
  322.   case WM_CTLCOLORSCROLLBAR:
  323.   case WM_CTLCOLORSTATIC:
  324.     return (BOOL)Ctl3dCtlColorEx(wMsg, wParam, lParam);
  326.   case WM_SETTEXT:
  327.   case WM_NCPAINT:
  328.   case WM_NCACTIVATE:
  329.     SetWindowLong(hdlg, DWL_MSGRESULT,
  330.   Ctl3dDlgFramePaint(hdlg, wMsg, wParam, lParam));
  331.     return TRUE;
  332. #endif
  334.   case WM_SYSCOLORCHANGE:
  335.     return Ctl3dColorChange();
  337.     // Process buttons
  338.   case WM_COMMAND:
  339.     switch (GET_WM_COMMAND_ID(wParam, lParam)) {
  340.       // Ensure the OK button is enabled only when a data source name
  341.       // is entered
  342.     case IDC_DSNAME:
  343.       if (GET_WM_COMMAND_CMD(wParam, lParam) == EN_CHANGE)
  344.       {
  345. char szItem[MAXDSNAME]; // Edit control text
  347. // Enable/disable the OK button
  348. EnableWindow(GetDlgItem(hdlg, IDOK),
  349.      GetDlgItemText(hdlg, IDC_DSNAME,
  350.     szItem, sizeof(szItem)));
  351. return TRUE;
  352.       }
  353.       break;
  356.     // Accept results
  357.     case IDOK:
  358.     {
  359.       LPSETUPDLG lpsetupdlg;
  361.       lpsetupdlg = (LPSETUPDLG)GetWindowLong(hdlg, DWL_USER);
  362.       // Retrieve dialog values
  363.       if (!lpsetupdlg->fDefault)
  364. GetDlgItemText(hdlg, IDC_DSNAME,
  365.        lpsetupdlg->aAttr[KEY_DSN].szAttr,
  366.        sizeof(lpsetupdlg->aAttr[KEY_DSN].szAttr));
  367.       GetDlgItemText(hdlg, IDC_DESC,
  368.      lpsetupdlg->aAttr[KEY_DESC].szAttr,
  369.      sizeof(lpsetupdlg->aAttr[KEY_DESC].szAttr));
  370.       GetDlgItemText(hdlg, IDC_DB,
  371.      lpsetupdlg->aAttr[KEY_DB].szAttr,
  372.      sizeof(lpsetupdlg->aAttr[KEY_DB].szAttr));
  373.       GetDlgItemText(hdlg, IDC_SERVER,
  374.      lpsetupdlg->aAttr[KEY_SERVER].szAttr,
  375.      sizeof(lpsetupdlg->aAttr[KEY_SERVER].szAttr));
  376.       GetDlgItemText(hdlg, IDC_USER,
  377.      lpsetupdlg->aAttr[KEY_USER].szAttr,
  378.      sizeof(lpsetupdlg->aAttr[KEY_USER].szAttr));
  379.       GetDlgItemText(hdlg, IDC_PASSWORD,
  380.      lpsetupdlg->aAttr[KEY_PASSWORD].szAttr,
  381.      sizeof(lpsetupdlg->aAttr[KEY_PASSWORD].szAttr));
  382.       GetDlgItemText(hdlg, IDC_PORT,
  383.      lpsetupdlg->aAttr[KEY_PORT].szAttr,
  384.      sizeof(lpsetupdlg->aAttr[KEY_PORT].szAttr));
  385.       GetDlgItemText(hdlg, IDC_FLAG,
  386.      lpsetupdlg->aAttr[KEY_FLAG].szAttr,
  387.      sizeof(lpsetupdlg->aAttr[KEY_FLAG].szAttr));
  389.       // Update ODBC.INI
  390.       SetDSNAttributes(hdlg, lpsetupdlg);
  391.     }
  393.     // Return to caller
  394.     case IDCANCEL:
  395.       Ctl3dUnregister (s_hModule);
  396.       EndDialog(hdlg, wParam);
  397.       return TRUE;
  398.     }
  399.     break;
  400.   }
  402.   // Message not processed
  403.   return FALSE;
  404. }
  407. /* ParseAttributes ---------------------------------------------------------
  408.   Description:  Parse attribute string moving values into the aAttr array
  409.   Input      :  lpszAttributes - Pointer to attribute string
  410.   Output     :  None (global aAttr normally updated)
  411. --------------------------------------------------------------------------*/
  412. void INTFUNC ParseAttributes(LPCSTR lpszAttributes, LPSETUPDLG lpsetupdlg)
  413. {
  414.   LPCSTR lpsz;
  415.   LPCSTR lpszStart;
  416.   char aszKey[MAXKEYLEN];
  417.   int iElement;
  418.   int cbKey;
  419.   DBUG_ENTER("ParseAttributes");
  420.   DBUG_PRINT("enter",("attributes: %s",lpszAttributes));
  422.   for (lpsz=lpszAttributes; *lpsz;)
  423.   {  //  Extract key name (e.g., DSN), it must be terminated by an equals
  424.     lpszStart = lpsz;
  425.     if (!(lpsz=strchr(lpsz,'=')))
  426.       DBUG_VOID_RETURN;
  427.     // Determine the keys index in the key table (-1 if not found)
  428.     iElement = -1;
  429.     cbKey  = lpsz - lpszStart;
  430.     if (cbKey < sizeof(aszKey))
  431.     {
  432.       register int j;
  434.       _fmemcpy(aszKey, lpszStart, cbKey);
  435.       aszKey[cbKey] = '';
  436.       for (j = 0; *s_aLookup[j].szKey; j++)
  437.       {
  438. if (!lstrcmpi(s_aLookup[j].szKey, aszKey))
  439. {
  440.   iElement = s_aLookup[j].iKey;
  441.   break;
  442. }
  443.       }
  444.     }
  446.     // Locate end of key value
  447.     lpszStart = ++lpsz;
  448.     lpsz=strcend(lpsz,';');
  450.     // Save value if key is known
  451.     // NOTE: This code assumes the szAttr buffers in aAttr have been
  452.     //    zero initialized
  453.     if (iElement >= 0)
  454.     {
  455.       lpsetupdlg->aAttr[iElement].fSupplied = TRUE;
  456.       strmake(lpsetupdlg->aAttr[iElement].szAttr,lpszStart,
  457.       min(lpsz-lpszStart, sizeof(lpsetupdlg->aAttr[0].szAttr)-1));
  458.     }
  459.     if (*lpsz)
  460.       lpsz++;               /* Skipp ';' */
  461.   }
  462.   DBUG_VOID_RETURN;
  463. }
  466. void set_attributes(LPSETUPDLG lpsetupdlg)
  467. {
  468.   LPCSTR lpszDSN=lpsetupdlg->aAttr[KEY_DSN].szAttr;
  470.   if (!lpsetupdlg->aAttr[KEY_DESC].fSupplied)
  471.     SQLGetPrivateProfileString(lpszDSN, INI_KDESC,
  472.        EMPTYSTR,
  473.        lpsetupdlg->aAttr[KEY_DESC].szAttr,
  474.        sizeof(lpsetupdlg->aAttr[KEY_DESC].szAttr),
  475.        ODBC_INI);
  476.   if (!lpsetupdlg->aAttr[KEY_DB].fSupplied)
  477.     SQLGetPrivateProfileString(lpszDSN, INI_KDB,
  478.        lpszDSN,
  479.        lpsetupdlg->aAttr[KEY_DB].szAttr,
  480.        sizeof(lpsetupdlg->aAttr[KEY_DB].szAttr),
  481.        ODBC_INI);
  482.   if (!lpsetupdlg->aAttr[KEY_SERVER].fSupplied)
  483.     SQLGetPrivateProfileString(lpszDSN, INI_KSERVER,
  484.        EMPTYSTR,
  485.        lpsetupdlg->aAttr[KEY_SERVER].szAttr,
  486.        sizeof(lpsetupdlg->aAttr[KEY_SERVER].szAttr),
  487.        ODBC_INI);
  488.   if (!lpsetupdlg->aAttr[KEY_USER].fSupplied)
  489.     SQLGetPrivateProfileString(lpszDSN, INI_KUSER,
  490.        EMPTYSTR,
  491.        lpsetupdlg->aAttr[KEY_USER].szAttr,
  492.        sizeof(lpsetupdlg->aAttr[KEY_USER].szAttr),
  493.        ODBC_INI);
  494.   if (!lpsetupdlg->aAttr[KEY_PASSWORD].fSupplied)
  495.     SQLGetPrivateProfileString(lpszDSN, INI_KPASSWORD,
  496.        EMPTYSTR,
  497.        lpsetupdlg->aAttr[KEY_PASSWORD].szAttr,
  498.        sizeof(lpsetupdlg->aAttr[KEY_PASSWORD].
  499.       szAttr),
  500.        ODBC_INI);
  501.   if (!lpsetupdlg->aAttr[KEY_PORT].fSupplied)
  502.   {
  503.     char buff[10];
  504.     int2str(MYSQL_PORT,buff,10);
  505.     SQLGetPrivateProfileString(lpszDSN, INI_KPORT,
  506.        buff,
  507.        lpsetupdlg->aAttr[KEY_PORT].szAttr,
  508.        sizeof(lpsetupdlg->aAttr[KEY_PORT].
  509.       szAttr),
  510.        ODBC_INI);
  511.   }
  512.   if (!lpsetupdlg->aAttr[KEY_FLAG].fSupplied)
  513.     SQLGetPrivateProfileString(lpszDSN, INI_KFLAG,
  514.        "0",
  515.        lpsetupdlg->aAttr[KEY_FLAG].szAttr,
  516.        sizeof(lpsetupdlg->aAttr[KEY_FLAG].
  517.       szAttr),
  518.        ODBC_INI);
  519. }
  522. /* SetDSNAttributes --------------------------------------------------------
  523.   Description:  Write data source attributes to ODBC.INI
  524.   Input      :  hwnd - Parent window handle (plus globals)
  525.   Output     :  TRUE if successful, FALSE otherwise
  526. --------------------------------------------------------------------------*/
  527. BOOL INTFUNC SetDSNAttributes(HWND hwndParent, LPSETUPDLG lpsetupdlg)
  528. {
  529.   LPCSTR lpszDSN=lpsetupdlg->aAttr[KEY_DSN].szAttr;  // Pointer to data source name
  531.   // Validate arguments
  532.   if (lpsetupdlg->fNewDSN && !*lpsetupdlg->aAttr[KEY_DSN].szAttr)
  533.     return FALSE;
  535.   // Write the data source name
  536.   if (!SQLWriteDSNToIni(lpszDSN, lpsetupdlg->lpszDrvr))
  537.   {
  538.     if (hwndParent)
  539.     {
  540.       char  szBuf[MAXPATHLEN];
  541.       char  szMsg[MAXPATHLEN];
  543.       LoadString(s_hModule, IDS_BADDSN, szBuf, sizeof(szBuf));
  544.       wsprintf(szMsg, szBuf, lpszDSN);
  545.       LoadString(s_hModule, IDS_MSGTITLE, szBuf, sizeof(szBuf));
  546.       MessageBox(hwndParent, szMsg, szBuf, MB_ICONEXCLAMATION | MB_OK);
  547.     }
  548.     return FALSE;
  549.   }
  551.   // Update ODBC.INI
  552.   // Save the value if the data source is new, if it was edited, or if
  553.   // it was explicitly supplied
  554.   if (hwndParent || lpsetupdlg->aAttr[KEY_DESC].fSupplied )
  555.     SQLWritePrivateProfileString(lpszDSN,
  556.  INI_KDESC,
  557.  lpsetupdlg->aAttr[KEY_DESC].szAttr,
  558.  ODBC_INI);
  559.   if (hwndParent || lpsetupdlg->aAttr[KEY_DB].fSupplied )
  560.     SQLWritePrivateProfileString(lpszDSN,
  561.  INI_KDB,
  562.  lpsetupdlg->aAttr[KEY_DB].szAttr,
  563.  ODBC_INI);
  564.   if (hwndParent || lpsetupdlg->aAttr[KEY_SERVER].fSupplied )
  565.     SQLWritePrivateProfileString(lpszDSN,
  566.  INI_KSERVER,
  567.  lpsetupdlg->aAttr[KEY_SERVER].szAttr,
  568.  ODBC_INI);
  569.   if (hwndParent || lpsetupdlg->aAttr[KEY_USER].fSupplied )
  570.     SQLWritePrivateProfileString(lpszDSN,
  571.  INI_KUSER,
  572.  lpsetupdlg->aAttr[KEY_USER].szAttr,
  573.  ODBC_INI);
  574.   if (hwndParent || lpsetupdlg->aAttr[KEY_PASSWORD].fSupplied )
  575.     SQLWritePrivateProfileString(lpszDSN,
  576.  INI_KPASSWORD,
  577.  lpsetupdlg->aAttr[KEY_PASSWORD].szAttr,
  578.  ODBC_INI);
  579.   if (hwndParent || lpsetupdlg->aAttr[KEY_PORT].fSupplied )
  580.   {
  581.     SQLWritePrivateProfileString(lpszDSN,
  582.  INI_KPORT,
  583.  lpsetupdlg->aAttr[KEY_PORT].szAttr,
  584.  ODBC_INI);
  585.   }
  586.   if (hwndParent || lpsetupdlg->aAttr[KEY_FLAG].fSupplied )
  587.   {
  588.     SQLWritePrivateProfileString(lpszDSN,
  589.  INI_KFLAG,
  590.  lpsetupdlg->aAttr[KEY_FLAG].szAttr,
  591.  ODBC_INI);
  592.   }
  593.   // If the data source name has changed, remove the old name
  594.   if (lpsetupdlg->aAttr[KEY_DSN].fSupplied &&
  595.       lstrcmpi(lpsetupdlg->szDSN, lpsetupdlg->aAttr[KEY_DSN].szAttr))
  596.   {
  597.     SQLRemoveDSNFromIni(lpsetupdlg->szDSN);
  598.   }
  599.   return TRUE;
  600. }
  603. /*****************************************************************************
  604. ** Connect dialog for driver connect
  605. *****************************************************************************/
  607. BOOL FAR PASCAL DriverConnectProc(HWND hdlg,WORD wMsg, WPARAM wParam,
  608.   LPARAM lParam)
  609. {
  610.   LPSETUPDLG lpsetupdlg;
  611.   switch (wMsg) {
  612.   case WM_INITDIALOG:
  613.     Ctl3dRegister (s_hModule);
  614.     SetWindowLong(hdlg, DWL_USER, lParam);
  615. #ifdef WIN32
  616.     Ctl3dSubclassDlg(hdlg, CTL3D_ALL);
  617. #else
  618.     Ctl3dSubclassDlgEx(hdlg, CTL3D_ALL);
  619. #endif
  620.     CenterDialog(hdlg);
  621.     lpsetupdlg = (LPSETUPDLG) lParam;
  622.     set_attributes(lpsetupdlg);
  623.     SetDlgItemText(hdlg, IDC_DSNAME,  lpsetupdlg->aAttr[KEY_DSN].szAttr);
  624.     SetDlgItemText(hdlg, IDC_DB,  lpsetupdlg->aAttr[KEY_DB].szAttr);
  625.     SetDlgItemText(hdlg, IDC_SERVER,  lpsetupdlg->aAttr[KEY_SERVER].szAttr);
  626.     SetDlgItemText(hdlg, IDC_USER,  lpsetupdlg->aAttr[KEY_USER].szAttr);
  627.     SetDlgItemText(hdlg, IDC_PASSWORD,  lpsetupdlg->aAttr[KEY_PASSWORD].szAttr);
  628.     SetDlgItemText(hdlg, IDC_PORT, lpsetupdlg->aAttr[KEY_PORT].szAttr);
  629.     SetDlgItemText(hdlg, IDC_FLAG, lpsetupdlg->aAttr[KEY_FLAG].szAttr);
  630.     SendDlgItemMessage(hdlg, IDC_DSNAME,
  631.        EM_LIMITTEXT, (WPARAM)(MAXDSNAME-1), 0L);
  632.     SendDlgItemMessage(hdlg, IDC_DESC,
  633.        EM_LIMITTEXT, (WPARAM)(MAXDESC-1), 0L);
  634.     SendDlgItemMessage(hdlg, IDC_DB,
  635.        EM_LIMITTEXT, (WORD)(NAME_LEN), 0L);
  636.     SendDlgItemMessage(hdlg, IDC_SERVER,
  637.        EM_LIMITTEXT, (WORD)(MAXSERVER-1), 0L);
  638.     SendDlgItemMessage(hdlg, IDC_USER,
  639.        EM_LIMITTEXT, (WORD)(MAXUSER-1), 0L);
  640.     SendDlgItemMessage(hdlg, IDC_PASSWORD,
  641.        EM_LIMITTEXT, (WORD)(MAXPASSWORD-1), 0L);
  642.     SendDlgItemMessage(hdlg, IDC_PORT,
  643.        EM_LIMITTEXT, (WORD)5, 0L);
  644.     SendDlgItemMessage(hdlg, IDC_FLAG,
  645.        EM_LIMITTEXT, (WORD)5, 0L);
  646.     return TRUE;
  648. #ifdef WIN32
  649.   case WM_CTLCOLORBTN:
  650.   case WM_CTLCOLORDLG:
  651.   case WM_CTLCOLOREDIT:
  652.   case WM_CTLCOLORLISTBOX:
  653.   case WM_CTLCOLORMSGBOX:
  654.   case WM_CTLCOLORSCROLLBAR:
  655.   case WM_CTLCOLORSTATIC:
  656.     return (BOOL)Ctl3dCtlColorEx(wMsg, wParam, lParam);
  658.   case WM_SETTEXT:
  659.   case WM_NCPAINT:
  660.   case WM_NCACTIVATE:
  661.     SetWindowLong(hdlg, DWL_MSGRESULT,
  662.   Ctl3dDlgFramePaint(hdlg, wMsg, wParam, lParam));
  663.     return TRUE;
  664. #endif
  666.   case WM_SYSCOLORCHANGE:
  667.     return Ctl3dColorChange();
  669.   case WM_COMMAND:
  670.     switch (GET_WM_COMMAND_ID(wParam, lParam)) {
  671.     case IDC_DSNAME:
  672.       if (GET_WM_COMMAND_CMD(wParam, lParam) == EN_CHANGE)
  673.       {
  674. char szItem[MAXDSNAME];
  675. EnableWindow(GetDlgItem(hdlg, IDOK),
  676.      GetDlgItemText(hdlg, IDC_DSNAME,szItem, sizeof(szItem)));
  677. return TRUE;
  678.       }
  679.       break;
  680.     case IDOK:
  681.       lpsetupdlg = (LPSETUPDLG)GetWindowLong(hdlg, DWL_USER);
  683.       GetDlgItemText(hdlg, IDC_DSNAME,
  684.      lpsetupdlg->aAttr[KEY_DSN].szAttr,
  685.      sizeof(lpsetupdlg->aAttr[KEY_DSN].szAttr));
  686.       GetDlgItemText(hdlg, IDC_DB,
  687.      lpsetupdlg->aAttr[KEY_DB].szAttr,
  688.      sizeof(lpsetupdlg->aAttr[KEY_DB].szAttr));
  689.       GetDlgItemText(hdlg, IDC_SERVER,
  690.      lpsetupdlg->aAttr[KEY_SERVER].szAttr,
  691.      sizeof(lpsetupdlg->aAttr[KEY_SERVER].szAttr));
  692.       GetDlgItemText(hdlg, IDC_USER,
  693.      lpsetupdlg->aAttr[KEY_USER].szAttr,
  694.      sizeof(lpsetupdlg->aAttr[KEY_USER].szAttr));
  695.       GetDlgItemText(hdlg, IDC_PASSWORD,
  696.      lpsetupdlg->aAttr[KEY_PASSWORD].szAttr,
  697.      sizeof(lpsetupdlg->aAttr[KEY_PASSWORD].szAttr));
  698.       GetDlgItemText(hdlg, IDC_PORT,
  699.      lpsetupdlg->aAttr[KEY_PORT].szAttr,
  700.      sizeof(lpsetupdlg->aAttr[KEY_PORT].szAttr));
  701.       GetDlgItemText(hdlg, IDC_FLAG,
  702.      lpsetupdlg->aAttr[KEY_FLAG].szAttr,
  703.      sizeof(lpsetupdlg->aAttr[KEY_FLAG].szAttr));
  705.     case IDCANCEL:
  706.       Ctl3dUnregister (s_hModule);
  707.       EndDialog(hdlg, GET_WM_COMMAND_ID(wParam, lParam) == IDOK);
  708.       return TRUE;
  709.     }
  710.   }
  711.   return FALSE;
  712. }