firebird-mirror/src/journal/archive.cpp

/*
 *	PROGRAM:	JRD Journal Server
 *	MODULE:		archive.c
 *	DESCRIPTION:
 *
 * The contents of this file are subject to the Interbase Public
 * License Version 1.0 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy
 * of the License at http://www.Inprise.com/IPL.html
 *
 * Software distributed under the License is distributed on an
 * "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, either express
 * or implied. See the License for the specific language governing
 * rights and limitations under the License.
 *
 * The Original Code was created by Inprise Corporation
 * and its predecessors. Portions created by Inprise Corporation are
 * Copyright (C) Inprise Corporation.
 *
 * All Rights Reserved.
 * Contributor(s): ______________________________________.
 *
 * 2002.10.29 Sean Leyne - Removed obsolete "Netware" port
 *
 */

#include "firebird.h"
#include "../jrd/ib_stdio.h"
#include <stdlib.h>
#include <string.h>
#include "../jrd/common.h"

#include "../wal/wal.h"
#include "../jrd/jrn.h"
#include "../jrd/llio.h"
#include "../jrd/jrn_proto.h"
#include "../jrd/llio_proto.h"

static bool copy_file(SLONG, SLONG, SLONG);
static void error_exit(ISC_STATUS *, JRN *, SLONG, SLONG, SLONG);
static bool open_file(TEXT *, SLONG, USHORT, SLONG *);


int CLIB_ROUTINE main(int argc,
					  char *argv[])
{
/**************************************
 *
 *	m a i n
 *
 **************************************
 *
 * Functional description
 *    backup a file to another disk, update journal database.
 *
 **************************************/
	TEXT *d_file;
	TEXT *s_file;
	TEXT *j_dir;
	SLONG file_id;
	SLONG s_fd, d_fd;
	SLONG p_offset;
	USHORT j_length;
	bool ret_val;
	SLONG size;
	SLONG db_id;
	ISC_STATUS_ARRAY status;
	JRN journal = NULL;

	argv++;

	db_id = atoi(*argv++);
	file_id = atoi(*argv++);
	p_offset = atoi(*argv++);
	size = atoi(*argv++);
	s_file = *argv++;
	d_file = *argv++;
	j_dir = *argv++;
	j_length = strlen(j_dir);

// Attach with journal server

	if (JRN_archive_begin(status, &journal, db_id, file_id, j_dir, j_length)
		!= FB_SUCCESS)
	{
		error_exit(status, &journal, db_id, file_id, 235);
		// msg 235: Archive process unable to attach to journal server.
	}

// Check in with database

	if (!open_file(s_file, p_offset, LLIO_OPEN_R, &s_fd))
		error_exit(status, &journal, db_id, file_id, 236);
		// msg 236: Archive process unable to open log file.

	if (!open_file(d_file, 0L, LLIO_OPEN_NEW_RW, &d_fd)) {
		LLIO_close(0, s_fd);
		error_exit(status, &journal, db_id, file_id, 237);
		// msg 237: Archive process unable to create archive file.
	}

	ret_val = copy_file(s_fd, d_fd, size);

	ret_val |= (LLIO_close(0, s_fd) == FB_SUCCESS);
	ret_val |= (LLIO_close(0, d_fd) == FB_SUCCESS);

	if (!ret_val)
		error_exit(status, &journal, db_id, file_id, 238);
		// msg 238: Archive process unable to close log and/or archive files.

// Now mark the file for deletion

	JRN_archive_end(status, &journal, db_id, file_id);

	exit(FINI_OK);
	// make compiler happy
	return 0;
}


static bool copy_file(SLONG s_fd,
						SLONG d_fd,
						SLONG size)
{
/**************************************
 *
 *	c o p y _ f i l e
 *
 **************************************
 *
 * Functional description
 *	copy source file to destination file
 *	Return true on success, false on failure
 *
 **************************************/
	UCHAR buff[1024];
	SLONG l, len, read, written;

	len = size;

	l = 1024;

	while (len) {
		if (len < 1024)
			l = len;

		len -= l;

		if (LLIO_read(0, s_fd, 0, 0L, LLIO_SEEK_NONE, buff, l, &read) ||
			l != read)
		{
			return false;
		}
		if (LLIO_write(0, d_fd, 0, 0L, LLIO_SEEK_NONE, buff, l, &written) ||
			l != written)
		{
			return false;
		}
	}

	return true;
}


static void error_exit(ISC_STATUS * status_vector,
					   JRN * ret_journal,
					   SLONG db_id,
					   SLONG file_id,
					   SLONG error_code)
{
/**************************************
 *
 *	e r r o r _ e x i t
 *
 **************************************
 *
 * Functional description
 *
 **************************************/

	JRN_archive_error(status_vector, ret_journal, db_id, file_id, error_code);
	exit(FINI_ERROR);
}


static bool open_file(TEXT * full_name,
					  SLONG p_offset,
					  USHORT mode,
					  SLONG * fd)
{
/**************************************
 *
 *	o p e n _ f i l e
 *
 **************************************
 *
 * Functional description
 *	open a file with a given mask
 *	Return true if success false if failure.
 *	The output file descriptor return in fd parameter.
 *
 **************************************/

	if (LLIO_open(0, full_name, mode, FALSE, fd))
		return false;

	if (p_offset) {
		if (LLIO_seek(0, *fd, 0, p_offset, LLIO_SEEK_BEGIN))
			return false;
	}

	return true;
}