qnetd-instance.c 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. /*
  2. * Copyright (c) 2015-2016 Red Hat, Inc.
  3. *
  4. * All rights reserved.
  5. *
  6. * Author: Jan Friesse (jfriesse@redhat.com)
  7. *
  8. * This software licensed under BSD license, the text of which follows:
  9. *
  10. * Redistribution and use in source and binary forms, with or without
  11. * modification, are permitted provided that the following conditions are met:
  12. *
  13. * - Redistributions of source code must retain the above copyright notice,
  14. * this list of conditions and the following disclaimer.
  15. * - Redistributions in binary form must reproduce the above copyright notice,
  16. * this list of conditions and the following disclaimer in the documentation
  17. * and/or other materials provided with the distribution.
  18. * - Neither the name of the Red Hat, Inc. nor the names of its
  19. * contributors may be used to endorse or promote products derived from this
  20. * software without specific prior written permission.
  21. *
  22. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  23. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  26. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  27. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  28. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  29. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  30. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  31. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  32. * THE POSSIBILITY OF SUCH DAMAGE.
  33. */
  34. #include <sys/types.h>
  35. #include <pk11func.h>
  36. #include "qnetd-instance.h"
  37. #include "qnetd-client.h"
  38. #include "qnetd-algorithm.h"
  39. #include "qnetd-log-debug.h"
  40. #include "qnetd-dpd-timer.h"
  41. #include "qnetd-poll-array-user-data.h"
  42. #include "qnetd-client-algo-timer.h"
  43. int
  44. qnetd_instance_init(struct qnetd_instance *instance, size_t max_client_receive_size,
  45. size_t max_client_send_buffers, size_t max_client_send_size,
  46. enum tlv_tls_supported tls_supported, int tls_client_cert_required, size_t max_clients)
  47. {
  48. memset(instance, 0, sizeof(*instance));
  49. pr_poll_array_init(&instance->poll_array, sizeof(struct qnetd_poll_array_user_data));
  50. qnetd_client_list_init(&instance->clients);
  51. qnetd_cluster_list_init(&instance->clusters);
  52. instance->max_client_receive_size = max_client_receive_size;
  53. instance->max_client_send_buffers = max_client_send_buffers;
  54. instance->max_client_send_size = max_client_send_size;
  55. instance->tls_supported = tls_supported;
  56. instance->tls_client_cert_required = tls_client_cert_required;
  57. instance->max_clients = max_clients;
  58. timer_list_init(&instance->main_timer_list);
  59. if (qnetd_dpd_timer_init(instance) != 0) {
  60. return (0);
  61. }
  62. return (0);
  63. }
  64. int
  65. qnetd_instance_destroy(struct qnetd_instance *instance)
  66. {
  67. struct qnetd_client *client;
  68. struct qnetd_client *client_next;
  69. qnetd_dpd_timer_destroy(instance);
  70. client = TAILQ_FIRST(&instance->clients);
  71. while (client != NULL) {
  72. client_next = TAILQ_NEXT(client, entries);
  73. qnetd_instance_client_disconnect(instance, client, 1);
  74. client = client_next;
  75. }
  76. pr_poll_array_destroy(&instance->poll_array);
  77. qnetd_cluster_list_free(&instance->clusters);
  78. qnetd_client_list_free(&instance->clients);
  79. timer_list_free(&instance->main_timer_list);
  80. return (0);
  81. }
  82. void
  83. qnetd_instance_client_disconnect(struct qnetd_instance *instance, struct qnetd_client *client,
  84. int server_going_down)
  85. {
  86. qnetd_log_debug_client_disconnect(client, server_going_down);
  87. if (client->init_received) {
  88. qnetd_algorithm_client_disconnect(client, server_going_down);
  89. }
  90. PR_Close(client->socket);
  91. if (client->cluster != NULL) {
  92. qnetd_cluster_list_del_client(&instance->clusters, client->cluster, client);
  93. }
  94. qnetd_client_algo_timer_abort(client);
  95. qnetd_client_list_del(&instance->clients, client);
  96. }
  97. int
  98. qnetd_instance_init_certs(struct qnetd_instance *instance)
  99. {
  100. instance->server.cert = PK11_FindCertFromNickname(QNETD_CERT_NICKNAME, NULL);
  101. if (instance->server.cert == NULL) {
  102. return (-1);
  103. }
  104. instance->server.private_key = PK11_FindKeyByAnyCert(instance->server.cert, NULL);
  105. if (instance->server.private_key == NULL) {
  106. return (-1);
  107. }
  108. return (0);
  109. }