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

C ext: build on ruby 1.9 and fix objectid generation

This commit is contained in:
Mike Dirolf 2009-11-13 11:41:50 -05:00
parent 6cc595c61d
commit 572bf10110
1 changed files with 22 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
ext/cbson/cbson.c
View File

@ -42,6 +42,7 @@
#include <time.h> #include <time.h>
#define INITIAL_BUFFER_SIZE 256 #define INITIAL_BUFFER_SIZE 256
#define MAX_HOSTNAME_LENGTH 256
static VALUE Binary; static VALUE Binary;
static VALUE Time; static VALUE Time;
@ -749,8 +750,8 @@ static VALUE fast_pack(VALUE self)
res = rb_str_buf_new(0); res = rb_str_buf_new(0);
for (i = 0; i < RARRAY(self)->len; i++) { for (i = 0; i < RARRAY_LEN(self); i++) {
c = FIX2LONG(RARRAY(self)->ptr[i]); c = FIX2LONG(RARRAY_PTR(self)[i]);
rb_str_buf_cat(res, &c, sizeof(char)); rb_str_buf_cat(res, &c, sizeof(char));
} }
@ -760,33 +761,33 @@ static VALUE fast_pack(VALUE self)
static VALUE objectid_generate(VALUE self) static VALUE objectid_generate(VALUE self)
{ {
VALUE ary, digest, hostname; VALUE oid, digest;
unsigned char *string; char hostname[MAX_HOSTNAME_LENGTH];
unsigned long t, pid, inc; unsigned char oid_bytes[12];
unsigned long t, inc;
unsigned short pid;
int i; int i;
string = ALLOC_N(unsigned char, 13);
hostname = rb_str_buf_new(64);
t = htonl(time(NULL)); t = htonl(time(NULL));
MEMCPY(string, &t, char, 4); MEMCPY(&oid_bytes, &t, unsigned char, 4);
gethostname(RSTRING(hostname)->ptr, 64); if (gethostname(&hostname, MAX_HOSTNAME_LENGTH) != 0) {
RSTRING(hostname)->len = strlen(RSTRING(hostname)->ptr); rb_raise(rb_eRuntimeError, "failed to get hostname");
digest = rb_funcall(DigestMD5, rb_intern("hexdigest"), 1, hostname); }
MEMCPY(string+4, RSTRING(digest)->ptr, unsigned char, 3); digest = rb_funcall(DigestMD5, rb_intern("digest"), 1, rb_str_new2(hostname));
MEMCPY(&oid_bytes[4], RSTRING_PTR(digest), unsigned char, 3);
pid = htonl(getpid()); pid = htons(getpid());
MEMCPY(string+7, &pid, char, 2); MEMCPY(&oid_bytes[7], &pid, unsigned char, 2);
inc = htonl(FIX2ULONG(rb_funcall(self, rb_intern("get_inc"), 0))); inc = htonl(FIX2ULONG(rb_funcall(self, rb_intern("get_inc"), 0)));
MEMCPY(string+9, ((unsigned char*)&inc + 1), unsigned char, 3); MEMCPY(&oid_bytes[9], ((unsigned char*)&inc + 1), unsigned char, 3);
ary = rb_ary_new2(12); oid = rb_ary_new2(12);
for(i = 0; i < 12; i++) for(i = 0; i < 12; i++) {
rb_ary_store(ary, i, INT2FIX((unsigned int)string[i])); rb_ary_store(oid, i, INT2FIX((unsigned int)oid_bytes[i]));
free(string); }
return ary; return oid;
} }