binyo 0.0.1

data/ext/binyo/binyo.h

@@ -0,0 +1,75 @@
+/*
+ * binyo - Fast binary IO for Ruby
+ *
+ * Copyright (c) 2012-2013
+ * Martin Bosslet <martin.bosslet@gmail.com>
+ * All rights reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#if !defined(_BINYO_H_)
+#define _BINYO_H_
+
+#include RUBY_EXTCONF_H
+
+#if defined(__cplusplus)
+extern "C" {
+#endif
+
+#define RSTRING_NOT_MODIFIED 1
+#define RUBY_READONLY_STRING 1
+
+#include <ruby.h>
+
+#if defined(HAVE_RUBY_IO_H)
+#include <ruby/io.h>
+#endif
+
+#include "binyo-missing.h"
+
+#ifdef _WIN32
+#define binyo_last_sys_error()	GetLastError()
+#define binyo_clear_sys_error()	SetLastError(0)
+#else
+#define binyo_last_sys_error()	errno
+#define binyo_clear_sys_error()	errno=0
+#endif
+
+/* include headers */
+#include "binyo-mem.h"
+#include "binyo-error.h"
+#include "binyo-error-internal.h"
+#include "binyo-io.h"
+#include "bytelist.h"
+#include "byte.h"
+
+extern VALUE mBinyo;
+
+extern VALUE eBinyoError;
+void Init_binyo(void);
+void Init_binyo_io(void);
+
+#if defined(__cplusplus)
+}
+#endif
+
+#endif /* _BINYO_H_ */
+

data/ext/binyo/byte.c

@@ -0,0 +1,116 @@
+/*
+ * binyo - Fast binary IO for Ruby
+ *
+ * Copyright (c) 2012-2013
+ * Martin Bosslet <martin.bosslet@gmail.com>
+ * All rights reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include "binyo.h"
+
+VALUE cBinyoByte;
+
+#define int_byte_get(obj, b)					\
+do { 								\
+    Data_Get_Struct((obj), uint8_t, (b));			\
+    if (!(b)) { 						\
+	rb_raise(eBinyoError, "Uninitialized byte");		\
+    } 								\
+} while (0)
+
+static void
+int_byte_free(uint8_t *b)
+{
+    if (b) BINYO_FREE(b);
+}
+
+static VALUE
+binyo_byte_alloc(VALUE klass)
+{
+    VALUE obj;
+
+    obj = Data_Wrap_Struct(klass, 0, int_byte_free, 0);
+    return obj;
+}
+
+VALUE
+binyo_byte_new(uint8_t input)
+{
+    VALUE obj;
+    uint8_t *b;
+
+    b = BINYO_ALLOC(uint8_t);
+    *b = input;
+    obj = Data_Wrap_Struct(cBinyoByte, 0, int_byte_free, b);
+    return obj;
+}
+
+static VALUE
+binyo_byte_initialize(VALUE self, VALUE byteval)
+{
+    uint8_t *b;
+
+    b = BINYO_ALLOC(uint8_t);
+    *b = NUM2INT(byteval) & 0xff;
+    DATA_PTR(self) = b;
+    return self;
+}
+
+static VALUE
+binyo_byte_shiftr(VALUE self, VALUE vshift)
+{
+    uint8_t *b;
+
+    int_byte_get(self, b);
+    return binyo_byte_new(*b >> (NUM2INT(vshift)));
+}
+
+static VALUE
+binyo_byte_and(VALUE self, VALUE vmask)
+{
+    uint8_t *b;
+
+    int_byte_get(self, b);
+    return binyo_byte_new(*b & (NUM2INT(vmask)));
+}
+
+static VALUE
+binyo_byte_to_i(VALUE self)
+{
+    uint8_t *b;
+
+    int_byte_get(self, b);
+    return INT2NUM(*b);
+}
+
+void
+Init_binyo_byte(void)
+{
+    cBinyoByte = rb_define_class_under(mBinyo, "Byte", rb_cObject);
+
+    rb_define_alloc_func(cBinyoByte, binyo_byte_alloc);
+    rb_define_method(cBinyoByte, "initialize", binyo_byte_initialize, 1);
+    rb_define_method(cBinyoByte, ">>", binyo_byte_shiftr, 1);
+    rb_define_method(cBinyoByte, "&", binyo_byte_and, 1);
+    rb_define_method(cBinyoByte, "to_i", binyo_byte_to_i, 0);
+}
+

data/ext/binyo/byte.h

@@ -0,0 +1,39 @@
+/*
+ * binyo - Fast binary IO for Ruby
+ *
+ * Copyright (c) 2012-2013
+ * Martin Bosslet <martin.bosslet@gmail.com>
+ * All rights reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#if !defined(_BYTE_H_)
+#define _BYTE_H_
+
+extern VALUE cBinyoByte;
+
+VALUE binyo_byte_new(uint8_t input);
+
+void Init_binyo_byte(void);
+
+#endif /* _BYTE_H_ */
+
+

