« back to all changes in this revision
Viewing changes to plugin/mysql_protocol/viosocket.cc
- browse files at revision 1218
- compare with another revision
- compare with revision 2422
- download diff
- download tarball
- view history from revision 1218
- Committer: Brian Aker
- Date: 2009-11-13 18:25:26 UTC
- mfrom: (971.7.11 eday-dev)
- Revision ID: brian@gaz-20091113182526-evaib6rmmndruddq
Merge Eric
- files added:
- plugin/mysql_protocol
- files renamed:
- plugin/oldlibdrizzle/ => plugin/drizzle_protocol/
- files modified:
- drizzled/drizzled.cc
added
removed
plugin/mysql_protocol/viosocket.cc
1
/* Copyright (C) 2000 MySQL AB
2
3
This program is free software; you can redistribute it and/or modify
4
it under the terms of the GNU General Public License as published by
5
the Free Software Foundation; version 2 of the License.
6
7
This program is distributed in the hope that it will be useful,
8
but WITHOUT ANY WARRANTY; without even the implied warranty of
9
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10
GNU General Public License for more details.
11
12
You should have received a copy of the GNU General Public License
13
along with this program; if not, write to the Free Software
14
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
15
16
/*
17
Note that we can't have assertion on file descriptors; The reason for
18
this is that during mysql shutdown, another thread can close a file
19
we are working on. In this case we should just return read errors from
20
the file descriptior.
21
*/
22
23
#define DONT_MAP_VIO
24
#include <drizzled/global.h>
25
#include "vio.h"
26
#include <drizzled/util/test.h>
27
28
#include <sys/socket.h>
29
#include <string.h>
30
31
#include <netinet/tcp.h>
32
#include <netinet/in.h>
33
#include <sys/poll.h>
34
35
#include <netdb.h>
36
37
#include <algorithm>
38
39
using namespace std;
40
41
int drizzleclient_vio_errno(Vio *vio)
42
{
43
(void)vio;
44
return errno;
45
}
46
47
48
size_t drizzleclient_vio_read(Vio * vio, unsigned char* buf, size_t size)
49
{
50
size_t r;
51
52
/* Ensure nobody uses drizzleclient_vio_read_buff and drizzleclient_vio_read simultaneously */
53
assert(vio->read_end == vio->read_pos);
54
r= read(vio->sd, buf, size);
55
56
return r;
57
}
58
59
60
/*
61
Buffered read: if average read size is small it may
62
reduce number of syscalls.
63
*/
64
65
size_t drizzleclient_vio_read_buff(Vio *vio, unsigned char* buf, size_t size)
66
{
67
size_t rc;
68
#define VIO_UNBUFFERED_READ_MIN_SIZE 2048
69
70
if (vio->read_pos < vio->read_end)
71
{
72
rc= min((size_t) (vio->read_end - vio->read_pos), size);
73
memcpy(buf, vio->read_pos, rc);
74
vio->read_pos+= rc;
75
/*
76
Do not try to read from the socket now even if rc < size:
77
drizzleclient_vio_read can return -1 due to an error or non-blocking mode, and
78
the safest way to handle it is to move to a separate branch.
79
*/
80
}
81
else if (size < VIO_UNBUFFERED_READ_MIN_SIZE)
82
{
83
rc= drizzleclient_vio_read(vio, (unsigned char*) vio->read_buffer, VIO_READ_BUFFER_SIZE);
84
if (rc != 0 && rc != (size_t) -1)
85
{
86
if (rc > size)
87
{
88
vio->read_pos= vio->read_buffer + size;
89
vio->read_end= vio->read_buffer + rc;
90
rc= size;
91
}
92
memcpy(buf, vio->read_buffer, rc);
93
}
94
}
95
else
96
rc= drizzleclient_vio_read(vio, buf, size);
97
98
return rc;
99
#undef VIO_UNBUFFERED_READ_MIN_SIZE
100
}
101
102
103
size_t drizzleclient_vio_write(Vio * vio, const unsigned char* buf, size_t size)
104
{
105
size_t r;
106
107
r = write(vio->sd, buf, size);
108
109
return r;
110
}
111
112
int drizzleclient_vio_blocking(Vio * vio, bool set_blocking_mode, bool *old_mode)
113
{
114
int r=0;
115
116
*old_mode= test(!(vio->fcntl_mode & O_NONBLOCK));
117
118
if (vio->sd >= 0)
119
{
120
int old_fcntl=vio->fcntl_mode;
121
if (set_blocking_mode)
122
vio->fcntl_mode &= ~O_NONBLOCK; /* clear bit */
123
else
124
vio->fcntl_mode |= O_NONBLOCK; /* set bit */
125
if (old_fcntl != vio->fcntl_mode)
126
{
127
r= fcntl(vio->sd, F_SETFL, vio->fcntl_mode);
128
if (r == -1)
129
{
130
vio->fcntl_mode= old_fcntl;
131
}
132
}
133
}
134
135
return r;
136
}
137
138
bool
139
drizzleclient_vio_is_blocking(Vio * vio)
140
{
141
bool r;
142
r = !(vio->fcntl_mode & O_NONBLOCK);
143
144
return r;
145
}
146
147
148
int drizzleclient_vio_fastsend(Vio * vio)
149
{
150
(void)vio;
151
int nodelay = 1;
152
int error;
153
154
error= setsockopt(vio->sd, IPPROTO_TCP, TCP_NODELAY,
155
&nodelay, sizeof(nodelay));
156
if (error != 0)
157
{
158
perror("setsockopt");
159
assert(error == 0);
160
}
161
162
return error;
163
}
164
165
int32_t drizzleclient_vio_keepalive(Vio* vio, bool set_keep_alive)
166
{
167
int r= 0;
168
uint32_t opt= 0;
169
170
if (set_keep_alive)
171
opt= 1;
172
173
r= setsockopt(vio->sd, SOL_SOCKET, SO_KEEPALIVE, (char *) &opt, sizeof(opt));
174
if (r != 0)
175
{
176
perror("setsockopt");
177
assert(r == 0);
178
}
179
180
return r;
181
}
182
183
184
bool
185
drizzleclient_vio_should_retry(Vio * vio)
186
{
187
(void)vio;
188
int en = errno;
189
return (en == EAGAIN || en == EINTR ||
190
en == EWOULDBLOCK);
191
}
192
193
194
bool
195
drizzleclient_vio_was_interrupted(Vio *vio)
196
{
197
(void)vio;
198
int en= errno;
199
return (en == EAGAIN || en == EINTR ||
200
en == EWOULDBLOCK || en == ETIMEDOUT);
201
}
202
203
204
int drizzleclient_vio_close(Vio * vio)
205
{
206
int r=0;
207
if (vio->type != VIO_CLOSED)
208
{
209
assert(vio->sd >= 0);
210
if (shutdown(vio->sd, SHUT_RDWR))
211
r= -1;
212
if (close(vio->sd))
213
r= -1;
214
}
215
vio->type= VIO_CLOSED;
216
vio->sd= -1;
217
218
return r;
219
}
220
221
222
const char *drizzleclient_vio_description(Vio * vio)
223
{
224
return vio->desc;
225
}
226
227
enum enum_vio_type drizzleclient_vio_type(Vio* vio)
228
{
229
return vio->type;
230
}
231
232
int drizzleclient_vio_fd(Vio* vio)
233
{
234
return vio->sd;
235
}
236
237
bool drizzleclient_vio_peer_addr(Vio *vio, char *buf, uint16_t *port, size_t buflen)
238
{
239
int error;
240
char port_buf[NI_MAXSERV];
241
socklen_t addrLen = sizeof(vio->remote);
242
243
if (getpeername(vio->sd, (struct sockaddr *) (&vio->remote),
244
&addrLen) != 0)
245
{
246
return true;
247
}
248
vio->addrLen= (int)addrLen;
249
250
if ((error= getnameinfo((struct sockaddr *)(&vio->remote),
251
addrLen,
252
buf, buflen,
253
port_buf, NI_MAXSERV, NI_NUMERICHOST|NI_NUMERICSERV)))
254
{
255
return true;
256
}
257
258
*port= (uint16_t)strtol(port_buf, (char **)NULL, 10);
259
260
return false;
261
}
262
263
264
/* Return 0 if there is data to be read */
265
266
bool drizzleclient_vio_poll_read(Vio *vio, int32_t timeout)
267
{
268
struct pollfd fds;
269
int res;
270
271
fds.fd=vio->sd;
272
fds.events=POLLIN;
273
fds.revents=0;
274
if ((res=poll(&fds,1,(int) timeout*1000)) <= 0)
275
{
276
return res < 0 ? false : true; /* Don't return 1 on errors */
277
}
278
return (fds.revents & (POLLIN | POLLERR | POLLHUP) ? false : true);
279
}
280
281
282
bool drizzleclient_vio_peek_read(Vio *vio, uint32_t *bytes)
283
{
284
char buf[1024];
285
ssize_t res= recv(vio->sd, &buf, sizeof(buf), MSG_PEEK);
286
287
if (res < 0)
288
return true;
289
*bytes= (uint32_t)res;
290
return false;
291
}
292
293
void drizzleclient_vio_timeout(Vio *vio, bool is_sndtimeo, int32_t timeout)
294
{
295
int error;
296
297
/* POSIX specifies time as struct timeval. */
298
struct timeval wait_timeout;
299
wait_timeout.tv_sec= timeout;
300
wait_timeout.tv_usec= 0;
301
302
assert(timeout >= 0 && timeout <= INT32_MAX);
303
assert(vio->sd != -1);
304
error= setsockopt(vio->sd, SOL_SOCKET, is_sndtimeo ? SO_SNDTIMEO : SO_RCVTIMEO,
305
&wait_timeout,
306
(socklen_t)sizeof(struct timeval));
307
if (error == -1 && errno != ENOPROTOOPT)
308
{
309
perror("setsockopt");
310
assert(error == 0);
311
}
312
}
Loggerhead is a web-based interface for Breezy
Version: 2.0.1