github-migration/mongo-ruby-driver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

regexp decoding

Browse Source
This commit is contained in:
Mike Dirolf 2009-03-10 15:33:03 -04:00
parent 173ebdbf48
commit 12e959eb15
1 changed files with 37 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
ext/cbson/cbson.c
View File

@ -32,6 +32,7 @@ static VALUE Undefined;
static VALUE Time; static VALUE Time;
static VALUE ObjectID; static VALUE ObjectID;
static VALUE DBRef; static VALUE DBRef;
static VALUE RegexpOfHolding;
typedef struct { typedef struct {
char* buffer; char* buffer;
@ -483,11 +484,44 @@ static VALUE get_value(const char* buffer, int* position, int type) {
value = Qnil; value = Qnil;
break; break;
} }
case 11:
{
int pattern_length = strlen(buffer + *position);
VALUE pattern = rb_str_new(buffer + *position, pattern_length);
*position += pattern_length + 1;
int flags_length = strlen(buffer + *position);
int i = 0;
char flags[4];
flags[0] = 0;
char extra[10];
extra[0] = 0;
for (i = 0; i < flags_length; i++) {
char flag = buffer[*position + i];
if (flag == 'i' || flag == 'm' || flag == 'x') {
if (strlen(flags) < 3) {
strncat(flags, &flag, 1);
}
}
else if (strlen(extra) < 9) {
strncat(extra, &flag, 1);
}
}
VALUE argv[3] = {
pattern,
rb_str_new2(flags),
rb_str_new2(extra)
};
value = rb_class_new_instance(3, argv, RegexpOfHolding);
*position += flags_length + 1;
break;
}
case 12: case 12:
{ {
*position += 4; *position += 4;
int collection_length = strlen(buffer + *position); int collection_length = strlen(buffer + *position);
VALUE collection = rb_str_new(buffer+ *position, collection_length); VALUE collection = rb_str_new(buffer + *position, collection_length);
*position += collection_length + 1; *position += collection_length + 1;
VALUE str = rb_str_new(buffer + *position, 12); VALUE str = rb_str_new(buffer + *position, 12);
@ -564,6 +598,8 @@ void Init_cbson() {
ObjectID = rb_const_get(driver, rb_intern("ObjectID")); ObjectID = rb_const_get(driver, rb_intern("ObjectID"));
rb_require("mongo/types/dbref"); rb_require("mongo/types/dbref");
DBRef = rb_const_get(driver, rb_intern("DBRef")); DBRef = rb_const_get(driver, rb_intern("DBRef"));
rb_require("mongo/types/regexp_of_holding");
RegexpOfHolding = rb_const_get(driver, rb_intern("RegexpOfHolding"));
VALUE CBson = rb_define_module("CBson"); VALUE CBson = rb_define_module("CBson");
rb_define_module_function(CBson, "serialize", method_serialize, 1); rb_define_module_function(CBson, "serialize", method_serialize, 1);