--- embedtools/src/direr.c 2010/07/12 15:20:41 1.1 +++ embedtools/src/direr.c 2010/07/12 15:20:41 1.1.2.1 @@ -0,0 +1,150 @@ +/************************************************************************* + * (C) 2010 AITNET - Sofia/Bulgaria - + * by Michael Pounov + * + * $Author: misho $ + * $Id: direr.c,v 1.1.2.1 2010/07/12 15:20:41 misho Exp $ + * + *************************************************************************/ +#include "global.h" + + +extern char compiled[], compiledby[], compilehost[]; + +static void +Usage() +{ + printf( "-= Direr =- Tool for compare directories by checksum\n" + "=== %s === %s@%s ===\n\n" + " Syntax: direr [options] [cmp_dir]\n\n" + "\t-s \t\tCompare dir with this checksum ...\n" + "\n", compiled, compiledby, compilehost); +} + + +int +calcDir(const char *csDir, u_char **md) +{ + DIR *dir; + struct dirent d, *pd; + MD5_CTX ctx; + + *md = malloc(MD5_DIGEST_LENGTH); + if (!*md) { + printf("Error:: %s(%d) #%d - %s\n", __func__, __LINE__, errno, strerror(errno)); + return -1; + } else + memset(*md, 0, MD5_DIGEST_LENGTH); + + dir = opendir(csDir); + if (!dir) { + printf("Error:: %s(%d) #%d - %s\n", __func__, __LINE__, errno, strerror(errno)); + 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); + } + MD5_Final(*md, &ctx); + + closedir(dir); + return 0; +} + +inline int +convMD5(u_char *md, char **str) +{ + register int i; + + *str = malloc(MD5_DIGEST_LENGTH * 2 + 1); + if (!*str) { + printf("Error:: %s(%d) #%d - %s\n", __func__, __LINE__, errno, strerror(errno)); + return -1; + } else + (*str)[MD5_DIGEST_LENGTH * 2] = 0; + + for (i = 0; i < MD5_DIGEST_LENGTH; i++) + snprintf(&(*str)[i * 2], 3, "%02x", md[i]); + + return 0; +} + +// ---------------------------------------------------- + +int +main(int argc, char **argv) +{ + u_char *md[2], mode = 0; + char szCmp[MAXPATHLEN], *str[2], ch; + int ret = 0; + + memset(md, 0, sizeof md); + memset(str, 0, sizeof str); + memset(szCmp, 0, MAXPATHLEN); + while ((ch = getopt(argc, argv, "hs:")) != -1) + switch (ch) { + case 's': + str[1] = strdup(optarg); + mode = 1; + break; + case 'h': + default: + Usage(); + return 1; + } + argc -= optind; + argv += optind; + + if (argc < 1) { + Usage(); + return 1; + } + if (argc > 1) { + strlcpy(szCmp, argv[1], MAXPATHLEN); + mode |= 2; + } + + if (calcDir(argv[0], &md[0]) == -1) + return 1; + if (convMD5(md[0], &str[0]) == -1) { + free(md[0]); + return 1; + } + printf("1.Directory '%s': %s\n", argv[0], str[0]); + + if (mode & 1 && *str[1]) { + ret |= !strcmp((char*) str[0], (char*) str[1]) << 1; + printf("Result checksum:: %s %s %s\n\n", str[0], ret & 1 ? "==" : "!=", str[1]); + free(str[1]); + } + + if (mode & 2 && *szCmp) { + if (calcDir(szCmp, &md[1]) == -1) + return 1; + if (convMD5(md[1], &str[1]) == -1) { + free(md[0]); + free(str[0]); + free(md[1]); + return 1; + } + printf("2.Directory '%s': %s\n", szCmp, str[1]); + + ret |= (!strcmp((char*) str[0], (char*) str[1]) << 2); + printf("Result directory:: %s %s %s\n\n", argv[0], ret & 2 ? "==" : "!=", szCmp); + + if (str[1]) + free(str[1]); + if (md[1]) + free(md[1]); + } + + if (str[0]) + free(str[0]); + if (md[0]) + free(md[0]); + return ret; +}