add size checks to utils

This commit is contained in:
Jethro Grassie 2019-05-05 23:53:36 -04:00
parent eeaeb1acdc
commit e407929555
No known key found for this signature in database
GPG key ID: DE8ED755616565BB
3 changed files with 16 additions and 17 deletions

View file

@ -1027,7 +1027,7 @@ target_to_hex(uint64_t target, char *target_hex)
if (target & 0xFFFFFFFF00000000) if (target & 0xFFFFFFFF00000000)
{ {
log_debug("High target requested: %"PRIu64, target); log_debug("High target requested: %"PRIu64, target);
bin_to_hex((const char*)&target, 8, &target_hex[0]); bin_to_hex((const char*)&target, 8, &target_hex[0], 16);
target_hex[16] = '\0'; target_hex[16] = '\0';
return; return;
} }
@ -1043,7 +1043,7 @@ target_to_hex(uint64_t target, char *target_hex)
BN_div(diff, NULL, base_diff, bnt, bn_ctx); BN_div(diff, NULL, base_diff, bnt, bn_ctx);
BN_rshift(diff, diff, 224); BN_rshift(diff, diff, 224);
uint32_t w = BN_get_word(diff); uint32_t w = BN_get_word(diff);
bin_to_hex((const char*)&w, 4, &target_hex[0]); bin_to_hex((const char*)&w, 4, &target_hex[0], 8);
target_hex[8] = '\0'; target_hex[8] = '\0';
BN_free(bnt); BN_free(bnt);
BN_free(diff); BN_free(diff);
@ -1056,7 +1056,7 @@ stratum_get_proxy_job_body(char *body, const client_t *client, const char *block
const char *client_id = client->client_id; const char *client_id = client->client_id;
const job_t *job = &client->active_jobs[0]; const job_t *job = &client->active_jobs[0];
char job_id[33]; char job_id[33];
bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id); bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id, 32);
uint64_t target = job->target; uint64_t target = job->target;
char target_hex[17]; char target_hex[17];
target_to_hex(target, &target_hex[0]); target_to_hex(target, &target_hex[0]);
@ -1091,7 +1091,7 @@ stratum_get_job_body(char *body, const client_t *client, bool response)
const char *client_id = client->client_id; const char *client_id = client->client_id;
const job_t *job = &client->active_jobs[0]; const job_t *job = &client->active_jobs[0];
char job_id[33]; char job_id[33];
bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id); bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id, 32);
const char *blob = job->blob; const char *blob = job->blob;
uint64_t target = job->target; uint64_t target = job->target;
uint64_t height = job->block_template->height; uint64_t height = job->block_template->height;
@ -1776,7 +1776,7 @@ client_send_job(client_t *client, bool response)
/* Make hex */ /* Make hex */
job->blob = calloc((hashing_blob_size << 1) +1, sizeof(char)); job->blob = calloc((hashing_blob_size << 1) +1, sizeof(char));
bin_to_hex(hashing_blob, hashing_blob_size, job->blob); bin_to_hex(hashing_blob, hashing_blob_size, job->blob, hashing_blob_size << 1);
log_trace("Miner hashing blob: %s", job->blob); log_trace("Miner hashing blob: %s", job->blob);
/* Save a job id */ /* Save a job id */
@ -1787,7 +1787,7 @@ client_send_job(client_t *client, bool response)
/* Send */ /* Send */
char job_id[33]; char job_id[33];
bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id); bin_to_hex((const char*)job->id, sizeof(uuid_t), job_id, 32);
/* Retarget */ /* Retarget */
double duration = difftime(time(NULL), client->connected_since); double duration = difftime(time(NULL), client->connected_since);
@ -1804,7 +1804,7 @@ client_send_job(client_t *client, bool response)
else else
{ {
char *block_hex = calloc(bin_size+1, sizeof(char)); char *block_hex = calloc(bin_size+1, sizeof(char));
bin_to_hex(block, bin_size, block_hex); bin_to_hex(block, bin_size, block_hex, bin_size << 1);
stratum_get_proxy_job_body(body, client, block_hex, response); stratum_get_proxy_job_body(body, client, block_hex, response);
free(block_hex); free(block_hex);
} }
@ -1875,7 +1875,7 @@ client_on_login(json_object *message, client_t *client)
strncpy(client->worker_id, worker_id, sizeof(client->worker_id)); strncpy(client->worker_id, worker_id, sizeof(client->worker_id));
uuid_t cid; uuid_t cid;
uuid_generate(cid); uuid_generate(cid);
bin_to_hex((const char*)cid, sizeof(uuid_t), client->client_id); bin_to_hex((const char*)cid, sizeof(uuid_t), client->client_id, 32);
char status[256]; char status[256];
snprintf(status, 256, "Logged in: %s %s\n", worker_id, address); snprintf(status, 256, "Logged in: %s %s\n", worker_id, address);
client_send_job(client, true); client_send_job(client, true);
@ -2051,7 +2051,7 @@ client_on_submit(json_object *message, client_t *client)
/* Yay! Mined a block so submit to network */ /* Yay! Mined a block so submit to network */
log_info("+++ MINED A BLOCK +++"); log_info("+++ MINED A BLOCK +++");
char *block_hex = calloc((bin_size << 1)+1, sizeof(char)); char *block_hex = calloc((bin_size << 1)+1, sizeof(char));
bin_to_hex(block, bin_size, block_hex); bin_to_hex(block, bin_size, block_hex, bin_size << 1);
char body[RPC_BODY_MAX]; char body[RPC_BODY_MAX];
snprintf(body, RPC_BODY_MAX, snprintf(body, RPC_BODY_MAX,
"{\"jsonrpc\":\"2.0\",\"id\":\"0\",\"method\":\"submit_block\", \"params\":[\"%s\"]}", "{\"jsonrpc\":\"2.0\",\"id\":\"0\",\"method\":\"submit_block\", \"params\":[\"%s\"]}",
@ -2063,7 +2063,7 @@ client_on_submit(json_object *message, client_t *client)
block_t* b = (block_t*)callback->data; block_t* b = (block_t*)callback->data;
b->height = bt->height; b->height = bt->height;
bin_to_hex(submitted_hash, 32, b->hash); bin_to_hex(submitted_hash, 32, b->hash, 64);
memcpy(b->prev_hash, bt->prev_hash, 64); memcpy(b->prev_hash, bt->prev_hash, 64);
b->difficulty = bt->difficulty; b->difficulty = bt->difficulty;
b->status = BLOCK_LOCKED; b->status = BLOCK_LOCKED;

View file

@ -48,12 +48,9 @@ is_hex_string(const char *str)
const char *cp = str; const char *cp = str;
while (*cp) while (*cp)
{ {
if (!isxdigit(*cp)) if (!isxdigit(*cp++))
{
return -2; return -2;
} }
cp++;
}
return 0; return 0;
} }
@ -64,7 +61,8 @@ hex_to_bin(const char *hex, char *bin, size_t bin_size)
assert(len % 2 == 0); assert(len % 2 == 0);
assert(bin_size >= len / 2); assert(bin_size >= len / 2);
const char *ph = hex; const char *ph = hex;
while (*ph) char *end = bin + bin_size;
while (*ph && bin < end)
{ {
sscanf(ph, "%2hhx", bin++); sscanf(ph, "%2hhx", bin++);
ph += 2; ph += 2;
@ -72,8 +70,9 @@ hex_to_bin(const char *hex, char *bin, size_t bin_size)
} }
void void
bin_to_hex(const char *bin, size_t bin_size, char *hex) bin_to_hex(const char *bin, size_t bin_size, char *hex, size_t hex_size)
{ {
assert(bin_size << 1 == hex_size);
const char *hex_chars = "0123456789abcdef"; const char *hex_chars = "0123456789abcdef";
char *ph = hex; char *ph = hex;
const char *pb = bin; const char *pb = bin;

View file

@ -35,7 +35,7 @@
int is_hex_string(const char *str); int is_hex_string(const char *str);
void hex_to_bin(const char *hex, char *bin, size_t bin_size); void hex_to_bin(const char *hex, char *bin, size_t bin_size);
void bin_to_hex(const char *bin, size_t bin_size, char *hex); void bin_to_hex(const char *bin, size_t bin_size, char *hex, size_t hex_size);
void reverse_bin(char *bin, size_t len); void reverse_bin(char *bin, size_t len);
char *stecpy(char *dst, const char *src, const char *end); char *stecpy(char *dst, const char *src, const char *end);