/*	$NetBSD: ypxfr.c,v 1.2 1996/08/15 22:17:57 chuck Exp $	*/

/*
 * Copyright (c) 1994 Mats O Jansson <moj@stacken.kth.se>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by Mats O Jansson
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>

#include <netinet/in.h>
#include <arpa/inet.h>

#include <fcntl.h>
#include <netdb.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <rpc/rpc.h>
#include <rpc/xdr.h>
#include <rpcsvc/yp_prot.h>
#include <rpcsvc/ypclnt.h>

#include "yplib_host.h"
#include "yplog.h"
#include "ypdb.h"
#include "ypdef.h"

extern	char *__progname;		/* from crt0.o */
extern	char *optarg;
extern	int optind;

DBM	*db;

static	int ypxfr_foreach __P((int, char *, int, char *, int, int *));

int	get_local_ordernum __P((char *, char *, u_int *));
int	get_remote_ordernum __P((CLIENT *, char *, char *, u_int, u_int *));
void	get_map __P((CLIENT *, char *, char *, struct ypall_callback *));
DBM	*create_db __P((char *, char *, char *));
int	install_db __P((char *, char *, char *));
int	unlink_db __P((char *, char *, char *));
int	add_order __P((DBM *, u_int));
int	add_master __P((CLIENT *, char *, char *, DBM *));
int	add_interdomain __P((CLIENT *, char *, char *, DBM *));
int	add_secure __P((CLIENT *, char *, char *, DBM *));
int	send_clear __P((CLIENT *));
int	send_reply __P((CLIENT *, int, int));

int
main(argc,argv)
	int argc;
	char **argv;
{
	int need_usage = 0, cflag = 0, fflag = 0, Cflag = 0;
	int ch;
	char *domain;
	char *host = NULL;
	char *srcdomain = NULL;
	char *tid = NULL;
	char *prog = NULL;
	char *ipadd = NULL;
	char *port = NULL;
	char *map = NULL;
	u_int ordernum, new_ordernum;
	struct ypall_callback callback;
	CLIENT *client;
	char mapname[] = "ypdbXXXXXX";
	int status, xfr_status;
	
	status = YPPUSH_SUCC;
	client = NULL;

	if (yp_get_default_domain(&domain))
		errx(1, "can't get YP domain name");

	while ((ch = getopt(argc, argv, "cd:fh:s:C:")) != -1) {
		switch (ch) {
		case 'c':
			cflag = 1;
			break;

		case 'd':
			domain = optarg;
			break;

		case 'f':
			fflag = 1;
			break;

		case 'h':
			host = optarg;
			break;

		case 's':
			srcdomain = optarg;
			break;

		case 'C':
			if (optind + 3 >= argc) {
				need_usage = 1;
				optind = argc;
				break;
			}
			Cflag = 1;
			tid = optarg;
			prog = argv[optind++];
			ipadd = argv[optind++];
			port = argv[optind++];
			break;

		default:
			need_usage = 1;
		}
	}
	argc -= optind; argv += optind;

	if (argc != 1)
		need_usage = 1;

	map = argv[0];

	if (need_usage) {
		status = YPPUSH_BADARGS;
		fprintf(stderr, "usage: %s [-cf] [-d domain] [-h host] %s\n",
		   __progname, "[-s domain] [-C tid prog ipadd port] mapname");
		goto punt;
	}

	ypopenlog();

	yplog("ypxfr: Arguments:");
	yplog("YP clear to local: %s", (cflag) ? "no" : "yes");
	yplog("   Force transfer: %s", (fflag) ? "yes" : "no");
	yplog("           domain: %s", domain); 
	yplog("             host: %s", host);
	yplog("    source domain: %s", srcdomain);
	yplog("          transid: %s", tid);
	yplog("             prog: %s", prog);
	yplog("             port: %s", port);
	yplog("            ipadd: %s", ipadd);
	yplog("              map: %s", map);

	if (fflag != 0)
		ordernum = 0;
	else {
		status = get_local_ordernum(domain, map, &ordernum);
		if (status < 0)
			goto punt;
	}

        yplog("Get Master");

	if (host == NULL) {
		if (srcdomain == NULL)
			status = yp_master(domain,map,&host);
	        else
			status = yp_master(srcdomain,map,&host);

		if (status == 0)
			status = YPPUSH_SUCC;
		else {
			status = -status;
			goto punt;
		}
	}

        yplog("Connect host: %s", host); 

	client = yp_bind_host(host,YPPROG,YPVERS,0,1);

	status = get_remote_ordernum(client,domain,map,
	    ordernum,&new_ordernum);
	

	if (status == YPPUSH_SUCC) {
		/* Create temporary db */
		mktemp(mapname);
		db = create_db(domain,map,mapname);
		if (db == NULL)
			status = YPPUSH_DBM;

	  	/* Add ORDER */
		if (status > 0)
			status = add_order(db, new_ordernum);
		
		/* Add MASTER */
		if (status > 0)
			status = add_master(client, domain, map, db);
		
	        /* Add INTERDOMAIN */
		if (status > 0)
			status = add_interdomain(client, domain, map, db);
		
	        /* Add SECURE */
		if (status > 0)
			status = add_secure(client, domain, map, db);
		
		if (status > 0) {
			callback.foreach = (int(*)())ypxfr_foreach;
			get_map(client, domain, map, &callback);
		}

		/* Close db */
		if (db != NULL)
			ypdb_close(db);

		/* Rename db */
		if (status > 0)
			status = install_db(domain, map, mapname);
		else
			status = unlink_db(domain, map, mapname);
	}
	
 punt:
	xfr_status = status;

	if (client != NULL)
		clnt_destroy(client);

	/* YP_CLEAR */
	if (!cflag) {
		client = yp_bind_local(YPPROG, YPVERS);
		status = send_clear(client);
		clnt_destroy(client);
	}

	if (Cflag > 0) {
		/* Send Response */
		client = yp_bind_host(ipadd, atoi(prog), 1, atoi(port), 0);
		status = send_reply(client, xfr_status, atoi(tid));
		clnt_destroy(client);
	}

	exit (0);
}

