/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

/*
 * Service routines
 */

#include "idmapd.h"
#include "idmap_priv.h"
#include <signal.h>
#include <thread.h>
#include <string.h>
#include <strings.h>
#include <errno.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <ucred.h>
#include <pwd.h>
#include <auth_attr.h>
#include <secdb.h>

#define	_VALIDATE_LIST_CB_DATA(col, val, siz)\
	retcode = validate_list_cb_data(cb_data, argc, argv, col,\
			(uchar_t **)val, siz);\
	if (retcode == IDMAP_NEXT) {\
		result->retcode = IDMAP_NEXT;\
		return (0);\
	} else if (retcode < 0) {\
		result->retcode = retcode;\
		return (1);\
	}

#define	PROCESS_LIST_SVC_SQL(rcode, db, sql, limit, cb, res, len)\
	rcode = process_list_svc_sql(db, sql, limit, cb, res);\
	if (rcode == IDMAP_ERR_BUSY)\
		res->retcode = IDMAP_ERR_BUSY;\
	else if (rcode == IDMAP_SUCCESS && len == 0)\
		res->retcode = IDMAP_ERR_NOTFOUND;


/* ARGSUSED */
bool_t
idmap_null_1_svc(void *result, struct svc_req *rqstp) {
	return (TRUE);
}

#define	IS_BATCH_SID(batch, i)\
	batch.idmap_mapping_batch_val[i].id1.idtype == IDMAP_SID

#define	IS_BATCH_UID(batch, i)\
	batch.idmap_mapping_batch_val[i].id1.idtype == IDMAP_UID

#define	IS_BATCH_GID(batch, i)\
	batch.idmap_mapping_batch_val[i].id1.idtype == IDMAP_GID

#define	IS_REQUEST_SID(request)\
	request.id1.idtype == IDMAP_SID

#define	IS_REQUEST_UID(request)\
	request.id1.idtype == IDMAP_UID

#define	IS_REQUEST_GID(request)\
	request.id1.idtype == IDMAP_GID

