#include "php.h"

#include "php_ini.h"

#include <ext/standard/php_smart_str.h>

#ifndef REDIS_COMMON_H

#define REDIS_COMMON_H

/* NULL check so Eclipse doesn't go crazy */

#ifndef NULL

#define NULL ((void *) 0)

#endif

#define redis_sock_name "Redis Socket Buffer"

#define REDIS_SOCK_STATUS_FAILED 0

#define REDIS_SOCK_STATUS_DISCONNECTED 1

#define REDIS_SOCK_STATUS_UNKNOWN 2

#define REDIS_SOCK_STATUS_CONNECTED 3

#define redis_multi_access_type_name "Redis Multi type access"

#define _NL "\r\n"

/* properties */

#define REDIS_NOT_FOUND 0

#define REDIS_STRING 1

#define REDIS_SET 2

#define REDIS_LIST 3

#define REDIS_ZSET 4

#define REDIS_HASH 5

#ifdef PHP_WIN32

#define PHP_REDIS_API __declspec(dllexport)

#else

#define PHP_REDIS_API

#endif

/* reply types */

typedef enum _REDIS_REPLY_TYPE {

TYPE_EOF = -1,

TYPE_LINE = '+',

TYPE_INT = ':',

TYPE_ERR = '-',

TYPE_BULK = '$',

TYPE_MULTIBULK = '*'

} REDIS_REPLY_TYPE;

/* SCAN variants */

typedef enum _REDIS_SCAN_TYPE {

TYPE_SCAN,

TYPE_SSCAN,

TYPE_HSCAN,

TYPE_ZSCAN

} REDIS_SCAN_TYPE;

/* PUBSUB subcommands */

typedef enum _PUBSUB_TYPE {

PUBSUB_CHANNELS,

PUBSUB_NUMSUB,

PUBSUB_NUMPAT

} PUBSUB_TYPE;

/* options */

#define REDIS_OPT_SERIALIZER 1

#define REDIS_OPT_PREFIX 2

#define REDIS_OPT_READ_TIMEOUT 3

#define REDIS_OPT_SCAN 4

/* cluster options */

#define REDIS_OPT_FAILOVER 5

#define REDIS_FAILOVER_NONE 0

#define REDIS_FAILOVER_ERROR 1

#define REDIS_FAILOVER_DISTRIBUTE 2

#define REDIS_FAILOVER_DISTRIBUTE_SLAVES 3

/* serializers */

#define REDIS_SERIALIZER_NONE 0

#define REDIS_SERIALIZER_PHP 1

#define REDIS_SERIALIZER_IGBINARY 2

/* SCAN options */

#define REDIS_SCAN_NORETRY 0

#define REDIS_SCAN_RETRY 1

/* GETBIT/SETBIT offset range limits */

#define BITOP_MIN_OFFSET 0

#define BITOP_MAX_OFFSET 4294967295

/* Specific error messages we want to throw against */

#define REDIS_ERR_LOADING_MSG "LOADING Redis is loading the dataset in memory"

#define REDIS_ERR_LOADING_KW "LOADING"

#define REDIS_ERR_AUTH_MSG "NOAUTH Authentication required."

#define REDIS_ERR_AUTH_KW "NOAUTH"

#define REDIS_ERR_SYNC_MSG "MASTERDOWN Link with MASTER is down and slave-serve-stale-data is set to 'no'"

#define REDIS_ERR_SYNC_KW "MASTERDOWN"

#define IF_MULTI() if(redis_sock->mode == MULTI)

#define IF_MULTI_OR_ATOMIC() if(redis_sock->mode == MULTI || redis_sock->mode == ATOMIC)\

#define IF_MULTI_OR_PIPELINE() if(redis_sock->mode == MULTI || redis_sock->mode == PIPELINE)

#define IF_PIPELINE() if(redis_sock->mode == PIPELINE)

#define IF_NOT_MULTI() if(redis_sock->mode != MULTI)

#define IF_NOT_ATOMIC() if(redis_sock->mode != ATOMIC)

