1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
namespace LeKoala\Encrypt; |
4
|
|
|
|
5
|
|
|
use Exception; |
6
|
|
|
use SilverStripe\ORM\DataObject; |
7
|
|
|
use SilverStripe\Forms\TextField; |
8
|
|
|
use ParagonIE\CipherSweet\BlindIndex; |
9
|
|
|
use SilverStripe\ORM\Queries\SQLSelect; |
10
|
|
|
use ParagonIE\CipherSweet\EncryptedField; |
11
|
|
|
use SilverStripe\ORM\FieldType\DBComposite; |
12
|
|
|
use ParagonIE\CipherSweet\Exception\InvalidCiphertextException; |
13
|
|
|
|
14
|
|
|
/** |
15
|
|
|
* Value will be set on parent record through built in getField |
16
|
|
|
* mechanisms for composite fields |
17
|
|
|
*/ |
18
|
|
|
class EncryptedDBField extends DBComposite |
19
|
|
|
{ |
20
|
|
|
const LARGE_INDEX_SIZE = 32; |
21
|
|
|
const SMALL_INDEX_SIZE = 16; |
22
|
|
|
const VALUE_SUFFIX = "Value"; |
23
|
|
|
const INDEX_SUFFIX = "BlindIndex"; |
24
|
|
|
|
25
|
|
|
/** |
26
|
|
|
* @config |
27
|
|
|
* @var int |
28
|
|
|
*/ |
29
|
|
|
private static $output_size = 15; |
|
|
|
|
30
|
|
|
|
31
|
|
|
/** |
32
|
|
|
* @config |
33
|
|
|
* @var int |
34
|
|
|
*/ |
35
|
|
|
private static $domain_size = 127; |
|
|
|
|
36
|
|
|
|
37
|
|
|
/** |
38
|
|
|
* @param array |
39
|
|
|
*/ |
40
|
|
|
private static $composite_db = array( |
|
|
|
|
41
|
|
|
"Value" => "Varchar(191)", |
42
|
|
|
"BlindIndex" => 'Varchar(32)', |
43
|
|
|
); |
44
|
|
|
|
45
|
|
|
/** |
46
|
|
|
* @var Exception |
47
|
|
|
*/ |
48
|
|
|
protected $encryptionException; |
49
|
|
|
|
50
|
|
|
/** |
51
|
|
|
* @return Exception |
52
|
|
|
*/ |
53
|
|
|
public function getEncryptionException() |
54
|
|
|
{ |
55
|
|
|
return $this->encryptionException; |
56
|
|
|
} |
57
|
|
|
|
58
|
|
|
/** |
59
|
|
|
* Output size is the number of bits (not bytes) of a blind index. |
60
|
|
|
* Eg: 4 for a 4 digits year |
61
|
|
|
* Note: the larger the output size, the smaller the index should be |
62
|
|
|
* @return int |
63
|
|
|
*/ |
64
|
|
|
public function getOutputSize() |
65
|
|
|
{ |
66
|
|
|
if (array_key_exists('output_size', $this->options)) { |
67
|
|
|
$outputSize = $this->options['output_size']; |
68
|
|
|
} else { |
69
|
|
|
$outputSize = static::config()->get('output_size'); |
70
|
|
|
} |
71
|
|
|
return $outputSize; |
72
|
|
|
} |
73
|
|
|
|
74
|
|
|
/** |
75
|
|
|
* Input domain is the set of all possible distinct inputs. |
76
|
|
|
* Eg : 4 digits have 10,000 possible values (10^4). The log (base 2) of 10,000 is 13.2877; you would want to always round up (so 14). |
77
|
|
|
* @return int |
78
|
|
|
*/ |
79
|
|
|
public function getDomainSize() |
80
|
|
|
{ |
81
|
|
|
if (array_key_exists('domain_size', $this->options)) { |
82
|
|
|
$domainSize = $this->options['domain_size']; |
83
|
|
|
} else { |
84
|
|
|
$domainSize = static::config()->get('domain_size'); |
85
|
|
|
} |
86
|
|
|
return $domainSize; |
87
|
|
|
} |
88
|
|
|
|
89
|
|
|
/** |
90
|
|
|
* @param int $default |
91
|
|
|
* @return int |
92
|
|
|
*/ |
93
|
|
|
public function getIndexSize($default = null) |
94
|
|
|
{ |
95
|
|
|
if (array_key_exists('index_size', $this->options)) { |
96
|
|
|
return $this->options['index_size']; |
97
|
|
|
} |
98
|
|
|
if ($default) { |
|
|
|
|
99
|
|
|
return $default; |
100
|
|
|
} |
101
|
|
|
return self::LARGE_INDEX_SIZE; |
102
|
|
|
} |
103
|
|
|
|
104
|
|
|
/** |
105
|
|
|
* @return string |
106
|
|
|
*/ |
107
|
|
|
public function getValueField() |
108
|
|
|
{ |
109
|
|
|
return $this->getField(self::VALUE_SUFFIX); |
110
|
|
|
} |
111
|
|
|
|
112
|
|
|
/** |
113
|
|
|
* @return $this |
114
|
|
|
*/ |
115
|
|
|
public function setValueField($value, $markChanged = true) |
116
|
|
|
{ |
117
|
|
|
return $this->setField(self::VALUE_SUFFIX, $value, $markChanged); |
118
|
|
|
} |
119
|
|
|
|
120
|
|
|
/** |
121
|
|
|
* @return string |
122
|
|
|
*/ |
123
|
|
|
public function getBlindIndexField() |
124
|
|
|
{ |
125
|
|
|
return $this->getField(self::INDEX_SUFFIX); |
126
|
|
|
} |
127
|
|
|
|
128
|
|
|
/** |
129
|
|
|
* @return $this |
130
|
|
|
*/ |
131
|
|
|
public function setBlindIndexField($value, $markChanged = true) |
132
|
|
|
{ |
133
|
|
|
return $this->setField(self::INDEX_SUFFIX, $value, $markChanged); |
134
|
|
|
} |
135
|
|
|
|
136
|
|
|
/** |
137
|
|
|
* @param CipherSweet $engine |
|
|
|
|
138
|
|
|
* @return EncryptedField |
139
|
|
|
*/ |
140
|
|
|
public function getEncryptedField($engine = null) |
141
|
|
|
{ |
142
|
|
|
if ($engine === null) { |
143
|
|
|
$engine = EncryptHelper::getCipherSweet(); |
144
|
|
|
} |
145
|
|
|
$indexSize = $this->getIndexSize(self::LARGE_INDEX_SIZE); |
146
|
|
|
// fieldName needs to match exact db name for row rotator to work properly |
147
|
|
|
$encryptedField = (new EncryptedField($engine, $this->tableName, $this->name . self::VALUE_SUFFIX)) |
148
|
|
|
->addBlindIndex(new BlindIndex($this->name . self::INDEX_SUFFIX, [], $indexSize)); |
149
|
|
|
return $encryptedField; |
150
|
|
|
} |
151
|
|
|
|
152
|
|
|
/** |
153
|
|
|
* This is not called anymore, we rely on saveInto for now |
154
|
|
|
* @link https://github.com/silverstripe/silverstripe-framework/issues/8800 |
155
|
|
|
* @param array $manipulation |
156
|
|
|
* @return void |
157
|
|
|
*/ |
158
|
|
|
public function writeToManipulation(&$manipulation) |
159
|
|
|
{ |
160
|
|
|
$encryptedField = $this->getEncryptedField(); |
161
|
|
|
|
162
|
|
|
if ($this->value) { |
163
|
|
|
$dataForStorage = $encryptedField->prepareForStorage($this->value); |
164
|
|
|
$encryptedValue = $this->prepValueForDB($dataForStorage[0]); |
165
|
|
|
$blindIndexes = $dataForStorage[1]; |
166
|
|
|
} else { |
167
|
|
|
$encryptedValue = null; |
168
|
|
|
$blindIndexes = []; |
169
|
|
|
} |
170
|
|
|
|
171
|
|
|
$manipulation['fields'][$this->name . self::VALUE_SUFFIX] = $encryptedValue; |
172
|
|
|
$manipulation['fields'][$this->name . self::INDEX_SUFFIX] = $blindIndexes[$this->name . self::INDEX_SUFFIX] ?? null; |
173
|
|
|
} |
174
|
|
|
|
175
|
|
|
/** |
176
|
|
|
* @param SQLSelect $query |
177
|
|
|
*/ |
178
|
|
|
public function addToQuery(&$query) |
179
|
|
|
{ |
180
|
|
|
parent::addToQuery($query); |
181
|
|
|
$query->selectField(sprintf('"%s' . self::VALUE_SUFFIX . '"', $this->name)); |
182
|
|
|
$query->selectField(sprintf('"%s' . self::INDEX_SUFFIX . '"', $this->name)); |
183
|
|
|
} |
184
|
|
|
|
185
|
|
|
/** |
186
|
|
|
* Return the blind index value to search in the database |
187
|
|
|
* |
188
|
|
|
* @param string $val The unencrypted value |
189
|
|
|
* @param string $indexSuffix The blind index. Defaults to full index |
190
|
|
|
* @return string |
191
|
|
|
*/ |
192
|
|
|
public function getSearchValue($val, $indexSuffix = null) |
193
|
|
|
{ |
194
|
|
|
if (!$this->tableName && $this->record) { |
195
|
|
|
$this->tableName = DataObject::getSchema()->tableName(get_class($this->record)); |
|
|
|
|
196
|
|
|
} |
197
|
|
|
if (!$this->tableName) { |
198
|
|
|
throw new Exception("Table name not set for search value. Please set a dataobject."); |
199
|
|
|
} |
200
|
|
|
if (!$this->name) { |
201
|
|
|
throw new Exception("Name not set for search value"); |
202
|
|
|
} |
203
|
|
|
if ($indexSuffix === null) { |
204
|
|
|
$indexSuffix = self::INDEX_SUFFIX; |
205
|
|
|
} |
206
|
|
|
$field = $this->getEncryptedField(); |
207
|
|
|
$index = $field->getBlindIndex($val, $this->name . $indexSuffix); |
208
|
|
|
return $index; |
|
|
|
|
209
|
|
|
} |
210
|
|
|
|
211
|
|
|
/** |
212
|
|
|
* Return a ready to use array params for a where clause |
213
|
|
|
* |
214
|
|
|
* @param string $val The unencrypted value |
215
|
|
|
* @param string $indexSuffix The blind index. Defaults to full index |
216
|
|
|
* @return array |
217
|
|
|
*/ |
218
|
|
|
public function getSearchParams($val, $indexSuffix = null) |
219
|
|
|
{ |
220
|
|
|
if (!$indexSuffix) { |
221
|
|
|
$indexSuffix = self::INDEX_SUFFIX; |
222
|
|
|
} |
223
|
|
|
$searchValue = $this->getSearchValue($val, $indexSuffix); |
224
|
|
|
$blindIndexField = $this->name . $indexSuffix; |
225
|
|
|
return array($blindIndexField . ' = ?' => $searchValue); |
226
|
|
|
} |
227
|
|
|
|
228
|
|
|
/** |
229
|
|
|
* @param string $val The unencrypted value |
230
|
|
|
* @param string $indexSuffix The blind index. Defaults to full index |
231
|
|
|
* @return DataList |
|
|
|
|
232
|
|
|
*/ |
233
|
|
|
public function fetchDataList($val, $indexSuffix = null) |
234
|
|
|
{ |
235
|
|
|
if (!$this->record) { |
236
|
|
|
throw new Exception("No record set for this field"); |
237
|
|
|
} |
238
|
|
|
if (!$indexSuffix) { |
239
|
|
|
$indexSuffix = self::INDEX_SUFFIX; |
240
|
|
|
} |
241
|
|
|
$class = get_class($this->record); |
|
|
|
|
242
|
|
|
|
243
|
|
|
// A blind index can return false positives |
244
|
|
|
$params = $this->getSearchParams($val, $indexSuffix); |
245
|
|
|
$blindIndexes = $this->getEncryptedField()->getBlindIndexObjects(); |
|
|
|
|
246
|
|
|
$list = $class::get()->where($params); |
247
|
|
|
return $list; |
248
|
|
|
} |
249
|
|
|
|
250
|
|
|
/** |
251
|
|
|
* @param string $val The unencrypted value |
252
|
|
|
* @param string $indexSuffix The blind index. Defaults to full index |
253
|
|
|
* @return DataObject |
254
|
|
|
*/ |
255
|
|
|
public function fetchRecord($val, $indexSuffix = null) |
256
|
|
|
{ |
257
|
|
|
if (!$indexSuffix) { |
258
|
|
|
$indexSuffix = self::INDEX_SUFFIX; |
259
|
|
|
} |
260
|
|
|
$list = $this->fetchDataList($val, $indexSuffix); |
261
|
|
|
$blindIndexes = $this->getEncryptedField()->getBlindIndexObjects(); |
262
|
|
|
$blindIndex = $blindIndexes[$this->name . $indexSuffix]; |
263
|
|
|
$name = $this->name; |
264
|
|
|
/** @var DataObject $record */ |
265
|
|
|
foreach ($list as $record) { |
266
|
|
|
$obj = $record->dbObject($name); |
267
|
|
|
// Value might be transformed |
268
|
|
|
if ($blindIndex->getTransformed($obj->getValue()) == $val) { |
269
|
|
|
return $record; |
270
|
|
|
} |
271
|
|
|
} |
272
|
|
|
// throw exception if there where matches but none with the right value |
273
|
|
|
if ($list->count()) { |
274
|
|
|
throw new Exception($list->count() . " records were found but none matched the right value"); |
275
|
|
|
} |
276
|
|
|
return false; |
|
|
|
|
277
|
|
|
} |
278
|
|
|
|
279
|
|
|
public function setValue($value, $record = null, $markChanged = true) |
280
|
|
|
{ |
281
|
|
|
if ($markChanged) { |
282
|
|
|
$this->isChanged = true; |
283
|
|
|
} |
284
|
|
|
|
285
|
|
|
// When given a dataobject, bind this field to that |
286
|
|
|
if ($record instanceof DataObject) { |
287
|
|
|
$this->bindTo($record); |
288
|
|
|
} |
289
|
|
|
|
290
|
|
|
// Convert an object to an array |
291
|
|
|
if ($record && $record instanceof DataObject) { |
292
|
|
|
$record = $record->getQueriedDatabaseFields(); |
293
|
|
|
if (!$record) { |
|
|
|
|
294
|
|
|
throw new Exception("Could not convert record to array"); |
295
|
|
|
} |
296
|
|
|
} |
297
|
|
|
|
298
|
|
|
// Set the table name if it was not set earlier |
299
|
|
|
if (!$this->tableName && $record) { |
300
|
|
|
$this->tableName = DataObject::getSchema()->tableName(get_class($record)); |
301
|
|
|
if (!$this->tableName) { |
302
|
|
|
throw new Exception("Could not get table name from record from " . gettype($record)); |
303
|
|
|
} |
304
|
|
|
} |
305
|
|
|
|
306
|
|
|
// Value will store the decrypted value |
307
|
|
|
if ($value instanceof EncryptedDBField) { |
308
|
|
|
$this->value = $value->getValue(); |
309
|
|
|
} elseif ($record && isset($record[$this->name . self::VALUE_SUFFIX])) { |
310
|
|
|
// In that case, the value come from the database and might be encrypted |
311
|
|
|
$encryptedValue = $record[$this->name . self::VALUE_SUFFIX]; |
312
|
|
|
|
313
|
|
|
// It should always be encrypted from the db, but just in case... |
314
|
|
|
if ($encryptedValue && EncryptHelper::isEncrypted($encryptedValue)) { |
315
|
|
|
try { |
316
|
|
|
$this->value = $this->getEncryptedField()->decryptValue($encryptedValue); |
317
|
|
|
} catch (InvalidCiphertextException $ex) { |
318
|
|
|
$this->encryptionException = $ex; |
319
|
|
|
// rotate backend ? |
320
|
|
|
if (EncryptHelper::getAutomaticRotation()) { |
321
|
|
|
$encryption = EncryptHelper::getEncryption($encryptedValue); |
322
|
|
|
$engine = EncryptHelper::getEngineForEncryption($encryption); |
323
|
|
|
$oldEncryptedField = $this->getEncryptedField($engine); |
324
|
|
|
$this->value = $oldEncryptedField->decryptValue($encryptedValue); |
325
|
|
|
} else { |
326
|
|
|
$this->value = $encryptedValue; |
327
|
|
|
} |
328
|
|
|
} catch (Exception $ex) { |
329
|
|
|
$this->encryptionException = $ex; |
330
|
|
|
// We cannot decrypt |
331
|
|
|
$this->value = $this->nullValue(); |
|
|
|
|
332
|
|
|
} |
333
|
|
|
} else { |
334
|
|
|
if ($encryptedValue) { |
335
|
|
|
$this->value = $encryptedValue; |
336
|
|
|
} else { |
337
|
|
|
$this->value = $this->nullValue(); |
|
|
|
|
338
|
|
|
} |
339
|
|
|
} |
340
|
|
|
} elseif (is_array($value)) { |
341
|
|
|
if (array_key_exists(self::VALUE_SUFFIX, $value)) { |
342
|
|
|
$this->value = $value; |
343
|
|
|
} |
344
|
|
|
} elseif (is_string($value) || !$value) { |
345
|
|
|
$this->value = $value; |
346
|
|
|
} else { |
347
|
|
|
throw new Exception("Unexcepted value of type " . gettype($value)); |
348
|
|
|
} |
349
|
|
|
|
350
|
|
|
// Forward changes since writeToManipulation are not called |
351
|
|
|
// $this->setValueField($value, $markChanged); |
352
|
|
|
|
353
|
|
|
return $this; |
354
|
|
|
} |
355
|
|
|
|
356
|
|
|
/** |
357
|
|
|
* @return string |
358
|
|
|
*/ |
359
|
|
|
public function Nice($options = array()) |
|
|
|
|
360
|
|
|
{ |
361
|
|
|
return $this->getValue(); |
362
|
|
|
} |
363
|
|
|
|
364
|
|
|
/** |
365
|
|
|
* @return boolean |
366
|
|
|
*/ |
367
|
|
|
public function exists() |
368
|
|
|
{ |
369
|
|
|
return strlen($this->value) > 0; |
370
|
|
|
} |
371
|
|
|
|
372
|
|
|
/** |
373
|
|
|
* This is called by getChangedFields() to check if a field is changed |
374
|
|
|
* |
375
|
|
|
* @return boolean |
376
|
|
|
*/ |
377
|
|
|
public function isChanged() |
378
|
|
|
{ |
379
|
|
|
return $this->isChanged; |
380
|
|
|
} |
381
|
|
|
|
382
|
|
|
/** |
383
|
|
|
* If we pass a DBField to the setField method, it will |
384
|
|
|
* trigger this method |
385
|
|
|
* |
386
|
|
|
* We save encrypted value on sub fields. They will be collected |
387
|
|
|
* by write() operation by prepareManipulationTable |
388
|
|
|
* |
389
|
|
|
* Currently prepareManipulationTable ignores composite fields |
390
|
|
|
* so we rely on the sub field mechanisms |
391
|
|
|
* |
392
|
|
|
* @param DataObject $dataObject |
393
|
|
|
* @return void |
394
|
|
|
*/ |
395
|
|
|
public function saveInto($dataObject) |
396
|
|
|
{ |
397
|
|
|
$encryptedField = $this->getEncryptedField(); |
398
|
|
|
|
399
|
|
|
if ($this->value) { |
400
|
|
|
$dataForStorage = $encryptedField->prepareForStorage($this->value); |
401
|
|
|
$encryptedValue = $this->prepValueForDB($dataForStorage[0]); |
402
|
|
|
$blindIndexes = $dataForStorage[1]; |
403
|
|
|
} else { |
404
|
|
|
$encryptedValue = null; |
405
|
|
|
$blindIndexes = []; |
406
|
|
|
} |
407
|
|
|
|
408
|
|
|
// This cause infinite loops |
409
|
|
|
// $dataObject->setField($this->getName(), $this->value); |
410
|
|
|
|
411
|
|
|
// Encrypt value |
412
|
|
|
$key = $this->getName() . 'Value'; |
413
|
|
|
$dataObject->setField($key, $encryptedValue); |
414
|
|
|
|
415
|
|
|
// Build blind index |
416
|
|
|
$key = $this->getName() . self::INDEX_SUFFIX; |
417
|
|
|
if (isset($blindIndexes[$key])) { |
418
|
|
|
$dataObject->setField($key, $blindIndexes[$key]); |
419
|
|
|
} |
420
|
|
|
} |
421
|
|
|
|
422
|
|
|
/** |
423
|
|
|
* @param string $title Optional. Localized title of the generated instance |
424
|
|
|
* @return FormField |
|
|
|
|
425
|
|
|
*/ |
426
|
|
|
public function scaffoldFormField($title = null, $params = null) |
427
|
|
|
{ |
428
|
|
|
$field = TextField::create($this->name); |
429
|
|
|
return $field; |
430
|
|
|
} |
431
|
|
|
|
432
|
|
|
/** |
433
|
|
|
* Returns the string value |
434
|
|
|
*/ |
435
|
|
|
public function __toString() |
436
|
|
|
{ |
437
|
|
|
return (string) $this->getValue(); |
438
|
|
|
} |
439
|
|
|
|
440
|
|
|
public function scalarValueOnly() |
441
|
|
|
{ |
442
|
|
|
return false; |
443
|
|
|
} |
444
|
|
|
} |
445
|
|
|
|