zipruby 0.1.0

data/ext/zipruby_file.c

@@ -0,0 +1,368 @@
+#include <errno.h>
+
+#ifndef _WIN32
+#include <alloca.h>
+#else
+#include <malloc.h>
+#endif
+
+#include "zip.h"
+#include "zipruby.h"
+#include "zipruby_archive.h"
+#include "zipruby_file.h"
+#include "zipruby_stat.h"
+#include "ruby.h"
+
+static VALUE zipruby_file(VALUE klass);
+static VALUE zipruby_file_alloc(VALUE klass);
+static void zipruby_file_mark(struct zipruby_file *p);
+static void zipruby_file_free(struct zipruby_file *p);
+static VALUE zipruby_file_initialize(int argc, VALUE *argv, VALUE self);
+static VALUE zipruby_file_close(VALUE self);
+static VALUE zipruby_file_read(VALUE self);
+static VALUE zipruby_file_stat(VALUE self);
+static VALUE zipruby_file_get_comment(int argc, VALUE *argv, VALUE self);
+static VALUE zipruby_file_set_comment(VALUE self, VALUE comment);
+static VALUE zipruby_file_delete(VALUE self);
+static VALUE zipruby_file_rename(VALUE self, VALUE name);
+static VALUE zipruby_file_unchange(VALUE self);
+static VALUE zipruby_file_name(VALUE self);
+static VALUE zipruby_file_index(VALUE self);
+static VALUE zipruby_file_crc(VALUE self);
+static VALUE zipruby_file_size(VALUE self);
+static VALUE zipruby_file_mtime(VALUE self);
+static VALUE zipruby_file_comp_size(VALUE self);
+static VALUE zipruby_file_comp_method(VALUE self);
+static VALUE zipruby_file_encryption_method(VALUE self);
+
+extern VALUE Zip;
+extern VALUE Archive;
+VALUE File;
+extern VALUE Stat;
+extern VALUE Error;
+
+void Init_zipruby_file() {
+  File = rb_define_class_under(Zip, "File", rb_cObject);
+  rb_define_alloc_func(File, zipruby_file_alloc);
+  rb_define_method(File, "initialize", zipruby_file_initialize, -1);
+  rb_define_method(File, "close", zipruby_file_close, 0);
+  rb_define_method(File, "read", zipruby_file_read, 0);
+  rb_define_method(File, "stat", zipruby_file_stat, 0);
+  rb_define_method(File, "get_comment", zipruby_file_get_comment, -1);
+  rb_define_method(File, "comment", zipruby_file_get_comment, -1);
+  rb_define_method(File, "comment=", zipruby_file_set_comment, 1);
+  rb_define_method(File, "delete", zipruby_file_delete, 0);
+  rb_define_method(File, "rename", zipruby_file_rename, 1);
+  rb_define_method(File, "unchange", zipruby_file_unchange, 1);
+  rb_define_method(File, "revert", zipruby_file_unchange, 1);
+  rb_define_method(File, "name", zipruby_file_name, 0);
+  rb_define_method(File, "index", zipruby_file_index, 0);
+  rb_define_method(File, "crc", zipruby_file_crc, 0);
+  rb_define_method(File, "size", zipruby_file_size, 0);
+  rb_define_method(File, "mtime", zipruby_file_mtime, 0);
+  rb_define_method(File, "comp_size", zipruby_file_comp_size, 0);
+  rb_define_method(File, "comp_method", zipruby_file_comp_method, 0);
+  rb_define_method(File, "encryption_method", zipruby_file_encryption_method, 0);
+}
+
+static VALUE zipruby_file_alloc(VALUE klass) {
+  struct zipruby_file *p = ALLOC(struct zipruby_file);
+
+  p->archive = NULL;
+  p->file = NULL;
+  p->sb = NULL;
+
+  return Data_Wrap_Struct(klass, zipruby_file_mark, zipruby_file_free, p);
+}
+
+static void zipruby_file_mark(struct zipruby_file *p) {
+  if (p->archive) { rb_gc_mark(p->v_archive); }
+  if (p->sb) { rb_gc_mark(p->v_sb); }
+}
+
+static void zipruby_file_free(struct zipruby_file *p) {
+  xfree(p);
+}
+
+/* */
+static VALUE zipruby_file_initialize(int argc, VALUE *argv, VALUE self) {
+  VALUE archive, index, flags, stat_flags;
+  struct zipruby_archive *p_archive;
+  struct zipruby_file *p_file;
+  struct zipruby_stat *p_stat;
+  struct zip_file *fzip;
+  char *fname = NULL;
+  int i_index = -1, i_flags = 0;
+
+  rb_scan_args(argc, argv, "22", &archive,  &index, &flags, &stat_flags);
+
+  if (!rb_obj_is_instance_of(archive, Archive)) {
+    rb_raise(rb_eTypeError, "wrong argument type %s (expected Zip::Archive)", rb_class2name(CLASS_OF(archive)));
+  }
+
+  switch (TYPE(index)) {
+  case T_STRING: fname = StringValuePtr(index); break;
+  case T_FIXNUM: i_index = NUM2INT(index); break;
+  default:
+    rb_raise(rb_eTypeError, "wrong argument type %s (expected String or Fixnum)", rb_class2name(CLASS_OF(index)));
+  }
+
+  if (!NIL_P(flags)) {
+    i_flags = NUM2INT(flags);
+  }
+
+  Data_Get_Struct(archive, struct zipruby_archive, p_archive);
+  Check_Archive(p_archive);
+
+  if (fname) {
+    fzip = zip_fopen(p_archive->archive, fname, i_flags);
+
+    if (fzip == NULL) {
+      rb_raise(Error, "Open file failed - %s: %s", fname, zip_strerror(p_archive->archive));
+    }
+  } else {
+    fzip = zip_fopen_index(p_archive->archive, i_index, i_flags);
+
+    if (fzip == NULL) {
+      rb_raise(Error, "Open file failed at %d", i_index, zip_strerror(p_archive->archive));
+    }
+  }
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  p_file->v_archive = archive;
+  p_file->archive = p_archive->archive;
+  p_file->file = fzip;
+  p_file->v_sb = rb_funcall(Stat, rb_intern("new"), 3, archive, index, stat_flags);
+  Data_Get_Struct(p_file->v_sb, struct zipruby_stat, p_stat);
+  p_file->sb = p_stat->sb;
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_close(VALUE self) {
+  struct zipruby_file *p_file;
+  int error;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  if ((error = zip_fclose(p_file->file)) != 0) {
+    char errstr[ERRSTR_BUFSIZE];
+    zip_error_to_str(errstr, ERRSTR_BUFSIZE, error, errno);
+    zip_unchange(p_file->archive, p_file->sb->index);
+    rb_raise(Error, "Close file failed: %s", errstr);
+  }
+
+  p_file->archive = NULL;
+  p_file->file = NULL;
+  p_file->sb = NULL;
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_read(VALUE self) {
+  struct zipruby_file *p_file;
+  struct zip_stat sb;
+  char *buf;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  zip_stat_init(&sb);
+
+  if (zip_stat_index(p_file->archive, p_file->sb->index, 0, &sb)) {
+    rb_raise(Error, "Read file failed: %s", zip_strerror(p_file->archive));
+  }
+
+  buf = alloca(sb.size);
+  
+  if (zip_fread(p_file->file, buf, sb.size) == -1) {
+    rb_raise(Error, "Read file failed: %s", zip_file_strerror(p_file->file));
+  }
+
+  return rb_str_new(buf, sb.size);
+}
+
+/* */
+static VALUE zipruby_file_stat(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return p_file->v_sb;
+}
+
+/* */
+static VALUE zipruby_file_get_comment(int argc, VALUE *argv, VALUE self) {
+  VALUE flags;
+  struct zipruby_file *p_file;
+  const char *comment;
+  int lenp, i_flags = 0;
+
+  rb_scan_args(argc, argv, "01", &flags);
+
+  if (!NIL_P(flags)) {
+    i_flags = NUM2INT(flags);
+  }
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  // XXX: How is the error checked?
+  comment = zip_get_file_comment(p_file->archive, p_file->sb->index, &lenp, i_flags);
+
+  return comment ? rb_str_new(comment, lenp) : Qnil;
+}
+
+/* */
+static VALUE zipruby_file_set_comment(VALUE self, VALUE comment) {
+  struct zipruby_file *p_file;
+  char *s_comment = NULL;
+  int len = 0;
+
+  if (!NIL_P(comment)) {
+    Check_Type(comment, T_STRING);
+    s_comment = StringValuePtr(comment);
+    len = RSTRING(comment)->len;
+  }
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  if (zip_set_file_comment(p_file->archive, p_file->sb->index, s_comment, len) == -1) {
+    zip_unchange_all(p_file->archive);
+    zip_unchange_archive(p_file->archive);
+    rb_raise(Error, "Comment file failed - %s: %s", p_file->sb->name, zip_strerror(p_file->archive));
+  }
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_delete(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  if (zip_delete(p_file->archive, p_file->sb->index) == -1) {
+    zip_unchange_all(p_file->archive);
+    zip_unchange_archive(p_file->archive);
+    rb_raise(Error, "Delete file failed - %s: %s", p_file->sb->name, zip_strerror(p_file->archive));
+  }
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_rename(VALUE self, VALUE name) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  if (zip_rename(p_file->archive, p_file->sb->index, StringValuePtr(name)) == -1) {
+    zip_unchange_all(p_file->archive);
+    zip_unchange_archive(p_file->archive);
+    rb_raise(Error, "Rename file failed - %s: %s", p_file->sb->name, zip_strerror(p_file->archive));
+  }
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_unchange(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  if (zip_unchange(p_file->archive, p_file->sb->index) == -1) {
+    rb_raise(Error, "Unchange file failed - %s: %s", p_file->sb->name, zip_strerror(p_file->archive));
+  }
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_file_name(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("name"), 0);
+}
+
+/* */
+static VALUE zipruby_file_index(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("index"), 0);
+}
+
+/* */
+static VALUE zipruby_file_crc(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("crc"), 0);
+}
+
+/* */
+static VALUE zipruby_file_size(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("size"), 0);
+}
+
+/* */
+static VALUE zipruby_file_mtime(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("mtime"), 0);
+}
+
+/* */
+static VALUE zipruby_file_comp_size(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("comp_size"), 0);
+}
+
+/* */
+static VALUE zipruby_file_comp_method(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("comp_method"), 0);
+}
+
+/* */
+static VALUE zipruby_file_encryption_method(VALUE self) {
+  struct zipruby_file *p_file;
+
+  Data_Get_Struct(self, struct zipruby_file, p_file);
+  Check_File(p_file);
+
+  return rb_funcall(p_file->v_sb, rb_intern("encryption_method"), 0);
+}

