]> pd.if.org Git - pdclib/blob - stdio/tmpfile.c
Cosmetic comment fixes.
[pdclib] / stdio / tmpfile.c
1 /* $Id$ */
2
3 /* tmpfile( void )
4
5    This file is part of the Public Domain C Library (PDCLib).
6    Permission is granted to use, modify, and / or redistribute at will.
7 */
8
9 #include <stdio.h>
10
11 #ifndef REGTEST
12
13 #include <inttypes.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <_PDCLIB_glue.h>
17 #include <unistd.h>
18 #ifdef __linux__
19 /* get O_CLOEXEC without sys/types.h being awful */
20 #include <asm/fcntl.h>
21 int open(const char *fname, int flags, ...);
22 #else
23 #include <fcntl.h>
24 #endif
25
26 extern const _PDCLIB_fileops_t _PDCLIB_fileops;
27
28 FILE* _PDCLIB_nothrow tmpfile( void )
29 {
30
31     /* Good quality random source */
32     int urandom = open( "/dev/urandom", O_RDONLY | O_CLOEXEC );
33     if(urandom == -1)
34     {
35         // TODO: errno!
36         return NULL;
37     }
38
39     int fd;
40     char filename[ L_tmpnam ];
41     for ( ;; )
42     {
43         long long randnum;
44         if( read(urandom, &randnum, sizeof randnum ) != sizeof randnum )
45         {
46             // TODO: errno!
47             close( urandom );
48             return NULL;
49         }
50
51         sprintf( filename, "/tmp/%llx.tmp", randnum );
52         /* Check if file of this name exists. Note that fopen() is a very weak
53            check, which does not take e.g. access permissions into account
54            (file might exist but not readable). Replace with something more
55            appropriate.
56         */
57         fd = open( filename, O_CREAT | O_EXCL | O_RDWR, 0600 );
58         if ( fd != -1 )
59         {
60             break;
61         }
62     }
63     close( urandom );
64
65     FILE* rc = _PDCLIB_fvopen(((_PDCLIB_fd_t){ .sval = fd}), &_PDCLIB_fileops,
66                                 _PDCLIB_FWRITE | _PDCLIB_FRW |
67                                 _PDCLIB_DELONCLOSE, filename);
68     if( rc == NULL )
69     {
70         close( fd );
71         return NULL;
72     }
73
74     return rc;
75 }
76
77 #endif
78
79 #ifdef TEST
80 #include <_PDCLIB_test.h>
81 #include <string.h>
82
83 int main( void )
84 {
85     FILE * fh;
86 #ifndef REGTEST
87     char filename[ L_tmpnam ];
88     FILE * fhtest;
89 #endif
90     TESTCASE( ( fh = tmpfile() ) != NULL );
91     TESTCASE( fputc( 'x', fh ) == 'x' );
92     /* Checking that file is actually there */
93     TESTCASE_NOREG( strcpy( filename, fh->filename ) == filename );
94     TESTCASE_NOREG( ( fhtest = fopen( filename, "r" ) ) != NULL );
95     TESTCASE_NOREG( fclose( fhtest ) == 0 );
96     /* Closing tmpfile */
97     TESTCASE( fclose( fh ) == 0 );
98     /* Checking that file was deleted */
99     TESTCASE_NOREG( fopen( filename, "r" ) == NULL );
100     return TEST_RESULTS;
101 }
102
103 #endif
104