/*	$NetBSD: wapbl.c,v 1.6 2022/07/25 05:15:08 dholland Exp $	*/

/*-
 * Copyright (c) 2005,2008 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Wasabi Systems, Inc.
 *
 * 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.
 */

/* This file contains fsck support for wapbl
 */
#define WAPBL_INTERNAL

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: wapbl.c,v 1.6 2022/07/25 05:15:08 dholland Exp $");

#include <sys/stat.h>
#include <sys/time.h>
#include <sys/uio.h>

#include <assert.h>
#include <errno.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <ufs/ufs/dinode.h>
#include <ufs/ufs/dir.h>
#include <ufs/ufs/ufs_bswap.h>
#include <ufs/ufs/ufs_wapbl.h>
#include <ufs/ffs/fs.h>
#include <ufs/ffs/ffs_extern.h>

#include <sys/wapbl.h>

#include "fsck.h"
#include "fsutil.h"
#include "extern.h"
#include "exitvalues.h"

int
wapbl_write(void *data, size_t len, struct vnode *devvp, daddr_t pbn)
{

	WAPBL_PRINTF(WAPBL_PRINT_IO,
		("wapbl_write: %zd bytes at block %"PRId64" on fd 0x%x\n",
		len, pbn, fswritefd));
	bwrite(fswritefd, data, pbn, len);
	return 0;
}

int
wapbl_read(void *data, size_t len, struct vnode *devvp, daddr_t pbn)
{

	WAPBL_PRINTF(WAPBL_PRINT_IO,
		("wapbl_read: %zd bytes at block %"PRId64" on fd 0x%x\n",
		len, pbn, fsreadfd));
	bread(fsreadfd, data, pbn, len);
	return 0;
}

struct wapbl_replay *wapbl_replay;

void
replay_wapbl(void)
{
	int error;

	if (nflag) {
		/*
		 * XXX: we ought to have a mode where we can replay
		 * the journal to memory, similar to what happens in
		 * the kernel with a readonly mount. For now though
		 * just print that we aren't doing it so as to avoid
		 * lying to the user.
		 */
		pwarn("CANNOT REPLAY JOURNAL IN -n MODE; continuing anyway\n");
	} else {
		error = wapbl_replay_write(wapbl_replay, 0);
		if (error) {
			pfatal("UNABLE TO REPLAY JOURNAL BLOCKS");
			if (reply("CONTINUE") == 0) {
				exit(FSCK_EXIT_CHECK_FAILED);
			}
		} else {
			wapbl_replay_stop(wapbl_replay);
		}
	}
	{
		int i;
		for (i = 0; i < wapbl_replay->wr_inodescnt; i++) {
			WAPBL_PRINTF(WAPBL_PRINT_REPLAY,("wapbl_replay: "
			    "not cleaning inode %"PRIu32" mode %"PRIo32"\n",
			    wapbl_replay->wr_inodes[i].wr_inumber,
			    wapbl_replay->wr_inodes[i].wr_imode));
		}
	}
}

void
cleanup_wapbl(void)
{

	if (wapbl_replay) {
		if (wapbl_replay_isopen(wapbl_replay))
			wapbl_replay_stop(wapbl_replay);
		wapbl_replay_free(wapbl_replay);
		wapbl_replay = 0;
	}
}

int
read_wapbl(char *buf, long size, daddr_t blk)
{

	if (!wapbl_replay || !wapbl_replay_isopen(wapbl_replay))
		return 0;
	return wapbl_replay_read(wapbl_replay, buf, blk, size);
}

int
is_journal_inode(ino_t ino)
{
	union dinode *dp;

	dp = ginode(ino);
	if ((iswap32(DIP(dp, flags)) & SF_LOG) != 0 &&
	    sblock->fs_journal_version == UFS_WAPBL_VERSION &&
	    sblock->fs_journal_location == UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM &&
	    sblock->fs_journallocs[UFS_WAPBL_INFS_INO] == ino)
		return 1;

	return 0;
}

