Initial libuv support build.

src/net/Client.cpp

@@ -0,0 +1,311 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include <jansson.h>
+
+
+#include "Console.h"
+#include "interfaces/IClientListener.h"
+#include "net/Client.h"
+#include "net/Url.h"
+
+
+Client::Client(IClientListener *listener) :
+    m_host(nullptr),
+    m_listener(listener),
+    m_recvBufPos(0),
+    m_state(UnconnectedState),
+    m_port(0),
+    m_stream(nullptr),
+    m_socket(nullptr)
+{
+    m_resolver.data = this;
+
+    m_hints.ai_family   = PF_INET;
+    m_hints.ai_socktype = SOCK_STREAM;
+    m_hints.ai_protocol = IPPROTO_TCP;
+    m_hints.ai_flags    = 0;
+
+    m_recvBuf.base = static_cast<char*>(malloc(kRecvBufSize));
+    m_recvBuf.len  = kRecvBufSize;
+}
+
+
+Client::~Client()
+{
+    free(m_recvBuf.base);
+    free(m_socket);
+    free(m_host);
+}
+
+
+/**
+ * @brief Connect to server.
+ *
+ * @param host
+ * @param port
+ */
+void Client::connect(const char *host, uint16_t port)
+{
+    m_host = strdup(host);
+    m_port = port;
+
+    LOG_DEBUG("[%s:%u] connect", m_host, m_port);
+
+    resolve(host);
+}
+
+
+/**
+ * @brief Connect to server.
+ *
+ * @param url
+ */
+void Client::connect(const Url *url)
+{
+    connect(url->host(), url->port());
+}
+
+
+void Client::disconnect()
+{
+    close();
+}
+
+
+void Client::login(const char *user, const char *pass, const char *agent)
+{
+    const size_t size = 96 + strlen(user) + strlen(pass) + strlen(agent);
+    char *req = static_cast<char*>(malloc(size));
+    snprintf(req, size, "{\"method\":\"login\",\"params\":{\"login\":\"%s\",\"pass\":\"%s\",\"agent\":\"%s\"},\"id\":1}\n", user, pass, agent);
+
+    send(req);
+}
+
+
+/**
+ * @brief Send raw data to server.
+ *
+ * @param data
+ */
+void Client::send(char *data)
+{
+    LOG_DEBUG("[%s:%u] send (%d bytes): \"%s\"", m_host, m_port, strlen(data), data);
+    if (state() != ConnectedState) {
+        LOG_DEBUG_ERR("[%s:%u] send failed, invalid state: %d", m_host, m_port, m_state);
+        return;
+    }
+
+    uv_buf_t buf = uv_buf_init(data, strlen(data));
+
+    uv_write_t *req = static_cast<uv_write_t*>(malloc(sizeof(uv_write_t)));
+    req->data = buf.base;
+
+    uv_write(req, m_stream, &buf, 1, [](uv_write_t *req, int status) {
+        if (status) {
+            auto client = getClient(req->data);
+            LOG_ERR("[%s:%u] write error: \"%s\"", client->m_host, client->m_port, uv_strerror(status));
+        }
+
+        free(req->data);
+        free(req);
+    });
+}
+
+
+int Client::resolve(const char *host)
+{
+    setState(HostLookupState);
+
+    m_recvBufPos = 0;
+
+    const int r = uv_getaddrinfo(uv_default_loop(), &m_resolver, Client::onResolved, host, NULL, &m_hints);
+    if (r) {
+        LOG_ERR("[%s:%u] getaddrinfo error: \"%s\"", host, m_port, uv_strerror(r));
+        return 1;
+    }
+
+    return 0;
+}
+
+
+void Client::close()
+{
+    if (m_state == UnconnectedState || m_state == ClosingState || !m_socket) {
+        return;
+    }
+
+    setState(ClosingState);
+    uv_close(reinterpret_cast<uv_handle_t*>(m_socket), Client::onClose);
+}
+
+
+void Client::connect(struct sockaddr *addr)
+{
+    setState(ConnectingState);
+
+    reinterpret_cast<struct sockaddr_in*>(addr)->sin_port = htons(m_port);
+    free(m_socket);
+
+    uv_connect_t *req = (uv_connect_t*) malloc(sizeof(uv_connect_t));
+    req->data = this;
+
+    m_socket = static_cast<uv_tcp_t*>(malloc(sizeof(uv_tcp_t)));
+    m_socket->data = this;
+
+    uv_tcp_init(uv_default_loop(), m_socket);
+    uv_tcp_nodelay(m_socket, 1);
+    uv_tcp_keepalive(m_socket, 1, 60);
+
+    uv_tcp_connect(req, m_socket, (const sockaddr*) addr, Client::onConnect);
+}
+
+
+void Client::parse(char *line, size_t len)
+{
+    line[len - 1] = '\0';
+
+    LOG_DEBUG("[%s:%u] received (%d bytes): \"%s\"", m_host, m_port, len, line);
+
+    json_error_t err;
+    json_t *val = json_loads(line, 0, &err);
+
+    if (!val) {
+        LOG_ERR("[%s:%u] JSON decode failed: \"%s\"", m_host, m_port, err.text);
+        return;
+    }
+}
+
+
+void Client::setState(SocketState state)
+{
+    LOG_DEBUG("[%s:%u] state: %d", m_host, m_port, state);
+
+    if (m_state == state) {
+        return;
+    }
+
+    m_state = state;
+}
+
+
+void Client::onAllocBuffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf)
+{
+    auto client = getClient(handle->data);
+
+    buf->base = &client->m_recvBuf.base[client->m_recvBufPos];
+    buf->len  = client->m_recvBuf.len - client->m_recvBufPos;
+}
+
+
+void Client::onClose(uv_handle_t *handle)
+{
+    auto client = getClient(handle->data);
+
+    free(client->m_socket);
+
+    client->m_stream = nullptr;
+    client->m_socket = nullptr;
+    client->setState(UnconnectedState);
+}
+
+
+void Client::onConnect(uv_connect_t *req, int status)
+{
+    auto client = getClient(req->data);
+    if (status < 0) {
+        LOG_ERR("[%s:%u] connect error: \"%s\"", client->m_host, client->m_port, uv_strerror(status));
+        free(req);
+        return;
+    }
+
+    client->m_stream = static_cast<uv_stream_t*>(req->handle);
+    client->m_stream->data = req->data;
+    client->setState(ConnectedState);
+
+    uv_read_start(client->m_stream, Client::onAllocBuffer, Client::onRead);
+    free(req);
+
+    client->m_listener->onLoginCredentialsRequired(client);
+}
+
+
+void Client::onRead(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf)
+{
+    auto client = getClient(stream->data);
+    if (nread < 0) {
+        if (nread != UV_EOF) {
+            LOG_ERR("[%s:%u] read error: \"%s\"", client->m_host, client->m_port, uv_strerror(nread));
+        }
+
+        client->close();
+        return;
+    }
+
+    client->m_recvBufPos += nread;
+
+    char* end;
+    char* start = client->m_recvBuf.base;
+    size_t remaining = client->m_recvBufPos;
+
+    while ((end = static_cast<char*>(memchr(start, '\n', remaining))) != nullptr) {
+        end++;
+        size_t len = end - start;
+        client->parse(start, len);
+
+        remaining -= len;
+        start = end;
+    }
+
+    if (remaining == 0) {
+        client->m_recvBufPos = 0;
+        return;
+    }
+
+    if (start == client->m_recvBuf.base) {
+        return;
+    }
+
+    memcpy(client->m_recvBuf.base, start, remaining);
+    client->m_recvBufPos = remaining;
+}
+
+
+void Client::onResolved(uv_getaddrinfo_t *req, int status, struct addrinfo *res)
+{
+    auto client = getClient(req->data);
+    if (status < 0) {
+        LOG_ERR("[%s:%u] DNS error: \"%s\"", client->m_host, client->m_port, uv_strerror(status));
+        return;
+    }
+
+    client->connect(res->ai_addr);
+    uv_freeaddrinfo(res);
+}
+
+
+Client *Client::getClient(void *data)
+{
+    return static_cast<Client*>(data);
+}

