Blame SOURCES/autofs-5.0.9-amd-lookup-update-lookup-nisplus-to-handle-amd-keys.patch

6bbd11
autofs-5.0.9 - amd lookup update lookup nisplus to handle amd keys
6bbd11
6bbd11
From: Ian Kent <raven@themaw.net>
6bbd11
6bbd11
6bbd11
---
6bbd11
 modules/lookup_nisplus.c |  208 +++++++++++++++++++++++++++++++++++++++++++---
6bbd11
 1 file changed, 196 insertions(+), 12 deletions(-)
6bbd11
6bbd11
diff --git a/modules/lookup_nisplus.c b/modules/lookup_nisplus.c
6bbd11
index 42c9ccc..e9444c9 100644
6bbd11
--- a/modules/lookup_nisplus.c
6bbd11
+++ b/modules/lookup_nisplus.c
6bbd11
@@ -243,7 +243,19 @@ int lookup_read_map(struct autofs_point *ap, time_t age, void *context)
6bbd11
 		if (*key == '+')
6bbd11
 			continue;
6bbd11
 
6bbd11
-		s_key = sanitize_path(key, len, ap->type, ap->logopt);
6bbd11
+		if (!(source->flags & MAP_FLAG_FORMAT_AMD))
6bbd11
+			s_key = sanitize_path(key, len, ap->type, ap->logopt);
6bbd11
+		else {
6bbd11
+			if (!strcmp(key, "/defaults")) {
6bbd11
+				mapent = ENTRY_VAL(this, 1);
6bbd11
+				cache_writelock(mc);
6bbd11
+				cache_update(mc, source, key, mapent, age);
6bbd11
+				cache_unlock(mc);
6bbd11
+				continue;
6bbd11
+			}
6bbd11
+			/* Don't fail on "/" in key => type == 0 */
6bbd11
+			s_key = sanitize_path(key, len, 0, ap->logopt);
6bbd11
+		}
6bbd11
 		if (!s_key)
6bbd11
 			continue;
6bbd11
 
6bbd11
@@ -322,6 +334,66 @@ static int lookup_one(struct autofs_point *ap,
6bbd11
 	return ret;
6bbd11
 }
6bbd11
 
6bbd11
+static int match_key(struct autofs_point *ap,
6bbd11
+		     struct map_source *source,
6bbd11
+		     const char *key, int key_len,
6bbd11
+		     struct lookup_context *ctxt)
6bbd11
+{
6bbd11
+	char buf[MAX_ERR_BUF];
6bbd11
+	char *lkp_key;
6bbd11
+	char *prefix;
6bbd11
+	int ret;
6bbd11
+
6bbd11
+	ret = lookup_one(ap, source, key, key_len, ctxt);
6bbd11
+	if (ret < 0)
6bbd11
+		return ret;
6bbd11
+	if (ret == CHE_OK || ret == CHE_UPDATED)
6bbd11
+		return ret;
6bbd11
+
6bbd11
+	if (!(source->flags & MAP_FLAG_FORMAT_AMD))
6bbd11
+		return CHE_FAIL;
6bbd11
+
6bbd11
+	lkp_key = strdup(key);
6bbd11
+	if (!lkp_key) {
6bbd11
+		char *estr = strerror_r(errno, buf, MAX_ERR_BUF);
6bbd11
+		error(ap->logopt, MODPREFIX "strdup: %s", estr);
6bbd11
+		return CHE_FAIL;
6bbd11
+	}
6bbd11
+
6bbd11
+	ret = CHE_MISSING;
6bbd11
+
6bbd11
+	/*
6bbd11
+	 * Now strip successive directory components and try a
6bbd11
+	 * match against map entries ending with a wildcard and
6bbd11
+	 * finally try the wilcard entry itself.
6bbd11
+	 */
6bbd11
+	while ((prefix = strrchr(lkp_key, '/'))) {
6bbd11
+		char *match;
6bbd11
+		size_t len;
6bbd11
+		*prefix = '\0';
6bbd11
+		len = strlen(lkp_key) + 3;
6bbd11
+		match = malloc(len);
6bbd11
+		if (!match) {
6bbd11
+			char *estr = strerror_r(errno, buf, MAX_ERR_BUF);
6bbd11
+			error(ap->logopt, MODPREFIX "malloc: %s", estr);
6bbd11
+			ret = CHE_FAIL;
6bbd11
+			goto done;
6bbd11
+		}
6bbd11
+		len--;
6bbd11
+		strcpy(match, lkp_key);
6bbd11
+		strcat(match, "/*");
6bbd11
+		ret = lookup_one(ap, source, match, len, ctxt);
6bbd11
+		free(match);
6bbd11
+		if (ret < 0)
6bbd11
+			goto done;
6bbd11
+		if (ret == CHE_OK || ret == CHE_UPDATED)
6bbd11
+			goto done;
6bbd11
+	}
6bbd11
+done:
6bbd11
+	free(lkp_key);
6bbd11
+	return ret;
6bbd11
+}
6bbd11
+
6bbd11
 static int lookup_wild(struct autofs_point *ap,
6bbd11
 		       struct map_source *source, struct lookup_context *ctxt)
