1a4de8b
From 9c1f74400c04267dea4f1bd7f62de8ba5e8d2b0e Mon Sep 17 00:00:00 2001
1a4de8b
From: Mark Andrews <marka@isc.org>
1a4de8b
Date: Thu, 28 Nov 2019 10:24:12 +1100
1a4de8b
Subject: [PATCH 4/4] rdataset_setownercase and rdataset_getownercase need to
1a4de8b
 obtain a node lock
1a4de8b
1a4de8b
(cherry picked from commit 637b2c4e517b466900a8c00b52f7a15727e12ae9)
1a4de8b
(cherry picked from commit 1c61f129c3b12071723a2154d33f74628bf80998)
1a4de8b
---
1a4de8b
 lib/dns/rbtdb.c | 35 ++++++++++++++++++++++++++++-------
1a4de8b
 1 file changed, 28 insertions(+), 7 deletions(-)
1a4de8b
1a4de8b
diff --git a/lib/dns/rbtdb.c b/lib/dns/rbtdb.c
1a4de8b
index d3bb8d7fe7..199ff08011 100644
1a4de8b
--- a/lib/dns/rbtdb.c
1a4de8b
+++ b/lib/dns/rbtdb.c
1a4de8b
@@ -10109,11 +10109,18 @@ setownercase(rdatasetheader_t *header, const dns_name_t *name) {
1a4de8b
 
1a4de8b
 static void
1a4de8b
 rdataset_setownercase(dns_rdataset_t *rdataset, const dns_name_t *name) {
1a4de8b
+	dns_rbtdb_t *rbtdb = rdataset->private1;
1a4de8b
+	dns_rbtnode_t *rbtnode = rdataset->private2;
1a4de8b
 	unsigned char *raw = rdataset->private3;        /* RDATASLAB */
1a4de8b
 	rdatasetheader_t *header;
1a4de8b
 
1a4de8b
 	header = (struct rdatasetheader *)(raw - sizeof(*header));
1a4de8b
+
1a4de8b
+	NODE_LOCK(&rbtdb->node_locks[rbtnode->locknum].lock,
1a4de8b
+		  isc_rwlocktype_write);
1a4de8b
 	setownercase(header, name);
1a4de8b
+	NODE_UNLOCK(&rbtdb->node_locks[rbtnode->locknum].lock,
1a4de8b
+		    isc_rwlocktype_write);
1a4de8b
 }
1a4de8b
 
1a4de8b
 static const unsigned char charmask[] = {
1a4de8b
@@ -10188,6 +10195,8 @@ static unsigned char maptolower[] = {
1a4de8b
 
1a4de8b
 static void
1a4de8b
 rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
+	dns_rbtdb_t *rbtdb = rdataset->private1;
1a4de8b
+	dns_rbtnode_t *rbtnode = rdataset->private2;
1a4de8b
 	const unsigned char *raw = rdataset->private3;        /* RDATASLAB */
1a4de8b
 	const rdatasetheader_t *header;
1a4de8b
 	unsigned int i, j;
1a4de8b
@@ -10196,8 +10205,12 @@ rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
 
1a4de8b
 	header = (const struct rdatasetheader *)(raw - sizeof(*header));
1a4de8b
 
1a4de8b
-	if (!CASESET(header))
1a4de8b
-		return;
1a4de8b
+	NODE_LOCK(&rbtdb->node_locks[rbtnode->locknum].lock,
1a4de8b
+		  isc_rwlocktype_read);
1a4de8b
+
1a4de8b
+	if (!CASESET(header)) {
1a4de8b
+		goto unlock;
1a4de8b
+	}
1a4de8b
 
1a4de8b
 #if 0
1a4de8b
 	/*
1a4de8b
@@ -10210,10 +10223,13 @@ rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
 		 */
1a4de8b
 		if (name->ndata[i] >= 0x61 && name->ndata[i] <= 0x7a &&
1a4de8b
 		    (header->upper[i/8] & (1 << (i%8))) != 0)
1a4de8b
+		{
1a4de8b
 			name->ndata[i] &= ~0x20; /* clear the lower case bit */
1a4de8b
-		else if (name->ndata[i] >= 0x41 && name->ndata[i] <= 0x5a &&
1a4de8b
-			 (header->upper[i/8] & (1 << (i%8))) == 0)
1a4de8b
+		} else if (name->ndata[i] >= 0x41 && name->ndata[i] <= 0x5a &&
1a4de8b
+			   (header->upper[i/8] & (1 << (i%8))) == 0)
1a4de8b
+		{
1a4de8b
 			name->ndata[i] |= 0x20; /* set the lower case bit */
1a4de8b
+		}
1a4de8b
 	}
1a4de8b
 #else
1a4de8b
 	if (ISC_LIKELY(CASEFULLYLOWER(header))) {
1a4de8b
@@ -10236,7 +10252,7 @@ rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
 			c = *bp;
1a4de8b
 			*bp++ = maptolower[c];
1a4de8b
 		}
1a4de8b
-		return;
1a4de8b
+		goto unlock;
1a4de8b
 	}
1a4de8b
 
1a4de8b
 	i = 0;
1a4de8b
@@ -10257,8 +10273,9 @@ rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
 		}
1a4de8b
 	}
1a4de8b
 
1a4de8b
-	if (ISC_UNLIKELY(i == name->length))
1a4de8b
-		return;
1a4de8b
+	if (ISC_UNLIKELY(i == name->length)) {
1a4de8b
+		goto unlock;
1a4de8b
+	}
1a4de8b
 
1a4de8b
 	bits = ~(header->upper[j]);
1a4de8b
 
1a4de8b
@@ -10272,6 +10289,10 @@ rdataset_getownercase(const dns_rdataset_t *rdataset, dns_name_t *name) {
1a4de8b
 		bits >>= 1;
1a4de8b
 	}
1a4de8b
 #endif
1a4de8b
+
1a4de8b
+ unlock:
1a4de8b
+	NODE_UNLOCK(&rbtdb->node_locks[rbtnode->locknum].lock,
1a4de8b
+		    isc_rwlocktype_read);
1a4de8b
 }
1a4de8b
 
1a4de8b
 /*%
1a4de8b
-- 
1a4de8b
2.21.0
1a4de8b