src/net/Client.h

@@ -0,0 +1,87 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef __CLIENT_H__
+#define __CLIENT_H__
+
+
+#include <uv.h>
+
+
+class Url;
+class IClientListener;
+
+
+class Client
+{
+public:
+    enum SocketState {
+        UnconnectedState,
+        HostLookupState,
+        ConnectingState,
+        ConnectedState,
+        ClosingState
+    };
+
+    Client(IClientListener *listener);
+    ~Client();
+
+    void connect(const char *host, uint16_t port);
+    void connect(const Url *url);
+    void disconnect();
+    void login(const char *user, const char *pass, const char *agent);
+    void send(char *data);
+
+    inline SocketState state() const { return m_state; }
+
+private:
+    constexpr static size_t kRecvBufSize = 4096;
+
+    int resolve(const char *host);
+    void close();
+    void connect(struct sockaddr *addr);
+    void parse(char *line, size_t len);
+    void setState(SocketState state);
+
+    static void onAllocBuffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf);
+    static void onClose(uv_handle_t *handle);
+    static void onConnect(uv_connect_t *req, int status);
+    static void onRead(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf);
+    static void onResolved(uv_getaddrinfo_t *req, int status, struct addrinfo *res);
+
+    static Client *getClient(void *data);
+
+    char *m_host;
+    IClientListener *m_listener;
+    size_t m_recvBufPos;
+    SocketState m_state;
+    struct addrinfo m_hints;
+    uint16_t m_port;
+    uv_buf_t m_recvBuf;
+    uv_getaddrinfo_t m_resolver;
+    uv_stream_t *m_stream;
+    uv_tcp_t *m_socket;
+};
+
+
+#endif /* __CLIENT_H__ */

