56239529eb2918a664c0d6971b51c92d1bc796f2
[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 if (fds.revents & POLLHUP)
217 {
218 break;
219 }
220 assert_msg(active_fd == 1 , "poll() returned an unexpected value");
221 return MEMCACHED_SUCCESS;
222 }
223 else if (active_fd == 0)
224 {
225 ptr->io_wait_count.timeouts++;
226 return memcached_set_error(*ptr, MEMCACHED_TIMEOUT, MEMCACHED_AT);
227 }
228
229 // Only an error should result in this code being called.
230 local_errno= get_socket_errno(); // We cache in case memcached_quit_server() modifies errno
231 assert_msg(active_fd == -1 , "poll() returned an unexpected value");
232 switch (local_errno)
233 {
234 #ifdef TARGET_OS_LINUX
235 case ERESTART:
236 #endif
237 case EINTR:
238 continue;
239
240 case EFAULT:
241 case ENOMEM:
242 return memcached_set_error(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT);
243
244 case EINVAL:
245 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"));
246
247 default:
248 if (fds.revents & POLLERR)
249 {
250 int err;
251 socklen_t len= sizeof (err);
252 if (getsockopt(ptr->fd, SOL_SOCKET, SO_ERROR, &err, &len) == 0)
253 {
254 if (err == 0) // treat this as EINTR
255 {
256 continue;
257 }
258 local_errno= err;
259 }
260 }
261 break;
262 }
263
264 break; // should only occur from poll error
265 }
266
267 memcached_quit_server(ptr, true);
268
269 return memcached_set_errno(*ptr, local_errno, MEMCACHED_AT);
270 }
271
272 static bool io_flush(memcached_server_write_instance_st ptr,
273 const bool with_flush,
274 memcached_return_t& error)
275 {
276 /*
277 ** We might want to purge the input buffer if we haven't consumed
278 ** any output yet... The test for the limits is the purge is inline
279 ** in the purge function to avoid duplicating the logic..
280 */
281 {
282 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
283
284 if (memcached_purge(ptr) == false)
285 {
286 return false;
287 }
288 }
289 char *local_write_ptr= ptr->write_buffer;
290 size_t write_length= ptr->write_buffer_offset;
291
292 error= MEMCACHED_SUCCESS;
293
294 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
295
296 /* Looking for memory overflows */
297 #if defined(DEBUG)
298 if (write_length == MEMCACHED_MAX_BUFFER)
299 WATCHPOINT_ASSERT(ptr->write_buffer == local_write_ptr);
300 WATCHPOINT_ASSERT((ptr->write_buffer + MEMCACHED_MAX_BUFFER) >= (local_write_ptr + write_length));
301 #endif
302
303 while (write_length)
304 {
305 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
306 WATCHPOINT_ASSERT(write_length > 0);
307
308 int flags= with_flush ? MSG_NOSIGNAL|MSG_DONTWAIT : MSG_NOSIGNAL|MSG_DONTWAIT|MSG_MORE;
309 ssize_t sent_length= ::send(ptr->fd, local_write_ptr, write_length, flags);
310
311 if (sent_length == SOCKET_ERROR)
312 {
313 #if 0 // @todo I should look at why we hit this bit of code hard frequently
314 WATCHPOINT_ERRNO(get_socket_errno());
315 WATCHPOINT_NUMBER(get_socket_errno());
316 #endif
317 switch (get_socket_errno())
318 {
319 case ENOBUFS:
320 continue;
321
322 #if EWOULDBLOCK != EAGAIN
323 case EWOULDBLOCK:
324 #endif
325 case EAGAIN:
326 {
327 /*
328 * We may be blocked on write because the input buffer
329 * is full. Let's check if we have room in our input
330 * buffer for more data and retry the write before
331 * waiting..
332 */
333 if (repack_input_buffer(ptr) or process_input_buffer(ptr))
334 {
335 continue;
336 }
337
338 memcached_return_t rc= io_wait(ptr, MEM_WRITE);
339 if (memcached_success(rc))
340 {
341 continue;
342 }
343 else if (rc == MEMCACHED_TIMEOUT)
344 {
345 return false;
346 }
347
348 memcached_quit_server(ptr, true);
349 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
350 return false;
351 }
352 case ENOTCONN:
353 case EPIPE:
354 default:
355 memcached_quit_server(ptr, true);
356 error= memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
357 WATCHPOINT_ASSERT(ptr->fd == INVALID_SOCKET);
358 return false;
359 }
360 }
361
362 ptr->io_bytes_sent+= uint32_t(sent_length);
363
364 local_write_ptr+= sent_length;
365 write_length-= uint32_t(sent_length);
366 }
367
368 WATCHPOINT_ASSERT(write_length == 0);
369 ptr->write_buffer_offset= 0;
370
371 return true;
372 }
373
374 memcached_return_t memcached_io_wait_for_write(memcached_server_write_instance_st ptr)
375 {
376 return io_wait(ptr, MEM_WRITE);
377 }
378
379 static memcached_return_t _io_fill(memcached_server_write_instance_st ptr)
380 {
381 ssize_t data_read;
382 do
383 {
384 data_read= ::recv(ptr->fd, ptr->read_buffer, MEMCACHED_MAX_BUFFER, MSG_DONTWAIT);
385 if (data_read == SOCKET_ERROR)
386 {
387 switch (get_socket_errno())
388 {
389 case EINTR: // We just retry
390 continue;
391
392 case ETIMEDOUT: // OSX
393 #if EWOULDBLOCK != EAGAIN
394 case EWOULDBLOCK:
395 #endif
396 case EAGAIN:
397 #ifdef TARGET_OS_LINUX
398 case ERESTART:
399 #endif
400 {
401 memcached_return_t io_wait_ret;
402 if (memcached_success(io_wait_ret= io_wait(ptr, MEM_READ)))
403 {
404 continue;
405 }
406
407 return io_wait_ret;
408 }
409
410 /* fall through */
411
412 case ENOTCONN: // Programmer Error
413 WATCHPOINT_ASSERT(0);
414 case ENOTSOCK:
415 WATCHPOINT_ASSERT(0);
416 case EBADF:
417 assert_msg(ptr->fd != INVALID_SOCKET, "Programmer error, invalid socket");
418 case EINVAL:
419 case EFAULT:
420 case ECONNREFUSED:
421 default:
422 memcached_quit_server(ptr, true);
423 memcached_set_errno(*ptr, get_socket_errno(), MEMCACHED_AT);
424 break;
425 }
426
427 return memcached_server_error_return(ptr);
428 }
429 else if (data_read == 0)
430 {
431 /*
432 EOF. Any data received so far is incomplete
433 so discard it. This always reads by byte in case of TCP
434 and protocol enforcement happens at memcached_response()
435 looking for '\n'. We do not care for UDB which requests 8 bytes
436 at once. Generally, this means that connection went away. Since
437 for blocking I/O we do not return 0 and for non-blocking case
438 it will return EGAIN if data is not immediatly available.
439 */
440 memcached_quit_server(ptr, true);
441 return memcached_set_error(*ptr, MEMCACHED_CONNECTION_FAILURE, MEMCACHED_AT,
442 memcached_literal_param("::rec() returned zero, server has disconnected"));
443 }
444 ptr->io_wait_count._bytes_read+= data_read;
445 } while (data_read <= 0);
446
447 ptr->io_bytes_sent= 0;
448 ptr->read_data_length= (size_t) data_read;
449 ptr->read_buffer_length= (size_t) data_read;
450 ptr->read_ptr= ptr->read_buffer;
451
452 return MEMCACHED_SUCCESS;
453 }
454
455 memcached_return_t memcached_io_read(memcached_server_write_instance_st ptr,
456 void *buffer, size_t length, ssize_t& nread)
457 {
458 assert(memcached_is_udp(ptr->root) == false);
459 assert_msg(ptr, "Programmer error, memcached_io_read() recieved an invalid memcached_server_write_instance_st"); // Programmer error
460 char *buffer_ptr= static_cast<char *>(buffer);
461
462 if (ptr->fd == INVALID_SOCKET)
463 {
464 #if 0
465 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Programmer error, invalid socket state");
466 #endif
467 return MEMCACHED_CONNECTION_FAILURE;
468 }
469
470 while (length)
471 {
472 if (ptr->read_buffer_length == 0)
473 {
474 memcached_return_t io_fill_ret;
475 if (memcached_fatal(io_fill_ret= _io_fill(ptr)))
476 {
477 nread= -1;
478 return io_fill_ret;
479 }
480 }
481
482 if (length > 1)
483 {
484 size_t difference= (length > ptr->read_buffer_length) ? ptr->read_buffer_length : length;
485
486 memcpy(buffer_ptr, ptr->read_ptr, difference);
487 length -= difference;
488 ptr->read_ptr+= difference;
489 ptr->read_buffer_length-= difference;
490 buffer_ptr+= difference;
491 }
492 else
493 {
494 *buffer_ptr= *ptr->read_ptr;
495 ptr->read_ptr++;
496 ptr->read_buffer_length--;
497 buffer_ptr++;
498 break;
499 }
500 }
501
502 nread= ssize_t(buffer_ptr - (char*)buffer);
503
504 return MEMCACHED_SUCCESS;
505 }
506
507 memcached_return_t memcached_io_slurp(memcached_server_write_instance_st ptr)
508 {
509 assert_msg(ptr, "Programmer error, invalid memcached_server_write_instance_st");
510 assert(memcached_is_udp(ptr->root) == false);
511
512 if (ptr->fd == INVALID_SOCKET)
513 {
514 assert_msg(int(ptr->state) <= int(MEMCACHED_SERVER_STATE_ADDRINFO), "Invalid socket state");
515 return MEMCACHED_CONNECTION_FAILURE;
516 }
517
518 ssize_t data_read;
519 char buffer[MEMCACHED_MAX_BUFFER];
520 do
521 {
522 data_read= recv(ptr->fd, ptr->read_buffer, sizeof(buffer), MSG_DONTWAIT);
523 if (data_read == SOCKET_ERROR)
524 {
525 switch (get_socket_errno())
526 {
527 case EINTR: // We just retry
528 continue;
529
530 case ETIMEDOUT: // OSX
531 #if EWOULDBLOCK != EAGAIN
532 case EWOULDBLOCK:
533 #endif
534 case EAGAIN:
535 #ifdef TARGET_OS_LINUX
536 case ERESTART:
537 #endif
538 if (memcached_success(io_wait(ptr, MEM_READ)))
539 {
540 continue;
541 }
542 return MEMCACHED_IN_PROGRESS;
543
544 /* fall through */
545
546 case ENOTCONN: // Programmer Error
547 assert(0);
548 case ENOTSOCK:
549 assert(0);
550 case EBADF:
551 assert_msg(ptr->fd != INVALID_SOCKET, "Invalid socket state");
552 case EINVAL:
553 case EFAULT:
554 case ECONNREFUSED:
555 default:
556 return MEMCACHED_CONNECTION_FAILURE; // We want this!
557 }
558 }
559 } while (data_read > 0);
560
561 return MEMCACHED_CONNECTION_FAILURE;
562 }
563
564 static bool _io_write(memcached_server_write_instance_st ptr,
565 const void *buffer, size_t length, bool with_flush,
566 size_t& written)
567 {
568 assert(ptr->fd != INVALID_SOCKET);
569 assert(memcached_is_udp(ptr->root) == false);
570
571 const char *buffer_ptr= static_cast<const char *>(buffer);
572
573 const size_t original_length= length;
574
575 while (length)
576 {
577 char *write_ptr;
578 size_t buffer_end= MEMCACHED_MAX_BUFFER;
579 size_t should_write= buffer_end -ptr->write_buffer_offset;
580 should_write= (should_write < length) ? should_write : length;
581
582 write_ptr= ptr->write_buffer + ptr->write_buffer_offset;
583 memcpy(write_ptr, buffer_ptr, should_write);
584 ptr->write_buffer_offset+= should_write;
585 buffer_ptr+= should_write;
586 length-= should_write;
587
588 if (ptr->write_buffer_offset == buffer_end)
589 {
590 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
591
592 memcached_return_t rc;
593 if (io_flush(ptr, with_flush, rc) == false)
594 {
595 written= original_length -length;
596 return false;
597 }
598 }
599 }
600
601 if (with_flush)
602 {
603 memcached_return_t rc;
604 WATCHPOINT_ASSERT(ptr->fd != INVALID_SOCKET);
605 if (io_flush(ptr, with_flush, rc) == false)
606 {
607 written= original_length -length;
608 return false;
609 }
610 }
611
612 written= original_length -length;
613
614 return true;
615 }
616
617 bool memcached_io_write(memcached_server_write_instance_st ptr)
618 {
619 size_t written;
620 return _io_write(ptr, NULL, 0, true, written);
621 }
622
623 ssize_t memcached_io_write(memcached_server_write_instance_st ptr,
624 const void *buffer, const size_t length, const bool with_flush)
625 {
626 size_t written;
627
628 if (_io_write(ptr, buffer, length, with_flush, written) == false)
629 {
630 return -1;
631 }
632
633 return ssize_t(written);
634 }
635
636 bool memcached_io_writev(memcached_server_write_instance_st ptr,
637 libmemcached_io_vector_st vector[],
638 const size_t number_of, const bool with_flush)
639 {
640 ssize_t complete_total= 0;
641 ssize_t total= 0;
642
643 for (size_t x= 0; x < number_of; x++, vector++)
644 {
645 complete_total+= vector->length;
646 if (vector->length)
647 {
648 size_t written;
649 if ((_io_write(ptr, vector->buffer, vector->length, false, written)) == false)
650 {
651 return false;
652 }
653 total+= written;
654 }
655 }
656
657 if (with_flush)
658 {
659 if (memcached_io_write(ptr) == false)
660 {
661 return false;
662 }
663 }
664
665 return (complete_total == total);
666 }
667
668
669 void memcached_io_close(memcached_server_write_instance_st ptr)
670 {
671 if (ptr->fd == INVALID_SOCKET)
672 {
673 return;
674 }
675
676 /* in case of death shutdown to avoid blocking at close() */
677 if (shutdown(ptr->fd, SHUT_RDWR) == SOCKET_ERROR && get_socket_errno() != ENOTCONN)
678 {
679 WATCHPOINT_NUMBER(ptr->fd);
680 WATCHPOINT_ERRNO(get_socket_errno());
681 WATCHPOINT_ASSERT(get_socket_errno());
682 }
683
684 if (closesocket(ptr->fd) == SOCKET_ERROR)
685 {
686 WATCHPOINT_ERRNO(get_socket_errno());
687 }
688 ptr->state= MEMCACHED_SERVER_STATE_NEW;
689 ptr->fd= INVALID_SOCKET;
690 }
691
692 memcached_server_write_instance_st memcached_io_get_readable_server(memcached_st *memc)
693 {
694 #define MAX_SERVERS_TO_POLL 100
695 struct pollfd fds[MAX_SERVERS_TO_POLL];
696 nfds_t host_index= 0;
697
698 for (uint32_t x= 0; x < memcached_server_count(memc) and host_index < MAX_SERVERS_TO_POLL; ++x)
699 {
700 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, x);
701
702 if (instance->read_buffer_length > 0) /* I have data in the buffer */
703 {
704 return instance;
705 }
706
707 if (memcached_server_response_count(instance) > 0)
708 {
709 fds[host_index].events = POLLIN;
710 fds[host_index].revents = 0;
711 fds[host_index].fd = instance->fd;
712 ++host_index;
713 }
714 }
715
716 if (host_index < 2)
717 {
718 /* We have 0 or 1 server with pending events.. */
719 for (uint32_t x= 0; x< memcached_server_count(memc); ++x)
720 {
721 memcached_server_write_instance_st instance=
722 memcached_server_instance_fetch(memc, x);
723
724 if (memcached_server_response_count(instance) > 0)
725 {
726 return instance;
727 }
728 }
729
730 return NULL;
731 }
732
733 int error= poll(fds, host_index, memc->poll_timeout);
734 switch (error)
735 {
736 case -1:
737 memcached_set_errno(*memc, get_socket_errno(), MEMCACHED_AT);
738 /* FALLTHROUGH */
739 case 0:
740 break;
741
742 default:
743 for (nfds_t x= 0; x < host_index; ++x)
744 {
745 if (fds[x].revents & POLLIN)
746 {
747 for (uint32_t y= 0; y < memcached_server_count(memc); ++y)
748 {
749 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, y);
750
751 if (instance->fd == fds[x].fd)
752 {
753 return instance;
754 }
755 }
756 }
757 }
758 }
759
760 return NULL;
761 }
762
763 /*
764 Eventually we will just kill off the server with the problem.
765 */
766 void memcached_io_reset(memcached_server_write_instance_st ptr)
767 {
768 memcached_quit_server(ptr, true);
769 }
770
771 /**
772 * Read a given number of bytes from the server and place it into a specific
773 * buffer. Reset the IO channel on this server if an error occurs.
774 */
775 memcached_return_t memcached_safe_read(memcached_server_write_instance_st ptr,
776 void *dta,
777 const size_t size)
778 {
779 size_t offset= 0;
780 char *data= static_cast<char *>(dta);
781
782 while (offset < size)
783 {
784 ssize_t nread;
785 memcached_return_t rc;
786
787 while (memcached_continue(rc= memcached_io_read(ptr, data + offset, size - offset, nread))) { };
788
789 if (memcached_failed(rc))
790 {
791 return rc;
792 }
793
794 offset+= size_t(nread);
795 }
796
797 return MEMCACHED_SUCCESS;
798 }
799
800 memcached_return_t memcached_io_readline(memcached_server_write_instance_st ptr,
801 char *buffer_ptr,
802 size_t size,
803 size_t& total_nr)
804 {
805 total_nr= 0;
806 bool line_complete= false;
807
808 while (line_complete == false)
809 {
810 if (ptr->read_buffer_length == 0)
811 {
812 /*
813 * We don't have any data in the buffer, so let's fill the read
814 * buffer. Call the standard read function to avoid duplicating
815 * the logic.
816 */
817 ssize_t nread;
818 memcached_return_t rc= memcached_io_read(ptr, buffer_ptr, 1, nread);
819 if (memcached_failed(rc) and rc == MEMCACHED_IN_PROGRESS)
820 {
821 memcached_quit_server(ptr, true);
822 return memcached_set_error(*ptr, rc, MEMCACHED_AT);
823 }
824 else if (memcached_failed(rc))
825 {
826 return rc;
827 }
828
829 if (*buffer_ptr == '\n')
830 {
831 line_complete= true;
832 }
833
834 ++buffer_ptr;
835 ++total_nr;
836 }
837
838 /* Now let's look in the buffer and copy as we go! */
839 while (ptr->read_buffer_length and total_nr < size and line_complete == false)
840 {
841 *buffer_ptr = *ptr->read_ptr;
842 if (*buffer_ptr == '\n')
843 {
844 line_complete = true;
845 }
846 --ptr->read_buffer_length;
847 ++ptr->read_ptr;
848 ++total_nr;
849 ++buffer_ptr;
850 }
851
852 if (total_nr == size)
853 {
854 return MEMCACHED_PROTOCOL_ERROR;
855 }
856 }
857
858 return MEMCACHED_SUCCESS;
859 }