xref: /src/sbin/fsck/progress.c

/*	$NetBSD: progress.c,v 1.5 2009/04/11 06:48:36 lukem Exp $	*/

/*-
 * Copyright (c) 1997-2004 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Luke Mewburn; by Chris Gilbert; and by Jason R. Thorpe.
 *
 * 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 NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
 */

#ifndef SMALL
#include <sys/cdefs.h>
__RCSID("$NetBSD: progress.c,v 1.5 2009/04/11 06:48:36 lukem Exp $");

/*
 * File system independent fsck progress bar routines.
 */

#include <sys/param.h>
#include <sys/tty.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "progress.h"

static size_t	ttywidth = 80;

static int	progress_onoff;
static int	progress_lowlim;
static int	progress_highlim;

#define	BUFLEFT		(sizeof(buf) - len)

void
progress_switch(int onoff)
{
	progress_onoff = onoff;
}

void
progress_init(void)
{
	progress_setrange(0, 100);
}

/* Set both low and high limit. */
void
progress_setrange(int lowlim, int highlim)
{
	progress_lowlim = lowlim;
	progress_highlim = highlim;
}

/* Previous high limit becomes new low limit; set new high limit. */
void
progress_sethighlim(int highlim)
{
	progress_setrange(progress_highlim, highlim);
}

/*
 * Display a progress bar, assuming that current/total represents a
 * percentage in the range [progress_lowlim .. progress_highlim].
 */
void
progress_bar(const char *dev, const char *label, off_t current, off_t total)
{
	static int lastpercentage = -1;
	char buf[256];
	int len, percentage;
	int barlength;
	int i;
	int lengthextras;

#define	BAROVERHEAD	10	/* non-* portion of progress bar */

	/*
	 * stars should contain at least sizeof(buf) - BAROVERHEAD
	 * entries.
	 */
	static const char stars[] =
"*****************************************************************************"
"*****************************************************************************"
"*****************************************************************************";

	if (progress_onoff == 0)
		return;

	len = 0;
	lengthextras = strlen(dev) + (label != NULL ? strlen(label) : 0);
	percentage = progress_lowlim +
		(current * (progress_highlim - progress_lowlim)) / total;
	percentage = MAX(percentage, 0);
	percentage = MIN(percentage, 100);

	if (percentage == lastpercentage)
		return;
	lastpercentage = percentage;

	len += snprintf(buf + len, BUFLEFT, "%s: ", dev);
	if (label != NULL)
		len += snprintf(buf + len, BUFLEFT, "%s ", label);

	barlength = MIN(sizeof(buf) - 1, ttywidth) - BAROVERHEAD - lengthextras;
	if (barlength > 0) {
		i = barlength * percentage / 100;
		len += snprintf(buf + len, BUFLEFT,
		    "|%.*s%*s| ", i, stars, barlength - i, "");
	}
	len += snprintf(buf + len, BUFLEFT, "%3d%%\r", percentage);
	write(fileno(stdout), buf, len);
}

void
progress_done(void)
{
	char buf[256];
	int len;

	if (progress_onoff == 0)
		return;

	len = MIN(sizeof(buf) - 2, ttywidth);
	memset(buf, ' ', len);
	buf[len] = '\r';
	buf[len + 1] = '\0';
	write(fileno(stdout), buf, len + 1);
}

void
progress_ttywidth(int a)
{
	struct winsize winsize;
	int oerrno = errno;

	if (ioctl(fileno(stdout), TIOCGWINSZ, &winsize) != -1 &&
	    winsize.ws_col != 0)
	    	ttywidth = winsize.ws_col;
	else
		ttywidth = 80;
	errno = oerrno;
}

#endif /* ! SMALL */