record/zonefile: malloc checks

2 files changed, 7 insertions, 1 deletions

diff --git a/src/record.c b/src/record.c
index e323718..e6e70dc 100644
--- a/src/record.c
+++ b/src/record.c
@@ -155,11 +155,13 @@ static ssize_t record_rdata_a(char* _str, void** _rdata) {
 
 	len = strlen(_str) + 1;
 	str = malloc(len);
+	if ( !str )
+		return -1;
 	strncpy(str, _str, len);
 
 	*_rdata = malloc(4);
 	if ( !*_rdata )
-		return -1;
+		goto err;
 
 	tok = strtok(str, ".");
 	if( !tok )
diff --git a/src/zonefile.c b/src/zonefile.c
index 1b0d036..1701ff8 100644
--- a/src/zonefile.c
+++ b/src/zonefile.c
@@ -52,6 +52,10 @@ int zonefile_parse_line(database_t *_database, char *_line) {
 		return -1;
 	}
 	qname = malloc( (unsigned)fqdn_len+2 );
+	if( !qname ) {
+		LOGPRINTF(_LOG_ERRNO, "malloc()");
+		return -1;
+	}
 	if ( fqdn_to_qname(parts[0], fqdn_len, qname, fqdn_len+2) < 0) {
 		LOGPRINTF(_LOG_ERROR, "Failed to convert to QNAME. This is a bug.");
 		return -1;