~drizzle-trunk/drizzle/development

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
/* Copyright (C) 2000 MySQL AB

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/*
 * Vio Lite.
 * Purpose: include file for Vio that will work with C and C++
 */

#ifndef PLUGIN_MYSQL_PROTOCOL_VIO_H
#define	PLUGIN_MYSQL_PROTOCOL_VIO_H

#include <sys/socket.h>
#include <errno.h>

/* Simple vio interface in C;  The functions are implemented in violite.c */

enum enum_vio_type
{
  VIO_CLOSED, VIO_TYPE_TCPIP, VIO_TYPE_SOCKET, VIO_TYPE_NAMEDPIPE,
  VIO_TYPE_SSL, VIO_TYPE_SHARED_MEMORY
};

typedef struct st_vio Vio;

#define VIO_BUFFERED_READ 2                     /* use buffered read */
#define VIO_READ_BUFFER_SIZE 16384              /* size of read buffer */

Vio*	drizzleclient_vio_new(int sd, enum enum_vio_type type, unsigned int flags);

void	drizzleclient_vio_delete(Vio* vio);
int	drizzleclient_vio_close(Vio* vio);
void    drizzleclient_vio_reset(Vio* vio, enum enum_vio_type type, int sd, uint32_t flags);
size_t	drizzleclient_vio_read(Vio *vio, unsigned char *	buf, size_t size);
size_t  drizzleclient_vio_read_buff(Vio *vio, unsigned char * buf, size_t size);
size_t	drizzleclient_vio_write(Vio *vio, const unsigned char * buf, size_t size);
int	drizzleclient_vio_blocking(Vio *vio, bool onoff, bool *old_mode);
bool	drizzleclient_vio_is_blocking(Vio *vio);
/* setsockopt TCP_NODELAY at IPPROTO_TCP level, when possible */
int	drizzleclient_vio_fastsend(Vio *vio);
/* setsockopt SO_KEEPALIVE at SOL_SOCKET level, when possible */
int32_t	drizzleclient_vio_keepalive(Vio *vio, bool	onoff);
/* Whenever we should retry the last read/write operation. */
bool	drizzleclient_vio_should_retry(Vio *vio);
/* Check that operation was timed out */
bool	drizzleclient_vio_was_interrupted(Vio *vio);
/* Short text description of the socket for those, who are curious.. */
const char* drizzleclient_vio_description(Vio *vio);
/* Return the type of the connection */
enum enum_vio_type drizzleclient_vio_type(Vio* vio);
/* Return last error number */
int	drizzleclient_vio_errno(Vio*vio);
/* Get socket number */
int drizzleclient_vio_fd(Vio*vio);
/* Remote peer's address and name in text form */
bool drizzleclient_vio_peer_addr(Vio *vio, char *buf, uint16_t *port, size_t buflen);
bool drizzleclient_vio_poll_read(Vio *vio, int timeout);
bool drizzleclient_vio_peek_read(Vio *vio, unsigned int *bytes);

void drizzleclient_vio_end(void);

void drizzleclient_vio_ignore_timeout(Vio *vio, bool is_sndtimeo, int32_t timeout);
void drizzleclient_vio_timeout(Vio *vio, bool is_sndtimeo, int32_t timeout);

#if !defined(DONT_MAP_VIO)
#define drizzleclient_vio_delete(vio) 			(vio)->viodelete(vio)
#define drizzleclient_vio_errno(vio)	 			(vio)->vioerrno(vio)
#define drizzleclient_vio_read(vio, buf, size)                ((vio)->read)(vio,buf,size)
#define drizzleclient_vio_write(vio, buf, size)               ((vio)->write)(vio, buf, size)
#define drizzleclient_vio_blocking(vio, set_blocking_mode, old_mode)\
 	(vio)->vioblocking(vio, set_blocking_mode, old_mode)
#define drizzleclient_vio_is_blocking(vio) 			(vio)->is_blocking(vio)
#define drizzleclient_vio_fastsend(vio)			(vio)->fastsend(vio)
#define drizzleclient_vio_keepalive(vio, set_keep_alive)	(vio)->viokeepalive(vio, set_keep_alive)
#define drizzleclient_vio_should_retry(vio) 			(vio)->should_retry(vio)
#define drizzleclient_vio_was_interrupted(vio) 		(vio)->was_interrupted(vio)
#define drizzleclient_vio_close(vio)				((vio)->vioclose)(vio)
#define drizzleclient_vio_peer_addr(vio, buf, prt, buflen)	(vio)->peer_addr(vio, buf, prt, buflen)
#define drizzleclient_vio_timeout(vio, which, seconds)	(vio)->timeout(vio, which, seconds)
#endif /* !defined(DONT_MAP_VIO) */

/* This enumerator is used in parser - should be always visible */
enum SSL_type
{
  SSL_TYPE_NOT_SPECIFIED= -1,
  SSL_TYPE_NONE,
  SSL_TYPE_ANY,
  SSL_TYPE_X509,
  SSL_TYPE_SPECIFIED
};

/* HFTODO - hide this if we don't want client in embedded server */
/* This structure is for every connection on both sides */
struct st_vio
{
  int		sd;		/* int - real or imaginary */
  int			fcntl_mode;	/* Buffered fcntl(sd,F_GETFL) */
  struct sockaddr_storage	local;		/* Local internet address */
  struct sockaddr_storage	remote;		/* Remote internet address */
  int addrLen;                          /* Length of remote address */
  enum enum_vio_type	type;		/* Type of connection */
  char			desc[30];	/* String description */
  char                  *read_pos;      /* start of unfetched data in the
                                           read buffer */
  char                  *read_end;      /* end of unfetched data */

  /* function pointers. They are similar for socket/SSL/whatever */
  void    (*viodelete)(Vio*);
  int32_t     (*vioerrno)(Vio*);
  size_t  (*read)(Vio*, unsigned char *, size_t);
  size_t  (*write)(Vio*, const unsigned char *, size_t);
  int32_t     (*vioblocking)(Vio*, bool, bool *);
  bool (*is_blocking)(Vio*);
  int32_t     (*viokeepalive)(Vio*, bool);
  int32_t     (*fastsend)(Vio*);
  bool (*peer_addr)(Vio*, char *, uint16_t *, size_t);
  void    (*in_addr)(Vio*, struct sockaddr_storage*);
  bool (*should_retry)(Vio*);
  bool (*was_interrupted)(Vio*);
  int32_t     (*vioclose)(Vio*);
  void	  (*timeout)(Vio*, bool is_sndtimeo, int32_t timeout);
  char                  *read_buffer;   /* buffer for drizzleclient_vio_read_buff */
};

#endif /* PLUGIN_MYSQL_PROTOCOL_VIO_H */