diff options
author | Jeff Garzik <jeff@garzik.org> | 2010-11-23 23:33:20 -0500 |
---|---|---|
committer | Jeff Garzik <jgarzik@redhat.com> | 2010-11-23 23:33:20 -0500 |
commit | 9599867d8b9ddd909ea3dc37679b34cab5de5674 (patch) | |
tree | 2ebcd04fc081f56aa4b2364c8ad1341a000e9807 /cpu-miner.c |
Initial revision of CPU miner.
NOTE: non-functional (incorrectly thinks it finds solution, too quickly)
Diffstat (limited to 'cpu-miner.c')
-rw-r--r-- | cpu-miner.c | 391 |
1 files changed, 391 insertions, 0 deletions
diff --git a/cpu-miner.c b/cpu-miner.c new file mode 100644 index 0000000..cb61ae2 --- /dev/null +++ b/cpu-miner.c @@ -0,0 +1,391 @@ + +#define _GNU_SOURCE +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <stdbool.h> +#include <unistd.h> +#include <jansson.h> +#include <curl/curl.h> +#include <openssl/bn.h> + +#include "sha256_generic.c" + +static const bool opt_verbose = true; +static const bool opt_debug = true; + +struct data_buffer { + void *buf; + size_t len; +}; + +struct upload_buffer { + const void *buf; + size_t len; +}; + +struct work { + unsigned char midstate[32]; + unsigned char data[128]; + unsigned char hash[64]; + unsigned char hash1[64]; + BIGNUM *target; +}; + +static void databuf_free(struct data_buffer *db) +{ + if (!db) + return; + + free(db->buf); + + memset(db, 0, sizeof(*db)); +} + +static size_t all_data_cb(const void *ptr, size_t size, size_t nmemb, + void *user_data) +{ + struct data_buffer *db = user_data; + size_t len = size * nmemb; + size_t oldlen, newlen; + void *newmem; + static const unsigned char zero; + + if (opt_debug) + fprintf(stderr, "DBG(%s): %p, %lu, %lu, %p\n", + __func__, ptr, (unsigned long) size, + (unsigned long) nmemb, user_data); + + oldlen = db->len; + newlen = oldlen + len; + + newmem = realloc(db->buf, newlen + 1); + if (!newmem) + return 0; + + db->buf = newmem; + db->len = newlen; + memcpy(db->buf + oldlen, ptr, len); + memcpy(db->buf + newlen, &zero, 1); /* null terminate */ + + return len; +} + +static size_t upload_data_cb(void *ptr, size_t size, size_t nmemb, + void *user_data) +{ + struct upload_buffer *ub = user_data; + int len = size * nmemb; + + if (opt_debug) + fprintf(stderr, "DBG(%s): %p, %lu, %lu, %p\n", + __func__, ptr, (unsigned long) size, + (unsigned long) nmemb, user_data); + + if (len > ub->len) + len = ub->len; + + if (len) { + memcpy(ptr, ub->buf, len); + ub->buf += len; + ub->len -= len; + } + + return len; +} + +static json_t *json_rpc_call(const char *url, const char *userpass, + const char *rpc_req) +{ + CURL *curl; + json_t *val; + int rc; + struct data_buffer all_data = { }; + struct upload_buffer upload_data; + json_error_t err = { }; + struct curl_slist *headers = NULL; + char len_hdr[64]; + + curl = curl_easy_init(); + if (!curl) + return NULL; + + if (opt_verbose) + curl_easy_setopt(curl, CURLOPT_VERBOSE, 1); + curl_easy_setopt(curl, CURLOPT_URL, url); + curl_easy_setopt(curl, CURLOPT_ENCODING, ""); + curl_easy_setopt(curl, CURLOPT_FAILONERROR, 1); + curl_easy_setopt(curl, CURLOPT_TCP_NODELAY, 1); + curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, all_data_cb); + curl_easy_setopt(curl, CURLOPT_WRITEDATA, &all_data); + curl_easy_setopt(curl, CURLOPT_READFUNCTION, upload_data_cb); + curl_easy_setopt(curl, CURLOPT_READDATA, &upload_data); + if (userpass) { + curl_easy_setopt(curl, CURLOPT_USERPWD, userpass); + curl_easy_setopt(curl, CURLOPT_HTTPAUTH, CURLAUTH_BASIC); + } + curl_easy_setopt(curl, CURLOPT_POST, 1); + + upload_data.buf = rpc_req; + upload_data.len = strlen(rpc_req); + sprintf(len_hdr, "Content-Length: %lu", + (unsigned long) upload_data.len); + + headers = curl_slist_append(headers, + "Content-type: application/json"); + headers = curl_slist_append(headers, len_hdr); + headers = curl_slist_append(headers, "Expect:"); /* disable Expect hdr*/ + + curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers); + + rc = curl_easy_perform(curl); + if (rc) + goto err_out; + + if (opt_debug) + printf("====\nSERVER RETURNS:\n%s\n====\n", + (char *) all_data.buf); + + val = json_loads(all_data.buf, &err); + if (!val) { + fprintf(stderr, "JSON failed(%d): %s\n", err.line, err.text); + goto err_out; + } + + databuf_free(&all_data); + curl_slist_free_all(headers); + curl_easy_cleanup(curl); + return val; + +err_out: + databuf_free(&all_data); + curl_slist_free_all(headers); + curl_easy_cleanup(curl); + return NULL; +} + +static bool hex2bin(unsigned char *p, const char *hexstr, size_t len) +{ + while (*hexstr && len) { + char hex_byte[3]; + unsigned int v; + + if (!hexstr[1]) { + fprintf(stderr, "hex2bin str truncated\n"); + return false; + } + + hex_byte[0] = hexstr[0]; + hex_byte[1] = hexstr[1]; + hex_byte[2] = 0; + + if (sscanf(hex_byte, "%x", &v) != 1) { + fprintf(stderr, "hex2bin sscanf '%s' failed\n", + hex_byte); + return false; + } + + *p = (unsigned char) v; + + p++; + hexstr += 2; + len--; + } + + return (len == 0 && *hexstr == 0) ? true : false; +} + +static bool jobj_binary(const json_t *obj, const char *key, + void *buf, size_t buflen) +{ + const char *hexstr; + json_t *tmp; + + tmp = json_object_get(obj, key); + if (!tmp) { + fprintf(stderr, "JSON key '%s' not found\n", key); + return false; + } + hexstr = json_string_value(tmp); + if (!hexstr) { + fprintf(stderr, "JSON key '%s' is not a string\n", key); + return false; + } + if (!hex2bin(buf, hexstr, buflen)) + return false; + + return true; +} + +static void work_free(struct work *work) +{ + if (!work) + return; + + if (work->target) + BN_free(work->target); + + free(work); +} + +static struct work *work_decode(const json_t *val) +{ + struct work *work; + + work = calloc(1, sizeof(*work)); + if (!work) + return NULL; + + if (!jobj_binary(val, "midstate", + work->midstate, sizeof(work->midstate))) { + fprintf(stderr, "JSON inval midstate\n"); + goto err_out; + } + + if (!jobj_binary(val, "data", work->data, sizeof(work->data))) { + fprintf(stderr, "JSON inval data\n"); + goto err_out; + } + + if (!jobj_binary(val, "hash1", work->hash1, sizeof(work->hash1))) { + fprintf(stderr, "JSON inval hash1\n"); + goto err_out; + } + + if (!BN_hex2bn(&work->target, + json_string_value(json_object_get(val, "target")))) { + fprintf(stderr, "JSON inval target\n"); + goto err_out; + } + + return work; + +err_out: + work_free(work); + return NULL; +} + +static void runhash(void *state, void *input, const void *init) +{ + memcpy(state, init, 32); + sha256_transform(state, input); +} + +static const uint32_t init_state[8] = { + 0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, + 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19 +}; + +static uint32_t scanhash(unsigned char *midstate, unsigned char *data, + unsigned char *hash1, unsigned char *hash) +{ + uint32_t *nonce = (uint32_t *)(data + 12); + uint32_t n; + + while (1) { + n = *nonce; + n++; + *nonce = n; + + runhash(hash1, data, midstate); + runhash(hash, hash1, init_state); + + if (((uint16_t *)hash)[14] == 0) { + if (opt_debug) + fprintf(stderr, "DBG: found zeroes in hash\n"); + return n; + } + + if ((n & 0xffffff) == 0) { + if (opt_debug) + fprintf(stderr, "DBG: end of nonce range\n"); + return 0; + } + } +} + +static const char *url = "http://127.0.0.1:8332/"; +static const char *userpass = "pretzel:smooth"; + +static void submit_work(struct work *work) +{ + char hexstr[256 + 1], *s; + int i; + json_t *val; + + printf("PROOF OF WORK FOUND\n"); + + for (i = 0; i < sizeof(work->data); i++) + sprintf(hexstr + (i * 2), "%02x", work->data[i]); + + if (asprintf(&s, + "{\"method\": \"getwork\", \"params\": [ \"%s\" ], \"id\":1}\r\n", + hexstr) < 0) { + fprintf(stderr, "asprintf failed\n"); + return; + } + + if (opt_debug) + fprintf(stderr, "DBG: sending RPC call:\n%s", s); + + val = json_rpc_call(url, userpass, s); + if (!val) { + fprintf(stderr, "submit_work json_rpc_call failed\n"); + return; + } + + free(s); + json_decref(val); +} + +static int main_loop(void) +{ + static const char *rpc_req = + "{\"method\": \"getwork\", \"params\": [], \"id\":0}\r\n"; + + while (1) { + json_t *val; + struct work *work; + uint32_t nonce; + + val = json_rpc_call(url, userpass, rpc_req); + if (!val) { + fprintf(stderr, "json_rpc_call failed\n"); + return 1; + } + + if (opt_verbose) { + char *s = json_dumps(val, JSON_INDENT(2)); + printf("JSON output:\n%s\n", s); + free(s); + } + + work = work_decode(json_object_get(val, "result")); + if (!work) { + fprintf(stderr, "work decode failed\n"); + return 1; + } + + json_decref(val); + + nonce = scanhash(work->midstate, work->data + 64, + work->hash1, work->hash); + + if (nonce) { + submit_work(work); + + fprintf(stderr, "sleeping, after proof-of-work...\n"); + sleep(20); + } + + work_free(work); + } + + return 0; +} + +int main (int argc, char *argv[]) +{ + return main_loop(); +} + |