|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace palma; |
|
4
|
|
|
|
|
5
|
|
|
// Copyright (C) 2014 Universitätsbibliothek Mannheim |
|
6
|
|
|
// See file LICENSE for license details. |
|
7
|
|
|
|
|
8
|
|
|
// Authors: Alexander Wagner, Stefan Weil |
|
9
|
|
|
|
|
10
|
|
|
class DBConnector extends \SQLite3 |
|
11
|
|
|
{ |
|
12
|
|
|
private const SQL_CREATE_TABLES = <<< eod |
|
13
|
|
|
BEGIN EXCLUSIVE TRANSACTION; |
|
14
|
|
|
PRAGMA foreign_keys = ON; |
|
15
|
|
|
|
|
16
|
|
|
CREATE TABLE IF NOT EXISTS setting ( |
|
17
|
|
|
key VARCHAR (10) PRIMARY KEY, |
|
18
|
|
|
value VARCHAR (20) |
|
19
|
|
|
); |
|
20
|
|
|
INSERT OR IGNORE INTO setting VALUES ('layout', 'g1x1'); |
|
21
|
|
|
INSERT OR IGNORE INTO setting VALUES ('pin', ''); |
|
22
|
|
|
|
|
23
|
|
|
CREATE TABLE IF NOT EXISTS user ( |
|
24
|
|
|
userid INTEGER PRIMARY KEY, |
|
25
|
|
|
name VARCHAR (30) UNIQUE, |
|
26
|
|
|
count INTEGER, |
|
27
|
|
|
enabled INTEGER |
|
28
|
|
|
); |
|
29
|
|
|
|
|
30
|
|
|
-- Table with user name, IP address and device type (laptop, tablet, mobile). |
|
31
|
|
|
CREATE TABLE IF NOT EXISTS address ( |
|
32
|
|
|
userid INTEGER, |
|
33
|
|
|
address VARCHAR (30), |
|
34
|
|
|
device VARCHAR (6), |
|
35
|
|
|
FOREIGN KEY(userid) REFERENCES user(userid) |
|
36
|
|
|
); |
|
37
|
|
|
|
|
38
|
|
|
-- userid INTEGER REFERENCES user(userid) |
|
39
|
|
|
-- The software currently does not handle this correctly, so we had to remove |
|
40
|
|
|
-- the reference because it fails with pragma foreign_keys. |
|
41
|
|
|
CREATE TABLE IF NOT EXISTS window ( |
|
42
|
|
|
id INTEGER PRIMARY KEY, |
|
43
|
|
|
win_id VARCHAR (3), |
|
44
|
|
|
section INTEGER, |
|
45
|
|
|
state VARCHAR (10), |
|
46
|
|
|
file VARCHAR (255), |
|
47
|
|
|
handler VARCHAR (255), |
|
48
|
|
|
userid INTEGER, |
|
49
|
|
|
date DATETIME |
|
50
|
|
|
); |
|
51
|
|
|
END TRANSACTION; |
|
52
|
|
|
eod; |
|
53
|
|
|
|
|
54
|
|
|
private const SQL_RESET_TABLES = <<< eod |
|
55
|
|
|
DROP TABLE address; |
|
56
|
|
|
DROP TABLE setting; |
|
57
|
|
|
DROP TABLE IF EXISTS settings; |
|
58
|
|
|
DROP TABLE user; |
|
59
|
|
|
DROP TABLE window; |
|
60
|
|
|
eod; |
|
61
|
|
|
|
|
62
|
|
|
private static ?DBConnector $instance = null; |
|
63
|
|
|
|
|
64
|
|
|
public static function getInstance(): DBConnector |
|
65
|
|
|
{ |
|
66
|
|
|
if (is_null(self::$instance)) { |
|
67
|
|
|
self::$instance = new DBConnector('palma.db'); |
|
68
|
|
|
} |
|
69
|
|
|
return self::$instance; |
|
|
|
|
|
|
70
|
|
|
} |
|
71
|
|
|
|
|
72
|
|
|
private function __construct(string $filename) |
|
73
|
|
|
{ |
|
74
|
|
|
parent::__construct($filename); |
|
75
|
|
|
|
|
76
|
|
|
// Wait up to 10000 ms when the database is locked. |
|
77
|
|
|
$this->busyTimeout(10000); |
|
78
|
|
|
|
|
79
|
|
|
// Create any missing tables. |
|
80
|
|
|
$this->exec(self::SQL_CREATE_TABLES); |
|
81
|
|
|
} |
|
82
|
|
|
|
|
83
|
|
|
public function resetTables(): void |
|
84
|
|
|
{ |
|
85
|
|
|
$this->exec(self::SQL_RESET_TABLES . self::SQL_CREATE_TABLES); |
|
86
|
|
|
} |
|
87
|
|
|
|
|
88
|
|
|
public function countWindows(): int |
|
89
|
|
|
{ |
|
90
|
|
|
$numRows = $this->querySingle('SELECT count(*) FROM window WHERE state="active"'); |
|
91
|
|
|
return $numRows; |
|
92
|
|
|
} |
|
93
|
|
|
|
|
94
|
|
|
public function nextID(): int |
|
95
|
|
|
{ |
|
96
|
|
|
// Find the first unused monitor section and return its number. |
|
97
|
|
|
$quadrant_ids = array(1, 2, 3, 4); |
|
98
|
|
|
$window_db_ids = array(); |
|
99
|
|
|
|
|
100
|
|
|
$window_keys = @$this->query('SELECT DISTINCT(id) FROM window'); |
|
101
|
|
|
while ($row = $window_keys->fetchArray()) { |
|
102
|
|
|
array_push($window_db_ids, $row['id']); |
|
103
|
|
|
} |
|
104
|
|
|
$window_keys->finalize(); |
|
105
|
|
|
$next_ids = array_values(array_diff($quadrant_ids, $window_db_ids)); |
|
106
|
|
|
if (count($next_ids) > 0) { |
|
107
|
|
|
$next = $next_ids[0]; |
|
108
|
|
|
} else { |
|
109
|
|
|
$next = $this->querySingle('SELECT MAX(id) FROM window') + 1; |
|
110
|
|
|
} |
|
111
|
|
|
return $next; |
|
112
|
|
|
} |
|
113
|
|
|
|
|
114
|
|
|
public function ipAddress(): string |
|
115
|
|
|
{ |
|
116
|
|
|
$ip = 'unknown'; |
|
117
|
|
|
if (isset($_SERVER['HTTP_X_FORWARED_FOR'])) { |
|
118
|
|
|
// Server is hidden behind a proxy. |
|
119
|
|
|
// Proxy for loopback ip might indicate spoofing |
|
120
|
|
|
if (!in_array($_SERVER['HTTP_X_FORWARED_FOR'], array('127.0.0.1', '::1'))) { |
|
121
|
|
|
$ip = $_SERVER['HTTP_X_FORWARED_FOR']; |
|
122
|
|
|
} |
|
123
|
|
|
} elseif (isset($_SERVER['REMOTE_ADDR'])) { |
|
124
|
|
|
// Client has direct access to the server. |
|
125
|
|
|
$ip = $_SERVER['REMOTE_ADDR']; |
|
126
|
|
|
} |
|
127
|
|
|
return $ip; |
|
128
|
|
|
} |
|
129
|
|
|
|
|
130
|
|
|
public function checkPermission(): bool |
|
131
|
|
|
{ |
|
132
|
|
|
// Check whether the remote IP address is allowed to run commands. |
|
133
|
|
|
// Only localhost or hosts of registered users are allowed. |
|
134
|
|
|
$ip = $this->ipAddress(); |
|
135
|
|
|
// Currently PalMA also makes local connections. |
|
136
|
|
|
$ip_list = array('127.0.0.1', '::1'); |
|
137
|
|
|
$rows = $this->query("SELECT address FROM address"); |
|
138
|
|
|
while ($row = $rows->fetchArray(SQLITE3_ASSOC)) { |
|
139
|
|
|
array_push($ip_list, $row['address']); |
|
140
|
|
|
} |
|
141
|
|
|
$rows->finalize(); |
|
142
|
|
|
$allowed = in_array($ip, $ip_list); |
|
143
|
|
|
return $allowed; |
|
144
|
|
|
} |
|
145
|
|
|
|
|
146
|
|
|
public function addUser(string $username, string $address, string $device = 'laptop'): void |
|
147
|
|
|
{ |
|
148
|
|
|
// Add a new user with his/her address and the device to the database. |
|
149
|
|
|
$this->exec("INSERT OR IGNORE INTO user VALUES (NULL, '$username', 1, 0)"); |
|
150
|
|
|
$usercount = intval($this->querySingle("SELECT COUNT(*) FROM user")); |
|
151
|
|
|
$userid = $this->querySingle("SELECT userid from user where name='" . $username . "'"); |
|
152
|
|
|
$this->exec("INSERT INTO address VALUES ('$userid', '$address', '$device')"); |
|
153
|
|
|
require_once 'globals.php'; |
|
154
|
|
|
debug("DBConnector::addUser: user $username connected with $device, $usercount user(s) now connected"); |
|
155
|
|
|
if ($usercount == 1) { |
|
156
|
|
|
// First user connected. Always enable this person. |
|
157
|
|
|
$this->enableUser($username); |
|
158
|
|
|
} |
|
159
|
|
|
} |
|
160
|
|
|
|
|
161
|
|
|
public function delUser(string $username, string $address): void |
|
|
|
|
|
|
162
|
|
|
{ |
|
163
|
|
|
// Remove an existing user with his/her address from the database. |
|
164
|
|
|
$ip = $this->ipAddress(); |
|
165
|
|
|
$userid = $this->querySingle("SELECT userid from user where name='$username'"); |
|
166
|
|
|
if (!is_null($userid)) { |
|
167
|
|
|
// Kill VNC connection associated with the user |
|
168
|
|
|
$this->deleteVNCWindow($userid); |
|
169
|
|
|
$this->exec("DELETE FROM user WHERE userid = '$userid'"); |
|
170
|
|
|
} |
|
171
|
|
|
$this->exec("DELETE FROM address WHERE userid = '$userid' AND address = '$ip'"); |
|
172
|
|
|
$usercount = intval($this->querySingle("SELECT COUNT(*) FROM user")); |
|
173
|
|
|
require_once 'globals.php'; |
|
174
|
|
|
debug("DBConnector::delUser: user $username disconnected, $usercount user(s) now connected"); |
|
175
|
|
|
if ($usercount == 0) { |
|
176
|
|
|
// Last user disconnected. |
|
177
|
|
|
// Clean some tables, just to be sure that nothing is left. |
|
178
|
|
|
$this->exec("DELETE FROM address; DELETE FROM window;"); |
|
179
|
|
|
} |
|
180
|
|
|
} |
|
181
|
|
|
|
|
182
|
|
|
public function enableUser(string $username): void |
|
183
|
|
|
{ |
|
184
|
|
|
$state = $this->exec("UPDATE user SET enabled=1 WHERE name='$username'"); |
|
185
|
|
|
if (!$state) { |
|
186
|
|
|
require_once 'globals.php'; |
|
187
|
|
|
debug("DBConnector::enableUser: failed for $username"); |
|
188
|
|
|
} |
|
189
|
|
|
} |
|
190
|
|
|
|
|
191
|
|
|
/** @return array<string> */ |
|
192
|
|
|
public function getUsers(): array |
|
193
|
|
|
{ |
|
194
|
|
|
$users = array(); |
|
195
|
|
|
$rows = $this->query("SELECT name FROM user"); |
|
196
|
|
|
while ($row = $rows->fetchArray(SQLITE3_ASSOC)) { |
|
197
|
|
|
array_push($users, $row['name']); |
|
198
|
|
|
} |
|
199
|
|
|
$rows->finalize(); |
|
200
|
|
|
return $users; |
|
201
|
|
|
} |
|
202
|
|
|
|
|
203
|
|
|
public function checkUser(string $username): bool |
|
204
|
|
|
{ |
|
205
|
|
|
$users = $this->getUsers(); |
|
206
|
|
|
// empty db (e.g. after PalMA restart) or old session |
|
207
|
|
|
if (count($users) === 0 || !in_array($username, $users)) { |
|
208
|
|
|
return false; |
|
209
|
|
|
} |
|
210
|
|
|
return true; |
|
211
|
|
|
} |
|
212
|
|
|
|
|
213
|
|
|
/** @return array<array<string>> */ |
|
214
|
|
|
public function getWindows(): array |
|
215
|
|
|
{ |
|
216
|
|
|
// Get list of all windows, ordered by their section. |
|
217
|
|
|
$window_objs = array(); |
|
218
|
|
|
$windows = @$this->query('SELECT * FROM window ORDER BY section ASC'); |
|
219
|
|
|
while ($row = $windows->fetchArray()) { |
|
220
|
|
|
array_push($window_objs, $row); |
|
221
|
|
|
} |
|
222
|
|
|
$windows->finalize(); |
|
223
|
|
|
|
|
224
|
|
|
return $window_objs; |
|
225
|
|
|
} |
|
226
|
|
|
|
|
227
|
|
|
public function getWindowIDBySection(string $section): ?string |
|
228
|
|
|
{ |
|
229
|
|
|
$id = $this->querySingle("SELECT win_id FROM window WHERE section='$section'"); |
|
230
|
|
|
return $id; |
|
231
|
|
|
} |
|
232
|
|
|
|
|
233
|
|
|
/** @return array<array<string>> */ |
|
234
|
|
|
public function getVNCClientInfo(): array |
|
235
|
|
|
{ |
|
236
|
|
|
|
|
237
|
|
|
$info = array(); |
|
238
|
|
|
|
|
239
|
|
|
$i = @$this->query('SELECT * FROM window WHERE handler="vnc"'); |
|
240
|
|
|
while ($row = $i->fetchArray()) { |
|
241
|
|
|
array_push($info, $row); |
|
242
|
|
|
} |
|
243
|
|
|
$i->finalize(); |
|
244
|
|
|
|
|
245
|
|
|
return $info; |
|
246
|
|
|
} |
|
247
|
|
|
|
|
248
|
|
|
/* |
|
249
|
|
|
public function getVNC_ClientWindowIDs() { |
|
250
|
|
|
$ids = array(); |
|
251
|
|
|
$rows = @$this->query('SELECT win_id FROM window WHERE handler="vnc"'); |
|
252
|
|
|
while ($row = $rows->fetchArray()) { |
|
253
|
|
|
array_push($ids, $row['win_id']); |
|
254
|
|
|
} |
|
255
|
|
|
$rows->finalize(); |
|
256
|
|
|
return $ids; |
|
257
|
|
|
} |
|
258
|
|
|
*/ |
|
259
|
|
|
|
|
260
|
|
|
public function getWindowState(string $window_id): string |
|
261
|
|
|
{ |
|
262
|
|
|
$state = @$this->querySingle('SELECT state FROM window WHERE win_id="' . $window_id . '"'); |
|
263
|
|
|
return $state; |
|
264
|
|
|
} |
|
265
|
|
|
|
|
266
|
|
|
public function setWindowState(string $window_id, string $state): void |
|
267
|
|
|
{ |
|
268
|
|
|
$this->exec('UPDATE window SET state="' . $state . '" WHERE win_id="' . $window_id . '"'); |
|
269
|
|
|
} |
|
270
|
|
|
|
|
271
|
|
|
/** |
|
272
|
|
|
* @param array<string> $window |
|
273
|
|
|
*/ |
|
274
|
|
|
public function insertWindow(array $window): void |
|
275
|
|
|
{ |
|
276
|
|
|
// transfer ob complete window object/array necessary |
|
277
|
|
|
$sql = 'INSERT INTO window (id, win_id, section, state, file, handler, userid, date) ' . |
|
278
|
|
|
'VALUES ' . '("' . |
|
279
|
|
|
$window[0] . '", "' . $window[1] . '", "' . |
|
280
|
|
|
$window[2] . '", "' . $window[3] . '", "' . |
|
281
|
|
|
$window[4] . '", "' . $window[5] . '", "' . |
|
282
|
|
|
$window[6] . '", "' . $window[7] . '")'; |
|
283
|
|
|
$new = $this->exec($sql); |
|
284
|
|
|
require_once 'globals.php'; |
|
285
|
|
|
debug("DBConnector::insertWindow: sql=$sql, result=$new"); |
|
286
|
|
|
} |
|
287
|
|
|
|
|
288
|
|
|
public function deleteWindow(string $window_id): void |
|
289
|
|
|
{ |
|
290
|
|
|
$this->exec('DELETE FROM window WHERE win_id="' . $window_id . '"'); |
|
291
|
|
|
} |
|
292
|
|
|
|
|
293
|
|
|
public function deleteVNCWindow(string $userid): void |
|
294
|
|
|
{ |
|
295
|
|
|
$winid = $this->querySingle('SELECT win_id FROM window WHERE handler="vnc" AND userid="' . $userid . '"'); |
|
296
|
|
|
if (is_null($winid)) { |
|
297
|
|
|
return; |
|
298
|
|
|
} |
|
299
|
|
|
require_once 'globals.php'; |
|
300
|
|
|
wmClose($winid); |
|
301
|
|
|
$this->deleteWindow($winid); |
|
302
|
|
|
//$this->exec('DELETE FROM window WHERE handler="vnc" AND userid="'.$userid.'"'); |
|
303
|
|
|
} |
|
304
|
|
|
|
|
305
|
|
|
public function deleteDebug(string $table, string $id, string $gt): void |
|
306
|
|
|
{ |
|
307
|
|
|
$this->exec('DELETE FROM ' . $table . ' WHERE ' . $id . ' >"' . $gt . '"'); |
|
308
|
|
|
} |
|
309
|
|
|
|
|
310
|
|
|
public function updateWindow(?string $window_id, string $field, string $value): void |
|
311
|
|
|
{ |
|
312
|
|
|
if (is_null($window_id)) { |
|
313
|
|
|
return; |
|
314
|
|
|
} |
|
315
|
|
|
$this->exec('UPDATE window SET ' . $field . '="' . $value . '" WHERE win_id="' . $window_id . '"'); |
|
316
|
|
|
} |
|
317
|
|
|
} |
|
318
|
|
|
|