/* ARGSUSED */
bool_t
idmap_get_mapped_ids_1_svc(idmap_mapping_batch batch,
		idmap_ids_res *result, struct svc_req *rqstp) {
	sqlite		*cache = NULL, *db = NULL;
	lookup_state_t	state;
	idmap_retcode	retcode, winrc;
	int		i;

	/* Init */
	(void) memset(result, 0, sizeof (*result));
	(void) memset(&state, 0, sizeof (state));

	/* Return success if nothing was requested */
	if (batch.idmap_mapping_batch_len < 1)
		goto out;

	/* Get cache handle */
	result->retcode = get_cache_handle(&cache);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	/* Get db handle */
	result->retcode = get_db_handle(&db);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	/* Allocate result array */
	result->ids.ids_val = calloc(batch.idmap_mapping_batch_len,
			sizeof (idmap_id_res));
	if (result->ids.ids_val == NULL) {
		idmapdlog(LOG_ERR, "Out of memory");
		result->retcode = IDMAP_ERR_MEMORY;
		goto out;
	}
	result->ids.ids_len = batch.idmap_mapping_batch_len;

	/* Init our 'done' flags */
	state.sid2pid_done = state.pid2sid_done = TRUE;

	/* First stage */
	for (i = 0; i < batch.idmap_mapping_batch_len; i++) {
		if (IS_BATCH_SID(batch, i)) {
			retcode = sid2pid_first_pass(
				&state,
				cache,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i]);
		} else if (IS_BATCH_UID(batch, i)) {
			retcode = pid2sid_first_pass(
				&state,
				cache,
				db,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i], 1, 0);
		} else if (IS_BATCH_GID(batch, i)) {
			retcode = pid2sid_first_pass(
				&state,
				cache,
				db,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i], 0, 0);
		} else {
			result->ids.ids_val[i].retcode = IDMAP_ERR_IDTYPE;
			continue;
		}
		if (IDMAP_FATAL_ERROR(retcode)) {
			result->retcode = retcode;
			goto out;
		}
	}

	/* Check if we are done */
	if (state.sid2pid_done == TRUE && state.pid2sid_done == TRUE)
		goto out;

	/* Process Windows server lookups for sid2name */
	if (state.ad_nqueries) {
		winrc = lookup_win_batch_sid2name(&state, &batch,
				result);
		if (IDMAP_FATAL_ERROR(winrc)) {
			result->retcode = winrc;
			goto out;
		}
	} else
		winrc = IDMAP_SUCCESS;

	/* Reset sid2pid 'done' flag */
	state.sid2pid_done = TRUE;

	/* Second stage */
	for (i = 0; i < batch.idmap_mapping_batch_len; i++) {
		/* Process sid to pid ONLY */
		if (IS_BATCH_SID(batch, i)) {
			if (IDMAP_ERROR(winrc))
				result->ids.ids_val[i].retcode = winrc;
			retcode = sid2pid_second_pass(
				&state,
				cache,
				db,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i]);
			if (IDMAP_FATAL_ERROR(retcode)) {
				result->retcode = retcode;
				goto out;
			}
		}
	}

	/* Check if we are done */
	if (state.sid2pid_done == TRUE && state.pid2sid_done == TRUE)
		goto out;

	/* Reset our 'done' flags */
	state.sid2pid_done = state.pid2sid_done = TRUE;

	/* Update cache in a single transaction */
	if (sql_exec_no_cb(cache, "BEGIN TRANSACTION;") != IDMAP_SUCCESS)
		goto out;

	for (i = 0; i < batch.idmap_mapping_batch_len; i++) {
		if (IS_BATCH_SID(batch, i)) {
			(void) update_cache_sid2pid(
				&state,
				cache,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i]);
		} else if ((IS_BATCH_UID(batch, i)) ||
				(IS_BATCH_GID(batch, i))) {
			(void) update_cache_pid2sid(
				&state,
				cache,
				&batch.idmap_mapping_batch_val[i],
				&result->ids.ids_val[i]);
		}
	}

	/* Commit if we have atleast one successful update */
	if (state.sid2pid_done == FALSE || state.pid2sid_done == FALSE)
		(void) sql_exec_no_cb(cache, "COMMIT TRANSACTION;");
	else
		(void) sql_exec_no_cb(cache, "END TRANSACTION;");

out:
	if (IDMAP_ERROR(result->retcode)) {
		xdr_free(xdr_idmap_ids_res, (caddr_t)result);
		result->ids.ids_len = 0;
		result->ids.ids_val = NULL;
	}
	if (cache)
		(void) sqlite_close(cache);
	if (db)
		(void) sqlite_close(db);
	result->retcode = idmap_stat4prot(result->retcode);
	return (TRUE);
}


