kjuulh/rhai
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects Releases Wiki Activity

Add missing data size check.

Browse Source
This commit is contained in:
Stephen Chung
2022-01-28 08:28:31 +08:00
parent b9f2fdb635
commit b86dd3f586
2 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/eval/stmt.rs
View File

@@ -127,6 +127,7 @@ impl Engine {
if let Some(OpAssignment {
hash_op_assign,
hash_op,
op_assign,
op,
}) = op_info
{
@@ -148,21 +149,23 @@ impl Engine {
let hash = hash_op_assign;
let args = &mut [lhs_ptr_inner, &mut new_val];
match self.call_native_fn(global, state, lib, op, hash, args, true, true, op_pos) {
match self.call_native_fn(
global, state, lib, op_assign, hash, args, true, true, op_pos,
) {
Ok(_) => {
#[cfg(not(feature = "unchecked"))]
self.check_data_size(&mut args[0], root.1)?;
self.check_data_size(&args[0], root.1)?;
}
Err(err) if matches!(*err, ERR::ErrorFunctionNotFound(ref f, _) if f.starts_with(op)) =>
Err(err) if matches!(*err, ERR::ErrorFunctionNotFound(ref f, _) if f.starts_with(op_assign)) =>
{
// Expand to `var = var op rhs`
let op = &op[..op.len() - 1]; // extract operator without =
// Run function
let (value, _) = self.call_native_fn(
global, state, lib, op, hash_op, args, true, false, op_pos,
)?;
#[cfg(not(feature = "unchecked"))]
self.check_data_size(&value, root.1)?;
*args[0] = value.flatten();
}
Err(err) => return Err(err),