[tarantool-patches] [PATCH v2 2/7] sql: rework sql_src_list_enlarge to set diag
Kirill Shcherbatov
kshcherbatov at tarantool.org
Wed Feb 27 14:13:13 MSK 2019
Refactored sql_src_list_enlarge routine to use diag_set in case
of memory allocation error. This will ensure that the
sqlSrcListAppend function throws an error using diag in
subsequent patches.
Needed for #3931
---
src/box/sql/build.c | 109 ++++++++++++++++---------------------------
src/box/sql/select.c | 28 ++++++-----
src/box/sql/sqlInt.h | 27 ++++++++++-
3 files changed, 83 insertions(+), 81 deletions(-)
diff --git a/src/box/sql/build.c b/src/box/sql/build.c
index 3a65eb5a1..5337df450 100644
--- a/src/box/sql/build.c
+++ b/src/box/sql/build.c
@@ -2597,76 +2597,45 @@ sqlIdListIndex(IdList * pList, const char *zName)
return -1;
}
-/*
- * Expand the space allocazted for the given SrcList object by
- * creating nExtra new slots beginning at iStart. iStart is zero based.
- * New slots are zeroed.
- *
- * For example, suppose a SrcList initially contains two entries: A,B.
- * To append 3 new entries onto the end, do this:
- *
- * sqlSrcListEnlarge(db, pSrclist, 3, 2);
- *
- * After the call above it would contain: A, B, nil, nil, nil.
- * If the iStart argument had been 1 instead of 2, then the result
- * would have been: A, nil, nil, nil, B. To prepend the new slots,
- * the iStart value would be 0. The result then would
- * be: nil, nil, nil, A, B.
- *
- * If a memory allocation fails the SrcList is unchanged. The
- * db->mallocFailed flag will be set to true.
- */
-SrcList *
-sqlSrcListEnlarge(sql * db, /* Database connection to notify of OOM errors */
- SrcList * pSrc, /* The SrcList to be enlarged */
- int nExtra, /* Number of new slots to add to pSrc->a[] */
- int iStart /* Index in pSrc->a[] of first new slot */
- )
-{
- int i;
-
- /* Sanity checking on calling parameters */
- assert(iStart >= 0);
- assert(nExtra >= 1);
- assert(pSrc != 0);
- assert(iStart <= pSrc->nSrc);
-
- /* Allocate additional space if needed */
- if ((u32) pSrc->nSrc + nExtra > pSrc->nAlloc) {
- SrcList *pNew;
- int nAlloc = pSrc->nSrc * 2 + nExtra;
- int nGot;
- pNew = sqlDbRealloc(db, pSrc,
- sizeof(*pSrc) + (nAlloc -
- 1) *
- sizeof(pSrc->a[0]));
- if (pNew == 0) {
- assert(db->mallocFailed);
- return pSrc;
+struct SrcList *
+sql_src_list_enlarge(struct sql *db, struct SrcList *src_list, int new_slots,
+ int start_idx)
+{
+ assert(start_idx >= 0);
+ assert(new_slots >= 1);
+ assert(src_list != NULL);
+ assert(start_idx <= src_list->nSrc);
+
+ /* Allocate additional space if needed. */
+ if (src_list->nSrc + new_slots > (int)src_list->nAlloc) {
+ int to_alloc = src_list->nSrc * 2 + new_slots;
+ int size = sizeof(*src_list) +
+ (to_alloc - 1) * sizeof(src_list->a[0]);
+ src_list = sqlDbRealloc(db, src_list, size);
+ if (src_list == NULL) {
+ diag_set(OutOfMemory, size, "sqlDbRealloc", "src_list");
+ return NULL;
}
- pSrc = pNew;
- nGot =
- (sqlDbMallocSize(db, pNew) -
- sizeof(*pSrc)) / sizeof(pSrc->a[0]) + 1;
- pSrc->nAlloc = nGot;
+ src_list->nAlloc =
+ (sqlDbMallocSize(db, src_list) -
+ sizeof(*src_list)) / sizeof(src_list->a[0]) + 1;;
}
- /* Move existing slots that come after the newly inserted slots
- * out of the way
+ /*
+ * Move existing slots that come after the newly inserted
+ * slots out of the way.
*/
- for (i = pSrc->nSrc - 1; i >= iStart; i--) {
- pSrc->a[i + nExtra] = pSrc->a[i];
- }
- pSrc->nSrc += nExtra;
+ for (int i = src_list->nSrc - 1; i >= start_idx; i--)
+ src_list->a[i + new_slots] = src_list->a[i];
+ src_list->nSrc += new_slots;
- /* Zero the newly allocated slots */
- memset(&pSrc->a[iStart], 0, sizeof(pSrc->a[0]) * nExtra);
- for (i = iStart; i < iStart + nExtra; i++) {
- pSrc->a[i].iCursor = -1;
- }
+ /* Zero the newly allocated slots. */
+ memset(&src_list->a[start_idx], 0, sizeof(src_list->a[0]) * new_slots);
+ for (int i = start_idx; i < start_idx + new_slots; i++)
+ src_list->a[i].iCursor = -1;
- /* Return a pointer to the enlarged SrcList */
- return pSrc;
+ /* Return a pointer to the enlarged SrcList. */
+ return src_list;
}
struct SrcList *
@@ -2732,11 +2701,13 @@ sqlSrcListAppend(sql * db, /* Connection to notify of malloc failures */
if (pList == 0)
return 0;
} else {
- pList = sqlSrcListEnlarge(db, pList, 1, pList->nSrc);
- }
- if (db->mallocFailed) {
- sqlSrcListDelete(db, pList);
- return 0;
+ struct SrcList *new_list =
+ sql_src_list_enlarge(db, pList, 1, pList->nSrc);
+ if (new_list == NULL) {
+ sqlSrcListDelete(db, pList);
+ return NULL;
+ }
+ pList = new_list;
}
pItem = &pList->a[pList->nSrc - 1];
pItem->zName = sqlNameFromToken(db, pTable);
diff --git a/src/box/sql/select.c b/src/box/sql/select.c
index 6a465a616..0baf8fb57 100644
--- a/src/box/sql/select.c
+++ b/src/box/sql/select.c
@@ -279,18 +279,22 @@ src_list_append_unique(struct sql *db, struct SrcList *list,
if (name != NULL && strcmp(new_name, name) == 0)
return list;
}
- list = sqlSrcListEnlarge(db, list, 1, list->nSrc);
- if (db->mallocFailed) {
- sqlSrcListDelete(db, list);
- return NULL;
- }
+ struct SrcList *new_list =
+ sql_src_list_enlarge(db, list, 1, list->nSrc);
+ if (new_list == NULL)
+ goto error;
+ list = new_list;
struct SrcList_item *pItem = &list->a[list->nSrc - 1];
pItem->zName = sqlDbStrNDup(db, new_name, strlen(new_name));
if (pItem->zName == NULL) {
- sqlSrcListDelete(db, list);
- return NULL;
+ diag_set(OutOfMemory, strlen(new_name), "sqlDbStrNDup",
+ "pItem->zName");
+ goto error;
}
return list;
+error:
+ sqlSrcListDelete(db, list);
+ return NULL;
}
static int
@@ -4135,12 +4139,14 @@ flattenSubquery(Parse * pParse, /* Parsing context */
* for the two elements in the FROM clause of the subquery.
*/
if (nSubSrc > 1) {
- pParent->pSrc = pSrc =
- sqlSrcListEnlarge(db, pSrc, nSubSrc - 1,
- iFrom + 1);
- if (db->mallocFailed) {
+ struct SrcList *new_list =
+ sql_src_list_enlarge(db, pSrc, nSubSrc - 1,
+ iFrom + 1);
+ if (new_list == NULL) {
+ sql_parser_error(pParse);
break;
}
+ pParent->pSrc = pSrc = new_list;
}
/* Transfer the FROM clause terms from the subquery into the
diff --git a/src/box/sql/sqlInt.h b/src/box/sql/sqlInt.h
index ce29dc0b1..ea6476931 100644
--- a/src/box/sql/sqlInt.h
+++ b/src/box/sql/sqlInt.h
@@ -3426,7 +3426,32 @@ void sqlInsert(Parse *, SrcList *, Select *, IdList *,
void *sqlArrayAllocate(sql *, void *, int, int *, int *);
IdList *sqlIdListAppend(sql *, IdList *, Token *);
int sqlIdListIndex(IdList *, const char *);
-SrcList *sqlSrcListEnlarge(sql *, SrcList *, int, int);
+
+/**
+ * Expand the space allocated for the given SrcList object by
+ * creating new_slots new slots beginning at start_idx.
+ * The start_idx is zero based. New slots are zeroed.
+ *
+ * For example, suppose a SrcList initially contains two entries:
+ * A,B.
+ * To append 3 new entries onto the end, do this:
+ * sql_src_list_enlarge(db, pSrclist, 3, 2);
+ *
+ * After the call above it would contain: A, B, nil, nil, nil.
+ * If the iStart argument had been 1 instead of 2, then the result
+ * would have been: A, nil, nil, nil, B. To prepend the new slots,
+ * the iStart value would be 0. The result then would
+ * be: nil, nil, nil, A, B.
+ * @param db The database connection.
+ * @param src_list The SrcList to be enlarged.
+ * @param new_slots Number of new slots to add to src_list->a[].
+ * @param start_idx Index in pSrc->a[] of first new slot.
+ * @retval not NULL SrcList pointer on success.
+ * @retval NULL otherwise.
+ */
+struct SrcList *
+sql_src_list_enlarge(struct sql *db, struct SrcList *src_list, int new_slots,
+ int start_idx);
/**
* Allocate a new empty SrcList object.
--
2.20.1
More information about the Tarantool-patches
mailing list