src/net/Network.cpp

@@ -0,0 +1,65 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include <uv.h>
+
+
+#include "Console.h"
+#include "net/Client.h"
+#include "net/Network.h"
+#include "Options.h"
+
+
+Network::Network(const Options *options) :
+    m_backupPool(nullptr),
+    m_donatePool(nullptr),
+    m_pool(nullptr),
+    m_options(options)
+{
+    m_agent = userAgent();
+    m_pool = new Client(this);
+}
+
+
+Network::~Network()
+{
+    delete m_pool;
+    delete m_donatePool;
+    delete m_backupPool;
+
+    free(m_agent);
+}
+
+
+void Network::connect()
+{
+    m_pool->connect(m_options->url());
+//    LOG_DEBUG("XX %s", m_options->url());
+}
+
+
+void Network::onLoginCredentialsRequired(Client *client)
+{
+    client->login(m_options->user(), m_options->pass(), m_agent);
+}

src/net/Network.h

@@ -0,0 +1,56 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef __NETWORK_H__
+#define __NETWORK_H__
+
+
+#include "interfaces/IClientListener.h"
+
+
+class Options;
+
+
+class Network : public IClientListener
+{
+public:
+  Network(const Options *options);
+  ~Network();
+
+  void connect();
+
+  static char *userAgent();
+
+protected:
+  void onLoginCredentialsRequired(Client *client) override;
+
+private:
+  char *m_agent;
+  Client *m_backupPool;
+  Client *m_donatePool;
+  Client *m_pool;
+  const Options *m_options;
+};
+
+
+#endif /* __NETWORK_H__ */

src/net/Network_win.cpp

