mf_tempfile.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This library is free software; you can redistribute it and/or
  4.    modify it under the terms of the GNU Library General Public
  5.    License as published by the Free Software Foundation; either
  6.    version 2 of the License, or (at your option) any later version.
  7.    
  8.    This library is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11.    Library General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU Library General Public
  14.    License along with this library; if not, write to the Free
  15.    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  16.    MA 02111-1307, USA */
  17. #include "mysys_priv.h"
  18. #include <m_string.h>
  19. #include "my_static.h"
  20. #include "mysys_err.h"
  21. #include <errno.h>
  22. #ifdef HAVE_PATHS_H
  23. #include <paths.h>
  24. #endif
  25. #ifdef HAVE_TEMPNAM
  26. #ifndef MSDOS
  27. extern char **environ;
  28. #endif
  29. #endif
  30. /*
  31.   Create a temporary file in a given directory
  32.   This function should be used instead of my_tempnam() !
  33. */
  34. File create_temp_file(char *to, const char *dir, const char *prefix,
  35.       int mode __attribute__((unused)),
  36.       myf MyFlags __attribute__((unused)))
  37. {
  38.   File file= -1;
  39.   DBUG_ENTER("create_temp_file");
  40. #if defined(_MSC_VER)
  41.   {
  42.     char temp[FN_REFLEN],*end,*res,**old_env,*temp_env[1];
  43.     old_env=environ;
  44.     if (dir)
  45.     {
  46.       end=strend(dir)-1;
  47.       if (!dir[0])
  48.       { /* Change empty string to current dir */
  49. to[0]= FN_CURLIB;
  50. to[1]= 0;
  51. dir=to;
  52.       }
  53.       else if (*end == FN_DEVCHAR)
  54.       { /* Get current dir for drive */
  55. _fullpath(temp,dir,FN_REFLEN);
  56. dir=to;
  57.       }
  58.       else if (*end == FN_LIBCHAR && dir < end && end[-1] != FN_DEVCHAR)
  59.       {
  60. strmake(to,dir,(uint) (end-dir)); /* Copy and remove last '' */
  61. dir=to;
  62.       }
  63.       environ=temp_env; /* Force use of dir (dir not checked) */
  64.       temp_env[0]=0;
  65.     }
  66.     if ((res=tempnam((char*) dir,(char *) prefix)))
  67.     {
  68.       strmake(to,res,FN_REFLEN-1);
  69.       (*free)(res);
  70.       file=my_create(to,0, mode, MyFlags);
  71.     }
  72.     environ=old_env;
  73.   }
  74. #elif defined(_ZTC__)
  75.   if (!dir)
  76.     dir=getenv("TMPDIR");
  77.   if ((res=tempnam((char*) dir,(char *) prefix)))
  78.   {
  79.     strmake(to,res,FN_REFLEN-1);
  80.     (*free)(res);
  81.     file=my_create(to, 0, mode, MyFlags);
  82.   }
  83. #elif defined(HAVE_MKSTEMP)
  84.   {
  85.     char prefix_buff[30];
  86.     uint pfx_len;
  87.     File org_file;
  88.     pfx_len=(strmov(strnmov(prefix_buff,
  89.     prefix ? prefix : "tmp.",
  90.     sizeof(prefix_buff)-7),"XXXXXX") - prefix_buff);
  91.     if (!dir && ! (dir =getenv("TMPDIR")))
  92.       dir=P_tmpdir;
  93.     if (strlen(dir)+ pfx_len > FN_REFLEN-2)
  94.     {
  95.       errno=my_errno= ENAMETOOLONG;
  96.       return 1;
  97.     }
  98.     strmov(to,dir);
  99.     strmov(convert_dirname(to),prefix_buff);
  100.     org_file=mkstemp(to);
  101.     file=my_register_filename(org_file, to, FILE_BY_MKSTEMP,
  102.       EE_CANTCREATEFILE, MyFlags);
  103.     /* If we didn't manage to register the name, remove the temp file */
  104.     if (org_file >= 0 && file < 0)
  105.     {
  106.       int tmp=my_errno;
  107.       (void) my_delete(to, MYF(MY_WME | ME_NOINPUT));
  108.       my_errno=tmp;
  109.     }
  110.   }
  111. #elif defined(HAVE_TEMPNAM)
  112.   {
  113.     char *res,**old_env,*temp_env[1];
  114.     if (dir && !dir[0])
  115.     { /* Change empty string to current dir */
  116.       to[0]= FN_CURLIB;
  117.       to[1]= 0;
  118.       dir=to;
  119.     }
  120.     old_env=environ;
  121.     if (dir)
  122.     { /* Don't use TMPDIR if dir is given */
  123.       environ=temp_env;
  124.       temp_env[0]=0;
  125.     }
  126.     if ((res=tempnam((char*) dir, (char*) prefix)))
  127.     {    
  128.       strmake(to,res,FN_REFLEN-1);
  129.       (*free)(res);
  130.       file=my_create(to,0,
  131.      (int) (O_RDWR | O_BINARY | O_TRUNC |
  132.     O_TEMPORARY | O_SHORT_LIVED),
  133.      MYF(MY_WME));
  134.     }
  135.     else
  136.     {
  137.       DBUG_PRINT("error",("Got error: %d from tempnam",errno));
  138.     }
  139.     environ=old_env;
  140.   }
  141. #else
  142.   {
  143.     register long uniq;
  144.     register int length;
  145.     my_string pos,end_pos;
  146.     /* Make an unique number */
  147.     pthread_mutex_lock(&THR_LOCK_open);
  148.     uniq= ((long) getpid() << 20) + (long) _my_tempnam_used++ ;
  149.     pthread_mutex_unlock(&THR_LOCK_open);
  150.     if (!dir && !(dir=getenv("TMPDIR"))) /* Use this if possibly */
  151.       dir=P_tmpdir; /* Use system default */
  152.     length=strlen(dir)+strlen(pfx)+1;
  153.     DBUG_PRINT("test",("mallocing %d byte",length+8+sizeof(TMP_EXT)+1));
  154.     if (length+8+sizeof(TMP_EXT)+1 > FN_REFLENGTH)
  155.       errno=my_errno= ENAMETOOLONG;
  156.     else
  157.     {
  158.       end_pos=strmov(to,dir);
  159.       if (end_pos != to && end_pos[-1] != FN_LIBCHAR)
  160. *end_pos++=FN_LIBCHAR;
  161.       end_pos=strmov(end_pos,pfx);
  162.       
  163.       for (length=0 ; length < 8 && uniq ; length++)
  164.       {
  165. *end_pos++= _dig_vec[(int) (uniq & 31)];
  166. uniq >>= 5;
  167.       }
  168.       (void) strmov(end_pos,TMP_EXT);
  169.       file=my_create(to,0,
  170.      (int) (O_RDWR | O_BINARY | O_TRUNC |
  171.     O_TEMPORARY | O_SHORT_LIVED),
  172.      MYF(MY_WME));
  173.     }
  174.   }
  175. #endif
  176.   if (file >= 0)
  177.     thread_safe_increment(my_tmp_file_created,&THR_LOCK_open);
  178.   DBUG_RETURN(file);
  179. }