git.m6w6.name Git - m6w6/libmemcached/blob - tests/server.c

   1 /* LibMemcached
   2  * Copyright (C) 2006-2009 Brian Aker
   3  * All rights reserved.
   4  *
   5  * Use and distribution licensed under the BSD license.  See
   6  * the COPYING file in the parent directory for full text.
   7  *
   8  * Summary:
   9  *
  10  */
  11
  12 /*
  13   Startup, and shutdown the memcached servers.
  14 */
  15
  16 #define TEST_PORT_BASE MEMCACHED_DEFAULT_PORT+10
  17
  18 #include "config.h"
  19
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <time.h>
  24 #include <assert.h>
  25 #include <signal.h>
  26 #include <libmemcached/memcached.h>
  27 #include <unistd.h>
  28 #include "server.h"
  29
  30 void server_startup(server_startup_st *construct)
  31 {
  32   unsigned int x;
  33
  34   if ((construct->server_list= getenv("MEMCACHED_SERVERS")))
  35   {
  36     printf("servers %s\n", construct->server_list);
  37     construct->servers= memcached_servers_parse(construct->server_list);
  38     construct->server_list= NULL;
  39     construct->count= 0;
  40   }
  41   else
  42   {
  43     {
  44       char server_string_buffer[8096];
  45       char *end_ptr;
  46       end_ptr= server_string_buffer;
  47
  48       for (x= 0; x < construct->count; x++)
  49       {
  50         char buffer[1024]; /* Nothing special for number */
  51         int count;
  52         int status;
  53
  54         sprintf(buffer, "/tmp/%umemc.pid", x);
  55         if (access(buffer, F_OK) == 0)
  56         {
  57           FILE *fp= fopen(buffer, "r");
  58           remove(buffer);
  59
  60           if (fp != NULL)
  61           {
  62             if (fgets(buffer, sizeof(buffer), fp) != NULL)
  63             {
  64               pid_t pid= (pid_t)atoi(buffer);
  65               if (pid != 0)
  66                 kill(pid, SIGTERM);
  67             }
  68
  69             fclose(fp);
  70           }
  71         }
  72
  73         if (x == 0)
  74         {
  75           sprintf(buffer, "%s -d -P /tmp/%umemc.pid -t 1 -p %u -U %u -m 128",
  76                     MEMCACHED_BINARY, x, x + TEST_PORT_BASE, x + TEST_PORT_BASE);
  77         }
  78         else
  79         {
  80           sprintf(buffer, "%s -d -P /tmp/%umemc.pid -t 1 -p %u -U %u",
  81                     MEMCACHED_BINARY, x, x + TEST_PORT_BASE, x + TEST_PORT_BASE);
  82         }
  83         fprintf(stderr, "STARTING SERVER: %s\n", buffer);
  84         status= system(buffer);
  85         count= sprintf(end_ptr, "localhost:%u,", x + TEST_PORT_BASE);
  86         end_ptr+= count;
  87       }
  88       *end_ptr= 0;
  89
  90       construct->server_list= strdup(server_string_buffer);
  91     }
  92     printf("servers %s\n", construct->server_list);
  93     construct->servers= memcached_servers_parse(construct->server_list);
  94   }
  95
  96   assert(construct->servers);
  97
  98   srandom((unsigned int)time(NULL));
  99
 100   for (x= 0; x < memcached_server_list_count(construct->servers); x++)
 101   {
 102     printf("\t%s : %d\n", memcached_server_name(&construct->servers[x]), memcached_server_port(&construct->servers[x]));
 103     assert(construct->servers[x].fd == -1);
 104     assert(construct->servers[x].cursor_active == 0);
 105   }
 106
 107   printf("\n");
 108 }
 109
 110 void server_shutdown(server_startup_st *construct)
 111 {
 112   unsigned int x;
 113
 114   if (construct->server_list)
 115   {
 116     for (x= 0; x < construct->count; x++)
 117     {
 118       char buffer[1024]; /* Nothing special for number */
 119       sprintf(buffer, "cat /tmp/%umemc.pid | xargs kill", x);
 120       /* We have to check the return value of this or the compiler will yell */
 121       int sys_ret= system(buffer);
 122       assert(sys_ret != -1);
 123       sprintf(buffer, "/tmp/%umemc.pid", x);
 124       unlink(buffer);
 125     }
 126
 127     free(construct->server_list);
 128   }
 129 }