summaryrefslogtreecommitdiff
path: root/src/lib/pac.c
blob: 9403d3974895967add63e3a8b53d9a68a1e8f16a (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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
/*******************************************************************************
 * libproxy - A library for proxy configuration
 * Copyright (C) 2006 Nathaniel McCallum <nathaniel@natemccallum.com>
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 ******************************************************************************/

#include <unistd.h>
#include <string.h>
#include <time.h>
#include <sys/types.h>
#include <sys/socket.h>

#include "url.h"
#include "misc.h"
#include "pac.h"

#define PAC_MIME_TYPE "application/x-ns-proxy-autoconfig"

/**
 * ProxyAutoConfig object.  All fields are private.
 */
struct _pxPAC {
	pxURL     *url;
	char      *cache;
	time_t     expires;
};

/**
 * Frees memory used by the ProxyAutoConfig.
 */
void
px_pac_free(pxPAC *self)
{
	if (!self) return;
	px_url_free(self->url);
	px_free(self->cache);
	px_free(self);
}

/**
 * Get the URL which the pxPAC uses.
 * @return The URL that the pxPAC came from
 */
const pxURL *
px_pac_get_url(pxPAC *self)
{
	return self->url;
}

/**
 * Create a new ProxyAutoConfig.
 * @url The URL where the PAC file is found
 * @return A new ProxyAutoConfig or NULL on error
 */
pxPAC *
px_pac_new(pxURL *url)
{
	if (!url) return NULL;
	
	// Allocate the object
	pxPAC *self = px_malloc0(sizeof(pxPAC));
	
	// Copy the given URL
	self->url = px_url_new(px_url_to_string(url)); // Always returns valid value
		
	// Make sure we have a real pxPAC
	if (!px_pac_reload(self)) { px_pac_free(self); return NULL; }
	
	return self;
}

/**
 * Create a new ProxyAutoConfig (from a string for convenience).
 * @url The url (string) where the PAC file is found
 * @return A new ProxyAutoConfig or NULL on error
 */
pxPAC *
px_pac_new_from_string(char *url)
{
	// Create temporary URL
	pxURL *tmpurl = px_url_new(url);
	if (!tmpurl) return NULL;
	
	// Create pac
	pxPAC *self = px_pac_new(tmpurl);
	px_url_free(tmpurl); // Free no longer used URL
	if (!self) return NULL;
	return self;
}

/**
 * Returns the Javascript code which the pxPAC uses.
 * @return The Javascript code used by the pxPAC
 */
const char *
px_pac_to_string(pxPAC *self)
{
	return self->cache;
}

/**
 * Download the latest version of the pxPAC file
 * @return Whether the download was successful or not
 */
bool
px_pac_reload(pxPAC *self)
{
	int sock;
	char *line = NULL;
	const char *headers[3] = { "Accept: " PAC_MIME_TYPE, "Connection: close", NULL };
	bool correct_mime_type;
	unsigned long int content_length = 0;
	
	// Get the pxPAC
	sock = px_url_get(self->url, headers);
	if (sock < 0) return false;

	// Verify status line
	line = px_readline(sock);
	if (!line)                                                    goto error;
	if (strncmp(line, "HTTP", strlen("HTTP")))                    goto error; // Check valid HTTP response
	if (!strchr(line, ' ') || atoi(strchr(line, ' ') + 1) != 200) goto error; // Check status code
	
	// Check for correct mime type and content length
	while (strcmp(line, "\r")) {
		// Check for content type
		if (strstr(line, "Content-Type: ") == line && strstr(line, PAC_MIME_TYPE))
			correct_mime_type = true;
			
		// Check for content length
		else if (strstr(line, "Content-Length: ") == line)
			content_length = atoi(line + strlen("Content-Length: "));
		
		// Get new line
		px_free(line);
		line = px_readline(sock);
		if (!line) goto error;
	}
	
	// Get content
	if (!content_length || !correct_mime_type) goto error;
	px_free(line); line = NULL;
	px_free(self->cache);
	self->cache = px_malloc0(content_length+1);
	for (int recvd=0 ; recvd != content_length ; )
		recvd += recv(sock, self->cache + recvd, content_length - recvd, 0);
	
	// Clean up
	close(sock);
	return true;
	
	error:
		px_free(self->cache); self->cache = NULL;
		if (sock >= 0) close(sock);
		px_free(line);
		return false;
}