ext/mkdtemp.c

   1 // Copyright 2007-2010 Wincent Colaiuta. All rights reserved.
   2 //
   3 // Redistribution and use in source and binary forms, with or without
   4 // modification, are permitted provided that the following conditions are met:
   5 //
   6 // 1. Redistributions of source code must retain the above copyright notice,
   7 //    this list of conditions and the following disclaimer.
   8 // 2. Redistributions in binary form must reproduce the above copyright notice,
   9 //    this list of conditions and the following disclaimer in the documentation
  10 //    and/or other materials provided with the distribution.
  11 //
  12 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  13 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  14 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  15 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
  16 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  17 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  18 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  19 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  20 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  21 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  22 // POSSIBILITY OF SUCH DAMAGE.
  23
  24 #include <ruby.h>
  25 #include <errno.h>
  26 #include <unistd.h>
  27 #include "ruby_compat.h"
  28
  29 // helper function needed by rb_iterate; see:
  30 //  http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/144100
  31 VALUE call_chdir(VALUE dir)
  32 {
  33     return rb_funcall(rb_cDir, rb_intern("chdir"), 1, dir);
  34 }
  35
  36 // helper function needed by rb_iterate
  37 VALUE yield_block(VALUE ignored, VALUE block)
  38 {
  39     return rb_funcall(block, rb_intern("call"), 0);
  40 }
  41 /*
  42  * Document-method: mkdtemp
  43  * call-seq:
  44  *     Dir.mkdtemp([string]) -> string
  45  *     Dir.mkdtemp([string]) { ... } -> string
  46  *
  47  * This method securely creates temporary directories. It is a wrapper for the
  48  * mkdtemp() function in the standard C library. It takes an optional String
  49  * parameter as a template describing the desired form of the directory name
  50  * and overwriting the template in-place; if no template is supplied then
  51  * "/tmp/temp.XXXXXX" is used as a default.
  52  *
  53  * If supplied a block, performs a Dir.chdir into the created directory and
  54  * yields to the block:
  55  *
  56  *      # this:            # is a shorthand for:
  57  *      Dir.mkdtemp do     #   dir = Dir.mkdtemp
  58  *        puts Dir.pwd     #   Dir.chdir dir do
  59  *      end                #     puts Dir.pwd
  60  *                         #   end
  61  *
  62  * Note that the exact implementation of mkdtemp() may vary depending on the
  63  * target system. For example, on Mac OS X at the time of writing, the man page
  64  * states that the template may contain "some number" of "Xs" on the end of the
  65  * string, whereas on Red Hat Enterprise Linux it states that the template
  66  * suffix "must be XXXXXX".
  67  */
  68 static VALUE dir_mkdtemp_m(int argc, VALUE *argv, VALUE self)
  69 {
  70     VALUE template, block;
  71     char *c_template;
  72     char *path;
  73
  74     // process arguments
  75     if (rb_scan_args(argc, argv, "01&", &template, &block) == 0)    // 0 mandatory, 1 optional, 1 block
  76         template = Qnil;                                            // default to nil if no argument passed
  77     if (NIL_P(template))
  78         template = rb_str_new2("/tmp/temp.XXXXXX");                 // fallback to this template if passed nil
  79     SafeStringValue(template);                                      // raises if template is tainted and SAFE level > 0
  80     template = StringValue(template);                               // duck typing support
  81
  82     // create temporary storage
  83     c_template = malloc(RSTRING_LEN(template) + 1);
  84     if (!c_template)
  85         rb_raise(rb_eNoMemError, "failed to allocate %ld bytes of template storage", RSTRING_LEN(template) + 1);
  86     strncpy(c_template, RSTRING_PTR(template), RSTRING_LEN(template));
  87     c_template[RSTRING_LEN(template)] = 0;              // NUL-terminate
  88
  89     // fill out template
  90     path = mkdtemp(c_template);
  91     if (path)
  92         template = rb_str_new2(path);
  93     free(c_template);
  94     if (path == NULL)
  95         rb_raise(rb_eSystemCallError, "mkdtemp failed (error #%d: %s)", errno, strerror(errno));
  96
  97     // yield to block if given, inside Dir.chdir
  98     if (rb_block_given_p() == Qtrue)
  99         rb_iterate(call_chdir, template, yield_block, block);
 100     return template;
 101 }
 102
 103 void Init_mkdtemp()
 104 {
 105 #if 0
 106     // for YARD, need to fake this here
 107     VALUE rb_cDir = rb_define_class("Dir", rb_cObject);
 108 #endif
 109     rb_define_singleton_method(rb_cDir, "mkdtemp", dir_mkdtemp_m, -1);
 110 }