/*
 * UnDUPE.C - Clean the dupefile of unwanted data Works as far as i know
 * evilution @ efnet 02-25-99
 * 
 * Fixed USAGE spelling error - Turranius 2004-01-04
 *
 * Modded/stripped for use with pzs-ng - psxc 2004-07-14
 * Fixed issues on 64bit - DuReX 2007-12-17
 * Compatibility for 64bit glftpds - Sked 2011-09-16
 * Try renaming prior to streaming, some cleanups - Sked 2011-09-20
 *
 */

#include <stdio.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <errno.h>
#include "../conf/zsconfig.h"
#include "../include/zsconfig.defaults.h"

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

#ifndef HAVE_STRLCPY
# include "strl/strl.h"
#endif

/* Force structure alignment to 4 bytes (for 64bit support). */
#if ( GLVERSION != 20164 )
#pragma pack(push, 4)
#endif

/* 32-bit time values (for 64bit support). */
typedef int32_t time32_t;

#if ( GLVERSION == 20164 )
struct dupeentry {
	char		filename[256];
	time_t		timeup;
	char		uploader[25];
};
#else
struct dupeentry {
	char		filename[256];
	time32_t	timeup;
	char		uploader[25];
};
#endif

/* Restore default structure alignment for non-critical structures. */
#if ( GLVERSION != 20164 )
#pragma pack(pop)
#endif

int 
main(int argc, char *argv[])
{
	FILE		*fp, *fp2;
	char		dupename[1024], data2[1024], dupefile[1024];
	struct dupeentry buffer;

	if (argc != 2) {
		printf("Please give a filename to undupe as well\n");
		return 1;
	}

	strlcpy(dupefile, dupepath, 1024);
	strlcpy(dupename, argv[1], 1024);
	sprintf(data2, "%s/dupefile.%d", storage, (int)getuid());

	if (!(fp = fopen(dupefile, "r+b"))) {
		printf("FATAL ERROR: Unable to open dupefile (%s)\n", dupefile);
		return 1;
	}
	if (!(fp2 = fopen(data2, "w+b"))) {
		printf("FATAL ERROR: Unable to write to tempfile (%s)\n", data2);
		fclose(fp);
		return 1;
	}

	while (!feof(fp)) {
		if (fread(&buffer, sizeof(struct dupeentry), 1, fp) < 1)
			break;
		/* If we found the file, delete it */
		if (strcmp(buffer.filename, dupename) == 0)
			fflush(fp);
		/* if not, write it to the new file, if we can */
		else if (fwrite(&buffer, sizeof(struct dupeentry), 1, fp2) < 1)
			break;
	}

	fclose(fp);
	fclose(fp2);

	/*
	 * Time to put back the remainder of the dupefile. We try renaming the
	 * tempfile first, if this fails we try streaming the contents of the
	 * tempfile to the original file. Streaming will work where renaming
	 * fails if the directory is not writable for the user.
	 */

	if (rename(data2, dupefile) == -1) {
		if (!(fp = fopen(data2, "r+b"))) {
			printf("FATAL ERROR: Unable to open tempfile (%s)\n", data2);
			return 1;
		}
		if (!(fp2 = fopen(dupefile, "w+b"))) {
			printf("FATAL ERROR: Unable to write to dupefile (%s)\n", dupefile);
			fclose(fp);
			return 1;
		}

		while (!feof(fp)) {
			if (fread(&buffer, sizeof(struct dupeentry), 1, fp) < 1)
				break;
			if (fwrite(&buffer, sizeof(struct dupeentry), 1, fp2) < 1)
				break;
		}

		fclose(fp);
		fclose(fp2);

		if (chmod(dupefile, 0666))
			printf("WARNING: Failed to chmod %s: %s\n", dupefile, strerror(errno));

		if (unlink(data2) > 0) {
			printf("FATAL ERROR: Unable to delete tempfile (%s)\n", data2);
			return 1;
		}
	}

	return 0;
}