/* $Id$ */
/* Copyright (c) 2007-2015 Pierre Pronchery <khorben@defora.org> */
/* This file is part of DeforaOS Unix others */
/* 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 3 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, see <http://www.gnu.org/licenses/>. */



#include <sys/statvfs.h>
#ifdef ST_WAIT /* NetBSD */
# include <sys/param.h>
#endif
#include <sys/mount.h>
#include <ctype.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#ifndef PROGNAME
# define PROGNAME "umount"
#endif


/* umount */
/* private */
/* types */
typedef int Prefs;
#define PREFS_a 0x1
#define PREFS_f 0x2


/* prototypes */
static int _umount(Prefs * prefs, int pathc, char * pathv[]);

static int _umount_error(char const * message, int ret);
static int _umount_usage(void);


/* functions */
/* umount */
static int _umount_all(Prefs * prefs);
static int _umount_do(Prefs * prefs, char const * pathname);

static int _umount(Prefs * prefs, int pathc, char * pathv[])
{
	int ret = 0;
	int i;

	if(*prefs & PREFS_a && pathc == 0)
		return _umount_all(prefs);
	for(i = 0; i < pathc; i++)
		ret |= _umount_do(prefs, pathv[i]);
	return ret;
}

#ifdef ST_WAIT
static int _umount_all(Prefs * prefs)
{
	int ret = 0;
	int cnt;
	struct statvfs * f;
	int i;

	if((cnt = getvfsstat(NULL, 0, ST_WAIT)) < 0)
		return _umount_error("getvfsstat", 1);
	if((f = malloc(sizeof(*f) * cnt)) == NULL)
		return _umount_error("malloc", 1);
	if((cnt = getvfsstat(f, sizeof(*f) * cnt, ST_WAIT)) < 0)
	{
		free(f);
		return _umount_error("getvfsstat", 1);
	}
	for(i = cnt - 1; i >= 0; i--)
		if(strcmp("/", f[i].f_mntonname) == 0)
			continue;
		else
			ret |= _umount_do(prefs, f[i].f_mntonname);
	free(f);
	return ret;
#else
# include <errno.h>
static int _umount_all(Prefs * prefs)
{
	int ret = 0;
	char const path[] = "/proc/mounts";
	FILE * fp;
	unsigned char buf[1024];
	size_t i;
	size_t j;
	int c;

	if((fp = fopen(path, "r")) == NULL)
		return -_umount_error(path, 1);
	while(fgets(buf, sizeof(buf), fp) != NULL)
	{
		/* skip device */
		for(i = 0; buf[i] != '\0' && !isspace((c = buf[i])); i++);
		for(; isspace((c = buf[i])); i++);
		/* determine mountpoint */
		for(j = i; buf[j] != '\0' && !isspace((c = buf[j])); j++);
		if(j > i && buf[j] != '\0' && strncmp(&buf[i], "/", j - i) != 0)
		{
			buf[j++] = '\0';
			ret |= _umount_do(prefs, &buf[i]);
		}
		for(; buf[j] != '\0' && buf[j] != '\n'; j++);
		if(buf[j] == '\n')
			continue;
		/* flush longer lines */
		for(c = fgetc(fp); c != EOF && c != '\n'; c = fgetc(fp));
	}
	if(ferror(fp))
		ret |= -_umount_error(path, 1);
	fclose(fp);
	return ret;
#endif
}

static int _umount_do(Prefs * prefs, char const * pathname)
{
	int flags = 0;

#ifdef MNT_FORCE
	if(*prefs & PREFS_f)
		flags |= MNT_FORCE;
#endif
#ifdef MS_RDONLY /* Linux */
	if(umount(pathname) == 0)
#else
	if(unmount(pathname, flags) == 0)
#endif
		return 0;
	return _umount_error(pathname, 1);
}


/* umount_error */
static int _umount_error(char const * message, int ret)
{
	fputs(PROGNAME ": ", stderr);
	perror(message);
	return ret;
}


/* umount_usage */
static int _umount_usage(void)
{
	fputs("Usage: " PROGNAME " -a [-f]\n"
"       " PROGNAME " [-f] special | node ...\n", stderr);
	return 1;
}


/* public */
/* functions */
/* main */
int main(int argc, char * argv[])
{
	Prefs prefs;
	int o;

	prefs = 0;
	while((o = getopt(argc, argv, "af")) != -1)
		switch(o)
		{
			case 'a':
				prefs |= PREFS_a;
				break;
			case 'f':
				prefs |= PREFS_f;
				break;
			default:
				return _umount_usage();
		}
	if(optind == argc && (prefs & PREFS_a) != PREFS_a)
		return _umount_usage();
	return (_umount(&prefs, argc - optind, &argv[optind]) == 0) ? 0 : 2;
}