extattr 0.1.2-x86-mingw32 → 0.2-x86-mingw32

data/ext/extattr.windows

@@ -1,460 +0,0 @@
-#include <windows.h>
-#include <windowsx.h>
-#include <winnt.h>
-#include <ddk/ntifs.h>
-#include <ddk/winddk.h>
-#include <ntdef.h>
-#include <psapi.h>
-#include <wchar.h>
-#include <ruby/win32.h>
-
-#define EXTATTR_NAMESPACE_USER 0
-#define EXTATTR_NAMESPACE_SYSTEM 1
-
-/*
- * - TODO:
- *   バッファサイズの確保量が結構大きめ。最適化をするべき。
- * - TODO:
- *   Win9X シリーズだとこのライブラリ読み込みすら出来ない。
- *   メソッド定義だけは行うようにするべき? (実際に呼び出したら NotImplementedError 例外を投げる、とか)
- * - リパースポイントって、ADSはつけられないのかな? 操作方法がわからない。
- */
-
-/*
- * ADS は普通のファイルのように扱えるから、extattr とみなして扱う場合は容量の制限を設けることにする。
- *
- * ruby-extattr で不十分な巨大なデータを扱う場合は File.open で開くことが出来るので必要であればそちらで。
- *
- * TODO: 最適値を探すべき
- */
-static const size_t EXTATTR_MAX = 65536;
-
-
-static rb_encoding *ENCutf8p;
-static VALUE ENCutf8;
-static rb_encoding *ENCutf16lep;
-static VALUE ENCutf16le;
-
-
-static VALUE
-str2wcs(VALUE str)
-{
-    str = rb_str_encode(str, ENCutf16le, 0, Qnil);
-    rb_str_buf_cat(str, "\0", 1);
-    return str;
-}
-
-static VALUE
-str2wpath(VALUE str)
-{
-    str = str2wcs(str);
-    wchar_t *p = (wchar_t *)RSTRING_PTR(str);
-    const wchar_t *end = p + RSTRING_LEN(str) / 2;
-    for (; p < end; p ++) {
-        if (*p == L'/') { *p = L'\\'; }
-    }
-    return str;
-}
-
-#define STR2WPATH(str)                     \
-    ({                                     \
-        (str) = str2wpath(str);            \
-        (const wchar_t *)RSTRING_PTR(str); \
-    })                                     \
-
-static VALUE
-wcs2str(const wchar_t str[], size_t size)
-{
-    if (!str) { return Qnil; }
-    VALUE v = rb_str_new((const char *)str, size * sizeof(str[0]));
-    rb_enc_associate(v, ENCutf16lep);
-
-    return rb_str_encode(v, ENCutf8, ECONV_INVALID_REPLACE | ECONV_UNDEF_REPLACE, Qnil);
-}
-
-static VALUE
-wpath2str(const wchar_t path[], size_t size)
-{
-    VALUE v = wcs2str(path, size);
-    if (NIL_P(v)) { return Qnil; }
-    char *p = RSTRING_PTR(v);
-    const char *end = p + RSTRING_LEN(v);
-    for (; p < end; p ++) {
-        if (*p == '\\') { *p = '/'; }
-    }
-    return v;
-}
-
-/*
- * ADS 名の先端のコロン『:』と、終端の『:$DATA』を除去してrbuyの文字列を返す。
- */
-static VALUE
-adsname2str(const wchar_t name[], size_t size)
-{
-    if (name[0] == L':') {
-        name ++;
-        size --;
-        if (size >= 6 && wcsncmp(name + size - 6, L":$DATA", 6) == 0) {
-            size -= 6;
-        } else {
-            size --;
-        }
-    }
-
-    if (size > 0) {
-        return wpath2str(name, size);
-    } else {
-        return Qnil;
-    }
-}
-
-
-static void
-raise_win32_error(DWORD status, VALUE name)
-{
-    if (NIL_P(name)) {
-        VALUE args[] = { INT2NUM(rb_w32_map_errno(status)), };
-        rb_exc_raise(rb_class_new_instance(1, args, rb_eSystemCallError));
-    } else {
-        // TODO: name がファイルオブジェクトなのであればパス名を取り出す。パス名がない場合は fd を取り出す。
-
-        VALUE args[] = { StringValue(name), INT2NUM(rb_w32_map_errno(status)), };
-        rb_exc_raise(rb_class_new_instance(2, args, rb_eSystemCallError));
-    }
-}
-
-static void
-raise_ntstatus_error(NTSTATUS status, VALUE name)
-{
-    raise_win32_error(RtlNtStatusToDosError(status), name);
-}
-
-static void
-check_status_error(NTSTATUS status)
-{
-    if (status != STATUS_SUCCESS) {
-        raise_ntstatus_error(status, Qnil);
-    }
-}
-
-
-static VALUE
-get_filepath(HANDLE file)
-{
-    DWORD size = 65536;
-    VALUE buf = rb_str_buf_new(size);
-    NTSTATUS status = NtQueryObject(file, 
-                                    ObjectNameInformation,
-                                    RSTRING_PTR(buf),
-                                    size,
-                                    &size);
-    check_status_error(status);
-
-    const OBJECT_NAME_INFORMATION *info = (const OBJECT_NAME_INFORMATION *)RSTRING_PTR(buf);
-    if (wcsnicmp(info->Name.Buffer, L"\\Device\\", 8) == 0) {
-        // 先頭の "/Device/" を "//?/" に置き換える
-        wcsncpy(info->Name.Buffer + 4, L"\\\\?\\", 4);
-        return wpath2str(info->Name.Buffer + 4, info->Name.Length / 2 - 4);
-    } else {
-        return wpath2str(info->Name.Buffer, info->Name.Length / 2);
-    }
-}
-
-static uint64_t
-get_filesize(HANDLE file)
-{
-    BY_HANDLE_FILE_INFORMATION info;
-
-    if (!GetFileInformationByHandle(file, &info)) {
-        raise_win32_error(GetLastError(), Qnil);
-    }
-
-    return ((uint64_t)info.nFileSizeHigh << 32) | info.nFileSizeLow;
-}
-
-static VALUE
-file_close(HANDLE file)
-{
-    /*
-     * rb_ensure から直接呼びたかったけど、呼び出し規約が違うから無理だよね。
-     */
-    CloseHandle(file);
-
-    return Qnil;
-}
-
-static void
-check_namespace(int namespace, VALUE path)
-{
-    if (namespace != EXTATTR_NAMESPACE_USER) {
-        errno = EPERM;
-        if (NIL_P(path)) {
-            rb_sys_fail(NULL);
-        } else {
-            rb_sys_fail(StringValueCStr(path));
-        }
-    }
-}
-
-
-static void
-extattr_list_name(const char *ptr, void *(*yield)(void *user, VALUE name), void *user)
-{
-    for (;;) {
-        const FILE_STREAM_INFORMATION *info = (const FILE_STREAM_INFORMATION *)ptr;
-        VALUE name = adsname2str(info->StreamName, info->StreamNameLength / 2);
-        if (!NIL_P(name) && RSTRING_LEN(name) > 0) {
-            yield(user, name);
-        }
-        size_t size = info->NextEntryOffset;
-        if (size == 0) { break; }
-        ptr += size;
-    }
-}
-
-static VALUE
-extattr_list0_try(HANDLE file)
-{
-    VALUE iostatusblock = rb_str_buf_new(4096);
-    size_t size = 65536; // TODO: 最適値を見つける
-    VALUE infobuf = rb_str_buf_new(size);
-    char *ptr = RSTRING_PTR(infobuf);
-    memset(ptr, 0, sizeof(FILE_STREAM_INFORMATION));
-    NTSTATUS status = NtQueryInformationFile(file,
-                                             (PIO_STATUS_BLOCK)RSTRING_PTR(iostatusblock),
-                                             ptr, size,
-                                             FileStreamInformation);
-    check_status_error(status);
-
-    if (rb_block_given_p()) {
-        extattr_list_name(ptr,
-                          (void *(*)(void *, VALUE))rb_yield_values,
-                          (void *)1);
-        return Qnil;
-    } else {
-        VALUE list = rb_ary_new();
-        extattr_list_name(ptr,
-                          (void *(*)(void *, VALUE))rb_ary_push,
-                          (void *)list);
-        return list;
-    }
-}
-
-static VALUE
-extattr_list0(VALUE path, DWORD flags, int namespace)
-{
-    check_namespace(namespace, path);
-    VALUE path1 = path;
-    HANDLE file = CreateFileW(STR2WPATH(path1), 0,
-                              FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
-                              NULL, 3,
-                              FILE_FLAG_BACKUP_SEMANTICS | flags,
-                              0);
-    if (file == INVALID_HANDLE_VALUE) { raise_win32_error(GetLastError(), path); }
-    return rb_ensure(extattr_list0_try, (VALUE)file, file_close, (VALUE)file);
-}
-
-static VALUE
-file_extattr_list0(VALUE file, int fd, int namespace)
-{
-    check_namespace(namespace, Qnil);
-    HANDLE file1 = (HANDLE)_get_osfhandle(fd);
-    return extattr_list0_try(file1);
-}
-
-static VALUE
-file_s_extattr_list0(VALUE path, int namespace)
-{
-    return extattr_list0(path, 0, namespace);
-}
-
-static VALUE
-file_s_extattr_list_link0(VALUE path, int namespace)
-{
-    return extattr_list0(path, FILE_FLAG_OPEN_REPARSE_POINT, namespace);
-}
-
-
-static VALUE
-extattr_size0_try(HANDLE file)
-{
-    return ULL2NUM(get_filesize(file));
-}
-
-static VALUE
-extattr_size0(VALUE path, DWORD flags, int namespace, VALUE name)
-{
-    check_namespace(namespace, path);
-    VALUE path1 = rb_str_plus(path, rb_str_new(":", 1));
-    rb_str_append(path1, name);
-    HANDLE file = CreateFileW(STR2WPATH(path1), 0,
-                              FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
-                              NULL, 3,
-                              FILE_FLAG_BACKUP_SEMANTICS | flags,
-                              0);
-    if (file == INVALID_HANDLE_VALUE) { raise_win32_error(GetLastError(), path); }
-    return rb_ensure(extattr_size0_try, (VALUE)file, file_close, (VALUE)file);
-}
-
-static VALUE
-file_extattr_size0(VALUE file, int fd, int namespace, VALUE name)
-{
-    return file_s_extattr_size0(get_filepath((HANDLE)_get_osfhandle(fd)),
-                                namespace, name);
-}
-
-static VALUE
-file_s_extattr_size0(VALUE path, int namespace, VALUE name)
-{
-    return extattr_size0(path, 0, namespace, name);
-}
-
-static VALUE
-file_s_extattr_size_link0(VALUE path, int namespace, VALUE name)
-{
-    return extattr_size0(path, FILE_FLAG_OPEN_REPARSE_POINT, namespace, name);
-}
-
-
-static VALUE
-extattr_get_try(VALUE args[3])
-{
-    HANDLE file = (HANDLE)args[0];
-    int namespace = (int)args[1];
-    VALUE name = args[2];
-    uint64_t size = get_filesize(file);
-    if (size > EXTATTR_MAX) { rb_raise(rb_eSystemCallError, "extattr too huge"); }
-    VALUE buf = rb_str_buf_new(size);
-    DWORD readsize = 0;
-    if (!ReadFile(file, RSTRING_PTR(buf), size, &readsize, NULL)) {
-        raise_win32_error(GetLastError(), Qnil);
-    }
-    rb_str_set_len(buf, readsize);
-    return buf;
-}
-
-static VALUE
-extattr_get0(VALUE path, int flags, int namespace, VALUE name)
-{
-    check_namespace(namespace, path);
-    VALUE path1 = rb_str_plus(path, rb_str_new(":", 1));
-    rb_str_append(path1, name);
-    HANDLE file = CreateFileW(STR2WPATH(path1), GENERIC_READ,
-                              FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
-                              NULL, 3,
-                              FILE_FLAG_BACKUP_SEMANTICS | flags,
-                              0);
-    if (file == INVALID_HANDLE_VALUE) { raise_win32_error(GetLastError(), path); }
-    VALUE args[] = { (VALUE)file, (VALUE)namespace, name, };
-    return rb_ensure(extattr_get_try, (VALUE)args, file_close, (VALUE)file);
-}
-
-static VALUE
-file_extattr_get0(VALUE file, int fd, int namespace, VALUE name)
-{
-    return file_s_extattr_get0(get_filepath((HANDLE)_get_osfhandle(fd)),
-                               namespace, name);
-}
-
-static VALUE
-file_s_extattr_get0(VALUE path, int namespace, VALUE name)
-{
-    return extattr_get0(path, 0, namespace, name);
-}
-
-static VALUE
-file_s_extattr_get_link0(VALUE path, int namespace, VALUE name)
-{
-    return extattr_get0(path, FILE_FLAG_OPEN_REPARSE_POINT, namespace, name);
-}
-
-
-static VALUE
-extattr_set0_try(VALUE args[])
-{
-    HANDLE file = (HANDLE)args[0];
-    int namespace = (int)args[1];
-    VALUE name = (VALUE)args[2];
-    VALUE data = (VALUE)args[3];
-    uint64_t size = RSTRING_LEN(data);
-    if (size > EXTATTR_MAX) { rb_raise(rb_eSystemCallError, "extattr too huge"); }
-    DWORD wrotesize = size;
-    if (!WriteFile(file, RSTRING_PTR(data), size, &wrotesize, NULL)) {
-        raise_win32_error(GetLastError(), Qnil);
-    }
-    return Qnil;
-}
-
-static VALUE
-extattr_set0(VALUE path, DWORD flags, int namespace, VALUE name, VALUE data)
-{
-    check_namespace(namespace, path);
-    VALUE path1 = rb_str_plus(path, rb_str_new(":", 1));
-    rb_str_append(path1, name);
-    HANDLE file = CreateFileW(STR2WPATH(path1), GENERIC_WRITE,
-                              FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
-                              NULL, CREATE_ALWAYS,
-                              FILE_FLAG_BACKUP_SEMANTICS | flags,
-                              0);
-    if (file == INVALID_HANDLE_VALUE) { raise_win32_error(GetLastError(), path); }
-    VALUE args[] = { (VALUE)file, (VALUE)namespace, name, data };
-    return rb_ensure(extattr_set0_try, (VALUE)args, file_close, (VALUE)file);
-}
-
-static VALUE
-file_extattr_set0(VALUE file, int fd, int namespace, VALUE name, VALUE data)
-{
-    return file_s_extattr_set0(get_filepath((HANDLE)_get_osfhandle(fd)),
-                               namespace, name, data);
-}
-
-static VALUE
-file_s_extattr_set0(VALUE path, int namespace, VALUE name, VALUE data)
-{
-    return extattr_set0(path, 0, namespace, name, data);
-}
-
-static VALUE
-file_s_extattr_set_link0(VALUE path, int namespace, VALUE name, VALUE data)
-{
-    return extattr_set0(path, FILE_FLAG_OPEN_REPARSE_POINT, namespace, name, data);
-}
-
-
-static VALUE
-file_extattr_delete0(VALUE file, int fd, int namespace, VALUE name)
-{
-    return file_s_extattr_delete0(get_filepath((HANDLE)_get_osfhandle(fd)),
-                                  namespace, name);
-}
-
-static VALUE
-file_s_extattr_delete0(VALUE path, int namespace, VALUE name)
-{
-    check_namespace(namespace, path);
-    VALUE path1 = rb_str_plus(path, rb_str_new(":", 1));
-    rb_str_append(path1, name);
-    if (!DeleteFileW(STR2WPATH(path1))) {
-        raise_win32_error(GetLastError(), path);
-    }
-    return Qnil;
-}
-
-static VALUE
-file_s_extattr_delete_link0(VALUE path, int namespace, VALUE name)
-{
-    return file_s_extattr_delete0(path, namespace, name);
-}
-
-
-static void
-setup(void)
-{
-    ENCutf8p = rb_enc_find("UTF-8");
-    ENCutf8 = rb_enc_from_encoding(ENCutf8p);
-    rb_gc_register_address(&ENCutf8);
-
-    ENCutf16lep = rb_enc_find("UTF-16LE");
-    ENCutf16le = rb_enc_from_encoding(ENCutf16lep);
-    rb_gc_register_address(&ENCutf16le);
-}