/*************************************************************************
* (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
*  by Michael Pounov <misho@openbsd-bg.org>
*
* $Author: misho $
* $Id: dir.c,v 1.1.2.3 2011/03/16 17:17:44 misho Exp $
*
*************************************************************************/
#include "global.h"


static int
func_comp(struct tagDirName const *d1, struct tagDirName const *d2)
{
	return d1->tag - d2->tag;
}

static struct tagDirName *
find_tag(int const * __restrict tags, struct tagDirName const * __restrict l, u_short t, u_int hash)
{
	struct tagDirName *find = NULL;
	register int i;

	// search in index tags
	if (tags[t] != -1 && l[tags[t]].tag == t) {
		// search in sorted hashes
		for (i = 0; l[tags[t] + i].tag == t; i++)
			if (l[tags[t] + i].hash == hash) {
				// finded & marked for delete!
				find = (struct tagDirName*) &l[tags[t] + i];
				find->ch = '*';
				break;
			}
	}

	return find;
}

static int *
create_tags()
{
	int *tags;

	tags = calloc(TABLESIZ, sizeof(int));
	if (!tags) {
		SETERR;
	} else
		memset(tags, -1, TABLESIZ * sizeof(int));

	return tags;
}

static int
create_diridx(const char *csDir, int lm, int *tags, struct tagDirName **list)
{
	struct tagDirName *l = *list;
	DIR *dir;
	struct dirent d, *pd;
	int n;
	char szStr[STRSIZ], szType[STRSIZ], *old;
	struct stat sb;
	register int i;

	l = malloc(sizeof(struct tagDirName));
	if (!l) {
		SETERR;
		*list = NULL;
		return -1;
	} else {
		n = 0;
		memset(l, 0, sizeof(struct tagDirName));
	}

	old = getcwd(NULL, 0);
	if (chdir(csDir) == -1) {
		SETERR;
		free(old);
		free(l);
		*list = NULL;
		return -1;
	}
	dir = opendir(".");
	if (!dir) {
		SETERR;
		chdir(old);
		free(old);
		free(l);
		*list = NULL;
		return -1;
	}
	while (!readdir_r(dir, &d, &pd) && pd) {
		if (d.d_type == DT_DIR && (!strcmp(d.d_name, ".") || !strcmp(d.d_name, "..")))
			continue;

		l = realloc(l, sizeof(struct tagDirName) * (n + 2));
		if (!l) {
			SETERR;
			chdir(old);
			free(old);
			free(l);
			*list = NULL;
			closedir(dir);
			return -1;
		} else
			memset(&l[n + 1], 0, sizeof(struct tagDirName));

		l[n].ch = '<';
		l[n].tag = crcFletcher16((u_short*) d.d_name, d.d_namlen / 2 + d.d_namlen % 2);
		l[n].hash = crcAdler((u_char*) d.d_name, d.d_namlen);
		strlcpy(l[n].name, d.d_name, MAXPATHLEN);
		if (lm & 1) {
			if (lstat(d.d_name, &sb) != -1) {
				memset(szStr, 0, STRSIZ);
				strftime(szStr, STRSIZ, "%Y-%m-%d %H:%M:%S", localtime((time_t*) &sb.st_mtim));
				switch (d.d_type) {
					case DT_FIFO:
						strlcpy(szType, "fifo", STRSIZ);
						break;
					case DT_CHR:
						strlcpy(szType, "char", STRSIZ);
						break;
					case DT_DIR:
						strlcpy(szType, "dir", STRSIZ);
						break;
					case DT_BLK:
						strlcpy(szType, "block", STRSIZ);
						break;
					case DT_REG:
						strlcpy(szType, "file", STRSIZ);
						break;
					case DT_LNK:
						strlcpy(szType, "link", STRSIZ);
						break;
					case DT_SOCK:
						strlcpy(szType, "socket", STRSIZ);
						break;
/* OpenBSD does not have this type */
#ifdef DT_WHT
					case DT_WHT:
						strlcpy(szType, "wht", STRSIZ);
						break;
#endif
					case DT_UNKNOWN:
					default:
						strlcpy(szType, "unknown", STRSIZ);
						break;
				}
				snprintf(l[n].extra, STRSIZ, "%s links=%d inode=%u %d:%d perm=0%o size=%llu %s", 
						szType, sb.st_nlink, sb.st_ino, sb.st_uid, sb.st_gid, 
						sb.st_mode & 0x1fff, sb.st_size, szStr);
			}
		}

		n++;
	}
	closedir(dir);

	qsort(l, n, sizeof(struct tagDirName), (int (*)(const void*, const void*)) func_comp);
	for (i = n - 1; i > -1; i--)
		tags[l[i].tag] = i;

	chdir(old);
	free(old);

	*list = l;
	return n;
}

