qnetd-cluster-list.c 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167
  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 <inttypes.h>
  35. #include <stdlib.h>
  36. #include <string.h>
  37. #include "qnetd-cluster-list.h"
  38. void
  39. qnetd_cluster_list_init(struct qnetd_cluster_list *list)
  40. {
  41. TAILQ_INIT(list);
  42. }
  43. struct qnetd_cluster *
  44. qnetd_cluster_list_find_by_name(struct qnetd_cluster_list *list,
  45. const char *cluster_name, size_t cluster_name_len)
  46. {
  47. struct qnetd_cluster *cluster;
  48. TAILQ_FOREACH(cluster, list, entries) {
  49. if (cluster->cluster_name_len == cluster_name_len &&
  50. memcmp(cluster->cluster_name, cluster_name, cluster_name_len) == 0) {
  51. return (cluster);
  52. }
  53. }
  54. return (NULL);
  55. }
  56. static int
  57. client_node_id_cmp(const struct qnetd_client *entry1, const struct qnetd_client *entry2)
  58. {
  59. int res;
  60. res = 0;
  61. if (entry1->node_id < entry2->node_id) res = -1;
  62. if (entry1->node_id > entry2->node_id) res = 1;
  63. return (res);
  64. }
  65. struct qnetd_cluster *
  66. qnetd_cluster_list_add_client(struct qnetd_cluster_list *list, struct qnetd_client *client)
  67. {
  68. struct qnetd_cluster *cluster;
  69. struct qnetd_client *entry;
  70. cluster = qnetd_cluster_list_find_by_name(list, client->cluster_name,
  71. client->cluster_name_len);
  72. if (cluster == NULL) {
  73. cluster = (struct qnetd_cluster *)malloc(sizeof(*cluster));
  74. if (cluster == NULL) {
  75. return (NULL);
  76. }
  77. if (qnetd_cluster_init(cluster, client->cluster_name,
  78. client->cluster_name_len) != 0) {
  79. free(cluster);
  80. return (NULL);
  81. }
  82. TAILQ_INSERT_TAIL(list, cluster, entries);
  83. }
  84. /*
  85. * Sort by nodeid to keep consistent results when listing clients
  86. */
  87. entry = TAILQ_FIRST(&cluster->client_list);
  88. while (entry != NULL) {
  89. if (client_node_id_cmp(entry, client) > 0) {
  90. TAILQ_INSERT_BEFORE(entry, client, cluster_entries);
  91. break;
  92. }
  93. entry = TAILQ_NEXT(entry, cluster_entries);
  94. }
  95. if (entry == NULL) {
  96. TAILQ_INSERT_TAIL(&cluster->client_list, client, cluster_entries);
  97. }
  98. return (cluster);
  99. }
  100. void
  101. qnetd_cluster_list_del_client(struct qnetd_cluster_list *list, struct qnetd_cluster *cluster,
  102. struct qnetd_client *client)
  103. {
  104. TAILQ_REMOVE(&cluster->client_list, client, cluster_entries);
  105. if (TAILQ_EMPTY(&cluster->client_list)) {
  106. TAILQ_REMOVE(list, cluster, entries);
  107. qnetd_cluster_destroy(cluster);
  108. free(cluster);
  109. }
  110. }
  111. void
  112. qnetd_cluster_list_free(struct qnetd_cluster_list *list)
  113. {
  114. struct qnetd_cluster *cluster;
  115. struct qnetd_cluster *cluster_next;
  116. cluster = TAILQ_FIRST(list);
  117. while (cluster != NULL) {
  118. cluster_next = TAILQ_NEXT(cluster, entries);
  119. qnetd_cluster_destroy(cluster);
  120. free(cluster);
  121. cluster = cluster_next;
  122. }
  123. TAILQ_INIT(list);
  124. }
  125. size_t
  126. qnetd_cluster_list_size(const struct qnetd_cluster_list *list)
  127. {
  128. size_t res;
  129. struct qnetd_cluster *cluster;
  130. res = 0;
  131. TAILQ_FOREACH(cluster, list, entries) {
  132. res++;
  133. }
  134. return (res);
  135. }