~clint-fewbar/ubuntu/precise/gearmand/drop-unneeded-patches

« back to all changes in this revision

Viewing changes to libgearman/connection.cc

  • Committer: Bazaar Package Importer
  • Author(s): Monty Taylor
  • Date: 2009-09-28 21:43:31 UTC
  • mto: (1.2.3 upstream) (6.1.1 sid)
  • mto: This revision was merged to the branch mainline in revision 5.
  • Revision ID: james.westby@ubuntu.com-20090928214331-9bku0d3v1b1ypgp4
ImportĀ upstreamĀ versionĀ 0.10

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2
 
 * 
3
 
 *  Gearmand client and server library.
4
 
 *
5
 
 *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
6
 
 *  Copyright (C) 2008 Brian Aker, Eric Day
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
 
 * @file
41
 
 * @brief Connection Definitions
42
 
 */
43
 
 
44
 
#include <libgearman/common.h>
45
 
 
46
 
#include <cassert>
47
 
#include <cerrno>
48
 
#include <cstdio>
49
 
#include <cstdlib>
50
 
#include <cstring>
51
 
#include <memory>
52
 
#include <unistd.h>
53
 
 
54
 
#if HAVE_NETINET_TCP_H
55
 
#include <netinet/tcp.h>    /* for TCP_NODELAY */
56
 
#endif
57
 
#ifdef HAVE_FCNTL_H
58
 
#include <fcntl.h>
59
 
#endif
60
 
 
61
 
static gearman_return_t gearman_connection_set_option(gearman_connection_st *connection,
62
 
                                                      gearman_connection_options_t options,
63
 
                                                      bool value);
64
 
 
65
 
 
66
 
 
67
 
gearman_return_t gearman_connection_st::connect_poll()
68
 
{
69
 
  struct pollfd fds[1];
70
 
  fds[0].fd= fd;
71
 
  fds[0].events= POLLOUT;
72
 
 
73
 
  size_t loop_max= 5;
74
 
 
75
 
#if 0
76
 
  if (universal.timeout == 0)
77
 
  {
78
 
    return gearman_error(universal, GEARMAN_TIMEOUT, "not connected");
79
 
  }
80
 
#endif
81
 
 
82
 
  while (--loop_max) // Should only loop on cases of ERESTART or EINTR
83
 
  {
84
 
    int error= poll(fds, 1, GEARMAN_DEFAULT_CONNECT_TIMEOUT);
85
 
    switch (error)
86
 
    {
87
 
    case 1:
88
 
      {
89
 
        int err;
90
 
        socklen_t len= sizeof (err);
91
 
        // We replace errno with err if getsockopt() passes, but err has been
92
 
        // set.
93
 
        if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &len) == 0)
94
 
        {
95
 
          // We check the value to see what happened wth the socket.
96
 
          if (err == 0)
97
 
          {
98
 
            return GEARMAN_SUCCESS;
99
 
          }
100
 
          errno= err;
101
 
        }
102
 
 
103
 
        return gearman_perror(universal, "getsockopt() failed");
104
 
      }
105
 
 
106
 
    case 0:
107
 
      {
108
 
        return gearman_error(universal, GEARMAN_TIMEOUT, "timeout occurred while trying to connect");
109
 
      }
110
 
 
111
 
    default: // A real error occurred and we need to completely bail
112
 
      switch (get_socket_errno())
113
 
      {
114
 
#ifdef TARGET_OS_LINUX
115
 
      case ERESTART:
116
 
#endif
117
 
      case EINTR:
118
 
        continue;
119
 
 
120
 
      case EFAULT:
121
 
      case ENOMEM:
122
 
        return gearman_perror(universal, "poll() failure");
123
 
 
124
 
      case EINVAL:
125
 
        return gearman_perror(universal, "RLIMIT_NOFILE exceeded, or if OSX the timeout value was invalid");
126
 
 
127
 
      default: // This should not happen
128
 
        if (fds[0].revents & POLLERR)
129
 
        {
130
 
          int err;
131
 
          socklen_t len= sizeof (err);
132
 
          (void)getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &len);
133
 
          errno= err;
134
 
        }
135
 
        else
136
 
        {
137
 
          errno= get_socket_errno();
138
 
        }
139
 
 
140
 
        assert_msg(fd != INVALID_SOCKET, "poll() was passed an invalid file descriptor");
141
 
 
142
 
        return gearman_perror(universal, "socket error occurred");
143
 
      }
144
 
    }
145
 
  }
146
 
 
147
 
  // This should only be possible from ERESTART or EINTR;
148
 
  return gearman_perror(universal, "connection failed (error should be from either ERESTART or EINTR");
149
 
}
150
 
 
151
 
/**
152
 
 * @addtogroup gearman_connection_static Static Connection Declarations
153
 
 * @ingroup gearman_connection
154
 
 * @{
155
 
 */
156
 
 
157
 
gearman_connection_st::gearman_connection_st(gearman_universal_st &universal_arg,
158
 
                                             gearman_connection_options_t *options_args) :
159
 
  state(GEARMAN_CON_UNIVERSAL_ADDRINFO),
160
 
  send_state(GEARMAN_CON_SEND_STATE_NONE),
161
 
  recv_state(GEARMAN_CON_RECV_UNIVERSAL_NONE),
162
 
  port(0),
163
 
  events(0),
164
 
  revents(0),
165
 
  fd(-1),
166
 
  cached_errno(0),
167
 
  created_id(0),
168
 
  created_id_next(0),
