mt-libcouchbase 1.4.0

data/ext/libcouchbase/tests/htparse/t_basic.cc

@@ -0,0 +1,173 @@
+#include <gtest/gtest.h>
+#include <lcbht/lcbht.h>
+#include <sstream>
+#include <map>
+#include "settings.h"
+
+using std::string;
+using namespace lcb::htparse;
+
+class HtparseTest : public ::testing::Test {
+};
+
+TEST_F(HtparseTest, testBasic)
+{
+    lcb_settings *settings = lcb_settings_new();
+    // Allocate a parser
+    Parser *parser = new Parser(settings);
+    ASSERT_FALSE(parser == NULL);
+
+    // Feed the parser some stuff
+    unsigned state;
+
+    string buf;
+    buf = "HTTP/1.0 200 OK\r\n";
+
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_EQ(0, state);
+
+    buf = "Connec";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_EQ(0, state);
+    buf = "tion: Keep-Alive\r\n";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_EQ(0, state);
+    buf += "Content-Length: 5\r\n\r\n";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_EQ(Parser::S_HEADER|Parser::S_HTSTATUS, state);
+
+    Response& resp = parser->get_cur_response();
+    ASSERT_EQ(200, resp.status);
+
+    // Add some data into the body
+    buf = "H";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_EQ("H", resp.body);
+
+    buf = "ello";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_DONE);
+    ASSERT_EQ("Hello", resp.body);
+
+    // Now find the header
+    delete parser;
+    lcb_settings_unref(settings);
+}
+
+TEST_F(HtparseTest, testHeaderFunctions)
+{
+    lcb_settings *settings = lcb_settings_new();
+    Parser *parser = new Parser(settings);
+
+    string buf = "HTTP/1.0 200 OK\r\n"
+            "Connection: keep-alive\r\n"
+            "X-Server: dummy/1.0\r\n"
+            "Content-Type: application/json\r\n"
+            "Content-Length: 0\r\n"
+            "\r\n";
+    unsigned state;
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_DONE);
+
+    Response &resp = parser->get_cur_response();
+    // Now fine the header value stuff
+    ASSERT_STREQ("keep-alive", resp.get_header_value("Connection"));
+    ASSERT_STREQ("dummy/1.0", resp.get_header_value("X-Server"));
+    ASSERT_STREQ("application/json", resp.get_header_value("Content-Type"));
+    delete parser;
+    lcb_settings_unref(settings);
+}
+
+TEST_F(HtparseTest, testParseErrors)
+{
+    lcb_settings *settings = lcb_settings_new();
+    Parser *parser = new Parser(settings);
+
+    string buf = "blahblahblah";
+    unsigned state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_ERROR);
+    delete parser;
+    lcb_settings_unref(settings);
+}
+
+
+TEST_F(HtparseTest, testParseExtended)
+{
+    lcb_settings *settings = lcb_settings_new();
+    Parser *parser = new Parser(settings);
+
+    const char *body;
+    unsigned nbody, nused;
+
+    string buf = "HTTP/1.0 200 OK\r\n"
+            "Connection: keep-alive\r\n"
+            "Content-Length: 5\r\n";
+
+    unsigned state;
+    state = parser->parse_ex(buf.c_str(), buf.size(), &nused, &nbody, &body);
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_EQ(NULL, body);
+    ASSERT_EQ(buf.size(), nused);
+    ASSERT_EQ(0, nbody);
+
+    Response& resp = parser->get_cur_response();
+    buf = "\r\nHello";
+    // Feed the buffer
+    state = parser->parse_ex(buf.c_str(), buf.size(), &nused, &nbody, &body);
+    ASSERT_EQ(0, state & Parser::S_DONE);
+    ASSERT_EQ(5, nbody);
+    ASSERT_FALSE(NULL == body);
+    ASSERT_STREQ("Hello", body);
+    ASSERT_EQ(buf.size()-1, nused);
+
+    size_t off = nused;
+
+    // Parse again
+    state = parser->parse_ex(buf.c_str()+off, buf.size()-off, &nused, &nbody, &body);
+    ASSERT_EQ(nused, buf.size()-off);
+    ASSERT_TRUE(body == NULL);
+    ASSERT_EQ(0, nbody);
+    ASSERT_NE(0, state & Parser::S_DONE);
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_EQ(0, resp.body.size());
+    delete parser;
+    lcb_settings_unref(settings);
+}
+
+TEST_F(HtparseTest, testCanKeepalive)
+{
+    lcb_settings *settings = lcb_settings_new();
+    Parser *parser = new Parser(settings);
+    string buf = "HTTP/1.0 200 OK\r\n"
+            "Content-Length: 0\r\n"
+            "\r\n";
+    unsigned state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_DONE);
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_FALSE(parser->can_keepalive());
+
+    // Use HTTP/1.1 with Connection: close
+    parser->reset();
+    buf = "HTTP/1.1 200 OK\r\n"
+            "Content-Length: 0\r\n"
+            "Connection: close\r\n"
+            "\r\n";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_DONE);
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_FALSE(parser->can_keepalive());
+
+    parser->reset();
+    // Default HTTP/1.1
+    buf = "HTTP/1.1 200 OK\r\n"
+            "Content-Length: 0\r\n"
+            "\r\n";
+    state = parser->parse(buf.c_str(), buf.size());
+    ASSERT_NE(0, state & Parser::S_DONE);
+    ASSERT_EQ(0, state & Parser::S_ERROR);
+    ASSERT_TRUE(parser->can_keepalive());
+
+    delete parser;
+    lcb_settings_unref(settings);
+}

