2 Basic socket buffered IO
6 #include "memcached_io.h"
7 #include <sys/select.h>
10 static int io_wait(memcached_st
*ptr
, unsigned int server_key
, unsigned read_or_write
)
16 flags
= POLLOUT
| POLLERR
;
18 flags
= POLLIN
| POLLERR
;
20 memset(&fds
, 0, sizeof(struct pollfd
));
21 fds
[0].fd
= ptr
->hosts
[server_key
].fd
;
24 if (poll(fds
, 1, -1) < 0)
25 return MEMCACHED_FAILURE
;
27 return MEMCACHED_SUCCESS
;
30 ssize_t
memcached_io_read(memcached_st
*ptr
, unsigned int server_key
,
31 char *buffer
, size_t length
)
39 if (!ptr
->hosts
[server_key
].read_buffer_length
)
45 if (ptr
->flags
& MEM_NO_BLOCK
)
49 rc
= io_wait(ptr
, server_key
, 0);
50 if (rc
!= MEMCACHED_SUCCESS
)
54 data_read
= read(ptr
->hosts
[server_key
].fd
,
55 ptr
->hosts
[server_key
].read_buffer
,
56 MEMCACHED_MAX_BUFFER
);
65 ptr
->cached_errno
= errno
;
72 /* If zero, just keep looping */
75 ptr
->hosts
[server_key
].read_buffer_length
= data_read
;
76 ptr
->hosts
[server_key
].read_ptr
= ptr
->hosts
[server_key
].read_buffer
;
79 *buffer_ptr
= *ptr
->hosts
[server_key
].read_ptr
;
81 ptr
->hosts
[server_key
].read_ptr
++;
82 ptr
->hosts
[server_key
].read_buffer_length
--;
86 return (size_t)(buffer_ptr
- buffer
);
89 ssize_t
memcached_io_write(memcached_st
*ptr
, unsigned int server_key
,
90 char *buffer
, size_t length
, char with_flush
)
94 for (x
= 0; x
< length
; x
++)
96 ptr
->hosts
[server_key
].write_buffer
[ptr
->hosts
[server_key
].write_buffer_offset
]= buffer
[x
];
97 ptr
->hosts
[server_key
].write_buffer_offset
++;
99 if (ptr
->hosts
[server_key
].write_buffer_offset
== MEMCACHED_MAX_BUFFER
)
103 sent_length
= memcached_io_flush(ptr
, server_key
);
105 WATCHPOINT_ASSERT(sent_length
== MEMCACHED_MAX_BUFFER
);
106 ptr
->hosts
[server_key
].write_buffer_offset
= 0;
112 if (memcached_io_flush(ptr
, server_key
) == -1)
119 ssize_t
memcached_io_flush(memcached_st
*ptr
, unsigned int server_key
)
122 size_t return_length
;
123 char *write_ptr
= ptr
->hosts
[server_key
].write_buffer
;
124 size_t write_length
= ptr
->hosts
[server_key
].write_buffer_offset
;
125 unsigned int loop
= 1;
127 if (ptr
->hosts
[server_key
].write_buffer_offset
== 0)
133 if (ptr
->flags
& MEM_NO_BLOCK
)
137 rc
= io_wait(ptr
, server_key
, 1);
138 if (rc
!= MEMCACHED_SUCCESS
)
143 if ((ssize_t
)(sent_length
= write(ptr
->hosts
[server_key
].fd
, write_ptr
,
144 write_length
)) == -1)
157 /* Yes, we want to fall through */
159 ptr
->cached_errno
= errno
;
164 write_ptr
+= sent_length
;
165 write_length
-= sent_length
;
166 return_length
+= sent_length
;
169 WATCHPOINT_ASSERT(write_length
== 0);
170 WATCHPOINT_ASSERT(return_length
== ptr
->hosts
[server_key
].write_buffer_offset
);
171 ptr
->hosts
[server_key
].write_buffer_offset
= 0;
173 return return_length
;
177 Eventually we will just kill off the server with the problem.
179 void memcached_io_reset(memcached_st
*ptr
, unsigned int server_key
)
181 ptr
->hosts
[server_key
].write_buffer_offset
= 0;