87 lines
2.7 KiB
Diff
87 lines
2.7 KiB
Diff
|
From 808d7ed1f82a24f7367006e43174c0c322e24590 Mon Sep 17 00:00:00 2001
|
||
|
|
From: SQLite Maintainers
|
||
|
|
Date: Thu, 15 Aug 2019 15:08:23 +0800
|
||
|
|
Subject: [PATCH] fix out of bounds read
|
||
|
|
|
||
|
|
Enhance the rtreenode() function of rtree (used for testing)
|
||
|
|
so that it uses the newer sqlite3_str object for better performance and
|
||
|
|
improved error reporting.
|
||
|
|
|
||
|
|
---
|
||
|
|
ext/rtree/rtree.c | 35 ++++++++++++++++-------------------
|
||
|
|
1 file changed, 16 insertions(+), 19 deletions(-)
|
||
|
|
|
||
|
|
diff --git a/ext/rtree/rtree.c b/ext/rtree/rtree.c
|
||
|
|
index 56bf836..53c7adf 100644
|
||
|
|
--- a/ext/rtree/rtree.c
|
||
|
|
+++ b/ext/rtree/rtree.c
|
||
|
|
@@ -3689,49 +3689,46 @@ rtreeInit_fail:
|
||
|
|
** <num-dimension>*2 coordinates.
|
||
|
|
*/
|
||
|
|
static void rtreenode(sqlite3_context *ctx, int nArg, sqlite3_value **apArg){
|
||
|
|
- char *zText = 0;
|
||
|
|
RtreeNode node;
|
||
|
|
Rtree tree;
|
||
|
|
int ii;
|
||
|
|
+ int nData;
|
||
|
|
+ int errCode;
|
||
|
|
+ sqlite3_str *pOut;
|
||
|
|
|
||
|
|
UNUSED_PARAMETER(nArg);
|
||
|
|
memset(&node, 0, sizeof(RtreeNode));
|
||
|
|
memset(&tree, 0, sizeof(Rtree));
|
||
|
|
tree.nDim = (u8)sqlite3_value_int(apArg[0]);
|
||
|
|
+ if( tree.nDim<1 || tree.nDim>5 ) return;
|
||
|
|
tree.nDim2 = tree.nDim*2;
|
||
|
|
tree.nBytesPerCell = 8 + 8 * tree.nDim;
|
||
|
|
node.zData = (u8 *)sqlite3_value_blob(apArg[1]);
|
||
|
|
+ nData = sqlite3_value_bytes(apArg[1]);
|
||
|
|
+ if( nData<4 ) return;
|
||
|
|
+ if( nData<NCELL(&node)*tree.nBytesPerCell ) return;
|
||
|
|
|
||
|
|
+ pOut = sqlite3_str_new(0);
|
||
|
|
for(ii=0; ii<NCELL(&node); ii++){
|
||
|
|
- char zCell[512];
|
||
|
|
- int nCell = 0;
|
||
|
|
RtreeCell cell;
|
||
|
|
int jj;
|
||
|
|
|
||
|
|
nodeGetCell(&tree, &node, ii, &cell);
|
||
|
|
- sqlite3_snprintf(512-nCell,&zCell[nCell],"%lld", cell.iRowid);
|
||
|
|
- nCell = (int)strlen(zCell);
|
||
|
|
+ if( ii>0 ) sqlite3_str_append(pOut, " ", 1);
|
||
|
|
+ sqlite3_str_appendf(pOut, "{%lld", cell.iRowid);
|
||
|
|
for(jj=0; jj<tree.nDim2; jj++){
|
||
|
|
#ifndef SQLITE_RTREE_INT_ONLY
|
||
|
|
- sqlite3_snprintf(512-nCell,&zCell[nCell], " %g",
|
||
|
|
- (double)cell.aCoord[jj].f);
|
||
|
|
+ sqlite3_str_appendf(pOut, " %g", (double)cell.aCoord[jj].f);
|
||
|
|
#else
|
||
|
|
- sqlite3_snprintf(512-nCell,&zCell[nCell], " %d",
|
||
|
|
- cell.aCoord[jj].i);
|
||
|
|
+ sqlite3_str_appendf(pOut, " %d", cell.aCoord[jj].i);
|
||
|
|
#endif
|
||
|
|
- nCell = (int)strlen(zCell);
|
||
|
|
- }
|
||
|
|
-
|
||
|
|
- if( zText ){
|
||
|
|
- char *zTextNew = sqlite3_mprintf("%s {%s}", zText, zCell);
|
||
|
|
- sqlite3_free(zText);
|
||
|
|
- zText = zTextNew;
|
||
|
|
- }else{
|
||
|
|
- zText = sqlite3_mprintf("{%s}", zCell);
|
||
|
|
}
|
||
|
|
+ sqlite3_str_append(pOut, "}", 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
- sqlite3_result_text(ctx, zText, -1, sqlite3_free);
|
||
|
|
+ errCode = sqlite3_str_errcode(pOut);
|
||
|
|
+ sqlite3_result_text(ctx, sqlite3_str_finish(pOut), -1, sqlite3_free);
|
||
|
|
+ sqlite3_result_error_code(ctx, errCode);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* This routine implements an SQL function that returns the "depth" parameter
|
||
|
|
--
|
||
|
|
1.8.3.1
|
||
|
|
|