Adding additional END, RESET, DEBUG, and INCLUDE options.
[m6w6/libmemcached] / libmemcached / options / parser.yy
index d4275a09e32ad2eae697ea7ef127cf6d460474b7..bff5b28df69bb7eaea48033927040399f2547b76 100644 (file)
  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+%error-verbose
+%debug
+%defines
+%expect 0
+%output "libmemcached/options/parser.cc"
+%defines "libmemcached/options/parser.h"
+%lex-param { yyscan_t *scanner }
+%name-prefix="libmemcached_"
+%parse-param { Context *context }
+%parse-param { yyscan_t *scanner }
+%locations
+%pure-parser
+%require "2.2"
+%start begin
+%verbose
+
 %{
 
+#include <config.h>
+
 #include <stdint.h>
 #include <iostream>
+#include <sstream>
+#include <string>
 
-#include <libmemcached/options/type.h>
+#include <libmemcached/options/context.h>
 #include <libmemcached/options/string.h>
 #include <libmemcached/options/symbol.h>
 
 #pragma GCC diagnostic ignored "-Wold-style-cast"
 #include <libmemcached/options/scanner.h>
 
-inline int libmemcached_error(YYLTYPE *locp, type_st *parser, yyscan_t *scanner, const char *str)
+int libmemcached_lex(YYSTYPE* lvalp, YYLTYPE* llocp, void* scanner);
+
+#define parser_abort(A, B) do { parser_abort_func((A), (B)); YYABORT; } while (0) 
+
+inline void parser_abort_func(Context *context, const char *error)
 {
-#if 0
-  std::cerr << str << std::endl;
-#endif
-  return 0;
+  (void)error;
+  if (context->rc == MEMCACHED_SUCCESS)
+    context->rc= MEMCACHED_PARSE_ERROR;
+
+  std::string error_message;
+  error_message+= context->begin;
+  error_message+= " (";
+  error_message+= memcached_strerror(NULL, context->rc);
+  error_message+= ")";
+
+  memcached_set_error_string(context->memc, context->rc, error_message.c_str(), error_message.size());
 }
 
+inline void libmemcached_error(YYLTYPE *locp, Context *context, yyscan_t *scanner, const char *error)
+{
+  if (not context->end())
+    parser_abort_func(context, error);
+}
 
 %}
 
-%error-verbose
-%debug
-%defines
-%expect 0
-%output "libmemcached/options/parser.cc"
-%defines "libmemcached/options/parser.h"
-%lex-param { yyscan_t *scanner }
-%name-prefix="libmemcached_"
-%parse-param { type_st *parser }
-%parse-param { yyscan_t *scanner }
-%locations
-%pure-parser
-%require "2.2"
-%start statement
-%verbose
-
+%token COMMENT
+%token END
+%token RESET
+%token DEBUG
+%token INCLUDE
+%token CONFIGURE_FILE
+%token EMPTY_LINE
 %token SERVER
 %token SERVERS
+%token SERVERS_OPTION
+%token UNKNOWN_OPTION
 %token UNKNOWN
 
-%token DASH_OPTION
-
 /* All behavior options */
 %token AUTO_EJECT_HOSTS
 %token BINARY_PROTOCOL
