mt-libcouchbase 1.4.0

data/ext/libcouchbase/plugins/io/libuv/CMakeLists.txt

@@ -0,0 +1,44 @@
+IF(LCB_INSTALL_HEADERS)
+  INSTALL(
+      FILES
+      plugin-libuv.c
+      plugin-internal.h
+      libuv_compat.h
+      DESTINATION
+      include/libcouchbase/plugins/io/libuv)
+
+  INSTALL(
+      FILES
+      libuv_io_opts.h
+      DESTINATION
+      include/libcouchbase)
+ENDIF(LCB_INSTALL_HEADERS)
+
+IF(NOT HAVE_LIBUV OR LCB_NO_PLUGINS OR NOT LCB_BUILD_LIBUV)
+    RETURN()
+ENDIF()
+
+ADD_LIBRARY(couchbase_libuv SHARED plugin-libuv.c)
+
+# The macros abstracting UV versions dynamically inject special signatures. We
+# can't have the compiler whine about that.
+
+# Unfortunately, Older GCC (EL5) barfs at this option.
+#IF(CMAKE_COMPILER_IS_GNUCC)
+#    SET(_lcbuv_cflags "-Wno-unused-parameters")
+#ELSE()
+#    SET(_lcbuv_cflags "")
+#ENDIF()
+SET_TARGET_PROPERTIES(couchbase_libuv
+    PROPERTIES
+    PREFIX "lib"
+    IMPORT_PREFIX "lib"
+    COMPILE_FLAGS "${CMAKE_C_FLAGS} ${LCB_CORE_FLAGS} ${_lcbuv_cflags}")
+
+TARGET_LINK_LIBRARIES(couchbase_libuv ${LIBUV_LIBRARIES} couchbase)
+INCLUDE_DIRECTORIES(AFTER ${LIBUV_INCLUDE_DIR})
+ADD_DEFINITIONS(-DLIBCOUCHBASE_INTERNAL=1)
+INSTALL(TARGETS
+    couchbase_libuv
+    LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
+    RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})

data/ext/libcouchbase/plugins/io/libuv/libuv_compat.h

