/*      $OpenBSD: ex_preserve.c,v 1.7 2016/01/06 22:28:52 millert Exp $ */

/*-
 * Copyright (c) 1993, 1994
 *      The Regents of the University of California.  All rights reserved.
 * Copyright (c) 1993, 1994, 1995, 1996
 *      Keith Bostic.  All rights reserved.
 *
 * See the LICENSE file for redistribution information.
 */

#include "config.h"

#include <sys/types.h>
#include <sys/queue.h>

#include <bitstring.h>
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <string.h>

#include "../common/common.h"

/*
 * ex_preserve -- :pre[serve]
 *      Push the file to recovery.
 *
 * PUBLIC: int ex_preserve(SCR *, EXCMD *);
 */
int
ex_preserve(SCR *sp, EXCMD *cmdp)
{
        recno_t lno;

        NEEDFILE(sp, cmdp);

        if (!F_ISSET(sp->ep, F_RCV_ON)) {
                msgq(sp, M_ERR, "Preservation of this file not possible");
                return (1);
        }

        /* If recovery not initialized, do so. */
        if (F_ISSET(sp->ep, F_FIRSTMODIFY) && rcv_init(sp))
                return (1);

        /* Force the file to be read in, in case it hasn't yet. */
        if (db_last(sp, &lno))
                return (1);

        /* Sync to disk. */
        if (rcv_sync(sp, RCV_SNAPSHOT))
                return (1);

        msgq(sp, M_INFO, "File preserved");
        return (0);
}

/*
 * ex_recover -- :rec[over][!] file
 *      Recover the file.
 *
 * PUBLIC: int ex_recover(SCR *, EXCMD *);
 */
int
ex_recover(SCR *sp, EXCMD *cmdp)
{
        ARGS *ap;
        FREF *frp;

        ap = cmdp->argv[0];

        /* Set the alternate file name. */
        set_alt_name(sp, ap->bp);

        /*
         * Check for modifications.  Autowrite did not historically
         * affect :recover.
         */
        if (file_m2(sp, FL_ISSET(cmdp->iflags, E_C_FORCE)))
                return (1);

        /* Get a file structure for the file. */
        if ((frp = file_add(sp, ap->bp)) == NULL)
                return (1);

        /* Set the recover bit. */
        F_SET(frp, FR_RECOVER);

        /* Switch files. */
        if (file_init(sp, frp, NULL, FS_SETALT |
            (FL_ISSET(cmdp->iflags, E_C_FORCE) ? FS_FORCE : 0)))
                return (1);

        F_SET(sp, SC_FSWITCH);
        return (0);
}