data/ext/libcouchbase/tests/ioserver/connection.cc

@@ -0,0 +1,166 @@
+#include "ioserver.h"
+#include <cassert>
+#include <algorithm>
+
+using namespace LCBTest;
+
+extern "C" {
+static void
+client_runfunc(void *arg)
+{
+    TestConnection *conn = (TestConnection *)arg;
+    conn->_doRun();
+}
+}
+
+void TestConnection::_doRun() { run(); }
+
+void
+TestConnection::setCommon(void *src, void **target)
+{
+    mutex.lock();
+    assert(*target == NULL);
+    *target = src;
+    char dummy = 0;
+    ctlfd_user->send(&dummy, 1);
+    mutex.unlock();
+}
+
+void
+TestConnection::sendData()
+{
+    f_send->startUpdate();
+    do {
+#ifdef _WIN32
+        const char *outbuf;
+#else
+        void *outbuf;
+#endif
+        size_t n = f_send->getBuf((void**)&outbuf);
+        ssize_t nw = datasock->send(outbuf, n);
+        if (nw < 0) {
+            f_send->bail();
+        } else {
+            f_send->setSent(nw);
+        }
+    } while (!f_send->shouldEnd());
+    f_send->endUpdate();
+    f_send = NULL;
+}
+
+void
+TestConnection::recvData()
+{
+    f_recv->startUpdate();
+    char buf[32768];
+
+    do {
+        size_t required = f_recv->getRequired();
+        size_t rdsize = std::min(required, sizeof(buf));
+        ssize_t nr = datasock->recv(buf, rdsize);
+        if (nr < 0) {
+            f_recv->bail();
+        } else {
+            f_recv->setReceived(buf, nr);
+        }
+    } while (!f_recv->shouldEnd());
+
+    f_recv->endUpdate();
+    f_recv = NULL;
+}
+
+void
+TestConnection::handleClose()
+{
+    f_close->startUpdate();
+    datasock->close();
+    f_close->setDone();
+    f_close->endUpdate();
+    f_close = NULL;
+}
+
+void
+TestConnection::run()
+{
+    char dummy = 0;
+    mutex.lock();
+    ctlfd_loop = ctlfd_lsn->acceptClient();
+    initcond.signal();
+    mutex.unlock();
+
+    while (::recv(*ctlfd_loop, &dummy, 1, 0) == 1) {
+        mutex.lock();
+        if (f_close && f_close->getType() == CloseFuture::BEFORE_IO) {
+            handleClose();
+        }
+
+        if (f_send) {
+            sendData();
+        }
+        if (f_recv) {
+            recvData();
+        }
+
+        if (f_close && f_close->getType() == CloseFuture::AFTER_IO) {
+            handleClose();
+        }
+        mutex.unlock();
+    }
+
+    mutex.lock();
+    if (f_recv) {
+        f_recv->updateFailed();
+        f_recv = NULL;
+    }
+    if (f_send) {
+        f_send->updateFailed();
+        f_send = NULL;
+    }
+    if (f_close) {
+        f_close->updateFailed();
+        f_close = NULL;
+    }
+    mutex.unlock();
+}
+
+TestConnection::TestConnection(TestServer *server, SockFD *newsock)
+{
+    datasock = newsock;
+    datasock->loadRemoteAddr();
+    ctlfd_lsn = SockFD::newListener();
+    ctlfd_loop = NULL;
+
+    f_send = NULL;
+    f_recv = NULL;
+    f_close = NULL;
+    parent = server;
+
+    thr = new Thread(client_runfunc, this);
+    ctlfd_user = SockFD::newClient(ctlfd_lsn);
+    ctlfd_user->setNodelay(true);
+
+    mutex.lock();
+    while (!ctlfd_loop) {
+        initcond.wait(mutex);
+    }
+    mutex.unlock();
+}
+
+TestConnection::~TestConnection()
+{
+    ctlfd_loop->close();
+    ctlfd_user->close();
+    ctlfd_lsn->close();
+    datasock->close();
+    // We don't want to explicitly call join() here since that
+    // gets called in the destructor.  This is unncessary
+    // and broken on musl.
+    // thr->join();
+    delete thr;
+    mutex.close();
+    initcond.close();
+    delete ctlfd_loop;
+    delete ctlfd_user;
+    delete ctlfd_lsn;
+    delete datasock;
+}