// ------------------------------------------------------

/*
 * sync_dirCSum() Calculate checksum of directory
 * @csDir = Directory
 * @md = Message digest allocated memory, must be free after use!
 * return: -1 error or !=-1 ok
 */
int
sync_dirCSum(const char *csDir, u_char **md)
{
	DIR *dir;
	struct dirent d, *pd;
	MD5_CTX ctx;
	register int ret = 0;

	*md = malloc(MD5_DIGEST_LENGTH);
	if (!*md) {
		SETERR;
		return -1;
	} else
		memset(*md, 0, MD5_DIGEST_LENGTH);

	dir = opendir(csDir);
	if (!dir) {
		SETERR;
		free(*md);
		return -1;
	}

	MD5_Init(&ctx);
	while (!readdir_r(dir, &d, &pd) && pd) {
		if (d.d_type == DT_DIR && (!strcmp(d.d_name, ".") || !strcmp(d.d_name, "..")))
			continue;
		MD5_Update(&ctx, d.d_name, d.d_namlen);
		ret++;
	}
	MD5_Final(*md, &ctx);

	closedir(dir);
	return ret;
}

/*
 * sync_dircmp() Compare directories
 * @csDir1 = Directory 1
 * @csDir2 = Directory 2
 * return: -1 error, 0 is equal or 1 different
 */
int
sync_dircmp(const char *csDir1, const char *csDir2)
{
	u_char *md[2] = { NULL, NULL };
	int ret = -1;

	if (!csDir1 || !csDir2)
		return ret;

	if (sync_dirCSum(csDir1, &md[0]) == -1)
		return ret;
	if (sync_dirCSum(csDir2, &md[1]) == -1) {
		free(md[0]);
		return ret;
	}

	if (!memcmp(md[0], md[1], MD5_DIGEST_LENGTH))
		ret = 0;
	else
		ret = 1;

	free(md[1]);
	free(md[0]);
	return ret;
}

/*
 * sync_dircmpList() Compare directories or directory and file list
 * @csDir1 = Directory 1
 * @csDir2 = Directory 2 or File list, if "-" get input from console
 * @lm = Long mode options, 1 long output
 * @list = Output diff list, after use must be free!
 * return: -1 error, 0 is equal or >0 count of returned list items
 */