/* ARGSUSED */
static int
list_mappings_cb(void *parg, int argc, char **argv, char **colnames) {
	list_cb_data_t		*cb_data;
	char			*str;
	idmap_mappings_res	*result;
	idmap_utf8str		*ptr;
	idmap_retcode		retcode;
	int			w2u, u2w;
	char			*end;

	cb_data = (list_cb_data_t *)parg;
	result = (idmap_mappings_res *)cb_data->result;

	_VALIDATE_LIST_CB_DATA(9, &result->mappings.mappings_val,
		sizeof (idmap_mapping));

	result->mappings.mappings_len++;

	if ((str = strdup(argv[1])) == NULL)
		return (1);
	result->mappings.mappings_val[cb_data->next].id1.idmap_id_u.sid.prefix =
		str;
	result->mappings.mappings_val[cb_data->next].id1.idmap_id_u.sid.rid =
		strtoul(argv[2], &end, 10);
	result->mappings.mappings_val[cb_data->next].id1.idtype = IDMAP_SID;

	result->mappings.mappings_val[cb_data->next].id2.idmap_id_u.uid =
		strtoul(argv[3], &end, 10);
	result->mappings.mappings_val[cb_data->next].id2.idtype = IDMAP_UID;

	w2u = argv[4]?strtol(argv[4], &end, 10):0;
	u2w = argv[5]?strtol(argv[5], &end, 10):0;

	if (w2u > 0 && u2w == 0)
		result->mappings.mappings_val[cb_data->next].direction =
		    IDMAP_DIRECTION_W2U;
	else if (w2u == 0 && u2w > 0)
		result->mappings.mappings_val[cb_data->next].direction =
		    IDMAP_DIRECTION_U2W;
	else
		result->mappings.mappings_val[cb_data->next].direction =
		    IDMAP_DIRECTION_BI;

	ptr = &result->mappings.mappings_val[cb_data->next].id1domain;
	if (idmap_str2utf8(&ptr, argv[6], 0) != IDMAP_SUCCESS)
		return (1);

	ptr = &result->mappings.mappings_val[cb_data->next].id1name;
	if (idmap_str2utf8(&ptr, argv[7], 0) != IDMAP_SUCCESS)
		return (1);

	ptr = &result->mappings.mappings_val[cb_data->next].id2name;
	if (idmap_str2utf8(&ptr, argv[8], 0) != IDMAP_SUCCESS)
		return (1);

	result->lastrowid = strtoll(argv[0], &end, 10);
	cb_data->next++;
	result->retcode = IDMAP_SUCCESS;
	return (0);
}


/* ARGSUSED */
bool_t
idmap_list_mappings_1_svc(bool_t is_user, int64_t lastrowid,
		uint64_t limit, idmap_mappings_res *result,
		struct svc_req *rqstp) {
	sqlite		*cache = NULL;
	char		lbuf[30], rbuf[30];
	uint64_t	maxlimit;
	idmap_retcode	retcode;
	char		*sql = NULL;

	(void) memset(result, 0, sizeof (*result));
	lbuf[0] = rbuf[0] = 0;

	RDLOCK_CONFIG();
	maxlimit = _idmapdstate.cfg->pgcfg.list_size_limit;
	UNLOCK_CONFIG();

	/* Get cache handle */
	result->retcode = get_cache_handle(&cache);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	result->retcode = IDMAP_ERR_INTERNAL;

	/* Create LIMIT expression. */
	if (limit == 0 || (maxlimit > 0 && maxlimit < limit))
		limit = maxlimit;
	if (limit > 0)
		(void) snprintf(lbuf, sizeof (lbuf),
			"LIMIT %" PRIu64, limit + 1ULL);

	(void) snprintf(rbuf, sizeof (rbuf), "rowid > %" PRIu64, lastrowid);

	/*
	 * Combine all the above into a giant SELECT statement that
	 * will return the requested mappings
	 */
	sql = sqlite_mprintf("SELECT rowid, sidprefix, rid, pid, w2u, u2w,"
			" windomain, winname, unixname"
			" FROM idmap_cache WHERE "
			" %s AND is_user = %d %s;",
			rbuf, is_user?1:0, lbuf);
	if (sql == NULL) {
		idmapdlog(LOG_ERR, "Out of memory");
		goto out;
	}

	/* Execute the SQL statement and update the return buffer */
	PROCESS_LIST_SVC_SQL(retcode, cache, sql, limit, list_mappings_cb,
		result, result->mappings.mappings_len);

out:
	if (sql)
		sqlite_freemem(sql);
	if (IDMAP_ERROR(result->retcode))
		(void) xdr_free(xdr_idmap_mappings_res, (caddr_t)result);
	if (cache)
		(void) sqlite_close(cache);
	result->retcode = idmap_stat4prot(result->retcode);
	return (TRUE);
}


