/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright (c) 1997-1999 by Sun Microsystems, Inc.
 * All rights reserved.
 */

#pragma ident	"@(#)io.c	1.3	05/06/08 SMI"


#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/param.h>
#include "parser.h"
#include "trace.h"
#include "db.h"
#include "util.h"
#include "errlog.h"

/* Types and Globals */
FILE	*Bodyfp = NULL;
FILE	*Headfp = NULL;
FILE	*Mapfp = NULL;

static char headfile_name[MAXLINE]; /* Saved for later. */
static char mapfile_name[MAXLINE]; /* Saved for later. */

/* File globals. */
static int alt_code_file(void);
static void abort_code_file(void);

/*
 * open_code_file - open the code file and the invisible temp file.
 */
int
open_code_file(void)
{
	char	*dir = db_get_target_directory();
	char	*body_file_name;
	int	rc = YES;

	errlog(BEGIN, "open_code_file() {");

	/* Open the Head file, which gets the headers, includes and */
	/* definitions, and eventually gets the body concatenated to it. */
	(void) snprintf(headfile_name, sizeof (headfile_name), "%s.c",
		db_get_output_file());
	if ((Headfp = fopen(headfile_name, "w")) == NULL) {
		errlog(FATAL, "%s: %s", headfile_name, strerror(errno));
	}

	(void) snprintf(mapfile_name, sizeof (mapfile_name), "%s-vers",
	    db_get_output_file());

	if ((Mapfp = fopen(mapfile_name, "w")) == NULL) {
		errlog(FATAL, "%s: %s", mapfile_name, strerror(errno));
	}
	(void) fputs("SUNWabi_1.1 {\n    global:\n", Mapfp);

	/* Now the Body file, which is an ephemeral temp-file. */
	if ((body_file_name = tempnam(dir, NULL)) == NULL) {
		errlog(FATAL, "out of memory creating a temp-file name");
	}

	if ((Bodyfp = fopen(body_file_name, "w+")) == NULL) {
		errlog(FATAL, "%s: %s", body_file_name, strerror(errno));
	}

	if (unlink(body_file_name) != 0) {
		errlog(FATAL, "unlink %s: %s", body_file_name, strerror(errno));
	}

	(void) free(body_file_name);
	errlog(END, "}");
	return (rc);
}

/*
 * abort_code_file -- close and discard files.
 * this function is also called from alt_code_file, so
 * it is not cool to unlink the code file or the mapfile
 */
static void
abort_code_file(void)
{
	errlog(BEGIN, "abort_code_file() {");
	(void) fclose(Bodyfp);
	(void) fclose(Headfp);
	if (unlink(headfile_name) != 0) {
		errlog(FATAL, "unlink %s: %s", headfile_name, strerror(errno));
	}
	errlog(END, "}");
}

int
alt_code_file(void)
{
	char hfn[MAXLINE];
	FILE *hfp;

	abort_code_file();
	(void) snprintf(hfn, sizeof (hfn), "%s.c", db_get_output_file());
	if ((hfp = fopen(hfn, "w")) == NULL) {
		errlog(FATAL, "%s: %s", headfile_name, strerror(errno));
	}

	(void) fputs("static int __abi_place_holder;\n", hfp);
	(void) fclose(hfp);

	return (YES);
}

/*
 * commit_code_file -- close and commit files that have advanced
 *	beyond byte position 0.
 */
int
commit_code_file(void)
{
	char	copy_buffer[BUFSIZ*8];
	size_t	n;

	errlog(BEGIN, "commit_code_file() {");
	/*
	 * We unconditionally want a .pf and a -vers file
	 */
	(void) fputs("    local:\n\t*;\n};\n", Mapfp);
	if (fclose(Mapfp) != 0) {
		errlog(FATAL, "fclose %s: %s", mapfile_name, strerror(errno));
	}
	if (ftell(Bodyfp) == 0) {
		/*
		 * special case, redo C file with place holder
		 * so that makefiles won't break...
		 */
		errlog(END, "}");
		return (alt_code_file());
	} else {
		/* Concatenate body file to head file, close both. */
		rewind(Bodyfp);
		while ((n = fread(copy_buffer, 1,
		    sizeof (copy_buffer), Bodyfp)) != 0) {
			(void) fwrite(copy_buffer, 1, n, Headfp);
		}
		(void) fclose(Bodyfp);
		if (fclose(Headfp) != 0) {
			errlog(FATAL, "fclose <temp file>: %s",
			    strerror(errno));
		}
	}

	errlog(END, "}");
	return (YES);
}