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