]>
Commit | Line | Data |
---|---|---|
e2641e09 | 1 | #include "redis.h" |
2 | #include "sha1.h" /* SHA1 is used for DEBUG DIGEST */ | |
3 | ||
3688d7f3 | 4 | #include <arpa/inet.h> |
5 | ||
e2641e09 | 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) */ | |
14 | void 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 | ||
27 | void 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 | */ | |
47 | void 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 | ||
57 | void 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. */ | |
69 | void 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 | time_t expiretime; | |
95 | ||
96 | memset(digest,0,20); /* This key-val digest */ | |
97 | key = dictGetEntryKey(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 */ | |
f13cb0d9 | 103 | o = dictGetEntryVal(de); |
e2641e09 | 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) { | |
cb72d0f1 | 122 | setTypeIterator *si = setTypeInitIterator(o); |
2767f1c0 | 123 | robj *ele; |
1b508da7 | 124 | while((ele = setTypeNextObject(si)) != NULL) { |
2767f1c0 PN |
125 | xorObjectDigest(digest,ele); |
126 | decrRefCount(ele); | |
e2641e09 | 127 | } |
2767f1c0 | 128 | setTypeReleaseIterator(si); |
e2641e09 | 129 | } else if (o->type == REDIS_ZSET) { |
dddf5335 PN |
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 | } | |
100ed062 | 162 | } else if (o->encoding == REDIS_ENCODING_SKIPLIST) { |
dddf5335 PN |
163 | zset *zs = o->ptr; |
164 | dictIterator *di = dictGetIterator(zs->dict); | |
165 | dictEntry *de; | |
166 | ||
167 | while((de = dictNext(di)) != NULL) { | |
168 | robj *eleobj = dictGetEntryKey(de); | |
169 | double *score = dictGetEntryVal(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"); | |
e2641e09 | 180 | } |
e2641e09 | 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); | |
8c304be3 | 190 | obj = hashTypeCurrentObject(hi,REDIS_HASH_KEY); |
e2641e09 | 191 | mixObjectDigest(eledigest,obj); |
192 | decrRefCount(obj); | |
8c304be3 | 193 | obj = hashTypeCurrentObject(hi,REDIS_HASH_VALUE); |
e2641e09 | 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 | ||
212 | void debugCommand(redisClient *c) { | |
213 | if (!strcasecmp(c->argv[1]->ptr,"segfault")) { | |
214 | *((char*)-1) = 'x'; | |
e3e69935 | 215 | } else if (!strcasecmp(c->argv[1]->ptr,"assert")) { |
216 | if (c->argc >= 3) c->argv[2] = tryObjectEncoding(c->argv[2]); | |
217 | redisAssertWithClientInfo(c,1 == 2); | |
e2641e09 | 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 | addReply(c,shared.err); | |
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) { | |
b39619d8 | 239 | dictEntry *de; |
e2641e09 | 240 | robj *val; |
5ef64098 | 241 | char *strenc; |
e2641e09 | 242 | |
b39619d8 | 243 | if ((de = dictFind(c->db->dict,c->argv[2]->ptr)) == NULL) { |
e2641e09 | 244 | addReply(c,shared.nokeyerr); |
245 | return; | |
246 | } | |
247 | val = dictGetEntryVal(de); | |
5ef64098 | 248 | strenc = strEncoding(val->encoding); |
3be00d7e | 249 | |
5ef64098 | 250 | addReplyStatusFormat(c, |
251 | "Value at:%p refcount:%d " | |
252 | "encoding:%s serializedlength:%lld " | |
3be00d7e | 253 | "lru:%d lru_seconds_idle:%lu", |
5ef64098 | 254 | (void*)val, val->refcount, |
255 | strenc, (long long) rdbSavedObjectLen(val), | |
3be00d7e | 256 | val->lru, estimateObjectIdleTime(val)); |
e2641e09 | 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]; | |
3ab20376 | 279 | sds d = sdsempty(); |
e2641e09 | 280 | int j; |
281 | ||
282 | computeDatasetDigest(digest); | |
283 | for (j = 0; j < 20; j++) | |
284 | d = sdscatprintf(d, "%02x",digest[j]); | |
3ab20376 PN |
285 | addReplyStatus(c,d); |
286 | sdsfree(d); | |
404345d8 | 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); | |
e2641e09 | 293 | } else { |
3ab20376 PN |
294 | addReplyError(c, |
295 | "Syntax error, try DEBUG [SEGFAULT|OBJECT <key>|SWAPIN <key>|SWAPOUT <key>|RELOAD]"); | |
e2641e09 | 296 | } |
297 | } | |
298 | ||
299 | void _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 | ||
e3e69935 | 308 | void _redisAssertPrintClientInfo(redisClient *c) { |
309 | if (c) { | |
310 | int j; | |
311 | ||
312 | redisLog(REDIS_WARNING,"=== ASSERTION FAILED CLIENT CONTEXT ==="); | |
313 | redisLog(REDIS_WARNING,"client->flags = %d", c->flags); | |
314 | redisLog(REDIS_WARNING,"client->fd = %d", c->fd); | |
315 | redisLog(REDIS_WARNING,"client->argc = %d", c->argc); | |
316 | for (j=0; j < c->argc; j++) { | |
317 | char buf[128]; | |
318 | char *arg; | |
319 | ||
320 | if (c->argv[j]->type == REDIS_STRING && | |
321 | c->argv[j]->encoding == REDIS_ENCODING_RAW) | |
322 | { | |
323 | arg = (char*) c->argv[j]->ptr; | |
324 | } else { | |
325 | snprintf(buf,sizeof(buf),"Object type: %d, encoding: %d", | |
326 | c->argv[j]->type, c->argv[j]->encoding); | |
327 | arg = buf; | |
328 | } | |
329 | redisLog(REDIS_WARNING,"client->argv[%d] = \"%s\" (refcount: %d)", | |
330 | j, arg, c->argv[j]->refcount); | |
331 | } | |
332 | } | |
333 | } | |
334 | ||
335 | void _redisAssertWithClientInfo(redisClient *c, char *estr, char *file, int line) { | |
336 | _redisAssertPrintClientInfo(c); | |
337 | _redisAssert(estr,file,line); | |
338 | } | |
339 | ||
e2641e09 | 340 | void _redisPanic(char *msg, char *file, int line) { |
120b9ba8 | 341 | redisLog(REDIS_WARNING,"------------------------------------------------"); |
e2641e09 | 342 | redisLog(REDIS_WARNING,"!!! Software Failure. Press left mouse button to continue"); |
343 | redisLog(REDIS_WARNING,"Guru Meditation: %s #%s:%d",msg,file,line); | |
344 | #ifdef HAVE_BACKTRACE | |
345 | redisLog(REDIS_WARNING,"(forcing SIGSEGV in order to print the stack trace)"); | |
120b9ba8 | 346 | redisLog(REDIS_WARNING,"------------------------------------------------"); |
e2641e09 | 347 | *((char*)-1) = 'x'; |
348 | #endif | |
349 | } |