]> git.saurik.com Git - redis.git/blame_incremental - src/networking.c
Encode sorted set after loading from dump
[redis.git] / src / networking.c
... / ...
CommitLineData
1#include "redis.h"
2#include <sys/uio.h>
3
4void *dupClientReplyValue(void *o) {
5 incrRefCount((robj*)o);
6 return o;
7}
8
9int listMatchObjects(void *a, void *b) {
10 return equalStringObjects(a,b);
11}
12
13redisClient *createClient(int fd) {
14 redisClient *c = zmalloc(sizeof(redisClient));
15 c->bufpos = 0;
16
17 anetNonBlock(NULL,fd);
18 anetTcpNoDelay(NULL,fd);
19 if (!c) return NULL;
20 if (aeCreateFileEvent(server.el,fd,AE_READABLE,
21 readQueryFromClient, c) == AE_ERR)
22 {
23 close(fd);
24 zfree(c);
25 return NULL;
26 }
27
28 selectDb(c,0);
29 c->fd = fd;
30 c->querybuf = sdsempty();
31 c->reqtype = 0;
32 c->argc = 0;
33 c->argv = NULL;
34 c->multibulklen = 0;
35 c->bulklen = -1;
36 c->sentlen = 0;
37 c->flags = 0;
38 c->lastinteraction = time(NULL);
39 c->authenticated = 0;
40 c->replstate = REDIS_REPL_NONE;
41 c->reply = listCreate();
42 listSetFreeMethod(c->reply,decrRefCount);
43 listSetDupMethod(c->reply,dupClientReplyValue);
44 c->bpop.keys = NULL;
45 c->bpop.count = 0;
46 c->bpop.timeout = 0;
47 c->bpop.target = NULL;
48 c->io_keys = listCreate();
49 c->watched_keys = listCreate();
50 listSetFreeMethod(c->io_keys,decrRefCount);
51 c->pubsub_channels = dictCreate(&setDictType,NULL);
52 c->pubsub_patterns = listCreate();
53 listSetFreeMethod(c->pubsub_patterns,decrRefCount);
54 listSetMatchMethod(c->pubsub_patterns,listMatchObjects);
55 listAddNodeTail(server.clients,c);
56 initClientMultiState(c);
57 return c;
58}
59
60/* Set the event loop to listen for write events on the client's socket.
61 * Typically gets called every time a reply is built. */
62int _installWriteEvent(redisClient *c) {
63 /* When CLOSE_AFTER_REPLY is set, no more replies may be added! */
64 redisAssert(!(c->flags & REDIS_CLOSE_AFTER_REPLY));
65
66 if (c->fd <= 0) return REDIS_ERR;
67 if (c->bufpos == 0 && listLength(c->reply) == 0 &&
68 (c->replstate == REDIS_REPL_NONE ||
69 c->replstate == REDIS_REPL_ONLINE) &&
70 aeCreateFileEvent(server.el, c->fd, AE_WRITABLE,
71 sendReplyToClient, c) == AE_ERR) return REDIS_ERR;
72 return REDIS_OK;
73}
74
75/* Create a duplicate of the last object in the reply list when
76 * it is not exclusively owned by the reply list. */
77robj *dupLastObjectIfNeeded(list *reply) {
78 robj *new, *cur;
79 listNode *ln;
80 redisAssert(listLength(reply) > 0);
81 ln = listLast(reply);
82 cur = listNodeValue(ln);
83 if (cur->refcount > 1) {
84 new = dupStringObject(cur);
85 decrRefCount(cur);
86 listNodeValue(ln) = new;
87 }
88 return listNodeValue(ln);
89}
90
91int _addReplyToBuffer(redisClient *c, char *s, size_t len) {
92 size_t available = sizeof(c->buf)-c->bufpos;
93
94 /* If there already are entries in the reply list, we cannot
95 * add anything more to the static buffer. */
96 if (listLength(c->reply) > 0) return REDIS_ERR;
97
98 /* Check that the buffer has enough space available for this string. */
99 if (len > available) return REDIS_ERR;
100
101 memcpy(c->buf+c->bufpos,s,len);
102 c->bufpos+=len;
103 return REDIS_OK;
104}
105
106void _addReplyObjectToList(redisClient *c, robj *o) {
107 robj *tail;
108 if (listLength(c->reply) == 0) {
109 incrRefCount(o);
110 listAddNodeTail(c->reply,o);
111 } else {
112 tail = listNodeValue(listLast(c->reply));
113
114 /* Append to this object when possible. */
115 if (tail->ptr != NULL &&
116 sdslen(tail->ptr)+sdslen(o->ptr) <= REDIS_REPLY_CHUNK_BYTES)
117 {
118 tail = dupLastObjectIfNeeded(c->reply);
119 tail->ptr = sdscatlen(tail->ptr,o->ptr,sdslen(o->ptr));
120 } else {
121 incrRefCount(o);
122 listAddNodeTail(c->reply,o);
123 }
124 }
125}
126
127/* This method takes responsibility over the sds. When it is no longer
128 * needed it will be free'd, otherwise it ends up in a robj. */
129void _addReplySdsToList(redisClient *c, sds s) {
130 robj *tail;
131 if (listLength(c->reply) == 0) {
132 listAddNodeTail(c->reply,createObject(REDIS_STRING,s));
133 } else {
134 tail = listNodeValue(listLast(c->reply));
135
136 /* Append to this object when possible. */
137 if (tail->ptr != NULL &&
138 sdslen(tail->ptr)+sdslen(s) <= REDIS_REPLY_CHUNK_BYTES)
139 {
140 tail = dupLastObjectIfNeeded(c->reply);
141 tail->ptr = sdscatlen(tail->ptr,s,sdslen(s));
142 sdsfree(s);
143 } else {
144 listAddNodeTail(c->reply,createObject(REDIS_STRING,s));
145 }
146 }
147}
148
149void _addReplyStringToList(redisClient *c, char *s, size_t len) {
150 robj *tail;
151 if (listLength(c->reply) == 0) {
152 listAddNodeTail(c->reply,createStringObject(s,len));
153 } else {
154 tail = listNodeValue(listLast(c->reply));
155
156 /* Append to this object when possible. */
157 if (tail->ptr != NULL &&
158 sdslen(tail->ptr)+len <= REDIS_REPLY_CHUNK_BYTES)
159 {
160 tail = dupLastObjectIfNeeded(c->reply);
161 tail->ptr = sdscatlen(tail->ptr,s,len);
162 } else {
163 listAddNodeTail(c->reply,createStringObject(s,len));
164 }
165 }
166}
167
168void addReply(redisClient *c, robj *obj) {
169 if (_installWriteEvent(c) != REDIS_OK) return;
170
171 /* This is an important place where we can avoid copy-on-write
172 * when there is a saving child running, avoiding touching the
173 * refcount field of the object if it's not needed.
174 *
175 * If the encoding is RAW and there is room in the static buffer
176 * we'll be able to send the object to the client without
177 * messing with its page. */
178 if (obj->encoding == REDIS_ENCODING_RAW) {
179 if (_addReplyToBuffer(c,obj->ptr,sdslen(obj->ptr)) != REDIS_OK)
180 _addReplyObjectToList(c,obj);
181 } else {
182 /* FIXME: convert the long into string and use _addReplyToBuffer()
183 * instead of calling getDecodedObject. As this place in the
184 * code is too performance critical. */
185 obj = getDecodedObject(obj);
186 if (_addReplyToBuffer(c,obj->ptr,sdslen(obj->ptr)) != REDIS_OK)
187 _addReplyObjectToList(c,obj);
188 decrRefCount(obj);
189 }
190}
191
192void addReplySds(redisClient *c, sds s) {
193 if (_installWriteEvent(c) != REDIS_OK) {
194 /* The caller expects the sds to be free'd. */
195 sdsfree(s);
196 return;
197 }
198 if (_addReplyToBuffer(c,s,sdslen(s)) == REDIS_OK) {
199 sdsfree(s);
200 } else {
201 /* This method free's the sds when it is no longer needed. */
202 _addReplySdsToList(c,s);
203 }
204}
205
206void addReplyString(redisClient *c, char *s, size_t len) {
207 if (_installWriteEvent(c) != REDIS_OK) return;
208 if (_addReplyToBuffer(c,s,len) != REDIS_OK)
209 _addReplyStringToList(c,s,len);
210}
211
212void _addReplyError(redisClient *c, char *s, size_t len) {
213 addReplyString(c,"-ERR ",5);
214 addReplyString(c,s,len);
215 addReplyString(c,"\r\n",2);
216}
217
218void addReplyError(redisClient *c, char *err) {
219 _addReplyError(c,err,strlen(err));
220}
221
222void addReplyErrorFormat(redisClient *c, const char *fmt, ...) {
223 va_list ap;
224 va_start(ap,fmt);
225 sds s = sdscatvprintf(sdsempty(),fmt,ap);
226 va_end(ap);
227 _addReplyError(c,s,sdslen(s));
228 sdsfree(s);
229}
230
231void _addReplyStatus(redisClient *c, char *s, size_t len) {
232 addReplyString(c,"+",1);
233 addReplyString(c,s,len);
234 addReplyString(c,"\r\n",2);
235}
236
237void addReplyStatus(redisClient *c, char *status) {
238 _addReplyStatus(c,status,strlen(status));
239}
240
241void addReplyStatusFormat(redisClient *c, const char *fmt, ...) {
242 va_list ap;
243 va_start(ap,fmt);
244 sds s = sdscatvprintf(sdsempty(),fmt,ap);
245 va_end(ap);
246 _addReplyStatus(c,s,sdslen(s));
247 sdsfree(s);
248}
249
250/* Adds an empty object to the reply list that will contain the multi bulk
251 * length, which is not known when this function is called. */
252void *addDeferredMultiBulkLength(redisClient *c) {
253 /* Note that we install the write event here even if the object is not
254 * ready to be sent, since we are sure that before returning to the
255 * event loop setDeferredMultiBulkLength() will be called. */
256 if (_installWriteEvent(c) != REDIS_OK) return NULL;
257 listAddNodeTail(c->reply,createObject(REDIS_STRING,NULL));
258 return listLast(c->reply);
259}
260
261/* Populate the length object and try glueing it to the next chunk. */
262void setDeferredMultiBulkLength(redisClient *c, void *node, long length) {
263 listNode *ln = (listNode*)node;
264 robj *len, *next;
265
266 /* Abort when *node is NULL (see addDeferredMultiBulkLength). */
267 if (node == NULL) return;
268
269 len = listNodeValue(ln);
270 len->ptr = sdscatprintf(sdsempty(),"*%ld\r\n",length);
271 if (ln->next != NULL) {
272 next = listNodeValue(ln->next);
273
274 /* Only glue when the next node is non-NULL (an sds in this case) */
275 if (next->ptr != NULL) {
276 len->ptr = sdscatlen(len->ptr,next->ptr,sdslen(next->ptr));
277 listDelNode(c->reply,ln->next);
278 }
279 }
280}
281
282/* Add a duble as a bulk reply */
283void addReplyDouble(redisClient *c, double d) {
284 char dbuf[128], sbuf[128];
285 int dlen, slen;
286 dlen = snprintf(dbuf,sizeof(dbuf),"%.17g",d);
287 slen = snprintf(sbuf,sizeof(sbuf),"$%d\r\n%s\r\n",dlen,dbuf);
288 addReplyString(c,sbuf,slen);
289}
290
291/* Add a long long as integer reply or bulk len / multi bulk count.
292 * Basically this is used to output <prefix><long long><crlf>. */
293void _addReplyLongLong(redisClient *c, long long ll, char prefix) {
294 char buf[128];
295 int len;
296 buf[0] = prefix;
297 len = ll2string(buf+1,sizeof(buf)-1,ll);
298 buf[len+1] = '\r';
299 buf[len+2] = '\n';
300 addReplyString(c,buf,len+3);
301}
302
303void addReplyLongLong(redisClient *c, long long ll) {
304 _addReplyLongLong(c,ll,':');
305}
306
307void addReplyMultiBulkLen(redisClient *c, long length) {
308 _addReplyLongLong(c,length,'*');
309}
310
311/* Create the length prefix of a bulk reply, example: $2234 */
312void addReplyBulkLen(redisClient *c, robj *obj) {
313 size_t len;
314
315 if (obj->encoding == REDIS_ENCODING_RAW) {
316 len = sdslen(obj->ptr);
317 } else {
318 long n = (long)obj->ptr;
319
320 /* Compute how many bytes will take this integer as a radix 10 string */
321 len = 1;
322 if (n < 0) {
323 len++;
324 n = -n;
325 }
326 while((n = n/10) != 0) {
327 len++;
328 }
329 }
330 _addReplyLongLong(c,len,'$');
331}
332
333/* Add a Redis Object as a bulk reply */
334void addReplyBulk(redisClient *c, robj *obj) {
335 addReplyBulkLen(c,obj);
336 addReply(c,obj);
337 addReply(c,shared.crlf);
338}
339
340/* Add a C buffer as bulk reply */
341void addReplyBulkCBuffer(redisClient *c, void *p, size_t len) {
342 _addReplyLongLong(c,len,'$');
343 addReplyString(c,p,len);
344 addReply(c,shared.crlf);
345}
346
347/* Add a C nul term string as bulk reply */
348void addReplyBulkCString(redisClient *c, char *s) {
349 if (s == NULL) {
350 addReply(c,shared.nullbulk);
351 } else {
352 addReplyBulkCBuffer(c,s,strlen(s));
353 }
354}
355
356/* Add a long long as a bulk reply */
357void addReplyBulkLongLong(redisClient *c, long long ll) {
358 char buf[64];
359 int len;
360
361 len = ll2string(buf,64,ll);
362 addReplyBulkCBuffer(c,buf,len);
363}
364
365static void acceptCommonHandler(int fd) {
366 redisClient *c;
367 if ((c = createClient(fd)) == NULL) {
368 redisLog(REDIS_WARNING,"Error allocating resoures for the client");
369 close(fd); /* May be already closed, just ingore errors */
370 return;
371 }
372 /* If maxclient directive is set and this is one client more... close the
373 * connection. Note that we create the client instead to check before
374 * for this condition, since now the socket is already set in nonblocking
375 * mode and we can send an error for free using the Kernel I/O */
376 if (server.maxclients && listLength(server.clients) > server.maxclients) {
377 char *err = "-ERR max number of clients reached\r\n";
378
379 /* That's a best effort error message, don't check write errors */
380 if (write(c->fd,err,strlen(err)) == -1) {
381 /* Nothing to do, Just to avoid the warning... */
382 }
383 freeClient(c);
384 return;
385 }
386 server.stat_numconnections++;
387}
388
389void acceptTcpHandler(aeEventLoop *el, int fd, void *privdata, int mask) {
390 int cport, cfd;
391 char cip[128];
392 REDIS_NOTUSED(el);
393 REDIS_NOTUSED(mask);
394 REDIS_NOTUSED(privdata);
395
396 cfd = anetTcpAccept(server.neterr, fd, cip, &cport);
397 if (cfd == AE_ERR) {
398 redisLog(REDIS_VERBOSE,"Accepting client connection: %s", server.neterr);
399 return;
400 }
401 redisLog(REDIS_VERBOSE,"Accepted %s:%d", cip, cport);
402 acceptCommonHandler(cfd);
403}
404
405void acceptUnixHandler(aeEventLoop *el, int fd, void *privdata, int mask) {
406 int cfd;
407 REDIS_NOTUSED(el);
408 REDIS_NOTUSED(mask);
409 REDIS_NOTUSED(privdata);
410
411 cfd = anetUnixAccept(server.neterr, fd);
412 if (cfd == AE_ERR) {
413 redisLog(REDIS_VERBOSE,"Accepting client connection: %s", server.neterr);
414 return;
415 }
416 redisLog(REDIS_VERBOSE,"Accepted connection to %s", server.unixsocket);
417 acceptCommonHandler(cfd);
418}
419
420
421static void freeClientArgv(redisClient *c) {
422 int j;
423 for (j = 0; j < c->argc; j++)
424 decrRefCount(c->argv[j]);
425 c->argc = 0;
426}
427
428void freeClient(redisClient *c) {
429 listNode *ln;
430
431 /* Note that if the client we are freeing is blocked into a blocking
432 * call, we have to set querybuf to NULL *before* to call
433 * unblockClientWaitingData() to avoid processInputBuffer() will get
434 * called. Also it is important to remove the file events after
435 * this, because this call adds the READABLE event. */
436 sdsfree(c->querybuf);
437 c->querybuf = NULL;
438 if (c->flags & REDIS_BLOCKED)
439 unblockClientWaitingData(c);
440
441 /* UNWATCH all the keys */
442 unwatchAllKeys(c);
443 listRelease(c->watched_keys);
444 /* Unsubscribe from all the pubsub channels */
445 pubsubUnsubscribeAllChannels(c,0);
446 pubsubUnsubscribeAllPatterns(c,0);
447 dictRelease(c->pubsub_channels);
448 listRelease(c->pubsub_patterns);
449 /* Obvious cleanup */
450 aeDeleteFileEvent(server.el,c->fd,AE_READABLE);
451 aeDeleteFileEvent(server.el,c->fd,AE_WRITABLE);
452 listRelease(c->reply);
453 freeClientArgv(c);
454 close(c->fd);
455 /* Remove from the list of clients */
456 ln = listSearchKey(server.clients,c);
457 redisAssert(ln != NULL);
458 listDelNode(server.clients,ln);
459 /* When client was just unblocked because of a blocking operation,
460 * remove it from the list with unblocked clients. */
461 if (c->flags & REDIS_UNBLOCKED) {
462 ln = listSearchKey(server.unblocked_clients,c);
463 redisAssert(ln != NULL);
464 listDelNode(server.unblocked_clients,ln);
465 }
466 /* Remove from the list of clients waiting for swapped keys, or ready
467 * to be restarted, but not yet woken up again. */
468 if (c->flags & REDIS_IO_WAIT) {
469 redisAssert(server.ds_enabled);
470 if (listLength(c->io_keys) == 0) {
471 ln = listSearchKey(server.io_ready_clients,c);
472
473 /* When this client is waiting to be woken up (REDIS_IO_WAIT),
474 * it should be present in the list io_ready_clients */
475 redisAssert(ln != NULL);
476 listDelNode(server.io_ready_clients,ln);
477 } else {
478 while (listLength(c->io_keys)) {
479 ln = listFirst(c->io_keys);
480 dontWaitForSwappedKey(c,ln->value);
481 }
482 }
483 server.cache_blocked_clients--;
484 }
485 listRelease(c->io_keys);
486 /* Master/slave cleanup.
487 * Case 1: we lost the connection with a slave. */
488 if (c->flags & REDIS_SLAVE) {
489 if (c->replstate == REDIS_REPL_SEND_BULK && c->repldbfd != -1)
490 close(c->repldbfd);
491 list *l = (c->flags & REDIS_MONITOR) ? server.monitors : server.slaves;
492 ln = listSearchKey(l,c);
493 redisAssert(ln != NULL);
494 listDelNode(l,ln);
495 }
496
497 /* Case 2: we lost the connection with the master. */
498 if (c->flags & REDIS_MASTER) {
499 server.master = NULL;
500 server.replstate = REDIS_REPL_CONNECT;
501 /* Since we lost the connection with the master, we should also
502 * close the connection with all our slaves if we have any, so
503 * when we'll resync with the master the other slaves will sync again
504 * with us as well. Note that also when the slave is not connected
505 * to the master it will keep refusing connections by other slaves. */
506 while (listLength(server.slaves)) {
507 ln = listFirst(server.slaves);
508 freeClient((redisClient*)ln->value);
509 }
510 }
511 /* Release memory */
512 zfree(c->argv);
513 freeClientMultiState(c);
514 zfree(c);
515}
516
517void sendReplyToClient(aeEventLoop *el, int fd, void *privdata, int mask) {
518 redisClient *c = privdata;
519 int nwritten = 0, totwritten = 0, objlen;
520 robj *o;
521 REDIS_NOTUSED(el);
522 REDIS_NOTUSED(mask);
523
524 while(c->bufpos > 0 || listLength(c->reply)) {
525 if (c->bufpos > 0) {
526 if (c->flags & REDIS_MASTER) {
527 /* Don't reply to a master */
528 nwritten = c->bufpos - c->sentlen;
529 } else {
530 nwritten = write(fd,c->buf+c->sentlen,c->bufpos-c->sentlen);
531 if (nwritten <= 0) break;
532 }
533 c->sentlen += nwritten;
534 totwritten += nwritten;
535
536 /* If the buffer was sent, set bufpos to zero to continue with
537 * the remainder of the reply. */
538 if (c->sentlen == c->bufpos) {
539 c->bufpos = 0;
540 c->sentlen = 0;
541 }
542 } else {
543 o = listNodeValue(listFirst(c->reply));
544 objlen = sdslen(o->ptr);
545
546 if (objlen == 0) {
547 listDelNode(c->reply,listFirst(c->reply));
548 continue;
549 }
550
551 if (c->flags & REDIS_MASTER) {
552 /* Don't reply to a master */
553 nwritten = objlen - c->sentlen;
554 } else {
555 nwritten = write(fd, ((char*)o->ptr)+c->sentlen,objlen-c->sentlen);
556 if (nwritten <= 0) break;
557 }
558 c->sentlen += nwritten;
559 totwritten += nwritten;
560
561 /* If we fully sent the object on head go to the next one */
562 if (c->sentlen == objlen) {
563 listDelNode(c->reply,listFirst(c->reply));
564 c->sentlen = 0;
565 }
566 }
567 /* Note that we avoid to send more thank REDIS_MAX_WRITE_PER_EVENT
568 * bytes, in a single threaded server it's a good idea to serve
569 * other clients as well, even if a very large request comes from
570 * super fast link that is always able to accept data (in real world
571 * scenario think about 'KEYS *' against the loopback interfae) */
572 if (totwritten > REDIS_MAX_WRITE_PER_EVENT) break;
573 }
574 if (nwritten == -1) {
575 if (errno == EAGAIN) {
576 nwritten = 0;
577 } else {
578 redisLog(REDIS_VERBOSE,
579 "Error writing to client: %s", strerror(errno));
580 freeClient(c);
581 return;
582 }
583 }
584 if (totwritten > 0) c->lastinteraction = time(NULL);
585 if (listLength(c->reply) == 0) {
586 c->sentlen = 0;
587 aeDeleteFileEvent(server.el,c->fd,AE_WRITABLE);
588
589 /* Close connection after entire reply has been sent. */
590 if (c->flags & REDIS_CLOSE_AFTER_REPLY) freeClient(c);
591 }
592}
593
594/* resetClient prepare the client to process the next command */
595void resetClient(redisClient *c) {
596 freeClientArgv(c);
597 c->reqtype = 0;
598 c->multibulklen = 0;
599 c->bulklen = -1;
600}
601
602void closeTimedoutClients(void) {
603 redisClient *c;
604 listNode *ln;
605 time_t now = time(NULL);
606 listIter li;
607
608 listRewind(server.clients,&li);
609 while ((ln = listNext(&li)) != NULL) {
610 c = listNodeValue(ln);
611 if (server.maxidletime &&
612 !(c->flags & REDIS_SLAVE) && /* no timeout for slaves */
613 !(c->flags & REDIS_MASTER) && /* no timeout for masters */
614 !(c->flags & REDIS_BLOCKED) && /* no timeout for BLPOP */
615 dictSize(c->pubsub_channels) == 0 && /* no timeout for pubsub */
616 listLength(c->pubsub_patterns) == 0 &&
617 (now - c->lastinteraction > server.maxidletime))
618 {
619 redisLog(REDIS_VERBOSE,"Closing idle client");
620 freeClient(c);
621 } else if (c->flags & REDIS_BLOCKED) {
622 if (c->bpop.timeout != 0 && c->bpop.timeout < now) {
623 addReply(c,shared.nullmultibulk);
624 unblockClientWaitingData(c);
625 }
626 }
627 }
628}
629
630int processInlineBuffer(redisClient *c) {
631 char *newline = strstr(c->querybuf,"\r\n");
632 int argc, j;
633 sds *argv;
634 size_t querylen;
635
636 /* Nothing to do without a \r\n */
637 if (newline == NULL)
638 return REDIS_ERR;
639
640 /* Split the input buffer up to the \r\n */
641 querylen = newline-(c->querybuf);
642 argv = sdssplitlen(c->querybuf,querylen," ",1,&argc);
643
644 /* Leave data after the first line of the query in the buffer */
645 c->querybuf = sdsrange(c->querybuf,querylen+2,-1);
646
647 /* Setup argv array on client structure */
648 if (c->argv) zfree(c->argv);
649 c->argv = zmalloc(sizeof(robj*)*argc);
650
651 /* Create redis objects for all arguments. */
652 for (c->argc = 0, j = 0; j < argc; j++) {
653 if (sdslen(argv[j])) {
654 c->argv[c->argc] = createObject(REDIS_STRING,argv[j]);
655 c->argc++;
656 } else {
657 sdsfree(argv[j]);
658 }
659 }
660 zfree(argv);
661 return REDIS_OK;
662}
663
664/* Helper function. Trims query buffer to make the function that processes
665 * multi bulk requests idempotent. */
666static void setProtocolError(redisClient *c, int pos) {
667 c->flags |= REDIS_CLOSE_AFTER_REPLY;
668 c->querybuf = sdsrange(c->querybuf,pos,-1);
669}
670
671int processMultibulkBuffer(redisClient *c) {
672 char *newline = NULL;
673 char *eptr;
674 int pos = 0, tolerr;
675 long bulklen;
676
677 if (c->multibulklen == 0) {
678 /* The client should have been reset */
679 redisAssert(c->argc == 0);
680
681 /* Multi bulk length cannot be read without a \r\n */
682 newline = strstr(c->querybuf,"\r\n");
683 if (newline == NULL)
684 return REDIS_ERR;
685
686 /* We know for sure there is a whole line since newline != NULL,
687 * so go ahead and find out the multi bulk length. */
688 redisAssert(c->querybuf[0] == '*');
689 c->multibulklen = strtol(c->querybuf+1,&eptr,10);
690 pos = (newline-c->querybuf)+2;
691 if (c->multibulklen <= 0) {
692 c->querybuf = sdsrange(c->querybuf,pos,-1);
693 return REDIS_OK;
694 } else if (c->multibulklen > 1024*1024) {
695 addReplyError(c,"Protocol error: invalid multibulk length");
696 setProtocolError(c,pos);
697 return REDIS_ERR;
698 }
699
700 /* Setup argv array on client structure */
701 if (c->argv) zfree(c->argv);
702 c->argv = zmalloc(sizeof(robj*)*c->multibulklen);
703
704 /* Search new newline */
705 newline = strstr(c->querybuf+pos,"\r\n");
706 }
707
708 redisAssert(c->multibulklen > 0);
709 while(c->multibulklen) {
710 /* Read bulk length if unknown */
711 if (c->bulklen == -1) {
712 newline = strstr(c->querybuf+pos,"\r\n");
713 if (newline != NULL) {
714 if (c->querybuf[pos] != '$') {
715 addReplyErrorFormat(c,
716 "Protocol error: expected '$', got '%c'",
717 c->querybuf[pos]);
718 setProtocolError(c,pos);
719 return REDIS_ERR;
720 }
721
722 bulklen = strtol(c->querybuf+pos+1,&eptr,10);
723 tolerr = (eptr[0] != '\r');
724 if (tolerr || bulklen == LONG_MIN || bulklen == LONG_MAX ||
725 bulklen < 0 || bulklen > 512*1024*1024)
726 {
727 addReplyError(c,"Protocol error: invalid bulk length");
728 setProtocolError(c,pos);
729 return REDIS_ERR;
730 }
731 pos += eptr-(c->querybuf+pos)+2;
732 c->bulklen = bulklen;
733 } else {
734 /* No newline in current buffer, so wait for more data */
735 break;
736 }
737 }
738
739 /* Read bulk argument */
740 if (sdslen(c->querybuf)-pos < (unsigned)(c->bulklen+2)) {
741 /* Not enough data (+2 == trailing \r\n) */
742 break;
743 } else {
744 c->argv[c->argc++] = createStringObject(c->querybuf+pos,c->bulklen);
745 pos += c->bulklen+2;
746 c->bulklen = -1;
747 c->multibulklen--;
748 }
749 }
750
751 /* Trim to pos */
752 c->querybuf = sdsrange(c->querybuf,pos,-1);
753
754 /* We're done when c->multibulk == 0 */
755 if (c->multibulklen == 0) {
756 return REDIS_OK;
757 }
758 return REDIS_ERR;
759}
760
761void processInputBuffer(redisClient *c) {
762 /* Keep processing while there is something in the input buffer */
763 while(sdslen(c->querybuf)) {
764 /* Immediately abort if the client is in the middle of something. */
765 if (c->flags & REDIS_BLOCKED || c->flags & REDIS_IO_WAIT) return;
766
767 /* REDIS_CLOSE_AFTER_REPLY closes the connection once the reply is
768 * written to the client. Make sure to not let the reply grow after
769 * this flag has been set (i.e. don't process more commands). */
770 if (c->flags & REDIS_CLOSE_AFTER_REPLY) return;
771
772 /* Determine request type when unknown. */
773 if (!c->reqtype) {
774 if (c->querybuf[0] == '*') {
775 c->reqtype = REDIS_REQ_MULTIBULK;
776 } else {
777 c->reqtype = REDIS_REQ_INLINE;
778 }
779 }
780
781 if (c->reqtype == REDIS_REQ_INLINE) {
782 if (processInlineBuffer(c) != REDIS_OK) break;
783 } else if (c->reqtype == REDIS_REQ_MULTIBULK) {
784 if (processMultibulkBuffer(c) != REDIS_OK) break;
785 } else {
786 redisPanic("Unknown request type");
787 }
788
789 /* Multibulk processing could see a <= 0 length. */
790 if (c->argc == 0) {
791 resetClient(c);
792 } else {
793 /* Only reset the client when the command was executed. */
794 if (processCommand(c) == REDIS_OK)
795 resetClient(c);
796 }
797 }
798}
799
800void readQueryFromClient(aeEventLoop *el, int fd, void *privdata, int mask) {
801 redisClient *c = (redisClient*) privdata;
802 char buf[REDIS_IOBUF_LEN];
803 int nread;
804 REDIS_NOTUSED(el);
805 REDIS_NOTUSED(mask);
806
807 nread = read(fd, buf, REDIS_IOBUF_LEN);
808 if (nread == -1) {
809 if (errno == EAGAIN) {
810 nread = 0;
811 } else {
812 redisLog(REDIS_VERBOSE, "Reading from client: %s",strerror(errno));
813 freeClient(c);
814 return;
815 }
816 } else if (nread == 0) {
817 redisLog(REDIS_VERBOSE, "Client closed connection");
818 freeClient(c);
819 return;
820 }
821 if (nread) {
822 c->querybuf = sdscatlen(c->querybuf,buf,nread);
823 c->lastinteraction = time(NULL);
824 } else {
825 return;
826 }
827 processInputBuffer(c);
828}
829
830void getClientsMaxBuffers(unsigned long *longest_output_list,
831 unsigned long *biggest_input_buffer) {
832 redisClient *c;
833 listNode *ln;
834 listIter li;
835 unsigned long lol = 0, bib = 0;
836
837 listRewind(server.clients,&li);
838 while ((ln = listNext(&li)) != NULL) {
839 c = listNodeValue(ln);
840
841 if (listLength(c->reply) > lol) lol = listLength(c->reply);
842 if (sdslen(c->querybuf) > bib) bib = sdslen(c->querybuf);
843 }
844 *longest_output_list = lol;
845 *biggest_input_buffer = bib;
846}
847