int
sync_dircmpList(const char *csDir1, const char *csDir2, int lm, struct tagDirName **list)
{
	struct tagDirName *l, *find;
	int n, cx;
	DIR *dir;
	FILE *f = stdin;
	struct dirent d, *pd;
	int *tags;
	register int i;
	u_short t;
	u_int hash;
	struct stat sb;
	char szLine[STRSIZ], szStr[STRSIZ], szType[STRSIZ], *str, *pbrk, *old;

	if (!csDir1 || !list || !(tags = create_tags()))
		return -1;

	n = create_diridx(csDir1, lm, tags, &l);
	if (n == -1 || !csDir2) {
		*list = l;
		return n;
	}

	if (lstat(csDir2, &sb) == -1) {
		SETERR;
		free(l);
		return -1;
	}
	if (S_ISDIR(sb.st_mode)) {
		old = getcwd(NULL, 0);
		if (chdir(csDir2) == -1) {
			SETERR;
			chdir(old);
			free(old);
			free(l);
			return -1;
		}
		dir = opendir(".");
		if (!dir) {
			SETERR;
			chdir(old);
			free(old);
			free(l);
			return -1;
		}
		while (!readdir_r(dir, &d, &pd) && pd) {
			if (d.d_type == DT_DIR && (!strcmp(d.d_name, ".") || !strcmp(d.d_name, "..")))
				continue;
			else {
				t = crcFletcher16((u_short*) d.d_name, d.d_namlen / 2 + d.d_namlen % 2);
				hash = crcAdler((u_char*) d.d_name, d.d_namlen);
			}

			find = find_tag(tags, l, t, hash);
			// element not find in dir1, added
			if (!find) {
				l = realloc(l, sizeof(struct tagDirName) * (n + 2));
				if (!l) {
					SETERR;
					chdir(old);
					free(old);
					closedir(dir);
					return -1;
				} else
					memset(&l[n + 1], 0, sizeof(struct tagDirName));

				l[n].ch = '>';
				l[n].tag = t;
				l[n].hash = hash;
				strlcpy(l[n].name, d.d_name, MAXPATHLEN);
				if (lm & 1) {
					if (lstat(d.d_name, &sb) != -1) {
						memset(szStr, 0, STRSIZ);
						strftime(szStr, STRSIZ, "%Y-%m-%d %H:%M:%S", 
								localtime((time_t*) &sb.st_mtim));
						switch (d.d_type) {
							case DT_FIFO:
								strlcpy(szType, "fifo", STRSIZ);
								break;
							case DT_CHR:
								strlcpy(szType, "char", STRSIZ);
								break;
							case DT_DIR:
								strlcpy(szType, "dir", STRSIZ);
								break;
							case DT_BLK:
								strlcpy(szType, "block", STRSIZ);
								break;
							case DT_REG:
								strlcpy(szType, "file", STRSIZ);
								break;
							case DT_LNK:
								strlcpy(szType, "link", STRSIZ);
								break;
							case DT_SOCK:
								strlcpy(szType, "socket", STRSIZ);
								break;
/* OpenBSD does not have this type */
#ifdef DT_WHT
							case DT_WHT:
								strlcpy(szType, "wht", STRSIZ);
								break;
#endif
							case DT_UNKNOWN:
							default:
								strlcpy(szType, "unknown", STRSIZ);
								break;
						}
						snprintf(l[n].extra, STRSIZ, 
								"%s links=%d inode=%u %d:%d perm=0%o size=%llu %s", 
								szType, sb.st_nlink, sb.st_ino, sb.st_uid, sb.st_gid, 
								sb.st_mode & 0x1fff, sb.st_size, szStr);
					}
				}

				n++;
			}
		}
		closedir(dir);
		chdir(old);
		free(old);
	} else {
		if (strcmp(csDir2, "-")) {
			f = fopen(csDir2, "r");
			if (!f) {
				SETERR;
				free(l);
				return -1;
			}
		}
		while (fgets(szLine, STRSIZ, f)) {
			if (!*szLine || *szLine == '#')
				continue;

			str = strtok_r(szLine, " \t", &pbrk);
			if (!str)
				continue;
			str = strtok_r(NULL, " \t", &pbrk);
			if (!str)
				continue;
			else {
				i = strlen(str);
				t = crcFletcher16((u_short*) str, i / 2 + i % 2);
				hash = crcAdler((u_char*) str, i);
			}

			find = find_tag(tags, l, t, hash);
			// element not find in dir1, added
			if (!find) {
				l = realloc(l, sizeof(struct tagDirName) * (n + 2));
				if (!l) {
					SETERR;
					if (strcmp(csDir2, "-"))
						fclose(f);
					return -1;
				} else
					memset(&l[n + 1], 0, sizeof(struct tagDirName));

				l[n].ch = '>';
				l[n].tag = t;
				l[n].hash = hash;
				strlcpy(l[n].name, str, MAXPATHLEN);
				if (lm & 1 && (str = strtok_r(NULL, "\r\n", &pbrk)))
					strlcpy(l[n].extra, str, STRSIZ);

				n++;
			}
		}
		if (strcmp(csDir2, "-"))
			fclose(f);
	}

	// delete equal elemets !!!
	for (i = cx = 0; i < n; i++)
		if (l[i].ch == '*') {
			memmove(&l[i], &l[i + 1], (n - i + 1) * sizeof(struct tagDirName));
			cx++;
			i--;
		}
	n -= cx;

	*list = l;
	return n;
}