set talloc parent to sock for thread safety
[freeradius.git] / src / main / tls_listen.c
index 2322177..32cf564 100644 (file)
@@ -33,6 +33,7 @@ USES_APPLE_DEPRECATED_API     /* OpenSSL API has been deprecated by Apple */
 #include <sys/stat.h>
 #endif
 
+#ifdef WITH_TCP
 #ifdef WITH_TLS
 #ifdef HAVE_OPENSSL_RAND_H
 #include <openssl/rand.h>
@@ -54,7 +55,7 @@ static void dump_hex(char const *msg, uint8_t const *data, size_t data_len)
 {
        size_t i;
 
-       if (debug_flag < 3) return;
+       if (rad_debug_lvl < 3) return;
 
        printf("%s %d\n", msg, (int) data_len);
        if (data_len > 256) data_len = 256;
@@ -72,26 +73,16 @@ static void tls_socket_close(rad_listen_t *listener)
 {
        listen_socket_t *sock = listener->data;
 
-       listener->status = RAD_LISTEN_STATUS_REMOVE_NOW;
-       listener->tls = NULL; /* parent owns this! */
+       SSL_shutdown(sock->ssn->ssl);
 
-       if (sock->parent) {
-               /*
-                *      Decrement the number of connections.
-                */
-               if (sock->parent->limit.num_connections > 0) {
-                       sock->parent->limit.num_connections--;
-               }
-               if (sock->client->limit.num_connections > 0) {
-                       sock->client->limit.num_connections--;
-               }
-       }
+       listener->status = RAD_LISTEN_STATUS_EOL;
+       listener->tls = NULL; /* parent owns this! */
 
        /*
         *      Tell the event handler that an FD has disappeared.
         */
        DEBUG("Client has closed connection");
-       event_new_fd(listener);
+       radius_update_listener(listener);
 
        /*
         *      Do NOT free the listener here.  It's in use by
@@ -102,7 +93,7 @@ static void tls_socket_close(rad_listen_t *listener)
         */
 }
 
-static int tls_socket_write(rad_listen_t *listener, REQUEST *request)
+static int CC_HINT(nonnull) tls_socket_write(rad_listen_t *listener, REQUEST *request)
 {
        uint8_t *p;
        ssize_t rcode;
@@ -140,7 +131,7 @@ static int tls_socket_recv(rad_listen_t *listener)
        RADCLIENT *client = sock->client;
 
        if (!sock->packet) {
-               sock->packet = rad_alloc(sock, 0);
+               sock->packet = rad_alloc(sock, false);
                if (!sock->packet) return 0;
 
                sock->packet->sockfd = listener->fd;
@@ -149,17 +140,14 @@ static int tls_socket_recv(rad_listen_t *listener)
                sock->packet->dst_ipaddr = sock->my_ipaddr;
                sock->packet->dst_port = sock->my_port;
 
-               if (sock->request) {
-                       (void) talloc_steal(sock->request, sock->packet);
-                       sock->request->packet = sock->packet;
-               }
+               if (sock->request) sock->request->packet = talloc_steal(sock->request, sock->packet);
        }
 
        /*
-        *      Allocate a REQUEST for debugging.
+        *      Allocate a REQUEST for debugging, and initialize the TLS session.
         */
        if (!sock->request) {
-               sock->request = request = request_alloc(NULL);
+               sock->request = request = request_alloc(sock);
                if (!sock->request) {
                        ERROR("Out of memory");
                        return 0;
@@ -167,30 +155,27 @@ static int tls_socket_recv(rad_listen_t *listener)
 
                rad_assert(request->packet == NULL);
                rad_assert(sock->packet != NULL);
-               request->packet = sock->packet;
+               request->packet = talloc_steal(request, sock->packet);
 
                request->component = "<core>";
                request->component = "<tls-connect>";
 
-               /*
-                *      Not sure if we should do this on every packet...
-                */
-               request->reply = rad_alloc(request, 0);
+               request->reply = rad_alloc(request, false);
                if (!request->reply) return 0;
 
                rad_assert(sock->ssn == NULL);
 
-               sock->ssn = tls_new_session(listener->tls, sock->request,
+               sock->ssn = tls_new_session(sock, listener->tls, sock->request,
                                            listener->tls->require_client_cert);
                if (!sock->ssn) {
-                       request_free(&sock->request);
+                       TALLOC_FREE(sock->request);
                        sock->packet = NULL;
                        return 0;
                }
 
-               (void) talloc_steal(sock, sock->ssn);
                SSL_set_ex_data(sock->ssn->ssl, FR_TLS_EX_INDEX_REQUEST, (void *)request);
-               SSL_set_ex_data(sock->ssn->ssl, FR_TLS_EX_INDEX_CERTS, (void *)&request->packet->vps);
+               SSL_set_ex_data(sock->ssn->ssl, fr_tls_ex_index_certs, (void *) &sock->certs);
+               SSL_set_ex_data(sock->ssn->ssl, FR_TLS_EX_INDEX_TALLOC, sock);
 
                doing_init = true;
        }
@@ -209,8 +194,9 @@ static int tls_socket_recv(rad_listen_t *listener)
                     sizeof(sock->ssn->dirty_in.data));
        if ((rcode < 0) && (errno == ECONNRESET)) {
        do_close:
-               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
+               DEBUG("Closing TLS socket from client port %u", sock->other_port);
                tls_socket_close(listener);
+               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
                return 0;
        }
 
@@ -237,28 +223,33 @@ static int tls_socket_recv(rad_listen_t *listener)
        }
 
        /*
-        *      Skip ahead to reading application data.
+        *      If we need to do more initialization, do that here.
         */
-       if (SSL_is_init_finished(sock->ssn->ssl)) goto app;
+       if (!SSL_is_init_finished(sock->ssn->ssl)) {
+               if (!tls_handshake_recv(request, sock->ssn)) {
+                       RDEBUG("FAILED in TLS handshake receive");
+                       goto do_close;
+               }
 
-       if (!tls_handshake_recv(request, sock->ssn)) {
-               RDEBUG("FAILED in TLS handshake receive");
-               goto do_close;
-       }
+               /*
+                *      More ACK data to send.  Do so.
+                */
+               if (sock->ssn->dirty_out.used > 0) {
+                       tls_socket_write(listener, request);
+                       PTHREAD_MUTEX_UNLOCK(&sock->mutex);
+                       return 0;
+               }
 
-       if (sock->ssn->dirty_out.used > 0) {
-               tls_socket_write(listener, request);
-               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
-               return 0;
+               /*
+                *      FIXME: Run the request through a virtual
+                *      server in order to see if we like the
+                *      certificate presented by the client.
+                */
        }
 
-app:
        /*
-        *      FIXME: Run the packet through a virtual server in
-        *      order to see if we like the certificate presented by
-        *      the client.
+        *      Try to get application data.
         */
-
        status = tls_application_data(sock->ssn, request);
        RDEBUG("Application data status %d", status);
 
@@ -272,7 +263,10 @@ app:
                return 0;
        }
 
-       dump_hex("TUNNELED DATA", sock->ssn->clean_out.data, sock->ssn->clean_out.used);
+       /*
+        *      We now have a bunch of application data.
+        */
+       dump_hex("TUNNELED DATA > ", sock->ssn->clean_out.data, sock->ssn->clean_out.used);
 
        /*
         *      If the packet is a complete RADIUS packet, return it to
@@ -280,7 +274,7 @@ app:
         */
        if ((sock->ssn->clean_out.used < 20) ||
            (((sock->ssn->clean_out.data[2] << 8) | sock->ssn->clean_out.data[3]) != (int) sock->ssn->clean_out.used)) {
-               RDEBUG("Received bad packet: Length %d contents %d",
+               RDEBUG("Received bad packet: Length %zd contents %d",
                       sock->ssn->clean_out.used,
                       (sock->ssn->clean_out.data[2] << 8) | sock->ssn->clean_out.data[3]);
                goto do_close;
@@ -293,19 +287,22 @@ app:
        packet->vps = NULL;
        PTHREAD_MUTEX_UNLOCK(&sock->mutex);
 
-       if (!rad_packet_ok(packet, 0)) {
-               RDEBUG("Received bad packet: %s", fr_strerror());
+       if (!rad_packet_ok(packet, 0, NULL)) {
+               if (DEBUG_ENABLED) ERROR("Receive - %s", fr_strerror());
+               DEBUG("Closing TLS socket from client");
+               PTHREAD_MUTEX_LOCK(&sock->mutex);
                tls_socket_close(listener);
+               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
                return 0;       /* do_close unlocks the mutex */
        }
 
        /*
         *      Copied from src/lib/radius.c, rad_recv();
         */
-       if (fr_debug_flag) {
+       if (fr_debug_lvl) {
                char host_ipaddr[128];
 
-               if ((packet->code > 0) && (packet->code < FR_MAX_PACKET_CODE)) {
+               if (is_radius_code(packet->code)) {
                        RDEBUG("tls_recv: %s packet from host %s port %d, id=%d, length=%d",
                               fr_packet_codes[packet->code],
                               inet_ntop(packet->src_ipaddr.af,
@@ -333,23 +330,22 @@ app:
 int dual_tls_recv(rad_listen_t *listener)
 {
        RADIUS_PACKET *packet;
-       REQUEST *request;
        RAD_REQUEST_FUNP fun = NULL;
        listen_socket_t *sock = listener->data;
        RADCLIENT       *client = sock->client;
 
+       if (listener->status != RAD_LISTEN_STATUS_KNOWN) return 0;
+
        if (!tls_socket_recv(listener)) {
                return 0;
        }
 
-       rad_assert(sock->request != NULL);
-       rad_assert(sock->request->packet != NULL);
        rad_assert(sock->packet != NULL);
        rad_assert(sock->ssn != NULL);
        rad_assert(client != NULL);
 
-       request = sock->request;
-       packet = sock->packet;
+       packet = talloc_steal(NULL, sock->packet);
+       sock->packet = NULL;
 
        /*
         *      Some sanity checks, based on the packet code.
@@ -357,15 +353,15 @@ int dual_tls_recv(rad_listen_t *listener)
         *      "auth+acct" are marked as "auth", with the "dual" flag
         *      set.
         */
-       switch(packet->code) {
-       case PW_AUTHENTICATION_REQUEST:
+       switch (packet->code) {
+       case PW_CODE_ACCESS_REQUEST:
                if (listener->type != RAD_LISTEN_AUTH) goto bad_packet;
                FR_STATS_INC(auth, total_requests);
                fun = rad_authenticate;
                break;
 
 #ifdef WITH_ACCOUNTING
-       case PW_ACCOUNTING_REQUEST:
+       case PW_CODE_ACCOUNTING_REQUEST:
                if (listener->type != RAD_LISTEN_ACCT) {
                        /*
                         *      Allow auth + dual.  Disallow
@@ -381,12 +377,11 @@ int dual_tls_recv(rad_listen_t *listener)
                break;
 #endif
 
-       case PW_STATUS_SERVER:
-               if (!mainconfig.status_server) {
+       case PW_CODE_STATUS_SERVER:
+               if (!main_config.status_server) {
                        FR_STATS_INC(auth, total_unknown_types);
-                       WDEBUG("Ignoring Status-Server request due to security configuration");
-                       rad_free(&sock->packet);
-                       request->packet = NULL;
+                       WARN("Ignoring Status-Server request due to security configuration");
+                       rad_free(&packet);
                        return 0;
                }
                fun = rad_status_server;
@@ -398,21 +393,16 @@ int dual_tls_recv(rad_listen_t *listener)
 
                DEBUG("Invalid packet code %d sent from client %s port %d : IGNORED",
                      packet->code, client->shortname, packet->src_port);
-               rad_free(&sock->packet);
-               request->packet = NULL;
+               rad_free(&packet);
                return 0;
        } /* switch over packet types */
 
-       if (!request_receive(listener, packet, client, fun)) {
+       if (!request_receive(NULL, listener, packet, client, fun)) {
                FR_STATS_INC(auth, total_packets_dropped);
-               rad_free(&sock->packet);
-               request->packet = NULL;
+               rad_free(&packet);
                return 0;
        }
 
-       sock->packet = NULL;    /* we have no need for more partial reads */
-       request->packet = NULL;
-
        return 1;
 }
 
@@ -424,9 +414,13 @@ int dual_tls_send(rad_listen_t *listener, REQUEST *request)
 {
        listen_socket_t *sock = listener->data;
 
+       VERIFY_REQUEST(request);
+
        rad_assert(request->listener == listener);
        rad_assert(listener->send == dual_tls_send);
 
+       if (listener->status != RAD_LISTEN_STATUS_KNOWN) return 0;
+
        /*
         *      Accounting reject's are silently dropped.
         *
@@ -444,6 +438,11 @@ int dual_tls_send(rad_listen_t *listener, REQUEST *request)
                return 0;
        }
 
+       if (request->reply->data_len > (MAX_PACKET_LEN - 100)) {
+               RWARN("Packet is large, and possibly truncated - %zd vs max %d",
+                     request->reply->data_len, MAX_PACKET_LEN);
+       }
+
        /*
         *      Sign the packet.
         */
@@ -454,12 +453,15 @@ int dual_tls_send(rad_listen_t *listener, REQUEST *request)
        }
 
        PTHREAD_MUTEX_LOCK(&sock->mutex);
+
        /*
         *      Write the packet to the SSL buffers.
         */
        sock->ssn->record_plus(&sock->ssn->clean_in,
                               request->reply->data, request->reply->data_len);
 
+       dump_hex("TUNNELED DATA < ", sock->ssn->clean_in.data, sock->ssn->clean_in.used);
+
        /*
         *      Do SSL magic to get encrypted data.
         */
@@ -492,7 +494,7 @@ int dual_tls_send(rad_listen_t *listener, REQUEST *request)
  *
  *     Called with the mutex held.
  */
-static int proxy_tls_read(rad_listen_t *listener)
+static ssize_t proxy_tls_read(rad_listen_t *listener)
 {
        int rcode;
        size_t length;
@@ -503,7 +505,7 @@ static int proxy_tls_read(rad_listen_t *listener)
         *      Get the maximum size of data to receive.
         */
        if (!sock->data) sock->data = talloc_array(sock, uint8_t,
-                                                  sock->ssn->offset);
+                                                  sock->ssn->mtu);
 
        data = sock->data;
 
@@ -526,11 +528,8 @@ static int proxy_tls_read(rad_listen_t *listener)
                                return -1;
 
                        default:
-                               while ((err = ERR_get_error())) {
-                                       DEBUG("proxy recv says %s",
-                                             ERR_error_string(err, NULL));
-                               }
-                               
+                               tls_error_log(NULL, "Failed in proxy receive");
+
                                goto do_close;
                        }
                }
@@ -553,7 +552,7 @@ static int proxy_tls_read(rad_listen_t *listener)
                 *      FIXME: allocate a RADIUS_PACKET, and set
                 *      "data" to be as large as necessary.
                 */
-               if (length > sock->ssn->offset) {
+               if (length > sock->ssn->mtu) {
                        INFO("Received packet will be too large! Set \"fragment_size = %u\"",
                             (data[2] << 8) | data[3]);
                        goto do_close;
@@ -594,36 +593,39 @@ static int proxy_tls_read(rad_listen_t *listener)
                return 0;
        }
 
-       sock->partial = 0;
-
-       return 1;
+       sock->partial = 0;      /* we've now read the packet */
+       return length;
 }
 
 
 int proxy_tls_recv(rad_listen_t *listener)
 {
-       int rcode;
        listen_socket_t *sock = listener->data;
        char buffer[256];
        RADIUS_PACKET *packet;
        uint8_t *data;
+       ssize_t data_len;
+
+       if (listener->status != RAD_LISTEN_STATUS_KNOWN) return 0;
 
        DEBUG3("Proxy SSL socket has data to read");
        PTHREAD_MUTEX_LOCK(&sock->mutex);
-       rcode = proxy_tls_read(listener);
+       data_len = proxy_tls_read(listener);
        PTHREAD_MUTEX_UNLOCK(&sock->mutex);
 
-       if (rcode < 0) {
-               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
+       if (data_len < 0) {
+               DEBUG("Closing TLS socket to home server");
+               PTHREAD_MUTEX_LOCK(&sock->mutex);
                tls_socket_close(listener);
+               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
                return 0;
        }
 
-       if (rcode == 0) return 0; /* no data to read */
+       if (data_len == 0) return 0; /* not done yet */
 
        data = sock->data;
 
-       packet = rad_alloc(sock, 0);
+       packet = rad_alloc(sock, false);
        packet->sockfd = listener->fd;
        packet->src_ipaddr = sock->other_ipaddr;
        packet->src_port = sock->other_port;
@@ -631,7 +633,7 @@ int proxy_tls_recv(rad_listen_t *listener)
        packet->dst_port = sock->my_port;
        packet->code = data[0];
        packet->id = data[1];
-       packet->data_len = (data[2] << 8) | data[3];
+       packet->data_len = data_len;
        packet->data = talloc_array(packet, uint8_t, packet->data_len);
        memcpy(packet->data, data, packet->data_len);
        memcpy(packet->vector, packet->data + 4, 16);
@@ -639,14 +641,14 @@ int proxy_tls_recv(rad_listen_t *listener)
        /*
         *      FIXME: Client MIB updates?
         */
-       switch(packet->code) {
-       case PW_AUTHENTICATION_ACK:
-       case PW_ACCESS_CHALLENGE:
-       case PW_AUTHENTICATION_REJECT:
+       switch (packet->code) {
+       case PW_CODE_ACCESS_ACCEPT:
+       case PW_CODE_ACCESS_CHALLENGE:
+       case PW_CODE_ACCESS_REJECT:
                break;
 
 #ifdef WITH_ACCOUNTING
-       case PW_ACCOUNTING_RESPONSE:
+       case PW_CODE_ACCOUNTING_RESPONSE:
                break;
 #endif
 
@@ -677,6 +679,11 @@ int proxy_tls_send(rad_listen_t *listener, REQUEST *request)
        int rcode;
        listen_socket_t *sock = listener->data;
 
+       VERIFY_REQUEST(request);
+
+       if ((listener->status != RAD_LISTEN_STATUS_INIT) &&
+           (listener->status != RAD_LISTEN_STATUS_KNOWN)) return 0;
+
        /*
         *      Normal proxying calls us with the data already
         *      encoded.  The "ping home server" code does not.  So,
@@ -690,16 +697,25 @@ int proxy_tls_send(rad_listen_t *listener, REQUEST *request)
        DEBUG3("Proxy is writing %u bytes to SSL",
               (unsigned int) request->proxy->data_len);
        PTHREAD_MUTEX_LOCK(&sock->mutex);
-       while ((rcode = SSL_write(sock->ssn->ssl, request->proxy->data,
-                                 request->proxy->data_len)) < 0) {
+       rcode = SSL_write(sock->ssn->ssl, request->proxy->data,
+                         request->proxy->data_len);
+       if (rcode < 0) {
                int err;
-               while ((err = ERR_get_error())) {
-                       DEBUG("proxy SSL_write says %s",
-                             ERR_error_string(err, NULL));
+
+               err = ERR_get_error();
+               switch (err) {
+               case SSL_ERROR_NONE:
+               case SSL_ERROR_WANT_READ:
+               case SSL_ERROR_WANT_WRITE:
+                       break;  /* let someone else retry */
+
+               default:
+                       tls_error_log(NULL, "Failed in proxy send");
+                       DEBUG("Closing TLS socket to home server");
+                       tls_socket_close(listener);
+                       PTHREAD_MUTEX_UNLOCK(&sock->mutex);
+                       return 0;
                }
-               PTHREAD_MUTEX_UNLOCK(&sock->mutex);
-               tls_socket_close(listener);
-               return 0;
        }
        PTHREAD_MUTEX_UNLOCK(&sock->mutex);
 
@@ -708,3 +724,4 @@ int proxy_tls_send(rad_listen_t *listener, REQUEST *request)
 #endif /* WITH_PROXY */
 
 #endif /* WITH_TLS */
+#endif /* WITH_TCP */