git.m6w6.name Git - awesomized/libmemcached/blob - tests/function.c

   1 /*
   2   Sample test application.
   3 */
   4 #include <assert.h>
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8 #include <sys/time.h>
   9 #include <sys/types.h>
  10 #include <sys/stat.h>
  11 #include <unistd.h>
  12 #include <time.h>
  13 #include "server.h"
  14 #include "../clients/generator.h"
  15 #include "../clients/execute.h"
  16
  17 #ifndef INT64_MAX
  18 #define INT64_MAX LONG_MAX
  19 #endif
  20 #ifndef INT32_MAX
  21 #define INT32_MAX INT_MAX
  22 #endif
  23
  24
  25 #include "test.h"
  26
  27 #define GLOBAL_COUNT 100000
  28 #define GLOBAL2_COUNT 1000
  29 static uint32_t global_count;
  30
  31 static pairs_st *global_pairs;
  32 static char *global_keys[GLOBAL_COUNT];
  33 static size_t global_keys_length[GLOBAL_COUNT];
  34
  35 uint8_t init_test(memcached_st *not_used)
  36 {
  37   memcached_st memc;
  38
  39   (void)memcached_create(&memc);
  40   memcached_free(&memc);
  41
  42   return 0;
  43 }
  44
  45 uint8_t server_list_null_test(memcached_st *ptr)
  46 {
  47   memcached_server_st *server_list;
  48   memcached_return rc;
  49
  50   server_list= memcached_server_list_append(NULL, NULL, 0, NULL);
  51   assert(server_list == NULL);
  52
  53   server_list= memcached_server_list_append(NULL, "localhost", 0, NULL);
  54   assert(server_list == NULL);
  55
  56   server_list= memcached_server_list_append(NULL, NULL, 0, &rc);
  57   assert(server_list == NULL);
  58
  59   return 0;
  60 }
  61
  62 uint8_t server_sort_test(memcached_st *ptr)
  63 {
  64   memcached_server_st *server_list;
  65   memcached_return rc;
  66
  67   memcached_behavior_set(ptr, MEMCACHED_BEHAVIOR_SORT_HOSTS, 1);
  68
  69   server_list= memcached_server_list_append(NULL, "arg", 0, &rc);
  70   assert(server_list);
  71
  72   server_list= memcached_server_list_append(server_list, "localhost", 0, &rc);
  73   assert(server_list);
  74
  75   server_list= memcached_server_list_append(server_list, "c", 0, &rc);
  76   assert(server_list);
  77
  78   server_list= memcached_server_list_append(server_list, "abba", 0, &rc);
  79   assert(server_list);
  80
  81   free(server_list);
  82
  83   return 0;
  84 }
  85
  86 uint8_t allocation_test(memcached_st *not_used)
  87 {
  88   memcached_st *memc;
  89   memc= memcached_create(NULL);
  90   assert(memc);
  91   memcached_free(memc);
  92
  93   return 0;
  94 }
  95
  96 uint8_t clone_test(memcached_st *memc)
  97 {
  98   /* All null? */
  99   {
 100     memcached_st *clone;
 101     clone= memcached_clone(NULL, NULL);
 102     assert(clone);
 103     memcached_free(clone);
 104   }
 105
 106   /* Can we init from null? */
 107   {
 108     memcached_st *clone;
 109     clone= memcached_clone(NULL, memc);
 110     assert(clone);
 111     memcached_free(clone);
 112   }
 113
 114   /* Can we init from struct? */
 115   {
 116     memcached_st declared_clone;
 117     memcached_st *clone;
 118     clone= memcached_clone(&declared_clone, NULL);
 119     assert(clone);
 120     memcached_free(clone);
 121   }
 122
 123   /* Can we init from struct? */
 124   {
 125     memcached_st declared_clone;
 126     memcached_st *clone;
 127     clone= memcached_clone(&declared_clone, memc);
 128     assert(clone);
 129     memcached_free(clone);
 130   }
 131
 132   return 0;
 133 }
 134
 135 uint8_t connection_test(memcached_st *memc)
 136 {
 137   memcached_return rc;
 138
 139   rc= memcached_server_add(memc, "localhost", 0);
 140   assert(rc == MEMCACHED_SUCCESS);
 141
 142   return 0;
 143 }
 144
 145 uint8_t error_test(memcached_st *memc)
 146 {
 147   memcached_return rc;
 148
 149   for (rc= MEMCACHED_SUCCESS; rc < MEMCACHED_MAXIMUM_RETURN; rc++)
 150   {
 151     printf("Error %d -> %s\n", rc, memcached_strerror(memc, rc));
 152   }
 153
 154   return 0;
 155 }
 156
 157 uint8_t set_test(memcached_st *memc)
 158 {
 159   memcached_return rc;
 160   char *key= "foo";
 161   char *value= "when we sanitize";
 162
 163   rc= memcached_set(memc, key, strlen(key),
 164                     value, strlen(value),
 165                     (time_t)0, (uint32_t)0);
 166   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 167
 168   return 0;
 169 }
 170
 171 uint8_t append_test(memcached_st *memc)
 172 {
 173   memcached_return rc;
 174   char *key= "fig";
 175   char *value= "we";
 176   size_t value_length;
 177   uint32_t flags;
 178
 179   rc= memcached_flush(memc, 0);
 180   assert(rc == MEMCACHED_SUCCESS);
 181
 182   rc= memcached_set(memc, key, strlen(key),
 183                     value, strlen(value),
 184                     (time_t)0, (uint32_t)0);
 185   assert(rc == MEMCACHED_SUCCESS);
 186
 187   rc= memcached_append(memc, key, strlen(key),
 188                        " the", strlen(" the"),
 189                        (time_t)0, (uint32_t)0);
 190   assert(rc == MEMCACHED_SUCCESS);
 191
 192   rc= memcached_append(memc, key, strlen(key),
 193                        " people", strlen(" people"),
 194                        (time_t)0, (uint32_t)0);
 195   assert(rc == MEMCACHED_SUCCESS);
 196
 197   value= memcached_get(memc, key, strlen(key),
 198                        &value_length, &flags, &rc);
 199   assert(!memcmp(value, "we the people", strlen("we the people")));
 200   assert(strlen("we the people") == value_length);
 201   assert(rc == MEMCACHED_SUCCESS);
 202   free(value);
 203
 204   return 0;
 205 }
 206
 207 uint8_t append_binary_test(memcached_st *memc)
 208 {
 209   memcached_return rc;
 210   char *key= "numbers";
 211   unsigned int *store_ptr;
 212   unsigned int store_list[] = { 23, 56, 499, 98, 32847, 0 };
 213   char *value;
 214   size_t value_length;
 215   uint32_t flags;
 216   unsigned int x;
 217
 218   rc= memcached_flush(memc, 0);
 219   assert(rc == MEMCACHED_SUCCESS);
 220
 221   rc= memcached_set(memc,
 222                     key, strlen(key),
 223                     NULL, 0,
 224                     (time_t)0, (uint32_t)0);
 225   assert(rc == MEMCACHED_SUCCESS);
 226
 227   for (x= 0; store_list[x] ; x++)
 228   {
 229     rc= memcached_append(memc,
 230                          key, strlen(key),
 231                          (char *)&store_list[x], sizeof(unsigned int),
 232                          (time_t)0, (uint32_t)0);
 233     assert(rc == MEMCACHED_SUCCESS);
 234   }
 235
 236   value= memcached_get(memc, key, strlen(key),
 237                        &value_length, &flags, &rc);
 238   assert((value_length == (sizeof(unsigned int) * x)));
 239   assert(rc == MEMCACHED_SUCCESS);
 240
 241   store_ptr= (unsigned int *)value;
 242   x= 0;
 243   while ((size_t)store_ptr < (size_t)(value + value_length))
 244   {
 245     assert(*store_ptr == store_list[x++]);
 246     store_ptr++;
 247   }
 248   free(value);
 249
 250   return 0;
 251 }
 252
 253 uint8_t cas2_test(memcached_st *memc)
 254 {
 255   memcached_return rc;
 256   char *keys[]= {"fudge", "son", "food"};
 257   size_t key_length[]= {5, 3, 4};
 258   char *value= "we the people";
 259   size_t value_length= strlen("we the people");
 260   unsigned int x;
 261   memcached_result_st results_obj;
 262   memcached_result_st *results;
 263   unsigned int set= 1;
 264
 265   rc= memcached_flush(memc, 0);
 266   assert(rc == MEMCACHED_SUCCESS);
 267
 268   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, set);
 269
 270   for (x= 0; x < 3; x++)
 271   {
 272     rc= memcached_set(memc, keys[x], key_length[x],
 273                       keys[x], key_length[x],
 274                       (time_t)50, (uint32_t)9);
 275     assert(rc == MEMCACHED_SUCCESS);
 276   }
 277
 278   rc= memcached_mget(memc, keys, key_length, 3);
 279
 280   results= memcached_result_create(memc, &results_obj);
 281
 282   results= memcached_fetch_result(memc, &results_obj, &rc);
 283   assert(results);
 284   assert(results->cas);
 285   assert(rc == MEMCACHED_SUCCESS);
 286   WATCHPOINT_ASSERT(memcached_result_cas(results));
 287
 288   assert(!memcmp(value, "we the people", strlen("we the people")));
 289   assert(strlen("we the people") == value_length);
 290   assert(rc == MEMCACHED_SUCCESS);
 291
 292   memcached_result_free(&results_obj);
 293
 294   return 0;
 295 }
 296
 297 uint8_t cas_test(memcached_st *memc)
 298 {
 299   memcached_return rc;
 300   char *key= "fun";
 301   size_t key_length= strlen("fun");
 302   char *value= "we the people";
 303   size_t value_length= strlen("we the people");
 304   memcached_result_st results_obj;
 305   memcached_result_st *results;
 306   unsigned int set= 1;
 307
 308   rc= memcached_flush(memc, 0);
 309   assert(rc == MEMCACHED_SUCCESS);
 310
 311   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, set);
 312
 313   rc= memcached_set(memc, key, strlen(key),
 314                     value, strlen(value),
 315                     (time_t)0, (uint32_t)0);
 316   assert(rc == MEMCACHED_SUCCESS);
 317
 318   rc= memcached_mget(memc, &key, &key_length, 1);
 319
 320   results= memcached_result_create(memc, &results_obj);
 321
 322   results= memcached_fetch_result(memc, &results_obj, &rc);
 323   assert(results);
 324   assert(rc == MEMCACHED_SUCCESS);
 325   WATCHPOINT_ASSERT(memcached_result_cas(results));
 326
 327   assert(!memcmp(value, "we the people", strlen("we the people")));
 328   assert(strlen("we the people") == value_length);
 329   assert(rc == MEMCACHED_SUCCESS);
 330
 331   rc= memcached_cas(memc, key, key_length,
 332                     "change the value", strlen("change the value"),
 333                     0, 0, memcached_result_cas(results));
 334
 335   assert(rc == MEMCACHED_SUCCESS);
 336
 337   rc= memcached_cas(memc, key, key_length,
 338                     "change the value", strlen("change the value"),
 339                     0, 0, 23);
 340
 341   assert(rc == MEMCACHED_DATA_EXISTS);
 342
 343
 344   memcached_result_free(&results_obj);
 345
 346   return 0;
 347 }
 348
 349 uint8_t prepend_test(memcached_st *memc)
 350 {
 351   memcached_return rc;
 352   char *key= "fig";
 353   char *value= "people";
 354   size_t value_length;
 355   uint32_t flags;
 356
 357   rc= memcached_flush(memc, 0);
 358   assert(rc == MEMCACHED_SUCCESS);
 359
 360   rc= memcached_set(memc, key, strlen(key),
 361                     value, strlen(value),
 362                     (time_t)0, (uint32_t)0);
 363   assert(rc == MEMCACHED_SUCCESS);
 364
 365   rc= memcached_prepend(memc, key, strlen(key),
 366                        "the ", strlen("the "),
 367                        (time_t)0, (uint32_t)0);
 368   assert(rc == MEMCACHED_SUCCESS);
 369
 370   rc= memcached_prepend(memc, key, strlen(key),
 371                        "we ", strlen("we "),
 372                        (time_t)0, (uint32_t)0);
 373   assert(rc == MEMCACHED_SUCCESS);
 374
 375   value= memcached_get(memc, key, strlen(key),
 376                        &value_length, &flags, &rc);
 377   assert(!memcmp(value, "we the people", strlen("we the people")));
 378   assert(strlen("we the people") == value_length);
 379   assert(rc == MEMCACHED_SUCCESS);
 380   free(value);
 381
 382   return 0;
 383 }
 384
 385 /*
 386   Set the value, then quit to make sure it is flushed.
 387   Come back in and test that add fails.
 388 */
 389 uint8_t add_test(memcached_st *memc)
 390 {
 391   memcached_return rc;
 392   char *key= "foo";
 393   char *value= "when we sanitize";
 394   unsigned long long setting_value;
 395
 396   setting_value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
 397
 398   rc= memcached_set(memc, key, strlen(key),
 399                     value, strlen(value),
 400                     (time_t)0, (uint32_t)0);
 401   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 402   memcached_quit(memc);
 403   rc= memcached_add(memc, key, strlen(key),
 404                     value, strlen(value),
 405                     (time_t)0, (uint32_t)0);
 406
 407   /* Too many broken OS'es have broken loopback in async, so we can't be sure of the result */
 408   if (setting_value)
 409     assert(rc == MEMCACHED_NOTSTORED || MEMCACHED_STORED);
 410   else
 411     assert(rc == MEMCACHED_NOTSTORED);
 412
 413   return 0;
 414 }
 415
 416 uint8_t add_wrapper(memcached_st *memc)
 417 {
 418   unsigned int x;
 419
 420   for (x= 0; x < 10000; x++)
 421     add_test(memc);
 422
 423   return 0;
 424 }
 425
 426 uint8_t replace_test(memcached_st *memc)
 427 {
 428   memcached_return rc;
 429   char *key= "foo";
 430   char *value= "when we sanitize";
 431   char *original= "first we insert some data";
 432
 433   rc= memcached_set(memc, key, strlen(key),
 434                     original, strlen(original),
 435                     (time_t)0, (uint32_t)0);
 436   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 437
 438   rc= memcached_replace(memc, key, strlen(key),
 439                     value, strlen(value),
 440                     (time_t)0, (uint32_t)0);
 441   assert(rc == MEMCACHED_SUCCESS);
 442
 443   return 0;
 444 }
 445
 446 uint8_t delete_test(memcached_st *memc)
 447 {
 448   memcached_return rc;
 449   char *key= "foo";
 450   char *value= "when we sanitize";
 451
 452   rc= memcached_set(memc, key, strlen(key),
 453                     value, strlen(value),
 454                     (time_t)0, (uint32_t)0);
 455   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 456
 457   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 458   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 459
 460   return 0;
 461 }
 462
 463 uint8_t flush_test(memcached_st *memc)
 464 {
 465   memcached_return rc;
 466
 467   rc= memcached_flush(memc, 0);
 468   assert(rc == MEMCACHED_SUCCESS);
 469
 470   return 0;
 471 }
 472
 473 memcached_return server_function(memcached_st *ptr, memcached_server_st *server, void *context)
 474 {
 475   /* Do Nothing */
 476
 477   return MEMCACHED_SUCCESS;
 478 }
 479
 480 uint8_t memcached_server_cursor_test(memcached_st *memc)
 481 {
 482   char *context= "foo bad";
 483   memcached_server_function callbacks[1];
 484
 485   callbacks[0]= server_function;
 486   memcached_server_cursor(memc, callbacks, context,  1);
 487
 488   return 0;
 489 }
 490
 491 uint8_t bad_key_test(memcached_st *memc)
 492 {
 493   memcached_return rc;
 494   char *key= "foo bad";
 495   char *string;
 496   size_t string_length;
 497   uint32_t flags;
 498   memcached_st *clone;
 499   unsigned int set= 1;
 500
 501   clone= memcached_clone(NULL, memc);
 502   assert(clone);
 503
 504   (void)memcached_behavior_set(clone, MEMCACHED_BEHAVIOR_VERIFY_KEY, set);
 505
 506   string= memcached_get(clone, key, strlen(key),
 507                         &string_length, &flags, &rc);
 508   assert(rc == MEMCACHED_BAD_KEY_PROVIDED);
 509   assert(string_length ==  0);
 510   assert(!string);
 511
 512   set= 0;
 513   (void)memcached_behavior_set(clone, MEMCACHED_BEHAVIOR_VERIFY_KEY, set);
 514   string= memcached_get(clone, key, strlen(key),
 515                         &string_length, &flags, &rc);
 516   assert(rc == MEMCACHED_NOTFOUND);
 517   assert(string_length ==  0);
 518   assert(!string);
 519
 520   memcached_free(clone);
 521
 522   return 0;
 523 }
 524
 525 #define READ_THROUGH_VALUE "set for me"
 526 memcached_return read_through_trigger(memcached_st *memc,
 527                                       char *key, size_t key_length,
 528                                       memcached_result_st *result)
 529 {
 530
 531   return memcached_result_set_value(result, READ_THROUGH_VALUE, strlen(READ_THROUGH_VALUE));
 532 }
 533
 534 uint8_t read_through(memcached_st *memc)
 535 {
 536   memcached_return rc;
 537   char *key= "foo";
 538   char *string;
 539   size_t string_length;
 540   uint32_t flags;
 541
 542   string= memcached_get(memc, key, strlen(key),
 543                         &string_length, &flags, &rc);
 544
 545   assert(rc == MEMCACHED_NOTFOUND);
 546   assert(string_length ==  0);
 547   assert(!string);
 548
 549   rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_GET_FAILURE, read_through_trigger);
 550   assert(rc == MEMCACHED_SUCCESS);
 551
 552   string= memcached_get(memc, key, strlen(key),
 553                         &string_length, &flags, &rc);
 554
 555   assert(rc == MEMCACHED_SUCCESS);
 556   assert(string_length ==  strlen(READ_THROUGH_VALUE));
 557   assert(!strcmp(READ_THROUGH_VALUE, string));
 558
 559   string= memcached_get(memc, key, strlen(key),
 560                         &string_length, &flags, &rc);
 561
 562   assert(rc == MEMCACHED_SUCCESS);
 563   assert(string_length ==  strlen(READ_THROUGH_VALUE));
 564   assert(!strcmp(READ_THROUGH_VALUE, string));
 565
 566   return 0;
 567 }
 568
 569 uint8_t get_test(memcached_st *memc)
 570 {
 571   memcached_return rc;
 572   char *key= "foo";
 573   char *string;
 574   size_t string_length;
 575   uint32_t flags;
 576
 577   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 578   assert(rc == MEMCACHED_BUFFERED || rc == MEMCACHED_NOTFOUND);
 579
 580   string= memcached_get(memc, key, strlen(key),
 581                         &string_length, &flags, &rc);
 582
 583   assert(rc == MEMCACHED_NOTFOUND);
 584   assert(string_length ==  0);
 585   assert(!string);
 586
 587   return 0;
 588 }
 589
 590 uint8_t get_test2(memcached_st *memc)
 591 {
 592   memcached_return rc;
 593   char *key= "foo";
 594   char *value= "when we sanitize";
 595   char *string;
 596   size_t string_length;
 597   uint32_t flags;
 598
 599   rc= memcached_set(memc, key, strlen(key),
 600                     value, strlen(value),
 601                     (time_t)0, (uint32_t)0);
 602   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 603
 604   string= memcached_get(memc, key, strlen(key),
 605                         &string_length, &flags, &rc);
 606
 607   assert(string);
 608   assert(rc == MEMCACHED_SUCCESS);
 609   assert(string_length == strlen(value));
 610   assert(!memcmp(string, value, string_length));
 611
 612   free(string);
 613
 614   return 0;
 615 }
 616
 617 uint8_t set_test2(memcached_st *memc)
 618 {
 619   memcached_return rc;
 620   char *key= "foo";
 621   char *value= "train in the brain";
 622   size_t value_length= strlen(value);
 623   unsigned int x;
 624
 625   for (x= 0; x < 10; x++)
 626   {
 627     rc= memcached_set(memc, key, strlen(key),
 628                       value, value_length,
 629                       (time_t)0, (uint32_t)0);
 630     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 631   }
 632
 633   return 0;
 634 }
 635
 636 uint8_t set_test3(memcached_st *memc)
 637 {
 638   memcached_return rc;
 639   char *key= "foo";
 640   char *value;
 641   size_t value_length= 8191;
 642   unsigned int x;
 643
 644   value = (char*)malloc(value_length);
 645   assert(value);
 646
 647   for (x= 0; x < value_length; x++)
 648     value[x] = (char) (x % 127);
 649
 650   for (x= 0; x < 1; x++)
 651   {
 652     rc= memcached_set(memc, key, strlen(key),
 653                       value, value_length,
 654                       (time_t)0, (uint32_t)0);
 655     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 656   }
 657
 658   free(value);
 659
 660   return 0;
 661 }
 662
 663 uint8_t get_test3(memcached_st *memc)
 664 {
 665   memcached_return rc;
 666   char *key= "foo";
 667   char *value;
 668   size_t value_length= 8191;
 669   char *string;
 670   size_t string_length;
 671   uint32_t flags;
 672   int x;
 673
 674   value = (char*)malloc(value_length);
 675   assert(value);
 676
 677   for (x= 0; x < value_length; x++)
 678     value[x] = (char) (x % 127);
 679
 680   rc= memcached_set(memc, key, strlen(key),
 681                     value, value_length,
 682                     (time_t)0, (uint32_t)0);
 683   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 684
 685   string= memcached_get(memc, key, strlen(key),
 686                         &string_length, &flags, &rc);
 687
 688   assert(rc == MEMCACHED_SUCCESS);
 689   assert(string);
 690   assert(string_length == value_length);
 691   assert(!memcmp(string, value, string_length));
 692
 693   free(string);
 694   free(value);
 695
 696   return 0;
 697 }
 698
 699 uint8_t get_test4(memcached_st *memc)
 700 {
 701   memcached_return rc;
 702   char *key= "foo";
 703   char *value;
 704   size_t value_length= 8191;
 705   char *string;
 706   size_t string_length;
 707   uint32_t flags;
 708   int x;
 709
 710   value = (char*)malloc(value_length);
 711   assert(value);
 712
 713   for (x= 0; x < value_length; x++)
 714     value[x] = (char) (x % 127);
 715
 716   rc= memcached_set(memc, key, strlen(key),
 717                     value, value_length,
 718                     (time_t)0, (uint32_t)0);
 719   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 720
 721   for (x= 0; x < 10; x++)
 722   {
 723     string= memcached_get(memc, key, strlen(key),
 724                           &string_length, &flags, &rc);
 725
 726     assert(rc == MEMCACHED_SUCCESS);
 727     assert(string);
 728     assert(string_length == value_length);
 729     assert(!memcmp(string, value, string_length));
 730     free(string);
 731   }
 732
 733   free(value);
 734
 735   return 0;
 736 }
 737
 738 /* Do not copy the style of this code, I just access hosts to testthis function */
 739 uint8_t stats_servername_test(memcached_st *memc)
 740 {
 741   memcached_return rc;
 742   memcached_stat_st stat;
 743   rc= memcached_stat_servername(&stat, NULL,
 744                                  memc->hosts[0].hostname,
 745                                  memc->hosts[0].port);
 746
 747   return 0;
 748 }
 749
 750 uint8_t increment_test(memcached_st *memc)
 751 {
 752   uint64_t new_number;
 753   memcached_return rc;
 754   char *key= "number";
 755   char *value= "0";
 756
 757   rc= memcached_set(memc, key, strlen(key),
 758                     value, strlen(value),
 759                     (time_t)0, (uint32_t)0);
 760   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 761
 762   rc= memcached_increment(memc, key, strlen(key),
 763                           1, &new_number);
 764   assert(rc == MEMCACHED_SUCCESS);
 765   assert(new_number == 1);
 766
 767   rc= memcached_increment(memc, key, strlen(key),
 768                           1, &new_number);
 769   assert(rc == MEMCACHED_SUCCESS);
 770   assert(new_number == 2);
 771
 772   return 0;
 773 }
 774
 775 uint8_t decrement_test(memcached_st *memc)
 776 {
 777   uint64_t new_number;
 778   memcached_return rc;
 779   char *key= "number";
 780   char *value= "3";
 781
 782   rc= memcached_set(memc, key, strlen(key),
 783                     value, strlen(value),
 784                     (time_t)0, (uint32_t)0);
 785   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 786
 787   rc= memcached_decrement(memc, key, strlen(key),
 788                           1, &new_number);
 789   assert(rc == MEMCACHED_SUCCESS);
 790   assert(new_number == 2);
 791
 792   rc= memcached_decrement(memc, key, strlen(key),
 793                           1, &new_number);
 794   assert(rc == MEMCACHED_SUCCESS);
 795   assert(new_number == 1);
 796
 797   return 0;
 798 }
 799
 800 uint8_t quit_test(memcached_st *memc)
 801 {
 802   memcached_return rc;
 803   char *key= "fudge";
 804   char *value= "sanford and sun";
 805
 806   rc= memcached_set(memc, key, strlen(key),
 807                     value, strlen(value),
 808                     (time_t)10, (uint32_t)3);
 809   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 810   memcached_quit(memc);
 811
 812   rc= memcached_set(memc, key, strlen(key),
 813                     value, strlen(value),
 814                     (time_t)50, (uint32_t)9);
 815   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 816
 817   return 0;
 818 }
 819
 820 uint8_t mget_result_test(memcached_st *memc)
 821 {
 822   memcached_return rc;
 823   char *keys[]= {"fudge", "son", "food"};
 824   size_t key_length[]= {5, 3, 4};
 825   unsigned int x;
 826
 827   memcached_result_st results_obj;
 828   memcached_result_st *results;
 829
 830   results= memcached_result_create(memc, &results_obj);
 831   assert(results);
 832   assert(&results_obj == results);
 833
 834   /* We need to empty the server before continueing test */
 835   rc= memcached_flush(memc, 0);
 836   assert(rc == MEMCACHED_SUCCESS);
 837
 838   rc= memcached_mget(memc, keys, key_length, 3);
 839   assert(rc == MEMCACHED_SUCCESS);
 840
 841   while ((results= memcached_fetch_result(memc, &results_obj, &rc)) != NULL)
 842   {
 843     assert(results);
 844   }
 845
 846   while ((results= memcached_fetch_result(memc, &results_obj, &rc)) != NULL)
 847   assert(!results);
 848   assert(rc == MEMCACHED_END);
 849
 850   for (x= 0; x < 3; x++)
 851   {
 852     rc= memcached_set(memc, keys[x], key_length[x],
 853                       keys[x], key_length[x],
 854                       (time_t)50, (uint32_t)9);
 855     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 856   }
 857
 858   rc= memcached_mget(memc, keys, key_length, 3);
 859   assert(rc == MEMCACHED_SUCCESS);
 860
 861   while ((results= memcached_fetch_result(memc, &results_obj, &rc)))
 862   {
 863     assert(results);
 864     assert(&results_obj == results);
 865     assert(rc == MEMCACHED_SUCCESS);
 866     assert(memcached_result_key_length(results) == memcached_result_length(results));
 867     assert(!memcmp(memcached_result_key_value(results),
 868                    memcached_result_value(results),
 869                    memcached_result_length(results)));
 870   }
 871
 872   memcached_result_free(&results_obj);
 873
 874   return 0;
 875 }
 876
 877 uint8_t mget_result_alloc_test(memcached_st *memc)
 878 {
 879   memcached_return rc;
 880   char *keys[]= {"fudge", "son", "food"};
 881   size_t key_length[]= {5, 3, 4};
 882   unsigned int x;
 883
 884   memcached_result_st *results;
 885
 886   /* We need to empty the server before continueing test */
 887   rc= memcached_flush(memc, 0);
 888   assert(rc == MEMCACHED_SUCCESS);
 889
 890   rc= memcached_mget(memc, keys, key_length, 3);
 891   assert(rc == MEMCACHED_SUCCESS);
 892
 893   while ((results= memcached_fetch_result(memc, NULL, &rc)) != NULL)
 894   {
 895     assert(results);
 896   }
 897   assert(!results);
 898   assert(rc == MEMCACHED_END);
 899
 900   for (x= 0; x < 3; x++)
 901   {
 902     rc= memcached_set(memc, keys[x], key_length[x],
 903                       keys[x], key_length[x],
 904                       (time_t)50, (uint32_t)9);
 905     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 906   }
 907
 908   rc= memcached_mget(memc, keys, key_length, 3);
 909   assert(rc == MEMCACHED_SUCCESS);
 910
 911   x= 0;
 912   while ((results= memcached_fetch_result(memc, NULL, &rc)))
 913   {
 914     assert(results);
 915     assert(rc == MEMCACHED_SUCCESS);
 916     assert(memcached_result_key_length(results) == memcached_result_length(results));
 917     assert(!memcmp(memcached_result_key_value(results),
 918                    memcached_result_value(results),
 919                    memcached_result_length(results)));
 920     memcached_result_free(results);
 921     x++;
 922   }
 923
 924   return 0;
 925 }
 926
 927 /* Count the results */
 928 unsigned int callback_counter(memcached_st *ptr, memcached_result_st *result, void *context)
 929 {
 930   unsigned int *counter= (unsigned int *)context;
 931
 932   *counter= *counter + 1;
 933
 934   return 0;
 935 }
 936
 937 uint8_t mget_result_function(memcached_st *memc)
 938 {
 939   memcached_return rc;
 940   char *keys[]= {"fudge", "son", "food"};
 941   size_t key_length[]= {5, 3, 4};
 942   unsigned int x;
 943   unsigned int counter;
 944   memcached_execute_function callbacks[1];
 945
 946   /* We need to empty the server before continueing test */
 947   rc= memcached_flush(memc, 0);
 948   for (x= 0; x < 3; x++)
 949   {
 950     rc= memcached_set(memc, keys[x], key_length[x],
 951                       keys[x], key_length[x],
 952                       (time_t)50, (uint32_t)9);
 953     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 954   }
 955
 956   rc= memcached_mget(memc, keys, key_length, 3);
 957   assert(rc == MEMCACHED_SUCCESS);
 958
 959   callbacks[0]= &callback_counter;
 960   counter= 0;
 961   rc= memcached_fetch_execute(memc, callbacks, (void *)&counter, 1);
 962
 963   assert(counter == 3);
 964
 965   return 0;
 966 }
 967
 968 uint8_t mget_test(memcached_st *memc)
 969 {
 970   memcached_return rc;
 971   char *keys[]= {"fudge", "son", "food"};
 972   size_t key_length[]= {5, 3, 4};
 973   unsigned int x;
 974   uint32_t flags;
 975
 976   char return_key[MEMCACHED_MAX_KEY];
 977   size_t return_key_length;
 978   char *return_value;
 979   size_t return_value_length;
 980
 981   /* We need to empty the server before continueing test */
 982   rc= memcached_flush(memc, 0);
 983   assert(rc == MEMCACHED_SUCCESS);
 984
 985   rc= memcached_mget(memc, keys, key_length, 3);
 986   assert(rc == MEMCACHED_SUCCESS);
 987
 988   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 989                       &return_value_length, &flags, &rc)) != NULL)
 990   {
 991     assert(return_value);
 992   }
 993   assert(!return_value);
 994   assert(return_value_length == 0);
 995   assert(rc == MEMCACHED_END);
 996
 997   for (x= 0; x < 3; x++)
 998   {
 999     rc= memcached_set(memc, keys[x], key_length[x],
1000                       keys[x], key_length[x],
1001                       (time_t)50, (uint32_t)9);
1002     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
1003   }
1004
1005   rc= memcached_mget(memc, keys, key_length, 3);
1006   assert(rc == MEMCACHED_SUCCESS);
1007
1008   x= 0;
1009   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1010                                         &return_value_length, &flags, &rc)))
1011   {
1012     assert(return_value);
1013     assert(rc == MEMCACHED_SUCCESS);
1014     assert(return_key_length == return_value_length);
1015     assert(!memcmp(return_value, return_key, return_value_length));
1016     free(return_value);
1017     x++;
1018   }
1019
1020   return 0;
1021 }
1022
1023 uint8_t get_stats_keys(memcached_st *memc)
1024 {
1025  char **list;
1026  char **ptr;
1027  memcached_stat_st stat;
1028  memcached_return rc;
1029
1030  list= memcached_stat_get_keys(memc, &stat, &rc);
1031  assert(rc == MEMCACHED_SUCCESS);
1032  for (ptr= list; *ptr; ptr++)
1033    assert(*ptr);
1034  fflush(stdout);
1035
1036  free(list);
1037
1038  return 0;
1039 }
1040
1041 uint8_t version_string_test(memcached_st *memc)
1042 {
1043   const char *version_string;
1044
1045   version_string= memcached_lib_version();
1046
1047   assert(!strcmp(version_string, LIBMEMCACHED_VERSION_STRING));
1048
1049   return 0;
1050 }
1051
1052 uint8_t get_stats(memcached_st *memc)
1053 {
1054  unsigned int x;
1055  char **list;
1056  char **ptr;
1057  memcached_return rc;
1058  memcached_stat_st *stat;
1059
1060  stat= memcached_stat(memc, NULL, &rc);
1061  assert(rc == MEMCACHED_SUCCESS);
1062
1063  assert(rc == MEMCACHED_SUCCESS);
1064  assert(stat);
1065
1066  for (x= 0; x < memcached_server_count(memc); x++)
1067  {
1068    list= memcached_stat_get_keys(memc, stat+x, &rc);
1069    assert(rc == MEMCACHED_SUCCESS);
1070    for (ptr= list; *ptr; ptr++);
1071
1072    free(list);
1073  }
1074
1075  memcached_stat_free(NULL, stat);
1076
1077   return 0;
1078 }
1079
1080 uint8_t add_host_test(memcached_st *memc)
1081 {
1082   unsigned int x;
1083   memcached_server_st *servers;
1084   memcached_return rc;
1085   char servername[]= "0.example.com";
1086
1087   servers= memcached_server_list_append(NULL, servername, 400, &rc);
1088   assert(servers);
1089   assert(1 == memcached_server_list_count(servers));
1090
1091   for (x= 2; x < 20; x++)
1092   {
1093     char buffer[SMALL_STRING_LEN];
1094
1095     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
1096     servers= memcached_server_list_append(servers, buffer, 401,
1097                                      &rc);
1098     assert(rc == MEMCACHED_SUCCESS);
1099     assert(x == memcached_server_list_count(servers));
1100   }
1101
1102   rc= memcached_server_push(memc, servers);
1103   assert(rc == MEMCACHED_SUCCESS);
1104   rc= memcached_server_push(memc, servers);
1105   assert(rc == MEMCACHED_SUCCESS);
1106
1107   memcached_server_list_free(servers);
1108
1109   return 0;
1110 }
1111
1112 memcached_return clone_test_callback(memcached_st *parent, memcached_st *clone)
1113 {
1114   return MEMCACHED_SUCCESS;
1115 }
1116
1117 memcached_return cleanup_test_callback(memcached_st *ptr)
1118 {
1119   return MEMCACHED_SUCCESS;
1120 }
1121
1122 uint8_t callback_test(memcached_st *memc)
1123 {
1124   /* Test User Data */
1125   {
1126     int x= 5;
1127     int *test_ptr;
1128     memcached_return rc;
1129
1130     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_USER_DATA, &x);
1131     assert(rc == MEMCACHED_SUCCESS);
1132     test_ptr= (int *)memcached_callback_get(memc, MEMCACHED_CALLBACK_USER_DATA, &rc);
1133     assert(*test_ptr == x);
1134   }
1135
1136   /* Test Clone Callback */
1137   {
1138     memcached_clone_func temp_function;
1139     memcached_return rc;
1140
1141     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, clone_test_callback);
1142     assert(rc == MEMCACHED_SUCCESS);
1143     temp_function= (memcached_clone_func)memcached_callback_get(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, &rc);
1144     assert(temp_function == clone_test_callback);
1145   }
1146
1147   /* Test Cleanup Callback */
1148   {
1149     memcached_cleanup_func temp_function;
1150     memcached_return rc;
1151
1152     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, cleanup_test_callback);
1153     assert(rc == MEMCACHED_SUCCESS);
1154     temp_function= (memcached_cleanup_func)memcached_callback_get(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, &rc);
1155     assert(temp_function == cleanup_test_callback);
1156   }
1157
1158   return 0;
1159 }
1160
1161 /* We don't test the behavior itself, we test the switches */
1162 uint8_t behavior_test(memcached_st *memc)
1163 {
1164   unsigned long long value;
1165   unsigned int set= 1;
1166
1167   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, set);
1168   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
1169   assert(value == 1);
1170
1171   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, set);
1172   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
1173   assert(value == 1);
1174
1175   set= MEMCACHED_HASH_MD5;
1176   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, set);
1177   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1178   assert(value == MEMCACHED_HASH_MD5);
1179
1180   set= 0;
1181
1182   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, set);
1183   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
1184   assert(value == 0);
1185
1186   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, set);
1187   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
1188   assert(value == 0);
1189
1190   set= MEMCACHED_HASH_DEFAULT;
1191   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, set);
1192   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1193   assert(value == MEMCACHED_HASH_DEFAULT);
1194
1195   set= MEMCACHED_HASH_CRC;
1196   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, set);
1197   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1198   assert(value == MEMCACHED_HASH_CRC);
1199
1200   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1201   assert(value > 0);
1202
1203   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1204   assert(value > 0);
1205
1206   return 0;
1207 }
1208
1209 /* Test case provided by Cal Haldenbrand */
1210 uint8_t user_supplied_bug1(memcached_st *memc)
1211 {
1212   unsigned int setter= 1;
1213   unsigned int x;
1214
1215   unsigned long long total= 0;
1216   int size= 0;
1217   char key[10];
1218   char randomstuff[6 * 1024];
1219   memcached_return rc;
1220
1221   memset(randomstuff, 0, 6 * 1024);
1222
1223   /* We just keep looking at the same values over and over */
1224   srandom(10);
1225
1226   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, setter);
1227   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, setter);
1228
1229
1230   /* add key */
1231   for (x= 0 ; total < 20 * 1024576 ; x++ )
1232   {
1233     unsigned int j= 0;
1234
1235     size= (rand() % ( 5 * 1024 ) ) + 400;
1236     memset(randomstuff, 0, 6 * 1024);
1237     assert(size < 6 * 1024); /* Being safe here */
1238
1239     for (j= 0 ; j < size ;j++)
1240       randomstuff[j] = (char) (rand() % 26) + 97;
1241
1242     total += size;
1243     sprintf(key, "%d", x);
1244     rc = memcached_set(memc, key, strlen(key),
1245                        randomstuff, strlen(randomstuff), 10, 0);
1246     /* If we fail, lets try again */
1247     if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_BUFFERED)
1248       rc = memcached_set(memc, key, strlen(key),
1249                          randomstuff, strlen(randomstuff), 10, 0);
1250     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
1251   }
1252
1253   return 0;
1254 }
1255
1256 /* Test case provided by Cal Haldenbrand */
1257 uint8_t user_supplied_bug2(memcached_st *memc)
1258 {
1259   int errors;
1260   unsigned int setter;
1261   unsigned int x;
1262   unsigned long long total;
1263
1264   setter= 1;
1265   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, setter);
1266   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, setter);
1267 #ifdef NOT_YET
1268   setter = 20 * 1024576;
1269   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, setter);
1270   setter = 20 * 1024576;
1271   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, setter);
1272   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1273   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1274
1275   for (x= 0, errors= 0, total= 0 ; total < 20 * 1024576 ; x++)
1276 #endif
1277
1278   for (x= 0, errors= 0, total= 0 ; total < 24576 ; x++)
1279   {
1280     memcached_return rc= MEMCACHED_SUCCESS;
1281     char buffer[SMALL_STRING_LEN];
1282     uint32_t flags= 0;
1283     size_t val_len= 0;
1284     char *getval;
1285
1286     memset(buffer, 0, SMALL_STRING_LEN);
1287
1288     snprintf(buffer, SMALL_STRING_LEN, "%u", x);
1289     getval= memcached_get(memc, buffer, strlen(buffer),
1290                            &val_len, &flags, &rc);
1291     if (rc != MEMCACHED_SUCCESS)
1292     {
1293       if (rc == MEMCACHED_NOTFOUND)
1294         errors++;
1295       else
1296       {
1297         WATCHPOINT_ERROR(rc);
1298         assert(0);
1299       }
1300
1301       continue;
1302     }
1303     total+= val_len;
1304     errors= 0;
1305     free(getval);
1306   }
1307
1308   return 0;
1309 }
1310
1311 /* Do a large mget() over all the keys we think exist */
1312 #define KEY_COUNT 3000 // * 1024576
1313 uint8_t user_supplied_bug3(memcached_st *memc)
1314 {
1315   memcached_return rc;
1316   unsigned int setter;
1317   unsigned int x;
1318   char **keys;
1319   size_t key_lengths[KEY_COUNT];
1320
1321   setter= 1;
1322   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, setter);
1323   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, setter);
1324 #ifdef NOT_YET
1325   setter = 20 * 1024576;
1326   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, setter);
1327   setter = 20 * 1024576;
1328   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, setter);
1329   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1330   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1331 #endif
1332
1333   keys= (char **)malloc(sizeof(char *) * KEY_COUNT);
1334   assert(keys);
1335   memset(keys, 0, (sizeof(char *) * KEY_COUNT));
1336   for (x= 0; x < KEY_COUNT; x++)
1337   {
1338     char buffer[30];
1339
1340     snprintf(buffer, 30, "%u", x);
1341     keys[x]= strdup(buffer);
1342     key_lengths[x]= strlen(keys[x]);
1343   }
1344
1345   rc= memcached_mget(memc, keys, key_lengths, KEY_COUNT);
1346   assert(rc == MEMCACHED_SUCCESS);
1347
1348   /* Turn this into a help function */
1349   {
1350     char return_key[MEMCACHED_MAX_KEY];
1351     size_t return_key_length;
1352     char *return_value;
1353     size_t return_value_length;
1354     uint32_t flags;
1355
1356     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1357                                           &return_value_length, &flags, &rc)))
1358     {
1359       assert(return_value);
1360       assert(rc == MEMCACHED_SUCCESS);
1361       free(return_value);
1362     }
1363   }
1364
1365   for (x= 0; x < KEY_COUNT; x++)
1366     free(keys[x]);
1367   free(keys);
1368
1369   return 0;
1370 }
1371
1372 /* Make sure we behave properly if server list has no values */
1373 uint8_t user_supplied_bug4(memcached_st *memc)
1374 {
1375   memcached_return rc;
1376   char *keys[]= {"fudge", "son", "food"};
1377   size_t key_length[]= {5, 3, 4};
1378   unsigned int x;
1379   uint32_t flags;
1380   char return_key[MEMCACHED_MAX_KEY];
1381   size_t return_key_length;
1382   char *return_value;
1383   size_t return_value_length;
1384
1385   /* Here we free everything before running a bunch of mget tests */
1386   {
1387     memcached_server_list_free(memc->hosts);
1388     memc->hosts= NULL;
1389     memc->number_of_hosts= 0;
1390   }
1391
1392
1393   /* We need to empty the server before continueing test */
1394   rc= memcached_flush(memc, 0);
1395   assert(rc == MEMCACHED_NO_SERVERS);
1396
1397   rc= memcached_mget(memc, keys, key_length, 3);
1398   assert(rc == MEMCACHED_NO_SERVERS);
1399
1400   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1401                       &return_value_length, &flags, &rc)) != NULL)
1402   {
1403     assert(return_value);
1404   }
1405   assert(!return_value);
1406   assert(return_value_length == 0);
1407   assert(rc == MEMCACHED_NO_SERVERS);
1408
1409   for (x= 0; x < 3; x++)
1410   {
1411     rc= memcached_set(memc, keys[x], key_length[x],
1412                       keys[x], key_length[x],
1413                       (time_t)50, (uint32_t)9);
1414     assert(rc == MEMCACHED_NO_SERVERS);
1415   }
1416
1417   rc= memcached_mget(memc, keys, key_length, 3);
1418   assert(rc == MEMCACHED_NO_SERVERS);
1419
1420   x= 0;
1421   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1422                                         &return_value_length, &flags, &rc)))
1423   {
1424     assert(return_value);
1425     assert(rc == MEMCACHED_SUCCESS);
1426     assert(return_key_length == return_value_length);
1427     assert(!memcmp(return_value, return_key, return_value_length));
1428     free(return_value);
1429     x++;
1430   }
1431
1432   return 0;
1433 }
1434
1435 #define VALUE_SIZE_BUG5 1048064
1436 uint8_t user_supplied_bug5(memcached_st *memc)
1437 {
1438   memcached_return rc;
1439   char *keys[]= {"036790384900", "036790384902", "036790384904", "036790384906"};
1440   size_t key_length[]=  {strlen("036790384900"), strlen("036790384902"), strlen("036790384904"), strlen("036790384906")};
1441   char return_key[MEMCACHED_MAX_KEY];
1442   size_t return_key_length;
1443   char *value;
1444   size_t value_length;
1445   uint32_t flags;
1446   unsigned int count;
1447   unsigned int x;
1448   char insert_data[VALUE_SIZE_BUG5];
1449
1450   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1451     insert_data[x]= rand();
1452
1453   memcached_flush(memc, 0);
1454   value= memcached_get(memc, keys[0], key_length[0],
1455                         &value_length, &flags, &rc);
1456   assert(value == NULL);
1457   rc= memcached_mget(memc, keys, key_length, 4);
1458
1459   count= 0;
1460   while ((value= memcached_fetch(memc, return_key, &return_key_length,
1461                                         &value_length, &flags, &rc)))
1462     count++;
1463   assert(count == 0);
1464
1465   for (x= 0; x < 4; x++)
1466   {
1467     rc= memcached_set(memc, keys[x], key_length[x],
1468                       insert_data, VALUE_SIZE_BUG5,
1469                       (time_t)0, (uint32_t)0);
1470     assert(rc == MEMCACHED_SUCCESS);
1471   }
1472
1473   for (x= 0; x < 10; x++)
1474   {
1475     value= memcached_get(memc, keys[0], key_length[0],
1476                          &value_length, &flags, &rc);
1477     assert(value);
1478     free(value);
1479
1480     rc= memcached_mget(memc, keys, key_length, 4);
1481     count= 0;
1482     while ((value= memcached_fetch(memc, return_key, &return_key_length,
1483                                           &value_length, &flags, &rc)))
1484     {
1485       count++;
1486       free(value);
1487     }
1488     assert(count == 4);
1489   }
1490
1491   return 0;
1492 }
1493
1494 uint8_t user_supplied_bug6(memcached_st *memc)
1495 {
1496   memcached_return rc;
1497   char *keys[]= {"036790384900", "036790384902", "036790384904", "036790384906"};
1498   size_t key_length[]=  {strlen("036790384900"), strlen("036790384902"), strlen("036790384904"), strlen("036790384906")};
1499   char return_key[MEMCACHED_MAX_KEY];
1500   size_t return_key_length;
1501   char *value;
1502   size_t value_length;
1503   uint32_t flags;
1504   unsigned int count;
1505   unsigned int x;
1506   char insert_data[VALUE_SIZE_BUG5];
1507
1508   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1509     insert_data[x]= rand();
1510
1511   memcached_flush(memc, 0);
1512   value= memcached_get(memc, keys[0], key_length[0],
1513                         &value_length, &flags, &rc);
1514   assert(value == NULL);
1515   assert(rc == MEMCACHED_NOTFOUND);
1516   rc= memcached_mget(memc, keys, key_length, 4);
1517   assert(rc == MEMCACHED_SUCCESS);
1518
1519   count= 0;
1520   while ((value= memcached_fetch(memc, return_key, &return_key_length,
1521                                         &value_length, &flags, &rc)))
1522     count++;
1523   assert(count == 0);
1524   assert(rc == MEMCACHED_END);
1525
1526   for (x= 0; x < 4; x++)
1527   {
1528     rc= memcached_set(memc, keys[x], key_length[x],
1529                       insert_data, VALUE_SIZE_BUG5,
1530                       (time_t)0, (uint32_t)0);
1531     assert(rc == MEMCACHED_SUCCESS);
1532   }
1533
1534   for (x= 0; x < 2; x++)
1535   {
1536     value= memcached_get(memc, keys[0], key_length[0],
1537                          &value_length, &flags, &rc);
1538     assert(value);
1539     free(value);
1540
1541     rc= memcached_mget(memc, keys, key_length, 4);
1542     assert(rc == MEMCACHED_SUCCESS);
1543     count= 3;
1544     /* We test for purge of partial complete fetches */
1545     for (count= 3; count; count--)
1546     {
1547       value= memcached_fetch(memc, return_key, &return_key_length,
1548                              &value_length, &flags, &rc);
1549       assert(rc == MEMCACHED_SUCCESS);
1550       assert(!(memcmp(value, insert_data, value_length)));
1551       assert(value_length);
1552       free(value);
1553     }
1554   }
1555
1556   return 0;
1557 }
1558
1559 uint8_t user_supplied_bug8(memcached_st *memc)
1560 {
1561   memcached_return rc;
1562   memcached_st *mine;
1563   memcached_st *clone;
1564
1565   memcached_server_st *servers;
1566   char *server_list= "memcache1.memcache.bk.sapo.pt:11211, memcache1.memcache.bk.sapo.pt:11212, memcache1.memcache.bk.sapo.pt:11213, memcache1.memcache.bk.sapo.pt:11214, memcache2.memcache.bk.sapo.pt:11211, memcache2.memcache.bk.sapo.pt:11212, memcache2.memcache.bk.sapo.pt:11213, memcache2.memcache.bk.sapo.pt:11214";
1567
1568   servers= memcached_servers_parse(server_list);
1569   assert(servers);
1570
1571   mine= memcached_create(NULL);
1572   rc= memcached_server_push(mine, servers);
1573   assert(rc == MEMCACHED_SUCCESS);
1574   memcached_server_list_free(servers);
1575
1576   assert(mine);
1577   clone= memcached_clone(NULL, mine);
1578
1579   memcached_quit(mine);
1580   memcached_quit(clone);
1581
1582
1583   memcached_free(mine);
1584   memcached_free(clone);
1585
1586   return 0;
1587 }
1588
1589 /* Test flag store/retrieve */
1590 uint8_t user_supplied_bug7(memcached_st *memc)
1591 {
1592   memcached_return rc;
1593   char *keys= "036790384900";
1594   size_t key_length=  strlen("036790384900");
1595   char return_key[MEMCACHED_MAX_KEY];
1596   size_t return_key_length;
1597   char *value;
1598   size_t value_length;
1599   uint32_t flags;
1600   unsigned int x;
1601   char insert_data[VALUE_SIZE_BUG5];
1602
1603   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1604     insert_data[x]= rand();
1605
1606   memcached_flush(memc, 0);
1607
1608   flags= 245;
1609   rc= memcached_set(memc, keys, key_length,
1610                     insert_data, VALUE_SIZE_BUG5,
1611                     (time_t)0, flags);
1612   assert(rc == MEMCACHED_SUCCESS);
1613
1614   flags= 0;
1615   value= memcached_get(memc, keys, key_length,
1616                         &value_length, &flags, &rc);
1617   assert(flags == 245);
1618   assert(value);
1619   free(value);
1620
1621   rc= memcached_mget(memc, &keys, &key_length, 1);
1622
1623   flags= 0;
1624   value= memcached_fetch(memc, return_key, &return_key_length,
1625                          &value_length, &flags, &rc);
1626   assert(flags == 245);
1627   assert(value);
1628   free(value);
1629
1630
1631   return 0;
1632 }
1633
1634 uint8_t user_supplied_bug9(memcached_st *memc)
1635 {
1636   memcached_return rc;
1637   char *keys[]= {"UDATA:edevil@sapo.pt", "fudge&*@#", "for^#@&$not"};
1638   size_t key_length[3];
1639   unsigned int x;
1640   uint32_t flags;
1641   unsigned count= 0;
1642
1643   char return_key[MEMCACHED_MAX_KEY];
1644   size_t return_key_length;
1645   char *return_value;
1646   size_t return_value_length;
1647
1648
1649   key_length[0]= strlen("UDATA:edevil@sapo.pt");
1650   key_length[1]= strlen("fudge&*@#");
1651   key_length[2]= strlen("for^#@&$not");
1652
1653
1654   for (x= 0; x < 3; x++)
1655   {
1656     rc= memcached_set(memc, keys[x], key_length[x],
1657                       keys[x], key_length[x],
1658                       (time_t)50, (uint32_t)9);
1659     assert(rc == MEMCACHED_SUCCESS);
1660   }
1661
1662   rc= memcached_mget(memc, keys, key_length, 3);
1663   assert(rc == MEMCACHED_SUCCESS);
1664
1665   /* We need to empty the server before continueing test */
1666   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1667                       &return_value_length, &flags, &rc)) != NULL)
1668   {
1669     assert(return_value);
1670     free(return_value);
1671     count++;
1672   }
1673   assert(count == 3);
1674
1675   return 0;
1676 }
1677
1678 /* We are testing with aggressive timeout to get failures */
1679 uint8_t user_supplied_bug10(memcached_st *memc)
1680 {
1681   char *key= "foo";
1682   char *value;
1683   size_t value_length= 512;
1684   unsigned int x;
1685   int key_len= 3;
1686   memcached_return rc;
1687   unsigned int set= 1;
1688   memcached_st *mclone= memcached_clone(NULL, memc);
1689   int32_t timeout;
1690
1691   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_NO_BLOCK, set);
1692   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_TCP_NODELAY, set);
1693   timeout= 2;
1694   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, timeout);
1695
1696   value = (char*)malloc(value_length * sizeof(char));
1697
1698   for (x= 0; x < value_length; x++)
1699     value[x]= (char) (x % 127);
1700
1701   for (x= 1; x <= 100000; ++x)
1702   {
1703     rc= memcached_set(mclone, key, key_len,value, value_length, 0, 0);
1704
1705     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_WRITE_FAILURE || rc == MEMCACHED_BUFFERED);
1706
1707     if (rc == MEMCACHED_WRITE_FAILURE)
1708       x--;
1709   }
1710
1711   free(value);
1712   memcached_free(mclone);
1713
1714   return 0;
1715 }
1716
1717 /*
1718   We are looking failures in the async protocol
1719 */
1720 uint8_t user_supplied_bug11(memcached_st *memc)
1721 {
1722   char *key= "foo";
1723   char *value;
1724   size_t value_length= 512;
1725   unsigned int x;
1726   int key_len= 3;
1727   memcached_return rc;
1728   unsigned int set= 1;
1729   int32_t timeout;
1730   memcached_st *mclone= memcached_clone(NULL, memc);
1731
1732   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_NO_BLOCK, set);
1733   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_TCP_NODELAY, set);
1734   timeout= -1;
1735   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, timeout);
1736
1737   timeout= (int32_t)memcached_behavior_get(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT);
1738
1739   assert(timeout == -1);
1740
1741   value = (char*)malloc(value_length * sizeof(char));
1742
1743   for (x= 0; x < value_length; x++)
1744     value[x]= (char) (x % 127);
1745
1746   for (x= 1; x <= 100000; ++x)
1747   {
1748     rc= memcached_set(mclone, key, key_len,value, value_length, 0, 0);
1749   }
1750
1751   free(value);
1752   memcached_free(mclone);
1753
1754   return 0;
1755 }
1756
1757 /*
1758   Bug found where incr was not returning MEMCACHED_NOTFOUND when object did not exist.
1759 */
1760 uint8_t user_supplied_bug12(memcached_st *memc)
1761 {
1762   memcached_return rc;
1763   uint32_t flags;
1764   size_t value_length;
1765   char *value;
1766   uint64_t number_value;
1767
1768   value= memcached_get(memc, "autoincrement", strlen("autoincrement"),
1769                         &value_length, &flags, &rc);
1770   assert(value == NULL);
1771   assert(rc == MEMCACHED_NOTFOUND);
1772
1773   rc= memcached_increment(memc, "autoincrement", strlen("autoincrement"),
1774                           1, &number_value);
1775
1776   assert(value == NULL);
1777   assert(rc == MEMCACHED_NOTFOUND);
1778
1779   rc= memcached_set(memc, "autoincrement", strlen("autoincrement"), "1", 1, 0, 0);
1780
1781   value= memcached_get(memc, "autoincrement", strlen("autoincrement"),
1782                         &value_length, &flags, &rc);
1783   assert(value);
1784   assert(rc == MEMCACHED_SUCCESS);
1785   free(value);
1786
1787   rc= memcached_increment(memc, "autoincrement", strlen("autoincrement"),
1788                           1, &number_value);
1789   assert(number_value == 2);
1790   assert(rc == MEMCACHED_SUCCESS);
1791
1792   return 0;
1793 }
1794
1795 /*
1796   Bug found where command total one more than MEMCACHED_MAX_BUFFER
1797   set key34567890 0 0 8169 \r\n is sent followed by buffer of size 8169, followed by 8169
1798  */
1799 uint8_t user_supplied_bug13(memcached_st *memc)
1800 {
1801   char key[] = "key34567890";
1802   char *overflow;
1803   memcached_return rc;
1804   size_t overflowSize;
1805
1806   char commandFirst[]= "set key34567890 0 0 ";
1807   char commandLast[] = " \r\n"; /* first line of command sent to server */
1808   size_t commandLength;
1809   size_t testSize;
1810
1811   commandLength = strlen(commandFirst) + strlen(commandLast) + 4; /* 4 is number of characters in size, probably 8196 */
1812
1813   overflowSize = MEMCACHED_MAX_BUFFER - commandLength;
1814
1815   for (testSize= overflowSize - 1; testSize < overflowSize + 1; testSize++)
1816   {
1817     overflow= malloc(testSize);
1818     assert(overflow != NULL);
1819
1820     memset(overflow, 'x', testSize);
1821     rc= memcached_set(memc, key, strlen(key),
1822                       overflow, testSize, 0, 0);
1823     assert(rc == MEMCACHED_SUCCESS);
1824     free(overflow);
1825   }
1826
1827   return 0;
1828 }
1829
1830
1831 /*
1832   Test values of many different sizes
1833   Bug found where command total one more than MEMCACHED_MAX_BUFFER
1834   set key34567890 0 0 8169 \r\n
1835   is sent followed by buffer of size 8169, followed by 8169
1836  */
1837 uint8_t user_supplied_bug14(memcached_st *memc)
1838 {
1839   int setter= 1;
1840   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, setter);
1841   memcached_return rc;
1842   char *key= "foo";
1843   char *value;
1844   size_t value_length= 18000;
1845   char *string;
1846   size_t string_length;
1847   uint32_t flags;
1848   unsigned int x;
1849   size_t current_length;
1850
1851   value = (char*)malloc(value_length);
1852   assert(value);
1853
1854   for (x= 0; x < value_length; x++)
1855     value[x] = (char) (x % 127);
1856
1857   for (current_length= 0; current_length < value_length; current_length++)
1858   {
1859     rc= memcached_set(memc, key, strlen(key),
1860                       value, current_length,
1861                       (time_t)0, (uint32_t)0);
1862     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
1863
1864     string= memcached_get(memc, key, strlen(key),
1865                           &string_length, &flags, &rc);
1866
1867     assert(rc == MEMCACHED_SUCCESS);
1868     assert(string);
1869     assert(string_length == current_length);
1870     assert(!memcmp(string, value, string_length));
1871
1872     free(string);
1873   }
1874
1875   free(value);
1876
1877   return 0;
1878 }
1879
1880 /*
1881   Look for zero length value problems
1882   */
1883 uint8_t user_supplied_bug15(memcached_st *memc)
1884 {
1885   uint32_t x;
1886   memcached_return rc;
1887   char *key= "mykey";
1888   char *value;
1889   size_t length;
1890   uint32_t flags;
1891
1892   for (x= 0; x < 2; x++)
1893   {
1894     rc= memcached_set(memc, key, strlen(key),
1895                       NULL, 0,
1896                       (time_t)0, (uint32_t)0);
1897
1898     assert(rc == MEMCACHED_SUCCESS);
1899
1900     value= memcached_get(memc, key, strlen(key),
1901                          &length, &flags, &rc);
1902
1903     assert(rc == MEMCACHED_SUCCESS);
1904     assert(value == NULL);
1905     assert(length == 0);
1906     assert(flags == 0);
1907
1908     value= memcached_get(memc, key, strlen(key),
1909                          &length, &flags, &rc);
1910
1911     assert(rc == MEMCACHED_SUCCESS);
1912     assert(value == NULL);
1913     assert(length == 0);
1914     assert(flags == 0);
1915   }
1916
1917   return 0;
1918 }
1919
1920 uint8_t result_static(memcached_st *memc)
1921 {
1922   memcached_result_st result;
1923   memcached_result_st *result_ptr;
1924
1925   result_ptr= memcached_result_create(memc, &result);
1926   assert(result.is_allocated == MEMCACHED_NOT_ALLOCATED);
1927   assert(result_ptr);
1928   memcached_result_free(&result);
1929
1930   return 0;
1931 }
1932
1933 uint8_t result_alloc(memcached_st *memc)
1934 {
1935   memcached_result_st *result;
1936
1937   result= memcached_result_create(memc, NULL);
1938   assert(result);
1939   memcached_result_free(result);
1940
1941   return 0;
1942 }
1943
1944 uint8_t string_static_null(memcached_st *memc)
1945 {
1946   memcached_string_st string;
1947   memcached_string_st *string_ptr;
1948
1949   string_ptr= memcached_string_create(memc, &string, 0);
1950   assert(string.is_allocated == MEMCACHED_NOT_ALLOCATED);
1951   assert(string_ptr);
1952   memcached_string_free(&string);
1953
1954   return 0;
1955 }
1956
1957 uint8_t string_alloc_null(memcached_st *memc)
1958 {
1959   memcached_string_st *string;
1960
1961   string= memcached_string_create(memc, NULL, 0);
1962   assert(string);
1963   memcached_string_free(string);
1964
1965   return 0;
1966 }
1967
1968 uint8_t string_alloc_with_size(memcached_st *memc)
1969 {
1970   memcached_string_st *string;
1971
1972   string= memcached_string_create(memc, NULL, 1024);
1973   assert(string);
1974   memcached_string_free(string);
1975
1976   return 0;
1977 }
1978
1979 uint8_t string_alloc_with_size_toobig(memcached_st *memc)
1980 {
1981   memcached_string_st *string;
1982
1983   string= memcached_string_create(memc, NULL, INT64_MAX);
1984   assert(string == NULL);
1985
1986   return 0;
1987 }
1988
1989 uint8_t string_alloc_append(memcached_st *memc)
1990 {
1991   unsigned int x;
1992   char buffer[SMALL_STRING_LEN];
1993   memcached_string_st *string;
1994
1995   /* Ring the bell! */
1996   memset(buffer, 6, SMALL_STRING_LEN);
1997
1998   string= memcached_string_create(memc, NULL, 100);
1999   assert(string);
2000
2001   for (x= 0; x < 1024; x++)
2002   {
2003     memcached_return rc;
2004     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
2005     assert(rc == MEMCACHED_SUCCESS);
2006   }
2007   memcached_string_free(string);
2008
2009   return 0;
2010 }
2011
2012 uint8_t string_alloc_append_toobig(memcached_st *memc)
2013 {
2014   memcached_return rc;
2015   unsigned int x;
2016   char buffer[SMALL_STRING_LEN];
2017   memcached_string_st *string;
2018
2019   /* Ring the bell! */
2020   memset(buffer, 6, SMALL_STRING_LEN);
2021
2022   string= memcached_string_create(memc, NULL, 100);
2023   assert(string);
2024
2025   for (x= 0; x < 1024; x++)
2026   {
2027     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
2028     assert(rc == MEMCACHED_SUCCESS);
2029   }
2030   rc= memcached_string_append(string, buffer, INT64_MAX);
2031   assert(rc == MEMCACHED_MEMORY_ALLOCATION_FAILURE);
2032   memcached_string_free(string);
2033
2034   return 0;
2035 }
2036
2037 uint8_t cleanup_pairs(memcached_st *memc)
2038 {
2039   pairs_free(global_pairs);
2040
2041   return 0;
2042 }
2043
2044 uint8_t generate_pairs(memcached_st *memc)
2045 {
2046   unsigned long long x;
2047   global_pairs= pairs_generate(GLOBAL_COUNT, 400);
2048   global_count= GLOBAL_COUNT;
2049
2050   for (x= 0; x < global_count; x++)
2051   {
2052     global_keys[x]= global_pairs[x].key;
2053     global_keys_length[x]=  global_pairs[x].key_length;
2054   }
2055
2056   return 0;
2057 }
2058
2059 uint8_t generate_large_pairs(memcached_st *memc)
2060 {
2061   unsigned long long x;
2062   global_pairs= pairs_generate(GLOBAL2_COUNT, MEMCACHED_MAX_BUFFER+10);
2063   global_count= GLOBAL2_COUNT;
2064
2065   for (x= 0; x < global_count; x++)
2066   {
2067     global_keys[x]= global_pairs[x].key;
2068     global_keys_length[x]=  global_pairs[x].key_length;
2069   }
2070
2071   return 0;
2072 }
2073
2074 uint8_t generate_data(memcached_st *memc)
2075 {
2076   execute_set(memc, global_pairs, global_count);
2077
2078   return 0;
2079 }
2080
2081 uint8_t generate_buffer_data(memcached_st *memc)
2082 {
2083   int latch= 0;
2084
2085   latch= 1;
2086   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, latch);
2087   generate_data(memc);
2088
2089   return 0;
2090 }
2091
2092 uint8_t get_read(memcached_st *memc)
2093 {
2094   unsigned int x;
2095   memcached_return rc;
2096
2097   {
2098     char *return_value;
2099     size_t return_value_length;
2100     uint32_t flags;
2101
2102     for (x= 0; x < global_count; x++)
2103     {
2104       return_value= memcached_get(memc, global_keys[x], global_keys_length[x],
2105                                   &return_value_length, &flags, &rc);
2106       /*
2107       assert(return_value);
2108       assert(rc == MEMCACHED_SUCCESS);
2109     */
2110       if (rc == MEMCACHED_SUCCESS && return_value)
2111         free(return_value);
2112     }
2113   }
2114
2115   return 0;
2116 }
2117
2118 uint8_t mget_read(memcached_st *memc)
2119 {
2120   memcached_return rc;
2121
2122   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
2123   assert(rc == MEMCACHED_SUCCESS);
2124   /* Turn this into a help function */
2125   {
2126     char return_key[MEMCACHED_MAX_KEY];
2127     size_t return_key_length;
2128     char *return_value;
2129     size_t return_value_length;
2130     uint32_t flags;
2131
2132     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
2133                                           &return_value_length, &flags, &rc)))
2134     {
2135       assert(return_value);
2136       assert(rc == MEMCACHED_SUCCESS);
2137       free(return_value);
2138     }
2139   }
2140
2141   return 0;
2142 }
2143
2144 uint8_t mget_read_result(memcached_st *memc)
2145 {
2146   memcached_return rc;
2147
2148   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
2149   assert(rc == MEMCACHED_SUCCESS);
2150   /* Turn this into a help function */
2151   {
2152     memcached_result_st results_obj;
2153     memcached_result_st *results;
2154
2155     results= memcached_result_create(memc, &results_obj);
2156
2157     while ((results= memcached_fetch_result(memc, &results_obj, &rc)))
2158     {
2159       assert(results);
2160       assert(rc == MEMCACHED_SUCCESS);
2161     }
2162
2163     memcached_result_free(&results_obj);
2164   }
2165
2166   return 0;
2167 }
2168
2169 uint8_t mget_read_function(memcached_st *memc)
2170 {
2171   memcached_return rc;
2172   unsigned int counter;
2173   unsigned int (*callbacks[1])(memcached_st *, memcached_result_st *, void *);
2174
2175   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
2176   assert(rc == MEMCACHED_SUCCESS);
2177
2178   callbacks[0]= &callback_counter;
2179   counter= 0;
2180   rc= memcached_fetch_execute(memc, callbacks, (void *)&counter, 1);
2181
2182   return 0;
2183 }
2184
2185 uint8_t delete_generate(memcached_st *memc)
2186 {
2187   unsigned int x;
2188
2189   for (x= 0; x < global_count; x++)
2190   {
2191     (void)memcached_delete(memc, global_keys[x], global_keys_length[x], (time_t)0);
2192   }
2193
2194   return 0;
2195 }
2196
2197 uint8_t delete_buffer_generate(memcached_st *memc)
2198 {
2199   int latch= 0;
2200   unsigned int x;
2201
2202   latch= 1;
2203   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, latch);
2204
2205   for (x= 0; x < global_count; x++)
2206   {
2207     (void)memcached_delete(memc, global_keys[x], global_keys_length[x], (time_t)0);
2208   }
2209
2210   return 0;
2211 }
2212
2213 uint8_t free_data(memcached_st *memc)
2214 {
2215   pairs_free(global_pairs);
2216
2217   return 0;
2218 }
2219
2220 uint8_t add_host_test1(memcached_st *memc)
2221 {
2222   unsigned int x;
2223   memcached_return rc;
2224   char servername[]= "0.example.com";
2225   memcached_server_st *servers;
2226
2227   servers= memcached_server_list_append(NULL, servername, 400, &rc);
2228   assert(servers);
2229   assert(1 == memcached_server_list_count(servers));
2230
2231   for (x= 2; x < 20; x++)
2232   {
2233     char buffer[SMALL_STRING_LEN];
2234
2235     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
2236     servers= memcached_server_list_append(servers, buffer, 401,
2237                                      &rc);
2238     assert(rc == MEMCACHED_SUCCESS);
2239     assert(x == memcached_server_list_count(servers));
2240   }
2241
2242   rc= memcached_server_push(memc, servers);
2243   assert(rc == MEMCACHED_SUCCESS);
2244   rc= memcached_server_push(memc, servers);
2245   assert(rc == MEMCACHED_SUCCESS);
2246
2247   memcached_server_list_free(servers);
2248
2249   return 0;
2250 }
2251
2252 memcached_return pre_nonblock(memcached_st *memc)
2253 {
2254   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
2255
2256   return MEMCACHED_SUCCESS;
2257 }
2258
2259 memcached_return pre_murmur(memcached_st *memc)
2260 {
2261   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_MURMUR);
2262
2263   return MEMCACHED_SUCCESS;
2264 }
2265
2266 memcached_return pre_md5(memcached_st *memc)
2267 {
2268   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_MD5);
2269
2270   return MEMCACHED_SUCCESS;
2271 }
2272
2273 memcached_return pre_crc(memcached_st *memc)
2274 {
2275   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_CRC);
2276
2277   return MEMCACHED_SUCCESS;
2278 }
2279
2280 memcached_return pre_hsieh(memcached_st *memc)
2281 {
2282   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_HSIEH);
2283
2284   return MEMCACHED_SUCCESS;
2285 }
2286
2287 memcached_return pre_hash_fnv1_64(memcached_st *memc)
2288 {
2289   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1_64);
2290
2291   return MEMCACHED_SUCCESS;
2292 }
2293
2294 memcached_return pre_hash_fnv1a_64(memcached_st *memc)
2295 {
2296   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1A_64);
2297
2298   return MEMCACHED_SUCCESS;
2299 }
2300
2301 memcached_return pre_hash_fnv1_32(memcached_st *memc)
2302 {
2303   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1_32);
2304
2305   return MEMCACHED_SUCCESS;
2306 }
2307
2308 memcached_return pre_hash_fnv1a_32(memcached_st *memc)
2309 {
2310   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1A_32);
2311
2312   return MEMCACHED_SUCCESS;
2313 }
2314
2315 memcached_return pre_hash_ketama(memcached_st *memc)
2316 {
2317   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_KETAMA);
2318
2319   return MEMCACHED_SUCCESS;
2320 }
2321
2322 void my_free(memcached_st *ptr, void *mem)
2323 {
2324   free(mem);
2325 }
2326
2327 void *my_malloc(memcached_st *ptr, const size_t size)
2328 {
2329   return malloc(size);
2330 }
2331
2332 void *my_realloc(memcached_st *ptr, void *mem, const size_t size)
2333 {
2334   return realloc(mem, size);
2335 }
2336
2337 memcached_return set_memory_alloc(memcached_st *memc)
2338 {
2339   {
2340     memcached_malloc_function test_ptr;
2341     memcached_return rc;
2342
2343     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_MALLOC_FUNCTION, &my_malloc);
2344     assert(rc == MEMCACHED_SUCCESS);
2345     test_ptr= (memcached_malloc_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_MALLOC_FUNCTION, &rc);
2346     assert(rc == MEMCACHED_SUCCESS);
2347     assert(test_ptr == my_malloc);
2348   }
2349
2350   {
2351     memcached_realloc_function test_ptr;
2352     memcached_return rc;
2353
2354     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_REALLOC_FUNCTION, &my_realloc);
2355     assert(rc == MEMCACHED_SUCCESS);
2356     test_ptr= (memcached_realloc_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_REALLOC_FUNCTION, &rc);
2357     assert(rc == MEMCACHED_SUCCESS);
2358     assert(test_ptr == my_realloc);
2359   }
2360
2361   {
2362     memcached_free_function test_ptr;
2363     memcached_return rc;
2364
2365     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_FREE_FUNCTION, my_free);
2366     assert(rc == MEMCACHED_SUCCESS);
2367     test_ptr= (memcached_free_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_FREE_FUNCTION, &rc);
2368     assert(rc == MEMCACHED_SUCCESS);
2369     assert(test_ptr == my_free);
2370   }
2371
2372   return MEMCACHED_SUCCESS;
2373 }
2374
2375 memcached_return enable_consistent(memcached_st *memc)
2376 {
2377   memcached_server_distribution value= MEMCACHED_DISTRIBUTION_CONSISTENT;
2378   memcached_hash hash;
2379   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, value);
2380   pre_hsieh(memc);
2381
2382   value= (memcached_server_distribution)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION);
2383   assert(value == MEMCACHED_DISTRIBUTION_CONSISTENT);
2384
2385   hash= (memcached_hash)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
2386   assert(hash == MEMCACHED_HASH_HSIEH);
2387
2388
2389   return MEMCACHED_SUCCESS;
2390 }
2391
2392 memcached_return enable_cas(memcached_st *memc)
2393 {
2394   unsigned int set= 1;
2395
2396   memcached_version(memc);
2397
2398   if (memc->hosts[0].major_version >= 1 &&
2399       memc->hosts[0].minor_version >= 2 &&
2400       memc->hosts[0].micro_version >= 4)
2401   {
2402     memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, set);
2403
2404     return MEMCACHED_SUCCESS;
2405   }
2406
2407   return MEMCACHED_FAILURE;
2408 }
2409
2410 memcached_return check_for_1_2_3(memcached_st *memc)
2411 {
2412   memcached_version(memc);
2413
2414   if (memc->hosts[0].major_version >= 1 &&
2415       memc->hosts[0].minor_version >= 2 &&
2416       memc->hosts[0].micro_version >= 4)
2417     return MEMCACHED_SUCCESS;
2418
2419   return MEMCACHED_FAILURE;
2420 }
2421
2422 memcached_return pre_unix_socket(memcached_st *memc)
2423 {
2424   memcached_return rc;
2425   struct stat buf;
2426
2427   memcached_server_list_free(memc->hosts);
2428   memc->hosts= NULL;
2429   memc->number_of_hosts= 0;
2430
2431   if (stat("/tmp/memcached.socket", &buf))
2432     return MEMCACHED_FAILURE;
2433
2434   rc= memcached_server_add_unix_socket(memc, "/tmp/memcached.socket");
2435
2436   return rc;
2437 }
2438
2439 memcached_return pre_udp(memcached_st *memc)
2440 {
2441   memcached_return rc;
2442
2443   memcached_server_list_free(memc->hosts);
2444   memc->hosts= NULL;
2445   memc->number_of_hosts= 0;
2446
2447   if (0)
2448     return MEMCACHED_FAILURE;
2449
2450   rc= memcached_server_add_udp(memc, "localhost", MEMCACHED_DEFAULT_PORT);
2451
2452   return rc;
2453 }
2454
2455 memcached_return pre_nodelay(memcached_st *memc)
2456 {
2457   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
2458   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, 0);
2459
2460   return MEMCACHED_SUCCESS;
2461 }
2462
2463 memcached_return poll_timeout(memcached_st *memc)
2464 {
2465   int32_t timeout;
2466
2467   timeout= 100;
2468
2469   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, timeout);
2470
2471   timeout= (int32_t)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_POLL_TIMEOUT);
2472
2473   assert(timeout == 100);
2474
2475   return MEMCACHED_SUCCESS;
2476 }
2477
2478
2479 /* Clean the server before beginning testing */
2480 test_st tests[] ={
2481   {"flush", 0, flush_test },
2482   {"init", 0, init_test },
2483   {"allocation", 0, allocation_test },
2484   {"server_list_null_test", 0, server_list_null_test},
2485   {"server_sort", 0, server_sort_test},
2486   {"clone_test", 0, clone_test },
2487   {"error", 0, error_test },
2488   {"set", 0, set_test },
2489   {"set2", 0, set_test2 },
2490   {"set3", 0, set_test3 },
2491   {"add", 1, add_test },
2492   {"replace", 1, replace_test },
2493   {"delete", 1, delete_test },
2494   {"get", 1, get_test },
2495   {"get2", 0, get_test2 },
2496   {"get3", 0, get_test3 },
2497   {"get4", 0, get_test4 },
2498   {"stats_servername", 0, stats_servername_test },
2499   {"increment", 0, increment_test },
2500   {"decrement", 0, decrement_test },
2501   {"quit", 0, quit_test },
2502   {"mget", 1, mget_test },
2503   {"mget_result", 1, mget_result_test },
2504   {"mget_result_alloc", 1, mget_result_alloc_test },
2505   {"mget_result_function", 1, mget_result_function },
2506   {"get_stats", 0, get_stats },
2507   {"add_host_test", 0, add_host_test },
2508   {"get_stats_keys", 0, get_stats_keys },
2509   {"behavior_test", 0, get_stats_keys },
2510   {"callback_test", 0, get_stats_keys },
2511   {"version_string_test", 0, version_string_test},
2512   {"bad_key", 1, bad_key_test },
2513   {"memcached_server_cursor", 1, memcached_server_cursor_test },
2514   {"read_through", 1, read_through },
2515   {0, 0, 0}
2516 };
2517
2518 test_st async_tests[] ={
2519   {"add", 1, add_wrapper },
2520   {0, 0, 0}
2521 };
2522
2523 test_st string_tests[] ={
2524   {"string static with null", 0, string_static_null },
2525   {"string alloc with null", 0, string_alloc_null },
2526   {"string alloc with 1K", 0, string_alloc_with_size },
2527   {"string alloc with malloc failure", 0, string_alloc_with_size_toobig },
2528   {"string append", 0, string_alloc_append },
2529   {"string append failure (too big)", 0, string_alloc_append_toobig },
2530   {0, 0, 0}
2531 };
2532
2533 test_st result_tests[] ={
2534   {"result static", 0, result_static},
2535   {"result alloc", 0, result_alloc},
2536   {0, 0, 0}
2537 };
2538
2539 test_st version_1_2_3[] ={
2540   {"append", 0, append_test },
2541   {"prepend", 0, prepend_test },
2542   {"cas", 0, cas_test },
2543   {"cas2", 0, cas2_test },
2544   {"append_binary", 0, append_binary_test },
2545   {0, 0, 0}
2546 };
2547
2548 test_st user_tests[] ={
2549   {"user_supplied_bug1", 0, user_supplied_bug1 },
2550   {"user_supplied_bug2", 0, user_supplied_bug2 },
2551   {"user_supplied_bug3", 0, user_supplied_bug3 },
2552   {"user_supplied_bug4", 0, user_supplied_bug4 },
2553   {"user_supplied_bug5", 1, user_supplied_bug5 },
2554   {"user_supplied_bug6", 1, user_supplied_bug6 },
2555   {"user_supplied_bug7", 1, user_supplied_bug7 },
2556   {"user_supplied_bug8", 1, user_supplied_bug8 },
2557   {"user_supplied_bug9", 1, user_supplied_bug9 },
2558   {"user_supplied_bug10", 1, user_supplied_bug10 },
2559   {"user_supplied_bug11", 1, user_supplied_bug11 },
2560   {"user_supplied_bug12", 1, user_supplied_bug12 },
2561   {"user_supplied_bug13", 1, user_supplied_bug13 },
2562   {"user_supplied_bug14", 1, user_supplied_bug14 },
2563   {"user_supplied_bug15", 1, user_supplied_bug15 },
2564   {0, 0, 0}
2565 };
2566
2567 test_st generate_tests[] ={
2568   {"generate_pairs", 1, generate_pairs },
2569   {"generate_data", 1, generate_data },
2570   {"get_read", 0, get_read },
2571   {"delete_generate", 0, delete_generate },
2572   {"generate_buffer_data", 1, generate_buffer_data },
2573   {"delete_buffer", 0, delete_buffer_generate},
2574   {"generate_data", 1, generate_data },
2575   {"mget_read", 0, mget_read },
2576   {"mget_read_result", 0, mget_read_result },
2577   {"mget_read_function", 0, mget_read_function },
2578   {"cleanup", 1, cleanup_pairs },
2579   {"generate_large_pairs", 1, generate_large_pairs },
2580   {"generate_data", 1, generate_data },
2581   {"generate_buffer_data", 1, generate_buffer_data },
2582   {"cleanup", 1, cleanup_pairs },
2583   {0, 0, 0}
2584 };
2585
2586
2587 collection_st collection[] ={
2588   {"block", 0, 0, tests},
2589   {"nonblock", pre_nonblock, 0, tests},
2590   {"nodelay", pre_nodelay, 0, tests},
2591   {"md5", pre_md5, 0, tests},
2592   {"crc", pre_crc, 0, tests},
2593   {"hsieh", pre_hsieh, 0, tests},
2594   {"fnv1_64", pre_hash_fnv1_64, 0, tests},
2595   {"fnv1a_64", pre_hash_fnv1a_64, 0, tests},
2596   {"fnv1_32", pre_hash_fnv1_32, 0, tests},
2597   {"fnv1a_32", pre_hash_fnv1a_32, 0, tests},
2598   {"ketama", pre_hash_ketama, 0, tests},
2599   {"unix_socket", pre_unix_socket, 0, tests},
2600   {"unix_socket_nodelay", pre_nodelay, 0, tests},
2601   {"poll_timeout", poll_timeout, 0, tests},
2602   {"gets", enable_cas, 0, tests},
2603   {"consistent", enable_consistent, 0, tests},
2604   {"memory_allocators", set_memory_alloc, 0, tests},
2605 //  {"udp", pre_udp, 0, tests},
2606   {"version_1_2_3", check_for_1_2_3, 0, version_1_2_3},
2607   {"string", 0, 0, string_tests},
2608   {"result", 0, 0, result_tests},
2609   {"async", pre_nonblock, 0, async_tests},
2610   {"user", 0, 0, user_tests},
2611   {"generate", 0, 0, generate_tests},
2612   {"generate_hsieh", pre_hsieh, 0, generate_tests},
2613   {"generate_hsieh_consistent", enable_consistent, 0, generate_tests},
2614   {"generate_md5", pre_md5, 0, generate_tests},
2615   {"generate_murmur", pre_murmur, 0, generate_tests},
2616   {"generate_nonblock", pre_nonblock, 0, generate_tests},
2617   {0, 0, 0, 0}
2618 };
2619
2620 #define SERVERS_TO_CREATE 5
2621
2622 void *world_create(void)
2623 {
2624   server_startup_st *construct;
2625
2626   construct= (server_startup_st *)malloc(sizeof(server_startup_st));
2627   memset(construct, 0, sizeof(server_startup_st));
2628   construct->count= SERVERS_TO_CREATE;
2629   construct->udp= 0;
2630   server_startup(construct);
2631
2632   return construct;
2633 }
2634
2635 void world_destroy(void *p)
2636 {
2637   server_startup_st *construct= (server_startup_st *)p;
2638   memcached_server_st *servers= (memcached_server_st *)construct->servers;
2639   memcached_server_list_free(servers);
2640
2641   server_shutdown(construct);
2642   free(construct);
2643 }
2644
2645 void get_world(world_st *world)
2646 {
2647   world->collections= collection;
2648   world->create= world_create;
2649   world->destroy= world_destroy;
2650 }