hiredis-client 0.4.0

data/ext/redis_client/hiredis/hiredis_connection.c

@@ -0,0 +1,708 @@
+// Extensive parts of this code is taken from `hiredis-rb`, so we're keeping the
+// initial copyright:
+//
+// Copyright (c) 2010-2012, Pieter Noordhuis
+//
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are met:
+//
+// * Redistributions of source code must retain the above copyright notice, this
+//   list of conditions and the following disclaimer.
+//
+// * Redistributions in binary form must reproduce the above copyright notice,
+//   this list of conditions and the following disclaimer in the documentation
+//   and/or other materials provided with the distribution.
+//
+// * Neither the name of Redis nor the names of its contributors may be used to
+//   endorse or promote products derived from this software without specific prior
+//   written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+// ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+// ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+#include "ruby.h"
+#include "ruby/encoding.h"
+#include <errno.h>
+#include <sys/socket.h>
+#include <stdbool.h>
+#include "hiredis.h"
+#include "hiredis_ssl.h"
+
+#if !defined(HAVE_RB_HASH_NEW_CAPA)
+static inline VALUE rb_hash_new_capa(long capa)
+{
+  return rb_hash_new();
+}
+#endif
+
+static VALUE rb_cSet, rb_eRedisClientCommandError, rb_eRedisClientConnectionError;
+static VALUE rb_eRedisClientConnectTimeoutError, rb_eRedisClientReadTimeoutError, rb_eRedisClientWriteTimeoutError;
+static ID id_parse, id_add, id_new;
+
+typedef struct {
+    redisSSLContext *context;
+} hiredis_ssl_context_t;
+
+#define ENSURE_CONNECTED(connection) if (!connection->context) rb_raise(rb_eRuntimeError, "[BUG] not connected");
+
+#define SSL_CONTEXT(from, name) \
+    hiredis_ssl_context_t *name = NULL; \
+    TypedData_Get_Struct(from, hiredis_ssl_context_t, &hiredis_ssl_context_data_type, name); \
+    if(name == NULL) { \
+        rb_raise(rb_eArgError, "NULL found for " # name " when shouldn't be."); \
+    }
+
+void hiredis_ssl_context_mark(void *ptr) { }
+
+void hiredis_ssl_context_free(void *ptr) {
+    hiredis_ssl_context_t *ssl_context = (hiredis_ssl_context_t *)ptr;
+    if (ssl_context->context) {
+        redisFreeSSLContext(ssl_context->context);
+        ssl_context = NULL;
+    }
+}
+
+static size_t hiredis_ssl_context_memsize(const void *ptr) {
+    size_t size = sizeof(hiredis_ssl_context_t);
+    // Note: I couldn't find a way to measure the SSLContext size.
+    return size;
+}
+
+static const rb_data_type_t hiredis_ssl_context_data_type = {
+    .wrap_struct_name = "redis-client:hiredis_ssl_context",
+    .function = {
+        .dmark = hiredis_ssl_context_mark,
+        .dfree = hiredis_ssl_context_free,
+        .dsize = hiredis_ssl_context_memsize,
+#ifdef HAS_GC_COMPACT
+        .dcompact = NULL
+#endif
+    },
+    .flags = RUBY_TYPED_FREE_IMMEDIATELY
+};
+
+static VALUE hiredis_ssl_context_alloc(VALUE klass) {
+    hiredis_ssl_context_t *ssl_context;
+    return TypedData_Make_Struct(klass, hiredis_ssl_context_t, &hiredis_ssl_context_data_type, ssl_context);
+}
+
+static VALUE hiredis_ssl_context_init(VALUE self, VALUE ca_file, VALUE ca_path, VALUE cert, VALUE key, VALUE hostname) {
+    redisSSLContextError ssl_error = 0;
+    SSL_CONTEXT(self, ssl_context);
+
+    ssl_context->context = redisCreateSSLContext(
+        RTEST(ca_file) ? StringValueCStr(ca_file) : NULL,
+        RTEST(ca_path) ? StringValueCStr(ca_path) : NULL,
+        RTEST(cert) ? StringValueCStr(cert) : NULL,
+        RTEST(key) ? StringValueCStr(key) : NULL,
+        RTEST(hostname) ? StringValueCStr(hostname) : NULL,
+        &ssl_error
+    );
+
+    if (ssl_error) {
+        return rb_str_new_cstr(redisSSLContextGetError(ssl_error));
+    }
+
+    if (!ssl_context->context) {
+        return rb_str_new_cstr("Unknown error while creating SSLContext");
+    }
+
+    return Qnil;
+}
+
+static void *reply_append(const redisReadTask *task, VALUE value) {
+    if (task && task->parent) {
+        VALUE parent = (VALUE)task->parent->obj;
+
+        switch (task->parent->type) {
+            case REDIS_REPLY_ARRAY:
+            case REDIS_REPLY_PUSH:
+                rb_ary_store(parent, task->idx, value);
+                break;
+            case REDIS_REPLY_MAP:
+                if (task->idx % 2) {
+                    VALUE key = (VALUE)task->parent->privdata;
+                    task->parent->privdata = NULL;
+                    rb_hash_aset(parent, key, value);
+                } else {
+                    task->parent->privdata = (void*)value;
+                }
+                break;
+            case REDIS_REPLY_SET:
+                rb_funcall(parent, id_add, 1, value);
+                break;
+            default:
+                rb_bug("[hiredis] Unexpected task parent type %d", task->parent->type);
+                break;
+        }
+    }
+    return (void*)value;
+}
+
+static void *reply_create_string(const redisReadTask *task, char *cstr, size_t len) {
+    VALUE string = rb_external_str_new(cstr, len);
+    if (rb_enc_str_coderange(string) == ENC_CODERANGE_BROKEN) {
+        rb_enc_associate(string, rb_ascii8bit_encoding());
+    }
+
+    if (task->type == REDIS_REPLY_STATUS) {
+      rb_str_freeze(string);
+    }
+
+    if (task->type == REDIS_REPLY_ERROR) {
+        string = rb_funcall(rb_eRedisClientCommandError, id_parse, 1, string);
+    }
+
+    return reply_append(task, string);
+}
+static void *reply_create_array(const redisReadTask *task, size_t elements) {
+    VALUE value = Qnil;
+    switch (task->type) {
+        case REDIS_REPLY_PUSH:
+        case REDIS_REPLY_ARRAY:
+            value = rb_ary_new_capa(elements);
+            break;
+        case REDIS_REPLY_MAP:
+            value = rb_hash_new_capa(elements / 2);
+            break;
+        case REDIS_REPLY_SET:
+            value = rb_funcallv(rb_cSet, id_new, 0, NULL);
+            break;
+        default:
+            rb_bug("[hiredis] Unexpected create array type %d", task->parent->type);
+            break;
+    }
+
+    return reply_append(task, value);
+}
+static void *reply_create_integer(const redisReadTask *task, long long value) {
+    return reply_append(task, LL2NUM(value));
+}
+static void *reply_create_double(const redisReadTask *task, double value, char *str, size_t len) {
+    return reply_append(task, DBL2NUM(value));
+}
+static void *reply_create_nil(const redisReadTask *task) {
+    return reply_append(task, Qnil);
+}
+static void *reply_create_bool(const redisReadTask *task, int bval) {
+    reply_append(task, bval ? Qtrue : Qfalse);
+    // Qfalse == NULL, so we can't return Qfalse as it would be interpreted as out of memory error.
+    // So we return Qnil instead.
+    return (void*)(bval ? Qtrue : Qnil);
+}
+
+static void reply_free(void *ptr) {
+    // we let GC handle it.
+}
+
+/* Default set of functions to build the reply. Keep in mind that such a
+ * function returning NULL is interpreted as OOM. */
+static redisReplyObjectFunctions reply_functions = {
+    reply_create_string,
+    reply_create_array,
+    reply_create_integer,
+    reply_create_double,
+    reply_create_nil,
+    reply_create_bool,
+    reply_free,
+};
+
+#define CONNECTION(from, name) \
+    hiredis_connection_t *name = NULL; \
+    TypedData_Get_Struct(from, hiredis_connection_t, &hiredis_connection_data_type, name); \
+    if(name == NULL) { \
+        rb_raise(rb_eArgError, "NULL found for " # name " when shouldn't be."); \
+    }
+
+
+typedef struct {
+    redisContext *context;
+    struct timeval connect_timeout;
+    struct timeval read_timeout;
+    struct timeval write_timeout;
+} hiredis_connection_t;
+
+void hiredis_connection_mark_task(redisReadTask *task) {
+    while (task) {
+        if (task->obj) { rb_gc_mark((VALUE)task->obj); }
+        if (task->privdata) { rb_gc_mark((VALUE)task->privdata); }
+        task = task->parent;
+    }
+}
+
+void hiredis_connection_mark(void *ptr) {
+    hiredis_connection_t *connection = ptr;
+    if (connection->context) {
+        redisReader *reader = connection->context->reader;
+        for (int index = 0; index < reader->tasks; index++) {
+            hiredis_connection_mark_task(reader->task[index]);
+        }
+    }
+}
+void hiredis_connection_free(void *ptr) {
+    hiredis_connection_t *connection = ptr;
+    if (connection) {
+         if (connection->context) {
+             redisFree(connection->context);
+         }
+         xfree(connection);
+    }
+}
+
+static size_t hiredis_connection_task_memsize(redisReadTask *task) {
+    size_t size = 0;
+    while (task) {
+        size += sizeof(redisReadTask);
+        task = task->parent;
+    }
+    return size;
+}
+
+static size_t hiredis_connection_memsize(const void *ptr) {
+    hiredis_connection_t *connection = (hiredis_connection_t *)ptr;
+
+    size_t size = sizeof(hiredis_connection_t);
+
+    if (connection->context) {
+        size += sizeof(redisContext);
+        if (connection->context->reader) {
+            redisReader *reader = connection->context->reader;
+            size += sizeof(redisReader);
+            size += reader->maxbuf;
+
+            for (int index = 0; index < reader->tasks; index++) {
+                size += hiredis_connection_task_memsize(reader->task[index]);
+            }
+        }
+    }
+
+    return size;
+}
+
+static const rb_data_type_t hiredis_connection_data_type = {
+    .wrap_struct_name = "redis-client:hiredis_connection",
+    .function = {
+        .dmark = hiredis_connection_mark,
+        .dfree = hiredis_connection_free,
+        .dsize = hiredis_connection_memsize,
+#ifdef HAS_GC_COMPACT
+        .dcompact = NULL
+#endif
+    },
+    .flags = RUBY_TYPED_FREE_IMMEDIATELY
+};
+
+static VALUE hiredis_alloc(VALUE klass) {
+    hiredis_connection_t *connection;
+    return TypedData_Make_Struct(klass, hiredis_connection_t, &hiredis_connection_data_type, connection);
+}
+
+static inline void redis_raise_error_and_disconnect(redisContext *context, VALUE timeout_error) {
+    if (!context) return;
+
+    int err = context->err;
+    char errstr[128];
+    MEMCPY(context->errstr, errstr, char, 128);
+    redisFree(context);
+
+    // OpenSSL bug: The SSL_ERROR_SYSCALL with errno value of 0 indicates unexpected EOF from the peer.
+    if (errno == EAGAIN || (err == REDIS_ERR_IO && errno == 0)) {
+        errno = 0;
+        rb_raise(timeout_error, "Resource temporarily unavailable");
+    }
+
+    switch(err) {
+    case REDIS_ERR_IO:
+        rb_sys_fail(0);
+        break;
+    default:
+        /* Raise something else */
+        rb_raise(rb_eRedisClientConnectionError, "%s", errstr);
+    }
+}
+
+static inline void hiredis_raise_error_and_disconnect(hiredis_connection_t *connection, VALUE timeout_error) {
+    redisContext *context = connection->context;
+    if (!context) return;
+    connection->context = NULL;
+    redis_raise_error_and_disconnect(context, timeout_error);
+}
+
+static VALUE hiredis_set_connect_timeout(VALUE self, VALUE timeout_us) {
+    CONNECTION(self, connection);
+    connection->connect_timeout.tv_sec = NUM2INT(timeout_us) / 1000000;
+    connection->connect_timeout.tv_usec = NUM2INT(timeout_us) % 1000000;
+    return timeout_us;
+}
+
+static VALUE hiredis_set_read_timeout(VALUE self, VALUE timeout_us) {
+    CONNECTION(self, connection);
+    connection->read_timeout.tv_sec = NUM2INT(timeout_us) / 1000000;
+    connection->read_timeout.tv_usec = NUM2INT(timeout_us) % 1000000;
+    return timeout_us;
+}
+
+static VALUE hiredis_set_write_timeout(VALUE self, VALUE timeout_us) {
+    CONNECTION(self, connection);
+    connection->write_timeout.tv_sec = NUM2INT(timeout_us) / 1000000;
+    connection->write_timeout.tv_usec = NUM2INT(timeout_us) % 1000000;
+    return timeout_us;
+}
+
+static int hiredis_wait_readable(int fd, const struct timeval *timeout, int *isset) {
+    struct timeval to;
+    struct timeval *toptr = NULL;
+
+    rb_fdset_t fds;
+
+    /* Be cautious: a call to rb_fd_init to initialize the rb_fdset_t structure
+     * must be paired with a call to rb_fd_term to free it. */
+    rb_fd_init(&fds);
+    rb_fd_set(fd, &fds);
+
+    /* rb_thread_{fd_,}select modifies the passed timeval, so we pass a copy */
+    if (timeout != NULL && (timeout->tv_sec || timeout->tv_usec)) {
+        memcpy(&to, timeout, sizeof(to));
+        toptr = &to;
+    }
+
+    if (rb_thread_fd_select(fd + 1, &fds, NULL, NULL, toptr) < 0) {
+        rb_fd_term(&fds);
+        return -1;
+    }
+
+    if (rb_fd_isset(fd, &fds) && isset) {
+        *isset = 1;
+    }
+
+    rb_fd_term(&fds);
+    return 0;
+}
+
+static int hiredis_wait_writable(int fd, const struct timeval *timeout, int *isset) {
+    struct timeval to;
+    struct timeval *toptr = NULL;
+
+    /* Be cautious: a call to rb_fd_init to initialize the rb_fdset_t structure
+     * must be paired with a call to rb_fd_term to free it. */
+    rb_fdset_t fds;
+    rb_fd_init(&fds);
+    rb_fd_set(fd, &fds);
+
+    /* rb_thread_{fd_,}select modifies the passed timeval, so we pass a copy */
+    if (timeout != NULL && (timeout->tv_sec || timeout->tv_usec)) {
+        memcpy(&to, timeout, sizeof(to));
+        toptr = &to;
+    }
+
+    if (rb_thread_fd_select(fd + 1, NULL, &fds, NULL, toptr) < 0) {
+        rb_fd_term(&fds);
+        return -1;
+    }
+
+    if (rb_fd_isset(fd, &fds) && isset) {
+        *isset = 1;
+    }
+
+    rb_fd_term(&fds);
+    return 0;
+}
+
+static VALUE hiredis_connect_finish(hiredis_connection_t *connection, redisContext *context) {
+    if (context->err) {
+        redis_raise_error_and_disconnect(context, rb_eRedisClientConnectTimeoutError);
+    }
+
+    int writable = 0;
+    int optval = 0;
+    socklen_t optlen = sizeof(optval);
+
+    /* Wait for socket to become writable */
+    if (hiredis_wait_writable(context->fd, &connection->connect_timeout, &writable) < 0) {
+        redis_raise_error_and_disconnect(context, rb_eRedisClientConnectTimeoutError);
+    }
+
+    if (!writable) {
+        errno = ETIMEDOUT;
+        redis_raise_error_and_disconnect(context, rb_eRedisClientConnectTimeoutError);
+    }
+
+    /* Check for socket error */
+    if (getsockopt(context->fd, SOL_SOCKET, SO_ERROR, &optval, &optlen) < 0) {
+        context->err = REDIS_ERR_IO;
+        redis_raise_error_and_disconnect(context, rb_eRedisClientConnectTimeoutError);
+    }
+
+    if (optval) {
+        errno = optval;
+        redis_raise_error_and_disconnect(context, rb_eRedisClientConnectTimeoutError);
+    }
+
+    context->reader->fn = &reply_functions;
+    redisSetPushCallback(context, NULL);
+    connection->context = context;
+    return Qtrue;
+}
+
+static VALUE hiredis_connect_tcp(VALUE self, VALUE host, VALUE port) {
+    CONNECTION(self, connection);
+    if (connection->context) {
+        redisFree(connection->context);
+        connection->context = NULL;
+    }
+    return hiredis_connect_finish(connection, redisConnectNonBlock(StringValuePtr(host), NUM2INT(port)));
+}
+
+static VALUE hiredis_connect_unix(VALUE self, VALUE path) {
+    CONNECTION(self, connection);
+    if (connection->context) {
+        redisFree(connection->context);
+        connection->context = NULL;
+    }
+    return hiredis_connect_finish(connection, redisConnectUnixNonBlock(StringValuePtr(path)));
+}
+
+static VALUE hiredis_init_ssl(VALUE self, VALUE ssl_param) {
+    CONNECTION(self, connection);
+    SSL_CONTEXT(ssl_param, ssl_context)
+
+    if (redisInitiateSSLWithContext(connection->context, ssl_context->context) != REDIS_OK) {
+        hiredis_raise_error_and_disconnect(connection, rb_eRedisClientConnectTimeoutError);
+    }
+
+    redisSSL *redis_ssl = redisGetSSLSocket(connection->context);
+ 
+    if (redis_ssl->wantRead) {
+        int readable = 0;
+        if (hiredis_wait_readable(connection->context->fd, &connection->connect_timeout, &readable) < 0) {
+            hiredis_raise_error_and_disconnect(connection, rb_eRedisClientConnectTimeoutError);
+        }
+        if (!readable) {
+            errno = EAGAIN;
+            hiredis_raise_error_and_disconnect(connection, rb_eRedisClientConnectTimeoutError);
+        }
+
+        if (redisInitiateSSLContinue(connection->context) != REDIS_OK) {
+            hiredis_raise_error_and_disconnect(connection, rb_eRedisClientConnectTimeoutError);
+        };
+    }
+
+    return Qtrue;
+}
+
+static VALUE hiredis_connected_p(VALUE self) {
+    CONNECTION(self, connection);
+
+    return connection->context ? Qtrue : Qfalse;
+}
+
+static VALUE hiredis_write(VALUE self, VALUE command) {
+    Check_Type(command, T_ARRAY);
+
+    CONNECTION(self, connection);
+    ENSURE_CONNECTED(connection);
+
+    int size = (int)RARRAY_LEN(command);
+    VALUE _argv_handle;
+    char **argv = RB_ALLOCV_N(char *, _argv_handle, size);
+
+    VALUE _argv_len_handle;
+    size_t *argv_len = RB_ALLOCV_N(size_t, _argv_len_handle, size);
+
+    for (int index = 0; index < size; index++) {
+        VALUE arg = rb_ary_entry(command, index);
+        Check_Type(arg, T_STRING);
+        argv[index] = RSTRING_PTR(arg);
+        argv_len[index] = RSTRING_LEN(arg);
+    }
+
+    redisAppendCommandArgv(connection->context, size, (const char **)argv, argv_len);
+    return Qnil;
+}
+
+static VALUE hiredis_flush(VALUE self) {
+    CONNECTION(self, connection);
+    ENSURE_CONNECTED(connection);
+
+    int wdone = 0;
+    while (!wdone) {
+        errno = 0;
+        if (redisBufferWrite(connection->context, &wdone) == REDIS_ERR) {
+            if (errno == EAGAIN) {
+                int writable = 0;
+
+                if (hiredis_wait_writable(connection->context->fd, &connection->write_timeout, &writable) < 0) {
+                    hiredis_raise_error_and_disconnect(connection, rb_eRedisClientWriteTimeoutError);
+                }
+
+                if (!writable) {
+                    errno = EAGAIN;
+                    hiredis_raise_error_and_disconnect(connection, rb_eRedisClientWriteTimeoutError);
+                }
+            } else {
+                hiredis_raise_error_and_disconnect(connection, rb_eRedisClientWriteTimeoutError);
+            }
+        }
+    }
+
+    return Qtrue;
+}
+
+static int hiredis_read_internal(hiredis_connection_t *connection, VALUE *reply) {
+    void *redis_reply = NULL;
+    int wdone = 0;
+
+    /* Try to read pending replies */
+    if (redisGetReplyFromReader(connection->context, &redis_reply) == REDIS_ERR) {
+        /* Protocol error */
+        return -1;
+    }
+
+    if (redis_reply == NULL) {
+        /* Write until the write buffer is drained */
+        while (!wdone) {
+            errno = 0;
+
+            if (redisBufferWrite(connection->context, &wdone) == REDIS_ERR) {
+                /* Socket error */
+                return -1;
+            }
+
+            if (errno == EAGAIN) {
+                int writable = 0;
+
+                if (hiredis_wait_writable(connection->context->fd, &connection->write_timeout, &writable) < 0) {
+                    return -2;
+                }
+
+                if (!writable) {
+                    errno = EAGAIN;
+                    return -2;
+                }
+            }
+        }
+
+        /* Read until there is a full reply */
+        while (redis_reply == NULL) {
+            errno = 0;
+
+            if (redisBufferRead(connection->context) == REDIS_ERR) {
+                /* Socket error */
+                return -1;
+            }
+
+            if (errno == EAGAIN) {
+                int readable = 0;
+
+                if (hiredis_wait_readable(connection->context->fd, &connection->read_timeout, &readable) < 0) {
+                    return -2;
+                }
+
+                if (!readable) {
+                    errno = EAGAIN;
+                    return -2;
+                }
+
+                /* Retry */
+                continue;
+            }
+
+            if (redisGetReplyFromReader(connection->context, &redis_reply) == REDIS_ERR) {
+                /* Protocol error */
+                return -1;
+            }
+        }
+    }
+
+    /* Set reply object */
+    if (reply != NULL) {
+        *reply = (VALUE)redis_reply;
+    }
+
+    return 0;
+}
+
+static VALUE hiredis_read(VALUE self) {
+    CONNECTION(self, connection);
+    ENSURE_CONNECTED(connection);
+
+    VALUE reply = Qnil;
+    if (hiredis_read_internal(connection, &reply)) {
+        hiredis_raise_error_and_disconnect(connection, rb_eRedisClientReadTimeoutError);
+    }
+    return reply;
+}
+
+static VALUE hiredis_close(VALUE self) {
+    CONNECTION(self, connection);
+    if (connection->context) {
+        redisFree(connection->context);
+        connection->context = NULL;
+    }
+    return Qnil;
+}
+
+void Init_hiredis_connection(void) {
+#ifdef RUBY_ASSERT
+        // Qfalse == NULL, so we can't return Qfalse in `reply_create_bool()`
+        RUBY_ASSERT((void *)Qfalse == NULL);
+        RUBY_ASSERT((void *)Qnil != NULL);
+#endif
+
+    redisInitOpenSSL();
+
+    id_parse = rb_intern("parse");
+    id_add = rb_intern("add");
+    id_new = rb_intern("new");
+
+    rb_cSet = rb_const_get(rb_cObject, rb_intern("Set"));
+    rb_global_variable(&rb_cSet);
+
+    VALUE rb_cRedisClient = rb_const_get(rb_cObject, rb_intern("RedisClient"));
+
+    rb_eRedisClientCommandError = rb_const_get(rb_cRedisClient, rb_intern("CommandError"));
+    rb_global_variable(&rb_eRedisClientCommandError);
+
+    rb_eRedisClientConnectionError = rb_const_get(rb_cRedisClient, rb_intern("ConnectionError"));
+    rb_global_variable(&rb_eRedisClientConnectionError);
+
+    rb_eRedisClientConnectTimeoutError = rb_const_get(rb_cRedisClient, rb_intern("ConnectTimeoutError"));
+    rb_global_variable(&rb_eRedisClientConnectTimeoutError);
+
+    rb_eRedisClientReadTimeoutError = rb_const_get(rb_cRedisClient, rb_intern("ReadTimeoutError"));
+    rb_global_variable(&rb_eRedisClientReadTimeoutError);
+
+    rb_eRedisClientWriteTimeoutError = rb_const_get(rb_cRedisClient, rb_intern("WriteTimeoutError"));
+    rb_global_variable(&rb_eRedisClientWriteTimeoutError);
+
+    VALUE rb_cHiredisConnection = rb_define_class_under(rb_cRedisClient, "HiredisConnection", rb_cObject);
+    rb_define_alloc_func(rb_cHiredisConnection, hiredis_alloc);
+
+    rb_define_private_method(rb_cHiredisConnection, "connect_timeout_us=", hiredis_set_connect_timeout, 1);
+    rb_define_private_method(rb_cHiredisConnection, "read_timeout_us=", hiredis_set_read_timeout, 1);
+    rb_define_private_method(rb_cHiredisConnection, "write_timeout_us=", hiredis_set_write_timeout, 1);
+
+    rb_define_private_method(rb_cHiredisConnection, "connect_tcp", hiredis_connect_tcp, 2);
+    rb_define_private_method(rb_cHiredisConnection, "connect_unix", hiredis_connect_unix, 1);
+    rb_define_private_method(rb_cHiredisConnection, "init_ssl", hiredis_init_ssl, 1);
+    rb_define_method(rb_cHiredisConnection, "connected?", hiredis_connected_p, 0);
+
+    rb_define_private_method(rb_cHiredisConnection, "_write", hiredis_write, 1);
+    rb_define_private_method(rb_cHiredisConnection, "_read", hiredis_read, 0);
+    rb_define_private_method(rb_cHiredisConnection, "flush", hiredis_flush, 0);
+    rb_define_method(rb_cHiredisConnection, "close", hiredis_close, 0);
+
+    VALUE rb_cHiredisSSLContext = rb_define_class_under(rb_cHiredisConnection, "SSLContext", rb_cObject);
+    rb_define_alloc_func(rb_cHiredisSSLContext, hiredis_ssl_context_alloc);
+    rb_define_private_method(rb_cHiredisSSLContext, "init", hiredis_ssl_context_init, 5);
+}