data/ext/zipruby_file.h

@@ -0,0 +1,23 @@
+#ifndef __ZIPRUBY_FILE_H__
+#define __ZIPRUBY_FILE_H__
+
+#include "zip.h"
+#include "ruby.h"
+
+struct zipruby_file {
+  VALUE v_archive;
+  struct zip *archive;
+  struct zip_file *file;
+  VALUE v_sb;
+  struct zip_stat *sb;
+};
+
+void Init_zipruby_file();
+
+#define Check_File(p) do { \
+  if ((p)->archive == NULL || (p)->file == NULL || (p)->sb == NULL) { \
+    rb_raise(rb_eRuntimeError, "invalid Zip::File"); \
+  } \
+} while(0)
+
+#endif

data/ext/zipruby_stat.c

@@ -0,0 +1,164 @@
+#include "zip.h"
+#include "zipruby.h"
+#include "zipruby_archive.h"
+#include "zipruby_stat.h"
+#include "ruby.h"
+
+static VALUE zipruby_stat_alloc(VALUE klass);
+static void zipruby_stat_free(struct zipruby_stat *p);
+static VALUE zipruby_stat_initialize(int argc, VALUE *argv, VALUE self);
+static VALUE zipruby_stat_name(VALUE self);
+static VALUE zipruby_stat_index(VALUE self);
+static VALUE zipruby_stat_crc(VALUE self);
+static VALUE zipruby_stat_size(VALUE self);
+static VALUE zipruby_stat_mtime(VALUE self);
+static VALUE zipruby_stat_comp_size(VALUE self);
+static VALUE zipruby_stat_comp_method(VALUE self);
+static VALUE zipruby_stat_encryption_method(VALUE self);
+
+extern VALUE Zip;
+extern VALUE Archive;
+VALUE Stat;
+extern VALUE Error;
+
+void Init_zipruby_stat() {
+  Stat = rb_define_class_under(Zip, "Stat", rb_cObject);
+  rb_define_alloc_func(Stat, zipruby_stat_alloc);
+  rb_define_method(Stat, "initialize", zipruby_stat_initialize, -1);
+  rb_define_method(Stat, "name", zipruby_stat_name, 0);
+  rb_define_method(Stat, "index", zipruby_stat_index, 0);
+  rb_define_method(Stat, "crc", zipruby_stat_crc, 0);
+  rb_define_method(Stat, "size", zipruby_stat_size, 0);
+  rb_define_method(Stat, "mtime", zipruby_stat_mtime, 0);
+  rb_define_method(Stat, "comp_size", zipruby_stat_comp_size, 0);
+  rb_define_method(Stat, "comp_method", zipruby_stat_comp_method, 0);
+  rb_define_method(Stat, "encryption_method", zipruby_stat_encryption_method, 0);
+}
+
+static VALUE zipruby_stat_alloc(VALUE klass) {
+  struct zipruby_stat *p = ALLOC(struct zipruby_stat);
+
+  p->sb = ALLOC(struct zip_stat);
+  zip_stat_init(p->sb);
+
+  return Data_Wrap_Struct(klass, 0, zipruby_stat_free, p);
+}
+
+static void zipruby_stat_free(struct zipruby_stat *p) {
+  xfree(p->sb);
+  xfree(p);
+}
+
+/* */
+static VALUE zipruby_stat_initialize(int argc, VALUE *argv, VALUE self) {
+  VALUE archive, index, flags;
+  struct zipruby_archive *p_archive;
+  struct zipruby_stat *p_stat;
+  char *fname = NULL;
+  int i_index = -1, i_flags = 0;
+
+  rb_scan_args(argc, argv, "21", &archive, &index, &flags);
+
+  if (!rb_obj_is_instance_of(archive, Archive)) {
+    rb_raise(rb_eTypeError, "wrong argument type %s (expected Zip::Archive)", rb_class2name(CLASS_OF(archive)));
+  }
+
+  switch (TYPE(index)) {
+  case T_STRING: fname = StringValuePtr(index); break;
+  case T_FIXNUM: i_index = NUM2INT(index); break;
+  default:
+    rb_raise(rb_eTypeError, "wrong argument type %s (expected String or Fixnum)", rb_class2name(CLASS_OF(index)));
+  }
+
+  if (!NIL_P(flags)) {
+    i_flags = NUM2INT(flags);
+  }
+
+  Data_Get_Struct(archive, struct zipruby_archive, p_archive);
+  Check_Archive(p_archive);
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  if (fname) {
+    if (zip_stat(p_archive->archive, fname, i_flags, p_stat->sb) != 0) {
+      rb_raise(Error, "Obtain file status failed - %s: %s", fname, zip_strerror(p_archive->archive));
+    }
+  } else {
+    if (zip_stat_index(p_archive->archive, i_index, i_flags, p_stat->sb) != 0) {
+      rb_raise(Error, "Obtain file status failed at %d: %s", i_index, zip_strerror(p_archive->archive));
+    }
+  }
+
+  return Qnil;
+}
+
+/* */
+static VALUE zipruby_stat_name(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return p_stat->sb->name ? rb_str_new2(p_stat->sb->name) : Qnil;
+}
+
+/* */
+static VALUE zipruby_stat_index(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return INT2NUM(p_stat->sb->index);
+}
+
+/* */
+static VALUE zipruby_stat_crc(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return UINT2NUM(p_stat->sb->crc);
+}
+
+/* */
+static VALUE zipruby_stat_size(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return LONG2NUM(p_stat->sb->size);
+}
+
+/* */
+static VALUE zipruby_stat_mtime(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return rb_funcall(rb_cTime, rb_intern("at"), 1,  LONG2NUM((long) p_stat->sb->mtime));
+}
+
+/* */
+static VALUE zipruby_stat_comp_size(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return LONG2NUM(p_stat->sb->comp_size);
+}
+
+/* */
+static VALUE zipruby_stat_comp_method(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return INT2NUM(p_stat->sb->comp_method);
+}
+
+/* */
+static VALUE zipruby_stat_encryption_method(VALUE self) {
+  struct zipruby_stat *p_stat;
+
+  Data_Get_Struct(self, struct zipruby_stat, p_stat);
+
+  return INT2NUM(p_stat->sb->encryption_method);
+}