@@ -0,0 +1,64 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include <uv.h>
+#include <windows.h>
+
+
+#include "net/Network.h"
+#include "version.h"
+
+
+static inline OSVERSIONINFOEX winOsVersion()
+{
+    typedef NTSTATUS (NTAPI *RtlGetVersionFunction)(LPOSVERSIONINFO);
+    OSVERSIONINFOEX result = { sizeof(OSVERSIONINFOEX), 0, 0, 0, 0, {'\0'}, 0, 0, 0, 0, 0};
+
+    HMODULE ntdll = GetModuleHandleW(L"ntdll.dll");
+    if (ntdll ) {
+        RtlGetVersionFunction pRtlGetVersion = reinterpret_cast<RtlGetVersionFunction>(GetProcAddress(ntdll, "RtlGetVersion"));
+
+        if (pRtlGetVersion) {
+            pRtlGetVersion((LPOSVERSIONINFO) &result);
+        }
+    }
+
+    return result;
+}
+
+
+char *Network::userAgent()
+{
+    const auto osver = winOsVersion();
+
+    char *buf = static_cast<char*>(malloc(128));
+
+#   ifdef __GNUC__
+    snprintf(buf, 128, "%s/%s (Windows NT %lu.%lu; Win64; x64) libuv/%s gcc/%d.%d.%d", APP_NAME, APP_VERSION, osver.dwMajorVersion, osver.dwMinorVersion, uv_version_string(), __GNUC__, __GNUC_MINOR__, __GNUC_PATCHLEVEL__);
+#   else
+    snprintf(buf, 128, "%s/%s (Windows NT %lu.%lu; Win64; x64) libuv/%s", APP_NAME, APP_VERSION, osver.dwMajorVersion, osver.dwMinorVersion, uv_version_string());
+#   endif
+
+    return buf;
+}

src/net/Url.cpp

@@ -0,0 +1,93 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include <string.h>
+#include <stdlib.h>
+
+
+#include "net/Url.h"
+
+
+/**
+ * @brief Parse url.
+ *
+ * Valid urls:
+ * example.com
+ * example.com:3333
+ * stratum+tcp://example.com
+ * stratum+tcp://example.com:3333
+ *
+ * @param url
+ */
+Url::Url(const char *url) :
+    m_host(nullptr),
+    m_port(3333)
+{
+    const char *p = strstr(url, "://");
+    const char *base = url;
+
+    if (p) {
+        if (strncasecmp(url, "stratum+tcp://", 14)) {
+            return;
+        }
+
+        base = url + 14;
+    }
+
+    if (!strlen(base) || *base == '/') {
+        return;
+    }
+
+    char *port = strchr(base, ':');
+    if (!port) {
+        m_host = strdup(base);
+        return;
+    }
+
+    const size_t size = port++ - base + 1;
+    m_host = static_cast<char*>(malloc(size));
+    memcpy(m_host, base, size - 1);
+    m_host[size - 1] = '\0';
+
+    m_port = strtol(port, nullptr, 10);
+}
+
+
+Url::Url(const char *host, uint16_t port) :
+    m_port(port)
+{
+    m_host = strdup(host);
+}
+
+
+Url::~Url()
+{
+    free(m_host);
+}
+
+
+bool Url::isNicehash() const
+{
+    return isValid() && strstr(m_host, ".nicehash.com");
+}

src/net/Url.h

@@ -0,0 +1,49 @@
+/* XMRig
+ * Copyright 2010      Jeff Garzik <jgarzik@pobox.com>
+ * Copyright 2012-2014 pooler      <pooler@litecoinpool.org>
+ * Copyright 2014      Lucas Jones <https://github.com/lucasjones>
+ * Copyright 2014-2016 Wolf9466    <https://github.com/OhGodAPet>
+ * Copyright 2016      Jay D Dee   <jayddee246@gmail.com>
+ * Copyright 2016-2017 XMRig       <support@xmrig.com>
+ *
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   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. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef __URL_H__
+#define __URL_H__
+
+
+#include <stdint.h>
+
+
+class Url
+{
+public:
+    Url(const char *url);
+    Url(const char *host, uint16_t port);
+    ~Url();
+
+    bool isNicehash() const;
+
+    inline bool isValid() const     { return m_host && m_port > 0; }
+    inline const char *host() const { return m_host; }
+    inline uint16_t port() const    { return m_port; }
+
+private:
+    char *m_host;
+    uint16_t m_port;
+};
+
+#endif /* __URL_H__ */