int
check_wapbl(void)
{
	uint64_t addr = 0, count = 0, blksize = 0;
	int error;
	int ret = 0;
	if (debug)
		wapbl_debug_print = WAPBL_PRINT_ERROR | WAPBL_PRINT_REPLAY;
	if (debug > 1)
		wapbl_debug_print |= WAPBL_PRINT_IO;

	if (sblock->fs_flags & FS_DOWAPBL) {
		if (sblock->fs_journal_version != UFS_WAPBL_VERSION) {
			pfatal("INVALID JOURNAL VERSION %d",
			    sblock->fs_journal_version);
			if (reply("CONTINUE") == 0) {
				exit(FSCK_EXIT_CHECK_FAILED);
			}
			pwarn("CLEARING EXISTING JOURNAL\n");
			sblock->fs_flags &= ~FS_DOWAPBL;
			sblock->fs_journal_flags = UFS_WAPBL_FLAGS_CLEAR_LOG;
			sbdirty();
			ret = FSCK_EXIT_CHECK_FAILED;
		} else {
			switch(sblock->fs_journal_location) {
			case UFS_WAPBL_JOURNALLOC_END_PARTITION:
				addr =
				  sblock->fs_journallocs[UFS_WAPBL_EPART_ADDR];
				count =
				  sblock->fs_journallocs[UFS_WAPBL_EPART_COUNT];
				blksize =
				  sblock->fs_journallocs[UFS_WAPBL_EPART_BLKSZ];
				break;
			case UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM:
				addr =
				  sblock->fs_journallocs[UFS_WAPBL_INFS_ADDR];
				count =
				  sblock->fs_journallocs[UFS_WAPBL_INFS_COUNT];
				blksize =
				  sblock->fs_journallocs[UFS_WAPBL_INFS_BLKSZ];
				break;
			default:
				pfatal("INVALID JOURNAL LOCATION %d",
				    sblock->fs_journal_location);
				if (reply("CONTINUE") == 0) {
					exit(FSCK_EXIT_CHECK_FAILED);
				}
				pwarn("CLEARING EXISTING JOURNAL\n");
				sblock->fs_flags &= ~FS_DOWAPBL;
				sblock->fs_journal_flags = UFS_WAPBL_FLAGS_CLEAR_LOG;
				sblock->fs_journal_location =
				    UFS_WAPBL_JOURNALLOC_NONE;
				sbdirty();
				ret = FSCK_EXIT_CHECK_FAILED;
				break;
			}
			if (sblock->fs_flags & FS_DOWAPBL) {
				error = wapbl_replay_start(
				    &wapbl_replay, 0, addr, count, blksize);
				if (error) {
					pfatal(
					   "UNABLE TO READ JOURNAL FOR REPLAY");
					if (reply("CONTINUE") == 0) {
						exit(FSCK_EXIT_CHECK_FAILED);
					}
					pwarn("CLEARING EXISTING JOURNAL\n");
					sblock->fs_flags &= ~FS_DOWAPBL;
					sblock->fs_journal_flags = UFS_WAPBL_FLAGS_CLEAR_LOG;
					sbdirty();
					ret = FSCK_EXIT_CHECK_FAILED;
				}
			}
		}
	}
	/*
	 * at this time fs_journal_flags can only be 0,
	 * UFS_WAPBL_FLAGS_CREATE_LOG or UFS_WAPBL_FLAGS_CLEAR_LOG.
	 * We can't have both flags at the same time.
	 */
	switch (sblock->fs_journal_flags) {
	case 0:
		break;
	case UFS_WAPBL_FLAGS_CREATE_LOG:
	case UFS_WAPBL_FLAGS_CLEAR_LOG:
		switch(sblock->fs_journal_location) {
		case UFS_WAPBL_JOURNALLOC_END_PARTITION:
		case UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM:
			break;
		case UFS_WAPBL_JOURNALLOC_NONE:
			if (sblock->fs_journal_flags ==
			    UFS_WAPBL_FLAGS_CLEAR_LOG)
				break;
			/* FALLTHROUGH */
		default:
			pfatal("INVALID JOURNAL LOCATION %d",
			    sblock->fs_journal_location);
			if (reply("CONTINUE") == 0) {
				exit(FSCK_EXIT_CHECK_FAILED);
			}
			pwarn("CLEARING JOURNAL FLAGS\n");
			sblock->fs_journal_flags = 0;
			sblock->fs_journal_location =
			    UFS_WAPBL_JOURNALLOC_NONE;
			sbdirty();
			ret = FSCK_EXIT_CHECK_FAILED;
			break;
		}
		break;
	default:
		pfatal("INVALID JOURNAL FLAGS %d",
		    sblock->fs_journal_flags);
		if (reply("CONTINUE") == 0) {
			exit(FSCK_EXIT_CHECK_FAILED);
		}
		pwarn("CLEARING JOURNAL FLAGS\n");
		sblock->fs_journal_flags = 0;
		sblock->fs_journal_location =
		    UFS_WAPBL_JOURNALLOC_NONE;
		sbdirty();
		ret = FSCK_EXIT_CHECK_FAILED;
		break;
	}
	return ret;
}