Devart Forums

Code: Select all

void __fastcall TForm1::Button3Click(TObject *Sender)
{
  TFileStream       *fs     = NULL;
  TSQLiteConnection *con    = NULL;
  UnicodeString     encKey  = L"abc123Ä"; // Will result in corruption of 'Db2.sqlite'


  // Create 2 empty database files
  // -----------------------------

  UnicodeString db1 = ExtractFilePath( ParamStr(0) ) + L"Db1.sqlite";
  UnicodeString db2 = ExtractFilePath( ParamStr(0) ) + L"Db2.sqlite";

  DeleteFile( db1 );
  fs = new TFileStream( db1, fmCreate | fmShareExclusive );
  delete( fs );
  fs = NULL;

  DeleteFile( db2 );
  fs = new TFileStream( db2, fmCreate | fmShareExclusive );
  delete( fs );
  fs = NULL;


  // Connect to database 'Db1' with blowfish encryption
  // --------------------------------------------------

  con = new TSQLiteConnection;

  con->SetProp( prDatabase,       db1 );
  con->SetProp( prStaticLibrary,  true );         // Always use static library
  con->SetProp( prBusyTimeout,    1000 );         // Add 1000ms busy timeout

  con->SetProp( prEncryptionAlgorithm, leBlowfish);
  con->SetProp( prEncryptionKey,  encKey );

  con->Connect( L"" );


  // Create a table in Db1
  // ---------------------

  con->ExecCommand( "CREATE TABLE Table1 ( Id INTEGER PRIMARY KEY, Field1 TEXT, Field2 TEXT );" );


  // Attach database Db2
  // -------------------

  UnicodeString attachCmd = UnicodeString::Format(  L"ATTACH DATABASE '%s' AS '%s' KEY '%s';",
                                            ARRAYOFCONST((
                                              db2,
                                              L"Db2",
                                              encKey )) );

  con->ExecCommand( attachCmd );

  // Create a table in Db2
  // ---------------------

  con->ExecCommand( L"CREATE TABLE Db2.Table2 ( Id INTEGER PRIMARY KEY, Field1 TEXT, Field2 TEXT );" );


  // Close databases and destroy connection object
  // ---------------------------------------------

  con->Disconnect();
  delete( con );
  con = NULL;


  // Verify databases by opening them
  // --------------------------------

  bool success = false;

  for( int idx=0; idx<2; idx++ ){

    UnicodeString dbFile = ( idx == 0 ) ? db1 : db2;

    con = new TSQLiteConnection;

    con->SetProp( prDatabase,       dbFile );
    con->SetProp( prStaticLibrary,  true );         // Always use static library
    con->SetProp( prBusyTimeout,    1000 );         // Add 1000ms busy timeout

    if( encKey.IsEmpty() == false ){
      con->SetProp( prEncryptionAlgorithm, leBlowfish );
      con->SetProp( prEncryptionKey, encKey );
    }

    try{
      con->Connect( L"" );

      // Db is successfully opened!

      ShowMessage( UnicodeString::Format( L"Success opening database %s!",
                                          ARRAYOFCONST((
                                            ExtractFileName( dbFile )
                                          )) ) );
    }
    catch( ... ){
      ShowMessage( UnicodeString::Format( L"Failure opening database %s!",
                                          ARRAYOFCONST((
                                            ExtractFileName( dbFile )
                                          )) ) );
    }

    delete( con );
    con = NULL;
  }
}