169
 
  send_buffer_size(0),
170
 
  send_data_size(0),
171
 
  send_data_offset(0),
172
 
  recv_buffer_size(0),
173
 
  recv_data_size(0),
174
 
  recv_data_offset(0),
175
 
  universal(universal_arg)
176
 
{
177
 
  options.ready= false;
178
 
  options.packet_in_use= false;
179
 
 
180
 
  if (options_args)
181
 
  {
182
 
    while (*options_args != GEARMAN_CON_MAX)
183
 
    {
184
 
      gearman_connection_set_option(this, *options_args, true);
185
 
      options_args++;
186
 
    }
187
 
  }
188
 
 
189
 
  if (universal.con_list)
190
 
  {
191
 
    universal.con_list->prev= this;
192
 
  }
193
 
  next= universal.con_list;
194
 
  prev= NULL;
195
 
  universal.con_list= this;
196
 
  universal.con_count++;
197
 
 
198
 
  context= NULL;
199
 
  addrinfo= NULL;
200
 
  addrinfo_next= NULL;
201
 
  send_buffer_ptr= send_buffer;
202
 
  recv_packet= NULL;
203
 
  recv_buffer_ptr= recv_buffer;
204
 
  host[0]= 0;
205
 
}
206
 
 
207
 
gearman_connection_st *gearman_connection_create(gearman_universal_st &universal,
208
 
                                                 gearman_connection_options_t *options)
209
 
{
210
 
  gearman_connection_st *connection= new (std::nothrow) gearman_connection_st(universal, options);
211
 
  if (connection == NULL)
212
 
  {
213
 
    gearman_perror(universal, "gearman_connection_st new");
214
 
    return NULL;
215
 
  }
216
 
 
217
 
  return connection;
218
 
}
219
 
 
220
 
gearman_connection_st *gearman_connection_create_args(gearman_universal_st& universal,
221
 
                                                      const char *host, in_port_t port)
222
 
{
223
 
  gearman_connection_st *connection= gearman_connection_create(universal, NULL);
224
 
  if (connection == NULL)
225
 
  {
226
 
    return NULL;
227
 
  }
228
 
 
229
 
  connection->set_host(host, port);
230
 
 
231
 
  if (gearman_failed(connection->lookup()))
232
 
  {
233
 
    delete connection;
234
 
    return NULL;
235
 
  }
236
 
 
237
 
  return connection;
238
 
}
239
 
 
240
 
gearman_connection_st *gearman_connection_copy(gearman_universal_st& universal,
241
 
                                               const gearman_connection_st& from)
242
 
{
243
 
  gearman_connection_st *connection= gearman_connection_create(universal, NULL);
244
 
 
245
 
  if (connection == NULL)
246
 
  {
247
 
    return NULL;
248
 
  }
249
 
 
250
 
  connection->options.ready= from.options.ready;
251
 
  // @todo Is this right?
252
 
  connection->options.packet_in_use= from.options.packet_in_use;
253
 
 
254
 
  strcpy(connection->host, from.host);
255
 
  connection->port= from.port;
256
 
 
257
 
  return connection;
258
 
}
259
 
 
260
 
gearman_connection_st::~gearman_connection_st()
261
 
{
262
 
  if (fd != INVALID_SOCKET)
263
 
  {
264
 
    close_socket();
265
 
  }
266
 
 
267
 
  reset_addrinfo();
268
 
 
269
 
  { // Remove from universal list
270
 
    if (universal.con_list == this)
271
 
    {
272
 
      universal.con_list= next;
273
 
    }
274
 
 
275
 
    if (prev)
276
 
    {
277
 
      prev->next= next;
278
 
    }
279
 
 
280
 
    if (next)
281
 
    {
282
 
      next->prev= prev;
283
 
    }
284
 
 
285
 
    universal.con_count--;
286
 
  }
287
 
 
288
 
  free_private_packet();
289
 
}
290
 
 
291
 
void gearman_connection_st::free_private_packet()
292
 
{
293
 
  if (options.packet_in_use)
294
 
  {
295
 
    gearman_packet_free(&_packet);
296
 
    options.packet_in_use= false;
297
 
  }
298
 
}
299
 
 
300
 
gearman_return_t gearman_connection_set_option(gearman_connection_st *connection,
301
 
                                               gearman_connection_options_t options,
302
 
                                               bool value)
303
 
{
304
 
  switch (options)
305
 
  {
306
 
  case GEARMAN_CON_READY:
307
 
    connection->options.ready= value;
308
 
    break;
309
 
 
310
 
  case GEARMAN_CON_PACKET_IN_USE:
311
 
    connection->options.packet_in_use= value;
312
 
    break;
313
 
 
314
 
  case GEARMAN_CON_IGNORE_LOST_CONNECTION:
315
 
    break;
316
 
 
317
 
  case GEARMAN_CON_CLOSE_AFTER_FLUSH:
318
 
    break;
319
 
 
320
 
  case GEARMAN_CON_EXTERNAL_FD:
321
 
  case GEARMAN_CON_MAX:
322
 
  default:
323
 
    return GEARMAN_INVALID_COMMAND;
324
 
  }
325
 
 
326
 
  return GEARMAN_SUCCESS;
327
 
}
328
 
 
329
 
/**
330
 
 * Set socket options for a connection.
331
 
 */
332
 
static gearman_return_t _con_setsockopt(gearman_connection_st *connection);
333
 
 
334
 
/** @} */
335
 
 
336
 
/*
337
 
 * Public Definitions
338
 
 */
