Files
harbour-core/harbour/src/rtl/fstemp.c
Viktor Szakats 1b080a03f9 2009-10-09 16:53 UTC+0200 Viktor Szakats (harbour.01 syenar.hu)
- source
  + src
  * src/Makefile
  * src/dynlib/mt/Makefile
  * src/dynlib/Makefile
  * INSTALL
  * Makefile
  * ChangeLog
  * harbour.spec
  * mpkg_tgz.sh
    * Renamed 'source' dir to 'src' to move closer to other FOSS 
      projects. It's also easier to type, plus it has the side 
      effect that some cmdlines will be shorter in the 
      build process.
2009-10-09 14:54:14 +00:00

383 lines
11 KiB
C

/*
* $Id$
*/
/*
* Harbour Project source code:
* HB_FTEMPCREATE() function
*
* Copyright 2000-2001 Jose Lalin <dezac@corevia.com>
* Viktor Szakats (harbour.01 syenar.hu)
* www - http://www.harbour-project.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this software; see the file COPYING. If not, write to
* the Free Software Foundation, Inc., 59 Temple Place, Suite 330,
* Boston, MA 02111-1307 USA (or visit the web site http://www.gnu.org/).
*
* As a special exception, the Harbour Project gives permission for
* additional uses of the text contained in its release of Harbour.
*
* The exception is that, if you link the Harbour libraries with other
* files to produce an executable, this does not by itself cause the
* resulting executable to be covered by the GNU General Public License.
* Your use of that executable is in no way restricted on account of
* linking the Harbour library code into it.
*
* This exception does not however invalidate any other reasons why
* the executable file might be covered by the GNU General Public License.
*
* This exception applies only to the code released by the Harbour
* Project under the name Harbour. If you copy code from other
* Harbour Project or Free Software Foundation releases into a copy of
* Harbour, as the General Public License permits, the exception does
* not apply to the code that you add in this way. To avoid misleading
* anyone as to the status of such modified files, you must delete
* this exception notice from them.
*
* If you write modifications of your own for Harbour, it is your choice
* whether to permit this exception to apply to your modifications.
* If you do not wish that, delete this exception notice.
*
*/
/* *nixes */
#if !defined( _LARGEFILE64_SOURCE )
# define _LARGEFILE64_SOURCE
#endif
#if !defined( _GNU_SOURCE )
# define _GNU_SOURCE
#endif
/* Windows */
#define HB_OS_WIN_USED
#include "hbapi.h"
#include "hbapifs.h"
#include "hbvm.h"
#include "hbmath.h"
#include "hbset.h"
#if defined( HB_OS_UNIX )
#include <stdlib.h>
#include <unistd.h> /* We need for mkstemp() on BSD */
#endif
#if ( defined( HB_OS_LINUX ) && !defined( __WATCOMC__ ) ) || \
defined( HB_OS_BSD ) || defined( HB_OS_DARWIN ) || defined( HB_OS_SUNOS )
#define HB_HAS_MKSTEMP
#if ( defined( HB_OS_BSD ) && !defined( __NetBSD__ ) ) || \
defined( HB_OS_DARWIN )
#define HB_HAS_MKSTEMPS
#endif
#endif
#if !defined( HB_USE_LARGEFILE64 ) && defined( HB_OS_UNIX )
#if defined( __USE_LARGEFILE64 )
/*
* The macro: __USE_LARGEFILE64 is set when _LARGEFILE64_SOURCE is
* defined and efectively enables lseek64/flock64/ftruncate64 functions
* on 32bit machines.
*/
#define HB_USE_LARGEFILE64
#elif defined( HB_OS_UNIX ) && defined( O_LARGEFILE ) && ! defined( __WATCOMC__ )
#define HB_USE_LARGEFILE64
#endif
#endif
#if !defined( HB_OS_WIN )
static BOOL fsGetTempDirByCase( char * pszName, const char * pszTempDir )
{
BOOL fOK = FALSE;
if( pszTempDir && *pszTempDir != '\0' )
{
hb_strncpy( pszName, pszTempDir, HB_PATH_MAX - 1 );
switch( hb_setGetDirCase() )
{
case HB_SET_CASE_LOWER:
hb_strLower( pszName, strlen( pszName ) );
fOK = strcmp( pszName, pszTempDir ) == 0;
break;
case HB_SET_CASE_UPPER:
hb_strUpper( pszName, strlen( pszName ) );
fOK = strcmp( pszName, pszTempDir ) == 0;
break;
default:
fOK = TRUE;
break;
}
}
# if defined( __DJGPP__ )
if( fOK )
{
/* convert '/' to '\' */
char * pszDelim;
while( ( pszDelim = strchr( pszName, '/' ) ) != NULL )
*pszDelim = '\\';
}
# endif
return fOK;
}
#endif
static HB_FHANDLE hb_fsCreateTempLow( const char * pszDir, const char * pszPrefix, ULONG ulAttr, char * pszName, const char * pszExt )
{
/* less attemps */
int iAttemptLeft = 99, iLen;
HB_FHANDLE fd;
do
{
pszName[ 0 ] = '\0';
if( pszDir && pszDir[ 0 ] != '\0' )
{
hb_strncpy( pszName, pszDir, HB_PATH_MAX - 1 );
}
else
{
#if defined( HB_OS_WIN )
if( ! GetTempPathA( ( DWORD ) ( HB_PATH_MAX - 1 ), pszName ) )
{
pszName[ 0 ] = '.';
pszName[ 1 ] = '\0';
}
#else
char * pszTmpDir = hb_getenv( "TMPDIR" );
if( !fsGetTempDirByCase( pszName, pszTmpDir ) )
{
#ifdef P_tmpdir
if( !fsGetTempDirByCase( pszName, P_tmpdir ) )
#endif
{
pszName[ 0 ] = '.';
pszName[ 1 ] = '\0';
}
}
if( pszTmpDir )
hb_xfree( pszTmpDir );
#endif
}
if( pszName[0] != '\0' )
{
int len = strlen( pszName );
if( pszName[ len - 1 ] != HB_OS_PATH_DELIM_CHR )
{
pszName[ len ] = HB_OS_PATH_DELIM_CHR;
pszName[ len + 1 ] = '\0';
}
}
if( pszPrefix )
hb_strncat( pszName, pszPrefix, HB_PATH_MAX - 1 );
iLen = ( int ) strlen( pszName );
if( iLen > ( HB_PATH_MAX - 1 ) - 6 )
return FS_ERROR;
#if defined( HB_HAS_MKSTEMP )
if( hb_setGetFileCase() != HB_SET_CASE_LOWER &&
hb_setGetFileCase() != HB_SET_CASE_UPPER &&
hb_setGetDirCase() != HB_SET_CASE_LOWER &&
hb_setGetDirCase() != HB_SET_CASE_UPPER
#if !defined( HB_HAS_MKSTEMPS )
&& ( pszExt == NULL || *pszExt == 0 )
#endif
)
{
hb_vmUnlock();
hb_strncat( pszName, "XXXXXX", HB_PATH_MAX - 1 );
#if defined( HB_HAS_MKSTEMPS )
if( pszExt && *pszExt )
{
hb_strncat( pszName, pszExt, HB_PATH_MAX - 1 );
#if defined( HB_USE_LARGEFILE64 )
fd = ( HB_FHANDLE ) mkstemps64( pszName, ( int ) strlen( pszExt ) );
#else
fd = ( HB_FHANDLE ) mkstemps( pszName, ( int ) strlen( pszExt ) );
#endif
}
else
#endif
#if defined( HB_USE_LARGEFILE64 )
fd = ( HB_FHANDLE ) mkstemp64( pszName );
#else
fd = ( HB_FHANDLE ) mkstemp( pszName );
#endif
hb_fsSetIOError( fd != ( HB_FHANDLE ) -1, 0 );
hb_vmLock();
}
else
#endif /* HB_HAS_MKSTEMP */
{
int i, n;
double d = hb_random_num(), x;
for( i = 0; i < 6; i++ )
{
d = d * 36;
n = ( int ) d;
d = modf( d, &x );
pszName[ iLen++ ] = ( char ) ( n + ( n > 9 ? 'a' - 10 : '0' ) );
}
pszName[ iLen ] = '\0';
if( pszExt )
hb_strncat( pszName, pszExt, HB_PATH_MAX - 1 );
hb_fsNameConv( pszName, NULL );
fd = hb_fsCreateEx( pszName, ulAttr, FO_EXCLUSIVE | FO_EXCL );
}
if( fd != ( HB_FHANDLE ) FS_ERROR )
return fd;
}
while( --iAttemptLeft );
return FS_ERROR;
}
/* NOTE: The buffer must be at least HB_PATH_MAX chars long */
#if !defined( HB_OS_UNIX )
static BOOL hb_fsTempName( char * pszBuffer, const char * pszDir, const char * pszPrefix )
{
BOOL fResult;
hb_vmUnlock();
#if defined( HB_IO_WIN )
{
char szTempDir[ HB_PATH_MAX ];
if( pszDir && pszDir[ 0 ] != '\0' )
hb_strncpy( szTempDir, pszDir, sizeof( szTempDir ) - 1 );
else
{
if( ! GetTempPathA( ( DWORD ) HB_PATH_MAX, szTempDir ) )
{
hb_fsSetIOError( FALSE, 0 );
return FALSE;
}
}
szTempDir[ HB_PATH_MAX - 1 ] = '\0';
fResult = GetTempFileNameA( szTempDir, pszPrefix ? pszPrefix : "hb", 0, pszBuffer );
}
#else
/* TODO: Implement these: */
HB_SYMBOL_UNUSED( pszDir );
HB_SYMBOL_UNUSED( pszPrefix );
/* TOFIX: The spec says to reserve L_tmpnam number of characters for the
passed buffer. It will be needed to fix HB_PATH_MAX - 1 to be
at least this large. */
pszBuffer[ 0 ] = '\0';
fResult = ( tmpnam( pszBuffer ) != NULL );
# if defined( __DJGPP__ )
{
/* convert '/' to '\' */
char * pszDelim;
while( ( pszDelim = strchr( pszBuffer, '/' ) ) != NULL )
*pszDelim = '\\';
}
# endif
#endif
hb_fsSetIOError( fResult, 0 );
hb_vmLock();
return fResult;
}
/* NOTE: The pszName buffer must be at least HB_PATH_MAX chars long */
HB_FHANDLE hb_fsCreateTemp( const char * pszDir, const char * pszPrefix, ULONG ulAttr, char * pszName )
{
int iAttemptLeft = 999;
while( --iAttemptLeft )
{
if( hb_fsTempName( pszName, pszDir, pszPrefix ) )
{
#if defined( HB_OS_WIN )
/* Using FO_TRUNC on win platforms as hb_fsTempName() uses GetTempFileName(),
which creates the file, so FO_EXCL would fail at this point. [vszakats] */
HB_FHANDLE fhnd = hb_fsCreateEx( pszName, ulAttr, FO_EXCLUSIVE | FO_TRUNC );
#else
HB_FHANDLE fhnd = hb_fsCreateEx( pszName, ulAttr, FO_EXCLUSIVE | FO_EXCL );
#endif
/* This function may fail, if the generated filename got
used between generation and the file creation. */
if( fhnd != FS_ERROR )
return fhnd;
}
else
{
/* Don't attempt to retry if the filename generator is
failing for some reason. */
break;
}
}
return FS_ERROR;
}
#else /* HB_OS_UNIX */
HB_FHANDLE hb_fsCreateTemp( const char * pszDir, const char * pszPrefix, ULONG ulAttr, char * pszName )
{
return hb_fsCreateTempLow( pszDir, pszPrefix, ulAttr, pszName, NULL );
}
#endif
HB_FUNC( HB_FTEMPCREATE )
{
char szName[ HB_PATH_MAX ];
hb_retnint( ( HB_NHANDLE ) hb_fsCreateTemp( hb_parc( 1 ),
hb_parc( 2 ),
( ULONG ) ( HB_ISNUM( 3 ) ? ( ULONG ) hb_parnl( 3 ) : FC_NORMAL ),
szName ) );
hb_storc( szName, 4 );
}
HB_FHANDLE hb_fsCreateTempEx( char * pszName, const char * pszDir, const char * pszPrefix, const char * pszExt, ULONG ulAttr )
{
return hb_fsCreateTempLow( pszDir, pszPrefix, ulAttr, pszName, pszExt );
}
HB_FUNC( HB_FTEMPCREATEEX )
{
char szName[ HB_PATH_MAX ];
hb_retnint( ( HB_NHANDLE ) hb_fsCreateTempEx( szName,
hb_parc( 2 ),
hb_parc( 3 ),
hb_parc( 4 ),
( ULONG ) ( HB_ISNUM( 5 ) ? ( ULONG ) hb_parnl( 5 ) : FC_NORMAL ) ) );
hb_storc( szName, 1 );
}