-
Notifications
You must be signed in to change notification settings - Fork 125
Expand file tree
/
Copy pathcore_ext.c
More file actions
129 lines (114 loc) · 3.89 KB
/
Copy pathcore_ext.c
File metadata and controls
129 lines (114 loc) · 3.89 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
/*
* MessagePack for Ruby
*
* Copyright (C) 2008-2012 FURUHASHI Sadayuki
*
* 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.
*/
#include "core_ext.h"
#include "packer.h"
#include "packer_class.h"
static inline VALUE delegete_to_pack(int argc, VALUE* argv, VALUE self)
{
if(argc == 0) {
return MessagePack_pack(1, &self);
} else if(argc == 1) {
/* write to io */
VALUE argv2[2];
argv2[0] = self;
argv2[1] = argv[0];
return MessagePack_pack(2, argv2);
} else {
rb_raise(rb_eArgError, "wrong number of arguments (%d for 0..1)", argc);
}
}
#define ENSURE_PACKER(argc, argv, packer, pk) \
if(argc != 1 || rb_class_of(argv[0]) != cMessagePack_Packer) { \
return delegete_to_pack(argc, argv, self); \
} \
VALUE packer = argv[0]; \
msgpack_packer_t *pk; \
Data_Get_Struct(packer, msgpack_packer_t, pk);
static VALUE NilClass_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_nil(pk);
return packer;
}
static VALUE TrueClass_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_true(pk);
return packer;
}
static VALUE FalseClass_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_false(pk);
return packer;
}
static VALUE Fixnum_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_fixnum_value(pk, self);
return packer;
}
static VALUE Bignum_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_bignum_value(pk, self);
return packer;
}
static VALUE Float_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_float_value(pk, self);
return packer;
}
static VALUE String_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_string_value(pk, self);
return packer;
}
static VALUE Array_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_array_value(pk, self);
return packer;
}
static VALUE Hash_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_hash_value(pk, self);
return packer;
}
static VALUE Symbol_to_msgpack(int argc, VALUE* argv, VALUE self)
{
ENSURE_PACKER(argc, argv, packer, pk);
msgpack_packer_write_symbol_value(pk, self);
return packer;
}
void MessagePack_core_ext_module_init()
{
rb_define_method(rb_cNilClass, "to_msgpack", NilClass_to_msgpack, -1);
rb_define_method(rb_cTrueClass, "to_msgpack", TrueClass_to_msgpack, -1);
rb_define_method(rb_cFalseClass, "to_msgpack", FalseClass_to_msgpack, -1);
rb_define_method(rb_cFixnum, "to_msgpack", Fixnum_to_msgpack, -1);
rb_define_method(rb_cBignum, "to_msgpack", Bignum_to_msgpack, -1);
rb_define_method(rb_cFloat, "to_msgpack", Float_to_msgpack, -1);
rb_define_method(rb_cString, "to_msgpack", String_to_msgpack, -1);
rb_define_method(rb_cArray, "to_msgpack", Array_to_msgpack, -1);
rb_define_method(rb_cHash, "to_msgpack", Hash_to_msgpack, -1);
rb_define_method(rb_cSymbol, "to_msgpack", Symbol_to_msgpack, -1);
}