@@ -0,0 +1,218 @@
+/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2013 Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#ifndef LIBUV_COMPAT_H
+#define LIBUV_COMPAT_H
+
+#ifdef UV_VERSION_MAJOR
+#ifndef UV_VERSION_PATCH
+#define UV_VERSION_PATCH 0
+#endif
+#define UV_VERSION  ((UV_VERSION_MAJOR << 16) | \
+                     (UV_VERSION_MINOR <<  8) | \
+                     (UV_VERSION_PATCH))
+#else
+#define UV_VERSION 0x000b00
+#endif
+
+#if defined(_WIN32) && defined(LIBCOUCHBASE_INTERNAL)
+#include <libcouchbase/plugins/io/wsaerr.h>
+#endif
+
+#ifndef UNKNOWN
+#define UNKNOWN -1
+#endif
+
+#ifndef EAIFAMNOSUPPORT
+#define EAIFAMNOSUPPORT EAI_FAMILY
+#endif
+
+#ifndef EAISERVICE
+#define EAISERVICE EAI_SERVICE
+#endif
+
+#ifndef EAI_SYSTEM
+#define EAI_SYSTEM -11
+#endif
+#ifndef EADDRINFO
+#define EADDRINFO EAI_SYSTEM
+#endif
+
+#ifndef EAISOCKTYPE
+#define EAISOCKTYPE EAI_SOCKTYPE
+#endif
+
+#ifndef ECHARSET
+#define ECHARSET 0
+#endif
+
+#ifndef EOF
+#define EOF -1
+#endif
+
+#ifndef ENONET
+#define ENONET ENETDOWN
+#endif
+
+#ifndef ESHUTDOWN
+#define ESHUTDOWN WSAESHUTDOWN
+#endif
+
+#ifndef EHOSTDOWN
+/* missing only on Windows */
+#define EHOSTDOWN WSAEHOSTDOWN
+#endif
+
+/* Not all systems have these error codes */
+#ifndef EAI_FAIL
+#define EAI_FAIL (-1)
+#endif
+#ifndef EAI_CANCELED
+#define EAI_CANCELED -101
+#endif
+#ifndef EAI_ADDRFAMILY
+#define EAI_ADDRFAMILY -9
+#endif
+#ifndef EREMOTEIO
+#define EREMOTEIO -121
+#endif
+#ifndef EAI_BADHINTS
+#define EAI_BADHINTS EAI_FAIL
+#endif
+#ifndef EAI_NODATA
+#define EAI_NODATA EAI_FAIL
+#endif
+#ifndef EAI_PROTOCOL
+#define EAI_PROTOCOL EAI_FAIL
+#endif
+#ifndef EAI_AGAIN
+#define EAI_AGAIN EAI_FAIL
+#endif
+#ifndef EAI_BADFLAGS
+#define EAI_BADFLAGS EAI_FAIL
+#endif
+#ifndef EAI_MEMORY
+#define EAI_MEMORY EAI_FAIL
+#endif
+#ifndef EAI_OVERFLOW
+#define EAI_OVERFLOW EAI_FAIL
+#endif
+#ifndef EFTYPE
+#define EFTYPE EAI_FAIL
+#endif
+
+#define OK 0
+
+#if UV_VERSION < 0x000900
+    #define UVC_RUN_ONCE(l) uv_run_once(l)
+    #define UVC_RUN_DEFAULT(l) uv_run(l)
+#else
+    #define UVC_RUN_ONCE(l) uv_run(l, UV_RUN_ONCE)
+    #define UVC_RUN_DEFAULT(l) uv_run(l, UV_RUN_DEFAULT)
+#endif
+
+#if UV_VERSION < 0x000b00
+
+  #define UVC_TCP_CONNECT(req, handle, addr, cb) \
+      uv_tcp_connect(req, handle, *(struct sockaddr_in *)addr, cb);
+
+  #define UVC_TCP_CONNECT6(req, handle, addr, cb) \
+      uv_tcp_connect6(req, handle, *(struct sockaddr_in6 *)addr, cb);
+
+  #define UVC_ALLOC_CB(func) \
+      uv_buf_t func(uv_handle_t *handle, size_t suggested_size)
+
+  #define UVC_ALLOC_CB_VARS() \
+      uv_buf_t _buf; uv_buf_t *buf = &_buf;
+
+  #define UVC_ALLOC_CB_RETURN() \
+      return _buf;
+
+  #define UVC_READ_CB(func) \
+      void func(uv_stream_t *stream, ssize_t nread, const uv_buf_t _buf)
+
+  #define UVC_READ_CB_VARS() \
+      const uv_buf_t *buf = &_buf;
+
+  #define UVC_TIMER_CB(func) \
+      void func(uv_timer_t *timer, int status)
+
+  static int uvc_uv2syserr(int status) {
+      #define X(errnum,errname,errdesc) \
+        if (status == UV_##errname) { return errname; }
+      UV_ERRNO_MAP(X);
+      #undef X
+      return 0;
+  }
+
+  static int uvc_is_eof(uv_loop_t *loop, int error) {
+      error = uv_last_error(loop).code;
+      return error == UV_EOF;
+  }
+
+  static int uvc_last_errno(uv_loop_t *loop, int error) {
+      int uverr = 0;
+
+      if (!error) {
+        return 0;
+      }
+
+      uverr = uv_last_error(loop).code;
+      return uvc_uv2syserr(uverr);
+  }
+
+#else
+
+  #define UVC_TCP_CONNECT(req, handle, addr, cb) \
+      uv_tcp_connect(req, handle, addr, cb);
+
+  #define UVC_TCP_CONNECT6(req, handle, addr, cb) \
+      uv_tcp_connect(req, handle, addr, cb);
+
+  #define UVC_ALLOC_CB(func) \
+      void func(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf)
+
+  #define UVC_ALLOC_CB_VARS()
+
+  #define UVC_ALLOC_CB_RETURN()
+
+  #define UVC_READ_CB(func) \
+      void func(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf)
+
+  #define UVC_READ_CB_VARS()
+
+  #define UVC_TIMER_CB(func) \
+      void func(uv_timer_t *timer)
+
+  static int uv_uv2syserr(int status) {
+      #define X(name, desc) if (status == UV_##name) { return name; }
+      UV_ERRNO_MAP(X)
+      #undef X
+      return 0;
+  }
+
+  static int uvc_last_errno(uv_loop_t *loop, int error) {
+      return uv_uv2syserr(error);
+  }
+
+  static int uvc_is_eof(uv_loop_t *loop, int error) {
+      (void) loop;
+      return error == UV_EOF;
+  }
+
+#endif
+#endif

data/ext/libcouchbase/plugins/io/libuv/libuv_io_opts.h

@@ -0,0 +1,118 @@
+/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2013 Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+/**
+ * @file
+ * libuv integration with libcouchbase
+ * @author Mark Nunberg
+ */
+
+/**
+ * @ingroup lcb-io-plugin-api
+ * @defgroup lcb-libuv libuv
+ * @brief libuv integration
+ *
+ * @details
+ * libuv is a cross platform event framework using a completion-based programming
+ * model. Since most distributions do not offer up-to-date libuv binary packages
+ * and because libuv is constantly evolving itself, binary packages are not
+ * provided. The source code for this plugin is actually shipped with the
+ * headers of libcouchbase.
+ *
+ * If you built libuv together with libcouchbase (and thus there exists a
+ * library called `libcouchbase_libuv` then you may simply include this file
+ * and initialize the iops structure.
+ *
+ * Otherwise you should ensure that `<libcouchbase/plugins/io/libuv/plugin-libuv.c>`
+ * is compiled into an object, _and_ that the `LCBUV_EMBEDDED_SOURCE` macro
+ * is defined for both the compiled object and any code in your application
+ * that is using the module, thus:
+ *
+ * @code{.c}
+ * //uv-stub.c
+ * #include <libcouchbase/plugins/io/libuv/plugin-libuv.c>
+ * @endcode
+ *
+ * Then, in your application
+ * @code{.c}
+ * #include <libcouchbase/libuv_io_opts.h>
+ *
+ * lcb_io_opt_t *io;
+ * lcbuv_options_t options;
+ * options.v.v0.loop = uv_default_loop();
+ * options.v.v0.startsop_noop = 1;
+ * lcb_create_libuv_io_opts(0, &io, &options);
+ * @endcode
+ *
+ * And then compile as
+ * @code
+ * $ gcc -o myapp uv-stub.c main.c -DLCBUV_EMBEDDED_SOURCE
+ * @endcode
+ *
+ * @addtogroup lcb-libuv
+ * @{
+ */
+
+#ifndef LCB_PLUGIN_UV_H
+#define LCB_PLUGIN_UV_H
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <libcouchbase/couchbase.h>
+#include <uv.h>
+
+#ifdef LCBUV_EMBEDDED_SOURCE
+#define LCBUV_API
+#else
+#define LCBUV_API LIBCOUCHBASE_API
+#endif
+
+    /**Options passed to the iops constructure. You will most likely want
+     * to set the 'startsop_noop' field to true if you are using an async
+     * application.*/
+    typedef struct lcbuv_options_st {
+        int version;
+        union {
+            struct {
+                /** External loop to be used (if not default) */
+                uv_loop_t *loop;
+
+                /** Whether run_event_loop/stop_event_loop should do anything */
+                int startsop_noop;
+            } v0;
+        } v;
+    } lcbuv_options_t;
+
+    /**
+     * Use this if using an existing uv_loop_t
+     * @param version Set this to `0`
+     * @param [out] io a pointer to an io pointer. Will be populated on success
+     * @param options the options to be passed. From libcouchbase this is a
+     * `void*` parameter.
+     */
+    LCBUV_API
+    lcb_error_t lcb_create_libuv_io_opts(int version,
+                                         lcb_io_opt_t *io,
+                                         lcbuv_options_t *options);
+
+#ifdef __cplusplus
+}
+#endif
+#endif
+
+/**@}*/