339
 
 
340
 
void gearman_connection_st::set_host(const char *host_arg, const in_port_t port_arg)
341
 
{
342
 
  reset_addrinfo();
343
 
 
344
 
  strncpy(host, host_arg == NULL ? GEARMAN_DEFAULT_TCP_HOST : host_arg, GEARMAN_NI_MAXHOST);
345
 
  host[GEARMAN_NI_MAXHOST - 1]= 0;
346
 
 
347
 
  port= in_port_t(port_arg == 0 ? GEARMAN_DEFAULT_TCP_PORT : port_arg);
348
 
}
349
 
 
350
 
void gearman_connection_st::close_socket()
351
 
{
352
 
  if (fd == INVALID_SOCKET)
353
 
  {
354
 
    return;
355
 
  }
356
 
 
357
 
  /* in case of death shutdown to avoid blocking at close_socket() */
358
 
  if (shutdown(fd, SHUT_RDWR) == SOCKET_ERROR && get_socket_errno() != ENOTCONN)
359
 
  {
360
 
#if 0
361
 
    gearman_perror(universal, "shutdown");
362
 
    assert(errno != ENOTSOCK);
363
 
#endif
364
 
  }
365
 
  else
366
 
  {
367
 
    if (closesocket(fd) == SOCKET_ERROR)
368
 
    {
369
 
#if 0
370
 
      gearman_perror(universal, "close");
371
 
#endif
372
 
    }
373
 
  }
374
 
 
375
 
  state= GEARMAN_CON_UNIVERSAL_ADDRINFO;
376
 
  fd= INVALID_SOCKET;
377
 
  events= 0;
378
 
  revents= 0;
379
 
 
380
 
  send_state= GEARMAN_CON_SEND_STATE_NONE;
381
 
  send_buffer_ptr= send_buffer;
382
 
  send_buffer_size= 0;
383
 
  send_data_size= 0;
384
 
  send_data_offset= 0;
385
 
 
386
 
  recv_state= GEARMAN_CON_RECV_UNIVERSAL_NONE;
387
 
  if (recv_packet)
388
 
  {
389
 
    gearman_packet_free(recv_packet);
390
 
    recv_packet= NULL;
391
 
  }
392
 
 
393
 
  recv_buffer_ptr= recv_buffer;
394
 
  recv_buffer_size= 0;
395
 
}
396
 
 
397
 
void gearman_connection_st::reset_addrinfo()
398
 
{
399
 
  if (addrinfo)
400
 
  {
401
 
    freeaddrinfo(addrinfo);
402
 
    addrinfo= NULL;
403
 
  }
404
 
 
405
 
  addrinfo_next= NULL;
406
 
}
407
 
 
408
 
gearman_return_t gearman_connection_st::send_packet(const gearman_packet_st& packet_arg, const bool flush_buffer)
409
 
