Files
harbour-core/contrib/hbsqlit3/tests/authoriz.prg
Viktor Szakats 9f16c2bf8e 2017-08-13 18:27 UTC Viktor Szakats (vszakats users.noreply.github.com)
* *
    * update copyright headers with new FSF postal address
    * COPYING.txt -> LICENSE.txt (rest of repo to be synced)
2017-08-13 18:38:59 +00:00

228 lines
6.7 KiB
Plaintext

/*
* SQLite3 Demo. Using sqlite3_set_authorizer()
*
* Copyright 2009 P.Chornyj <myorg63@mail.ru>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; see the file LICENSE.txt. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301 USA (or visit https://www.gnu.org/licenses/).
*
* As a special exception, the Harbour Project gives permission for
* additional uses of the text contained in its release of Harbour.
*
* The exception is that, if you link the Harbour libraries with other
* files to produce an executable, this does not by itself cause the
* resulting executable to be covered by the GNU General Public License.
* Your use of that executable is in no way restricted on account of
* linking the Harbour library code into it.
*
* This exception does not however invalidate any other reasons why
* the executable file might be covered by the GNU General Public License.
*
* This exception applies only to the code released by the Harbour
* Project under the name Harbour. If you copy code from other
* Harbour Project or Free Software Foundation releases into a copy of
* Harbour, as the General Public License permits, the exception does
* not apply to the code that you add in this way. To avoid misleading
* anyone as to the status of such modified files, you must delete
* this exception notice from them.
*
* If you write modifications of your own for Harbour, it is your choice
* whether to permit this exception to apply to your modifications.
* If you do not wish that, delete this exception notice.
*
*/
/*
* Using sqlite3_set_authorizer()
*
* This routine registers a authorizer callback with a particular
* database connection, supplied in the first argument.
* The authorizer callback is invoked as SQL statements are being compiled
* by sqlite3_prepare().
*
* When the callback returns SQLITE_OK, that means the operation requested
* is ok.
* When the callback returns SQLITE_DENY, the sqlite3_prepare() or
* equivalent call that triggered the authorizer will fail with an error
* message explaining that access is denied.
* If the authorizer code is SQLITE_READ and the callback returns
* SQLITE_IGNORE then the prepared statement statement is constructed to
* substitute a NULL value in place of the table column that would have
* been read if SQLITE_OK had been returned.
* The SQLITE_IGNORE return can be used to deny an untrusted user access
* to individual columns of a table.
*
* The first parameter to the authorizer callback is an integer
* action code that specifies the particular action to be authorized.
* The second through fourth parameters to the callback are strings
* that contain additional details about the action to be authorized.
*/
#require "hbsqlit3"
PROCEDURE Main()
LOCAL cFile := ":memory:"
LOCAL cSQLTEXT
LOCAL pDb, cb
CLS
IF Empty( pDb := PrepareDB( cFile ) )
ErrorLevel( 1 )
RETURN
ENDIF
// Authorizer1
sqlite3_set_authorizer( pDb, @Authorizer() /*"Authorizer"*/ )
? cSQLTEXT := "SELECT * FROM main.person WHERE age BETWEEN 20 AND 40"
cb := @CallBack() // "CallBack"
? cErrorMsg( sqlite3_exec( pDb, cSQLTEXT, cb ) )
sqlite3_sleep( 3000 )
// Authorizer2
? cErrorMsg( sqlite3_set_authorizer( pDb, @Authorizer2() /*"Authorizer2"*/ ) )
? cSQLTEXT := "SELECT * FROM main.person WHERE age BETWEEN 20 AND 40"
? cErrorMsg( sqlite3_exec( pDb, cSQLTEXT, cb ) )
sqlite3_sleep( 3000 )
// Authorizer3
? cErrorMsg( sqlite3_set_authorizer( pDb, @Authorizer3() /*"Authorizer3"*/ ) )
? cSQLTEXT := "SELECT * FROM main.person WHERE age BETWEEN 20 AND 40"
? cErrorMsg( sqlite3_exec( pDb, cSQLTEXT, cb ), .F. )
sqlite3_sleep( 3000 )
pDb := NIL // close database
RETURN
/**
*/
FUNCTION Authorizer( nAction, cName1, cName2, cDatabaseName, cTriggerOrViewName )
LOCAL oldColor := SetColor( "R/N" )
? "=>", StrZero( nAction, 2 ), cName1, cName2, cDatabaseName, cTriggerOrViewName
SetColor( oldColor )
RETURN SQLITE_OK
/**
*/
FUNCTION Authorizer2( nAction, cName1, cName2, cDatabaseName, cTriggerOrViewName )
LOCAL oldColor := SetColor( "R/N" )
? "=>", StrZero( nAction, 2 ), cName1, cName2, cDatabaseName, cTriggerOrViewName
SetColor( oldColor )
RETURN iif( cName2 == "pasw", SQLITE_IGNORE, SQLITE_OK )
/**
*/
FUNCTION Authorizer3( nAction, cName1, cName2, cDatabaseName, cTriggerOrViewName )
HB_SYMBOL_UNUSED( cName1 )
HB_SYMBOL_UNUSED( cName2 )
HB_SYMBOL_UNUSED( cDatabaseName )
HB_SYMBOL_UNUSED( cTriggerOrViewName )
RETURN iif( nAction == SQLITE_SELECT, SQLITE_DENY, SQLITE_OK )
/**
*/
FUNCTION CallBack( nColCount, aValue, aColName )
LOCAL nI
LOCAL oldColor := SetColor( "G/N" )
FOR nI := 1 TO nColCount
? PadR( aColName[ nI ], 5 ), " == ", aValue[ nI ]
NEXT
SetColor( oldColor )
RETURN 0
/**
*/
STATIC FUNCTION cErrorMsg( nError, lShortMsg )
hb_default( @lShortMsg, .T. )
RETURN iif( lShortMsg, hb_sqlite3_errstr_short( nError ), sqlite3_errstr( nError ) )
/**
*/
STATIC FUNCTION PrepareDB( cFile )
LOCAL cSQLTEXT
LOCAL pDb, pStmt
LOCAL hPerson := { ;
"Bob" => 52, ;
"Fred" => 32, ;
"Sasha" => 17, ;
"Andy" => 20, ;
"Ivet" => 28 ;
}, enum
pDb := sqlite3_open( cFile, .T. )
IF Empty( pDb )
? "Can't open/create database : ", cFile
RETURN NIL
ENDIF
cSQLTEXT := "CREATE TABLE person( name TEXT, age INTEGER, pasw TEXT(32) )"
IF sqlite3_exec( pDb, cSQLTEXT ) != SQLITE_OK
? "Can't create table : person"
pDb := NIL // close database
RETURN NIL
ENDIF
cSQLTEXT := "INSERT INTO person( name, age, pasw ) VALUES( :name, :age, :pasw )"
pStmt := sqlite3_prepare( pDb, cSQLTEXT )
IF Empty( pStmt )
? "Can't prepare statement : ", cSQLTEXT
pDb := NIL
RETURN NIL
ENDIF
FOR EACH enum IN hPerson
sqlite3_reset( pStmt )
sqlite3_bind_text( pStmt, 1, enum:__enumKey() )
sqlite3_bind_int( pStmt, 2, enum:__enumValue() )
sqlite3_bind_text( pStmt, 3, hb_MD5( enum:__enumKey() ) )
sqlite3_step( pStmt )
NEXT
sqlite3_clear_bindings( pStmt )
sqlite3_finalize( pStmt )
RETURN pDb