/* ARGSUSED */
static int
list_namerules_cb(void *parg, int argc, char **argv, char **colnames) {
	list_cb_data_t		*cb_data;
	idmap_namerules_res	*result;
	idmap_retcode		retcode;
	idmap_utf8str		*ptr;
	int			w2u_order, u2w_order;
	char			*end;

	cb_data = (list_cb_data_t *)parg;
	result = (idmap_namerules_res *)cb_data->result;

	_VALIDATE_LIST_CB_DATA(8, &result->rules.rules_val,
		sizeof (idmap_namerule));

	result->rules.rules_len++;

	result->rules.rules_val[cb_data->next].is_user =
		strtol(argv[1], &end, 10);

	ptr = &result->rules.rules_val[cb_data->next].windomain;
	if (idmap_str2utf8(&ptr, argv[2], 0) != IDMAP_SUCCESS)
		return (1);

	ptr = &result->rules.rules_val[cb_data->next].winname;
	if (idmap_str2utf8(&ptr, argv[3], 0) != IDMAP_SUCCESS)
		return (1);

	result->rules.rules_val[cb_data->next].is_nt4 =
		strtol(argv[4], &end, 10);

	ptr = &result->rules.rules_val[cb_data->next].unixname;
	if (idmap_str2utf8(&ptr, argv[5], 0) != IDMAP_SUCCESS)
		return (1);

	w2u_order = argv[6]?strtol(argv[6], &end, 10):0;
	u2w_order = argv[7]?strtol(argv[7], &end, 10):0;

	if (w2u_order > 0 && u2w_order == 0)
		result->rules.rules_val[cb_data->next].direction =
		    IDMAP_DIRECTION_W2U;
	else if (w2u_order == 0 && u2w_order > 0)
		result->rules.rules_val[cb_data->next].direction =
		    IDMAP_DIRECTION_U2W;
	else
		result->rules.rules_val[cb_data->next].direction =
		    IDMAP_DIRECTION_BI;

	result->lastrowid = strtoll(argv[0], &end, 10);
	cb_data->next++;
	result->retcode = IDMAP_SUCCESS;
	return (0);
}


