| /*****************************************************************************\ | 
 |  *  cluster_report_functions.c - Interface to functions dealing with cluster | 
 |  *                               reports. | 
 |  ****************************************************************************** | 
 |  *  Copyright (C) 2010 Lawrence Livermore National Security. | 
 |  *  Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER). | 
 |  *  Written by Danny Auble da@llnl.gov, et. al. | 
 |  *  CODE-OCEC-09-009. All rights reserved. | 
 |  * | 
 |  *  This file is part of Slurm, a resource management program. | 
 |  *  For details, see <https://slurm.schedmd.com/>. | 
 |  *  Please also read the included file: DISCLAIMER. | 
 |  * | 
 |  *  Slurm is free software; you can redistribute it and/or modify it under | 
 |  *  the terms of the GNU General Public License as published by the Free | 
 |  *  Software Foundation; either version 2 of the License, or (at your option) | 
 |  *  any later version. | 
 |  * | 
 |  *  In addition, as a special exception, the copyright holders give permission | 
 |  *  to link the code of portions of this program with the OpenSSL library under | 
 |  *  certain conditions as described in each individual source file, and | 
 |  *  distribute linked combinations including the two. You must obey the GNU | 
 |  *  General Public License in all respects for all of the code used other than | 
 |  *  OpenSSL. If you modify file(s) with this exception, you may extend this | 
 |  *  exception to your version of the file(s), but you are not obligated to do | 
 |  *  so. If you do not wish to do so, delete this exception statement from your | 
 |  *  version.  If you delete this exception statement from all source files in | 
 |  *  the program, then also delete it here. | 
 |  * | 
 |  *  Slurm is distributed in the hope that it will be useful, but WITHOUT ANY | 
 |  *  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS | 
 |  *  FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more | 
 |  *  details. | 
 |  * | 
 |  *  You should have received a copy of the GNU General Public License along | 
 |  *  with Slurm; if not, write to the Free Software Foundation, Inc., | 
 |  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA. | 
 | \*****************************************************************************/ | 
 |  | 
 | #include "slurm/slurm.h" | 
 | #include "slurm/slurm_errno.h" | 
 |  | 
 | #include "src/common/slurmdb_defs.h" | 
 | #include "src/interfaces/accounting_storage.h" | 
 | #include "src/common/xstring.h" | 
 |  | 
 | typedef enum { | 
 | 	CLUSTER_REPORT_UA, | 
 | 	CLUSTER_REPORT_AU, | 
 | 	CLUSTER_REPORT_UW, | 
 | 	CLUSTER_REPORT_WU | 
 | } cluster_report_t; | 
 |  | 
 | static void _process_ua(list_t *user_list, slurmdb_assoc_rec_t *assoc) | 
 | { | 
 | 	list_itr_t *itr = NULL; | 
 | 	slurmdb_report_user_rec_t *slurmdb_report_user = NULL; | 
 |  | 
 | 	/* make sure we add all associations to this | 
 | 	   user rec because we could have some in | 
 | 	   partitions which would create another | 
 | 	   record otherwise | 
 | 	*/ | 
 | 	itr = list_iterator_create(user_list); | 
 | 	while((slurmdb_report_user = list_next(itr))) { | 
 | 		if (!xstrcmp(slurmdb_report_user->name, assoc->user) | 
 | 		   && !xstrcmp(slurmdb_report_user->acct, assoc->acct)) | 
 | 			break; | 
 | 	} | 
 | 	list_iterator_destroy(itr); | 
 |  | 
 | 	if (!slurmdb_report_user) { | 
 | 		struct passwd *passwd_ptr = NULL; | 
 | 		uid_t uid = NO_VAL; | 
 | 		passwd_ptr = getpwnam(assoc->user); | 
 | 		if (passwd_ptr) | 
 | 			uid = passwd_ptr->pw_uid; | 
 | 		/* In this report we are using the slurmdb_report user | 
 | 		   structure to store the information we want | 
 | 		   since it is already available and will do | 
 | 		   pretty much what we want. | 
 | 		*/ | 
 | 		slurmdb_report_user = | 
 | 			xmalloc(sizeof(slurmdb_report_user_rec_t)); | 
 | 		slurmdb_report_user->name = xstrdup(assoc->user); | 
 | 		slurmdb_report_user->uid = uid; | 
 | 		slurmdb_report_user->acct = xstrdup(assoc->acct); | 
 |  | 
 | 		list_append(user_list, slurmdb_report_user); | 
 | 	} | 
 |  | 
 | 	/* get the amount of time this assoc used | 
 | 	   during the time we are looking at */ | 
 | 	slurmdb_transfer_acct_list_2_tres(assoc->accounting_list, | 
 | 					  &slurmdb_report_user->tres_list); | 
 | } | 
 |  | 
 | static int _find_assoc_in_report(void *x, void *key) | 
 | { | 
 | 	slurmdb_report_assoc_rec_t *slurmdb_report_assoc = x; | 
 | 	slurmdb_accounting_rec_t *accting = key; | 
 |  | 
 | 	if ((slurmdb_report_assoc->id == accting->id) && | 
 | 	    (slurmdb_report_assoc->id_alt == accting->id_alt)) | 
 | 		return 1; | 
 | 	return 0; | 
 | } | 
 |  | 
 | static void _process_au(list_t *assoc_list, slurmdb_assoc_rec_t *assoc) | 
 | { | 
 | 	list_itr_t *itr; | 
 | 	slurmdb_accounting_rec_t *accting = NULL; | 
 | 	slurmdb_report_assoc_rec_t *slurmdb_report_assoc = NULL; | 
 |  | 
 | 	itr = list_iterator_create(assoc->accounting_list); | 
 | 	while ((accting = list_next(itr))) { | 
 | 		if (slurmdb_report_assoc && | 
 | 		    _find_assoc_in_report(slurmdb_report_assoc, accting)) { | 
 | 			/* Same report as before, no need to look it up again */ | 
 | 		} else if (!(slurmdb_report_assoc = list_find_first( | 
 | 				     assoc_list, | 
 | 				     _find_assoc_in_report, | 
 | 				     accting))) { | 
 | 			slurmdb_report_assoc = | 
 | 				xmalloc(sizeof(*slurmdb_report_assoc)); | 
 |  | 
 | 			list_append(assoc_list, slurmdb_report_assoc); | 
 |  | 
 | 			slurmdb_report_assoc->acct = xstrdup(assoc->acct); | 
 | 			slurmdb_report_assoc->cluster = | 
 | 				xstrdup(assoc->cluster); | 
 | 			slurmdb_report_assoc->parent_acct = | 
 | 				xstrdup(assoc->parent_acct); | 
 | 			slurmdb_report_assoc->user = xstrdup(assoc->user); | 
 | 			slurmdb_report_assoc->id = accting->id; | 
 | 			slurmdb_report_assoc->id_alt = accting->id_alt; | 
 | 		} | 
 |  | 
 | 		slurmdb_add_accounting_to_tres_list( | 
 | 			accting, &slurmdb_report_assoc->tres_list); | 
 | 	} | 
 | 	list_iterator_destroy(itr); | 
 | } | 
 |  | 
 | static void _process_uw(list_t *user_list, slurmdb_wckey_rec_t *wckey) | 
 | { | 
 | 	slurmdb_report_user_rec_t *slurmdb_report_user = NULL; | 
 | 	struct passwd *passwd_ptr = NULL; | 
 | 	uid_t uid = NO_VAL; | 
 |  | 
 | 	passwd_ptr = getpwnam(wckey->user); | 
 | 	if (passwd_ptr) | 
 | 		uid = passwd_ptr->pw_uid; | 
 | 	/* In this report we are using the slurmdb_report user | 
 | 	   structure to store the information we want | 
 | 	   since it is already available and will do | 
 | 	   pretty much what we want. | 
 | 	*/ | 
 | 	slurmdb_report_user = | 
 | 		xmalloc(sizeof(slurmdb_report_user_rec_t)); | 
 | 	slurmdb_report_user->name = xstrdup(wckey->user); | 
 | 	slurmdb_report_user->uid = uid; | 
 | 	slurmdb_report_user->acct = xstrdup(wckey->name); | 
 |  | 
 | 	list_append(user_list, slurmdb_report_user); | 
 |  | 
 | 	/* get the amount of time this wckey used | 
 | 	   during the time we are looking at */ | 
 | 	slurmdb_transfer_acct_list_2_tres(wckey->accounting_list, | 
 | 					  &slurmdb_report_user->tres_list); | 
 | } | 
 |  | 
 | static void _process_wu(list_t *assoc_list, slurmdb_wckey_rec_t *wckey) | 
 | { | 
 | 	slurmdb_report_assoc_rec_t *slurmdb_report_assoc = NULL, | 
 | 		*parent_assoc = NULL; | 
 | 	list_itr_t *itr = NULL; | 
 |  | 
 | 	/* find the parent */ | 
 | 	itr = list_iterator_create(assoc_list); | 
 | 	while((parent_assoc = list_next(itr))) { | 
 | 		if (!parent_assoc->user | 
 | 		   && !xstrcmp(parent_assoc->acct, wckey->name)) | 
 | 			break; | 
 | 	} | 
 | 	list_iterator_destroy(itr); | 
 | 	if (!parent_assoc) { | 
 | 		parent_assoc = xmalloc(sizeof(slurmdb_report_assoc_rec_t)); | 
 |  | 
 | 		list_append(assoc_list, | 
 | 			    parent_assoc); | 
 | 		parent_assoc->acct = xstrdup(wckey->name); | 
 | 	} | 
 |  | 
 | 	/* now add one for the user */ | 
 | 	slurmdb_report_assoc = xmalloc(sizeof(slurmdb_report_assoc_rec_t)); | 
 | 	list_append(assoc_list, slurmdb_report_assoc); | 
 |  | 
 | 	slurmdb_report_assoc->acct = xstrdup(wckey->name); | 
 | 	slurmdb_report_assoc->user = xstrdup(wckey->user); | 
 |  | 
 | 	/* get the amount of time this wckey used | 
 | 	   during the time we are looking at */ | 
 | 	slurmdb_transfer_acct_list_2_tres(wckey->accounting_list, | 
 | 					  &slurmdb_report_assoc->tres_list); | 
 | 	slurmdb_transfer_acct_list_2_tres(wckey->accounting_list, | 
 | 					  &parent_assoc->tres_list); | 
 | } | 
 |  | 
 | static void _process_assoc_type( | 
 | 	list_itr_t *itr, | 
 | 	slurmdb_report_cluster_rec_t *slurmdb_report_cluster, | 
 | 	char *cluster_name, | 
 | 	cluster_report_t type) | 
 | { | 
 | 	slurmdb_assoc_rec_t *assoc = NULL; | 
 |  | 
 | 	/* now add the associations of interest here by user */ | 
 | 	while((assoc = list_next(itr))) { | 
 | 		if (!assoc->accounting_list | 
 | 		   || !list_count(assoc->accounting_list) | 
 | 		   || ((type == CLUSTER_REPORT_UA) && !assoc->user)) { | 
 | 			list_delete_item(itr); | 
 | 			continue; | 
 | 		} | 
 |  | 
 | 		if (xstrcmp(cluster_name, assoc->cluster)) | 
 | 			continue; | 
 |  | 
 | 		if (type == CLUSTER_REPORT_UA) | 
 | 			_process_ua(slurmdb_report_cluster->user_list, | 
 | 				    assoc); | 
 | 		else if (type == CLUSTER_REPORT_AU) | 
 | 			_process_au(slurmdb_report_cluster->assoc_list, | 
 | 				    assoc); | 
 |  | 
 | 		list_delete_item(itr); | 
 | 	} | 
 | } | 
 |  | 
 | static void _process_wckey_type( | 
 | 	list_itr_t *itr, | 
 | 	slurmdb_report_cluster_rec_t *slurmdb_report_cluster, | 
 | 	char *cluster_name, | 
 | 	cluster_report_t type) | 
 | { | 
 | 	slurmdb_wckey_rec_t *wckey = NULL; | 
 |  | 
 | 	/* now add the wckeyiations of interest here by user */ | 
 | 	while((wckey = list_next(itr))) { | 
 | 		if (!wckey->accounting_list | 
 | 		   || !list_count(wckey->accounting_list) | 
 | 		   || ((type == CLUSTER_REPORT_UW) && !wckey->user)) { | 
 | 			list_delete_item(itr); | 
 | 			continue; | 
 | 		} | 
 |  | 
 | 		if (xstrcmp(cluster_name, wckey->cluster)) | 
 | 			continue; | 
 |  | 
 | 		if (type == CLUSTER_REPORT_UW) | 
 | 			_process_uw(slurmdb_report_cluster->user_list, | 
 | 				    wckey); | 
 | 		else if (type == CLUSTER_REPORT_WU) | 
 | 			_process_wu(slurmdb_report_cluster->assoc_list, | 
 | 				    wckey); | 
 |  | 
 | 		list_delete_item(itr); | 
 | 	} | 
 | } | 
 |  | 
 | static list_t *_process_util_by_report(void *db_conn, char *calling_name, | 
 | 				       void *cond, cluster_report_t type) | 
 | { | 
 | 	list_itr_t *itr = NULL; | 
 | 	list_itr_t *type_itr = NULL; | 
 | 	slurmdb_cluster_cond_t cluster_cond; | 
 | 	list_t *type_list = NULL; | 
 | 	list_t *cluster_list = NULL; | 
 | 	list_t *first_list = NULL; | 
 | 	slurmdb_cluster_rec_t *cluster = NULL; | 
 | 	slurmdb_report_cluster_rec_t *slurmdb_report_cluster = NULL; | 
 | 	time_t start_time, end_time; | 
 |  | 
 | 	int exit_code = 0; | 
 |  | 
 | 	uid_t my_uid = getuid(); | 
 | 	list_t *ret_list = list_create(slurmdb_destroy_report_cluster_rec); | 
 |  | 
 | 	slurmdb_init_cluster_cond(&cluster_cond, 0); | 
 |  | 
 | 	cluster_cond.with_deleted = 1; | 
 | 	cluster_cond.with_usage = 1; | 
 | 	if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) { | 
 | 		start_time = ((slurmdb_assoc_cond_t *)cond)->usage_start; | 
 | 		end_time = ((slurmdb_assoc_cond_t *)cond)->usage_end; | 
 |  | 
 | 		cluster_cond.cluster_list = | 
 | 			((slurmdb_assoc_cond_t *)cond)->cluster_list; | 
 | 	} else if ((type == CLUSTER_REPORT_UW) || (type == CLUSTER_REPORT_WU)) { | 
 | 		start_time = ((slurmdb_wckey_cond_t *)cond)->usage_start; | 
 | 		end_time = ((slurmdb_wckey_cond_t *)cond)->usage_end; | 
 |  | 
 | 		cluster_cond.cluster_list = | 
 | 			((slurmdb_wckey_cond_t *)cond)->cluster_list; | 
 | 	} else { | 
 | 		error("unknown report type %d", type); | 
 | 		return NULL; | 
 | 	} | 
 |  | 
 | 	/* This needs to be done on some systems to make sure | 
 | 	   cluster_cond isn't messed.  This has happened on some 64 | 
 | 	   bit machines and this is here to be on the safe side. | 
 | 	*/ | 
 | 	slurmdb_report_set_start_end_time(&start_time, &end_time); | 
 | 	cluster_cond.usage_end = end_time; | 
 | 	cluster_cond.usage_start = start_time; | 
 |  | 
 |  | 
 | 	cluster_list = acct_storage_g_get_clusters( | 
 | 		db_conn, my_uid, &cluster_cond); | 
 |  | 
 | 	if (!cluster_list) { | 
 | 		exit_code=1; | 
 | 		fprintf(stderr, "%s: Problem with cluster query.\n", | 
 | 			calling_name); | 
 | 		goto end_it; | 
 | 	} | 
 |  | 
 | 	if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) { | 
 | 		((slurmdb_assoc_cond_t *)cond)->usage_start = start_time; | 
 | 		((slurmdb_assoc_cond_t *)cond)->usage_end = end_time; | 
 | 		type_list = acct_storage_g_get_assocs( | 
 | 			db_conn, my_uid, cond); | 
 | 	} else if ((type == CLUSTER_REPORT_UW) || (type == CLUSTER_REPORT_WU)) { | 
 | 		((slurmdb_wckey_cond_t *)cond)->usage_start = start_time; | 
 | 		((slurmdb_wckey_cond_t *)cond)->usage_end = end_time; | 
 | 		type_list = acct_storage_g_get_wckeys( | 
 | 			db_conn, my_uid, cond); | 
 | 	} | 
 |  | 
 | 	if (!type_list) { | 
 | 		exit_code=1; | 
 | 		fprintf(stderr, "%s: Problem with get query.\n", calling_name); | 
 | 		goto end_it; | 
 | 	} | 
 |  | 
 | 	if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) { | 
 | 		first_list = type_list; | 
 | 		type_list = slurmdb_get_hierarchical_sorted_assoc_list( | 
 | 			first_list); | 
 | 	} | 
 |  | 
 | 	/* set up the structures for easy retrieval later */ | 
 | 	itr = list_iterator_create(cluster_list); | 
 | 	type_itr = list_iterator_create(type_list); | 
 | 	while((cluster = list_next(itr))) { | 
 | 		/* check to see if this cluster is around during the | 
 | 		   time we are looking at */ | 
 | 		if (!cluster->accounting_list | 
 | 		   || !list_count(cluster->accounting_list)) | 
 | 			continue; | 
 |  | 
 | 		slurmdb_report_cluster = slurmdb_cluster_rec_2_report(cluster); | 
 |  | 
 | 		list_append(ret_list, slurmdb_report_cluster); | 
 |  | 
 | 		if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_UW)) | 
 | 			slurmdb_report_cluster->user_list = | 
 | 				list_create(slurmdb_destroy_report_user_rec); | 
 | 		else if ((type == CLUSTER_REPORT_AU) | 
 | 			|| (type == CLUSTER_REPORT_WU)) | 
 | 			slurmdb_report_cluster->assoc_list = | 
 | 				list_create(slurmdb_destroy_report_assoc_rec); | 
 |  | 
 | 		if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) | 
 | 			_process_assoc_type(type_itr, slurmdb_report_cluster, | 
 | 					    cluster->name, type); | 
 | 		else if ((type == CLUSTER_REPORT_UW) | 
 | 			|| (type == CLUSTER_REPORT_WU)) | 
 | 			_process_wckey_type(type_itr, slurmdb_report_cluster, | 
 | 					    cluster->name, type); | 
 | 		list_iterator_reset(type_itr); | 
 | 	} | 
 | 	list_iterator_destroy(type_itr); | 
 | 	list_iterator_destroy(itr); | 
 |  | 
 | end_it: | 
 | 	FREE_NULL_LIST(type_list); | 
 | 	FREE_NULL_LIST(first_list); | 
 | 	FREE_NULL_LIST(cluster_list); | 
 | 	if (exit_code) | 
 | 		FREE_NULL_LIST(ret_list); | 
 | 	return ret_list; | 
 | } | 
 |  | 
 |  | 
 | extern list_t *slurmdb_report_cluster_account_by_user( | 
 | 	void *db_conn, slurmdb_assoc_cond_t *assoc_cond) | 
 | { | 
 | 	return _process_util_by_report(db_conn, | 
 | 				       "slurmdb_report_cluster_account_by_user", | 
 | 				       assoc_cond, CLUSTER_REPORT_AU); | 
 | } | 
 |  | 
 | extern list_t *slurmdb_report_cluster_user_by_account( | 
 | 	void *db_conn, slurmdb_assoc_cond_t *assoc_cond) | 
 | { | 
 | 	return _process_util_by_report(db_conn, | 
 | 				       "slurmdb_report_cluster_user_by_account", | 
 | 				       assoc_cond, CLUSTER_REPORT_UA); | 
 | } | 
 |  | 
 | extern list_t *slurmdb_report_cluster_wckey_by_user( | 
 | 	void *db_conn, slurmdb_wckey_cond_t *wckey_cond) | 
 | { | 
 | 	return _process_util_by_report(db_conn, | 
 | 				       "slurmdb_report_cluster_wckey_by_user", | 
 | 				       wckey_cond, CLUSTER_REPORT_WU); | 
 | } | 
 |  | 
 | extern list_t *slurmdb_report_cluster_user_by_wckey( | 
 | 	void *db_conn, slurmdb_wckey_cond_t *wckey_cond) | 
 | { | 
 | 	return _process_util_by_report(db_conn, | 
 | 				       "slurmdb_report_cluster_user_by_wckey", | 
 | 				       wckey_cond, CLUSTER_REPORT_UW); | 
 | } |