123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872 |
- /* rhash.c - implementation of LibRHash library calls
- *
- * Copyright: 2008-2012 Aleksey Kravchenko <rhash.admin@gmail.com>
- *
- * 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.
- *
- * This program 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. Use this program at your own risk!
- */
- /* macros for large file support, must be defined before any include file */
- #define _LARGEFILE64_SOURCE
- #define _FILE_OFFSET_BITS 64
- #include <string.h> /* memset() */
- #include <stdlib.h> /* free() */
- #include <stddef.h> /* ptrdiff_t */
- #include <stdio.h>
- #include <assert.h>
- #include <errno.h>
- /* modifier for Windows DLL */
- #if defined(_WIN32) && defined(RHASH_EXPORTS)
- # define RHASH_API __declspec(dllexport)
- #endif
- #include "byte_order.h"
- #include "algorithms.h"
- #include "util.h"
- #include "hex.h"
- #include "rhash.h" /* RHash library interface */
- #define STATE_ACTIVE 0xb01dbabe
- #define STATE_STOPED 0xdeadbeef
- #define STATE_DELETED 0xdecea5ed
- #define RCTX_AUTO_FINAL 0x1
- #define RCTX_FINALIZED 0x2
- #define RCTX_FINALIZED_MASK (RCTX_AUTO_FINAL | RCTX_FINALIZED)
- #define RHPR_FORMAT (RHPR_RAW | RHPR_HEX | RHPR_BASE32 | RHPR_BASE64)
- #define RHPR_MODIFIER (RHPR_UPPERCASE | RHPR_REVERSE)
- /**
- * Initialize static data of rhash algorithms
- */
- void rhash_library_init(void)
- {
- rhash_init_algorithms(RHASH_ALL_HASHES);
- #ifdef USE_OPENSSL
- rhash_plug_openssl();
- #endif
- }
- /**
- * Returns the number of supported hash algorithms.
- *
- * @return the number of supported hash functions
- */
- int RHASH_API rhash_count(void)
- {
- return rhash_info_size;
- }
- /* Lo-level rhash library functions */
- /**
- * Allocate and initialize RHash context for calculating hash(es).
- * After initializing rhash_update()/rhash_final() functions should be used.
- * Then the context must be freed by calling rhash_free().
- *
- * @param hash_id union of bit flags, containing ids of hashes to calculate.
- * @return initialized rhash context, NULL on error and errno is set
- */
- RHASH_API rhash rhash_init(unsigned hash_id)
- {
- unsigned tail_bit_index; /* index of hash_id trailing bit */
- unsigned num = 0; /* number of hashes to compute */
- rhash_context_ext *rctx = NULL; /* allocated rhash context */
- size_t hash_size_sum = 0; /* size of hash contexts to store in rctx */
- unsigned i, bit_index, id;
- struct rhash_hash_info* info;
- size_t aligned_size;
- char* phash_ctx;
- hash_id &= RHASH_ALL_HASHES;
- if (hash_id == 0) {
- errno = EINVAL;
- return NULL;
- }
- tail_bit_index = rhash_ctz(hash_id); /* get trailing bit index */
- assert(tail_bit_index < RHASH_HASH_COUNT);
- id = 1 << tail_bit_index;
- if (hash_id == id) {
- /* handle the most common case of only one hash */
- num = 1;
- info = &rhash_info_table[tail_bit_index];
- hash_size_sum = info->context_size;
- } else {
- /* another case: hash_id contains several hashes */
- for (bit_index = tail_bit_index; id <= hash_id; bit_index++, id = id << 1) {
- assert(id != 0);
- assert(bit_index < RHASH_HASH_COUNT);
- info = &rhash_info_table[bit_index];
- if (hash_id & id) {
- /* align sizes by 8 bytes */
- aligned_size = (info->context_size + 7) & ~7;
- hash_size_sum += aligned_size;
- num++;
- }
- }
- assert(num > 1);
- }
- /* align the size of the rhash context common part */
- aligned_size = (offsetof(rhash_context_ext, vector[num]) + 7) & ~7;
- assert(aligned_size >= sizeof(rhash_context_ext));
- /* allocate rhash context with enough memory to store contexts of all used hashes */
- rctx = (rhash_context_ext*)malloc(aligned_size + hash_size_sum);
- if (rctx == NULL) return NULL;
- /* initialize common fields of the rhash context */
- memset(rctx, 0, sizeof(rhash_context_ext));
- rctx->rc.hash_id = hash_id;
- rctx->flags = RCTX_AUTO_FINAL; /* turn on auto-final by default */
- rctx->state = STATE_ACTIVE;
- rctx->hash_vector_size = num;
- /* aligned hash contexts follows rctx->vector[num] in the same memory block */
- phash_ctx = (char*)rctx + aligned_size;
- assert(phash_ctx >= (char*)&rctx->vector[num]);
- /* initialize context for every hash in a loop */
- for (bit_index = tail_bit_index, id = 1 << tail_bit_index, i = 0;
- id <= hash_id; bit_index++, id = id << 1)
- {
- /* check if a hash function with given id shall be included into rctx */
- if ((hash_id & id) != 0) {
- info = &rhash_info_table[bit_index];
- assert(info->context_size > 0);
- assert(((phash_ctx - (char*)0) & 7) == 0); /* hash context is aligned */
- assert(info->init != NULL);
- rctx->vector[i].hash_info = info;
- rctx->vector[i].context = phash_ctx;
- #if 0
- /* BTIH initialization is complex, save pointer for later */
- if ((id & RHASH_BTIH) != 0) rctx->bt_ctx = phash_ctx;
- #endif
- phash_ctx += (info->context_size + 7) & ~7;
- /* initialize the i-th hash context */
- info->init(rctx->vector[i].context);
- i++;
- }
- }
- return &rctx->rc; /* return allocated and initialized rhash context */
- }
- /**
- * Free RHash context memory.
- *
- * @param ctx the context to free.
- */
- void rhash_free(rhash ctx)
- {
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- unsigned i;
- if (ctx == 0) return;
- assert(ectx->hash_vector_size <= RHASH_HASH_COUNT);
- ectx->state = STATE_DELETED; /* mark memory block as being removed */
- /* clean the hash functions, which require additional clean up */
- for (i = 0; i < ectx->hash_vector_size; i++) {
- struct rhash_hash_info* info = ectx->vector[i].hash_info;
- if (info->cleanup != 0) {
- info->cleanup(ectx->vector[i].context);
- }
- }
- free(ectx);
- }
- /**
- * Re-initialize RHash context to reuse it.
- * Useful to speed up processing of many small messages.
- *
- * @param ctx context to reinitialize
- */
- RHASH_API void rhash_reset(rhash ctx)
- {
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- unsigned i;
- assert(ectx->hash_vector_size > 0);
- assert(ectx->hash_vector_size <= RHASH_HASH_COUNT);
- ectx->state = STATE_ACTIVE; /* re-activate the structure */
- /* re-initialize every hash in a loop */
- for (i = 0; i < ectx->hash_vector_size; i++) {
- struct rhash_hash_info* info = ectx->vector[i].hash_info;
- if (info->cleanup != 0) {
- info->cleanup(ectx->vector[i].context);
- }
- assert(info->init != NULL);
- info->init(ectx->vector[i].context);
- }
- ectx->flags &= ~RCTX_FINALIZED; /* clear finalized state */
- }
- /**
- * Calculate hashes of message.
- * Can be called repeatedly with chunks of the message to be hashed.
- *
- * @param ctx the rhash context
- * @param message message chunk
- * @param length length of the message chunk
- * @return 0 on success; On fail return -1 and set errno
- */
- RHASH_API int rhash_update(rhash ctx, const void* message, size_t length)
- {
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- unsigned i;
- assert(ectx->hash_vector_size <= RHASH_HASH_COUNT);
- if (ectx->state != STATE_ACTIVE) return 0; /* do nothing if canceled */
- ctx->msg_size += length;
- /* call update method for every algorithm */
- for (i = 0; i < ectx->hash_vector_size; i++) {
- struct rhash_hash_info* info = ectx->vector[i].hash_info;
- assert(info->update != 0);
- info->update(ectx->vector[i].context, message, length);
- }
- return 0; /* no error processing at the moment */
- }
- /**
- * Finalize hash calculation and optionally store the first hash.
- *
- * @param ctx the rhash context
- * @param first_result optional buffer to store a calculated hash with the lowest available id
- * @return 0 on success; On fail return -1 and set errno
- */
- RHASH_API int rhash_final(rhash ctx, unsigned char* first_result)
- {
- unsigned i = 0;
- unsigned char buffer[130];
- unsigned char* out = (first_result ? first_result : buffer);
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- assert(ectx->hash_vector_size <= RHASH_HASH_COUNT);
- /* skip final call if already finalized and auto-final is on */
- if ((ectx->flags & RCTX_FINALIZED_MASK) ==
- (RCTX_AUTO_FINAL | RCTX_FINALIZED)) return 0;
- /* call final method for every algorithm */
- for (i = 0; i < ectx->hash_vector_size; i++) {
- struct rhash_hash_info* info = ectx->vector[i].hash_info;
- assert(info->final != 0);
- assert(info->info->digest_size < sizeof(buffer));
- info->final(ectx->vector[i].context, out);
- out = buffer;
- }
- ectx->flags |= RCTX_FINALIZED;
- return 0; /* no error processing at the moment */
- }
- /**
- * Store digest for given hash_id.
- * If hash_id is zero, function stores digest for a hash with the lowest id found in the context.
- * For nonzero hash_id the context must contain it, otherwise function silently does nothing.
- *
- * @param ctx rhash context
- * @param hash_id id of hash to retrieve or zero for hash with the lowest available id
- * @param result buffer to put the hash into
- */
- static void rhash_put_digest(rhash ctx, unsigned hash_id, unsigned char* result)
- {
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- unsigned i;
- rhash_vector_item *item;
- struct rhash_hash_info* info;
- unsigned char* digest;
- assert(ectx);
- assert(ectx->hash_vector_size > 0 && ectx->hash_vector_size <= RHASH_HASH_COUNT);
- /* finalize context if not yet finalized and auto-final is on */
- if ((ectx->flags & RCTX_FINALIZED_MASK) == RCTX_AUTO_FINAL) {
- rhash_final(ctx, NULL);
- }
- if (hash_id == 0) {
- item = &ectx->vector[0]; /* get the first hash */
- info = item->hash_info;
- } else {
- for (i = 0;; i++) {
- if (i >= ectx->hash_vector_size) {
- return; /* hash_id not found, do nothing */
- }
- item = &ectx->vector[i];
- info = item->hash_info;
- if (info->info->hash_id == hash_id) break;
- }
- }
- digest = ((unsigned char*)item->context + info->digest_diff);
- if (info->info->flags & F_SWAP32) {
- assert((info->info->digest_size & 3) == 0);
- /* NB: the next call is correct only for multiple of 4 byte size */
- rhash_swap_copy_str_to_u32(result, 0, digest, info->info->digest_size);
- } else if (info->info->flags & F_SWAP64) {
- rhash_swap_copy_u64_to_str(result, digest, info->info->digest_size);
- } else {
- memcpy(result, digest, info->info->digest_size);
- }
- }
- /**
- * Set the callback function to be called from the
- * rhash_file() and rhash_file_update() functions
- * on processing every file block. The file block
- * size is set internally by rhash and now is 8 KiB.
- *
- * @param ctx rhash context
- * @param callback pointer to the callback function
- * @param callback_data pointer to data passed to the callback
- */
- RHASH_API void rhash_set_callback(rhash ctx, rhash_callback_t callback, void* callback_data)
- {
- ((rhash_context_ext*)ctx)->callback = callback;
- ((rhash_context_ext*)ctx)->callback_data = callback_data;
- }
- /* hi-level message hashing interface */
- /**
- * Compute a hash of the given message.
- *
- * @param hash_id id of hash sum to compute
- * @param message the message to process
- * @param length message length
- * @param result buffer to receive binary hash string
- * @return 0 on success, -1 on error
- */
- RHASH_API int rhash_msg(unsigned hash_id, const void* message, size_t length, unsigned char* result)
- {
- rhash ctx;
- hash_id &= RHASH_ALL_HASHES;
- ctx = rhash_init(hash_id);
- if (ctx == NULL) return -1;
- rhash_update(ctx, message, length);
- rhash_final(ctx, result);
- rhash_free(ctx);
- return 0;
- }
- /**
- * Hash a file or stream. Multiple hashes can be computed.
- * First, inintialize ctx parameter with rhash_init() before calling
- * rhash_file_update(). Then use rhash_final() and rhash_print()
- * to retrive hash values. Finaly call rhash_free() on ctx
- * to free allocated memory or call rhash_reset() to reuse ctx.
- *
- * @param ctx rhash context
- * @param fd descriptor of the file to hash
- * @return 0 on success, -1 on error and errno is set
- */
- RHASH_API int rhash_file_update(rhash ctx, FILE* fd)
- {
- rhash_context_ext* const ectx = (rhash_context_ext*)ctx;
- const size_t block_size = 8192;
- unsigned char *buffer, *pmem;
- size_t length = 0, align8;
- int res = 0;
- if (ectx->state != STATE_ACTIVE) return 0; /* do nothing if canceled */
- if (ctx == NULL) {
- errno = EINVAL;
- return -1;
- }
- pmem = (unsigned char*)malloc(block_size + 8);
- if (!pmem) return -1; /* errno is set to ENOMEM according to UNIX 98 */
- align8 = ((unsigned char*)0 - pmem) & 7;
- buffer = pmem + align8;
- while (!feof(fd)) {
- /* stop if canceled */
- if (ectx->state != STATE_ACTIVE) break;
- length = fread(buffer, 1, block_size, fd);
- if (ferror(fd)) {
- res = -1; /* note: errno contains error code */
- break;
- } else if (length) {
- rhash_update(ctx, buffer, length);
- if (ectx->callback) {
- ((rhash_callback_t)ectx->callback)(ectx->callback_data, ectx->rc.msg_size);
- }
- }
- }
- free(buffer);
- return res;
- }
- /**
- * Compute a single hash for given file.
- *
- * @param hash_id id of hash sum to compute
- * @param filepath path to the file to hash
- * @param result buffer to receive hash value with the lowest requested id
- * @return 0 on success, -1 on error and errno is set
- */
- RHASH_API int rhash_file(unsigned hash_id, const char* filepath, unsigned char* result)
- {
- FILE* fd;
- rhash ctx;
- int res;
- hash_id &= RHASH_ALL_HASHES;
- if (hash_id == 0) {
- errno = EINVAL;
- return -1;
- }
- if ((fd = fopen(filepath, "rb")) == NULL) return -1;
- if ((ctx = rhash_init(hash_id)) == NULL) return -1;
- res = rhash_file_update(ctx, fd); /* hash the file */
- fclose(fd);
- rhash_final(ctx, result);
- rhash_free(ctx);
- return res;
- }
- #ifdef _WIN32 /* windows only function */
- #include <share.h>
- /**
- * Compute a single hash for given file.
- *
- * @param hash_id id of hash sum to compute
- * @param filepath path to the file to hash
- * @param result buffer to receive hash value with the lowest requested id
- * @return 0 on success, -1 on error, -1 on error and errno is set
- */
- RHASH_API int rhash_wfile(unsigned hash_id, const wchar_t* filepath, unsigned char* result)
- {
- FILE* fd;
- rhash ctx;
- int res;
- hash_id &= RHASH_ALL_HASHES;
- if (hash_id == 0) {
- errno = EINVAL;
- return -1;
- }
- if ((fd = _wfsopen(filepath, L"rb", _SH_DENYWR)) == NULL) return -1;
- if ((ctx = rhash_init(hash_id)) == NULL) return -1;
- res = rhash_file_update(ctx, fd); /* hash the file */
- fclose(fd);
- rhash_final(ctx, result);
- rhash_free(ctx);
- return res;
- }
- #endif
- /* RHash information functions */
- /**
- * Returns information about a hash function by its hash_id.
- *
- * @param hash_id the id of hash algorithm
- * @return pointer to the rhash_info structure containing the information
- */
- const rhash_info* rhash_info_by_id(unsigned hash_id)
- {
- hash_id &= RHASH_ALL_HASHES;
- /* check that only one bit is set */
- if (hash_id != (hash_id & -(int)hash_id)) return NULL;
- /* note: alternative condition is (hash_id == 0 || (hash_id & (hash_id - 1)) != 0) */
- return rhash_info_table[rhash_ctz(hash_id)].info;
- }
- #if 0
- /**
- * Detect default digest output format for given hash algorithm.
- *
- * @param hash_id the id of hash algorithm
- * @return 1 for base32 format, 0 for hexadecimal
- */
- RHASH_API int rhash_is_base32(unsigned hash_id)
- {
- /* fast method is just to test a bit-mask */
- return ((hash_id & (RHASH_TTH | RHASH_AICH)) != 0);
- }
- #endif
- /**
- * Returns size of binary digest for given hash algorithm.
- *
- * @param hash_id the id of hash algorithm
- * @return digest size in bytes
- */
- RHASH_API int rhash_get_digest_size(unsigned hash_id)
- {
- hash_id &= RHASH_ALL_HASHES;
- if (hash_id == 0 || (hash_id & (hash_id - 1)) != 0) return -1;
- return (int)rhash_info_table[rhash_ctz(hash_id)].info->digest_size;
- }
- /**
- * Returns length of digest hash string in default output format.
- *
- * @param hash_id the id of hash algorithm
- * @return the length of hash string
- */
- RHASH_API int rhash_get_hash_length(unsigned hash_id)
- {
- const rhash_info* info = rhash_info_by_id(hash_id);
- return (int)(info ? (info->flags & F_BS32 ?
- BASE32_LENGTH(info->digest_size) : info->digest_size * 2) : 0);
- }
- /**
- * Returns a name of given hash algorithm.
- *
- * @param hash_id the id of hash algorithm
- * @return algorithm name
- */
- RHASH_API const char* rhash_get_name(unsigned hash_id)
- {
- const rhash_info* info = rhash_info_by_id(hash_id);
- return (info ? info->name : 0);
- }
- /**
- * Returns a name part of magnet urn of the given hash algorithm.
- * Such magnet_name is used to generate a magnet link of the form
- * urn:<magnet_name>=<hash_value>.
- *
- * @param hash_id the id of hash algorithm
- * @return name
- */
- RHASH_API const char* rhash_get_magnet_name(unsigned hash_id)
- {
- const rhash_info* info = rhash_info_by_id(hash_id);
- return (info ? info->magnet_name : 0);
- }
- #if 0
- static size_t rhash_get_magnet_url_size(const char* filepath,
- rhash context, unsigned hash_mask, int flags)
- {
- size_t size = 0; /* count terminating '\0' */
- unsigned bit, hash = context->hash_id & hash_mask;
- /* RHPR_NO_MAGNET, RHPR_FILESIZE */
- if ((flags & RHPR_NO_MAGNET) == 0) {
- size += 8;
- }
- if ((flags & RHPR_FILESIZE) != 0) {
- uint64_t num = context->msg_size;
- size += 4;
- if (num == 0) size++;
- else {
- for (; num; num /= 10, size++);
- }
- }
- if (filepath) {
- size += 4 + rhash_urlencode(NULL, filepath);
- }
- /* loop through hash values */
- for (bit = hash & -(int)hash; bit <= hash; bit <<= 1) {
- const char* name;
- if ((bit & hash) == 0) continue;
- if ((name = rhash_get_magnet_name(bit)) == 0) continue;
- size += (7 + 2) + strlen(name);
- size += rhash_print(NULL, context, bit,
- (bit & (RHASH_SHA1 | RHASH_BTIH) ? RHPR_BASE32 : 0));
- }
- return size;
- }
- /**
- * Print magnet link with given filepath and calculated hash sums into the
- * output buffer. The hash_mask can limit which hash values will be printed.
- * The function returns the size of the required buffer.
- * If output is NULL the .
- *
- * @param output a string buffer to receive the magnet link or NULL
- * @param filepath the file path to be printed or NULL
- * @param context algorithms state
- * @param hash_mask bit mask of the hash sums to add to the link
- * @param flags can be combination of bits RHPR_UPPERCASE, RHPR_NO_MAGNET,
- * RHPR_FILESIZE
- * @return number of written characters, including terminating '\0' on success, 0 on fail
- */
- RHASH_API size_t rhash_print_magnet(char* output, const char* filepath,
- rhash context, unsigned hash_mask, int flags)
- {
- int i;
- const char* begin = output;
- if (output == NULL) return rhash_get_magnet_url_size(
- filepath, context, hash_mask, flags);
- /* RHPR_NO_MAGNET, RHPR_FILESIZE */
- if ((flags & RHPR_NO_MAGNET) == 0) {
- strcpy(output, "magnet:?");
- output += 8;
- }
- if ((flags & RHPR_FILESIZE) != 0) {
- strcpy(output, "xl=");
- output += 3;
- output += rhash_sprintI64(output, context->msg_size);
- *(output++) = '&';
- }
- if (filepath) {
- strcpy(output, "dn=");
- output += 3;
- output += rhash_urlencode(output, filepath);
- *(output++) = '&';
- }
- flags &= RHPR_UPPERCASE;
- for (i = 0; i < 2; i++) {
- unsigned bit;
- unsigned hash = context->hash_id & hash_mask;
- hash = (i == 0 ? hash & (RHASH_ED2K | RHASH_AICH)
- : hash & ~(RHASH_ED2K | RHASH_AICH));
- if (!hash) continue;
- /* loop through hash values */
- for (bit = hash & -(int)hash; bit <= hash; bit <<= 1) {
- const char* name;
- if ((bit & hash) == 0) continue;
- if (!(name = rhash_get_magnet_name(bit))) continue;
- strcpy(output, "xt=urn:");
- output += 7;
- strcpy(output, name);
- output += strlen(name);
- *(output++) = ':';
- output += rhash_print(output, context, bit,
- (bit & (RHASH_SHA1 | RHASH_BTIH) ? flags | RHPR_BASE32 : flags));
- *(output++) = '&';
- }
- }
- output[-1] = '\0'; /* terminate the line */
- return (output - begin);
- }
- /* hash sum output */
- /**
- * Print a text presentation of a given hash sum to the specified buffer,
- *
- * @param output a buffer to print the hash to
- * @param bytes a hash sum to print
- * @param size a size of hash sum in bytes
- * @param flags a bit-mask controlling how to format the hash sum,
- * can be a mix of the flags: RHPR_RAW, RHPR_HEX, RHPR_BASE32,
- * RHPR_BASE64, RHPR_UPPERCASE, RHPR_REVERSE
- * @return the number of written characters
- */
- size_t rhash_print_bytes(char* output, const unsigned char* bytes,
- size_t size, int flags)
- {
- size_t str_len;
- int upper_case = (flags & RHPR_UPPERCASE);
- int format = (flags & ~RHPR_MODIFIER);
- switch (format) {
- case RHPR_HEX:
- str_len = size * 2;
- rhash_byte_to_hex(output, bytes, (unsigned)size, upper_case);
- break;
- case RHPR_BASE32:
- str_len = BASE32_LENGTH(size);
- rhash_byte_to_base32(output, bytes, (unsigned)size, upper_case);
- break;
- case RHPR_BASE64:
- str_len = BASE64_LENGTH(size);
- rhash_byte_to_base64(output, bytes, (unsigned)size);
- break;
- default:
- str_len = size;
- memcpy(output, bytes, size);
- break;
- }
- return str_len;
- }
- /**
- * Print text presentation of a hash sum with given hash_id to the specified
- * output buffer. If the hash_id is zero, then print the hash sum with
- * the lowest id stored in the hash context.
- * The function call fails if the context doesn't include a hash with the
- * given hash_id.
- *
- * @param output a buffer to print the hash to
- * @param context algorithms state
- * @param hash_id id of the hash sum to print or 0 to print the first hash
- * saved in the context.
- * @param flags a bitmask controlling how to print the hash. Can contain flags
- * RHPR_UPPERCASE, RHPR_HEX, RHPR_BASE32, RHPR_BASE64, etc.
- * @return the number of written characters on success or 0 on fail
- */
- size_t RHASH_API rhash_print(char* output, rhash context, unsigned hash_id, int flags)
- {
- const rhash_info* info;
- unsigned char digest[80];
- size_t digest_size;
- info = (hash_id != 0 ? rhash_info_by_id(hash_id) :
- ((rhash_context_ext*)context)->vector[0].hash_info->info);
- if (info == NULL) return 0;
- digest_size = info->digest_size;
- assert(digest_size <= 64);
- flags &= (RHPR_FORMAT | RHPR_MODIFIER);
- if ((flags & RHPR_FORMAT) == 0) {
- /* use default format if not specified by flags */
- flags |= (info->flags & RHASH_INFO_BASE32 ? RHPR_BASE32 : RHPR_HEX);
- }
- if (output == NULL) {
- switch (flags & RHPR_FORMAT) {
- case RHPR_HEX:
- return (digest_size * 2);
- case RHPR_BASE32:
- return BASE32_LENGTH(digest_size);
- case RHPR_BASE64:
- return BASE64_LENGTH(digest_size);
- default:
- return digest_size;
- }
- }
- /* note: use info->hash_id, cause hash_id can be 0 */
- rhash_put_digest(context, info->hash_id, digest);
- if ((flags & ~RHPR_UPPERCASE) == (RHPR_REVERSE | RHPR_HEX)) {
- /* reverse the digest */
- unsigned char *p = digest, *r = digest + digest_size - 1;
- char tmp;
- for (; p < r; p++, r--) {
- tmp = *p;
- *p = *r;
- *r = tmp;
- }
- }
- return rhash_print_bytes(output, digest, digest_size, flags);
- }
- #if defined(_WIN32) && defined(RHASH_EXPORTS)
- #include <windows.h>
- BOOL APIENTRY DllMain(HMODULE hModule, DWORD reason, LPVOID reserved);
- BOOL APIENTRY DllMain(HMODULE hModule, DWORD reason, LPVOID reserved)
- {
- (void)hModule;
- (void)reserved;
- switch (reason) {
- case DLL_PROCESS_ATTACH:
- rhash_library_init();
- break;
- case DLL_PROCESS_DETACH:
- /*rhash_library_free();*/
- case DLL_THREAD_ATTACH:
- case DLL_THREAD_DETACH:
- break;
- }
- return TRUE;
- }
- #endif
- #define PVOID2UPTR(p) ((rhash_uptr_t)((char*)p - 0))
- /**
- * Process a rhash message.
- *
- * @param msg_id message identifier
- * @param dst message destination (can be NULL for generic messages)
- * @param ldata data depending on message
- * @param rdata data depending on message
- * @return message-specific data
- */
- RHASH_API rhash_uptr_t rhash_transmit(unsigned msg_id, void* dst, rhash_uptr_t ldata, rhash_uptr_t rdata)
- {
- /* for messages working with rhash context */
- rhash_context_ext* const ctx = (rhash_context_ext*)dst;
- switch (msg_id) {
- case RMSG_GET_CONTEXT:
- {
- unsigned i;
- for (i = 0; i < ctx->hash_vector_size; i++) {
- struct rhash_hash_info* info = ctx->vector[i].hash_info;
- if (info->info->hash_id == (unsigned)ldata)
- return PVOID2UPTR(ctx->vector[i].context);
- }
- return (rhash_uptr_t)0;
- }
- case RMSG_CANCEL:
- /* mark rhash context as canceled, in a multithreaded program */
- atomic_compare_and_swap(&ctx->state, STATE_ACTIVE, STATE_STOPED);
- return 0;
- case RMSG_IS_CANCELED:
- return (ctx->state == STATE_STOPED);
- case RMSG_GET_FINALIZED:
- return ((ctx->flags & RCTX_FINALIZED) != 0);
- case RMSG_SET_AUTOFINAL:
- ctx->flags &= ~RCTX_AUTO_FINAL;
- if (ldata) ctx->flags |= RCTX_AUTO_FINAL;
- break;
- /* OpenSSL related messages */
- #ifdef USE_OPENSSL
- case RMSG_SET_OPENSSL_MASK:
- rhash_openssl_hash_mask = (unsigned)ldata;
- break;
- case RMSG_GET_OPENSSL_MASK:
- return rhash_openssl_hash_mask;
- #endif
- default:
- return RHASH_ERROR; /* unknown message */
- }
- return 0;
- }
- #endif
|