static int
ypxfr_foreach(status, keystr, keylen, valstr, vallen, data)
	int status;
	char *keystr;
	int keylen;
	char *valstr;
	int vallen;
	int *data;
{
	datum key, val;

	if (status == YP_NOMORE)
		return (0);

	keystr[keylen] = '\0';
	valstr[vallen] = '\0';

	key.dptr = keystr;
	key.dsize = strlen(keystr);

	val.dptr = valstr;
	val.dsize = strlen(valstr);

	ypdb_store(db, key, val, YPDB_INSERT);

	return (0);
}

int
get_local_ordernum(domain, map, lordernum)
	char *domain, *map;
	u_int *lordernum;
{
	char map_path[1024];
	char order_key[] = YP_LAST_KEY;
	char order[MAX_LAST_LEN+1];
	struct stat finfo;
	DBM *db;
	datum k,v;
	int status;

	status = YPPUSH_SUCC;

	snprintf(map_path, sizeof(map_path), "%s/%s", YP_DB_PATH, domain);
	map_path[sizeof(map_path) - 1] = '\0';

	/* Make sure we serve the domain. */
	if ((stat(map_path, &finfo)) != 0 ||
	    (S_ISDIR(finfo.st_mode) == 0)) {
		warnx("domain `%s' not found locally");
		status = YPPUSH_NODOM;
		goto out;
	}

	/* Make sure we serve the map. */
	snprintf(map_path, sizeof(map_path), "%s/%s/%s%s",
	    YP_DB_PATH, domain, map, YPDB_SUFFIX);
	map_path[sizeof(map_path) - 1] = '\0';
	if (stat(map_path, &finfo) != 0) {
		status = YPPUSH_NOMAP;
		goto out;
	}

	/* Open the map file. */
	snprintf(map_path, sizeof(map_path), "%s/%s/%s",
	    YP_DB_PATH, domain, map);
	map_path[sizeof(map_path) - 1] = '\0';
	db = ypdb_open(map_path, O_RDONLY, 0444);
	if (db == NULL) {
		status = YPPUSH_DBM;
		goto out;
	}

	k.dptr = (char *)&order_key;
	k.dsize = YP_LAST_LEN;

	v = ypdb_fetch(db, k);
	ypdb_close(db);

	if (v.dptr == NULL)
		*lordernum = 0;
	else {
		strncpy(order, v.dptr, v.dsize);
		order[v.dsize] = '\0';
		*lordernum = (u_int)atoi((char *)&order);
	}

 out:
	if ((status == YPPUSH_NOMAP) || (status == YPPUSH_DBM)) {
		*lordernum = 0;
		status = YPPUSH_SUCC;
	}

	return (status);
}

int
get_remote_ordernum(client, domain, map, lordernum, rordernum)
	CLIENT *client;
	char *domain, *map;
	u_int lordernum, *rordernum;
{
	int status;

	status = yp_order_host(client, domain, map, (int *)rordernum);

	if (status == 0)
		if (*rordernum <= lordernum)
			status = YPPUSH_AGE;
		else
			status = YPPUSH_SUCC;

	return status;
}

void
get_map(client, domain, map, incallback)
	CLIENT *client;
	char *domain, *map;
	struct ypall_callback *incallback;
{

	(void)yp_all_host(client, domain, map, incallback);
}

DBM *
create_db(domain, map, temp_map)
	char *domain, *map, *temp_map;
{
	char db_temp[255];
	DBM *db;

	snprintf(db_temp, sizeof(db_temp), "%s/%s/%s",
	    YP_DB_PATH, domain, temp_map);
	db_temp[sizeof(db_temp) - 1] = '\0';

	db = ypdb_open(db_temp, O_RDWR|O_CREAT, 0444);

	return db;
}

