mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-06-15 17:03:45 +02:00
2.0 compiles with postgres/redis controller code
Probably doesn't work yet
This commit is contained in:
parent
2bceabdfa5
commit
17f0dc9ba2
54 changed files with 507 additions and 9267 deletions
|
@ -27,17 +27,22 @@ if(NOT CMAKE_BUILD_TYPE AND NOT CMAKE_CONFIGURATION_TYPES)
|
||||||
set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
|
set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
#option(BUILD_CENTRAL_CONTROLLER "Build ZeroTier Central Controller" OFF)
|
option(BUILD_CENTRAL_CONTROLLER "Build ZeroTier Central Controller" OFF)
|
||||||
#if (BUILD_CENTRAL_CONTROLLER)
|
if(BUILD_CENTRAL_CONTROLLER)
|
||||||
# find_package(PostgreSQL REQUIRED)
|
find_package(PkgConfig REQUIRED)
|
||||||
# set(ENABLE_SSL_SUPPORT OFF)
|
if(APPLE)
|
||||||
# set(BUILD_SHARED_LIBS OFF)
|
set(CMAKE_PREFIX_PATH
|
||||||
# set(BUILD_EXAMPLES OFF)
|
${CMAKE_PREFIX_PATH}
|
||||||
# set(BUILD_TOOLS OFF)
|
/usr/local/opt/libpq
|
||||||
# set(BUILD_TESTS OFF)
|
/usr/local/lib
|
||||||
# set(BUILD_API_DOCS OFF)
|
)
|
||||||
# add_subdirectory("ext/librabbitmq")
|
endif(APPLE)
|
||||||
#endif(BUILD_CENTRAL_CONTROLLER)
|
find_package(PostgreSQL REQUIRED)
|
||||||
|
|
||||||
|
pkg_check_modules(hiredis REQUIRED IMPORTED_TARGET hiredis)
|
||||||
|
|
||||||
|
add_subdirectory(controller/thirdparty/redis-plus-plus-1.1.1)
|
||||||
|
endif(BUILD_CENTRAL_CONTROLLER)
|
||||||
|
|
||||||
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
||||||
add_definitions(-DZT_DEBUG)
|
add_definitions(-DZT_DEBUG)
|
||||||
|
@ -95,15 +100,15 @@ if (
|
||||||
add_compile_options(-maes -mrdrnd -mpclmul -msse -msse2 -mssse3)
|
add_compile_options(-maes -mrdrnd -mpclmul -msse -msse2 -mssse3)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
if(BUILD_CENTRAL_CONTROLLER)
|
||||||
|
add_definitions(-DZT_CONTROLLER_USE_LIBPQ=1)
|
||||||
|
endif(BUILD_CENTRAL_CONTROLLER)
|
||||||
|
|
||||||
add_subdirectory(core)
|
add_subdirectory(core)
|
||||||
add_subdirectory(controller)
|
add_subdirectory(controller)
|
||||||
add_subdirectory(osdep)
|
add_subdirectory(osdep)
|
||||||
add_subdirectory(serviceiocore)
|
add_subdirectory(serviceiocore)
|
||||||
|
|
||||||
#if(BUILD_CENTRAL_CONTROLLER)
|
|
||||||
# set(libs ${libs} rabbitmq-static ${PostgreSQL_LIBRARIES})
|
|
||||||
#endif(BUILD_CENTRAL_CONTROLLER)
|
|
||||||
|
|
||||||
set(
|
set(
|
||||||
zt_core
|
zt_core
|
||||||
zt_osdep
|
zt_osdep
|
||||||
|
@ -117,4 +122,5 @@ add_custom_target(zerotier ALL
|
||||||
BYPRODUCTS zerotier
|
BYPRODUCTS zerotier
|
||||||
)
|
)
|
||||||
add_dependencies(zerotier zt_osdep zt_core zt_controller zt_service_io_core)
|
add_dependencies(zerotier zt_osdep zt_core zt_controller zt_service_io_core)
|
||||||
|
|
||||||
set(ADDITIONAL_MAKE_CLEAN_FILES zerotier)
|
set(ADDITIONAL_MAKE_CLEAN_FILES zerotier)
|
||||||
|
|
6
Makefile
6
Makefile
|
@ -6,13 +6,13 @@ all: setup
|
||||||
cd ${BUILDDIR} && $(MAKE) -j$(shell getconf _NPROCESSORS_ONLN)
|
cd ${BUILDDIR} && $(MAKE) -j$(shell getconf _NPROCESSORS_ONLN)
|
||||||
|
|
||||||
setup:
|
setup:
|
||||||
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Release
|
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Release ${CMAKE_ARGS}
|
||||||
|
|
||||||
setup-debug:
|
setup-debug:
|
||||||
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Debug
|
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Debug ${CMAKE_ARGS}
|
||||||
|
|
||||||
debug:
|
debug:
|
||||||
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Debug && $(MAKE)
|
mkdir -p ${BUILDDIR} && cd ${BUILDDIR} && cmake .. -DCMAKE_BUILD_TYPE=Debug ${CMAKE_ARGS} && $(MAKE)
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
rm -rf ${BUILDDIR} cmake-build-*
|
rm -rf ${BUILDDIR} cmake-build-*
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
cmake_minimum_required(VERSION 2.8)
|
cmake_minimum_required(VERSION 2.8)
|
||||||
project(zt_controller)
|
project(zt_controller)
|
||||||
|
|
||||||
|
|
||||||
if(WIN32)
|
if(WIN32)
|
||||||
add_definitions(-DNOMINMAX)
|
add_definitions(-DNOMINMAX)
|
||||||
endif(WIN32)
|
endif(WIN32)
|
||||||
|
@ -11,7 +12,6 @@ set(ctl_src
|
||||||
EmbeddedNetworkController.cpp
|
EmbeddedNetworkController.cpp
|
||||||
FileDB.cpp
|
FileDB.cpp
|
||||||
LFDB.cpp
|
LFDB.cpp
|
||||||
RabbitMQ.cpp
|
|
||||||
)
|
)
|
||||||
|
|
||||||
set(ctl_hdr
|
set(ctl_hdr
|
||||||
|
@ -20,17 +20,17 @@ set(ctl_hdr
|
||||||
EmbeddedNetworkController.hpp
|
EmbeddedNetworkController.hpp
|
||||||
FileDB.hpp
|
FileDB.hpp
|
||||||
LFDB.hpp
|
LFDB.hpp
|
||||||
RabbitMQ.hpp
|
Redis.hpp
|
||||||
)
|
)
|
||||||
|
|
||||||
if(BUILD_CENTRAL_CONTROLLER)
|
if(BUILD_CENTRAL_CONTROLLER)
|
||||||
add_definitions(-DZT_CONTROLLER_USE_LIBPQ)
|
|
||||||
include_directories("../ext/librabbitmq/librabbitmq" ${PostgreSQL_INCLUDE_DIRS})
|
|
||||||
|
|
||||||
set(ctl_src ${ctl_src} PostgreSQL.cpp)
|
set(ctl_src ${ctl_src} PostgreSQL.cpp)
|
||||||
set(ctl_hdr ${ctl_hdr} PostgreSQL.hpp)
|
set(ctl_hdr ${ctl_hdr} PostgreSQL.hpp)
|
||||||
endif(BUILD_CENTRAL_CONTROLLER)
|
endif(BUILD_CENTRAL_CONTROLLER)
|
||||||
|
|
||||||
add_library(${PROJECT_NAME} STATIC ${ctl_src} ${ctl_hdr})
|
add_library(${PROJECT_NAME} STATIC ${ctl_src} ${ctl_hdr} ${PostgreSQL_LIBRARIES} ${STATIC_LIB} ${hiredis_libraries})
|
||||||
|
target_include_directories(${PROJECT_NAME}
|
||||||
|
PUBLIC ${hiredis_INCLUDE_DIRS}/../ ${CMAKE_CURRENT_LIST_DIR}/thirdparty/redis-plus-plus-1.1.1/src/sw ${PostgreSQL_INCLUDE_DIRS}
|
||||||
|
PRIVATE ${CMAKE_BINARY_DIR}/core)
|
||||||
target_compile_features(${PROJECT_NAME} PUBLIC cxx_std_11)
|
target_compile_features(${PROJECT_NAME} PUBLIC cxx_std_11)
|
||||||
|
|
||||||
|
|
|
@ -446,7 +446,7 @@ static bool _parseRule(json &r,ZT_VirtualNetworkRule &rule)
|
||||||
|
|
||||||
} // anonymous namespace
|
} // anonymous namespace
|
||||||
|
|
||||||
EmbeddedNetworkController::EmbeddedNetworkController(Node *node,const char *ztPath,const char *dbPath, int listenPort, MQConfig *mqc) :
|
EmbeddedNetworkController::EmbeddedNetworkController(Node *node,const char *ztPath,const char *dbPath, int listenPort, RedisConfig *rc) :
|
||||||
_startTime(OSUtils::now()),
|
_startTime(OSUtils::now()),
|
||||||
_listenPort(listenPort),
|
_listenPort(listenPort),
|
||||||
_node(node),
|
_node(node),
|
||||||
|
@ -454,7 +454,7 @@ EmbeddedNetworkController::EmbeddedNetworkController(Node *node,const char *ztPa
|
||||||
_path(dbPath),
|
_path(dbPath),
|
||||||
_sender((NetworkController::Sender *)0),
|
_sender((NetworkController::Sender *)0),
|
||||||
_db(this),
|
_db(this),
|
||||||
_mqc(mqc)
|
_rc(rc)
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -476,7 +476,7 @@ void EmbeddedNetworkController::init(const Identity &signingId,Sender *sender)
|
||||||
|
|
||||||
#ifdef ZT_CONTROLLER_USE_LIBPQ
|
#ifdef ZT_CONTROLLER_USE_LIBPQ
|
||||||
if ((_path.length() > 9)&&(_path.substr(0,9) == "postgres:")) {
|
if ((_path.length() > 9)&&(_path.substr(0,9) == "postgres:")) {
|
||||||
_db.addDB(std::shared_ptr<DB>(new PostgreSQL(_signingId,_path.substr(9).c_str(), _listenPort, _mqc)));
|
_db.addDB(std::shared_ptr<DB>(new PostgreSQL(_signingId,_path.substr(9).c_str(), _listenPort, _rc)));
|
||||||
} else {
|
} else {
|
||||||
#endif
|
#endif
|
||||||
_db.addDB(std::shared_ptr<DB>(new FileDB(_path.c_str())));
|
_db.addDB(std::shared_ptr<DB>(new FileDB(_path.c_str())));
|
||||||
|
|
|
@ -44,7 +44,7 @@ namespace ZeroTier {
|
||||||
|
|
||||||
class Node;
|
class Node;
|
||||||
|
|
||||||
struct MQConfig;
|
struct RedisConfig;
|
||||||
|
|
||||||
class EmbeddedNetworkController : public NetworkController,public DB::ChangeListener
|
class EmbeddedNetworkController : public NetworkController,public DB::ChangeListener
|
||||||
{
|
{
|
||||||
|
@ -54,7 +54,7 @@ public:
|
||||||
* @param ztPath ZeroTier base path
|
* @param ztPath ZeroTier base path
|
||||||
* @param dbPath Database path (file path or database credentials)
|
* @param dbPath Database path (file path or database credentials)
|
||||||
*/
|
*/
|
||||||
EmbeddedNetworkController(Node *node,const char *ztPath,const char *dbPath, int listenPort, MQConfig *mqc = NULL);
|
EmbeddedNetworkController(Node *node,const char *ztPath,const char *dbPath, int listenPort, RedisConfig *rc = NULL);
|
||||||
virtual ~EmbeddedNetworkController();
|
virtual ~EmbeddedNetworkController();
|
||||||
|
|
||||||
virtual void init(const Identity &signingId,Sender *sender);
|
virtual void init(const Identity &signingId,Sender *sender);
|
||||||
|
@ -147,7 +147,7 @@ private:
|
||||||
std::unordered_map< _MemberStatusKey,_MemberStatus,_MemberStatusHash > _memberStatus;
|
std::unordered_map< _MemberStatusKey,_MemberStatus,_MemberStatusHash > _memberStatus;
|
||||||
std::mutex _memberStatus_l;
|
std::mutex _memberStatus_l;
|
||||||
|
|
||||||
MQConfig *_mqc;
|
RedisConfig *_rc;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace ZeroTier
|
} // namespace ZeroTier
|
||||||
|
|
|
@ -1,10 +1,10 @@
|
||||||
/*
|
/*
|
||||||
* Copyright (c)2013-2020 ZeroTier, Inc.
|
* Copyright (c)2019 ZeroTier, Inc.
|
||||||
*
|
*
|
||||||
* Use of this software is governed by the Business Source License included
|
* Use of this software is governed by the Business Source License included
|
||||||
* in the LICENSE.TXT file in the project's root directory.
|
* in the LICENSE.TXT file in the project's root directory.
|
||||||
*
|
*
|
||||||
* Change Date: 2024-01-01
|
* Change Date: 2023-01-01
|
||||||
*
|
*
|
||||||
* On the date above, in accordance with the Business Source License, use
|
* On the date above, in accordance with the Business Source License, use
|
||||||
* of this software will be governed by version 2.0 of the Apache License.
|
* of this software will be governed by version 2.0 of the Apache License.
|
||||||
|
@ -17,13 +17,13 @@
|
||||||
|
|
||||||
#include "../core/Constants.hpp"
|
#include "../core/Constants.hpp"
|
||||||
#include "EmbeddedNetworkController.hpp"
|
#include "EmbeddedNetworkController.hpp"
|
||||||
#include "RabbitMQ.hpp"
|
#include "version.h"
|
||||||
#include "../version.h"
|
#include "Redis.hpp"
|
||||||
|
|
||||||
#include <libpq-fe.h>
|
#include <libpq-fe.h>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <amqp.h>
|
#include <climits>
|
||||||
#include <amqp_tcp_socket.h>
|
|
||||||
|
|
||||||
using json = nlohmann::json;
|
using json = nlohmann::json;
|
||||||
|
|
||||||
|
@ -69,7 +69,11 @@ std::string join(const std::vector<std::string> &elements, const char * const se
|
||||||
|
|
||||||
using namespace ZeroTier;
|
using namespace ZeroTier;
|
||||||
|
|
||||||
PostgreSQL::PostgreSQL(const Identity &myId, const char *path, int listenPort, MQConfig *mqc)
|
using Attrs = std::vector<std::pair<std::string, std::string>>;
|
||||||
|
using Item = std::pair<std::string, Attrs>;
|
||||||
|
using ItemStream = std::vector<Item>;
|
||||||
|
|
||||||
|
PostgreSQL::PostgreSQL(const Identity &myId, const char *path, int listenPort, RedisConfig *rc)
|
||||||
: DB()
|
: DB()
|
||||||
, _myId(myId)
|
, _myId(myId)
|
||||||
, _myAddress(myId.address())
|
, _myAddress(myId.address())
|
||||||
|
@ -78,7 +82,9 @@ PostgreSQL::PostgreSQL(const Identity &myId, const char *path, int listenPort, M
|
||||||
, _run(1)
|
, _run(1)
|
||||||
, _waitNoticePrinted(false)
|
, _waitNoticePrinted(false)
|
||||||
, _listenPort(listenPort)
|
, _listenPort(listenPort)
|
||||||
, _mqc(mqc)
|
, _rc(rc)
|
||||||
|
, _redis(NULL)
|
||||||
|
, _cluster(NULL)
|
||||||
{
|
{
|
||||||
char myAddress[64];
|
char myAddress[64];
|
||||||
_myAddressStr = myId.address().toString(myAddress);
|
_myAddressStr = myId.address().toString(myAddress);
|
||||||
|
@ -108,13 +114,37 @@ PostgreSQL::PostgreSQL(const Identity &myId, const char *path, int listenPort, M
|
||||||
fprintf(stderr, "Central database schema version too low. This controller version requires a minimum schema version of %d. Please upgrade your Central instance", DB_MINIMUM_VERSION);
|
fprintf(stderr, "Central database schema version too low. This controller version requires a minimum schema version of %d. Please upgrade your Central instance", DB_MINIMUM_VERSION);
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
res = NULL;
|
res = NULL;
|
||||||
|
|
||||||
|
if (_rc != NULL) {
|
||||||
|
sw::redis::ConnectionOptions opts;
|
||||||
|
sw::redis::ConnectionPoolOptions poolOpts;
|
||||||
|
opts.host = _rc->hostname;
|
||||||
|
opts.port = _rc->port;
|
||||||
|
opts.password = _rc->password;
|
||||||
|
opts.db = 0;
|
||||||
|
poolOpts.size = 10;
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
fprintf(stderr, "Using Redis in Cluster Mode\n");
|
||||||
|
_cluster = std::make_shared<sw::redis::RedisCluster>(opts, poolOpts);
|
||||||
|
} else {
|
||||||
|
fprintf(stderr, "Using Redis in Standalone Mode\n");
|
||||||
|
_redis = std::make_shared<sw::redis::Redis>(opts, poolOpts);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_readyLock.lock();
|
||||||
|
|
||||||
|
fprintf(stderr, "[%s] NOTICE: %.10llx controller PostgreSQL waiting for initial data download..." ZT_EOL_S, ::_timestr(), (unsigned long long)_myAddress.toInt());
|
||||||
|
_waitNoticePrinted = true;
|
||||||
|
|
||||||
|
initializeNetworks(conn);
|
||||||
|
initializeMembers(conn);
|
||||||
|
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
conn = NULL;
|
conn = NULL;
|
||||||
|
|
||||||
_readyLock.lock();
|
|
||||||
_heartbeatThread = std::thread(&PostgreSQL::heartbeat, this);
|
_heartbeatThread = std::thread(&PostgreSQL::heartbeat, this);
|
||||||
_membersDbWatcher = std::thread(&PostgreSQL::membersDbWatcher, this);
|
_membersDbWatcher = std::thread(&PostgreSQL::membersDbWatcher, this);
|
||||||
_networksDbWatcher = std::thread(&PostgreSQL::networksDbWatcher, this);
|
_networksDbWatcher = std::thread(&PostgreSQL::networksDbWatcher, this);
|
||||||
|
@ -132,21 +162,17 @@ PostgreSQL::~PostgreSQL()
|
||||||
_heartbeatThread.join();
|
_heartbeatThread.join();
|
||||||
_membersDbWatcher.join();
|
_membersDbWatcher.join();
|
||||||
_networksDbWatcher.join();
|
_networksDbWatcher.join();
|
||||||
|
_commitQueue.stop();
|
||||||
for (int i = 0; i < ZT_CENTRAL_CONTROLLER_COMMIT_THREADS; ++i) {
|
for (int i = 0; i < ZT_CENTRAL_CONTROLLER_COMMIT_THREADS; ++i) {
|
||||||
_commitThread[i].join();
|
_commitThread[i].join();
|
||||||
}
|
}
|
||||||
_onlineNotificationThread.join();
|
_onlineNotificationThread.join();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
bool PostgreSQL::waitForReady()
|
bool PostgreSQL::waitForReady()
|
||||||
{
|
{
|
||||||
while (_ready < 2) {
|
while (_ready < 2) {
|
||||||
if (!_waitNoticePrinted) {
|
|
||||||
_waitNoticePrinted = true;
|
|
||||||
fprintf(stderr, "[%s] NOTICE: %.10llx controller PostgreSQL waiting for initial data download..." ZT_EOL_S, ::_timestr(), (unsigned long long)_myAddress.toInt());
|
|
||||||
}
|
|
||||||
_readyLock.lock();
|
_readyLock.lock();
|
||||||
_readyLock.unlock();
|
_readyLock.unlock();
|
||||||
}
|
}
|
||||||
|
@ -166,24 +192,24 @@ bool PostgreSQL::save(nlohmann::json &record,bool notifyListeners)
|
||||||
return false;
|
return false;
|
||||||
const std::string objtype = record["objtype"];
|
const std::string objtype = record["objtype"];
|
||||||
if (objtype == "network") {
|
if (objtype == "network") {
|
||||||
const uint64_t nwid = DB::jsonIntHex(record["id"],0ULL);
|
const uint64_t nwid = jsonIntHex(record["id"],0ULL);
|
||||||
if (nwid) {
|
if (nwid) {
|
||||||
nlohmann::json old;
|
nlohmann::json old;
|
||||||
get(nwid,old);
|
get(nwid,old);
|
||||||
if ((!old.is_object())||(!_compareRecords(old,record))) {
|
if ((!old.is_object())||(!_compareRecords(old,record))) {
|
||||||
record["revision"] = DB::jsonInt(record["revision"],0ULL) + 1ULL;
|
record["revision"] = jsonInt(record["revision"],0ULL) + 1ULL;
|
||||||
_commitQueue.post(std::pair<nlohmann::json,bool>(record,notifyListeners));
|
_commitQueue.post(std::pair<nlohmann::json,bool>(record,notifyListeners));
|
||||||
modified = true;
|
modified = true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if (objtype == "member") {
|
} else if (objtype == "member") {
|
||||||
const uint64_t nwid = DB::jsonIntHex(record["nwid"],0ULL);
|
const uint64_t nwid = jsonIntHex(record["nwid"],0ULL);
|
||||||
const uint64_t id = DB::jsonIntHex(record["id"],0ULL);
|
const uint64_t id = jsonIntHex(record["id"],0ULL);
|
||||||
if ((id)&&(nwid)) {
|
if ((id)&&(nwid)) {
|
||||||
nlohmann::json network,old;
|
nlohmann::json network,old;
|
||||||
get(nwid,network,id,old);
|
get(nwid,network,id,old);
|
||||||
if ((!old.is_object())||(!_compareRecords(old,record))) {
|
if ((!old.is_object())||(!_compareRecords(old,record))) {
|
||||||
record["revision"] = DB::jsonInt(record["revision"],0ULL) + 1ULL;
|
record["revision"] = jsonInt(record["revision"],0ULL) + 1ULL;
|
||||||
_commitQueue.post(std::pair<nlohmann::json,bool>(record,notifyListeners));
|
_commitQueue.post(std::pair<nlohmann::json,bool>(record,notifyListeners));
|
||||||
modified = true;
|
modified = true;
|
||||||
}
|
}
|
||||||
|
@ -207,12 +233,15 @@ void PostgreSQL::eraseNetwork(const uint64_t networkId)
|
||||||
tmp.first["objtype"] = "_delete_network";
|
tmp.first["objtype"] = "_delete_network";
|
||||||
tmp.second = true;
|
tmp.second = true;
|
||||||
_commitQueue.post(tmp);
|
_commitQueue.post(tmp);
|
||||||
|
nlohmann::json nullJson;
|
||||||
|
_networkChanged(tmp.first, nullJson, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::eraseMember(const uint64_t networkId, const uint64_t memberId)
|
void PostgreSQL::eraseMember(const uint64_t networkId, const uint64_t memberId)
|
||||||
{
|
{
|
||||||
char tmp2[24];
|
char tmp2[24];
|
||||||
std::pair<nlohmann::json,bool> tmp;
|
waitForReady();
|
||||||
|
std::pair<nlohmann::json,bool> tmp, nw;
|
||||||
Utils::hex(networkId, tmp2);
|
Utils::hex(networkId, tmp2);
|
||||||
tmp.first["nwid"] = tmp2;
|
tmp.first["nwid"] = tmp2;
|
||||||
Utils::hex(memberId, tmp2);
|
Utils::hex(memberId, tmp2);
|
||||||
|
@ -220,6 +249,8 @@ void PostgreSQL::eraseMember(const uint64_t networkId, const uint64_t memberId)
|
||||||
tmp.first["objtype"] = "_delete_member";
|
tmp.first["objtype"] = "_delete_member";
|
||||||
tmp.second = true;
|
tmp.second = true;
|
||||||
_commitQueue.post(tmp);
|
_commitQueue.post(tmp);
|
||||||
|
nlohmann::json nullJson;
|
||||||
|
_memberChanged(tmp.first, nullJson, true);
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress &physicalAddress)
|
void PostgreSQL::nodeIsOnline(const uint64_t networkId, const uint64_t memberId, const InetAddress &physicalAddress)
|
||||||
|
@ -240,10 +271,29 @@ void PostgreSQL::initializeNetworks(PGconn *conn)
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string setKey = "networks:{" + _myAddressStr + "}";
|
||||||
|
|
||||||
|
if (_rc != NULL) {
|
||||||
|
try {
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->del(setKey);
|
||||||
|
} else {
|
||||||
|
_redis->del(setKey);
|
||||||
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
// del can throw an error if the key doesn't exist
|
||||||
|
// swallow it and move along
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::unordered_set<std::string> networkSet;
|
||||||
|
|
||||||
const char *params[1] = {
|
const char *params[1] = {
|
||||||
_myAddressStr.c_str()
|
_myAddressStr.c_str()
|
||||||
};
|
};
|
||||||
|
|
||||||
|
fprintf(stderr, "Initializing Networks...\n");
|
||||||
|
|
||||||
PGresult *res = PQexecParams(conn, "SELECT id, EXTRACT(EPOCH FROM creation_time AT TIME ZONE 'UTC')*1000, capabilities, "
|
PGresult *res = PQexecParams(conn, "SELECT id, EXTRACT(EPOCH FROM creation_time AT TIME ZONE 'UTC')*1000, capabilities, "
|
||||||
"enable_broadcast, EXTRACT(EPOCH FROM last_modified AT TIME ZONE 'UTC')*1000, mtu, multicast_limit, name, private, remote_trace_level, "
|
"enable_broadcast, EXTRACT(EPOCH FROM last_modified AT TIME ZONE 'UTC')*1000, mtu, multicast_limit, name, private, remote_trace_level, "
|
||||||
"remote_trace_target, revision, rules, tags, v4_assign_mode, v6_assign_mode FROM ztc_network "
|
"remote_trace_target, revision, rules, tags, v4_assign_mode, v6_assign_mode FROM ztc_network "
|
||||||
|
@ -269,9 +319,12 @@ void PostgreSQL::initializeNetworks(PGconn *conn)
|
||||||
const char *nwidparam[1] = {
|
const char *nwidparam[1] = {
|
||||||
PQgetvalue(res, i, 0)
|
PQgetvalue(res, i, 0)
|
||||||
};
|
};
|
||||||
|
std::string nwid = PQgetvalue(res, i, 0);
|
||||||
|
|
||||||
config["id"] = PQgetvalue(res, i, 0);
|
networkSet.insert(nwid);
|
||||||
config["nwid"] = PQgetvalue(res, i, 0);
|
|
||||||
|
config["id"] = nwid;
|
||||||
|
config["nwid"] = nwid;
|
||||||
try {
|
try {
|
||||||
config["creationTime"] = std::stoull(PQgetvalue(res, i, 1));
|
config["creationTime"] = std::stoull(PQgetvalue(res, i, 1));
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
|
@ -384,14 +437,29 @@ void PostgreSQL::initializeNetworks(PGconn *conn)
|
||||||
|
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
|
|
||||||
|
if(!networkSet.empty()) {
|
||||||
|
if (_rc && _rc->clusterMode) {
|
||||||
|
auto tx = _cluster->transaction(_myAddressStr, true);
|
||||||
|
tx.sadd(setKey, networkSet.begin(), networkSet.end());
|
||||||
|
tx.exec();
|
||||||
|
} else if (_rc && !_rc->clusterMode) {
|
||||||
|
auto tx = _redis->transaction(true);
|
||||||
|
tx.sadd(setKey, networkSet.begin(), networkSet.end());
|
||||||
|
tx.exec();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (++this->_ready == 2) {
|
if (++this->_ready == 2) {
|
||||||
if (_waitNoticePrinted) {
|
if (_waitNoticePrinted) {
|
||||||
fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());
|
fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());
|
||||||
}
|
}
|
||||||
_readyLock.unlock();
|
_readyLock.unlock();
|
||||||
}
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
fprintf(stderr, "ERROR: Error initializing networks in Redis: %s\n", e.what());
|
||||||
|
exit(-1);
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
fprintf(stderr, "ERROR: Error initializing networks: %s", e.what());
|
fprintf(stderr, "ERROR: Error initializing networks: %s\n", e.what());
|
||||||
exit(-1);
|
exit(-1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -403,11 +471,44 @@ void PostgreSQL::initializeMembers(PGconn *conn)
|
||||||
fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));
|
fprintf(stderr, "Bad Database Connection: %s", PQerrorMessage(conn));
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
std::string setKeyBase = "network-nodes-all:{" + _myAddressStr + "}:";
|
||||||
|
|
||||||
|
if (_rc != NULL) {
|
||||||
|
std::lock_guard<std::mutex> l(_networks_l);
|
||||||
|
std::unordered_set<std::string> deletes;
|
||||||
|
for ( auto it : _networks) {
|
||||||
|
uint64_t nwid_i = it.first;
|
||||||
|
char nwidTmp[64] = {0};
|
||||||
|
OSUtils::ztsnprintf(nwidTmp, sizeof(nwidTmp), "%.16llx", nwid_i);
|
||||||
|
std::string nwid(nwidTmp);
|
||||||
|
std::string key = setKeyBase + nwid;
|
||||||
|
deletes.insert(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!deletes.empty()) {
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
auto tx = _cluster->transaction(_myAddressStr, true);
|
||||||
|
for (std::string k : deletes) {
|
||||||
|
tx.del(k);
|
||||||
|
}
|
||||||
|
tx.exec();
|
||||||
|
} else {
|
||||||
|
auto tx = _redis->transaction(true);
|
||||||
|
for (std::string k : deletes) {
|
||||||
|
tx.del(k);
|
||||||
|
}
|
||||||
|
tx.exec();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
const char *params[1] = {
|
const char *params[1] = {
|
||||||
_myAddressStr.c_str()
|
_myAddressStr.c_str()
|
||||||
};
|
};
|
||||||
|
|
||||||
|
std::unordered_map<std::string, std::string> networkMembers;
|
||||||
|
|
||||||
|
fprintf(stderr, "Initializing Members...\n");
|
||||||
PGresult *res = PQexecParams(conn,
|
PGresult *res = PQexecParams(conn,
|
||||||
"SELECT m.id, m.network_id, m.active_bridge, m.authorized, m.capabilities, EXTRACT(EPOCH FROM m.creation_time AT TIME ZONE 'UTC')*1000, m.identity, "
|
"SELECT m.id, m.network_id, m.active_bridge, m.authorized, m.capabilities, EXTRACT(EPOCH FROM m.creation_time AT TIME ZONE 'UTC')*1000, m.identity, "
|
||||||
" EXTRACT(EPOCH FROM m.last_authorized_time AT TIME ZONE 'UTC')*1000, "
|
" EXTRACT(EPOCH FROM m.last_authorized_time AT TIME ZONE 'UTC')*1000, "
|
||||||
|
@ -438,6 +539,9 @@ void PostgreSQL::initializeMembers(PGconn *conn)
|
||||||
|
|
||||||
std::string memberId(PQgetvalue(res, i, 0));
|
std::string memberId(PQgetvalue(res, i, 0));
|
||||||
std::string networkId(PQgetvalue(res, i, 1));
|
std::string networkId(PQgetvalue(res, i, 1));
|
||||||
|
|
||||||
|
networkMembers.insert(std::pair<std::string, std::string>(setKeyBase+networkId, memberId));
|
||||||
|
|
||||||
std::string ctime = PQgetvalue(res, i, 5);
|
std::string ctime = PQgetvalue(res, i, 5);
|
||||||
config["id"] = memberId;
|
config["id"] = memberId;
|
||||||
config["nwid"] = networkId;
|
config["nwid"] = networkId;
|
||||||
|
@ -530,7 +634,12 @@ void PostgreSQL::initializeMembers(PGconn *conn)
|
||||||
|
|
||||||
int n = PQntuples(r2);
|
int n = PQntuples(r2);
|
||||||
for (int j = 0; j < n; ++j) {
|
for (int j = 0; j < n; ++j) {
|
||||||
config["ipAssignments"].push_back(PQgetvalue(r2, j, 0));
|
std::string ipaddr = PQgetvalue(r2, j, 0);
|
||||||
|
std::size_t pos = ipaddr.find('/');
|
||||||
|
if (pos != std::string::npos) {
|
||||||
|
ipaddr = ipaddr.substr(0, pos);
|
||||||
|
}
|
||||||
|
config["ipAssignments"].push_back(ipaddr);
|
||||||
}
|
}
|
||||||
|
|
||||||
_memberChanged(empty, config, false);
|
_memberChanged(empty, config, false);
|
||||||
|
@ -538,12 +647,31 @@ void PostgreSQL::initializeMembers(PGconn *conn)
|
||||||
|
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
|
|
||||||
|
if (!networkMembers.empty()) {
|
||||||
|
if (_rc != NULL) {
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
auto tx = _cluster->transaction(_myAddressStr, true);
|
||||||
|
for (auto it : networkMembers) {
|
||||||
|
tx.sadd(it.first, it.second);
|
||||||
|
}
|
||||||
|
tx.exec();
|
||||||
|
} else {
|
||||||
|
auto tx = _redis->transaction(true);
|
||||||
|
for (auto it : networkMembers) {
|
||||||
|
tx.sadd(it.first, it.second);
|
||||||
|
}
|
||||||
|
tx.exec();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
if (++this->_ready == 2) {
|
if (++this->_ready == 2) {
|
||||||
if (_waitNoticePrinted) {
|
if (_waitNoticePrinted) {
|
||||||
fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());
|
fprintf(stderr,"[%s] NOTICE: %.10llx controller PostgreSQL data download complete." ZT_EOL_S,_timestr(),(unsigned long long)_myAddress.toInt());
|
||||||
}
|
}
|
||||||
_readyLock.unlock();
|
_readyLock.unlock();
|
||||||
}
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
fprintf(stderr, "ERROR: Error initializing members (redis): %s\n", e.what());
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
fprintf(stderr, "ERROR: Error initializing members: %s\n", e.what());
|
fprintf(stderr, "ERROR: Error initializing members: %s\n", e.what());
|
||||||
exit(-1);
|
exit(-1);
|
||||||
|
@ -581,14 +709,15 @@ void PostgreSQL::heartbeat()
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
exit(6);
|
exit(6);
|
||||||
}
|
}
|
||||||
|
int64_t ts = OSUtils::now();
|
||||||
if (conn) {
|
if (conn) {
|
||||||
std::string major = std::to_string(ZEROTIER_ONE_VERSION_MAJOR);
|
std::string major = std::to_string(ZEROTIER_VERSION_MAJOR);
|
||||||
std::string minor = std::to_string(ZEROTIER_ONE_VERSION_MINOR);
|
std::string minor = std::to_string(ZEROTIER_VERSION_MINOR);
|
||||||
std::string rev = std::to_string(ZEROTIER_ONE_VERSION_REVISION);
|
std::string rev = std::to_string(ZEROTIER_VERSION_REVISION);
|
||||||
std::string build = std::to_string(ZEROTIER_ONE_VERSION_BUILD);
|
std::string build = std::to_string(ZEROTIER_VERSION_BUILD);
|
||||||
std::string now = std::to_string(OSUtils::now());
|
std::string now = std::to_string(ts);
|
||||||
std::string host_port = std::to_string(_listenPort);
|
std::string host_port = std::to_string(_listenPort);
|
||||||
std::string use_rabbitmq = (_mqc != NULL) ? "true" : "false";
|
std::string use_redis = (_rc != NULL) ? "true" : "false";
|
||||||
const char *values[10] = {
|
const char *values[10] = {
|
||||||
controllerId,
|
controllerId,
|
||||||
hostname,
|
hostname,
|
||||||
|
@ -599,16 +728,16 @@ void PostgreSQL::heartbeat()
|
||||||
rev.c_str(),
|
rev.c_str(),
|
||||||
build.c_str(),
|
build.c_str(),
|
||||||
host_port.c_str(),
|
host_port.c_str(),
|
||||||
use_rabbitmq.c_str()
|
use_redis.c_str()
|
||||||
};
|
};
|
||||||
|
|
||||||
PGresult *res = PQexecParams(conn,
|
PGresult *res = PQexecParams(conn,
|
||||||
"INSERT INTO ztc_controller (id, cluster_host, last_alive, public_identity, v_major, v_minor, v_rev, v_build, host_port, use_rabbitmq) "
|
"INSERT INTO ztc_controller (id, cluster_host, last_alive, public_identity, v_major, v_minor, v_rev, v_build, host_port, use_redis) "
|
||||||
"VALUES ($1, $2, TO_TIMESTAMP($3::double precision/1000), $4, $5, $6, $7, $8, $9, $10) "
|
"VALUES ($1, $2, TO_TIMESTAMP($3::double precision/1000), $4, $5, $6, $7, $8, $9, $10) "
|
||||||
"ON CONFLICT (id) DO UPDATE SET cluster_host = EXCLUDED.cluster_host, last_alive = EXCLUDED.last_alive, "
|
"ON CONFLICT (id) DO UPDATE SET cluster_host = EXCLUDED.cluster_host, last_alive = EXCLUDED.last_alive, "
|
||||||
"public_identity = EXCLUDED.public_identity, v_major = EXCLUDED.v_major, v_minor = EXCLUDED.v_minor, "
|
"public_identity = EXCLUDED.public_identity, v_major = EXCLUDED.v_major, v_minor = EXCLUDED.v_minor, "
|
||||||
"v_rev = EXCLUDED.v_rev, v_build = EXCLUDED.v_rev, host_port = EXCLUDED.host_port, "
|
"v_rev = EXCLUDED.v_rev, v_build = EXCLUDED.v_rev, host_port = EXCLUDED.host_port, "
|
||||||
"use_rabbitmq = EXCLUDED.use_rabbitmq",
|
"use_redis = EXCLUDED.use_redis",
|
||||||
10, // number of parameters
|
10, // number of parameters
|
||||||
NULL, // oid field. ignore
|
NULL, // oid field. ignore
|
||||||
values, // values for substitution
|
values, // values for substitution
|
||||||
|
@ -621,12 +750,20 @@ void PostgreSQL::heartbeat()
|
||||||
}
|
}
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
}
|
}
|
||||||
|
if (_rc != NULL) {
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->zadd("controllers", controllerId, ts);
|
||||||
|
} else {
|
||||||
|
_redis->zadd("controllers", controllerId, ts);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::this_thread::sleep_for(std::chrono::milliseconds(1000));
|
std::this_thread::sleep_for(std::chrono::milliseconds(1000));
|
||||||
}
|
}
|
||||||
|
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
conn = NULL;
|
conn = NULL;
|
||||||
|
fprintf(stderr, "Exited heartbeat thread\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::membersDbWatcher()
|
void PostgreSQL::membersDbWatcher()
|
||||||
|
@ -638,12 +775,10 @@ void PostgreSQL::membersDbWatcher()
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
initializeMembers(conn);
|
if (_rc) {
|
||||||
|
|
||||||
if (this->_mqc != NULL) {
|
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
conn = NULL;
|
conn = NULL;
|
||||||
_membersWatcher_RabbitMQ();
|
_membersWatcher_Redis();
|
||||||
} else {
|
} else {
|
||||||
_membersWatcher_Postgres(conn);
|
_membersWatcher_Postgres(conn);
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
|
@ -698,41 +833,62 @@ void PostgreSQL::_membersWatcher_Postgres(PGconn *conn) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::_membersWatcher_RabbitMQ() {
|
void PostgreSQL::_membersWatcher_Redis() {
|
||||||
char buf[11] = {0};
|
char buf[11] = {0};
|
||||||
std::string qname = "member_"+ std::string(_myAddress.toString(buf));
|
std::string key = "member-stream:{" + std::string(_myAddress.toString(buf)) + "}";
|
||||||
RabbitMQ rmq(_mqc, qname.c_str());
|
|
||||||
try {
|
|
||||||
rmq.init();
|
|
||||||
} catch (std::runtime_error &e) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());
|
|
||||||
exit(11);
|
|
||||||
}
|
|
||||||
while (_run == 1) {
|
while (_run == 1) {
|
||||||
try {
|
try {
|
||||||
std::string msg = rmq.consume();
|
json tmp;
|
||||||
// fprintf(stderr, "Got Member Update: %s\n", msg.c_str());
|
std::unordered_map<std::string, ItemStream> result;
|
||||||
if (msg.empty()) {
|
if (_rc->clusterMode) {
|
||||||
continue;
|
_cluster->xread(key, "$", std::chrono::seconds(1), 0, std::inserter(result, result.end()));
|
||||||
|
} else {
|
||||||
|
_redis->xread(key, "$", std::chrono::seconds(1), 0, std::inserter(result, result.end()));
|
||||||
}
|
}
|
||||||
json tmp(json::parse(msg));
|
if (!result.empty()) {
|
||||||
json &ov = tmp["old_val"];
|
for (auto element : result) {
|
||||||
json &nv = tmp["new_val"];
|
#ifdef ZT_TRACE
|
||||||
json oldConfig, newConfig;
|
fprintf(stdout, "Received notification from: %s\n", element.first.c_str());
|
||||||
if (ov.is_object()) oldConfig = ov;
|
#endif
|
||||||
if (nv.is_object()) newConfig = nv;
|
for (auto rec : element.second) {
|
||||||
if (oldConfig.is_object() || newConfig.is_object()) {
|
std::string id = rec.first;
|
||||||
_memberChanged(oldConfig,newConfig,(this->_ready>=2));
|
auto attrs = rec.second;
|
||||||
|
#ifdef ZT_TRACE
|
||||||
|
fprintf(stdout, "Record ID: %s\n", id.c_str());
|
||||||
|
fprintf(stdout, "attrs len: %lu\n", attrs.size());
|
||||||
|
#endif
|
||||||
|
for (auto a : attrs) {
|
||||||
|
#ifdef ZT_TRACE
|
||||||
|
fprintf(stdout, "key: %s\nvalue: %s\n", a.first.c_str(), a.second.c_str());
|
||||||
|
#endif
|
||||||
|
try {
|
||||||
|
tmp = json::parse(a.second);
|
||||||
|
json &ov = tmp["old_val"];
|
||||||
|
json &nv = tmp["new_val"];
|
||||||
|
json oldConfig, newConfig;
|
||||||
|
if (ov.is_object()) oldConfig = ov;
|
||||||
|
if (nv.is_object()) newConfig = nv;
|
||||||
|
if (oldConfig.is_object()||newConfig.is_object()) {
|
||||||
|
_memberChanged(oldConfig,newConfig,(this->_ready >= 2));
|
||||||
|
}
|
||||||
|
} catch (...) {
|
||||||
|
fprintf(stderr, "json parse error in networkWatcher_Redis\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->xdel(key, id);
|
||||||
|
} else {
|
||||||
|
_redis->xdel(key, id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} catch (std::runtime_error &e) {
|
} catch (sw::redis::Error &e) {
|
||||||
fprintf(stderr, "RABBITMQ ERROR member change: %s\n", e.what());
|
fprintf(stderr, "Error in Redis members watcher: %s\n", e.what());
|
||||||
break;
|
|
||||||
} catch(std::exception &e ) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR member change: %s\n", e.what());
|
|
||||||
} catch(...) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR member change: unknown error\n");
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
fprintf(stderr, "membersWatcher ended\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::networksDbWatcher()
|
void PostgreSQL::networksDbWatcher()
|
||||||
|
@ -744,12 +900,10 @@ void PostgreSQL::networksDbWatcher()
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
initializeNetworks(conn);
|
if (_rc) {
|
||||||
|
|
||||||
if (this->_mqc != NULL) {
|
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
conn = NULL;
|
conn = NULL;
|
||||||
_networksWatcher_RabbitMQ();
|
_networksWatcher_Redis();
|
||||||
} else {
|
} else {
|
||||||
_networksWatcher_Postgres(conn);
|
_networksWatcher_Postgres(conn);
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
|
@ -802,41 +956,63 @@ void PostgreSQL::_networksWatcher_Postgres(PGconn *conn) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::_networksWatcher_RabbitMQ() {
|
void PostgreSQL::_networksWatcher_Redis() {
|
||||||
char buf[11] = {0};
|
char buf[11] = {0};
|
||||||
std::string qname = "network_"+ std::string(_myAddress.toString(buf));
|
std::string key = "network-stream:{" + std::string(_myAddress.toString(buf)) + "}";
|
||||||
RabbitMQ rmq(_mqc, qname.c_str());
|
|
||||||
try {
|
|
||||||
rmq.init();
|
|
||||||
} catch (std::runtime_error &e) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());
|
|
||||||
exit(11);
|
|
||||||
}
|
|
||||||
while (_run == 1) {
|
while (_run == 1) {
|
||||||
try {
|
try {
|
||||||
std::string msg = rmq.consume();
|
json tmp;
|
||||||
if (msg.empty()) {
|
std::unordered_map<std::string, ItemStream> result;
|
||||||
continue;
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->xread(key, "$", std::chrono::seconds(1), 0, std::inserter(result, result.end()));
|
||||||
|
} else {
|
||||||
|
_redis->xread(key, "$", std::chrono::seconds(1), 0, std::inserter(result, result.end()));
|
||||||
}
|
}
|
||||||
// fprintf(stderr, "Got network update: %s\n", msg.c_str());
|
|
||||||
json tmp(json::parse(msg));
|
if (!result.empty()) {
|
||||||
json &ov = tmp["old_val"];
|
for (auto element : result) {
|
||||||
json &nv = tmp["new_val"];
|
#ifdef ZT_TRACE
|
||||||
json oldConfig, newConfig;
|
fprintf(stdout, "Received notification from: %s\n", element.first.c_str());
|
||||||
if (ov.is_object()) oldConfig = ov;
|
#endif
|
||||||
if (nv.is_object()) newConfig = nv;
|
for (auto rec : element.second) {
|
||||||
if (oldConfig.is_object()||newConfig.is_object()) {
|
std::string id = rec.first;
|
||||||
_networkChanged(oldConfig,newConfig,(this->_ready >= 2));
|
auto attrs = rec.second;
|
||||||
|
#ifdef ZT_TRACE
|
||||||
|
fprintf(stdout, "Record ID: %s\n", id.c_str());
|
||||||
|
fprintf(stdout, "attrs len: %lu\n", attrs.size());
|
||||||
|
#endif
|
||||||
|
for (auto a : attrs) {
|
||||||
|
#ifdef ZT_TRACE
|
||||||
|
fprintf(stdout, "key: %s\nvalue: %s\n", a.first.c_str(), a.second.c_str());
|
||||||
|
#endif
|
||||||
|
try {
|
||||||
|
tmp = json::parse(a.second);
|
||||||
|
json &ov = tmp["old_val"];
|
||||||
|
json &nv = tmp["new_val"];
|
||||||
|
json oldConfig, newConfig;
|
||||||
|
if (ov.is_object()) oldConfig = ov;
|
||||||
|
if (nv.is_object()) newConfig = nv;
|
||||||
|
if (oldConfig.is_object()||newConfig.is_object()) {
|
||||||
|
_networkChanged(oldConfig,newConfig,(this->_ready >= 2));
|
||||||
|
}
|
||||||
|
} catch (...) {
|
||||||
|
fprintf(stderr, "json parse error in networkWatcher_Redis\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->xdel(key, id);
|
||||||
|
} else {
|
||||||
|
_redis->xdel(key, id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} catch (std::runtime_error &e) {
|
} catch (sw::redis::Error &e) {
|
||||||
fprintf(stderr, "RABBITMQ ERROR: %s\n", e.what());
|
fprintf(stderr, "Error in Redis networks watcher: %s\n", e.what());
|
||||||
break;
|
|
||||||
} catch (std::exception &e) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR network watcher: %s\n", e.what());
|
|
||||||
} catch(...) {
|
|
||||||
fprintf(stderr, "RABBITMQ ERROR network watcher: unknown error\n");
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
fprintf(stderr, "networksWatcher ended\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::commitThread()
|
void PostgreSQL::commitThread()
|
||||||
|
@ -872,12 +1048,12 @@ void PostgreSQL::commitThread()
|
||||||
target = (*config)["remoteTraceTarget"];
|
target = (*config)["remoteTraceTarget"];
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string caps = DB::jsonDump((*config)["capabilities"], -1);
|
std::string caps = jsonDump((*config)["capabilities"], -1);
|
||||||
std::string lastAuthTime = std::to_string((long long)(*config)["lastAuthorizedTime"]);
|
std::string lastAuthTime = std::to_string((long long)(*config)["lastAuthorizedTime"]);
|
||||||
std::string lastDeauthTime = std::to_string((long long)(*config)["lastDeauthorizedTime"]);
|
std::string lastDeauthTime = std::to_string((long long)(*config)["lastDeauthorizedTime"]);
|
||||||
std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);
|
std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);
|
||||||
std::string rev = std::to_string((unsigned long long)(*config)["revision"]);
|
std::string rev = std::to_string((unsigned long long)(*config)["revision"]);
|
||||||
std::string tags = DB::jsonDump((*config)["tags"], -1);
|
std::string tags = jsonDump((*config)["tags"], -1);
|
||||||
std::string vmajor = std::to_string((int)(*config)["vMajor"]);
|
std::string vmajor = std::to_string((int)(*config)["vMajor"]);
|
||||||
std::string vminor = std::to_string((int)(*config)["vMinor"]);
|
std::string vminor = std::to_string((int)(*config)["vMinor"]);
|
||||||
std::string vrev = std::to_string((int)(*config)["vRev"]);
|
std::string vrev = std::to_string((int)(*config)["vRev"]);
|
||||||
|
@ -924,7 +1100,7 @@ void PostgreSQL::commitThread()
|
||||||
|
|
||||||
if (PQresultStatus(res) != PGRES_COMMAND_OK) {
|
if (PQresultStatus(res) != PGRES_COMMAND_OK) {
|
||||||
fprintf(stderr, "ERROR: Error updating member: %s\n", PQresultErrorMessage(res));
|
fprintf(stderr, "ERROR: Error updating member: %s\n", PQresultErrorMessage(res));
|
||||||
fprintf(stderr, "%s", DB::jsonDump(*config, 2).c_str());
|
fprintf(stderr, "%s", jsonDump(*config, 2).c_str());
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
delete config;
|
delete config;
|
||||||
config = nullptr;
|
config = nullptr;
|
||||||
|
@ -984,7 +1160,7 @@ void PostgreSQL::commitThread()
|
||||||
};
|
};
|
||||||
|
|
||||||
res = PQexecParams(conn,
|
res = PQexecParams(conn,
|
||||||
"INSERT INTO ztc_member_ip_assignment (member_id, network_id, address) VALUES ($1, $2, $3)",
|
"INSERT INTO ztc_member_ip_assignment (member_id, network_id, address) VALUES ($1, $2, $3) ON CONFLICT (network_id, member_id, address) DO NOTHING",
|
||||||
3,
|
3,
|
||||||
NULL,
|
NULL,
|
||||||
v3,
|
v3,
|
||||||
|
@ -998,6 +1174,7 @@ void PostgreSQL::commitThread()
|
||||||
PQclear(PQexec(conn, "ROLLBACK"));
|
PQclear(PQexec(conn, "ROLLBACK"));
|
||||||
break;;
|
break;;
|
||||||
}
|
}
|
||||||
|
assignments.push_back(addr);
|
||||||
}
|
}
|
||||||
|
|
||||||
res = PQexec(conn, "COMMIT");
|
res = PQexec(conn, "COMMIT");
|
||||||
|
@ -1007,8 +1184,8 @@ void PostgreSQL::commitThread()
|
||||||
|
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
|
|
||||||
const uint64_t nwidInt = DB::jsonIntHex((*config)["nwid"], 0ULL);
|
const uint64_t nwidInt = jsonIntHex((*config)["nwid"], 0ULL);
|
||||||
const uint64_t memberidInt = DB::jsonIntHex((*config)["id"], 0ULL);
|
const uint64_t memberidInt = jsonIntHex((*config)["id"], 0ULL);
|
||||||
if (nwidInt && memberidInt) {
|
if (nwidInt && memberidInt) {
|
||||||
nlohmann::json nwOrig;
|
nlohmann::json nwOrig;
|
||||||
nlohmann::json memOrig;
|
nlohmann::json memOrig;
|
||||||
|
@ -1038,15 +1215,15 @@ void PostgreSQL::commitThread()
|
||||||
if ((*config)["rulesSource"].is_string()) {
|
if ((*config)["rulesSource"].is_string()) {
|
||||||
rulesSource = (*config)["rulesSource"];
|
rulesSource = (*config)["rulesSource"];
|
||||||
}
|
}
|
||||||
std::string caps = DB::jsonDump((*config)["capabilitles"], -1);
|
std::string caps = jsonDump((*config)["capabilitles"], -1);
|
||||||
std::string now = std::to_string(OSUtils::now());
|
std::string now = std::to_string(OSUtils::now());
|
||||||
std::string mtu = std::to_string((int)(*config)["mtu"]);
|
std::string mtu = std::to_string((int)(*config)["mtu"]);
|
||||||
std::string mcastLimit = std::to_string((int)(*config)["multicastLimit"]);
|
std::string mcastLimit = std::to_string((int)(*config)["multicastLimit"]);
|
||||||
std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);
|
std::string rtraceLevel = std::to_string((int)(*config)["remoteTraceLevel"]);
|
||||||
std::string rules = DB::jsonDump((*config)["rules"], -1);
|
std::string rules = jsonDump((*config)["rules"], -1);
|
||||||
std::string tags = DB::jsonDump((*config)["tags"], -1);
|
std::string tags = jsonDump((*config)["tags"], -1);
|
||||||
std::string v4mode = DB::jsonDump((*config)["v4AssignMode"],-1);
|
std::string v4mode = jsonDump((*config)["v4AssignMode"],-1);
|
||||||
std::string v6mode = DB::jsonDump((*config)["v6AssignMode"], -1);
|
std::string v6mode = jsonDump((*config)["v6AssignMode"], -1);
|
||||||
bool enableBroadcast = (*config)["enableBroadcast"];
|
bool enableBroadcast = (*config)["enableBroadcast"];
|
||||||
bool isPrivate = (*config)["private"];
|
bool isPrivate = (*config)["private"];
|
||||||
|
|
||||||
|
@ -1253,7 +1430,7 @@ void PostgreSQL::commitThread()
|
||||||
}
|
}
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
|
|
||||||
const uint64_t nwidInt = DB::jsonIntHex((*config)["nwid"], 0ULL);
|
const uint64_t nwidInt = jsonIntHex((*config)["nwid"], 0ULL);
|
||||||
if (nwidInt) {
|
if (nwidInt) {
|
||||||
nlohmann::json nwOrig;
|
nlohmann::json nwOrig;
|
||||||
nlohmann::json nwNew(*config);
|
nlohmann::json nwNew(*config);
|
||||||
|
@ -1268,6 +1445,20 @@ void PostgreSQL::commitThread()
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());
|
fprintf(stderr, "ERROR: Error updating member: %s\n", e.what());
|
||||||
}
|
}
|
||||||
|
if (_rc != NULL) {
|
||||||
|
try {
|
||||||
|
std::string id = (*config)["id"];
|
||||||
|
std::string controllerId = _myAddressStr.c_str();
|
||||||
|
std::string key = "networks:{" + controllerId + "}";
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->sadd(key, id);
|
||||||
|
} else {
|
||||||
|
_redis->sadd(key, id);
|
||||||
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
fprintf(stderr, "ERROR: Error adding network to Redis: %s\n", e.what());
|
||||||
|
}
|
||||||
|
}
|
||||||
} else if (objtype == "_delete_network") {
|
} else if (objtype == "_delete_network") {
|
||||||
try {
|
try {
|
||||||
std::string networkId = (*config)["nwid"];
|
std::string networkId = (*config)["nwid"];
|
||||||
|
@ -1291,6 +1482,22 @@ void PostgreSQL::commitThread()
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
fprintf(stderr, "ERROR: Error deleting network: %s\n", e.what());
|
fprintf(stderr, "ERROR: Error deleting network: %s\n", e.what());
|
||||||
}
|
}
|
||||||
|
if (_rc != NULL) {
|
||||||
|
try {
|
||||||
|
std::string id = (*config)["id"];
|
||||||
|
std::string controllerId = _myAddressStr.c_str();
|
||||||
|
std::string key = "networks:{" + controllerId + "}";
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->srem(key, id);
|
||||||
|
_cluster->del("network-nodes-online:{"+controllerId+"}:"+id);
|
||||||
|
} else {
|
||||||
|
_redis->srem(key, id);
|
||||||
|
_redis->del("network-nodes-online:{"+controllerId+"}:"+id);
|
||||||
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
fprintf(stderr, "ERROR: Error adding network to Redis: %s\n", e.what());
|
||||||
|
}
|
||||||
|
}
|
||||||
} else if (objtype == "_delete_member") {
|
} else if (objtype == "_delete_member") {
|
||||||
try {
|
try {
|
||||||
std::string memberId = (*config)["id"];
|
std::string memberId = (*config)["id"];
|
||||||
|
@ -1318,6 +1525,23 @@ void PostgreSQL::commitThread()
|
||||||
} catch (std::exception &e) {
|
} catch (std::exception &e) {
|
||||||
fprintf(stderr, "ERROR: Error deleting member: %s\n", e.what());
|
fprintf(stderr, "ERROR: Error deleting member: %s\n", e.what());
|
||||||
}
|
}
|
||||||
|
if (_rc != NULL) {
|
||||||
|
try {
|
||||||
|
std::string memberId = (*config)["id"];
|
||||||
|
std::string networkId = (*config)["nwid"];
|
||||||
|
std::string controllerId = _myAddressStr.c_str();
|
||||||
|
std::string key = "network-nodes-all:{" + controllerId + "}:" + networkId;
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
_cluster->srem(key, memberId);
|
||||||
|
_cluster->del("member:{"+controllerId+"}:"+networkId+":"+memberId);
|
||||||
|
} else {
|
||||||
|
_redis->srem(key, memberId);
|
||||||
|
_redis->del("member:{"+controllerId+"}:"+networkId+":"+memberId);
|
||||||
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
fprintf(stderr, "ERROR: Error deleting member from Redis: %s\n", e.what());
|
||||||
|
}
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
fprintf(stderr, "ERROR: unknown objtype");
|
fprintf(stderr, "ERROR: unknown objtype");
|
||||||
}
|
}
|
||||||
|
@ -1333,9 +1557,21 @@ void PostgreSQL::commitThread()
|
||||||
fprintf(stderr, "ERROR: %s commitThread should still be running! Exiting Controller.\n", _myAddressStr.c_str());
|
fprintf(stderr, "ERROR: %s commitThread should still be running! Exiting Controller.\n", _myAddressStr.c_str());
|
||||||
exit(7);
|
exit(7);
|
||||||
}
|
}
|
||||||
|
fprintf(stderr, "commitThread finished\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
void PostgreSQL::onlineNotificationThread()
|
void PostgreSQL::onlineNotificationThread()
|
||||||
|
{
|
||||||
|
waitForReady();
|
||||||
|
|
||||||
|
if (_rc != NULL) {
|
||||||
|
onlineNotification_Redis();
|
||||||
|
} else {
|
||||||
|
onlineNotification_Postgres();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void PostgreSQL::onlineNotification_Postgres()
|
||||||
{
|
{
|
||||||
PGconn *conn = getPgConn();
|
PGconn *conn = getPgConn();
|
||||||
if (PQstatus(conn) == CONNECTION_BAD) {
|
if (PQstatus(conn) == CONNECTION_BAD) {
|
||||||
|
@ -1345,9 +1581,7 @@ void PostgreSQL::onlineNotificationThread()
|
||||||
}
|
}
|
||||||
_connected = 1;
|
_connected = 1;
|
||||||
|
|
||||||
//int64_t lastUpdatedNetworkStatus = 0;
|
nlohmann::json jtmp1, jtmp2;
|
||||||
std::unordered_map< std::pair<uint64_t,uint64_t>,int64_t,_PairHasher > lastOnlineCumulative;
|
|
||||||
|
|
||||||
while (_run == 1) {
|
while (_run == 1) {
|
||||||
if (PQstatus(conn) != CONNECTION_OK) {
|
if (PQstatus(conn) != CONNECTION_OK) {
|
||||||
fprintf(stderr, "ERROR: Online Notification thread lost connection to Postgres.");
|
fprintf(stderr, "ERROR: Online Notification thread lost connection to Postgres.");
|
||||||
|
@ -1355,9 +1589,6 @@ void PostgreSQL::onlineNotificationThread()
|
||||||
exit(5);
|
exit(5);
|
||||||
}
|
}
|
||||||
|
|
||||||
// map used to send notifications to front end
|
|
||||||
std::unordered_map<std::string, std::vector<std::string>> updateMap;
|
|
||||||
|
|
||||||
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > lastOnline;
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > lastOnline;
|
||||||
{
|
{
|
||||||
std::lock_guard<std::mutex> l(_lastOnline_l);
|
std::lock_guard<std::mutex> l(_lastOnline_l);
|
||||||
|
@ -1378,20 +1609,13 @@ void PostgreSQL::onlineNotificationThread()
|
||||||
OSUtils::ztsnprintf(nwidTmp,sizeof(nwidTmp), "%.16llx", nwid_i);
|
OSUtils::ztsnprintf(nwidTmp,sizeof(nwidTmp), "%.16llx", nwid_i);
|
||||||
OSUtils::ztsnprintf(memTmp,sizeof(memTmp), "%.10llx", i->first.second);
|
OSUtils::ztsnprintf(memTmp,sizeof(memTmp), "%.10llx", i->first.second);
|
||||||
|
|
||||||
auto found = _networks.find(nwid_i);
|
if(!get(nwid_i, jtmp1, i->first.second, jtmp2)) {
|
||||||
if (found == _networks.end()) {
|
continue; // skip non existent networks/members
|
||||||
continue; // skip members trying to join non-existant networks
|
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string networkId(nwidTmp);
|
std::string networkId(nwidTmp);
|
||||||
std::string memberId(memTmp);
|
std::string memberId(memTmp);
|
||||||
|
|
||||||
std::vector<std::string> &members = updateMap[networkId];
|
|
||||||
members.push_back(memberId);
|
|
||||||
|
|
||||||
lastOnlineCumulative[i->first] = i->second.first;
|
|
||||||
|
|
||||||
|
|
||||||
const char *qvals[2] = {
|
const char *qvals[2] = {
|
||||||
networkId.c_str(),
|
networkId.c_str(),
|
||||||
memberId.c_str()
|
memberId.c_str()
|
||||||
|
@ -1451,7 +1675,7 @@ void PostgreSQL::onlineNotificationThread()
|
||||||
PQclear(res);
|
PQclear(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||||
}
|
}
|
||||||
fprintf(stderr, "%s: Fell out of run loop in onlineNotificationThread\n", _myAddressStr.c_str());
|
fprintf(stderr, "%s: Fell out of run loop in onlineNotificationThread\n", _myAddressStr.c_str());
|
||||||
PQfinish(conn);
|
PQfinish(conn);
|
||||||
|
@ -1461,6 +1685,95 @@ void PostgreSQL::onlineNotificationThread()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void PostgreSQL::onlineNotification_Redis()
|
||||||
|
{
|
||||||
|
_connected = 1;
|
||||||
|
|
||||||
|
char buf[11] = {0};
|
||||||
|
std::string controllerId = std::string(_myAddress.toString(buf));
|
||||||
|
|
||||||
|
while (_run == 1) {
|
||||||
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > lastOnline;
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> l(_lastOnline_l);
|
||||||
|
lastOnline.swap(_lastOnline);
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
if (!lastOnline.empty()) {
|
||||||
|
if (_rc->clusterMode) {
|
||||||
|
auto tx = _cluster->transaction(controllerId, true);
|
||||||
|
_doRedisUpdate(tx, controllerId, lastOnline);
|
||||||
|
} else {
|
||||||
|
auto tx = _redis->transaction(true);
|
||||||
|
_doRedisUpdate(tx, controllerId, lastOnline);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (sw::redis::Error &e) {
|
||||||
|
#ifdef ZT_TRACE
|
||||||
|
fprintf(stderr, "Error in online notification thread (redis): %s\n", e.what());
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void PostgreSQL::_doRedisUpdate(sw::redis::Transaction &tx, std::string &controllerId,
|
||||||
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > &lastOnline)
|
||||||
|
|
||||||
|
{
|
||||||
|
nlohmann::json jtmp1, jtmp2;
|
||||||
|
for (auto i=lastOnline.begin(); i != lastOnline.end(); ++i) {
|
||||||
|
uint64_t nwid_i = i->first.first;
|
||||||
|
uint64_t memberid_i = i->first.second;
|
||||||
|
char nwidTmp[64];
|
||||||
|
char memTmp[64];
|
||||||
|
char ipTmp[64];
|
||||||
|
OSUtils::ztsnprintf(nwidTmp,sizeof(nwidTmp), "%.16llx", nwid_i);
|
||||||
|
OSUtils::ztsnprintf(memTmp,sizeof(memTmp), "%.10llx", memberid_i);
|
||||||
|
|
||||||
|
if (!get(nwid_i, jtmp1, memberid_i, jtmp2)){
|
||||||
|
continue; // skip non existent members/networks
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string networkId(nwidTmp);
|
||||||
|
std::string memberId(memTmp);
|
||||||
|
|
||||||
|
int64_t ts = i->second.first;
|
||||||
|
std::string ipAddr = i->second.second.toIpString(ipTmp);
|
||||||
|
std::string timestamp = std::to_string(ts);
|
||||||
|
|
||||||
|
std::unordered_map<std::string, std::string> record = {
|
||||||
|
{"id", memberId},
|
||||||
|
{"address", ipAddr},
|
||||||
|
{"last_updated", std::to_string(ts)}
|
||||||
|
};
|
||||||
|
tx.zadd("nodes-online:{"+controllerId+"}", memberId, ts)
|
||||||
|
.zadd("nodes-online2:{"+controllerId+"}", networkId+"-"+memberId, ts)
|
||||||
|
.zadd("network-nodes-online:{"+controllerId+"}:"+networkId, memberId, ts)
|
||||||
|
.zadd("active-networks:{"+controllerId+"}", networkId, ts)
|
||||||
|
.sadd("network-nodes-all:{"+controllerId+"}:"+networkId, memberId)
|
||||||
|
.hmset("member:{"+controllerId+"}:"+networkId+":"+memberId, record.begin(), record.end());
|
||||||
|
}
|
||||||
|
|
||||||
|
// expire records from all-nodes and network-nodes member list
|
||||||
|
uint64_t expireOld = OSUtils::now() - 300000;
|
||||||
|
|
||||||
|
tx.zremrangebyscore("nodes-online:{"+controllerId+"}", sw::redis::RightBoundedInterval<double>(expireOld, sw::redis::BoundType::LEFT_OPEN));
|
||||||
|
tx.zremrangebyscore("nodes-online2:{"+controllerId+"}", sw::redis::RightBoundedInterval<double>(expireOld, sw::redis::BoundType::LEFT_OPEN));
|
||||||
|
tx.zremrangebyscore("active-networks:{"+controllerId+"}", sw::redis::RightBoundedInterval<double>(expireOld, sw::redis::BoundType::LEFT_OPEN));
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> l(_networks_l);
|
||||||
|
for (const auto &it : _networks) {
|
||||||
|
uint64_t nwid_i = it.first;
|
||||||
|
char nwidTmp[64];
|
||||||
|
OSUtils::ztsnprintf(nwidTmp,sizeof(nwidTmp), "%.16llx", nwid_i);
|
||||||
|
tx.zremrangebyscore("network-nodes-online:{"+controllerId+"}:"+nwidTmp,
|
||||||
|
sw::redis::RightBoundedInterval<double>(expireOld, sw::redis::BoundType::LEFT_OPEN));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tx.exec();
|
||||||
|
}
|
||||||
|
|
||||||
PGconn *PostgreSQL::getPgConn(OverrideMode m)
|
PGconn *PostgreSQL::getPgConn(OverrideMode m)
|
||||||
{
|
{
|
||||||
if (m == ALLOW_PGBOUNCER_OVERRIDE) {
|
if (m == ALLOW_PGBOUNCER_OVERRIDE) {
|
||||||
|
|
|
@ -1,10 +1,10 @@
|
||||||
/*
|
/*
|
||||||
* Copyright (c)2013-2020 ZeroTier, Inc.
|
* Copyright (c)2019 ZeroTier, Inc.
|
||||||
*
|
*
|
||||||
* Use of this software is governed by the Business Source License included
|
* Use of this software is governed by the Business Source License included
|
||||||
* in the LICENSE.TXT file in the project's root directory.
|
* in the LICENSE.TXT file in the project's root directory.
|
||||||
*
|
*
|
||||||
* Change Date: 2024-01-01
|
* Change Date: 2023-01-01
|
||||||
*
|
*
|
||||||
* On the date above, in accordance with the Business Source License, use
|
* On the date above, in accordance with the Business Source License, use
|
||||||
* of this software will be governed by version 2.0 of the Apache License.
|
* of this software will be governed by version 2.0 of the Apache License.
|
||||||
|
@ -20,13 +20,16 @@
|
||||||
|
|
||||||
#define ZT_CENTRAL_CONTROLLER_COMMIT_THREADS 4
|
#define ZT_CENTRAL_CONTROLLER_COMMIT_THREADS 4
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
#include <redis++/redis++.h>
|
||||||
|
|
||||||
extern "C" {
|
extern "C" {
|
||||||
typedef struct pg_conn PGconn;
|
typedef struct pg_conn PGconn;
|
||||||
}
|
}
|
||||||
|
|
||||||
namespace ZeroTier {
|
namespace ZeroTier {
|
||||||
|
|
||||||
struct MQConfig;
|
struct RedisConfig;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* A controller database driver that talks to PostgreSQL
|
* A controller database driver that talks to PostgreSQL
|
||||||
|
@ -37,7 +40,7 @@ struct MQConfig;
|
||||||
class PostgreSQL : public DB
|
class PostgreSQL : public DB
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
PostgreSQL(const Identity &myId, const char *path, int listenPort, MQConfig *mqc = NULL);
|
PostgreSQL(const Identity &myId, const char *path, int listenPort, RedisConfig *rc);
|
||||||
virtual ~PostgreSQL();
|
virtual ~PostgreSQL();
|
||||||
|
|
||||||
virtual bool waitForReady();
|
virtual bool waitForReady();
|
||||||
|
@ -59,13 +62,18 @@ private:
|
||||||
void heartbeat();
|
void heartbeat();
|
||||||
void membersDbWatcher();
|
void membersDbWatcher();
|
||||||
void _membersWatcher_Postgres(PGconn *conn);
|
void _membersWatcher_Postgres(PGconn *conn);
|
||||||
void _membersWatcher_RabbitMQ();
|
|
||||||
void networksDbWatcher();
|
void networksDbWatcher();
|
||||||
void _networksWatcher_Postgres(PGconn *conn);
|
void _networksWatcher_Postgres(PGconn *conn);
|
||||||
void _networksWatcher_RabbitMQ();
|
|
||||||
|
void _membersWatcher_Redis();
|
||||||
|
void _networksWatcher_Redis();
|
||||||
|
|
||||||
void commitThread();
|
void commitThread();
|
||||||
void onlineNotificationThread();
|
void onlineNotificationThread();
|
||||||
|
void onlineNotification_Postgres();
|
||||||
|
void onlineNotification_Redis();
|
||||||
|
void _doRedisUpdate(sw::redis::Transaction &tx, std::string &controllerId,
|
||||||
|
std::unordered_map< std::pair<uint64_t,uint64_t>,std::pair<int64_t,InetAddress>,_PairHasher > &lastOnline);
|
||||||
|
|
||||||
enum OverrideMode {
|
enum OverrideMode {
|
||||||
ALLOW_PGBOUNCER_OVERRIDE = 0,
|
ALLOW_PGBOUNCER_OVERRIDE = 0,
|
||||||
|
@ -96,7 +104,9 @@ private:
|
||||||
|
|
||||||
int _listenPort;
|
int _listenPort;
|
||||||
|
|
||||||
MQConfig *_mqc;
|
RedisConfig *_rc;
|
||||||
|
std::shared_ptr<sw::redis::Redis> _redis;
|
||||||
|
std::shared_ptr<sw::redis::RedisCluster> _cluster;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace ZeroTier
|
} // namespace ZeroTier
|
||||||
|
|
|
@ -1,120 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c)2013-2020 ZeroTier, Inc.
|
|
||||||
*
|
|
||||||
* Use of this software is governed by the Business Source License included
|
|
||||||
* in the LICENSE.TXT file in the project's root directory.
|
|
||||||
*
|
|
||||||
* Change Date: 2024-01-01
|
|
||||||
*
|
|
||||||
* On the date above, in accordance with the Business Source License, use
|
|
||||||
* of this software will be governed by version 2.0 of the Apache License.
|
|
||||||
*/
|
|
||||||
/****/
|
|
||||||
|
|
||||||
#include "RabbitMQ.hpp"
|
|
||||||
|
|
||||||
#ifdef ZT_CONTROLLER_USE_LIBPQ
|
|
||||||
|
|
||||||
#include <amqp.h>
|
|
||||||
#include <amqp_tcp_socket.h>
|
|
||||||
#include <stdexcept>
|
|
||||||
#include <cstring>
|
|
||||||
|
|
||||||
namespace ZeroTier
|
|
||||||
{
|
|
||||||
|
|
||||||
RabbitMQ::RabbitMQ(MQConfig *cfg, const char *queueName)
|
|
||||||
: _mqc(cfg)
|
|
||||||
, _qName(queueName)
|
|
||||||
, _socket(NULL)
|
|
||||||
, _status(0)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
RabbitMQ::~RabbitMQ()
|
|
||||||
{
|
|
||||||
amqp_channel_close(_conn, _channel, AMQP_REPLY_SUCCESS);
|
|
||||||
amqp_connection_close(_conn, AMQP_REPLY_SUCCESS);
|
|
||||||
amqp_destroy_connection(_conn);
|
|
||||||
}
|
|
||||||
|
|
||||||
void RabbitMQ::init()
|
|
||||||
{
|
|
||||||
struct timeval tval;
|
|
||||||
memset(&tval, 0, sizeof(struct timeval));
|
|
||||||
tval.tv_sec = 5;
|
|
||||||
|
|
||||||
fprintf(stderr, "Initializing RabbitMQ %s\n", _qName);
|
|
||||||
_conn = amqp_new_connection();
|
|
||||||
_socket = amqp_tcp_socket_new(_conn);
|
|
||||||
if (!_socket) {
|
|
||||||
throw std::runtime_error("Can't create socket for RabbitMQ");
|
|
||||||
}
|
|
||||||
|
|
||||||
_status = amqp_socket_open_noblock(_socket, _mqc->host.c_str(), _mqc->port, &tval);
|
|
||||||
if (_status) {
|
|
||||||
throw std::runtime_error("Can't connect to RabbitMQ");
|
|
||||||
}
|
|
||||||
|
|
||||||
amqp_rpc_reply_t r = amqp_login(_conn, "/", 0, 131072, 0, AMQP_SASL_METHOD_PLAIN,
|
|
||||||
_mqc->username.c_str(), _mqc->password.c_str());
|
|
||||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) {
|
|
||||||
throw std::runtime_error("RabbitMQ Login Error");
|
|
||||||
}
|
|
||||||
|
|
||||||
static int chan = 0;
|
|
||||||
{
|
|
||||||
Mutex::Lock l(_chan_m);
|
|
||||||
_channel = ++chan;
|
|
||||||
}
|
|
||||||
amqp_channel_open(_conn, _channel);
|
|
||||||
r = amqp_get_rpc_reply(_conn);
|
|
||||||
if(r.reply_type != AMQP_RESPONSE_NORMAL) {
|
|
||||||
throw std::runtime_error("Error opening communication channel");
|
|
||||||
}
|
|
||||||
|
|
||||||
_q = amqp_queue_declare(_conn, _channel, amqp_cstring_bytes(_qName), 0, 0, 0, 0, amqp_empty_table);
|
|
||||||
r = amqp_get_rpc_reply(_conn);
|
|
||||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) {
|
|
||||||
throw std::runtime_error("Error declaring queue " + std::string(_qName));
|
|
||||||
}
|
|
||||||
|
|
||||||
amqp_basic_consume(_conn, _channel, amqp_cstring_bytes(_qName), amqp_empty_bytes, 0, 1, 0, amqp_empty_table);
|
|
||||||
r = amqp_get_rpc_reply(_conn);
|
|
||||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) {
|
|
||||||
throw std::runtime_error("Error consuming queue " + std::string(_qName));
|
|
||||||
}
|
|
||||||
fprintf(stderr, "RabbitMQ Init OK %s\n", _qName);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string RabbitMQ::consume()
|
|
||||||
{
|
|
||||||
amqp_rpc_reply_t res;
|
|
||||||
amqp_envelope_t envelope;
|
|
||||||
amqp_maybe_release_buffers(_conn);
|
|
||||||
|
|
||||||
struct timeval timeout;
|
|
||||||
timeout.tv_sec = 1;
|
|
||||||
timeout.tv_usec = 0;
|
|
||||||
|
|
||||||
res = amqp_consume_message(_conn, &envelope, &timeout, 0);
|
|
||||||
if (res.reply_type != AMQP_RESPONSE_NORMAL) {
|
|
||||||
if (res.reply_type == AMQP_RESPONSE_LIBRARY_EXCEPTION && res.library_error == AMQP_STATUS_TIMEOUT) {
|
|
||||||
// timeout waiting for message. Return empty string
|
|
||||||
return "";
|
|
||||||
} else {
|
|
||||||
throw std::runtime_error("Error getting message");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string msg(
|
|
||||||
(const char*)envelope.message.body.bytes,
|
|
||||||
envelope.message.body.len
|
|
||||||
);
|
|
||||||
amqp_destroy_envelope(&envelope);
|
|
||||||
return msg;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // ZT_CONTROLLER_USE_LIBPQ
|
|
|
@ -1,69 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c)2013-2020 ZeroTier, Inc.
|
|
||||||
*
|
|
||||||
* Use of this software is governed by the Business Source License included
|
|
||||||
* in the LICENSE.TXT file in the project's root directory.
|
|
||||||
*
|
|
||||||
* Change Date: 2024-01-01
|
|
||||||
*
|
|
||||||
* On the date above, in accordance with the Business Source License, use
|
|
||||||
* of this software will be governed by version 2.0 of the Apache License.
|
|
||||||
*/
|
|
||||||
/****/
|
|
||||||
|
|
||||||
#ifndef ZT_CONTROLLER_RABBITMQ_HPP
|
|
||||||
#define ZT_CONTROLLER_RABBITMQ_HPP
|
|
||||||
|
|
||||||
#include "DB.hpp"
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
namespace ZeroTier
|
|
||||||
{
|
|
||||||
struct MQConfig {
|
|
||||||
std::string host;
|
|
||||||
int port;
|
|
||||||
std::string username;
|
|
||||||
std::string password;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef ZT_CONTROLLER_USE_LIBPQ
|
|
||||||
|
|
||||||
#include "../core/Mutex.hpp"
|
|
||||||
|
|
||||||
#include <amqp.h>
|
|
||||||
#include <amqp_tcp_socket.h>
|
|
||||||
|
|
||||||
|
|
||||||
namespace ZeroTier
|
|
||||||
{
|
|
||||||
|
|
||||||
class RabbitMQ {
|
|
||||||
public:
|
|
||||||
RabbitMQ(MQConfig *cfg, const char *queueName);
|
|
||||||
~RabbitMQ();
|
|
||||||
|
|
||||||
void init();
|
|
||||||
|
|
||||||
std::string consume();
|
|
||||||
|
|
||||||
private:
|
|
||||||
MQConfig *_mqc;
|
|
||||||
const char *_qName;
|
|
||||||
|
|
||||||
amqp_socket_t *_socket;
|
|
||||||
amqp_connection_state_t _conn;
|
|
||||||
amqp_queue_declare_ok_t *_q;
|
|
||||||
int _status;
|
|
||||||
|
|
||||||
int _channel;
|
|
||||||
|
|
||||||
Mutex _chan_m;
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // ZT_CONTROLLER_USE_LIBPQ
|
|
||||||
|
|
||||||
#endif // ZT_CONTROLLER_RABBITMQ_HPP
|
|
||||||
|
|
15
controller/Redis.hpp
Normal file
15
controller/Redis.hpp
Normal file
|
@ -0,0 +1,15 @@
|
||||||
|
#ifndef ZT_CONTROLLER_REDIS_HPP
|
||||||
|
#define ZT_CONTROLLER_REDIS_HPP
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
namespace ZeroTier {
|
||||||
|
struct RedisConfig {
|
||||||
|
std::string hostname;
|
||||||
|
int port;
|
||||||
|
std::string password;
|
||||||
|
bool clusterMode;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
|
@ -1,7 +0,0 @@
|
||||||
/hiredis-test
|
|
||||||
/examples/hiredis-example*
|
|
||||||
/*.o
|
|
||||||
/*.so
|
|
||||||
/*.dylib
|
|
||||||
/*.a
|
|
||||||
/*.pc
|
|
45
controller/thirdparty/hiredis-0.14.1/.travis.yml
vendored
45
controller/thirdparty/hiredis-0.14.1/.travis.yml
vendored
|
@ -1,45 +0,0 @@
|
||||||
language: c
|
|
||||||
sudo: false
|
|
||||||
compiler:
|
|
||||||
- gcc
|
|
||||||
- clang
|
|
||||||
|
|
||||||
os:
|
|
||||||
- linux
|
|
||||||
- osx
|
|
||||||
|
|
||||||
branches:
|
|
||||||
only:
|
|
||||||
- staging
|
|
||||||
- trying
|
|
||||||
- master
|
|
||||||
|
|
||||||
before_script:
|
|
||||||
- if [ "$TRAVIS_OS_NAME" == "osx" ] ; then brew update; brew install redis; fi
|
|
||||||
|
|
||||||
addons:
|
|
||||||
apt:
|
|
||||||
packages:
|
|
||||||
- libc6-dbg
|
|
||||||
- libc6-dev
|
|
||||||
- libc6:i386
|
|
||||||
- libc6-dev-i386
|
|
||||||
- libc6-dbg:i386
|
|
||||||
- gcc-multilib
|
|
||||||
- valgrind
|
|
||||||
|
|
||||||
env:
|
|
||||||
- CFLAGS="-Werror"
|
|
||||||
- PRE="valgrind --track-origins=yes --leak-check=full"
|
|
||||||
- TARGET="32bit" TARGET_VARS="32bit-vars" CFLAGS="-Werror"
|
|
||||||
- TARGET="32bit" TARGET_VARS="32bit-vars" PRE="valgrind --track-origins=yes --leak-check=full"
|
|
||||||
|
|
||||||
matrix:
|
|
||||||
exclude:
|
|
||||||
- os: osx
|
|
||||||
env: PRE="valgrind --track-origins=yes --leak-check=full"
|
|
||||||
|
|
||||||
- os: osx
|
|
||||||
env: TARGET="32bit" TARGET_VARS="32bit-vars" PRE="valgrind --track-origins=yes --leak-check=full"
|
|
||||||
|
|
||||||
script: make $TARGET CFLAGS="$CFLAGS" && make check PRE="$PRE" && make $TARGET_VARS hiredis-example
|
|
190
controller/thirdparty/hiredis-0.14.1/CHANGELOG.md
vendored
190
controller/thirdparty/hiredis-0.14.1/CHANGELOG.md
vendored
|
@ -1,190 +0,0 @@
|
||||||
**NOTE: BREAKING CHANGES upgrading from 0.13.x to 0.14.x **:
|
|
||||||
|
|
||||||
* Bulk and multi-bulk lengths less than -1 or greater than `LLONG_MAX` are now
|
|
||||||
protocol errors. This is consistent with the RESP specification. On 32-bit
|
|
||||||
platforms, the upper bound is lowered to `SIZE_MAX`.
|
|
||||||
|
|
||||||
* Change `redisReply.len` to `size_t`, as it denotes the the size of a string
|
|
||||||
|
|
||||||
User code should compare this to `size_t` values as well. If it was used to
|
|
||||||
compare to other values, casting might be necessary or can be removed, if
|
|
||||||
casting was applied before.
|
|
||||||
|
|
||||||
### 0.14.1 (2020-03-13)
|
|
||||||
|
|
||||||
* Adds safe allocation wrappers (CVE-2020-7105, #747, #752) (Michael Grunder)
|
|
||||||
|
|
||||||
### 0.14.0 (2018-09-25)
|
|
||||||
|
|
||||||
* Make string2ll static to fix conflict with Redis (Tom Lee [c3188b])
|
|
||||||
* Use -dynamiclib instead of -shared for OSX (Ryan Schmidt [a65537])
|
|
||||||
* Use string2ll from Redis w/added tests (Michael Grunder [7bef04, 60f622])
|
|
||||||
* Makefile - OSX compilation fixes (Ryan Schmidt [881fcb, 0e9af8])
|
|
||||||
* Remove redundant NULL checks (Justin Brewer [54acc8, 58e6b8])
|
|
||||||
* Fix bulk and multi-bulk length truncation (Justin Brewer [109197])
|
|
||||||
* Fix SIGSEGV in OpenBSD by checking for NULL before calling freeaddrinfo (Justin Brewer [546d94])
|
|
||||||
* Several POSIX compatibility fixes (Justin Brewer [bbeab8, 49bbaa, d1c1b6])
|
|
||||||
* Makefile - Compatibility fixes (Dimitri Vorobiev [3238cf, 12a9d1])
|
|
||||||
* Makefile - Fix make install on FreeBSD (Zach Shipko [a2ef2b])
|
|
||||||
* Makefile - don't assume $(INSTALL) is cp (Igor Gnatenko [725a96])
|
|
||||||
* Separate side-effect causing function from assert and small cleanup (amallia [b46413, 3c3234])
|
|
||||||
* Don't send negative values to `__redisAsyncCommand` (Frederik Deweerdt [706129])
|
|
||||||
* Fix leak if setsockopt fails (Frederik Deweerdt [e21c9c])
|
|
||||||
* Fix libevent leak (zfz [515228])
|
|
||||||
* Clean up GCC warning (Ichito Nagata [2ec774])
|
|
||||||
* Keep track of errno in `__redisSetErrorFromErrno()` as snprintf may use it (Jin Qing [25cd88])
|
|
||||||
* Solaris compilation fix (Donald Whyte [41b07d])
|
|
||||||
* Reorder linker arguments when building examples (Tustfarm-heart [06eedd])
|
|
||||||
* Keep track of subscriptions in case of rapid subscribe/unsubscribe (Hyungjin Kim [073dc8, be76c5, d46999])
|
|
||||||
* libuv use after free fix (Paul Scott [cbb956])
|
|
||||||
* Properly close socket fd on reconnect attempt (WSL [64d1ec])
|
|
||||||
* Skip valgrind in OSX tests (Jan-Erik Rediger [9deb78])
|
|
||||||
* Various updates for Travis testing OSX (Ted Nyman [fa3774, 16a459, bc0ea5])
|
|
||||||
* Update libevent (Chris Xin [386802])
|
|
||||||
* Change sds.h for building in C++ projects (Ali Volkan ATLI [f5b32e])
|
|
||||||
* Use proper format specifier in redisFormatSdsCommandArgv (Paulino Huerta, Jan-Erik Rediger [360a06, 8655a6])
|
|
||||||
* Better handling of NULL reply in example code (Jan-Erik Rediger [1b8ed3])
|
|
||||||
* Prevent overflow when formatting an error (Jan-Erik Rediger [0335cb])
|
|
||||||
* Compatibility fix for strerror_r (Tom Lee [bb1747])
|
|
||||||
* Properly detect integer parse/overflow errors (Justin Brewer [93421f])
|
|
||||||
* Adds CI for Windows and cygwin fixes (owent, [6c53d6, 6c3e40])
|
|
||||||
* Catch a buffer overflow when formatting the error message
|
|
||||||
* Import latest upstream sds. This breaks applications that are linked against the old hiredis v0.13
|
|
||||||
* Fix warnings, when compiled with -Wshadow
|
|
||||||
* Make hiredis compile in Cygwin on Windows, now CI-tested
|
|
||||||
|
|
||||||
**BREAKING CHANGES**:
|
|
||||||
|
|
||||||
* Remove backwards compatibility macro's
|
|
||||||
|
|
||||||
This removes the following old function aliases, use the new name now:
|
|
||||||
|
|
||||||
| Old | New |
|
|
||||||
| --------------------------- | ---------------------- |
|
|
||||||
| redisReplyReaderCreate | redisReaderCreate |
|
|
||||||
| redisReplyReaderCreate | redisReaderCreate |
|
|
||||||
| redisReplyReaderFree | redisReaderFree |
|
|
||||||
| redisReplyReaderFeed | redisReaderFeed |
|
|
||||||
| redisReplyReaderGetReply | redisReaderGetReply |
|
|
||||||
| redisReplyReaderSetPrivdata | redisReaderSetPrivdata |
|
|
||||||
| redisReplyReaderGetObject | redisReaderGetObject |
|
|
||||||
| redisReplyReaderGetError | redisReaderGetError |
|
|
||||||
|
|
||||||
* The `DEBUG` variable in the Makefile was renamed to `DEBUG_FLAGS`
|
|
||||||
|
|
||||||
Previously it broke some builds for people that had `DEBUG` set to some arbitrary value,
|
|
||||||
due to debugging other software.
|
|
||||||
By renaming we avoid unintentional name clashes.
|
|
||||||
|
|
||||||
Simply rename `DEBUG` to `DEBUG_FLAGS` in your environment to make it working again.
|
|
||||||
|
|
||||||
### 0.13.3 (2015-09-16)
|
|
||||||
|
|
||||||
* Revert "Clear `REDIS_CONNECTED` flag when connection is closed".
|
|
||||||
* Make tests pass on FreeBSD (Thanks, Giacomo Olgeni)
|
|
||||||
|
|
||||||
|
|
||||||
If the `REDIS_CONNECTED` flag is cleared,
|
|
||||||
the async onDisconnect callback function will never be called.
|
|
||||||
This causes problems as the disconnect is never reported back to the user.
|
|
||||||
|
|
||||||
### 0.13.2 (2015-08-25)
|
|
||||||
|
|
||||||
* Prevent crash on pending replies in async code (Thanks, @switch-st)
|
|
||||||
* Clear `REDIS_CONNECTED` flag when connection is closed (Thanks, Jerry Jacobs)
|
|
||||||
* Add MacOS X addapter (Thanks, @dizzus)
|
|
||||||
* Add Qt adapter (Thanks, Pietro Cerutti)
|
|
||||||
* Add Ivykis adapter (Thanks, Gergely Nagy)
|
|
||||||
|
|
||||||
All adapters are provided as is and are only tested where possible.
|
|
||||||
|
|
||||||
### 0.13.1 (2015-05-03)
|
|
||||||
|
|
||||||
This is a bug fix release.
|
|
||||||
The new `reconnect` method introduced new struct members, which clashed with pre-defined names in pre-C99 code.
|
|
||||||
Another commit forced C99 compilation just to make it work, but of course this is not desirable for outside projects.
|
|
||||||
Other non-C99 code can now use hiredis as usual again.
|
|
||||||
Sorry for the inconvenience.
|
|
||||||
|
|
||||||
* Fix memory leak in async reply handling (Salvatore Sanfilippo)
|
|
||||||
* Rename struct member to avoid name clash with pre-c99 code (Alex Balashov, ncopa)
|
|
||||||
|
|
||||||
### 0.13.0 (2015-04-16)
|
|
||||||
|
|
||||||
This release adds a minimal Windows compatibility layer.
|
|
||||||
The parser, standalone since v0.12.0, can now be compiled on Windows
|
|
||||||
(and thus used in other client libraries as well)
|
|
||||||
|
|
||||||
* Windows compatibility layer for parser code (tzickel)
|
|
||||||
* Properly escape data printed to PKGCONF file (Dan Skorupski)
|
|
||||||
* Fix tests when assert() undefined (Keith Bennett, Matt Stancliff)
|
|
||||||
* Implement a reconnect method for the client context, this changes the structure of `redisContext` (Aaron Bedra)
|
|
||||||
|
|
||||||
### 0.12.1 (2015-01-26)
|
|
||||||
|
|
||||||
* Fix `make install`: DESTDIR support, install all required files, install PKGCONF in proper location
|
|
||||||
* Fix `make test` as 32 bit build on 64 bit platform
|
|
||||||
|
|
||||||
### 0.12.0 (2015-01-22)
|
|
||||||
|
|
||||||
* Add optional KeepAlive support
|
|
||||||
|
|
||||||
* Try again on EINTR errors
|
|
||||||
|
|
||||||
* Add libuv adapter
|
|
||||||
|
|
||||||
* Add IPv6 support
|
|
||||||
|
|
||||||
* Remove possibility of multiple close on same fd
|
|
||||||
|
|
||||||
* Add ability to bind source address on connect
|
|
||||||
|
|
||||||
* Add redisConnectFd() and redisFreeKeepFd()
|
|
||||||
|
|
||||||
* Fix getaddrinfo() memory leak
|
|
||||||
|
|
||||||
* Free string if it is unused (fixes memory leak)
|
|
||||||
|
|
||||||
* Improve redisAppendCommandArgv performance 2.5x
|
|
||||||
|
|
||||||
* Add support for SO_REUSEADDR
|
|
||||||
|
|
||||||
* Fix redisvFormatCommand format parsing
|
|
||||||
|
|
||||||
* Add GLib 2.0 adapter
|
|
||||||
|
|
||||||
* Refactor reading code into read.c
|
|
||||||
|
|
||||||
* Fix errno error buffers to not clobber errors
|
|
||||||
|
|
||||||
* Generate pkgconf during build
|
|
||||||
|
|
||||||
* Silence _BSD_SOURCE warnings
|
|
||||||
|
|
||||||
* Improve digit counting for multibulk creation
|
|
||||||
|
|
||||||
|
|
||||||
### 0.11.0
|
|
||||||
|
|
||||||
* Increase the maximum multi-bulk reply depth to 7.
|
|
||||||
|
|
||||||
* Increase the read buffer size from 2k to 16k.
|
|
||||||
|
|
||||||
* Use poll(2) instead of select(2) to support large fds (>= 1024).
|
|
||||||
|
|
||||||
### 0.10.1
|
|
||||||
|
|
||||||
* Makefile overhaul. Important to check out if you override one or more
|
|
||||||
variables using environment variables or via arguments to the "make" tool.
|
|
||||||
|
|
||||||
* Issue #45: Fix potential memory leak for a multi bulk reply with 0 elements
|
|
||||||
being created by the default reply object functions.
|
|
||||||
|
|
||||||
* Issue #43: Don't crash in an asynchronous context when Redis returns an error
|
|
||||||
reply after the connection has been made (this happens when the maximum
|
|
||||||
number of connections is reached).
|
|
||||||
|
|
||||||
### 0.10.0
|
|
||||||
|
|
||||||
* See commit log.
|
|
||||||
|
|
29
controller/thirdparty/hiredis-0.14.1/COPYING
vendored
29
controller/thirdparty/hiredis-0.14.1/COPYING
vendored
|
@ -1,29 +0,0 @@
|
||||||
Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
|
|
||||||
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.
|
|
214
controller/thirdparty/hiredis-0.14.1/Makefile
vendored
214
controller/thirdparty/hiredis-0.14.1/Makefile
vendored
|
@ -1,214 +0,0 @@
|
||||||
# Hiredis Makefile
|
|
||||||
# Copyright (C) 2010-2011 Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
# Copyright (C) 2010-2011 Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
# This file is released under the BSD license, see the COPYING file
|
|
||||||
|
|
||||||
OBJ=net.o hiredis.o sds.o async.o read.o alloc.o
|
|
||||||
EXAMPLES=hiredis-example hiredis-example-libevent hiredis-example-libev hiredis-example-glib
|
|
||||||
TESTS=hiredis-test
|
|
||||||
LIBNAME=libhiredis
|
|
||||||
PKGCONFNAME=hiredis.pc
|
|
||||||
|
|
||||||
HIREDIS_MAJOR=$(shell grep HIREDIS_MAJOR hiredis.h | awk '{print $$3}')
|
|
||||||
HIREDIS_MINOR=$(shell grep HIREDIS_MINOR hiredis.h | awk '{print $$3}')
|
|
||||||
HIREDIS_PATCH=$(shell grep HIREDIS_PATCH hiredis.h | awk '{print $$3}')
|
|
||||||
HIREDIS_SONAME=$(shell grep HIREDIS_SONAME hiredis.h | awk '{print $$3}')
|
|
||||||
|
|
||||||
# Installation related variables and target
|
|
||||||
PREFIX?=/usr/local
|
|
||||||
INCLUDE_PATH?=include/hiredis
|
|
||||||
LIBRARY_PATH?=lib
|
|
||||||
PKGCONF_PATH?=pkgconfig
|
|
||||||
INSTALL_INCLUDE_PATH= $(DESTDIR)$(PREFIX)/$(INCLUDE_PATH)
|
|
||||||
INSTALL_LIBRARY_PATH= $(DESTDIR)$(PREFIX)/$(LIBRARY_PATH)
|
|
||||||
INSTALL_PKGCONF_PATH= $(INSTALL_LIBRARY_PATH)/$(PKGCONF_PATH)
|
|
||||||
|
|
||||||
# redis-server configuration used for testing
|
|
||||||
REDIS_PORT=56379
|
|
||||||
REDIS_SERVER=redis-server
|
|
||||||
define REDIS_TEST_CONFIG
|
|
||||||
daemonize yes
|
|
||||||
pidfile /tmp/hiredis-test-redis.pid
|
|
||||||
port $(REDIS_PORT)
|
|
||||||
bind 127.0.0.1
|
|
||||||
unixsocket /tmp/hiredis-test-redis.sock
|
|
||||||
endef
|
|
||||||
export REDIS_TEST_CONFIG
|
|
||||||
|
|
||||||
# Fallback to gcc when $CC is not in $PATH.
|
|
||||||
CC:=$(shell sh -c 'type $${CC%% *} >/dev/null 2>/dev/null && echo $(CC) || echo gcc')
|
|
||||||
CXX:=$(shell sh -c 'type $${CXX%% *} >/dev/null 2>/dev/null && echo $(CXX) || echo g++')
|
|
||||||
OPTIMIZATION?=-O3
|
|
||||||
WARNINGS=-Wall -W -Wstrict-prototypes -Wwrite-strings
|
|
||||||
DEBUG_FLAGS?= -g -ggdb
|
|
||||||
REAL_CFLAGS=$(OPTIMIZATION) -fPIC $(CFLAGS) $(WARNINGS) $(DEBUG_FLAGS)
|
|
||||||
REAL_LDFLAGS=$(LDFLAGS)
|
|
||||||
|
|
||||||
DYLIBSUFFIX=so
|
|
||||||
STLIBSUFFIX=a
|
|
||||||
DYLIB_MINOR_NAME=$(LIBNAME).$(DYLIBSUFFIX).$(HIREDIS_SONAME)
|
|
||||||
DYLIB_MAJOR_NAME=$(LIBNAME).$(DYLIBSUFFIX).$(HIREDIS_MAJOR)
|
|
||||||
DYLIBNAME=$(LIBNAME).$(DYLIBSUFFIX)
|
|
||||||
DYLIB_MAKE_CMD=$(CC) -shared -Wl,-soname,$(DYLIB_MINOR_NAME) -o $(DYLIBNAME) $(LDFLAGS)
|
|
||||||
STLIBNAME=$(LIBNAME).$(STLIBSUFFIX)
|
|
||||||
STLIB_MAKE_CMD=ar rcs $(STLIBNAME)
|
|
||||||
|
|
||||||
# Platform-specific overrides
|
|
||||||
uname_S := $(shell sh -c 'uname -s 2>/dev/null || echo not')
|
|
||||||
ifeq ($(uname_S),SunOS)
|
|
||||||
REAL_LDFLAGS+= -ldl -lnsl -lsocket
|
|
||||||
DYLIB_MAKE_CMD=$(CC) -G -o $(DYLIBNAME) -h $(DYLIB_MINOR_NAME) $(LDFLAGS)
|
|
||||||
endif
|
|
||||||
ifeq ($(uname_S),Darwin)
|
|
||||||
DYLIBSUFFIX=dylib
|
|
||||||
DYLIB_MINOR_NAME=$(LIBNAME).$(HIREDIS_SONAME).$(DYLIBSUFFIX)
|
|
||||||
DYLIB_MAKE_CMD=$(CC) -dynamiclib -Wl,-install_name,$(PREFIX)/$(LIBRARY_PATH)/$(DYLIB_MINOR_NAME) -o $(DYLIBNAME) $(LDFLAGS)
|
|
||||||
endif
|
|
||||||
|
|
||||||
all: $(DYLIBNAME) $(STLIBNAME) hiredis-test $(PKGCONFNAME)
|
|
||||||
|
|
||||||
# Deps (use make dep to generate this)
|
|
||||||
alloc.o: alloc.c fmacros.h alloc.h
|
|
||||||
async.o: async.c fmacros.h alloc.h async.h hiredis.h read.h sds.h net.h dict.c dict.h
|
|
||||||
dict.o: dict.c fmacros.h alloc.h dict.h
|
|
||||||
hiredis.o: hiredis.c fmacros.h hiredis.h read.h sds.h alloc.h net.h
|
|
||||||
net.o: net.c fmacros.h net.h hiredis.h read.h sds.h alloc.h
|
|
||||||
read.o: read.c fmacros.h read.h sds.h
|
|
||||||
sds.o: sds.c sds.h sdsalloc.h
|
|
||||||
test.o: test.c fmacros.h hiredis.h read.h sds.h alloc.h net.h
|
|
||||||
|
|
||||||
$(DYLIBNAME): $(OBJ)
|
|
||||||
$(DYLIB_MAKE_CMD) $(OBJ)
|
|
||||||
|
|
||||||
$(STLIBNAME): $(OBJ)
|
|
||||||
$(STLIB_MAKE_CMD) $(OBJ)
|
|
||||||
|
|
||||||
dynamic: $(DYLIBNAME)
|
|
||||||
static: $(STLIBNAME)
|
|
||||||
|
|
||||||
# Binaries:
|
|
||||||
hiredis-example-libevent: examples/example-libevent.c adapters/libevent.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< -levent $(STLIBNAME)
|
|
||||||
|
|
||||||
hiredis-example-libev: examples/example-libev.c adapters/libev.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< -lev $(STLIBNAME)
|
|
||||||
|
|
||||||
hiredis-example-glib: examples/example-glib.c adapters/glib.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< $(shell pkg-config --cflags --libs glib-2.0) $(STLIBNAME)
|
|
||||||
|
|
||||||
hiredis-example-ivykis: examples/example-ivykis.c adapters/ivykis.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< -livykis $(STLIBNAME)
|
|
||||||
|
|
||||||
hiredis-example-macosx: examples/example-macosx.c adapters/macosx.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< -framework CoreFoundation $(STLIBNAME)
|
|
||||||
|
|
||||||
ifndef AE_DIR
|
|
||||||
hiredis-example-ae:
|
|
||||||
@echo "Please specify AE_DIR (e.g. <redis repository>/src)"
|
|
||||||
@false
|
|
||||||
else
|
|
||||||
hiredis-example-ae: examples/example-ae.c adapters/ae.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. -I$(AE_DIR) $< $(AE_DIR)/ae.o $(AE_DIR)/zmalloc.o $(AE_DIR)/../deps/jemalloc/lib/libjemalloc.a -pthread $(STLIBNAME)
|
|
||||||
endif
|
|
||||||
|
|
||||||
ifndef LIBUV_DIR
|
|
||||||
hiredis-example-libuv:
|
|
||||||
@echo "Please specify LIBUV_DIR (e.g. ../libuv/)"
|
|
||||||
@false
|
|
||||||
else
|
|
||||||
hiredis-example-libuv: examples/example-libuv.c adapters/libuv.h $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. -I$(LIBUV_DIR)/include $< $(LIBUV_DIR)/.libs/libuv.a -lpthread -lrt $(STLIBNAME)
|
|
||||||
endif
|
|
||||||
|
|
||||||
ifeq ($(and $(QT_MOC),$(QT_INCLUDE_DIR),$(QT_LIBRARY_DIR)),)
|
|
||||||
hiredis-example-qt:
|
|
||||||
@echo "Please specify QT_MOC, QT_INCLUDE_DIR AND QT_LIBRARY_DIR"
|
|
||||||
@false
|
|
||||||
else
|
|
||||||
hiredis-example-qt: examples/example-qt.cpp adapters/qt.h $(STLIBNAME)
|
|
||||||
$(QT_MOC) adapters/qt.h -I. -I$(QT_INCLUDE_DIR) -I$(QT_INCLUDE_DIR)/QtCore | \
|
|
||||||
$(CXX) -x c++ -o qt-adapter-moc.o -c - $(REAL_CFLAGS) -I. -I$(QT_INCLUDE_DIR) -I$(QT_INCLUDE_DIR)/QtCore
|
|
||||||
$(QT_MOC) examples/example-qt.h -I. -I$(QT_INCLUDE_DIR) -I$(QT_INCLUDE_DIR)/QtCore | \
|
|
||||||
$(CXX) -x c++ -o qt-example-moc.o -c - $(REAL_CFLAGS) -I. -I$(QT_INCLUDE_DIR) -I$(QT_INCLUDE_DIR)/QtCore
|
|
||||||
$(CXX) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. -I$(QT_INCLUDE_DIR) -I$(QT_INCLUDE_DIR)/QtCore -L$(QT_LIBRARY_DIR) qt-adapter-moc.o qt-example-moc.o $< -pthread $(STLIBNAME) -lQtCore
|
|
||||||
endif
|
|
||||||
|
|
||||||
hiredis-example: examples/example.c $(STLIBNAME)
|
|
||||||
$(CC) -o examples/$@ $(REAL_CFLAGS) $(REAL_LDFLAGS) -I. $< $(STLIBNAME)
|
|
||||||
|
|
||||||
examples: $(EXAMPLES)
|
|
||||||
|
|
||||||
hiredis-test: test.o $(STLIBNAME)
|
|
||||||
|
|
||||||
hiredis-%: %.o $(STLIBNAME)
|
|
||||||
$(CC) $(REAL_CFLAGS) -o $@ $(REAL_LDFLAGS) $< $(STLIBNAME)
|
|
||||||
|
|
||||||
test: hiredis-test
|
|
||||||
./hiredis-test
|
|
||||||
|
|
||||||
check: hiredis-test
|
|
||||||
@echo "$$REDIS_TEST_CONFIG" | $(REDIS_SERVER) -
|
|
||||||
$(PRE) ./hiredis-test -h 127.0.0.1 -p $(REDIS_PORT) -s /tmp/hiredis-test-redis.sock || \
|
|
||||||
( kill `cat /tmp/hiredis-test-redis.pid` && false )
|
|
||||||
kill `cat /tmp/hiredis-test-redis.pid`
|
|
||||||
|
|
||||||
.c.o:
|
|
||||||
$(CC) -std=c99 -pedantic -c $(REAL_CFLAGS) $<
|
|
||||||
|
|
||||||
clean:
|
|
||||||
rm -rf $(DYLIBNAME) $(STLIBNAME) $(TESTS) $(PKGCONFNAME) examples/hiredis-example* *.o *.gcda *.gcno *.gcov
|
|
||||||
|
|
||||||
dep:
|
|
||||||
$(CC) -MM *.c
|
|
||||||
|
|
||||||
INSTALL?= cp -pPR
|
|
||||||
|
|
||||||
$(PKGCONFNAME): hiredis.h
|
|
||||||
@echo "Generating $@ for pkgconfig..."
|
|
||||||
@echo prefix=$(PREFIX) > $@
|
|
||||||
@echo exec_prefix=\$${prefix} >> $@
|
|
||||||
@echo libdir=$(PREFIX)/$(LIBRARY_PATH) >> $@
|
|
||||||
@echo includedir=$(PREFIX)/$(INCLUDE_PATH) >> $@
|
|
||||||
@echo >> $@
|
|
||||||
@echo Name: hiredis >> $@
|
|
||||||
@echo Description: Minimalistic C client library for Redis. >> $@
|
|
||||||
@echo Version: $(HIREDIS_MAJOR).$(HIREDIS_MINOR).$(HIREDIS_PATCH) >> $@
|
|
||||||
@echo Libs: -L\$${libdir} -lhiredis >> $@
|
|
||||||
@echo Cflags: -I\$${includedir} -D_FILE_OFFSET_BITS=64 >> $@
|
|
||||||
|
|
||||||
install: $(DYLIBNAME) $(STLIBNAME) $(PKGCONFNAME)
|
|
||||||
mkdir -p $(INSTALL_INCLUDE_PATH) $(INSTALL_INCLUDE_PATH)/adapters $(INSTALL_LIBRARY_PATH)
|
|
||||||
$(INSTALL) hiredis.h async.h read.h sds.h alloc.h $(INSTALL_INCLUDE_PATH)
|
|
||||||
$(INSTALL) adapters/*.h $(INSTALL_INCLUDE_PATH)/adapters
|
|
||||||
$(INSTALL) $(DYLIBNAME) $(INSTALL_LIBRARY_PATH)/$(DYLIB_MINOR_NAME)
|
|
||||||
cd $(INSTALL_LIBRARY_PATH) && ln -sf $(DYLIB_MINOR_NAME) $(DYLIBNAME)
|
|
||||||
$(INSTALL) $(STLIBNAME) $(INSTALL_LIBRARY_PATH)
|
|
||||||
mkdir -p $(INSTALL_PKGCONF_PATH)
|
|
||||||
$(INSTALL) $(PKGCONFNAME) $(INSTALL_PKGCONF_PATH)
|
|
||||||
|
|
||||||
32bit:
|
|
||||||
@echo ""
|
|
||||||
@echo "WARNING: if this fails under Linux you probably need to install libc6-dev-i386"
|
|
||||||
@echo ""
|
|
||||||
$(MAKE) CFLAGS="-m32" LDFLAGS="-m32"
|
|
||||||
|
|
||||||
32bit-vars:
|
|
||||||
$(eval CFLAGS=-m32)
|
|
||||||
$(eval LDFLAGS=-m32)
|
|
||||||
|
|
||||||
gprof:
|
|
||||||
$(MAKE) CFLAGS="-pg" LDFLAGS="-pg"
|
|
||||||
|
|
||||||
gcov:
|
|
||||||
$(MAKE) CFLAGS="-fprofile-arcs -ftest-coverage" LDFLAGS="-fprofile-arcs"
|
|
||||||
|
|
||||||
coverage: gcov
|
|
||||||
make check
|
|
||||||
mkdir -p tmp/lcov
|
|
||||||
lcov -d . -c -o tmp/lcov/hiredis.info
|
|
||||||
genhtml --legend -o tmp/lcov/report tmp/lcov/hiredis.info
|
|
||||||
|
|
||||||
noopt:
|
|
||||||
$(MAKE) OPTIMIZATION=""
|
|
||||||
|
|
||||||
.PHONY: all test check clean dep install 32bit 32bit-vars gprof gcov noopt
|
|
410
controller/thirdparty/hiredis-0.14.1/README.md
vendored
410
controller/thirdparty/hiredis-0.14.1/README.md
vendored
|
@ -1,410 +0,0 @@
|
||||||
[](https://travis-ci.org/redis/hiredis)
|
|
||||||
|
|
||||||
**This Readme reflects the latest changed in the master branch. See [v0.14.1](https://github.com/redis/hiredis/tree/v0.14.1) for the Readme and documentation for the latest release.**
|
|
||||||
|
|
||||||
# HIREDIS
|
|
||||||
|
|
||||||
Hiredis is a minimalistic C client library for the [Redis](http://redis.io/) database.
|
|
||||||
|
|
||||||
It is minimalistic because it just adds minimal support for the protocol, but
|
|
||||||
at the same time it uses a high level printf-alike API in order to make it
|
|
||||||
much higher level than otherwise suggested by its minimal code base and the
|
|
||||||
lack of explicit bindings for every Redis command.
|
|
||||||
|
|
||||||
Apart from supporting sending commands and receiving replies, it comes with
|
|
||||||
a reply parser that is decoupled from the I/O layer. It
|
|
||||||
is a stream parser designed for easy reusability, which can for instance be used
|
|
||||||
in higher level language bindings for efficient reply parsing.
|
|
||||||
|
|
||||||
Hiredis only supports the binary-safe Redis protocol, so you can use it with any
|
|
||||||
Redis version >= 1.2.0.
|
|
||||||
|
|
||||||
The library comes with multiple APIs. There is the
|
|
||||||
*synchronous API*, the *asynchronous API* and the *reply parsing API*.
|
|
||||||
|
|
||||||
## IMPORTANT: Breaking changes when upgrading from 0.13.x -> 0.14.x
|
|
||||||
|
|
||||||
Bulk and multi-bulk lengths less than -1 or greater than `LLONG_MAX` are now
|
|
||||||
protocol errors. This is consistent with the RESP specification. On 32-bit
|
|
||||||
platforms, the upper bound is lowered to `SIZE_MAX`.
|
|
||||||
|
|
||||||
Change `redisReply.len` to `size_t`, as it denotes the the size of a string
|
|
||||||
|
|
||||||
User code should compare this to `size_t` values as well. If it was used to
|
|
||||||
compare to other values, casting might be necessary or can be removed, if
|
|
||||||
casting was applied before.
|
|
||||||
|
|
||||||
For a detailed list of changes please view our [Changelog](CHANGELOG.md).
|
|
||||||
|
|
||||||
## Synchronous API
|
|
||||||
|
|
||||||
To consume the synchronous API, there are only a few function calls that need to be introduced:
|
|
||||||
|
|
||||||
```c
|
|
||||||
redisContext *redisConnect(const char *ip, int port);
|
|
||||||
void *redisCommand(redisContext *c, const char *format, ...);
|
|
||||||
void freeReplyObject(void *reply);
|
|
||||||
```
|
|
||||||
|
|
||||||
### Connecting
|
|
||||||
|
|
||||||
The function `redisConnect` is used to create a so-called `redisContext`. The
|
|
||||||
context is where Hiredis holds state for a connection. The `redisContext`
|
|
||||||
struct has an integer `err` field that is non-zero when the connection is in
|
|
||||||
an error state. The field `errstr` will contain a string with a description of
|
|
||||||
the error. More information on errors can be found in the **Errors** section.
|
|
||||||
After trying to connect to Redis using `redisConnect` you should
|
|
||||||
check the `err` field to see if establishing the connection was successful:
|
|
||||||
```c
|
|
||||||
redisContext *c = redisConnect("127.0.0.1", 6379);
|
|
||||||
if (c == NULL || c->err) {
|
|
||||||
if (c) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
// handle error
|
|
||||||
} else {
|
|
||||||
printf("Can't allocate redis context\n");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
*Note: A `redisContext` is not thread-safe.*
|
|
||||||
|
|
||||||
### Sending commands
|
|
||||||
|
|
||||||
There are several ways to issue commands to Redis. The first that will be introduced is
|
|
||||||
`redisCommand`. This function takes a format similar to printf. In the simplest form,
|
|
||||||
it is used like this:
|
|
||||||
```c
|
|
||||||
reply = redisCommand(context, "SET foo bar");
|
|
||||||
```
|
|
||||||
|
|
||||||
The specifier `%s` interpolates a string in the command, and uses `strlen` to
|
|
||||||
determine the length of the string:
|
|
||||||
```c
|
|
||||||
reply = redisCommand(context, "SET foo %s", value);
|
|
||||||
```
|
|
||||||
When you need to pass binary safe strings in a command, the `%b` specifier can be
|
|
||||||
used. Together with a pointer to the string, it requires a `size_t` length argument
|
|
||||||
of the string:
|
|
||||||
```c
|
|
||||||
reply = redisCommand(context, "SET foo %b", value, (size_t) valuelen);
|
|
||||||
```
|
|
||||||
Internally, Hiredis splits the command in different arguments and will
|
|
||||||
convert it to the protocol used to communicate with Redis.
|
|
||||||
One or more spaces separates arguments, so you can use the specifiers
|
|
||||||
anywhere in an argument:
|
|
||||||
```c
|
|
||||||
reply = redisCommand(context, "SET key:%s %s", myid, value);
|
|
||||||
```
|
|
||||||
|
|
||||||
### Using replies
|
|
||||||
|
|
||||||
The return value of `redisCommand` holds a reply when the command was
|
|
||||||
successfully executed. When an error occurs, the return value is `NULL` and
|
|
||||||
the `err` field in the context will be set (see section on **Errors**).
|
|
||||||
Once an error is returned the context cannot be reused and you should set up
|
|
||||||
a new connection.
|
|
||||||
|
|
||||||
The standard replies that `redisCommand` are of the type `redisReply`. The
|
|
||||||
`type` field in the `redisReply` should be used to test what kind of reply
|
|
||||||
was received:
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_STATUS`**:
|
|
||||||
* The command replied with a status reply. The status string can be accessed using `reply->str`.
|
|
||||||
The length of this string can be accessed using `reply->len`.
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_ERROR`**:
|
|
||||||
* The command replied with an error. The error string can be accessed identical to `REDIS_REPLY_STATUS`.
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_INTEGER`**:
|
|
||||||
* The command replied with an integer. The integer value can be accessed using the
|
|
||||||
`reply->integer` field of type `long long`.
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_NIL`**:
|
|
||||||
* The command replied with a **nil** object. There is no data to access.
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_STRING`**:
|
|
||||||
* A bulk (string) reply. The value of the reply can be accessed using `reply->str`.
|
|
||||||
The length of this string can be accessed using `reply->len`.
|
|
||||||
|
|
||||||
* **`REDIS_REPLY_ARRAY`**:
|
|
||||||
* A multi bulk reply. The number of elements in the multi bulk reply is stored in
|
|
||||||
`reply->elements`. Every element in the multi bulk reply is a `redisReply` object as well
|
|
||||||
and can be accessed via `reply->element[..index..]`.
|
|
||||||
Redis may reply with nested arrays but this is fully supported.
|
|
||||||
|
|
||||||
Replies should be freed using the `freeReplyObject()` function.
|
|
||||||
Note that this function will take care of freeing sub-reply objects
|
|
||||||
contained in arrays and nested arrays, so there is no need for the user to
|
|
||||||
free the sub replies (it is actually harmful and will corrupt the memory).
|
|
||||||
|
|
||||||
**Important:** the current version of hiredis (0.10.0) frees replies when the
|
|
||||||
asynchronous API is used. This means you should not call `freeReplyObject` when
|
|
||||||
you use this API. The reply is cleaned up by hiredis _after_ the callback
|
|
||||||
returns. This behavior will probably change in future releases, so make sure to
|
|
||||||
keep an eye on the changelog when upgrading (see issue #39).
|
|
||||||
|
|
||||||
### Cleaning up
|
|
||||||
|
|
||||||
To disconnect and free the context the following function can be used:
|
|
||||||
```c
|
|
||||||
void redisFree(redisContext *c);
|
|
||||||
```
|
|
||||||
This function immediately closes the socket and then frees the allocations done in
|
|
||||||
creating the context.
|
|
||||||
|
|
||||||
### Sending commands (cont'd)
|
|
||||||
|
|
||||||
Together with `redisCommand`, the function `redisCommandArgv` can be used to issue commands.
|
|
||||||
It has the following prototype:
|
|
||||||
```c
|
|
||||||
void *redisCommandArgv(redisContext *c, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
```
|
|
||||||
It takes the number of arguments `argc`, an array of strings `argv` and the lengths of the
|
|
||||||
arguments `argvlen`. For convenience, `argvlen` may be set to `NULL` and the function will
|
|
||||||
use `strlen(3)` on every argument to determine its length. Obviously, when any of the arguments
|
|
||||||
need to be binary safe, the entire array of lengths `argvlen` should be provided.
|
|
||||||
|
|
||||||
The return value has the same semantic as `redisCommand`.
|
|
||||||
|
|
||||||
### Pipelining
|
|
||||||
|
|
||||||
To explain how Hiredis supports pipelining in a blocking connection, there needs to be
|
|
||||||
understanding of the internal execution flow.
|
|
||||||
|
|
||||||
When any of the functions in the `redisCommand` family is called, Hiredis first formats the
|
|
||||||
command according to the Redis protocol. The formatted command is then put in the output buffer
|
|
||||||
of the context. This output buffer is dynamic, so it can hold any number of commands.
|
|
||||||
After the command is put in the output buffer, `redisGetReply` is called. This function has the
|
|
||||||
following two execution paths:
|
|
||||||
|
|
||||||
1. The input buffer is non-empty:
|
|
||||||
* Try to parse a single reply from the input buffer and return it
|
|
||||||
* If no reply could be parsed, continue at *2*
|
|
||||||
2. The input buffer is empty:
|
|
||||||
* Write the **entire** output buffer to the socket
|
|
||||||
* Read from the socket until a single reply could be parsed
|
|
||||||
|
|
||||||
The function `redisGetReply` is exported as part of the Hiredis API and can be used when a reply
|
|
||||||
is expected on the socket. To pipeline commands, the only things that needs to be done is
|
|
||||||
filling up the output buffer. For this cause, two commands can be used that are identical
|
|
||||||
to the `redisCommand` family, apart from not returning a reply:
|
|
||||||
```c
|
|
||||||
void redisAppendCommand(redisContext *c, const char *format, ...);
|
|
||||||
void redisAppendCommandArgv(redisContext *c, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
```
|
|
||||||
After calling either function one or more times, `redisGetReply` can be used to receive the
|
|
||||||
subsequent replies. The return value for this function is either `REDIS_OK` or `REDIS_ERR`, where
|
|
||||||
the latter means an error occurred while reading a reply. Just as with the other commands,
|
|
||||||
the `err` field in the context can be used to find out what the cause of this error is.
|
|
||||||
|
|
||||||
The following examples shows a simple pipeline (resulting in only a single call to `write(2)` and
|
|
||||||
a single call to `read(2)`):
|
|
||||||
```c
|
|
||||||
redisReply *reply;
|
|
||||||
redisAppendCommand(context,"SET foo bar");
|
|
||||||
redisAppendCommand(context,"GET foo");
|
|
||||||
redisGetReply(context,&reply); // reply for SET
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisGetReply(context,&reply); // reply for GET
|
|
||||||
freeReplyObject(reply);
|
|
||||||
```
|
|
||||||
This API can also be used to implement a blocking subscriber:
|
|
||||||
```c
|
|
||||||
reply = redisCommand(context,"SUBSCRIBE foo");
|
|
||||||
freeReplyObject(reply);
|
|
||||||
while(redisGetReply(context,&reply) == REDIS_OK) {
|
|
||||||
// consume message
|
|
||||||
freeReplyObject(reply);
|
|
||||||
}
|
|
||||||
```
|
|
||||||
### Errors
|
|
||||||
|
|
||||||
When a function call is not successful, depending on the function either `NULL` or `REDIS_ERR` is
|
|
||||||
returned. The `err` field inside the context will be non-zero and set to one of the
|
|
||||||
following constants:
|
|
||||||
|
|
||||||
* **`REDIS_ERR_IO`**:
|
|
||||||
There was an I/O error while creating the connection, trying to write
|
|
||||||
to the socket or read from the socket. If you included `errno.h` in your
|
|
||||||
application, you can use the global `errno` variable to find out what is
|
|
||||||
wrong.
|
|
||||||
|
|
||||||
* **`REDIS_ERR_EOF`**:
|
|
||||||
The server closed the connection which resulted in an empty read.
|
|
||||||
|
|
||||||
* **`REDIS_ERR_PROTOCOL`**:
|
|
||||||
There was an error while parsing the protocol.
|
|
||||||
|
|
||||||
* **`REDIS_ERR_OTHER`**:
|
|
||||||
Any other error. Currently, it is only used when a specified hostname to connect
|
|
||||||
to cannot be resolved.
|
|
||||||
|
|
||||||
In every case, the `errstr` field in the context will be set to hold a string representation
|
|
||||||
of the error.
|
|
||||||
|
|
||||||
## Asynchronous API
|
|
||||||
|
|
||||||
Hiredis comes with an asynchronous API that works easily with any event library.
|
|
||||||
Examples are bundled that show using Hiredis with [libev](http://software.schmorp.de/pkg/libev.html)
|
|
||||||
and [libevent](http://monkey.org/~provos/libevent/).
|
|
||||||
|
|
||||||
### Connecting
|
|
||||||
|
|
||||||
The function `redisAsyncConnect` can be used to establish a non-blocking connection to
|
|
||||||
Redis. It returns a pointer to the newly created `redisAsyncContext` struct. The `err` field
|
|
||||||
should be checked after creation to see if there were errors creating the connection.
|
|
||||||
Because the connection that will be created is non-blocking, the kernel is not able to
|
|
||||||
instantly return if the specified host and port is able to accept a connection.
|
|
||||||
|
|
||||||
*Note: A `redisAsyncContext` is not thread-safe.*
|
|
||||||
|
|
||||||
```c
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
// handle error
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
The asynchronous context can hold a disconnect callback function that is called when the
|
|
||||||
connection is disconnected (either because of an error or per user request). This function should
|
|
||||||
have the following prototype:
|
|
||||||
```c
|
|
||||||
void(const redisAsyncContext *c, int status);
|
|
||||||
```
|
|
||||||
On a disconnect, the `status` argument is set to `REDIS_OK` when disconnection was initiated by the
|
|
||||||
user, or `REDIS_ERR` when the disconnection was caused by an error. When it is `REDIS_ERR`, the `err`
|
|
||||||
field in the context can be accessed to find out the cause of the error.
|
|
||||||
|
|
||||||
The context object is always freed after the disconnect callback fired. When a reconnect is needed,
|
|
||||||
the disconnect callback is a good point to do so.
|
|
||||||
|
|
||||||
Setting the disconnect callback can only be done once per context. For subsequent calls it will
|
|
||||||
return `REDIS_ERR`. The function to set the disconnect callback has the following prototype:
|
|
||||||
```c
|
|
||||||
int redisAsyncSetDisconnectCallback(redisAsyncContext *ac, redisDisconnectCallback *fn);
|
|
||||||
```
|
|
||||||
### Sending commands and their callbacks
|
|
||||||
|
|
||||||
In an asynchronous context, commands are automatically pipelined due to the nature of an event loop.
|
|
||||||
Therefore, unlike the synchronous API, there is only a single way to send commands.
|
|
||||||
Because commands are sent to Redis asynchronously, issuing a command requires a callback function
|
|
||||||
that is called when the reply is received. Reply callbacks should have the following prototype:
|
|
||||||
```c
|
|
||||||
void(redisAsyncContext *c, void *reply, void *privdata);
|
|
||||||
```
|
|
||||||
The `privdata` argument can be used to curry arbitrary data to the callback from the point where
|
|
||||||
the command is initially queued for execution.
|
|
||||||
|
|
||||||
The functions that can be used to issue commands in an asynchronous context are:
|
|
||||||
```c
|
|
||||||
int redisAsyncCommand(
|
|
||||||
redisAsyncContext *ac, redisCallbackFn *fn, void *privdata,
|
|
||||||
const char *format, ...);
|
|
||||||
int redisAsyncCommandArgv(
|
|
||||||
redisAsyncContext *ac, redisCallbackFn *fn, void *privdata,
|
|
||||||
int argc, const char **argv, const size_t *argvlen);
|
|
||||||
```
|
|
||||||
Both functions work like their blocking counterparts. The return value is `REDIS_OK` when the command
|
|
||||||
was successfully added to the output buffer and `REDIS_ERR` otherwise. Example: when the connection
|
|
||||||
is being disconnected per user-request, no new commands may be added to the output buffer and `REDIS_ERR` is
|
|
||||||
returned on calls to the `redisAsyncCommand` family.
|
|
||||||
|
|
||||||
If the reply for a command with a `NULL` callback is read, it is immediately freed. When the callback
|
|
||||||
for a command is non-`NULL`, the memory is freed immediately following the callback: the reply is only
|
|
||||||
valid for the duration of the callback.
|
|
||||||
|
|
||||||
All pending callbacks are called with a `NULL` reply when the context encountered an error.
|
|
||||||
|
|
||||||
### Disconnecting
|
|
||||||
|
|
||||||
An asynchronous connection can be terminated using:
|
|
||||||
```c
|
|
||||||
void redisAsyncDisconnect(redisAsyncContext *ac);
|
|
||||||
```
|
|
||||||
When this function is called, the connection is **not** immediately terminated. Instead, new
|
|
||||||
commands are no longer accepted and the connection is only terminated when all pending commands
|
|
||||||
have been written to the socket, their respective replies have been read and their respective
|
|
||||||
callbacks have been executed. After this, the disconnection callback is executed with the
|
|
||||||
`REDIS_OK` status and the context object is freed.
|
|
||||||
|
|
||||||
### Hooking it up to event library *X*
|
|
||||||
|
|
||||||
There are a few hooks that need to be set on the context object after it is created.
|
|
||||||
See the `adapters/` directory for bindings to *libev* and *libevent*.
|
|
||||||
|
|
||||||
## Reply parsing API
|
|
||||||
|
|
||||||
Hiredis comes with a reply parsing API that makes it easy for writing higher
|
|
||||||
level language bindings.
|
|
||||||
|
|
||||||
The reply parsing API consists of the following functions:
|
|
||||||
```c
|
|
||||||
redisReader *redisReaderCreate(void);
|
|
||||||
void redisReaderFree(redisReader *reader);
|
|
||||||
int redisReaderFeed(redisReader *reader, const char *buf, size_t len);
|
|
||||||
int redisReaderGetReply(redisReader *reader, void **reply);
|
|
||||||
```
|
|
||||||
The same set of functions are used internally by hiredis when creating a
|
|
||||||
normal Redis context, the above API just exposes it to the user for a direct
|
|
||||||
usage.
|
|
||||||
|
|
||||||
### Usage
|
|
||||||
|
|
||||||
The function `redisReaderCreate` creates a `redisReader` structure that holds a
|
|
||||||
buffer with unparsed data and state for the protocol parser.
|
|
||||||
|
|
||||||
Incoming data -- most likely from a socket -- can be placed in the internal
|
|
||||||
buffer of the `redisReader` using `redisReaderFeed`. This function will make a
|
|
||||||
copy of the buffer pointed to by `buf` for `len` bytes. This data is parsed
|
|
||||||
when `redisReaderGetReply` is called. This function returns an integer status
|
|
||||||
and a reply object (as described above) via `void **reply`. The returned status
|
|
||||||
can be either `REDIS_OK` or `REDIS_ERR`, where the latter means something went
|
|
||||||
wrong (either a protocol error, or an out of memory error).
|
|
||||||
|
|
||||||
The parser limits the level of nesting for multi bulk payloads to 7. If the
|
|
||||||
multi bulk nesting level is higher than this, the parser returns an error.
|
|
||||||
|
|
||||||
### Customizing replies
|
|
||||||
|
|
||||||
The function `redisReaderGetReply` creates `redisReply` and makes the function
|
|
||||||
argument `reply` point to the created `redisReply` variable. For instance, if
|
|
||||||
the response of type `REDIS_REPLY_STATUS` then the `str` field of `redisReply`
|
|
||||||
will hold the status as a vanilla C string. However, the functions that are
|
|
||||||
responsible for creating instances of the `redisReply` can be customized by
|
|
||||||
setting the `fn` field on the `redisReader` struct. This should be done
|
|
||||||
immediately after creating the `redisReader`.
|
|
||||||
|
|
||||||
For example, [hiredis-rb](https://github.com/pietern/hiredis-rb/blob/master/ext/hiredis_ext/reader.c)
|
|
||||||
uses customized reply object functions to create Ruby objects.
|
|
||||||
|
|
||||||
### Reader max buffer
|
|
||||||
|
|
||||||
Both when using the Reader API directly or when using it indirectly via a
|
|
||||||
normal Redis context, the redisReader structure uses a buffer in order to
|
|
||||||
accumulate data from the server.
|
|
||||||
Usually this buffer is destroyed when it is empty and is larger than 16
|
|
||||||
KiB in order to avoid wasting memory in unused buffers
|
|
||||||
|
|
||||||
However when working with very big payloads destroying the buffer may slow
|
|
||||||
down performances considerably, so it is possible to modify the max size of
|
|
||||||
an idle buffer changing the value of the `maxbuf` field of the reader structure
|
|
||||||
to the desired value. The special value of 0 means that there is no maximum
|
|
||||||
value for an idle buffer, so the buffer will never get freed.
|
|
||||||
|
|
||||||
For instance if you have a normal Redis context you can set the maximum idle
|
|
||||||
buffer to zero (unlimited) just with:
|
|
||||||
```c
|
|
||||||
context->reader->maxbuf = 0;
|
|
||||||
```
|
|
||||||
This should be done only in order to maximize performances when working with
|
|
||||||
large payloads. The context should be set back to `REDIS_READER_MAX_BUF` again
|
|
||||||
as soon as possible in order to prevent allocation of useless memory.
|
|
||||||
|
|
||||||
## AUTHORS
|
|
||||||
|
|
||||||
Hiredis was written by Salvatore Sanfilippo (antirez at gmail) and
|
|
||||||
Pieter Noordhuis (pcnoordhuis at gmail) and is released under the BSD license.
|
|
||||||
Hiredis is currently maintained by Matt Stancliff (matt at genges dot com) and
|
|
||||||
Jan-Erik Rediger (janerik at fnordig dot com)
|
|
127
controller/thirdparty/hiredis-0.14.1/adapters/ae.h
vendored
127
controller/thirdparty/hiredis-0.14.1/adapters/ae.h
vendored
|
@ -1,127 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_AE_H__
|
|
||||||
#define __HIREDIS_AE_H__
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <ae.h>
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct redisAeEvents {
|
|
||||||
redisAsyncContext *context;
|
|
||||||
aeEventLoop *loop;
|
|
||||||
int fd;
|
|
||||||
int reading, writing;
|
|
||||||
} redisAeEvents;
|
|
||||||
|
|
||||||
static void redisAeReadEvent(aeEventLoop *el, int fd, void *privdata, int mask) {
|
|
||||||
((void)el); ((void)fd); ((void)mask);
|
|
||||||
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
redisAsyncHandleRead(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeWriteEvent(aeEventLoop *el, int fd, void *privdata, int mask) {
|
|
||||||
((void)el); ((void)fd); ((void)mask);
|
|
||||||
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
redisAsyncHandleWrite(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeAddRead(void *privdata) {
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
aeEventLoop *loop = e->loop;
|
|
||||||
if (!e->reading) {
|
|
||||||
e->reading = 1;
|
|
||||||
aeCreateFileEvent(loop,e->fd,AE_READABLE,redisAeReadEvent,e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeDelRead(void *privdata) {
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
aeEventLoop *loop = e->loop;
|
|
||||||
if (e->reading) {
|
|
||||||
e->reading = 0;
|
|
||||||
aeDeleteFileEvent(loop,e->fd,AE_READABLE);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeAddWrite(void *privdata) {
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
aeEventLoop *loop = e->loop;
|
|
||||||
if (!e->writing) {
|
|
||||||
e->writing = 1;
|
|
||||||
aeCreateFileEvent(loop,e->fd,AE_WRITABLE,redisAeWriteEvent,e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeDelWrite(void *privdata) {
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
aeEventLoop *loop = e->loop;
|
|
||||||
if (e->writing) {
|
|
||||||
e->writing = 0;
|
|
||||||
aeDeleteFileEvent(loop,e->fd,AE_WRITABLE);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisAeCleanup(void *privdata) {
|
|
||||||
redisAeEvents *e = (redisAeEvents*)privdata;
|
|
||||||
redisAeDelRead(privdata);
|
|
||||||
redisAeDelWrite(privdata);
|
|
||||||
free(e);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisAeAttach(aeEventLoop *loop, redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisAeEvents *e;
|
|
||||||
|
|
||||||
/* Nothing should be attached when something is already attached */
|
|
||||||
if (ac->ev.data != NULL)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Create container for context and r/w events */
|
|
||||||
e = (redisAeEvents*)hi_malloc(sizeof(*e));
|
|
||||||
e->context = ac;
|
|
||||||
e->loop = loop;
|
|
||||||
e->fd = c->fd;
|
|
||||||
e->reading = e->writing = 0;
|
|
||||||
|
|
||||||
/* Register functions to start/stop listening for events */
|
|
||||||
ac->ev.addRead = redisAeAddRead;
|
|
||||||
ac->ev.delRead = redisAeDelRead;
|
|
||||||
ac->ev.addWrite = redisAeAddWrite;
|
|
||||||
ac->ev.delWrite = redisAeDelWrite;
|
|
||||||
ac->ev.cleanup = redisAeCleanup;
|
|
||||||
ac->ev.data = e;
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
#endif
|
|
153
controller/thirdparty/hiredis-0.14.1/adapters/glib.h
vendored
153
controller/thirdparty/hiredis-0.14.1/adapters/glib.h
vendored
|
@ -1,153 +0,0 @@
|
||||||
#ifndef __HIREDIS_GLIB_H__
|
|
||||||
#define __HIREDIS_GLIB_H__
|
|
||||||
|
|
||||||
#include <glib.h>
|
|
||||||
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct
|
|
||||||
{
|
|
||||||
GSource source;
|
|
||||||
redisAsyncContext *ac;
|
|
||||||
GPollFD poll_fd;
|
|
||||||
} RedisSource;
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_add_read (gpointer data)
|
|
||||||
{
|
|
||||||
RedisSource *source = (RedisSource *)data;
|
|
||||||
g_return_if_fail(source);
|
|
||||||
source->poll_fd.events |= G_IO_IN;
|
|
||||||
g_main_context_wakeup(g_source_get_context((GSource *)data));
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_del_read (gpointer data)
|
|
||||||
{
|
|
||||||
RedisSource *source = (RedisSource *)data;
|
|
||||||
g_return_if_fail(source);
|
|
||||||
source->poll_fd.events &= ~G_IO_IN;
|
|
||||||
g_main_context_wakeup(g_source_get_context((GSource *)data));
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_add_write (gpointer data)
|
|
||||||
{
|
|
||||||
RedisSource *source = (RedisSource *)data;
|
|
||||||
g_return_if_fail(source);
|
|
||||||
source->poll_fd.events |= G_IO_OUT;
|
|
||||||
g_main_context_wakeup(g_source_get_context((GSource *)data));
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_del_write (gpointer data)
|
|
||||||
{
|
|
||||||
RedisSource *source = (RedisSource *)data;
|
|
||||||
g_return_if_fail(source);
|
|
||||||
source->poll_fd.events &= ~G_IO_OUT;
|
|
||||||
g_main_context_wakeup(g_source_get_context((GSource *)data));
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_cleanup (gpointer data)
|
|
||||||
{
|
|
||||||
RedisSource *source = (RedisSource *)data;
|
|
||||||
|
|
||||||
g_return_if_fail(source);
|
|
||||||
|
|
||||||
redis_source_del_read(source);
|
|
||||||
redis_source_del_write(source);
|
|
||||||
/*
|
|
||||||
* It is not our responsibility to remove ourself from the
|
|
||||||
* current main loop. However, we will remove the GPollFD.
|
|
||||||
*/
|
|
||||||
if (source->poll_fd.fd >= 0) {
|
|
||||||
g_source_remove_poll((GSource *)data, &source->poll_fd);
|
|
||||||
source->poll_fd.fd = -1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static gboolean
|
|
||||||
redis_source_prepare (GSource *source,
|
|
||||||
gint *timeout_)
|
|
||||||
{
|
|
||||||
RedisSource *redis = (RedisSource *)source;
|
|
||||||
*timeout_ = -1;
|
|
||||||
return !!(redis->poll_fd.events & redis->poll_fd.revents);
|
|
||||||
}
|
|
||||||
|
|
||||||
static gboolean
|
|
||||||
redis_source_check (GSource *source)
|
|
||||||
{
|
|
||||||
RedisSource *redis = (RedisSource *)source;
|
|
||||||
return !!(redis->poll_fd.events & redis->poll_fd.revents);
|
|
||||||
}
|
|
||||||
|
|
||||||
static gboolean
|
|
||||||
redis_source_dispatch (GSource *source,
|
|
||||||
GSourceFunc callback,
|
|
||||||
gpointer user_data)
|
|
||||||
{
|
|
||||||
RedisSource *redis = (RedisSource *)source;
|
|
||||||
|
|
||||||
if ((redis->poll_fd.revents & G_IO_OUT)) {
|
|
||||||
redisAsyncHandleWrite(redis->ac);
|
|
||||||
redis->poll_fd.revents &= ~G_IO_OUT;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ((redis->poll_fd.revents & G_IO_IN)) {
|
|
||||||
redisAsyncHandleRead(redis->ac);
|
|
||||||
redis->poll_fd.revents &= ~G_IO_IN;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (callback) {
|
|
||||||
return callback(user_data);
|
|
||||||
}
|
|
||||||
|
|
||||||
return TRUE;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
redis_source_finalize (GSource *source)
|
|
||||||
{
|
|
||||||
RedisSource *redis = (RedisSource *)source;
|
|
||||||
|
|
||||||
if (redis->poll_fd.fd >= 0) {
|
|
||||||
g_source_remove_poll(source, &redis->poll_fd);
|
|
||||||
redis->poll_fd.fd = -1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static GSource *
|
|
||||||
redis_source_new (redisAsyncContext *ac)
|
|
||||||
{
|
|
||||||
static GSourceFuncs source_funcs = {
|
|
||||||
.prepare = redis_source_prepare,
|
|
||||||
.check = redis_source_check,
|
|
||||||
.dispatch = redis_source_dispatch,
|
|
||||||
.finalize = redis_source_finalize,
|
|
||||||
};
|
|
||||||
redisContext *c = &ac->c;
|
|
||||||
RedisSource *source;
|
|
||||||
|
|
||||||
g_return_val_if_fail(ac != NULL, NULL);
|
|
||||||
|
|
||||||
source = (RedisSource *)g_source_new(&source_funcs, sizeof *source);
|
|
||||||
source->ac = ac;
|
|
||||||
source->poll_fd.fd = c->fd;
|
|
||||||
source->poll_fd.events = 0;
|
|
||||||
source->poll_fd.revents = 0;
|
|
||||||
g_source_add_poll((GSource *)source, &source->poll_fd);
|
|
||||||
|
|
||||||
ac->ev.addRead = redis_source_add_read;
|
|
||||||
ac->ev.delRead = redis_source_del_read;
|
|
||||||
ac->ev.addWrite = redis_source_add_write;
|
|
||||||
ac->ev.delWrite = redis_source_del_write;
|
|
||||||
ac->ev.cleanup = redis_source_cleanup;
|
|
||||||
ac->ev.data = source;
|
|
||||||
|
|
||||||
return (GSource *)source;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif /* __HIREDIS_GLIB_H__ */
|
|
|
@ -1,81 +0,0 @@
|
||||||
#ifndef __HIREDIS_IVYKIS_H__
|
|
||||||
#define __HIREDIS_IVYKIS_H__
|
|
||||||
#include <iv.h>
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct redisIvykisEvents {
|
|
||||||
redisAsyncContext *context;
|
|
||||||
struct iv_fd fd;
|
|
||||||
} redisIvykisEvents;
|
|
||||||
|
|
||||||
static void redisIvykisReadEvent(void *arg) {
|
|
||||||
redisAsyncContext *context = (redisAsyncContext *)arg;
|
|
||||||
redisAsyncHandleRead(context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisWriteEvent(void *arg) {
|
|
||||||
redisAsyncContext *context = (redisAsyncContext *)arg;
|
|
||||||
redisAsyncHandleWrite(context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisAddRead(void *privdata) {
|
|
||||||
redisIvykisEvents *e = (redisIvykisEvents*)privdata;
|
|
||||||
iv_fd_set_handler_in(&e->fd, redisIvykisReadEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisDelRead(void *privdata) {
|
|
||||||
redisIvykisEvents *e = (redisIvykisEvents*)privdata;
|
|
||||||
iv_fd_set_handler_in(&e->fd, NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisAddWrite(void *privdata) {
|
|
||||||
redisIvykisEvents *e = (redisIvykisEvents*)privdata;
|
|
||||||
iv_fd_set_handler_out(&e->fd, redisIvykisWriteEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisDelWrite(void *privdata) {
|
|
||||||
redisIvykisEvents *e = (redisIvykisEvents*)privdata;
|
|
||||||
iv_fd_set_handler_out(&e->fd, NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisIvykisCleanup(void *privdata) {
|
|
||||||
redisIvykisEvents *e = (redisIvykisEvents*)privdata;
|
|
||||||
|
|
||||||
iv_fd_unregister(&e->fd);
|
|
||||||
free(e);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisIvykisAttach(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisIvykisEvents *e;
|
|
||||||
|
|
||||||
/* Nothing should be attached when something is already attached */
|
|
||||||
if (ac->ev.data != NULL)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Create container for context and r/w events */
|
|
||||||
e = (redisIvykisEvents*)hi_malloc(sizeof(*e));
|
|
||||||
e->context = ac;
|
|
||||||
|
|
||||||
/* Register functions to start/stop listening for events */
|
|
||||||
ac->ev.addRead = redisIvykisAddRead;
|
|
||||||
ac->ev.delRead = redisIvykisDelRead;
|
|
||||||
ac->ev.addWrite = redisIvykisAddWrite;
|
|
||||||
ac->ev.delWrite = redisIvykisDelWrite;
|
|
||||||
ac->ev.cleanup = redisIvykisCleanup;
|
|
||||||
ac->ev.data = e;
|
|
||||||
|
|
||||||
/* Initialize and install read/write events */
|
|
||||||
IV_FD_INIT(&e->fd);
|
|
||||||
e->fd.fd = c->fd;
|
|
||||||
e->fd.handler_in = redisIvykisReadEvent;
|
|
||||||
e->fd.handler_out = redisIvykisWriteEvent;
|
|
||||||
e->fd.handler_err = NULL;
|
|
||||||
e->fd.cookie = e->context;
|
|
||||||
|
|
||||||
iv_fd_register(&e->fd);
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
#endif
|
|
|
@ -1,147 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_LIBEV_H__
|
|
||||||
#define __HIREDIS_LIBEV_H__
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <ev.h>
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct redisLibevEvents {
|
|
||||||
redisAsyncContext *context;
|
|
||||||
struct ev_loop *loop;
|
|
||||||
int reading, writing;
|
|
||||||
ev_io rev, wev;
|
|
||||||
} redisLibevEvents;
|
|
||||||
|
|
||||||
static void redisLibevReadEvent(EV_P_ ev_io *watcher, int revents) {
|
|
||||||
#if EV_MULTIPLICITY
|
|
||||||
((void)loop);
|
|
||||||
#endif
|
|
||||||
((void)revents);
|
|
||||||
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)watcher->data;
|
|
||||||
redisAsyncHandleRead(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevWriteEvent(EV_P_ ev_io *watcher, int revents) {
|
|
||||||
#if EV_MULTIPLICITY
|
|
||||||
((void)loop);
|
|
||||||
#endif
|
|
||||||
((void)revents);
|
|
||||||
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)watcher->data;
|
|
||||||
redisAsyncHandleWrite(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevAddRead(void *privdata) {
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)privdata;
|
|
||||||
struct ev_loop *loop = e->loop;
|
|
||||||
((void)loop);
|
|
||||||
if (!e->reading) {
|
|
||||||
e->reading = 1;
|
|
||||||
ev_io_start(EV_A_ &e->rev);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevDelRead(void *privdata) {
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)privdata;
|
|
||||||
struct ev_loop *loop = e->loop;
|
|
||||||
((void)loop);
|
|
||||||
if (e->reading) {
|
|
||||||
e->reading = 0;
|
|
||||||
ev_io_stop(EV_A_ &e->rev);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevAddWrite(void *privdata) {
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)privdata;
|
|
||||||
struct ev_loop *loop = e->loop;
|
|
||||||
((void)loop);
|
|
||||||
if (!e->writing) {
|
|
||||||
e->writing = 1;
|
|
||||||
ev_io_start(EV_A_ &e->wev);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevDelWrite(void *privdata) {
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)privdata;
|
|
||||||
struct ev_loop *loop = e->loop;
|
|
||||||
((void)loop);
|
|
||||||
if (e->writing) {
|
|
||||||
e->writing = 0;
|
|
||||||
ev_io_stop(EV_A_ &e->wev);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibevCleanup(void *privdata) {
|
|
||||||
redisLibevEvents *e = (redisLibevEvents*)privdata;
|
|
||||||
redisLibevDelRead(privdata);
|
|
||||||
redisLibevDelWrite(privdata);
|
|
||||||
free(e);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisLibevAttach(EV_P_ redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisLibevEvents *e;
|
|
||||||
|
|
||||||
/* Nothing should be attached when something is already attached */
|
|
||||||
if (ac->ev.data != NULL)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Create container for context and r/w events */
|
|
||||||
e = (redisLibevEvents*)hi_malloc(sizeof(*e));
|
|
||||||
e->context = ac;
|
|
||||||
#if EV_MULTIPLICITY
|
|
||||||
e->loop = loop;
|
|
||||||
#else
|
|
||||||
e->loop = NULL;
|
|
||||||
#endif
|
|
||||||
e->reading = e->writing = 0;
|
|
||||||
e->rev.data = e;
|
|
||||||
e->wev.data = e;
|
|
||||||
|
|
||||||
/* Register functions to start/stop listening for events */
|
|
||||||
ac->ev.addRead = redisLibevAddRead;
|
|
||||||
ac->ev.delRead = redisLibevDelRead;
|
|
||||||
ac->ev.addWrite = redisLibevAddWrite;
|
|
||||||
ac->ev.delWrite = redisLibevDelWrite;
|
|
||||||
ac->ev.cleanup = redisLibevCleanup;
|
|
||||||
ac->ev.data = e;
|
|
||||||
|
|
||||||
/* Initialize read/write events */
|
|
||||||
ev_io_init(&e->rev,redisLibevReadEvent,c->fd,EV_READ);
|
|
||||||
ev_io_init(&e->wev,redisLibevWriteEvent,c->fd,EV_WRITE);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
|
@ -1,108 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_LIBEVENT_H__
|
|
||||||
#define __HIREDIS_LIBEVENT_H__
|
|
||||||
#include <event2/event.h>
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct redisLibeventEvents {
|
|
||||||
redisAsyncContext *context;
|
|
||||||
struct event *rev, *wev;
|
|
||||||
} redisLibeventEvents;
|
|
||||||
|
|
||||||
static void redisLibeventReadEvent(int fd, short event, void *arg) {
|
|
||||||
((void)fd); ((void)event);
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)arg;
|
|
||||||
redisAsyncHandleRead(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventWriteEvent(int fd, short event, void *arg) {
|
|
||||||
((void)fd); ((void)event);
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)arg;
|
|
||||||
redisAsyncHandleWrite(e->context);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventAddRead(void *privdata) {
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
||||||
event_add(e->rev,NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventDelRead(void *privdata) {
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
||||||
event_del(e->rev);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventAddWrite(void *privdata) {
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
||||||
event_add(e->wev,NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventDelWrite(void *privdata) {
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
||||||
event_del(e->wev);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisLibeventCleanup(void *privdata) {
|
|
||||||
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
||||||
event_free(e->rev);
|
|
||||||
event_free(e->wev);
|
|
||||||
free(e);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisLibeventAttach(redisAsyncContext *ac, struct event_base *base) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisLibeventEvents *e;
|
|
||||||
|
|
||||||
/* Nothing should be attached when something is already attached */
|
|
||||||
if (ac->ev.data != NULL)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Create container for context and r/w events */
|
|
||||||
e = (redisLibeventEvents*)hi_calloc(1, sizeof(*e));
|
|
||||||
e->context = ac;
|
|
||||||
|
|
||||||
/* Register functions to start/stop listening for events */
|
|
||||||
ac->ev.addRead = redisLibeventAddRead;
|
|
||||||
ac->ev.delRead = redisLibeventDelRead;
|
|
||||||
ac->ev.addWrite = redisLibeventAddWrite;
|
|
||||||
ac->ev.delWrite = redisLibeventDelWrite;
|
|
||||||
ac->ev.cleanup = redisLibeventCleanup;
|
|
||||||
ac->ev.data = e;
|
|
||||||
|
|
||||||
/* Initialize and install read/write events */
|
|
||||||
e->rev = event_new(base, c->fd, EV_READ, redisLibeventReadEvent, e);
|
|
||||||
e->wev = event_new(base, c->fd, EV_WRITE, redisLibeventWriteEvent, e);
|
|
||||||
event_add(e->rev, NULL);
|
|
||||||
event_add(e->wev, NULL);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
#endif
|
|
|
@ -1,122 +0,0 @@
|
||||||
#ifndef __HIREDIS_LIBUV_H__
|
|
||||||
#define __HIREDIS_LIBUV_H__
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <uv.h>
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
#include <string.h>
|
|
||||||
|
|
||||||
typedef struct redisLibuvEvents {
|
|
||||||
redisAsyncContext* context;
|
|
||||||
uv_poll_t handle;
|
|
||||||
int events;
|
|
||||||
} redisLibuvEvents;
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvPoll(uv_poll_t* handle, int status, int events) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)handle->data;
|
|
||||||
|
|
||||||
if (status != 0) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (p->context != NULL && (events & UV_READABLE)) {
|
|
||||||
redisAsyncHandleRead(p->context);
|
|
||||||
}
|
|
||||||
if (p->context != NULL && (events & UV_WRITABLE)) {
|
|
||||||
redisAsyncHandleWrite(p->context);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvAddRead(void *privdata) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)privdata;
|
|
||||||
|
|
||||||
p->events |= UV_READABLE;
|
|
||||||
|
|
||||||
uv_poll_start(&p->handle, p->events, redisLibuvPoll);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvDelRead(void *privdata) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)privdata;
|
|
||||||
|
|
||||||
p->events &= ~UV_READABLE;
|
|
||||||
|
|
||||||
if (p->events) {
|
|
||||||
uv_poll_start(&p->handle, p->events, redisLibuvPoll);
|
|
||||||
} else {
|
|
||||||
uv_poll_stop(&p->handle);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvAddWrite(void *privdata) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)privdata;
|
|
||||||
|
|
||||||
p->events |= UV_WRITABLE;
|
|
||||||
|
|
||||||
uv_poll_start(&p->handle, p->events, redisLibuvPoll);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvDelWrite(void *privdata) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)privdata;
|
|
||||||
|
|
||||||
p->events &= ~UV_WRITABLE;
|
|
||||||
|
|
||||||
if (p->events) {
|
|
||||||
uv_poll_start(&p->handle, p->events, redisLibuvPoll);
|
|
||||||
} else {
|
|
||||||
uv_poll_stop(&p->handle);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void on_close(uv_handle_t* handle) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)handle->data;
|
|
||||||
|
|
||||||
free(p);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static void redisLibuvCleanup(void *privdata) {
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)privdata;
|
|
||||||
|
|
||||||
p->context = NULL; // indicate that context might no longer exist
|
|
||||||
uv_close((uv_handle_t*)&p->handle, on_close);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static int redisLibuvAttach(redisAsyncContext* ac, uv_loop_t* loop) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
|
|
||||||
if (ac->ev.data != NULL) {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
ac->ev.addRead = redisLibuvAddRead;
|
|
||||||
ac->ev.delRead = redisLibuvDelRead;
|
|
||||||
ac->ev.addWrite = redisLibuvAddWrite;
|
|
||||||
ac->ev.delWrite = redisLibuvDelWrite;
|
|
||||||
ac->ev.cleanup = redisLibuvCleanup;
|
|
||||||
|
|
||||||
redisLibuvEvents* p = (redisLibuvEvents*)malloc(sizeof(*p));
|
|
||||||
|
|
||||||
if (!p) {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
memset(p, 0, sizeof(*p));
|
|
||||||
|
|
||||||
if (uv_poll_init(loop, &p->handle, c->fd) != 0) {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
ac->ev.data = p;
|
|
||||||
p->handle.data = p;
|
|
||||||
p->context = ac;
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
#endif
|
|
|
@ -1,114 +0,0 @@
|
||||||
//
|
|
||||||
// Created by Дмитрий Бахвалов on 13.07.15.
|
|
||||||
// Copyright (c) 2015 Dmitry Bakhvalov. All rights reserved.
|
|
||||||
//
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_MACOSX_H__
|
|
||||||
#define __HIREDIS_MACOSX_H__
|
|
||||||
|
|
||||||
#include <CoreFoundation/CoreFoundation.h>
|
|
||||||
|
|
||||||
#include "../hiredis.h"
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
redisAsyncContext *context;
|
|
||||||
CFSocketRef socketRef;
|
|
||||||
CFRunLoopSourceRef sourceRef;
|
|
||||||
} RedisRunLoop;
|
|
||||||
|
|
||||||
static int freeRedisRunLoop(RedisRunLoop* redisRunLoop) {
|
|
||||||
if( redisRunLoop != NULL ) {
|
|
||||||
if( redisRunLoop->sourceRef != NULL ) {
|
|
||||||
CFRunLoopSourceInvalidate(redisRunLoop->sourceRef);
|
|
||||||
CFRelease(redisRunLoop->sourceRef);
|
|
||||||
}
|
|
||||||
if( redisRunLoop->socketRef != NULL ) {
|
|
||||||
CFSocketInvalidate(redisRunLoop->socketRef);
|
|
||||||
CFRelease(redisRunLoop->socketRef);
|
|
||||||
}
|
|
||||||
free(redisRunLoop);
|
|
||||||
}
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSAddRead(void *privdata) {
|
|
||||||
RedisRunLoop *redisRunLoop = (RedisRunLoop*)privdata;
|
|
||||||
CFSocketEnableCallBacks(redisRunLoop->socketRef, kCFSocketReadCallBack);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSDelRead(void *privdata) {
|
|
||||||
RedisRunLoop *redisRunLoop = (RedisRunLoop*)privdata;
|
|
||||||
CFSocketDisableCallBacks(redisRunLoop->socketRef, kCFSocketReadCallBack);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSAddWrite(void *privdata) {
|
|
||||||
RedisRunLoop *redisRunLoop = (RedisRunLoop*)privdata;
|
|
||||||
CFSocketEnableCallBacks(redisRunLoop->socketRef, kCFSocketWriteCallBack);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSDelWrite(void *privdata) {
|
|
||||||
RedisRunLoop *redisRunLoop = (RedisRunLoop*)privdata;
|
|
||||||
CFSocketDisableCallBacks(redisRunLoop->socketRef, kCFSocketWriteCallBack);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSCleanup(void *privdata) {
|
|
||||||
RedisRunLoop *redisRunLoop = (RedisRunLoop*)privdata;
|
|
||||||
freeRedisRunLoop(redisRunLoop);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void redisMacOSAsyncCallback(CFSocketRef __unused s, CFSocketCallBackType callbackType, CFDataRef __unused address, const void __unused *data, void *info) {
|
|
||||||
redisAsyncContext* context = (redisAsyncContext*) info;
|
|
||||||
|
|
||||||
switch (callbackType) {
|
|
||||||
case kCFSocketReadCallBack:
|
|
||||||
redisAsyncHandleRead(context);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case kCFSocketWriteCallBack:
|
|
||||||
redisAsyncHandleWrite(context);
|
|
||||||
break;
|
|
||||||
|
|
||||||
default:
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisMacOSAttach(redisAsyncContext *redisAsyncCtx, CFRunLoopRef runLoop) {
|
|
||||||
redisContext *redisCtx = &(redisAsyncCtx->c);
|
|
||||||
|
|
||||||
/* Nothing should be attached when something is already attached */
|
|
||||||
if( redisAsyncCtx->ev.data != NULL ) return REDIS_ERR;
|
|
||||||
|
|
||||||
RedisRunLoop* redisRunLoop = (RedisRunLoop*) calloc(1, sizeof(RedisRunLoop));
|
|
||||||
if( !redisRunLoop ) return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Setup redis stuff */
|
|
||||||
redisRunLoop->context = redisAsyncCtx;
|
|
||||||
|
|
||||||
redisAsyncCtx->ev.addRead = redisMacOSAddRead;
|
|
||||||
redisAsyncCtx->ev.delRead = redisMacOSDelRead;
|
|
||||||
redisAsyncCtx->ev.addWrite = redisMacOSAddWrite;
|
|
||||||
redisAsyncCtx->ev.delWrite = redisMacOSDelWrite;
|
|
||||||
redisAsyncCtx->ev.cleanup = redisMacOSCleanup;
|
|
||||||
redisAsyncCtx->ev.data = redisRunLoop;
|
|
||||||
|
|
||||||
/* Initialize and install read/write events */
|
|
||||||
CFSocketContext socketCtx = { 0, redisAsyncCtx, NULL, NULL, NULL };
|
|
||||||
|
|
||||||
redisRunLoop->socketRef = CFSocketCreateWithNative(NULL, redisCtx->fd,
|
|
||||||
kCFSocketReadCallBack | kCFSocketWriteCallBack,
|
|
||||||
redisMacOSAsyncCallback,
|
|
||||||
&socketCtx);
|
|
||||||
if( !redisRunLoop->socketRef ) return freeRedisRunLoop(redisRunLoop);
|
|
||||||
|
|
||||||
redisRunLoop->sourceRef = CFSocketCreateRunLoopSource(NULL, redisRunLoop->socketRef, 0);
|
|
||||||
if( !redisRunLoop->sourceRef ) return freeRedisRunLoop(redisRunLoop);
|
|
||||||
|
|
||||||
CFRunLoopAddSource(runLoop, redisRunLoop->sourceRef, kCFRunLoopDefaultMode);
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
135
controller/thirdparty/hiredis-0.14.1/adapters/qt.h
vendored
135
controller/thirdparty/hiredis-0.14.1/adapters/qt.h
vendored
|
@ -1,135 +0,0 @@
|
||||||
/*-
|
|
||||||
* Copyright (C) 2014 Pietro Cerutti <gahr@gahr.ch>
|
|
||||||
*
|
|
||||||
* Redistribution and use in source and binary forms, with or without
|
|
||||||
* modification, are permitted provided that the following conditions
|
|
||||||
* are met:
|
|
||||||
* 1. Redistributions of source code must retain the above copyright
|
|
||||||
* notice, this list of conditions and the following disclaimer.
|
|
||||||
* 2. 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.
|
|
||||||
*
|
|
||||||
* THIS SOFTWARE IS PROVIDED BY AUTHOR 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 AUTHOR 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_QT_H__
|
|
||||||
#define __HIREDIS_QT_H__
|
|
||||||
#include <QSocketNotifier>
|
|
||||||
#include "../async.h"
|
|
||||||
|
|
||||||
static void RedisQtAddRead(void *);
|
|
||||||
static void RedisQtDelRead(void *);
|
|
||||||
static void RedisQtAddWrite(void *);
|
|
||||||
static void RedisQtDelWrite(void *);
|
|
||||||
static void RedisQtCleanup(void *);
|
|
||||||
|
|
||||||
class RedisQtAdapter : public QObject {
|
|
||||||
|
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
friend
|
|
||||||
void RedisQtAddRead(void * adapter) {
|
|
||||||
RedisQtAdapter * a = static_cast<RedisQtAdapter *>(adapter);
|
|
||||||
a->addRead();
|
|
||||||
}
|
|
||||||
|
|
||||||
friend
|
|
||||||
void RedisQtDelRead(void * adapter) {
|
|
||||||
RedisQtAdapter * a = static_cast<RedisQtAdapter *>(adapter);
|
|
||||||
a->delRead();
|
|
||||||
}
|
|
||||||
|
|
||||||
friend
|
|
||||||
void RedisQtAddWrite(void * adapter) {
|
|
||||||
RedisQtAdapter * a = static_cast<RedisQtAdapter *>(adapter);
|
|
||||||
a->addWrite();
|
|
||||||
}
|
|
||||||
|
|
||||||
friend
|
|
||||||
void RedisQtDelWrite(void * adapter) {
|
|
||||||
RedisQtAdapter * a = static_cast<RedisQtAdapter *>(adapter);
|
|
||||||
a->delWrite();
|
|
||||||
}
|
|
||||||
|
|
||||||
friend
|
|
||||||
void RedisQtCleanup(void * adapter) {
|
|
||||||
RedisQtAdapter * a = static_cast<RedisQtAdapter *>(adapter);
|
|
||||||
a->cleanup();
|
|
||||||
}
|
|
||||||
|
|
||||||
public:
|
|
||||||
RedisQtAdapter(QObject * parent = 0)
|
|
||||||
: QObject(parent), m_ctx(0), m_read(0), m_write(0) { }
|
|
||||||
|
|
||||||
~RedisQtAdapter() {
|
|
||||||
if (m_ctx != 0) {
|
|
||||||
m_ctx->ev.data = NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int setContext(redisAsyncContext * ac) {
|
|
||||||
if (ac->ev.data != NULL) {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
m_ctx = ac;
|
|
||||||
m_ctx->ev.data = this;
|
|
||||||
m_ctx->ev.addRead = RedisQtAddRead;
|
|
||||||
m_ctx->ev.delRead = RedisQtDelRead;
|
|
||||||
m_ctx->ev.addWrite = RedisQtAddWrite;
|
|
||||||
m_ctx->ev.delWrite = RedisQtDelWrite;
|
|
||||||
m_ctx->ev.cleanup = RedisQtCleanup;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
void addRead() {
|
|
||||||
if (m_read) return;
|
|
||||||
m_read = new QSocketNotifier(m_ctx->c.fd, QSocketNotifier::Read, 0);
|
|
||||||
connect(m_read, SIGNAL(activated(int)), this, SLOT(read()));
|
|
||||||
}
|
|
||||||
|
|
||||||
void delRead() {
|
|
||||||
if (!m_read) return;
|
|
||||||
delete m_read;
|
|
||||||
m_read = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void addWrite() {
|
|
||||||
if (m_write) return;
|
|
||||||
m_write = new QSocketNotifier(m_ctx->c.fd, QSocketNotifier::Write, 0);
|
|
||||||
connect(m_write, SIGNAL(activated(int)), this, SLOT(write()));
|
|
||||||
}
|
|
||||||
|
|
||||||
void delWrite() {
|
|
||||||
if (!m_write) return;
|
|
||||||
delete m_write;
|
|
||||||
m_write = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void cleanup() {
|
|
||||||
delRead();
|
|
||||||
delWrite();
|
|
||||||
}
|
|
||||||
|
|
||||||
private slots:
|
|
||||||
void read() { redisAsyncHandleRead(m_ctx); }
|
|
||||||
void write() { redisAsyncHandleWrite(m_ctx); }
|
|
||||||
|
|
||||||
private:
|
|
||||||
redisAsyncContext * m_ctx;
|
|
||||||
QSocketNotifier * m_read;
|
|
||||||
QSocketNotifier * m_write;
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif /* !__HIREDIS_QT_H__ */
|
|
65
controller/thirdparty/hiredis-0.14.1/alloc.c
vendored
65
controller/thirdparty/hiredis-0.14.1/alloc.c
vendored
|
@ -1,65 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2020, Michael Grunder <michael dot grunder at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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 "fmacros.h"
|
|
||||||
#include "alloc.h"
|
|
||||||
#include <string.h>
|
|
||||||
|
|
||||||
void *hi_malloc(size_t size) {
|
|
||||||
void *ptr = malloc(size);
|
|
||||||
if (ptr == NULL)
|
|
||||||
HIREDIS_OOM_HANDLER;
|
|
||||||
|
|
||||||
return ptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
void *hi_calloc(size_t nmemb, size_t size) {
|
|
||||||
void *ptr = calloc(nmemb, size);
|
|
||||||
if (ptr == NULL)
|
|
||||||
HIREDIS_OOM_HANDLER;
|
|
||||||
|
|
||||||
return ptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
void *hi_realloc(void *ptr, size_t size) {
|
|
||||||
void *newptr = realloc(ptr, size);
|
|
||||||
if (newptr == NULL)
|
|
||||||
HIREDIS_OOM_HANDLER;
|
|
||||||
|
|
||||||
return newptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
char *hi_strdup(const char *str) {
|
|
||||||
char *newstr = strdup(str);
|
|
||||||
if (newstr == NULL)
|
|
||||||
HIREDIS_OOM_HANDLER;
|
|
||||||
|
|
||||||
return newstr;
|
|
||||||
}
|
|
53
controller/thirdparty/hiredis-0.14.1/alloc.h
vendored
53
controller/thirdparty/hiredis-0.14.1/alloc.h
vendored
|
@ -1,53 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2020, Michael Grunder <michael dot grunder at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef HIREDIS_ALLOC_H
|
|
||||||
#define HIREDIS_ALLOC_H
|
|
||||||
|
|
||||||
#include <stdlib.h> /* for size_t */
|
|
||||||
|
|
||||||
#ifndef HIREDIS_OOM_HANDLER
|
|
||||||
#define HIREDIS_OOM_HANDLER abort()
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
void *hi_malloc(size_t size);
|
|
||||||
void *hi_calloc(size_t nmemb, size_t size);
|
|
||||||
void *hi_realloc(void *ptr, size_t size);
|
|
||||||
char *hi_strdup(const char *str);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif /* HIREDIS_ALLOC_H */
|
|
|
@ -1,23 +0,0 @@
|
||||||
# Appveyor configuration file for CI build of hiredis on Windows (under Cygwin)
|
|
||||||
environment:
|
|
||||||
matrix:
|
|
||||||
- CYG_BASH: C:\cygwin64\bin\bash
|
|
||||||
CC: gcc
|
|
||||||
- CYG_BASH: C:\cygwin\bin\bash
|
|
||||||
CC: gcc
|
|
||||||
TARGET: 32bit
|
|
||||||
TARGET_VARS: 32bit-vars
|
|
||||||
|
|
||||||
clone_depth: 1
|
|
||||||
|
|
||||||
# Attempt to ensure we don't try to convert line endings to Win32 CRLF as this will cause build to fail
|
|
||||||
init:
|
|
||||||
- git config --global core.autocrlf input
|
|
||||||
|
|
||||||
# Install needed build dependencies
|
|
||||||
install:
|
|
||||||
- '%CYG_BASH% -lc "cygcheck -dc cygwin"'
|
|
||||||
|
|
||||||
build_script:
|
|
||||||
- 'echo building...'
|
|
||||||
- '%CYG_BASH% -lc "cd $APPVEYOR_BUILD_FOLDER; exec 0</dev/null; make LDFLAGS=$LDFLAGS CC=$CC $TARGET CFLAGS=$CFLAGS && make LDFLAGS=$LDFLAGS CC=$CC $TARGET_VARS hiredis-example"'
|
|
717
controller/thirdparty/hiredis-0.14.1/async.c
vendored
717
controller/thirdparty/hiredis-0.14.1/async.c
vendored
|
@ -1,717 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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 "fmacros.h"
|
|
||||||
#include "alloc.h"
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <strings.h>
|
|
||||||
#include <assert.h>
|
|
||||||
#include <ctype.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include "async.h"
|
|
||||||
#include "net.h"
|
|
||||||
#include "dict.c"
|
|
||||||
#include "sds.h"
|
|
||||||
|
|
||||||
#define _EL_ADD_READ(ctx) do { \
|
|
||||||
if ((ctx)->ev.addRead) (ctx)->ev.addRead((ctx)->ev.data); \
|
|
||||||
} while(0)
|
|
||||||
#define _EL_DEL_READ(ctx) do { \
|
|
||||||
if ((ctx)->ev.delRead) (ctx)->ev.delRead((ctx)->ev.data); \
|
|
||||||
} while(0)
|
|
||||||
#define _EL_ADD_WRITE(ctx) do { \
|
|
||||||
if ((ctx)->ev.addWrite) (ctx)->ev.addWrite((ctx)->ev.data); \
|
|
||||||
} while(0)
|
|
||||||
#define _EL_DEL_WRITE(ctx) do { \
|
|
||||||
if ((ctx)->ev.delWrite) (ctx)->ev.delWrite((ctx)->ev.data); \
|
|
||||||
} while(0)
|
|
||||||
#define _EL_CLEANUP(ctx) do { \
|
|
||||||
if ((ctx)->ev.cleanup) (ctx)->ev.cleanup((ctx)->ev.data); \
|
|
||||||
} while(0);
|
|
||||||
|
|
||||||
/* Forward declaration of function in hiredis.c */
|
|
||||||
int __redisAppendCommand(redisContext *c, const char *cmd, size_t len);
|
|
||||||
|
|
||||||
/* Functions managing dictionary of callbacks for pub/sub. */
|
|
||||||
static unsigned int callbackHash(const void *key) {
|
|
||||||
return dictGenHashFunction((const unsigned char *)key,
|
|
||||||
sdslen((const sds)key));
|
|
||||||
}
|
|
||||||
|
|
||||||
static void *callbackValDup(void *privdata, const void *src) {
|
|
||||||
((void) privdata);
|
|
||||||
redisCallback *dup = hi_malloc(sizeof(*dup));
|
|
||||||
memcpy(dup,src,sizeof(*dup));
|
|
||||||
return dup;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int callbackKeyCompare(void *privdata, const void *key1, const void *key2) {
|
|
||||||
int l1, l2;
|
|
||||||
((void) privdata);
|
|
||||||
|
|
||||||
l1 = sdslen((const sds)key1);
|
|
||||||
l2 = sdslen((const sds)key2);
|
|
||||||
if (l1 != l2) return 0;
|
|
||||||
return memcmp(key1,key2,l1) == 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void callbackKeyDestructor(void *privdata, void *key) {
|
|
||||||
((void) privdata);
|
|
||||||
sdsfree((sds)key);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void callbackValDestructor(void *privdata, void *val) {
|
|
||||||
((void) privdata);
|
|
||||||
free(val);
|
|
||||||
}
|
|
||||||
|
|
||||||
static dictType callbackDict = {
|
|
||||||
callbackHash,
|
|
||||||
NULL,
|
|
||||||
callbackValDup,
|
|
||||||
callbackKeyCompare,
|
|
||||||
callbackKeyDestructor,
|
|
||||||
callbackValDestructor
|
|
||||||
};
|
|
||||||
|
|
||||||
static redisAsyncContext *redisAsyncInitialize(redisContext *c) {
|
|
||||||
redisAsyncContext *ac;
|
|
||||||
|
|
||||||
ac = realloc(c,sizeof(redisAsyncContext));
|
|
||||||
if (ac == NULL)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
c = &(ac->c);
|
|
||||||
|
|
||||||
/* The regular connect functions will always set the flag REDIS_CONNECTED.
|
|
||||||
* For the async API, we want to wait until the first write event is
|
|
||||||
* received up before setting this flag, so reset it here. */
|
|
||||||
c->flags &= ~REDIS_CONNECTED;
|
|
||||||
|
|
||||||
ac->err = 0;
|
|
||||||
ac->errstr = NULL;
|
|
||||||
ac->data = NULL;
|
|
||||||
|
|
||||||
ac->ev.data = NULL;
|
|
||||||
ac->ev.addRead = NULL;
|
|
||||||
ac->ev.delRead = NULL;
|
|
||||||
ac->ev.addWrite = NULL;
|
|
||||||
ac->ev.delWrite = NULL;
|
|
||||||
ac->ev.cleanup = NULL;
|
|
||||||
|
|
||||||
ac->onConnect = NULL;
|
|
||||||
ac->onDisconnect = NULL;
|
|
||||||
|
|
||||||
ac->replies.head = NULL;
|
|
||||||
ac->replies.tail = NULL;
|
|
||||||
ac->sub.invalid.head = NULL;
|
|
||||||
ac->sub.invalid.tail = NULL;
|
|
||||||
ac->sub.channels = dictCreate(&callbackDict,NULL);
|
|
||||||
ac->sub.patterns = dictCreate(&callbackDict,NULL);
|
|
||||||
return ac;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* We want the error field to be accessible directly instead of requiring
|
|
||||||
* an indirection to the redisContext struct. */
|
|
||||||
static void __redisAsyncCopyError(redisAsyncContext *ac) {
|
|
||||||
if (!ac)
|
|
||||||
return;
|
|
||||||
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
ac->err = c->err;
|
|
||||||
ac->errstr = c->errstr;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncContext *redisAsyncConnect(const char *ip, int port) {
|
|
||||||
redisContext *c;
|
|
||||||
redisAsyncContext *ac;
|
|
||||||
|
|
||||||
c = redisConnectNonBlock(ip,port);
|
|
||||||
if (c == NULL)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
ac = redisAsyncInitialize(c);
|
|
||||||
if (ac == NULL) {
|
|
||||||
redisFree(c);
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
__redisAsyncCopyError(ac);
|
|
||||||
return ac;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncContext *redisAsyncConnectBind(const char *ip, int port,
|
|
||||||
const char *source_addr) {
|
|
||||||
redisContext *c = redisConnectBindNonBlock(ip,port,source_addr);
|
|
||||||
redisAsyncContext *ac = redisAsyncInitialize(c);
|
|
||||||
__redisAsyncCopyError(ac);
|
|
||||||
return ac;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncContext *redisAsyncConnectBindWithReuse(const char *ip, int port,
|
|
||||||
const char *source_addr) {
|
|
||||||
redisContext *c = redisConnectBindNonBlockWithReuse(ip,port,source_addr);
|
|
||||||
redisAsyncContext *ac = redisAsyncInitialize(c);
|
|
||||||
__redisAsyncCopyError(ac);
|
|
||||||
return ac;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncContext *redisAsyncConnectUnix(const char *path) {
|
|
||||||
redisContext *c;
|
|
||||||
redisAsyncContext *ac;
|
|
||||||
|
|
||||||
c = redisConnectUnixNonBlock(path);
|
|
||||||
if (c == NULL)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
ac = redisAsyncInitialize(c);
|
|
||||||
if (ac == NULL) {
|
|
||||||
redisFree(c);
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
__redisAsyncCopyError(ac);
|
|
||||||
return ac;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisAsyncSetConnectCallback(redisAsyncContext *ac, redisConnectCallback *fn) {
|
|
||||||
if (ac->onConnect == NULL) {
|
|
||||||
ac->onConnect = fn;
|
|
||||||
|
|
||||||
/* The common way to detect an established connection is to wait for
|
|
||||||
* the first write event to be fired. This assumes the related event
|
|
||||||
* library functions are already set. */
|
|
||||||
_EL_ADD_WRITE(ac);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisAsyncSetDisconnectCallback(redisAsyncContext *ac, redisDisconnectCallback *fn) {
|
|
||||||
if (ac->onDisconnect == NULL) {
|
|
||||||
ac->onDisconnect = fn;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Helper functions to push/shift callbacks */
|
|
||||||
static int __redisPushCallback(redisCallbackList *list, redisCallback *source) {
|
|
||||||
redisCallback *cb;
|
|
||||||
|
|
||||||
/* Copy callback from stack to heap */
|
|
||||||
cb = malloc(sizeof(*cb));
|
|
||||||
if (cb == NULL)
|
|
||||||
return REDIS_ERR_OOM;
|
|
||||||
|
|
||||||
if (source != NULL) {
|
|
||||||
memcpy(cb,source,sizeof(*cb));
|
|
||||||
cb->next = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Store callback in list */
|
|
||||||
if (list->head == NULL)
|
|
||||||
list->head = cb;
|
|
||||||
if (list->tail != NULL)
|
|
||||||
list->tail->next = cb;
|
|
||||||
list->tail = cb;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int __redisShiftCallback(redisCallbackList *list, redisCallback *target) {
|
|
||||||
redisCallback *cb = list->head;
|
|
||||||
if (cb != NULL) {
|
|
||||||
list->head = cb->next;
|
|
||||||
if (cb == list->tail)
|
|
||||||
list->tail = NULL;
|
|
||||||
|
|
||||||
/* Copy callback from heap to stack */
|
|
||||||
if (target != NULL)
|
|
||||||
memcpy(target,cb,sizeof(*cb));
|
|
||||||
free(cb);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void __redisRunCallback(redisAsyncContext *ac, redisCallback *cb, redisReply *reply) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
if (cb->fn != NULL) {
|
|
||||||
c->flags |= REDIS_IN_CALLBACK;
|
|
||||||
cb->fn(ac,reply,cb->privdata);
|
|
||||||
c->flags &= ~REDIS_IN_CALLBACK;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Helper function to free the context. */
|
|
||||||
static void __redisAsyncFree(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisCallback cb;
|
|
||||||
dictIterator *it;
|
|
||||||
dictEntry *de;
|
|
||||||
|
|
||||||
/* Execute pending callbacks with NULL reply. */
|
|
||||||
while (__redisShiftCallback(&ac->replies,&cb) == REDIS_OK)
|
|
||||||
__redisRunCallback(ac,&cb,NULL);
|
|
||||||
|
|
||||||
/* Execute callbacks for invalid commands */
|
|
||||||
while (__redisShiftCallback(&ac->sub.invalid,&cb) == REDIS_OK)
|
|
||||||
__redisRunCallback(ac,&cb,NULL);
|
|
||||||
|
|
||||||
/* Run subscription callbacks callbacks with NULL reply */
|
|
||||||
it = dictGetIterator(ac->sub.channels);
|
|
||||||
while ((de = dictNext(it)) != NULL)
|
|
||||||
__redisRunCallback(ac,dictGetEntryVal(de),NULL);
|
|
||||||
dictReleaseIterator(it);
|
|
||||||
dictRelease(ac->sub.channels);
|
|
||||||
|
|
||||||
it = dictGetIterator(ac->sub.patterns);
|
|
||||||
while ((de = dictNext(it)) != NULL)
|
|
||||||
__redisRunCallback(ac,dictGetEntryVal(de),NULL);
|
|
||||||
dictReleaseIterator(it);
|
|
||||||
dictRelease(ac->sub.patterns);
|
|
||||||
|
|
||||||
/* Signal event lib to clean up */
|
|
||||||
_EL_CLEANUP(ac);
|
|
||||||
|
|
||||||
/* Execute disconnect callback. When redisAsyncFree() initiated destroying
|
|
||||||
* this context, the status will always be REDIS_OK. */
|
|
||||||
if (ac->onDisconnect && (c->flags & REDIS_CONNECTED)) {
|
|
||||||
if (c->flags & REDIS_FREEING) {
|
|
||||||
ac->onDisconnect(ac,REDIS_OK);
|
|
||||||
} else {
|
|
||||||
ac->onDisconnect(ac,(ac->err == 0) ? REDIS_OK : REDIS_ERR);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Cleanup self */
|
|
||||||
redisFree(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Free the async context. When this function is called from a callback,
|
|
||||||
* control needs to be returned to redisProcessCallbacks() before actual
|
|
||||||
* free'ing. To do so, a flag is set on the context which is picked up by
|
|
||||||
* redisProcessCallbacks(). Otherwise, the context is immediately free'd. */
|
|
||||||
void redisAsyncFree(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
c->flags |= REDIS_FREEING;
|
|
||||||
if (!(c->flags & REDIS_IN_CALLBACK))
|
|
||||||
__redisAsyncFree(ac);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Helper function to make the disconnect happen and clean up. */
|
|
||||||
static void __redisAsyncDisconnect(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
|
|
||||||
/* Make sure error is accessible if there is any */
|
|
||||||
__redisAsyncCopyError(ac);
|
|
||||||
|
|
||||||
if (ac->err == 0) {
|
|
||||||
/* For clean disconnects, there should be no pending callbacks. */
|
|
||||||
int ret = __redisShiftCallback(&ac->replies,NULL);
|
|
||||||
assert(ret == REDIS_ERR);
|
|
||||||
} else {
|
|
||||||
/* Disconnection is caused by an error, make sure that pending
|
|
||||||
* callbacks cannot call new commands. */
|
|
||||||
c->flags |= REDIS_DISCONNECTING;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* For non-clean disconnects, __redisAsyncFree() will execute pending
|
|
||||||
* callbacks with a NULL-reply. */
|
|
||||||
__redisAsyncFree(ac);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Tries to do a clean disconnect from Redis, meaning it stops new commands
|
|
||||||
* from being issued, but tries to flush the output buffer and execute
|
|
||||||
* callbacks for all remaining replies. When this function is called from a
|
|
||||||
* callback, there might be more replies and we can safely defer disconnecting
|
|
||||||
* to redisProcessCallbacks(). Otherwise, we can only disconnect immediately
|
|
||||||
* when there are no pending callbacks. */
|
|
||||||
void redisAsyncDisconnect(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
c->flags |= REDIS_DISCONNECTING;
|
|
||||||
if (!(c->flags & REDIS_IN_CALLBACK) && ac->replies.head == NULL)
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int __redisGetSubscribeCallback(redisAsyncContext *ac, redisReply *reply, redisCallback *dstcb) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
dict *callbacks;
|
|
||||||
redisCallback *cb;
|
|
||||||
dictEntry *de;
|
|
||||||
int pvariant;
|
|
||||||
char *stype;
|
|
||||||
sds sname;
|
|
||||||
|
|
||||||
/* Custom reply functions are not supported for pub/sub. This will fail
|
|
||||||
* very hard when they are used... */
|
|
||||||
if (reply->type == REDIS_REPLY_ARRAY) {
|
|
||||||
assert(reply->elements >= 2);
|
|
||||||
assert(reply->element[0]->type == REDIS_REPLY_STRING);
|
|
||||||
stype = reply->element[0]->str;
|
|
||||||
pvariant = (tolower(stype[0]) == 'p') ? 1 : 0;
|
|
||||||
|
|
||||||
if (pvariant)
|
|
||||||
callbacks = ac->sub.patterns;
|
|
||||||
else
|
|
||||||
callbacks = ac->sub.channels;
|
|
||||||
|
|
||||||
/* Locate the right callback */
|
|
||||||
assert(reply->element[1]->type == REDIS_REPLY_STRING);
|
|
||||||
sname = sdsnewlen(reply->element[1]->str,reply->element[1]->len);
|
|
||||||
de = dictFind(callbacks,sname);
|
|
||||||
if (de != NULL) {
|
|
||||||
cb = dictGetEntryVal(de);
|
|
||||||
|
|
||||||
/* If this is an subscribe reply decrease pending counter. */
|
|
||||||
if (strcasecmp(stype+pvariant,"subscribe") == 0) {
|
|
||||||
cb->pending_subs -= 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
memcpy(dstcb,cb,sizeof(*dstcb));
|
|
||||||
|
|
||||||
/* If this is an unsubscribe message, remove it. */
|
|
||||||
if (strcasecmp(stype+pvariant,"unsubscribe") == 0) {
|
|
||||||
if (cb->pending_subs == 0)
|
|
||||||
dictDelete(callbacks,sname);
|
|
||||||
|
|
||||||
/* If this was the last unsubscribe message, revert to
|
|
||||||
* non-subscribe mode. */
|
|
||||||
assert(reply->element[2]->type == REDIS_REPLY_INTEGER);
|
|
||||||
|
|
||||||
/* Unset subscribed flag only when no pipelined pending subscribe. */
|
|
||||||
if (reply->element[2]->integer == 0
|
|
||||||
&& dictSize(ac->sub.channels) == 0
|
|
||||||
&& dictSize(ac->sub.patterns) == 0)
|
|
||||||
c->flags &= ~REDIS_SUBSCRIBED;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sdsfree(sname);
|
|
||||||
} else {
|
|
||||||
/* Shift callback for invalid commands. */
|
|
||||||
__redisShiftCallback(&ac->sub.invalid,dstcb);
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
void redisProcessCallbacks(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisCallback cb = {NULL, NULL, 0, NULL};
|
|
||||||
void *reply = NULL;
|
|
||||||
int status;
|
|
||||||
|
|
||||||
while((status = redisGetReply(c,&reply)) == REDIS_OK) {
|
|
||||||
if (reply == NULL) {
|
|
||||||
/* When the connection is being disconnected and there are
|
|
||||||
* no more replies, this is the cue to really disconnect. */
|
|
||||||
if (c->flags & REDIS_DISCONNECTING && sdslen(c->obuf) == 0
|
|
||||||
&& ac->replies.head == NULL) {
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* If monitor mode, repush callback */
|
|
||||||
if(c->flags & REDIS_MONITORING) {
|
|
||||||
__redisPushCallback(&ac->replies,&cb);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* When the connection is not being disconnected, simply stop
|
|
||||||
* trying to get replies and wait for the next loop tick. */
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Even if the context is subscribed, pending regular callbacks will
|
|
||||||
* get a reply before pub/sub messages arrive. */
|
|
||||||
if (__redisShiftCallback(&ac->replies,&cb) != REDIS_OK) {
|
|
||||||
/*
|
|
||||||
* A spontaneous reply in a not-subscribed context can be the error
|
|
||||||
* reply that is sent when a new connection exceeds the maximum
|
|
||||||
* number of allowed connections on the server side.
|
|
||||||
*
|
|
||||||
* This is seen as an error instead of a regular reply because the
|
|
||||||
* server closes the connection after sending it.
|
|
||||||
*
|
|
||||||
* To prevent the error from being overwritten by an EOF error the
|
|
||||||
* connection is closed here. See issue #43.
|
|
||||||
*
|
|
||||||
* Another possibility is that the server is loading its dataset.
|
|
||||||
* In this case we also want to close the connection, and have the
|
|
||||||
* user wait until the server is ready to take our request.
|
|
||||||
*/
|
|
||||||
if (((redisReply*)reply)->type == REDIS_REPLY_ERROR) {
|
|
||||||
c->err = REDIS_ERR_OTHER;
|
|
||||||
snprintf(c->errstr,sizeof(c->errstr),"%s",((redisReply*)reply)->str);
|
|
||||||
c->reader->fn->freeObject(reply);
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
/* No more regular callbacks and no errors, the context *must* be subscribed or monitoring. */
|
|
||||||
assert((c->flags & REDIS_SUBSCRIBED || c->flags & REDIS_MONITORING));
|
|
||||||
if(c->flags & REDIS_SUBSCRIBED)
|
|
||||||
__redisGetSubscribeCallback(ac,reply,&cb);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (cb.fn != NULL) {
|
|
||||||
__redisRunCallback(ac,&cb,reply);
|
|
||||||
c->reader->fn->freeObject(reply);
|
|
||||||
|
|
||||||
/* Proceed with free'ing when redisAsyncFree() was called. */
|
|
||||||
if (c->flags & REDIS_FREEING) {
|
|
||||||
__redisAsyncFree(ac);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
/* No callback for this reply. This can either be a NULL callback,
|
|
||||||
* or there were no callbacks to begin with. Either way, don't
|
|
||||||
* abort with an error, but simply ignore it because the client
|
|
||||||
* doesn't know what the server will spit out over the wire. */
|
|
||||||
c->reader->fn->freeObject(reply);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Disconnect when there was an error reading the reply */
|
|
||||||
if (status != REDIS_OK)
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Internal helper function to detect socket status the first time a read or
|
|
||||||
* write event fires. When connecting was not successful, the connect callback
|
|
||||||
* is called with a REDIS_ERR status and the context is free'd. */
|
|
||||||
static int __redisAsyncHandleConnect(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
|
|
||||||
if (redisCheckSocketError(c) == REDIS_ERR) {
|
|
||||||
/* Try again later when connect(2) is still in progress. */
|
|
||||||
if (errno == EINPROGRESS)
|
|
||||||
return REDIS_OK;
|
|
||||||
|
|
||||||
if (ac->onConnect) ac->onConnect(ac,REDIS_ERR);
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Mark context as connected. */
|
|
||||||
c->flags |= REDIS_CONNECTED;
|
|
||||||
if (ac->onConnect) ac->onConnect(ac,REDIS_OK);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* This function should be called when the socket is readable.
|
|
||||||
* It processes all replies that can be read and executes their callbacks.
|
|
||||||
*/
|
|
||||||
void redisAsyncHandleRead(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
|
|
||||||
if (!(c->flags & REDIS_CONNECTED)) {
|
|
||||||
/* Abort connect was not successful. */
|
|
||||||
if (__redisAsyncHandleConnect(ac) != REDIS_OK)
|
|
||||||
return;
|
|
||||||
/* Try again later when the context is still not connected. */
|
|
||||||
if (!(c->flags & REDIS_CONNECTED))
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redisBufferRead(c) == REDIS_ERR) {
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
} else {
|
|
||||||
/* Always re-schedule reads */
|
|
||||||
_EL_ADD_READ(ac);
|
|
||||||
redisProcessCallbacks(ac);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void redisAsyncHandleWrite(redisAsyncContext *ac) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
int done = 0;
|
|
||||||
|
|
||||||
if (!(c->flags & REDIS_CONNECTED)) {
|
|
||||||
/* Abort connect was not successful. */
|
|
||||||
if (__redisAsyncHandleConnect(ac) != REDIS_OK)
|
|
||||||
return;
|
|
||||||
/* Try again later when the context is still not connected. */
|
|
||||||
if (!(c->flags & REDIS_CONNECTED))
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redisBufferWrite(c,&done) == REDIS_ERR) {
|
|
||||||
__redisAsyncDisconnect(ac);
|
|
||||||
} else {
|
|
||||||
/* Continue writing when not done, stop writing otherwise */
|
|
||||||
if (!done)
|
|
||||||
_EL_ADD_WRITE(ac);
|
|
||||||
else
|
|
||||||
_EL_DEL_WRITE(ac);
|
|
||||||
|
|
||||||
/* Always schedule reads after writes */
|
|
||||||
_EL_ADD_READ(ac);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Sets a pointer to the first argument and its length starting at p. Returns
|
|
||||||
* the number of bytes to skip to get to the following argument. */
|
|
||||||
static const char *nextArgument(const char *start, const char **str, size_t *len) {
|
|
||||||
const char *p = start;
|
|
||||||
if (p[0] != '$') {
|
|
||||||
p = strchr(p,'$');
|
|
||||||
if (p == NULL) return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
*len = (int)strtol(p+1,NULL,10);
|
|
||||||
p = strchr(p,'\r');
|
|
||||||
assert(p);
|
|
||||||
*str = p+2;
|
|
||||||
return p+2+(*len)+2;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Helper function for the redisAsyncCommand* family of functions. Writes a
|
|
||||||
* formatted command to the output buffer and registers the provided callback
|
|
||||||
* function with the context. */
|
|
||||||
static int __redisAsyncCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *cmd, size_t len) {
|
|
||||||
redisContext *c = &(ac->c);
|
|
||||||
redisCallback cb;
|
|
||||||
struct dict *cbdict;
|
|
||||||
dictEntry *de;
|
|
||||||
redisCallback *existcb;
|
|
||||||
int pvariant, hasnext;
|
|
||||||
const char *cstr, *astr;
|
|
||||||
size_t clen, alen;
|
|
||||||
const char *p;
|
|
||||||
sds sname;
|
|
||||||
int ret;
|
|
||||||
|
|
||||||
/* Don't accept new commands when the connection is about to be closed. */
|
|
||||||
if (c->flags & (REDIS_DISCONNECTING | REDIS_FREEING)) return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Setup callback */
|
|
||||||
cb.fn = fn;
|
|
||||||
cb.privdata = privdata;
|
|
||||||
cb.pending_subs = 1;
|
|
||||||
|
|
||||||
/* Find out which command will be appended. */
|
|
||||||
p = nextArgument(cmd,&cstr,&clen);
|
|
||||||
assert(p != NULL);
|
|
||||||
hasnext = (p[0] == '$');
|
|
||||||
pvariant = (tolower(cstr[0]) == 'p') ? 1 : 0;
|
|
||||||
cstr += pvariant;
|
|
||||||
clen -= pvariant;
|
|
||||||
|
|
||||||
if (hasnext && strncasecmp(cstr,"subscribe\r\n",11) == 0) {
|
|
||||||
c->flags |= REDIS_SUBSCRIBED;
|
|
||||||
|
|
||||||
/* Add every channel/pattern to the list of subscription callbacks. */
|
|
||||||
while ((p = nextArgument(p,&astr,&alen)) != NULL) {
|
|
||||||
sname = sdsnewlen(astr,alen);
|
|
||||||
if (pvariant)
|
|
||||||
cbdict = ac->sub.patterns;
|
|
||||||
else
|
|
||||||
cbdict = ac->sub.channels;
|
|
||||||
|
|
||||||
de = dictFind(cbdict,sname);
|
|
||||||
|
|
||||||
if (de != NULL) {
|
|
||||||
existcb = dictGetEntryVal(de);
|
|
||||||
cb.pending_subs = existcb->pending_subs + 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = dictReplace(cbdict,sname,&cb);
|
|
||||||
|
|
||||||
if (ret == 0) sdsfree(sname);
|
|
||||||
}
|
|
||||||
} else if (strncasecmp(cstr,"unsubscribe\r\n",13) == 0) {
|
|
||||||
/* It is only useful to call (P)UNSUBSCRIBE when the context is
|
|
||||||
* subscribed to one or more channels or patterns. */
|
|
||||||
if (!(c->flags & REDIS_SUBSCRIBED)) return REDIS_ERR;
|
|
||||||
|
|
||||||
/* (P)UNSUBSCRIBE does not have its own response: every channel or
|
|
||||||
* pattern that is unsubscribed will receive a message. This means we
|
|
||||||
* should not append a callback function for this command. */
|
|
||||||
} else if(strncasecmp(cstr,"monitor\r\n",9) == 0) {
|
|
||||||
/* Set monitor flag and push callback */
|
|
||||||
c->flags |= REDIS_MONITORING;
|
|
||||||
__redisPushCallback(&ac->replies,&cb);
|
|
||||||
} else {
|
|
||||||
if (c->flags & REDIS_SUBSCRIBED)
|
|
||||||
/* This will likely result in an error reply, but it needs to be
|
|
||||||
* received and passed to the callback. */
|
|
||||||
__redisPushCallback(&ac->sub.invalid,&cb);
|
|
||||||
else
|
|
||||||
__redisPushCallback(&ac->replies,&cb);
|
|
||||||
}
|
|
||||||
|
|
||||||
__redisAppendCommand(c,cmd,len);
|
|
||||||
|
|
||||||
/* Always schedule a write when the write buffer is non-empty */
|
|
||||||
_EL_ADD_WRITE(ac);
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisvAsyncCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *format, va_list ap) {
|
|
||||||
char *cmd;
|
|
||||||
int len;
|
|
||||||
int status;
|
|
||||||
len = redisvFormatCommand(&cmd,format,ap);
|
|
||||||
|
|
||||||
/* We don't want to pass -1 or -2 to future functions as a length. */
|
|
||||||
if (len < 0)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
status = __redisAsyncCommand(ac,fn,privdata,cmd,len);
|
|
||||||
free(cmd);
|
|
||||||
return status;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisAsyncCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *format, ...) {
|
|
||||||
va_list ap;
|
|
||||||
int status;
|
|
||||||
va_start(ap,format);
|
|
||||||
status = redisvAsyncCommand(ac,fn,privdata,format,ap);
|
|
||||||
va_end(ap);
|
|
||||||
return status;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisAsyncCommandArgv(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, int argc, const char **argv, const size_t *argvlen) {
|
|
||||||
sds cmd;
|
|
||||||
int len;
|
|
||||||
int status;
|
|
||||||
len = redisFormatSdsCommandArgv(&cmd,argc,argv,argvlen);
|
|
||||||
if (len < 0)
|
|
||||||
return REDIS_ERR;
|
|
||||||
status = __redisAsyncCommand(ac,fn,privdata,cmd,len);
|
|
||||||
sdsfree(cmd);
|
|
||||||
return status;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisAsyncFormattedCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *cmd, size_t len) {
|
|
||||||
int status = __redisAsyncCommand(ac,fn,privdata,cmd,len);
|
|
||||||
return status;
|
|
||||||
}
|
|
130
controller/thirdparty/hiredis-0.14.1/async.h
vendored
130
controller/thirdparty/hiredis-0.14.1/async.h
vendored
|
@ -1,130 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_ASYNC_H
|
|
||||||
#define __HIREDIS_ASYNC_H
|
|
||||||
#include "hiredis.h"
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
struct redisAsyncContext; /* need forward declaration of redisAsyncContext */
|
|
||||||
struct dict; /* dictionary header is included in async.c */
|
|
||||||
|
|
||||||
/* Reply callback prototype and container */
|
|
||||||
typedef void (redisCallbackFn)(struct redisAsyncContext*, void*, void*);
|
|
||||||
typedef struct redisCallback {
|
|
||||||
struct redisCallback *next; /* simple singly linked list */
|
|
||||||
redisCallbackFn *fn;
|
|
||||||
int pending_subs;
|
|
||||||
void *privdata;
|
|
||||||
} redisCallback;
|
|
||||||
|
|
||||||
/* List of callbacks for either regular replies or pub/sub */
|
|
||||||
typedef struct redisCallbackList {
|
|
||||||
redisCallback *head, *tail;
|
|
||||||
} redisCallbackList;
|
|
||||||
|
|
||||||
/* Connection callback prototypes */
|
|
||||||
typedef void (redisDisconnectCallback)(const struct redisAsyncContext*, int status);
|
|
||||||
typedef void (redisConnectCallback)(const struct redisAsyncContext*, int status);
|
|
||||||
|
|
||||||
/* Context for an async connection to Redis */
|
|
||||||
typedef struct redisAsyncContext {
|
|
||||||
/* Hold the regular context, so it can be realloc'ed. */
|
|
||||||
redisContext c;
|
|
||||||
|
|
||||||
/* Setup error flags so they can be used directly. */
|
|
||||||
int err;
|
|
||||||
char *errstr;
|
|
||||||
|
|
||||||
/* Not used by hiredis */
|
|
||||||
void *data;
|
|
||||||
|
|
||||||
/* Event library data and hooks */
|
|
||||||
struct {
|
|
||||||
void *data;
|
|
||||||
|
|
||||||
/* Hooks that are called when the library expects to start
|
|
||||||
* reading/writing. These functions should be idempotent. */
|
|
||||||
void (*addRead)(void *privdata);
|
|
||||||
void (*delRead)(void *privdata);
|
|
||||||
void (*addWrite)(void *privdata);
|
|
||||||
void (*delWrite)(void *privdata);
|
|
||||||
void (*cleanup)(void *privdata);
|
|
||||||
} ev;
|
|
||||||
|
|
||||||
/* Called when either the connection is terminated due to an error or per
|
|
||||||
* user request. The status is set accordingly (REDIS_OK, REDIS_ERR). */
|
|
||||||
redisDisconnectCallback *onDisconnect;
|
|
||||||
|
|
||||||
/* Called when the first write event was received. */
|
|
||||||
redisConnectCallback *onConnect;
|
|
||||||
|
|
||||||
/* Regular command callbacks */
|
|
||||||
redisCallbackList replies;
|
|
||||||
|
|
||||||
/* Subscription callbacks */
|
|
||||||
struct {
|
|
||||||
redisCallbackList invalid;
|
|
||||||
struct dict *channels;
|
|
||||||
struct dict *patterns;
|
|
||||||
} sub;
|
|
||||||
} redisAsyncContext;
|
|
||||||
|
|
||||||
/* Functions that proxy to hiredis */
|
|
||||||
redisAsyncContext *redisAsyncConnect(const char *ip, int port);
|
|
||||||
redisAsyncContext *redisAsyncConnectBind(const char *ip, int port, const char *source_addr);
|
|
||||||
redisAsyncContext *redisAsyncConnectBindWithReuse(const char *ip, int port,
|
|
||||||
const char *source_addr);
|
|
||||||
redisAsyncContext *redisAsyncConnectUnix(const char *path);
|
|
||||||
int redisAsyncSetConnectCallback(redisAsyncContext *ac, redisConnectCallback *fn);
|
|
||||||
int redisAsyncSetDisconnectCallback(redisAsyncContext *ac, redisDisconnectCallback *fn);
|
|
||||||
void redisAsyncDisconnect(redisAsyncContext *ac);
|
|
||||||
void redisAsyncFree(redisAsyncContext *ac);
|
|
||||||
|
|
||||||
/* Handle read/write events */
|
|
||||||
void redisAsyncHandleRead(redisAsyncContext *ac);
|
|
||||||
void redisAsyncHandleWrite(redisAsyncContext *ac);
|
|
||||||
|
|
||||||
/* Command functions for an async context. Write the command to the
|
|
||||||
* output buffer and register the provided callback. */
|
|
||||||
int redisvAsyncCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *format, va_list ap);
|
|
||||||
int redisAsyncCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *format, ...);
|
|
||||||
int redisAsyncCommandArgv(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
int redisAsyncFormattedCommand(redisAsyncContext *ac, redisCallbackFn *fn, void *privdata, const char *cmd, size_t len);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
339
controller/thirdparty/hiredis-0.14.1/dict.c
vendored
339
controller/thirdparty/hiredis-0.14.1/dict.c
vendored
|
@ -1,339 +0,0 @@
|
||||||
/* Hash table implementation.
|
|
||||||
*
|
|
||||||
* This file implements in memory hash tables with insert/del/replace/find/
|
|
||||||
* get-random-element operations. Hash tables will auto resize if needed
|
|
||||||
* tables of power of two in size are used, collisions are handled by
|
|
||||||
* chaining. See the source code for more information... :)
|
|
||||||
*
|
|
||||||
* Copyright (c) 2006-2010, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* 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 "fmacros.h"
|
|
||||||
#include "alloc.h"
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <assert.h>
|
|
||||||
#include <limits.h>
|
|
||||||
#include "dict.h"
|
|
||||||
|
|
||||||
/* -------------------------- private prototypes ---------------------------- */
|
|
||||||
|
|
||||||
static int _dictExpandIfNeeded(dict *ht);
|
|
||||||
static unsigned long _dictNextPower(unsigned long size);
|
|
||||||
static int _dictKeyIndex(dict *ht, const void *key);
|
|
||||||
static int _dictInit(dict *ht, dictType *type, void *privDataPtr);
|
|
||||||
|
|
||||||
/* -------------------------- hash functions -------------------------------- */
|
|
||||||
|
|
||||||
/* Generic hash function (a popular one from Bernstein).
|
|
||||||
* I tested a few and this was the best. */
|
|
||||||
static unsigned int dictGenHashFunction(const unsigned char *buf, int len) {
|
|
||||||
unsigned int hash = 5381;
|
|
||||||
|
|
||||||
while (len--)
|
|
||||||
hash = ((hash << 5) + hash) + (*buf++); /* hash * 33 + c */
|
|
||||||
return hash;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* ----------------------------- API implementation ------------------------- */
|
|
||||||
|
|
||||||
/* Reset an hashtable already initialized with ht_init().
|
|
||||||
* NOTE: This function should only called by ht_destroy(). */
|
|
||||||
static void _dictReset(dict *ht) {
|
|
||||||
ht->table = NULL;
|
|
||||||
ht->size = 0;
|
|
||||||
ht->sizemask = 0;
|
|
||||||
ht->used = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Create a new hash table */
|
|
||||||
static dict *dictCreate(dictType *type, void *privDataPtr) {
|
|
||||||
dict *ht = hi_malloc(sizeof(*ht));
|
|
||||||
_dictInit(ht,type,privDataPtr);
|
|
||||||
return ht;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Initialize the hash table */
|
|
||||||
static int _dictInit(dict *ht, dictType *type, void *privDataPtr) {
|
|
||||||
_dictReset(ht);
|
|
||||||
ht->type = type;
|
|
||||||
ht->privdata = privDataPtr;
|
|
||||||
return DICT_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Expand or create the hashtable */
|
|
||||||
static int dictExpand(dict *ht, unsigned long size) {
|
|
||||||
dict n; /* the new hashtable */
|
|
||||||
unsigned long realsize = _dictNextPower(size), i;
|
|
||||||
|
|
||||||
/* the size is invalid if it is smaller than the number of
|
|
||||||
* elements already inside the hashtable */
|
|
||||||
if (ht->used > size)
|
|
||||||
return DICT_ERR;
|
|
||||||
|
|
||||||
_dictInit(&n, ht->type, ht->privdata);
|
|
||||||
n.size = realsize;
|
|
||||||
n.sizemask = realsize-1;
|
|
||||||
n.table = calloc(realsize,sizeof(dictEntry*));
|
|
||||||
|
|
||||||
/* Copy all the elements from the old to the new table:
|
|
||||||
* note that if the old hash table is empty ht->size is zero,
|
|
||||||
* so dictExpand just creates an hash table. */
|
|
||||||
n.used = ht->used;
|
|
||||||
for (i = 0; i < ht->size && ht->used > 0; i++) {
|
|
||||||
dictEntry *he, *nextHe;
|
|
||||||
|
|
||||||
if (ht->table[i] == NULL) continue;
|
|
||||||
|
|
||||||
/* For each hash entry on this slot... */
|
|
||||||
he = ht->table[i];
|
|
||||||
while(he) {
|
|
||||||
unsigned int h;
|
|
||||||
|
|
||||||
nextHe = he->next;
|
|
||||||
/* Get the new element index */
|
|
||||||
h = dictHashKey(ht, he->key) & n.sizemask;
|
|
||||||
he->next = n.table[h];
|
|
||||||
n.table[h] = he;
|
|
||||||
ht->used--;
|
|
||||||
/* Pass to the next element */
|
|
||||||
he = nextHe;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
assert(ht->used == 0);
|
|
||||||
free(ht->table);
|
|
||||||
|
|
||||||
/* Remap the new hashtable in the old */
|
|
||||||
*ht = n;
|
|
||||||
return DICT_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Add an element to the target hash table */
|
|
||||||
static int dictAdd(dict *ht, void *key, void *val) {
|
|
||||||
int index;
|
|
||||||
dictEntry *entry;
|
|
||||||
|
|
||||||
/* Get the index of the new element, or -1 if
|
|
||||||
* the element already exists. */
|
|
||||||
if ((index = _dictKeyIndex(ht, key)) == -1)
|
|
||||||
return DICT_ERR;
|
|
||||||
|
|
||||||
/* Allocates the memory and stores key */
|
|
||||||
entry = hi_malloc(sizeof(*entry));
|
|
||||||
entry->next = ht->table[index];
|
|
||||||
ht->table[index] = entry;
|
|
||||||
|
|
||||||
/* Set the hash entry fields. */
|
|
||||||
dictSetHashKey(ht, entry, key);
|
|
||||||
dictSetHashVal(ht, entry, val);
|
|
||||||
ht->used++;
|
|
||||||
return DICT_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Add an element, discarding the old if the key already exists.
|
|
||||||
* Return 1 if the key was added from scratch, 0 if there was already an
|
|
||||||
* element with such key and dictReplace() just performed a value update
|
|
||||||
* operation. */
|
|
||||||
static int dictReplace(dict *ht, void *key, void *val) {
|
|
||||||
dictEntry *entry, auxentry;
|
|
||||||
|
|
||||||
/* Try to add the element. If the key
|
|
||||||
* does not exists dictAdd will succeed. */
|
|
||||||
if (dictAdd(ht, key, val) == DICT_OK)
|
|
||||||
return 1;
|
|
||||||
/* It already exists, get the entry */
|
|
||||||
entry = dictFind(ht, key);
|
|
||||||
/* Free the old value and set the new one */
|
|
||||||
/* Set the new value and free the old one. Note that it is important
|
|
||||||
* to do that in this order, as the value may just be exactly the same
|
|
||||||
* as the previous one. In this context, think to reference counting,
|
|
||||||
* you want to increment (set), and then decrement (free), and not the
|
|
||||||
* reverse. */
|
|
||||||
auxentry = *entry;
|
|
||||||
dictSetHashVal(ht, entry, val);
|
|
||||||
dictFreeEntryVal(ht, &auxentry);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Search and remove an element */
|
|
||||||
static int dictDelete(dict *ht, const void *key) {
|
|
||||||
unsigned int h;
|
|
||||||
dictEntry *de, *prevde;
|
|
||||||
|
|
||||||
if (ht->size == 0)
|
|
||||||
return DICT_ERR;
|
|
||||||
h = dictHashKey(ht, key) & ht->sizemask;
|
|
||||||
de = ht->table[h];
|
|
||||||
|
|
||||||
prevde = NULL;
|
|
||||||
while(de) {
|
|
||||||
if (dictCompareHashKeys(ht,key,de->key)) {
|
|
||||||
/* Unlink the element from the list */
|
|
||||||
if (prevde)
|
|
||||||
prevde->next = de->next;
|
|
||||||
else
|
|
||||||
ht->table[h] = de->next;
|
|
||||||
|
|
||||||
dictFreeEntryKey(ht,de);
|
|
||||||
dictFreeEntryVal(ht,de);
|
|
||||||
free(de);
|
|
||||||
ht->used--;
|
|
||||||
return DICT_OK;
|
|
||||||
}
|
|
||||||
prevde = de;
|
|
||||||
de = de->next;
|
|
||||||
}
|
|
||||||
return DICT_ERR; /* not found */
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Destroy an entire hash table */
|
|
||||||
static int _dictClear(dict *ht) {
|
|
||||||
unsigned long i;
|
|
||||||
|
|
||||||
/* Free all the elements */
|
|
||||||
for (i = 0; i < ht->size && ht->used > 0; i++) {
|
|
||||||
dictEntry *he, *nextHe;
|
|
||||||
|
|
||||||
if ((he = ht->table[i]) == NULL) continue;
|
|
||||||
while(he) {
|
|
||||||
nextHe = he->next;
|
|
||||||
dictFreeEntryKey(ht, he);
|
|
||||||
dictFreeEntryVal(ht, he);
|
|
||||||
free(he);
|
|
||||||
ht->used--;
|
|
||||||
he = nextHe;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
/* Free the table and the allocated cache structure */
|
|
||||||
free(ht->table);
|
|
||||||
/* Re-initialize the table */
|
|
||||||
_dictReset(ht);
|
|
||||||
return DICT_OK; /* never fails */
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Clear & Release the hash table */
|
|
||||||
static void dictRelease(dict *ht) {
|
|
||||||
_dictClear(ht);
|
|
||||||
free(ht);
|
|
||||||
}
|
|
||||||
|
|
||||||
static dictEntry *dictFind(dict *ht, const void *key) {
|
|
||||||
dictEntry *he;
|
|
||||||
unsigned int h;
|
|
||||||
|
|
||||||
if (ht->size == 0) return NULL;
|
|
||||||
h = dictHashKey(ht, key) & ht->sizemask;
|
|
||||||
he = ht->table[h];
|
|
||||||
while(he) {
|
|
||||||
if (dictCompareHashKeys(ht, key, he->key))
|
|
||||||
return he;
|
|
||||||
he = he->next;
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static dictIterator *dictGetIterator(dict *ht) {
|
|
||||||
dictIterator *iter = hi_malloc(sizeof(*iter));
|
|
||||||
|
|
||||||
iter->ht = ht;
|
|
||||||
iter->index = -1;
|
|
||||||
iter->entry = NULL;
|
|
||||||
iter->nextEntry = NULL;
|
|
||||||
return iter;
|
|
||||||
}
|
|
||||||
|
|
||||||
static dictEntry *dictNext(dictIterator *iter) {
|
|
||||||
while (1) {
|
|
||||||
if (iter->entry == NULL) {
|
|
||||||
iter->index++;
|
|
||||||
if (iter->index >=
|
|
||||||
(signed)iter->ht->size) break;
|
|
||||||
iter->entry = iter->ht->table[iter->index];
|
|
||||||
} else {
|
|
||||||
iter->entry = iter->nextEntry;
|
|
||||||
}
|
|
||||||
if (iter->entry) {
|
|
||||||
/* We need to save the 'next' here, the iterator user
|
|
||||||
* may delete the entry we are returning. */
|
|
||||||
iter->nextEntry = iter->entry->next;
|
|
||||||
return iter->entry;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void dictReleaseIterator(dictIterator *iter) {
|
|
||||||
free(iter);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* ------------------------- private functions ------------------------------ */
|
|
||||||
|
|
||||||
/* Expand the hash table if needed */
|
|
||||||
static int _dictExpandIfNeeded(dict *ht) {
|
|
||||||
/* If the hash table is empty expand it to the initial size,
|
|
||||||
* if the table is "full" dobule its size. */
|
|
||||||
if (ht->size == 0)
|
|
||||||
return dictExpand(ht, DICT_HT_INITIAL_SIZE);
|
|
||||||
if (ht->used == ht->size)
|
|
||||||
return dictExpand(ht, ht->size*2);
|
|
||||||
return DICT_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Our hash table capability is a power of two */
|
|
||||||
static unsigned long _dictNextPower(unsigned long size) {
|
|
||||||
unsigned long i = DICT_HT_INITIAL_SIZE;
|
|
||||||
|
|
||||||
if (size >= LONG_MAX) return LONG_MAX;
|
|
||||||
while(1) {
|
|
||||||
if (i >= size)
|
|
||||||
return i;
|
|
||||||
i *= 2;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Returns the index of a free slot that can be populated with
|
|
||||||
* an hash entry for the given 'key'.
|
|
||||||
* If the key already exists, -1 is returned. */
|
|
||||||
static int _dictKeyIndex(dict *ht, const void *key) {
|
|
||||||
unsigned int h;
|
|
||||||
dictEntry *he;
|
|
||||||
|
|
||||||
/* Expand the hashtable if needed */
|
|
||||||
if (_dictExpandIfNeeded(ht) == DICT_ERR)
|
|
||||||
return -1;
|
|
||||||
/* Compute the key hash value */
|
|
||||||
h = dictHashKey(ht, key) & ht->sizemask;
|
|
||||||
/* Search if this slot does not already contain the given key */
|
|
||||||
he = ht->table[h];
|
|
||||||
while(he) {
|
|
||||||
if (dictCompareHashKeys(ht, key, he->key))
|
|
||||||
return -1;
|
|
||||||
he = he->next;
|
|
||||||
}
|
|
||||||
return h;
|
|
||||||
}
|
|
||||||
|
|
126
controller/thirdparty/hiredis-0.14.1/dict.h
vendored
126
controller/thirdparty/hiredis-0.14.1/dict.h
vendored
|
@ -1,126 +0,0 @@
|
||||||
/* Hash table implementation.
|
|
||||||
*
|
|
||||||
* This file implements in memory hash tables with insert/del/replace/find/
|
|
||||||
* get-random-element operations. Hash tables will auto resize if needed
|
|
||||||
* tables of power of two in size are used, collisions are handled by
|
|
||||||
* chaining. See the source code for more information... :)
|
|
||||||
*
|
|
||||||
* Copyright (c) 2006-2010, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __DICT_H
|
|
||||||
#define __DICT_H
|
|
||||||
|
|
||||||
#define DICT_OK 0
|
|
||||||
#define DICT_ERR 1
|
|
||||||
|
|
||||||
/* Unused arguments generate annoying warnings... */
|
|
||||||
#define DICT_NOTUSED(V) ((void) V)
|
|
||||||
|
|
||||||
typedef struct dictEntry {
|
|
||||||
void *key;
|
|
||||||
void *val;
|
|
||||||
struct dictEntry *next;
|
|
||||||
} dictEntry;
|
|
||||||
|
|
||||||
typedef struct dictType {
|
|
||||||
unsigned int (*hashFunction)(const void *key);
|
|
||||||
void *(*keyDup)(void *privdata, const void *key);
|
|
||||||
void *(*valDup)(void *privdata, const void *obj);
|
|
||||||
int (*keyCompare)(void *privdata, const void *key1, const void *key2);
|
|
||||||
void (*keyDestructor)(void *privdata, void *key);
|
|
||||||
void (*valDestructor)(void *privdata, void *obj);
|
|
||||||
} dictType;
|
|
||||||
|
|
||||||
typedef struct dict {
|
|
||||||
dictEntry **table;
|
|
||||||
dictType *type;
|
|
||||||
unsigned long size;
|
|
||||||
unsigned long sizemask;
|
|
||||||
unsigned long used;
|
|
||||||
void *privdata;
|
|
||||||
} dict;
|
|
||||||
|
|
||||||
typedef struct dictIterator {
|
|
||||||
dict *ht;
|
|
||||||
int index;
|
|
||||||
dictEntry *entry, *nextEntry;
|
|
||||||
} dictIterator;
|
|
||||||
|
|
||||||
/* This is the initial size of every hash table */
|
|
||||||
#define DICT_HT_INITIAL_SIZE 4
|
|
||||||
|
|
||||||
/* ------------------------------- Macros ------------------------------------*/
|
|
||||||
#define dictFreeEntryVal(ht, entry) \
|
|
||||||
if ((ht)->type->valDestructor) \
|
|
||||||
(ht)->type->valDestructor((ht)->privdata, (entry)->val)
|
|
||||||
|
|
||||||
#define dictSetHashVal(ht, entry, _val_) do { \
|
|
||||||
if ((ht)->type->valDup) \
|
|
||||||
entry->val = (ht)->type->valDup((ht)->privdata, _val_); \
|
|
||||||
else \
|
|
||||||
entry->val = (_val_); \
|
|
||||||
} while(0)
|
|
||||||
|
|
||||||
#define dictFreeEntryKey(ht, entry) \
|
|
||||||
if ((ht)->type->keyDestructor) \
|
|
||||||
(ht)->type->keyDestructor((ht)->privdata, (entry)->key)
|
|
||||||
|
|
||||||
#define dictSetHashKey(ht, entry, _key_) do { \
|
|
||||||
if ((ht)->type->keyDup) \
|
|
||||||
entry->key = (ht)->type->keyDup((ht)->privdata, _key_); \
|
|
||||||
else \
|
|
||||||
entry->key = (_key_); \
|
|
||||||
} while(0)
|
|
||||||
|
|
||||||
#define dictCompareHashKeys(ht, key1, key2) \
|
|
||||||
(((ht)->type->keyCompare) ? \
|
|
||||||
(ht)->type->keyCompare((ht)->privdata, key1, key2) : \
|
|
||||||
(key1) == (key2))
|
|
||||||
|
|
||||||
#define dictHashKey(ht, key) (ht)->type->hashFunction(key)
|
|
||||||
|
|
||||||
#define dictGetEntryKey(he) ((he)->key)
|
|
||||||
#define dictGetEntryVal(he) ((he)->val)
|
|
||||||
#define dictSlots(ht) ((ht)->size)
|
|
||||||
#define dictSize(ht) ((ht)->used)
|
|
||||||
|
|
||||||
/* API */
|
|
||||||
static unsigned int dictGenHashFunction(const unsigned char *buf, int len);
|
|
||||||
static dict *dictCreate(dictType *type, void *privDataPtr);
|
|
||||||
static int dictExpand(dict *ht, unsigned long size);
|
|
||||||
static int dictAdd(dict *ht, void *key, void *val);
|
|
||||||
static int dictReplace(dict *ht, void *key, void *val);
|
|
||||||
static int dictDelete(dict *ht, const void *key);
|
|
||||||
static void dictRelease(dict *ht);
|
|
||||||
static dictEntry * dictFind(dict *ht, const void *key);
|
|
||||||
static dictIterator *dictGetIterator(dict *ht);
|
|
||||||
static dictEntry *dictNext(dictIterator *iter);
|
|
||||||
static void dictReleaseIterator(dictIterator *iter);
|
|
||||||
|
|
||||||
#endif /* __DICT_H */
|
|
|
@ -1,62 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/ae.h>
|
|
||||||
|
|
||||||
/* Put event loop in the global scope, so it can be explicitly stopped */
|
|
||||||
static aeEventLoop *loop;
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
aeStop(loop);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
aeStop(loop);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
aeStop(loop);
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
loop = aeCreateEventLoop(64);
|
|
||||||
redisAeAttach(loop, c);
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
aeMain(loop);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,73 +0,0 @@
|
||||||
#include <stdlib.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/glib.h>
|
|
||||||
|
|
||||||
static GMainLoop *mainloop;
|
|
||||||
|
|
||||||
static void
|
|
||||||
connect_cb (const redisAsyncContext *ac G_GNUC_UNUSED,
|
|
||||||
int status)
|
|
||||||
{
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
g_printerr("Failed to connect: %s\n", ac->errstr);
|
|
||||||
g_main_loop_quit(mainloop);
|
|
||||||
} else {
|
|
||||||
g_printerr("Connected...\n");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
disconnect_cb (const redisAsyncContext *ac G_GNUC_UNUSED,
|
|
||||||
int status)
|
|
||||||
{
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
g_error("Failed to disconnect: %s", ac->errstr);
|
|
||||||
} else {
|
|
||||||
g_printerr("Disconnected...\n");
|
|
||||||
g_main_loop_quit(mainloop);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
command_cb(redisAsyncContext *ac,
|
|
||||||
gpointer r,
|
|
||||||
gpointer user_data G_GNUC_UNUSED)
|
|
||||||
{
|
|
||||||
redisReply *reply = r;
|
|
||||||
|
|
||||||
if (reply) {
|
|
||||||
g_print("REPLY: %s\n", reply->str);
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncDisconnect(ac);
|
|
||||||
}
|
|
||||||
|
|
||||||
gint
|
|
||||||
main (gint argc G_GNUC_UNUSED,
|
|
||||||
gchar *argv[] G_GNUC_UNUSED)
|
|
||||||
{
|
|
||||||
redisAsyncContext *ac;
|
|
||||||
GMainContext *context = NULL;
|
|
||||||
GSource *source;
|
|
||||||
|
|
||||||
ac = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (ac->err) {
|
|
||||||
g_printerr("%s\n", ac->errstr);
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
source = redis_source_new(ac);
|
|
||||||
mainloop = g_main_loop_new(context, FALSE);
|
|
||||||
g_source_attach(source, context);
|
|
||||||
|
|
||||||
redisAsyncSetConnectCallback(ac, connect_cb);
|
|
||||||
redisAsyncSetDisconnectCallback(ac, disconnect_cb);
|
|
||||||
redisAsyncCommand(ac, command_cb, NULL, "SET key 1234");
|
|
||||||
redisAsyncCommand(ac, command_cb, NULL, "GET key");
|
|
||||||
|
|
||||||
g_main_loop_run(mainloop);
|
|
||||||
|
|
||||||
return EXIT_SUCCESS;
|
|
||||||
}
|
|
|
@ -1,58 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/ivykis.h>
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
|
|
||||||
iv_init();
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisIvykisAttach(c);
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
|
|
||||||
iv_main();
|
|
||||||
|
|
||||||
iv_deinit();
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
|
@ -1,52 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/libev.h>
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisLibevAttach(EV_DEFAULT_ c);
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
ev_loop(EV_DEFAULT_ 0);
|
|
||||||
return 0;
|
|
||||||
}
|
|
|
@ -1,53 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/libevent.h>
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
struct event_base *base = event_base_new();
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisLibeventAttach(c,base);
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
event_base_dispatch(base);
|
|
||||||
return 0;
|
|
||||||
}
|
|
|
@ -1,53 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <signal.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/libuv.h>
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
uv_loop_t* loop = uv_default_loop();
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisLibuvAttach(c,loop);
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
uv_run(loop, UV_RUN_DEFAULT);
|
|
||||||
return 0;
|
|
||||||
}
|
|
|
@ -1,66 +0,0 @@
|
||||||
//
|
|
||||||
// Created by Дмитрий Бахвалов on 13.07.15.
|
|
||||||
// Copyright (c) 2015 Dmitry Bakhvalov. All rights reserved.
|
|
||||||
//
|
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
#include <async.h>
|
|
||||||
#include <adapters/macosx.h>
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *c, void *r, void *privdata) {
|
|
||||||
redisReply *reply = r;
|
|
||||||
if (reply == NULL) return;
|
|
||||||
printf("argv[%s]: %s\n", (char*)privdata, reply->str);
|
|
||||||
|
|
||||||
/* Disconnect after receiving the reply to GET */
|
|
||||||
redisAsyncDisconnect(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
void connectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
printf("Connected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
void disconnectCallback(const redisAsyncContext *c, int status) {
|
|
||||||
if (status != REDIS_OK) {
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
CFRunLoopStop(CFRunLoopGetCurrent());
|
|
||||||
printf("Disconnected...\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
|
|
||||||
CFRunLoopRef loop = CFRunLoopGetCurrent();
|
|
||||||
if( !loop ) {
|
|
||||||
printf("Error: Cannot get current run loop\n");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379);
|
|
||||||
if (c->err) {
|
|
||||||
/* Let *c leak for now... */
|
|
||||||
printf("Error: %s\n", c->errstr);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
redisMacOSAttach(c, loop);
|
|
||||||
|
|
||||||
redisAsyncSetConnectCallback(c,connectCallback);
|
|
||||||
redisAsyncSetDisconnectCallback(c,disconnectCallback);
|
|
||||||
|
|
||||||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1]));
|
|
||||||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key");
|
|
||||||
|
|
||||||
CFRunLoopRun();
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,46 +0,0 @@
|
||||||
#include <iostream>
|
|
||||||
using namespace std;
|
|
||||||
|
|
||||||
#include <QCoreApplication>
|
|
||||||
#include <QTimer>
|
|
||||||
|
|
||||||
#include "example-qt.h"
|
|
||||||
|
|
||||||
void getCallback(redisAsyncContext *, void * r, void * privdata) {
|
|
||||||
|
|
||||||
redisReply * reply = static_cast<redisReply *>(r);
|
|
||||||
ExampleQt * ex = static_cast<ExampleQt *>(privdata);
|
|
||||||
if (reply == nullptr || ex == nullptr) return;
|
|
||||||
|
|
||||||
cout << "key: " << reply->str << endl;
|
|
||||||
|
|
||||||
ex->finish();
|
|
||||||
}
|
|
||||||
|
|
||||||
void ExampleQt::run() {
|
|
||||||
|
|
||||||
m_ctx = redisAsyncConnect("localhost", 6379);
|
|
||||||
|
|
||||||
if (m_ctx->err) {
|
|
||||||
cerr << "Error: " << m_ctx->errstr << endl;
|
|
||||||
redisAsyncFree(m_ctx);
|
|
||||||
emit finished();
|
|
||||||
}
|
|
||||||
|
|
||||||
m_adapter.setContext(m_ctx);
|
|
||||||
|
|
||||||
redisAsyncCommand(m_ctx, NULL, NULL, "SET key %s", m_value);
|
|
||||||
redisAsyncCommand(m_ctx, getCallback, this, "GET key");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main (int argc, char **argv) {
|
|
||||||
|
|
||||||
QCoreApplication app(argc, argv);
|
|
||||||
|
|
||||||
ExampleQt example(argv[argc-1]);
|
|
||||||
|
|
||||||
QObject::connect(&example, SIGNAL(finished()), &app, SLOT(quit()));
|
|
||||||
QTimer::singleShot(0, &example, SLOT(run()));
|
|
||||||
|
|
||||||
return app.exec();
|
|
||||||
}
|
|
|
@ -1,32 +0,0 @@
|
||||||
#ifndef __HIREDIS_EXAMPLE_QT_H
|
|
||||||
#define __HIREDIS_EXAMPLE_QT_H
|
|
||||||
|
|
||||||
#include <adapters/qt.h>
|
|
||||||
|
|
||||||
class ExampleQt : public QObject {
|
|
||||||
|
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
public:
|
|
||||||
ExampleQt(const char * value, QObject * parent = 0)
|
|
||||||
: QObject(parent), m_value(value) {}
|
|
||||||
|
|
||||||
signals:
|
|
||||||
void finished();
|
|
||||||
|
|
||||||
public slots:
|
|
||||||
void run();
|
|
||||||
|
|
||||||
private:
|
|
||||||
void finish() { emit finished(); }
|
|
||||||
|
|
||||||
private:
|
|
||||||
const char * m_value;
|
|
||||||
redisAsyncContext * m_ctx;
|
|
||||||
RedisQtAdapter m_adapter;
|
|
||||||
|
|
||||||
friend
|
|
||||||
void getCallback(redisAsyncContext *, void *, void *);
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif /* !__HIREDIS_EXAMPLE_QT_H */
|
|
|
@ -1,78 +0,0 @@
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
|
|
||||||
#include <hiredis.h>
|
|
||||||
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
unsigned int j;
|
|
||||||
redisContext *c;
|
|
||||||
redisReply *reply;
|
|
||||||
const char *hostname = (argc > 1) ? argv[1] : "127.0.0.1";
|
|
||||||
int port = (argc > 2) ? atoi(argv[2]) : 6379;
|
|
||||||
|
|
||||||
struct timeval timeout = { 1, 500000 }; // 1.5 seconds
|
|
||||||
c = redisConnectWithTimeout(hostname, port, timeout);
|
|
||||||
if (c == NULL || c->err) {
|
|
||||||
if (c) {
|
|
||||||
printf("Connection error: %s\n", c->errstr);
|
|
||||||
redisFree(c);
|
|
||||||
} else {
|
|
||||||
printf("Connection error: can't allocate redis context\n");
|
|
||||||
}
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* PING server */
|
|
||||||
reply = redisCommand(c,"PING");
|
|
||||||
printf("PING: %s\n", reply->str);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Set a key */
|
|
||||||
reply = redisCommand(c,"SET %s %s", "foo", "hello world");
|
|
||||||
printf("SET: %s\n", reply->str);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Set a key using binary safe API */
|
|
||||||
reply = redisCommand(c,"SET %b %b", "bar", (size_t) 3, "hello", (size_t) 5);
|
|
||||||
printf("SET (binary API): %s\n", reply->str);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Try a GET and two INCR */
|
|
||||||
reply = redisCommand(c,"GET foo");
|
|
||||||
printf("GET foo: %s\n", reply->str);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
reply = redisCommand(c,"INCR counter");
|
|
||||||
printf("INCR counter: %lld\n", reply->integer);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
/* again ... */
|
|
||||||
reply = redisCommand(c,"INCR counter");
|
|
||||||
printf("INCR counter: %lld\n", reply->integer);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Create a list of numbers, from 0 to 9 */
|
|
||||||
reply = redisCommand(c,"DEL mylist");
|
|
||||||
freeReplyObject(reply);
|
|
||||||
for (j = 0; j < 10; j++) {
|
|
||||||
char buf[64];
|
|
||||||
|
|
||||||
snprintf(buf,64,"%u",j);
|
|
||||||
reply = redisCommand(c,"LPUSH mylist element-%s", buf);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Let's check what we have inside the list */
|
|
||||||
reply = redisCommand(c,"LRANGE mylist 0 -1");
|
|
||||||
if (reply->type == REDIS_REPLY_ARRAY) {
|
|
||||||
for (j = 0; j < reply->elements; j++) {
|
|
||||||
printf("%u) %s\n", j, reply->element[j]->str);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Disconnects and frees the context */
|
|
||||||
redisFree(c);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
12
controller/thirdparty/hiredis-0.14.1/fmacros.h
vendored
12
controller/thirdparty/hiredis-0.14.1/fmacros.h
vendored
|
@ -1,12 +0,0 @@
|
||||||
#ifndef __HIREDIS_FMACRO_H
|
|
||||||
#define __HIREDIS_FMACRO_H
|
|
||||||
|
|
||||||
#define _XOPEN_SOURCE 600
|
|
||||||
#define _POSIX_C_SOURCE 200112L
|
|
||||||
|
|
||||||
#if defined(__APPLE__) && defined(__MACH__)
|
|
||||||
/* Enable TCP_KEEPALIVE */
|
|
||||||
#define _DARWIN_C_SOURCE
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
1006
controller/thirdparty/hiredis-0.14.1/hiredis.c
vendored
1006
controller/thirdparty/hiredis-0.14.1/hiredis.c
vendored
File diff suppressed because it is too large
Load diff
200
controller/thirdparty/hiredis-0.14.1/hiredis.h
vendored
200
controller/thirdparty/hiredis-0.14.1/hiredis.h
vendored
|
@ -1,200 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2014, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
* Copyright (c) 2015, Matt Stancliff <matt at genges dot com>,
|
|
||||||
* Jan-Erik Rediger <janerik at fnordig dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_H
|
|
||||||
#define __HIREDIS_H
|
|
||||||
#include "read.h"
|
|
||||||
#include <stdarg.h> /* for va_list */
|
|
||||||
#include <sys/time.h> /* for struct timeval */
|
|
||||||
#include <stdint.h> /* uintXX_t, etc */
|
|
||||||
#include "sds.h" /* for sds */
|
|
||||||
#include "alloc.h" /* for allocation wrappers */
|
|
||||||
|
|
||||||
#define HIREDIS_MAJOR 0
|
|
||||||
#define HIREDIS_MINOR 14
|
|
||||||
#define HIREDIS_PATCH 1
|
|
||||||
#define HIREDIS_SONAME 0.14
|
|
||||||
|
|
||||||
/* Connection type can be blocking or non-blocking and is set in the
|
|
||||||
* least significant bit of the flags field in redisContext. */
|
|
||||||
#define REDIS_BLOCK 0x1
|
|
||||||
|
|
||||||
/* Connection may be disconnected before being free'd. The second bit
|
|
||||||
* in the flags field is set when the context is connected. */
|
|
||||||
#define REDIS_CONNECTED 0x2
|
|
||||||
|
|
||||||
/* The async API might try to disconnect cleanly and flush the output
|
|
||||||
* buffer and read all subsequent replies before disconnecting.
|
|
||||||
* This flag means no new commands can come in and the connection
|
|
||||||
* should be terminated once all replies have been read. */
|
|
||||||
#define REDIS_DISCONNECTING 0x4
|
|
||||||
|
|
||||||
/* Flag specific to the async API which means that the context should be clean
|
|
||||||
* up as soon as possible. */
|
|
||||||
#define REDIS_FREEING 0x8
|
|
||||||
|
|
||||||
/* Flag that is set when an async callback is executed. */
|
|
||||||
#define REDIS_IN_CALLBACK 0x10
|
|
||||||
|
|
||||||
/* Flag that is set when the async context has one or more subscriptions. */
|
|
||||||
#define REDIS_SUBSCRIBED 0x20
|
|
||||||
|
|
||||||
/* Flag that is set when monitor mode is active */
|
|
||||||
#define REDIS_MONITORING 0x40
|
|
||||||
|
|
||||||
/* Flag that is set when we should set SO_REUSEADDR before calling bind() */
|
|
||||||
#define REDIS_REUSEADDR 0x80
|
|
||||||
|
|
||||||
#define REDIS_KEEPALIVE_INTERVAL 15 /* seconds */
|
|
||||||
|
|
||||||
/* number of times we retry to connect in the case of EADDRNOTAVAIL and
|
|
||||||
* SO_REUSEADDR is being used. */
|
|
||||||
#define REDIS_CONNECT_RETRIES 10
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/* This is the reply object returned by redisCommand() */
|
|
||||||
typedef struct redisReply {
|
|
||||||
int type; /* REDIS_REPLY_* */
|
|
||||||
long long integer; /* The integer when type is REDIS_REPLY_INTEGER */
|
|
||||||
size_t len; /* Length of string */
|
|
||||||
char *str; /* Used for both REDIS_REPLY_ERROR and REDIS_REPLY_STRING */
|
|
||||||
size_t elements; /* number of elements, for REDIS_REPLY_ARRAY */
|
|
||||||
struct redisReply **element; /* elements vector for REDIS_REPLY_ARRAY */
|
|
||||||
} redisReply;
|
|
||||||
|
|
||||||
redisReader *redisReaderCreate(void);
|
|
||||||
|
|
||||||
/* Function to free the reply objects hiredis returns by default. */
|
|
||||||
void freeReplyObject(void *reply);
|
|
||||||
|
|
||||||
/* Functions to format a command according to the protocol. */
|
|
||||||
int redisvFormatCommand(char **target, const char *format, va_list ap);
|
|
||||||
int redisFormatCommand(char **target, const char *format, ...);
|
|
||||||
int redisFormatCommandArgv(char **target, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
int redisFormatSdsCommandArgv(sds *target, int argc, const char ** argv, const size_t *argvlen);
|
|
||||||
void redisFreeCommand(char *cmd);
|
|
||||||
void redisFreeSdsCommand(sds cmd);
|
|
||||||
|
|
||||||
enum redisConnectionType {
|
|
||||||
REDIS_CONN_TCP,
|
|
||||||
REDIS_CONN_UNIX
|
|
||||||
};
|
|
||||||
|
|
||||||
/* Context for a connection to Redis */
|
|
||||||
typedef struct redisContext {
|
|
||||||
int err; /* Error flags, 0 when there is no error */
|
|
||||||
char errstr[128]; /* String representation of error when applicable */
|
|
||||||
int fd;
|
|
||||||
int flags;
|
|
||||||
char *obuf; /* Write buffer */
|
|
||||||
redisReader *reader; /* Protocol reader */
|
|
||||||
|
|
||||||
enum redisConnectionType connection_type;
|
|
||||||
struct timeval *timeout;
|
|
||||||
|
|
||||||
struct {
|
|
||||||
char *host;
|
|
||||||
char *source_addr;
|
|
||||||
int port;
|
|
||||||
} tcp;
|
|
||||||
|
|
||||||
struct {
|
|
||||||
char *path;
|
|
||||||
} unix_sock;
|
|
||||||
|
|
||||||
} redisContext;
|
|
||||||
|
|
||||||
redisContext *redisConnect(const char *ip, int port);
|
|
||||||
redisContext *redisConnectWithTimeout(const char *ip, int port, const struct timeval tv);
|
|
||||||
redisContext *redisConnectNonBlock(const char *ip, int port);
|
|
||||||
redisContext *redisConnectBindNonBlock(const char *ip, int port,
|
|
||||||
const char *source_addr);
|
|
||||||
redisContext *redisConnectBindNonBlockWithReuse(const char *ip, int port,
|
|
||||||
const char *source_addr);
|
|
||||||
redisContext *redisConnectUnix(const char *path);
|
|
||||||
redisContext *redisConnectUnixWithTimeout(const char *path, const struct timeval tv);
|
|
||||||
redisContext *redisConnectUnixNonBlock(const char *path);
|
|
||||||
redisContext *redisConnectFd(int fd);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Reconnect the given context using the saved information.
|
|
||||||
*
|
|
||||||
* This re-uses the exact same connect options as in the initial connection.
|
|
||||||
* host, ip (or path), timeout and bind address are reused,
|
|
||||||
* flags are used unmodified from the existing context.
|
|
||||||
*
|
|
||||||
* Returns REDIS_OK on successful connect or REDIS_ERR otherwise.
|
|
||||||
*/
|
|
||||||
int redisReconnect(redisContext *c);
|
|
||||||
|
|
||||||
int redisSetTimeout(redisContext *c, const struct timeval tv);
|
|
||||||
int redisEnableKeepAlive(redisContext *c);
|
|
||||||
void redisFree(redisContext *c);
|
|
||||||
int redisFreeKeepFd(redisContext *c);
|
|
||||||
int redisBufferRead(redisContext *c);
|
|
||||||
int redisBufferWrite(redisContext *c, int *done);
|
|
||||||
|
|
||||||
/* In a blocking context, this function first checks if there are unconsumed
|
|
||||||
* replies to return and returns one if so. Otherwise, it flushes the output
|
|
||||||
* buffer to the socket and reads until it has a reply. In a non-blocking
|
|
||||||
* context, it will return unconsumed replies until there are no more. */
|
|
||||||
int redisGetReply(redisContext *c, void **reply);
|
|
||||||
int redisGetReplyFromReader(redisContext *c, void **reply);
|
|
||||||
|
|
||||||
/* Write a formatted command to the output buffer. Use these functions in blocking mode
|
|
||||||
* to get a pipeline of commands. */
|
|
||||||
int redisAppendFormattedCommand(redisContext *c, const char *cmd, size_t len);
|
|
||||||
|
|
||||||
/* Write a command to the output buffer. Use these functions in blocking mode
|
|
||||||
* to get a pipeline of commands. */
|
|
||||||
int redisvAppendCommand(redisContext *c, const char *format, va_list ap);
|
|
||||||
int redisAppendCommand(redisContext *c, const char *format, ...);
|
|
||||||
int redisAppendCommandArgv(redisContext *c, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
|
|
||||||
/* Issue a command to Redis. In a blocking context, it is identical to calling
|
|
||||||
* redisAppendCommand, followed by redisGetReply. The function will return
|
|
||||||
* NULL if there was an error in performing the request, otherwise it will
|
|
||||||
* return the reply. In a non-blocking context, it is identical to calling
|
|
||||||
* only redisAppendCommand and will always return NULL. */
|
|
||||||
void *redisvCommand(redisContext *c, const char *format, va_list ap);
|
|
||||||
void *redisCommand(redisContext *c, const char *format, ...);
|
|
||||||
void *redisCommandArgv(redisContext *c, int argc, const char **argv, const size_t *argvlen);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
477
controller/thirdparty/hiredis-0.14.1/net.c
vendored
477
controller/thirdparty/hiredis-0.14.1/net.c
vendored
|
@ -1,477 +0,0 @@
|
||||||
/* Extracted from anet.c to work properly with Hiredis error reporting.
|
|
||||||
*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2014, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
* Copyright (c) 2015, Matt Stancliff <matt at genges dot com>,
|
|
||||||
* Jan-Erik Rediger <janerik at fnordig dot com>
|
|
||||||
*
|
|
||||||
* 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 "fmacros.h"
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <sys/select.h>
|
|
||||||
#include <sys/un.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <netinet/tcp.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <fcntl.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <netdb.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <stdarg.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <poll.h>
|
|
||||||
#include <limits.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
|
|
||||||
#include "net.h"
|
|
||||||
#include "sds.h"
|
|
||||||
|
|
||||||
/* Defined in hiredis.c */
|
|
||||||
void __redisSetError(redisContext *c, int type, const char *str);
|
|
||||||
|
|
||||||
static void redisContextCloseFd(redisContext *c) {
|
|
||||||
if (c && c->fd >= 0) {
|
|
||||||
close(c->fd);
|
|
||||||
c->fd = -1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void __redisSetErrorFromErrno(redisContext *c, int type, const char *prefix) {
|
|
||||||
int errorno = errno; /* snprintf() may change errno */
|
|
||||||
char buf[128] = { 0 };
|
|
||||||
size_t len = 0;
|
|
||||||
|
|
||||||
if (prefix != NULL)
|
|
||||||
len = snprintf(buf,sizeof(buf),"%s: ",prefix);
|
|
||||||
strerror_r(errorno, (char *)(buf + len), sizeof(buf) - len);
|
|
||||||
__redisSetError(c,type,buf);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisSetReuseAddr(redisContext *c) {
|
|
||||||
int on = 1;
|
|
||||||
if (setsockopt(c->fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,NULL);
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisCreateSocket(redisContext *c, int type) {
|
|
||||||
int s;
|
|
||||||
if ((s = socket(type, SOCK_STREAM, 0)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,NULL);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
c->fd = s;
|
|
||||||
if (type == AF_INET) {
|
|
||||||
if (redisSetReuseAddr(c) == REDIS_ERR) {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisSetBlocking(redisContext *c, int blocking) {
|
|
||||||
int flags;
|
|
||||||
|
|
||||||
/* Set the socket nonblocking.
|
|
||||||
* Note that fcntl(2) for F_GETFL and F_SETFL can't be
|
|
||||||
* interrupted by a signal. */
|
|
||||||
if ((flags = fcntl(c->fd, F_GETFL)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"fcntl(F_GETFL)");
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (blocking)
|
|
||||||
flags &= ~O_NONBLOCK;
|
|
||||||
else
|
|
||||||
flags |= O_NONBLOCK;
|
|
||||||
|
|
||||||
if (fcntl(c->fd, F_SETFL, flags) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"fcntl(F_SETFL)");
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisKeepAlive(redisContext *c, int interval) {
|
|
||||||
int val = 1;
|
|
||||||
int fd = c->fd;
|
|
||||||
|
|
||||||
if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &val, sizeof(val)) == -1){
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,strerror(errno));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
val = interval;
|
|
||||||
|
|
||||||
#if defined(__APPLE__) && defined(__MACH__)
|
|
||||||
if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPALIVE, &val, sizeof(val)) < 0) {
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,strerror(errno));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
#if defined(__GLIBC__) && !defined(__FreeBSD_kernel__)
|
|
||||||
if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &val, sizeof(val)) < 0) {
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,strerror(errno));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
val = interval/3;
|
|
||||||
if (val == 0) val = 1;
|
|
||||||
if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, &val, sizeof(val)) < 0) {
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,strerror(errno));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
val = 3;
|
|
||||||
if (setsockopt(fd, IPPROTO_TCP, TCP_KEEPCNT, &val, sizeof(val)) < 0) {
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,strerror(errno));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisSetTcpNoDelay(redisContext *c) {
|
|
||||||
int yes = 1;
|
|
||||||
if (setsockopt(c->fd, IPPROTO_TCP, TCP_NODELAY, &yes, sizeof(yes)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"setsockopt(TCP_NODELAY)");
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
#define __MAX_MSEC (((LONG_MAX) - 999) / 1000)
|
|
||||||
|
|
||||||
static int redisContextTimeoutMsec(redisContext *c, long *result)
|
|
||||||
{
|
|
||||||
const struct timeval *timeout = c->timeout;
|
|
||||||
long msec = -1;
|
|
||||||
|
|
||||||
/* Only use timeout when not NULL. */
|
|
||||||
if (timeout != NULL) {
|
|
||||||
if (timeout->tv_usec > 1000000 || timeout->tv_sec > __MAX_MSEC) {
|
|
||||||
*result = msec;
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
msec = (timeout->tv_sec * 1000) + ((timeout->tv_usec + 999) / 1000);
|
|
||||||
|
|
||||||
if (msec < 0 || msec > INT_MAX) {
|
|
||||||
msec = INT_MAX;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
*result = msec;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int redisContextWaitReady(redisContext *c, long msec) {
|
|
||||||
struct pollfd wfd[1];
|
|
||||||
|
|
||||||
wfd[0].fd = c->fd;
|
|
||||||
wfd[0].events = POLLOUT;
|
|
||||||
|
|
||||||
if (errno == EINPROGRESS) {
|
|
||||||
int res;
|
|
||||||
|
|
||||||
if ((res = poll(wfd, 1, msec)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c, REDIS_ERR_IO, "poll(2)");
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
} else if (res == 0) {
|
|
||||||
errno = ETIMEDOUT;
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,NULL);
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redisCheckSocketError(c) != REDIS_OK)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,NULL);
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisCheckSocketError(redisContext *c) {
|
|
||||||
int err = 0;
|
|
||||||
socklen_t errlen = sizeof(err);
|
|
||||||
|
|
||||||
if (getsockopt(c->fd, SOL_SOCKET, SO_ERROR, &err, &errlen) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"getsockopt(SO_ERROR)");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (err) {
|
|
||||||
errno = err;
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,NULL);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisContextSetTimeout(redisContext *c, const struct timeval tv) {
|
|
||||||
if (setsockopt(c->fd,SOL_SOCKET,SO_RCVTIMEO,&tv,sizeof(tv)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"setsockopt(SO_RCVTIMEO)");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
if (setsockopt(c->fd,SOL_SOCKET,SO_SNDTIMEO,&tv,sizeof(tv)) == -1) {
|
|
||||||
__redisSetErrorFromErrno(c,REDIS_ERR_IO,"setsockopt(SO_SNDTIMEO)");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int _redisContextConnectTcp(redisContext *c, const char *addr, int port,
|
|
||||||
const struct timeval *timeout,
|
|
||||||
const char *source_addr) {
|
|
||||||
int s, rv, n;
|
|
||||||
char _port[6]; /* strlen("65535"); */
|
|
||||||
struct addrinfo hints, *servinfo, *bservinfo, *p, *b;
|
|
||||||
int blocking = (c->flags & REDIS_BLOCK);
|
|
||||||
int reuseaddr = (c->flags & REDIS_REUSEADDR);
|
|
||||||
int reuses = 0;
|
|
||||||
long timeout_msec = -1;
|
|
||||||
|
|
||||||
servinfo = NULL;
|
|
||||||
c->connection_type = REDIS_CONN_TCP;
|
|
||||||
c->tcp.port = port;
|
|
||||||
|
|
||||||
/* We need to take possession of the passed parameters
|
|
||||||
* to make them reusable for a reconnect.
|
|
||||||
* We also carefully check we don't free data we already own,
|
|
||||||
* as in the case of the reconnect method.
|
|
||||||
*
|
|
||||||
* This is a bit ugly, but atleast it works and doesn't leak memory.
|
|
||||||
**/
|
|
||||||
if (c->tcp.host != addr) {
|
|
||||||
free(c->tcp.host);
|
|
||||||
|
|
||||||
c->tcp.host = hi_strdup(addr);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (timeout) {
|
|
||||||
if (c->timeout != timeout) {
|
|
||||||
if (c->timeout == NULL)
|
|
||||||
c->timeout = hi_malloc(sizeof(struct timeval));
|
|
||||||
|
|
||||||
memcpy(c->timeout, timeout, sizeof(struct timeval));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
free(c->timeout);
|
|
||||||
c->timeout = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redisContextTimeoutMsec(c, &timeout_msec) != REDIS_OK) {
|
|
||||||
__redisSetError(c, REDIS_ERR_IO, "Invalid timeout specified");
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (source_addr == NULL) {
|
|
||||||
free(c->tcp.source_addr);
|
|
||||||
c->tcp.source_addr = NULL;
|
|
||||||
} else if (c->tcp.source_addr != source_addr) {
|
|
||||||
free(c->tcp.source_addr);
|
|
||||||
c->tcp.source_addr = hi_strdup(source_addr);
|
|
||||||
}
|
|
||||||
|
|
||||||
snprintf(_port, 6, "%d", port);
|
|
||||||
memset(&hints,0,sizeof(hints));
|
|
||||||
hints.ai_family = AF_INET;
|
|
||||||
hints.ai_socktype = SOCK_STREAM;
|
|
||||||
|
|
||||||
/* Try with IPv6 if no IPv4 address was found. We do it in this order since
|
|
||||||
* in a Redis client you can't afford to test if you have IPv6 connectivity
|
|
||||||
* as this would add latency to every connect. Otherwise a more sensible
|
|
||||||
* route could be: Use IPv6 if both addresses are available and there is IPv6
|
|
||||||
* connectivity. */
|
|
||||||
if ((rv = getaddrinfo(c->tcp.host,_port,&hints,&servinfo)) != 0) {
|
|
||||||
hints.ai_family = AF_INET6;
|
|
||||||
if ((rv = getaddrinfo(addr,_port,&hints,&servinfo)) != 0) {
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,gai_strerror(rv));
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (p = servinfo; p != NULL; p = p->ai_next) {
|
|
||||||
addrretry:
|
|
||||||
if ((s = socket(p->ai_family,p->ai_socktype,p->ai_protocol)) == -1)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
c->fd = s;
|
|
||||||
if (redisSetBlocking(c,0) != REDIS_OK)
|
|
||||||
goto error;
|
|
||||||
if (c->tcp.source_addr) {
|
|
||||||
int bound = 0;
|
|
||||||
/* Using getaddrinfo saves us from self-determining IPv4 vs IPv6 */
|
|
||||||
if ((rv = getaddrinfo(c->tcp.source_addr, NULL, &hints, &bservinfo)) != 0) {
|
|
||||||
char buf[128];
|
|
||||||
snprintf(buf,sizeof(buf),"Can't get addr: %s",gai_strerror(rv));
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,buf);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (reuseaddr) {
|
|
||||||
n = 1;
|
|
||||||
if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*) &n,
|
|
||||||
sizeof(n)) < 0) {
|
|
||||||
freeaddrinfo(bservinfo);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (b = bservinfo; b != NULL; b = b->ai_next) {
|
|
||||||
if (bind(s,b->ai_addr,b->ai_addrlen) != -1) {
|
|
||||||
bound = 1;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
freeaddrinfo(bservinfo);
|
|
||||||
if (!bound) {
|
|
||||||
char buf[128];
|
|
||||||
snprintf(buf,sizeof(buf),"Can't bind socket: %s",strerror(errno));
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,buf);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (connect(s,p->ai_addr,p->ai_addrlen) == -1) {
|
|
||||||
if (errno == EHOSTUNREACH) {
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
continue;
|
|
||||||
} else if (errno == EINPROGRESS && !blocking) {
|
|
||||||
/* This is ok. */
|
|
||||||
} else if (errno == EADDRNOTAVAIL && reuseaddr) {
|
|
||||||
if (++reuses >= REDIS_CONNECT_RETRIES) {
|
|
||||||
goto error;
|
|
||||||
} else {
|
|
||||||
redisContextCloseFd(c);
|
|
||||||
goto addrretry;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (redisContextWaitReady(c,timeout_msec) != REDIS_OK)
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (blocking && redisSetBlocking(c,1) != REDIS_OK)
|
|
||||||
goto error;
|
|
||||||
if (redisSetTcpNoDelay(c) != REDIS_OK)
|
|
||||||
goto error;
|
|
||||||
|
|
||||||
c->flags |= REDIS_CONNECTED;
|
|
||||||
rv = REDIS_OK;
|
|
||||||
goto end;
|
|
||||||
}
|
|
||||||
if (p == NULL) {
|
|
||||||
char buf[128];
|
|
||||||
snprintf(buf,sizeof(buf),"Can't create socket: %s",strerror(errno));
|
|
||||||
__redisSetError(c,REDIS_ERR_OTHER,buf);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
error:
|
|
||||||
rv = REDIS_ERR;
|
|
||||||
end:
|
|
||||||
if(servinfo) {
|
|
||||||
freeaddrinfo(servinfo);
|
|
||||||
}
|
|
||||||
|
|
||||||
return rv; // Need to return REDIS_OK if alright
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisContextConnectTcp(redisContext *c, const char *addr, int port,
|
|
||||||
const struct timeval *timeout) {
|
|
||||||
return _redisContextConnectTcp(c, addr, port, timeout, NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisContextConnectBindTcp(redisContext *c, const char *addr, int port,
|
|
||||||
const struct timeval *timeout,
|
|
||||||
const char *source_addr) {
|
|
||||||
return _redisContextConnectTcp(c, addr, port, timeout, source_addr);
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisContextConnectUnix(redisContext *c, const char *path, const struct timeval *timeout) {
|
|
||||||
int blocking = (c->flags & REDIS_BLOCK);
|
|
||||||
struct sockaddr_un sa;
|
|
||||||
long timeout_msec = -1;
|
|
||||||
|
|
||||||
if (redisCreateSocket(c,AF_UNIX) < 0)
|
|
||||||
return REDIS_ERR;
|
|
||||||
if (redisSetBlocking(c,0) != REDIS_OK)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
c->connection_type = REDIS_CONN_UNIX;
|
|
||||||
if (c->unix_sock.path != path)
|
|
||||||
c->unix_sock.path = hi_strdup(path);
|
|
||||||
|
|
||||||
if (timeout) {
|
|
||||||
if (c->timeout != timeout) {
|
|
||||||
if (c->timeout == NULL)
|
|
||||||
c->timeout = hi_malloc(sizeof(struct timeval));
|
|
||||||
|
|
||||||
memcpy(c->timeout, timeout, sizeof(struct timeval));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
free(c->timeout);
|
|
||||||
c->timeout = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redisContextTimeoutMsec(c,&timeout_msec) != REDIS_OK)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
sa.sun_family = AF_UNIX;
|
|
||||||
strncpy(sa.sun_path,path,sizeof(sa.sun_path)-1);
|
|
||||||
if (connect(c->fd, (struct sockaddr*)&sa, sizeof(sa)) == -1) {
|
|
||||||
if (errno == EINPROGRESS && !blocking) {
|
|
||||||
/* This is ok. */
|
|
||||||
} else {
|
|
||||||
if (redisContextWaitReady(c,timeout_msec) != REDIS_OK)
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Reset socket to be blocking after connect(2). */
|
|
||||||
if (blocking && redisSetBlocking(c,1) != REDIS_OK)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
c->flags |= REDIS_CONNECTED;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
49
controller/thirdparty/hiredis-0.14.1/net.h
vendored
49
controller/thirdparty/hiredis-0.14.1/net.h
vendored
|
@ -1,49 +0,0 @@
|
||||||
/* Extracted from anet.c to work properly with Hiredis error reporting.
|
|
||||||
*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2014, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
* Copyright (c) 2015, Matt Stancliff <matt at genges dot com>,
|
|
||||||
* Jan-Erik Rediger <janerik at fnordig dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __NET_H
|
|
||||||
#define __NET_H
|
|
||||||
|
|
||||||
#include "hiredis.h"
|
|
||||||
|
|
||||||
int redisCheckSocketError(redisContext *c);
|
|
||||||
int redisContextSetTimeout(redisContext *c, const struct timeval tv);
|
|
||||||
int redisContextConnectTcp(redisContext *c, const char *addr, int port, const struct timeval *timeout);
|
|
||||||
int redisContextConnectBindTcp(redisContext *c, const char *addr, int port,
|
|
||||||
const struct timeval *timeout,
|
|
||||||
const char *source_addr);
|
|
||||||
int redisContextConnectUnix(redisContext *c, const char *path, const struct timeval *timeout);
|
|
||||||
int redisKeepAlive(redisContext *c, int interval);
|
|
||||||
|
|
||||||
#endif
|
|
598
controller/thirdparty/hiredis-0.14.1/read.c
vendored
598
controller/thirdparty/hiredis-0.14.1/read.c
vendored
|
@ -1,598 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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 "fmacros.h"
|
|
||||||
#include <string.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#ifndef _MSC_VER
|
|
||||||
#include <unistd.h>
|
|
||||||
#endif
|
|
||||||
#include <assert.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <ctype.h>
|
|
||||||
#include <limits.h>
|
|
||||||
|
|
||||||
#include "read.h"
|
|
||||||
#include "sds.h"
|
|
||||||
|
|
||||||
static void __redisReaderSetError(redisReader *r, int type, const char *str) {
|
|
||||||
size_t len;
|
|
||||||
|
|
||||||
if (r->reply != NULL && r->fn && r->fn->freeObject) {
|
|
||||||
r->fn->freeObject(r->reply);
|
|
||||||
r->reply = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Clear input buffer on errors. */
|
|
||||||
sdsfree(r->buf);
|
|
||||||
r->buf = NULL;
|
|
||||||
r->pos = r->len = 0;
|
|
||||||
|
|
||||||
/* Reset task stack. */
|
|
||||||
r->ridx = -1;
|
|
||||||
|
|
||||||
/* Set error. */
|
|
||||||
r->err = type;
|
|
||||||
len = strlen(str);
|
|
||||||
len = len < (sizeof(r->errstr)-1) ? len : (sizeof(r->errstr)-1);
|
|
||||||
memcpy(r->errstr,str,len);
|
|
||||||
r->errstr[len] = '\0';
|
|
||||||
}
|
|
||||||
|
|
||||||
static size_t chrtos(char *buf, size_t size, char byte) {
|
|
||||||
size_t len = 0;
|
|
||||||
|
|
||||||
switch(byte) {
|
|
||||||
case '\\':
|
|
||||||
case '"':
|
|
||||||
len = snprintf(buf,size,"\"\\%c\"",byte);
|
|
||||||
break;
|
|
||||||
case '\n': len = snprintf(buf,size,"\"\\n\""); break;
|
|
||||||
case '\r': len = snprintf(buf,size,"\"\\r\""); break;
|
|
||||||
case '\t': len = snprintf(buf,size,"\"\\t\""); break;
|
|
||||||
case '\a': len = snprintf(buf,size,"\"\\a\""); break;
|
|
||||||
case '\b': len = snprintf(buf,size,"\"\\b\""); break;
|
|
||||||
default:
|
|
||||||
if (isprint(byte))
|
|
||||||
len = snprintf(buf,size,"\"%c\"",byte);
|
|
||||||
else
|
|
||||||
len = snprintf(buf,size,"\"\\x%02x\"",(unsigned char)byte);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return len;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void __redisReaderSetErrorProtocolByte(redisReader *r, char byte) {
|
|
||||||
char cbuf[8], sbuf[128];
|
|
||||||
|
|
||||||
chrtos(cbuf,sizeof(cbuf),byte);
|
|
||||||
snprintf(sbuf,sizeof(sbuf),
|
|
||||||
"Protocol error, got %s as reply type byte", cbuf);
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,sbuf);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void __redisReaderSetErrorOOM(redisReader *r) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_OOM,"Out of memory");
|
|
||||||
}
|
|
||||||
|
|
||||||
static char *readBytes(redisReader *r, unsigned int bytes) {
|
|
||||||
char *p;
|
|
||||||
if (r->len-r->pos >= bytes) {
|
|
||||||
p = r->buf+r->pos;
|
|
||||||
r->pos += bytes;
|
|
||||||
return p;
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Find pointer to \r\n. */
|
|
||||||
static char *seekNewline(char *s, size_t len) {
|
|
||||||
int pos = 0;
|
|
||||||
int _len = len-1;
|
|
||||||
|
|
||||||
/* Position should be < len-1 because the character at "pos" should be
|
|
||||||
* followed by a \n. Note that strchr cannot be used because it doesn't
|
|
||||||
* allow to search a limited length and the buffer that is being searched
|
|
||||||
* might not have a trailing NULL character. */
|
|
||||||
while (pos < _len) {
|
|
||||||
while(pos < _len && s[pos] != '\r') pos++;
|
|
||||||
if (pos==_len) {
|
|
||||||
/* Not found. */
|
|
||||||
return NULL;
|
|
||||||
} else {
|
|
||||||
if (s[pos+1] == '\n') {
|
|
||||||
/* Found. */
|
|
||||||
return s+pos;
|
|
||||||
} else {
|
|
||||||
/* Continue searching. */
|
|
||||||
pos++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Convert a string into a long long. Returns REDIS_OK if the string could be
|
|
||||||
* parsed into a (non-overflowing) long long, REDIS_ERR otherwise. The value
|
|
||||||
* will be set to the parsed value when appropriate.
|
|
||||||
*
|
|
||||||
* Note that this function demands that the string strictly represents
|
|
||||||
* a long long: no spaces or other characters before or after the string
|
|
||||||
* representing the number are accepted, nor zeroes at the start if not
|
|
||||||
* for the string "0" representing the zero number.
|
|
||||||
*
|
|
||||||
* Because of its strictness, it is safe to use this function to check if
|
|
||||||
* you can convert a string into a long long, and obtain back the string
|
|
||||||
* from the number without any loss in the string representation. */
|
|
||||||
static int string2ll(const char *s, size_t slen, long long *value) {
|
|
||||||
const char *p = s;
|
|
||||||
size_t plen = 0;
|
|
||||||
int negative = 0;
|
|
||||||
unsigned long long v;
|
|
||||||
|
|
||||||
if (plen == slen)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Special case: first and only digit is 0. */
|
|
||||||
if (slen == 1 && p[0] == '0') {
|
|
||||||
if (value != NULL) *value = 0;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (p[0] == '-') {
|
|
||||||
negative = 1;
|
|
||||||
p++; plen++;
|
|
||||||
|
|
||||||
/* Abort on only a negative sign. */
|
|
||||||
if (plen == slen)
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* First digit should be 1-9, otherwise the string should just be 0. */
|
|
||||||
if (p[0] >= '1' && p[0] <= '9') {
|
|
||||||
v = p[0]-'0';
|
|
||||||
p++; plen++;
|
|
||||||
} else if (p[0] == '0' && slen == 1) {
|
|
||||||
*value = 0;
|
|
||||||
return REDIS_OK;
|
|
||||||
} else {
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
while (plen < slen && p[0] >= '0' && p[0] <= '9') {
|
|
||||||
if (v > (ULLONG_MAX / 10)) /* Overflow. */
|
|
||||||
return REDIS_ERR;
|
|
||||||
v *= 10;
|
|
||||||
|
|
||||||
if (v > (ULLONG_MAX - (p[0]-'0'))) /* Overflow. */
|
|
||||||
return REDIS_ERR;
|
|
||||||
v += p[0]-'0';
|
|
||||||
|
|
||||||
p++; plen++;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Return if not all bytes were used. */
|
|
||||||
if (plen < slen)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
if (negative) {
|
|
||||||
if (v > ((unsigned long long)(-(LLONG_MIN+1))+1)) /* Overflow. */
|
|
||||||
return REDIS_ERR;
|
|
||||||
if (value != NULL) *value = -v;
|
|
||||||
} else {
|
|
||||||
if (v > LLONG_MAX) /* Overflow. */
|
|
||||||
return REDIS_ERR;
|
|
||||||
if (value != NULL) *value = v;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
static char *readLine(redisReader *r, int *_len) {
|
|
||||||
char *p, *s;
|
|
||||||
int len;
|
|
||||||
|
|
||||||
p = r->buf+r->pos;
|
|
||||||
s = seekNewline(p,(r->len-r->pos));
|
|
||||||
if (s != NULL) {
|
|
||||||
len = s-(r->buf+r->pos);
|
|
||||||
r->pos += len+2; /* skip \r\n */
|
|
||||||
if (_len) *_len = len;
|
|
||||||
return p;
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void moveToNextTask(redisReader *r) {
|
|
||||||
redisReadTask *cur, *prv;
|
|
||||||
while (r->ridx >= 0) {
|
|
||||||
/* Return a.s.a.p. when the stack is now empty. */
|
|
||||||
if (r->ridx == 0) {
|
|
||||||
r->ridx--;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
cur = &(r->rstack[r->ridx]);
|
|
||||||
prv = &(r->rstack[r->ridx-1]);
|
|
||||||
assert(prv->type == REDIS_REPLY_ARRAY);
|
|
||||||
if (cur->idx == prv->elements-1) {
|
|
||||||
r->ridx--;
|
|
||||||
} else {
|
|
||||||
/* Reset the type because the next item can be anything */
|
|
||||||
assert(cur->idx < prv->elements);
|
|
||||||
cur->type = -1;
|
|
||||||
cur->elements = -1;
|
|
||||||
cur->idx++;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static int processLineItem(redisReader *r) {
|
|
||||||
redisReadTask *cur = &(r->rstack[r->ridx]);
|
|
||||||
void *obj;
|
|
||||||
char *p;
|
|
||||||
int len;
|
|
||||||
|
|
||||||
if ((p = readLine(r,&len)) != NULL) {
|
|
||||||
if (cur->type == REDIS_REPLY_INTEGER) {
|
|
||||||
if (r->fn && r->fn->createInteger) {
|
|
||||||
long long v;
|
|
||||||
if (string2ll(p, len, &v) == REDIS_ERR) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"Bad integer value");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
obj = r->fn->createInteger(cur,v);
|
|
||||||
} else {
|
|
||||||
obj = (void*)REDIS_REPLY_INTEGER;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
/* Type will be error or status. */
|
|
||||||
if (r->fn && r->fn->createString)
|
|
||||||
obj = r->fn->createString(cur,p,len);
|
|
||||||
else
|
|
||||||
obj = (void*)(size_t)(cur->type);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (obj == NULL) {
|
|
||||||
__redisReaderSetErrorOOM(r);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Set reply if this is the root object. */
|
|
||||||
if (r->ridx == 0) r->reply = obj;
|
|
||||||
moveToNextTask(r);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int processBulkItem(redisReader *r) {
|
|
||||||
redisReadTask *cur = &(r->rstack[r->ridx]);
|
|
||||||
void *obj = NULL;
|
|
||||||
char *p, *s;
|
|
||||||
long long len;
|
|
||||||
unsigned long bytelen;
|
|
||||||
int success = 0;
|
|
||||||
|
|
||||||
p = r->buf+r->pos;
|
|
||||||
s = seekNewline(p,r->len-r->pos);
|
|
||||||
if (s != NULL) {
|
|
||||||
p = r->buf+r->pos;
|
|
||||||
bytelen = s-(r->buf+r->pos)+2; /* include \r\n */
|
|
||||||
|
|
||||||
if (string2ll(p, bytelen - 2, &len) == REDIS_ERR) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"Bad bulk string length");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (len < -1 || (LLONG_MAX > SIZE_MAX && len > (long long)SIZE_MAX)) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"Bulk string length out of range");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (len == -1) {
|
|
||||||
/* The nil object can always be created. */
|
|
||||||
if (r->fn && r->fn->createNil)
|
|
||||||
obj = r->fn->createNil(cur);
|
|
||||||
else
|
|
||||||
obj = (void*)REDIS_REPLY_NIL;
|
|
||||||
success = 1;
|
|
||||||
} else {
|
|
||||||
/* Only continue when the buffer contains the entire bulk item. */
|
|
||||||
bytelen += len+2; /* include \r\n */
|
|
||||||
if (r->pos+bytelen <= r->len) {
|
|
||||||
if (r->fn && r->fn->createString)
|
|
||||||
obj = r->fn->createString(cur,s+2,len);
|
|
||||||
else
|
|
||||||
obj = (void*)REDIS_REPLY_STRING;
|
|
||||||
success = 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Proceed when obj was created. */
|
|
||||||
if (success) {
|
|
||||||
if (obj == NULL) {
|
|
||||||
__redisReaderSetErrorOOM(r);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
r->pos += bytelen;
|
|
||||||
|
|
||||||
/* Set reply if this is the root object. */
|
|
||||||
if (r->ridx == 0) r->reply = obj;
|
|
||||||
moveToNextTask(r);
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int processMultiBulkItem(redisReader *r) {
|
|
||||||
redisReadTask *cur = &(r->rstack[r->ridx]);
|
|
||||||
void *obj;
|
|
||||||
char *p;
|
|
||||||
long long elements;
|
|
||||||
int root = 0, len;
|
|
||||||
|
|
||||||
/* Set error for nested multi bulks with depth > 7 */
|
|
||||||
if (r->ridx == 8) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"No support for nested multi bulk replies with depth > 7");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ((p = readLine(r,&len)) != NULL) {
|
|
||||||
if (string2ll(p, len, &elements) == REDIS_ERR) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"Bad multi-bulk length");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
root = (r->ridx == 0);
|
|
||||||
|
|
||||||
if (elements < -1 || elements > INT_MAX) {
|
|
||||||
__redisReaderSetError(r,REDIS_ERR_PROTOCOL,
|
|
||||||
"Multi-bulk length out of range");
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (elements == -1) {
|
|
||||||
if (r->fn && r->fn->createNil)
|
|
||||||
obj = r->fn->createNil(cur);
|
|
||||||
else
|
|
||||||
obj = (void*)REDIS_REPLY_NIL;
|
|
||||||
|
|
||||||
if (obj == NULL) {
|
|
||||||
__redisReaderSetErrorOOM(r);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
moveToNextTask(r);
|
|
||||||
} else {
|
|
||||||
if (r->fn && r->fn->createArray)
|
|
||||||
obj = r->fn->createArray(cur,elements);
|
|
||||||
else
|
|
||||||
obj = (void*)REDIS_REPLY_ARRAY;
|
|
||||||
|
|
||||||
if (obj == NULL) {
|
|
||||||
__redisReaderSetErrorOOM(r);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Modify task stack when there are more than 0 elements. */
|
|
||||||
if (elements > 0) {
|
|
||||||
cur->elements = elements;
|
|
||||||
cur->obj = obj;
|
|
||||||
r->ridx++;
|
|
||||||
r->rstack[r->ridx].type = -1;
|
|
||||||
r->rstack[r->ridx].elements = -1;
|
|
||||||
r->rstack[r->ridx].idx = 0;
|
|
||||||
r->rstack[r->ridx].obj = NULL;
|
|
||||||
r->rstack[r->ridx].parent = cur;
|
|
||||||
r->rstack[r->ridx].privdata = r->privdata;
|
|
||||||
} else {
|
|
||||||
moveToNextTask(r);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Set reply if this is the root object. */
|
|
||||||
if (root) r->reply = obj;
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int processItem(redisReader *r) {
|
|
||||||
redisReadTask *cur = &(r->rstack[r->ridx]);
|
|
||||||
char *p;
|
|
||||||
|
|
||||||
/* check if we need to read type */
|
|
||||||
if (cur->type < 0) {
|
|
||||||
if ((p = readBytes(r,1)) != NULL) {
|
|
||||||
switch (p[0]) {
|
|
||||||
case '-':
|
|
||||||
cur->type = REDIS_REPLY_ERROR;
|
|
||||||
break;
|
|
||||||
case '+':
|
|
||||||
cur->type = REDIS_REPLY_STATUS;
|
|
||||||
break;
|
|
||||||
case ':':
|
|
||||||
cur->type = REDIS_REPLY_INTEGER;
|
|
||||||
break;
|
|
||||||
case '$':
|
|
||||||
cur->type = REDIS_REPLY_STRING;
|
|
||||||
break;
|
|
||||||
case '*':
|
|
||||||
cur->type = REDIS_REPLY_ARRAY;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
__redisReaderSetErrorProtocolByte(r,*p);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
/* could not consume 1 byte */
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* process typed item */
|
|
||||||
switch(cur->type) {
|
|
||||||
case REDIS_REPLY_ERROR:
|
|
||||||
case REDIS_REPLY_STATUS:
|
|
||||||
case REDIS_REPLY_INTEGER:
|
|
||||||
return processLineItem(r);
|
|
||||||
case REDIS_REPLY_STRING:
|
|
||||||
return processBulkItem(r);
|
|
||||||
case REDIS_REPLY_ARRAY:
|
|
||||||
return processMultiBulkItem(r);
|
|
||||||
default:
|
|
||||||
assert(NULL);
|
|
||||||
return REDIS_ERR; /* Avoid warning. */
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
redisReader *redisReaderCreateWithFunctions(redisReplyObjectFunctions *fn) {
|
|
||||||
redisReader *r;
|
|
||||||
|
|
||||||
r = calloc(1,sizeof(redisReader));
|
|
||||||
if (r == NULL)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
r->fn = fn;
|
|
||||||
r->buf = sdsempty();
|
|
||||||
r->maxbuf = REDIS_READER_MAX_BUF;
|
|
||||||
if (r->buf == NULL) {
|
|
||||||
free(r);
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
r->ridx = -1;
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
|
|
||||||
void redisReaderFree(redisReader *r) {
|
|
||||||
if (r == NULL)
|
|
||||||
return;
|
|
||||||
if (r->reply != NULL && r->fn && r->fn->freeObject)
|
|
||||||
r->fn->freeObject(r->reply);
|
|
||||||
sdsfree(r->buf);
|
|
||||||
free(r);
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisReaderFeed(redisReader *r, const char *buf, size_t len) {
|
|
||||||
sds newbuf;
|
|
||||||
|
|
||||||
/* Return early when this reader is in an erroneous state. */
|
|
||||||
if (r->err)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Copy the provided buffer. */
|
|
||||||
if (buf != NULL && len >= 1) {
|
|
||||||
/* Destroy internal buffer when it is empty and is quite large. */
|
|
||||||
if (r->len == 0 && r->maxbuf != 0 && sdsavail(r->buf) > r->maxbuf) {
|
|
||||||
sdsfree(r->buf);
|
|
||||||
r->buf = sdsempty();
|
|
||||||
r->pos = 0;
|
|
||||||
|
|
||||||
/* r->buf should not be NULL since we just free'd a larger one. */
|
|
||||||
assert(r->buf != NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
newbuf = sdscatlen(r->buf,buf,len);
|
|
||||||
if (newbuf == NULL) {
|
|
||||||
__redisReaderSetErrorOOM(r);
|
|
||||||
return REDIS_ERR;
|
|
||||||
}
|
|
||||||
|
|
||||||
r->buf = newbuf;
|
|
||||||
r->len = sdslen(r->buf);
|
|
||||||
}
|
|
||||||
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
||||||
|
|
||||||
int redisReaderGetReply(redisReader *r, void **reply) {
|
|
||||||
/* Default target pointer to NULL. */
|
|
||||||
if (reply != NULL)
|
|
||||||
*reply = NULL;
|
|
||||||
|
|
||||||
/* Return early when this reader is in an erroneous state. */
|
|
||||||
if (r->err)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* When the buffer is empty, there will never be a reply. */
|
|
||||||
if (r->len == 0)
|
|
||||||
return REDIS_OK;
|
|
||||||
|
|
||||||
/* Set first item to process when the stack is empty. */
|
|
||||||
if (r->ridx == -1) {
|
|
||||||
r->rstack[0].type = -1;
|
|
||||||
r->rstack[0].elements = -1;
|
|
||||||
r->rstack[0].idx = -1;
|
|
||||||
r->rstack[0].obj = NULL;
|
|
||||||
r->rstack[0].parent = NULL;
|
|
||||||
r->rstack[0].privdata = r->privdata;
|
|
||||||
r->ridx = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Process items in reply. */
|
|
||||||
while (r->ridx >= 0)
|
|
||||||
if (processItem(r) != REDIS_OK)
|
|
||||||
break;
|
|
||||||
|
|
||||||
/* Return ASAP when an error occurred. */
|
|
||||||
if (r->err)
|
|
||||||
return REDIS_ERR;
|
|
||||||
|
|
||||||
/* Discard part of the buffer when we've consumed at least 1k, to avoid
|
|
||||||
* doing unnecessary calls to memmove() in sds.c. */
|
|
||||||
if (r->pos >= 1024) {
|
|
||||||
sdsrange(r->buf,r->pos,-1);
|
|
||||||
r->pos = 0;
|
|
||||||
r->len = sdslen(r->buf);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Emit a reply when there is one. */
|
|
||||||
if (r->ridx == -1) {
|
|
||||||
if (reply != NULL)
|
|
||||||
*reply = r->reply;
|
|
||||||
r->reply = NULL;
|
|
||||||
}
|
|
||||||
return REDIS_OK;
|
|
||||||
}
|
|
111
controller/thirdparty/hiredis-0.14.1/read.h
vendored
111
controller/thirdparty/hiredis-0.14.1/read.h
vendored
|
@ -1,111 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (c) 2009-2011, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
|
|
||||||
*
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef __HIREDIS_READ_H
|
|
||||||
#define __HIREDIS_READ_H
|
|
||||||
#include <stdio.h> /* for size_t */
|
|
||||||
|
|
||||||
#define REDIS_ERR -1
|
|
||||||
#define REDIS_OK 0
|
|
||||||
|
|
||||||
/* When an error occurs, the err flag in a context is set to hold the type of
|
|
||||||
* error that occurred. REDIS_ERR_IO means there was an I/O error and you
|
|
||||||
* should use the "errno" variable to find out what is wrong.
|
|
||||||
* For other values, the "errstr" field will hold a description. */
|
|
||||||
#define REDIS_ERR_IO 1 /* Error in read or write */
|
|
||||||
#define REDIS_ERR_EOF 3 /* End of file */
|
|
||||||
#define REDIS_ERR_PROTOCOL 4 /* Protocol error */
|
|
||||||
#define REDIS_ERR_OOM 5 /* Out of memory */
|
|
||||||
#define REDIS_ERR_OTHER 2 /* Everything else... */
|
|
||||||
|
|
||||||
#define REDIS_REPLY_STRING 1
|
|
||||||
#define REDIS_REPLY_ARRAY 2
|
|
||||||
#define REDIS_REPLY_INTEGER 3
|
|
||||||
#define REDIS_REPLY_NIL 4
|
|
||||||
#define REDIS_REPLY_STATUS 5
|
|
||||||
#define REDIS_REPLY_ERROR 6
|
|
||||||
|
|
||||||
#define REDIS_READER_MAX_BUF (1024*16) /* Default max unused reader buffer. */
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
typedef struct redisReadTask {
|
|
||||||
int type;
|
|
||||||
int elements; /* number of elements in multibulk container */
|
|
||||||
int idx; /* index in parent (array) object */
|
|
||||||
void *obj; /* holds user-generated value for a read task */
|
|
||||||
struct redisReadTask *parent; /* parent task */
|
|
||||||
void *privdata; /* user-settable arbitrary field */
|
|
||||||
} redisReadTask;
|
|
||||||
|
|
||||||
typedef struct redisReplyObjectFunctions {
|
|
||||||
void *(*createString)(const redisReadTask*, char*, size_t);
|
|
||||||
void *(*createArray)(const redisReadTask*, int);
|
|
||||||
void *(*createInteger)(const redisReadTask*, long long);
|
|
||||||
void *(*createNil)(const redisReadTask*);
|
|
||||||
void (*freeObject)(void*);
|
|
||||||
} redisReplyObjectFunctions;
|
|
||||||
|
|
||||||
typedef struct redisReader {
|
|
||||||
int err; /* Error flags, 0 when there is no error */
|
|
||||||
char errstr[128]; /* String representation of error when applicable */
|
|
||||||
|
|
||||||
char *buf; /* Read buffer */
|
|
||||||
size_t pos; /* Buffer cursor */
|
|
||||||
size_t len; /* Buffer length */
|
|
||||||
size_t maxbuf; /* Max length of unused buffer */
|
|
||||||
|
|
||||||
redisReadTask rstack[9];
|
|
||||||
int ridx; /* Index of current read task */
|
|
||||||
void *reply; /* Temporary reply pointer */
|
|
||||||
|
|
||||||
redisReplyObjectFunctions *fn;
|
|
||||||
void *privdata;
|
|
||||||
} redisReader;
|
|
||||||
|
|
||||||
/* Public API for the protocol parser. */
|
|
||||||
redisReader *redisReaderCreateWithFunctions(redisReplyObjectFunctions *fn);
|
|
||||||
void redisReaderFree(redisReader *r);
|
|
||||||
int redisReaderFeed(redisReader *r, const char *buf, size_t len);
|
|
||||||
int redisReaderGetReply(redisReader *r, void **reply);
|
|
||||||
|
|
||||||
#define redisReaderSetPrivdata(_r, _p) (int)(((redisReader*)(_r))->privdata = (_p))
|
|
||||||
#define redisReaderGetObject(_r) (((redisReader*)(_r))->reply)
|
|
||||||
#define redisReaderGetError(_r) (((redisReader*)(_r))->errstr)
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
1272
controller/thirdparty/hiredis-0.14.1/sds.c
vendored
1272
controller/thirdparty/hiredis-0.14.1/sds.c
vendored
File diff suppressed because it is too large
Load diff
273
controller/thirdparty/hiredis-0.14.1/sds.h
vendored
273
controller/thirdparty/hiredis-0.14.1/sds.h
vendored
|
@ -1,273 +0,0 @@
|
||||||
/* SDSLib 2.0 -- A C dynamic strings library
|
|
||||||
*
|
|
||||||
* Copyright (c) 2006-2015, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2015, Oran Agra
|
|
||||||
* Copyright (c) 2015, Redis Labs, Inc
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef __SDS_H
|
|
||||||
#define __SDS_H
|
|
||||||
|
|
||||||
#define SDS_MAX_PREALLOC (1024*1024)
|
|
||||||
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <stdarg.h>
|
|
||||||
#include <stdint.h>
|
|
||||||
|
|
||||||
typedef char *sds;
|
|
||||||
|
|
||||||
/* Note: sdshdr5 is never used, we just access the flags byte directly.
|
|
||||||
* However is here to document the layout of type 5 SDS strings. */
|
|
||||||
struct __attribute__ ((__packed__)) sdshdr5 {
|
|
||||||
unsigned char flags; /* 3 lsb of type, and 5 msb of string length */
|
|
||||||
char buf[];
|
|
||||||
};
|
|
||||||
struct __attribute__ ((__packed__)) sdshdr8 {
|
|
||||||
uint8_t len; /* used */
|
|
||||||
uint8_t alloc; /* excluding the header and null terminator */
|
|
||||||
unsigned char flags; /* 3 lsb of type, 5 unused bits */
|
|
||||||
char buf[];
|
|
||||||
};
|
|
||||||
struct __attribute__ ((__packed__)) sdshdr16 {
|
|
||||||
uint16_t len; /* used */
|
|
||||||
uint16_t alloc; /* excluding the header and null terminator */
|
|
||||||
unsigned char flags; /* 3 lsb of type, 5 unused bits */
|
|
||||||
char buf[];
|
|
||||||
};
|
|
||||||
struct __attribute__ ((__packed__)) sdshdr32 {
|
|
||||||
uint32_t len; /* used */
|
|
||||||
uint32_t alloc; /* excluding the header and null terminator */
|
|
||||||
unsigned char flags; /* 3 lsb of type, 5 unused bits */
|
|
||||||
char buf[];
|
|
||||||
};
|
|
||||||
struct __attribute__ ((__packed__)) sdshdr64 {
|
|
||||||
uint64_t len; /* used */
|
|
||||||
uint64_t alloc; /* excluding the header and null terminator */
|
|
||||||
unsigned char flags; /* 3 lsb of type, 5 unused bits */
|
|
||||||
char buf[];
|
|
||||||
};
|
|
||||||
|
|
||||||
#define SDS_TYPE_5 0
|
|
||||||
#define SDS_TYPE_8 1
|
|
||||||
#define SDS_TYPE_16 2
|
|
||||||
#define SDS_TYPE_32 3
|
|
||||||
#define SDS_TYPE_64 4
|
|
||||||
#define SDS_TYPE_MASK 7
|
|
||||||
#define SDS_TYPE_BITS 3
|
|
||||||
#define SDS_HDR_VAR(T,s) struct sdshdr##T *sh = (struct sdshdr##T *)((s)-(sizeof(struct sdshdr##T)));
|
|
||||||
#define SDS_HDR(T,s) ((struct sdshdr##T *)((s)-(sizeof(struct sdshdr##T))))
|
|
||||||
#define SDS_TYPE_5_LEN(f) ((f)>>SDS_TYPE_BITS)
|
|
||||||
|
|
||||||
static inline size_t sdslen(const sds s) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5:
|
|
||||||
return SDS_TYPE_5_LEN(flags);
|
|
||||||
case SDS_TYPE_8:
|
|
||||||
return SDS_HDR(8,s)->len;
|
|
||||||
case SDS_TYPE_16:
|
|
||||||
return SDS_HDR(16,s)->len;
|
|
||||||
case SDS_TYPE_32:
|
|
||||||
return SDS_HDR(32,s)->len;
|
|
||||||
case SDS_TYPE_64:
|
|
||||||
return SDS_HDR(64,s)->len;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline size_t sdsavail(const sds s) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5: {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
case SDS_TYPE_8: {
|
|
||||||
SDS_HDR_VAR(8,s);
|
|
||||||
return sh->alloc - sh->len;
|
|
||||||
}
|
|
||||||
case SDS_TYPE_16: {
|
|
||||||
SDS_HDR_VAR(16,s);
|
|
||||||
return sh->alloc - sh->len;
|
|
||||||
}
|
|
||||||
case SDS_TYPE_32: {
|
|
||||||
SDS_HDR_VAR(32,s);
|
|
||||||
return sh->alloc - sh->len;
|
|
||||||
}
|
|
||||||
case SDS_TYPE_64: {
|
|
||||||
SDS_HDR_VAR(64,s);
|
|
||||||
return sh->alloc - sh->len;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void sdssetlen(sds s, size_t newlen) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5:
|
|
||||||
{
|
|
||||||
unsigned char *fp = ((unsigned char*)s)-1;
|
|
||||||
*fp = SDS_TYPE_5 | (newlen << SDS_TYPE_BITS);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_8:
|
|
||||||
SDS_HDR(8,s)->len = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_16:
|
|
||||||
SDS_HDR(16,s)->len = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_32:
|
|
||||||
SDS_HDR(32,s)->len = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_64:
|
|
||||||
SDS_HDR(64,s)->len = newlen;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void sdsinclen(sds s, size_t inc) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5:
|
|
||||||
{
|
|
||||||
unsigned char *fp = ((unsigned char*)s)-1;
|
|
||||||
unsigned char newlen = SDS_TYPE_5_LEN(flags)+inc;
|
|
||||||
*fp = SDS_TYPE_5 | (newlen << SDS_TYPE_BITS);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_8:
|
|
||||||
SDS_HDR(8,s)->len += inc;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_16:
|
|
||||||
SDS_HDR(16,s)->len += inc;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_32:
|
|
||||||
SDS_HDR(32,s)->len += inc;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_64:
|
|
||||||
SDS_HDR(64,s)->len += inc;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* sdsalloc() = sdsavail() + sdslen() */
|
|
||||||
static inline size_t sdsalloc(const sds s) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5:
|
|
||||||
return SDS_TYPE_5_LEN(flags);
|
|
||||||
case SDS_TYPE_8:
|
|
||||||
return SDS_HDR(8,s)->alloc;
|
|
||||||
case SDS_TYPE_16:
|
|
||||||
return SDS_HDR(16,s)->alloc;
|
|
||||||
case SDS_TYPE_32:
|
|
||||||
return SDS_HDR(32,s)->alloc;
|
|
||||||
case SDS_TYPE_64:
|
|
||||||
return SDS_HDR(64,s)->alloc;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static inline void sdssetalloc(sds s, size_t newlen) {
|
|
||||||
unsigned char flags = s[-1];
|
|
||||||
switch(flags&SDS_TYPE_MASK) {
|
|
||||||
case SDS_TYPE_5:
|
|
||||||
/* Nothing to do, this type has no total allocation info. */
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_8:
|
|
||||||
SDS_HDR(8,s)->alloc = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_16:
|
|
||||||
SDS_HDR(16,s)->alloc = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_32:
|
|
||||||
SDS_HDR(32,s)->alloc = newlen;
|
|
||||||
break;
|
|
||||||
case SDS_TYPE_64:
|
|
||||||
SDS_HDR(64,s)->alloc = newlen;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
sds sdsnewlen(const void *init, size_t initlen);
|
|
||||||
sds sdsnew(const char *init);
|
|
||||||
sds sdsempty(void);
|
|
||||||
sds sdsdup(const sds s);
|
|
||||||
void sdsfree(sds s);
|
|
||||||
sds sdsgrowzero(sds s, size_t len);
|
|
||||||
sds sdscatlen(sds s, const void *t, size_t len);
|
|
||||||
sds sdscat(sds s, const char *t);
|
|
||||||
sds sdscatsds(sds s, const sds t);
|
|
||||||
sds sdscpylen(sds s, const char *t, size_t len);
|
|
||||||
sds sdscpy(sds s, const char *t);
|
|
||||||
|
|
||||||
sds sdscatvprintf(sds s, const char *fmt, va_list ap);
|
|
||||||
#ifdef __GNUC__
|
|
||||||
sds sdscatprintf(sds s, const char *fmt, ...)
|
|
||||||
__attribute__((format(printf, 2, 3)));
|
|
||||||
#else
|
|
||||||
sds sdscatprintf(sds s, const char *fmt, ...);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
sds sdscatfmt(sds s, char const *fmt, ...);
|
|
||||||
sds sdstrim(sds s, const char *cset);
|
|
||||||
void sdsrange(sds s, int start, int end);
|
|
||||||
void sdsupdatelen(sds s);
|
|
||||||
void sdsclear(sds s);
|
|
||||||
int sdscmp(const sds s1, const sds s2);
|
|
||||||
sds *sdssplitlen(const char *s, int len, const char *sep, int seplen, int *count);
|
|
||||||
void sdsfreesplitres(sds *tokens, int count);
|
|
||||||
void sdstolower(sds s);
|
|
||||||
void sdstoupper(sds s);
|
|
||||||
sds sdsfromlonglong(long long value);
|
|
||||||
sds sdscatrepr(sds s, const char *p, size_t len);
|
|
||||||
sds *sdssplitargs(const char *line, int *argc);
|
|
||||||
sds sdsmapchars(sds s, const char *from, const char *to, size_t setlen);
|
|
||||||
sds sdsjoin(char **argv, int argc, char *sep);
|
|
||||||
sds sdsjoinsds(sds *argv, int argc, const char *sep, size_t seplen);
|
|
||||||
|
|
||||||
/* Low level functions exposed to the user API */
|
|
||||||
sds sdsMakeRoomFor(sds s, size_t addlen);
|
|
||||||
void sdsIncrLen(sds s, int incr);
|
|
||||||
sds sdsRemoveFreeSpace(sds s);
|
|
||||||
size_t sdsAllocSize(sds s);
|
|
||||||
void *sdsAllocPtr(sds s);
|
|
||||||
|
|
||||||
/* Export the allocator used by SDS to the program using SDS.
|
|
||||||
* Sometimes the program SDS is linked to, may use a different set of
|
|
||||||
* allocators, but may want to allocate or free things that SDS will
|
|
||||||
* respectively free or allocate. */
|
|
||||||
void *sds_malloc(size_t size);
|
|
||||||
void *sds_realloc(void *ptr, size_t size);
|
|
||||||
void sds_free(void *ptr);
|
|
||||||
|
|
||||||
#ifdef REDIS_TEST
|
|
||||||
int sdsTest(int argc, char *argv[]);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
42
controller/thirdparty/hiredis-0.14.1/sdsalloc.h
vendored
42
controller/thirdparty/hiredis-0.14.1/sdsalloc.h
vendored
|
@ -1,42 +0,0 @@
|
||||||
/* SDSLib 2.0 -- A C dynamic strings library
|
|
||||||
*
|
|
||||||
* Copyright (c) 2006-2015, Salvatore Sanfilippo <antirez at gmail dot com>
|
|
||||||
* Copyright (c) 2015, Oran Agra
|
|
||||||
* Copyright (c) 2015, Redis Labs, Inc
|
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
|
|
||||||
/* SDS allocator selection.
|
|
||||||
*
|
|
||||||
* This file is used in order to change the SDS allocator at compile time.
|
|
||||||
* Just define the following defines to what you want to use. Also add
|
|
||||||
* the include of your alternate allocator if needed (not needed in order
|
|
||||||
* to use the default libc allocator). */
|
|
||||||
|
|
||||||
#define s_malloc malloc
|
|
||||||
#define s_realloc realloc
|
|
||||||
#define s_free free
|
|
923
controller/thirdparty/hiredis-0.14.1/test.c
vendored
923
controller/thirdparty/hiredis-0.14.1/test.c
vendored
|
@ -1,923 +0,0 @@
|
||||||
#include "fmacros.h"
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <strings.h>
|
|
||||||
#include <sys/time.h>
|
|
||||||
#include <assert.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <signal.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <limits.h>
|
|
||||||
|
|
||||||
#include "hiredis.h"
|
|
||||||
#include "net.h"
|
|
||||||
|
|
||||||
enum connection_type {
|
|
||||||
CONN_TCP,
|
|
||||||
CONN_UNIX,
|
|
||||||
CONN_FD
|
|
||||||
};
|
|
||||||
|
|
||||||
struct config {
|
|
||||||
enum connection_type type;
|
|
||||||
|
|
||||||
struct {
|
|
||||||
const char *host;
|
|
||||||
int port;
|
|
||||||
struct timeval timeout;
|
|
||||||
} tcp;
|
|
||||||
|
|
||||||
struct {
|
|
||||||
const char *path;
|
|
||||||
} unix_sock;
|
|
||||||
};
|
|
||||||
|
|
||||||
/* The following lines make up our testing "framework" :) */
|
|
||||||
static int tests = 0, fails = 0;
|
|
||||||
#define test(_s) { printf("#%02d ", ++tests); printf(_s); }
|
|
||||||
#define test_cond(_c) if(_c) printf("\033[0;32mPASSED\033[0;0m\n"); else {printf("\033[0;31mFAILED\033[0;0m\n"); fails++;}
|
|
||||||
|
|
||||||
static long long usec(void) {
|
|
||||||
struct timeval tv;
|
|
||||||
gettimeofday(&tv,NULL);
|
|
||||||
return (((long long)tv.tv_sec)*1000000)+tv.tv_usec;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* The assert() calls below have side effects, so we need assert()
|
|
||||||
* even if we are compiling without asserts (-DNDEBUG). */
|
|
||||||
#ifdef NDEBUG
|
|
||||||
#undef assert
|
|
||||||
#define assert(e) (void)(e)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
static redisContext *select_database(redisContext *c) {
|
|
||||||
redisReply *reply;
|
|
||||||
|
|
||||||
/* Switch to DB 9 for testing, now that we know we can chat. */
|
|
||||||
reply = redisCommand(c,"SELECT 9");
|
|
||||||
assert(reply != NULL);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Make sure the DB is emtpy */
|
|
||||||
reply = redisCommand(c,"DBSIZE");
|
|
||||||
assert(reply != NULL);
|
|
||||||
if (reply->type == REDIS_REPLY_INTEGER && reply->integer == 0) {
|
|
||||||
/* Awesome, DB 9 is empty and we can continue. */
|
|
||||||
freeReplyObject(reply);
|
|
||||||
} else {
|
|
||||||
printf("Database #9 is not empty, test can not continue\n");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
return c;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int disconnect(redisContext *c, int keep_fd) {
|
|
||||||
redisReply *reply;
|
|
||||||
|
|
||||||
/* Make sure we're on DB 9. */
|
|
||||||
reply = redisCommand(c,"SELECT 9");
|
|
||||||
assert(reply != NULL);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
reply = redisCommand(c,"FLUSHDB");
|
|
||||||
assert(reply != NULL);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* Free the context as well, but keep the fd if requested. */
|
|
||||||
if (keep_fd)
|
|
||||||
return redisFreeKeepFd(c);
|
|
||||||
redisFree(c);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
static redisContext *connect(struct config config) {
|
|
||||||
redisContext *c = NULL;
|
|
||||||
|
|
||||||
if (config.type == CONN_TCP) {
|
|
||||||
c = redisConnect(config.tcp.host, config.tcp.port);
|
|
||||||
} else if (config.type == CONN_UNIX) {
|
|
||||||
c = redisConnectUnix(config.unix_sock.path);
|
|
||||||
} else if (config.type == CONN_FD) {
|
|
||||||
/* Create a dummy connection just to get an fd to inherit */
|
|
||||||
redisContext *dummy_ctx = redisConnectUnix(config.unix_sock.path);
|
|
||||||
if (dummy_ctx) {
|
|
||||||
int fd = disconnect(dummy_ctx, 1);
|
|
||||||
printf("Connecting to inherited fd %d\n", fd);
|
|
||||||
c = redisConnectFd(fd);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
assert(NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (c == NULL) {
|
|
||||||
printf("Connection error: can't allocate redis context\n");
|
|
||||||
exit(1);
|
|
||||||
} else if (c->err) {
|
|
||||||
printf("Connection error: %s\n", c->errstr);
|
|
||||||
redisFree(c);
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
return select_database(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_format_commands(void) {
|
|
||||||
char *cmd;
|
|
||||||
int len;
|
|
||||||
|
|
||||||
test("Format command without interpolation: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET foo bar");
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with %%s string interpolation: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %s %s","foo","bar");
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with %%s and an empty string: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %s %s","foo","");
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$0\r\n\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(0+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with an empty string in between proper interpolations: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %s %s","","foo");
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$0\r\n\r\n$3\r\nfoo\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(0+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with %%b string interpolation: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %b %b","foo",(size_t)3,"b\0r",(size_t)3);
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$3\r\nb\0r\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with %%b and an empty string: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %b %b","foo",(size_t)3,"",(size_t)0);
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$0\r\n\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(0+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command with literal %%: ");
|
|
||||||
len = redisFormatCommand(&cmd,"SET %% %%");
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$1\r\n%\r\n$1\r\n%\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(1+2)+4+(1+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
/* Vararg width depends on the type. These tests make sure that the
|
|
||||||
* width is correctly determined using the format and subsequent varargs
|
|
||||||
* can correctly be interpolated. */
|
|
||||||
#define INTEGER_WIDTH_TEST(fmt, type) do { \
|
|
||||||
type value = 123; \
|
|
||||||
test("Format command with printf-delegation (" #type "): "); \
|
|
||||||
len = redisFormatCommand(&cmd,"key:%08" fmt " str:%s", value, "hello"); \
|
|
||||||
test_cond(strncmp(cmd,"*2\r\n$12\r\nkey:00000123\r\n$9\r\nstr:hello\r\n",len) == 0 && \
|
|
||||||
len == 4+5+(12+2)+4+(9+2)); \
|
|
||||||
free(cmd); \
|
|
||||||
} while(0)
|
|
||||||
|
|
||||||
#define FLOAT_WIDTH_TEST(type) do { \
|
|
||||||
type value = 123.0; \
|
|
||||||
test("Format command with printf-delegation (" #type "): "); \
|
|
||||||
len = redisFormatCommand(&cmd,"key:%08.3f str:%s", value, "hello"); \
|
|
||||||
test_cond(strncmp(cmd,"*2\r\n$12\r\nkey:0123.000\r\n$9\r\nstr:hello\r\n",len) == 0 && \
|
|
||||||
len == 4+5+(12+2)+4+(9+2)); \
|
|
||||||
free(cmd); \
|
|
||||||
} while(0)
|
|
||||||
|
|
||||||
INTEGER_WIDTH_TEST("d", int);
|
|
||||||
INTEGER_WIDTH_TEST("hhd", char);
|
|
||||||
INTEGER_WIDTH_TEST("hd", short);
|
|
||||||
INTEGER_WIDTH_TEST("ld", long);
|
|
||||||
INTEGER_WIDTH_TEST("lld", long long);
|
|
||||||
INTEGER_WIDTH_TEST("u", unsigned int);
|
|
||||||
INTEGER_WIDTH_TEST("hhu", unsigned char);
|
|
||||||
INTEGER_WIDTH_TEST("hu", unsigned short);
|
|
||||||
INTEGER_WIDTH_TEST("lu", unsigned long);
|
|
||||||
INTEGER_WIDTH_TEST("llu", unsigned long long);
|
|
||||||
FLOAT_WIDTH_TEST(float);
|
|
||||||
FLOAT_WIDTH_TEST(double);
|
|
||||||
|
|
||||||
test("Format command with invalid printf format: ");
|
|
||||||
len = redisFormatCommand(&cmd,"key:%08p %b",(void*)1234,"foo",(size_t)3);
|
|
||||||
test_cond(len == -1);
|
|
||||||
|
|
||||||
const char *argv[3];
|
|
||||||
argv[0] = "SET";
|
|
||||||
argv[1] = "foo\0xxx";
|
|
||||||
argv[2] = "bar";
|
|
||||||
size_t lens[3] = { 3, 7, 3 };
|
|
||||||
int argc = 3;
|
|
||||||
|
|
||||||
test("Format command by passing argc/argv without lengths: ");
|
|
||||||
len = redisFormatCommandArgv(&cmd,argc,argv,NULL);
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
test("Format command by passing argc/argv with lengths: ");
|
|
||||||
len = redisFormatCommandArgv(&cmd,argc,argv,lens);
|
|
||||||
test_cond(strncmp(cmd,"*3\r\n$3\r\nSET\r\n$7\r\nfoo\0xxx\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(7+2)+4+(3+2));
|
|
||||||
free(cmd);
|
|
||||||
|
|
||||||
sds sds_cmd;
|
|
||||||
|
|
||||||
sds_cmd = sdsempty();
|
|
||||||
test("Format command into sds by passing argc/argv without lengths: ");
|
|
||||||
len = redisFormatSdsCommandArgv(&sds_cmd,argc,argv,NULL);
|
|
||||||
test_cond(strncmp(sds_cmd,"*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(3+2)+4+(3+2));
|
|
||||||
sdsfree(sds_cmd);
|
|
||||||
|
|
||||||
sds_cmd = sdsempty();
|
|
||||||
test("Format command into sds by passing argc/argv with lengths: ");
|
|
||||||
len = redisFormatSdsCommandArgv(&sds_cmd,argc,argv,lens);
|
|
||||||
test_cond(strncmp(sds_cmd,"*3\r\n$3\r\nSET\r\n$7\r\nfoo\0xxx\r\n$3\r\nbar\r\n",len) == 0 &&
|
|
||||||
len == 4+4+(3+2)+4+(7+2)+4+(3+2));
|
|
||||||
sdsfree(sds_cmd);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_append_formatted_commands(struct config config) {
|
|
||||||
redisContext *c;
|
|
||||||
redisReply *reply;
|
|
||||||
char *cmd;
|
|
||||||
int len;
|
|
||||||
|
|
||||||
c = connect(config);
|
|
||||||
|
|
||||||
test("Append format command: ");
|
|
||||||
|
|
||||||
len = redisFormatCommand(&cmd, "SET foo bar");
|
|
||||||
|
|
||||||
test_cond(redisAppendFormattedCommand(c, cmd, len) == REDIS_OK);
|
|
||||||
|
|
||||||
assert(redisGetReply(c, (void*)&reply) == REDIS_OK);
|
|
||||||
|
|
||||||
free(cmd);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
disconnect(c, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_reply_reader(void) {
|
|
||||||
redisReader *reader;
|
|
||||||
void *reply;
|
|
||||||
int ret;
|
|
||||||
int i;
|
|
||||||
|
|
||||||
test("Error handling in reply parser: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader,(char*)"@foo\r\n",6);
|
|
||||||
ret = redisReaderGetReply(reader,NULL);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Protocol error, got \"@\" as reply type byte") == 0);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
/* when the reply already contains multiple items, they must be free'd
|
|
||||||
* on an error. valgrind will bark when this doesn't happen. */
|
|
||||||
test("Memory cleanup in reply parser: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader,(char*)"*2\r\n",4);
|
|
||||||
redisReaderFeed(reader,(char*)"$5\r\nhello\r\n",11);
|
|
||||||
redisReaderFeed(reader,(char*)"@foo\r\n",6);
|
|
||||||
ret = redisReaderGetReply(reader,NULL);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Protocol error, got \"@\" as reply type byte") == 0);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error on nested multi bulks with depth > 7: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
|
|
||||||
for (i = 0; i < 9; i++) {
|
|
||||||
redisReaderFeed(reader,(char*)"*1\r\n",4);
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = redisReaderGetReply(reader,NULL);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strncasecmp(reader->errstr,"No support for",14) == 0);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Correctly parses LLONG_MAX: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, ":9223372036854775807\r\n",22);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_OK &&
|
|
||||||
((redisReply*)reply)->type == REDIS_REPLY_INTEGER &&
|
|
||||||
((redisReply*)reply)->integer == LLONG_MAX);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error when > LLONG_MAX: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, ":9223372036854775808\r\n",22);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Bad integer value") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Correctly parses LLONG_MIN: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, ":-9223372036854775808\r\n",23);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_OK &&
|
|
||||||
((redisReply*)reply)->type == REDIS_REPLY_INTEGER &&
|
|
||||||
((redisReply*)reply)->integer == LLONG_MIN);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error when < LLONG_MIN: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, ":-9223372036854775809\r\n",23);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Bad integer value") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error when array < -1: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, "*-2\r\n+asdf\r\n",12);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Multi-bulk length out of range") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error when bulk < -1: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, "$-2\r\nasdf\r\n",11);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Bulk string length out of range") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Set error when array > INT_MAX: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, "*9223372036854775807\r\n+asdf\r\n",29);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Multi-bulk length out of range") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
#if LLONG_MAX > SIZE_MAX
|
|
||||||
test("Set error when bulk > SIZE_MAX: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader, "$9223372036854775807\r\nasdf\r\n",28);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR &&
|
|
||||||
strcasecmp(reader->errstr,"Bulk string length out of range") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
test("Works with NULL functions for reply: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
reader->fn = NULL;
|
|
||||||
redisReaderFeed(reader,(char*)"+OK\r\n",5);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_OK && reply == (void*)REDIS_REPLY_STATUS);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Works when a single newline (\\r\\n) covers two calls to feed: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
reader->fn = NULL;
|
|
||||||
redisReaderFeed(reader,(char*)"+OK\r",4);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
assert(ret == REDIS_OK && reply == NULL);
|
|
||||||
redisReaderFeed(reader,(char*)"\n",1);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_OK && reply == (void*)REDIS_REPLY_STATUS);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
test("Don't reset state after protocol error: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
reader->fn = NULL;
|
|
||||||
redisReaderFeed(reader,(char*)"x",1);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
assert(ret == REDIS_ERR);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_ERR && reply == NULL);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
|
|
||||||
/* Regression test for issue #45 on GitHub. */
|
|
||||||
test("Don't do empty allocation for empty multi bulk: ");
|
|
||||||
reader = redisReaderCreate();
|
|
||||||
redisReaderFeed(reader,(char*)"*0\r\n",4);
|
|
||||||
ret = redisReaderGetReply(reader,&reply);
|
|
||||||
test_cond(ret == REDIS_OK &&
|
|
||||||
((redisReply*)reply)->type == REDIS_REPLY_ARRAY &&
|
|
||||||
((redisReply*)reply)->elements == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
redisReaderFree(reader);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_free_null(void) {
|
|
||||||
void *redisCtx = NULL;
|
|
||||||
void *reply = NULL;
|
|
||||||
|
|
||||||
test("Don't fail when redisFree is passed a NULL value: ");
|
|
||||||
redisFree(redisCtx);
|
|
||||||
test_cond(redisCtx == NULL);
|
|
||||||
|
|
||||||
test("Don't fail when freeReplyObject is passed a NULL value: ");
|
|
||||||
freeReplyObject(reply);
|
|
||||||
test_cond(reply == NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_blocking_connection_errors(void) {
|
|
||||||
redisContext *c;
|
|
||||||
|
|
||||||
test("Returns error when host cannot be resolved: ");
|
|
||||||
c = redisConnect((char*)"idontexist.test", 6379);
|
|
||||||
test_cond(c->err == REDIS_ERR_OTHER &&
|
|
||||||
(strcmp(c->errstr,"Name or service not known") == 0 ||
|
|
||||||
strcmp(c->errstr,"Can't resolve: idontexist.test") == 0 ||
|
|
||||||
strcmp(c->errstr,"nodename nor servname provided, or not known") == 0 ||
|
|
||||||
strcmp(c->errstr,"No address associated with hostname") == 0 ||
|
|
||||||
strcmp(c->errstr,"Temporary failure in name resolution") == 0 ||
|
|
||||||
strcmp(c->errstr,"hostname nor servname provided, or not known") == 0 ||
|
|
||||||
strcmp(c->errstr,"no address associated with name") == 0));
|
|
||||||
redisFree(c);
|
|
||||||
|
|
||||||
test("Returns error when the port is not open: ");
|
|
||||||
c = redisConnect((char*)"localhost", 1);
|
|
||||||
test_cond(c->err == REDIS_ERR_IO &&
|
|
||||||
strcmp(c->errstr,"Connection refused") == 0);
|
|
||||||
redisFree(c);
|
|
||||||
|
|
||||||
test("Returns error when the unix_sock socket path doesn't accept connections: ");
|
|
||||||
c = redisConnectUnix((char*)"/tmp/idontexist.sock");
|
|
||||||
test_cond(c->err == REDIS_ERR_IO); /* Don't care about the message... */
|
|
||||||
redisFree(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_blocking_connection(struct config config) {
|
|
||||||
redisContext *c;
|
|
||||||
redisReply *reply;
|
|
||||||
|
|
||||||
c = connect(config);
|
|
||||||
|
|
||||||
test("Is able to deliver commands: ");
|
|
||||||
reply = redisCommand(c,"PING");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_STATUS &&
|
|
||||||
strcasecmp(reply->str,"pong") == 0)
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("Is a able to send commands verbatim: ");
|
|
||||||
reply = redisCommand(c,"SET foo bar");
|
|
||||||
test_cond (reply->type == REDIS_REPLY_STATUS &&
|
|
||||||
strcasecmp(reply->str,"ok") == 0)
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("%%s String interpolation works: ");
|
|
||||||
reply = redisCommand(c,"SET %s %s","foo","hello world");
|
|
||||||
freeReplyObject(reply);
|
|
||||||
reply = redisCommand(c,"GET foo");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_STRING &&
|
|
||||||
strcmp(reply->str,"hello world") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("%%b String interpolation works: ");
|
|
||||||
reply = redisCommand(c,"SET %b %b","foo",(size_t)3,"hello\x00world",(size_t)11);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
reply = redisCommand(c,"GET foo");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_STRING &&
|
|
||||||
memcmp(reply->str,"hello\x00world",11) == 0)
|
|
||||||
|
|
||||||
test("Binary reply length is correct: ");
|
|
||||||
test_cond(reply->len == 11)
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("Can parse nil replies: ");
|
|
||||||
reply = redisCommand(c,"GET nokey");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_NIL)
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* test 7 */
|
|
||||||
test("Can parse integer replies: ");
|
|
||||||
reply = redisCommand(c,"INCR mycounter");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_INTEGER && reply->integer == 1)
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("Can parse multi bulk replies: ");
|
|
||||||
freeReplyObject(redisCommand(c,"LPUSH mylist foo"));
|
|
||||||
freeReplyObject(redisCommand(c,"LPUSH mylist bar"));
|
|
||||||
reply = redisCommand(c,"LRANGE mylist 0 -1");
|
|
||||||
test_cond(reply->type == REDIS_REPLY_ARRAY &&
|
|
||||||
reply->elements == 2 &&
|
|
||||||
!memcmp(reply->element[0]->str,"bar",3) &&
|
|
||||||
!memcmp(reply->element[1]->str,"foo",3))
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
/* m/e with multi bulk reply *before* other reply.
|
|
||||||
* specifically test ordering of reply items to parse. */
|
|
||||||
test("Can handle nested multi bulk replies: ");
|
|
||||||
freeReplyObject(redisCommand(c,"MULTI"));
|
|
||||||
freeReplyObject(redisCommand(c,"LRANGE mylist 0 -1"));
|
|
||||||
freeReplyObject(redisCommand(c,"PING"));
|
|
||||||
reply = (redisCommand(c,"EXEC"));
|
|
||||||
test_cond(reply->type == REDIS_REPLY_ARRAY &&
|
|
||||||
reply->elements == 2 &&
|
|
||||||
reply->element[0]->type == REDIS_REPLY_ARRAY &&
|
|
||||||
reply->element[0]->elements == 2 &&
|
|
||||||
!memcmp(reply->element[0]->element[0]->str,"bar",3) &&
|
|
||||||
!memcmp(reply->element[0]->element[1]->str,"foo",3) &&
|
|
||||||
reply->element[1]->type == REDIS_REPLY_STATUS &&
|
|
||||||
strcasecmp(reply->element[1]->str,"pong") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
disconnect(c, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_blocking_connection_timeouts(struct config config) {
|
|
||||||
redisContext *c;
|
|
||||||
redisReply *reply;
|
|
||||||
ssize_t s;
|
|
||||||
const char *cmd = "DEBUG SLEEP 3\r\n";
|
|
||||||
struct timeval tv;
|
|
||||||
|
|
||||||
c = connect(config);
|
|
||||||
test("Successfully completes a command when the timeout is not exceeded: ");
|
|
||||||
reply = redisCommand(c,"SET foo fast");
|
|
||||||
freeReplyObject(reply);
|
|
||||||
tv.tv_sec = 0;
|
|
||||||
tv.tv_usec = 10000;
|
|
||||||
redisSetTimeout(c, tv);
|
|
||||||
reply = redisCommand(c, "GET foo");
|
|
||||||
test_cond(reply != NULL && reply->type == REDIS_REPLY_STRING && memcmp(reply->str, "fast", 4) == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
disconnect(c, 0);
|
|
||||||
|
|
||||||
c = connect(config);
|
|
||||||
test("Does not return a reply when the command times out: ");
|
|
||||||
s = write(c->fd, cmd, strlen(cmd));
|
|
||||||
tv.tv_sec = 0;
|
|
||||||
tv.tv_usec = 10000;
|
|
||||||
redisSetTimeout(c, tv);
|
|
||||||
reply = redisCommand(c, "GET foo");
|
|
||||||
test_cond(s > 0 && reply == NULL && c->err == REDIS_ERR_IO && strcmp(c->errstr, "Resource temporarily unavailable") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("Reconnect properly reconnects after a timeout: ");
|
|
||||||
redisReconnect(c);
|
|
||||||
reply = redisCommand(c, "PING");
|
|
||||||
test_cond(reply != NULL && reply->type == REDIS_REPLY_STATUS && strcmp(reply->str, "PONG") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
test("Reconnect properly uses owned parameters: ");
|
|
||||||
config.tcp.host = "foo";
|
|
||||||
config.unix_sock.path = "foo";
|
|
||||||
redisReconnect(c);
|
|
||||||
reply = redisCommand(c, "PING");
|
|
||||||
test_cond(reply != NULL && reply->type == REDIS_REPLY_STATUS && strcmp(reply->str, "PONG") == 0);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
|
|
||||||
disconnect(c, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_blocking_io_errors(struct config config) {
|
|
||||||
redisContext *c;
|
|
||||||
redisReply *reply;
|
|
||||||
void *_reply;
|
|
||||||
int major, minor;
|
|
||||||
|
|
||||||
/* Connect to target given by config. */
|
|
||||||
c = connect(config);
|
|
||||||
{
|
|
||||||
/* Find out Redis version to determine the path for the next test */
|
|
||||||
const char *field = "redis_version:";
|
|
||||||
char *p, *eptr;
|
|
||||||
|
|
||||||
reply = redisCommand(c,"INFO");
|
|
||||||
p = strstr(reply->str,field);
|
|
||||||
major = strtol(p+strlen(field),&eptr,10);
|
|
||||||
p = eptr+1; /* char next to the first "." */
|
|
||||||
minor = strtol(p,&eptr,10);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
}
|
|
||||||
|
|
||||||
test("Returns I/O error when the connection is lost: ");
|
|
||||||
reply = redisCommand(c,"QUIT");
|
|
||||||
if (major > 2 || (major == 2 && minor > 0)) {
|
|
||||||
/* > 2.0 returns OK on QUIT and read() should be issued once more
|
|
||||||
* to know the descriptor is at EOF. */
|
|
||||||
test_cond(strcasecmp(reply->str,"OK") == 0 &&
|
|
||||||
redisGetReply(c,&_reply) == REDIS_ERR);
|
|
||||||
freeReplyObject(reply);
|
|
||||||
} else {
|
|
||||||
test_cond(reply == NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* On 2.0, QUIT will cause the connection to be closed immediately and
|
|
||||||
* the read(2) for the reply on QUIT will set the error to EOF.
|
|
||||||
* On >2.0, QUIT will return with OK and another read(2) needed to be
|
|
||||||
* issued to find out the socket was closed by the server. In both
|
|
||||||
* conditions, the error will be set to EOF. */
|
|
||||||
assert(c->err == REDIS_ERR_EOF &&
|
|
||||||
strcmp(c->errstr,"Server closed the connection") == 0);
|
|
||||||
redisFree(c);
|
|
||||||
|
|
||||||
c = connect(config);
|
|
||||||
test("Returns I/O error on socket timeout: ");
|
|
||||||
struct timeval tv = { 0, 1000 };
|
|
||||||
assert(redisSetTimeout(c,tv) == REDIS_OK);
|
|
||||||
test_cond(redisGetReply(c,&_reply) == REDIS_ERR &&
|
|
||||||
c->err == REDIS_ERR_IO && errno == EAGAIN);
|
|
||||||
redisFree(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_invalid_timeout_errors(struct config config) {
|
|
||||||
redisContext *c;
|
|
||||||
|
|
||||||
test("Set error when an invalid timeout usec value is given to redisConnectWithTimeout: ");
|
|
||||||
|
|
||||||
config.tcp.timeout.tv_sec = 0;
|
|
||||||
config.tcp.timeout.tv_usec = 10000001;
|
|
||||||
|
|
||||||
c = redisConnectWithTimeout(config.tcp.host, config.tcp.port, config.tcp.timeout);
|
|
||||||
|
|
||||||
test_cond(c->err == REDIS_ERR_IO && strcmp(c->errstr, "Invalid timeout specified") == 0);
|
|
||||||
redisFree(c);
|
|
||||||
|
|
||||||
test("Set error when an invalid timeout sec value is given to redisConnectWithTimeout: ");
|
|
||||||
|
|
||||||
config.tcp.timeout.tv_sec = (((LONG_MAX) - 999) / 1000) + 1;
|
|
||||||
config.tcp.timeout.tv_usec = 0;
|
|
||||||
|
|
||||||
c = redisConnectWithTimeout(config.tcp.host, config.tcp.port, config.tcp.timeout);
|
|
||||||
|
|
||||||
test_cond(c->err == REDIS_ERR_IO && strcmp(c->errstr, "Invalid timeout specified") == 0);
|
|
||||||
redisFree(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void test_throughput(struct config config) {
|
|
||||||
redisContext *c = connect(config);
|
|
||||||
redisReply **replies;
|
|
||||||
int i, num;
|
|
||||||
long long t1, t2;
|
|
||||||
|
|
||||||
test("Throughput:\n");
|
|
||||||
for (i = 0; i < 500; i++)
|
|
||||||
freeReplyObject(redisCommand(c,"LPUSH mylist foo"));
|
|
||||||
|
|
||||||
num = 1000;
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
replies[i] = redisCommand(c,"PING");
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_STATUS);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx PING: %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
replies[i] = redisCommand(c,"LRANGE mylist 0 499");
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_ARRAY);
|
|
||||||
assert(replies[i] != NULL && replies[i]->elements == 500);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx LRANGE with 500 elements: %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
replies[i] = redisCommand(c, "INCRBY incrkey %d", 1000000);
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_INTEGER);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx INCRBY: %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
num = 10000;
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
for (i = 0; i < num; i++)
|
|
||||||
redisAppendCommand(c,"PING");
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
assert(redisGetReply(c, (void*)&replies[i]) == REDIS_OK);
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_STATUS);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx PING (pipelined): %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
for (i = 0; i < num; i++)
|
|
||||||
redisAppendCommand(c,"LRANGE mylist 0 499");
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
assert(redisGetReply(c, (void*)&replies[i]) == REDIS_OK);
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_ARRAY);
|
|
||||||
assert(replies[i] != NULL && replies[i]->elements == 500);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx LRANGE with 500 elements (pipelined): %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
replies = malloc(sizeof(redisReply*)*num);
|
|
||||||
for (i = 0; i < num; i++)
|
|
||||||
redisAppendCommand(c,"INCRBY incrkey %d", 1000000);
|
|
||||||
t1 = usec();
|
|
||||||
for (i = 0; i < num; i++) {
|
|
||||||
assert(redisGetReply(c, (void*)&replies[i]) == REDIS_OK);
|
|
||||||
assert(replies[i] != NULL && replies[i]->type == REDIS_REPLY_INTEGER);
|
|
||||||
}
|
|
||||||
t2 = usec();
|
|
||||||
for (i = 0; i < num; i++) freeReplyObject(replies[i]);
|
|
||||||
free(replies);
|
|
||||||
printf("\t(%dx INCRBY (pipelined): %.3fs)\n", num, (t2-t1)/1000000.0);
|
|
||||||
|
|
||||||
disconnect(c, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
// static long __test_callback_flags = 0;
|
|
||||||
// static void __test_callback(redisContext *c, void *privdata) {
|
|
||||||
// ((void)c);
|
|
||||||
// /* Shift to detect execution order */
|
|
||||||
// __test_callback_flags <<= 8;
|
|
||||||
// __test_callback_flags |= (long)privdata;
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// static void __test_reply_callback(redisContext *c, redisReply *reply, void *privdata) {
|
|
||||||
// ((void)c);
|
|
||||||
// /* Shift to detect execution order */
|
|
||||||
// __test_callback_flags <<= 8;
|
|
||||||
// __test_callback_flags |= (long)privdata;
|
|
||||||
// if (reply) freeReplyObject(reply);
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// static redisContext *__connect_nonblock() {
|
|
||||||
// /* Reset callback flags */
|
|
||||||
// __test_callback_flags = 0;
|
|
||||||
// return redisConnectNonBlock("127.0.0.1", port, NULL);
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// static void test_nonblocking_connection() {
|
|
||||||
// redisContext *c;
|
|
||||||
// int wdone = 0;
|
|
||||||
//
|
|
||||||
// test("Calls command callback when command is issued: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisSetCommandCallback(c,__test_callback,(void*)1);
|
|
||||||
// redisCommand(c,"PING");
|
|
||||||
// test_cond(__test_callback_flags == 1);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("Calls disconnect callback on redisDisconnect: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisSetDisconnectCallback(c,__test_callback,(void*)2);
|
|
||||||
// redisDisconnect(c);
|
|
||||||
// test_cond(__test_callback_flags == 2);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("Calls disconnect callback and free callback on redisFree: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisSetDisconnectCallback(c,__test_callback,(void*)2);
|
|
||||||
// redisSetFreeCallback(c,__test_callback,(void*)4);
|
|
||||||
// redisFree(c);
|
|
||||||
// test_cond(__test_callback_flags == ((2 << 8) | 4));
|
|
||||||
//
|
|
||||||
// test("redisBufferWrite against empty write buffer: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// test_cond(redisBufferWrite(c,&wdone) == REDIS_OK && wdone == 1);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("redisBufferWrite against not yet connected fd: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisCommand(c,"PING");
|
|
||||||
// test_cond(redisBufferWrite(c,NULL) == REDIS_ERR &&
|
|
||||||
// strncmp(c->error,"write:",6) == 0);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("redisBufferWrite against closed fd: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisCommand(c,"PING");
|
|
||||||
// redisDisconnect(c);
|
|
||||||
// test_cond(redisBufferWrite(c,NULL) == REDIS_ERR &&
|
|
||||||
// strncmp(c->error,"write:",6) == 0);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("Process callbacks in the right sequence: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisCommandWithCallback(c,__test_reply_callback,(void*)1,"PING");
|
|
||||||
// redisCommandWithCallback(c,__test_reply_callback,(void*)2,"PING");
|
|
||||||
// redisCommandWithCallback(c,__test_reply_callback,(void*)3,"PING");
|
|
||||||
//
|
|
||||||
// /* Write output buffer */
|
|
||||||
// wdone = 0;
|
|
||||||
// while(!wdone) {
|
|
||||||
// usleep(500);
|
|
||||||
// redisBufferWrite(c,&wdone);
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// /* Read until at least one callback is executed (the 3 replies will
|
|
||||||
// * arrive in a single packet, causing all callbacks to be executed in
|
|
||||||
// * a single pass). */
|
|
||||||
// while(__test_callback_flags == 0) {
|
|
||||||
// assert(redisBufferRead(c) == REDIS_OK);
|
|
||||||
// redisProcessCallbacks(c);
|
|
||||||
// }
|
|
||||||
// test_cond(__test_callback_flags == 0x010203);
|
|
||||||
// redisFree(c);
|
|
||||||
//
|
|
||||||
// test("redisDisconnect executes pending callbacks with NULL reply: ");
|
|
||||||
// c = __connect_nonblock();
|
|
||||||
// redisSetDisconnectCallback(c,__test_callback,(void*)1);
|
|
||||||
// redisCommandWithCallback(c,__test_reply_callback,(void*)2,"PING");
|
|
||||||
// redisDisconnect(c);
|
|
||||||
// test_cond(__test_callback_flags == 0x0201);
|
|
||||||
// redisFree(c);
|
|
||||||
// }
|
|
||||||
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
struct config cfg = {
|
|
||||||
.tcp = {
|
|
||||||
.host = "127.0.0.1",
|
|
||||||
.port = 6379
|
|
||||||
},
|
|
||||||
.unix_sock = {
|
|
||||||
.path = "/tmp/redis.sock"
|
|
||||||
}
|
|
||||||
};
|
|
||||||
int throughput = 1;
|
|
||||||
int test_inherit_fd = 1;
|
|
||||||
|
|
||||||
/* Ignore broken pipe signal (for I/O error tests). */
|
|
||||||
signal(SIGPIPE, SIG_IGN);
|
|
||||||
|
|
||||||
/* Parse command line options. */
|
|
||||||
argv++; argc--;
|
|
||||||
while (argc) {
|
|
||||||
if (argc >= 2 && !strcmp(argv[0],"-h")) {
|
|
||||||
argv++; argc--;
|
|
||||||
cfg.tcp.host = argv[0];
|
|
||||||
} else if (argc >= 2 && !strcmp(argv[0],"-p")) {
|
|
||||||
argv++; argc--;
|
|
||||||
cfg.tcp.port = atoi(argv[0]);
|
|
||||||
} else if (argc >= 2 && !strcmp(argv[0],"-s")) {
|
|
||||||
argv++; argc--;
|
|
||||||
cfg.unix_sock.path = argv[0];
|
|
||||||
} else if (argc >= 1 && !strcmp(argv[0],"--skip-throughput")) {
|
|
||||||
throughput = 0;
|
|
||||||
} else if (argc >= 1 && !strcmp(argv[0],"--skip-inherit-fd")) {
|
|
||||||
test_inherit_fd = 0;
|
|
||||||
} else {
|
|
||||||
fprintf(stderr, "Invalid argument: %s\n", argv[0]);
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
argv++; argc--;
|
|
||||||
}
|
|
||||||
|
|
||||||
test_format_commands();
|
|
||||||
test_reply_reader();
|
|
||||||
test_blocking_connection_errors();
|
|
||||||
test_free_null();
|
|
||||||
|
|
||||||
printf("\nTesting against TCP connection (%s:%d):\n", cfg.tcp.host, cfg.tcp.port);
|
|
||||||
cfg.type = CONN_TCP;
|
|
||||||
test_blocking_connection(cfg);
|
|
||||||
test_blocking_connection_timeouts(cfg);
|
|
||||||
test_blocking_io_errors(cfg);
|
|
||||||
test_invalid_timeout_errors(cfg);
|
|
||||||
test_append_formatted_commands(cfg);
|
|
||||||
if (throughput) test_throughput(cfg);
|
|
||||||
|
|
||||||
printf("\nTesting against Unix socket connection (%s):\n", cfg.unix_sock.path);
|
|
||||||
cfg.type = CONN_UNIX;
|
|
||||||
test_blocking_connection(cfg);
|
|
||||||
test_blocking_connection_timeouts(cfg);
|
|
||||||
test_blocking_io_errors(cfg);
|
|
||||||
if (throughput) test_throughput(cfg);
|
|
||||||
|
|
||||||
if (test_inherit_fd) {
|
|
||||||
printf("\nTesting against inherited fd (%s):\n", cfg.unix_sock.path);
|
|
||||||
cfg.type = CONN_FD;
|
|
||||||
test_blocking_connection(cfg);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if (fails) {
|
|
||||||
printf("*** %d TESTS FAILED ***\n", fails);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
printf("ALL TESTS PASSED\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
42
controller/thirdparty/hiredis-0.14.1/win32.h
vendored
42
controller/thirdparty/hiredis-0.14.1/win32.h
vendored
|
@ -1,42 +0,0 @@
|
||||||
#ifndef _WIN32_HELPER_INCLUDE
|
|
||||||
#define _WIN32_HELPER_INCLUDE
|
|
||||||
#ifdef _MSC_VER
|
|
||||||
|
|
||||||
#ifndef inline
|
|
||||||
#define inline __inline
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef va_copy
|
|
||||||
#define va_copy(d,s) ((d) = (s))
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef snprintf
|
|
||||||
#define snprintf c99_snprintf
|
|
||||||
|
|
||||||
__inline int c99_vsnprintf(char* str, size_t size, const char* format, va_list ap)
|
|
||||||
{
|
|
||||||
int count = -1;
|
|
||||||
|
|
||||||
if (size != 0)
|
|
||||||
count = _vsnprintf_s(str, size, _TRUNCATE, format, ap);
|
|
||||||
if (count == -1)
|
|
||||||
count = _vscprintf(format, ap);
|
|
||||||
|
|
||||||
return count;
|
|
||||||
}
|
|
||||||
|
|
||||||
__inline int c99_snprintf(char* str, size_t size, const char* format, ...)
|
|
||||||
{
|
|
||||||
int count;
|
|
||||||
va_list ap;
|
|
||||||
|
|
||||||
va_start(ap, format);
|
|
||||||
count = c99_vsnprintf(str, size, format, ap);
|
|
||||||
va_end(ap);
|
|
||||||
|
|
||||||
return count;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
||||||
#endif
|
|
|
@ -34,8 +34,6 @@ set_target_properties(${SHARED_LIB} PROPERTIES OUTPUT_NAME ${PROJECT_NAME})
|
||||||
set_target_properties(${STATIC_LIB} PROPERTIES CLEAN_DIRECT_OUTPUT 1)
|
set_target_properties(${STATIC_LIB} PROPERTIES CLEAN_DIRECT_OUTPUT 1)
|
||||||
set_target_properties(${SHARED_LIB} PROPERTIES CLEAN_DIRECT_OUTPUT 1)
|
set_target_properties(${SHARED_LIB} PROPERTIES CLEAN_DIRECT_OUTPUT 1)
|
||||||
|
|
||||||
add_subdirectory(test)
|
|
||||||
|
|
||||||
# Install static lib.
|
# Install static lib.
|
||||||
install(TARGETS ${STATIC_LIB}
|
install(TARGETS ${STATIC_LIB}
|
||||||
ARCHIVE DESTINATION lib)
|
ARCHIVE DESTINATION lib)
|
||||||
|
|
Loading…
Add table
Reference in a new issue