129 lines
4.1 KiB
C
129 lines
4.1 KiB
C
/* tinyproxy - A fast light-weight HTTP proxy
|
|
* Copyright (C) 2001 Robert James Kaes <rjkaes@users.sourceforge.net>
|
|
*
|
|
* 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.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/* Create and free the connection structure. One day there could be
|
|
* other connection related tasks put here, but for now the header
|
|
* file and this file are only used for create/free functions and the
|
|
* connection structure definition.
|
|
*/
|
|
|
|
#include "main.h"
|
|
|
|
#include "buffer.h"
|
|
#include "conns.h"
|
|
#include "heap.h"
|
|
#include "log.h"
|
|
#include "stats.h"
|
|
|
|
void conn_struct_init(struct conn_s *connptr) {
|
|
connptr->error_number = -1;
|
|
connptr->client_fd = -1;
|
|
connptr->server_fd = -1;
|
|
/* There is _no_ content length initially */
|
|
connptr->content_length.server = connptr->content_length.client = -1;
|
|
}
|
|
|
|
int conn_init_contents (struct conn_s *connptr, const char *ipaddr,
|
|
const char *sock_ipaddr)
|
|
{
|
|
struct buffer_s *cbuffer, *sbuffer;
|
|
|
|
assert (connptr->client_fd >= 0);
|
|
|
|
/*
|
|
* Allocate the memory for all the internal components
|
|
*/
|
|
cbuffer = new_buffer ();
|
|
sbuffer = new_buffer ();
|
|
|
|
if (!cbuffer || !sbuffer)
|
|
goto error_exit;
|
|
|
|
connptr->cbuffer = cbuffer;
|
|
connptr->sbuffer = sbuffer;
|
|
|
|
connptr->server_ip_addr = (sock_ipaddr ?
|
|
safestrdup (sock_ipaddr) : NULL);
|
|
connptr->client_ip_addr = safestrdup (ipaddr);
|
|
|
|
update_stats (STAT_OPEN);
|
|
|
|
return 1;
|
|
|
|
error_exit:
|
|
/*
|
|
* If we got here, there was a problem allocating memory
|
|
*/
|
|
if (cbuffer)
|
|
delete_buffer (cbuffer);
|
|
if (sbuffer)
|
|
delete_buffer (sbuffer);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void conn_destroy_contents (struct conn_s *connptr)
|
|
{
|
|
assert (connptr != NULL);
|
|
|
|
if (connptr->client_fd != -1)
|
|
if (close (connptr->client_fd) < 0)
|
|
log_message (LOG_INFO, "Client (%d) close message: %s",
|
|
connptr->client_fd, strerror (errno));
|
|
connptr->client_fd = -1;
|
|
if (connptr->server_fd != -1)
|
|
if (close (connptr->server_fd) < 0)
|
|
log_message (LOG_INFO, "Server (%d) close message: %s",
|
|
connptr->server_fd, strerror (errno));
|
|
connptr->server_fd = -1;
|
|
|
|
if (connptr->cbuffer)
|
|
delete_buffer (connptr->cbuffer);
|
|
if (connptr->sbuffer)
|
|
delete_buffer (connptr->sbuffer);
|
|
|
|
if (connptr->request_line)
|
|
safefree (connptr->request_line);
|
|
|
|
if (connptr->error_variables) {
|
|
char *k;
|
|
htab_value *v;
|
|
size_t it = 0;
|
|
while((it = htab_next(connptr->error_variables, it, &k, &v))) {
|
|
safefree(v->p);
|
|
safefree(k);
|
|
}
|
|
htab_destroy (connptr->error_variables);
|
|
}
|
|
|
|
if (connptr->error_string)
|
|
safefree (connptr->error_string);
|
|
|
|
if (connptr->server_ip_addr)
|
|
safefree (connptr->server_ip_addr);
|
|
if (connptr->client_ip_addr)
|
|
safefree (connptr->client_ip_addr);
|
|
|
|
#ifdef REVERSE_SUPPORT
|
|
if (connptr->reversepath)
|
|
safefree (connptr->reversepath);
|
|
#endif
|
|
|
|
update_stats (STAT_CLOSE);
|
|
}
|