{
410
 
  switch (send_state)
411
 
  {
412
 
  case GEARMAN_CON_SEND_STATE_NONE:
413
 
    if (packet_arg.options.complete == false)
414
 
    {
415
 
      return gearman_error(universal, GEARMAN_INVALID_PACKET, "packet not complete");
416
 
    }
417
 
 
418
 
    /* Pack first part of packet, which is everything but the payload. */
419
 
    while (1)
420
 
    {
421
 
      gearman_return_t rc;
422
 
      { // Scoping to shut compiler up about switch/case jump
423
 
        size_t send_size= gearman_packet_pack(packet_arg,
424
 
                                              send_buffer +send_buffer_size,
425
 
                                              GEARMAN_SEND_BUFFER_SIZE -send_buffer_size, rc);
426
 
 
427
 
        if (gearman_success(rc))
428
 
        {
429
 
          send_buffer_size+= send_size;
430
 
          break;
431
 
        }
432
 
        else if (rc == GEARMAN_IGNORE_PACKET)
433
 
        {
434
 
          return GEARMAN_SUCCESS;
435
 
        }
436
 
        else if (rc != GEARMAN_FLUSH_DATA)
437
 
        {
438
 
          return rc;
439
 
        }
440
 
      }
441
 
 
442
 
      /* We were asked to flush when the buffer is already flushed! */
443
 
      if (send_buffer_size == 0)
444
 
      {
445
 
        return gearman_universal_set_error(universal, GEARMAN_SEND_BUFFER_TOO_SMALL, __func__, AT,
446
 
                                           "send buffer too small (%u)", GEARMAN_SEND_BUFFER_SIZE);
447
 
      }
448
 
 
449
 
      /* Flush buffer now if first part of packet won't fit in. */
450
 
      send_state= GEARMAN_CON_SEND_UNIVERSAL_PRE_FLUSH;
451
 
 
452
 
    case GEARMAN_CON_SEND_UNIVERSAL_PRE_FLUSH:
453
 
      {
454
 
        gearman_return_t ret= flush();
455
 
        if (gearman_failed(ret))
456
 
        {
457
 
          return ret;
458
 
        }
459
 
      }
460
 
    }
461
 
 
462
 
    /* Return here if we have no data to send. */
463
 
    if (packet_arg.data_size == 0)
464
 
    {
465
 
      break;
466
 
    }
467
 
 
468
 
    /* If there is any room in the buffer, copy in data. */
469
 
    if (packet_arg.data and (GEARMAN_SEND_BUFFER_SIZE - send_buffer_size) > 0)
470
 
    {
471
 
      send_data_offset= GEARMAN_SEND_BUFFER_SIZE - send_buffer_size;
472
 
      if (send_data_offset > packet_arg.data_size)
473
 
      {
474
 
        send_data_offset= packet_arg.data_size;
475
 
      }
476
 
 
477
 
      memcpy(send_buffer + send_buffer_size, packet_arg.data, send_data_offset);
478
 
      send_buffer_size+= send_data_offset;
479
 
 
480
 
      /* Return if all data fit in the send buffer. */
481
 
      if (send_data_offset == packet_arg.data_size)
482
 
      {
483
 
        send_data_offset= 0;
484
 
        break;
485
 
      }
486
 
    }
487
 
 
488
 
    /* Flush buffer now so we can start writing directly from data buffer. */
489
 
    send_state= GEARMAN_CON_SEND_UNIVERSAL_FORCE_FLUSH;
490
 
 
491
 
  case GEARMAN_CON_SEND_UNIVERSAL_FORCE_FLUSH:
492
 
    {
493
 
      gearman_return_t ret= flush();
494
 
      if (gearman_failed(ret))
495
 
      {
496
 
        return ret;
497
 
      }
498
 
    }
499
 
 
500
 
    send_data_size= packet_arg.data_size;
501
 
 
502
 
    /* If this is NULL, then gearman_connection_send_data function will be used. */
503
 
    if (not packet_arg.data)
504
 
    {
505
 
      send_state= GEARMAN_CON_SEND_UNIVERSAL_FLUSH_DATA;
506
 
      return GEARMAN_SUCCESS;
507
 
    }
508
 
 
509
 
    /* Copy into the buffer if it fits, otherwise flush from packet buffer. */
510
 
    send_buffer_size= packet_arg.data_size - send_data_offset;
511
 
    if (send_buffer_size < GEARMAN_SEND_BUFFER_SIZE)
512
 
    {
513
 
      memcpy(send_buffer,
514
 
             static_cast<char *>(const_cast<void *>(packet_arg.data)) + send_data_offset,
515
 
             send_buffer_size);
516
 
      send_data_size= 0;
517
 
      send_data_offset= 0;
518
 
      break;
519
 
    }
520
 
 
521
 
    send_buffer_ptr= static_cast<char *>(const_cast<void *>(packet_arg.data)) + send_data_offset;
522
 
    send_state= GEARMAN_CON_SEND_UNIVERSAL_FLUSH_DATA;
523
 
 
524
 
  case GEARMAN_CON_SEND_UNIVERSAL_FLUSH:
525
 
  case GEARMAN_CON_SEND_UNIVERSAL_FLUSH_DATA:
526
 
    return flush();
527
 
  }
528
 
 
529
 
  if (flush_buffer)
530
 
  {
531
 
    send_state= GEARMAN_CON_SEND_UNIVERSAL_FLUSH;
532
 
    return flush();
533
 
  }
534
 
 
535
 
  send_state= GEARMAN_CON_SEND_STATE_NONE;
536
 
 
537
 
  return GEARMAN_SUCCESS;
538
 
}
539
 
 
540
 
size_t gearman_connection_st::send_and_flush(const void *data, size_t data_size, gearman_return_t *ret_ptr)
541
 
{
542
 
  if (send_state != GEARMAN_CON_SEND_UNIVERSAL_FLUSH_DATA)
543
 
  {
544
 
    return gearman_error(universal, GEARMAN_NOT_FLUSHING, "not flushing");
545
 
  }
546
 
 
547
 
  if (data_size > (send_data_size - send_data_offset))
548
 
  {
549
 
    return gearman_error(universal, GEARMAN_DATA_TOO_LARGE, "data too large");
550
 
  }
551
 
 
552
 
  send_buffer_ptr= static_cast<char *>(const_cast<void *>(data));
553
 
  send_buffer_size= data_size;
554
 
 
555
 
  *ret_ptr= flush();
556
 
 
557
 
  return data_size -send_buffer_size;
558
 
}
559
 
 
560
 
gearman_return_t gearman_connection_st::lookup()
561
 
{
562
 
  if (addrinfo)
563
 
  {
564
 
    freeaddrinfo(addrinfo);
565
 
    addrinfo= NULL;
566
 
  }
567
 
 
568
 
  char port_str[GEARMAN_NI_MAXSERV];
569
 
  snprintf(port_str, GEARMAN_NI_MAXSERV, "%hu", uint16_t(port));
570
 
 
571
 
  struct addrinfo ai;
572
 
  memset(&ai, 0, sizeof(struct addrinfo));
573
 
  ai.ai_socktype= SOCK_STREAM;
574
 
  ai.ai_protocol= IPPROTO_TCP;
575
 
 
576
 
  int ret;
577
 
  if ((ret= getaddrinfo(host, port_str, &ai, &(addrinfo))))
578
 
  {
579
 
    return gearman_universal_set_error(universal, GEARMAN_GETADDRINFO, AT, "getaddrinfo:%s", gai_strerror(ret));
580
 
  }
581
 
 
582
 
  addrinfo_next= addrinfo;
583
 
  state= GEARMAN_CON_UNIVERSAL_CONNECT;
584
 
 
585
 
  return GEARMAN_SUCCESS;
586
 
}
587
 
 
588
 
gearman_return_t gearman_connection_st::flush()
589
 
