Fixing memory leak if no more memory available for allocating columns

mariano · Jun 3, 2011 · 0afd864 · 0afd864
1 parent 7e13d0a
commit 0afd864
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/result.cc b/src/result.cc
@@ -95,6 +95,10 @@ node_db_mysql::Result::Result(MYSQL* connection) throw(node_db::Exception&)
 
             for (uint16_t i = 0; i < this->totalColumns; i++) {
                 this->columns[i] = new Column(fields[i]);
+                if (this->columns[i] == NULL) {
+                    delete [] this->columns;
+                    throw node_db::Exception("Could not allocate storage for column");
+                }
             }
         }