/*
 *  Copyright (C) 2009 Sony Computer Entertainment Inc.
 *  Copyright 2009 Sony Corp.
 *
 *  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; version 2 of the License.
 *
 *  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#if defined(HAVE_CONFIG_H)
#include "config.h"
#endif

#include <assert.h>
#include <string.h>

#include "log/log.h"
#include "talloc/talloc.h"
#include "url.h"

/**
 * pb_scheme_info - Helper for parsing URLs.
 */

struct pb_scheme_info {
	enum pb_url_scheme scheme;
	const char *str;
	unsigned int str_len;
	bool has_host;
};

static const struct pb_scheme_info schemes[] = {
	{
		.scheme = pb_url_file,
		.str = "file",
		.str_len = sizeof("file") - 1,
		.has_host = false,
	},
	{
		.scheme = pb_url_ftp,
		.str = "ftp",
		.str_len = sizeof("ftp") - 1,
		.has_host = true,
	},
	{
		.scheme = pb_url_http,
		.str = "http",
		.str_len = sizeof("http") - 1,
		.has_host = true,
	},
	{
		.scheme = pb_url_https,
		.str = "https",
		.str_len = sizeof("https") - 1,
		.has_host = true,
	},
	{
		.scheme = pb_url_nfs,
		.str = "nfs",
		.str_len = sizeof("nfs") - 1,
		.has_host = true,
	},
	{
		.scheme = pb_url_sftp,
		.str = "sftp",
		.str_len = sizeof("sftp") - 1,
		.has_host = true,
	},
	{
		.scheme = pb_url_tftp,
		.str = "tftp",
		.str_len = sizeof("tftp") - 1,
		.has_host = true,
	},
};

static const struct pb_scheme_info *file_scheme = &schemes[0];

/**
 * pb_url_find_scheme - Find the pb_scheme_info for a URL string.
 */

static const struct pb_scheme_info *pb_url_scheme_info(
		enum pb_url_scheme scheme)
{
	unsigned int i;

	for (i = 0; i < sizeof(schemes) / sizeof(schemes[0]); i++) {
		const struct pb_scheme_info *info = &schemes[i];

		if (info->scheme == scheme)
			return info;

	}
	return NULL;
}

static const struct pb_scheme_info *pb_url_find_scheme(const char *url)
{
	static const int sep_len = sizeof("://") - 1;
	static const char *sep = "://";
	unsigned int i, url_len;

	url_len = strlen(url);

	for (i = 0; i < sizeof(schemes) / sizeof(schemes[0]); i++) {
		const struct pb_scheme_info *scheme = &schemes[i];

		if (url_len < scheme->str_len + sep_len)
			continue;

		if (strncmp(url + scheme->str_len, sep, sep_len))
			continue;

		if (strncasecmp(url, scheme->str, scheme->str_len))
			continue;

		return scheme;
	}

	return NULL;
}

static void pb_url_parse_path(struct pb_url *url)
{
	const char *p = strrchr(url->path, '/');

	talloc_free(url->dir);
	talloc_free(url->file);

	if (p) {
		p++;
		url->dir = talloc_strndup(url, url->path, p - url->path);
		url->file = talloc_strdup(url, p);
	} else {
		url->dir = NULL;
		url->file = talloc_strdup(url, url->path);
	}
}

/**
 * pb_url_parse - Parse a remote file URL.
 * @ctx: The talloc context to associate with the returned string.
 *
 * Returns a talloc'ed struct pb_url instance on success, or NULL on error.
 */

struct pb_url *pb_url_parse(void *ctx, const char *url_str)
{
	const struct pb_scheme_info *si;
	struct pb_url *url;
	const char *p;

	if (!url_str || !*url_str) {
		assert(0 && "bad url");
		return NULL;
	}

	url = talloc_zero(ctx, struct pb_url);

	if (!url)
		return NULL;

	si = pb_url_find_scheme(url_str);
	if (si) {
		url->scheme = si->scheme;
		p = url_str + si->str_len + strlen("://");
	} else {
		url->scheme = file_scheme->scheme;
		p = url_str;
	}

	url->full = talloc_strdup(url, url_str);

