/*
 * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * This program 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 any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

FILE_LICENCE ( GPL2_OR_LATER );

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <stddef.h>
#include <string.h>
#include <assert.h>
#include <getopt.h>
#include <gpxe/netdevice.h>
#include <gpxe/in.h>
#include <gpxe/command.h>
#include <usr/dhcpmgmt.h>
#include <usr/ifmgmt.h>

/** @file
 *
 * DHCP management commands
 *
 */

/**
 * "dhcp" command syntax message
 *
 * @v argv		Argument list
 */
static void dhcp_syntax ( char **argv ) {
	printf ( "Usage:\n"
		 "  %s <interface> [<interface> ...]\n"
		 "  %s any\n"
		 "\n"
		 "Configure a network interface using DHCP\n",
		 argv[0], argv[0] );
}

/**
 * Attempt to configure a device with dhcp
 *
 * @v netdev		Device to configure
 * @ret rc		Exit code
 */
static int dhcp_one_device ( struct net_device *netdev ) {
	int rc;

	/* Perform DHCP */
	if ( ( rc = dhcp ( netdev ) ) != 0 ) {
		/* Close the device on error to avoid out-of-memory */
		netdev_close ( netdev );

		printf ( "Could not configure %s: %s\n", netdev->name,
			 strerror ( rc ) );
		return 1;
	}

	return 0;
}

/**
 * Call dhcp_one_device() for each name in argv
 *
 * @v argc		Number of devices
 * @v argv		List of device names
 * @ret rc		Exit code
 */
static int dhcp_each_device_name ( int argc, char **argv ) {
	int i;
	char *netdev_name;
	struct net_device *netdev;

	for ( i = 0; i < argc; i++ ) {
		netdev_name = argv[i];
		netdev = find_netdev ( netdev_name );

		if ( ! netdev ) {
			printf ( "No such interface: %s\n", netdev_name );
			continue;
		}

		if ( dhcp_one_device ( netdev ) == 0 )
			return 0;
	}

	printf ( "Could not configure any interface.\n" );
	return 1;
}

/**
 * Call dhcp_one_device() for each device in net_devices
 *
 * @ret rc		Exit code
 */
static int dhcp_each_device ( void ) {
	struct net_device *netdev;

	for_each_netdev ( netdev ) {
		if ( dhcp_one_device ( netdev ) == 0 )
			return 0;
	}

	printf ( "Could not configure any interface.\n" );
	return 1;
}

/**
 * The "dhcp" command
 *
 * @v argc		Argument count
 * @v argv		Argument list
 * @ret rc		Exit code
 */
static int dhcp_exec ( int argc, char **argv ) {
	static struct option longopts[] = {
		{ "help", 0, NULL, 'h' },
		{ NULL, 0, NULL, 0 },
	};
	int c;

	/* Parse options */
	while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
		switch ( c ) {
		case 'h':
			/* Display help text */
		default:
			/* Unrecognised/invalid option */
			dhcp_syntax ( argv );
			return 1;
		}
	}

	/* Need one or more interface names remaining after the options */
	if ( ( argc - optind ) < 1 ) {
		dhcp_syntax ( argv );
		return 1;
	}

	if ( strcmp ( argv[optind], "any" ) == 0 )
		return dhcp_each_device();

	return dhcp_each_device_name ( argc - optind, argv + optind );
}

/**
 * "pxebs" command syntax message
 *
 * @v argv		Argument list
 */
static void pxebs_syntax ( char **argv ) {
	printf ( "Usage:\n"
		 "  %s <interface> <server_type>\n"
		 "\n"
		 "Perform PXE Boot Server discovery\n",
		 argv[0] );
}

/**
 * The "pxebs" command
 *
 * @v argc		Argument count
 * @v argv		Argument list
 * @ret rc		Exit code
 */
static int pxebs_exec ( int argc, char **argv ) {
	static struct option longopts[] = {
		{ "help", 0, NULL, 'h' },
		{ NULL, 0, NULL, 0 },
	};
	const char *netdev_txt;
	const char *pxe_type_txt;
	struct net_device *netdev;
	unsigned int pxe_type;
	char *end;
	int c;
	int rc;

	/* Parse options */
	while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
		switch ( c ) {
		case 'h':
			/* Display help text */
		default:
			/* Unrecognised/invalid option */
			pxebs_syntax ( argv );
			return 1;
		}
	}
	if ( optind != ( argc - 2 ) ) {
		pxebs_syntax ( argv );
		return 1;
	}
	netdev_txt = argv[optind];
	pxe_type_txt = argv[ optind + 1 ];

	/* Parse arguments */
	netdev = find_netdev ( netdev_txt );
	if ( ! netdev ) {
		printf ( "No such interface: %s\n", netdev_txt );
		return 1;
	}
	pxe_type = strtoul ( pxe_type_txt, &end, 0 );
	if ( *end ) {
		printf ( "Bad server type: %s\n", pxe_type_txt );
		return 1;
	}

	/* Perform Boot Server Discovery */
	if ( ( rc = pxebs ( netdev, pxe_type ) ) != 0 ) {
		printf ( "Could not discover boot server on %s: %s\n",
			 netdev->name, strerror ( rc ) );
		return 1;
	}

	return 0;
}

/** DHCP management commands */
struct command dhcp_commands[] __command = {
	{
		.name = "dhcp",
		.exec = dhcp_exec,
	},
	{
		.name = "pxebs",
		.exec = pxebs_exec,
	},
};