/*
 * Copyright (c) 2017-2018 The DragonFly Project.  All rights reserved.
 *
 * This code is derived from software contributed to The DragonFly Project
 * by Matthew Dillon <dillon@dragonflybsd.org>
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``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
 * COPYRIGHT HOLDERS OR CONTRIBUTORS 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 "hammer2.h"

static int docleanup(const char *path);
static int sameh2prefix(const char *from);
static void saveh2prefix(const char *from);
static void freeh2prefixes(void);

int
cmd_cleanup(const char *sel_path)
{
	struct statfs *fsary;
	char *fstype;
	char *path;
	char *from;
	int i;
	int n;
	int r;
	int rc;

	if (sel_path)
		return (docleanup(sel_path));

	n = getmntinfo(&fsary, MNT_NOWAIT);
	if (n <= 0) {
		fprintf(stderr, "hammer2 cleanup: no HAMMER2 mounts\n");
		return 0;
	}
	rc = 0;
	for (i = 0; i < n; ++i) {
		fstype = fsary[i].f_fstypename;
		path = fsary[i].f_mntonname;
		from = fsary[i].f_mntfromname;

		if (strcmp(fstype, "hammer2") != 0)
			continue;
		if (sameh2prefix(from)) {
			printf("hammer2 cleanup \"%s\" (same partition)\n",
				path);
		} else {
			r = docleanup(path);
			if (r)
				rc = r;
			saveh2prefix(from);
		}
	}
	freeh2prefixes();

	return rc;
}

static
int
docleanup(const char *path)
{
	int rc;

	printf("hammer2 cleanup \"%s\"\n", path);
	rc = cmd_bulkfree(path);

	return rc;
}

static char **h2prefixes;
static int h2count;
static int h2limit;

static
int
sameh2prefix(const char *from)
{
	char *ptr;
	int rc = 0;
	int i;

	ptr = strdup(from);
	if (strchr(ptr, '@'))
		*strchr(ptr, '@') = 0;
	for (i = 0; i < h2count; ++i) {
		if (strcmp(h2prefixes[i], ptr) == 0) {
			rc = 1;
			break;
		}
	}
	free(ptr);

	return rc;
}

static
void
saveh2prefix(const char *from)
{
	char *ptr;

	if (h2count >= h2limit) {
		h2limit = (h2limit + 8) << 1;
		h2prefixes = realloc(h2prefixes, sizeof(char *) * h2limit);
	}
	ptr = strdup(from);
	if (strchr(ptr, '@'))
		*strchr(ptr, '@') = 0;
	h2prefixes[h2count++] = ptr;
}

static
void
freeh2prefixes(void)
{
	int i;

	for (i = 0; i < h2count; ++i)
		free(h2prefixes[i]);
	free(h2prefixes);
	h2prefixes = NULL;
	h2count = 0;
	h2limit = 0;
}