]> granicus.if.org Git - apache/blob - modules/proxy/mod_proxy.h
Add pre_request and post_request proxy hooks.
[apache] / modules / proxy / mod_proxy.h
1 /* Copyright 1999-2004 The Apache Software Foundation
2  *
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15
16 #ifndef MOD_PROXY_H
17 #define MOD_PROXY_H 
18
19 /*
20  * Main include file for the Apache proxy
21  */
22
23 /*
24
25    Also note numerous FIXMEs and CHECKMEs which should be eliminated.
26
27    This code is once again experimental!
28
29    Things to do:
30
31    1. Make it completely work (for FTP too)
32
33    2. HTTP/1.1
34
35    Chuck Murcko <chuck@topsail.org> 02-06-01
36
37  */
38
39 #define CORE_PRIVATE
40
41 #include "apr_hooks.h"
42 #include "apr.h"
43 #include "apr_lib.h"
44 #include "apr_strings.h"
45 #include "apr_buckets.h"
46 #include "apr_md5.h"
47 #include "apr_network_io.h"
48 #include "apr_pools.h"
49 #include "apr_strings.h"
50 #include "apr_uri.h"
51 #include "apr_date.h"
52 #include "apr_strmatch.h"
53 #include "apr_fnmatch.h"
54 #include "apr_reslist.h"
55 #define APR_WANT_STRFUNC
56 #include "apr_want.h"
57
58 #include "httpd.h"
59 #include "http_config.h"
60 #include "ap_config.h"
61 #include "http_core.h"
62 #include "http_protocol.h"
63 #include "http_request.h"
64 #include "http_vhost.h"
65 #include "http_main.h"
66 #include "http_log.h"
67 #include "http_connection.h"
68 #include "util_filter.h"
69 #include "util_ebcdic.h"
70
71 #if APR_HAVE_NETINET_IN_H
72 #include <netinet/in.h>
73 #endif
74 #if APR_HAVE_ARPA_INET_H
75 #include <arpa/inet.h>
76 #endif
77
78 /* for proxy_canonenc() */
79 enum enctype {
80     enc_path, enc_search, enc_user, enc_fpath, enc_parm
81 };
82
83 #if APR_CHARSET_EBCDIC
84 #define CRLF   "\r\n"
85 #else /*APR_CHARSET_EBCDIC*/
86 #define CRLF   "\015\012"
87 #endif /*APR_CHARSET_EBCDIC*/
88
89 /* default Max-Forwards header setting */
90 #define DEFAULT_MAX_FORWARDS    10
91
92 /* static information about a remote proxy */
93 struct proxy_remote {
94     const char *scheme;         /* the schemes handled by this proxy, or '*' */
95     const char *protocol;       /* the scheme used to talk to this proxy */
96     const char *hostname;       /* the hostname of this proxy */
97     apr_port_t  port;           /* the port for this proxy */
98     regex_t *regexp;            /* compiled regex (if any) for the remote */
99     int use_regex;              /* simple boolean. True if we have a regex pattern */
100 };
101
102 struct proxy_alias {
103     const char  *real;
104     const char  *fake;
105 };
106
107 struct dirconn_entry {
108     char *name;
109     struct in_addr addr, mask;
110     struct apr_sockaddr_t *hostaddr;
111     int (*matcher) (struct dirconn_entry * This, request_rec *r);
112 };
113
114 struct noproxy_entry {
115     const char *name;
116     struct apr_sockaddr_t *addr;
117 };
118
119 typedef struct {
120     apr_array_header_t *proxies;
121     apr_array_header_t *sec_proxy;
122     apr_array_header_t *aliases;
123     apr_array_header_t *raliases;
124     apr_array_header_t *noproxies;
125     apr_array_header_t *dirconn;
126     apr_array_header_t *allowed_connect_ports;
127     apr_array_header_t *workers;
128     apr_array_header_t *balancers;
129     const char *domain;         /* domain name to use in absence of a domain name in the request */
130     int req;                    /* true if proxy requests are enabled */
131     char req_set;
132     enum {
133       via_off,
134       via_on,
135       via_block,
136       via_full
137     } viaopt;                   /* how to deal with proxy Via: headers */
138     char viaopt_set;
139     apr_size_t recv_buffer_size;
140     char recv_buffer_size_set;
141     apr_size_t io_buffer_size;
142     char io_buffer_size_set;
143     long maxfwd;
144     char maxfwd_set;
145     /** 
146      * the following setting masks the error page
147      * returned from the 'proxied server' and just 
148      * forwards the status code upwards.
149      * This allows the main server (us) to generate
150      * the error page, (so it will look like a error
151      * returned from the rest of the system 
152      */
153     int error_override;
154     int error_override_set;
155     int preserve_host;
156     int preserve_host_set;
157     apr_interval_time_t timeout;
158     apr_interval_time_t timeout_set;
159     enum {
160       bad_error,
161       bad_ignore,
162       bad_body
163     } badopt;                   /* how to deal with bad headers */
164     char badopt_set;
165 /* putting new stuff on the end maximises binary back-compatibility.
166  * the strmatch_patterns are really a const just to have a
167  * case-independent strstr.
168  */
169     apr_array_header_t* cookie_paths;
170     apr_array_header_t* cookie_domains;
171     const apr_strmatch_pattern* cookie_path_str;
172     const apr_strmatch_pattern* cookie_domain_str;
173
174 } proxy_server_conf;
175
176 typedef struct {
177     const char *p;            /* The path */
178     int         p_is_fnmatch; /* Is this path an fnmatch candidate? */
179     regex_t    *r;            /* Is this a regex? */
180 } proxy_dir_conf;
181
182 typedef struct {
183     conn_rec *connection;
184     char *hostname;
185     apr_port_t port;
186     int is_ssl;
187 } proxy_conn_rec;
188
189 typedef struct {
190         float cache_completion; /* completion percentage */
191         int content_length; /* length of the content */
192 } proxy_completion;
193
194 /* Physical connection to the backend server */
195 typedef struct {
196     apr_pool_t   *pool; /* Subpool used for creating socket */
197     apr_socket_t *sock;
198     int          close; /* Close 'this' connection */
199 } proxy_conn;
200
201 /* Connection pool */
202 typedef struct {
203     apr_pool_t     *pool;   /* The pool used in constructor and destructor calls */
204     apr_sockaddr_t *addr;   /* Preparsed remote address info */
205 #if APR_HAS_THREADS
206     apr_reslist_t  *res;    /* Connection resource list */
207 #else
208     proxy_conn     *conn;   /* Single connection for prefork mpm's */
209 #endif
210 } proxy_conn_pool;
211
212 /* Worker configuration */
213 typedef struct {
214     int             status;
215     apr_time_t      error_time; /* time of the last error */
216     apr_interval_time_t retry;  /* retry interval */
217     int             retries;    /* number of retries on this worker */
218     int             lbfactor;   /* initial load balancing factor */
219     const char      *name;
220     const char      *scheme;    /* scheme to use ajp|http|https */
221     const char      *hostname;  /* remote backend address */
222     apr_port_t      port;
223     int             min;        /* Desired minimum number of available connections */
224     int             smax;       /* Soft maximum on the total number of connections */
225     int             hmax;       /* Hard maximum on the total number of connections */
226     apr_interval_time_t ttl;    /* maximum amount of time in seconds a connection
227                                  * may be available while exceeding the soft limit */
228     proxy_conn_pool *cp;        /* Connection pool to use */
229     void            *opaque;    /* per scheme worker data */
230 } proxy_worker;
231
232 /* Runtime worker status informations. Shared in scoreboard */
233 typedef struct {
234     proxy_worker    *w;
235     double          lbfactor;   /* dynamic lbfactor */
236     double          lbsatus;    /* Current lbstatus */
237     apr_size_t      transfered; /* Number of bytes transfered to remote */
238     apr_size_t      readed;     /* Number of bytes readed from remote */
239 } proxy_runtime_worker;
240
241 struct proxy_balancer {
242     apr_array_header_t *workers; /* array of proxy_runtime_workers */
243     const char *name;            /* name of the load balancer */
244     const char *sticky;          /* sticky session identifier */
245     int         sticky_force;    /* Disable failover for sticky sessions */
246     apr_interval_time_t timeout; /* Timeout for waiting on free connection */
247 #if APR_HAS_THREADS
248     apr_thread_mutex_t  *mutex;  /* Thread lock for updating lb params */
249 #endif
250 };
251
252 /* data structure for set/get module_config */
253 typedef struct {
254     char                  *url;         /* rewtitten url */
255     struct proxy_balancer *balancer;    /* load balancer to use */
256     proxy_worker          *worker;      /* most suitable worker */
257     void                  *opaque;      /* module private data */
258 } proxy_module_conf;
259
260 /* hooks */
261
262 /* Create a set of PROXY_DECLARE(type), PROXY_DECLARE_NONSTD(type) and 
263  * PROXY_DECLARE_DATA with appropriate export and import tags for the platform
264  */
265 #if !defined(WIN32)
266 #define PROXY_DECLARE(type)            type
267 #define PROXY_DECLARE_NONSTD(type)     type
268 #define PROXY_DECLARE_DATA
269 #elif defined(PROXY_DECLARE_STATIC)
270 #define PROXY_DECLARE(type)            type __stdcall
271 #define PROXY_DECLARE_NONSTD(type)     type
272 #define PROXY_DECLARE_DATA
273 #elif defined(PROXY_DECLARE_EXPORT)
274 #define PROXY_DECLARE(type)            __declspec(dllexport) type __stdcall
275 #define PROXY_DECLARE_NONSTD(type)     __declspec(dllexport) type
276 #define PROXY_DECLARE_DATA             __declspec(dllexport)
277 #else
278 #define PROXY_DECLARE(type)            __declspec(dllimport) type __stdcall
279 #define PROXY_DECLARE_NONSTD(type)     __declspec(dllimport) type
280 #define PROXY_DECLARE_DATA             __declspec(dllimport)
281 #endif
282
283 /**
284  * Hook an optional proxy hook.  Unlike static hooks, this uses a macro
285  * instead of a function.
286  */
287 #define PROXY_OPTIONAL_HOOK(name,fn,pre,succ,order) \
288         APR_OPTIONAL_HOOK(proxy,name,fn,pre,succ,order)
289
290 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, scheme_handler, (request_rec *r, 
291                           proxy_server_conf *conf, char *url, 
292                           const char *proxyhost, apr_port_t proxyport))
293 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, canon_handler, (request_rec *r, 
294                           char *url))
295
296 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, create_req, (request_rec *r, request_rec *pr))
297 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, fixups, (request_rec *r)) 
298
299 /**
300  * pre request hook.
301  * It will return the most suitable worker at the moment
302  * and coresponding balancer.
303  * The url is rewritten from balancer://cluster/uri to scheme://host:port/uri
304  * and then the scheme_handler is called.
305  *
306  */
307 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, pre_request, (proxy_worker **worker,
308                           struct proxy_balancer **balancer,
309                           request_rec *r,
310                           proxy_server_conf *conf, char **url))                          
311 /**
312  * post request hook.
313  * It is called after request for updating runtime balancer status.
314  */
315 APR_DECLARE_EXTERNAL_HOOK(proxy, PROXY, int, post_request, (proxy_worker *worker,
316                           struct proxy_balancer *balancer, request_rec *r,
317                           proxy_server_conf *conf))
318
319
320 /* proxy_util.c */
321
322 PROXY_DECLARE(request_rec *)ap_proxy_make_fake_req(conn_rec *c, request_rec *r);
323 PROXY_DECLARE(int) ap_proxy_hex2c(const char *x);
324 PROXY_DECLARE(void) ap_proxy_c2hex(int ch, char *x);
325 PROXY_DECLARE(char *)ap_proxy_canonenc(apr_pool_t *p, const char *x, int len, enum enctype t,
326                         int isenc);
327 PROXY_DECLARE(char *)ap_proxy_canon_netloc(apr_pool_t *p, char **const urlp, char **userp,
328                          char **passwordp, char **hostp, apr_port_t *port);
329 PROXY_DECLARE(const char *)ap_proxy_date_canon(apr_pool_t *p, const char *x);
330 PROXY_DECLARE(int) ap_proxy_liststr(const char *list, const char *val);
331 PROXY_DECLARE(char *)ap_proxy_removestr(apr_pool_t *pool, const char *list, const char *val);
332 PROXY_DECLARE(int) ap_proxy_hex2sec(const char *x);
333 PROXY_DECLARE(void) ap_proxy_sec2hex(int t, char *y);
334 PROXY_DECLARE(int) ap_proxyerror(request_rec *r, int statuscode, const char *message);
335 PROXY_DECLARE(int) ap_proxy_is_ipaddr(struct dirconn_entry *This, apr_pool_t *p);
336 PROXY_DECLARE(int) ap_proxy_is_domainname(struct dirconn_entry *This, apr_pool_t *p);
337 PROXY_DECLARE(int) ap_proxy_is_hostname(struct dirconn_entry *This, apr_pool_t *p);
338 PROXY_DECLARE(int) ap_proxy_is_word(struct dirconn_entry *This, apr_pool_t *p);
339 PROXY_DECLARE(int) ap_proxy_checkproxyblock(request_rec *r, proxy_server_conf *conf, apr_sockaddr_t *uri_addr);
340 PROXY_DECLARE(int) ap_proxy_pre_http_request(conn_rec *c, request_rec *r);
341 PROXY_DECLARE(apr_status_t) ap_proxy_string_read(conn_rec *c, apr_bucket_brigade *bb, char *buff, size_t bufflen, int *eos);
342 PROXY_DECLARE(void) ap_proxy_table_unmerge(apr_pool_t *p, apr_table_t *t, char *key);
343 PROXY_DECLARE(int) ap_proxy_connect_to_backend(apr_socket_t **, const char *, apr_sockaddr_t *, const char *, proxy_server_conf *, server_rec *, apr_pool_t *);
344 PROXY_DECLARE(int) ap_proxy_ssl_enable(conn_rec *c);
345 PROXY_DECLARE(int) ap_proxy_ssl_disable(conn_rec *c);
346
347 /* Connection pool API */
348 PROXY_DECLARE(proxy_worker *) ap_proxy_get_worker(apr_pool_t *p, proxy_server_conf *conf, const char *url);
349 PROXY_DECLARE(const char *) ap_proxy_add_worker(proxy_worker **worker, apr_pool_t *p, proxy_server_conf *conf, const char *url);
350 PROXY_DECLARE(struct proxy_balancer *) ap_proxy_get_balancer(apr_pool_t *p, proxy_server_conf *conf, const char *url);
351 PROXY_DECLARE(const char *) ap_proxy_add_balancer(struct proxy_balancer **balancer, apr_pool_t *p, proxy_server_conf *conf, const char *url);
352 PROXY_DECLARE(void) ap_proxy_add_worker_to_balancer(struct proxy_balancer *balancer, proxy_worker *worker);
353
354
355 /* For proxy_util */
356 extern module PROXY_DECLARE_DATA proxy_module;
357
358 #endif /*MOD_PROXY_H*/