1 /*****************************************************************************
2 * RRDtool 1.1.x Copyright Tobias Oetiker, 1997 - 2002
3 *****************************************************************************
4 * rrd_open.c Open an RRD File
5 *****************************************************************************
6 * $Id$
7 * $Log$
8 * Revision 1.5 2002/06/20 00:21:03 jake
9 * More Win32 build changes; thanks to Kerry Calvert.
10 *
11 * Revision 1.4 2002/02/01 20:34:49 oetiker
12 * fixed version number and date/time
13 *
14 * Revision 1.3 2001/03/04 13:01:55 oetiker
15 * Aberrant Behavior Detection support. A brief overview added to rrdtool.pod.
16 * Major updates to rrd_update.c, rrd_create.c. Minor update to other core files.
17 * This is backwards compatible! But new files using the Aberrant stuff are not readable
18 * by old rrdtool versions. See http://cricket.sourceforge.net/aberrant/rrd_hw.htm
19 * -- Jake Brutlag <jakeb@corp.webtv.net>
20 *
21 * Revision 1.2 2001/03/04 10:29:20 oetiker
22 * fixed filedescriptor leak
23 * -- Mike Franusich <mike@franusich.com>
24 *
25 * Revision 1.1.1.1 2001/02/25 22:25:05 oetiker
26 * checkin
27 *
28 *****************************************************************************/
30 #include "rrd_tool.h"
31 #define MEMBLK 8192
33 /* open a database file, return its header and a open filehandle */
34 /* positioned to the first cdp in the first rra */
36 int
37 rrd_open(char *file_name, FILE **in_file, rrd_t *rrd, int rdwr)
38 {
41 char *mode = NULL;
42 rrd_init(rrd);
43 if (rdwr == RRD_READONLY) {
44 #ifndef WIN32
45 mode = "r";
46 #else
47 mode = "rb";
48 #endif
49 } else {
50 #ifndef WIN32
51 mode = "r+";
52 #else
53 mode = "rb+";
54 #endif
55 }
57 if (((*in_file) = fopen(file_name,mode)) == NULL ){
58 rrd_set_error("opening '%s': %s",file_name, strerror(errno));
59 return (-1);
60 }
61 /*
62 if (rdwr == RRD_READWRITE)
63 {
64 if (setvbuf((*in_file),NULL,_IONBF,2)) {
65 rrd_set_error("failed to disable the stream buffer\n");
66 return (-1);
67 }
68 }
69 */
71 #define MYFREAD(MYVAR,MYVART,MYCNT) \
72 if ((MYVAR = malloc(sizeof(MYVART) * MYCNT)) == NULL) {\
73 rrd_set_error("" #MYVAR " malloc"); \
74 fclose(*in_file); \
75 return (-1); } \
76 fread(MYVAR,sizeof(MYVART),MYCNT, *in_file);
79 MYFREAD(rrd->stat_head, stat_head_t, 1)
81 /* lets do some test if we are on track ... */
82 if (strncmp(rrd->stat_head->cookie,RRD_COOKIE,4) != 0){
83 rrd_set_error("'%s' is not an RRD file",file_name);
84 free(rrd->stat_head);
85 fclose(*in_file);
86 return(-1);}
88 if (atoi(rrd->stat_head->version) > atoi(RRD_VERSION)){
89 rrd_set_error("can't handle RRD file version %s",
90 rrd->stat_head->version);
91 free(rrd->stat_head);
92 fclose(*in_file);
93 return(-1);}
95 if (rrd->stat_head->float_cookie != FLOAT_COOKIE){
96 rrd_set_error("This RRD was created on other architecture");
97 free(rrd->stat_head);
98 fclose(*in_file);
99 return(-1);}
101 MYFREAD(rrd->ds_def, ds_def_t, rrd->stat_head->ds_cnt)
102 MYFREAD(rrd->rra_def, rra_def_t, rrd->stat_head->rra_cnt)
103 MYFREAD(rrd->live_head, live_head_t, 1)
104 MYFREAD(rrd->pdp_prep, pdp_prep_t, rrd->stat_head->ds_cnt)
105 MYFREAD(rrd->cdp_prep, cdp_prep_t, (rrd->stat_head->rra_cnt
106 * rrd->stat_head->ds_cnt))
107 MYFREAD(rrd->rra_ptr, rra_ptr_t, rrd->stat_head->rra_cnt)
108 #undef MYFREAD
110 return(0);
111 }
113 void rrd_init(rrd_t *rrd)
114 {
115 rrd->stat_head = NULL;
116 rrd->ds_def = NULL;
117 rrd->rra_def = NULL;
118 rrd->live_head = NULL;
119 rrd->rra_ptr = NULL;
120 rrd->pdp_prep = NULL;
121 rrd->cdp_prep = NULL;
122 rrd->rrd_value = NULL;
123 }
125 void rrd_free(rrd_t *rrd)
126 {
127 free(rrd->stat_head);
128 free(rrd->ds_def);
129 free(rrd->rra_def);
130 free(rrd->live_head);
131 free(rrd->rra_ptr);
132 free(rrd->pdp_prep);
133 free(rrd->cdp_prep);
134 free(rrd->rrd_value);
135 }
137 /* routine used by external libraries to free memory allocated by
138 * rrd library */
139 void rrd_freemem(void *mem)
140 {
142 free(mem);
143 }
145 int readfile(char *file_name, char **buffer, int skipfirst){
146 long writecnt=0,totalcnt = MEMBLK;
147 FILE *input=NULL;
148 char c ;
149 if ((strcmp("-",file_name) == 0)) { input = stdin; }
150 else {
151 if ((input = fopen(file_name,"rb")) == NULL ){
152 rrd_set_error("opening '%s': %s",file_name,strerror(errno));
153 return (-1);
154 }
155 }
156 if (skipfirst){
157 do { c = getc(input); } while (c != '\n' && ! feof(input));
158 }
159 if (((*buffer) = (char *) malloc((MEMBLK+4)*sizeof(char))) == NULL) {
160 perror("Allocate Buffer:");
161 exit(1);
162 };
163 do{
164 writecnt += fread((*buffer)+writecnt, 1, MEMBLK * sizeof(char) ,input);
165 if (writecnt >= totalcnt){
166 totalcnt += MEMBLK;
167 if (((*buffer)=rrd_realloc((*buffer), (totalcnt+4) * sizeof(char)))==NULL){
168 perror("Realloc Buffer:");
169 exit(1);
170 };
171 }
172 } while (! feof(input));
173 (*buffer)[writecnt] = '\0';
174 if (strcmp("-",file_name) != 0) {fclose(input);};
175 return writecnt;
176 }