Adding a bit more checking on file operations.
[debian/gnuradio] / gnuradio-core / src / lib / runtime / gr_preferences.cc
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2003,2010 Free Software Foundation, Inc.
4  * 
5  * This file is part of GNU Radio
6  * 
7  * GNU Radio is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3, or (at your option)
10  * any later version.
11  * 
12  * GNU Radio is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with GNU Radio; see the file COPYING.  If not, write to
19  * the Free Software Foundation, Inc., 51 Franklin Street,
20  * Boston, MA 02110-1301, USA.
21  */
22
23 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26
27 #include <gr_preferences.h>
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <string.h>
31 #include <sys/types.h>
32 #include <sys/stat.h>
33 #include <unistd.h>
34 #include <string.h>
35
36
37 #ifdef MKDIR_TAKES_ONE_ARG
38 #define gr_mkdir(pathname, mode) mkdir(pathname)
39 #else
40 #define gr_mkdir(pathname, mode) mkdir((pathname), (mode))
41 #endif
42
43 /*
44  * The simplest thing that could possibly work:
45  *  the key is the filename; the value is the file contents.
46  */
47
48 static const char *
49 pathname (const char *key)
50 {
51   static char buf[200];
52   snprintf (buf, sizeof (buf), "%s/.gnuradio/prefs/%s", getenv ("HOME"), key);
53   return buf;
54 }
55
56 static void
57 ensure_dir_path ()
58 {
59   char path[200];
60   struct stat   statbuf;
61
62   snprintf (path, sizeof (path), "%s/.gnuradio/prefs", getenv ("HOME"));
63   if (stat (path, &statbuf) == 0 && S_ISDIR (statbuf.st_mode))
64     return;
65
66   // blindly try to make it     // FIXME make this robust. C++ SUCKS!
67
68   snprintf (path, sizeof (path), "%s/.gnuradio", getenv ("HOME"));
69   gr_mkdir (path, 0750);
70   snprintf (path, sizeof (path), "%s/.gnuradio/prefs", getenv ("HOME"));
71   gr_mkdir (path, 0750);
72 }
73
74 const char *
75 gr_preferences::get (const char *key)
76 {
77   static char buf[1024];
78
79   FILE  *fp = fopen (pathname (key), "r");
80   if (fp == 0) {
81     perror (pathname (key));
82     return 0;
83   }
84
85   memset (buf, 0, sizeof (buf));
86   size_t ret = fread (buf, 1, sizeof (buf) - 1, fp);
87   if(ret == 0) {
88     if(ferror(fp) != 0) {
89       perror (pathname (key));
90       fclose (fp);
91       return 0;
92     }
93   }
94   fclose (fp);
95   return buf;
96 }
97
98 void 
99 gr_preferences::set (const char *key, const char *value)
100 {
101   ensure_dir_path ();
102
103   FILE *fp = fopen (pathname (key), "w");
104   if (fp == 0){
105     perror (pathname (key));
106     return;
107   }
108
109   size_t ret = fwrite (value, 1, strlen (value), fp);
110   if(ret == 0) {
111     if(ferror(fp) != 0) {
112       perror (pathname (key));
113       fclose (fp);
114       return;
115     }
116   }
117   fclose (fp);
118 };