1 | #include <sys/types.h>
|
---|
2 | #include <sys/stat.h>
|
---|
3 |
|
---|
4 | #include <dirent.h>
|
---|
5 | #include <unistd.h>
|
---|
6 |
|
---|
7 | #include <stdio.h>
|
---|
8 | #include <stdlib.h>
|
---|
9 | #include <string.h>
|
---|
10 |
|
---|
11 | #include "mruby.h"
|
---|
12 | #include "mruby/string.h"
|
---|
13 | #include "mruby/variable.h"
|
---|
14 |
|
---|
15 |
|
---|
16 | mrb_value
|
---|
17 | mrb_dirtest_setup(mrb_state *mrb, mrb_value klass)
|
---|
18 | {
|
---|
19 | mrb_value s;
|
---|
20 | char buf[1024];
|
---|
21 | const char *aname = "a";
|
---|
22 | const char *bname = "b";
|
---|
23 |
|
---|
24 | /* save current working directory */
|
---|
25 | if (getcwd(buf, sizeof(buf)) == NULL) {
|
---|
26 | mrb_raise(mrb, E_RUNTIME_ERROR, "getcwd() failed");
|
---|
27 | }
|
---|
28 | mrb_cv_set(mrb, klass, mrb_intern_cstr(mrb, "pwd"), mrb_str_new_cstr(mrb, buf));
|
---|
29 |
|
---|
30 | /* create sandbox */
|
---|
31 | #if defined(_WIN32) || defined(_WIN64)
|
---|
32 | snprintf(buf, sizeof(buf), "%s\\mruby-dir-test.XXXXXX", _getcwd(NULL,0));
|
---|
33 | if ((mktemp(buf) == NULL) || mkdir(buf) != 0) {
|
---|
34 | mrb_raisef(mrb, E_RUNTIME_ERROR, "mkdtemp(%S) failed", mrb_str_new_cstr(mrb, buf));
|
---|
35 | }
|
---|
36 | #else
|
---|
37 | snprintf(buf, sizeof(buf), "%s/mruby-dir-test.XXXXXX", P_tmpdir);
|
---|
38 | if (mkdtemp(buf) == NULL) {
|
---|
39 | mrb_raisef(mrb, E_RUNTIME_ERROR, "mkdtemp(%S) failed", mrb_str_new_cstr(mrb, buf));
|
---|
40 | }
|
---|
41 | #endif
|
---|
42 | s = mrb_str_new_cstr(mrb, buf);
|
---|
43 | mrb_cv_set(mrb, klass, mrb_intern_cstr(mrb, "sandbox"), s);
|
---|
44 |
|
---|
45 | /* go to sandbox */
|
---|
46 | if (chdir(buf) == -1) {
|
---|
47 | rmdir(buf);
|
---|
48 | mrb_raisef(mrb, E_RUNTIME_ERROR, "chdir(%S) failed", s);
|
---|
49 | }
|
---|
50 |
|
---|
51 | /* make some directories in the sandbox */
|
---|
52 | #if defined(_WIN32) || defined(_WIN64)
|
---|
53 | if (mkdir(aname) == -1) {
|
---|
54 | #else
|
---|
55 | if (mkdir(aname, 0) == -1) {
|
---|
56 | #endif
|
---|
57 | chdir("..");
|
---|
58 | rmdir(buf);
|
---|
59 | mrb_raisef(mrb, E_RUNTIME_ERROR, "mkdir(%S) failed", mrb_str_new_cstr(mrb, aname));
|
---|
60 | }
|
---|
61 | #if defined(_WIN32) || defined(_WIN64)
|
---|
62 | if (mkdir(bname) == -1) {
|
---|
63 | #else
|
---|
64 | if (mkdir(bname, 0) == -1) {
|
---|
65 | #endif
|
---|
66 | rmdir(aname);
|
---|
67 | chdir("..");
|
---|
68 | rmdir(buf);
|
---|
69 | mrb_raisef(mrb, E_RUNTIME_ERROR, "mkdir(%S) failed", mrb_str_new_cstr(mrb, bname));
|
---|
70 | }
|
---|
71 |
|
---|
72 | return mrb_true_value();
|
---|
73 | }
|
---|
74 |
|
---|
75 | mrb_value
|
---|
76 | mrb_dirtest_teardown(mrb_state *mrb, mrb_value klass)
|
---|
77 | {
|
---|
78 | mrb_value d, sandbox;
|
---|
79 | DIR *dirp;
|
---|
80 | struct dirent *dp;
|
---|
81 | const char *path;
|
---|
82 |
|
---|
83 | /* cleanup sandbox */
|
---|
84 | sandbox = mrb_cv_get(mrb, klass, mrb_intern_cstr(mrb, "sandbox"));
|
---|
85 | path = mrb_str_to_cstr(mrb, sandbox);
|
---|
86 |
|
---|
87 | dirp = opendir(path);
|
---|
88 | while ((dp = readdir(dirp)) != NULL) {
|
---|
89 | if (strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..") == 0)
|
---|
90 | continue;
|
---|
91 | if (rmdir(dp->d_name) == -1) {
|
---|
92 | mrb_raisef(mrb, E_RUNTIME_ERROR, "rmdir(%S) failed", mrb_str_new_cstr(mrb, dp->d_name));
|
---|
93 | }
|
---|
94 | }
|
---|
95 | closedir(dirp);
|
---|
96 |
|
---|
97 | /* back to original pwd */
|
---|
98 | d = mrb_cv_get(mrb, klass, mrb_intern_cstr(mrb, "pwd"));
|
---|
99 | path = mrb_str_to_cstr(mrb, d);
|
---|
100 | if (chdir(path) == -1) {
|
---|
101 | mrb_raisef(mrb, E_RUNTIME_ERROR, "chdir(%S) failed", d);
|
---|
102 | }
|
---|
103 |
|
---|
104 | /* remove sandbox directory */
|
---|
105 | sandbox = mrb_cv_get(mrb, klass, mrb_intern_cstr(mrb, "sandbox"));
|
---|
106 | path = mrb_str_to_cstr(mrb, sandbox);
|
---|
107 | if (rmdir(path) == -1) {
|
---|
108 | mrb_raisef(mrb, E_RUNTIME_ERROR, "rmdir(%S) failed", sandbox);
|
---|
109 | }
|
---|
110 |
|
---|
111 | return mrb_true_value();
|
---|
112 | }
|
---|
113 |
|
---|
114 | mrb_value
|
---|
115 | mrb_dirtest_sandbox(mrb_state *mrb, mrb_value klass)
|
---|
116 | {
|
---|
117 | return mrb_cv_get(mrb, klass, mrb_intern_cstr(mrb, "sandbox"));
|
---|
118 | }
|
---|
119 |
|
---|
120 | void
|
---|
121 | mrb_mruby_dir_gem_test(mrb_state *mrb)
|
---|
122 | {
|
---|
123 | struct RClass *c = mrb_define_module(mrb, "DirTest");
|
---|
124 |
|
---|
125 | mrb_define_class_method(mrb, c, "sandbox", mrb_dirtest_sandbox, MRB_ARGS_NONE());
|
---|
126 | mrb_define_class_method(mrb, c, "setup", mrb_dirtest_setup, MRB_ARGS_NONE());
|
---|
127 | mrb_define_class_method(mrb, c, "teardown", mrb_dirtest_teardown, MRB_ARGS_NONE());
|
---|
128 | }
|
---|
129 |
|
---|