[PATCH 1/1] TFTP: Add support for custom server and port

[Oleksij Rempel]

Am 21.08.2017 um 18:23 schrieb Clément Leger:
> From de97046a780acf75263f932b34b32eb33a9d6eea Mon Sep 17 00:00:00 2001
> From: Clement Leger <clement.leger at kalray.eu>
> Date: Mon, 21 Aug 2017 18:08:01 +0200
> Subject: [PATCH] TFTP: Add support for custom server and port
> 
> This patch add argument to the tftp command which allow to specify a
> custom tftp server and port. These options are now given to the tftp
> filesystem by using serveri[:port] format.
> In order to maintain backward compatibity with direct mount usage, the
> port is optional and defaults to the standard tftp port if not given.
> These parameters are useful when using a different server than the 
> one provided by dhcp and not running on a standard port.

Hm... it should be possible to overwrite serverip in the interface file
/env/network/eth0, even if dhcp is used:
http://www.barebox.org/doc/latest/user/booting-linux.html#network-boot

is it not working for you? or are there some other reasons to have this
parameter?

> Signed-off-by: Clement Leger <clement.leger at kalray.eu>
> ---
>  commands/tftp.c | 31 +++++++++++++++++++++++++------
>  fs/tftp.c       | 19 ++++++++++++++++---
>  2 files changed, 41 insertions(+), 9 deletions(-)
> 
> diff --git a/commands/tftp.c b/commands/tftp.c
> index 08366b4..3c6ec72 100644
> --- a/commands/tftp.c
> +++ b/commands/tftp.c
> @@ -28,6 +28,10 @@
>  
>  #define TFTP_MOUNT_PATH	"/.tftp_tmp_path"
>  
> +#define IPV4_STR_SIZE	sizeof("255.255.255.255:12345")
> +
> +#define TFTP_PORT	69
> +
>  static int do_tftpb(int argc, char *argv[])
>  {
>  	char *source, *dest, *freep;
> @@ -36,13 +40,22 @@ static int do_tftpb(int argc, char *argv[])
>  	int tftp_push = 0;
>  	int ret;
>  	IPaddr_t ip;
> -	char ip4_str[sizeof("255.255.255.255")];
> +	int port = TFTP_PORT;
> +	char server_str[IPV4_STR_SIZE] = {0};
> +	int server_given = 0;
>  
> -	while ((opt = getopt(argc, argv, "p")) > 0) {
> +	while ((opt = getopt(argc, argv, "pP:s:")) > 0) {
>  		switch(opt) {
>  		case 'p':
>  			tftp_push = 1;
>  			break;
> +		case 'P':
> +			port = simple_strtol(optarg, NULL, 0);
> +			break;
> +		case 's':
> +			server_given = 1;
> +			strncpy(server_str, optarg, IPV4_STR_SIZE - 1);
> +			break;
>  		default:
>  			return COMMAND_ERROR_USAGE;
>  		}
> @@ -73,9 +86,13 @@ static int do_tftpb(int argc, char *argv[])
>  	if (ret)
>  		goto err_free;
>  
> -	ip = net_get_serverip();
> -	sprintf(ip4_str, "%pI4", &ip);
> -	ret = mount(ip4_str, "tftp", TFTP_MOUNT_PATH, NULL);
> +	if (!server_given) {
> +		ip = net_get_serverip();
> +		sprintf(server_str, "%pI4", &ip);
> +	}
> +
> +	snprintf(server_str + strlen(server_str), IPV4_STR_SIZE - 1, ":%d", port);
> +	ret = mount(server_str, "tftp", TFTP_MOUNT_PATH, NULL);
>  	if (ret)
>  		goto err_rmdir;
>  
> @@ -100,12 +117,14 @@ BAREBOX_CMD_HELP_TEXT("server address is taken from the environment (ethX.server
>  BAREBOX_CMD_HELP_TEXT("")
>  BAREBOX_CMD_HELP_TEXT("Options:")
>  BAREBOX_CMD_HELP_OPT ("-p", "push to TFTP server")
> +BAREBOX_CMD_HELP_OPT ("-P", "TFTP port")
> +BAREBOX_CMD_HELP_OPT ("-s", "Server IP")
>  BAREBOX_CMD_HELP_END
>  
>  BAREBOX_CMD_START(tftp)
>  	.cmd		= do_tftpb,
>  	BAREBOX_CMD_DESC("load (or save) a file using TFTP")
> -	BAREBOX_CMD_OPTS("[-p] SOURCE [DEST]")
> +	BAREBOX_CMD_OPTS("[-pPs] SOURCE [DEST]")
>  	BAREBOX_CMD_GROUP(CMD_GRP_NET)
>  	BAREBOX_CMD_HELP(cmd_tftp_help)
>  BAREBOX_CMD_END
> diff --git a/fs/tftp.c b/fs/tftp.c
> index 847921a..c4a2769 100644
> --- a/fs/tftp.c
> +++ b/fs/tftp.c
> @@ -87,6 +87,7 @@ struct file_priv {
>  
>  struct tftp_priv {
>  	IPaddr_t server;
> +	uint16_t port;
>  };
>  
>  static int tftp_create(struct device_d *dev, const char *pathname, mode_t mode)
> @@ -423,7 +424,7 @@ static struct file_priv *tftp_do_open(struct device_d *dev,
>  		goto out;
>  	}
>  
> -	priv->tftp_con = net_udp_new(tpriv->server, TFTP_PORT, tftp_handler,
> +	priv->tftp_con = net_udp_new(tpriv->server, tpriv->port, tftp_handler,
>  			priv);
>  	if (IS_ERR(priv->tftp_con)) {
>  		ret = PTR_ERR(priv->tftp_con);
> @@ -647,12 +648,24 @@ static int tftp_probe(struct device_d *dev)
>  {
>  	struct fs_device_d *fsdev = dev_to_fs_device(dev);
>  	struct tftp_priv *priv = xzalloc(sizeof(struct tftp_priv));
> +	char *tmp = xstrdup(fsdev->backingstore);
> +	char *port;
> +	int ret = 0;
>  
>  	dev->priv = priv;
>  
> -	priv->server = resolv(fsdev->backingstore);
> +	port = strchr(tmp, ':');
> +	if (!port) {
> +		priv->port = TFTP_PORT;
> +	} else {
> +		*port = 0;
> +		priv->port = simple_strtol(port + 1, NULL, 0);
> +	}
>  
> -	return 0;
> +	priv->server = resolv(tmp);
> +
> +	free(tmp);
> +	return ret;
>  }
>  
>  static void tftp_remove(struct device_d *dev)
> 


-- 
Regards,
Oleksij

-------------- next part --------------
A non-text attachment was scrubbed...
Name: signature.asc
Type: application/pgp-signature
Size: 195 bytes
Desc: OpenPGP digital signature
URL: <http://lists.infradead.org/pipermail/barebox/attachments/20170822/43600253/attachment.sig>