summaryrefslogtreecommitdiffhomepage
path: root/src/conns.c
blob: 19aaa49c78bdb60ca6f9ee28474b5f30836788b6 (plain)
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
128
/* 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);
}