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, int permissions) 47 : m_statement(statement) 48 , m_arguments(arguments) 49 , m_permissions(permissions) 50 { 51 ASSERT(!m_statement.isEmpty()); 52 } 53 54 PassRefPtr<SQLResultSet> SQLStatementSync::execute(DatabaseSync* db, ExceptionCode& ec) 55 { 56 db->setAuthorizerPermissions(m_permissions); 57 58 SQLiteDatabase* database = &db->sqliteDatabase(); 59 60 SQLiteStatement statement(*database, m_statement); 61 int result = statement.prepare(); 62 if (result != SQLResultOk) { 63 ec = (result == SQLResultInterrupt ? SQLException::DATABASE_ERR : SQLException::SYNTAX_ERR); 64 return 0; 65 } 66 67 if (statement.bindParameterCount() != m_arguments.size()) { 68 ec = (db->isInterrupted()? SQLException::DATABASE_ERR : SQLException::SYNTAX_ERR); 69 return 0; 70 } 71 72 for (unsigned i = 0; i < m_arguments.size(); ++i) { 73 result = statement.bindValue(i + 1, m_arguments[i]); 74 if (result == SQLResultFull) { 75 ec = SQLException::QUOTA_ERR; 76 return 0; 77 } 78 79 if (result != SQLResultOk) { 80 ec = SQLException::DATABASE_ERR; 81 return 0; 82 } 83 } 84 85 RefPtr<SQLResultSet> resultSet = SQLResultSet::create(); 86 87 // Step so we can fetch the column names. 88 result = statement.step(); 89 if (result == SQLResultRow) { 90 int columnCount = statement.columnCount(); 91 SQLResultSetRowList* rows = resultSet->rows(); 92 93 for (int i = 0; i < columnCount; i++) 94 rows->addColumn(statement.getColumnName(i)); 95 96 do { 97 for (int i = 0; i < columnCount; i++) 98 rows->addResult(statement.getColumnValue(i)); 99 100 result = statement.step(); 101 } while (result == SQLResultRow); 102 103 if (result != SQLResultDone) { 104 ec = SQLException::DATABASE_ERR; 105 return 0; 106 } 107 } else if (result == SQLResultDone) { 108 // Didn't find anything, or was an insert. 109 if (db->lastActionWasInsert()) 110 resultSet->setInsertId(database->lastInsertRowID()); 111 } else if (result == SQLResultFull) { 112 // Quota error, the delegate will be asked for more space and this statement might be re-run. 113 ec = SQLException::QUOTA_ERR; 114 return 0; 115 } else { 116 ec = SQLException::DATABASE_ERR; 117 return 0; 118 } 119 120 resultSet->setRowsAffected(database->lastChanges()); 121 return resultSet.release(); 122 } 123 124 } // namespace WebCore 125 126 #endif // ENABLE(DATABASE) 127