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
|
/*
neon-specific test utils
Copyright (C) 2001-2009, Joe Orton <joe@manyfish.co.uk>
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; either version 2 of the License, or
(at your option) any later version.
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., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#ifndef UTILS_H
#define UTILS_H 1
#include "ne_request.h"
#include "child.h"
#define ONREQ(x) do { int _ret = (x); if (_ret) { t_context("line %d: HTTP error:\n%s", __LINE__, ne_get_error(sess)); return FAIL; } } while (0);
int single_serve_string(ne_socket *s, void *userdata);
struct many_serve_args {
int count;
const char *str;
};
/* Serves args->str response args->count times down a single
* connection. */
int many_serve_string(ne_socket *s, void *userdata);
/* Run a request using URI on the session. */
int any_request(ne_session *sess, const char *uri);
/* Run a request using URI on the session; fail on a non-2xx response.
*/
int any_2xx_request(ne_session *sess, const char *uri);
/* As above but with a request body. */
int any_2xx_request_body(ne_session *sess, const char *uri);
/* makes *session, spawns server which will run 'fn(userdata,
* socket)'. sets error context if returns non-zero, i.e use like:
* CALL(make_session(...)); */
int make_session(ne_session **sess, server_fn fn, void *userdata);
/* Server which sleeps for 10 seconds then closes the socket. */
int sleepy_server(ne_socket *sock, void *userdata);
struct string {
char *data;
size_t len;
};
struct double_serve_args {
struct string first, second;
};
/* Serve a struct string. */
int serve_sstring(ne_socket *sock, void *ud);
/* Discards an HTTP request, serves response ->first, discards another
* HTTP request, then serves response ->second. */
int double_serve_sstring(ne_socket *s, void *userdata);
/* Serve a struct string slowly. */
int serve_sstring_slowly(ne_socket *sock, void *ud);
struct infinite {
const char *header, *repeat;
};
/* Pass a "struct infinite *" as userdata, this function sends
* ->header and then loops sending ->repeat forever. */
int serve_infinite(ne_socket *sock, void *ud);
/* SOCKS server stuff. */
struct socks_server {
enum ne_sock_sversion version;
enum socks_failure {
fail_none = 0,
fail_init_vers,
fail_init_close,
fail_init_trunc,
fail_no_auth,
fail_bogus_auth,
fail_auth_close,
fail_auth_denied
} failure;
unsigned int expect_port;
ne_inet_addr *expect_addr;
const char *expect_fqdn;
const char *username;
const char *password;
int say_hello;
server_fn server;
void *userdata;
};
int socks_server(ne_socket *sock, void *userdata);
int full_write(ne_socket *sock, const char *data, size_t len);
/* Create a session with server process running fn(userdata). Sets
* test suite error on failure; initializes *sess with a new session
* on success. Uses an unspecified hostname/port for the server. */
int session_server(ne_session **sess, server_fn fn, void *userdata);
/* Create a session with server process running fn(userdata). Sets
* test suite error on failure; initializes *sess with a new session
* on success. Uses an unspecified hostname/port for the server;
* session is created as if using origin 'host:fakeport' via proxy to
* spawned server. */
int proxied_session_server(ne_session **sess, const char *scheme,
const char *host, unsigned int fakeport,
server_fn fn, void *userdata);
#endif /* UTILS_H */
|