Ignore *.exe
[awesomized/libmemcached] / 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 <libmemcached/util.h>
28 #include <unistd.h>
29 #include <sys/time.h>
30
31 #include "server.h"
32
33 void server_startup(server_startup_st *construct)
34 {
35 if ((construct->server_list= getenv("MEMCACHED_SERVERS")))
36 {
37 printf("servers %s\n", construct->server_list);
38 construct->servers= memcached_servers_parse(construct->server_list);
39 construct->server_list= NULL;
40 construct->count= 0;
41 }
42 else
43 {
44 {
45 char server_string_buffer[8096];
46 char *end_ptr;
47 end_ptr= server_string_buffer;
48
49 for (uint32_t x= 0; x < construct->count; x++)
50 {
51 char buffer[1024]; /* Nothing special for number */
52 int count;
53 int status;
54
55 sprintf(buffer, "/tmp/%umemc.pid", x);
56 if (access(buffer, F_OK) == 0)
57 {
58 FILE *fp= fopen(buffer, "r");
59 remove(buffer);
60
61 if (fp != NULL)
62 {
63 if (fgets(buffer, sizeof(buffer), fp) != NULL)
64 {
65 pid_t pid= (pid_t)atoi(buffer);
66 if (pid != 0)
67 kill(pid, SIGTERM);
68 }
69
70 fclose(fp);
71 }
72 }
73
74 if (x == 0)
75 {
76 sprintf(buffer, "%s -d -P /tmp/%umemc.pid -t 1 -p %u -U %u -m 128",
77 MEMCACHED_BINARY, x, x + TEST_PORT_BASE, x + TEST_PORT_BASE);
78 }
79 else
80 {
81 sprintf(buffer, "%s -d -P /tmp/%umemc.pid -t 1 -p %u -U %u",
82 MEMCACHED_BINARY, x, x + TEST_PORT_BASE, x + TEST_PORT_BASE);
83 }
84 if (libmemcached_util_ping("localhost", (in_port_t)(x + TEST_PORT_BASE), NULL))
85 {
86 fprintf(stderr, "Server on port %u already exists\n", x + TEST_PORT_BASE);
87 }
88 else
89 {
90 status= system(buffer);
91 fprintf(stderr, "STARTING SERVER: %s status:%d\n", buffer, status);
92 }
93 count= sprintf(end_ptr, "localhost:%u,", x + TEST_PORT_BASE);
94 end_ptr+= count;
95 }
96 *end_ptr= 0;
97
98 for (uint32_t x= 0; x < construct->count; x++)
99 {
100 uint32_t counter= 3;
101 char buffer[1024]; /* Nothing special for number */
102
103 snprintf(buffer, sizeof(buffer), "/tmp/%umemc.pid", x);
104
105 while (--counter)
106 {
107 int memcached_pid;
108
109 FILE *file;
110 file= fopen(buffer, "r");
111 if (file == NULL)
112 {
113 #ifndef WIN32
114 struct timespec req= { .tv_sec= 0, .tv_nsec= 5000 };
115 nanosleep(&req, NULL);
116 #endif
117 continue;
118 }
119 char *found= fgets(buffer, sizeof(buffer), file);
120 if (!found)
121 {
122 abort();
123 }
124 // Yes, we currently pitch this and don't make use of it.
125 memcached_pid= atoi(buffer);
126 fclose(file);
127 }
128
129
130 }
131
132 construct->server_list= strdup(server_string_buffer);
133 }
134 printf("servers %s\n", construct->server_list);
135 construct->servers= memcached_servers_parse(construct->server_list);
136 }
137
138 assert(construct->servers);
139
140 srandom((unsigned int)time(NULL));
141
142 for (uint32_t x= 0; x < memcached_server_list_count(construct->servers); x++)
143 {
144 printf("\t%s : %d\n", memcached_server_name(&construct->servers[x]), memcached_server_port(&construct->servers[x]));
145 assert(construct->servers[x].fd == -1);
146 assert(construct->servers[x].cursor_active == 0);
147 }
148
149 printf("\n");
150 }
151
152 void server_shutdown(server_startup_st *construct)
153 {
154 if (construct->server_list)
155 {
156 for (uint32_t x= 0; x < construct->count; x++)
157 {
158 char buffer[1024]; /* Nothing special for number */
159 sprintf(buffer, "cat /tmp/%umemc.pid | xargs kill", x);
160 /* We have to check the return value of this or the compiler will yell */
161 int sys_ret= system(buffer);
162 assert(sys_ret != -1);
163 sprintf(buffer, "/tmp/%umemc.pid", x);
164 unlink(buffer);
165 }
166
167 free(construct->server_list);
168 }
169 }