crypto/pqueue/pqueue.c

   1 /* crypto/pqueue/pqueue.c */
   2 /*
   3  * DTLS implementation written by Nagendra Modadugu
   4  * (nagendra@cs.stanford.edu) for the OpenSSL project 2005.
   5  */
   6 /* ====================================================================
   7  * Copyright (c) 1999-2005 The OpenSSL Project.  All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  *
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  *
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in
  18  *    the documentation and/or other materials provided with the
  19  *    distribution.
  20  *
  21  * 3. All advertising materials mentioning features or use of this
  22  *    software must display the following acknowledgment:
  23  *    "This product includes software developed by the OpenSSL Project
  24  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
  25  *
  26  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  27  *    endorse or promote products derived from this software without
  28  *    prior written permission. For written permission, please contact
  29  *    openssl-core@OpenSSL.org.
  30  *
  31  * 5. Products derived from this software may not be called "OpenSSL"
  32  *    nor may "OpenSSL" appear in their names without prior written
  33  *    permission of the OpenSSL Project.
  34  *
  35  * 6. Redistributions of any form whatsoever must retain the following
  36  *    acknowledgment:
  37  *    "This product includes software developed by the OpenSSL Project
  38  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
  39  *
  40  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  41  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  43  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  44  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  45  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  46  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  47  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  49  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  50  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  51  * OF THE POSSIBILITY OF SUCH DAMAGE.
  52  * ====================================================================
  53  *
  54  * This product includes cryptographic software written by Eric Young
  55  * (eay@cryptsoft.com).  This product includes software written by Tim
  56  * Hudson (tjh@cryptsoft.com).
  57  *
  58  */
  59
  60 #include "internal/cryptlib.h"
  61 #include <openssl/bn.h>
  62 #include <openssl/pqueue.h>
  63
  64 typedef struct _pqueue {
  65     pitem *items;
  66     int count;
  67 } pqueue_s;
  68
  69 pitem *pitem_new(unsigned char *prio64be, void *data)
  70 {
  71     pitem *item = OPENSSL_malloc(sizeof(*item));
  72     if (item == NULL)
  73         return NULL;
  74
  75     memcpy(item->priority, prio64be, sizeof(item->priority));
  76
  77     item->data = data;
  78     item->next = NULL;
  79
  80     return item;
  81 }
  82
  83 void pitem_free(pitem *item)
  84 {
  85     OPENSSL_free(item);
  86 }
  87
  88 pqueue_s *pqueue_new()
  89 {
  90     pqueue_s *pq = OPENSSL_zalloc(sizeof(*pq));
  91
  92     return pq;
  93 }
  94
  95 void pqueue_free(pqueue_s *pq)
  96 {
  97     OPENSSL_free(pq);
  98 }
  99
 100 pitem *pqueue_insert(pqueue_s *pq, pitem *item)
 101 {
 102     pitem *curr, *next;
 103
 104     if (pq->items == NULL) {
 105         pq->items = item;
 106         return item;
 107     }
 108
 109     for (curr = NULL, next = pq->items;
 110          next != NULL; curr = next, next = next->next) {
 111         /*
 112          * we can compare 64-bit value in big-endian encoding with memcmp:-)
 113          */
 114         int cmp = memcmp(next->priority, item->priority, 8);
 115         if (cmp > 0) {          /* next > item */
 116             item->next = next;
 117
 118             if (curr == NULL)
 119                 pq->items = item;
 120             else
 121                 curr->next = item;
 122
 123             return item;
 124         }
 125
 126         else if (cmp == 0)      /* duplicates not allowed */
 127             return NULL;
 128     }
 129
 130     item->next = NULL;
 131     curr->next = item;
 132
 133     return item;
 134 }
 135
 136 pitem *pqueue_peek(pqueue_s *pq)
 137 {
 138     return pq->items;
 139 }
 140
 141 pitem *pqueue_pop(pqueue_s *pq)
 142 {
 143     pitem *item = pq->items;
 144
 145     if (pq->items != NULL)
 146         pq->items = pq->items->next;
 147
 148     return item;
 149 }
 150
 151 pitem *pqueue_find(pqueue_s *pq, unsigned char *prio64be)
 152 {
 153     pitem *next;
 154     pitem *found = NULL;
 155
 156     if (pq->items == NULL)
 157         return NULL;
 158
 159     for (next = pq->items; next->next != NULL; next = next->next) {
 160         if (memcmp(next->priority, prio64be, 8) == 0) {
 161             found = next;
 162             break;
 163         }
 164     }
 165
 166     /* check the one last node */
 167     if (memcmp(next->priority, prio64be, 8) == 0)
 168         found = next;
 169
 170     if (!found)
 171         return NULL;
 172
 173     return found;
 174 }
 175
 176 void pqueue_print(pqueue_s *pq)
 177 {
 178     pitem *item = pq->items;
 179
 180     while (item != NULL) {
 181         printf("item\t%02x%02x%02x%02x%02x%02x%02x%02x\n",
 182                item->priority[0], item->priority[1],
 183                item->priority[2], item->priority[3],
 184                item->priority[4], item->priority[5],
 185                item->priority[6], item->priority[7]);
 186         item = item->next;
 187     }
 188 }
 189
 190 pitem *pqueue_iterator(pqueue_s *pq)
 191 {
 192     return pqueue_peek(pq);
 193 }
 194
 195 pitem *pqueue_next(pitem **item)
 196 {
 197     pitem *ret;
 198
 199     if (item == NULL || *item == NULL)
 200         return NULL;
 201
 202     /* *item != NULL */
 203     ret = *item;
 204     *item = (*item)->next;
 205
 206     return ret;
 207 }
 208
 209 int pqueue_size(pqueue_s *pq)
 210 {
 211     pitem *item = pq->items;
 212     int count = 0;
 213
 214     while (item != NULL) {
 215         count++;
 216         item = item->next;
 217     }
 218     return count;
 219 }