6 #include "mruby/array.h"
7 #include "mruby/class.h"
8 #include "mruby/data.h"
9 #include "mruby/hash.h"
10 #include "mruby/string.h"
11 #include "mruby/variable.h"
12 #include "mruby/ext/io.h"
14 #if MRUBY_RELEASE_NO < 10000
17 #include "mruby/error.h"
20 #include <sys/types.h>
23 #if defined(_WIN32) || defined(_WIN64)
33 #define isatty _isatty
34 #define WEXITSTATUS(x) (x)
37 typedef long fsuseconds_t;
43 typedef size_t fsize_t;
44 typedef time_t ftime_t;
45 typedef suseconds_t fsuseconds_t;
46 typedef mode_t fmode_t;
50 typedef mrb_int pid_t;
60 static void mrb_io_free(mrb_state *mrb, void *ptr);
61 struct mrb_data_type mrb_io_type = { "IO", mrb_io_free };
64 static struct mrb_io *io_get_open_fptr(mrb_state *mrb, mrb_value self);
65 static int mrb_io_modestr_to_flags(mrb_state *mrb, const char *modestr);
66 static int mrb_io_flags_to_modenum(mrb_state *mrb, int flags);
67 static void fptr_finalize(mrb_state *mrb, struct mrb_io *fptr, int quiet);
69 #if MRUBY_RELEASE_NO < 10000
70 static struct RClass *
71 mrb_module_get(mrb_state *mrb, const char *name)
73 return mrb_class_get(mrb, name);
77 static struct mrb_io *
78 io_get_open_fptr(mrb_state *mrb, mrb_value self)
82 fptr = (struct mrb_io *)mrb_get_datatype(mrb, self, &mrb_io_type);
84 mrb_raise(mrb, E_IO_ERROR, "uninitialized stream.");
87 mrb_raise(mrb, E_IO_ERROR, "closed stream.");
93 io_set_process_status(mrb_state *mrb, pid_t pid, int status)
95 struct RClass *c_process, *c_status;
99 if (mrb_class_defined(mrb, "Process")) {
100 c_process = mrb_module_get(mrb, "Process");
101 if (mrb_const_defined(mrb, mrb_obj_value(c_process), mrb_intern_cstr(mrb, "Status"))) {
102 c_status = mrb_class_get_under(mrb, c_process, "Status");
105 if (c_status != NULL) {
106 v = mrb_funcall(mrb, mrb_obj_value(c_status), "new", 2, mrb_fixnum_value(pid), mrb_fixnum_value(status));
108 v = mrb_fixnum_value(WEXITSTATUS(status));
110 mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$?"), v);
114 mrb_io_modestr_to_flags(mrb_state *mrb, const char *mode)
117 const char *m = mode;
121 flags |= FMODE_READABLE;
124 flags |= FMODE_WRITABLE | FMODE_CREATE | FMODE_TRUNC;
127 flags |= FMODE_WRITABLE | FMODE_APPEND | FMODE_CREATE;
130 mrb_raisef(mrb, E_ARGUMENT_ERROR, "illegal access mode %S", mrb_str_new_cstr(mrb, mode));
136 flags |= FMODE_BINMODE;
139 flags |= FMODE_READWRITE;
142 /* XXX: PASSTHROUGH*/
144 mrb_raisef(mrb, E_ARGUMENT_ERROR, "illegal access mode %S", mrb_str_new_cstr(mrb, mode));
152 mrb_io_flags_to_modenum(mrb_state *mrb, int flags)
156 switch(flags & (FMODE_READABLE|FMODE_WRITABLE|FMODE_READWRITE)) {
163 case FMODE_READWRITE:
168 if (flags & FMODE_APPEND) {
171 if (flags & FMODE_TRUNC) {
174 if (flags & FMODE_CREATE) {
178 if (flags & FMODE_BINMODE) {
187 mrb_fd_cloexec(mrb_state *mrb, int fd)
189 #if defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
192 flags = fcntl(fd, F_GETFD);
194 mrb_bug(mrb, "mrb_fd_cloexec: fcntl(%S, F_GETFD) failed: %S",
195 mrb_fixnum_value(fd), mrb_fixnum_value(errno));
198 flags2 = flags & ~FD_CLOEXEC; /* Clear CLOEXEC for standard file descriptors: 0, 1, 2. */
201 flags2 = flags | FD_CLOEXEC; /* Set CLOEXEC for non-standard file descriptors: 3, 4, 5, ... */
203 if (flags != flags2) {
204 if (fcntl(fd, F_SETFD, flags2) == -1) {
205 mrb_bug(mrb, "mrb_fd_cloexec: fcntl(%S, F_SETFD, %S) failed: %S",
206 mrb_fixnum_value(fd), mrb_fixnum_value(flags2), mrb_fixnum_value(errno));
212 #if !defined(_WIN32) && !(defined(TARGET_OS_IPHONE) && TARGET_OS_IPHONE)
214 mrb_cloexec_pipe(mrb_state *mrb, int fildes[2])
220 mrb_fd_cloexec(mrb, fildes[0]);
221 mrb_fd_cloexec(mrb, fildes[1]);
226 mrb_pipe(mrb_state *mrb, int pipes[2])
229 ret = mrb_cloexec_pipe(mrb, pipes);
231 if (errno == EMFILE || errno == ENFILE) {
232 mrb_garbage_collect(mrb);
233 ret = mrb_cloexec_pipe(mrb, pipes);
240 mrb_proc_exec(const char *pname)
245 while (*s == ' ' || *s == '\t' || *s == '\n')
253 execl("/bin/sh", "sh", "-c", pname, (char *)NULL);
259 mrb_io_free(mrb_state *mrb, void *ptr)
261 struct mrb_io *io = (struct mrb_io *)ptr;
263 fptr_finalize(mrb, io, TRUE);
268 static struct mrb_io *
269 mrb_io_alloc(mrb_state *mrb)
273 fptr = (struct mrb_io *)mrb_malloc(mrb, sizeof(struct mrb_io));
289 option_to_fd(mrb_state *mrb, mrb_value obj, const char *key)
291 mrb_value opt = mrb_funcall(mrb, obj, "[]", 1, mrb_symbol_value(mrb_intern_static(mrb, key, strlen(key))));
292 if (mrb_nil_p(opt)) {
296 switch (mrb_type(opt)) {
297 case MRB_TT_DATA: /* IO */
298 return (int)mrb_fixnum(mrb_io_fileno(mrb, opt));
300 return (int)mrb_fixnum(opt);
302 mrb_raise(mrb, E_ARGUMENT_ERROR, "wrong exec redirect action");
305 return -1; /* never reached */
310 mrb_io_s_popen(mrb_state *mrb, mrb_value klass)
313 mrb_value mode = mrb_str_new_cstr(mrb, "r");
314 mrb_value opt = mrb_hash_new(mrb);
320 PROCESS_INFORMATION pi;
321 SECURITY_ATTRIBUTES saAttr;
327 int opt_in, opt_out, opt_err;
329 ifd[0] = INVALID_HANDLE_VALUE;
330 ifd[1] = INVALID_HANDLE_VALUE;
331 ofd[0] = INVALID_HANDLE_VALUE;
332 ofd[1] = INVALID_HANDLE_VALUE;
334 mrb_get_args(mrb, "S|SH", &cmd, &mode, &opt);
335 io = mrb_obj_value(mrb_data_object_alloc(mrb, mrb_class_ptr(klass), NULL, &mrb_io_type));
337 pname = mrb_string_value_cstr(mrb, &cmd);
338 flags = mrb_io_modestr_to_flags(mrb, mrb_string_value_cstr(mrb, &mode));
340 doexec = (strcmp("-", pname) != 0);
341 opt_in = option_to_fd(mrb, opt, "in");
342 opt_out = option_to_fd(mrb, opt, "out");
343 opt_err = option_to_fd(mrb, opt, "err");
345 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
346 saAttr.bInheritHandle = TRUE;
347 saAttr.lpSecurityDescriptor = NULL;
349 if (flags & FMODE_READABLE) {
350 if (!CreatePipe(&ofd[0], &ofd[1], &saAttr, 0)
351 || !SetHandleInformation(ofd[0], HANDLE_FLAG_INHERIT, 0)) {
352 mrb_sys_fail(mrb, "pipe");
356 if (flags & FMODE_WRITABLE) {
357 if (!CreatePipe(&ifd[0], &ifd[1], &saAttr, 0)
358 || !SetHandleInformation(ifd[1], HANDLE_FLAG_INHERIT, 0)) {
359 mrb_sys_fail(mrb, "pipe");
364 ZeroMemory(&pi, sizeof(pi));
365 ZeroMemory(&si, sizeof(si));
367 si.dwFlags |= STARTF_USESHOWWINDOW;
368 si.wShowWindow = SW_HIDE;
369 si.dwFlags |= STARTF_USESTDHANDLES;
370 if (flags & FMODE_READABLE) {
371 si.hStdOutput = ofd[1];
372 si.hStdError = ofd[1];
374 if (flags & FMODE_WRITABLE) {
375 si.hStdInput = ifd[0];
378 NULL, (char*)pname, NULL, NULL,
379 TRUE, CREATE_NEW_PROCESS_GROUP, NULL, NULL, &si, &pi)) {
384 mrb_raisef(mrb, E_IO_ERROR, "command not found: %S", cmd);
386 CloseHandle(pi.hThread);
389 pid = pi.dwProcessId;
392 mrb_iv_set(mrb, io, mrb_intern_cstr(mrb, "@buf"), mrb_str_new_cstr(mrb, ""));
394 fptr = mrb_io_alloc(mrb);
395 fptr->fd = _open_osfhandle((intptr_t)ofd[0], 0);
396 fptr->fd2 = _open_osfhandle((intptr_t)ifd[1], 0);
398 fptr->readable = ((flags & FMODE_READABLE) != 0);
399 fptr->writable = ((flags & FMODE_WRITABLE) != 0);
402 DATA_TYPE(io) = &mrb_io_type;
406 #elif defined(TARGET_OS_IPHONE) && TARGET_OS_IPHONE
408 mrb_io_s_popen(mrb_state *mrb, mrb_value klass)
410 mrb_raise(mrb, E_NOTIMP_ERROR, "IO#popen is not supported on the platform");
411 return mrb_false_value();
415 mrb_io_s_popen(mrb_state *mrb, mrb_value klass)
417 mrb_value cmd, io, result;
418 mrb_value mode = mrb_str_new_cstr(mrb, "r");
419 mrb_value opt = mrb_hash_new(mrb);
423 int pid, flags, fd, write_fd = -1;
424 int pr[2] = { -1, -1 };
425 int pw[2] = { -1, -1 };
428 int opt_in, opt_out, opt_err;
430 mrb_get_args(mrb, "S|SH", &cmd, &mode, &opt);
431 io = mrb_obj_value(mrb_data_object_alloc(mrb, mrb_class_ptr(klass), NULL, &mrb_io_type));
433 pname = mrb_string_value_cstr(mrb, &cmd);
434 flags = mrb_io_modestr_to_flags(mrb, mrb_string_value_cstr(mrb, &mode));
436 doexec = (strcmp("-", pname) != 0);
437 opt_in = option_to_fd(mrb, opt, "in");
438 opt_out = option_to_fd(mrb, opt, "out");
439 opt_err = option_to_fd(mrb, opt, "err");
441 if (flags & FMODE_READABLE) {
442 if (pipe(pr) == -1) {
443 mrb_sys_fail(mrb, "pipe");
445 mrb_fd_cloexec(mrb, pr[0]);
446 mrb_fd_cloexec(mrb, pr[1]);
449 if (flags & FMODE_WRITABLE) {
450 if (pipe(pw) == -1) {
451 if (pr[0] != -1) close(pr[0]);
452 if (pr[1] != -1) close(pr[1]);
453 mrb_sys_fail(mrb, "pipe");
455 mrb_fd_cloexec(mrb, pw[0]);
456 mrb_fd_cloexec(mrb, pw[1]);
466 result = mrb_nil_value();
467 switch (pid = fork()) {
478 if (flags & FMODE_READABLE) {
485 if (flags & FMODE_WRITABLE) {
493 for (fd = 3; fd < NOFILE; fd++) {
496 mrb_proc_exec(pname);
497 mrb_raisef(mrb, E_IO_ERROR, "command not found: %S", cmd);
500 result = mrb_nil_value();
503 default: /* parent */
504 if ((flags & FMODE_READABLE) && (flags & FMODE_WRITABLE)) {
509 } else if (flags & FMODE_READABLE) {
517 mrb_iv_set(mrb, io, mrb_intern_cstr(mrb, "@buf"), mrb_str_new_cstr(mrb, ""));
519 fptr = mrb_io_alloc(mrb);
521 fptr->fd2 = write_fd;
523 fptr->readable = ((flags & FMODE_READABLE) != 0);
524 fptr->writable = ((flags & FMODE_WRITABLE) != 0);
527 DATA_TYPE(io) = &mrb_io_type;
534 if (flags & FMODE_READABLE) {
538 if (flags & FMODE_WRITABLE) {
543 mrb_sys_fail(mrb, "pipe_open failed.");
551 mrb_dup(mrb_state *mrb, int fd, mrb_bool *failed)
560 if (new_fd > 0) *failed = FALSE;
565 mrb_io_initialize_copy(mrb_state *mrb, mrb_value copy)
569 struct mrb_io *fptr_copy;
570 struct mrb_io *fptr_orig;
571 mrb_bool failed = TRUE;
573 mrb_get_args(mrb, "o", &orig);
574 fptr_orig = io_get_open_fptr(mrb, orig);
575 fptr_copy = (struct mrb_io *)DATA_PTR(copy);
576 if (fptr_orig == fptr_copy) return copy;
577 if (fptr_copy != NULL) {
578 fptr_finalize(mrb, fptr_copy, FALSE);
579 mrb_free(mrb, fptr_copy);
581 fptr_copy = (struct mrb_io *)mrb_io_alloc(mrb);
583 DATA_TYPE(copy) = &mrb_io_type;
584 DATA_PTR(copy) = fptr_copy;
586 buf = mrb_iv_get(mrb, orig, mrb_intern_cstr(mrb, "@buf"));
587 mrb_iv_set(mrb, copy, mrb_intern_cstr(mrb, "@buf"), buf);
589 fptr_copy->fd = mrb_dup(mrb, fptr_orig->fd, &failed);
591 mrb_sys_fail(mrb, 0);
593 mrb_fd_cloexec(mrb, fptr_copy->fd);
595 if (fptr_orig->fd2 != -1) {
596 fptr_copy->fd2 = mrb_dup(mrb, fptr_orig->fd2, &failed);
598 close(fptr_copy->fd);
599 mrb_sys_fail(mrb, 0);
601 mrb_fd_cloexec(mrb, fptr_copy->fd2);
604 fptr_copy->pid = fptr_orig->pid;
605 fptr_copy->readable = fptr_orig->readable;
606 fptr_copy->writable = fptr_orig->writable;
607 fptr_copy->sync = fptr_orig->sync;
608 fptr_copy->is_socket = fptr_orig->is_socket;
614 mrb_io_initialize(mrb_state *mrb, mrb_value io)
621 mode = opt = mrb_nil_value();
623 mrb_get_args(mrb, "i|So", &fd, &mode, &opt);
624 if (mrb_nil_p(mode)) {
625 mode = mrb_str_new_cstr(mrb, "r");
627 if (mrb_nil_p(opt)) {
628 opt = mrb_hash_new(mrb);
631 flags = mrb_io_modestr_to_flags(mrb, mrb_string_value_cstr(mrb, &mode));
633 mrb_iv_set(mrb, io, mrb_intern_cstr(mrb, "@buf"), mrb_str_new_cstr(mrb, ""));
635 fptr = (struct mrb_io *)DATA_PTR(io);
637 fptr_finalize(mrb, fptr, TRUE);
640 fptr = mrb_io_alloc(mrb);
642 DATA_TYPE(io) = &mrb_io_type;
646 fptr->readable = ((flags & FMODE_READABLE) != 0);
647 fptr->writable = ((flags & FMODE_WRITABLE) != 0);
653 fptr_finalize(mrb_state *mrb, struct mrb_io *fptr, int quiet)
663 if (fptr->is_socket) {
664 if (closesocket(fptr->fd) != 0) {
665 saved_errno = WSAGetLastError();
670 if (fptr->fd != -1) {
671 if (close(fptr->fd) == -1) {
679 if (close(fptr->fd2) == -1) {
680 if (saved_errno == 0) {
687 if (fptr->pid != 0) {
688 #if !defined(_WIN32) && !defined(_WIN64)
692 pid = waitpid(fptr->pid, &status, 0);
693 } while (pid == -1 && errno == EINTR);
694 if (!quiet && pid == fptr->pid) {
695 io_set_process_status(mrb, pid, status);
698 HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, fptr->pid);
700 if (WaitForSingleObject(h, INFINITE) && GetExitCodeProcess(h, &status))
702 io_set_process_status(mrb, fptr->pid, (int)status);
706 /* Note: we don't raise an exception when waitpid(3) fails */
709 if (!quiet && saved_errno != 0) {
711 mrb_sys_fail(mrb, "fptr_finalize failed.");
716 mrb_io_check_readable(mrb_state *mrb, mrb_value self)
718 struct mrb_io *fptr = io_get_open_fptr(mrb, self);
719 if (! fptr->readable) {
720 mrb_raise(mrb, E_IO_ERROR, "not opened for reading");
722 return mrb_nil_value();
726 mrb_io_isatty(mrb_state *mrb, mrb_value self)
730 fptr = io_get_open_fptr(mrb, self);
731 if (isatty(fptr->fd) == 0)
732 return mrb_false_value();
733 return mrb_true_value();
737 mrb_io_s_for_fd(mrb_state *mrb, mrb_value klass)
739 struct RClass *c = mrb_class_ptr(klass);
740 enum mrb_vtype ttype = MRB_INSTANCE_TT(c);
743 /* copied from mrb_instance_alloc() */
744 if (ttype == 0) ttype = MRB_TT_OBJECT;
745 obj = mrb_obj_value((struct RObject*)mrb_obj_alloc(mrb, ttype, c));
746 return mrb_io_initialize(mrb, obj);
750 mrb_io_s_sysclose(mrb_state *mrb, mrb_value klass)
753 mrb_get_args(mrb, "i", &fd);
754 if (close((int)fd) == -1) {
755 mrb_sys_fail(mrb, "close");
757 return mrb_fixnum_value(0);
761 mrb_cloexec_open(mrb_state *mrb, const char *pathname, mrb_int flags, mrb_int mode)
764 int fd, retry = FALSE;
765 char* fname = mrb_locale_from_utf8(pathname, -1);
768 /* O_CLOEXEC is available since Linux 2.6.23. Linux 2.6.18 silently ignore it. */
770 #elif defined O_NOINHERIT
771 flags |= O_NOINHERIT;
774 fd = open(fname, (int)flags, (fmode_t)mode);
780 mrb_garbage_collect(mrb);
786 emsg = mrb_format(mrb, "open %S", mrb_str_new_cstr(mrb, pathname));
787 mrb_str_modify(mrb, mrb_str_ptr(emsg));
788 mrb_sys_fail(mrb, RSTRING_PTR(emsg));
790 mrb_locale_free(fname);
793 mrb_fd_cloexec(mrb, fd);
799 mrb_io_s_sysopen(mrb_state *mrb, mrb_value klass)
801 mrb_value path = mrb_nil_value();
802 mrb_value mode = mrb_nil_value();
803 mrb_int fd, perm = -1;
807 mrb_get_args(mrb, "S|Si", &path, &mode, &perm);
808 if (mrb_nil_p(mode)) {
809 mode = mrb_str_new_cstr(mrb, "r");
815 pat = mrb_string_value_cstr(mrb, &path);
816 flags = mrb_io_modestr_to_flags(mrb, mrb_string_value_cstr(mrb, &mode));
817 modenum = mrb_io_flags_to_modenum(mrb, flags);
818 fd = mrb_cloexec_open(mrb, pat, modenum, perm);
819 return mrb_fixnum_value(fd);
823 mrb_io_sysread(mrb_state *mrb, mrb_value io)
826 mrb_value buf = mrb_nil_value();
830 mrb_get_args(mrb, "i|S", &maxlen, &buf);
832 mrb_raise(mrb, E_ARGUMENT_ERROR, "negative expanding string size");
834 else if (maxlen == 0) {
835 return mrb_str_new(mrb, NULL, maxlen);
838 if (mrb_nil_p(buf)) {
839 buf = mrb_str_new(mrb, NULL, maxlen);
842 if (RSTRING_LEN(buf) != maxlen) {
843 buf = mrb_str_resize(mrb, buf, maxlen);
845 mrb_str_modify(mrb, RSTRING(buf));
848 fptr = (struct mrb_io *)io_get_open_fptr(mrb, io);
849 if (!fptr->readable) {
850 mrb_raise(mrb, E_IO_ERROR, "not opened for reading");
852 ret = read(fptr->fd, RSTRING_PTR(buf), (fsize_t)maxlen);
856 buf = mrb_str_new_cstr(mrb, "");
858 mrb_raise(mrb, E_EOF_ERROR, "sysread failed: End of File");
862 mrb_sys_fail(mrb, "sysread failed");
865 if (RSTRING_LEN(buf) != ret) {
866 buf = mrb_str_resize(mrb, buf, ret);
875 mrb_io_sysseek(mrb_state *mrb, mrb_value io)
879 mrb_int offset, whence = -1;
881 mrb_get_args(mrb, "i|i", &offset, &whence);
886 fptr = io_get_open_fptr(mrb, io);
887 pos = lseek(fptr->fd, (off_t)offset, (int)whence);
889 mrb_sys_fail(mrb, "sysseek");
891 if (pos > MRB_INT_MAX) {
892 #ifndef MRB_WITHOUT_FLOAT
893 return mrb_float_value(mrb, (mrb_float)pos);
895 mrb_raise(mrb, E_IO_ERROR, "sysseek reached too far for MRB_WITHOUT_FLOAT");
898 return mrb_fixnum_value(pos);
903 mrb_io_syswrite(mrb_state *mrb, mrb_value io)
909 fptr = io_get_open_fptr(mrb, io);
910 if (! fptr->writable) {
911 mrb_raise(mrb, E_IO_ERROR, "not opened for writing");
914 mrb_get_args(mrb, "S", &str);
915 if (mrb_type(str) != MRB_TT_STRING) {
916 buf = mrb_funcall(mrb, str, "to_s", 0);
921 if (fptr->fd2 == -1) {
926 length = write(fd, RSTRING_PTR(buf), (fsize_t)RSTRING_LEN(buf));
928 mrb_sys_fail(mrb, 0);
931 return mrb_fixnum_value(length);
935 mrb_io_close(mrb_state *mrb, mrb_value self)
938 fptr = io_get_open_fptr(mrb, self);
939 fptr_finalize(mrb, fptr, FALSE);
940 return mrb_nil_value();
944 mrb_io_close_write(mrb_state *mrb, mrb_value self)
947 fptr = io_get_open_fptr(mrb, self);
948 if (close((int)fptr->fd2) == -1) {
949 mrb_sys_fail(mrb, "close");
951 return mrb_nil_value();
955 mrb_io_closed(mrb_state *mrb, mrb_value io)
958 fptr = (struct mrb_io *)mrb_get_datatype(mrb, io, &mrb_io_type);
959 if (fptr == NULL || fptr->fd >= 0) {
960 return mrb_false_value();
963 return mrb_true_value();
967 mrb_io_pid(mrb_state *mrb, mrb_value io)
970 fptr = io_get_open_fptr(mrb, io);
973 return mrb_fixnum_value(fptr->pid);
976 return mrb_nil_value();
979 static struct timeval
980 time2timeval(mrb_state *mrb, mrb_value time)
982 struct timeval t = { 0, 0 };
984 switch (mrb_type(time)) {
986 t.tv_sec = (ftime_t)mrb_fixnum(time);
990 #ifndef MRB_WITHOUT_FLOAT
992 t.tv_sec = (ftime_t)mrb_float(time);
993 t.tv_usec = (fsuseconds_t)((mrb_float(time) - t.tv_sec) * 1000000.0);
998 mrb_raise(mrb, E_TYPE_ERROR, "wrong argument class");
1005 mrb_io_read_data_pending(mrb_state *mrb, mrb_value io)
1007 mrb_value buf = mrb_iv_get(mrb, io, mrb_intern_cstr(mrb, "@buf"));
1008 if (mrb_type(buf) == MRB_TT_STRING && RSTRING_LEN(buf) > 0) {
1014 #if !defined(_WIN32) && !(defined(TARGET_OS_IPHONE) && TARGET_OS_IPHONE)
1016 mrb_io_s_pipe(mrb_state *mrb, mrb_value klass)
1018 mrb_value r = mrb_nil_value();
1019 mrb_value w = mrb_nil_value();
1020 struct mrb_io *fptr_r;
1021 struct mrb_io *fptr_w;
1024 if (mrb_pipe(mrb, pipes) == -1) {
1025 mrb_sys_fail(mrb, "pipe");
1028 r = mrb_obj_value(mrb_data_object_alloc(mrb, mrb_class_ptr(klass), NULL, &mrb_io_type));
1029 mrb_iv_set(mrb, r, mrb_intern_cstr(mrb, "@buf"), mrb_str_new_cstr(mrb, ""));
1030 fptr_r = mrb_io_alloc(mrb);
1031 fptr_r->fd = pipes[0];
1032 fptr_r->readable = 1;
1033 fptr_r->writable = 0;
1035 DATA_TYPE(r) = &mrb_io_type;
1036 DATA_PTR(r) = fptr_r;
1038 w = mrb_obj_value(mrb_data_object_alloc(mrb, mrb_class_ptr(klass), NULL, &mrb_io_type));
1039 mrb_iv_set(mrb, w, mrb_intern_cstr(mrb, "@buf"), mrb_str_new_cstr(mrb, ""));
1040 fptr_w = mrb_io_alloc(mrb);
1041 fptr_w->fd = pipes[1];
1042 fptr_w->readable = 0;
1043 fptr_w->writable = 1;
1045 DATA_TYPE(w) = &mrb_io_type;
1046 DATA_PTR(w) = fptr_w;
1048 return mrb_assoc_new(mrb, r, w);
1053 mrb_io_s_select(mrb_state *mrb, mrb_value klass)
1057 mrb_value read, read_io, write, except, timeout, list;
1058 struct timeval *tp, timerec;
1059 fd_set pset, rset, wset, eset;
1060 fd_set *rp, *wp, *ep;
1061 struct mrb_io *fptr;
1065 int interrupt_flag = 0;
1068 mrb_get_args(mrb, "*", &argv, &argc);
1070 if (argc < 1 || argc > 4) {
1071 mrb_raisef(mrb, E_ARGUMENT_ERROR, "wrong number of arguments (%S for 1..4)", mrb_fixnum_value(argc));
1074 timeout = mrb_nil_value();
1075 except = mrb_nil_value();
1076 write = mrb_nil_value();
1085 if (mrb_nil_p(timeout)) {
1088 timerec = time2timeval(mrb, timeout);
1093 if (!mrb_nil_p(read)) {
1094 mrb_check_type(mrb, read, MRB_TT_ARRAY);
1097 for (i = 0; i < RARRAY_LEN(read); i++) {
1098 read_io = RARRAY_PTR(read)[i];
1099 fptr = io_get_open_fptr(mrb, read_io);
1100 FD_SET(fptr->fd, rp);
1101 if (mrb_io_read_data_pending(mrb, read_io)) {
1103 FD_SET(fptr->fd, &pset);
1109 timerec.tv_sec = timerec.tv_usec = 0;
1116 if (!mrb_nil_p(write)) {
1117 mrb_check_type(mrb, write, MRB_TT_ARRAY);
1120 for (i = 0; i < RARRAY_LEN(write); i++) {
1121 fptr = io_get_open_fptr(mrb, RARRAY_PTR(write)[i]);
1122 FD_SET(fptr->fd, wp);
1125 if (fptr->fd2 >= 0) {
1126 FD_SET(fptr->fd2, wp);
1127 if (max < fptr->fd2)
1135 if (!mrb_nil_p(except)) {
1136 mrb_check_type(mrb, except, MRB_TT_ARRAY);
1139 for (i = 0; i < RARRAY_LEN(except); i++) {
1140 fptr = io_get_open_fptr(mrb, RARRAY_PTR(except)[i]);
1141 FD_SET(fptr->fd, ep);
1144 if (fptr->fd2 >= 0) {
1145 FD_SET(fptr->fd2, ep);
1146 if (max < fptr->fd2)
1157 n = select(max, rp, wp, ep, tp);
1160 mrb_sys_fail(mrb, "select failed");
1166 if (!pending && n == 0)
1167 return mrb_nil_value();
1169 result = mrb_ary_new_capa(mrb, 3);
1170 mrb_ary_push(mrb, result, rp? mrb_ary_new(mrb) : mrb_ary_new_capa(mrb, 0));
1171 mrb_ary_push(mrb, result, wp? mrb_ary_new(mrb) : mrb_ary_new_capa(mrb, 0));
1172 mrb_ary_push(mrb, result, ep? mrb_ary_new(mrb) : mrb_ary_new_capa(mrb, 0));
1174 if (interrupt_flag == 0) {
1176 list = RARRAY_PTR(result)[0];
1177 for (i = 0; i < RARRAY_LEN(read); i++) {
1178 fptr = io_get_open_fptr(mrb, RARRAY_PTR(read)[i]);
1179 if (FD_ISSET(fptr->fd, rp) ||
1180 FD_ISSET(fptr->fd, &pset)) {
1181 mrb_ary_push(mrb, list, RARRAY_PTR(read)[i]);
1187 list = RARRAY_PTR(result)[1];
1188 for (i = 0; i < RARRAY_LEN(write); i++) {
1189 fptr = io_get_open_fptr(mrb, RARRAY_PTR(write)[i]);
1190 if (FD_ISSET(fptr->fd, wp)) {
1191 mrb_ary_push(mrb, list, RARRAY_PTR(write)[i]);
1192 } else if (fptr->fd2 >= 0 && FD_ISSET(fptr->fd2, wp)) {
1193 mrb_ary_push(mrb, list, RARRAY_PTR(write)[i]);
1199 list = RARRAY_PTR(result)[2];
1200 for (i = 0; i < RARRAY_LEN(except); i++) {
1201 fptr = io_get_open_fptr(mrb, RARRAY_PTR(except)[i]);
1202 if (FD_ISSET(fptr->fd, ep)) {
1203 mrb_ary_push(mrb, list, RARRAY_PTR(except)[i]);
1204 } else if (fptr->fd2 >= 0 && FD_ISSET(fptr->fd2, ep)) {
1205 mrb_ary_push(mrb, list, RARRAY_PTR(except)[i]);
1215 mrb_io_fileno(mrb_state *mrb, mrb_value io)
1217 struct mrb_io *fptr;
1218 fptr = io_get_open_fptr(mrb, io);
1219 return mrb_fixnum_value(fptr->fd);
1223 mrb_io_close_on_exec_p(mrb_state *mrb, mrb_value self)
1225 #if defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
1226 struct mrb_io *fptr;
1229 fptr = io_get_open_fptr(mrb, self);
1231 if (fptr->fd2 >= 0) {
1232 if ((ret = fcntl(fptr->fd2, F_GETFD)) == -1) mrb_sys_fail(mrb, "F_GETFD failed");
1233 if (!(ret & FD_CLOEXEC)) return mrb_false_value();
1236 if ((ret = fcntl(fptr->fd, F_GETFD)) == -1) mrb_sys_fail(mrb, "F_GETFD failed");
1237 if (!(ret & FD_CLOEXEC)) return mrb_false_value();
1238 return mrb_true_value();
1241 mrb_raise(mrb, E_NOTIMP_ERROR, "IO#close_on_exec? is not supported on the platform");
1242 return mrb_false_value();
1247 mrb_io_set_close_on_exec(mrb_state *mrb, mrb_value self)
1249 #if defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
1250 struct mrb_io *fptr;
1254 fptr = io_get_open_fptr(mrb, self);
1255 mrb_get_args(mrb, "b", &b);
1256 flag = b ? FD_CLOEXEC : 0;
1258 if (fptr->fd2 >= 0) {
1259 if ((ret = fcntl(fptr->fd2, F_GETFD)) == -1) mrb_sys_fail(mrb, "F_GETFD failed");
1260 if ((ret & FD_CLOEXEC) != flag) {
1261 ret = (ret & ~FD_CLOEXEC) | flag;
1262 ret = fcntl(fptr->fd2, F_SETFD, ret);
1264 if (ret == -1) mrb_sys_fail(mrb, "F_SETFD failed");
1268 if ((ret = fcntl(fptr->fd, F_GETFD)) == -1) mrb_sys_fail(mrb, "F_GETFD failed");
1269 if ((ret & FD_CLOEXEC) != flag) {
1270 ret = (ret & ~FD_CLOEXEC) | flag;
1271 ret = fcntl(fptr->fd, F_SETFD, ret);
1272 if (ret == -1) mrb_sys_fail(mrb, "F_SETFD failed");
1275 return mrb_bool_value(b);
1277 mrb_raise(mrb, E_NOTIMP_ERROR, "IO#close_on_exec= is not supported on the platform");
1278 return mrb_nil_value();
1283 mrb_io_set_sync(mrb_state *mrb, mrb_value self)
1285 struct mrb_io *fptr;
1288 fptr = io_get_open_fptr(mrb, self);
1289 mrb_get_args(mrb, "b", &b);
1291 return mrb_bool_value(b);
1295 mrb_io_sync(mrb_state *mrb, mrb_value self)
1297 struct mrb_io *fptr;
1298 fptr = io_get_open_fptr(mrb, self);
1299 return mrb_bool_value(fptr->sync);
1303 mrb_init_io(mrb_state *mrb)
1307 io = mrb_define_class(mrb, "IO", mrb->object_class);
1308 MRB_SET_INSTANCE_TT(io, MRB_TT_DATA);
1310 mrb_include_module(mrb, io, mrb_module_get(mrb, "Enumerable")); /* 15.2.20.3 */
1311 mrb_define_class_method(mrb, io, "_popen", mrb_io_s_popen, MRB_ARGS_ANY());
1312 mrb_define_class_method(mrb, io, "_sysclose", mrb_io_s_sysclose, MRB_ARGS_REQ(1));
1313 mrb_define_class_method(mrb, io, "for_fd", mrb_io_s_for_fd, MRB_ARGS_ANY());
1314 mrb_define_class_method(mrb, io, "select", mrb_io_s_select, MRB_ARGS_ANY());
1315 mrb_define_class_method(mrb, io, "sysopen", mrb_io_s_sysopen, MRB_ARGS_ANY());
1316 #if !defined(_WIN32) && !(defined(TARGET_OS_IPHONE) && TARGET_OS_IPHONE)
1317 mrb_define_class_method(mrb, io, "_pipe", mrb_io_s_pipe, MRB_ARGS_NONE());
1320 mrb_define_method(mrb, io, "initialize", mrb_io_initialize, MRB_ARGS_ANY()); /* 15.2.20.5.21 (x)*/
1321 mrb_define_method(mrb, io, "initialize_copy", mrb_io_initialize_copy, MRB_ARGS_REQ(1));
1322 mrb_define_method(mrb, io, "_check_readable", mrb_io_check_readable, MRB_ARGS_NONE());
1323 mrb_define_method(mrb, io, "isatty", mrb_io_isatty, MRB_ARGS_NONE());
1324 mrb_define_method(mrb, io, "sync", mrb_io_sync, MRB_ARGS_NONE());
1325 mrb_define_method(mrb, io, "sync=", mrb_io_set_sync, MRB_ARGS_REQ(1));
1326 mrb_define_method(mrb, io, "sysread", mrb_io_sysread, MRB_ARGS_ANY());
1327 mrb_define_method(mrb, io, "sysseek", mrb_io_sysseek, MRB_ARGS_REQ(1));
1328 mrb_define_method(mrb, io, "syswrite", mrb_io_syswrite, MRB_ARGS_REQ(1));
1329 mrb_define_method(mrb, io, "close", mrb_io_close, MRB_ARGS_NONE()); /* 15.2.20.5.1 */
1330 mrb_define_method(mrb, io, "close_write", mrb_io_close_write, MRB_ARGS_NONE());
1331 mrb_define_method(mrb, io, "close_on_exec=", mrb_io_set_close_on_exec, MRB_ARGS_REQ(1));
1332 mrb_define_method(mrb, io, "close_on_exec?", mrb_io_close_on_exec_p, MRB_ARGS_NONE());
1333 mrb_define_method(mrb, io, "closed?", mrb_io_closed, MRB_ARGS_NONE()); /* 15.2.20.5.2 */
1334 mrb_define_method(mrb, io, "pid", mrb_io_pid, MRB_ARGS_NONE()); /* 15.2.20.5.2 */
1335 mrb_define_method(mrb, io, "fileno", mrb_io_fileno, MRB_ARGS_NONE());
1338 mrb_gv_set(mrb, mrb_intern_cstr(mrb, "$/"), mrb_str_new_cstr(mrb, "\n"));