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) */ 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 */ 103 o = dictGetEntryVal(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 = 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"); 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 212 void debugCommand(redisClient *c) { 213 if (!strcasecmp(c->argv[1]->ptr,"segfault")) { 214 *((char*)-1) = 'x'; 215 } else if (!strcasecmp(c->argv[1]->ptr,"flushcache")) { 216 if (!server.ds_enabled) { 217 addReplyError(c, "DEBUG FLUSHCACHE called with diskstore off."); 218 return; 219 } else if (server.bgsavethread != (pthread_t) -1) { 220 addReplyError(c, "Can't flush cache while BGSAVE is in progress."); 221 return; 222 } else { 223 /* To flush the whole cache we need to wait for everything to 224 * be flushed on disk... */ 225 cacheForcePointInTime(); 226 emptyDb(); 227 addReply(c,shared.ok); 228 return; 229 } 230 } else if (!strcasecmp(c->argv[1]->ptr,"reload")) { 231 if (server.ds_enabled) { 232 addReply(c,shared.ok); 233 return; 234 } 235 if (rdbSave(server.dbfilename) != REDIS_OK) { 236 addReply(c,shared.err); 237 return; 238 } 239 emptyDb(); 240 if (rdbLoad(server.dbfilename) != REDIS_OK) { 241 addReply(c,shared.err); 242 return; 243 } 244 redisLog(REDIS_WARNING,"DB reloaded by DEBUG RELOAD"); 245 addReply(c,shared.ok); 246 } else if (!strcasecmp(c->argv[1]->ptr,"loadaof")) { 247 emptyDb(); 248 if (loadAppendOnlyFile(server.appendfilename) != REDIS_OK) { 249 addReply(c,shared.err); 250 return; 251 } 252 redisLog(REDIS_WARNING,"Append Only File loaded by DEBUG LOADAOF"); 253 addReply(c,shared.ok); 254 } else if (!strcasecmp(c->argv[1]->ptr,"object") && c->argc == 3) { 255 dictEntry *de; 256 robj *val; 257 char *strenc; 258 259 if (server.ds_enabled) lookupKeyRead(c->db,c->argv[2]); 260 if ((de = dictFind(c->db->dict,c->argv[2]->ptr)) == NULL) { 261 addReply(c,shared.nokeyerr); 262 return; 263 } 264 val = dictGetEntryVal(de); 265 strenc = strEncoding(val->encoding); 266 267 addReplyStatusFormat(c, 268 "Value at:%p refcount:%d " 269 "encoding:%s serializedlength:%lld " 270 "lru:%d lru_seconds_idle:%lu", 271 (void*)val, val->refcount, 272 strenc, (long long) rdbSavedObjectLen(val), 273 val->lru, estimateObjectIdleTime(val)); 274 } else if (!strcasecmp(c->argv[1]->ptr,"populate") && c->argc == 3) { 275 long keys, j; 276 robj *key, *val; 277 char buf[128]; 278 279 if (getLongFromObjectOrReply(c, c->argv[2], &keys, NULL) != REDIS_OK) 280 return; 281 for (j = 0; j < keys; j++) { 282 snprintf(buf,sizeof(buf),"key:%lu",j); 283 key = createStringObject(buf,strlen(buf)); 284 if (lookupKeyRead(c->db,key) != NULL) { 285 decrRefCount(key); 286 continue; 287 } 288 snprintf(buf,sizeof(buf),"value:%lu",j); 289 val = createStringObject(buf,strlen(buf)); 290 dbAdd(c->db,key,val); 291 decrRefCount(key); 292 } 293 addReply(c,shared.ok); 294 } else if (!strcasecmp(c->argv[1]->ptr,"digest") && c->argc == 2) { 295 unsigned char digest[20]; 296 sds d = sdsempty(); 297 int j; 298 299 computeDatasetDigest(digest); 300 for (j = 0; j < 20; j++) 301 d = sdscatprintf(d, "%02x",digest[j]); 302 addReplyStatus(c,d); 303 sdsfree(d); 304 } else { 305 addReplyError(c, 306 "Syntax error, try DEBUG [SEGFAULT|OBJECT <key>|SWAPIN <key>|SWAPOUT <key>|RELOAD]"); 307 } 308 } 309 310 void _redisAssert(char *estr, char *file, int line) { 311 redisLog(REDIS_WARNING,"=== ASSERTION FAILED ==="); 312 redisLog(REDIS_WARNING,"==> %s:%d '%s' is not true",file,line,estr); 313 #ifdef HAVE_BACKTRACE 314 redisLog(REDIS_WARNING,"(forcing SIGSEGV in order to print the stack trace)"); 315 *((char*)-1) = 'x'; 316 #endif 317 } 318 319 void _redisPanic(char *msg, char *file, int line) { 320 redisLog(REDIS_WARNING,"------------------------------------------------"); 321 redisLog(REDIS_WARNING,"!!! Software Failure. Press left mouse button to continue"); 322 redisLog(REDIS_WARNING,"Guru Meditation: %s #%s:%d",msg,file,line); 323 #ifdef HAVE_BACKTRACE 324 redisLog(REDIS_WARNING,"(forcing SIGSEGV in order to print the stack trace)"); 325 redisLog(REDIS_WARNING,"------------------------------------------------"); 326 *((char*)-1) = 'x'; 327 #endif 328 } 329