s4-drs: check if an optional feature is enabled

diff --git a/source4/dsdb/samdb/ldb_modules/util.c b/source4/dsdb/samdb/ldb_modules/util.c
index d3c58568acfbb4bd970e64cd5b8839ae72e0b411..8cf3335e15d7dc40f41a487a3c47004787928536 100644 (file)
--- a/source4/dsdb/samdb/ldb_modules/util.c
+++ b/source4/dsdb/samdb/ldb_modules/util.c
@@ -432,6 +432,65 @@ int dsdb_check_single_valued_link(const struct dsdb_attribute *attr,
        return LDB_SUCCESS;
 }
 
+int dsdb_check_optional_feature(struct ldb_module *module, struct ldb_dn *scope,
+                                       struct GUID op_feature_guid, bool *feature_enabled)
+{
+       TALLOC_CTX *tmp_ctx;
+       struct ldb_context *ldb = ldb_module_get_ctx(module);
+       struct ldb_result *res;
+       struct ldb_dn *search_dn;
+       struct GUID search_guid;
+       const char *attrs[] = {"msDS-EnabledFeature", NULL};
+       int ret;
+       int i;
+       struct ldb_message_element *el;
+
+       *feature_enabled = false;
+
+       tmp_ctx = talloc_new(ldb);
+
+       ret = ldb_search(ldb, tmp_ctx, &res,
+                                       scope, LDB_SCOPE_BASE, attrs,
+                                       NULL);
+       if (ret != LDB_SUCCESS) {
+               ldb_asprintf_errstring(ldb,
+                               "Could no find the scope object - dn: %s\n",
+                               ldb_dn_get_linearized(scope));
+               talloc_free(tmp_ctx);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+       if (res->msgs[0]->num_elements > 0) {
+
+               el = ldb_msg_find_element(res->msgs[0],"msDS-EnabledFeature");
+
+               attrs[0] = "msDS-OptionalFeatureGUID";
+
+               for (i=0; i<el->num_values; i++) {
+                       search_dn = ldb_dn_from_ldb_val(tmp_ctx, ldb, &el->values[i]);
+
+                       ret = ldb_search(ldb, tmp_ctx, &res,
+                                                       search_dn, LDB_SCOPE_BASE, attrs,
+                                                       NULL);
+                       if (ret != LDB_SUCCESS) {
+                               ldb_asprintf_errstring(ldb,
+                                               "Could no find object dn: %s\n",
+                                               ldb_dn_get_linearized(search_dn));
+                               talloc_free(tmp_ctx);
+                               return LDB_ERR_OPERATIONS_ERROR;
+                       }
+
+                       search_guid = samdb_result_guid(res->msgs[0], "msDS-OptionalFeatureGUID");
+
+                       if (GUID_compare(&search_guid, &op_feature_guid) == 0){
+                               *feature_enabled = true;
+                               break;
+                       }
+               }
+       }
+       talloc_free(tmp_ctx);
+       return LDB_SUCCESS;
+}
+
 
 /*
   find a 'reference' DN that points at another object