/*
 * 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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <libelf.h>
#include <sys/param.h>
#include <link.h>

#pragma init(ld_support_init)

#define	SUNPRO_DEPENDENCIES	"SUNPRO_DEPENDENCIES"

/*
 * Linked list of strings - used to keep lists of names
 * of directories or files.
 */
struct Stritem {
	char	*str;
	void	*next;
};

typedef struct Stritem 	Stritem;

static char 	*depend_file = NULL;
static Stritem	*list = NULL;

void
ld_support_init()
{
	depend_file = getenv(SUNPRO_DEPENDENCIES);
}

static void
prepend_str(Stritem **list, const char *str)
{
	Stritem		*new;
	char		*newstr;
	const char	*lib = "libldmake.so";

	if (!(new = calloc(1, sizeof (Stritem)))) {
		perror(lib);
		return;
	}

	if (!(newstr = malloc(strlen(str) + 1))) {
		perror(lib);
		return;
	}

	new->str = strcpy(newstr, str);
	new->next = *list;
	*list = new;

}

/* ARGSUSED */
void
ld_file(const char *file, const Elf_Kind ekind, int flags, Elf *elf)
{
	/*
	 * SUNPRO_DEPENDENCIES wasn't set, we don't collect .make.state
	 * information.
	 */
	if (!depend_file)
		return;

	if ((flags & LD_SUP_DERIVED) && !(flags & LD_SUP_EXTRACTED))
		prepend_str(&list, file);
}

void
ld_file64(const char *file, const Elf_Kind ekind, int flags, Elf *elf)
{
	ld_file(file, ekind, flags, elf);
}

void
ld_atexit(int exit_code)
{
	Stritem 	*cur;
	char		lockfile[MAXPATHLEN], *err, *space, *target;
	FILE		*ofp;
	extern char 	*file_lock(char *, char *, int);

	if (!depend_file || exit_code)
		return;

	if ((space = strchr(depend_file, ' ')) == NULL)
		return;
	*space = '\0';
	target = &space[1];

	(void) snprintf(lockfile, MAXPATHLEN, "%s.lock", depend_file);
	if ((err = file_lock(depend_file, lockfile, 0))) {
		(void) fprintf(stderr, "%s\n", err);
		return;
	}

	if (!(ofp = fopen(depend_file, "a")))
		return;

	if (list)
		(void) fprintf(ofp, "%s: ", target);

	for (cur = list; cur; cur = cur->next)
		(void) fprintf(ofp, " %s", cur->str);

	(void) fputc('\n', ofp);

	(void) fclose(ofp);
	(void) unlink(lockfile);
	*space = ' ';
}

void
ld_atexit64(int exit_code)
{
	ld_atexit(exit_code);
}