6bbd11
 {
6bbd11
@@ -374,6 +446,59 @@ static int lookup_wild(struct autofs_point *ap,
6bbd11
 	return ret;
6bbd11
 }
6bbd11
 
6bbd11
+static int lookup_amd_defaults(struct autofs_point *ap,
6bbd11
+			       struct map_source *source,
6bbd11
+			       struct lookup_context *ctxt)
6bbd11
+{
6bbd11
+	struct mapent_cache *mc = source->mc;
6bbd11
+	char *tablename;
6bbd11
+	nis_result *result;
6bbd11
+	nis_object *this;
6bbd11
+	char *mapent;
6bbd11
+	char buf[MAX_ERR_BUF];
6bbd11
+	int cur_state;
6bbd11
+	int ret;
6bbd11
+
6bbd11
+	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cur_state);
6bbd11
+	tablename = malloc(9 + strlen(ctxt->mapname) +
6bbd11
+			   strlen(ctxt->domainname) + 20);
6bbd11
+	if (!tablename) {
6bbd11
+		char *estr = strerror_r(errno, buf, MAX_ERR_BUF);
6bbd11
+		logerr(MODPREFIX "malloc: %s", estr);
6bbd11
+		pthread_setcancelstate(cur_state, NULL);
6bbd11
+		return CHE_FAIL;
6bbd11
+	}
6bbd11
+	sprintf(tablename, "[key=/defaults],%s.org_dir.%s",
6bbd11
+		ctxt->mapname, ctxt->domainname);
6bbd11
+
6bbd11
+	result = nis_list(tablename, FOLLOW_PATH | FOLLOW_LINKS, NULL, NULL);
6bbd11
+	if (result->status != NIS_SUCCESS && result->status != NIS_S_SUCCESS) {
6bbd11
+		nis_error rs = result->status;
6bbd11
+		nis_freeresult(result);
6bbd11
+		free(tablename);
6bbd11
+		pthread_setcancelstate(cur_state, NULL);
6bbd11
+		if (rs == NIS_NOTFOUND ||
6bbd11
+		    rs == NIS_S_NOTFOUND ||
6bbd11
+		    rs == NIS_PARTIAL)
6bbd11
+			return CHE_MISSING;
6bbd11
+
6bbd11
+		return -rs;
6bbd11
+	}
6bbd11
+
6bbd11
+	this = NIS_RES_OBJECT(result);
6bbd11
+	mapent = ENTRY_VAL(this, 1);
6bbd11
+
6bbd11
+	cache_writelock(mc);
6bbd11
+	ret = cache_update(mc, source, "/defaults", mapent, time(NULL));
6bbd11
+	cache_unlock(mc);
6bbd11
+
6bbd11
+	nis_freeresult(result);
6bbd11
+	free(tablename);
6bbd11
+	pthread_setcancelstate(cur_state, NULL);
6bbd11
+
6bbd11
+	return ret;
6bbd11
+}
6bbd11
+
6bbd11
 static int check_map_indirect(struct autofs_point *ap,
6bbd11
 			      struct map_source *source,
6bbd11
 			      char *key, int key_len,
6bbd11
@@ -387,8 +512,17 @@ static int check_map_indirect(struct autofs_point *ap,
6bbd11
 
6bbd11
 	mc = source->mc;
6bbd11
 
6bbd11
+	if (source->flags & MAP_FLAG_FORMAT_AMD) {
6bbd11
+		/* Check for a /defaults entry to update the map source */
6bbd11
+		if (lookup_amd_defaults(ap, source, ctxt) == CHE_FAIL) {
6bbd11
+			warn(ap->logopt, MODPREFIX
6bbd11
+			     "error getting /defaults from map %s",
6bbd11
+			     ctxt->mapname);
6bbd11
+		}
6bbd11
+	}
6bbd11
+
6bbd11
 	/* check map and if change is detected re-read map */
6bbd11
-	ret = lookup_one(ap, source, key, key_len, ctxt);
6bbd11
+	ret = match_key(ap, source, key, key_len, ctxt);
6bbd11
 	if (ret == CHE_FAIL)
6bbd11
 		return NSS_STATUS_NOTFOUND;
6bbd11
 
6bbd11
@@ -397,9 +531,16 @@ static int check_map_indirect(struct autofs_point *ap,
6bbd11
 		 * If the server is down and the entry exists in the cache
6bbd11
 		 * and belongs to this map return success and use the entry.
6bbd11
 		 */
6bbd11
-		exists = cache_lookup(mc, key);
6bbd11
-		if (exists && exists->source == source)
6bbd11
+		cache_readlock(mc);
6bbd11
+		if (source->flags & MAP_FLAG_FORMAT_AMD)
6bbd11
+			exists = match_cached_key(ap, MODPREFIX, source, key);
6bbd11
+		else
6bbd11
+			exists = cache_lookup(mc, key);
6bbd11
+		if (exists && exists->source == source) {
6bbd11
+			cache_unlock(mc);
6bbd11
 			return NSS_STATUS_SUCCESS;
6bbd11
+		}
6bbd11
+		cache_unlock(mc);
6bbd11
 
6bbd11
 		warn(ap->logopt,
6bbd11
 		     MODPREFIX "lookup for %s failed: %s",
6bbd11
@@ -418,6 +559,10 @@ static int check_map_indirect(struct autofs_point *ap,
6bbd11
 		}
6bbd11
 		me = cache_lookup_next(mc, me);
6bbd11
 	}
6bbd11
+	if (source->flags & MAP_FLAG_FORMAT_AMD)
6bbd11
+		exists = match_cached_key(ap, MODPREFIX, source, key);
6bbd11
+	else
6bbd11
+		exists = cache_lookup_distinct(mc, key);
6bbd11
 	exists = cache_lookup_distinct(mc, key);
6bbd11
 	/* Not found in the map but found in the cache */
6bbd11
 	if (exists && exists->source == source && ret & CHE_MISSING) {
6bbd11
@@ -474,9 +619,11 @@ int lookup_mount(struct autofs_point *ap, const char *name, int name_len, void *
6bbd11
 	struct mapent_cache *mc;
6bbd11
 	char key[KEY_MAX_LEN + 1];
6bbd11
 	int key_len;
6bbd11
+	char *lkp_key;
6bbd11
 	char *mapent = NULL;
6bbd11
 	int mapent_len;
6bbd11
 	struct mapent *me;
6bbd11
+	char buf[MAX_ERR_BUF];
6bbd11
 	int status;
6bbd11
 	int ret = 1;
6bbd11
 
6bbd11
@@ -488,9 +635,18 @@ int lookup_mount(struct autofs_point *ap, const char *name, int name_len, void *
6bbd11
 
6bbd11
 	debug(ap->logopt, MODPREFIX "looking up %s", name);
6bbd11
 
6bbd11
-	key_len = snprintf(key, KEY_MAX_LEN + 1, "%s", name);
6bbd11
-	if (key_len > KEY_MAX_LEN)
6bbd11
-		return NSS_STATUS_NOTFOUND;
6bbd11
+	if (!(source->flags & MAP_FLAG_FORMAT_AMD)) {
6bbd11
+		key_len = snprintf(key, KEY_MAX_LEN + 1, "%s", name);
6bbd11
+		if (key_len > KEY_MAX_LEN)
6bbd11
+			return NSS_STATUS_NOTFOUND;
6bbd11
+	} else {
6bbd11
+		key_len = expandamdent(name, NULL, NULL);
6bbd11
+		if (key_len > KEY_MAX_LEN)
6bbd11
+			return NSS_STATUS_NOTFOUND;
6bbd11
+		memset(key, 0, KEY_MAX_LEN + 1);
6bbd11
+		expandamdent(name, key, NULL);
6bbd11
+		debug(ap->logopt, MODPREFIX "expanded key: \"%s\"", key);
6bbd11
+	}
6bbd11
 
6bbd11
 	/* Check if we recorded a mount fail for this key anywhere */
6bbd11
 	me = lookup_source_mapent(ap, key, LKP_DISTINCT);
6bbd11
@@ -524,21 +680,30 @@ int lookup_mount(struct autofs_point *ap, const char *name, int name_len, void *
6bbd11
 	 * we never know about it.
6bbd11
 	 */
6bbd11
 	if (ap->type == LKP_INDIRECT && *key != '/') {
6bbd11
-		char *lkp_key;
6bbd11
-
6bbd11
 		cache_readlock(mc);
6bbd11
 		me = cache_lookup_distinct(mc, key);
6bbd11
 		if (me && me->multi)
6bbd11
 			lkp_key = strdup(me->multi->key);
6bbd11
-		else
6bbd11
+		else if (!ap->pref)
6bbd11
 			lkp_key = strdup(key);
6bbd11
+		else {
6bbd11
+			lkp_key = malloc(strlen(ap->pref) + strlen(key) + 1);
6bbd11
+			if (lkp_key) {
6bbd11
+				strcpy(lkp_key, ap->pref);
6bbd11
+				strcat(lkp_key, key);
6bbd11
+			}
6bbd11
+		}
6bbd11
 		cache_unlock(mc);
6bbd11
 
6bbd11
-		if (!lkp_key)
6bbd11
+		if (!lkp_key) {
6bbd11
+			char *estr = strerror_r(errno, buf, MAX_ERR_BUF);
6bbd11
+			error(ap->logopt, MODPREFIX "malloc: %s", estr);
6bbd11
 			return NSS_STATUS_UNKNOWN;
6bbd11
+		}
6bbd11
 
6bbd11
 		status = check_map_indirect(ap, source,
6bbd11
 					    lkp_key, strlen(lkp_key), ctxt);
6bbd11
+		free(lkp_key);
6bbd11
 		if (status)
6bbd11
 			return status;
6bbd11
 	}
6bbd11
@@ -555,7 +720,25 @@ int lookup_mount(struct autofs_point *ap, const char *name, int name_len, void *
6bbd11
 		cache_readlock(mc);
6bbd11
 	else
6bbd11
 		cache_writelock(mc);
6bbd11
-	me = cache_lookup(mc, key);
6bbd11
+
6bbd11
+	if (!ap->pref)
6bbd11
+		lkp_key = strdup(key);
6bbd11
+	else {
6bbd11
+		lkp_key = malloc(strlen(ap->pref) + strlen(key) + 1);
6bbd11
+		if (lkp_key) {
6bbd11
+			strcpy(lkp_key, ap->pref);
6bbd11
+			strcat(lkp_key, key);
6bbd11
+		}
6bbd11
+	}
6bbd11
+
6bbd11
+	if (!lkp_key) {
6bbd11
+		char *estr = strerror_r(errno, buf, MAX_ERR_BUF);
6bbd11
+		error(ap->logopt, MODPREFIX "malloc: %s", estr);
6bbd11
+		cache_unlock(mc);
6bbd11
+		return NSS_STATUS_UNKNOWN;
6bbd11
+	}
6bbd11
+
6bbd11
+	me = match_cached_key(ap, MODPREFIX, source, lkp_key);
6bbd11
 	/* Stale mapent => check for entry in alternate source or wildcard */
6bbd11
 	if (me && !me->mapent) {
6bbd11
 		while ((me = cache_lookup_key_next(me)))
6bbd11
@@ -583,6 +766,7 @@ int lookup_mount(struct autofs_point *ap, const char *name, int name_len, void *
6bbd11
 		}
6bbd11
 	}
6bbd11
 	cache_unlock(mc);
6bbd11
+	free(lkp_key);
6bbd11
 
6bbd11
 	if (!mapent)
6bbd11
 		return NSS_STATUS_TRYAGAIN;