@@ -98,6 +125,9 @@ inline int libmemcached_error(YYLTYPE *locp, type_st *parser, yyscan_t *scanner,
 %token USE_UDP
 %token VERIFY_KEY
 
+/* Callbacks */
+%token PREFIX_KEY
+
 /* Hash types */
 %token MD5
 %token CRC
@@ -114,193 +144,297 @@ inline int libmemcached_error(YYLTYPE *locp, type_st *parser, yyscan_t *scanner,
 %token MODULA
 %token RANDOM
 
+/* Boolean values */
+%token <boolean> TRUE
+%token <boolean> FALSE
+
 %nonassoc ','
 %nonassoc '='
 
 %token <number> NUMBER
 %token <number> FLOAT
-%token <string> IDENTIFIER
-%token <string> SERVER_WITH_PORT
+%token <string> HOSTNAME
+%token <string> HOSTNAME_WITH_PORT
 %token <string> IPADDRESS
 %token <string> IPADDRESS_WITH_PORT
+%token <string> STRING
+%token <string> QUOTED_STRING
 
 %type <server> server
+%type <string> string
 %type <distribution> distribution
 %type <hash> hash
+%type <behavior> behavior_boolean
+%type <behavior> behavior_number
 
 %%
 
+begin:
+          statement 
+        | statement ' ' statement
+        ;
+
 statement:
-         DASH_OPTION expression
+         expression
+          { }
+        | COMMENT
           { }
-        | statement ' ' DASH_OPTION expression
+        | EMPTY_LINE
           { }
+        | END
+          {
+            context->set_end();
+            YYACCEPT;
+          }
+        | RESET
+          {
+            memcached_reset(context->memc);
+          }
+        | RESET SERVERS
+          {
+            memcached_servers_reset(context->memc);
+          }
+        | DEBUG
+          {
+            yydebug= 1;
+          }
+        | DEBUG TRUE
+          {
+            yydebug= 1;
+          }
+        | DEBUG FALSE
+          {
+            yydebug= 0;
+          }
+        | INCLUDE string
+          {
+            if ((context->rc= memcached_parse_configure_file(context->memc, $2.c_str, $2.length)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);
+            }
+          }
         ;
 
 
 expression:
           SERVER '=' server
           { 
-            (void) memcached_server_add_parsed(parser->memc, $3.c_str, $3.length, $3.port, 0);
+            if ((context->rc= memcached_server_add_parsed(context->memc, $3.c_str, $3.length, $3.port, 0)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);
+            }
           }
-        | SERVERS '=' server_list
+        | SERVERS_OPTION '=' server_list
           {
           }
+        | CONFIGURE_FILE '=' string
+          {
+            memcached_set_configuration_file(context->memc, $3.c_str, $3.length);
+          }
         | behaviors
         ;
 
 behaviors:
-          AUTO_EJECT_HOSTS
+          PREFIX_KEY '=' string
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS, 1);
+            if ((context->rc= memcached_callback_set(context->memc, MEMCACHED_CALLBACK_PREFIX_KEY, std::string($3.c_str, $3.length).c_str())) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
-        | BINARY_PROTOCOL
+        | DISTRIBUTION '=' distribution
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL, 1);
+            if ((context->rc= memcached_behavior_set(context->memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, $3)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
-        | BUFFER_REQUESTS
+        | HASH '=' hash
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, 1);
+            if ((context->rc= memcached_behavior_set(context->memc, MEMCACHED_BEHAVIOR_HASH, $3)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);; 
+            }
           }
-        | CACHE_LOOKUPS
+        | KETAMA_HASH '=' hash
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_CACHE_LOOKUPS, 1);
+            if ((context->rc= memcached_behavior_set(context->memc, MEMCACHED_BEHAVIOR_KETAMA_HASH, $3)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
-        | CONNECT_TIMEOUT '=' NUMBER
+        | behavior_number '=' NUMBER
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT, $3);
+            if ((context->rc= memcached_behavior_set(context->memc, $1, $3)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
-        | _CORK
+        | behavior_boolean
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_CORK, 1);
+            if ((context->rc= memcached_behavior_set(context->memc, $1, true)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
-        | DISTRIBUTION '=' distribution
+        |  USER_DATA
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, $3);
           }
-        | HASH '=' hash
+        ;
+
+behavior_number:
+          CONNECT_TIMEOUT
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, $3);
+            $$= MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT;
           }
-        | HASH_WITH_PREFIX_KEY
+        | IO_MSG_WATERMARK
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY, 1);
+            $$= MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK;
           }
-        | IO_BYTES_WATERMARK '=' NUMBER
+        | IO_BYTES_WATERMARK
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK, $3);
+            $$= MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK;
           }
-        | IO_KEY_PREFETCH '=' NUMBER
+        | IO_KEY_PREFETCH
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH, $3);
+            $$= MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH;
           }
-        | IO_MSG_WATERMARK '=' NUMBER
+        | NUMBER_OF_REPLICAS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK, $3);
+            $$= MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS;
           }
-        | KETAMA
+        | POLL_TIMEOUT
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_KETAMA, true);
+            $$= MEMCACHED_BEHAVIOR_POLL_TIMEOUT;
           }
-        | KETAMA_HASH
+        |  RCV_TIMEOUT
           {
+            $$= MEMCACHED_BEHAVIOR_RCV_TIMEOUT;
           }
-        | KETAMA_WEIGHTED
+        |  RETRY_TIMEOUT
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED, true);
+            $$= MEMCACHED_BEHAVIOR_RETRY_TIMEOUT;
           }
-        | NOREPLY
+        |  SERVER_FAILURE_LIMIT
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_NOREPLY, 1);
+            $$= MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT;
           }
-        | NUMBER_OF_REPLICAS '=' NUMBER
+        |  SND_TIMEOUT
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS, $3);
+            $$= MEMCACHED_BEHAVIOR_SND_TIMEOUT;
           }
-        | POLL_TIMEOUT '=' NUMBER
+        |  SOCKET_RECV_SIZE
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, $3);
+            $$= MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE;
           }
