Resets lineno with new file
[tedtools.git] / connection.h
1 /*
2  * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *      notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *      notice, this list of conditions and the following disclaimer in the
12  *      documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the author nor the names of any co-contributors
14  *      may be used to endorse or promote products derived from this software
15  *      without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #ifndef __CONNECTION_H__
31 #define __CONNECTION_H__
32
33 #include <sys/types.h>
34 #include <sys/time.h>
35 #include <sys/socket.h>
36 #include <netinet/in.h>
37 #include <arpa/inet.h>
38
39 #include "tools.h"
40
41 #define CS_OK           0
42 #define CS_INPROCESS    1
43 #define CS_CONNECTED    2
44 #define CS_READ  3
45 #define CS_SEND  4
46 #define CS_WAIT  5
47 #define CS_ERROR        6
48 #define CS_FINISHSEND   7
49 #define CS_FINISHREAD   8
50 #define CS_TIMEOUT      9
51 #define CS_CLOSED       10      
52 #define CS_NOTINITED    11
53 #define CS_AGAIN        12
54 #define CS_FINISH       13
55
56
57 #define READ_INCRIMENT_BUFSIZ 1024
58
59 /*
60  * any message shoulbe compatible with TCMsg struct
61  */
62 typedef struct {
63                 /* len and type should be transferred in network byte order */
64         u_int32_t       len; 
65         u_int32_t       type;
66         char        data[1];
67 } TCMsg;
68
69 #define TCMSGHDRSZ      offsetof(TCMsg, data)
70
71 /* tcp */
72 typedef struct {
73         /* I/O buffer */
74         u_int32_t  len;
75         char    *ptr;
76         TCMsg   *buf; /* send buf */
77
78         /* internal fields */
79         int     fd;
80         u_int32_t
81                 readyio:1,  
82                 state:29;
83         struct sockaddr_in serv_addr;
84
85         /* external link */
86         void*           data;
87 } TC_Connection;
88
89 #define TCCONNHDRSZ  offsetof(TC_Connection, data)
90
91 TC_Connection *TC_fillConnection( TC_Connection *cs, char *name, u_int32_t port );
92 TC_Connection* TC_AcceptTcp(TC_Connection *cs);
93 u_int32_t TC_ClientInitConnection(TC_Connection *cs, char *name, u_int32_t port);
94 u_int32_t TC_ServerInitConnect( TC_Connection *cs );
95 u_int32_t TC_ServerConnect( TC_Connection *cs, int timeout );
96 /*
97  * TC_Send doesn't promise to keep TC_Connection->buf unchanged
98  */
99 u_int32_t TC_Send( TC_Connection *cs );
100 u_int32_t TC_Read( TC_Connection *cs, size_t maxsize );
101 u_int32_t TC_Talk( TC_Connection *cs, size_t maxsize );
102 void TC_FreeConnection( TC_Connection *cs );
103 int TC_ReadyIO( TC_Connection **cs, int number, int timeout );
104
105 /* udp */
106
107
108 typedef struct {
109         char    *host;
110         u_int32_t       port;
111         TCMsg   *msg;
112
113         /* private members */
114         int             sockfd;
115         struct sockaddr_in host_addr;
116 } Msg; 
117
118 /* 
119  * Udp server loop:
120  * TC_Connection  conn,*connptr;
121  * conn.fd = TC_AcceptUdp("127.0.0.1", 5432);
122  * conn.state = CS_READ;
123  * connptr = &conn;
124  * while(1) {
125  *      if ( TC_ReadyIO( &connptr, 1, 100 ) ) {
126  *              Msg m; 
127  *              if ( TC_getMsg(conn.fd, &m) == CS_OK ) {
128  *                      //do something
129  *              } 
130  *      }
131  * }
132  * close(conn.fd);
133  *
134  * Udp client send:
135  * Msg msg;
136  * msg.host     = "127.0.0.1";
137  * msg.port     = 5432;
138  * msg.sockfd =-1;
139  * msg.msg = GOTFILLEDPMSG();
140  * if ( TC_sendMsg(&msg)!=CS_OK ) {
141  *      //Very bad
142  * }
143  * msg.msg = GOTFILLEDPMSG();
144  * if ( TC_sendMsg(&msg)!=CS_OK ) {
145  *      //Very bad
146  * }
147  * TC_closefd(&msg);
148  */ 
149
150 int TC_AcceptUdp(char *host, int port);
151 u_int32_t TC_getMsg( int sockfd, Msg *msg );
152 u_int32_t TC_sendMsg( Msg *msg );
153 void      TC_closefd( Msg *msg );
154
155 /*
156  * Connection pool
157  */
158
159 typedef struct {
160         u_int32_t       len;
161         u_int32_t       number;
162         TC_Connection   **conn;
163 } PoolConnection;
164
165 void TC_addConnection(PoolConnection *pool, TC_Connection *c);
166 void TC_deleteConnectionByN(PoolConnection *pool, int n);
167 void TC_deleteConnectionByC(PoolConnection *pool, TC_Connection *c);
168
169 #ifndef HAVE_HSTRERROR
170 #define hstrerror       strerror
171 #define h_errno         errno
172 #endif
173
174 #endif
175