summaryrefslogtreecommitdiffstats
path: root/src/net/peerdist.c
blob: 3210ac0ec0b970090b43967ade01b4edca0b2ef8 (plain) (blame)
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
171
172
173
174
175
176
177
178
179
180
181
182
183
/*
 * Copyright (C) 2015 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * 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.
 *
 * You can also choose to distribute this program under the terms of
 * the Unmodified Binary Distribution Licence (as given in the file
 * COPYING.UBDL), provided that you have satisfied its requirements.
 */

FILE_LICENCE ( GPL2_OR_LATER_OR_UBDL );

#include <stdio.h>
#include <ipxe/http.h>
#include <ipxe/settings.h>
#include <ipxe/peermux.h>

/** @file
 *
 * Peer Content Caching and Retrieval (PeerDist) protocol
 *
 * This is quite possibly the ugliest protocol I have ever had the
 * misfortune to encounter, and I've encountered multicast TFTP.
 */

/** PeerDist is globally enabled */
static long peerdist_enabled = 1;

/**
 * Check whether or not to support PeerDist encoding for this request
 *
 * @v http		HTTP transaction
 * @ret supported	PeerDist encoding is supported for this request
 */
static int http_peerdist_supported ( struct http_transaction *http ) {

	/* Allow PeerDist to be globally enabled/disabled */
	if ( ! peerdist_enabled )
		return 0;

	/* Support PeerDist encoding only if we can directly access an
	 * underlying data transfer buffer.  Direct access is required
	 * in order to support decryption of data received via the
	 * retrieval protocol (which provides the AES initialisation
	 * vector only after all of the encrypted data has been
	 * received).
	 *
	 * This test simultaneously ensures that we do not attempt to
	 * use PeerDist encoding on a request which is itself a
	 * PeerDist individual block download, since the individual
	 * block downloads do not themselves provide direct access to
	 * an underlying data transfer buffer.
	 */
	return ( xfer_buffer ( &http->xfer ) != NULL );
}

/**
 * Format HTTP "X-P2P-PeerDist" header
 *
 * @v http		HTTP transaction
 * @v buf		Buffer
 * @v len		Length of buffer
 * @ret len		Length of header value, or negative error
 */
static int http_format_p2p_peerdist ( struct http_transaction *http,
				      char *buf, size_t len ) {
	int supported = http_peerdist_supported ( http );
	int missing;

	/* PeerDist wants us to inform the server whenever we make a
	 * request for data that was missing from local peers
	 * (presumably for statistical purposes only).  We use the
	 * heuristic of assuming that the combination of "this request
	 * may not itself use PeerDist content encoding" and "this is
	 * a range request" probably indicates that we are making a
	 * PeerDist block raw range request for missing data.
	 */
	missing = ( http->request.range.len && ( ! supported ) );

	/* Omit header if PeerDist encoding is not supported and we
	 * are not reporting a missing data request.
	 */
	if ( ! ( supported || missing ) )
		return 0;

	/* Construct header */
	return snprintf ( buf, len, "Version=1.1%s",
			  ( missing ? ", MissingDataRequest=true" : "" ) );
}

/** HTTP "X-P2P-PeerDist" header */
struct http_request_header http_request_p2p_peerdist __http_request_header = {
	.name = "X-P2P-PeerDist",
	.format = http_format_p2p_peerdist,
};

/**
 * Format HTTP "X-P2P-PeerDistEx" header
 *
 * @v http		HTTP transaction
 * @v buf		Buffer
 * @v len		Length of buffer
 * @ret len		Length of header value, or negative error
 */
static int http_format_p2p_peerdistex ( struct http_transaction *http,
					char *buf, size_t len ) {
	int supported = http_peerdist_supported ( http );

	/* Omit header if PeerDist encoding is not supported */
	if ( ! supported )
		return 0;

	/* Construct header */
	return snprintf ( buf, len, ( "MinContentInformation=1.0, "
				      "MaxContentInformation=2.0" ) );
}

/** HTTP "X-P2P-PeerDist" header */
struct http_request_header http_request_p2p_peerdistex __http_request_header = {
	.name = "X-P2P-PeerDistEx",
	.format = http_format_p2p_peerdistex,
};

/**
 * Initialise PeerDist content encoding
 *
 * @v http		HTTP transaction
 * @ret rc		Return status code
 */
static int http_peerdist_init ( struct http_transaction *http ) {

	return peermux_filter ( &http->content, &http->transfer, http->uri );
}

/** PeerDist HTTP content encoding */
struct http_content_encoding peerdist_encoding __http_content_encoding = {
	.name = "peerdist",
	.supported = http_peerdist_supported,
	.init = http_peerdist_init,
};

/** PeerDist enabled setting */
const struct setting peerdist_setting __setting ( SETTING_MISC, peerdist ) = {
	.name = "peerdist",
	.description = "PeerDist enabled",
	.type = &setting_type_int8,
};

/**
 * Apply PeerDist settings
 *
 * @ret rc		Return status code
 */
static int apply_peerdist_settings ( void ) {

	/* Fetch global PeerDist enabled setting */
	if ( fetch_int_setting ( NULL, &peerdist_setting,
				 &peerdist_enabled ) < 0 ) {
		peerdist_enabled = 1;
	}
	DBGC ( &peerdist_enabled, "PEERDIST is %s\n",
	       ( peerdist_enabled ? "enabled" : "disabled" ) );

	return 0;
}

/** PeerDist settings applicator */
struct settings_applicator peerdist_applicator __settings_applicator = {
	.apply = apply_peerdist_settings,
};