	if (url->scheme == pb_url_file) {
		url->port = NULL;
		url->host = NULL;
		url->path = talloc_strdup(url, p);
	} else {
		int len;
		const char *col;
		const char *path;

		path = strchr(p, '/');

		if (!path) {
			pb_log_fn("parse path failed '%s'\n" , p);
			goto fail;
		}

		col = strrchr(p, ':');
		if (col <= p)
			col = NULL;
		if (col && strchr(col , ']')) {
			/*
			 * This is likely an IPv6 address with no port.
			 * See https://www.ietf.org/rfc/rfc2732.txt
			 */
			col = NULL;
		}

		if (col) {
			len = path - col - 1;
			url->port = len ? talloc_strndup(url, col + 1, len)
				: NULL;
			len = col - p;
		} else {
			url->port = NULL;
			len = path - p;
		}

		if (p[0] == '[' && p[len - 1] == ']')
			/* IPv6 */
			url->host = talloc_strndup(url, p + 1, len - 2);
		else
			/* IPv4 */
			url->host = talloc_strndup(url, p, len);


		/* remove multiple leading slashes */
		for (; *path && *(path+1) == '/'; path++)
			;

		url->path = talloc_strdup(url, path);
	}

	pb_url_parse_path(url);

	return url;

fail:
	talloc_free(url);
	return NULL;
}

bool is_url(const char *str)
{
	return strstr(str, "://") != NULL;
}

int addr_scheme(const char *address)
{
	uint8_t buf[sizeof(struct in6_addr)];
	int rc;

	rc = inet_pton(AF_INET, address , buf);
	if (rc == 1)
		return AF_INET;
	rc = inet_pton(AF_INET6, address , buf);
	if (rc == 1)
		return AF_INET6;

	return 0;
}

char *pb_url_to_string(struct pb_url *url)
{
	const struct pb_scheme_info *scheme = pb_url_scheme_info(url->scheme);
	char *str, *port;
	assert(scheme);

	port = url->port ? talloc_asprintf(url, ":%s", url->port) : NULL;

	if (scheme->has_host && addr_scheme(url->host) == AF_INET6)
		str = talloc_asprintf(url, "%s://[%s]%s%s", scheme->str,
				url->host, port ?: "", url->path);
	else
		str = talloc_asprintf(url, "%s://%s%s%s", scheme->str,
				scheme->has_host ? url->host : "",
				port ?: "", url->path);

	talloc_free(port);
	return str;
}

static void pb_url_update_full(struct pb_url *url)
{
	talloc_free(url->full);
	url->full = pb_url_to_string(url);
}

struct pb_url *pb_url_copy(void *ctx, const struct pb_url *url)
{
	struct pb_url *new_url;

	new_url = talloc(ctx, struct pb_url);
	new_url->scheme = url->scheme;
	new_url->full = talloc_strdup(new_url, url->full);

	new_url->host = url->host ? talloc_strdup(new_url, url->host) : NULL;
	new_url->port = url->port ? talloc_strdup(new_url, url->port) : NULL;
	new_url->path = url->path ? talloc_strdup(new_url, url->path) : NULL;
	new_url->dir  = url->dir  ? talloc_strdup(new_url, url->dir)  : NULL;
	new_url->file = url->file ? talloc_strdup(new_url, url->file) : NULL;

	return new_url;
}

struct pb_url *pb_url_join(void *ctx, const struct pb_url *url, const char *s)
{
	struct pb_url *new_url;

	/* complete url: just parse all info from s */
	if (is_url(s))
		return pb_url_parse(ctx, s);

	new_url = pb_url_copy(ctx, url);

	if (s[0] == '/') {
		/* absolute path: replace path of new_url */
		talloc_free(new_url->path);
		new_url->path = talloc_strdup(new_url, s);

	} else {
		/* relative path: join s to existing path. We know that
		 * url->dir ends with a slash. */
		char *tmp = new_url->path;
		new_url->path = talloc_asprintf(new_url, "%s%s", url->dir, s);
		talloc_free(tmp);
	}

	/* replace ->dir and ->file with components from ->path */
	pb_url_parse_path(new_url);

	/* and re-generate the full URL */
	pb_url_update_full(new_url);

	return new_url;
}

const char *pb_url_scheme_name(enum pb_url_scheme scheme)
{
	const struct pb_scheme_info *info = pb_url_scheme_info(scheme);
	return info ? info->str : NULL;
}