data/ext/binyo/bytelist.c

@@ -0,0 +1,223 @@
+/*
+ * binyo - Fast binary IO for Ruby
+ *
+ * Copyright (c) 2012-2013
+ * Martin Bosslet <martin.bosslet@gmail.com>
+ * All rights reserved.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include "binyo.h"
+
+VALUE cBinyoByteList;
+
+struct binyo_bytelist_st
+{
+    uint8_t *bytes;
+    size_t len;
+    size_t limit;
+}; 
+
+#define int_bytelist_get(obj, bytelist)				\
+do { 								\
+    Data_Get_Struct((obj), binyo_bytelist, (bytelist));		\
+    if (!(bytelist)) { 						\
+	rb_raise(eBinyoError, "Uninitialized byte list");	\
+    } 								\
+} while (0)
+
+static binyo_bytelist *
+int_bytelist_new()
+{
+    binyo_bytelist *bytelist;
+
+    bytelist = BINYO_ALLOC(binyo_bytelist);
+    memset(bytelist, 0, sizeof(binyo_bytelist));
+    return bytelist;
+}
+
+binyo_bytelist *
+binyo_bytelist_new_size(size_t size)
+{
+    binyo_bytelist *bytelist = int_bytelist_new();
+
+    bytelist->bytes = BINYO_ALLOC_N(uint8_t, size);
+    bytelist->limit = size;
+    return bytelist;
+}
+
+binyo_bytelist *
+binyo_bytelist_new_size_default(size_t size, uint8_t default_value)
+{
+    size_t i;
+    binyo_bytelist *bytelist = binyo_bytelist_new_size(size);
+
+    for (i=0; i < size; i++) {
+	bytelist->bytes[i] = default_value;
+    }
+
+    bytelist->len = size;
+    return bytelist;
+}
+
+binyo_bytelist *
+binyo_bytelist_new_bytes(uint8_t *bytes, size_t len)
+{
+    binyo_bytelist *bytelist = binyo_bytelist_new_size(len);
+
+    memcpy(bytelist->bytes, bytes, len);
+    bytelist->len = len;
+    return bytelist;
+}
+
+void
+binyo_bytelist_free(binyo_bytelist *bytelist)
+{
+    if (bytelist) {
+       if (bytelist->bytes)
+	   BINYO_FREE(bytelist->bytes);
+       BINYO_FREE(bytelist);
+    }
+}
+
+static VALUE
+binyo_bytelist_alloc(VALUE klass)
+{
+    VALUE obj;
+
+    obj = Data_Wrap_Struct(klass, 0, binyo_bytelist_free, 0);
+    return obj;
+}
+
+static VALUE
+binyo_bytelist_initialize(int argc, VALUE *argv, VALUE self)
+{
+    VALUE vfirst;
+    VALUE vsize = Qnil;
+    binyo_bytelist *bytelist;
+    
+    rb_scan_args(argc, argv, "11", &vfirst, &vsize);
+
+    if (NIL_P(vfirst))
+	rb_raise(rb_eArgError, "First argument may not be nil");
+    if (argc == 2 && NIL_P(vsize))
+	rb_raise(rb_eArgError, "Size argument may not be nil");
+
+    if (TYPE(vfirst) == T_STRING) {
+	bytelist = binyo_bytelist_new_bytes((uint8_t *) RSTRING_PTR(vfirst), (size_t) RSTRING_LEN(vfirst));
+    } else {
+	bytelist = binyo_bytelist_new_size((size_t) NUM2LONG(vfirst));
+    }
+
+    DATA_PTR(self) = bytelist;
+    return self;
+}
+
+static VALUE
+binyo_bytelist_get_index(VALUE self, VALUE vidx)
+{
+    binyo_bytelist *bytelist;
+    size_t i = (size_t) NUM2LONG(vidx);
+
+    int_bytelist_get(self, bytelist);
+
+    if (i >= bytelist->len)
+	rb_raise(eBinyoError, "Index out of bounds: %lu", i); /* TODO treat it like Array */
+
+    return INT2NUM(bytelist->bytes[i]);
+}
+
+static VALUE
+binyo_bytelist_set_index(VALUE self, VALUE vidx, VALUE byteval)
+{
+    binyo_bytelist *bytelist;
+    size_t i = (size_t) NUM2LONG(vidx);
+
+    int_bytelist_get(self, bytelist);
+
+    if (i >= bytelist->limit)
+	rb_raise(eBinyoError, "Index out of bounds: %lu", i); /* TODO treat it like Array */
+
+    bytelist->bytes[i] = NUM2INT(byteval) & 0xff;
+    if (i >= bytelist->len)
+	bytelist->len = i + 1;
+
+    return byteval;
+}
+
+static VALUE
+binyo_bytelist_push(VALUE self, VALUE byteval)
+{
+    binyo_bytelist *bytelist;
+
+    int_bytelist_get(self, bytelist);
+
+    if (bytelist->len == bytelist->limit)
+	rb_raise(eBinyoError, "Cannot append to ByteList"); /* TODO auto-resize */
+
+    bytelist->bytes[bytelist->len++] = NUM2INT(byteval) & 0xff;
+    return self;
+}
+
+static VALUE
+binyo_bytelist_each(VALUE self)
+{
+    binyo_bytelist *bytelist;
+    size_t i;
+
+    if (!rb_block_given_p()) {
+	/* TODO */
+	rb_raise(eBinyoError, "No block given");
+    }
+
+    int_bytelist_get(self, bytelist);
+
+    for (i=0; i < bytelist->len; i++) {
+	rb_yield(INT2NUM(bytelist->bytes[i]));
+    }
+
+    return self;
+}
+
+static VALUE
+binyo_bytelist_to_s(VALUE self)
+{
+    binyo_bytelist *bytelist;
+
+    int_bytelist_get(self, bytelist);
+
+    return rb_str_new((const char *)bytelist->bytes, (long) bytelist->len);
+}
+
+void
+Init_binyo_bytelist(void)
+{
+    cBinyoByteList = rb_define_class_under(mBinyo, "ByteList", rb_cObject);
+
+    rb_define_alloc_func(cBinyoByteList, binyo_bytelist_alloc);
+    rb_define_method(cBinyoByteList, "initialize", binyo_bytelist_initialize, -1);
+    rb_define_method(cBinyoByteList, "each", binyo_bytelist_each, 0);
+    rb_define_method(cBinyoByteList, "[]", binyo_bytelist_get_index, 1);
+    rb_define_method(cBinyoByteList, "[]=", binyo_bytelist_set_index, 2);
+    rb_define_method(cBinyoByteList, "<<", binyo_bytelist_push, 1);
+    rb_define_method(cBinyoByteList, "to_s", binyo_bytelist_to_s, 0);
+}
+