int
install_db(domain,map,temp_map)
	char *domain, *map, *temp_map;
{
	char db_name[255], db_temp[255];

	snprintf(db_name, sizeof(db_name), "%s/%s/%s%s",
	    YP_DB_PATH, domain, map, YPDB_SUFFIX);
	db_name[sizeof(db_name) - 1] = '\0';

	snprintf(db_temp, sizeof(db_temp), "%s/%s/%s%s",
	    YP_DB_PATH, domain, temp_map, YPDB_SUFFIX);
	db_temp[sizeof(db_temp) - 1] = '\0';

	if (rename(db_temp, db_name)) {
		warn("can't rename `%s' -> `%s'", db_temp, db_name);
		return YPPUSH_YPERR;
	}

	return YPPUSH_SUCC;
}

int
unlink_db(domain, map, temp_map)
	char *domain, *map, *temp_map;
{
	char db_temp[255];

	snprintf(db_temp, sizeof(db_temp), "%s/%s/%s%s",
	    YP_DB_PATH, domain, temp_map, YPDB_SUFFIX);
	db_temp[sizeof(db_temp) - 1] = '\0';

	if (unlink(db_temp)) {
		warn("can't unlink `%s'", db_temp);
		return YPPUSH_YPERR;
	}

	return YPPUSH_SUCC;
}

int
add_order(db, ordernum)
	DBM *db;
	u_int ordernum;
{
	char datestr[11];
	datum key, val;
	char keystr[] = YP_LAST_KEY;
	int status;

	snprintf(datestr, sizeof(datestr), "%010d", ordernum);
	datestr[sizeof(datestr) - 1] = '\0';

	key.dptr = keystr;
	key.dsize = strlen(keystr);
	
	val.dptr = datestr;
	val.dsize = strlen(datestr);
	
	status = ypdb_store(db, key, val, YPDB_INSERT);
	if(status >= 0)
		status = YPPUSH_SUCC;
	else
		status = YPPUSH_DBM;

	return (status);
}

int
add_master(client, domain, map, db)
	CLIENT *client;
	char *domain, *map;
	DBM *db;
{
	char keystr[] = YP_MASTER_KEY;
	char *master;
	int status;
	datum key,val;

	master = NULL;

	/* Get MASTER */
	status = yp_master_host(client, domain, map, &master);
	
	if (master != NULL) {
		key.dptr = keystr;
		key.dsize = strlen(keystr);

		val.dptr = master;
		val.dsize = strlen(master);

		status = ypdb_store(db, key, val, YPDB_INSERT);
		if (status >= 0)
			status = YPPUSH_SUCC;
		else
			status = YPPUSH_DBM;
	}

	return status;
}

int
add_interdomain(client, domain, map, db)
	CLIENT *client;
	char *domain, *map;
	DBM *db;
{
	char keystr[] = YP_INTERDOMAIN_KEY;
	char *value;
	int vallen;
	int status;
	datum k, v;

	/* Get INTERDOMAIN */
	k.dptr = keystr;
	k.dsize = strlen(keystr);

	status = yp_match_host(client, domain, map,
	    k.dptr, k.dsize, &value, &vallen);

	if (status == 0 && value) {
		v.dptr = value;
		v.dsize = vallen;
		
		if (v.dptr != NULL) {
			status = ypdb_store(db, k, v, YPDB_INSERT);
			if (status >= 0)
				status = YPPUSH_SUCC;
			else
				status = YPPUSH_DBM;
		}
	}

	return status;
}

int
add_secure(client, domain, map, db)
	CLIENT *client;
	char *domain, *map;
	DBM *db;
{
	char keystr[] = YP_SECURE_KEY;
	char *value;
	int vallen;
	int status;
	datum k, v;

	/* Get SECURE */
	k.dptr = keystr;
	k.dsize = strlen(keystr);

	status = yp_match_host(client, domain, map,
	    k.dptr, k.dsize, &value, &vallen);
	
	if (status > 0) {
		v.dptr = value;
		v.dsize = vallen;
		
		if (v.dptr != NULL) {
			status = ypdb_store(db, k, v, YPDB_INSERT);
			if (status >= 0)
				status = YPPUSH_SUCC;
			else
				status = YPPUSH_DBM;
		}
	}

	return status;
}

int
send_clear(client)
	CLIENT *client;
{
	struct timeval tv;
	int r;
	int status;

	status = YPPUSH_SUCC;

	tv.tv_sec = 10;
	tv.tv_usec = 0;

	/* Send CLEAR */
	r = clnt_call(client, YPPROC_CLEAR, xdr_void, 0, xdr_void, 0, tv);
	if (r != RPC_SUCCESS) {
		clnt_perror(client, "yp_clear: clnt_call");
		status = YPPUSH_RPC;
	}

	return status;
}

int
send_reply(client, status, tid)
	CLIENT *client;
	int status, tid;
{
	struct timeval tv;
	struct ypresp_xfr resp;
	int r;

	tv.tv_sec = 10;
	tv.tv_usec = 0;

	resp.transid = tid;
	resp.xfrstat = status;

	/* Send XFRRESP */
	r = clnt_call(client, YPPUSHPROC_XFRRESP, xdr_ypresp_xfr, &resp,
	    xdr_void, 0, tv);
	if (r != RPC_SUCCESS) {
		clnt_perror(client, "yppushresp_xdr: clnt_call");
		status = YPPUSH_RPC;
	}

	return status;
}