73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync/* $Id$ */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync/** @file
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * IPRT Testcase - Temporary files and directories.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync/*
c7814cf6e1240a519cbec0441e033d0e2470ed00vboxsync * Copyright (C) 2009-2012 Oracle Corporation
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync *
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * available from http://www.virtualbox.org. This file is free software;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * you can redistribute it and/or modify it under the terms of the GNU
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * General Public License (GPL) as published by the Free Software
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync *
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * The contents of this file may alternatively be used under the terms
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * of the Common Development and Distribution License Version 1.0
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * VirtualBox OSE distribution, in which case the provisions of the
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * CDDL are applicable instead of those of the GPL.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync *
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * You may elect to license modified versions of this file under the
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * terms and conditions of either the GPL or the CDDL or both.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync/*******************************************************************************
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync* Header Files *
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync*******************************************************************************/
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/dir.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/file.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/path.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/err.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/initterm.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/mem.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/param.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/path.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/stream.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/string.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync#include <iprt/test.h>
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync/*******************************************************************************
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync* Global Variables *
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync*******************************************************************************/
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsyncstatic char g_szTempPath[RTPATH_MAX - 50];
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsyncstatic void tstObjectCreateTemp(const char *pszSubTest, const char *pszTemplate, bool fFile, RTFMODE fMode, unsigned cTimes, bool fSkipXCheck)
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync{
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTestISub(pszSubTest);
f455158a85f95b4886cbe7154b067144b815c125vboxsync const char *pcszAPI = fFile ? "RTFileCreateTemp" : "RTDirCreateTemp";
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /* Allocate the result array. */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync char **papszNames = (char **)RTMemTmpAllocZ(cTimes * sizeof(char *));
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTESTI_CHECK_RETV(papszNames != NULL);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /* The test loop. */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync unsigned i;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync for (i = 0; i < cTimes; i++)
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync {
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync int rc;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync char szName[RTPATH_MAX];
742e772d8dd4aaec6a12e08dbc0fb170270bc96bvboxsync RTFMODE fModeFinal;
742e772d8dd4aaec6a12e08dbc0fb170270bc96bvboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTESTI_CHECK_RC(rc = RTPathAppend(strcpy(szName, g_szTempPath), sizeof(szName), pszTemplate), VINF_SUCCESS);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync if (RT_FAILURE(rc))
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync break;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTESTI_CHECK(papszNames[i] = RTStrDup(szName));
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync if (!papszNames[i])
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync break;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsync rc = fFile
f455158a85f95b4886cbe7154b067144b815c125vboxsync ? RTFileCreateTemp(papszNames[i], fMode)
f455158a85f95b4886cbe7154b067144b815c125vboxsync : RTDirCreateTemp(papszNames[i], fMode);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync if (rc != VINF_SUCCESS)
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync {
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTestIFailed("%s(%s, %#o) call #%u -> %Rrc\n", pcszAPI, szName, (int)fMode, i, rc);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTStrFree(papszNames[i]);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync papszNames[i] = NULL;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync break;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync }
34b52389fec71ab796205b7d4972778499c239f9vboxsync /* Check that the final permissions are not more permissive than
34b52389fec71ab796205b7d4972778499c239f9vboxsync * the ones requested (less permissive is fine, c.f. umask etc.).
34b52389fec71ab796205b7d4972778499c239f9vboxsync * I mask out the group as I am not sure how we deal with that on
34b52389fec71ab796205b7d4972778499c239f9vboxsync * Windows. */
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_RC_OK(rc = RTPathGetMode(papszNames[i], &fModeFinal));
f455158a85f95b4886cbe7154b067144b815c125vboxsync if (RT_SUCCESS(rc))
f455158a85f95b4886cbe7154b067144b815c125vboxsync {
742e772d8dd4aaec6a12e08dbc0fb170270bc96bvboxsync fModeFinal &= (RTFS_UNIX_IRWXU | RTFS_UNIX_IRWXO);
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_MSG((fModeFinal & ~fMode) == 0,
f455158a85f95b4886cbe7154b067144b815c125vboxsync ("%s: szName %s\nfModeFinal ~= %#o, expected %#o\n",
f455158a85f95b4886cbe7154b067144b815c125vboxsync pcszAPI, szName, fModeFinal, (int)fMode));
f455158a85f95b4886cbe7154b067144b815c125vboxsync }
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTestIPrintf(RTTESTLVL_DEBUG, "%s: %s\n", pcszAPI, papszNames[i]);
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_MSG(strlen(szName) == strlen(papszNames[i]), ("%s: szName %s\nReturned %s\n", pcszAPI, szName, papszNames[i]));
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync if (!fSkipXCheck)
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_MSG(strchr(RTPathFilename(papszNames[i]), 'X') == NULL, ("%s: szName %s\nReturned %s\n", pcszAPI, szName, papszNames[i]));
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync }
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /* cleanup */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync while (i-- > 0)
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync {
f455158a85f95b4886cbe7154b067144b815c125vboxsync if (fFile)
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_RC(RTFileDelete(papszNames[i]), VINF_SUCCESS);
f455158a85f95b4886cbe7154b067144b815c125vboxsync else
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTTESTI_CHECK_RC(RTDirRemove(papszNames[i]), VINF_SUCCESS);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTStrFree(papszNames[i]);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync }
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTMemTmpFree(papszNames);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync}
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsyncstatic void tstFileCreateTemp(const char *pszSubTest, const char *pszTemplate, RTFMODE fMode, unsigned cTimes, bool fSkipXCheck)
f455158a85f95b4886cbe7154b067144b815c125vboxsync{
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstObjectCreateTemp(pszSubTest, pszTemplate, true /* fFile */, fMode,
f455158a85f95b4886cbe7154b067144b815c125vboxsync cTimes, fSkipXCheck);
f455158a85f95b4886cbe7154b067144b815c125vboxsync}
f455158a85f95b4886cbe7154b067144b815c125vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsyncstatic void tstDirCreateTemp(const char *pszSubTest, const char *pszTemplate, RTFMODE fMode, unsigned cTimes, bool fSkipXCheck)
f455158a85f95b4886cbe7154b067144b815c125vboxsync{
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstObjectCreateTemp(pszSubTest, pszTemplate, false /* fFile */, fMode,
f455158a85f95b4886cbe7154b067144b815c125vboxsync cTimes, fSkipXCheck);
f455158a85f95b4886cbe7154b067144b815c125vboxsync}
f455158a85f95b4886cbe7154b067144b815c125vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsyncstatic void tstBothCreateTemp(const char *pszSubTest, const char *pszTemplate, RTFMODE fMode, unsigned cTimes, bool fSkipXCheck)
f455158a85f95b4886cbe7154b067144b815c125vboxsync{
f455158a85f95b4886cbe7154b067144b815c125vboxsync char pszSubTestLong[128];
64eea8161bef2aa3c6516481383c830bca27abfevboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTStrPrintf(pszSubTestLong, sizeof(pszSubTestLong), "RTFileCreateTemp %s",
f455158a85f95b4886cbe7154b067144b815c125vboxsync pszSubTest);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstFileCreateTemp(pszSubTestLong, pszTemplate, fMode, cTimes,
f455158a85f95b4886cbe7154b067144b815c125vboxsync fSkipXCheck);
f455158a85f95b4886cbe7154b067144b815c125vboxsync RTStrPrintf(pszSubTestLong, sizeof(pszSubTestLong), "RTDirCreateTemp %s",
f455158a85f95b4886cbe7154b067144b815c125vboxsync pszSubTest);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstDirCreateTemp(pszSubTestLong, pszTemplate, fMode, cTimes, fSkipXCheck);
f455158a85f95b4886cbe7154b067144b815c125vboxsync}
f455158a85f95b4886cbe7154b067144b815c125vboxsync
f455158a85f95b4886cbe7154b067144b815c125vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsyncint main()
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync{
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTEST hTest;
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync int rc = RTTestInitAndCreate("tstRTTemp", &hTest);
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync if (rc)
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync return rc;
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTestBanner(hTest);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /*
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * Get the temp directory (this is essential to the testcase).
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync RTTESTI_CHECK_RC(rc = RTPathTemp(g_szTempPath, sizeof(g_szTempPath)), VINF_SUCCESS);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync if (RT_FAILURE(rc))
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync return RTTestSummaryAndDestroy(hTest);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /*
f455158a85f95b4886cbe7154b067144b815c125vboxsync * Create N temporary files and directories using RT(File|Dir)CreateTemp.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync */
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#1 (standard)", "rtRTTemp-XXXXXX", 0700, 128, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#2 (long)", "rtRTTemp-XXXXXXXXXXXXXXXXX", 0700, 128, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#3 (short)", "rtRTTemp-XX", 0777, 128, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#4 (very short)", "rtRTTemp-X", 0100, 26+10, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#5 (in-name)", "rtRTTemp-XXXt", 0301, 2, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#6 (in-name)", "XXX-rtRTTemp", 0355, 2, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#7 (in-name)", "rtRTTemp-XXXXXXXXX.tmp", 0755, 128, false /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#8 (in-name)", "rtRTTemp-XXXXXXX-X.tmp", 0700, 128, true /*fSkipXCheck*/);
f455158a85f95b4886cbe7154b067144b815c125vboxsync tstBothCreateTemp("#9 (in-name)", "rtRTTemp-XXXXXX-XX.tmp", 0700, 128, true /*fSkipXCheck*/);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync /*
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync * Summary.
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync */
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync return RTTestSummaryAndDestroy(hTest);
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync}
73647ac113a5872720ca6c0d73ac29b37a07b814vboxsync