← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to libdrizzle/libdrizzle_priv.c

  • Committer: Monty Taylor
  • Date: 2008-09-16 00:00:48 UTC
  • mto: This revision was merged to the branch mainline in revision 391.
  • Revision ID: monty@inaugust.com-20080916000048-3rvrv3gv9l0ad3gs
Fixed copyright headers in drizzled/

Show diffs side-by-side

added added

removed removed

Lines of Context:
libdrizzle/libdrizzle_priv.c
1
 
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems, Inc.
5
 
 *
6
 
 *  This program is free software; you can redistribute it and/or modify
7
 
 *  it under the terms of the GNU General Public License as published by
8
 
 *  the Free Software Foundation; version 2 of the License.
9
 
 *
10
 
 *  This program is distributed in the hope that it will be useful,
11
 
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
 *  GNU General Public License for more details.
14
 
 *
15
 
 *  You should have received a copy of the GNU General Public License
16
 
 *  along with this program; if not, write to the Free Software
17
 
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
 
 */
19
 
 
20
 
/* Various helper functions not intended to be part of a public API */
21
 
 
22
 
#include <config.h>
23
 
#include "libdrizzle_priv.h"
24
 
#include <poll.h>
25
 
#include <fcntl.h>
26
 
 
27
 
const char _dig_vec_upper[] =
28
 
  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
29
 
const char _dig_vec_lower[] =
30
 
  "0123456789abcdefghijklmnopqrstuvwxyz";
31
 
 
32
 
const char  *unknown_sqlstate= "HY000";
33
 
const char  *not_error_sqlstate= "00000";
34
 
const char  *cant_connect_sqlstate= "08001";
35
 
 
36
 
const char * sqlstate_get_unknown(void)
37
 
{
38
 
  return unknown_sqlstate;
39
 
}
40
 
 
41
 
const char * sqlstate_get_not_error(void)
42
 
{
43
 
  return not_error_sqlstate;
44
 
}
45
 
 
46
 
const char * sqlstate_get_cant_connect(void)
47
 
{
48
 
  return cant_connect_sqlstate;
49
 
}
50
 
 
51
 
/****************************************************************************
52
 
  A modified version of connect().  connect_with_timeout() allows you to specify
53
 
  a timeout value, in seconds, that we should wait until we
54
 
  derermine we can't connect to a particular host.  If timeout is 0,
55
 
  connect_with_timeout() will behave exactly like connect().
56
 
 
57
 
  Base version coded by Steve Bernacki, Jr. <steve@navinet.net>
58
 
*****************************************************************************/
59
 
 
60
 
int connect_with_timeout(int fd, const struct sockaddr *name, uint32_t namelen, int32_t timeout)
61
 
{
62
 
  int flags, res, s_err;
63
 
 
64
 
  /*
65
 
    If they passed us a timeout of zero, we should behave
66
 
    exactly like the normal connect() call does.
67
 
  */
68
 
 
69
 
  if (timeout == 0)
70
 
    return connect(fd, (struct sockaddr*) name, namelen);
71
 
 
72
 
  flags = fcntl(fd, F_GETFL, 0);    /* Set socket to not block */
73
 
#ifdef O_NONBLOCK
74
 
  fcntl(fd, F_SETFL, flags | O_NONBLOCK);  /* and save the flags..  */
75
 
#endif
76
 
 
77
 
  res= connect(fd, (struct sockaddr*) name, namelen);
78
 
  s_err= errno;      /* Save the error... */
79
 
  fcntl(fd, F_SETFL, flags);
80
 
  if ((res != 0) && (s_err != EINPROGRESS))
81
 
  {
82
 
    errno= s_err;      /* Restore it */
83
 
    return(-1);
84
 
  }
85
 
  if (res == 0)        /* Connected quickly! */
86
 
    return(0);
87
 
 
88
 
  return wait_for_data(fd, timeout);
89
 
}
90
 
 
91
 
/*
92
 
  Wait up to timeout seconds for a connection to be established.
93
 
 
94
 
  We prefer to do this with poll() as there is no limitations with this.
95
 
  If not, we will use select()
96
 
*/
97
 
 
98
 
int wait_for_data(int fd, int32_t timeout)
99
 
{
100
 
  struct pollfd ufds;
101
 
  int res;
102
 
 
103
 
  ufds.fd= fd;
104
 
  ufds.events= POLLIN | POLLPRI;
105
 
  if (!(res= poll(&ufds, 1, (int) timeout*1000)))
106
 
  {
107
 
    errno= EINTR;
108
 
    return -1;
109
 
  }
110
 
  if (res < 0 || !(ufds.revents & (POLLIN | POLLPRI)) || (ufds.revents & POLLHUP))
111
 
    return -1;
112
 
  return 0;
113
 
}