mysql_adaptor: add function mysql_adaptor_get_user_groups_rec

References: DESK-3893, DESK-3984

Author: jengelh

Makefile.am

@@ -320,7 +320,7 @@ tzd_files += data/Greenwich.tzd data/Haiti.tzd data/Hawaiian.tzd data/India.tzd
 tzd_files += data/Saratov.tzd data/Singapore.tzd data/South_Africa.tzd data/South_Sudan.tzd data/Sri_Lanka.tzd data/Sudan.tzd data/Syria.tzd data/Taipei.tzd data/Tasmania.tzd data/Tocantins.tzd data/Tokyo.tzd data/Tomsk.tzd data/Tonga.tzd data/Transbaikal.tzd data/Turkey.tzd data/Turks_And_Caicos.tzd data/US_Eastern.tzd data/US_Mountain.tzd data/UTC+12.tzd data/UTC+13.tzd data/UTC-02.tzd data/UTC-08.tzd data/UTC-09.tzd data/UTC-11.tzd data/UTC.tzd data/Ulaanbaatar.tzd data/Venezuela.tzd data/Vladivostok.tzd data/Volgograd.tzd data/W__Australia.tzd data/W__Central_Africa.tzd data/W__Europe.tzd data/W__Mongolia.tzd data/West_Asia.tzd data/West_Bank.tzd data/West_Pacific.tzd data/Yakutsk.tzd data/Yukon.tzd
 tzd_files += data/windowsZones.xml
 header_files = include/gromox/ab_tree.hpp include/gromox/arcfour.hpp include/gromox/archive.hpp include/gromox/atomic.hpp include/gromox/authmgr.hpp include/gromox/bounce_gen.hpp include/gromox/clock.hpp include/gromox/common_types.hpp include/gromox/config_file.hpp include/gromox/contexts_pool.hpp include/gromox/cookie_parser.hpp include/gromox/cryptoutil.hpp include/gromox/database.h include/gromox/database_mysql.hpp include/gromox/dbop.h include/gromox/dcerpc.hpp include/gromox/defs.h include/gromox/double_list.hpp include/gromox/dsn.hpp include/gromox/eid_array.hpp include/gromox/element_data.hpp include/gromox/exmdb_client.hpp include/gromox/exmdb_common_util.hpp include/gromox/exmdb_ext.hpp include/gromox/exmdb_idef.hpp include/gromox/exmdb_provider_client.hpp include/gromox/exmdb_rpc.hpp include/gromox/exmdb_server.hpp include/gromox/ext_buffer.hpp
-header_files += include/gromox/fileio.h include/gromox/flusher_common.h include/gromox/freebusy.hpp include/gromox/gab.hpp include/gromox/generic_connection.hpp include/gromox/hook_common.h include/gromox/hpm_common.h include/gromox/http.hpp include/gromox/ical.hpp include/gromox/icase.hpp include/gromox/json.hpp include/gromox/list_file.hpp include/gromox/lzxpress.hpp include/gromox/mail.hpp include/gromox/mail_func.hpp include/gromox/mapi_types.hpp include/gromox/mapidefs.h include/gromox/mapierr.hpp include/gromox/mapitags.hpp include/gromox/midb.hpp include/gromox/midb_agent.hpp include/gromox/mime.hpp include/gromox/mjson.hpp include/gromox/mysql_adaptor.hpp include/gromox/ndr.hpp include/gromox/ntlmssp.hpp include/gromox/oxcmail.hpp include/gromox/oxoabkt.hpp
+header_files += include/gromox/fileio.h include/gromox/flat_set.hpp include/gromox/flusher_common.h include/gromox/freebusy.hpp include/gromox/gab.hpp include/gromox/generic_connection.hpp include/gromox/hook_common.h include/gromox/hpm_common.h include/gromox/http.hpp include/gromox/ical.hpp include/gromox/icase.hpp include/gromox/json.hpp include/gromox/list_file.hpp include/gromox/lzxpress.hpp include/gromox/mail.hpp include/gromox/mail_func.hpp include/gromox/mapi_types.hpp include/gromox/mapidefs.h include/gromox/mapierr.hpp include/gromox/mapitags.hpp include/gromox/midb.hpp include/gromox/midb_agent.hpp include/gromox/mime.hpp include/gromox/mjson.hpp include/gromox/mysql_adaptor.hpp include/gromox/ndr.hpp include/gromox/ntlmssp.hpp include/gromox/oxcmail.hpp include/gromox/oxoabkt.hpp
 header_files += include/gromox/paths.h include/gromox/pcl.hpp include/gromox/plugin.hpp include/gromox/proc_common.h include/gromox/process.hpp include/gromox/proptag_array.hpp include/gromox/propval.hpp include/gromox/range_set.hpp include/gromox/resource_pool.hpp include/gromox/restriction.hpp include/gromox/rop_util.hpp include/gromox/rpc_types.hpp include/gromox/rule_actions.hpp include/gromox/safeint.hpp include/gromox/simple_tree.hpp include/gromox/sortorder_set.hpp include/gromox/stream.hpp include/gromox/svc_common.h include/gromox/svc_loader.hpp include/gromox/textmaps.hpp include/gromox/threads_pool.hpp include/gromox/tie.hpp include/gromox/tnef.hpp include/gromox/usercvt.hpp include/gromox/util.hpp include/gromox/vcard.hpp include/gromox/xarray2.hpp include/gromox/zcore_client.hpp include/gromox/zcore_rpc.hpp include/gromox/zcore_types.hpp include/gromox/zz_ndr_stack.hpp
 if ENABLE_PRIVATE_HEADERS
 pkginclude_HEADERS = ${header_files}