{
590
 
  while (1)
591
 
  {
592
 
    switch (state)
593
 
    {
594
 
    case GEARMAN_CON_UNIVERSAL_ADDRINFO:
595
 
      {
596
 
        gearman_return_t ret= lookup();
597
 
 
598
 
        if (gearman_failed(ret))
599
 
        {
600
 
          return ret;
601
 
        }
602
 
      }
603
 
 
604
 
    case GEARMAN_CON_UNIVERSAL_CONNECT:
605
 
      if (fd != INVALID_SOCKET)
606
 
      {
607
 
        close_socket();
608
 
      }
609
 
 
610
 
      if (addrinfo_next == NULL)
611
 
      {
612
 
        state= GEARMAN_CON_UNIVERSAL_ADDRINFO;
613
 
        return gearman_universal_set_error(universal, GEARMAN_COULD_NOT_CONNECT, GEARMAN_AT, "%s:%hu", host, uint16_t(port));
614
 
      }
615
 
 
616
 
      fd= socket(addrinfo_next->ai_family, addrinfo_next->ai_socktype, addrinfo_next->ai_protocol);
617
 
      if (fd == INVALID_SOCKET)
618
 
      {
619
 
        state= GEARMAN_CON_UNIVERSAL_ADDRINFO;
620
 
        return gearman_perror(universal, "socket");
621
 
      }
622
 
 
623
 
      {
624
 
        gearman_return_t gret= _con_setsockopt(this);
625
 
        if (gearman_failed(gret))
626
 
        {
627
 
          close_socket();
628
 
          return gret;
629
 
        }
630
 
      }
631
 
 
632
 
      while (1)
633
 
      {
634
 
        if (connect(fd, addrinfo_next->ai_addr, addrinfo_next->ai_addrlen) == 0)
635
 
        {
636
 
          state= GEARMAN_CON_UNIVERSAL_CONNECTED;
637
 
          addrinfo_next= NULL;
638
 
          break;
639
 
        }
640
 
 
641
 
        if (errno == EAGAIN || errno == EINTR)
642
 
        {
643
 
          continue;
644
 
        }
645
 
 
646
 
        if (errno == EINPROGRESS)
647
 
        {
648
 
          gearman_return_t gret= connect_poll();
649
 
          if (gearman_failed(gret))
650
 
          {
651
 
            assert_msg(universal.error.rc != GEARMAN_SUCCESS, "Programmer error, connect_poll() returned an error, but it was not set");
652
 
            close_socket();
653
 
            return gret;
654
 
          }
655
 
 
656
 
          state= GEARMAN_CON_UNIVERSAL_CONNECTING;
657
 
          break;
658
 
        }
659
 
 
660
 
        if (errno == ECONNREFUSED || errno == ENETUNREACH || errno == ETIMEDOUT)
661
 
        {
662
 
          state= GEARMAN_CON_UNIVERSAL_CONNECT;
663
 
          addrinfo_next= addrinfo_next->ai_next;
664
 
          break;
665
 
        }
666
 
 
667
 
        gearman_perror(universal, "connect");
668
 
        close_socket();
669
 
        return GEARMAN_COULD_NOT_CONNECT;
670
 
      }
671
 
 
672
 
      if (state != GEARMAN_CON_UNIVERSAL_CONNECTING)
673
 
      {
674
 
        break;
675
 
      }
676
 
 
677
 
    case GEARMAN_CON_UNIVERSAL_CONNECTING:
678
 
      while (1)
679
 
      {
680
 
        if (revents & (POLLERR | POLLHUP | POLLNVAL))
681
 
        {
682
 
          state= GEARMAN_CON_UNIVERSAL_CONNECT;
683
 
          addrinfo_next= addrinfo_next->ai_next;
684
 
          break;
685
 
        }
686
 
        else if (revents & POLLOUT)
687
 
        {
688
 
          state= GEARMAN_CON_UNIVERSAL_CONNECTED;
689
 
          break;
690
 
        }
691
 
 
692
 
        set_events(POLLOUT);
693
 
 
694
 
        if (gearman_universal_is_non_blocking(universal))
695
 
        {
696
 
          state= GEARMAN_CON_UNIVERSAL_CONNECTING;
697
 
          return gearman_gerror(universal, GEARMAN_IO_WAIT);
698
 
        }
699
 
 
700
 
        gearman_return_t gret= gearman_wait(universal);
701
 
        if (gearman_failed(gret))
702
 
        {
703
 
          return gret;
704
 
        }
705
 
      }
706
 
 
707
 
      if (state != GEARMAN_CON_UNIVERSAL_CONNECTED)
708
 
      {
709
 
        break;
710
 
      }
711
 
 
712
 
    case GEARMAN_CON_UNIVERSAL_CONNECTED:
713
 
      while (send_buffer_size != 0)
714
 
      {
715
 
        ssize_t write_size= ::send(fd, send_buffer_ptr, send_buffer_size, 
716
 
                                   gearman_universal_is_non_blocking(universal) ? MSG_NOSIGNAL| MSG_DONTWAIT : MSG_NOSIGNAL);
717
 
 
718
 
        if (write_size == 0) // Zero value on send()
719
 
        { }
720
 
        else if (write_size == -1)
721
 
        {
722
 
          if (errno == EAGAIN)
723
 
          {
724
 
            set_events(POLLOUT);
725
 
 
726
 
            if (gearman_universal_is_non_blocking(universal))
727
 
            {
728
 
              gearman_gerror(universal, GEARMAN_IO_WAIT);
729
 
              return GEARMAN_IO_WAIT;
730
 
            }
731
 
 
732
 
            gearman_return_t gret= gearman_wait(universal);
733
 
            if (gearman_failed(gret))
734
 
            {
735
 
              return gret;
736
 
            }
737
 
 
738
 
            continue;
739
 
          }
740
 
          else if (errno == EINTR)
741
 
          {
742
 
            continue;
743
 
          }
744
 
          else if (errno == EPIPE || errno == ECONNRESET || errno == EHOSTDOWN)
745
 
          {
746
 
            gearman_return_t ret= gearman_perror(universal, "lost connection to server during send");
747
 
            close_socket();
748
 
            return ret;
749
 
          }
750
 
 
751
 
          gearman_return_t ret= gearman_perror(universal, "send");
752
 
          close_socket();
753
 
 
754
 
          return ret;
755
 
        }
756
 
 
757
 
        send_buffer_size-= size_t(write_size);
758
 
        if (send_state == GEARMAN_CON_SEND_UNIVERSAL_FLUSH_DATA)
759
 
        {
760
 
          send_data_offset+= size_t(write_size);
761
 
          if (send_data_offset == send_data_size)
762
 
          {
763
 
            send_data_size= 0;
764
 
            send_data_offset= 0;
765
 
            break;
766
 
          }
767
 
 
768
 
          if (send_buffer_size == 0)
769
 
          {
770
 
            return GEARMAN_SUCCESS;
771
 
          }
772
 
        }
773
 
        else if (send_buffer_size == 0)
774
 
        {
775
 
          break;
776
 
        }
777
 
 
778
 
        send_buffer_ptr+= write_size;
779
 
      }
780
 
 
781
 
      send_state= GEARMAN_CON_SEND_STATE_NONE;
782
 
      send_buffer_ptr= send_buffer;
783
 
 
784
 
      return GEARMAN_SUCCESS;
785
 
    }
786
 
  }
787
 
}
788
 
 
789
 
