8000910e6b8abf74aaf907a57be5efa7a170bc3b
[awesomized/libmemcached] / clients / generator.c
1 #include "libmemcached/common.h"
2
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <stdint.h>
6 #include <string.h>
7
8 #include "generator.h"
9
10 /* Use this for string generation */
11 static const char ALPHANUMERICS[]=
12 "0123456789ABCDEFGHIJKLMNOPQRSTWXYZabcdefghijklmnopqrstuvwxyz";
13
14 #define ALPHANUMERICS_SIZE (sizeof(ALPHANUMERICS)-1)
15
16 static void get_random_string(char *buffer, size_t size)
17 {
18 char *buffer_ptr= buffer;
19
20 while (--size)
21 *buffer_ptr++= ALPHANUMERICS[random() % ALPHANUMERICS_SIZE];
22 *buffer_ptr++= ALPHANUMERICS[random() % ALPHANUMERICS_SIZE];
23 }
24
25 void pairs_free(pairs_st *pairs)
26 {
27 uint32_t x;
28
29 if (!pairs)
30 return;
31
32 /* We free until we hit the null pair we stores during creation */
33 for (x= 0; pairs[x].key; x++)
34 {
35 free(pairs[x].key);
36 free(pairs[x].value);
37 }
38
39 free(pairs);
40 }
41
42 pairs_st *pairs_generate(uint64_t number_of, size_t value_length)
43 {
44 unsigned int x;
45 pairs_st *pairs;
46
47 pairs= (pairs_st*)malloc(sizeof(pairs_st) * (number_of+1));
48
49 if (!pairs)
50 goto error;
51
52 memset(pairs, 0, sizeof(pairs_st) * (number_of+1));
53
54 for (x= 0; x < number_of; x++)
55 {
56 pairs[x].key= (char *)malloc(sizeof(char) * 100);
57 if (!pairs[x].key)
58 goto error;
59 get_random_string(pairs[x].key, 100);
60 pairs[x].key_length= 100;
61
62 pairs[x].value= (char *)malloc(sizeof(char) * value_length);
63 if (!pairs[x].value)
64 goto error;
65 get_random_string(pairs[x].value, value_length);
66 pairs[x].value_length= value_length;
67 }
68
69 return pairs;
70 error:
71 fprintf(stderr, "Memory Allocation failure in pairs_generate.\n");
72 exit(0);
73 }