mirror of https://github.com/GOSTSec/vanitygen
samr7
12 years ago
11 changed files with 3634 additions and 2582 deletions
@ -0,0 +1,35 @@ |
|||||||
|
/*
|
||||||
|
* Vanitygen, vanity bitcoin address generator |
||||||
|
* Copyright (C) 2011 <samr7@cs.washington.edu> |
||||||
|
* |
||||||
|
* Vanitygen is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU Affero General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* any later version. |
||||||
|
* |
||||||
|
* Vanitygen 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 Affero General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU Affero General Public License |
||||||
|
* along with Vanitygen. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
|
||||||
|
#if !defined (__VG_OCLENGINE_H__) |
||||||
|
#define __VG_OCLENGINE_H__ |
||||||
|
|
||||||
|
#include "pattern.h" |
||||||
|
|
||||||
|
typedef struct _vg_ocl_context_s vg_ocl_context_t; |
||||||
|
|
||||||
|
extern vg_ocl_context_t *vg_ocl_context_new( |
||||||
|
vg_context_t *vcp, int platformidx, int deviceidx, |
||||||
|
int safe_mode, int verify, |
||||||
|
int worksize, int nthreads, int nrows, int ncols, |
||||||
|
int invsize); |
||||||
|
extern void vg_ocl_context_free(vg_ocl_context_t *vocp); |
||||||
|
|
||||||
|
extern void *vg_opencl_loop(void *vocp); |
||||||
|
|
||||||
|
#endif /* !defined (__VG_OCLENGINE_H__) */ |
||||||
@ -0,0 +1,748 @@ |
|||||||
|
/*
|
||||||
|
* Vanitygen, vanity bitcoin address generator |
||||||
|
* Copyright (C) 2011 <samr7@cs.washington.edu> |
||||||
|
* |
||||||
|
* Vanitygen is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU Affero General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* any later version. |
||||||
|
* |
||||||
|
* Vanitygen 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 Affero General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU Affero General Public License |
||||||
|
* along with Vanitygen. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
|
||||||
|
#include <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
#include <math.h> |
||||||
|
#include <assert.h> |
||||||
|
|
||||||
|
#include <pthread.h> |
||||||
|
|
||||||
|
#include <openssl/ec.h> |
||||||
|
#include <openssl/bn.h> |
||||||
|
#include <openssl/rand.h> |
||||||
|
|
||||||
|
#include <curl/curl.h> |
||||||
|
|
||||||
|
#include "oclengine.h" |
||||||
|
#include "pattern.h" |
||||||
|
#include "util.h" |
||||||
|
|
||||||
|
|
||||||
|
const char *version = "0.17"; |
||||||
|
const int debug = 0; |
||||||
|
|
||||||
|
|
||||||
|
typedef struct workitem_s { |
||||||
|
const char *pattern; |
||||||
|
const char *comment; |
||||||
|
EC_POINT *pubkey; |
||||||
|
int addrtype; |
||||||
|
double difficulty; |
||||||
|
double reward; |
||||||
|
double value; |
||||||
|
} workitem_t; |
||||||
|
|
||||||
|
typedef struct server_request_s { |
||||||
|
int request_status; |
||||||
|
const EC_GROUP *group; |
||||||
|
|
||||||
|
char *part_buf; |
||||||
|
size_t part_off; |
||||||
|
size_t part_end; |
||||||
|
size_t part_size; |
||||||
|
|
||||||
|
workitem_t **items; |
||||||
|
int nitems; |
||||||
|
int nalloc; |
||||||
|
} server_request_t; |
||||||
|
|
||||||
|
void |
||||||
|
server_workitem_free(workitem_t *wip) |
||||||
|
{ |
||||||
|
if (wip->pubkey) |
||||||
|
EC_POINT_free(wip->pubkey); |
||||||
|
free(wip); |
||||||
|
} |
||||||
|
|
||||||
|
static workitem_t * |
||||||
|
server_workitem_new(server_request_t *reqp, |
||||||
|
const char *pfx, const char *pubkey_s, |
||||||
|
const char *addrtype_s, const char *reward_s, |
||||||
|
const char *comment) |
||||||
|
{ |
||||||
|
workitem_t *wip; |
||||||
|
EC_POINT *pubkey; |
||||||
|
int addrtype; |
||||||
|
double reward; |
||||||
|
double difficulty; |
||||||
|
|
||||||
|
addrtype = atoi(addrtype_s); |
||||||
|
if ((addrtype < 0) || (addrtype > 255)) |
||||||
|
return NULL; |
||||||
|
|
||||||
|
reward = strtod(reward_s, NULL); |
||||||
|
if (reward < 0.0) |
||||||
|
return NULL; |
||||||
|
|
||||||
|
difficulty = vg_prefix_get_difficulty(addrtype, pfx); |
||||||
|
if (difficulty == 0.0) |
||||||
|
return NULL; |
||||||
|
|
||||||
|
pubkey = EC_POINT_hex2point(reqp->group, pubkey_s, NULL, NULL); |
||||||
|
if (pubkey == NULL) |
||||||
|
return NULL; |
||||||
|
|
||||||
|
|
||||||
|
wip = (workitem_t *) malloc(sizeof(*wip) + |
||||||
|
strlen(pfx) + |
||||||
|
strlen(comment) + 2); |
||||||
|
memset(wip, 0, sizeof(*wip)); |
||||||
|
wip->pattern = (char *) (wip + 1); |
||||||
|
strcpy((char *)wip->pattern, pfx); |
||||||
|
wip->comment = wip->pattern + (strlen(wip->pattern) + 1); |
||||||
|
strcpy((char *) wip->comment, comment); |
||||||
|
wip->pubkey = pubkey; |
||||||
|
wip->addrtype = addrtype; |
||||||
|
wip->difficulty = difficulty; |
||||||
|
wip->reward = reward; |
||||||
|
wip->value = (reward * 1000000.0 * 3600.0) / difficulty; |
||||||
|
|
||||||
|
return wip; |
||||||
|
} |
||||||
|
|
||||||
|
static int |
||||||
|
server_workitem_ptr_comp(const void *pa, const void *pb) |
||||||
|
{ |
||||||
|
workitem_t *a = *(workitem_t **) pa, *b = *(workitem_t **) pb; |
||||||
|
return (a->value > b->value) ? -1 : ((a->value < b->value) ? 1 : 0); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
typedef struct server_context_s { |
||||||
|
EC_KEY *dummy_key; |
||||||
|
const char *url; |
||||||
|
const char *credit_addr; |
||||||
|
char *getwork; |
||||||
|
char *submit; |
||||||
|
int verbose; |
||||||
|
} server_context_t; |
||||||
|
|
||||||
|
static int |
||||||
|
server_workitem_equal(server_context_t *ctxp, workitem_t *a, workitem_t *b) |
||||||
|
{ |
||||||
|
if (strcmp(a->pattern, b->pattern)) |
||||||
|
return 0; |
||||||
|
if (EC_POINT_cmp(EC_KEY_get0_group(ctxp->dummy_key), |
||||||
|
a->pubkey, b->pubkey, NULL)) |
||||||
|
return 0; |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
void |
||||||
|
server_context_free(server_context_t *ctxp) |
||||||
|
{ |
||||||
|
if (ctxp->dummy_key) |
||||||
|
EC_KEY_free(ctxp->dummy_key); |
||||||
|
if (ctxp->getwork) |
||||||
|
free(ctxp->getwork); |
||||||
|
if (ctxp->submit) |
||||||
|
free(ctxp->submit); |
||||||
|
free(ctxp); |
||||||
|
} |
||||||
|
|
||||||
|
server_context_t * |
||||||
|
server_context_new(const char *url, const char *credit_addr) |
||||||
|
{ |
||||||
|
server_context_t *ctxp; |
||||||
|
int urllen = strlen(url); |
||||||
|
int addrlen = strlen(credit_addr); |
||||||
|
ctxp = (server_context_t *) |
||||||
|
malloc(sizeof(*ctxp) + urllen + addrlen + 2); |
||||||
|
memset(ctxp, 0, sizeof(*ctxp)); |
||||||
|
ctxp->url = (const char *) (ctxp + 1); |
||||||
|
ctxp->credit_addr = (const char *) (ctxp->url + urllen + 1); |
||||||
|
strcpy((char *) ctxp->url, url); |
||||||
|
strcpy((char *) ctxp->credit_addr, credit_addr); |
||||||
|
|
||||||
|
ctxp->dummy_key = vg_exec_context_new_key(); |
||||||
|
ctxp->getwork = (char *) malloc(urllen + 9); |
||||||
|
ctxp->submit = (char *) malloc(urllen + 7); |
||||||
|
if (url[urllen - 1] == '/') { |
||||||
|
snprintf(ctxp->getwork, urllen + 9, "%sgetWork", url); |
||||||
|
snprintf(ctxp->submit, urllen + 7, "%ssolve", url); |
||||||
|
} else { |
||||||
|
snprintf(ctxp->getwork, urllen + 9, "%s/getWork", url); |
||||||
|
snprintf(ctxp->submit, urllen + 7, "%s/solve", url); |
||||||
|
} |
||||||
|
|
||||||
|
return ctxp; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
int |
||||||
|
server_workitem_add(server_request_t *reqp, workitem_t *wip) |
||||||
|
{ |
||||||
|
int nalloc; |
||||||
|
|
||||||
|
if ((reqp->nitems + 1) >= reqp->nalloc) { |
||||||
|
nalloc = reqp->nalloc * 2; |
||||||
|
if (nalloc == 0) |
||||||
|
nalloc = 16; |
||||||
|
if (nalloc > 65536) |
||||||
|
return -1; |
||||||
|
reqp->items = (workitem_t **) |
||||||
|
realloc(reqp->items, nalloc * sizeof(*reqp->items)); |
||||||
|
if (reqp->items == NULL) |
||||||
|
return -1; |
||||||
|
reqp->nalloc = nalloc; |
||||||
|
} |
||||||
|
reqp->items[reqp->nitems++] = wip; |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
static int |
||||||
|
server_body_reader(const char *buf, size_t elemsize, size_t len, void *param) |
||||||
|
{ |
||||||
|
server_request_t *reqp = (server_request_t *) param; |
||||||
|
char *line, *sep, *pfx, *pubkey_s, *addrtype_s, *reward_s, *comment; |
||||||
|
workitem_t *wip; |
||||||
|
|
||||||
|
if (!len) |
||||||
|
return 0; |
||||||
|
|
||||||
|
if ((reqp->part_size < (reqp->part_end + len)) && |
||||||
|
(reqp->part_off > 0)) { |
||||||
|
memmove(reqp->part_buf, |
||||||
|
reqp->part_buf + reqp->part_off, |
||||||
|
reqp->part_end - reqp->part_off); |
||||||
|
reqp->part_end -= reqp->part_off; |
||||||
|
reqp->part_off = 0; |
||||||
|
} |
||||||
|
|
||||||
|
if (reqp->part_size < (reqp->part_end + len)) { |
||||||
|
if (reqp->part_size == 0) |
||||||
|
reqp->part_size = 4096; |
||||||
|
while (reqp->part_size < (reqp->part_end + len)) { |
||||||
|
reqp->part_size *= 2; |
||||||
|
if (reqp->part_size > (1024*1024)) { |
||||||
|
fprintf(stderr, "Line too long from server"); |
||||||
|
reqp->request_status = 0; |
||||||
|
return -1; |
||||||
|
} |
||||||
|
} |
||||||
|
reqp->part_buf = (char *) realloc(reqp->part_buf, |
||||||
|
reqp->part_size); |
||||||
|
if (!reqp->part_buf) { |
||||||
|
fprintf(stderr, "Out of memory"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
memcpy(reqp->part_buf + reqp->part_end, buf, len); |
||||||
|
reqp->part_end += len; |
||||||
|
|
||||||
|
line = reqp->part_buf; |
||||||
|
while (1) { |
||||||
|
sep = strchr(line, '\n'); |
||||||
|
if (!sep) |
||||||
|
break; |
||||||
|
pfx = line; |
||||||
|
*sep = '\0'; |
||||||
|
line = sep + 1; |
||||||
|
sep = strchr(pfx, ':'); |
||||||
|
if (!sep) |
||||||
|
goto bad_line; |
||||||
|
*sep = '\0'; sep += 1; |
||||||
|
pubkey_s = sep; |
||||||
|
sep = strchr(sep, ':'); |
||||||
|
if (!sep) |
||||||
|
goto bad_line; |
||||||
|
*sep = '\0'; sep += 1; |
||||||
|
addrtype_s = sep; |
||||||
|
sep = strchr(sep, ':'); |
||||||
|
if (!sep) |
||||||
|
goto bad_line; |
||||||
|
*sep = '\0'; sep += 1; |
||||||
|
reward_s = sep; |
||||||
|
sep = strchr(sep, ';'); |
||||||
|
if (!sep) |
||||||
|
goto bad_line; |
||||||
|
*sep = '\0'; sep += 1; |
||||||
|
comment = sep; |
||||||
|
|
||||||
|
wip = server_workitem_new(reqp, pfx, pubkey_s, addrtype_s, |
||||||
|
reward_s, comment); |
||||||
|
if (!wip) |
||||||
|
goto bad_line; |
||||||
|
if (server_workitem_add(reqp, wip)) { |
||||||
|
server_workitem_free(wip); |
||||||
|
goto bad_line; |
||||||
|
} |
||||||
|
continue; |
||||||
|
|
||||||
|
bad_line: |
||||||
|
; |
||||||
|
} |
||||||
|
|
||||||
|
reqp->part_off = line - reqp->part_buf; |
||||||
|
if (reqp->part_off == reqp->part_end) { |
||||||
|
reqp->part_off = 0; |
||||||
|
reqp->part_end = 0; |
||||||
|
} |
||||||
|
|
||||||
|
return len; |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
dump_work(workitem_t **workarray) |
||||||
|
{ |
||||||
|
workitem_t *wip; |
||||||
|
int i; |
||||||
|
printf("Available bounties:\n"); |
||||||
|
for (i = 0; workarray[i] != NULL; i++) { |
||||||
|
wip = workarray[i]; |
||||||
|
printf("Pattern: \"%s\" Reward: %f Value: %f BTC/MkeyHr\n", |
||||||
|
wip->pattern, |
||||||
|
wip->reward, |
||||||
|
wip->value); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
free_work_array(workitem_t **workarray, workitem_t *except) |
||||||
|
{ |
||||||
|
int i; |
||||||
|
if (workarray) { |
||||||
|
for (i = 0; workarray[i] != NULL; i++) { |
||||||
|
if (workarray[i] != except) |
||||||
|
server_workitem_free(workarray[i]); |
||||||
|
} |
||||||
|
free(workarray); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
workitem_t ** |
||||||
|
server_context_getwork(server_context_t *ctxp) |
||||||
|
{ |
||||||
|
CURLcode res; |
||||||
|
server_request_t *reqp; |
||||||
|
CURL *creq; |
||||||
|
|
||||||
|
reqp = (server_request_t *) malloc(sizeof(*reqp)); |
||||||
|
memset(reqp, 0, sizeof(*reqp)); |
||||||
|
|
||||||
|
reqp->group = EC_KEY_get0_group(ctxp->dummy_key); |
||||||
|
|
||||||
|
creq = curl_easy_init(); |
||||||
|
if (curl_easy_setopt(creq, CURLOPT_URL, ctxp->getwork) || |
||||||
|
curl_easy_setopt(creq, CURLOPT_VERBOSE, ctxp->verbose > 1) || |
||||||
|
curl_easy_setopt(creq, CURLOPT_WRITEFUNCTION, |
||||||
|
server_body_reader) || |
||||||
|
curl_easy_setopt(creq, CURLOPT_WRITEDATA, reqp)) { |
||||||
|
fprintf(stderr, "Failed to set up libcurl\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
|
||||||
|
res = curl_easy_perform(creq); |
||||||
|
if (res != CURLE_OK) { |
||||||
|
fprintf(stderr, "Get work request failed: %s\n", |
||||||
|
curl_easy_strerror(res)); |
||||||
|
curl_easy_cleanup(creq); |
||||||
|
free_work_array(reqp->items, NULL); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
if (reqp->items) { |
||||||
|
qsort(reqp->items, reqp->nitems, sizeof(*(reqp->items)), |
||||||
|
server_workitem_ptr_comp); |
||||||
|
reqp->items[reqp->nitems] = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
curl_easy_cleanup(creq); |
||||||
|
return reqp->items; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
int |
||||||
|
server_context_submit_solution(server_context_t *ctxp, |
||||||
|
workitem_t *work, |
||||||
|
const char *privkey) |
||||||
|
{ |
||||||
|
char urlbuf[8192]; |
||||||
|
char *pubhex; |
||||||
|
CURL *creq; |
||||||
|
CURLcode res; |
||||||
|
|
||||||
|
pubhex = EC_POINT_point2hex(EC_KEY_get0_group(ctxp->dummy_key), |
||||||
|
work->pubkey, |
||||||
|
POINT_CONVERSION_UNCOMPRESSED, |
||||||
|
NULL); |
||||||
|
snprintf(urlbuf, sizeof(urlbuf), |
||||||
|
"%s?key=%s%%3A%s&privateKey=%s&bitcoinAddress=%s", |
||||||
|
ctxp->submit, |
||||||
|
work->pattern, |
||||||
|
pubhex, |
||||||
|
privkey, |
||||||
|
ctxp->credit_addr); |
||||||
|
OPENSSL_free(pubhex); |
||||||
|
creq = curl_easy_init(); |
||||||
|
if (curl_easy_setopt(creq, CURLOPT_URL, urlbuf) || |
||||||
|
curl_easy_setopt(creq, CURLOPT_VERBOSE, ctxp->verbose > 1) || |
||||||
|
curl_easy_setopt(creq, CURLOPT_POST, 1)) { |
||||||
|
fprintf(stderr, "Failed to set up libcurl\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
|
||||||
|
res = curl_easy_perform(creq); |
||||||
|
if (res != CURLE_OK) { |
||||||
|
fprintf(stderr, "Submission failed: %s\n", |
||||||
|
curl_easy_strerror(res)); |
||||||
|
curl_easy_cleanup(creq); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
curl_easy_cleanup(creq); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static pthread_mutex_t soln_lock; |
||||||
|
static pthread_cond_t soln_cond; |
||||||
|
static char *soln_pattern = NULL; |
||||||
|
static char *soln_private_key = NULL; |
||||||
|
|
||||||
|
void |
||||||
|
free_soln() |
||||||
|
{ |
||||||
|
if (soln_pattern) { |
||||||
|
free(soln_pattern); |
||||||
|
soln_pattern = NULL; |
||||||
|
} |
||||||
|
if (soln_private_key) { |
||||||
|
OPENSSL_free(soln_private_key); |
||||||
|
soln_private_key = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
output_match_work_complete(vg_context_t *vcp, EC_KEY *pkey, const char *pattern) |
||||||
|
{ |
||||||
|
vg_output_match_console(vcp, pkey, pattern); |
||||||
|
pthread_mutex_lock(&soln_lock); |
||||||
|
free_soln(); |
||||||
|
soln_pattern = strdup(pattern); |
||||||
|
soln_private_key = BN_bn2hex(EC_KEY_get0_private_key(pkey)); |
||||||
|
|
||||||
|
/* Signal the generator to stop */ |
||||||
|
vcp->vc_halt = 1; |
||||||
|
|
||||||
|
/* Wake up the main thread, if it's sleeping */ |
||||||
|
pthread_cond_broadcast(&soln_cond); |
||||||
|
pthread_mutex_unlock(&soln_lock); |
||||||
|
} |
||||||
|
|
||||||
|
int |
||||||
|
check_solution(server_context_t *scp, workitem_t *wip) |
||||||
|
{ |
||||||
|
int res = 0; |
||||||
|
pthread_mutex_lock(&soln_lock); |
||||||
|
if (soln_private_key != NULL) { |
||||||
|
assert(!strcmp(soln_pattern, wip->pattern)); |
||||||
|
server_context_submit_solution(scp, wip, soln_private_key); |
||||||
|
free_soln(); |
||||||
|
res = 1; |
||||||
|
} |
||||||
|
pthread_mutex_unlock(&soln_lock); |
||||||
|
return res; |
||||||
|
} |
||||||
|
|
||||||
|
void |
||||||
|
usage(const char *name) |
||||||
|
{ |
||||||
|
fprintf(stderr, |
||||||
|
"oclVanityMiner %s (" OPENSSL_VERSION_TEXT ")\n" |
||||||
|
"Usage: %s -u <URL> -a <credit address>\n" |
||||||
|
"Organized vanity address mining client using OpenCL. Contacts the specified\n" |
||||||
|
"bounty pool server, downloads a list of active bounties, and attempts to\n" |
||||||
|
"generate the address with the best difficulty to reward ratio. Maintains\n" |
||||||
|
"contact with the bounty pool server and periodically refreshes the bounty\n" |
||||||
|
"list.\n" |
||||||
|
"\n" |
||||||
|
"Options:\n" |
||||||
|
"-u <URL> Bounty pool URL\n" |
||||||
|
"-a <address> Credit address for completed work\n" |
||||||
|
"-i <interval> Set server polling interval in seconds (default 90)\n" |
||||||
|
"-v Verbose output\n" |
||||||
|
"-q Quiet output\n" |
||||||
|
"-p <platform> Select OpenCL platform\n" |
||||||
|
"-d <device> Select OpenCL device\n" |
||||||
|
"-S Safe mode, disable OpenCL loop unrolling optimizations\n" |
||||||
|
"-w <worksize> Set work items per thread in a work unit\n" |
||||||
|
"-t <threads> Set target thread count per multiprocessor\n" |
||||||
|
"-g <x>x<y> Set grid size\n" |
||||||
|
"-b <invsize> Set modular inverse ops per thread\n" |
||||||
|
"-V Enable kernel/OpenCL/hardware verification (SLOW)\n", |
||||||
|
version, name); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
int |
||||||
|
main(int argc, char **argv) |
||||||
|
{ |
||||||
|
const char *url = NULL; |
||||||
|
const char *credit_addr = NULL; |
||||||
|
int opt; |
||||||
|
int platformidx = -1, deviceidx = -1; |
||||||
|
char *pend; |
||||||
|
int verbose = 1; |
||||||
|
int interval = 90; |
||||||
|
int nthreads = 0; |
||||||
|
int worksize = 0; |
||||||
|
int nrows = 0, ncols = 0; |
||||||
|
int invsize = 0; |
||||||
|
int verify_mode = 0; |
||||||
|
int safe_mode = 0; |
||||||
|
vg_context_t *vcp = NULL; |
||||||
|
vg_ocl_context_t *vocp = NULL; |
||||||
|
|
||||||
|
int res; |
||||||
|
int thread_started = 0; |
||||||
|
pthread_t thread; |
||||||
|
workitem_t *active_wip = NULL; |
||||||
|
|
||||||
|
server_context_t *scp = NULL; |
||||||
|
workitem_t *wip = NULL, **wipa; |
||||||
|
int wip_index; |
||||||
|
int was_sleeping = 0; |
||||||
|
|
||||||
|
struct timeval tv; |
||||||
|
struct timespec sleepy; |
||||||
|
|
||||||
|
pthread_mutex_init(&soln_lock, NULL); |
||||||
|
pthread_cond_init(&soln_cond, NULL); |
||||||
|
|
||||||
|
if (argc == 1) { |
||||||
|
usage(argv[0]); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
while ((opt = getopt(argc, argv, |
||||||
|
"u:a:vqp:d:w:t:g:b:VSh?i:")) != -1) { |
||||||
|
switch (opt) { |
||||||
|
case 'u': |
||||||
|
url = optarg; |
||||||
|
break; |
||||||
|
case 'a': |
||||||
|
credit_addr = optarg; |
||||||
|
break; |
||||||
|
case 'v': |
||||||
|
verbose = 2; |
||||||
|
break; |
||||||
|
case 'q': |
||||||
|
verbose = 0; |
||||||
|
break; |
||||||
|
case 'i': |
||||||
|
interval = atoi(optarg); |
||||||
|
if (interval < 60) { |
||||||
|
fprintf(stderr, |
||||||
|
"Invalid interval '%s'\n", optarg); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
case 'p': |
||||||
|
platformidx = atoi(optarg); |
||||||
|
break; |
||||||
|
case 'd': |
||||||
|
deviceidx = atoi(optarg); |
||||||
|
break; |
||||||
|
case 'w': |
||||||
|
worksize = atoi(optarg); |
||||||
|
if (worksize == 0) { |
||||||
|
fprintf(stderr, |
||||||
|
"Invalid work size '%s'\n", optarg); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
break; |
||||||
|
case 't': |
||||||
|
nthreads = atoi(optarg); |
||||||
|
if (nthreads == 0) { |
||||||
|
fprintf(stderr, |
||||||
|
"Invalid thread count '%s'\n", optarg); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
break; |
||||||
|
case 'g': |
||||||
|
nrows = 0; |
||||||
|
ncols = strtol(optarg, &pend, 0); |
||||||
|
if (pend && *pend == 'x') { |
||||||
|
nrows = strtol(pend+1, NULL, 0); |
||||||
|
} |
||||||
|
if (!nrows || !ncols) { |
||||||
|
fprintf(stderr, |
||||||
|
"Invalid grid size '%s'\n", optarg); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
break; |
||||||
|
case 'b': |
||||||
|
invsize = atoi(optarg); |
||||||
|
if (!invsize) { |
||||||
|
fprintf(stderr, |
||||||
|
"Invalid modular inverse size '%s'\n", |
||||||
|
optarg); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
if (invsize & (invsize - 1)) { |
||||||
|
fprintf(stderr, |
||||||
|
"Modular inverse size must be " |
||||||
|
"a power of 2\n"); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
break; |
||||||
|
case 'V': |
||||||
|
verify_mode = 1; |
||||||
|
break; |
||||||
|
case 'S': |
||||||
|
safe_mode = 1; |
||||||
|
break; |
||||||
|
default: |
||||||
|
usage(argv[0]); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#if OPENSSL_VERSION_NUMBER < 0x10000000L |
||||||
|
/* Complain about older versions of OpenSSL */ |
||||||
|
if (verbose > 0) { |
||||||
|
fprintf(stderr, |
||||||
|
"WARNING: Built with " OPENSSL_VERSION_TEXT "\n" |
||||||
|
"WARNING: Use OpenSSL 1.0.0d+ for best performance\n"); |
||||||
|
} |
||||||
|
#endif |
||||||
|
curl_easy_init(); |
||||||
|
|
||||||
|
vcp = vg_prefix_context_new(0, 128, 0); |
||||||
|
|
||||||
|
vcp->vc_verbose = verbose; |
||||||
|
|
||||||
|
vcp->vc_output_match = output_match_work_complete; |
||||||
|
vcp->vc_output_timing = vg_output_timing_console; |
||||||
|
|
||||||
|
|
||||||
|
if (!url) { |
||||||
|
fprintf(stderr, "ERROR: No server URL specified\n"); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
if (!credit_addr) { |
||||||
|
fprintf(stderr, "ERROR: No reward address specified\n"); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
if (!vg_b58_decode_check(credit_addr, NULL, 0)) { |
||||||
|
fprintf(stderr, "ERROR: Invalid reward address specified\n"); |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
scp = server_context_new(url, credit_addr); |
||||||
|
scp->verbose = verbose; |
||||||
|
wipa = NULL; |
||||||
|
|
||||||
|
while (1) { |
||||||
|
if (!wipa) { |
||||||
|
wipa = server_context_getwork(scp); |
||||||
|
wip_index = 0; |
||||||
|
} |
||||||
|
|
||||||
|
if (wipa) { |
||||||
|
wip = wipa[wip_index]; |
||||||
|
if (wip) |
||||||
|
wip_index += 1; |
||||||
|
} else |
||||||
|
wip = NULL; |
||||||
|
|
||||||
|
/* If the work item is the same as the one we're executing,
|
||||||
|
keep it */ |
||||||
|
if (wip && active_wip && |
||||||
|
server_workitem_equal(scp, active_wip, wip)) |
||||||
|
wip = active_wip; |
||||||
|
|
||||||
|
if (thread_started && (!active_wip || (wip != active_wip))) { |
||||||
|
/* If a thread is running, stop it */ |
||||||
|
vcp->vc_halt = 1; |
||||||
|
pthread_join(thread, NULL); |
||||||
|
thread_started = 0; |
||||||
|
vcp->vc_halt = 0; |
||||||
|
if (active_wip) { |
||||||
|
check_solution(scp, active_wip); |
||||||
|
active_wip = NULL; |
||||||
|
} |
||||||
|
vg_context_clear_all_patterns(vcp); |
||||||
|
} |
||||||
|
|
||||||
|
if (!wip) { |
||||||
|
if (!was_sleeping) { |
||||||
|
fprintf(stderr, |
||||||
|
"No work available, sleeping\n"); |
||||||
|
was_sleeping = 1; |
||||||
|
} |
||||||
|
|
||||||
|
} else if (!active_wip) { |
||||||
|
was_sleeping = 0; |
||||||
|
fprintf(stderr, |
||||||
|
"Searching for pattern: \"%s\" " |
||||||
|
"Reward: %f Value: %f BTC/MkeyHr\n", |
||||||
|
wip->pattern, |
||||||
|
wip->reward, |
||||||
|
wip->value); |
||||||
|
vcp->vc_addrtype = wip->addrtype; |
||||||
|
vcp->vc_pubkey_base = wip->pubkey; |
||||||
|
if (!vg_context_add_patterns(vcp, &wip->pattern, 1)) |
||||||
|
return 1; |
||||||
|
assert(vcp->vc_npatterns); |
||||||
|
|
||||||
|
if (!vocp) { |
||||||
|
vocp = vg_ocl_context_new(vcp, |
||||||
|
platformidx, deviceidx, |
||||||
|
safe_mode, verify_mode, |
||||||
|
worksize, nthreads, nrows, |
||||||
|
ncols, invsize); |
||||||
|
if (!vocp) |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
res = pthread_create(&thread, NULL, |
||||||
|
vg_opencl_loop, vocp); |
||||||
|
thread_started = 1; |
||||||
|
active_wip = wip; |
||||||
|
} |
||||||
|
|
||||||
|
/* Wait for something to happen */ |
||||||
|
gettimeofday(&tv, NULL); |
||||||
|
sleepy.tv_sec = tv.tv_sec; |
||||||
|
sleepy.tv_nsec = tv.tv_usec * 1000; |
||||||
|
sleepy.tv_sec += interval; |
||||||
|
|
||||||
|
pthread_mutex_lock(&soln_lock); |
||||||
|
if (!soln_private_key) |
||||||
|
res = pthread_cond_timedwait(&soln_cond, |
||||||
|
&soln_lock, &sleepy); |
||||||
|
pthread_mutex_unlock(&soln_lock); |
||||||
|
|
||||||
|
if (res == 0) { |
||||||
|
if (check_solution(scp, active_wip)) |
||||||
|
active_wip = NULL; |
||||||
|
} |
||||||
|
else if (res == ETIMEDOUT) { |
||||||
|
if (wipa) { |
||||||
|
free_work_array(wipa, active_wip); |
||||||
|
wipa = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
Loading…
Reference in new issue