← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to libdrizzle/viosocket.c

  • Committer: Daniel Nichter
  • Date: 2011-10-23 16:01:37 UTC
  • mto: This revision was merged to the branch mainline in revision 2448.
  • Revision ID: daniel@percona.com-20111023160137-7ac3blgz8z4tf8za
Add Administration Getting Started and Logging.  Capitalize SQL clause keywords.

Show diffs side-by-side

added added

removed removed

Lines of Context:
libdrizzle/viosocket.c
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 <libdrizzle/vio.h>
26
 
#include <sys/socket.h>
27
 
#include <drizzled/util/test.h>
28
 
#include <string.h>
29
 
 
30
 
#include <netinet/tcp.h>
31
 
#include <sys/poll.h>
32
 
 
33
 
#include <netdb.h>
34
 
 
35
 
int vio_errno(Vio *vio __attribute__((unused)))
36
 
{
37
 
  return errno;
38
 
}
39
 
 
40
 
 
41
 
size_t vio_read(Vio * vio, unsigned char* buf, size_t size)
42
 
{
43
 
  size_t r;
44
 
 
45
 
  /* Ensure nobody uses vio_read_buff and vio_read simultaneously */
46
 
  assert(vio->read_end == vio->read_pos);
47
 
  r= read(vio->sd, buf, size);
48
 
 
49
 
  return r;
50
 
}
51
 
 
52
 
 
53
 
/*
54
 
  Buffered read: if average read size is small it may
55
 
  reduce number of syscalls.
56
 
*/
57
 
 
58
 
size_t vio_read_buff(Vio *vio, unsigned char* buf, size_t size)
59
 
{
60
 
  size_t rc;
61
 
#define VIO_UNBUFFERED_READ_MIN_SIZE 2048
62
 
 
63
 
  if (vio->read_pos < vio->read_end)
64
 
  {
65
 
    rc= cmin((size_t) (vio->read_end - vio->read_pos), size);
66
 
    memcpy(buf, vio->read_pos, rc);
67
 
    vio->read_pos+= rc;
68
 
    /*
69
 
      Do not try to read from the socket now even if rc < size:
70
 
      vio_read can return -1 due to an error or non-blocking mode, and
71
 
      the safest way to handle it is to move to a separate branch.
72
 
    */
73
 
  }
74
 
  else if (size < VIO_UNBUFFERED_READ_MIN_SIZE)
75
 
  {
76
 
    rc= vio_read(vio, (unsigned char*) vio->read_buffer, VIO_READ_BUFFER_SIZE);
77
 
    if (rc != 0 && rc != (size_t) -1)
78
 
    {
79
 
      if (rc > size)
80
 
      {
81
 
        vio->read_pos= vio->read_buffer + size;
82
 
        vio->read_end= vio->read_buffer + rc;
83
 
        rc= size;
84
 
      }
85
 
      memcpy(buf, vio->read_buffer, rc);
86
 
    }
87
 
  }
88
 
  else
89
 
    rc= vio_read(vio, buf, size);
90
 
 
91
 
  return rc;
92
 
#undef VIO_UNBUFFERED_READ_MIN_SIZE
93
 
}
94
 
 
95
 
 
96
 
size_t vio_write(Vio * vio, const unsigned char* buf, size_t size)
97
 
{
98
 
  size_t r;
99
 
 
100
 
  r = write(vio->sd, buf, size);
101
 
 
102
 
  return r;
103
 
}
104
 
 
105
 
int vio_blocking(Vio * vio, bool set_blocking_mode, bool *old_mode)
106
 
{
107
 
  int r=0;
108
 
 
109
 
  *old_mode= test(!(vio->fcntl_mode & O_NONBLOCK));
110
 
 
111
 
  if (vio->sd >= 0)
112
 
  {
113
 
    int old_fcntl=vio->fcntl_mode;
114
 
    if (set_blocking_mode)
115
 
      vio->fcntl_mode &= ~O_NONBLOCK; /* clear bit */
116
 
    else
117
 
      vio->fcntl_mode |= O_NONBLOCK; /* set bit */
118
 
    if (old_fcntl != vio->fcntl_mode)
119
 
    {
120
 
      r= fcntl(vio->sd, F_SETFL, vio->fcntl_mode);
121
 
      if (r == -1)
122
 
      {
123
 
        vio->fcntl_mode= old_fcntl;
124
 
      }
125
 
    }
126
 
  }
127
 
 
128
 
  return r;
129
 
}
130
 
 
131
 
bool
132
 
vio_is_blocking(Vio * vio)
133
 
{
134
 
  bool r;
135
 
  r = !(vio->fcntl_mode & O_NONBLOCK);
136
 
 
137
 
  return r;
138
 
}
139
 
 
140
 
 
141
 
int vio_fastsend(Vio * vio __attribute__((unused)))
142
 
{
143
 
  int nodelay = 1;
144
 
  int error;
145
 
 
146
 
  error= setsockopt(vio->sd, IPPROTO_TCP, TCP_NODELAY,
147
 
                    &nodelay, sizeof(nodelay));
148
 
  if (error != 0)
149
 
  {
150
 
    perror("setsockopt");
151
 
    assert(error == 0);
152
 
  }
153
 
 
154
 
  return error;
155
 
}
156
 
 
157
 
