1 | #include <sys/types.h>
|
---|
2 | #include <errno.h>
|
---|
3 |
|
---|
4 | #if defined(_WIN32) || defined(_WIN64)
|
---|
5 | #include <winsock.h>
|
---|
6 | #include <io.h>
|
---|
7 | #else
|
---|
8 | #include <sys/socket.h>
|
---|
9 | #include <unistd.h>
|
---|
10 | #include <sys/un.h>
|
---|
11 | #endif
|
---|
12 |
|
---|
13 | #include <sys/stat.h>
|
---|
14 | #include <stdio.h>
|
---|
15 | #include <stdlib.h>
|
---|
16 |
|
---|
17 | #include "mruby.h"
|
---|
18 | #include "mruby/array.h"
|
---|
19 | #include "mruby/error.h"
|
---|
20 | #include "mruby/string.h"
|
---|
21 | #include "mruby/variable.h"
|
---|
22 |
|
---|
23 | static mrb_value
|
---|
24 | mrb_io_test_io_setup(mrb_state *mrb, mrb_value self)
|
---|
25 | {
|
---|
26 | char rfname[] = "tmp.mruby-io-test.XXXXXXXX";
|
---|
27 | char wfname[] = "tmp.mruby-io-test.XXXXXXXX";
|
---|
28 | char symlinkname[] = "tmp.mruby-io-test.XXXXXXXX";
|
---|
29 | char socketname[] = "/tmp/mruby-io-test.XXXXXXXX";
|
---|
30 | char msg[] = "mruby io test\n";
|
---|
31 | mode_t mask;
|
---|
32 | int fd0, fd1, fd2, fd3;
|
---|
33 | FILE *fp;
|
---|
34 |
|
---|
35 | #ifndef _WIN32
|
---|
36 | struct sockaddr_un sun0;
|
---|
37 | #endif
|
---|
38 |
|
---|
39 | mask = umask(077);
|
---|
40 | fd0 = mkstemp(rfname);
|
---|
41 | fd1 = mkstemp(wfname);
|
---|
42 | #ifndef _WIN32
|
---|
43 | fd2 = mkstemp(symlinkname);
|
---|
44 | fd3 = mkstemp(socketname);
|
---|
45 | #endif
|
---|
46 | if (fd0 == -1 || fd1 == -1 || fd2 == -1 || fd3 == -1) {
|
---|
47 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't create temporary file");
|
---|
48 | return mrb_nil_value();
|
---|
49 | }
|
---|
50 | umask(mask);
|
---|
51 |
|
---|
52 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_rfname"), mrb_str_new_cstr(mrb, rfname));
|
---|
53 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_wfname"), mrb_str_new_cstr(mrb, wfname));
|
---|
54 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_symlinkname"), mrb_str_new_cstr(mrb, symlinkname));
|
---|
55 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_socketname"), mrb_str_new_cstr(mrb, socketname));
|
---|
56 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_msg"), mrb_str_new_cstr(mrb, msg));
|
---|
57 |
|
---|
58 | fp = fopen(rfname, "wb");
|
---|
59 | if (fp == NULL) {
|
---|
60 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't open temporary file");
|
---|
61 | return mrb_nil_value();
|
---|
62 | }
|
---|
63 | fputs(msg, fp);
|
---|
64 | fclose(fp);
|
---|
65 |
|
---|
66 | fp = fopen(wfname, "wb");
|
---|
67 | if (fp == NULL) {
|
---|
68 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't open temporary file");
|
---|
69 | return mrb_nil_value();
|
---|
70 | }
|
---|
71 | fclose(fp);
|
---|
72 |
|
---|
73 | #ifndef _WIN32
|
---|
74 | unlink(symlinkname);
|
---|
75 | close(fd2);
|
---|
76 | if (symlink(rfname, symlinkname) == -1) {
|
---|
77 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't make a symbolic link");
|
---|
78 | }
|
---|
79 |
|
---|
80 | unlink(socketname);
|
---|
81 | close(fd3);
|
---|
82 | fd3 = socket(AF_UNIX, SOCK_STREAM, 0);
|
---|
83 | if (fd3 == -1) {
|
---|
84 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't make a socket");
|
---|
85 | }
|
---|
86 | sun0.sun_family = AF_UNIX;
|
---|
87 | snprintf(sun0.sun_path, sizeof(sun0.sun_path), "%s", socketname);
|
---|
88 | if (bind(fd3, (struct sockaddr *)&sun0, sizeof(sun0)) == -1) {
|
---|
89 | mrb_raisef(mrb, E_RUNTIME_ERROR, "can't bind AF_UNIX socket to %S: %S",
|
---|
90 | mrb_str_new_cstr(mrb, sun0.sun_path),
|
---|
91 | mrb_fixnum_value(errno));
|
---|
92 | }
|
---|
93 | close(fd3);
|
---|
94 | #endif
|
---|
95 |
|
---|
96 | return mrb_true_value();
|
---|
97 | }
|
---|
98 |
|
---|
99 | static mrb_value
|
---|
100 | mrb_io_test_io_cleanup(mrb_state *mrb, mrb_value self)
|
---|
101 | {
|
---|
102 | mrb_value rfname = mrb_gv_get(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_rfname"));
|
---|
103 | mrb_value wfname = mrb_gv_get(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_wfname"));
|
---|
104 | mrb_value symlinkname = mrb_gv_get(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_symlinkname"));
|
---|
105 | mrb_value socketname = mrb_gv_get(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_socketname"));
|
---|
106 |
|
---|
107 | if (mrb_type(rfname) == MRB_TT_STRING) {
|
---|
108 | remove(RSTRING_PTR(rfname));
|
---|
109 | }
|
---|
110 | if (mrb_type(wfname) == MRB_TT_STRING) {
|
---|
111 | remove(RSTRING_PTR(wfname));
|
---|
112 | }
|
---|
113 | if (mrb_type(symlinkname) == MRB_TT_STRING) {
|
---|
114 | remove(RSTRING_PTR(symlinkname));
|
---|
115 | }
|
---|
116 | if (mrb_type(socketname) == MRB_TT_STRING) {
|
---|
117 | remove(RSTRING_PTR(socketname));
|
---|
118 | }
|
---|
119 |
|
---|
120 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_rfname"), mrb_nil_value());
|
---|
121 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_wfname"), mrb_nil_value());
|
---|
122 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_symlinkname"), mrb_nil_value());
|
---|
123 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_socketname"), mrb_nil_value());
|
---|
124 | mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$mrbtest_io_msg"), mrb_nil_value());
|
---|
125 |
|
---|
126 | return mrb_nil_value();
|
---|
127 | }
|
---|
128 |
|
---|
129 | static mrb_value
|
---|
130 | mrb_io_test_file_setup(mrb_state *mrb, mrb_value self)
|
---|
131 | {
|
---|
132 | mrb_value ary = mrb_io_test_io_setup(mrb, self);
|
---|
133 | #ifndef _WIN32
|
---|
134 | if (symlink("/usr/bin", "test-bin") == -1) {
|
---|
135 | mrb_raise(mrb, E_RUNTIME_ERROR, "can't make a symbolic link");
|
---|
136 | }
|
---|
137 | #endif
|
---|
138 |
|
---|
139 | return ary;
|
---|
140 | }
|
---|
141 |
|
---|
142 | static mrb_value
|
---|
143 | mrb_io_test_file_cleanup(mrb_state *mrb, mrb_value self)
|
---|
144 | {
|
---|
145 | mrb_io_test_io_cleanup(mrb, self);
|
---|
146 | remove("test-bin");
|
---|
147 |
|
---|
148 | return mrb_nil_value();
|
---|
149 | }
|
---|
150 |
|
---|
151 | static mrb_value
|
---|
152 | mrb_io_test_mkdtemp(mrb_state *mrb, mrb_value klass)
|
---|
153 | {
|
---|
154 | mrb_value str;
|
---|
155 | char *cp;
|
---|
156 |
|
---|
157 | mrb_get_args(mrb, "S", &str);
|
---|
158 | cp = mrb_str_to_cstr(mrb, str);
|
---|
159 | if (mkdtemp(cp) == NULL) {
|
---|
160 | mrb_sys_fail(mrb, "mkdtemp");
|
---|
161 | }
|
---|
162 | return mrb_str_new_cstr(mrb, cp);
|
---|
163 | }
|
---|
164 |
|
---|
165 | static mrb_value
|
---|
166 | mrb_io_test_rmdir(mrb_state *mrb, mrb_value klass)
|
---|
167 | {
|
---|
168 | mrb_value str;
|
---|
169 | char *cp;
|
---|
170 |
|
---|
171 | mrb_get_args(mrb, "S", &str);
|
---|
172 | cp = mrb_str_to_cstr(mrb, str);
|
---|
173 | if (rmdir(cp) == -1) {
|
---|
174 | mrb_sys_fail(mrb, "rmdir");
|
---|
175 | }
|
---|
176 | return mrb_true_value();
|
---|
177 | }
|
---|
178 |
|
---|
179 | void
|
---|
180 | mrb_mruby_io_gem_test(mrb_state* mrb)
|
---|
181 | {
|
---|
182 | struct RClass *io_test = mrb_define_module(mrb, "MRubyIOTestUtil");
|
---|
183 | mrb_define_class_method(mrb, io_test, "io_test_setup", mrb_io_test_io_setup, MRB_ARGS_NONE());
|
---|
184 | mrb_define_class_method(mrb, io_test, "io_test_cleanup", mrb_io_test_io_cleanup, MRB_ARGS_NONE());
|
---|
185 |
|
---|
186 | mrb_define_class_method(mrb, io_test, "file_test_setup", mrb_io_test_file_setup, MRB_ARGS_NONE());
|
---|
187 | mrb_define_class_method(mrb, io_test, "file_test_cleanup", mrb_io_test_file_cleanup, MRB_ARGS_NONE());
|
---|
188 |
|
---|
189 | mrb_define_class_method(mrb, io_test, "mkdtemp", mrb_io_test_mkdtemp, MRB_ARGS_REQ(1));
|
---|
190 | mrb_define_class_method(mrb, io_test, "rmdir", mrb_io_test_rmdir, MRB_ARGS_REQ(1));
|
---|
191 | }
|
---|