/* ARGSUSED */
bool_t
idmap_list_namerules_1_svc(idmap_namerule rule, uint64_t lastrowid,
		uint64_t limit, idmap_namerules_res *result,
		struct svc_req *rqstp) {

	sqlite		*db = NULL;
	char		w2ubuf[15], u2wbuf[15];
	char		lbuf[30], rbuf[30];
	char		*sql = NULL;
	char		*s_windomain = NULL, *s_winname = NULL;
	char		*s_unixname = NULL;
	uint64_t	maxlimit;
	idmap_retcode	retcode;

	(void) memset(result, 0, sizeof (*result));
	lbuf[0] = rbuf[0] = 0;

	RDLOCK_CONFIG();
	maxlimit = _idmapdstate.cfg->pgcfg.list_size_limit;
	UNLOCK_CONFIG();

	/* Get db handle */
	result->retcode = get_db_handle(&db);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	result->retcode = IDMAP_ERR_INTERNAL;

	if (rule.direction < 0) {
		w2ubuf[0] = u2wbuf[0] = 0;
	} else if (rule.direction == IDMAP_DIRECTION_BI) {
		(void) snprintf(w2ubuf, sizeof (w2ubuf), "AND w2u_order > 0");
		(void) snprintf(u2wbuf, sizeof (u2wbuf), "AND u2w_order > 0");
	} else if (rule.direction == IDMAP_DIRECTION_W2U) {
		(void) snprintf(w2ubuf, sizeof (w2ubuf), "AND w2u_order > 0");
		(void) snprintf(u2wbuf, sizeof (u2wbuf),
				"AND (u2w_order = 0 OR u2w_order ISNULL)");
	} else if (rule.direction == IDMAP_DIRECTION_U2W) {
		(void) snprintf(w2ubuf, sizeof (w2ubuf),
				"AND (w2u_order = 0 OR w2u_order ISNULL)");
		(void) snprintf(u2wbuf, sizeof (u2wbuf), "AND u2w_order > 0");
	}

	/* Create where statement for windomain */
	if (rule.windomain.idmap_utf8str_len > 0) {
		if (gen_sql_expr_from_utf8str("AND", "windomain", "=",
				&rule.windomain,
				"", &s_windomain) != IDMAP_SUCCESS)
			goto out;
	}

	/* Create where statement for winname */
	if (rule.winname.idmap_utf8str_len > 0) {
		if (gen_sql_expr_from_utf8str("AND", "winname", "=",
				&rule.winname,
				"", &s_winname) != IDMAP_SUCCESS)
			goto out;
	}

	/* Create where statement for unixname */
	if (rule.unixname.idmap_utf8str_len > 0) {
		if (gen_sql_expr_from_utf8str("AND", "unixname", "=",
				&rule.unixname,
				"", &s_unixname) != IDMAP_SUCCESS)
			goto out;
	}

	/* Create LIMIT expression. */
	if (limit == 0 || (maxlimit > 0 && maxlimit < limit))
		limit = maxlimit;
	if (limit > 0)
		(void) snprintf(lbuf, sizeof (lbuf),
			"LIMIT %" PRIu64, limit + 1ULL);

	(void) snprintf(rbuf, sizeof (rbuf), "rowid > %" PRIu64, lastrowid);

	/*
	 * Combine all the above into a giant SELECT statement that
	 * will return the requested rules
	 */
	sql = sqlite_mprintf("SELECT rowid, is_user, windomain, winname, "
			"is_nt4, unixname, w2u_order, u2w_order "
			"FROM namerules WHERE "
			" %s AND is_user = %d %s %s %s %s %s %s;",
			rbuf, rule.is_user?1:0,
			s_windomain?s_windomain:"",
			s_winname?s_winname:"",
			s_unixname?s_unixname:"",
			w2ubuf, u2wbuf, lbuf);
	if (sql == NULL) {
		idmapdlog(LOG_ERR, "Out of memory");
		goto out;
	}

	/* Execute the SQL statement and update the return buffer */
	PROCESS_LIST_SVC_SQL(retcode, db, sql, limit, list_namerules_cb,
		result, result->rules.rules_len);

out:
	if (s_windomain)
		sqlite_freemem(s_windomain);
	if (s_winname)
		sqlite_freemem(s_winname);
	if (s_unixname)
		sqlite_freemem(s_unixname);
	if (sql)
		sqlite_freemem(sql);
	if (IDMAP_ERROR(result->retcode))
		(void) xdr_free(xdr_idmap_namerules_res, (caddr_t)result);
	if (db)
		(void) sqlite_close(db);
	result->retcode = idmap_stat4prot(result->retcode);
	return (TRUE);
}

#define	IDMAP_RULES_AUTH	"solaris.admin.idmap.rules"
static int
verify_rules_auth(struct svc_req *rqstp) {
	ucred_t		*uc = NULL;
	uid_t		uid;
	char		buf[1024];
	struct passwd	pwd;
	const char	*me = "verify_rules_auth";

	if (svc_getcallerucred(rqstp->rq_xprt, &uc) != 0) {
		idmapdlog(LOG_ERR,
			"%s: svc_getcallerucred failed (errno=%d)",
			me, errno);
		return (-1);
	}

	uid = ucred_geteuid(uc);
	if (uid == (uid_t)-1) {
		idmapdlog(LOG_ERR,
			"%s: ucred_geteuid failed (errno=%d)",
			me, errno);
		ucred_free(uc);
		return (-1);
	}

	if (getpwuid_r(uid, &pwd, buf, sizeof (buf)) == NULL) {
		idmapdlog(LOG_ERR,
			"%s: getpwuid_r(%u) failed (errno=%d)",
			me, uid, errno);
		ucred_free(uc);
		return (-1);
	}

	if (chkauthattr(IDMAP_RULES_AUTH, pwd.pw_name) != 1) {
		idmapdlog(LOG_INFO,
			"%s: %s does not have authorization.",
			me, pwd.pw_name);
		ucred_free(uc);
		return (-1);
	}

	ucred_free(uc);
	return (1);
}

