michael@0: // Copyright (c) 2012, Google Inc.
michael@0: // All rights reserved.
michael@0: //
michael@0: // Redistribution and use in source and binary forms, with or without
michael@0: // modification, are permitted provided that the following conditions are
michael@0: // met:
michael@0: //
michael@0: //     * Redistributions of source code must retain the above copyright
michael@0: // notice, this list of conditions and the following disclaimer.
michael@0: //     * Redistributions in binary form must reproduce the above
michael@0: // copyright notice, this list of conditions and the following disclaimer
michael@0: // in the documentation and/or other materials provided with the
michael@0: // distribution.
michael@0: //     * Neither the name of Google Inc. nor the names of its
michael@0: // contributors may be used to endorse or promote products derived from
michael@0: // this software without specific prior written permission.
michael@0: //
michael@0: // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
michael@0: // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
michael@0: // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
michael@0: // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
michael@0: // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
michael@0: // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
michael@0: // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
michael@0: // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
michael@0: // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
michael@0: // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
michael@0: // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0: 
michael@0: // Android doesn't provide mkdtemp(). Keep this implementation in an
michael@0: // C++ anonymous namespace to avoid conflicts on Chromium (which
michael@0: // already provides an extern "C" mkdtemp function).
michael@0: //
michael@0: // The reason this is inlined here is to avoid linking a new object file
michael@0: // into each unit test program (i.e. keep build files simple).
michael@0: 
michael@0: #ifndef GOOGLE_BREAKPAD_COMMON_ANDROID_TESTING_MKDTEMP_H
michael@0: #define GOOGLE_BREAKPAD_COMMON_ANDROID_TESTING_MKDTEMP_H
michael@0: 
michael@0: #include <assert.h>
michael@0: #include <errno.h>
michael@0: #include <stdlib.h>
michael@0: #include <stdio.h>
michael@0: #include <string.h>
michael@0: #include <sys/stat.h>
michael@0: 
michael@0: namespace {
michael@0: 
michael@0: char* mkdtemp(char* path) {
michael@0:   if (path == NULL) {
michael@0:     errno = EINVAL;
michael@0:     return NULL;
michael@0:   }
michael@0: 
michael@0:   // 'path' must be terminated with six 'X'
michael@0:   const char kSuffix[] = "XXXXXX";
michael@0:   const size_t kSuffixLen = strlen(kSuffix);
michael@0:   char* path_end = path + strlen(path);
michael@0: 
michael@0:   if (static_cast<size_t>(path_end - path) < kSuffixLen ||
michael@0:       memcmp(path_end - kSuffixLen, kSuffix, kSuffixLen) != 0) {
michael@0:     errno = EINVAL;
michael@0:     return NULL;
michael@0:   }
michael@0: 
michael@0:   // If 'path' contains a directory separator, check that it exists to
michael@0:   // avoid looping later.
michael@0:   char* sep = strrchr(path, '/');
michael@0:   if (sep != NULL) {
michael@0:     struct stat st;
michael@0:     int ret;
michael@0:     *sep = '\0';  // temporarily zero-terminate the dirname.
michael@0:     ret = stat(path, &st);
michael@0:     *sep = '/';   // restore full path.
michael@0:     if (ret < 0)
michael@0:       return NULL;
michael@0:     if (!S_ISDIR(st.st_mode)) {
michael@0:       errno = ENOTDIR;
michael@0:       return NULL;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   // Loop. On each iteration, replace the XXXXXX suffix with a random
michael@0:   // number.
michael@0:   int tries;
michael@0:   for (tries = 128; tries > 0; tries--) {
michael@0:     int random = rand() % 1000000;
michael@0: 
michael@0:     snprintf(path_end - kSuffixLen, kSuffixLen + 1, "%0d", random);
michael@0:     if (mkdir(path, 0700) == 0)
michael@0:       return path;  // Success
michael@0: 
michael@0:     if (errno != EEXIST)
michael@0:       return NULL;
michael@0:   }
michael@0: 
michael@0:   assert(errno == EEXIST);
michael@0:   return NULL;
michael@0: }
michael@0: 
michael@0: }  // namespace
michael@0: 
michael@0: #endif  // GOOGLE_BREAKPAD_COMMON_ANDROID_TESTING_MKDTEMP_H