mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
325 lines
10 KiB
325 lines
10 KiB
/***
|
|
*tempnam.c - generate unique file name
|
|
*
|
|
* Copyright (c) 1986-2001, Microsoft Corporation. All rights reserved.
|
|
*
|
|
*Purpose:
|
|
*
|
|
*Revision History:
|
|
* ??-??-?? TC initial version
|
|
* 04-17-86 JMB changed directory of last resort from \tmp to tmp.
|
|
* eliminated use of L_tmpnam (it was incoorectly defined
|
|
* in stdio.h and should not be used in tempnam; see
|
|
* System V definition of tempnam.
|
|
* 04-23-86 TC changed last try directory from tmp to current directory
|
|
* 04-29-86 JMB bug fix: pfxlength was being set from strlen(pfx)
|
|
* even if pfx was NULL. Fixed to set pfxlength to zero
|
|
* if pfx is NULL, strlen(pfx) otherwise.
|
|
* 05-28-86 TC changed stat's to access's, and optimized code a bit
|
|
* 12-01-86 JMB added support for Kanji file names until KANJI switch
|
|
* 12-15-86 JMB free malloced memory if (++_tmpoff == first)
|
|
* 07-15-87 JCR Re-init _tempoff based on length of pfx (fixes infinate
|
|
* loop bug; also, tempnam() now uses _tempoff instead of
|
|
* _tmpoff (used by tmpnam()).
|
|
* 10-16-87 JCR Fixed bug in _tempoff re-init code if pfx is NULL.
|
|
* 11-09-87 JCR Multi-thread version
|
|
* 12-11-87 JCR Added "_LOAD_DS" to declaration
|
|
* 05-27-88 PHG Merged DLL and normal versions
|
|
* 06-09-89 GJF Propagated MT's change of 05-17-89 (Kanji)
|
|
* 02-16-90 GJF Fixed copyright and indents
|
|
* 03-19-90 GJF Replaced _LOAD_DS and _CALLTYPE1 and added #include
|
|
* <cruntime.h>.
|
|
* 03-26-90 GJF Added #include <io.h>.
|
|
* 08-13-90 SBM Compiles cleanly with -W3, replaced explicit register
|
|
* declarations by REGn references
|
|
* 10-03-90 GJF New-style function declarator.
|
|
* 01-21-91 GJF ANSI naming.
|
|
* 08-19-91 JCR Allow quotes in TMP variable path
|
|
* 08-27-91 JCR ANSI naming
|
|
* 08-25-92 GJF Don't build for POSIX.
|
|
* 11-30-92 KRS Generalize KANJI support to MBCS. Port 16-bit bug fix.
|
|
* 04-06-93 SKS Replace _CRTAPI* with __cdecl
|
|
* 12-07-93 CFW Wide char enable.
|
|
* 01-10-95 CFW Debug CRT allocs.
|
|
* 01-23-95 CFW Debug: tempnam return freed by user.
|
|
* 02-21-95 GJF Appended Mac version of source file (somewhat cleaned
|
|
* up), with appropriate #ifdef-s. Also replaced WPRFLAG
|
|
* with _UNICODE.
|
|
* 03-10-95 CFW Made _tempnam() parameters const.
|
|
* 03-14-95 JCF Made pfin _TSCHAR for the Mac.
|
|
* 07-30-96 GJF Allow longer file names and changed the way we ensure
|
|
* a file name is not too long. In particular, this
|
|
* corrects the absurdly long loop (appeared infinite to
|
|
* to users) which occurred when the user supplied prefix
|
|
* was too long. Also, cleaned up the format a bit.
|
|
* 03-04-98 GJF Exception-safe locking.
|
|
* 01-04-99 GJF Changes for 64-bit size_t.
|
|
* 05-17-99 PML Remove all Macintosh support.
|
|
* 12-10-00 PML Fix double-free of qptr when _stripquote succeeds but
|
|
* result isn't a valid directory (vs7#5416).
|
|
* 02-20-01 PML vs7#172586 Avoid _RT_LOCK by preallocating all locks
|
|
* that will be required, and returning failure back on
|
|
* inability to allocate a lock.
|
|
* 07-07-01 BWT Initialize 's' in tempnam - in case the done2 branch is taken,
|
|
* 's' must be something reasonable.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifndef _POSIX_
|
|
|
|
#include <cruntime.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <io.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <malloc.h>
|
|
#include <string.h>
|
|
#include <internal.h>
|
|
#include <mtdll.h>
|
|
#include <tchar.h>
|
|
#include <dbgint.h>
|
|
|
|
#ifdef _MBCS
|
|
#include <mbstring.h>
|
|
#endif
|
|
|
|
/* local tchar */
|
|
#ifdef _UNICODE
|
|
#define _tP_tmpdir _wP_tmpdir
|
|
#else
|
|
#define _tP_tmpdir _P_tmpdir
|
|
#endif
|
|
|
|
#ifdef _UNICODE
|
|
static wchar_t * _wstripquote (wchar_t *);
|
|
#else
|
|
static char * _stripquote (char *);
|
|
#endif
|
|
|
|
/***
|
|
*_TSCHAR *_tempnam(dir, prefix) - create unique file name
|
|
*
|
|
*Purpose:
|
|
* Create a file name that is unique in the specified directory.
|
|
* The semantics of directory specification is as follows:
|
|
* Use the directory specified by the TMP environment variable
|
|
* if that exists, else use the dir argument if non-NULL, else
|
|
* use _P_tmpdir if that directory exists, else use the current
|
|
* working directory), else return NULL.
|
|
*
|
|
*Entry:
|
|
* _TSCHAR *dir - directory to be used for temp file if TMP env var
|
|
* not set
|
|
* _TSCHAR *prefix - user provided prefix for temp file name
|
|
*
|
|
*Exit:
|
|
* returns ptr to constructed file name if successful
|
|
* returns NULL if unsuccessful
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
_TSCHAR * __cdecl _ttempnam (
|
|
const _TSCHAR *dir,
|
|
const _TSCHAR *pfx
|
|
)
|
|
{
|
|
REG1 _TSCHAR *ptr;
|
|
REG2 unsigned int pfxlength = 0;
|
|
_TSCHAR *s = NULL;
|
|
_TSCHAR *pfin;
|
|
unsigned int first;
|
|
unsigned int bufsz;
|
|
_TSCHAR * qptr = NULL; /* ptr to TMP path with quotes stripped out */
|
|
|
|
#ifdef _MT
|
|
if ( !_mtinitlocknum( _TMPNAM_LOCK ))
|
|
return NULL;
|
|
#endif
|
|
|
|
/* try TMP path */
|
|
if ( ( ptr = _tgetenv( _T("TMP") ) ) && ( _taccess( ptr, 0 ) != -1 ) )
|
|
dir = ptr;
|
|
|
|
/* try stripping quotes out of TMP path */
|
|
#ifdef _UNICODE
|
|
else if ( (ptr != NULL) && (qptr = _wstripquote(ptr)) &&
|
|
#else
|
|
else if ( (ptr != NULL) && (qptr = _stripquote(ptr)) &&
|
|
#endif
|
|
(_taccess(qptr, 0) != -1 ) )
|
|
dir = qptr;
|
|
|
|
/* TMP path not available, use alternatives */
|
|
else if (!( dir != NULL && ( _taccess( dir, 0 ) != -1 ) ) )
|
|
/* do not "simplify" this depends on side effects!! */
|
|
{
|
|
if ( _taccess( _tP_tmpdir, 0 ) != -1 )
|
|
dir = _tP_tmpdir;
|
|
else
|
|
dir = _T(".");
|
|
}
|
|
|
|
|
|
if (pfx)
|
|
pfxlength = (unsigned)_tcslen(pfx);
|
|
|
|
if ( ((bufsz = (unsigned)_tcslen(dir) + pfxlength + 12) > FILENAME_MAX) ||
|
|
((s = malloc(bufsz * sizeof(_TSCHAR))) == NULL) )
|
|
/* the 12 above allows for a backslash, 10 char temp string and
|
|
a null terminator */
|
|
{
|
|
goto done2;
|
|
}
|
|
|
|
*s = _T('\0');
|
|
_tcscat( s, dir );
|
|
pfin = (_TSCHAR *)&(dir[ _tcslen( dir ) - 1 ]);
|
|
|
|
#ifdef _MBCS
|
|
if (*pfin == '\\') {
|
|
if (pfin != _mbsrchr(dir,'\\'))
|
|
/* *pfin is second byte of a double-byte char */
|
|
strcat( s, "\\" );
|
|
}
|
|
else if (*pfin != '/')
|
|
strcat( s, "\\" );
|
|
#else
|
|
if ( ( *pfin != _T('\\') ) && ( *pfin != _T('/') ) )
|
|
{
|
|
_tcscat( s, _T("\\") );
|
|
}
|
|
#endif
|
|
|
|
if ( pfx != NULL )
|
|
{
|
|
_tcscat( s, pfx );
|
|
}
|
|
ptr = &s[_tcslen( s )];
|
|
|
|
/*
|
|
Re-initialize _tempoff if necessary. If we don't re-init _tempoff, we
|
|
can get into an infinate loop (e.g., (a) _tempoff is a big number on
|
|
entry, (b) prefix is a long string (e.g., 8 chars) and all tempfiles
|
|
with that prefix exist, (c) _tempoff will never equal first and we'll
|
|
loop forever).
|
|
|
|
[NOTE: To avoid a conflict that causes the same bug as that discussed
|
|
above, _tempnam() uses _tempoff; tmpnam() uses _tmpoff]
|
|
*/
|
|
|
|
#ifdef _MT
|
|
_mlock(_TMPNAM_LOCK); /* Lock access to _old_pfxlen and _tempoff */
|
|
__try {
|
|
#endif
|
|
|
|
if (_old_pfxlen < pfxlength)
|
|
_tempoff = 1;
|
|
_old_pfxlen = pfxlength;
|
|
|
|
first = _tempoff;
|
|
|
|
do {
|
|
if ( (++_tempoff - first) > TMP_MAX ) {
|
|
free(s);
|
|
s = NULL;
|
|
goto done1;
|
|
}
|
|
/* the maximum length string returned by _ultot is 10 chars
|
|
(assuming 32-bit unsigned long) so there is enough room in
|
|
the tail of s (pointed to by ptr) for it */
|
|
_ultot( (unsigned long)_tempoff, ptr, 10 );
|
|
}
|
|
while ( (_taccess( s, 0 ) == 0 ) || (errno == EACCES) );
|
|
|
|
|
|
/* Common return */
|
|
done1:
|
|
|
|
#ifdef _MT
|
|
; }
|
|
__finally {
|
|
_munlock(_TMPNAM_LOCK); /* release tempnam lock */
|
|
}
|
|
#endif
|
|
|
|
done2:
|
|
_free_crt(qptr); /* free temp ptr, if non-NULL */
|
|
return(s);
|
|
}
|
|
|
|
|
|
|
|
/***
|
|
*_stripquote() - Strip quotes out of a string
|
|
*
|
|
*Purpose:
|
|
* This routine strips quotes out of a string. This is necessary
|
|
* in the case where a file/path name has embedded quotes (i.e.,
|
|
* new file system.)
|
|
*
|
|
* For example,
|
|
* c:\tmp\"a b c"\d --> c:\tmp\a b d\d
|
|
*
|
|
* NOTE: This routine makes a copy of the string since it may be
|
|
* passed a pointer to an environment variable that shouldn't be
|
|
* changed. It is up to the caller to free up the memory (if the
|
|
* return value is non-NULL).
|
|
*
|
|
*Entry:
|
|
* _TSCHAR * ptr = pointer to string
|
|
*
|
|
*Exit:
|
|
* _TSCHAR * ptr = pointer to copy of string with quotes gone.
|
|
* NULL = no quotes in string.
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifdef _UNICODE
|
|
static wchar_t * _wstripquote (
|
|
#else
|
|
static char * _stripquote (
|
|
#endif
|
|
_TSCHAR * src
|
|
)
|
|
{
|
|
_TSCHAR * dst;
|
|
_TSCHAR * ret;
|
|
unsigned int q = 0;
|
|
|
|
|
|
/* get a buffer for the new string */
|
|
|
|
if ((dst = _malloc_crt((_tcslen(src)+1) * sizeof(_TSCHAR))) == NULL)
|
|
return(NULL);
|
|
|
|
/* copy the string stripping out the quotes */
|
|
|
|
ret = dst; /* save base ptr */
|
|
|
|
while (*src) {
|
|
|
|
if (*src == _T('\"')) {
|
|
src++; q++;
|
|
}
|
|
else
|
|
*dst++ = *src++;
|
|
}
|
|
|
|
if (q) {
|
|
*dst = _T('\0'); /* final nul */
|
|
return(ret);
|
|
}
|
|
else {
|
|
_free_crt(ret);
|
|
return(NULL);
|
|
}
|
|
|
|
}
|
|
|
|
#endif /* _POSIX_ */
|