/* ARGSUSED */
bool_t
idmap_update_1_svc(idmap_update_batch batch, idmap_retcode *result,
		struct svc_req *rqstp) {
	sqlite		*db = NULL;
	idmap_update_op	*up;
	int		i;

	if (verify_rules_auth(rqstp) < 0) {
		*result = IDMAP_ERR_PERMISSION_DENIED;
		goto out;
	}

	if (batch.idmap_update_batch_len == 0 ||
			batch.idmap_update_batch_val == NULL) {
		*result = IDMAP_SUCCESS;
		goto out;
	}

	/* Get db handle */
	*result = get_db_handle(&db);
	if (*result != IDMAP_SUCCESS)
		goto out;

	*result = sql_exec_no_cb(db, "BEGIN TRANSACTION;");
	if (*result != IDMAP_SUCCESS)
		goto out;

	for (i = 0; i < batch.idmap_update_batch_len; i++) {
		up = &batch.idmap_update_batch_val[i];
		switch (up->opnum) {
		case OP_NONE:
			*result = IDMAP_SUCCESS;
			break;
		case OP_ADD_NAMERULE:
			*result = add_namerule(db,
				&up->idmap_update_op_u.rule);
			break;
		case OP_RM_NAMERULE:
			*result = rm_namerule(db,
				&up->idmap_update_op_u.rule);
			break;
		case OP_FLUSH_NAMERULES:
			*result = flush_namerules(db,
				up->idmap_update_op_u.is_user);
			break;
		default:
			*result = IDMAP_ERR_NOTSUPPORTED;
			goto out;
		};

		if (*result != IDMAP_SUCCESS)
			goto out;
	}

out:
	if (*result == IDMAP_SUCCESS && db) {
		*result = sql_exec_no_cb(db, "COMMIT TRANSACTION;");
	}

	if (db)
		(void) sqlite_close(db);
	*result = idmap_stat4prot(*result);
	return (TRUE);
}


/* ARGSUSED */
bool_t
idmap_get_mapped_id_by_name_1_svc(idmap_mapping request,
		idmap_mappings_res *result, struct svc_req *rqstp) {
	sqlite		*cache = NULL, *db = NULL;

	/* Init */
	(void) memset(result, 0, sizeof (*result));

	/* Get cache handle */
	result->retcode = get_cache_handle(&cache);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	/* Get db handle */
	result->retcode = get_db_handle(&db);
	if (result->retcode != IDMAP_SUCCESS)
		goto out;

	/* Allocate result */
	result->mappings.mappings_val = calloc(1, sizeof (idmap_mapping));
	if (result->mappings.mappings_val == NULL) {
		idmapdlog(LOG_ERR, "Out of memory");
		result->retcode = IDMAP_ERR_MEMORY;
		goto out;
	}
	result->mappings.mappings_len = 1;

	if (IS_REQUEST_SID(request)) {
		result->retcode = get_w2u_mapping(
			cache,
			db,
			&request,
			result->mappings.mappings_val);
	} else if (IS_REQUEST_UID(request)) {
		result->retcode = get_u2w_mapping(
			cache,
			db,
			&request,
			result->mappings.mappings_val,
			1);
	} else if (IS_REQUEST_GID(request)) {
		result->retcode = get_u2w_mapping(
			cache,
			db,
			&request,
			result->mappings.mappings_val,
			0);
	} else {
		result->retcode = IDMAP_ERR_IDTYPE;
	}

out:
	if (IDMAP_FATAL_ERROR(result->retcode)) {
		xdr_free(xdr_idmap_mappings_res, (caddr_t)result);
		result->mappings.mappings_len = 0;
		result->mappings.mappings_val = NULL;
	}
	if (cache)
		(void) sqlite_close(cache);
	if (db)
		(void) sqlite_close(db);
	result->retcode = idmap_stat4prot(result->retcode);
	return (TRUE);
}


/* ARGSUSED */
int
idmap_prog_1_freeresult(SVCXPRT *transp, xdrproc_t xdr_result,
		caddr_t result) {
	(void) xdr_free(xdr_result, result);
	return (TRUE);
}