94686d0590851a26c16b9cce32bc3858080b0661
[m6w6/libmemcached] / libmemcached / io.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * LibMemcached
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are
11 * met:
12 *
13 * * Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 *
16 * * Redistributions in binary form must reproduce the above
17 * copyright notice, this list of conditions and the following disclaimer
18 * in the documentation and/or other materials provided with the
19 * distribution.
20 *
21 * * The names of its contributors may not be used to endorse or
22 * promote products derived from this software without specific prior
23 * written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
28 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
29 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
35 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 *
37 */
38
39
40 #include <libmemcached/common.h>
41
42 enum memc_read_or_write {
43 MEM_READ,
44 MEM_WRITE
45 };
46
47 /**
48 * Try to fill the input buffer for a server with as much
49 * data as possible.
50 *
51 * @param ptr the server to pack
52 */
53 static bool repack_input_buffer(memcached_server_write_instance_st ptr)
54 {
55 if (ptr->read_ptr != ptr->read_buffer)
56 {
57 /* Move all of the data to the beginning of the buffer so
58 ** that we can fit more data into the buffer...
59 */
60 memmove(ptr->read_buffer, ptr->read_ptr, ptr->read_buffer_length);
61 ptr->read_ptr= ptr->read_buffer;
62 ptr->read_data_length= ptr->read_buffer_length;
63 }
64
65 /* There is room in the buffer, try to fill it! */
66 if (ptr->read_buffer_length != MEMCACHED_MAX_BUFFER)
67 {
68 do {
69 /* Just try a single read to grab what's available */
70 ssize_t nr= recv(ptr->fd,
71 ptr->read_ptr + ptr->read_data_length,
72 MEMCACHED_MAX_BUFFER - ptr->read_data_length,
73 MSG_DONTWAIT);
74
75 switch (nr)
76 {
77 case SOCKET_ERROR:
78 {
79 switch (get_socket_errno())
80 {
81 case EINTR:
82 continue;
83
84 #if EWOULDBLOCK != EAGAIN
85 case EWOULDBLOCK:
86 #endif
87 case EAGAIN:
88 #ifdef TARGET_OS_LINUX
89 case ERESTART:
90 #endif
91 break; // No IO is fine, we can just move on
92
93 default:
94 memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
95 }
96 }
97 break;
98
99 case 0: // Shutdown on the socket has occurred
100 {
101 memcached_set_error(*ptr, MEMCACHED_CONNECTION_FAILURE, MEMCACHED_AT);
102 }
103 break;
104
105 default:
106 {
107 ptr->read_data_length+= size_t(nr);
108 ptr->read_buffer_length+= size_t(nr);
109 return true;
110 }
111 break;
112 }
113 } while (0);
114 }
115 return false;
116 }
117
118 /**
119 * If the we have callbacks connected to this server structure
120 * we may start process the input queue and fire the callbacks
121 * for the incomming messages. This function is _only_ called
122 * when the input buffer is full, so that we _know_ that we have
123 * at least _one_ message to process.
124 *
125 * @param ptr the server to star processing iput messages for
126 * @return true if we processed anything, false otherwise
127 */
128 static bool process_input_buffer(memcached_server_write_instance_st ptr)
129 {
130 /*
131 ** We might be able to process some of the response messages if we
132 ** have a callback set up
133 */
134 if (ptr->root->callbacks != NULL)
135 {
136 /*
137 * We might have responses... try to read them out and fire
138 * callbacks
139 */
140 memcached_callback_st cb= *ptr->root->callbacks;
141
142 memcached_set_processing_input((memcached_st *)ptr->root, true);
143
144 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
145 memcached_st *root= (memcached_st *)ptr->root;
146 memcached_return_t error= memcached_response(ptr, buffer, sizeof(buffer), &root->result);
147
148 memcached_set_processing_input(root, false);
149
150 if (error == MEMCACHED_SUCCESS)
151 {
152 for (unsigned int x= 0; x < cb.number_of_callback; x++)
153 {
154 error= (*cb.callback[x])(ptr->root, &root->result, cb.context);
155 if (error != MEMCACHED_SUCCESS)
156 {
157 break;
158 }
159 }
160
161 /* @todo what should I do with the error message??? */
162 }
163 /* @todo what should I do with other error messages?? */
164 return true;
165 }
166
167 return false;
168 }
169
170 static memcached_return_t io_wait(memcached_server_write_instance_st ptr,
171 const memc_read_or_write read_or_write)
172 {
173 /*
174 ** We are going to block on write, but at least on Solaris we might block
175 ** on write if we haven't read anything from our input buffer..
176 ** Try to purge the input buffer if we don't do any flow control in the
177 ** application layer (just sending a lot of data etc)
178 ** The test is moved down in the purge function to avoid duplication of
179 ** the test.
180 */
181 if (read_or_write == MEM_WRITE)
182 {
183 if (memcached_fatal(memcached_purge(ptr)))
184 {
185 return MEMCACHED_FAILURE;
186 }
187 }
188
189 struct pollfd fds;
190 memset(&fds, 0, sizeof(pollfd));
191 fds.fd= ptr->fd;
192 fds.events= POLLIN;
193
194 if (read_or_write == MEM_WRITE) /* write */
195 {
196 fds.events= POLLOUT;
197 ptr->io_wait_count.write++;
198 }
199 else
200 {
201 ptr->io_wait_count.read++;
202 }
203
204 if (ptr->root->poll_timeout == 0) // Mimic 0 causes timeout behavior (not all platforms do this)
205 {
206 return memcached_set_error(*ptr, MEMCACHED_TIMEOUT, MEMCACHED_AT);
207 }
208
209 size_t loop_max= 5;
210 while (--loop_max) // While loop is for ERESTART or EINTR
211 {
212 int active_fd= poll(&fds, 1, ptr->root->poll_timeout);
213 assert_msg(active_fd <= 1 , "poll() returned an unexpected value");
214
215 if (active_fd == 1)
216 {
217 return MEMCACHED_SUCCESS;
218 }
219 else if (active_fd == 0)
220 {
221 ptr->io_wait_count.timeouts++;
222 return memcached_set_error(*ptr, MEMCACHED_TIMEOUT, MEMCACHED_AT);
223 }
224 else // -1
225 {
226 switch (get_socket_errno())
227 {
228 #ifdef TARGET_OS_LINUX
229 case ERESTART:
230 #endif
231 case EINTR:
232 break;
233
234 case EFAULT:
235 case ENOMEM:
236 return memcached_set_error(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT);
237
238 case EINVAL:
239 return memcached_set_error(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT, memcached_literal_param("RLIMIT_NOFILE exceeded, or if OSX the timeout value was invalid"));
240
241 default:
242 int local_errno= get_socket_errno(); // We cache in case memcached_quit_server() modifies errno
243 if (fds.revents & POLLERR)
244 {
245 int err;
246 socklen_t len= sizeof (err);
247 if (getsockopt(ptr->fd, SOL_SOCKET, SO_ERROR, &err, &len) == 0)
248 {
249 if (err == 0) // treat this as EINTR
250 {
251 continue;
252 }
253 local_errno= err;
254 }
255 }
256 memcached_quit_server(ptr, true);
257
258 return memcached_set_errno(*ptr, local_errno, MEMCACHED_AT);
259 }
260 }
261 }
262
263 memcached_quit_server(ptr, true);
264
265 return memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
266 }
267
268 static bool io_flush(memcached_server_write_instance_st ptr,
269 const bool with_flush,
270 memcached_return_t& error)
271 {
272 /*
273 ** We might want to purge the input buffer if we haven't consumed
274 ** any output yet... The test for the limits is the purge is inline
275 ** in the purge function to avoid duplicating the logic..
276 */
277 {
278 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
279 memcached_return_t rc= memcached_purge(ptr);
280
281 if (rc != MEMCACHED_SUCCESS and rc != MEMCACHED_STORED)
282 {
283 return false;
284 }
285 }
286 char *local_write_ptr= ptr->write_buffer;
287 size_t write_length= ptr->write_buffer_offset;
288
289 error= MEMCACHED_SUCCESS;
290
291 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
292
293 /* Looking for memory overflows */
294 #if defined(DEBUG)
295 if (write_length == MEMCACHED_MAX_BUFFER)
296 WATCHPOINT_ASSERT(ptr->write_buffer == local_write_ptr);
297 WATCHPOINT_ASSERT((ptr->write_buffer + MEMCACHED_MAX_BUFFER) >= (local_write_ptr + write_length));
298 #endif
299
300 while (write_length)
301 {
302 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
303 WATCHPOINT_ASSERT(write_length > 0);
304
305 ssize_t sent_length= 0;
306 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
307 if (with_flush)
308 {
309 sent_length= ::send(ptr->fd, local_write_ptr, write_length, MSG_NOSIGNAL|MSG_DONTWAIT);
310 }
311 else
312 {
313 sent_length= ::send(ptr->fd, local_write_ptr, write_length, MSG_NOSIGNAL|MSG_DONTWAIT|MSG_MORE);
314 }
315
316 if (sent_length == SOCKET_ERROR)
317 {
318 #if 0 // @todo I should look at why we hit this bit of code hard frequently
319 WATCHPOINT_ERRNO(get_socket_errno());
320 WATCHPOINT_NUMBER(get_socket_errno());
321 #endif
322 switch (get_socket_errno())
323 {
324 case ENOBUFS:
325 continue;
326
327 #if EWOULDBLOCK != EAGAIN
328 case EWOULDBLOCK:
329 #endif
330 case EAGAIN:
331 {
332 /*
333 * We may be blocked on write because the input buffer
334 * is full. Let's check if we have room in our input
335 * buffer for more data and retry the write before
336 * waiting..
337 */
338 if (repack_input_buffer(ptr) or process_input_buffer(ptr))
339 {
340 continue;
341 }
342
343 memcached_return_t rc= io_wait(ptr, MEM_WRITE);
344 if (memcached_success(rc))
345 {
346 continue;
347 }
348 else if (rc == MEMCACHED_TIMEOUT)
349 {
350 ptr->io_wait_count.timeouts++;
351 error= memcached_set_error(*ptr, MEMCACHED_TIMEOUT, MEMCACHED_AT);
352 return false;
353 }
354
355 memcached_quit_server(ptr, true);
356 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
357 return false;
358 }
359 case ENOTCONN:
360 case EPIPE:
361 default:
362 memcached_quit_server(ptr, true);
363 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
364 WATCHPOINT_ASSERT(ptr->fd == INVALID_SOCKET);
365 return false;
366 }
367 }
368
369 ptr->io_bytes_sent+= uint32_t(sent_length);
370
371 local_write_ptr+= sent_length;
372 write_length-= uint32_t(sent_length);
373 }
374
375 WATCHPOINT_ASSERT(write_length == 0);
376 ptr->write_buffer_offset= 0;
377
378 return true;
379 }
380
381 memcached_return_t memcached_io_wait_for_write(memcached_server_write_instance_st ptr)
382 {
383 return io_wait(ptr, MEM_WRITE);
384 }
385
386 memcached_return_t memcached_io_read(memcached_server_write_instance_st ptr,
387 void *buffer, size_t length, ssize_t& nread)
388 {
389 assert(memcached_is_udp(ptr->root) == false);
390 assert_msg(ptr, "Programmer error, memcached_io_read() recieved an invalid memcached_server_write_instance_st"); // Programmer error
391 char *buffer_ptr= static_cast<char *>(buffer);
392
393 if (ptr->fd == INVALID_SOCKET)
394 {
395 #if 0
396 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Programmer error, invalid socket state");
397 #endif
398 return MEMCACHED_CONNECTION_FAILURE;
399 }
400
401 while (length)
402 {
403 if (ptr->read_buffer_length == 0)
404 {
405 ssize_t data_read;
406 do
407 {
408 data_read= ::recv(ptr->fd, ptr->read_buffer, MEMCACHED_MAX_BUFFER, MSG_DONTWAIT);
409 if (data_read == SOCKET_ERROR)
410 {
411 switch (get_socket_errno())
412 {
413 case EINTR: // We just retry
414 continue;
415
416 case ETIMEDOUT: // OSX
417 #if EWOULDBLOCK != EAGAIN
418 case EWOULDBLOCK:
419 #endif
420 case EAGAIN:
421 #ifdef TARGET_OS_LINUX
422 case ERESTART:
423 #endif
424 {
425 memcached_return_t io_wait_ret;
426 if (memcached_success(io_wait_ret= io_wait(ptr, MEM_READ)))
427 {
428 continue;
429 }
430
431 return io_wait_ret;
432 }
433
434 /* fall through */
435
436 case ENOTCONN: // Programmer Error
437 WATCHPOINT_ASSERT(0);
438 case ENOTSOCK:
439 WATCHPOINT_ASSERT(0);
440 case EBADF:
441 assert_msg(ptr->fd != INVALID_SOCKET, "Programmer error, invalid socket");
442 case EINVAL:
443 case EFAULT:
444 case ECONNREFUSED:
445 default:
446 {
447 memcached_quit_server(ptr, true);
448 nread= -1;
449 return memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
450 }
451 }
452 }
453 else if (data_read == 0)
454 {
455 /*
456 EOF. Any data received so far is incomplete
457 so discard it. This always reads by byte in case of TCP
458 and protocol enforcement happens at memcached_response()
459 looking for '\n'. We do not care for UDB which requests 8 bytes
460 at once. Generally, this means that connection went away. Since
461 for blocking I/O we do not return 0 and for non-blocking case
462 it will return EGAIN if data is not immediatly available.
463 */
464 WATCHPOINT_STRING("We had a zero length recv()");
465 memcached_quit_server(ptr, true);
466 nread= -1;
467 return memcached_set_error(*ptr, MEMCACHED_CONNECTION_FAILURE, MEMCACHED_AT,
468 memcached_literal_param("::rec() returned zero, server has disconnected"));
469 }
470 } while (data_read <= 0);
471
472 ptr->io_bytes_sent = 0;
473 ptr->read_data_length= (size_t) data_read;
474 ptr->read_buffer_length= (size_t) data_read;
475 ptr->read_ptr= ptr->read_buffer;
476 }
477
478 if (length > 1)
479 {
480 size_t difference;
481
482 difference= (length > ptr->read_buffer_length) ? ptr->read_buffer_length : length;
483
484 memcpy(buffer_ptr, ptr->read_ptr, difference);
485 length -= difference;
486 ptr->read_ptr+= difference;
487 ptr->read_buffer_length-= difference;
488 buffer_ptr+= difference;
489 }
490 else
491 {
492 *buffer_ptr= *ptr->read_ptr;
493 ptr->read_ptr++;
494 ptr->read_buffer_length--;
495 buffer_ptr++;
496 break;
497 }
498 }
499
500 nread= ssize_t(buffer_ptr - (char*)buffer);
501
502 return MEMCACHED_SUCCESS;
503 }
504
505 memcached_return_t memcached_io_slurp(memcached_server_write_instance_st ptr)
506 {
507 assert_msg(ptr, "Programmer error, invalid memcached_server_write_instance_st");
508 assert(memcached_is_udp(ptr->root) == false);
509
510 if (ptr->fd == INVALID_SOCKET)
511 {
512 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Invalid socket state");
513 return MEMCACHED_CONNECTION_FAILURE;
514 }
515
516 ssize_t data_read;
517 char buffer[MEMCACHED_MAX_BUFFER];
518 do
519 {
520 data_read= recv(ptr->fd, ptr->read_buffer, sizeof(buffer), MSG_DONTWAIT);
521 if (data_read == SOCKET_ERROR)
522 {
523 switch (get_socket_errno())
524 {
525 case EINTR: // We just retry
526 continue;
527
528 case ETIMEDOUT: // OSX
529 #if EWOULDBLOCK != EAGAIN
530 case EWOULDBLOCK:
531 #endif
532 case EAGAIN:
533 #ifdef TARGET_OS_LINUX
534 case ERESTART:
535 #endif
536 if (memcached_success(io_wait(ptr, MEM_READ)))
537 {
538 continue;
539 }
540 return MEMCACHED_IN_PROGRESS;
541
542 /* fall through */
543
544 case ENOTCONN: // Programmer Error
545 WATCHPOINT_ASSERT(0);
546 case ENOTSOCK:
547 WATCHPOINT_ASSERT(0);
548 case EBADF:
549 assert_msg(ptr->fd != INVALID_SOCKET, "Invalid socket state");
550 case EINVAL:
551 case EFAULT:
552 case ECONNREFUSED:
553 default:
554 return MEMCACHED_CONNECTION_FAILURE; // We want this!
555 }
556 }
557 } while (data_read > 0);
558
559 return MEMCACHED_CONNECTION_FAILURE;
560 }
561
562 static ssize_t _io_write(memcached_server_write_instance_st ptr,
563 const void *buffer, size_t length, bool with_flush)
564 {
565 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
566 assert(memcached_is_udp(ptr->root) == false);
567
568 size_t original_length= length;
569 const char *buffer_ptr= static_cast<const char *>(buffer);
570
571 while (length)
572 {
573 char *write_ptr;
574 size_t buffer_end= MEMCACHED_MAX_BUFFER;
575 size_t should_write= buffer_end -ptr->write_buffer_offset;
576 should_write= (should_write < length) ? should_write : length;
577
578 write_ptr= ptr->write_buffer + ptr->write_buffer_offset;
579 memcpy(write_ptr, buffer_ptr, should_write);
580 ptr->write_buffer_offset+= should_write;
581 buffer_ptr+= should_write;
582 length-= should_write;
583
584 if (ptr->write_buffer_offset == buffer_end)
585 {
586 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
587
588 memcached_return_t rc;
589 if (io_flush(ptr, with_flush, rc) == false)
590 {
591 return -1;
592 }
593 }
594 }
595
596 if (with_flush)
597 {
598 memcached_return_t rc;
599 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
600 if (io_flush(ptr, with_flush, rc) == false)
601 {
602 return -1;
603 }
604 }
605
606 return ssize_t(original_length);
607 }
608
609 bool memcached_io_write(memcached_server_write_instance_st ptr)
610 {
611 return (_io_write(ptr, NULL, 0, true) >= 0);
612 }
613
614 ssize_t memcached_io_write(memcached_server_write_instance_st ptr,
615 const void *buffer, const size_t length, const bool with_flush)
616 {
617 return _io_write(ptr, buffer, length, with_flush);
618 }
619
620 ssize_t memcached_io_writev(memcached_server_write_instance_st ptr,
621 libmemcached_io_vector_st vector[],
622 const size_t number_of, const bool with_flush)
623 {
624 ssize_t total= 0;
625
626 for (size_t x= 0; x < number_of; x++, vector++)
627 {
628 ssize_t returnable;
629
630 if (vector->length)
631 {
632 if ((returnable= _io_write(ptr, vector->buffer, vector->length, false)) == -1)
633 {
634 return -1;
635 }
636 total+= returnable;
637 }
638 }
639
640 if (with_flush)
641 {
642 if (memcached_io_write(ptr) == false)
643 {
644 return -1;
645 }
646 }
647
648 return total;
649 }
650
651
652 void memcached_io_close(memcached_server_write_instance_st ptr)
653 {
654 if (ptr->fd == INVALID_SOCKET)
655 {
656 return;
657 }
658
659 /* in case of death shutdown to avoid blocking at close() */
660 if (shutdown(ptr->fd, SHUT_RDWR) == SOCKET_ERROR && get_socket_errno() != ENOTCONN)
661 {
662 WATCHPOINT_NUMBER(ptr->fd);
663 WATCHPOINT_ERRNO(get_socket_errno());
664 WATCHPOINT_ASSERT(get_socket_errno());
665 }
666
667 if (closesocket(ptr->fd) == SOCKET_ERROR)
668 {
669 WATCHPOINT_ERRNO(get_socket_errno());
670 }
671 ptr->state= MEMCACHED_SERVER_STATE_NEW;
672 ptr->fd= INVALID_SOCKET;
673 }
674
675 memcached_server_write_instance_st memcached_io_get_readable_server(memcached_st *memc)
676 {
677 #define MAX_SERVERS_TO_POLL 100
678 struct pollfd fds[MAX_SERVERS_TO_POLL];
679 nfds_t host_index= 0;
680
681 for (uint32_t x= 0; x < memcached_server_count(memc) and host_index < MAX_SERVERS_TO_POLL; ++x)
682 {
683 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, x);
684
685 if (instance->read_buffer_length > 0) /* I have data in the buffer */
686 {
687 return instance;
688 }
689
690 if (memcached_server_response_count(instance) > 0)
691 {
692 fds[host_index].events = POLLIN;
693 fds[host_index].revents = 0;
694 fds[host_index].fd = instance->fd;
695 ++host_index;
696 }
697 }
698
699 if (host_index < 2)
700 {
701 /* We have 0 or 1 server with pending events.. */
702 for (uint32_t x= 0; x< memcached_server_count(memc); ++x)
703 {
704 memcached_server_write_instance_st instance=
705 memcached_server_instance_fetch(memc, x);
706
707 if (memcached_server_response_count(instance) > 0)
708 {
709 return instance;
710 }
711 }
712
713 return NULL;
714 }
715
716 int error= poll(fds, host_index, memc->poll_timeout);
717 switch (error)
718 {
719 case -1:
720 memcached_set_errno(*memc, get_socket_errno(), MEMCACHED_AT);
721 /* FALLTHROUGH */
722 case 0:
723 break;
724
725 default:
726 for (nfds_t x= 0; x < host_index; ++x)
727 {
728 if (fds[x].revents & POLLIN)
729 {
730 for (uint32_t y= 0; y < memcached_server_count(memc); ++y)
731 {
732 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, y);
733
734 if (instance->fd == fds[x].fd)
735 {
736 return instance;
737 }
738 }
739 }
740 }
741 }
742
743 return NULL;
744 }
745
746 /*
747 Eventually we will just kill off the server with the problem.
748 */
749 void memcached_io_reset(memcached_server_write_instance_st ptr)
750 {
751 memcached_quit_server(ptr, true);
752 }
753
754 /**
755 * Read a given number of bytes from the server and place it into a specific
756 * buffer. Reset the IO channel on this server if an error occurs.
757 */
758 memcached_return_t memcached_safe_read(memcached_server_write_instance_st ptr,
759 void *dta,
760 const size_t size)
761 {
762 size_t offset= 0;
763 char *data= static_cast<char *>(dta);
764
765 while (offset < size)
766 {
767 ssize_t nread;
768 memcached_return_t rc;
769
770 while (memcached_continue(rc= memcached_io_read(ptr, data + offset, size - offset, nread))) { };
771
772 if (memcached_failed(rc))
773 {
774 return rc;
775 }
776
777 offset+= size_t(nread);
778 }
779
780 return MEMCACHED_SUCCESS;
781 }
782
783 memcached_return_t memcached_io_readline(memcached_server_write_instance_st ptr,
784 char *buffer_ptr,
785 size_t size,
786 size_t& total_nr)
787 {
788 total_nr= 0;
789 bool line_complete= false;
790
791 while (line_complete == false)
792 {
793 if (ptr->read_buffer_length == 0)
794 {
795 /*
796 * We don't have any data in the buffer, so let's fill the read
797 * buffer. Call the standard read function to avoid duplicating
798 * the logic.
799 */
800 ssize_t nread;
801 memcached_return_t rc= memcached_io_read(ptr, buffer_ptr, 1, nread);
802 if (memcached_failed(rc) and rc == MEMCACHED_IN_PROGRESS)
803 {
804 memcached_quit_server(ptr, true);
805 return memcached_set_error(*ptr, rc, MEMCACHED_AT);
806 }
807 else if (memcached_failed(rc))
808 {
809 return rc;
810 }
811
812 if (*buffer_ptr == '\n')
813 {
814 line_complete= true;
815 }
816
817 ++buffer_ptr;
818 ++total_nr;
819 }
820
821 /* Now let's look in the buffer and copy as we go! */
822 while (ptr->read_buffer_length && total_nr < size && !line_complete)
823 {
824 *buffer_ptr = *ptr->read_ptr;
825 if (*buffer_ptr == '\n')
826 {
827 line_complete = true;
828 }
829 --ptr->read_buffer_length;
830 ++ptr->read_ptr;
831 ++total_nr;
832 ++buffer_ptr;
833 }
834
835 if (total_nr == size)
836 {
837 return MEMCACHED_PROTOCOL_ERROR;
838 }
839 }
840
841 return MEMCACHED_SUCCESS;
842 }