+ int noCols = 0;
+
+ RETCODE retcode;
+
+ wxString UserID;
+ wxString TableName;
+
+ if (userID)
+ {
+ if (!wxStrlen(userID))
+ UserID = uid;
+ else
+ UserID = userID;
+ }
+ else
+ UserID = wxT("");
+
+ // dBase does not use user names, and some drivers fail if you try to pass one
+ if (Dbms() == dbmsDBASE)
+ UserID = wxT("");
+
+ // Oracle user names may only be in uppercase, so force
+ // the name to uppercase
+ if (Dbms() == dbmsORACLE)
+ UserID = UserID.Upper();
+
+ {
+ // Loop through each table name
+ {
+ TableName = tableName;
+ // Oracle table names are uppercase only, so force
+ // the name to uppercase just in case programmer forgot to do this
+ if (Dbms() == dbmsORACLE)
+ TableName = TableName.Upper();
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ // MySQL and Access cannot accept a user name when looking up column names, so we
+ // use the call below that leaves out the user name
+ if (wxStrcmp(UserID.c_str(),wxT("")) &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS)
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // Owner
+ (UCHAR *) TableName.c_str(), SQL_NTS,
+ NULL, 0); // All columns
+ }
+ else
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // Owner
+ (UCHAR *) TableName.c_str(), SQL_NTS,
+ NULL, 0); // All columns
+ }
+ if (retcode != SQL_SUCCESS)
+ { // Error occured, abort
+ DispAllErrors(henv, hdbc, hstmt);
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return(-1);
+ }
+
+ // Count the columns
+ while ((retcode = SQLFetch(hstmt)) == SQL_SUCCESS)
+ noCols++;
+
+ if (retcode != SQL_NO_DATA_FOUND)
+ { // Error occured, abort
+ DispAllErrors(henv, hdbc, hstmt);
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return(-1);
+ }
+ }
+ }
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return noCols;
+
+} // wxDb::GetColumnCount()
+
+
+/********** wxDb::GetCatalog() *******/
+wxDbInf *wxDb::GetCatalog(char *userID)
+/*
+ * ---------------------------------------------------------------------
+ * -- 19991203 : mj10777 : Create ------
+ * -- : Creates a wxDbInf with Tables / Cols Array ------
+ * -- : uses SQLTables and fills pTableInf; ------
+ * -- : pColInf is set to NULL and numCols to 0; ------
+ * -- : returns pDbInf (wxDbInf) ------
+ * -- - if unsuccesfull (pDbInf == NULL) ------
+ * -- : pColInf can be filled with GetColumns(..); ------
+ * -- : numCols can be filled with GetColumnCount(..); ------
+ * ---------------------------------------------------------------------
+ *
+ * userID is evaluated in the following manner:
+ * userID == NULL ... UserID is ignored
+ * userID == "" ... UserID set equal to 'this->uid'
+ * userID != "" ... UserID set equal to 'userID'
+ *
+ * NOTE: ALL column bindings associated with this wxDb instance are unbound
+ * by this function. This function should use its own wxDb instance
+ * to avoid undesired unbinding of columns.
+ */
+{
+ wxDbInf *pDbInf = NULL; // Array of catalog entries
+ int noTab = 0; // Counter while filling table entries
+ int pass;
+ RETCODE retcode;
+ SDWORD cb;
+ wxString tblNameSave;
+
+ wxString UserID;
+
+ if (userID)
+ {
+ if (!wxStrlen(userID))
+ UserID = uid;
+ else
+ UserID = userID;
+ }
+ else
+ UserID = wxT("");
+
+ // dBase does not use user names, and some drivers fail if you try to pass one
+ if (Dbms() == dbmsDBASE)
+ UserID = wxT("");
+
+ // Oracle user names may only be in uppercase, so force
+ // the name to uppercase
+ if (Dbms() == dbmsORACLE)
+ UserID = UserID.Upper();
+
+ //-------------------------------------------------------------
+ pDbInf = new wxDbInf; // Create the Database Arrray
+ //-------------------------------------------------------------
+ // Table Information
+ // Pass 1 - Determine how many Tables there are.
+ // Pass 2 - Create the Table array and fill it
+ // - Create the Cols array = NULL
+ //-------------------------------------------------------------
+
+ for (pass = 1; pass <= 2; pass++)
+ {
+ SQLFreeStmt(hstmt, SQL_CLOSE); // Close if Open
+ tblNameSave = wxT("");
+
+ if (wxStrcmp(UserID.c_str(),wxT("")) &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS)
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+ else
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+
+ if (retcode != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ pDbInf = NULL;
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return pDbInf;
+ }
+
+ while ((retcode = SQLFetch(hstmt)) == SQL_SUCCESS) // Table Information
+ {
+ if (pass == 1) // First pass, just count the Tables
+ {
+ if (pDbInf->numTables == 0)
+ {
+ GetData( 1, SQL_C_CHAR, (UCHAR*) pDbInf->catalog, 128+1, &cb);
+ GetData( 2, SQL_C_CHAR, (UCHAR*) pDbInf->schema, 128+1, &cb);
+ }
+ pDbInf->numTables++; // Counter for Tables
+ } // if (pass == 1)
+ if (pass == 2) // Create and fill the Table entries
+ {
+ if (pDbInf->pTableInf == NULL) // Has the Table Array been created
+ { // no, then create the Array
+ pDbInf->pTableInf = new wxDbTableInf[pDbInf->numTables];
+ noTab = 0;
+ } // if (pDbInf->pTableInf == NULL) // Has the Table Array been created
+
+ GetData( 3, SQL_C_CHAR, (UCHAR*) (pDbInf->pTableInf+noTab)->tableName, DB_MAX_TABLE_NAME_LEN+1, &cb);
+ GetData( 4, SQL_C_CHAR, (UCHAR*) (pDbInf->pTableInf+noTab)->tableType, 30+1, &cb);
+ GetData( 5, SQL_C_CHAR, (UCHAR*) (pDbInf->pTableInf+noTab)->tableRemarks, 254+1, &cb);
+
+ noTab++;
+ } // if
+ } // while
+ } // for
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ // Query how many columns are in each table
+ for (noTab=0;noTab<pDbInf->numTables;noTab++)
+ {
+ (pDbInf->pTableInf+noTab)->numCols = GetColumnCount((pDbInf->pTableInf+noTab)->tableName,UserID);
+ }
+
+ return pDbInf;
+
+} // wxDb::GetCatalog()
+
+
+/********** wxDb::Catalog() **********/
+bool wxDb::Catalog(const char *userID, const char *fileName)
+/*
+ * Creates the text file specified in 'filename' which will contain
+ * a minimal data dictionary of all tables accessible by the user specified
+ * in 'userID'
+ *
+ * userID is evaluated in the following manner:
+ * userID == NULL ... UserID is ignored
+ * userID == "" ... UserID set equal to 'this->uid'
+ * userID != "" ... UserID set equal to 'userID'
+ *
+ * NOTE: ALL column bindings associated with this wxDb instance are unbound
+ * by this function. This function should use its own wxDb instance
+ * to avoid undesired unbinding of columns.
+ */
+{
+ assert(fileName && wxStrlen(fileName));
+
+ RETCODE retcode;
+ SDWORD cb;
+ char tblName[DB_MAX_TABLE_NAME_LEN+1];
+ wxString tblNameSave;
+ char colName[DB_MAX_COLUMN_NAME_LEN+1];
+ SWORD sqlDataType;
+ char typeName[30+1];
+ SWORD precision, length;
+
+ wxString UserID;
+
+ FILE *fp = fopen(fileName,"wt");
+ if (fp == NULL)
+ return(FALSE);
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ if (userID)
+ {
+ if (!wxStrlen(userID))
+ UserID = uid;
+ else
+ UserID = userID;
+ }
+ else
+ UserID = wxT("");
+
+ // dBase does not use user names, and some drivers fail if you try to pass one
+ if (Dbms() == dbmsDBASE)
+ UserID = wxT("");
+
+ // Oracle user names may only be in uppercase, so force
+ // the name to uppercase
+ if (Dbms() == dbmsORACLE)
+ UserID = UserID.Upper();
+
+ if (wxStrcmp(UserID.c_str(),wxT("")) &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS)
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+ else
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+ if (retcode != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ fclose(fp);
+ return(FALSE);
+ }
+
+ wxString outStr;
+ tblNameSave = wxT("");
+ int cnt = 0;
+
+ while ((retcode = SQLFetch(hstmt)) == SQL_SUCCESS)
+ {
+ if (wxStrcmp(tblName,tblNameSave.c_str()))
+ {
+ if (cnt)
+ fputs("\n", fp);
+ fputs("================================ ", fp);
+ fputs("================================ ", fp);
+ fputs("===================== ", fp);
+ fputs("========= ", fp);
+ fputs("=========\n", fp);
+ outStr.sprintf(wxT("%-32s %-32s %-21s %9s %9s\n"),
+ wxT("TABLE NAME"), wxT("COLUMN NAME"), wxT("DATA TYPE"), wxT("PRECISION"), wxT("LENGTH"));
+ fputs(outStr.c_str(), fp);
+ fputs("================================ ", fp);
+ fputs("================================ ", fp);
+ fputs("===================== ", fp);
+ fputs("========= ", fp);
+ fputs("=========\n", fp);
+ tblNameSave = tblName;
+ }
+
+ GetData(3,SQL_C_CHAR, (UCHAR *)tblName, DB_MAX_TABLE_NAME_LEN+1, &cb);
+ GetData(4,SQL_C_CHAR, (UCHAR *)colName, DB_MAX_COLUMN_NAME_LEN+1,&cb);
+ GetData(5,SQL_C_SSHORT,(UCHAR *)&sqlDataType,0, &cb);
+ GetData(6,SQL_C_CHAR, (UCHAR *)typeName, sizeof(typeName), &cb);
+ GetData(7,SQL_C_SSHORT,(UCHAR *)&precision, 0, &cb);
+ GetData(8,SQL_C_SSHORT,(UCHAR *)&length, 0, &cb);
+
+ outStr.sprintf("%-32s %-32s (%04d)%-15s %9d %9d\n",
+ tblName, colName, sqlDataType, typeName, precision, length);
+ if (fputs(outStr.c_str(), fp) == EOF)
+ {
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ fclose(fp);
+ return(FALSE);
+ }
+ cnt++;
+ }
+
+ if (retcode != SQL_NO_DATA_FOUND)
+ DispAllErrors(henv, hdbc, hstmt);
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ fclose(fp);
+ return(retcode == SQL_NO_DATA_FOUND);
+
+} // wxDb::Catalog()
+
+
+bool wxDb::TableExists(const char *tableName, const char *userID, const char *tablePath)
+/*
+ * Table name can refer to a table, view, alias or synonym. Returns true
+ * if the object exists in the database. This function does not indicate
+ * whether or not the user has privleges to query or perform other functions
+ * on the table.
+ *
+ * userID is evaluated in the following manner:
+ * userID == NULL ... UserID is ignored
+ * userID == "" ... UserID set equal to 'this->uid'
+ * userID != "" ... UserID set equal to 'userID'
+ */
+{
+ wxString UserID;
+ wxString TableName;
+
+ assert(tableName && wxStrlen(tableName));
+
+ if (Dbms() == dbmsDBASE)
+ {
+ wxString dbName;
+ if (tablePath && wxStrlen(tablePath))
+ dbName.sprintf("%s\\%s.dbf",tablePath,tableName);
+ else
+ dbName.sprintf("%s.dbf",tableName);
+
+ bool exists;
+ exists = wxFileExists(dbName.c_str());
+ return exists;
+ }
+
+ if (userID)
+ {
+ if (!wxStrlen(userID))
+ UserID = uid;
+ else
+ UserID = userID;
+ }
+ else
+ UserID = "";
+
+ // Oracle user names may only be in uppercase, so force
+ // the name to uppercase
+ if (Dbms() == dbmsORACLE)
+ UserID = UserID.Upper();
+
+ TableName = tableName;
+ // Oracle table names are uppercase only, so force
+ // the name to uppercase just in case programmer forgot to do this
+ if (Dbms() == dbmsORACLE)
+ TableName = TableName.Upper();
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ RETCODE retcode;
+
+ // MySQL and Access cannot accept a user name when looking up table names, so we
+ // use the call below that leaves out the user name
+ if (wxStrcmp(UserID,"") &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS)
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // All owners
+ (UCHAR FAR *)TableName.c_str(), SQL_NTS,
+ NULL, 0); // All table types
+ }
+ else
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // All owners
+ (UCHAR FAR *)TableName.c_str(), SQL_NTS,
+ NULL, 0); // All table types
+ }
+ if (retcode != SQL_SUCCESS)
+ return(DispAllErrors(henv, hdbc, hstmt));
+
+ retcode = SQLFetch(hstmt);
+ if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
+ {
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return(DispAllErrors(henv, hdbc, hstmt));
+ }
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ return(TRUE);
+
+} // wxDb::TableExists()
+
+
+#if EXPERIMENTAL_WXDB_FUNCTIONS // will be added in 2.4
+/********** wxDB::TablePrivileges() **********/
+bool wxDB::TablePrivileges(const char *tableName, const char* priv,
+ const char *userID, const char *tablePath)
+{
+ wxDbTablePrivilegeInfo result;
+ SDWORD cbRetVal;
+ RETCODE retcode;
+
+ //We probably need to be able to dynamically set this based on
+ //the driver type, and state.
+ char curRole[]="public";
+
+ //Prologue here similar to db::TableExists()
+ wxString UserID;
+ wxString TableName;
+
+ assert(tableName && wxStrlen(tableName));
+
+ if (userID)
+ {
+ if (!wxStrlen(userID))
+ UserID = uid;
+ else
+ UserID = userID;
+ }
+ else
+ UserID = "";
+
+ // Oracle user names may only be in uppercase, so force
+ // the name to uppercase
+ if (Dbms() == dbmsORACLE)
+ UserID = UserID.Upper();
+
+ TableName = tableName;
+ // Oracle table names are uppercase only, so force
+ // the name to uppercase just in case programmer forgot to do this
+ if (Dbms() == dbmsORACLE)
+ TableName = TableName.Upper();
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ retcode = SQLTablePrivileges(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // All owners
+ (UCHAR FAR *)TableName.GetData(), SQL_NTS);
+
+#ifdef DBDEBUG_CONSOLE
+ fprintf(stderr ,"SQLTablePrivileges() returned %i \n",retcode);
+#endif
+ retcode = SQLBindCol (hstmt, 1, SQL_C_CHAR , &result.tableQual, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 2, SQL_C_CHAR , &result.tableOwner, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 3, SQL_C_CHAR , &result.tableName, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 4, SQL_C_CHAR , &result.grantor, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 5, SQL_C_CHAR , &result.grantee, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 6, SQL_C_CHAR , &result.privilege, 128, &cbRetVal);
+
+ retcode = SQLBindCol (hstmt, 7, SQL_C_CHAR , &result.grantable, 3, &cbRetVal);
+
+ retcode = SQLFetch(hstmt);
+ while (retcode == SQL_SUCCESS || retcode == SQL_SUCCESS_WITH_INFO)
+ {
+#ifdef DBDEBUG_CONSOLE
+ fprintf(stderr,"Scanning %s privilege on table %s.%s granted by %s to %s\n",
+ result.privilege,result.tabowner,result.tabname,
+ result.grantor, result.grantee);
+#endif
+ if (UserID.IsSameAs(result.tableOwner,false) )
+ return TRUE;
+
+ if (UserID.IsSameAs(result.grantee,false) &&
+ !strcmp(result.privilege,priv))
+ return TRUE;
+
+ if (!strcmp(result.grantee,curRole) &&
+ !strcmp(result.privilege,priv))
+ return TRUE;
+
+ retcode = SQLFetch(hstmt);
+ }
+
+ return FALSE;
+
+} // wxDB::TablePrivileges
+#endif
projects / wxWidgets.git / blobdiff