]> git.saurik.com Git - redis.git/blame - src/replication.c
Scripting: Reset Lua fake client reply_bytes after command execution.
[redis.git] / src / replication.c
CommitLineData
e2641e09 1#include "redis.h"
2
3#include <sys/time.h>
4#include <unistd.h>
5#include <fcntl.h>
6#include <sys/stat.h>
7
f4aa600b 8/* ---------------------------------- MASTER -------------------------------- */
9
e2641e09 10void replicationFeedSlaves(list *slaves, int dictid, robj **argv, int argc) {
11 listNode *ln;
12 listIter li;
632e4c09 13 int j;
e2641e09 14
e2641e09 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
632e4c09
PN
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. */
e2641e09 25 if (slave->slaveseldb != dictid) {
26 robj *selectcmd;
27
f892797e 28 if (dictid >= 0 && dictid < REDIS_SHARED_SELECT_CMDS) {
f892797e 29 selectcmd = shared.select[dictid];
c2672a06 30 incrRefCount(selectcmd);
f892797e 31 } else {
e2641e09 32 selectcmd = createObject(REDIS_STRING,
33 sdscatprintf(sdsempty(),"select %d\r\n",dictid));
e2641e09 34 }
35 addReply(slave,selectcmd);
f892797e 36 decrRefCount(selectcmd);
e2641e09 37 slave->slaveseldb = dictid;
38 }
632e4c09
PN
39 addReplyMultiBulkLen(slave,argc);
40 for (j = 0; j < argc; j++) addReplyBulk(slave,argv[j]);
e2641e09 41 }
e2641e09 42}
43
e31b615e 44void replicationFeedMonitors(redisClient *c, list *monitors, int dictid, robj **argv, int argc) {
e2641e09 45 listNode *ln;
46 listIter li;
e31b615e 47 int j, port;
e2641e09 48 sds cmdrepr = sdsnew("+");
49 robj *cmdobj;
e31b615e 50 char ip[32];
e2641e09 51 struct timeval tv;
52
53 gettimeofday(&tv,NULL);
2b2eca1f 54 cmdrepr = sdscatprintf(cmdrepr,"%ld.%06ld ",(long)tv.tv_sec,(long)tv.tv_usec);
e31b615e 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 }
e2641e09 61
62 for (j = 0; j < argc; j++) {
63 if (argv[j]->encoding == REDIS_ENCODING_INT) {
d3b958c3 64 cmdrepr = sdscatprintf(cmdrepr, "\"%ld\"", (long)argv[j]->ptr);
e2641e09 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
e2641e09 83void syncCommand(redisClient *c) {
84 /* ignore SYNC if aleady slave or in monitor mode */
85 if (c->flags & REDIS_SLAVE) return;
86
778b2210 87 /* Refuse SYNC requests if we are a slave but the link with our master
88 * is not ok... */
1844f990 89 if (server.masterhost && server.repl_state != REDIS_REPL_CONNECTED) {
3ab20376 90 addReplyError(c,"Can't SYNC while not connected with my master");
778b2210 91 return;
92 }
93
e2641e09 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) {
3ab20376 99 addReplyError(c,"SYNC is invalid with pending input");
e2641e09 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 */
f48cd4b9 106 if (server.rdb_child_pid != -1) {
e2641e09 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. */
1824e3a3 122 copyClientOutputBuffer(c,slave);
e2641e09 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");
f48cd4b9 134 if (rdbSaveBackground(server.rdb_filename) != REDIS_OK) {
e2641e09 135 redisLog(REDIS_NOTICE,"Replication failed, can't BGSAVE");
3ab20376 136 addReplyError(c,"Unable to perform background save");
e2641e09 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
3a328978 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. */
160void 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
e2641e09 188void 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 }
e2641e09 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. */
247void 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 }
f48cd4b9 267 if ((slave->repldbfd = open(server.rdb_filename,O_RDONLY)) == -1 ||
e2641e09 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) {
f48cd4b9 284 if (rdbSaveBackground(server.rdb_filename) != REDIS_OK) {
e2641e09 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
f4aa600b 299/* ----------------------------------- SLAVE -------------------------------- */
300
301/* Abort the async download of the bulk dataset while SYNC-ing with master */
302void replicationAbortSyncTransfer(void) {
1844f990 303 redisAssert(server.repl_state == REDIS_REPL_TRANSFER);
f4aa600b 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);
1844f990 310 server.repl_state = REDIS_REPL_CONNECT;
f4aa600b 311}
312
313/* Asynchronously read the SYNC payload we receive from a master */
784b9308 314#define REPL_MAX_WRITTEN_BEFORE_FSYNC (1024*1024*8) /* 8 MB */
f4aa600b 315void readSyncBulkPayload(aeEventLoop *el, int fd, void *privdata, int mask) {
26b33669 316 char buf[4096];
62ec599c 317 ssize_t nread, readlen;
784b9308 318 off_t left;
62ec599c 319 REDIS_NOTUSED(el);
320 REDIS_NOTUSED(privdata);
321 REDIS_NOTUSED(mask);
f4aa600b 322
784b9308 323 /* If repl_transfer_size == -1 we still have to read the bulk length
26b33669 324 * from the master reply. */
784b9308 325 if (server.repl_transfer_size == -1) {
9157549f 326 if (syncReadLine(fd,buf,1024,server.repl_syncio_timeout*1000) == -1) {
26b33669 327 redisLog(REDIS_WARNING,
328 "I/O error reading bulk count from MASTER: %s",
329 strerror(errno));
b075621f 330 goto error;
26b33669 331 }
b075621f 332
26b33669 333 if (buf[0] == '-') {
334 redisLog(REDIS_WARNING,
335 "MASTER aborted replication with an error: %s",
336 buf+1);
b075621f 337 goto error;
89a1433e 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. */
d1949054 342 server.repl_transfer_lastio = server.unixtime;
89a1433e 343 return;
26b33669 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?");
b075621f 346 goto error;
26b33669 347 }
784b9308 348 server.repl_transfer_size = strtol(buf+1,NULL,10);
26b33669 349 redisLog(REDIS_NOTICE,
350 "MASTER <-> SLAVE sync: receiving %ld bytes from master",
784b9308 351 server.repl_transfer_size);
26b33669 352 return;
353 }
354
355 /* Read bulk data */
784b9308 356 left = server.repl_transfer_size - server.repl_transfer_read;
357 readlen = (left < (signed)sizeof(buf)) ? left : (signed)sizeof(buf);
f4aa600b 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 }
d1949054 365 server.repl_transfer_lastio = server.unixtime;
f4aa600b 366 if (write(server.repl_transfer_fd,buf,nread) != nread) {
31788f50 367 redisLog(REDIS_WARNING,"Write error or short write writing to the DB dump file needed for MASTER <-> SLAVE synchronization: %s", strerror(errno));
b075621f 368 goto error;
f4aa600b 369 }
784b9308 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
f4aa600b 385 /* Check if the transfer is now complete */
784b9308 386 if (server.repl_transfer_read == server.repl_transfer_size) {
f48cd4b9 387 if (rename(server.repl_transfer_tmpfile,server.rdb_filename) == -1) {
f4aa600b 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 }
f6433915 392 redisLog(REDIS_NOTICE, "MASTER <-> SLAVE sync: Loading DB in memory");
f4aa600b 393 emptyDb();
9fd01051 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);
f48cd4b9 399 if (rdbLoad(server.rdb_filename) != REDIS_OK) {
f4aa600b 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 */
f4aa600b 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;
1844f990 410 server.repl_state = REDIS_REPL_CONNECTED;
26b33669 411 redisLog(REDIS_NOTICE, "MASTER <-> SLAVE sync: Finished with success");
e7a2e7c1 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. */
e394114d 415 if (server.aof_state != REDIS_AOF_OFF) {
e7a2e7c1 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 }
f4aa600b 428 }
b075621f
PN
429
430 return;
431
432error:
433 replicationAbortSyncTransfer();
434 return;
f4aa600b 435}
436
3a328978 437/* Send a synchronous command to the master. Used to send AUTH and
36def8fd 438 * REPLCONF commands before starting the replication with SYNC.
3a328978 439 *
440 * On success NULL is returned.
441 * On error an sds string describing the error is returned.
442 */
443char *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
a3309139 483void syncWithMaster(aeEventLoop *el, int fd, void *privdata, int mask) {
3a328978 484 char tmpfile[256], *err;
e2641e09 485 int dfd, maxtries = 5;
a3309139
PN
486 REDIS_NOTUSED(el);
487 REDIS_NOTUSED(privdata);
488 REDIS_NOTUSED(mask);
e2641e09 489
76e772f3 490 /* If this event fired after the user turned the instance into a master
491 * with SLAVEOF NO ONE we must just return ASAP. */
1844f990 492 if (server.repl_state == REDIS_REPL_NONE) {
76e772f3 493 close(fd);
494 return;
495 }
496
45029d37 497 redisLog(REDIS_NOTICE,"Non blocking connect for SYNC fired the event.");
b075621f
PN
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. */
45029d37 501 aeDeleteFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE);
e2641e09 502
503 /* AUTH with the master if required. */
504 if(server.masterauth) {
3a328978 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);
a3309139
PN
509 goto error;
510 }
3a328978 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);
e2641e09 525 }
526 }
527
528 /* Issue the SYNC command */
299290d3 529 if (syncWrite(fd,"SYNC\r\n",6,server.repl_syncio_timeout*1000) == -1) {
e2641e09 530 redisLog(REDIS_WARNING,"I/O error writing to MASTER: %s",
531 strerror(errno));
a3309139 532 goto error;
e2641e09 533 }
26b33669 534
535 /* Prepare a suitable temp file for bulk transfer */
e2641e09 536 while(maxtries--) {
537 snprintf(tmpfile,256,
d1949054 538 "temp-%d.%ld.rdb",(int)server.unixtime,(long int)getpid());
e2641e09 539 dfd = open(tmpfile,O_CREAT|O_WRONLY|O_EXCL,0644);
540 if (dfd != -1) break;
541 sleep(1);
542 }
543 if (dfd == -1) {
e2641e09 544 redisLog(REDIS_WARNING,"Opening the temp file needed for MASTER <-> SLAVE synchronization: %s",strerror(errno));
a3309139 545 goto error;
e2641e09 546 }
e2641e09 547
f4aa600b 548 /* Setup the non blocking download of the bulk file. */
a3309139 549 if (aeCreateFileEvent(server.el,fd, AE_READABLE,readSyncBulkPayload,NULL)
62ec599c 550 == AE_ERR)
f4aa600b 551 {
f4aa600b 552 redisLog(REDIS_WARNING,"Can't create readable event for SYNC");
a3309139 553 goto error;
e2641e09 554 }
a3309139 555
1844f990 556 server.repl_state = REDIS_REPL_TRANSFER;
784b9308 557 server.repl_transfer_size = -1;
558 server.repl_transfer_read = 0;
559 server.repl_transfer_last_fsync_off = 0;
f4aa600b 560 server.repl_transfer_fd = dfd;
d1949054 561 server.repl_transfer_lastio = server.unixtime;
f4aa600b 562 server.repl_transfer_tmpfile = zstrdup(tmpfile);
a3309139
PN
563 return;
564
565error:
1844f990 566 server.repl_state = REDIS_REPL_CONNECT;
a3309139
PN
567 close(fd);
568 return;
569}
570
571int 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
45029d37 581 if (aeCreateFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE,syncWithMaster,NULL) ==
b075621f 582 AE_ERR)
a3309139
PN
583 {
584 close(fd);
585 redisLog(REDIS_WARNING,"Can't create readable event for SYNC");
586 return REDIS_ERR;
587 }
588
d1949054 589 server.repl_transfer_lastio = server.unixtime;
a3309139 590 server.repl_transfer_s = fd;
1844f990 591 server.repl_state = REDIS_REPL_CONNECTING;
e2641e09 592 return REDIS_OK;
593}
594
27acd7aa 595/* This function can be called when a non blocking connection is currently
596 * in progress to undo it. */
597void undoConnectWithMaster(void) {
598 int fd = server.repl_transfer_s;
599
1844f990 600 redisAssert(server.repl_state == REDIS_REPL_CONNECTING);
27acd7aa 601 aeDeleteFileEvent(server.el,fd,AE_READABLE|AE_WRITABLE);
602 close(fd);
603 server.repl_transfer_s = -1;
1844f990 604 server.repl_state = REDIS_REPL_CONNECT;
27acd7aa 605}
606
e2641e09 607void 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);
1844f990 614 if (server.repl_state == REDIS_REPL_TRANSFER)
f4aa600b 615 replicationAbortSyncTransfer();
1844f990 616 else if (server.repl_state == REDIS_REPL_CONNECTING)
27acd7aa 617 undoConnectWithMaster();
1844f990 618 server.repl_state = REDIS_REPL_NONE;
e2641e09 619 redisLog(REDIS_NOTICE,"MASTER MODE enabled (user request)");
620 }
621 } else {
ebdfad69 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. */
e2641e09 636 sdsfree(server.masterhost);
637 server.masterhost = sdsdup(c->argv[1]->ptr);
ebdfad69 638 server.masterport = port;
e2641e09 639 if (server.master) freeClient(server.master);
179e54d2 640 disconnectSlaves(); /* Force our slaves to resync with us as well. */
1844f990 641 if (server.repl_state == REDIS_REPL_TRANSFER)
f4aa600b 642 replicationAbortSyncTransfer();
1844f990 643 server.repl_state = REDIS_REPL_CONNECT;
e2641e09 644 redisLog(REDIS_NOTICE,"SLAVE OF %s:%d enabled (user request)",
645 server.masterhost, server.masterport);
646 }
647 addReply(c,shared.ok);
648}
f4aa600b 649
650/* --------------------------- REPLICATION CRON ---------------------------- */
651
f4aa600b 652void replicationCron(void) {
27acd7aa 653 /* Non blocking connection timeout? */
1844f990 654 if (server.masterhost && server.repl_state == REDIS_REPL_CONNECTING &&
27acd7aa 655 (time(NULL)-server.repl_transfer_lastio) > server.repl_timeout)
656 {
657 redisLog(REDIS_WARNING,"Timeout connecting to the MASTER...");
658 undoConnectWithMaster();
659 }
660
f4aa600b 661 /* Bulk transfer I/O timeout? */
1844f990 662 if (server.masterhost && server.repl_state == REDIS_REPL_TRANSFER &&
8996bf77 663 (time(NULL)-server.repl_transfer_lastio) > server.repl_timeout)
f4aa600b 664 {
665 redisLog(REDIS_WARNING,"Timeout receiving bulk data from MASTER...");
666 replicationAbortSyncTransfer();
667 }
668
89a1433e 669 /* Timed out master when we are an already connected slave? */
1844f990 670 if (server.masterhost && server.repl_state == REDIS_REPL_CONNECTED &&
8996bf77 671 (time(NULL)-server.master->lastinteraction) > server.repl_timeout)
89a1433e 672 {
673 redisLog(REDIS_WARNING,"MASTER time out: no data nor PING received...");
674 freeClient(server.master);
675 }
676
f4aa600b 677 /* Check if we should connect to a MASTER */
1844f990 678 if (server.repl_state == REDIS_REPL_CONNECT) {
f4aa600b 679 redisLog(REDIS_NOTICE,"Connecting to MASTER...");
a3309139
PN
680 if (connectWithMaster() == REDIS_OK) {
681 redisLog(REDIS_NOTICE,"MASTER <-> SLAVE sync started");
f4aa600b 682 }
683 }
89a1433e 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. */
8996bf77 689 if (!(server.cronloops % (server.repl_ping_slave_period*10))) {
89a1433e 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 */
a950a843 702 addReplySds(slave,sdsnew("*1\r\n$4\r\nPING\r\n"));
89a1433e 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. */
f96a9f82 709 if (write(slave->fd, "\n", 1) == -1) {
710 /* Don't worry, it's just a ping. */
711 }
89a1433e 712 }
713 }
714 }
f4aa600b 715}