data/ext/zipruby_stat.h

@@ -0,0 +1,12 @@
+#ifndef __ZIPRUBY_STAT_H__
+#define __ZIPRUBY_STAT_H__
+
+#include "zip.h"
+
+struct zipruby_stat {
+  struct zip_stat *sb;
+};
+
+void Init_zipruby_stat();
+
+#endif

data/ext/zipruby_zip.c

@@ -0,0 +1,37 @@
+#include "ruby.h"
+#include "zip.h"
+#include "zipruby.h"
+#include "zipruby_zip.h"
+
+VALUE Zip;
+
+void Init_zipruby_zip() {
+  Zip = rb_define_module("Zip");
+  rb_define_const(Zip, "VERSION", rb_str_new2(VERSION));
+
+  rb_define_const(Zip, "CREATE",    INT2NUM(ZIP_CREATE));
+  rb_define_const(Zip, "EXCL",      INT2NUM(ZIP_EXCL));
+  rb_define_const(Zip, "CHECKCONS", INT2NUM(ZIP_CHECKCONS));
+
+  rb_define_const(Zip, "FL_NOCASE",     INT2NUM(ZIP_FL_NOCASE));
+  rb_define_const(Zip, "FL_NODIR",      INT2NUM(ZIP_FL_NODIR));
+  rb_define_const(Zip, "FL_COMPRESSED", INT2NUM(ZIP_FL_COMPRESSED));
+  rb_define_const(Zip, "FL_UNCHANGED",  INT2NUM(ZIP_FL_UNCHANGED));
+
+  rb_define_const(Zip, "CM_DEFAULT"   ,     INT2NUM(ZIP_CM_DEFAULT));
+  rb_define_const(Zip, "CM_STORE",          INT2NUM(ZIP_CM_STORE));
+  rb_define_const(Zip, "CM_SHRINK",         INT2NUM(ZIP_CM_SHRINK));
+  rb_define_const(Zip, "CM_REDUCE_1",       INT2NUM(ZIP_CM_REDUCE_1));
+  rb_define_const(Zip, "CM_REDUCE_2",       INT2NUM(ZIP_CM_REDUCE_2));
+  rb_define_const(Zip, "CM_REDUCE_3",       INT2NUM(ZIP_CM_REDUCE_3));
+  rb_define_const(Zip, "CM_REDUCE_4",       INT2NUM(ZIP_CM_REDUCE_4));
+  rb_define_const(Zip, "CM_IMPLODE",        INT2NUM(ZIP_CM_IMPLODE));
+  rb_define_const(Zip, "CM_DEFLATE",        INT2NUM(ZIP_CM_DEFLATE));
+  rb_define_const(Zip, "CM_DEFLATE64",      INT2NUM(ZIP_CM_DEFLATE64));
+  rb_define_const(Zip, "CM_PKWARE_IMPLODE", INT2NUM(ZIP_CM_PKWARE_IMPLODE));
+  rb_define_const(Zip, "CM_BZIP2",          INT2NUM(ZIP_CM_BZIP2));
+
+  rb_define_const(Zip, "EM_NONE",        INT2NUM(ZIP_EM_NONE));
+  rb_define_const(Zip, "EM_TRAD_PKWARE", INT2NUM(ZIP_EM_TRAD_PKWARE));
+  // XXX: Strong Encryption Header not parsed yet
+}

data/ext/zipruby_zip.h

@@ -0,0 +1,6 @@
+#ifndef __ZIPRUBY_ZIP_H__
+#define __ZIPRUBY_ZIP_H__
+
+void Init_zipruby_zip();
+
+#endif