agorise/c-protobuf
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Better allocation of memory for strings with zero bytes

Browse source
This commit is contained in:
jmjatlanta 2016-12-23 20:10:52 -05:00
parent 8d966178f4
commit 3b49ac2c5d
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
protobuf.c
View file

@ -116,8 +116,10 @@ int protobuf_decode_string(const unsigned char* buffer, size_t buffer_length, ch
// grab the field size // grab the field size
int length = varint_decode(&buffer[pos], buffer_length - pos, bytes_read); int length = varint_decode(&buffer[pos], buffer_length - pos, bytes_read);
pos += *bytes_read; pos += *bytes_read;
*bytes_read += length;
// allocate memory // allocate memory (if neccesary)
if (length > 0) {
*results = malloc(sizeof(char) * length + 1); *results = malloc(sizeof(char) * length + 1);
if ((*results) == NULL) if ((*results) == NULL)
return 0; return 0;
@ -128,9 +130,7 @@ int protobuf_decode_string(const unsigned char* buffer, size_t buffer_length, ch
memcpy((*results), &buffer[pos], length); memcpy((*results), &buffer[pos], length);
// don't forget the null // don't forget the null
(*results)[length] = 0; (*results)[length] = 0;
pos += length; }
*bytes_read = pos;
return 1; return 1;
} }