用C语言实现md5sum/sha256sum的功能

=Start=

缘由：

学习需要

正文：

参考解答：

MD5(支持文件和字符串)

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <openssl/md5.h>
int file_exist(char *filename) {
    if( access( filename, F_OK ) != -1 ) {
        return 1;   // file exists
    } else {
        return 0;   // file doesn't exist
    }
}
void get_file_hash(char* filename, unsigned char* digest){
    MD5_CTX c;
    char buf[1024];
    ssize_t bytes;
    int fd = open(filename, O_RDONLY);
    if (fd == -1)
        exit(-1);
    MD5_Init(&c);
    bytes = read(fd, buf, 1024);
    while(bytes > 0) {
        MD5_Update(&c, buf, bytes);
        bytes=read(fd, buf, 1024);
    }
    MD5_Final(digest, &c);
    close(fd);
}
void get_str_hash(char* str, unsigned char* digest){
    MD5_CTX c;
    MD5_Init(&c);
    MD5_Update(&c, str, strlen(str));
    MD5_Final(digest, &c);
}
// Print the MD5 sum as hex-digits.
void print_md5_sum(unsigned char* md) {
    int i;
    for(i=0; i <MD5_DIGEST_LENGTH; i++) {
        printf("%02x", md[i]);
    }
}
int main(int argc, char *argv[]) {
    printf("MD5_DIGEST_LENGTH = %d\n", MD5_DIGEST_LENGTH);
    unsigned char md5_digest[MD5_DIGEST_LENGTH];
    int i = 0;
    for (i = 1; i < argc; i++) {
        printf("argv[%d] = '%s'\n", i, argv[i]);
        if (file_exist(argv[i])) {
            get_file_hash(argv[i], md5_digest);
            print_md5_sum(md5_digest);
            printf("\tFile('%s')\n\n", argv[i]);
        } else {
            get_str_hash(argv[i], md5_digest);
            print_md5_sum(md5_digest);
            printf("\tString('%s')\n\n", argv[i]);
        }
    }
    return 0;
}

SHA256(支持文件和字符串)

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <openssl/sha.h>
int file_exist(char *filename) {
    if( access( filename, F_OK ) != -1 ) {
        return 1;   // file exists
    } else {
        return 0;   // file doesn't exist
    }
}
void get_file_hash(char* filename, unsigned char* digest){
    SHA256_CTX c;
    char buf[1024];
    ssize_t bytes;
    int fd = open(filename, O_RDONLY);
    if (fd == -1)
        exit(-1);
    SHA256_Init(&c);
    bytes = read(fd, buf, 1024);
    while(bytes > 0) {
        SHA256_Update(&c, buf, bytes);
        bytes=read(fd, buf, 1024);
    }
    SHA256_Final(digest, &c);
    close(fd);
}
void get_str_hash(char* str, unsigned char* digest){
    SHA256_CTX c;
    SHA256_Init(&c);
    SHA256_Update(&c, str, strlen(str));
    SHA256_Final(digest, &c);
}
// Print the MD5/SHA1/SHA256 sum as hex-digits.
void print_hex_hash(unsigned char* digest, int length) {
    int i;
    for(i=0; i <length; i++) {
        printf("%02x", digest[i]);
    }
}
int main(int argc, char *argv[]) {
    printf("SHA256_DIGEST_LENGTH = %d\n", SHA256_DIGEST_LENGTH);
    unsigned char sha256_digest[SHA256_DIGEST_LENGTH];
    int i = 0;
    for (i = 1; i < argc; i++) {
        printf("argv[%d] = '%s'\n", i, argv[i]);
        if (file_exist(argv[i])) {
            get_file_hash(argv[i], sha256_digest);
            print_hex_hash(sha256_digest, SHA256_DIGEST_LENGTH);
            printf("\tFile('%s')\n\n", argv[i]);
        } else {
            get_str_hash(argv[i], sha256_digest);
            print_hex_hash(sha256_digest, SHA256_DIGEST_LENGTH);
            printf("\tString('%s')\n\n", argv[i]);
        }
    }
    return 0;
}