data/ext/libcouchbase/tests/ioserver/future.cc

@@ -0,0 +1,50 @@
+#include "ioserver.h"
+using namespace LCBTest;
+
+Future::Future()
+{
+    failed = false;
+}
+
+void
+Future::wait()
+{
+    mutex.lock();
+    while (!isDone() && !failed) {
+        cond.wait(mutex);
+    }
+    mutex.unlock();
+}
+
+Future::~Future()
+{
+    mutex.close();
+    cond.close();
+}
+
+void
+Future::startUpdate()
+{
+    mutex.lock();
+}
+
+void
+Future::endUpdate()
+{
+    if (shouldEnd()) {
+        cond.signal();
+    }
+    mutex.unlock();
+}
+
+bool
+Future::checkDone()
+{
+    bool ret;
+    if (!mutex.tryLock()) {
+        return false;
+    }
+    ret = isDone();
+    mutex.unlock();
+    return ret;
+}

data/ext/libcouchbase/tests/ioserver/ioserver.cc

@@ -0,0 +1,117 @@
+#include <cassert>
+#include <sys/types.h>
+#include "ioserver.h"
+using namespace LCBTest;
+using std::list;
+
+extern "C" {
+static void
+server_runfunc(void *arg)
+{
+    TestServer *server = (TestServer *)arg;
+    server->run();
+}
+}
+
+void
+TestServer::run()
+{
+    if (closed) {
+        return;
+    }
+
+    fd_set fds;
+    struct timeval tmout = { 1, 0 };
+
+    FD_ZERO(&fds);
+    FD_SET(*lsn, &fds);
+
+    while (!closed) {
+        struct sockaddr_in newaddr;
+        socklen_t naddr = sizeof(newaddr);
+
+        if (select(*lsn + 1, &fds, NULL, NULL, &tmout) == 1) {
+            int newsock = accept(*lsn, (struct sockaddr *)&newaddr, &naddr);
+
+            if (newsock == -1) {
+                break;
+            }
+
+            TestConnection *newconn = new TestConnection(this, factory(newsock));
+            startConnection(newconn);
+        }
+    }
+}
+
+void
+TestServer::startConnection(TestConnection *conn)
+{
+    mutex.lock();
+    if (isClosed()) {
+        conn->close();
+        delete conn;
+    } else {
+        conns.push_back(conn);
+    }
+    mutex.unlock();
+}
+
+TestServer::TestServer()
+{
+    lsn = SockFD::newListener();
+    closed = false;
+    factory = plainSocketFactory;
+
+    // Now spin up a thread to start the accept loop
+    thr = new Thread(server_runfunc, this);
+}
+
+TestServer::~TestServer()
+{
+    close();
+    mutex.lock();
+    std::list<TestConnection*>::iterator iter = conns.begin();
+    for (; iter != conns.end(); ++iter) {
+        (*iter)->close();
+        delete *iter;
+    }
+    mutex.unlock();
+    // We don't want to explicitly call join() here since that
+    // gets called in the destructor.  This is unncessary
+    // and broken on musl.
+    // thr->join();
+    delete thr;
+    mutex.close();
+    delete lsn;
+}
+
+std::string
+TestServer::getPortString()
+{
+    char buf[4096];
+    sprintf(buf, "%d", lsn->getLocalPort());
+    return std::string(buf);
+}
+
+TestConnection *
+TestServer::findConnection(uint16_t port)
+{
+    TestConnection *ret = NULL;
+    list<TestConnection *>::iterator iter;
+
+    while (ret == NULL) {
+        sched_yield();
+        mutex.lock();
+        iter = conns.begin();
+
+        for (; iter != conns.end(); ++iter) {
+            if ((*iter)->getPeerPort() == port) {
+                ret = *iter;
+                break;
+            }
+        }
+        mutex.unlock();
+    };
+
+    return ret;
+}