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.2 2001/03/04 10:29:20 oetiker
9 * fixed filedescriptor leak
10 * -- Mike Franusich <mike@franusich.com>
11 *
12 * Revision 1.1.1.1 2001/02/25 22:25:05 oetiker
13 * checkin
14 *
15 *****************************************************************************/
17 #include "rrd_tool.h"
18 #define MEMBLK 8192
20 /* open a database file, return its header and a open filehandle */
21 /* positioned to the first cdp in the first rra */
23 int
24 rrd_open(char *file_name, FILE **in_file, rrd_t *rrd, int rdwr)
25 {
28 char *mode = NULL;
29 rrd_init(rrd);
30 if (rdwr == RRD_READONLY) {
31 #ifndef WIN32
32 mode = "r";
33 #else
34 mode = "rb";
35 #endif
36 } else {
37 #ifndef WIN32
38 mode = "r+";
39 #else
40 mode = "rb+";
41 #endif
42 }
44 if (((*in_file) = fopen(file_name,mode)) == NULL ){
45 rrd_set_error("opening '%s': %s",file_name, strerror(errno));
46 return (-1);
47 }
49 #define MYFREAD(MYVAR,MYVART,MYCNT) \
50 if ((MYVAR = malloc(sizeof(MYVART) * MYCNT)) == NULL) {\
51 rrd_set_error("" #MYVAR " malloc"); \
52 fclose(*in_file); \
53 return (-1); } \
54 fread(MYVAR,sizeof(MYVART),MYCNT, *in_file);
57 MYFREAD(rrd->stat_head, stat_head_t, 1)
59 /* lets do some test if we are on track ... */
60 if (strncmp(rrd->stat_head->cookie,RRD_COOKIE,4) != 0){
61 rrd_set_error("'%s' is not an RRD file",file_name);
62 free(rrd->stat_head);
63 fclose(*in_file);
64 return(-1);}
66 if (strncmp(rrd->stat_head->version,RRD_VERSION,5) != 0){
67 rrd_set_error("can't handle RRD file version %s",
68 rrd->stat_head->version);
69 free(rrd->stat_head);
70 fclose(*in_file);
71 return(-1);}
73 if (rrd->stat_head->float_cookie != FLOAT_COOKIE){
74 rrd_set_error("This RRD was created on other architecture");
75 free(rrd->stat_head);
76 fclose(*in_file);
77 return(-1);}
79 MYFREAD(rrd->ds_def, ds_def_t, rrd->stat_head->ds_cnt)
80 MYFREAD(rrd->rra_def, rra_def_t, rrd->stat_head->rra_cnt)
81 MYFREAD(rrd->live_head, live_head_t, 1)
82 MYFREAD(rrd->pdp_prep, pdp_prep_t, rrd->stat_head->ds_cnt)
83 MYFREAD(rrd->cdp_prep, cdp_prep_t, (rrd->stat_head->rra_cnt
84 * rrd->stat_head->ds_cnt))
85 MYFREAD(rrd->rra_ptr, rra_ptr_t, rrd->stat_head->rra_cnt)
86 #undef MYFREAD
88 return(0);
89 }
91 void rrd_init(rrd_t *rrd)
92 {
93 rrd->stat_head = NULL;
94 rrd->ds_def = NULL;
95 rrd->rra_def = NULL;
96 rrd->live_head = NULL;
97 rrd->rra_ptr = NULL;
98 rrd->pdp_prep = NULL;
99 rrd->cdp_prep = NULL;
100 rrd->rrd_value = NULL;
101 }
103 void rrd_free(rrd_t *rrd)
104 {
105 free(rrd->stat_head);
106 free(rrd->ds_def);
107 free(rrd->rra_def);
108 free(rrd->live_head);
109 free(rrd->rra_ptr);
110 free(rrd->pdp_prep);
111 free(rrd->cdp_prep);
112 free(rrd->rrd_value);
113 }
115 int readfile(char *file_name, char **buffer, int skipfirst){
116 long writecnt=0,totalcnt = MEMBLK;
117 FILE *input=NULL;
118 char c ;
119 if ((strcmp("-",file_name) == 0)) { input = stdin; }
120 else {
121 if ((input = fopen(file_name,"rb")) == NULL ){
122 rrd_set_error("opening '%s': %s",file_name,strerror(errno));
123 return (-1);
124 }
125 }
126 if (skipfirst){
127 do { c = getc(input); } while (c != '\n' && ! feof(input));
128 }
129 if (((*buffer) = (char *) malloc((MEMBLK+4)*sizeof(char))) == NULL) {
130 perror("Allocate Buffer:");
131 exit(1);
132 };
133 do{
134 writecnt += fread((*buffer)+writecnt, 1, MEMBLK * sizeof(char) ,input);
135 if (writecnt >= totalcnt){
136 totalcnt += MEMBLK;
137 if (((*buffer)=rrd_realloc((*buffer), (totalcnt+4) * sizeof(char)))==NULL){
138 perror("Realloc Buffer:");
139 exit(1);
140 };
141 }
142 } while (! feof(input));
143 (*buffer)[writecnt] = '\0';
144 if (strcmp("-",file_name) != 0) {fclose(input);};
145 return writecnt;
146 }