Skip to content

Commit 072ad36

Browse files
committed
Detect invalid content (#458)
1 parent 0df7d35 commit 072ad36

File tree

3 files changed

+3
-6
lines changed

3 files changed

+3
-6
lines changed

source/ncnn.js

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,6 @@ ncnn.ModelFactory = class {
3939
}
4040
}
4141
if (identifier.endsWith('.bin') || identifier.endsWith('.weights.ncnn')) {
42-
if (identifier == 'snapshot_blob.bin' || identifier === 'v8_context_snapshot.bin') {
43-
return undefined;
44-
}
4542
const stream = context.stream;
4643
if (stream.length > 4) {
4744
const buffer = stream.peek(4);

source/openvino.js

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,6 @@ openvino.ModelFactory = class {
88
const identifier = context.identifier;
99
const extension = identifier.split('.').pop().toLowerCase();
1010
if (extension === 'bin') {
11-
if (identifier === 'natives_blob.bin' || identifier === 'snapshot_blob.bin' || identifier === 'v8_context_snapshot.bin') {
12-
return undefined;
13-
}
1411
const stream = context.stream;
1512
const signature = [ 0x21, 0xA8, 0xEF, 0xBE, 0xAD, 0xDE ];
1613
if (signature.length <= stream.length && stream.peek(signature.length).every((value, index) => value === signature[index])) {

source/view.js

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5584,6 +5584,9 @@ view.ModelFactoryService = class {
55845584
{ name: 'TSD header', value: /^%TSD-Header-###%/ },
55855585
{ name: 'AppleDouble data', value: /^\x00\x05\x16\x07/ },
55865586
{ name: 'TensorFlow Hub module', value: /^\x08\x03$/, identifier: 'tfhub_module.pb' },
5587+
{ name: 'V8 snapshot', value: /^.\x00\x00\x00.\x00\x00\x00/, identifier: 'snapshot_blob.bin' },
5588+
{ name: 'V8 context snapshot', value: /^.\x00\x00\x00.\x00\x00\x00/, identifier: 'v8_context_snapshot.bin' },
5589+
{ name: 'V8 natives blob', value: /^./, identifier: 'natives_blob.bin' },
55875590
{ name: 'ViSQOL model', value: /^svm_type\s/ },
55885591
{ name: 'SenseTime model', value: /^STEF/ }
55895592
];

0 commit comments

Comments
 (0)