boinc/lib/md5_file.cpp

142 lines
3.3 KiB
C++

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
#if defined(_WIN32) && !defined(__STDWX_H__)
#include "boinc_win.h"
#elif defined(_WIN32) && defined(__STDWX_H__)
#include "stdwx.h"
#else
#include "config.h"
#ifdef _USING_FCGI_
#include "boinc_fcgi.h"
#else
#include <cstdio>
#endif
#endif
#ifdef _WIN32
#include <wincrypt.h>
#endif
#ifdef ANDROID
#include <stdlib.h>
#endif
#include "error_numbers.h"
#include "md5.h"
#include "md5_file.h"
int md5_file(const char* path, char* output, double& nbytes) {
unsigned char buf[4096];
unsigned char binout[16];
md5_state_t state;
int i, n;
nbytes = 0;
#ifndef _USING_FCGI_
FILE *f = fopen(path, "rb");
#else
FILE *f = FCGI::fopen(path, "rb");
#endif
if (!f) {
fprintf(stderr, "md5_file: can't open %s\n", path);
#ifndef _USING_FCGI_
std::perror("md5_file");
#else
FCGI::perror("md5_file");
#endif
return ERR_FOPEN;
}
md5_init(&state);
while (1) {
n = (int)fread(buf, 1, 4096, f);
if (n<=0) break;
nbytes += n;
md5_append(&state, buf, n);
}
md5_finish(&state, binout);
for (i=0; i<16; i++) {
sprintf(output+2*i, "%02x", binout[i]);
}
output[32] = 0;
fclose(f);
return 0;
}
int md5_block(const unsigned char* data, int nbytes, char* output) {
unsigned char binout[16];
int i;
md5_state_t state;
md5_init(&state);
md5_append(&state, data, nbytes);
md5_finish(&state, binout);
for (i=0; i<16; i++) {
sprintf(output+2*i, "%02x", binout[i]);
}
output[32] = 0;
return 0;
}
std::string md5_string(const unsigned char* data, int nbytes) {
char output[MD5_LEN];
md5_block(data, nbytes, output);
return std::string(output);
}
// make a random 32-char string
// (the MD5 of some quasi-random bits)
//
int make_random_string(char* out) {
char buf[256];
#ifdef _WIN32
HCRYPTPROV hCryptProv;
if(! CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_FULL, 0)) {
return -1;
}
if(! CryptGenRandom(hCryptProv, (DWORD) 32, (BYTE *) buf)) {
CryptReleaseContext(hCryptProv, 0);
return -1;
}
CryptReleaseContext(hCryptProv, 0);
#elif defined ANDROID
// /dev/random not available on Android, using stdlib function instead
int i = rand();
snprintf(buf, sizeof(buf), "%d", i);
#else
#ifndef _USING_FCGI_
FILE* f = fopen("/dev/random", "r");
#else
FILE* f = FCGI::fopen("/dev/random", "r");
#endif
if (!f) {
return -1;
}
size_t n = fread(buf, 32, 1, f);
fclose(f);
if (n != 1) return -1;
#endif
md5_block((const unsigned char*)buf, 32, out);
return 0;
}