int32_t vio_keepalive(Vio* vio, bool set_keep_alive)
158
 
{
159
 
  int r= 0;
160
 
  uint32_t opt= 0;
161
 
 
162
 
  if (set_keep_alive)
163
 
    opt= 1;
164
 
 
165
 
  r= setsockopt(vio->sd, SOL_SOCKET, SO_KEEPALIVE, (char *) &opt, sizeof(opt));
166
 
  if (r != 0)
167
 
  {
168
 
    perror("setsockopt");
169
 
    assert(r == 0);
170
 
  }
171
 
 
172
 
  return r;
173
 
}
174
 
 
175
 
 
176
 
bool
177
 
vio_should_retry(Vio * vio __attribute__((unused)))
178
 
{
179
 
  int en = errno;
180
 
  return (en == EAGAIN || en == EINTR ||
181
 
          en == EWOULDBLOCK);
182
 
}
183
 
 
184
 
 
185
 
bool
186
 
vio_was_interrupted(Vio *vio __attribute__((unused)))
187
 
{
188
 
  int en= errno;
189
 
  return (en == EAGAIN || en == EINTR ||
190
 
          en == EWOULDBLOCK || en == ETIMEDOUT);
191
 
}
192
 
 
193
 
 
194
 
int vio_close(Vio * vio)
195
 
{
196
 
  int r=0;
197
 
 if (vio->type != VIO_CLOSED)
198
 
  {
199
 
    assert(vio->sd >= 0);
200
 
    if (shutdown(vio->sd, SHUT_RDWR))
201
 
      r= -1;
202
 
    if (close(vio->sd))
203
 
      r= -1;
204
 
  }
205
 
  vio->type= VIO_CLOSED;
206
 
  vio->sd=   -1;
207
 
 
208
 
  return r;
209
 
}
210
 
 
211
 
 
212
 
const char *vio_description(Vio * vio)
213
 
{
214
 
  return vio->desc;
215
 
}
216
 
 
217
 
enum enum_vio_type vio_type(Vio* vio)
218
 
{
219
 
  return vio->type;
220
 
}
221
 
 
222
 
int vio_fd(Vio* vio)
223
 
{
224
 
  return vio->sd;
225
 
}
226
 
 
227
 
bool vio_peer_addr(Vio *vio, char *buf, uint16_t *port, size_t buflen)
228
 
{
229
 
  int error;
230
 
  char port_buf[NI_MAXSERV];
231
 
  socklen_t addrLen = sizeof(vio->remote);
232
 
 
233
 
  if (getpeername(vio->sd, (struct sockaddr *) (&vio->remote),
234
 
                  &addrLen) != 0)
235
 
  {
236
 
    return true;
237
 
  }
238
 
  vio->addrLen= (int)addrLen;
239
 
 
240
 
  if ((error= getnameinfo((struct sockaddr *)(&vio->remote), 
241
 
                          addrLen,
242
 
                          buf, buflen,
243
 
                          port_buf, NI_MAXSERV, NI_NUMERICHOST|NI_NUMERICSERV)))
244
 
  {
245
 
    return true;
246
 
  }
247
 
 
248
 
  *port= (uint16_t)strtol(port_buf, (char **)NULL, 10);
249
 
 
250
 
  return false;
251
 
}
252
 
 
253
 
 
254
 
/* Return 0 if there is data to be read */
255
 
 
256
 
bool vio_poll_read(Vio *vio, int32_t timeout)
257
 
{
258
 
  struct pollfd fds;
259
 
  int res;
260
 
 
261
 
  fds.fd=vio->sd;
262
 
  fds.events=POLLIN;
263
 
  fds.revents=0;
264
 
  if ((res=poll(&fds,1,(int) timeout*1000)) <= 0)
265
 
  {
266
 
    return res < 0 ? false : true;              /* Don't return 1 on errors */
267
 
  }
268
 
  return (fds.revents & (POLLIN | POLLERR | POLLHUP) ? false : true);
269
 
}
270
 
 
271
 
 
272
 
bool vio_peek_read(Vio *vio, uint32_t *bytes)
273
 
{
274
 
  char buf[1024];
275
 
  ssize_t res= recv(vio->sd, &buf, sizeof(buf), MSG_PEEK);
276
 
 
277
 
  if (res < 0)
278
 
    return true;
279
 
  *bytes= (uint32_t)res;
280
 
  return false;
281
 
}
282
 
 
283
 
void vio_timeout(Vio *vio, bool is_sndtimeo, int32_t timeout)
284
 
{
285
 
  int error;
286
 
 
287
 
  /* POSIX specifies time as struct timeval. */
288
 
  struct timeval wait_timeout;
289
 
  wait_timeout.tv_sec= timeout;
290
 
  wait_timeout.tv_usec= 0;
291
 
 
292
 
  assert(timeout >= 0 && timeout <= INT32_MAX);
293
 
  assert(vio->sd != -1);
294
 
  error= setsockopt(vio->sd, SOL_SOCKET, is_sndtimeo ? SO_SNDTIMEO : SO_RCVTIMEO,
295
 
                    &wait_timeout,
296
 
                    (socklen_t)sizeof(struct timeval));
297
 
  if (error != 0)
298
 
  {
299
 
    perror("setsockopt");
300
 
    assert(error == 0);
301
 
  }
302
 
}