asserts should not be used instead of ordinary input checks. Yet the native DNN backend did it: get_input_native() asserted that the first dimension was one, despite this value coming directly from the input file without having been sanitized.
Showing the changes to libavfilter/dnn (1 of 1 file). Show all
Unified · Split
| @@ -232,6 +232,8 @@ DNNModel *ff_dnn_load_model_native(const char *model_filename, DNNFunctionType f · 231 unmodified lines | |||
| 232 | oprd->dims[dim] = (int32_t)avio_rl32(model_file_context); | 232 | oprd->dims[dim] = (int32_t)avio_rl32(model_file_context); |
| 233 | dnn_size += 4; | 233 | dnn_size += 4; |
| 234 | } | 234 | } |
| 235 | if (oprd->type == DOT_INPUT && oprd->dims[0] != 1) | ||
| 236 | goto fail; | ||
| 235 | 237 | ||
| 236 | oprd->isNHWC = 1; | 238 | oprd->isNHWC = 1; |
| 237 | } | 239 | } |