summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/conns.c10
-rw-r--r--src/conns.h7
-rw-r--r--src/reqs.c33
-rw-r--r--src/utils.c27
-rw-r--r--src/utils.h5
5 files changed, 54 insertions, 28 deletions
diff --git a/src/conns.c b/src/conns.c
index 683e34b..227998a 100644
--- a/src/conns.c
+++ b/src/conns.c
@@ -1,4 +1,4 @@
-/* $Id: conns.c,v 1.8 2002-04-11 20:27:51 rjkaes Exp $
+/* $Id: conns.c,v 1.9 2002-04-15 02:07:27 rjkaes Exp $
*
* Create and free the connection structure. One day there could be
* other connnection related tasks put here, but for now the header
@@ -57,7 +57,10 @@ initialize_conn(int client_fd)
connptr->request_line = NULL;
- connptr->response_message_sent = FALSE;
+ /* These store any error strings */
+ connptr->error_string = NULL;
+ connptr->error_number = -1;
+
connptr->connect_method = FALSE;
connptr->protocol.major = connptr->protocol.minor = 0;
@@ -98,6 +101,9 @@ destroy_conn(struct conn_s *connptr)
if (connptr->request_line)
safefree(connptr->request_line);
+ if (connptr->error_string)
+ safefree(connptr->error_string);
+
safefree(connptr);
update_stats(STAT_CLOSE);
diff --git a/src/conns.h b/src/conns.h
index fa412fe..bc05540 100644
--- a/src/conns.h
+++ b/src/conns.h
@@ -1,4 +1,4 @@
-/* $Id: conns.h,v 1.7 2002-04-11 20:27:51 rjkaes Exp $
+/* $Id: conns.h,v 1.8 2002-04-15 02:07:27 rjkaes Exp $
*
* See 'conns.c' for a detailed description.
*
@@ -34,7 +34,10 @@ struct conn_s {
char *request_line;
bool_t connect_method;
- bool_t response_message_sent;
+
+ /* Store the error response if there is one */
+ char *error_string;
+ int error_number;
/* A Content-Length value from the remote server */
long remote_content_length;
diff --git a/src/reqs.c b/src/reqs.c
index c57184a..154298b 100644
--- a/src/reqs.c
+++ b/src/reqs.c
@@ -1,4 +1,4 @@
-/* $Id: reqs.c,v 1.57 2002-04-12 17:00:42 rjkaes Exp $
+/* $Id: reqs.c,v 1.58 2002-04-15 02:07:27 rjkaes Exp $
*
* This is where all the work in tinyproxy is actually done. Incoming
* connections have a new thread created for them. The thread then
@@ -348,7 +348,7 @@ process_request(struct conn_s *connptr)
log_message(LOG_ERR,
"process_request: Bad Request on file descriptor %d",
connptr->client_fd);
- httperr(connptr, 400, "Bad Request. No request found.");
+ indicate_http_error(connptr, 400, "Bad Request. No request found.");
safefree(url);
free_request_struct(request);
@@ -364,7 +364,7 @@ process_request(struct conn_s *connptr)
log_message(LOG_ERR,
"process_request: Null URL on file descriptor %d",
connptr->client_fd);
- httperr(connptr, 400, "Bad Request. Null URL.");
+ indicate_http_error(connptr, 400, "Bad Request. Null URL.");
safefree(url);
free_request_struct(request);
@@ -377,7 +377,7 @@ process_request(struct conn_s *connptr)
memcpy(url, "http", 4);
if (extract_http_url(url, request) < 0) {
- httperr(connptr, 400,
+ indicate_http_error(connptr, 400,
"Bad Request. Could not parse URL.");
safefree(url);
@@ -387,7 +387,7 @@ process_request(struct conn_s *connptr)
}
} else if (strcmp(request->method, "CONNECT") == 0) {
if (extract_ssl_url(url, request) < 0) {
- httperr(connptr, 400,
+ indicate_http_error(connptr, 400,
"Bad Request. Could not parse URL.");
safefree(url);
@@ -398,7 +398,7 @@ process_request(struct conn_s *connptr)
/* Verify that the port in the CONNECT method is allowed */
if (check_allowed_connect_ports(request->port) <= 0) {
- httperr(connptr, 403,
+ indicate_http_error(connptr, 403,
"CONNECT method not allowed with selected port.");
log_message(LOG_INFO, "Refused CONNECT method on port %d",
request->port);
@@ -414,7 +414,7 @@ process_request(struct conn_s *connptr)
log_message(LOG_ERR,
"process_request: Unknown URL type on file descriptor %d",
connptr->client_fd);
- httperr(connptr, 400, "Bad Request. Unknown URL type.");
+ indicate_http_error(connptr, 400, "Bad Request. Unknown URL type.");
safefree(url);
free_request_struct(request);
@@ -435,7 +435,7 @@ process_request(struct conn_s *connptr)
log_message(LOG_NOTICE,
"Proxying refused on filtered domain \"%s\"",
request->host);
- httperr(connptr, 404,
+ indicate_http_error(connptr, 404,
"Connection to filtered domain is now allowed.");
free_request_struct(request);
@@ -495,7 +495,7 @@ pull_client_data(struct conn_s *connptr, unsigned long int length)
return -1;
}
- if (!connptr->response_message_sent) {
+ if (!connptr->error_string) {
if (safe_write(connptr->server_fd, buffer, len) < 0) {
safefree(buffer);
return -1;
@@ -1022,7 +1022,7 @@ connect_to_upstream(struct conn_s *connptr, struct request_s *request)
if (connptr->server_fd < 0) {
log_message(LOG_WARNING,
"Could not connect to upstream proxy.");
- httperr(connptr, 404, "Unable to connect to upstream proxy.");
+ indicate_http_error(connptr, 404, "Unable to connect to upstream proxy.");
return -1;
}
@@ -1094,7 +1094,7 @@ connect_to_tunnel(struct conn_s *connptr)
if (connptr->server_fd < 0) {
log_message(LOG_WARNING,
"Could not connect to tunnel.");
- httperr(connptr, 404, "Unable to connect to tunnel.");
+ indicate_http_error(connptr, 404, "Unable to connect to tunnel.");
return -1;
}
@@ -1136,7 +1136,7 @@ handle_connection(int fd)
if (check_acl(fd) <= 0) {
update_stats(STAT_DENIED);
- httperr(connptr, 403,
+ indicate_http_error(connptr, 403,
"You do not have authorization for using this service.");
goto send_error;
}
@@ -1157,7 +1157,7 @@ handle_connection(int fd)
request = process_request(connptr);
if (!request) {
- if (!connptr->response_message_sent) {
+ if (!connptr->error_string) {
update_stats(STAT_BADCONN);
destroy_conn(connptr);
return;
@@ -1171,7 +1171,7 @@ handle_connection(int fd)
} else {
connptr->server_fd = opensock(request->host, request->port);
if (connptr->server_fd < 0) {
- httperr(connptr, 500, HTTP500ERROR);
+ indicate_http_error(connptr, 500, HTTP500ERROR);
goto send_error;
}
@@ -1188,13 +1188,14 @@ handle_connection(int fd)
if (process_client_headers(connptr) < 0) {
update_stats(STAT_BADCONN);
- if (!connptr->response_message_sent) {
+ if (!connptr->error_string) {
destroy_conn(connptr);
return;
}
}
- if (connptr->response_message_sent) {
+ if (connptr->error_string) {
+ send_http_error_message(connptr);
destroy_conn(connptr);
return;
}
diff --git a/src/utils.c b/src/utils.c
index 8d274e2..d99abaf 100644
--- a/src/utils.c
+++ b/src/utils.c
@@ -1,4 +1,4 @@
-/* $Id: utils.c,v 1.22 2002-04-07 21:37:07 rjkaes Exp $
+/* $Id: utils.c,v 1.23 2002-04-15 02:07:27 rjkaes Exp $
*
* Misc. routines which are used by the various functions to handle strings
* and memory allocation and pretty much anything else we can think of. Also,
@@ -100,8 +100,6 @@ send_http_message(struct conn_s *connptr, int http_code,
safe_write(connptr->client_fd, message, strlen(message));
- connptr->response_message_sent = TRUE;
-
return 0;
}
@@ -109,7 +107,7 @@ send_http_message(struct conn_s *connptr, int http_code,
* Display an error to the client.
*/
int
-httperr(struct conn_s *connptr, int err, const char *msg)
+send_http_error_message(struct conn_s *connptr)
{
static char *message = \
"<html><head><title>%s</title></head>\r\n" \
@@ -135,7 +133,9 @@ httperr(struct conn_s *connptr, int err, const char *msg)
* See the write_message() function in sock.c for more information.
*/
while (1) {
- n = snprintf(message_buffer, size, message, msg, err, msg, PACKAGE, VERSION);
+ n = snprintf(message_buffer, size, message,
+ connptr->error_string, connptr->error_number,
+ connptr->error_string, PACKAGE, VERSION);
if (n > -1 && n < size)
break;
@@ -152,11 +152,26 @@ httperr(struct conn_s *connptr, int err, const char *msg)
message_buffer = tmpbuf;
}
- ret = send_http_message(connptr, err, msg, message_buffer);
+ ret = send_http_message(connptr, connptr->error_number,
+ connptr->error_string, message_buffer);
safefree(message_buffer);
return ret;
}
+/*
+ * Add the error information to the conn structure.
+ */
+int
+indicate_http_error(struct conn_s* connptr, int number, const char* string)
+{
+ connptr->error_string = strdup(string);
+ if (!connptr->error_string)
+ return -1;
+ connptr->error_number = number;
+
+ return 0;
+}
+
void
makedaemon(void)
{
diff --git a/src/utils.h b/src/utils.h
index a401201..1e19bad 100644
--- a/src/utils.h
+++ b/src/utils.h
@@ -1,4 +1,4 @@
-/* $Id: utils.h,v 1.13 2001-11-25 02:22:05 rjkaes Exp $
+/* $Id: utils.h,v 1.14 2002-04-15 02:07:27 rjkaes Exp $
*
* See 'utils.h' for a detailed description.
*
@@ -32,7 +32,8 @@
extern int send_http_message(struct conn_s *connptr, int http_code,
const char *error_title, const char *message);
-extern int httperr(struct conn_s *connptr, int err, const char *msg);
+extern int send_http_error_message(struct conn_s *connptr);
+extern int indicate_http_error(struct conn_s* connptr, int number, const char *string);
extern void makedaemon(void);
extern void pidfile_create(const char *path);