freebsd-dev/contrib/ntp/ntpd/ntp_prio_q.c
Cy Schubert 2b15cb3d09 MFV ntp 4.2.8p1 (r258945, r275970, r276091, r276092, r276093, r278284)
Thanks to roberto for providing pointers to wedge this into HEAD.

Approved by:	roberto
2015-03-30 13:30:15 +00:00

239 lines
4.5 KiB
C

/* ntp_prio_q.c
*
* This file contains the priority queue implementation used by the
* discrete event simulator.
*
* Written By: Sachin Kamboj
* University of Delaware
* Newark, DE 19711
* Copyright (c) 2006
*/
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <ntp_stdlib.h>
#include <ntp_prio_q.h>
/* Priority Queue
* --------------
* Define a priority queue in which the relative priority of the elements
* is determined by a function 'get_order' which is supplied to the
* priority_queue
*/
queue *debug_create_priority_queue(
q_order_func get_order
#ifdef _CRTDBG_MAP_ALLOC
, const char * sourcefile
, int line_num
#endif
)
{
queue *my_queue;
#ifndef _CRTDBG_MAP_ALLOC
my_queue = emalloc(sizeof(queue));
#else
/* preserve original callsite __FILE__ and __LINE__ for leak report */
my_queue = debug_erealloc(NULL, sizeof(queue), sourcefile, line_num);
#endif
my_queue->get_order = get_order;
my_queue->front = NULL;
my_queue->no_of_elements = 0;
return my_queue;
}
/* Define a function to "destroy" a priority queue, freeing-up
* all the allocated resources in the process
*/
void destroy_queue(
queue *my_queue
)
{
node *temp = NULL;
/* Empty out the queue elements if they are not already empty */
while (my_queue->front != NULL) {
temp = my_queue->front;
my_queue->front = my_queue->front->node_next;
free(temp);
}
/* Now free the queue */
free(my_queue);
}
/* Define a function to allocate memory for one element
* of the queue. The allocated memory consists of size
* bytes plus the number of bytes needed for bookkeeping
*/
void *debug_get_node(
size_t size
#ifdef _CRTDBG_MAP_ALLOC
, const char * sourcefile
, int line_num
#endif
)
{
node *new_node;
#ifndef _CRTDBG_MAP_ALLOC
new_node = emalloc(sizeof(*new_node) + size);
#else
new_node = debug_erealloc(NULL, sizeof(*new_node) + size,
sourcefile, line_num);
#endif
new_node->node_next = NULL;
return new_node + 1;
}
/* Define a function to free the allocated memory for a queue node */
void free_node(
void *my_node
)
{
node *old_node = my_node;
free(old_node - 1);
}
void *
next_node(
void *pv
)
{
node *pn;
pn = pv;
pn--;
if (pn->node_next == NULL)
return NULL;
return pn->node_next + 1;
}
/* Define a function to check if the queue is empty. */
int empty(
queue *my_queue
)
{
return (!my_queue || !my_queue->front);
}
void *
queue_head(
queue *q
)
{
if (NULL == q || NULL == q->front)
return NULL;
return q->front + 1;
}
/* Define a function to add an element to the priority queue.
* The element is added according to its priority -
* relative priority is given by the get_order function
*/
queue *enqueue(
queue * my_queue,
void * my_node
)
{
node *new_node = (node *)my_node - 1;
node *i = NULL;
node *j = my_queue->front;
while (j != NULL &&
(*my_queue->get_order)(new_node + 1, j + 1) > 0) {
i = j;
j = j->node_next;
}
if (i == NULL) { /* Insert at beginning of the queue */
new_node->node_next = my_queue->front;
my_queue->front = new_node;
} else { /* Insert Elsewhere, including the end */
new_node->node_next = i->node_next;
i->node_next = new_node;
}
++my_queue->no_of_elements;
return my_queue;
}
/* Define a function to dequeue the first element from the priority
* queue and return it
*/
void *dequeue(
queue *my_queue
)
{
node *my_node = my_queue->front;
if (my_node != NULL) {
my_queue->front = my_node->node_next;
--my_queue->no_of_elements;
return my_node + 1;
} else
return NULL;
}
/* Define a function that returns the number of elements in the
* priority queue
*/
int get_no_of_elements(
queue *my_queue
)
{
return my_queue->no_of_elements;
}
/* Define a function to append a queue onto another.
* Note: there is a faster way (O(1) as opposed to O(n))
* to do this for simple (FIFO) queues, but we can't rely on
* that for priority queues. (Given the current representation)
*
* I don't anticipate this to be a problem. If it does turn
* out to be a bottleneck, I will consider replacing the
* current implementation with a binomial or fibonacci heap.
*/
void append_queue(
queue *q1,
queue *q2
)
{
while (!empty(q2))
enqueue(q1, dequeue(q2));
destroy_queue(q2);
}
/* FIFO Queue
* ----------
* Use the priority queue to create a traditional FIFO queue.
* The only extra function needed is the create_queue
*/
/* C is not Lisp and does not allow anonymous lambda functions :-(.
* So define a get_fifo_order function here
*/
int get_fifo_order(const void *el1, const void *el2)
{
return 1;
}