GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: usr.bin/dc/stack.c Lines: 147 162 90.7 %
Date: 2017-11-13 Branches: 55 68 80.9 %

Line Branch Exec Source
1
/*	$OpenBSD: stack.c,v 1.14 2016/03/27 15:55:13 otto Exp $	*/
2
3
/*
4
 * Copyright (c) 2003, Otto Moerbeek <otto@drijf.net>
5
 *
6
 * Permission to use, copy, modify, and distribute this software for any
7
 * purpose with or without fee is hereby granted, provided that the above
8
 * copyright notice and this permission notice appear in all copies.
9
 *
10
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17
 */
18
19
#include <err.h>
20
#include <stdlib.h>
21
#include <string.h>
22
23
#include "extern.h"
24
25
static __inline bool	stack_empty(const struct stack *);
26
static void		stack_grow(struct stack *);
27
static struct array	*array_new(void);
28
static __inline void	array_free(struct array *);
29
static struct array *	array_dup(const struct array *);
30
static __inline void	array_grow(struct array *, size_t);
31
static __inline void	array_assign(struct array *, size_t, const struct value *);
32
static __inline struct value	*array_retrieve(const struct array *, size_t);
33
34
void
35
stack_init(struct stack *stack)
36
{
37
17866180
	stack->size = 0;
38
8933090
	stack->sp = -1;
39
8933090
	stack->stack = NULL;
40
8933090
}
41
42
static __inline bool
43
stack_empty(const struct stack *stack)
44
{
45
900890
	bool empty = stack->sp == -1;
46
450445
	if (empty)
47
		warnx("stack empty");
48
450445
	return empty;
49
}
50
51
/* Clear number or string, but leave value itself */
52
void
53
stack_free_value(struct value *v)
54
{
55
246440
	switch (v->type) {
56
	case BCODE_NONE:
57
		break;
58
	case BCODE_NUMBER:
59
73890
		free_number(v->u.num);
60
73890
		break;
61
	case BCODE_STRING:
62
8200
		free(v->u.string);
63
8200
		break;
64
	}
65
82175
	array_free(v->array);
66
82175
	v->array = NULL;
67
82175
}
68
69
/* Copy number or string content into already allocated target */
70
struct value *
71
stack_dup_value(const struct value *a, struct value *copy)
72
{
73
387930
	copy->type = a->type;
74
75
258610
	switch (a->type) {
76
	case BCODE_NONE:
77
		break;
78
	case BCODE_NUMBER:
79
112960
		copy->u.num = dup_number(a->u.num);
80
112960
		break;
81
	case BCODE_STRING:
82
16330
		copy->u.string = strdup(a->u.string);
83
16330
		if (copy->u.string == NULL)
84
			err(1, NULL);
85
		break;
86
	}
87
88
258660
	copy->array = a->array == NULL ? NULL : array_dup(a->array);
89
90
129320
	return copy;
91
}
92
93
size_t
94
stack_size(const struct stack *stack)
95
{
96
10880
	return stack->sp + 1;
97
}
98
99
void
100
stack_dup(struct stack *stack)
101
{
102
	struct value	*value;
103
62770
	struct value	copy;
104
105
31385
	value = stack_tos(stack);
106
31385
	if (value == NULL) {
107
		warnx("stack empty");
108
		return;
109
	}
110
31385
	stack_push(stack, stack_dup_value(value, &copy));
111
62770
}
112
113
void
114
stack_swap(struct stack *stack)
115
{
116
70
	struct value	copy;
117
118
35
	if (stack->sp < 1) {
119
10
		warnx("stack empty");
120
10
		return;
121
	}
122
25
	copy = stack->stack[stack->sp];
123
25
	stack->stack[stack->sp] = stack->stack[stack->sp-1];
124
25
	stack->stack[stack->sp-1] = copy;
125
60
}
126
127
static void
128
stack_grow(struct stack *stack)
129
{
130
	size_t new_size;
131
132
905870
	if (++stack->sp == stack->size) {
133
51975
		new_size = stack->size * 2 + 1;
134
51975
		stack->stack = breallocarray(stack->stack,
135
		    new_size, sizeof(*stack->stack));
136
51975
		stack->size = new_size;
137
51975
	}
138
452935
}
139
140
void
141
stack_pushnumber(struct stack *stack, struct number *b)
142
{
143
628300
	stack_grow(stack);
144
314150
	stack->stack[stack->sp].type = BCODE_NUMBER;
145
314150
	stack->stack[stack->sp].u.num = b;
146
314150
	stack->stack[stack->sp].array = NULL;
147
314150
}
148
149
void
150
stack_pushstring(struct stack *stack, char *string)
151
{
152
277390
	stack_grow(stack);
153
138695
	stack->stack[stack->sp].type = BCODE_STRING;
154
138695
	stack->stack[stack->sp].u.string = string;
155
138695
	stack->stack[stack->sp].array = NULL;
156
138695
}
157
158
void
159
stack_push(struct stack *stack, struct value *v)
160
{
161

426075
	switch (v->type) {
162
	case BCODE_NONE:
163
90
		stack_grow(stack);
164
90
		stack->stack[stack->sp].type = BCODE_NONE;
165
90
		break;
166
	case BCODE_NUMBER:
167
125100
		stack_pushnumber(stack, v->u.num);
168
125100
		break;
169
	case BCODE_STRING:
170
16835
		stack_pushstring(stack, v->u.string);
171
16835
		break;
172
	}
173
284120
	stack->stack[stack->sp].array = v->array == NULL ?
174
70
	    NULL : array_dup(v->array);
175
142025
}
176
177
struct value *
178
stack_tos(const struct stack *stack)
179
{
180
338060
	if (stack->sp == -1)
181
		return NULL;
182
169030
	return &stack->stack[stack->sp];
183
169030
}
184
185
void
186
stack_set_tos(struct stack *stack, struct value *v)
187
{
188
128290
	if (stack->sp == -1)
189
2045
		stack_push(stack, v);
190
	else {
191
62100
		stack_free_value(&stack->stack[stack->sp]);
192
62100
		stack->stack[stack->sp] = *v;
193
124220
		stack->stack[stack->sp].array = v->array == NULL ?
194
20
		    NULL : array_dup(v->array);
195
	}
196
64145
}
197
198
struct value *
199
stack_pop(struct stack *stack)
200
{
201
189900
	if (stack_empty(stack))
202
		return NULL;
203
94950
	return &stack->stack[stack->sp--];
204
94950
}
205
206
struct number *
207
stack_popnumber(struct stack *stack)
208
{
209
452100
	if (stack_empty(stack))
210
		return NULL;
211
226050
	array_free(stack->stack[stack->sp].array);
212
226050
	stack->stack[stack->sp].array = NULL;
213
226050
	if (stack->stack[stack->sp].type != BCODE_NUMBER) {
214
		warnx("not a number"); /* XXX remove */
215
		return NULL;
216
	}
217
226050
	return stack->stack[stack->sp--].u.num;
218
226050
}
219
220
char *
221
stack_popstring(struct stack *stack)
222
{
223
258890
	if (stack_empty(stack))
224
		return NULL;
225
129445
	array_free(stack->stack[stack->sp].array);
226
129445
	stack->stack[stack->sp].array = NULL;
227
129445
	if (stack->stack[stack->sp].type != BCODE_STRING) {
228
		warnx("not a string"); /* XXX remove */
229
		return NULL;
230
	}
231
129445
	return stack->stack[stack->sp--].u.string;
232
129445
}
233
234
void
235
stack_clear(struct stack *stack)
236
{
237
76695
	while (stack->sp >= 0)
238
60
		stack_free_value(&stack->stack[stack->sp--]);
239
25525
	free(stack->stack);
240
25525
	stack_init(stack);
241
25525
}
242
243
void
244
stack_print(FILE *f, const struct stack *stack, const char *prefix, u_int base)
245
{
246
	ssize_t i;
247
248
500
	for (i = stack->sp; i >= 0; i--) {
249
190
		print_value(f, &stack->stack[i], prefix, base);
250
380
		(void)putc('\n', f);
251
	}
252
40
}
253
254
255
static struct array *
256
array_new(void)
257
{
258
	struct array *a;
259
260
290
	a = bmalloc(sizeof(*a));
261
145
	a->data = NULL;
262
145
	a->size = 0;
263
145
	return a;
264
}
265
266
static __inline void
267
array_free(struct array *a)
268
{
269
	size_t i;
270
271
875340
	if (a == NULL)
272
437635
		return;
273
200
	for (i = 0; i < a->size; i++)
274
65
		stack_free_value(&a->data[i]);
275
35
	free(a->data);
276
35
	free(a);
277
437705
}
278
279
static struct array *
280
array_dup(const struct array *a)
281
{
282
	struct array	*n;
283
	size_t		i;
284
285
220
	if (a == NULL)
286
		return NULL;
287
110
	n = array_new();
288
110
	array_grow(n, a->size);
289
660
	for (i = 0; i < a->size; i++)
290
220
		(void)stack_dup_value(&a->data[i], &n->data[i]);
291
110
	return n;
292
110
}
293
294
static __inline void
295
array_grow(struct array *array, size_t newsize)
296
{
297
	size_t i;
298
299
320
	array->data = breallocarray(array->data, newsize, sizeof(*array->data));
300
1880
	for (i = array->size; i < newsize; i++) {
301
780
		array->data[i].type = BCODE_NONE;
302
780
		array->data[i].array = NULL;
303
	}
304
160
	array->size = newsize;
305
160
}
306
307
static __inline void
308
array_assign(struct array *array, size_t index, const struct value *v)
309
{
310
130
	if (index >= array->size)
311
50
		array_grow(array, index+1);
312
65
	stack_free_value(&array->data[index]);
313
65
	array->data[index] = *v;
314
65
}
315
316
static __inline struct value *
317
array_retrieve(const struct array *array, size_t index)
318
{
319
90
	if (index >= array->size)
320
		return NULL;
321
45
	return &array->data[index];
322
45
}
323
324
void
325
frame_assign(struct stack *stack, size_t index, const struct value *v)
326
{
327
	struct array *a;
328
130
	struct value n;
329
330
65
	if (stack->sp == -1) {
331
25
		n.type = BCODE_NONE;
332
25
		n.array = NULL;
333
25
		stack_push(stack, &n);
334
25
	}
335
336
65
	a = stack->stack[stack->sp].array;
337
65
	if (a == NULL)
338
35
		a = stack->stack[stack->sp].array = array_new();
339
65
	array_assign(a, index, v);
340
65
}
341
342
struct value *
343
frame_retrieve(const struct stack *stack, size_t index)
344
{
345
	struct array *a;
346
347
120
	if (stack->sp == -1)
348
15
		return NULL;
349
45
	a = stack->stack[stack->sp].array;
350
45
	if (a == NULL)
351
		a = stack->stack[stack->sp].array = array_new();
352
45
	return array_retrieve(a, index);
353
60
}