gearman_packet_st *gearman_connection_st::receiving(gearman_packet_st& packet_arg,
790
 
                                                    gearman_return_t& ret,
791
 
                                                    const bool recv_data)
792
 
{
793
 
  switch (recv_state)
794
 
  {
795
 
  case GEARMAN_CON_RECV_UNIVERSAL_NONE:
796
 
    if (state != GEARMAN_CON_UNIVERSAL_CONNECTED)
797
 
    {
798
 
      gearman_error(universal, GEARMAN_NOT_CONNECTED, "not connected");
799
 
      ret= GEARMAN_NOT_CONNECTED;
800
 
      return NULL;
801
 
    }
802
 
 
803
 
    recv_packet= gearman_packet_create(universal, &packet_arg);
804
 
    if (recv_packet == NULL)
805
 
    {
806
 
      ret= GEARMAN_MEMORY_ALLOCATION_FAILURE;
807
 
      return NULL;
808
 
    }
809
 
 
810
 
    recv_state= GEARMAN_CON_RECV_UNIVERSAL_READ;
811
 
 
812
 
  case GEARMAN_CON_RECV_UNIVERSAL_READ:
813
 
    while (1)
814
 
    {
815
 
      if (recv_buffer_size > 0)
816
 
      {
817
 
        size_t recv_size= gearman_packet_unpack(*recv_packet,
818
 
                                                recv_buffer_ptr,
819
 
                                                recv_buffer_size, ret);
820
 
        recv_buffer_ptr+= recv_size;
821
 
        recv_buffer_size-= recv_size;
822
 
 
823
 
        if (gearman_success(ret))
824
 
        {
825
 
          break;
826
 
        }
827
 
        else if (ret != GEARMAN_IO_WAIT)
828
 
        {
829
 
          close_socket();
830
 
          return NULL;
831
 
        }
832
 
      }
833
 
 
834
 
      /* Shift buffer contents if needed. */
835
 
      if (recv_buffer_size > 0)
836
 
      {
837
 
        memmove(recv_buffer, recv_buffer_ptr, recv_buffer_size);
838
 
      }
839
 
      recv_buffer_ptr= recv_buffer;
840
 
 
841
 
      size_t recv_size= recv_socket(recv_buffer + recv_buffer_size, GEARMAN_RECV_BUFFER_SIZE - recv_buffer_size, ret);
842
 
      if (gearman_failed(ret))
843
 
      {
844
 
        return NULL;
845
 
      }
846
 
 
847
 
      recv_buffer_size+= recv_size;
848
 
    }
849
 
 
850
 
    if (packet_arg.data_size == 0)
851
 
    {
852
 
      recv_state= GEARMAN_CON_RECV_UNIVERSAL_NONE;
853
 
      break;
854
 
    }
855
 
 
856
 
    recv_data_size= packet_arg.data_size;
857
 
 
858
 
    if (not recv_data)
859
 
    {
860
 
      recv_state= GEARMAN_CON_RECV_STATE_READ_DATA;
861
 
      break;
862
 
    }
863
 
 
864
 
    assert(packet_arg.universal);
865
 
    packet_arg.data= gearman_malloc((*packet_arg.universal), packet_arg.data_size);
866
 
    if (not packet_arg.data)
867
 
    {
868
 
      ret= GEARMAN_MEMORY_ALLOCATION_FAILURE;
869
 
      close_socket();
870
 
      return NULL;
871
 
    }
872
 
 
873
 
    packet_arg.options.free_data= true;
874
 
    recv_state= GEARMAN_CON_RECV_STATE_READ_DATA;
875
 
 
876
 
  case GEARMAN_CON_RECV_STATE_READ_DATA:
877
 
    while (recv_data_size)
878
 
    {
879
 
      (void)receiving(static_cast<uint8_t *>(const_cast<void *>(packet_arg.data)) +
880
 
                      recv_data_offset,
881
 
                      packet_arg.data_size -recv_data_offset, ret);
882
 
      if (gearman_failed(ret))
883
 
      {
884
 
        return NULL;
885
 
      }
886
 
    }
887
 
 
888
 
    recv_state= GEARMAN_CON_RECV_UNIVERSAL_NONE;
889
 
    break;
890
 
  }
891
 
 
892
 
  gearman_packet_st *tmp_packet_arg= recv_packet;
893
 
  recv_packet= NULL;
894
 
 
895
 
  return tmp_packet_arg;
896
 
}
897
 
 
898
 
