2 Unix SMB/CIFS implementation.
4 transport layer security handling code
6 Copyright (C) Andrew Tridgell 2005
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 2 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program; if not, write to the Free Software
20 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "lib/events/events.h"
25 #include "lib/socket/socket.h"
26 #include "lib/tls/tls.h"
29 #include "gnutls/gnutls.h"
33 /* hold persistent tls data */
35 gnutls_certificate_credentials x509_cred;
36 gnutls_dh_params dh_params;
40 /* hold per connection tls data */
42 struct tls_params *params;
43 struct socket_context *socket;
45 gnutls_session session;
51 const char *plain_chars;
57 callback for reading from a socket
59 static ssize_t tls_pull(gnutls_transport_ptr ptr, void *buf, size_t size)
61 struct tls_context *tls = talloc_get_type(ptr, struct tls_context);
65 if (tls->have_first_byte) {
66 *(uint8_t *)buf = tls->first_byte;
67 tls->have_first_byte = False;
71 status = socket_recv(tls->socket, buf, size, &nread, 0);
72 if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) {
75 if (NT_STATUS_IS_ERR(status)) {
76 EVENT_FD_NOT_READABLE(tls->fde);
77 EVENT_FD_NOT_WRITEABLE(tls->fde);
81 if (!NT_STATUS_IS_OK(status)) {
82 EVENT_FD_READABLE(tls->fde);
83 EVENT_FD_NOT_WRITEABLE(tls->fde);
87 if (tls->output_pending) {
88 EVENT_FD_WRITEABLE(tls->fde);
91 EVENT_FD_READABLE(tls->fde);
97 callback for writing to a socket
99 static ssize_t tls_push(gnutls_transport_ptr ptr, const void *buf, size_t size)
101 struct tls_context *tls = talloc_get_type(ptr, struct tls_context);
106 if (!tls->tls_enabled) {
110 b.data = discard_const(buf);
113 status = socket_send(tls->socket, &b, &nwritten, 0);
114 if (!NT_STATUS_IS_OK(status)) {
115 EVENT_FD_WRITEABLE(tls->fde);
118 if (size != nwritten) {
119 EVENT_FD_WRITEABLE(tls->fde);
125 destroy a tls session
127 static int tls_destructor(void *ptr)
129 struct tls_context *tls = talloc_get_type(ptr, struct tls_context);
131 ret = gnutls_bye(tls->session, GNUTLS_SHUT_WR);
133 DEBUG(0,("TLS gnutls_bye failed - %s\n", gnutls_strerror(ret)));
140 possibly continue the handshake process
142 static NTSTATUS tls_handshake(struct tls_context *tls)
146 if (tls->done_handshake) {
150 ret = gnutls_handshake(tls->session);
151 if (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN) {
152 return STATUS_MORE_ENTRIES;
155 DEBUG(0,("TLS gnutls_handshake failed - %s\n", gnutls_strerror(ret)));
156 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
158 tls->done_handshake = True;
163 see how many bytes are pending on the connection
165 NTSTATUS tls_socket_pending(struct tls_context *tls, size_t *npending)
167 if (!tls->tls_enabled || tls->tls_detect) {
168 return socket_pending(tls->socket, npending);
170 *npending = gnutls_record_check_pending(tls->session);
171 if (*npending == 0) {
172 return socket_pending(tls->socket, npending);
178 receive data either by tls or normal socket_recv
180 NTSTATUS tls_socket_recv(struct tls_context *tls, void *buf, size_t wantlen,
185 if (tls->tls_enabled && tls->tls_detect) {
186 status = socket_recv(tls->socket, &tls->first_byte, 1, nread, 0);
187 NT_STATUS_NOT_OK_RETURN(status);
188 if (*nread == 0) return NT_STATUS_OK;
189 tls->tls_detect = False;
190 /* look for the first byte of a valid HTTP operation */
191 if (strchr(tls->plain_chars, tls->first_byte)) {
193 tls->tls_enabled = False;
194 *(uint8_t *)buf = tls->first_byte;
197 tls->have_first_byte = True;
200 if (!tls->tls_enabled) {
201 return socket_recv(tls->socket, buf, wantlen, nread, 0);
204 status = tls_handshake(tls);
205 NT_STATUS_NOT_OK_RETURN(status);
207 ret = gnutls_record_recv(tls->session, buf, wantlen);
208 if (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN) {
209 return STATUS_MORE_ENTRIES;
212 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
220 send data either by tls or normal socket_recv
222 NTSTATUS tls_socket_send(struct tls_context *tls, const DATA_BLOB *blob, size_t *sendlen)
227 if (!tls->tls_enabled) {
228 return socket_send(tls->socket, blob, sendlen, 0);
231 status = tls_handshake(tls);
232 NT_STATUS_NOT_OK_RETURN(status);
234 ret = gnutls_record_send(tls->session, blob->data, blob->length);
235 if (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN) {
236 return STATUS_MORE_ENTRIES;
239 DEBUG(0,("gnutls_record_send of %d failed - %s\n", blob->length, gnutls_strerror(ret)));
240 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
243 tls->output_pending = (ret < blob->length);
249 initialise global tls state
251 struct tls_params *tls_initialise(TALLOC_CTX *mem_ctx)
253 struct tls_params *params;
255 const char *keyfile = lp_tls_keyfile();
256 const char *certfile = lp_tls_certfile();
257 const char *cafile = lp_tls_cafile();
258 const char *crlfile = lp_tls_crlfile();
259 void tls_cert_generate(TALLOC_CTX *, const char *, const char *, const char *);
261 params = talloc(mem_ctx, struct tls_params);
262 if (params == NULL) return NULL;
264 if (!lp_tls_enabled() || keyfile == NULL || *keyfile == 0) {
265 params->tls_enabled = False;
269 if (!file_exist(cafile)) {
270 tls_cert_generate(params, keyfile, certfile, cafile);
273 ret = gnutls_global_init();
274 if (ret < 0) goto init_failed;
276 gnutls_certificate_allocate_credentials(¶ms->x509_cred);
277 if (ret < 0) goto init_failed;
279 if (cafile && *cafile) {
280 ret = gnutls_certificate_set_x509_trust_file(params->x509_cred, cafile,
281 GNUTLS_X509_FMT_PEM);
283 DEBUG(0,("TLS failed to initialise cafile %s\n", cafile));
288 if (crlfile && *crlfile) {
289 ret = gnutls_certificate_set_x509_crl_file(params->x509_cred,
291 GNUTLS_X509_FMT_PEM);
293 DEBUG(0,("TLS failed to initialise crlfile %s\n", crlfile));
298 ret = gnutls_certificate_set_x509_key_file(params->x509_cred,
300 GNUTLS_X509_FMT_PEM);
302 DEBUG(0,("TLS failed to initialise certfile %s and keyfile %s\n",
307 ret = gnutls_dh_params_init(¶ms->dh_params);
308 if (ret < 0) goto init_failed;
310 ret = gnutls_dh_params_generate2(params->dh_params, DH_BITS);
311 if (ret < 0) goto init_failed;
313 gnutls_certificate_set_dh_params(params->x509_cred, params->dh_params);
315 params->tls_enabled = True;
319 DEBUG(0,("GNUTLS failed to initialise - %s\n", gnutls_strerror(ret)));
320 params->tls_enabled = False;
326 setup for a new connection
328 struct tls_context *tls_init_server(struct tls_params *params,
329 struct socket_context *socket,
330 struct fd_event *fde,
331 const char *plain_chars)
333 struct tls_context *tls;
336 tls = talloc(socket, struct tls_context);
337 if (tls == NULL) return NULL;
339 tls->socket = socket;
342 if (!params->tls_enabled) {
343 tls->tls_enabled = False;
347 #define TLSCHECK(call) do { \
350 DEBUG(0,("TLS %s - %s\n", #call, gnutls_strerror(ret))); \
355 TLSCHECK(gnutls_init(&tls->session, GNUTLS_SERVER));
357 talloc_set_destructor(tls, tls_destructor);
359 TLSCHECK(gnutls_set_default_priority(tls->session));
360 TLSCHECK(gnutls_credentials_set(tls->session, GNUTLS_CRD_CERTIFICATE,
362 gnutls_certificate_server_set_request(tls->session, GNUTLS_CERT_REQUEST);
363 gnutls_dh_set_prime_bits(tls->session, DH_BITS);
364 gnutls_transport_set_ptr(tls->session, (gnutls_transport_ptr)tls);
365 gnutls_transport_set_pull_function(tls->session, (gnutls_pull_func)tls_pull);
366 gnutls_transport_set_push_function(tls->session, (gnutls_push_func)tls_push);
367 gnutls_transport_set_lowat(tls->session, 0);
369 tls->plain_chars = plain_chars;
371 tls->tls_detect = True;
373 tls->tls_detect = False;
376 tls->output_pending = False;
377 tls->params = params;
378 tls->done_handshake = False;
379 tls->have_first_byte = False;
380 tls->tls_enabled = True;
385 DEBUG(0,("TLS init connection failed - %s\n", gnutls_strerror(ret)));
386 tls->tls_enabled = False;
387 params->tls_enabled = False;
391 BOOL tls_enabled(struct tls_context *tls)
393 return tls->tls_enabled;
396 BOOL tls_support(struct tls_params *params)
398 return params->tls_enabled;
404 /* for systems without tls we just map the tls socket calls to the
405 normal socket calls */
407 struct tls_params *tls_initialise(TALLOC_CTX *mem_ctx)
409 return talloc_new(mem_ctx);
412 struct tls_context *tls_init_server(struct tls_params *params,
413 struct socket_context *sock,
414 struct fd_event *fde,
415 const char *plain_chars)
417 if (plain_chars == NULL) return NULL;
418 return (struct tls_context *)sock;
422 NTSTATUS tls_socket_recv(struct tls_context *tls, void *buf, size_t wantlen,
425 return socket_recv((struct socket_context *)tls, buf, wantlen, nread, 0);
428 NTSTATUS tls_socket_send(struct tls_context *tls, const DATA_BLOB *blob, size_t *sendlen)
430 return socket_send((struct socket_context *)tls, blob, sendlen, 0);
433 BOOL tls_enabled(struct tls_context *tls)
438 BOOL tls_support(struct tls_params *params)
443 NTSTATUS tls_socket_pending(struct tls_context *tls, size_t *npending)
445 return socket_pending((struct socket_context *)tls, npending);