-
Notifications
You must be signed in to change notification settings - Fork 125
Expand file tree
/
Copy pathunpacker.h
More file actions
133 lines (97 loc) · 3.25 KB
/
Copy pathunpacker.h
File metadata and controls
133 lines (97 loc) · 3.25 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/*
* MessagePack for Ruby
*
* Copyright (C) 2008-2013 Sadayuki Furuhashi
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MSGPACK_RUBY_UNPACKER_H__
#define MSGPACK_RUBY_UNPACKER_H__
#include "buffer.h"
#include "unpacker_ext_registry.h"
#ifndef MSGPACK_UNPACKER_STACK_CAPACITY
#define MSGPACK_UNPACKER_STACK_CAPACITY 128
#endif
struct msgpack_unpacker_t;
typedef struct msgpack_unpacker_t msgpack_unpacker_t;
enum stack_type_t {
STACK_TYPE_ARRAY,
STACK_TYPE_MAP_KEY,
STACK_TYPE_MAP_VALUE,
};
typedef struct {
size_t count;
enum stack_type_t type;
VALUE object;
VALUE key;
} msgpack_unpacker_stack_t;
#define MSGPACK_UNPACKER_STACK_SIZE (8+4+8+8) /* assumes size_t <= 64bit, enum <= 32bit, VALUE <= 64bit */
struct msgpack_unpacker_t {
msgpack_buffer_t buffer;
unsigned int head_byte;
msgpack_unpacker_stack_t* stack;
size_t stack_depth;
size_t stack_capacity;
VALUE last_object;
VALUE reading_raw;
size_t reading_raw_remaining;
int reading_raw_type;
VALUE buffer_ref;
msgpack_unpacker_ext_registry_t ext_registry;
/* options */
bool symbolize_keys;
bool allow_unknown_ext;
};
#define UNPACKER_BUFFER_(uk) (&(uk)->buffer)
enum msgpack_unpacker_object_type {
TYPE_NIL = 0,
TYPE_BOOLEAN,
TYPE_INTEGER,
TYPE_FLOAT,
TYPE_RAW,
TYPE_ARRAY,
TYPE_MAP,
};
void msgpack_unpacker_static_init();
void msgpack_unpacker_static_destroy();
void _msgpack_unpacker_init(msgpack_unpacker_t* uk);
void _msgpack_unpacker_destroy(msgpack_unpacker_t* uk);
void msgpack_unpacker_mark(msgpack_unpacker_t* uk);
void _msgpack_unpacker_reset(msgpack_unpacker_t* uk);
static inline void msgpack_unpacker_set_symbolized_keys(msgpack_unpacker_t* uk, bool enable)
{
uk->symbolize_keys = enable;
}
static inline void msgpack_unpacker_set_allow_unknown_ext(msgpack_unpacker_t* uk, bool enable)
{
uk->allow_unknown_ext = enable;
}
/* error codes */
#define PRIMITIVE_CONTAINER_START 1
#define PRIMITIVE_OBJECT_COMPLETE 0
#define PRIMITIVE_EOF -1
#define PRIMITIVE_INVALID_BYTE -2
#define PRIMITIVE_STACK_TOO_DEEP -3
#define PRIMITIVE_UNEXPECTED_TYPE -4
#define PRIMITIVE_UNEXPECTED_EXT_TYPE -5
int msgpack_unpacker_read(msgpack_unpacker_t* uk, size_t target_stack_depth);
int msgpack_unpacker_skip(msgpack_unpacker_t* uk, size_t target_stack_depth);
static inline VALUE msgpack_unpacker_get_last_object(msgpack_unpacker_t* uk)
{
return uk->last_object;
}
int msgpack_unpacker_peek_next_object_type(msgpack_unpacker_t* uk);
int msgpack_unpacker_skip_nil(msgpack_unpacker_t* uk);
int msgpack_unpacker_read_array_header(msgpack_unpacker_t* uk, uint32_t* result_size);
int msgpack_unpacker_read_map_header(msgpack_unpacker_t* uk, uint32_t* result_size);
#endif