tempnam.c

来自「开放源码的编译器open watcom 1.6.0版的源代码」· C语言 代码 · 共 131 行

/****************************************************************************
*
*                            Open Watcom Project
*
*    Portions Copyright (c) 1983-2002 Sybase, Inc. All Rights Reserved.
*
*  ========================================================================
*
*    This file contains Original Code and/or Modifications of Original
*    Code as defined in and that are subject to the Sybase Open Watcom
*    Public License version 1.0 (the 'License'). You may not use this file
*    except in compliance with the License. BY USING THIS FILE YOU AGREE TO
*    ALL TERMS AND CONDITIONS OF THE LICENSE. A copy of the License is
*    provided with the Original Code and Modifications, and is also
*    available at www.sybase.com/developer/opensource.
*
*    The Original Code and all software distributed under the License are
*    distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
*    EXPRESS OR IMPLIED, AND SYBASE AND ALL CONTRIBUTORS HEREBY DISCLAIM
*    ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF
*    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR
*    NON-INFRINGEMENT. Please see the License for the specific language
*    governing rights and limitations under the License.
*
*  ========================================================================
*
* Description:  WHEN YOU FIGURE OUT WHAT THIS FILE DOES, PLEASE
*               DESCRIBE IT HERE!
*
****************************************************************************/


#include "variety.h"
#include "widechar.h"
#include <io.h>
#include <malloc.h>
#include <mbstring.h>
#include <process.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define UNIQUE                  5


static CHAR_TYPE *try_dir( CHAR_TYPE *dir, CHAR_TYPE *prefix )
{
    CHAR_TYPE *         buf;
    size_t              numchars;
    int                 addslash = 0;
    unsigned            number;
    int                 alive;

    /*** Ensure the directory exists ***/
    if( dir[0] != '\0' ) {
        if( __F_NAME(access,_waccess)( dir, F_OK )  ==  -1 ) {
            return( NULL );
        }
    }

    /*** Initialize addslash ***/
    numchars = __F_NAME(_mbslen,wcslen)( dir );
    if( numchars > 0  &&  prefix[0] != '\\' ) {
        #ifdef __WIDECHAR__
            if( dir[numchars-1] != '\\' ) {
        #else
            if( _mbccmp( _mbsninc(dir,numchars-1), "\\" ) ) {
        #endif
                addslash = 1;
            }
    }

    /*** Ensure the filename wouldn't be too long ***/
    #ifdef __WIDECHAR__
        /* dir + slash if needed + prefix + room for unique characters */
        numchars = wcslen( dir )  +  addslash  +  wcslen( prefix )  +  UNIQUE;
    #else
        /* strlen is ok for DBCS, since we want the byte count */
        numchars = strlen( dir )  +  addslash  +  strlen( prefix )  +  UNIQUE;
    #endif
    if( numchars > _MAX_PATH )  return( NULL );

    /*** Try to build a unique filename ***/
    buf = malloc( CHARSIZE * (numchars+1) );
    for( number=1,alive=1; alive; number++ ) {
        __F_NAME(strcpy,wcscpy)( buf, dir );
        if( addslash ) {
            __F_NAME(strcat,wcscat)( buf, __F_NAME("\\",L"\\") );
        }
        __F_NAME(strcat,wcscat)( buf, prefix );
        __F_NAME(utoa,_utow)( number, buf + __F_NAME(strlen,wcslen)(buf), 10 );

        if( __F_NAME(access,_waccess)( buf, F_OK )  !=  0 ) {
            return( buf );
        }
        if( number == 65535u )  alive = 0;
    }

    return( NULL );
}


_WCRTLINK CHAR_TYPE *__F_NAME(_tempnam,_wtempnam)( CHAR_TYPE *dir, CHAR_TYPE *prefix )
{
    CHAR_TYPE *         p;
    CHAR_TYPE *         envp;

    /*** Try the directory given by the TMP environment variable ***/
    envp = __F_NAME(getenv,_wgetenv)( __F_NAME("TMP",L"TMP") );
    if( envp != NULL ) {
        p = try_dir( envp, prefix );
        if( p != NULL )  return( p );
    }

    /*** Try the directory passed to this function (dir) ***/
    if( dir != NULL ) {
        p = try_dir( dir, prefix );
        if( p != NULL )  return( p );
    }

    /*** Try the _P_tmpdir directory (#defined in stdio.h) ***/
    p = try_dir( __F_NAME(_P_tmpdir,_wP_tmpdir), prefix );
    if( p != NULL )  return( p );

    /*** Try the current directory ***/
    p = try_dir( __F_NAME("",L""), prefix );
    if( p != NULL )  return( p );

    return( NULL );
}