mktemp.c
上传用户:dayuan858
上传日期:2007-01-04
资源大小:194k
文件大小:3k
源码类别:

网络编程

开发平台:

Unix_Linux

  1. /*
  2.  * Copyright (c) 1987 Regents of the University of California.
  3.  * All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  * This product includes software developed by the University of
  16.  * California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33. #ifdef HAVE_CONFIG_H
  34. # include <config.h>
  35. #endif
  36. #include <sys/types.h>
  37. #include <sys/stat.h>
  38. #include <fcntl.h>
  39. #include <errno.h>
  40. #include <stdio.h>
  41. #include <ctype.h>
  42. #include <popper.h>
  43. #ifndef S_ISDIR
  44. #define S_ISDIR(mode)   ((mode&0xF000) == 0x4000)
  45. #endif
  46. static int _gettemp();
  47. mkstemp(path)
  48. char *path;
  49. {
  50. int fd;
  51. return (_gettemp(path, &fd) ? fd : -1);
  52. }
  53. char *
  54. mktemp(path)
  55. char *path;
  56. {
  57. return(_gettemp(path, (int *)NULL) ? path : (char *)NULL);
  58. }
  59. static
  60. _gettemp(path, doopen)
  61. char *path;
  62. register int *doopen;
  63. {
  64. extern int errno;
  65. register char *start, *trv;
  66. struct stat sbuf;
  67. PID_T pid;
  68. pid = getpid();
  69. for (trv = path; *trv; ++trv); /* extra X's get set to 0's */
  70. while (*--trv == 'X') {
  71. *trv = (pid % 10) + '0';
  72. pid /= 10;
  73. }
  74. /*
  75.  * check the target directory; if you have six X's and it
  76.  * doesn't exist this runs for a *very* long time.
  77.  */
  78. for (start = trv + 1;; --trv) {
  79. if (trv <= path)
  80. break;
  81. if (*trv == '/') {
  82. *trv = '';
  83. if (stat(path, &sbuf))
  84. return(0);
  85. if (!S_ISDIR(sbuf.st_mode)) {
  86. errno = ENOTDIR;
  87. return(0);
  88. }
  89. *trv = '/';
  90. break;
  91. }
  92. }
  93. for (;;) {
  94. if (doopen) {
  95. if ((*doopen =
  96.     open(path, O_CREAT|O_EXCL|O_RDWR, 0600)) >= 0)
  97. return(1);
  98. if (errno != EEXIST)
  99. return(0);
  100. }
  101. else if (stat(path, &sbuf))
  102. return(errno == ENOENT ? 1 : 0);
  103. /* tricky little algorithm for backward compatibility */
  104. for (trv = start;;) {
  105. if (!*trv)
  106. return(0);
  107. if (*trv == 'z')
  108. *trv++ = 'a';
  109. else {
  110. if (isdigit(*trv))
  111. *trv = 'a';
  112. else
  113. ++*trv;
  114. break;
  115. }
  116. }
  117. }
  118. /*NOTREACHED*/
  119. }