2.5: bugfix for BLI_is_writable, made render with Save Buffers
[blender.git] / source / blender / blenlib / intern / fileops.c
1 /*
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 #include <string.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33
34 #include "zlib.h"
35
36 #ifdef WIN32
37 #include "BLI_winstuff.h"
38 #include <io.h>
39 #else
40 #include <unistd.h> // for read close
41 #include <sys/param.h>
42 #endif
43
44 #include "BLI_blenlib.h"
45 #include "BLI_storage.h"
46 #include "BLI_fileops.h"
47 #include "BLI_callbacks.h"
48
49 #include <sys/types.h>
50 #include <sys/stat.h>
51 #include <fcntl.h>
52
53 #include "BKE_utildefines.h"
54 #include <errno.h>
55
56 #include "BLO_sys_types.h" // for intptr_t support
57
58 /* implementations: */
59 char *first_slash(char *string) {
60         char *ffslash, *fbslash;
61         
62         ffslash= strchr(string, '/');   
63         fbslash= strchr(string, '\\');
64         
65         if (!ffslash) return fbslash;
66         else if (!fbslash) return ffslash;
67         
68         if ((intptr_t)ffslash < (intptr_t)fbslash) return ffslash;
69         else return fbslash;
70 }
71
72 char *BLI_last_slash(const char *string) {
73         char *lfslash, *lbslash;
74         
75         lfslash= strrchr(string, '/');  
76         lbslash= strrchr(string, '\\');
77
78         if (!lfslash) return lbslash; 
79         else if (!lbslash) return lfslash;
80         
81         if ((intptr_t)lfslash < (intptr_t)lbslash) return lbslash;
82         else return lfslash;
83 }
84
85 /* adds a slash if there isnt one there alredy */
86 void BLI_add_slash(char *string) {
87         int len = strlen(string);
88 #ifdef WIN32
89         if (len==0 || string[len-1]!='\\') {
90                 string[len] = '\\';
91                 string[len+1] = '\0';
92         }
93 #else
94         if (len==0 || string[len-1]!='/') {
95                 string[len] = '/';
96                 string[len+1] = '\0';
97         }
98 #endif
99 }
100
101 /* removes a slash if there is one */
102 void BLI_del_slash(char *string) {
103         int len = strlen(string);
104         while (len) {
105 #ifdef WIN32
106                 if (string[len-1]=='\\') {
107 #else
108                 if (string[len-1]=='/') {
109 #endif
110                         string[len-1] = '\0';
111                         len--;
112                 } else {
113                         break;
114                 }
115         }
116 }
117
118 /* gzip the file in from and write it to "to". 
119  return -1 if zlib fails, -2 if the originating file does not exist
120  note: will remove the "from" file
121   */
122 int BLI_gzip(char *from, char *to) {
123         char buffer[10240];
124         int file;
125         int readsize = 0;
126         
127         gzFile gzfile = gzopen(to,"wb"); 
128         if (NULL == gzfile) return -1;
129         
130         file = open(from,O_BINARY|O_RDONLY);
131         
132         if ( -1 == file )       return -2;
133
134         while ( 1 )
135         {
136                 readsize = read(file, buffer, 10240);
137                 
138                 if (readsize <= 0) break;
139                 
140                 gzwrite(gzfile,buffer,readsize);
141         }
142         
143         gzclose(gzfile);
144         close(file);
145         
146         remove(from);
147
148         return 0;
149 }
150
151 /* return 1 when file can be written */
152 int BLI_is_writable(char *filename)
153 {
154         int file;
155         
156         /* first try to open without creating */
157         file = open(filename, O_BINARY | O_RDWR, 0666);
158         
159         if (file < 0) {
160                 /* now try to open and create. a test without actually
161                  * creating a file would be nice, but how? */
162                 file = open(filename, O_BINARY | O_RDWR | O_CREAT, 0666);
163                 
164                 if(file < 0) {
165                         return 0;
166                 }
167                 else {
168                         /* success, delete the file we create */
169                         close(file);
170                         BLI_delete(filename, 0, 0);
171                         return 1;
172                 }
173         }
174         else {
175                 close(file);
176                 return 1;
177         }
178 }
179
180 int BLI_touch(const char *file)
181 {
182    FILE *f = fopen(file,"r+b");
183    if (f != NULL) {
184                 char c = getc(f);
185                 rewind(f);
186                 putc(c,f);
187         } else {
188            f = fopen(file,"wb");
189         }
190         if (f) {
191                 fclose(f);
192                 return 1;
193         }
194         return 0;
195 }
196
197 #ifdef WIN32
198
199 static char str[MAXPATHLEN+12];
200
201 int BLI_delete(char *file, int dir, int recursive) {
202         int err;
203
204         if (recursive) {
205                 callLocalErrorCallBack("Recursive delete is unsupported on Windows");
206                 err= 1;
207         } else if (dir) {
208                 err= !RemoveDirectory(file);
209                 if (err) printf ("Unable to remove directory");
210         } else {
211                 err= !DeleteFile(file);
212                 if (err) callLocalErrorCallBack("Unable to delete file");
213         }
214
215         return err;
216 }
217
218 int BLI_move(char *file, char *to) {
219         int err;
220
221         // windows doesn't support moveing to a directory
222         // it has to be 'mv filename filename' and not
223         // 'mv filename destdir'
224
225         strcpy(str, to);
226         // points 'to' to a directory ?
227         if (BLI_last_slash(str) == (str + strlen(str) - 1)) {
228                 if (BLI_last_slash(file) != NULL) {
229                         strcat(str, BLI_last_slash(file) + 1);
230                 }
231         }
232
233         err= !MoveFile(file, str);
234         if (err) {
235                 callLocalErrorCallBack("Unable to move file");
236                 printf(" Move from '%s' to '%s' failed\n", file, str);
237         }
238
239         return err;
240 }
241
242
243 int BLI_copy_fileops(char *file, char *to) {
244         int err;
245
246         // windows doesn't support copying to a directory
247         // it has to be 'cp filename filename' and not
248         // 'cp filename destdir'
249
250         strcpy(str, to);
251         // points 'to' to a directory ?
252         if (BLI_last_slash(str) == (str + strlen(str) - 1)) {
253                 if (BLI_last_slash(file) != NULL) {
254                         strcat(str, BLI_last_slash(file) + 1);
255                 }
256         }
257
258         err= !CopyFile(file,str,FALSE);
259         
260         if (err) {
261                 callLocalErrorCallBack("Unable to copy file!");
262                 printf(" Copy from '%s' to '%s' failed\n", file, str);
263         }
264
265         return err;
266 }
267
268 int BLI_link(char *file, char *to) {
269         callLocalErrorCallBack("Linking files is unsupported on Windows");
270         
271         return 1;
272 }
273
274 int BLI_exists(char *file) {
275         return (GetFileAttributes(file) != 0xFFFFFFFF);
276 }
277
278 void BLI_recurdir_fileops(char *dirname) {
279         char *lslash;
280         char tmp[MAXPATHLEN];
281         
282         // First remove possible slash at the end of the dirname.
283         // This routine otherwise tries to create
284         // blah1/blah2/ (with slash) after creating
285         // blah1/blah2 (without slash)
286
287         strcpy(tmp, dirname);
288         lslash= BLI_last_slash(tmp);
289
290         if (lslash == tmp + strlen(tmp) - 1) {
291                 *lslash = 0;
292         }
293         
294         if (BLI_exists(tmp)) return;
295                 
296         lslash= BLI_last_slash(tmp);
297         if (lslash) {
298                         /* Split about the last slash and recurse */    
299                 *lslash = 0;
300                 BLI_recurdir_fileops(tmp);
301         }
302         
303         if(dirname[0]) /* patch, this recursive loop tries to create a nameless directory */
304                 if (!CreateDirectory(dirname, NULL))
305                         callLocalErrorCallBack("Unable to create directory\n");
306 }
307
308 int BLI_rename(char *from, char *to) {
309         if (!BLI_exists(from)) return 0;
310
311         /* make sure the filenames are different (case insensitive) before removing */
312         if (BLI_exists(to) && BLI_strcasecmp(from, to))
313                 if(BLI_delete(to, 0, 0)) return 1;
314                 
315         return rename(from, to);
316 }
317
318 #else /* The sane UNIX world */
319
320 /*
321  * but the sane UNIX world is tied to the interface, and the system
322  * timer, and... We implement a callback mechanism. The system will
323  * have to initialise the callback before the functions will work!
324  * */
325 static char str[MAXPATHLEN+12];
326
327 int BLI_delete(char *file, int dir, int recursive) 
328 {
329         if(strchr(file, '"')) {
330                 printf("Error: not deleted file %s because of quote!\n", file);
331         }
332         else {
333                 if (recursive) {
334                         sprintf(str, "/bin/rm -rf \"%s\"", file);
335                         return system(str);
336                 }
337                 else if (dir) {
338                         sprintf(str, "/bin/rmdir \"%s\"", file);
339                         return system(str);
340                 }
341                 else {
342                         return remove(file); //sprintf(str, "/bin/rm -f \"%s\"", file);
343                 }
344         }
345         return -1;
346 }
347
348 int BLI_move(char *file, char *to) {
349         sprintf(str, "/bin/mv -f \"%s\" \"%s\"", file, to);
350
351         return system(str);
352 }
353
354 int BLI_copy_fileops(char *file, char *to) {
355         sprintf(str, "/bin/cp -rf \"%s\" \"%s\"", file, to);
356
357         return system(str);
358 }
359
360 int BLI_link(char *file, char *to) {
361         sprintf(str, "/bin/ln -f \"%s\" \"%s\"", file, to);
362         
363         return system(str);
364 }
365
366 int BLI_exists(char *file) {
367         return BLI_exist(file);
368 }
369
370 void BLI_recurdir_fileops(char *dirname) {
371         char *lslash;
372         char tmp[MAXPATHLEN];
373                 
374         if (BLI_exists(dirname)) return;
375
376         strcpy(tmp, dirname);
377                 
378         lslash= BLI_last_slash(tmp);
379         if (lslash) {
380                         /* Split about the last slash and recurse */    
381                 *lslash = 0;
382                 BLI_recurdir_fileops(tmp);
383         }
384
385         mkdir(dirname, 0777);
386 }
387
388 int BLI_rename(char *from, char *to) {
389         if (!BLI_exists(from)) return 0;
390         
391         if (BLI_exists(to))     if(BLI_delete(to, 0, 0)) return 1;
392                 
393         return rename(from, to);
394 }
395
396 #endif