ISAR/rusqlite
3
0
Fork 0
mirror of https://github.com/isar/rusqlite.git synced 2024-11-24 09:59:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Interpret generate_series arguments as possibly NULL

Browse Source
This commit is contained in:
David Selassie 2023-07-06 17:23:28 -07:00
parent 84362c02c6
commit 3a2312e0bd
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/vtab/series.rs
View File

@ -8,7 +8,6 @@ use std::marker::PhantomData;
use std::os::raw::c_int; use std::os::raw::c_int;
use crate::ffi; use crate::ffi;
use crate::types::Type;
use crate::vtab::{ use crate::vtab::{
eponymous_only_module, Context, IndexConstraintOp, IndexInfo, VTab, VTabConfig, VTabConnection, eponymous_only_module, Context, IndexConstraintOp, IndexInfo, VTab, VTabConfig, VTabConnection,
VTabCursor, Values, VTabCursor, Values,
@ -198,21 +197,34 @@ impl SeriesTabCursor<'_> {
unsafe impl VTabCursor for SeriesTabCursor<'_> { unsafe impl VTabCursor for SeriesTabCursor<'_> {
fn filter(&mut self, idx_num: c_int, _idx_str: Option<&str>, args: &Values<'_>) -> Result<()> { fn filter(&mut self, idx_num: c_int, _idx_str: Option<&str>, args: &Values<'_>) -> Result<()> {
let mut idx_num = QueryPlanFlags::from_bits_truncate(idx_num); let mut idx_num = QueryPlanFlags::from_bits_truncate(idx_num);
let mut any_null = false;
let mut i = 0; let mut i = 0;
if idx_num.contains(QueryPlanFlags::START) { if idx_num.contains(QueryPlanFlags::START) {
self.min_value = args.get(i)?; if let Some(min_value) = args.get(i)? {
self.min_value = min_value;
} else {
any_null = true;
}
i += 1; i += 1;
} else { } else {
self.min_value = 0; self.min_value = 0;
} }
if idx_num.contains(QueryPlanFlags::STOP) { if idx_num.contains(QueryPlanFlags::STOP) {
self.max_value = args.get(i)?; if let Some(max_value) = args.get(i)? {
self.max_value = max_value;
} else {
any_null = true;
}
i += 1; i += 1;
} else { } else {
self.max_value = 0xffff_ffff; self.max_value = 0xffff_ffff;
} }
if idx_num.contains(QueryPlanFlags::STEP) { if idx_num.contains(QueryPlanFlags::STEP) {
self.step = args.get(i)?; if let Some(step) = args.get(i)? {
self.step = step;
} else {
any_null = true;
}
if self.step == 0 { if self.step == 0 {
self.step = 1; self.step = 1;
} else if self.step < 0 { } else if self.step < 0 {
@ -224,13 +236,11 @@ unsafe impl VTabCursor for SeriesTabCursor<'_> {
} else { } else {
self.step = 1; self.step = 1;
}; };
for arg in args.iter() { if any_null {
if arg.data_type() == Type::Null { // If any of the constraints have a NULL value, then
// If any of the constraints have a NULL value, then return no rows. // return no rows.
self.min_value = 1; self.min_value = 1;
self.max_value = 0; self.max_value = 0;
break;
}
} }
self.is_desc = idx_num.contains(QueryPlanFlags::DESC); self.is_desc = idx_num.contains(QueryPlanFlags::DESC);
if self.is_desc { if self.is_desc {