]> git.saurik.com Git - redis.git/blob - src/replication.c
543477b661a2813e31603fd35267f08311c7d845
[redis.git] / src / replication.c
1 #include "redis.h"
2
3 #include <sys/time.h>
4 #include <unistd.h>
5 #include <fcntl.h>
6 #include <sys/stat.h>
7
8 /* ---------------------------------- MASTER -------------------------------- */
9
10 void replicationFeedSlaves(list *slaves, int dictid, robj **argv, int argc) {
11 listNode *ln;
12 listIter li;
13 int j;
14
15 listRewind(slaves,&li);
16 while((ln = listNext(&li))) {
17 redisClient *slave = ln->value;
18
19 /* Don't feed slaves that are still waiting for BGSAVE to start */
20 if (slave->replstate == REDIS_REPL_WAIT_BGSAVE_START) continue;
21
22 /* Feed slaves that are waiting for the initial SYNC (so these commands
23 * are queued in the output buffer until the intial SYNC completes),
24 * or are already in sync with the master. */
25 if (slave->slaveseldb != dictid) {
26 robj *selectcmd;
27
28 if (dictid >= 0 && dictid < REDIS_SHARED_SELECT_CMDS) {
29 selectcmd = shared.select[dictid];
30 incrRefCount(selectcmd);
31 } else {
32 selectcmd = createObject(REDIS_STRING,
33 sdscatprintf(sdsempty(),"select %d\r\n",dictid));
34 }
35 addReply(slave,selectcmd);
36 decrRefCount(selectcmd);
37 slave->slaveseldb = dictid;
38 }
39 addReplyMultiBulkLen(slave,argc);
40 for (j = 0; j < argc; j++) addReplyBulk(slave,argv[j]);
41 }
42 }
43
44 void replicationFeedMonitors(redisClient *c, list *monitors, int dictid, robj **argv, int argc) {
45 listNode *ln;
46 listIter li;
47 int j, port;
48 sds cmdrepr = sdsnew("+");
49 robj *cmdobj;
50 char ip[32];
51 struct timeval tv;
52
53 gettimeofday(&tv,NULL);
54 cmdrepr = sdscatprintf(cmdrepr,"%ld.%06ld ",(long)tv.tv_sec,(long)tv.tv_usec);
55 if (c->flags & REDIS_LUA_CLIENT) {
56 cmdrepr = sdscatprintf(cmdrepr,"[%d lua] ", dictid);
57 } else {
58 anetPeerToString(c->fd,ip,&port);
59 cmdrepr = sdscatprintf(cmdrepr,"[%d %s:%d] ", dictid,ip,port);
60 }
61
62 for (j = 0; j < argc; j++) {
63 if (argv[j]->encoding == REDIS_ENCODING_INT) {
64 cmdrepr = sdscatprintf(cmdrepr, "\"%ld\"", (long)argv[j]->ptr);
65 } else {
66 cmdrepr = sdscatrepr(cmdrepr,(char*)argv[j]->ptr,
67 sdslen(argv[j]->ptr));
68 }
69 if (j != argc-1)
70 cmdrepr = sdscatlen(cmdrepr," ",1);
71 }
72 cmdrepr = sdscatlen(cmdrepr,"\r\n",2);
73 cmdobj = createObject(REDIS_STRING,cmdrepr);
74
75 listRewind(monitors,&li);
76 while((ln = listNext(&li))) {
77 redisClient *monitor = ln->value;
78 addReply(monitor,cmdobj);
79 }
80 decrRefCount(cmdobj);
81 }
82
83 void syncCommand(redisClient *c) {
84 /* ignore SYNC if aleady slave or in monitor mode */
85 if (c->flags & REDIS_SLAVE) return;
86
87 /* Refuse SYNC requests if we are a slave but the link with our master
88 * is not ok... */
89 if (server.masterhost && server.repl_state != REDIS_REPL_CONNECTED) {
90 addReplyError(c,"Can't SYNC while not connected with my master");
91 return;
92 }
93
94 /* SYNC can't be issued when the server has pending data to send to
95 * the client about already issued commands. We need a fresh reply
96 * buffer registering the differences between the BGSAVE and the current
97 * dataset, so that we can copy to other slaves if needed. */
98 if (listLength(c->reply) != 0) {
99 addReplyError(c,"SYNC is invalid with pending input");
100 return;
101 }
102
103 redisLog(REDIS_NOTICE,"Slave ask for synchronization");
104 /* Here we need to check if there is a background saving operation
105 * in progress, or if it is required to start one */
106 if (server.rdb_child_pid != -1) {
107 /* Ok a background save is in progress. Let's check if it is a good
108 * one for replication, i.e. if there is another slave that is
109 * registering differences since the server forked to save */
110 redisClient *slave;
111 listNode *ln;
112 listIter li;
113
114 listRewind(server.slaves,&li);
115 while((ln = listNext(&li))) {
116 slave = ln->value;
117 if (slave->replstate == REDIS_REPL_WAIT_BGSAVE_END) break;
118 }
119 if (ln) {
120 /* Perfect, the server is already registering differences for
121 * another slave. Set the right state, and copy the buffer. */
122 copyClientOutputBuffer(c,slave);
123 c->replstate = REDIS_REPL_WAIT_BGSAVE_END;
124 redisLog(REDIS_NOTICE,"Waiting for end of BGSAVE for SYNC");
125 } else {
126 /* No way, we need to wait for the next BGSAVE in order to
127 * register differences */
128 c->replstate = REDIS_REPL_WAIT_BGSAVE_START;
129 redisLog(REDIS_NOTICE,"Waiting for next BGSAVE for SYNC");
130 }
131 } else {
132 /* Ok we don't have a BGSAVE in progress, let's start one */
133 redisLog(REDIS_NOTICE,"Starting BGSAVE for SYNC");
134 if (rdbSaveBackground(server.rdb_filename) != REDIS_OK) {
135 redisLog(REDIS_NOTICE,"Replication failed, can't BGSAVE");
136 addReplyError(c,"Unable to perform background save");
137 return;
138 }
139 c->replstate = REDIS_REPL_WAIT_BGSAVE_END;
140 }
141 c->repldbfd = -1;
142 c->flags |= REDIS_SLAVE;
143 c->slaveseldb = 0;
144 listAddNodeTail(server.slaves,c);
145 return;
146 }
147
148 /* REPLCONF <option> <value> <option> <value> ...
149 * This command is used by a slave in order to configure the replication
150 * process before starting it with the SYNC command.
151 *
152 * Currently the only use of this command is to communicate to the master
153 * what is the listening port of the Slave redis instance, so that the
154 * master can accurately list slaves and their listening ports in
155 * the INFO output.
156 *
157 * In the future the same command can be used in order to configure
158 * the replication to initiate an incremental replication instead of a
159 * full resync. */
160 void replconfCommand(redisClient *c) {
161 int j;
162
163 if ((c->argc % 2) == 0) {
164 /* Number of arguments must be odd to make sure that every
165 * option has a corresponding value. */
166 addReply(c,shared.syntaxerr);
167 return;
168 }
169
170 /* Process every option-value pair. */
171 for (j = 1; j < c->argc; j+=2) {
172 if (!strcasecmp(c->argv[j]->ptr,"listening-port")) {
173 long port;
174
175 if ((getLongFromObjectOrReply(c,c->argv[j+1],
176 &port,NULL) != REDIS_OK))
177 return;
178 c->slave_listening_port = port;
179 } else {
180 addReplyErrorFormat(c,"Unrecognized REPLCONF option: %s",
181 (char*)c->argv[j]->ptr);
182 return;
183 }
184 }
185 addReply(c,shared.ok);
186 }
187
188 void sendBulkToSlave(aeEventLoop *el, int fd, void *privdata, int mask) {
189 redisClient *slave = privdata;
190 REDIS_NOTUSED(el);
191 REDIS_NOTUSED(mask);
192 char buf[REDIS_IOBUF_LEN];
193 ssize_t nwritten, buflen;
194
195 if (slave->repldboff == 0) {
196 /* Write the bulk write count before to transfer the DB. In theory here
197 * we don't know how much room there is in the output buffer of the
198 * socket, but in pratice SO_SNDLOWAT (the minimum count for output
199 * operations) will never be smaller than the few bytes we need. */
200 sds bulkcount;
201
202 bulkcount = sdscatprintf(sdsempty(),"$%lld\r\n",(unsigned long long)
203 slave->repldbsize);
204 if (write(fd,bulkcount,sdslen(bulkcount)) != (signed)sdslen(bulkcount))
205 {
206 sdsfree(bulkcount);
207 freeClient(slave);
208 return;
209 }
210 sdsfree(bulkcount);
211 }
212 lseek(slave->repldbfd,slave->repldboff,SEEK_SET);
213 buflen = read(slave->repldbfd,buf,REDIS_IOBUF_LEN);
214 if (buflen <= 0) {
215 redisLog(REDIS_WARNING,"Read error sending DB to slave: %s",
216 (buflen == 0) ? "premature EOF" : strerror(errno));
217 freeClient(slave);
218 return;
219 }
220 if ((nwritten = write(fd,buf,buflen)) == -1) {
221 redisLog(REDIS_VERBOSE,"Write error sending DB to slave: %s",
222 strerror(errno));
223 freeClient(slave);
224 return;
225 }
226 slave->repldboff += nwritten;
227 if (slave->repldboff == slave->repldbsize) {
228 close(slave->repldbfd);
229 slave->repldbfd = -1;
230 aeDeleteFileEvent(server.el,slave->fd,AE_WRITABLE);
231 slave->replstate = REDIS_REPL_ONLINE;
232 if (aeCreateFileEvent(server.el, slave->fd, AE_WRITABLE,
233 sendReplyToClient, slave) == AE_ERR) {
234 freeClient(slave);
235 return;
236 }
237 redisLog(REDIS_NOTICE,"Synchronization with slave succeeded");
238 }
239 }
240
241 /* This function is called at the end of every backgrond saving.
242 * The argument bgsaveerr is REDIS_OK if the background saving succeeded
243 * otherwise REDIS_ERR is passed to the function.
244 *
245 * The goal of this function is to handle slaves waiting for a successful
246 * background saving in order to perform non-blocking synchronization. */
247 void updateSlavesWaitingBgsave(int bgsaveerr) {
248 listNode *ln;
249 int startbgsave = 0;
250 listIter li;
251
252 listRewind(server.slaves,&li);
253 while((ln = listNext(&li))) {
254 redisClient *slave = ln->value;
255
256 if (slave->replstate == REDIS_REPL_WAIT_BGSAVE_START) {
257 startbgsave = 1;
258 slave->replstate = REDIS_REPL_WAIT_BGSAVE_END;
259 } else if (slave->replstate == REDIS_REPL_WAIT_BGSAVE_END) {
260 struct redis_stat buf;
261
262 if (bgsaveerr != REDIS_OK) {
263 freeClient(slave);
264 redisLog(REDIS_WARNING,"SYNC failed. BGSAVE child returned an error");
265 continue;
266 }
267 if ((slave->repldbfd = open(server.rdb_filename,O_RDONLY)) == -1 ||
268 redis_fstat(slave->repldbfd,&buf) == -1) {
269 freeClient(slave);
270 redisLog(REDIS_WARNING,"SYNC failed. Can't open/stat DB after BGSAVE: %s", strerror(errno));
271 continue;
272 }
273 slave->repldboff = 0;
274 slave->repldbsize = buf.st_size;
275 slave->replstate = REDIS_REPL_SEND_BULK;
276 aeDeleteFileEvent(server.el,slave->fd,AE_WRITABLE);
277 if (aeCreateFileEvent(server.el, slave->fd, AE_WRITABLE, sendBulkToSlave, slave) == AE_ERR) {
278 freeClient(slave);
279 continue;
280 }
281 }
282 }
283 if (startbgsave) {
284 if (rdbSaveBackground(server.rdb_filename) != REDIS_OK) {
285 listIter li;
286
287 listRewind(server.slaves,&li);
288 redisLog(REDIS_WARNING,"SYNC failed. BGSAVE failed");
289 while((ln = listNext(&li))) {
290 redisClient *slave = ln->value;
291
292 if (slave->replstate == REDIS_REPL_WAIT_BGSAVE_START)
293 freeClient(slave);
294 }
295 }
296 }
297 }
298
299 /* ----------------------------------- SLAVE -------------------------------- */
300
301 /* Abort the async download of the bulk dataset while SYNC-ing with master */
302 void replicationAbortSyncTransfer(void) {
303 redisAssert(server.repl_state == REDIS_REPL_TRANSFER);
304
305 aeDeleteFileEvent(server.el,server.repl_transfer_s,AE_READABLE);
306 close(server.repl_transfer_s);
307 close(server.repl_transfer_fd);
308 unlink(server.repl_transfer_tmpfile);
309 zfree(server.repl_transfer_tmpfile);
310 server.repl_state = REDIS_REPL_CONNECT;
311 }
312
313 /* Asynchronously read the SYNC payload we receive from a master */
314 #define REPL_MAX_WRITTEN_BEFORE_FSYNC (1024*1024*8) /* 8 MB */
315 void readSyncBulkPayload(aeEventLoop *el, int fd, void *privdata, int mask) {
316 char buf[4096];
317 ssize_t nread, readlen;
318 off_t left;
319 REDIS_NOTUSED(el);
320 REDIS_NOTUSED(privdata);
321 REDIS_NOTUSED(mask);
322
323 /* If repl_transfer_size == -1 we still have to read the bulk length
324 * from the master reply. */
325 if (server.repl_transfer_size == -1) {
326 if (syncReadLine(fd,buf,1024,server.repl_syncio_timeout*1000) == -1) {
327 redisLog(REDIS_WARNING,
328 "I/O error reading bulk count from MASTER: %s",
329 strerror(errno));
330 goto error;
331 }
332
333 if (buf[0] == '-') {
334 redisLog(REDIS_WARNING,
335 "MASTER aborted replication with an error: %s",
336 buf+1);
337 goto error;
338 } else if (buf[0] == '\0') {
339 /* At this stage just a newline works as a PING in order to take
340 * the connection live. So we refresh our last interaction
341 * timestamp. */
342 server.repl_transfer_lastio = server.unixtime;
343 return;
344 } else if (buf[0] != '$') {
345 redisLog(REDIS_WARNING,"Bad protocol from MASTER, the first byte is not '$', are you sure the host and port are right?");
346 goto error;
347 }
348 server.repl_transfer_size = strtol(buf+1,NULL,10);
349 redisLog(REDIS_NOTICE,
350 "MASTER <-> SLAVE sync: receiving %ld bytes from master",
351 server.repl_transfer_size);
352 return;
353 }
354
355 /* Read bulk data */
356 left = server.repl_transfer_size - server.repl_transfer_read;
357 readlen = (left < (signed)sizeof(buf)) ? left : (signed)sizeof(buf);
358 nread = read(fd,buf,readlen);
359 if (nread <= 0) {
360 redisLog(REDIS_WARNING,"I/O error trying to sync with MASTER: %s",
361 (nread == -1) ? strerror(errno) : "connection lost");
362 replicationAbortSyncTransfer();
363 return;
364 }
365 server.repl_transfer_lastio = server.unixtime;
366 if (write(server.repl_transfer_fd,buf,nread) != nread) {
367 redisLog(REDIS_WARNING,"Write error or short write writing to the DB dump file needed for MASTER <-> SLAVE synchronization: %s", strerror(errno));
368 goto error;
369 }
370 server.repl_transfer_read += nread;
371
372 /* Sync data on disk from time to time, otherwise at the end of the transfer
373 * we may suffer a big delay as the memory buffers are copied into the
374 * actual disk. */
375 if (server.repl_transfer_read >=
376 server.repl_transfer_last_fsync_off + REPL_MAX_WRITTEN_BEFORE_FSYNC)
377 {
378 off_t sync_size = server.repl_transfer_read -
379 server.repl_transfer_last_fsync_off;
380 rdb_fsync_range(server.repl_transfer_fd,
381 server.repl_transfer_last_fsync_off, sync_size);
382 server.repl_transfer_last_fsync_off += sync_size;
383 }
384
385 /* Check if the transfer is now complete */
386 if (server.repl_transfer_read == server.repl_transfer_size) {
387 if (rename(server.repl_transfer_tmpfile,server.rdb_filename) == -1) {
388 redisLog(REDIS_WARNING,"Failed trying to rename the temp DB into dump.rdb in MASTER <-> SLAVE synchronization: %s", strerror(errno));
389 replicationAbortSyncTransfer();
390 return;
391 }
392 redisLog(REDIS_NOTICE, "MASTER <-> SLAVE sync: Loading DB in memory");
393 emptyDb();
394 /* Before loading the DB into memory we need to delete the readable
395 * handler, otherwise it will get called recursively since
396 * rdbLoad() will call the event loop to process events from time to
397 * time for non blocking loading. */
398 aeDeleteFileEvent(server.el,server.repl_transfer_s,AE_READABLE);
399 if (rdbLoad(server.rdb_filename) != REDIS_OK) {
400 redisLog(REDIS_WARNING,"Failed trying to load the MASTER synchronization DB from disk");
401 replicationAbortSyncTransfer();
402 return;
403 }
404 /* Final setup of the connected slave <- master link */
405 zfree(server.repl_transfer_tmpfile);
406 close(server.repl_transfer_fd);
407 server.master = createClient(server.repl_transfer_s);
408 server.master->flags |= REDIS_MASTER;
409 server.master->authenticated = 1;
410 server.repl_state = REDIS_REPL_CONNECTED;
411 redisLog(REDIS_NOTICE, "MASTER <-> SLAVE sync: Finished with success");
412 /* Restart the AOF subsystem now that we finished the sync. This
413 * will trigger an AOF rewrite, and when done will start appending
414 * to the new file. */
415 if (server.aof_state != REDIS_AOF_OFF) {
416 int retry = 10;
417
418 stopAppendOnly();
419 while (retry-- && startAppendOnly() == REDIS_ERR) {
420 redisLog(REDIS_WARNING,"Failed enabling the AOF after successful master synchrnization! Trying it again in one second.");
421 sleep(1);
422 }
423 if (!retry) {
424 redisLog(REDIS_WARNING,"FATAL: this slave instance finished the synchronization with its master, but the AOF can't be turned on. Exiting now.");
425 exit(1);
426 }
427 }
428 }
429
430 return;
431
432 error:
433 replicationAbortSyncTransfer();
434 return;
435 }
436
437 /* Send a synchronous command to the master. Used to send AUTH and
438 * REPLCONF commands before starting the replication with SYNC.
439 *
440 * On success NULL is returned.
441 * On error an sds string describing the error is returned.
442 */
443 char *sendSynchronousCommand(int fd, ...) {
444 va_list ap;
445 sds cmd = sdsempty();
446 char *arg, buf[256];
447
448 /* Create the command to send to the master, we use simple inline
449 * protocol for simplicity as currently we only send simple strings. */
450 va_start(ap,fd);
451 while(1) {
452 arg = va_arg(ap, char*);
453 if (arg == NULL) break;
454
455 if (sdslen(cmd) != 0) cmd = sdscatlen(cmd," ",1);
456 cmd = sdscat(cmd,arg);
457 }
458 cmd = sdscatlen(cmd,"\r\n",2);
459
460 /* Transfer command to the server. */
461 if (syncWrite(fd,cmd,sdslen(cmd),server.repl_syncio_timeout*1000) == -1) {
462 sdsfree(cmd);
463 return sdscatprintf(sdsempty(),"Writing to master: %s",
464 strerror(errno));
465 }
466 sdsfree(cmd);
467
468 /* Read the reply from the server. */
469 if (syncReadLine(fd,buf,sizeof(buf),server.repl_syncio_timeout*1000) == -1)
470 {
471 return sdscatprintf(sdsempty(),"Reading from master: %s",
472 strerror(errno));
473 }
474
475 /* Check for errors from the server. */
476 if (buf[0] != '+') {
477 return sdscatprintf(sdsempty(),"Error from master: %s", buf);
478 }
479
480 return NULL; /* No errors. */
481 }
482
483 void syncWithMaster(aeEventLoop *el, int fd, void *privdata, int mask) {
484 char tmpfile[256], *err;
485 int dfd, maxtries = 5;
486 REDIS_NOTUSED(el);
487 REDIS_NOTUSED(privdata);
488 REDIS_NOTUSED(mask);
489
490 /* If this event fired after the user turned the instance into a master
491 * with SLAVEOF NO ONE we must just return ASAP. */
492 if (server.repl_state == REDIS_REPL_NONE) {
493 close(fd);
494 return;
495 }
496
497 redisLog(REDIS_NOTICE,"Non blocking connect for SYNC fired the event.");
498 /* This event should only be triggered once since it is used to have a
499 * non-blocking connect(2) to the master. It has been triggered when this
500 * function is called, so we can delete it. */
501 aeDeleteFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE);
502
503 /* AUTH with the master if required. */
504 if(server.masterauth) {
505 err = sendSynchronousCommand(fd,"AUTH",server.masterauth,NULL);
506 if (err) {
507 redisLog(REDIS_WARNING,"Unable to AUTH to MASTER: %s",err);
508 sdsfree(err);
509 goto error;
510 }
511 }
512
513 /* Set the slave port, so that Master's INFO command can list the
514 * slave listening port correctly. */
515 {
516 sds port = sdsfromlonglong(server.port);
517 err = sendSynchronousCommand(fd,"REPLCONF","listening-port",port,
518 NULL);
519 sdsfree(port);
520 /* Ignore the error if any, not all the Redis versions support
521 * REPLCONF listening-port. */
522 if (err) {
523 redisLog(REDIS_NOTICE,"(non critical): Master does not understand REPLCONF listening-port: %s", err);
524 sdsfree(err);
525 }
526 }
527
528 /* Issue the SYNC command */
529 if (syncWrite(fd,"SYNC\r\n",6,server.repl_syncio_timeout*1000) == -1) {
530 redisLog(REDIS_WARNING,"I/O error writing to MASTER: %s",
531 strerror(errno));
532 goto error;
533 }
534
535 /* Prepare a suitable temp file for bulk transfer */
536 while(maxtries--) {
537 snprintf(tmpfile,256,
538 "temp-%d.%ld.rdb",(int)server.unixtime,(long int)getpid());
539 dfd = open(tmpfile,O_CREAT|O_WRONLY|O_EXCL,0644);
540 if (dfd != -1) break;
541 sleep(1);
542 }
543 if (dfd == -1) {
544 redisLog(REDIS_WARNING,"Opening the temp file needed for MASTER <-> SLAVE synchronization: %s",strerror(errno));
545 goto error;
546 }
547
548 /* Setup the non blocking download of the bulk file. */
549 if (aeCreateFileEvent(server.el,fd, AE_READABLE,readSyncBulkPayload,NULL)
550 == AE_ERR)
551 {
552 redisLog(REDIS_WARNING,"Can't create readable event for SYNC");
553 goto error;
554 }
555
556 server.repl_state = REDIS_REPL_TRANSFER;
557 server.repl_transfer_size = -1;
558 server.repl_transfer_read = 0;
559 server.repl_transfer_last_fsync_off = 0;
560 server.repl_transfer_fd = dfd;
561 server.repl_transfer_lastio = server.unixtime;
562 server.repl_transfer_tmpfile = zstrdup(tmpfile);
563 return;
564
565 error:
566 server.repl_state = REDIS_REPL_CONNECT;
567 close(fd);
568 return;
569 }
570
571 int connectWithMaster(void) {
572 int fd;
573
574 fd = anetTcpNonBlockConnect(NULL,server.masterhost,server.masterport);
575 if (fd == -1) {
576 redisLog(REDIS_WARNING,"Unable to connect to MASTER: %s",
577 strerror(errno));
578 return REDIS_ERR;
579 }
580
581 if (aeCreateFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE,syncWithMaster,NULL) ==
582 AE_ERR)
583 {
584 close(fd);
585 redisLog(REDIS_WARNING,"Can't create readable event for SYNC");
586 return REDIS_ERR;
587 }
588
589 server.repl_transfer_lastio = server.unixtime;
590 server.repl_transfer_s = fd;
591 server.repl_state = REDIS_REPL_CONNECTING;
592 return REDIS_OK;
593 }
594
595 /* This function can be called when a non blocking connection is currently
596 * in progress to undo it. */
597 void undoConnectWithMaster(void) {
598 int fd = server.repl_transfer_s;
599
600 redisAssert(server.repl_state == REDIS_REPL_CONNECTING);
601 aeDeleteFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE);
602 close(fd);
603 server.repl_transfer_s = -1;
604 server.repl_state = REDIS_REPL_CONNECT;
605 }
606
607 void slaveofCommand(redisClient *c) {
608 if (!strcasecmp(c->argv[1]->ptr,"no") &&
609 !strcasecmp(c->argv[2]->ptr,"one")) {
610 if (server.masterhost) {
611 sdsfree(server.masterhost);
612 server.masterhost = NULL;
613 if (server.master) freeClient(server.master);
614 if (server.repl_state == REDIS_REPL_TRANSFER)
615 replicationAbortSyncTransfer();
616 else if (server.repl_state == REDIS_REPL_CONNECTING)
617 undoConnectWithMaster();
618 server.repl_state = REDIS_REPL_NONE;
619 redisLog(REDIS_NOTICE,"MASTER MODE enabled (user request)");
620 }
621 } else {
622 long port;
623
624 if ((getLongFromObjectOrReply(c, c->argv[2], &port, NULL) != REDIS_OK))
625 return;
626
627 /* Check if we are already attached to the specified slave */
628 if (server.masterhost && !strcasecmp(server.masterhost,c->argv[1]->ptr)
629 && server.masterport == port) {
630 redisLog(REDIS_NOTICE,"SLAVE OF would result into synchronization with the master we are already connected with. No operation performed.");
631 addReplySds(c,sdsnew("+OK Already connected to specified master\r\n"));
632 return;
633 }
634 /* There was no previous master or the user specified a different one,
635 * we can continue. */
636 sdsfree(server.masterhost);
637 server.masterhost = sdsdup(c->argv[1]->ptr);
638 server.masterport = port;
639 if (server.master) freeClient(server.master);
640 disconnectSlaves(); /* Force our slaves to resync with us as well. */
641 if (server.repl_state == REDIS_REPL_TRANSFER)
642 replicationAbortSyncTransfer();
643 server.repl_state = REDIS_REPL_CONNECT;
644 redisLog(REDIS_NOTICE,"SLAVE OF %s:%d enabled (user request)",
645 server.masterhost, server.masterport);
646 }
647 addReply(c,shared.ok);
648 }
649
650 /* --------------------------- REPLICATION CRON ---------------------------- */
651
652 void replicationCron(void) {
653 /* Non blocking connection timeout? */
654 if (server.masterhost && server.repl_state == REDIS_REPL_CONNECTING &&
655 (time(NULL)-server.repl_transfer_lastio) > server.repl_timeout)
656 {
657 redisLog(REDIS_WARNING,"Timeout connecting to the MASTER...");
658 undoConnectWithMaster();
659 }
660
661 /* Bulk transfer I/O timeout? */
662 if (server.masterhost && server.repl_state == REDIS_REPL_TRANSFER &&
663 (time(NULL)-server.repl_transfer_lastio) > server.repl_timeout)
664 {
665 redisLog(REDIS_WARNING,"Timeout receiving bulk data from MASTER...");
666 replicationAbortSyncTransfer();
667 }
668
669 /* Timed out master when we are an already connected slave? */
670 if (server.masterhost && server.repl_state == REDIS_REPL_CONNECTED &&
671 (time(NULL)-server.master->lastinteraction) > server.repl_timeout)
672 {
673 redisLog(REDIS_WARNING,"MASTER time out: no data nor PING received...");
674 freeClient(server.master);
675 }
676
677 /* Check if we should connect to a MASTER */
678 if (server.repl_state == REDIS_REPL_CONNECT) {
679 redisLog(REDIS_NOTICE,"Connecting to MASTER...");
680 if (connectWithMaster() == REDIS_OK) {
681 redisLog(REDIS_NOTICE,"MASTER <-> SLAVE sync started");
682 }
683 }
684
685 /* If we have attached slaves, PING them from time to time.
686 * So slaves can implement an explicit timeout to masters, and will
687 * be able to detect a link disconnection even if the TCP connection
688 * will not actually go down. */
689 if (!(server.cronloops % (server.repl_ping_slave_period*10))) {
690 listIter li;
691 listNode *ln;
692
693 listRewind(server.slaves,&li);
694 while((ln = listNext(&li))) {
695 redisClient *slave = ln->value;
696
697 /* Don't ping slaves that are in the middle of a bulk transfer
698 * with the master for first synchronization. */
699 if (slave->replstate == REDIS_REPL_SEND_BULK) continue;
700 if (slave->replstate == REDIS_REPL_ONLINE) {
701 /* If the slave is online send a normal ping */
702 addReplySds(slave,sdsnew("*1\r\n$4\r\nPING\r\n"));
703 } else {
704 /* Otherwise we are in the pre-synchronization stage.
705 * Just a newline will do the work of refreshing the
706 * connection last interaction time, and at the same time
707 * we'll be sure that being a single char there are no
708 * short-write problems. */
709 if (write(slave->fd, "\n", 1) == -1) {
710 /* Don't worry, it's just a ping. */
711 }
712 }
713 }
714 }
715 }