cartodb/node-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle NULL return from MallocCString

Browse Source
This commit is contained in:
Brian Carlson 2013-07-15 09:22:37 -05:00
parent 539d3bae54
commit 287d7e67a3
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/binding.cc
View File

@ -140,6 +140,11 @@ public:
Connection *self = ObjectWrap::Unwrap<Connection>(args.This()); Connection *self = ObjectWrap::Unwrap<Connection>(args.This());
char* inputStr = MallocCString(args[0]); char* inputStr = MallocCString(args[0]);
if(!inputStr) {
THROW("Unable to allocate memory for a string in EscapeIdentifier.")
}
char* escapedStr = self->EscapeIdentifier(inputStr); char* escapedStr = self->EscapeIdentifier(inputStr);
free(inputStr); free(inputStr);
@ -161,6 +166,11 @@ public:
Connection *self = ObjectWrap::Unwrap<Connection>(args.This()); Connection *self = ObjectWrap::Unwrap<Connection>(args.This());
char* inputStr = MallocCString(args[0]); char* inputStr = MallocCString(args[0]);
if(!inputStr) {
THROW("Unable to allocate memory for a string in EscapeIdentifier.")
}
char* escapedStr = self->EscapeLiteral(inputStr); char* escapedStr = self->EscapeLiteral(inputStr);
free(inputStr); free(inputStr);
@ -320,7 +330,7 @@ public:
Connection *self = ObjectWrap::Unwrap<Connection>(args.This()); Connection *self = ObjectWrap::Unwrap<Connection>(args.This());
//TODO handle errors in some way //TODO handle errors in some way
if (args.Length() < 1 && !Buffer::HasInstance(args[0])) { if (args.Length() < 1 && !Buffer::HasInstance(args[0])) {
THROW("SendCopyFromChunk requires 1 Buffer argument"); THROW("SendCopyFromChunk requires 1 Buffer argument");
} }
self->SendCopyFromChunk(args[0]->ToObject()); self->SendCopyFromChunk(args[0]->ToObject());
return Undefined(); return Undefined();