Fixed bug where we might not see the right return type for the distribution type.
[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;
71 if ((nr= recv(ptr->fd,
72 ptr->read_ptr + ptr->read_data_length,
73 MEMCACHED_MAX_BUFFER - ptr->read_data_length,
74 MSG_DONTWAIT)) <= 0)
75 {
76 if (nr == 0)
77 {
78 memcached_set_error(*ptr, MEMCACHED_CONNECTION_FAILURE, MEMCACHED_AT);
79 }
80 else
81 {
82 switch (get_socket_errno())
83 {
84 case EINTR:
85 continue;
86
87 #if EWOULDBLOCK != EAGAIN
88 case EWOULDBLOCK:
89 #endif
90 case EAGAIN:
91 #ifdef TARGET_OS_LINUX
92 case ERESTART:
93 #endif
94 break; // No IO is fine, we can just move on
95
96 default:
97 memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
98 }
99 }
100
101 break;
102 }
103 else // We read data, append to our read buffer
104 {
105 ptr->read_data_length+= size_t(nr);
106 ptr->read_buffer_length+= size_t(nr);
107
108 return true;
109 }
110 } while (false);
111 }
112
113 return false;
114 }
115
116 /**
117 * If the we have callbacks connected to this server structure
118 * we may start process the input queue and fire the callbacks
119 * for the incomming messages. This function is _only_ called
120 * when the input buffer is full, so that we _know_ that we have
121 * at least _one_ message to process.
122 *
123 * @param ptr the server to star processing iput messages for
124 * @return true if we processed anything, false otherwise
125 */
126 static bool process_input_buffer(memcached_server_write_instance_st ptr)
127 {
128 /*
129 ** We might be able to process some of the response messages if we
130 ** have a callback set up
131 */
132 if (ptr->root->callbacks != NULL)
133 {
134 /*
135 * We might have responses... try to read them out and fire
136 * callbacks
137 */
138 memcached_callback_st cb= *ptr->root->callbacks;
139
140 memcached_set_processing_input((memcached_st *)ptr->root, true);
141
142 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
143 memcached_st *root= (memcached_st *)ptr->root;
144 memcached_return_t error= memcached_response(ptr, buffer, sizeof(buffer), &root->result);
145
146 memcached_set_processing_input(root, false);
147
148 if (error == MEMCACHED_SUCCESS)
149 {
150 for (unsigned int x= 0; x < cb.number_of_callback; x++)
151 {
152 error= (*cb.callback[x])(ptr->root, &root->result, cb.context);
153 if (error != MEMCACHED_SUCCESS)
154 {
155 break;
156 }
157 }
158
159 /* @todo what should I do with the error message??? */
160 }
161 /* @todo what should I do with other error messages?? */
162 return true;
163 }
164
165 return false;
166 }
167
168 static memcached_return_t io_wait(memcached_server_write_instance_st ptr,
169 const memc_read_or_write read_or_write)
170 {
171 /*
172 ** We are going to block on write, but at least on Solaris we might block
173 ** on write if we haven't read anything from our input buffer..
174 ** Try to purge the input buffer if we don't do any flow control in the
175 ** application layer (just sending a lot of data etc)
176 ** The test is moved down in the purge function to avoid duplication of
177 ** the test.
178 */
179 if (read_or_write == MEM_WRITE)
180 {
181 if (memcached_purge(ptr) == false)
182 {
183 return MEMCACHED_FAILURE;
184 }
185 }
186
187 struct pollfd fds;
188 fds.fd= ptr->fd;
189 fds.events= POLLIN;
190 fds.revents= 0;
191
192 if (read_or_write == MEM_WRITE) /* write */
193 {
194 fds.events= POLLOUT;
195 ptr->io_wait_count.write++;
196 }
197 else
198 {
199 ptr->io_wait_count.read++;
200 }
201
202 if (ptr->root->poll_timeout == 0) // Mimic 0 causes timeout behavior (not all platforms do this)
203 {
204 ptr->io_wait_count.timeouts++;
205 return memcached_set_error(*ptr, MEMCACHED_TIMEOUT, MEMCACHED_AT);
206 }
207
208 int local_errno;
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
214 if (active_fd >= 1)
215 {
216 assert_msg(active_fd == 1 , "poll() returned an unexpected value");
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
225 // Only an error should result in this code being called.
226 local_errno= get_socket_errno(); // We cache in case memcached_quit_server() modifies errno
227 assert_msg(active_fd == -1 , "poll() returned an unexpected value");
228 switch (local_errno)
229 {
230 #ifdef TARGET_OS_LINUX
231 case ERESTART:
232 #endif
233 case EINTR:
234 continue;
235
236 case EFAULT:
237 case ENOMEM:
238 return memcached_set_error(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT);
239
240 case EINVAL:
241 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"));
242
243 default:
244 if (fds.revents & POLLERR)
245 {
246 int err;
247 socklen_t len= sizeof (err);
248 if (getsockopt(ptr->fd, SOL_SOCKET, SO_ERROR, &err, &len) == 0)
249 {
250 if (err == 0) // treat this as EINTR
251 {
252 continue;
253 }
254 local_errno= err;
255 }
256 }
257 break;
258 }
259
260 break; // should only occur from poll error
261 }
262
263 memcached_quit_server(ptr, true);
264
265 return memcached_set_errno(*ptr, local_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
280 if (memcached_purge(ptr) == false)
281 {
282 return false;
283 }
284 }
285 char *local_write_ptr= ptr->write_buffer;
286 size_t write_length= ptr->write_buffer_offset;
287
288 error= MEMCACHED_SUCCESS;
289
290 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
291
292 /* Looking for memory overflows */
293 #if defined(DEBUG)
294 if (write_length == MEMCACHED_MAX_BUFFER)
295 WATCHPOINT_ASSERT(ptr->write_buffer == local_write_ptr);
296 WATCHPOINT_ASSERT((ptr->write_buffer + MEMCACHED_MAX_BUFFER) >= (local_write_ptr + write_length));
297 #endif
298
299 while (write_length)
300 {
301 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
302 WATCHPOINT_ASSERT(write_length > 0);
303
304 int flags= with_flush ? MSG_NOSIGNAL|MSG_DONTWAIT : MSG_NOSIGNAL|MSG_DONTWAIT|MSG_MORE;
305 ssize_t sent_length= ::send(ptr->fd, local_write_ptr, write_length, flags);
306
307 if (sent_length == SOCKET_ERROR)
308 {
309 #if 0 // @todo I should look at why we hit this bit of code hard frequently
310 WATCHPOINT_ERRNO(get_socket_errno());
311 WATCHPOINT_NUMBER(get_socket_errno());
312 #endif
313 switch (get_socket_errno())
314 {
315 case ENOBUFS:
316 continue;
317
318 #if EWOULDBLOCK != EAGAIN
319 case EWOULDBLOCK:
320 #endif
321 case EAGAIN:
322 {
323 /*
324 * We may be blocked on write because the input buffer
325 * is full. Let's check if we have room in our input
326 * buffer for more data and retry the write before
327 * waiting..
328 */
329 if (repack_input_buffer(ptr) or process_input_buffer(ptr))
330 {
331 continue;
332 }
333
334 memcached_return_t rc= io_wait(ptr, MEM_WRITE);
335 if (memcached_success(rc))
336 {
337 continue;
338 }
339 else if (rc == MEMCACHED_TIMEOUT)
340 {
341 return false;
342 }
343
344 memcached_quit_server(ptr, true);
345 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
346 return false;
347 }
348 case ENOTCONN:
349 case EPIPE:
350 default:
351 memcached_quit_server(ptr, true);
352 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
353 WATCHPOINT_ASSERT(ptr->fd == INVALID_SOCKET);
354 return false;
355 }
356 }
357
358 ptr->io_bytes_sent+= uint32_t(sent_length);
359
360 local_write_ptr+= sent_length;
361 write_length-= uint32_t(sent_length);
362 }
363
364 WATCHPOINT_ASSERT(write_length == 0);
365 ptr->write_buffer_offset= 0;
366
367 return true;
368 }
369
370 memcached_return_t memcached_io_wait_for_write(memcached_server_write_instance_st ptr)
371 {
372 return io_wait(ptr, MEM_WRITE);
373 }
374
375 static memcached_return_t _io_fill(memcached_server_write_instance_st ptr)
376 {
377 ssize_t data_read;
378 do
379 {
380 data_read= ::recv(ptr->fd, ptr->read_buffer, MEMCACHED_MAX_BUFFER, MSG_DONTWAIT);
381 if (data_read == SOCKET_ERROR)
382 {
383 switch (get_socket_errno())
384 {
385 case EINTR: // We just retry
386 continue;
387
388 case ETIMEDOUT: // OSX
389 #if EWOULDBLOCK != EAGAIN
390 case EWOULDBLOCK:
391 #endif
392 case EAGAIN:
393 #ifdef TARGET_OS_LINUX
394 case ERESTART:
395 #endif
396 {
397 memcached_return_t io_wait_ret;
398 if (memcached_success(io_wait_ret= io_wait(ptr, MEM_READ)))
399 {
400 continue;
401 }
402
403 return io_wait_ret;
404 }
405
406 /* fall through */
407
408 case ENOTCONN: // Programmer Error
409 WATCHPOINT_ASSERT(0);
410 case ENOTSOCK:
411 WATCHPOINT_ASSERT(0);
412 case EBADF:
413 assert_msg(ptr->fd != INVALID_SOCKET, "Programmer error, invalid socket");
414 case EINVAL:
415 case EFAULT:
416 case ECONNREFUSED:
417 default:
418 memcached_quit_server(ptr, true);
419 memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
420 break;
421 }
422
423 return memcached_server_error_return(ptr);
424 }
425 else if (data_read == 0)
426 {
427 /*
428 EOF. Any data received so far is incomplete
429 so discard it. This always reads by byte in case of TCP
430 and protocol enforcement happens at memcached_response()
431 looking for '\n'. We do not care for UDB which requests 8 bytes
432 at once. Generally, this means that connection went away. Since
433 for blocking I/O we do not return 0 and for non-blocking case
434 it will return EGAIN if data is not immediatly available.
435 */
436 memcached_quit_server(ptr, true);
437 return memcached_set_error(*ptr, MEMCACHED_CONNECTION_FAILURE, MEMCACHED_AT,
438 memcached_literal_param("::rec() returned zero, server has disconnected"));
439 }
440 ptr->io_wait_count._bytes_read+= data_read;
441 } while (data_read <= 0);
442
443 ptr->io_bytes_sent= 0;
444 ptr->read_data_length= (size_t) data_read;
445 ptr->read_buffer_length= (size_t) data_read;
446 ptr->read_ptr= ptr->read_buffer;
447
448 return MEMCACHED_SUCCESS;
449 }
450
451 memcached_return_t memcached_io_read(memcached_server_write_instance_st ptr,
452 void *buffer, size_t length, ssize_t& nread)
453 {
454 assert(memcached_is_udp(ptr->root) == false);
455 assert_msg(ptr, "Programmer error, memcached_io_read() recieved an invalid memcached_server_write_instance_st"); // Programmer error
456 char *buffer_ptr= static_cast<char *>(buffer);
457
458 if (ptr->fd == INVALID_SOCKET)
459 {
460 #if 0
461 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Programmer error, invalid socket state");
462 #endif
463 return MEMCACHED_CONNECTION_FAILURE;
464 }
465
466 while (length)
467 {
468 if (ptr->read_buffer_length == 0)
469 {
470 memcached_return_t io_fill_ret;
471 if (memcached_fatal(io_fill_ret= _io_fill(ptr)))
472 {
473 nread= -1;
474 return io_fill_ret;
475 }
476 }
477
478 if (length > 1)
479 {
480 size_t difference= (length > ptr->read_buffer_length) ? ptr->read_buffer_length : length;
481
482 memcpy(buffer_ptr, ptr->read_ptr, difference);
483 length -= difference;
484 ptr->read_ptr+= difference;
485 ptr->read_buffer_length-= difference;
486 buffer_ptr+= difference;
487 }
488 else
489 {
490 *buffer_ptr= *ptr->read_ptr;
491 ptr->read_ptr++;
492 ptr->read_buffer_length--;
493 buffer_ptr++;
494 break;
495 }
496 }
497
498 nread= ssize_t(buffer_ptr - (char*)buffer);
499
500 return MEMCACHED_SUCCESS;
501 }
502
503 memcached_return_t memcached_io_slurp(memcached_server_write_instance_st ptr)
504 {
505 assert_msg(ptr, "Programmer error, invalid memcached_server_write_instance_st");
506 assert(memcached_is_udp(ptr->root) == false);
507
508 if (ptr->fd == INVALID_SOCKET)
509 {
510 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Invalid socket state");
511 return MEMCACHED_CONNECTION_FAILURE;
512 }
513
514 ssize_t data_read;
515 char buffer[MEMCACHED_MAX_BUFFER];
516 do
517 {
518 data_read= recv(ptr->fd, ptr->read_buffer, sizeof(buffer), MSG_DONTWAIT);
519 if (data_read == SOCKET_ERROR)
520 {
521 switch (get_socket_errno())
522 {
523 case EINTR: // We just retry
524 continue;
525
526 case ETIMEDOUT: // OSX
527 #if EWOULDBLOCK != EAGAIN
528 case EWOULDBLOCK:
529 #endif
530 case EAGAIN:
531 #ifdef TARGET_OS_LINUX
532 case ERESTART:
533 #endif
534 if (memcached_success(io_wait(ptr, MEM_READ)))
535 {
536 continue;
537 }
538 return MEMCACHED_IN_PROGRESS;
539
540 /* fall through */
541
542 case ENOTCONN: // Programmer Error
543 assert(0);
544 case ENOTSOCK:
545 assert(0);
546 case EBADF:
547 assert_msg(ptr->fd != INVALID_SOCKET, "Invalid socket state");
548 case EINVAL:
549 case EFAULT:
550 case ECONNREFUSED:
551 default:
552 return MEMCACHED_CONNECTION_FAILURE; // We want this!
553 }
554 }
555 } while (data_read > 0);
556
557 return MEMCACHED_CONNECTION_FAILURE;
558 }
559
560 static bool _io_write(memcached_server_write_instance_st ptr,
561 const void *buffer, size_t length, bool with_flush,
562 size_t& written)
563 {
564 assert(ptr->fd != INVALID_SOCKET);
565 assert(memcached_is_udp(ptr->root) == false);
566
567 const char *buffer_ptr= static_cast<const char *>(buffer);
568
569 const size_t original_length= length;
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 written= original_length -length;
592 return false;
593 }
594 }
595 }
596
597 if (with_flush)
598 {
599 memcached_return_t rc;
600 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
601 if (io_flush(ptr, with_flush, rc) == false)
602 {
603 written= original_length -length;
604 return false;
605 }
606 }
607
608 written= original_length -length;
609
610 return true;
611 }
612
613 bool memcached_io_write(memcached_server_write_instance_st ptr)
614 {
615 size_t written;
616 return _io_write(ptr, NULL, 0, true, written);
617 }
618
619 ssize_t memcached_io_write(memcached_server_write_instance_st ptr,
620 const void *buffer, const size_t length, const bool with_flush)
621 {
622 size_t written;
623
624 if (_io_write(ptr, buffer, length, with_flush, written) == false)
625 {
626 return -1;
627 }
628
629 return ssize_t(written);
630 }
631
632 bool memcached_io_writev(memcached_server_write_instance_st ptr,
633 libmemcached_io_vector_st vector[],
634 const size_t number_of, const bool with_flush)
635 {
636 ssize_t complete_total= 0;
637 ssize_t total= 0;
638
639 for (size_t x= 0; x < number_of; x++, vector++)
640 {
641 complete_total+= vector->length;
642 if (vector->length)
643 {
644 size_t written;
645 if ((_io_write(ptr, vector->buffer, vector->length, false, written)) == false)
646 {
647 return false;
648 }
649 total+= written;
650 }
651 }
652
653 if (with_flush)
654 {
655 if (memcached_io_write(ptr) == false)
656 {
657 return false;
658 }
659 }
660
661 return (complete_total == total);
662 }
663
664
665 void memcached_io_close(memcached_server_write_instance_st ptr)
666 {
667 if (ptr->fd == INVALID_SOCKET)
668 {
669 return;
670 }
671
672 /* in case of death shutdown to avoid blocking at close() */
673 if (shutdown(ptr->fd, SHUT_RDWR) == SOCKET_ERROR && get_socket_errno() != ENOTCONN)
674 {
675 WATCHPOINT_NUMBER(ptr->fd);
676 WATCHPOINT_ERRNO(get_socket_errno());
677 WATCHPOINT_ASSERT(get_socket_errno());
678 }
679
680 if (closesocket(ptr->fd) == SOCKET_ERROR)
681 {
682 WATCHPOINT_ERRNO(get_socket_errno());
683 }
684 ptr->state= MEMCACHED_SERVER_STATE_NEW;
685 ptr->fd= INVALID_SOCKET;
686 }
687
688 memcached_server_write_instance_st memcached_io_get_readable_server(memcached_st *memc)
689 {
690 #define MAX_SERVERS_TO_POLL 100
691 struct pollfd fds[MAX_SERVERS_TO_POLL];
692 nfds_t host_index= 0;
693
694 for (uint32_t x= 0; x < memcached_server_count(memc) and host_index < MAX_SERVERS_TO_POLL; ++x)
695 {
696 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, x);
697
698 if (instance->read_buffer_length > 0) /* I have data in the buffer */
699 {
700 return instance;
701 }
702
703 if (memcached_server_response_count(instance) > 0)
704 {
705 fds[host_index].events = POLLIN;
706 fds[host_index].revents = 0;
707 fds[host_index].fd = instance->fd;
708 ++host_index;
709 }
710 }
711
712 if (host_index < 2)
713 {
714 /* We have 0 or 1 server with pending events.. */
715 for (uint32_t x= 0; x< memcached_server_count(memc); ++x)
716 {
717 memcached_server_write_instance_st instance=
718 memcached_server_instance_fetch(memc, x);
719
720 if (memcached_server_response_count(instance) > 0)
721 {
722 return instance;
723 }
724 }
725
726 return NULL;
727 }
728
729 int error= poll(fds, host_index, memc->poll_timeout);
730 switch (error)
731 {
732 case -1:
733 memcached_set_errno(*memc, get_socket_errno(), MEMCACHED_AT);
734 /* FALLTHROUGH */
735 case 0:
736 break;
737
738 default:
739 for (nfds_t x= 0; x < host_index; ++x)
740 {
741 if (fds[x].revents & POLLIN)
742 {
743 for (uint32_t y= 0; y < memcached_server_count(memc); ++y)
744 {
745 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, y);
746
747 if (instance->fd == fds[x].fd)
748 {
749 return instance;
750 }
751 }
752 }
753 }
754 }
755
756 return NULL;
757 }
758
759 /*
760 Eventually we will just kill off the server with the problem.
761 */
762 void memcached_io_reset(memcached_server_write_instance_st ptr)
763 {
764 memcached_quit_server(ptr, true);
765 }
766
767 /**
768 * Read a given number of bytes from the server and place it into a specific
769 * buffer. Reset the IO channel on this server if an error occurs.
770 */
771 memcached_return_t memcached_safe_read(memcached_server_write_instance_st ptr,
772 void *dta,
773 const size_t size)
774 {
775 size_t offset= 0;
776 char *data= static_cast<char *>(dta);
777
778 while (offset < size)
779 {
780 ssize_t nread;
781 memcached_return_t rc;
782
783 while (memcached_continue(rc= memcached_io_read(ptr, data + offset, size - offset, nread))) { };
784
785 if (memcached_failed(rc))
786 {
787 return rc;
788 }
789
790 offset+= size_t(nread);
791 }
792
793 return MEMCACHED_SUCCESS;
794 }
795
796 memcached_return_t memcached_io_readline(memcached_server_write_instance_st ptr,
797 char *buffer_ptr,
798 size_t size,
799 size_t& total_nr)
800 {
801 total_nr= 0;
802 bool line_complete= false;
803
804 while (line_complete == false)
805 {
806 if (ptr->read_buffer_length == 0)
807 {
808 /*
809 * We don't have any data in the buffer, so let's fill the read
810 * buffer. Call the standard read function to avoid duplicating
811 * the logic.
812 */
813 ssize_t nread;
814 memcached_return_t rc= memcached_io_read(ptr, buffer_ptr, 1, nread);
815 if (memcached_failed(rc) and rc == MEMCACHED_IN_PROGRESS)
816 {
817 memcached_quit_server(ptr, true);
818 return memcached_set_error(*ptr, rc, MEMCACHED_AT);
819 }
820 else if (memcached_failed(rc))
821 {
822 return rc;
823 }
824
825 if (*buffer_ptr == '\n')
826 {
827 line_complete= true;
828 }
829
830 ++buffer_ptr;
831 ++total_nr;
832 }
833
834 /* Now let's look in the buffer and copy as we go! */
835 while (ptr->read_buffer_length and total_nr < size and line_complete == false)
836 {
837 *buffer_ptr = *ptr->read_ptr;
838 if (*buffer_ptr == '\n')
839 {
840 line_complete = true;
841 }
842 --ptr->read_buffer_length;
843 ++ptr->read_ptr;
844 ++total_nr;
845 ++buffer_ptr;
846 }
847
848 if (total_nr == size)
849 {
850 return MEMCACHED_PROTOCOL_ERROR;
851 }
852 }
853
854 return MEMCACHED_SUCCESS;
855 }