src/ucx/stack.c

Sun, 13 May 2018 12:06:41 +0200

author
Olaf Wintermann <olaf.wintermann@gmail.com>
date
Sun, 13 May 2018 12:06:41 +0200
changeset 197
23fa058ae54b
parent 135
471e28cca288
child 254
4784c14aa639
permissions
-rw-r--r--

fixes ssl accept error handling

99
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
1 /*
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
3 *
135
471e28cca288 ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents: 99
diff changeset
4 * Copyright 2016 Olaf Wintermann. All rights reserved.
99
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
5 *
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
6 * Redistribution and use in source and binary forms, with or without
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
7 * modification, are permitted provided that the following conditions are met:
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
8 *
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
9 * 1. Redistributions of source code must retain the above copyright
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
10 * notice, this list of conditions and the following disclaimer.
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
11 *
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
12 * 2. Redistributions in binary form must reproduce the above copyright
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
13 * notice, this list of conditions and the following disclaimer in the
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
14 * documentation and/or other materials provided with the distribution.
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
15 *
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
26 * POSSIBILITY OF SUCH DAMAGE.
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
27 */
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
28
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
29 #include "stack.h"
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
30 #include <string.h>
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
31
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
32 static size_t ucx_stack_align(size_t n) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
33 int align = n % sizeof(void*);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
34 if (align) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
35 n += sizeof(void*) - align;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
36 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
37 return n;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
38 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
39
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
40 void ucx_stack_init(UcxStack *stack, char* space, size_t size) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
41 stack->size = size - size % sizeof(void*);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
42 stack->space = space;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
43 stack->top = NULL;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
44
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
45 stack->allocator.pool = stack;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
46 stack->allocator.malloc = (ucx_allocator_malloc) ucx_stack_malloc;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
47 stack->allocator.calloc = (ucx_allocator_calloc) ucx_stack_calloc;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
48 stack->allocator.realloc = (ucx_allocator_realloc) ucx_stack_realloc;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
49 stack->allocator.free = (ucx_allocator_free) ucx_stack_free;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
50 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
51
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
52 void *ucx_stack_malloc(UcxStack *stack, size_t n) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
53
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
54 if (ucx_stack_avail(stack) < ucx_stack_align(n)) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
55 return NULL;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
56 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
57 char *prev = stack->top;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
58 if (stack->top) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
59 stack->top += ucx_stack_align(ucx_stack_topsize(stack));
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
60 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
61 stack->top = stack->space;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
62 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
63
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
64 ((struct ucx_stack_metadata*)stack->top)->prev = prev;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
65 ((struct ucx_stack_metadata*)stack->top)->size = n;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
66 stack->top += sizeof(struct ucx_stack_metadata);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
67
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
68 return stack->top;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
69 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
70 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
71
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
72 void *ucx_stack_calloc(UcxStack *stack, size_t nelem, size_t elsize) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
73 void *mem = ucx_stack_malloc(stack, nelem*elsize);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
74 memset(mem, 0, nelem*elsize);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
75 return mem;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
76 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
77
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
78 void *ucx_stack_realloc(UcxStack *stack, void *ptr, size_t n) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
79 if (ptr == stack->top) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
80 if (stack->size - (stack->top - stack->space) < ucx_stack_align(n)) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
81 return NULL;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
82 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
83 ((struct ucx_stack_metadata*)stack->top - 1)->size = n;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
84 return ptr;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
85 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
86 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
87 if (ucx_stack_align(((struct ucx_stack_metadata*)ptr - 1)->size) <
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
88 ucx_stack_align(n)) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
89 void *nptr = ucx_stack_malloc(stack, n);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
90 if (nptr) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
91 memcpy(nptr, ptr, n);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
92 ucx_stack_free(stack, ptr);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
93
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
94 return nptr;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
95 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
96 return NULL;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
97 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
98 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
99 ((struct ucx_stack_metadata*)ptr - 1)->size = n;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
100 return ptr;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
101 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
102 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
103 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
104
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
105 void ucx_stack_free(UcxStack *stack, void *ptr) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
106 if (ptr == stack->top) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
107 stack->top = ((struct ucx_stack_metadata*) stack->top - 1)->prev;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
108 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
109 struct ucx_stack_metadata *next = (struct ucx_stack_metadata*)(
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
110 (char*)ptr +
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
111 ucx_stack_align(((struct ucx_stack_metadata*) ptr - 1)->size)
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
112 );
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
113 next->prev = ((struct ucx_stack_metadata*) ptr - 1)->prev;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
114 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
115 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
116
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
117 void ucx_stack_popn(UcxStack *stack, void *dest, size_t n) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
118 if (ucx_stack_empty(stack)) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
119 return;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
120 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
121
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
122 size_t len = ucx_stack_topsize(stack);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
123 if (len > n) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
124 len = n;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
125 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
126
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
127 memcpy(dest, stack->top, len);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
128
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
129 ucx_stack_free(stack, stack->top);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
130 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
131
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
132 size_t ucx_stack_avail(UcxStack *stack) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
133 size_t avail = ((stack->top ? (stack->size
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
134 - (stack->top - stack->space)
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
135 - ucx_stack_align(ucx_stack_topsize(stack)))
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
136 : stack->size));
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
137
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
138 if (avail > sizeof(struct ucx_stack_metadata)) {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
139 return avail - sizeof(struct ucx_stack_metadata);
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
140 } else {
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
141 return 0;
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
142 }
b9a6af0ae41a ucx update
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
143 }

mercurial