#define IF_ATOMIC() if(redis_sock->mode == ATOMIC)

#define ELSE_IF_MULTI() else if(redis_sock->mode == MULTI) { \

if(redis_response_enqueued(redis_sock TSRMLS_CC) == 1) { \

RETURN_ZVAL(getThis(), 1, 0);\

} else { \

RETURN_FALSE; \

} \

}

#define ELSE_IF_PIPELINE() else IF_PIPELINE() { \

RETURN_ZVAL(getThis(), 1, 0);\

}

#define MULTI_RESPONSE(callback) IF_MULTI_OR_PIPELINE() { \

fold_item *f1, *current; \

f1 = malloc(sizeof(fold_item)); \

f1->fun = (void *)callback; \

f1->next = NULL; \

current = redis_sock->current;\

if(current) current->next = f1; \

redis_sock->current = f1; \

}

#define PIPELINE_ENQUEUE_COMMAND(cmd, cmd_len) request_item *tmp; \

struct request_item *current_request;\

tmp = malloc(sizeof(request_item));\

tmp->request_str = calloc(cmd_len, 1);\

memcpy(tmp->request_str, cmd, cmd_len);\

tmp->request_size = cmd_len;\

tmp->next = NULL;\

current_request = redis_sock->pipeline_current; \

if(current_request) {\

current_request->next = tmp;\

} \

redis_sock->pipeline_current = tmp; \

if(NULL == redis_sock->pipeline_head) { \

redis_sock->pipeline_head = redis_sock->pipeline_current;\

}

#define SOCKET_WRITE_COMMAND(redis_sock, cmd, cmd_len) \

if(redis_sock_write(redis_sock, cmd, cmd_len TSRMLS_CC) < 0) { \

efree(cmd); \

RETURN_FALSE; \

}

#define REDIS_SAVE_CALLBACK(callback, closure_context) \

IF_MULTI_OR_PIPELINE() { \

fold_item *f1, *current; \

f1 = malloc(sizeof(fold_item)); \

f1->fun = (void *)callback; \

f1->ctx = closure_context; \

f1->next = NULL; \

current = redis_sock->current;\

if(current) current->next = f1; \

redis_sock->current = f1; \

if(NULL == redis_sock->head) { \

redis_sock->head = redis_sock->current;\

}\

}

#define REDIS_ELSE_IF_MULTI(function, closure_context) \

else if(redis_sock->mode == MULTI) { \

if(redis_response_enqueued(redis_sock TSRMLS_CC) == 1) {\

REDIS_SAVE_CALLBACK(function, closure_context); \

RETURN_ZVAL(getThis(), 1, 0);\

} else {\

RETURN_FALSE;\

}\

}

#define REDIS_ELSE_IF_PIPELINE(function, closure_context) \

else IF_PIPELINE() { \

REDIS_SAVE_CALLBACK(function, closure_context); \

RETURN_ZVAL(getThis(), 1, 0); \

}

#define REDIS_PROCESS_REQUEST(redis_sock, cmd, cmd_len) \

IF_MULTI_OR_ATOMIC() { \

SOCKET_WRITE_COMMAND(redis_sock, cmd, cmd_len); \

efree(cmd); \

}\

IF_PIPELINE() { \

PIPELINE_ENQUEUE_COMMAND(cmd, cmd_len); \

efree(cmd); \

}

#define REDIS_PROCESS_RESPONSE_CLOSURE(function, closure_context) \

REDIS_ELSE_IF_MULTI(function, closure_context) \

REDIS_ELSE_IF_PIPELINE(function, closure_context);

#define REDIS_PROCESS_RESPONSE(function) \

REDIS_PROCESS_RESPONSE_CLOSURE(function, NULL)

/* Clear redirection info */

#define REDIS_MOVED_CLEAR(redis_sock) \

redis_sock->redir_slot = 0; \

redis_sock->redir_port = 0; \

redis_sock->redir_type = MOVED_NONE; \

/* Process a command assuming our command where our command building

* function is redis_<cmdname>_cmd */