exch/mysql_adaptor/sql2.cpp

@@ -31,6 +31,7 @@
 #include <gromox/dbop.h>
 #include <gromox/defs.h>
 #include <gromox/fileio.h>
+#include <gromox/flat_set.hpp>
 #include <gromox/mapidefs.h>
 #include <gromox/mysql_adaptor.hpp>
 #include <gromox/svc_common.h>
@@ -904,6 +905,42 @@ bool mysql_plugin::check_mlist_include(const char *mlist_name,
 	return false;
 }
 
+errno_t mysql_plugin::get_user_groups_rec(const char *username, std::vector<std::string> &groups) try
+{
+	gromox::maybe_flat_set<uint32_t> seen_ml;
+	auto conn = g_sqlconn_pool.get_wait();
+	if (!conn)
+		return EIO;
+	bool round = 0;
+	size_t rescan_begin = 0, rescan_end = 0;
+	for (size_t i = rescan_begin; i < rescan_end || round == 0; ++i) {
+		auto entity = round != 0 ? groups[i].c_str() : username;
+		auto qstr = "SELECT DISTINCT m.listname FROM associations AS a "
+			    "INNER JOIN mlists AS m ON a.list_id=m.id "
+			    "WHERE a.username='" + conn->quote(entity) + "'";
+		if (!conn->query(qstr))
+			return EIO;
+		auto result = conn->store_result();
+		DB_ROW row;
+		++rescan_begin;
+		round = 1;
+		while ((row = result.fetch_row()) != nullptr) {
+			if (row[1] == nullptr)
+				continue;
+			auto list_id = strtoul(row[0], nullptr, 0);
+			if (seen_ml.contains(list_id))
+				continue;
+			seen_ml.emplace(list_id);
+			groups.emplace_back(row[1]);
+			++rescan_end;
+		}
+	}
+	return 0;
+} catch (const std::bad_alloc &e) {
+	mlog(LV_ERR, "E-1731: ENOMEM");
+	return ENOMEM;
+}
+
 errno_t mysql_adaptor_meta(const char *u, unsigned int p, sql_meta_result &r)
 {
 	return le_mysql_plugin->meta(u, p, r);
@@ -1055,3 +1092,8 @@ errno_t mysql_adaptor_mda_domain_list(sql_domain_set &v)
 {
 	return le_mysql_plugin->mda_domain_list(v);
 }
+
+errno_t mysql_adaptor_get_user_groups_rec(const char *user, std::vector<std::string> &groups)
+{
+	return le_mysql_plugin->get_user_groups_rec(user, groups);
+}

exch/mysql_adaptor/sql2.hpp

@@ -80,6 +80,7 @@ struct mysql_plugin final {
 	int mbop_userlist(std::vector<sql_user> &);
 	gromox::errno_t mda_alias_list(gromox::sql_alias_map &, size_t &);
 	gromox::errno_t mda_domain_list(gromox::sql_domain_set &);
+	gromox::errno_t get_user_groups_rec(const char *, std::vector<std::string> &);
 
 	protected:
 	bool mlist_domain_contains(sqlconn *, const char *mlist, const char *account);

include/gromox/flat_set.hpp

@@ -0,0 +1,12 @@
+#pragma once
+#if __cplusplus >= 202300L
+#include <flat_set>
+namespace gromox {
+template<typename T> using maybe_flat_set = std::flat_set<T>;
+}
+#else
+#	include <set>
+namespace gromox {
+template<typename T> using maybe_flat_set = std::set<T>;
+}
+#endif

include/gromox/mysql_adaptor.hpp

@@ -147,6 +147,7 @@ extern GX_EXPORT int mysql_adaptor_domain_list_query(const char *dom);
 extern GX_EXPORT int mysql_adaptor_mbop_userlist(std::vector<sql_user> &);
 extern GX_EXPORT gromox::errno_t mysql_adaptor_mda_alias_list(gromox::sql_alias_map &, size_t &);
 extern GX_EXPORT gromox::errno_t mysql_adaptor_mda_domain_list(gromox::sql_domain_set &);
+extern GX_EXPORT gromox::errno_t mysql_adaptor_get_user_groups_rec(const char *, std::vector<std::string> &);
 
 /**
  * Determines whether an arbitrary actor can generally open/read the primary