]> git.saurik.com Git - redis.git/blame_incremental - src/debug.c
Merge pull request #192 from pietern/makefile
[redis.git] / src / debug.c
... / ...
CommitLineData
1#include "redis.h"
2#include "sha1.h" /* SHA1 is used for DEBUG DIGEST */
3
4#include <arpa/inet.h>
5
6/* ================================= Debugging ============================== */
7
8/* Compute the sha1 of string at 's' with 'len' bytes long.
9 * The SHA1 is then xored againt the string pointed by digest.
10 * Since xor is commutative, this operation is used in order to
11 * "add" digests relative to unordered elements.
12 *
13 * So digest(a,b,c,d) will be the same of digest(b,a,c,d) */
14void xorDigest(unsigned char *digest, void *ptr, size_t len) {
15 SHA1_CTX ctx;
16 unsigned char hash[20], *s = ptr;
17 int j;
18
19 SHA1Init(&ctx);
20 SHA1Update(&ctx,s,len);
21 SHA1Final(hash,&ctx);
22
23 for (j = 0; j < 20; j++)
24 digest[j] ^= hash[j];
25}
26
27void xorObjectDigest(unsigned char *digest, robj *o) {
28 o = getDecodedObject(o);
29 xorDigest(digest,o->ptr,sdslen(o->ptr));
30 decrRefCount(o);
31}
32
33/* This function instead of just computing the SHA1 and xoring it
34 * against diget, also perform the digest of "digest" itself and
35 * replace the old value with the new one.
36 *
37 * So the final digest will be:
38 *
39 * digest = SHA1(digest xor SHA1(data))
40 *
41 * This function is used every time we want to preserve the order so
42 * that digest(a,b,c,d) will be different than digest(b,c,d,a)
43 *
44 * Also note that mixdigest("foo") followed by mixdigest("bar")
45 * will lead to a different digest compared to "fo", "obar".
46 */
47void mixDigest(unsigned char *digest, void *ptr, size_t len) {
48 SHA1_CTX ctx;
49 char *s = ptr;
50
51 xorDigest(digest,s,len);
52 SHA1Init(&ctx);
53 SHA1Update(&ctx,digest,20);
54 SHA1Final(digest,&ctx);
55}
56
57void mixObjectDigest(unsigned char *digest, robj *o) {
58 o = getDecodedObject(o);
59 mixDigest(digest,o->ptr,sdslen(o->ptr));
60 decrRefCount(o);
61}
62
63/* Compute the dataset digest. Since keys, sets elements, hashes elements
64 * are not ordered, we use a trick: every aggregate digest is the xor
65 * of the digests of their elements. This way the order will not change
66 * the result. For list instead we use a feedback entering the output digest
67 * as input in order to ensure that a different ordered list will result in
68 * a different digest. */
69void computeDatasetDigest(unsigned char *final) {
70 unsigned char digest[20];
71 char buf[128];
72 dictIterator *di = NULL;
73 dictEntry *de;
74 int j;
75 uint32_t aux;
76
77 memset(final,0,20); /* Start with a clean result */
78
79 for (j = 0; j < server.dbnum; j++) {
80 redisDb *db = server.db+j;
81
82 if (dictSize(db->dict) == 0) continue;
83 di = dictGetIterator(db->dict);
84
85 /* hash the DB id, so the same dataset moved in a different
86 * DB will lead to a different digest */
87 aux = htonl(j);
88 mixDigest(final,&aux,sizeof(aux));
89
90 /* Iterate this DB writing every entry */
91 while((de = dictNext(di)) != NULL) {
92 sds key;
93 robj *keyobj, *o;
94 long long expiretime;
95
96 memset(digest,0,20); /* This key-val digest */
97 key = dictGetKey(de);
98 keyobj = createStringObject(key,sdslen(key));
99
100 mixDigest(digest,key,sdslen(key));
101
102 /* Make sure the key is loaded if VM is active */
103 o = dictGetVal(de);
104
105 aux = htonl(o->type);
106 mixDigest(digest,&aux,sizeof(aux));
107 expiretime = getExpire(db,keyobj);
108
109 /* Save the key and associated value */
110 if (o->type == REDIS_STRING) {
111 mixObjectDigest(digest,o);
112 } else if (o->type == REDIS_LIST) {
113 listTypeIterator *li = listTypeInitIterator(o,0,REDIS_TAIL);
114 listTypeEntry entry;
115 while(listTypeNext(li,&entry)) {
116 robj *eleobj = listTypeGet(&entry);
117 mixObjectDigest(digest,eleobj);
118 decrRefCount(eleobj);
119 }
120 listTypeReleaseIterator(li);
121 } else if (o->type == REDIS_SET) {
122 setTypeIterator *si = setTypeInitIterator(o);
123 robj *ele;
124 while((ele = setTypeNextObject(si)) != NULL) {
125 xorObjectDigest(digest,ele);
126 decrRefCount(ele);
127 }
128 setTypeReleaseIterator(si);
129 } else if (o->type == REDIS_ZSET) {
130 unsigned char eledigest[20];
131
132 if (o->encoding == REDIS_ENCODING_ZIPLIST) {
133 unsigned char *zl = o->ptr;
134 unsigned char *eptr, *sptr;
135 unsigned char *vstr;
136 unsigned int vlen;
137 long long vll;
138 double score;
139
140 eptr = ziplistIndex(zl,0);
141 redisAssert(eptr != NULL);
142 sptr = ziplistNext(zl,eptr);
143 redisAssert(sptr != NULL);
144
145 while (eptr != NULL) {
146 redisAssert(ziplistGet(eptr,&vstr,&vlen,&vll));
147 score = zzlGetScore(sptr);
148
149 memset(eledigest,0,20);
150 if (vstr != NULL) {
151 mixDigest(eledigest,vstr,vlen);
152 } else {
153 ll2string(buf,sizeof(buf),vll);
154 mixDigest(eledigest,buf,strlen(buf));
155 }
156
157 snprintf(buf,sizeof(buf),"%.17g",score);
158 mixDigest(eledigest,buf,strlen(buf));
159 xorDigest(digest,eledigest,20);
160 zzlNext(zl,&eptr,&sptr);
161 }
162 } else if (o->encoding == REDIS_ENCODING_SKIPLIST) {
163 zset *zs = o->ptr;
164 dictIterator *di = dictGetIterator(zs->dict);
165 dictEntry *de;
166
167 while((de = dictNext(di)) != NULL) {
168 robj *eleobj = dictGetKey(de);
169 double *score = dictGetVal(de);
170
171 snprintf(buf,sizeof(buf),"%.17g",*score);
172 memset(eledigest,0,20);
173 mixObjectDigest(eledigest,eleobj);
174 mixDigest(eledigest,buf,strlen(buf));
175 xorDigest(digest,eledigest,20);
176 }
177 dictReleaseIterator(di);
178 } else {
179 redisPanic("Unknown sorted set encoding");
180 }
181 } else if (o->type == REDIS_HASH) {
182 hashTypeIterator *hi;
183 robj *obj;
184
185 hi = hashTypeInitIterator(o);
186 while (hashTypeNext(hi) != REDIS_ERR) {
187 unsigned char eledigest[20];
188
189 memset(eledigest,0,20);
190 obj = hashTypeCurrentObject(hi,REDIS_HASH_KEY);
191 mixObjectDigest(eledigest,obj);
192 decrRefCount(obj);
193 obj = hashTypeCurrentObject(hi,REDIS_HASH_VALUE);
194 mixObjectDigest(eledigest,obj);
195 decrRefCount(obj);
196 xorDigest(digest,eledigest,20);
197 }
198 hashTypeReleaseIterator(hi);
199 } else {
200 redisPanic("Unknown object type");
201 }
202 /* If the key has an expire, add it to the mix */
203 if (expiretime != -1) xorDigest(digest,"!!expire!!",10);
204 /* We can finally xor the key-val digest to the final digest */
205 xorDigest(final,digest,20);
206 decrRefCount(keyobj);
207 }
208 dictReleaseIterator(di);
209 }
210}
211
212void debugCommand(redisClient *c) {
213 if (!strcasecmp(c->argv[1]->ptr,"segfault")) {
214 *((char*)-1) = 'x';
215 } else if (!strcasecmp(c->argv[1]->ptr,"assert")) {
216 if (c->argc >= 3) c->argv[2] = tryObjectEncoding(c->argv[2]);
217 redisAssertWithInfo(c,c->argv[0],1 == 2);
218 } else if (!strcasecmp(c->argv[1]->ptr,"reload")) {
219 if (rdbSave(server.dbfilename) != REDIS_OK) {
220 addReply(c,shared.err);
221 return;
222 }
223 emptyDb();
224 if (rdbLoad(server.dbfilename) != REDIS_OK) {
225 addReplyError(c,"Error trying to load the RDB dump");
226 return;
227 }
228 redisLog(REDIS_WARNING,"DB reloaded by DEBUG RELOAD");
229 addReply(c,shared.ok);
230 } else if (!strcasecmp(c->argv[1]->ptr,"loadaof")) {
231 emptyDb();
232 if (loadAppendOnlyFile(server.appendfilename) != REDIS_OK) {
233 addReply(c,shared.err);
234 return;
235 }
236 redisLog(REDIS_WARNING,"Append Only File loaded by DEBUG LOADAOF");
237 addReply(c,shared.ok);
238 } else if (!strcasecmp(c->argv[1]->ptr,"object") && c->argc == 3) {
239 dictEntry *de;
240 robj *val;
241 char *strenc;
242
243 if ((de = dictFind(c->db->dict,c->argv[2]->ptr)) == NULL) {
244 addReply(c,shared.nokeyerr);
245 return;
246 }
247 val = dictGetVal(de);
248 strenc = strEncoding(val->encoding);
249
250 addReplyStatusFormat(c,
251 "Value at:%p refcount:%d "
252 "encoding:%s serializedlength:%lld "
253 "lru:%d lru_seconds_idle:%lu",
254 (void*)val, val->refcount,
255 strenc, (long long) rdbSavedObjectLen(val),
256 val->lru, estimateObjectIdleTime(val));
257 } else if (!strcasecmp(c->argv[1]->ptr,"populate") && c->argc == 3) {
258 long keys, j;
259 robj *key, *val;
260 char buf[128];
261
262 if (getLongFromObjectOrReply(c, c->argv[2], &keys, NULL) != REDIS_OK)
263 return;
264 for (j = 0; j < keys; j++) {
265 snprintf(buf,sizeof(buf),"key:%lu",j);
266 key = createStringObject(buf,strlen(buf));
267 if (lookupKeyRead(c->db,key) != NULL) {
268 decrRefCount(key);
269 continue;
270 }
271 snprintf(buf,sizeof(buf),"value:%lu",j);
272 val = createStringObject(buf,strlen(buf));
273 dbAdd(c->db,key,val);
274 decrRefCount(key);
275 }
276 addReply(c,shared.ok);
277 } else if (!strcasecmp(c->argv[1]->ptr,"digest") && c->argc == 2) {
278 unsigned char digest[20];
279 sds d = sdsempty();
280 int j;
281
282 computeDatasetDigest(digest);
283 for (j = 0; j < 20; j++)
284 d = sdscatprintf(d, "%02x",digest[j]);
285 addReplyStatus(c,d);
286 sdsfree(d);
287 } else if (!strcasecmp(c->argv[1]->ptr,"sleep") && c->argc == 3) {
288 double dtime = strtod(c->argv[2]->ptr,NULL);
289 long long utime = dtime*1000000;
290
291 usleep(utime);
292 addReply(c,shared.ok);
293 } else {
294 addReplyError(c,
295 "Syntax error, try DEBUG [SEGFAULT|OBJECT <key>|SWAPIN <key>|SWAPOUT <key>|RELOAD]");
296 }
297}
298
299void _redisAssert(char *estr, char *file, int line) {
300 redisLog(REDIS_WARNING,"=== ASSERTION FAILED ===");
301 redisLog(REDIS_WARNING,"==> %s:%d '%s' is not true",file,line,estr);
302#ifdef HAVE_BACKTRACE
303 redisLog(REDIS_WARNING,"(forcing SIGSEGV in order to print the stack trace)");
304 *((char*)-1) = 'x';
305#endif
306}
307
308void _redisAssertPrintClientInfo(redisClient *c) {
309 int j;
310
311 redisLog(REDIS_WARNING,"=== ASSERTION FAILED CLIENT CONTEXT ===");
312 redisLog(REDIS_WARNING,"client->flags = %d", c->flags);
313 redisLog(REDIS_WARNING,"client->fd = %d", c->fd);
314 redisLog(REDIS_WARNING,"client->argc = %d", c->argc);
315 for (j=0; j < c->argc; j++) {
316 char buf[128];
317 char *arg;
318
319 if (c->argv[j]->type == REDIS_STRING &&
320 c->argv[j]->encoding == REDIS_ENCODING_RAW)
321 {
322 arg = (char*) c->argv[j]->ptr;
323 } else {
324 snprintf(buf,sizeof(buf),"Object type: %d, encoding: %d",
325 c->argv[j]->type, c->argv[j]->encoding);
326 arg = buf;
327 }
328 redisLog(REDIS_WARNING,"client->argv[%d] = \"%s\" (refcount: %d)",
329 j, arg, c->argv[j]->refcount);
330 }
331}
332
333void _redisAssertPrintObject(robj *o) {
334 redisLog(REDIS_WARNING,"=== ASSERTION FAILED OBJECT CONTEXT ===");
335 redisLog(REDIS_WARNING,"Object type: %d", o->type);
336 redisLog(REDIS_WARNING,"Object encoding: %d", o->encoding);
337 redisLog(REDIS_WARNING,"Object refcount: %d", o->refcount);
338 if (o->type == REDIS_STRING && o->encoding == REDIS_ENCODING_RAW) {
339 redisLog(REDIS_WARNING,"Object raw string len: %d", sdslen(o->ptr));
340 if (sdslen(o->ptr) < 4096)
341 redisLog(REDIS_WARNING,"Object raw string content: \"%s\"", (char*)o->ptr);
342 }
343}
344
345void _redisAssertWithInfo(redisClient *c, robj *o, char *estr, char *file, int line) {
346 if (c) _redisAssertPrintClientInfo(c);
347 if (o) _redisAssertPrintObject(o);
348 _redisAssert(estr,file,line);
349}
350
351void _redisPanic(char *msg, char *file, int line) {
352 redisLog(REDIS_WARNING,"------------------------------------------------");
353 redisLog(REDIS_WARNING,"!!! Software Failure. Press left mouse button to continue");
354 redisLog(REDIS_WARNING,"Guru Meditation: %s #%s:%d",msg,file,line);
355#ifdef HAVE_BACKTRACE
356 redisLog(REDIS_WARNING,"(forcing SIGSEGV in order to print the stack trace)");
357 redisLog(REDIS_WARNING,"------------------------------------------------");
358 *((char*)-1) = 'x';
359#endif
360}