size_t gearman_connection_st::receiving(void *data, size_t data_size, gearman_return_t& ret)
899
 
{
900
 
  size_t recv_size= 0;
901
 
 
902
 
  if (recv_data_size == 0)
903
 
  {
904
 
    ret= GEARMAN_SUCCESS;
905
 
    return 0;
906
 
  }
907
 
 
908
 
  if ((recv_data_size - recv_data_offset) < data_size)
909
 
    data_size= recv_data_size - recv_data_offset;
910
 
 
911
 
  if (recv_buffer_size > 0)
912
 
  {
913
 
    if (recv_buffer_size < data_size)
914
 
      recv_size= recv_buffer_size;
915
 
    else
916
 
      recv_size= data_size;
917
 
 
918
 
    memcpy(data, recv_buffer_ptr, recv_size);
919
 
    recv_buffer_ptr+= recv_size;
920
 
    recv_buffer_size-= recv_size;
921
 
  }
922
 
 
923
 
  if (data_size != recv_size)
924
 
  {
925
 
    recv_size+= recv_socket(static_cast<uint8_t *>(const_cast<void *>(data)) + recv_size, data_size - recv_size, ret);
926
 
    recv_data_offset+= recv_size;
927
 
  }
928
 
  else
929
 
  {
930
 
    recv_data_offset+= recv_size;
931
 
    ret= GEARMAN_SUCCESS;
932
 
  }
933
 
 
934
 
  if (recv_data_size == recv_data_offset)
935
 
  {
936
 
    recv_data_size= 0;
937
 
    recv_data_offset= 0;
938
 
    recv_state= GEARMAN_CON_RECV_UNIVERSAL_NONE;
939
 
  }
940
 
 
941
 
  return recv_size;
942
 
}
943
 
 
944
 
size_t gearman_connection_st::recv_socket(void *data, size_t data_size, gearman_return_t& ret)
945
 
{
946
 
  ssize_t read_size;
947
 
 
948
 
  while (1)
949
 
  {
950
 
    read_size= ::recv(fd, data, data_size, 0);
951
 
    if (read_size == 0)
952
 
    {
953
 
      gearman_error(universal, GEARMAN_LOST_CONNECTION, "lost connection to server (EOF)");
954
 
      close_socket();
955
 
      ret= GEARMAN_LOST_CONNECTION;
956
 
 
957
 
      return 0;
958
 
    }
959
 
    else if (read_size == -1)
960
 
    {
961
 
      if (errno == EAGAIN)
962
 
      {
963
 
        set_events(POLLIN);
964
 
 
965
 
        if (gearman_universal_is_non_blocking(universal))
966
 
        {
967
 
          gearman_gerror(universal, GEARMAN_IO_WAIT);
968
 
          ret= GEARMAN_IO_WAIT;
969
 
          return 0;
970
 
        }
971
 
 
972
 
        ret= gearman_wait(universal);
973
 
 
974
 
        if (ret == GEARMAN_SHUTDOWN_GRACEFUL)
975
 
        {
976
 
          ret= gearman_kill(gearman_universal_id(universal), GEARMAN_INTERRUPT);
977
 
 
978
 
          if (gearman_failed(ret))
979
 
          {
980
 
            ret= GEARMAN_SHUTDOWN;
981
 
          }
982
 
        }
983
 
        else if (ret == GEARMAN_SHUTDOWN)
984
 
        {
985
 
          close_socket();
986
 
          return 0;
987
 
        }
988
 
 
989
 
        if (gearman_failed(ret))
990
 
        {
991
 
          return 0;
992
 
        }
993
 
 
994
 
        continue;
995
 
      }
996
 
      else if (errno == EINTR)
997
 
      {
998
 
        continue;
999
 
      }
1000
 
      else if (errno == EPIPE || errno == ECONNRESET || errno == EHOSTDOWN)
1001
 
      {
1002
 
        gearman_perror(universal, "lost connection to server during read");
1003
 
        ret= GEARMAN_LOST_CONNECTION;
1004
 
      }
1005
 
      else
1006
 
      {
1007
 
        gearman_perror(universal, "read");
1008
 
        ret= GEARMAN_ERRNO;
1009
 
      }
1010
 
 
1011
 
      close_socket();
1012
 
      return 0;
1013
 
    }
1014
 
 
1015
 
    break;
1016
 
  }
1017
 
 
1018
 
  ret= GEARMAN_SUCCESS;
1019
 
  return size_t(read_size);
1020
 
}
1021
 
 
1022
 
void gearman_connection_st::set_events(short arg)
1023
 
{
1024
 
  if ((events | arg) == events)
1025
 
    return;
1026
 
 
1027
 
  events|= arg;
1028
 
}
1029
 
 
1030
 
void gearman_connection_st::set_revents(short arg)
1031
 