-        |  RANDOMIZE_REPLICA_READ
+        |  SOCKET_SEND_SIZE
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ, true);
+            $$= MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE;
           }
-        |  RCV_TIMEOUT '=' NUMBER
+        ;
+
+behavior_boolean: 
+          AUTO_EJECT_HOSTS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_RCV_TIMEOUT, $3);
+            $$= MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS;
           }
-        |  RETRY_TIMEOUT '=' NUMBER
+        | BINARY_PROTOCOL
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT, $3);
+            $$= MEMCACHED_BEHAVIOR_BINARY_PROTOCOL;
           }
-        |  SERVER_FAILURE_LIMIT '=' NUMBER
+        | BUFFER_REQUESTS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT, $3);
+            $$= MEMCACHED_BEHAVIOR_BUFFER_REQUESTS;
           }
-        |  SND_TIMEOUT '=' NUMBER
+        | CACHE_LOOKUPS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SND_TIMEOUT, $3);
+            $$= MEMCACHED_BEHAVIOR_CACHE_LOOKUPS;
           }
-        |  SOCKET_RECV_SIZE '=' NUMBER
+        | _CORK
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, $3);
+            $$= MEMCACHED_BEHAVIOR_CORK;
           }
-        |  SOCKET_SEND_SIZE '=' NUMBER
+        | HASH_WITH_PREFIX_KEY
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, $3);
+            $$= MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY;
+          }
+        | KETAMA
+          {
+            $$= MEMCACHED_BEHAVIOR_KETAMA;
+          }
+        | KETAMA_WEIGHTED
+          {
+            $$= MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED;
+          }
+        | NOREPLY
+          {
+            $$= MEMCACHED_BEHAVIOR_NOREPLY;
+          }
+        |  RANDOMIZE_REPLICA_READ
+          {
+            $$= MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ;
           }
         |  SORT_HOSTS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SORT_HOSTS, true);
+            $$= MEMCACHED_BEHAVIOR_SORT_HOSTS;
           }
         |  SUPPORT_CAS
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, true);
+            $$= MEMCACHED_BEHAVIOR_SUPPORT_CAS;
           }
         |  _TCP_NODELAY
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, true);
+            $$= MEMCACHED_BEHAVIOR_TCP_NODELAY;
           }
         |  _TCP_KEEPALIVE
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_TCP_KEEPALIVE, true);
+            $$= MEMCACHED_BEHAVIOR_TCP_KEEPALIVE;
           }
         |  _TCP_KEEPIDLE
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_TCP_KEEPIDLE, true);
-          }
-        |  USER_DATA
-          {
+            $$= MEMCACHED_BEHAVIOR_TCP_KEEPIDLE;
           }
         |  USE_UDP
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_USE_UDP, true);
+            $$= MEMCACHED_BEHAVIOR_USE_UDP;
           }
         |  VERIFY_KEY
           {
-            memcached_behavior_set(parser->memc, MEMCACHED_BEHAVIOR_VERIFY_KEY, true);
+            $$= MEMCACHED_BEHAVIOR_VERIFY_KEY;
           }
-        ;
+
 
 server_list:
            server
           {
-            (void) memcached_server_add_parsed(parser->memc, $1.c_str, $1.length, $1.port, 0);
+            if ((context->rc= memcached_server_add_parsed(context->memc, $1.c_str, $1.length, $1.port, 0)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
         | server_list ',' server
           {
-            (void) memcached_server_add_parsed(parser->memc, $3.c_str, $3.length, $3.port, 0);
+            if ((context->rc= memcached_server_add_parsed(context->memc, $3.c_str, $3.length, $3.port, 0)) != MEMCACHED_SUCCESS)
+            {
+              parser_abort(context, NULL);;
+            }
           }
         ;
 
 server:
-          SERVER_WITH_PORT NUMBER
+          HOSTNAME_WITH_PORT NUMBER
           {
             $$.c_str= $1.c_str;
             $$.length= $1.length -1;
             $$.port= $2;
           }
-        | IDENTIFIER
+        | HOSTNAME
+          {
+            $$.c_str= $1.c_str;
+            $$.length= $1.length;
+            $$.port= MEMCACHED_DEFAULT_PORT;
+          }
+        | STRING /* a match can be against "localhost" which is just a string */
           {
             $$.c_str= $1.c_str;
             $$.length= $1.length;
@@ -359,6 +493,18 @@ hash:
           }
         ;
 
+string:
+          STRING
+          {
+            $$= $1;
+          }
+        | QUOTED_STRING
+          {
+            $$.c_str= $1.c_str +1;
+            $$.length= $1.length -2;
+          }
+        ;
+
 distribution:
           CONSISTENT
           {