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