md5_test.c

/*
 * MIT License
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <md5.h>

#define HEX_STR "0123456789abcdef"

#define FILE_0_MD5 "d41d8cd98f00b204e9800998ecf8427e"
#define FILE_1_MD5 "60b725f10c9c85c70d97880dfe8191b3"
#define FILE_2_MD5 "daa8075d6ac5ff8d0c6d4650adb4ef29"
#define FILE_3_MD5 "3b41b17bcce0710e7b6f3984c3ce092b"

static char*
md5_one(const char *fn)
{
    unsigned char buf[4096], digest[16];
    //unsigned char result[32 + 1];
    char* result;
    MD5_CTX md5;
    int l;
    FILE *fp;

    result = malloc(sizeof(char) * (32 + 1));
    fp = strcmp(fn, "-")? fopen(fn, "r") : stdin;
    if (fp == 0) {
        fprintf(stderr, "md5sum: %s: No such file or directory\n", fn);
        exit(1);
    }
    MD5Init(&md5);
    while ((l = fread(buf, 1, 4096, fp)) > 0)
        MD5Update(&md5, buf, l);
    MD5Final(digest, &md5);
    if (fp != stdin) fclose(fp);
    for (l = 0; l < 16; ++l)
        sprintf(&result[l*2], "%c%c", HEX_STR[digest[l]>>4&0xf], HEX_STR[digest[l]&0xf]);
    return result;
}

int main_example(int argc, char *argv[])
{
    int i;
    char* result;
    if (argc == 1) {
        md5_one("-");
    } else {
        for (i = 1; i < argc; ++i) {
            result = md5_one(argv[i]);
            printf("sum= %s\n", result);
            free(result);
        }
    }

    return 0;
}

int main(int argc, char *argv[])
{
    char* result;
    int ret;
    result = md5_one("tests/file0");
    if (strcmp(result, FILE_0_MD5) != 0) {
        printf("error on file 0\n");
        free(result);
        return 1;
    }
    free(result);

    result = md5_one("tests/file1");
    if (strcmp(result, FILE_1_MD5) != 0) {
        printf("error on file 1\n");
        free(result);
        return 1;
    }
    free(result);

    result = md5_one("tests/file2");
    if (strcmp(result, FILE_2_MD5) != 0) {
        printf("error on file 2\n");
        free(result);
        return 1;
    } else {
    }
    free(result);

    result = md5_one("tests/file3");
    if (strcmp(result, FILE_3_MD5) != 0) {
        printf("error on file 3\n");
        free(result);
        return 1;
    }
    free(result);

    return 0;
}