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
SQLStatementSync(const String & statement,const Vector<SQLValue> & arguments,int permissions)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
execute(DatabaseSync * db,ExceptionCode & ec)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