data/ext/libcouchbase/plugins/io/libuv/plugin-internal.h

@@ -0,0 +1,155 @@
+/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+/*
+ *     Copyright 2013 Couchbase, Inc.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+#ifndef LCBUV_PLUGIN_INTERNAL_H
+#define LCBUV_PLUGIN_INTERNAL_H
+
+#include <libcouchbase/couchbase.h>
+#include <uv.h>
+
+#include <errno.h>
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#ifndef _WIN32
+#include <unistd.h>
+#endif
+
+#include "libuv_compat.h"
+
+#ifdef LCBUV_EMBEDDED_SOURCE
+#include <libcouchbase/libuv_io_opts.h>
+#else
+/** Load definitions from inside */
+#include "libuv_io_opts.h"
+#endif
+
+typedef void (*v0_callback_t)(lcb_socket_t, short, void *);
+typedef void (*generic_callback_t)(void);
+
+
+/**
+ * These structures come about the limitation that with -Werror -Wextra
+ * compilation fails because strictly speaking, a function pointer isn't
+ * convertible to a normal pointer.
+ */
+
+/**
+ * Macro - sometimes we might want to use the ->data field?
+ */
+#define CbREQ(mr) (mr)->callback
+typedef struct {
+    uv_tcp_t t;
+    lcb_ioC_read2_callback callback;
+} my_tcp_t;
+
+/**
+ * Wrapper for lcb_sockdata_t
+ */
+typedef struct {
+    lcb_sockdata_t base;
+
+    /**
+     * UV tcp handle. This is also a uv_stream_t.
+     * ->data field contains the read callback
+     */
+    my_tcp_t tcp;
+
+    /** Reference count */
+    unsigned int refcount;
+
+    /** Flag indicating whether uv_close has already been called  on the handle */
+    unsigned char uv_close_called;
+
+    lcb_IOV iov;
+    void *rdarg;
+
+    struct {
+        int read;
+        int write;
+    } pending;
+
+} my_sockdata_t;
+
+
+typedef struct {
+    uv_write_t w;
+    lcb_ioC_write2_callback callback;
+    my_sockdata_t *sock;
+} my_write_t;
+
+
+typedef struct {
+    struct lcb_io_opt_st base;
+    uv_loop_t *loop;
+
+    /** Refcount. When this hits zero we free this */
+    unsigned int iops_refcount;
+
+    /** Whether using a user-initiated loop */
+    int external_loop;
+
+    /** whether start/stop are noops */
+    int startstop_noop;
+
+    /** for 0.8 only, whether to stop */
+    int do_stop;
+} my_iops_t;
+
+typedef struct {
+    uv_timer_t uvt;
+    v0_callback_t callback;
+    void *cb_arg;
+    my_iops_t *parent;
+} my_timer_t;
+
+typedef struct {
+    union {
+        uv_connect_t conn;
+        uv_idle_t idle;
+    } uvreq;
+
+    union {
+        lcb_io_connect_cb conn;
+        generic_callback_t cb_;
+    } cb;
+
+    my_sockdata_t *socket;
+} my_uvreq_t;
+
+/******************************************************************************
+ ******************************************************************************
+ ** Common Macros                                                            **
+ ******************************************************************************
+ ******************************************************************************/
+#define PTR_FROM_FIELD(t, p, fld) ((t*)(void*)((char*)p-(offsetof(t, fld))))
+
+#define incref_iops(io) (io)->iops_refcount++
+
+#ifdef _WIN32
+  typedef ULONG lcb_uvbuf_len_t;
+#else
+  typedef size_t lcb_uvbuf_len_t;
+#endif
+
+#ifndef INVALID_SOCKET
+#define INVALID_SOCKET -1
+#endif
+
+#endif