1 /*

     2  * Copyright (C) 2007 Apple Inc. All rights reserved.

     3  * Copyright (C) 2010 Google Inc. All rights reserved.

     4  *

     5  * Redistribution and use in source and binary forms, with or without

     6  * modification, are permitted provided that the following conditions

     7  * are met:

     8  *

     9  * 1.  Redistributions of source code must retain the above copyright

    10  *     notice, this list of conditions and the following disclaimer.

    11  * 2.  Redistributions in binary form must reproduce the above copyright

    12  *     notice, this list of conditions and the following disclaimer in the

    13  *     documentation and/or other materials provided with the distribution.

    14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of

    15  *     its contributors may be used to endorse or promote products derived

    16  *     from this software without specific prior written permission.

    17  *

    18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY

    19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

    20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

    21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY

    22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

    23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

    24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

    25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

    27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    28  */

    29 

    30 #include "config.h"

    31 #include "SQLStatementSync.h"

    32 

    33 #if ENABLE(DATABASE)

    34 

    35 #include "DatabaseSync.h"

    36 #include "SQLException.h"

    37 #include "SQLResultSet.h"

    38 #include "SQLValue.h"

    39 #include "SQLiteDatabase.h"

    40 #include "SQLiteStatement.h"

    41 #include <wtf/PassRefPtr.h>

    42 #include <wtf/RefPtr.h>

    43 

    44 namespace WebCore {

    45 

    46 SQLStatementSync::SQLStatementSync(const String& statement, const Vector<SQLValue>& arguments, bool readOnly)

    47     : m_statement(statement)

    48     , m_arguments(arguments)

    49     , m_readOnly(readOnly)

    50 {

    51     ASSERT(!m_statement.isEmpty());

    52 }

    53 

    54 PassRefPtr<SQLResultSet> SQLStatementSync::execute(DatabaseSync* db, ExceptionCode& ec)

    55 {

    56     if (m_readOnly)

    57         db->setAuthorizerReadOnly();

    58 

    59     SQLiteDatabase* database = &db->sqliteDatabase();

    60 

    61     SQLiteStatement statement(*database, m_statement);

    62     int result = statement.prepare();

    63     if (result != SQLResultOk) {

    64         ec = SQLException::SYNTAX_ERR;

    65         return 0;

    66     }

    67 

    68     if (statement.bindParameterCount() != m_arguments.size()) {

    69         ec = SQLException::SYNTAX_ERR;

    70         return 0;

    71     }

    72 

    73     for (unsigned i = 0; i < m_arguments.size(); ++i) {

    74         result = statement.bindValue(i + 1, m_arguments[i]);

    75         if (result == SQLResultFull) {

    76             ec = SQLException::QUOTA_ERR;

    77             return 0;

    78         }

    79 

    80         if (result != SQLResultOk) {

    81             ec = SQLException::DATABASE_ERR;

    82             return 0;

    83         }

    84     }

    85 

    86     RefPtr<SQLResultSet> resultSet = SQLResultSet::create();

    87 

    88     // Step so we can fetch the column names.

    89     result = statement.step();

    90     if (result == SQLResultRow) {

    91         int columnCount = statement.columnCount();

    92         SQLResultSetRowList* rows = resultSet->rows();

    93 

    94         for (int i = 0; i < columnCount; i++)

    95             rows->addColumn(statement.getColumnName(i));

    96 

    97         do {

    98             for (int i = 0; i < columnCount; i++)

    99                 rows->addResult(statement.getColumnValue(i));

   100 

   101             result = statement.step();

   102         } while (result == SQLResultRow);

   103 

   104         if (result != SQLResultDone) {

   105             ec = SQLException::DATABASE_ERR;

   106             return 0;

   107         }

   108     } else if (result == SQLResultDone) {

   109         // Didn't find anything, or was an insert.

   110         if (db->lastActionWasInsert())

   111             resultSet->setInsertId(database->lastInsertRowID());

   112     } else if (result == SQLResultFull) {

   113         // Quota error, the delegate will be asked for more space and this statement might be re-run.

   114         ec = SQLException::QUOTA_ERR;

   115         return 0;

   116     } else {

   117         ec = SQLException::DATABASE_ERR;

   118         return 0;

   119     }

   120 

   121     resultSet->setRowsAffected(database->lastChanges());

   122     return resultSet.release();

   123 }

   124 

   125 } // namespace WebCore

   126 

   127 #endif // ENABLE(DATABASE)