{
1032
 
  if (arg)
1033
 
    options.ready= true;
1034
 
 
1035
 
  revents= arg;
1036
 
  events&= short(~arg);
1037
 
}
1038
 
 
1039
 
/*
1040
 
 * Static Definitions
1041
 
 */
1042
 
 
1043
 
static gearman_return_t _con_setsockopt(gearman_connection_st *connection)
1044
 
{
1045
 
  int ret;
1046
 
  struct linger linger;
1047
 
  struct timeval waittime;
1048
 
 
1049
 
  ret= 1;
1050
 
  ret= setsockopt(connection->fd, IPPROTO_TCP, TCP_NODELAY, &ret,
1051
 
                  socklen_t(sizeof(int)));
1052
 
  if (ret == -1 && errno != EOPNOTSUPP)
1053
 
  {
1054
 
    gearman_perror(connection->universal, "setsockopt(TCP_NODELAY)");
1055
 
    return GEARMAN_ERRNO;
1056
 
  }
1057
 
 
1058
 
  linger.l_onoff= 1;
1059
 
  linger.l_linger= GEARMAN_DEFAULT_SOCKET_TIMEOUT;
1060
 
  ret= setsockopt(connection->fd, SOL_SOCKET, SO_LINGER, &linger,
1061
 
                  socklen_t(sizeof(struct linger)));
1062
 
  if (ret == -1)
1063
 
  {
1064
 
    gearman_perror(connection->universal, "setsockopt(SO_LINGER)");
1065
 
    return GEARMAN_ERRNO;
1066
 
  }
1067
 
 
1068
 
  waittime.tv_sec= GEARMAN_DEFAULT_SOCKET_TIMEOUT;
1069
 
  waittime.tv_usec= 0;
1070
 
  ret= setsockopt(connection->fd, SOL_SOCKET, SO_SNDTIMEO, &waittime,
1071
 
                  socklen_t(sizeof(struct timeval)));
1072
 
  if (ret == -1 && errno != ENOPROTOOPT)
1073
 
  {
1074
 
    gearman_perror(connection->universal, "setsockopt(SO_SNDTIMEO)");
1075
 
    return GEARMAN_ERRNO;
1076
 
  }
1077
 
 
1078
 
  {
1079
 
    int optval= 1;
1080
 
    ret= setsockopt(connection->fd, SOL_SOCKET, SO_KEEPALIVE, &optval, sizeof(optval));
1081
 
    if (ret == -1 && errno != ENOPROTOOPT)
1082
 
    {
1083
 
      gearman_perror(connection->universal, "setsockopt(SO_RCVTIMEO)");
1084
 
      return GEARMAN_ERRNO;
1085
 
    }
1086
 
  }
1087
 
 
1088
 
 
1089
 
  ret= setsockopt(connection->fd, SOL_SOCKET, SO_RCVTIMEO, &waittime,
1090
 
                  socklen_t(sizeof(struct timeval)));
1091
 
  if (ret == -1 && errno != ENOPROTOOPT)
1092
 
  {
1093
 
    gearman_perror(connection->universal, "setsockopt(SO_RCVTIMEO)");
1094
 
    return GEARMAN_ERRNO;
1095
 
  }
1096
 
 
1097
 
  ret= GEARMAN_DEFAULT_SOCKET_SEND_SIZE;
1098
 
  ret= setsockopt(connection->fd, SOL_SOCKET, SO_SNDBUF, &ret, socklen_t(sizeof(int)));
1099
 
  if (ret == -1)
1100
 
  {
1101
 
    gearman_perror(connection->universal, "setsockopt(SO_SNDBUF)");
1102
 
    return GEARMAN_ERRNO;
1103
 
  }
1104
 
 
1105
 
#if defined(__MACH__) && defined(__APPLE__) || defined(__FreeBSD__)
1106
 
  {
1107
 
    ret= 1;
1108
 
    setsockopt(connection->fd, SOL_SOCKET, SO_NOSIGPIPE, static_cast<void *>(&ret), sizeof(int));
1109
 
 
1110
 
    // This is not considered a fatal error 
1111
 
    if (ret == -1)
1112
 
    {
1113
 
      gearman_perror(connection->universal, "setsockopt(SO_NOSIGPIPE)");
1114
 
    }
1115
 
  }
1116
 
#endif
1117
 
 
1118
 
  ret= GEARMAN_DEFAULT_SOCKET_RECV_SIZE;
1119
 
  ret= setsockopt(connection->fd, SOL_SOCKET, SO_RCVBUF, &ret, socklen_t(sizeof(int)));
1120
 
  if (ret == -1)
1121
 
  {
1122
 
    gearman_perror(connection->universal, "setsockopt(SO_RCVBUF)");
1123
 
    return GEARMAN_ERRNO;
1124
 
  }
1125
 
 
1126
 
  ret= fcntl(connection->fd, F_GETFL, 0);
1127
 
  if (ret == -1)
1128
 
  {
1129
 
    gearman_perror(connection->universal, "fcntl(F_GETFL)");
1130
 
    return GEARMAN_ERRNO;
1131
 
  }
1132
 
 
1133
 
  ret= fcntl(connection->fd, F_SETFL, ret | O_NONBLOCK);
1134
 
  if (ret == -1)
1135
 
  {
1136
 
    gearman_perror(connection->universal, "fcntl(F_SETFL)");
1137
 
    return GEARMAN_ERRNO;
1138
 
  }
1139
 
 
1140
 
  return GEARMAN_SUCCESS;
1141
 
}