f0d201ea3f7a737813249ed635620c2e4467d22d
[blender.git] / source / blender / blenkernel / intern / blender.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/blender.c
29  *  \ingroup bke
30  */
31
32
33 #ifndef _WIN32 
34 #  include <unistd.h> // for read close
35 #else
36 #  include <io.h> // for open close read
37 #  define open _open
38 #  define read _read
39 #  define close _close
40 #  define write _write
41 #endif
42
43 #include <stdlib.h>
44 #include <stdio.h>
45 #include <stddef.h>
46 #include <string.h>
47 #include <fcntl.h>  /* for open */
48 #include <errno.h>
49
50 #include "MEM_guardedalloc.h"
51
52 #include "DNA_userdef_types.h"
53 #include "DNA_scene_types.h"
54 #include "DNA_screen_types.h"
55 #include "DNA_sequence_types.h"
56 #include "DNA_sound_types.h"
57
58 #include "BLI_blenlib.h"
59 #include "BLI_bpath.h"
60 #include "BLI_dynstr.h"
61 #include "BLI_utildefines.h"
62 #include "BLI_callbacks.h"
63
64 #include "IMB_imbuf.h"
65 #include "IMB_moviecache.h"
66
67 #include "BKE_blender.h"
68 #include "BKE_context.h"
69 #include "BKE_depsgraph.h"
70 #include "BKE_displist.h"
71 #include "BKE_global.h"
72 #include "BKE_idprop.h"
73 #include "BKE_image.h"
74 #include "BKE_ipo.h"
75 #include "BKE_library.h"
76 #include "BKE_main.h"
77 #include "BKE_node.h"
78 #include "BKE_report.h"
79 #include "BKE_scene.h"
80 #include "BKE_screen.h"
81 #include "BKE_sequencer.h"
82 #include "BKE_sound.h"
83 #include "RE_pipeline.h"
84
85 #include "BLO_undofile.h"
86 #include "BLO_readfile.h" 
87 #include "BLO_writefile.h" 
88
89 #include "RNA_access.h"
90
91 #include "WM_api.h" // XXXXX BAD, very BAD dependency (bad level call) - remove asap, elubie
92
93 #include "IMB_colormanagement.h"
94
95 #ifdef WITH_PYTHON
96 #  include "BPY_extern.h"
97 #endif
98
99 Global G;
100 UserDef U;
101 /* ListBase = {NULL, NULL}; */
102
103 char versionstr[48] = "";
104
105 /* ********** free ********** */
106
107 /* only to be called on exit blender */
108 void free_blender(void)
109 {
110         /* samples are in a global list..., also sets G.main->sound->sample NULL */
111         free_main(G.main);
112         G.main = NULL;
113
114         BKE_spacetypes_free();      /* after free main, it uses space callbacks */
115         
116         IMB_exit();
117         BKE_images_exit();
118
119         BLI_callback_global_finalize();
120
121         BKE_sequencer_cache_destruct();
122         IMB_moviecache_destruct();
123         
124         free_nodesystem();
125 }
126
127 void initglobals(void)
128 {
129         memset(&G, 0, sizeof(Global));
130         
131         U.savetime = 1;
132
133         G.main = MEM_callocN(sizeof(Main), "initglobals");
134
135         strcpy(G.ima, "//");
136
137         if (BLENDER_SUBVERSION)
138                 BLI_snprintf(versionstr, sizeof(versionstr), "v%d.%02d.%d", BLENDER_VERSION / 100, BLENDER_VERSION % 100, BLENDER_SUBVERSION);
139         else
140                 BLI_snprintf(versionstr, sizeof(versionstr), "v%d.%02d", BLENDER_VERSION / 100, BLENDER_VERSION % 100);
141
142 #ifdef _WIN32   // FULLSCREEN
143         G.windowstate = G_WINDOWSTATE_USERDEF;
144 #endif
145
146         G.charstart = 0x0000;
147         G.charmin = 0x0000;
148         G.charmax = 0xffff;
149
150 #ifndef WITH_PYTHON_SECURITY /* default */
151         G.f |= G_SCRIPT_AUTOEXEC;
152 #else
153         G.f &= ~G_SCRIPT_AUTOEXEC;
154 #endif
155 }
156
157 /***/
158
159 static void clear_global(void) 
160 {
161 //      extern short winqueue_break;    /* screen.c */
162
163         free_main(G.main);          /* free all lib data */
164         
165 //      free_vertexpaint();
166
167         G.main = NULL;
168 }
169
170 static int clean_paths_visit_cb(void *UNUSED(userdata), char *path_dst, const char *path_src)
171 {
172         strcpy(path_dst, path_src);
173         BLI_clean(path_dst);
174         return (strcmp(path_dst, path_src) == 0) ? FALSE : TRUE;
175 }
176
177 /* make sure path names are correct for OS */
178 static void clean_paths(Main *main)
179 {
180         Scene *scene;
181
182         BLI_bpath_traverse_main(main, clean_paths_visit_cb, BLI_BPATH_TRAVERSE_SKIP_MULTIFILE, NULL);
183
184         for (scene = main->scene.first; scene; scene = scene->id.next) {
185                 BLI_clean(scene->r.pic);
186         }
187 }
188
189 /* context matching */
190 /* handle no-ui case */
191
192 /* note, this is called on Undo so any slow conversion functions here
193  * should be avoided or check (mode!='u') */
194
195 static void setup_app_data(bContext *C, BlendFileData *bfd, const char *filepath)
196 {
197         bScreen *curscreen = NULL;
198         Scene *curscene = NULL;
199         int recover;
200         char mode;
201
202         /* 'u' = undo save, 'n' = no UI load */
203         if (bfd->main->screen.first == NULL) mode = 'u';
204         else if (G.fileflags & G_FILE_NO_UI) mode = 'n';
205         else mode = 0;
206
207         recover = (G.fileflags & G_FILE_RECOVER);
208
209         /* Free all render results, without this stale data gets displayed after loading files */
210         if (mode != 'u') {
211                 RE_FreeAllRenderResults();
212         }
213
214         /* Only make filepaths compatible when loading for real (not undo) */
215         if (mode != 'u') {
216                 clean_paths(bfd->main);
217         }
218
219         /* XXX here the complex windowmanager matching */
220         
221         /* no load screens? */
222         if (mode) {
223                 /* comes from readfile.c */
224                 SWAP(ListBase, G.main->wm, bfd->main->wm);
225                 SWAP(ListBase, G.main->screen, bfd->main->screen);
226                 SWAP(ListBase, G.main->script, bfd->main->script);
227                 
228                 /* we re-use current screen */
229                 curscreen = CTX_wm_screen(C);
230                 /* but use new Scene pointer */
231                 curscene = bfd->curscene;
232                 if (curscene == NULL) curscene = bfd->main->scene.first;
233                 /* and we enforce curscene to be in current screen */
234                 if (curscreen) curscreen->scene = curscene;  /* can run in bgmode */
235
236                 /* clear_global will free G.main, here we can still restore pointers */
237                 blo_lib_link_screen_restore(bfd->main, curscreen, curscene);
238         }
239         
240         /* free G.main Main database */
241 //      CTX_wm_manager_set(C, NULL);
242         clear_global();
243         
244         /* clear old property update cache, in case some old references are left dangling */
245         RNA_property_update_cache_free();
246         
247         G.main = bfd->main;
248
249         CTX_data_main_set(C, G.main);
250
251         sound_init_main(G.main);
252         
253         if (bfd->user) {
254                 
255                 /* only here free userdef themes... */
256                 BKE_userdef_free();
257                 
258                 U = *bfd->user;
259                 MEM_freeN(bfd->user);
260         }
261         
262         /* case G_FILE_NO_UI or no screens in file */
263         if (mode) {
264                 /* leave entire context further unaltered? */
265                 CTX_data_scene_set(C, curscene);
266         }
267         else {
268                 G.winpos = bfd->winpos;
269                 G.displaymode = bfd->displaymode;
270                 G.fileflags = bfd->fileflags;
271                 CTX_wm_manager_set(C, G.main->wm.first);
272                 CTX_wm_screen_set(C, bfd->curscreen);
273                 CTX_data_scene_set(C, bfd->curscreen->scene);
274                 CTX_wm_area_set(C, NULL);
275                 CTX_wm_region_set(C, NULL);
276                 CTX_wm_menu_set(C, NULL);
277         }
278         
279         /* this can happen when active scene was lib-linked, and doesn't exist anymore */
280         if (CTX_data_scene(C) == NULL) {
281                 /* in case we don't even have a local scene, add one */
282                 if (!G.main->scene.first)
283                         BKE_scene_add("Scene");
284
285                 CTX_data_scene_set(C, G.main->scene.first);
286                 CTX_wm_screen(C)->scene = CTX_data_scene(C);
287                 curscene = CTX_data_scene(C);
288         }
289
290         /* special cases, override loaded flags: */
291         if (G.f != bfd->globalf) {
292                 const int flags_keep = (G_SWAP_EXCHANGE | G_SCRIPT_AUTOEXEC | G_SCRIPT_OVERRIDE_PREF);
293                 bfd->globalf = (bfd->globalf & ~flags_keep) | (G.f & flags_keep);
294         }
295
296
297         G.f = bfd->globalf;
298
299 #ifdef WITH_PYTHON
300         /* let python know about new main */
301         BPY_context_update(C);
302 #endif
303
304         if (!G.background) {
305                 //setscreen(G.curscreen);
306         }
307         
308         /* FIXME: this version patching should really be part of the file-reading code,
309          * but we still get too many unrelated data-corruption crashes otherwise... */
310         if (G.main->versionfile < 250)
311                 do_versions_ipos_to_animato(G.main);
312         
313         if (recover && bfd->filename[0] && G.relbase_valid) {
314                 /* in case of autosave or quit.blend, use original filename instead
315                  * use relbase_valid to make sure the file is saved, else we get <memory2> in the filename */
316                 filepath = bfd->filename;
317         }
318 #if 0
319         else if (!G.relbase_valid) {
320                 /* otherwise, use an empty string as filename, rather than <memory2> */
321                 filepath = "";
322         }
323 #endif
324         
325         /* these are the same at times, should never copy to the same location */
326         if (G.main->name != filepath)
327                 BLI_strncpy(G.main->name, filepath, FILE_MAX);
328
329         /* baseflags, groups, make depsgraph, etc */
330         BKE_scene_set_background(G.main, CTX_data_scene(C));
331
332         if (mode != 'u') {
333                 IMB_colormanagement_check_file_config(G.main);
334         }
335
336         MEM_freeN(bfd);
337
338         (void)curscene; /* quiet warning */
339 }
340
341 static int handle_subversion_warning(Main *main, ReportList *reports)
342 {
343         if (main->minversionfile > BLENDER_VERSION ||
344             (main->minversionfile == BLENDER_VERSION &&
345              main->minsubversionfile > BLENDER_SUBVERSION))
346         {
347                 BKE_reportf(reports, RPT_ERROR, "File written by newer Blender binary (%d.%d), expect loss of data!",
348                             main->minversionfile, main->minsubversionfile);
349         }
350
351         return 1;
352 }
353
354 static void keymap_item_free(wmKeyMapItem *kmi)
355 {
356         if (kmi->properties) {
357                 IDP_FreeProperty(kmi->properties);
358                 MEM_freeN(kmi->properties);
359         }
360         if (kmi->ptr)
361                 MEM_freeN(kmi->ptr);
362 }
363
364 void BKE_userdef_free(void)
365 {
366         wmKeyMap *km;
367         wmKeyMapItem *kmi;
368         wmKeyMapDiffItem *kmdi;
369
370         for (km = U.user_keymaps.first; km; km = km->next) {
371                 for (kmdi = km->diff_items.first; kmdi; kmdi = kmdi->next) {
372                         if (kmdi->add_item) {
373                                 keymap_item_free(kmdi->add_item);
374                                 MEM_freeN(kmdi->add_item);
375                         }
376                         if (kmdi->remove_item) {
377                                 keymap_item_free(kmdi->remove_item);
378                                 MEM_freeN(kmdi->remove_item);
379                         }
380                 }
381
382                 for (kmi = km->items.first; kmi; kmi = kmi->next)
383                         keymap_item_free(kmi);
384
385                 BLI_freelistN(&km->diff_items);
386                 BLI_freelistN(&km->items);
387         }
388         
389         BLI_freelistN(&U.uistyles);
390         BLI_freelistN(&U.uifonts);
391         BLI_freelistN(&U.themes);
392         BLI_freelistN(&U.user_keymaps);
393         BLI_freelistN(&U.addons);
394 }
395
396 int BKE_read_file(bContext *C, const char *filepath, ReportList *reports)
397 {
398         BlendFileData *bfd;
399         int retval = BKE_READ_FILE_OK;
400
401         if (strstr(filepath, BLENDER_STARTUP_FILE) == NULL) /* don't print user-pref loading */
402                 printf("read blend: %s\n", filepath);
403
404         bfd = BLO_read_from_file(filepath, reports);
405         if (bfd) {
406                 if (bfd->user) retval = BKE_READ_FILE_OK_USERPREFS;
407                 
408                 if (0 == handle_subversion_warning(bfd->main, reports)) {
409                         free_main(bfd->main);
410                         MEM_freeN(bfd);
411                         bfd = NULL;
412                         retval = BKE_READ_FILE_FAIL;
413                 }
414                 else
415                         setup_app_data(C, bfd, filepath);  // frees BFD
416         }
417         else
418                 BKE_reports_prependf(reports, "Loading '%s' failed: ", filepath);
419                 
420         return (bfd ? retval : BKE_READ_FILE_FAIL);
421 }
422
423 int BKE_read_file_from_memory(bContext *C, char *filebuf, int filelength, ReportList *reports)
424 {
425         BlendFileData *bfd;
426
427         bfd = BLO_read_from_memory(filebuf, filelength, reports);
428         if (bfd)
429                 setup_app_data(C, bfd, "<memory2>");
430         else
431                 BKE_reports_prepend(reports, "Loading failed: ");
432
433         return (bfd ? 1 : 0);
434 }
435
436 /* memfile is the undo buffer */
437 int BKE_read_file_from_memfile(bContext *C, MemFile *memfile, ReportList *reports)
438 {
439         BlendFileData *bfd;
440
441         bfd = BLO_read_from_memfile(CTX_data_main(C), G.main->name, memfile, reports);
442         if (bfd)
443                 setup_app_data(C, bfd, "<memory1>");
444         else
445                 BKE_reports_prepend(reports, "Loading failed: ");
446
447         return (bfd ? 1 : 0);
448 }
449
450
451 /* *****************  testing for break ************* */
452
453 static void (*blender_test_break_cb)(void) = NULL;
454
455 void set_blender_test_break_cb(void (*func)(void) )
456 {
457         blender_test_break_cb = func;
458 }
459
460
461 int blender_test_break(void)
462 {
463         if (!G.background) {
464                 if (blender_test_break_cb)
465                         blender_test_break_cb();
466         }
467         
468         return (G.is_break == TRUE);
469 }
470
471
472 /* ***************** GLOBAL UNDO *************** */
473
474 #define UNDO_DISK   0
475
476 typedef struct UndoElem {
477         struct UndoElem *next, *prev;
478         char str[FILE_MAX];
479         char name[BKE_UNDO_STR_MAX];
480         MemFile memfile;
481         uintptr_t undosize;
482 } UndoElem;
483
484 static ListBase undobase = {NULL, NULL};
485 static UndoElem *curundo = NULL;
486
487
488 static int read_undosave(bContext *C, UndoElem *uel)
489 {
490         char mainstr[sizeof(G.main->name)];
491         int success = 0, fileflags;
492         
493         /* This is needed so undoing/redoing doesn't crash with threaded previews going */
494         WM_jobs_kill_all_except(CTX_wm_manager(C), CTX_wm_screen(C));
495
496         BLI_strncpy(mainstr, G.main->name, sizeof(mainstr));    /* temporal store */
497
498         fileflags = G.fileflags;
499         G.fileflags |= G_FILE_NO_UI;
500
501         if (UNDO_DISK) 
502                 success = (BKE_read_file(C, uel->str, NULL) != BKE_READ_FILE_FAIL);
503         else
504                 success = BKE_read_file_from_memfile(C, &uel->memfile, NULL);
505
506         /* restore */
507         BLI_strncpy(G.main->name, mainstr, sizeof(G.main->name)); /* restore */
508         G.fileflags = fileflags;
509
510         if (success) {
511                 /* important not to update time here, else non keyed tranforms are lost */
512                 DAG_on_visible_update(G.main, FALSE);
513         }
514
515         return success;
516 }
517
518 /* name can be a dynamic string */
519 void BKE_write_undo(bContext *C, const char *name)
520 {
521         uintptr_t maxmem, totmem, memused;
522         int nr /*, success */ /* UNUSED */;
523         UndoElem *uel;
524         
525         if ((U.uiflag & USER_GLOBALUNDO) == 0) {
526                 return;
527         }
528
529         if (U.undosteps == 0) {
530                 return;
531         }
532         
533         /* remove all undos after (also when curundo == NULL) */
534         while (undobase.last != curundo) {
535                 uel = undobase.last;
536                 BLI_remlink(&undobase, uel);
537                 BLO_free_memfile(&uel->memfile);
538                 MEM_freeN(uel);
539         }
540         
541         /* make new */
542         curundo = uel = MEM_callocN(sizeof(UndoElem), "undo file");
543         BLI_strncpy(uel->name, name, sizeof(uel->name));
544         BLI_addtail(&undobase, uel);
545         
546         /* and limit amount to the maximum */
547         nr = 0;
548         uel = undobase.last;
549         while (uel) {
550                 nr++;
551                 if (nr == U.undosteps) break;
552                 uel = uel->prev;
553         }
554         if (uel) {
555                 while (undobase.first != uel) {
556                         UndoElem *first = undobase.first;
557                         BLI_remlink(&undobase, first);
558                         /* the merge is because of compression */
559                         BLO_merge_memfile(&first->memfile, &first->next->memfile);
560                         MEM_freeN(first);
561                 }
562         }
563
564
565         /* disk save version */
566         if (UNDO_DISK) {
567                 static int counter = 0;
568                 char filepath[FILE_MAX];
569                 char numstr[32];
570                 int fileflags = G.fileflags & ~(G_FILE_HISTORY); /* don't do file history on undo */
571
572                 /* calculate current filepath */
573                 counter++;
574                 counter = counter % U.undosteps;
575         
576                 BLI_snprintf(numstr, sizeof(numstr), "%d.blend", counter);
577                 BLI_make_file_string("/", filepath, BLI_temporary_dir(), numstr);
578         
579                 /* success = */ /* UNUSED */ BLO_write_file(CTX_data_main(C), filepath, fileflags, NULL, NULL);
580                 
581                 BLI_strncpy(curundo->str, filepath, sizeof(curundo->str));
582         }
583         else {
584                 MemFile *prevfile = NULL;
585                 
586                 if (curundo->prev) prevfile = &(curundo->prev->memfile);
587                 
588                 memused = MEM_get_memory_in_use();
589                 /* success = */ /* UNUSED */ BLO_write_file_mem(CTX_data_main(C), prevfile, &curundo->memfile, G.fileflags);
590                 curundo->undosize = MEM_get_memory_in_use() - memused;
591         }
592
593         if (U.undomemory != 0) {
594                 /* limit to maximum memory (afterwards, we can't know in advance) */
595                 totmem = 0;
596                 maxmem = ((uintptr_t)U.undomemory) * 1024 * 1024;
597
598                 /* keep at least two (original + other) */
599                 uel = undobase.last;
600                 while (uel && uel->prev) {
601                         totmem += uel->undosize;
602                         if (totmem > maxmem) break;
603                         uel = uel->prev;
604                 }
605
606                 if (uel) {
607                         if (uel->prev && uel->prev->prev)
608                                 uel = uel->prev;
609
610                         while (undobase.first != uel) {
611                                 UndoElem *first = undobase.first;
612                                 BLI_remlink(&undobase, first);
613                                 /* the merge is because of compression */
614                                 BLO_merge_memfile(&first->memfile, &first->next->memfile);
615                                 MEM_freeN(first);
616                         }
617                 }
618         }
619 }
620
621 /* 1 = an undo, -1 is a redo. we have to make sure 'curundo' remains at current situation */
622 void BKE_undo_step(bContext *C, int step)
623 {
624         
625         if (step == 0) {
626                 read_undosave(C, curundo);
627         }
628         else if (step == 1) {
629                 /* curundo should never be NULL, after restart or load file it should call undo_save */
630                 if (curundo == NULL || curundo->prev == NULL) {
631                         // XXX error("No undo available");
632                 }
633                 else {
634                         if (G.debug & G_DEBUG) printf("undo %s\n", curundo->name);
635                         curundo = curundo->prev;
636                         read_undosave(C, curundo);
637                 }
638         }
639         else {
640                 /* curundo has to remain current situation! */
641                 
642                 if (curundo == NULL || curundo->next == NULL) {
643                         // XXX error("No redo available");
644                 }
645                 else {
646                         read_undosave(C, curundo->next);
647                         curundo = curundo->next;
648                         if (G.debug & G_DEBUG) printf("redo %s\n", curundo->name);
649                 }
650         }
651 }
652
653 void BKE_reset_undo(void)
654 {
655         UndoElem *uel;
656         
657         uel = undobase.first;
658         while (uel) {
659                 BLO_free_memfile(&uel->memfile);
660                 uel = uel->next;
661         }
662         
663         BLI_freelistN(&undobase);
664         curundo = NULL;
665 }
666
667 /* based on index nr it does a restore */
668 void BKE_undo_number(bContext *C, int nr)
669 {
670         curundo = BLI_findlink(&undobase, nr);
671         BKE_undo_step(C, 0);
672 }
673
674 /* go back to the last occurance of name in stack */
675 void BKE_undo_name(bContext *C, const char *name)
676 {
677         UndoElem *uel = BLI_rfindstring(&undobase, name, offsetof(UndoElem, name));
678
679         if (uel && uel->prev) {
680                 curundo = uel->prev;
681                 BKE_undo_step(C, 0);
682         }
683 }
684
685 /* name optional */
686 int BKE_undo_valid(const char *name)
687 {
688         if (name) {
689                 UndoElem *uel = BLI_rfindstring(&undobase, name, offsetof(UndoElem, name));
690                 return uel && uel->prev;
691         }
692         
693         return undobase.last != undobase.first;
694 }
695
696 /* get name of undo item, return null if no item with this index */
697 /* if active pointer, set it to 1 if true */
698 const char *BKE_undo_get_name(int nr, int *active)
699 {
700         UndoElem *uel = BLI_findlink(&undobase, nr);
701         
702         if (active) *active = 0;
703         
704         if (uel) {
705                 if (active && uel == curundo)
706                         *active = 1;
707                 return uel->name;
708         }
709         return NULL;
710 }
711
712 char *BKE_undo_menu_string(void)
713 {
714         UndoElem *uel;
715         DynStr *ds = BLI_dynstr_new();
716         char *menu;
717
718         BLI_dynstr_append(ds, "Global Undo History %t");
719         
720         for (uel = undobase.first; uel; uel = uel->next) {
721                 BLI_dynstr_append(ds, "|");
722                 BLI_dynstr_append(ds, uel->name);
723         }
724
725         menu = BLI_dynstr_get_cstring(ds);
726         BLI_dynstr_free(ds);
727
728         return menu;
729 }
730
731 /* saves quit.blend */
732 void BKE_undo_save_quit(void)
733 {
734         UndoElem *uel;
735         MemFileChunk *chunk;
736         char str[FILE_MAX];
737         const int flag = O_BINARY + O_WRONLY + O_CREAT + O_TRUNC + O_EXCL;
738         int file;
739
740         if ((U.uiflag & USER_GLOBALUNDO) == 0) {
741                 return;
742         }
743
744         uel = curundo;
745         if (uel == NULL) {
746                 fprintf(stderr, "No undo buffer to save recovery file\n");
747                 return;
748         }
749
750         /* no undo state to save */
751         if (undobase.first == undobase.last) {
752                 return;
753         }
754
755         /* save the undo state as quit.blend */
756         BLI_make_file_string("/", str, BLI_temporary_dir(), "quit.blend");
757
758         /* first try create the file, if it exists call without 'O_CREAT',
759          * to avoid writing to a symlink - use 'O_EXCL' (CVE-2008-1103) */
760         errno = 0;
761         file = BLI_open(str, flag, 0666);
762         if (file == -1) {
763                 if (errno == EEXIST) {
764                         errno = 0;
765                         file = BLI_open(str, flag & ~O_CREAT, 0666);
766                 }
767         }
768
769         if (file == -1) {
770                 fprintf(stderr, "Unable to save '%s': %s\n",
771                         str, errno ? strerror(errno) : "Unknown error opening file");
772                 return;
773         }
774
775         for (chunk = uel->memfile.chunks.first; chunk; chunk = chunk->next) {
776                 if (write(file, chunk->buf, chunk->size) != chunk->size) {
777                         break;
778                 }
779         }
780
781         close(file);
782         
783         if (chunk) {
784                 fprintf(stderr, "Unable to save '%s': %s\n",
785                         str, errno ? strerror(errno) : "Unknown error writing file");
786         }
787         else {
788                 printf("Saved session recovery to '%s'\n", str);
789         }
790 }
791
792 /* sets curscene */
793 Main *BKE_undo_get_main(Scene **scene)
794 {
795         Main *mainp = NULL;
796         BlendFileData *bfd = BLO_read_from_memfile(G.main, G.main->name, &curundo->memfile, NULL);
797         
798         if (bfd) {
799                 mainp = bfd->main;
800                 if (scene)
801                         *scene = bfd->curscene;
802                 
803                 MEM_freeN(bfd);
804         }
805         
806         return mainp;
807 }
808