#define REDIS_PROCESS_CMD(cmdname, resp_func) \

RedisSock *redis_sock; char *cmd; int cmd_len; void *ctx=NULL; \

if(redis_sock_get(getThis(), &redis_sock TSRMLS_CC, 0)<0 || \

redis_##cmdname##_cmd(INTERNAL_FUNCTION_PARAM_PASSTHRU,redis_sock, \

&cmd, &cmd_len, NULL, &ctx)==FAILURE) { \

RETURN_FALSE; \

} \

REDIS_PROCESS_REQUEST(redis_sock, cmd, cmd_len); \

IF_ATOMIC() { \

resp_func(INTERNAL_FUNCTION_PARAM_PASSTHRU, redis_sock, NULL, ctx); \

} \

REDIS_PROCESS_RESPONSE_CLOSURE(resp_func,ctx);

/* Process a command but with a specific command building function

* and keyword which is passed to us*/

#define REDIS_PROCESS_KW_CMD(kw, cmdfunc, resp_func) \

RedisSock *redis_sock; char *cmd; int cmd_len; void *ctx=NULL; \

if(redis_sock_get(getThis(), &redis_sock TSRMLS_CC, 0)<0 || \

cmdfunc(INTERNAL_FUNCTION_PARAM_PASSTHRU, redis_sock, kw, &cmd, \

&cmd_len, NULL, &ctx)==FAILURE) { \

RETURN_FALSE; \

} \

REDIS_PROCESS_REQUEST(redis_sock, cmd, cmd_len); \

IF_ATOMIC() { \

resp_func(INTERNAL_FUNCTION_PARAM_PASSTHRU, redis_sock, NULL, ctx); \

} \

REDIS_PROCESS_RESPONSE_CLOSURE(resp_func,ctx);

/* Extended SET argument detection */

#define IS_EX_ARG(a) \

((a[0]=='e' || a[0]=='E') && (a[1]=='x' || a[1]=='X') && a[2]=='\0')

#define IS_PX_ARG(a) \

((a[0]=='p' || a[0]=='P') && (a[1]=='x' || a[1]=='X') && a[2]=='\0')

#define IS_NX_ARG(a) \

((a[0]=='n' || a[0]=='N') && (a[1]=='x' || a[1]=='X') && a[2]=='\0')

#define IS_XX_ARG(a) \

((a[0]=='x' || a[0]=='X') && (a[1]=='x' || a[1]=='X') && a[2]=='\0')

#define IS_EX_PX_ARG(a) (IS_EX_ARG(a) || IS_PX_ARG(a))

#define IS_NX_XX_ARG(a) (IS_NX_ARG(a) || IS_XX_ARG(a))

/* Given a string and length, validate a zRangeByLex argument. The semantics

* here are that the argument must start with '(' or '[' or be just the char

* '+' or '-' */

#define IS_LEX_ARG(s,l) \

(l>0 && (*s=='(' || *s=='[' || (l==1 && (*s=='+' || *s=='-'))))

typedef enum {ATOMIC, MULTI, PIPELINE} redis_mode;

typedef struct fold_item {

zval * (*fun)(INTERNAL_FUNCTION_PARAMETERS, void *, ...);

void *ctx;

struct fold_item *next;

} fold_item;

typedef struct request_item {

char *request_str;

int request_size; /* size_t */

struct request_item *next;

} request_item;

/* {{{ struct RedisSock */

typedef struct {

php_stream *stream;

char *host;

short port;

char *auth;

double timeout;

double read_timeout;

long retry_interval;

int failed;

int status;

int persistent;

int watching;

char *persistent_id;

int serializer;

long dbNumber;

char *prefix;

int prefix_len;

redis_mode mode;

fold_item *head;

fold_item *current;

request_item *pipeline_head;

request_item *pipeline_current;

char *err;

int err_len;

zend_bool lazy_connect;

int scan;

int readonly;

} RedisSock;

/* }}} */

void

free_reply_callbacks(zval *z_this, RedisSock *redis_sock);

#endif