2001-12-20 12:48:32 +08:00
|
|
|
/* $Id: reqs.c,v 1.50 2001-12-20 04:48:32 rjkaes Exp $
|
2000-02-17 01:32:49 +08:00
|
|
|
*
|
|
|
|
* This is where all the work in tinyproxy is actually done. Incoming
|
2000-09-12 08:04:42 +08:00
|
|
|
* connections have a new thread created for them. The thread then
|
|
|
|
* processes the headers from the client, the response from the server,
|
|
|
|
* and then relays the bytes between the two.
|
2001-08-28 12:32:14 +08:00
|
|
|
* If TUNNEL_SUPPORT is enabled, then tinyproxy will actually work
|
2000-09-12 08:04:42 +08:00
|
|
|
* as a simple buffering TCP tunnel. Very cool! (Robert actually uses
|
2001-05-27 10:29:06 +08:00
|
|
|
* this feature for a buffering NNTP tunnel.)
|
2000-02-17 01:32:49 +08:00
|
|
|
*
|
2000-09-12 08:04:42 +08:00
|
|
|
* Copyright (C) 1998 Steven Young
|
|
|
|
* Copyright (C) 1999,2000 Robert James Kaes (rjkaes@flarenet.com)
|
|
|
|
* Copyright (C) 2000 Chris Lightfoot (chris@ex-parrot.com)
|
2000-02-17 01:32:49 +08:00
|
|
|
*
|
|
|
|
* 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, 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "tinyproxy.h"
|
2000-09-12 08:04:42 +08:00
|
|
|
|
|
|
|
#include "acl.h"
|
|
|
|
#include "anonymous.h"
|
2000-02-17 01:32:49 +08:00
|
|
|
#include "buffer.h"
|
2001-10-26 00:58:50 +08:00
|
|
|
#include "conns.h"
|
2000-02-17 01:32:49 +08:00
|
|
|
#include "filter.h"
|
2000-09-12 08:04:42 +08:00
|
|
|
#include "log.h"
|
2000-02-17 01:32:49 +08:00
|
|
|
#include "regexp.h"
|
2000-09-12 08:04:42 +08:00
|
|
|
#include "reqs.h"
|
|
|
|
#include "sock.h"
|
|
|
|
#include "stats.h"
|
|
|
|
#include "utils.h"
|
2000-02-17 01:32:49 +08:00
|
|
|
|
|
|
|
#define HTTP400ERROR "Unrecognizable request. Only HTTP is allowed."
|
|
|
|
#define HTTP500ERROR "Unable to connect to remote server."
|
|
|
|
#define HTTP503ERROR "Internal server error."
|
|
|
|
|
2001-12-18 03:10:56 +08:00
|
|
|
/*
|
|
|
|
* Maximum length of a HTTP line
|
|
|
|
*/
|
2001-12-19 13:13:40 +08:00
|
|
|
#define HTTP_LINE_LENGTH (MAXBUFFSIZE / 6)
|
2001-12-18 03:10:56 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
/*
|
|
|
|
* Macro to help test if the Upstream proxy supported is compiled in and
|
|
|
|
* enabled.
|
|
|
|
*/
|
|
|
|
#ifdef UPSTREAM_SUPPORT
|
|
|
|
# define UPSTREAM_CONFIGURED() (config.upstream_name && config.upstream_port != -1)
|
|
|
|
#else
|
|
|
|
# define UPSTREAM_CONFIGURED() (0)
|
|
|
|
#endif
|
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
/*
|
|
|
|
* Macro to help test if tunnel support is compiled in, and is enabled.
|
|
|
|
*/
|
|
|
|
#ifdef TUNNEL_SUPPORT
|
|
|
|
# define TUNNEL_CONFIGURED() (config.tunnel_name && config.tunnel_port != -1)
|
|
|
|
#else
|
|
|
|
# define TUNNEL_CONFIGURED() (0)
|
|
|
|
#endif
|
|
|
|
|
2001-12-20 12:48:32 +08:00
|
|
|
/*
|
|
|
|
* Codify the test for the carriage return and new line characters.
|
|
|
|
*/
|
|
|
|
#define CHECK_CRLF(header, len) ((len == 1 && header[0] == '\n') || (len == 2 && header[0] == '\r' && header[1] == '\n'))
|
|
|
|
|
2000-02-17 01:32:49 +08:00
|
|
|
/*
|
2001-09-14 12:56:29 +08:00
|
|
|
* Read in the first line from the client (the request line for HTTP
|
|
|
|
* connections. The request line is allocated from the heap, but it must
|
|
|
|
* be freed in another function.
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static char *
|
|
|
|
read_request_line(struct conn_s *connptr)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2001-09-14 12:56:29 +08:00
|
|
|
char *request_buffer;
|
2001-05-27 10:29:06 +08:00
|
|
|
size_t len;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
retry:
|
2001-11-22 08:19:45 +08:00
|
|
|
len = readline(connptr->client_fd, &request_buffer);
|
2001-08-27 05:11:55 +08:00
|
|
|
if (len <= 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"read_request_line: Client (file descriptor: %d) closed socket before read.",
|
|
|
|
connptr->client_fd);
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(request_buffer);
|
|
|
|
return NULL;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-08-27 05:11:55 +08:00
|
|
|
/*
|
2001-09-14 12:56:29 +08:00
|
|
|
* Strip the new line and character return from the string.
|
2001-08-27 05:11:55 +08:00
|
|
|
*/
|
2001-12-17 08:11:32 +08:00
|
|
|
if (chomp(request_buffer, len) == len) {
|
|
|
|
/*
|
|
|
|
* If the number of characters removed is the same as the
|
|
|
|
* length then it was a blank line. Free the buffer and
|
|
|
|
* try again (since we're looking for a request line.)
|
|
|
|
*/
|
|
|
|
safefree(request_buffer);
|
|
|
|
goto retry;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
log_message(LOG_CONN, "Request (file descriptor %d): %s",
|
|
|
|
connptr->client_fd, request_buffer);
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
return request_buffer;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* This structure holds the information pulled from a URL request.
|
|
|
|
*/
|
|
|
|
struct request_s {
|
2001-09-17 04:10:19 +08:00
|
|
|
char *method;
|
|
|
|
char *protocol;
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
char *host;
|
|
|
|
char *path;
|
|
|
|
int port;
|
|
|
|
};
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
static void
|
|
|
|
free_request_struct(struct request_s *request)
|
2001-09-17 04:10:19 +08:00
|
|
|
{
|
2001-10-23 00:08:29 +08:00
|
|
|
if (!request)
|
|
|
|
return;
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
safefree(request->method);
|
|
|
|
safefree(request->protocol);
|
|
|
|
|
|
|
|
safefree(request->host);
|
|
|
|
safefree(request->path);
|
|
|
|
|
|
|
|
safefree(request);
|
|
|
|
}
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
/*
|
|
|
|
* Pull the information out of the URL line.
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
extract_http_url(const char *url, struct request_s *request)
|
2001-09-14 12:56:29 +08:00
|
|
|
{
|
|
|
|
request->host = safemalloc(strlen(url) + 1);
|
|
|
|
request->path = safemalloc(strlen(url) + 1);
|
|
|
|
|
|
|
|
if (!request->host || !request->path) {
|
|
|
|
safefree(request->host);
|
|
|
|
safefree(request->path);
|
|
|
|
|
|
|
|
return -1;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
if (sscanf
|
|
|
|
(url, "http://%[^:/]:%d%s", request->host, &request->port,
|
|
|
|
request->path) == 3) ;
|
2001-11-23 09:17:19 +08:00
|
|
|
else if (sscanf(url, "http://%[^/]%s", request->host, request->path) == 2)
|
2001-09-14 12:56:29 +08:00
|
|
|
request->port = 80;
|
2001-11-22 08:31:10 +08:00
|
|
|
else if (sscanf(url, "http://%[^:/]:%d", request->host, &request->port)
|
|
|
|
== 2)
|
2001-09-14 12:56:29 +08:00
|
|
|
strcpy(request->path, "/");
|
|
|
|
else if (sscanf(url, "http://%[^/]", request->host) == 1) {
|
|
|
|
request->port = 80;
|
|
|
|
strcpy(request->path, "/");
|
|
|
|
} else {
|
2001-10-24 08:37:23 +08:00
|
|
|
log_message(LOG_ERR, "extract_http_url: Can't parse URL.");
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
safefree(request->host);
|
|
|
|
safefree(request->path);
|
2001-11-22 08:31:10 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
return -1;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
/*
|
|
|
|
* Extract the URL from a SSL connection.
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
extract_ssl_url(const char *url, struct request_s *request)
|
2001-09-14 12:56:29 +08:00
|
|
|
{
|
|
|
|
request->host = safemalloc(strlen(url) + 1);
|
2001-09-16 05:26:14 +08:00
|
|
|
if (!request->host)
|
2001-09-14 12:56:29 +08:00
|
|
|
return -1;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
if (sscanf(url, "%[^:]:%d", request->host, &request->port) == 2) ;
|
2001-09-14 12:56:29 +08:00
|
|
|
else if (sscanf(url, "%s", request->host) == 1)
|
|
|
|
request->port = 443;
|
|
|
|
else {
|
2001-10-24 08:37:23 +08:00
|
|
|
log_message(LOG_ERR, "extract_ssl_url: Can't parse URL.");
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
safefree(request->host);
|
|
|
|
return -1;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2001-12-19 13:13:40 +08:00
|
|
|
* Send a "message" to the file descriptor provided. This handles the
|
|
|
|
* differences between the various implementations of vsnprintf. This code
|
|
|
|
* was basically stolen from the snprintf() man page of Debian Linux
|
|
|
|
* (although I did fix a memory leak. :)
|
2001-09-14 12:56:29 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
2001-12-19 13:13:40 +08:00
|
|
|
write_message(int fd, const char *fmt, ...)
|
2001-09-14 12:56:29 +08:00
|
|
|
{
|
2001-12-19 13:19:03 +08:00
|
|
|
ssize_t n;
|
|
|
|
size_t size = (1024 * 2); /* start with 2 KB and go from there */
|
2001-12-19 13:13:40 +08:00
|
|
|
char *buf, *tmpbuf;
|
|
|
|
va_list ap;
|
2001-12-17 08:11:32 +08:00
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
if ((buf = safemalloc(size)) == NULL)
|
2001-09-14 12:56:29 +08:00
|
|
|
return -1;
|
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
while (1) {
|
|
|
|
va_start(ap, fmt);
|
|
|
|
n = vsnprintf(buf, size, fmt, ap);
|
|
|
|
va_end(ap);
|
2001-12-17 08:11:32 +08:00
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
/* If that worked, break out so we can send the buffer */
|
|
|
|
if (n > -1 && n < size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
/* Else, try again with more space */
|
|
|
|
if (n > -1)
|
|
|
|
/* precisely what is needed (glibc2.1) */
|
|
|
|
size = n + 1;
|
|
|
|
else
|
|
|
|
/* twice the old size (glibc2.0) */
|
|
|
|
size *= 2;
|
|
|
|
|
|
|
|
if ((tmpbuf = saferealloc(buf, size)) == NULL) {
|
|
|
|
safefree(buf);
|
|
|
|
return -1;
|
|
|
|
} else
|
|
|
|
buf = tmpbuf;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (safe_write(fd, buf, n) < 0) {
|
|
|
|
safefree(buf);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
safefree(buf);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Create a connection for HTTP connections.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
establish_http_connection(struct conn_s *connptr, struct request_s *request)
|
|
|
|
{
|
|
|
|
if (write_message(connptr->server_fd,
|
|
|
|
"%s %s HTTP/1.0\r\n",
|
|
|
|
request->method, request->path) < 0)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (write_message(connptr->server_fd, "Host: %s\r\n", request->host) < 0)
|
|
|
|
return -1;
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-11-22 03:19:46 +08:00
|
|
|
/*
|
|
|
|
* Send the Connection header since we don't support persistant
|
|
|
|
* connections.
|
|
|
|
*/
|
|
|
|
if (safe_write(connptr->server_fd, "Connection: close\r\n", 19) < 0)
|
2001-12-19 13:13:40 +08:00
|
|
|
return -1;
|
2001-11-22 03:19:46 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-16 05:26:14 +08:00
|
|
|
/*
|
|
|
|
* These two defines are for the SSL tunnelling.
|
|
|
|
*/
|
|
|
|
#define SSL_CONNECTION_RESPONSE "HTTP/1.0 200 Connection established\r\n"
|
|
|
|
#define PROXY_AGENT "Proxy-agent: " PACKAGE "/" VERSION "\r\n"
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Send the appropriate response to the client to establish a SSL
|
|
|
|
* connection.
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static inline int
|
|
|
|
send_ssl_response(struct conn_s *connptr)
|
2001-09-16 05:26:14 +08:00
|
|
|
{
|
2001-11-22 08:31:10 +08:00
|
|
|
if (safe_write
|
|
|
|
(connptr->client_fd, SSL_CONNECTION_RESPONSE,
|
|
|
|
strlen(SSL_CONNECTION_RESPONSE)) < 0)
|
2001-09-16 05:26:14 +08:00
|
|
|
return -1;
|
|
|
|
|
2001-11-23 09:17:19 +08:00
|
|
|
if (safe_write(connptr->client_fd, PROXY_AGENT, strlen(PROXY_AGENT)) < 0)
|
2001-09-16 05:26:14 +08:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (safe_write(connptr->client_fd, "\r\n", 2) < 0)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
/*
|
|
|
|
* Break the request line apart and figure out where to connect and
|
|
|
|
* build a new request line. Finally connect to the remote server.
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static struct request_s *
|
|
|
|
process_request(struct conn_s *connptr, char *request_line)
|
2001-09-14 12:56:29 +08:00
|
|
|
{
|
|
|
|
char *url;
|
2001-09-17 04:10:19 +08:00
|
|
|
struct request_s *request;
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
size_t request_len;
|
|
|
|
|
|
|
|
/* NULL out all the fields so free's don't cause segfaults. */
|
2001-09-17 04:10:19 +08:00
|
|
|
request = safecalloc(1, sizeof(struct request_s));
|
|
|
|
if (!request)
|
|
|
|
return NULL;
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
request_len = strlen(request_line) + 1;
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
request->method = safemalloc(request_len);
|
2001-09-14 12:56:29 +08:00
|
|
|
url = safemalloc(request_len);
|
2001-09-17 04:10:19 +08:00
|
|
|
request->protocol = safemalloc(request_len);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
if (!request->method || !url || !request->protocol) {
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
|
|
|
|
|
|
|
return NULL;
|
2000-03-29 00:41:45 +08:00
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
ret =
|
|
|
|
sscanf(request_line, "%[^ ] %[^ ] %[^ ]", request->method, url,
|
|
|
|
request->protocol);
|
2001-09-14 12:56:29 +08:00
|
|
|
if (ret < 2) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"process_request: Bad Request on file descriptor %d",
|
|
|
|
connptr->client_fd);
|
2001-09-14 12:56:29 +08:00
|
|
|
httperr(connptr, 400, "Bad Request. No request found.");
|
|
|
|
|
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-12-17 08:11:32 +08:00
|
|
|
/*
|
|
|
|
* NOTE: We need to add code for the simple HTTP/0.9 style GET
|
|
|
|
* request.
|
|
|
|
*/
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
if (!url) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"process_request: Null URL on file descriptor %d",
|
|
|
|
connptr->client_fd);
|
2001-09-14 12:56:29 +08:00
|
|
|
httperr(connptr, 400, "Bad Request. Null URL.");
|
|
|
|
|
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
if (strncasecmp(url, "http://", 7) == 0) {
|
|
|
|
/* Make sure the first four characters are lowercase */
|
|
|
|
memcpy(url, "http", 4);
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
if (extract_http_url(url, request) < 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
httperr(connptr, 400,
|
|
|
|
"Bad Request. Could not parse URL.");
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-09-17 04:10:19 +08:00
|
|
|
} else if (strcmp(request->method, "CONNECT") == 0) {
|
|
|
|
if (extract_ssl_url(url, request) < 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
httperr(connptr, 400,
|
|
|
|
"Bad Request. Could not parse URL.");
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2000-04-01 06:55:22 +08:00
|
|
|
}
|
2001-12-17 08:11:32 +08:00
|
|
|
|
|
|
|
connptr->connect_method = TRUE;
|
2000-04-01 06:55:22 +08:00
|
|
|
} else {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"process_request: Unknown URL type on file descriptor %d",
|
|
|
|
connptr->client_fd);
|
2001-09-14 12:56:29 +08:00
|
|
|
httperr(connptr, 400, "Bad Request. Unknown URL type.");
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(url);
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(url);
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-14 12:56:29 +08:00
|
|
|
#ifdef FILTER_ENABLE
|
2000-09-12 08:04:42 +08:00
|
|
|
/*
|
2001-09-14 12:56:29 +08:00
|
|
|
* Filter restricted domains
|
2000-09-12 08:04:42 +08:00
|
|
|
*/
|
2001-09-14 12:56:29 +08:00
|
|
|
if (config.filter) {
|
2001-09-17 04:10:19 +08:00
|
|
|
if (filter_url(request->host)) {
|
|
|
|
update_stats(STAT_DENIED);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_NOTICE,
|
|
|
|
"Proxying refused on filtered domain \"%s\"",
|
|
|
|
request->host);
|
|
|
|
httperr(connptr, 404,
|
|
|
|
"Connection to filtered domain is now allowed.");
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-14 12:56:29 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2001-09-14 12:56:29 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-16 05:26:14 +08:00
|
|
|
/*
|
|
|
|
* Check to see if they're requesting the stat host
|
|
|
|
*/
|
2001-09-17 04:10:19 +08:00
|
|
|
if (config.stathost && strcmp(config.stathost, request->host) == 0) {
|
2001-10-24 08:37:23 +08:00
|
|
|
log_message(LOG_NOTICE, "Request for the stathost.");
|
2001-09-16 05:26:14 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
free_request_struct(request);
|
2001-09-16 05:26:14 +08:00
|
|
|
|
|
|
|
showstats(connptr);
|
2001-09-17 04:10:19 +08:00
|
|
|
return NULL;
|
2001-09-14 12:56:29 +08:00
|
|
|
}
|
|
|
|
|
2001-11-21 08:59:33 +08:00
|
|
|
/*
|
|
|
|
* Break apart the protocol and update the connection structure.
|
|
|
|
*/
|
|
|
|
if (strncasecmp(request->protocol, "http", 4) == 0) {
|
|
|
|
memcpy(request->protocol, "HTTP", 4);
|
2001-12-17 08:11:32 +08:00
|
|
|
sscanf(request->protocol, "HTTP/%u.%u",
|
2001-11-22 08:31:10 +08:00
|
|
|
&connptr->protocol.major, &connptr->protocol.minor);
|
2001-11-21 08:59:33 +08:00
|
|
|
}
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
return request;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Check to see if the line is allowed or not depending on the anonymous
|
2001-08-27 05:11:55 +08:00
|
|
|
* headers which are to be allowed. If the header is found in the
|
|
|
|
* anonymous list return 0, otherwise return -1.
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
compare_header(char *line)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2001-08-27 05:11:55 +08:00
|
|
|
char *buffer;
|
|
|
|
char *ptr;
|
2000-04-01 04:13:36 +08:00
|
|
|
int ret;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-08-31 00:51:10 +08:00
|
|
|
if ((ptr = strstr(line, ":")) == NULL)
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
2000-04-01 04:13:36 +08:00
|
|
|
|
2001-09-09 02:58:37 +08:00
|
|
|
if ((buffer = safemalloc(ptr - line + 1)) == NULL)
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
2000-04-01 04:13:36 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
memcpy(buffer, line, (size_t) (ptr - line));
|
2000-04-01 04:13:36 +08:00
|
|
|
buffer[ptr - line] = '\0';
|
|
|
|
|
2001-08-27 05:11:55 +08:00
|
|
|
ret = anonymous_search(buffer);
|
2000-09-12 08:04:42 +08:00
|
|
|
safefree(buffer);
|
2001-08-27 05:11:55 +08:00
|
|
|
|
|
|
|
return ret;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2000-09-12 08:04:42 +08:00
|
|
|
* pull_client_data is used to pull across any client data (like in a
|
|
|
|
* POST) which needs to be handled before an error can be reported, or
|
|
|
|
* server headers can be processed.
|
|
|
|
* - rjkaes
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
pull_client_data(struct conn_s *connptr, unsigned long int length)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2001-09-12 03:26:49 +08:00
|
|
|
char *buffer;
|
2001-05-27 10:29:06 +08:00
|
|
|
ssize_t len;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-12 03:26:49 +08:00
|
|
|
buffer = safemalloc(MAXBUFFSIZE);
|
|
|
|
if (!buffer)
|
|
|
|
return -1;
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
do {
|
2001-11-22 08:31:10 +08:00
|
|
|
len =
|
|
|
|
safe_read(connptr->client_fd, buffer,
|
|
|
|
min(MAXBUFFSIZE, length));
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
if (len <= 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(buffer);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (!connptr->send_response_message) {
|
2000-09-12 08:04:42 +08:00
|
|
|
if (safe_write(connptr->server_fd, buffer, len) < 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(buffer);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
length -= len;
|
|
|
|
} while (length > 0);
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(buffer);
|
2000-02-17 01:32:49 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
#ifdef XTINYPROXY_ENABLE
|
2000-02-17 01:32:49 +08:00
|
|
|
/*
|
2000-09-12 08:04:42 +08:00
|
|
|
* Add the X-Tinyproxy header to the collection of headers being sent to
|
|
|
|
* the server.
|
|
|
|
* -rjkaes
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
add_xtinyproxy_header(struct conn_s *connptr)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2000-09-12 08:04:42 +08:00
|
|
|
char ipaddr[PEER_IP_LENGTH];
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-11-22 03:19:46 +08:00
|
|
|
/*
|
|
|
|
* Don't try to send if we have an invalid server handle.
|
|
|
|
*/
|
|
|
|
if (connptr->server_fd == -1)
|
|
|
|
return 0;
|
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
return write_message(connptr->server_fd,
|
|
|
|
"X-Tinyproxy: %s\r\n",
|
|
|
|
getpeer_ip(connptr->client_fd, ipaddr));
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-11-22 08:31:10 +08:00
|
|
|
#endif /* XTINYPROXY */
|
2000-02-17 01:32:49 +08:00
|
|
|
|
|
|
|
/*
|
2000-09-12 08:04:42 +08:00
|
|
|
* Here we loop through all the headers the client is sending. If we
|
|
|
|
* are running in anonymous mode, we will _only_ send the headers listed
|
|
|
|
* (plus a few which are required for various methods).
|
|
|
|
* - rjkaes
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
process_client_headers(struct conn_s *connptr)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2001-09-12 03:26:49 +08:00
|
|
|
char *header;
|
2000-09-12 08:04:42 +08:00
|
|
|
long content_length = -1;
|
2001-11-21 08:59:33 +08:00
|
|
|
short int sent_via_header = 0;
|
2001-12-17 08:11:32 +08:00
|
|
|
ssize_t len;
|
2000-09-12 08:04:42 +08:00
|
|
|
|
2001-09-15 03:50:45 +08:00
|
|
|
static char *skipheaders[] = {
|
2001-11-22 03:19:46 +08:00
|
|
|
"host",
|
2001-12-18 13:01:03 +08:00
|
|
|
"connection",
|
2001-11-21 08:59:33 +08:00
|
|
|
"keep-alive",
|
|
|
|
"proxy-authenticate",
|
|
|
|
"proxy-authorization",
|
2001-12-18 13:01:03 +08:00
|
|
|
"te",
|
|
|
|
"trailers",
|
|
|
|
"transfer-encoding",
|
|
|
|
"upgrade"
|
2000-09-12 08:04:42 +08:00
|
|
|
};
|
|
|
|
int i;
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
for (;;) {
|
2001-12-17 08:11:32 +08:00
|
|
|
if ((len = readline(connptr->client_fd, &header)) <= 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
DEBUG2("Client (file descriptor %d) closed connection.",
|
|
|
|
connptr->client_fd);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
/*
|
|
|
|
* If we receive a CR LF (or just a LF) on a line by itself,
|
|
|
|
* the headers are finished.
|
|
|
|
*/
|
2001-12-20 12:48:32 +08:00
|
|
|
if (CHECK_CRLF(header, len))
|
2000-09-12 08:04:42 +08:00
|
|
|
break;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-15 05:16:56 +08:00
|
|
|
/*
|
2001-12-17 08:11:32 +08:00
|
|
|
* Don't send headers if there's already an error, or if
|
|
|
|
* this was a CONNECT method (unless upstream proxy is in
|
|
|
|
* use.)
|
2001-09-15 05:16:56 +08:00
|
|
|
*/
|
2001-12-17 08:11:32 +08:00
|
|
|
if (connptr->server_fd == -1
|
|
|
|
|| (connptr->connect_method && !UPSTREAM_CONFIGURED())) {
|
2001-11-22 08:19:45 +08:00
|
|
|
safefree(header);
|
2001-09-15 03:50:45 +08:00
|
|
|
continue;
|
2001-11-22 08:19:45 +08:00
|
|
|
}
|
2001-11-23 09:17:19 +08:00
|
|
|
|
2001-11-21 08:59:33 +08:00
|
|
|
/*
|
|
|
|
* If we find a Via header we need to append our information
|
|
|
|
* to the end of it.
|
|
|
|
*/
|
|
|
|
if (strncasecmp(header, "via", 3) == 0) {
|
|
|
|
char hostname[128];
|
|
|
|
char via_header_buffer[256];
|
2001-11-23 09:17:19 +08:00
|
|
|
char *new_header;
|
2001-11-21 08:59:33 +08:00
|
|
|
|
|
|
|
sent_via_header = 1;
|
|
|
|
|
|
|
|
gethostname(hostname, sizeof(hostname));
|
2001-11-22 08:31:10 +08:00
|
|
|
snprintf(via_header_buffer, sizeof(via_header_buffer),
|
|
|
|
", %hu.%hu %s (%s/%s)\r\n",
|
|
|
|
connptr->protocol.major,
|
|
|
|
connptr->protocol.minor, hostname, PACKAGE,
|
|
|
|
VERSION);
|
2001-11-21 08:59:33 +08:00
|
|
|
|
2001-11-23 09:17:19 +08:00
|
|
|
chomp(header, strlen(header));
|
2001-11-21 08:59:33 +08:00
|
|
|
|
2001-11-23 09:17:19 +08:00
|
|
|
new_header = safemalloc(strlen(header) + strlen(via_header_buffer) + 1);
|
|
|
|
strcpy(new_header, header);
|
|
|
|
strcat(new_header, via_header_buffer);
|
|
|
|
|
|
|
|
safefree(header);
|
|
|
|
|
|
|
|
header = new_header;
|
2001-11-21 08:59:33 +08:00
|
|
|
}
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
/*
|
|
|
|
* Don't send certain headers.
|
|
|
|
*/
|
|
|
|
for (i = 0; i < (sizeof(skipheaders) / sizeof(char *)); i++) {
|
2001-11-22 08:31:10 +08:00
|
|
|
if (strncasecmp
|
|
|
|
(header, skipheaders[i],
|
|
|
|
strlen(skipheaders[i])) == 0) {
|
2000-09-12 08:04:42 +08:00
|
|
|
break;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
}
|
2001-11-22 08:19:45 +08:00
|
|
|
if (i != (sizeof(skipheaders) / sizeof(char *))) {
|
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
continue;
|
2001-11-22 08:19:45 +08:00
|
|
|
}
|
2000-09-12 08:04:42 +08:00
|
|
|
|
2001-11-22 08:19:45 +08:00
|
|
|
if (is_anonymous_enabled() && compare_header(header) < 0) {
|
|
|
|
safefree(header);
|
2001-09-05 02:22:00 +08:00
|
|
|
continue;
|
2001-11-22 08:19:45 +08:00
|
|
|
}
|
2001-09-05 02:22:00 +08:00
|
|
|
|
2001-08-27 05:11:55 +08:00
|
|
|
if (content_length == -1
|
|
|
|
&& strncasecmp(header, "content-length", 14) == 0) {
|
2000-09-12 08:04:42 +08:00
|
|
|
char *content_ptr = strchr(header, ':') + 1;
|
|
|
|
content_length = atol(content_ptr);
|
|
|
|
}
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (safe_write(connptr->server_fd, header, strlen(header)) < 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
2001-09-12 03:26:49 +08:00
|
|
|
}
|
2001-11-22 08:19:45 +08:00
|
|
|
|
|
|
|
safefree(header);
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (!connptr->send_response_message
|
|
|
|
&& (!connptr->connect_method || UPSTREAM_CONFIGURED())) {
|
2000-09-12 08:04:42 +08:00
|
|
|
#ifdef XTINYPROXY_ENABLE
|
2001-11-22 08:31:10 +08:00
|
|
|
if (config.my_domain && add_xtinyproxy_header(connptr) < 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
|
|
|
}
|
2001-11-22 08:31:10 +08:00
|
|
|
#endif /* XTINYPROXY */
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (sent_via_header == 0) {
|
|
|
|
/*
|
|
|
|
* We're the first proxy so send the first Via header.
|
|
|
|
*/
|
|
|
|
char via_header_buffer[256];
|
|
|
|
char hostname[128];
|
|
|
|
|
|
|
|
gethostname(hostname, sizeof(hostname));
|
|
|
|
snprintf(via_header_buffer, sizeof(via_header_buffer),
|
|
|
|
"Via: %hu.%hu %s (%s/%s)\r\n", connptr->protocol.major,
|
|
|
|
connptr->protocol.minor, hostname, PACKAGE, VERSION);
|
|
|
|
|
|
|
|
safe_write(connptr->server_fd, via_header_buffer,
|
|
|
|
strlen(via_header_buffer));
|
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
if ((connptr->server_fd != -1)
|
|
|
|
&& safe_write(connptr->server_fd, header,
|
|
|
|
strlen(header)) < 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
2000-02-17 01:32:49 +08:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
/*
|
|
|
|
* Spin here pulling the data from the client.
|
|
|
|
*/
|
|
|
|
if (content_length >= 0)
|
2001-11-22 08:31:10 +08:00
|
|
|
return pull_client_data(connptr,
|
|
|
|
(unsigned long int) content_length);
|
2000-09-12 08:04:42 +08:00
|
|
|
else
|
|
|
|
return 0;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2000-09-12 08:04:42 +08:00
|
|
|
* Loop through all the headers (including the response code) from the
|
|
|
|
* server.
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
process_server_headers(struct conn_s *connptr)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2001-09-12 03:26:49 +08:00
|
|
|
char *header;
|
2001-12-20 04:40:23 +08:00
|
|
|
ssize_t len;
|
2001-09-12 03:26:49 +08:00
|
|
|
|
2001-12-20 04:40:23 +08:00
|
|
|
while (1) {
|
|
|
|
if ((len = readline(connptr->server_fd, &header)) <= 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
DEBUG2("Server (file descriptor %d) closed connection.",
|
|
|
|
connptr->server_fd);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-12-20 04:40:23 +08:00
|
|
|
if (safe_write(connptr->client_fd, header, len) < 0) {
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
return -1;
|
2000-03-12 04:37:44 +08:00
|
|
|
}
|
2001-11-22 08:31:10 +08:00
|
|
|
|
2001-12-20 12:48:32 +08:00
|
|
|
if (CHECK_CRLF(header, len))
|
2001-12-20 04:40:23 +08:00
|
|
|
break;
|
|
|
|
|
2001-09-12 03:26:49 +08:00
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
}
|
2001-09-12 03:26:49 +08:00
|
|
|
|
|
|
|
safefree(header);
|
2000-09-12 08:04:42 +08:00
|
|
|
return 0;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
/*
|
|
|
|
* Switch the sockets into nonblocking mode and begin relaying the bytes
|
|
|
|
* between the two connections. We continue to use the buffering code
|
|
|
|
* since we want to be able to buffer a certain amount for slower
|
|
|
|
* connections (as this was the reason why I originally modified
|
|
|
|
* tinyproxy oh so long ago...)
|
|
|
|
* - rjkaes
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static void
|
|
|
|
relay_connection(struct conn_s *connptr)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
2000-09-12 08:04:42 +08:00
|
|
|
fd_set rset, wset;
|
|
|
|
struct timeval tv;
|
|
|
|
time_t last_access;
|
|
|
|
int ret;
|
|
|
|
double tdiff;
|
2001-05-27 10:29:06 +08:00
|
|
|
int maxfd = max(connptr->client_fd, connptr->server_fd) + 1;
|
2000-09-12 08:04:42 +08:00
|
|
|
|
|
|
|
socket_nonblocking(connptr->client_fd);
|
|
|
|
socket_nonblocking(connptr->server_fd);
|
|
|
|
|
|
|
|
last_access = time(NULL);
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
for (;;) {
|
2000-09-12 08:04:42 +08:00
|
|
|
FD_ZERO(&rset);
|
|
|
|
FD_ZERO(&wset);
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
tv.tv_sec =
|
|
|
|
config.idletimeout - difftime(time(NULL), last_access);
|
2000-09-12 08:04:42 +08:00
|
|
|
tv.tv_usec = 0;
|
|
|
|
|
2001-11-05 23:23:34 +08:00
|
|
|
if (BUFFER_SIZE(connptr->sbuffer) > 0)
|
2000-09-12 08:04:42 +08:00
|
|
|
FD_SET(connptr->client_fd, &wset);
|
2001-11-05 23:23:34 +08:00
|
|
|
if (BUFFER_SIZE(connptr->cbuffer) > 0)
|
2000-09-12 08:04:42 +08:00
|
|
|
FD_SET(connptr->server_fd, &wset);
|
2001-11-05 23:23:34 +08:00
|
|
|
if (BUFFER_SIZE(connptr->sbuffer) < MAXBUFFSIZE)
|
2000-09-12 08:04:42 +08:00
|
|
|
FD_SET(connptr->server_fd, &rset);
|
2001-11-05 23:23:34 +08:00
|
|
|
if (BUFFER_SIZE(connptr->cbuffer) < MAXBUFFSIZE)
|
2000-09-12 08:04:42 +08:00
|
|
|
FD_SET(connptr->client_fd, &rset);
|
|
|
|
|
2001-05-27 10:29:06 +08:00
|
|
|
ret = select(maxfd, &rset, &wset, NULL, &tv);
|
2001-05-24 01:58:19 +08:00
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
if (ret == 0) {
|
|
|
|
tdiff = difftime(time(NULL), last_access);
|
|
|
|
if (tdiff > config.idletimeout) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_INFO,
|
|
|
|
"Idle Timeout (after select) as %g > %u.",
|
|
|
|
tdiff, config.idletimeout);
|
2000-02-17 01:32:49 +08:00
|
|
|
return;
|
2000-09-12 08:04:42 +08:00
|
|
|
} else {
|
|
|
|
continue;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2001-05-27 10:29:06 +08:00
|
|
|
} else if (ret < 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"relay_connection: select() error \"%s\". Closing connection (client_fd:%d, server_fd:%d)",
|
|
|
|
strerror(errno), connptr->client_fd,
|
|
|
|
connptr->server_fd);
|
2000-09-12 08:04:42 +08:00
|
|
|
return;
|
2001-05-27 10:29:06 +08:00
|
|
|
} else {
|
|
|
|
/*
|
|
|
|
* Okay, something was actually selected so mark it.
|
|
|
|
*/
|
|
|
|
last_access = time(NULL);
|
|
|
|
}
|
2001-11-22 08:31:10 +08:00
|
|
|
|
2001-05-27 10:29:06 +08:00
|
|
|
if (FD_ISSET(connptr->server_fd, &rset)
|
2001-12-17 08:11:32 +08:00
|
|
|
&& read_buffer(connptr->server_fd, connptr->sbuffer) < 0) {
|
2001-05-30 23:45:14 +08:00
|
|
|
break;
|
2000-09-12 08:04:42 +08:00
|
|
|
}
|
2001-05-27 10:29:06 +08:00
|
|
|
if (FD_ISSET(connptr->client_fd, &rset)
|
2001-12-17 08:11:32 +08:00
|
|
|
&& read_buffer(connptr->client_fd, connptr->cbuffer) < 0) {
|
2001-10-25 12:40:48 +08:00
|
|
|
break;
|
2000-09-12 08:04:42 +08:00
|
|
|
}
|
2001-05-27 10:29:06 +08:00
|
|
|
if (FD_ISSET(connptr->server_fd, &wset)
|
2001-12-17 08:11:32 +08:00
|
|
|
&& write_buffer(connptr->server_fd, connptr->cbuffer) < 0) {
|
2001-05-30 23:45:14 +08:00
|
|
|
break;
|
2000-09-12 08:04:42 +08:00
|
|
|
}
|
2001-05-27 10:29:06 +08:00
|
|
|
if (FD_ISSET(connptr->client_fd, &wset)
|
2001-12-17 08:11:32 +08:00
|
|
|
&& write_buffer(connptr->client_fd, connptr->sbuffer) < 0) {
|
2001-10-25 12:40:48 +08:00
|
|
|
break;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
/*
|
|
|
|
* Here the server has closed the connection... write the
|
|
|
|
* remainder to the client and then exit.
|
|
|
|
*/
|
|
|
|
socket_blocking(connptr->client_fd);
|
2001-11-05 23:23:34 +08:00
|
|
|
while (BUFFER_SIZE(connptr->sbuffer) > 0) {
|
2001-12-17 08:11:32 +08:00
|
|
|
if (write_buffer(connptr->client_fd, connptr->sbuffer) < 0)
|
2001-10-25 12:40:48 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Try to send any remaining data to the server if we can.
|
|
|
|
*/
|
|
|
|
socket_blocking(connptr->server_fd);
|
2001-11-05 23:23:34 +08:00
|
|
|
while (BUFFER_SIZE(connptr->cbuffer) > 0) {
|
2001-12-17 08:11:32 +08:00
|
|
|
if (write_buffer(connptr->client_fd, connptr->cbuffer) < 0)
|
2001-10-25 12:40:48 +08:00
|
|
|
break;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
2000-09-12 08:04:42 +08:00
|
|
|
|
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-10-20 02:03:49 +08:00
|
|
|
#ifdef UPSTREAM_SUPPORT
|
|
|
|
/*
|
|
|
|
* Establish a connection to the upstream proxy server.
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
static int
|
|
|
|
connect_to_upstream(struct conn_s *connptr, struct request_s *request)
|
2001-10-20 02:03:49 +08:00
|
|
|
{
|
2001-10-25 12:40:48 +08:00
|
|
|
char *combined_string;
|
2001-10-20 02:03:49 +08:00
|
|
|
int len;
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
connptr->server_fd =
|
|
|
|
opensock(config.upstream_name, config.upstream_port);
|
2001-10-20 02:03:49 +08:00
|
|
|
|
|
|
|
if (connptr->server_fd < 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_WARNING,
|
|
|
|
"Could not connect to upstream proxy.");
|
2001-10-20 02:03:49 +08:00
|
|
|
httperr(connptr, 404, "Unable to connect to upstream proxy.");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_CONN,
|
|
|
|
"Established connection to upstream proxy \"%s\" using file descriptor %d.",
|
|
|
|
config.upstream_name, connptr->server_fd);
|
2001-10-20 02:03:49 +08:00
|
|
|
|
2001-10-25 12:40:48 +08:00
|
|
|
/*
|
|
|
|
* We need to re-write the "path" part of the request so that we
|
|
|
|
* can reuse the establish_http_connection() function. It expects a
|
|
|
|
* method and path.
|
|
|
|
*/
|
2001-12-17 08:11:32 +08:00
|
|
|
if (connptr->connect_method) {
|
2001-10-25 12:40:48 +08:00
|
|
|
len = strlen(request->host) + 6;
|
2001-10-20 02:03:49 +08:00
|
|
|
|
2001-10-25 12:40:48 +08:00
|
|
|
combined_string = safemalloc(len + 1);
|
|
|
|
if (!combined_string) {
|
|
|
|
return -1;
|
|
|
|
}
|
2001-10-20 02:03:49 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
snprintf(combined_string, len, "%s:%d", request->host,
|
|
|
|
request->port);
|
2001-10-25 12:40:48 +08:00
|
|
|
} else {
|
|
|
|
len = strlen(request->host) + strlen(request->path) + 14;
|
|
|
|
combined_string = safemalloc(len + 1);
|
|
|
|
if (!combined_string) {
|
|
|
|
return -1;
|
|
|
|
}
|
2001-10-20 02:03:49 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
snprintf(combined_string, len, "http://%s:%d%s", request->host,
|
|
|
|
request->port, request->path);
|
2001-10-25 12:40:48 +08:00
|
|
|
}
|
2001-10-20 02:03:49 +08:00
|
|
|
|
|
|
|
safefree(request->path);
|
2001-10-25 12:40:48 +08:00
|
|
|
request->path = combined_string;
|
2001-11-22 08:31:10 +08:00
|
|
|
|
2001-10-20 02:03:49 +08:00
|
|
|
return establish_http_connection(connptr, request);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
#ifdef TUNNEL_SUPPORT
|
|
|
|
/*
|
|
|
|
* If tunnel has been configured then redirect any connections to it.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
connect_to_tunnel(struct conn_s *connptr)
|
|
|
|
{
|
|
|
|
char *request_buf;
|
|
|
|
size_t len;
|
|
|
|
int pos;
|
|
|
|
|
|
|
|
request_buf = safemalloc(HTTP_LINE_LENGTH);
|
|
|
|
if (request_buf) {
|
|
|
|
len = recv(connptr->client_fd, request_buf, HTTP_LINE_LENGTH - 1, MSG_PEEK);
|
|
|
|
for (pos = 0; pos < len && request_buf[pos] != '\n'; pos++)
|
|
|
|
;
|
|
|
|
request_buf[pos] = '\0';
|
|
|
|
|
|
|
|
log_message(LOG_CONN, "Request: %s", request_buf);
|
|
|
|
|
|
|
|
safefree(request_buf);
|
|
|
|
}
|
|
|
|
log_message(LOG_INFO, "Redirecting to %s:%d",
|
|
|
|
config.tunnel_name, config.tunnel_port);
|
|
|
|
|
|
|
|
connptr->server_fd =
|
|
|
|
opensock(config.tunnel_name, config.tunnel_port);
|
|
|
|
|
|
|
|
if (connptr->server_fd < 0) {
|
|
|
|
log_message(LOG_WARNING,
|
|
|
|
"Could not connect to tunnel.");
|
|
|
|
httperr(connptr, 404, "Unable to connect to tunnel.");
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
log_message(LOG_INFO,
|
|
|
|
"Established a connection to the tunnel \"%s\" using file descriptor %d.",
|
|
|
|
config.tunnel_name, connptr->server_fd);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2000-02-17 01:32:49 +08:00
|
|
|
/*
|
2000-09-12 08:04:42 +08:00
|
|
|
* This is the main drive for each connection. As you can tell, for the
|
|
|
|
* first few steps we are using a blocking socket. If you remember the
|
|
|
|
* older tinyproxy code, this use to be a very confusing state machine.
|
|
|
|
* Well, no more! :) The sockets are only switched into nonblocking mode
|
|
|
|
* when we start the relay portion. This makes most of the original
|
|
|
|
* tinyproxy code, which was confusing, redundant. Hail progress.
|
|
|
|
* - rjkaes
|
2000-02-17 01:32:49 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
void
|
|
|
|
handle_connection(int fd)
|
2000-02-17 01:32:49 +08:00
|
|
|
{
|
|
|
|
struct conn_s *connptr;
|
2001-10-20 02:03:49 +08:00
|
|
|
struct request_s *request = NULL;
|
2001-09-17 04:10:19 +08:00
|
|
|
|
2000-02-17 01:32:49 +08:00
|
|
|
char peer_ipaddr[PEER_IP_LENGTH];
|
2000-09-12 08:04:42 +08:00
|
|
|
char peer_string[PEER_STRING_LENGTH];
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
char *request_line = NULL;
|
2001-09-14 12:56:29 +08:00
|
|
|
|
|
|
|
log_message(LOG_CONN, "Connect (file descriptor %d): %s [%s]",
|
|
|
|
fd,
|
2001-08-27 05:11:55 +08:00
|
|
|
getpeer_string(fd, peer_string),
|
|
|
|
getpeer_ip(fd, peer_ipaddr));
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2001-09-09 02:58:37 +08:00
|
|
|
connptr = safemalloc(sizeof(struct conn_s));
|
2001-09-17 04:10:19 +08:00
|
|
|
if (!connptr)
|
2000-09-12 08:04:42 +08:00
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
initialize_conn(connptr);
|
|
|
|
connptr->client_fd = fd;
|
2000-02-17 01:32:49 +08:00
|
|
|
|
2000-09-12 08:04:42 +08:00
|
|
|
if (check_acl(fd) <= 0) {
|
|
|
|
update_stats(STAT_DENIED);
|
2001-11-22 08:31:10 +08:00
|
|
|
httperr(connptr, 403,
|
|
|
|
"You do not have authorization for using this service.");
|
2000-09-12 08:04:42 +08:00
|
|
|
goto send_error;
|
|
|
|
}
|
2001-12-18 03:10:56 +08:00
|
|
|
|
2001-12-19 13:13:40 +08:00
|
|
|
if (TUNNEL_CONFIGURED()) {
|
|
|
|
if (connect_to_tunnel(connptr) < 0)
|
2000-09-12 08:04:42 +08:00
|
|
|
goto internal_proxy;
|
2001-12-19 13:13:40 +08:00
|
|
|
else
|
|
|
|
goto relay_proxy;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
internal_proxy:
|
2001-09-14 12:56:29 +08:00
|
|
|
request_line = read_request_line(connptr);
|
|
|
|
if (!request_line) {
|
2001-09-17 04:10:19 +08:00
|
|
|
update_stats(STAT_BADCONN);
|
2000-09-12 08:04:42 +08:00
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
request = process_request(connptr, request_line);
|
2001-09-14 12:56:29 +08:00
|
|
|
safefree(request_line);
|
|
|
|
|
2001-09-17 04:10:19 +08:00
|
|
|
if (!request) {
|
2001-12-17 08:11:32 +08:00
|
|
|
if (!connptr->send_response_message) {
|
2001-10-23 00:08:29 +08:00
|
|
|
update_stats(STAT_BADCONN);
|
2001-09-17 04:10:19 +08:00
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
|
|
|
}
|
2001-10-23 00:08:29 +08:00
|
|
|
goto send_error;
|
|
|
|
}
|
2001-10-25 13:12:46 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (UPSTREAM_CONFIGURED()) {
|
2001-10-23 00:08:29 +08:00
|
|
|
if (connect_to_upstream(connptr, request) < 0)
|
|
|
|
goto send_error;
|
|
|
|
} else {
|
|
|
|
connptr->server_fd = opensock(request->host, request->port);
|
|
|
|
if (connptr->server_fd < 0) {
|
|
|
|
httperr(connptr, 500, HTTP500ERROR);
|
|
|
|
goto send_error;
|
|
|
|
}
|
2001-09-17 04:10:19 +08:00
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_CONN,
|
|
|
|
"Established connection to host \"%s\" using file descriptor %d.",
|
|
|
|
request->host, connptr->server_fd);
|
2001-10-20 02:03:49 +08:00
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (!connptr->connect_method)
|
2001-10-23 00:08:29 +08:00
|
|
|
establish_http_connection(connptr, request);
|
2001-09-17 04:10:19 +08:00
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
send_error:
|
2001-10-20 02:03:49 +08:00
|
|
|
free_request_struct(request);
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (process_client_headers(connptr) < 0) {
|
|
|
|
update_stats(STAT_BADCONN);
|
|
|
|
if (!connptr->send_response_message) {
|
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (connptr->send_response_message) {
|
2000-09-12 08:04:42 +08:00
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-12-17 08:11:32 +08:00
|
|
|
if (!connptr->connect_method || UPSTREAM_CONFIGURED()) {
|
2001-09-14 12:56:29 +08:00
|
|
|
if (process_server_headers(connptr) < 0) {
|
|
|
|
update_stats(STAT_BADCONN);
|
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
} else {
|
2001-09-16 05:26:14 +08:00
|
|
|
if (send_ssl_response(connptr) < 0) {
|
2001-11-22 08:31:10 +08:00
|
|
|
log_message(LOG_ERR,
|
|
|
|
"handle_connection: Could not send SSL greeting to client.");
|
2001-09-17 04:10:19 +08:00
|
|
|
update_stats(STAT_BADCONN);
|
2001-09-14 12:56:29 +08:00
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
|
|
|
}
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|
|
|
|
|
2001-11-22 08:31:10 +08:00
|
|
|
relay_proxy:
|
2000-09-12 08:04:42 +08:00
|
|
|
relay_connection(connptr);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* All done... close everything and go home... :)
|
|
|
|
*/
|
|
|
|
destroy_conn(connptr);
|
|
|
|
return;
|
2000-02-17 01:32:49 +08:00
|
|
|
}
|