1 /*****************************************************************************
2 * RRDtool 1.2.23 Copyright by Tobi Oetiker, 1997-2007
3 *****************************************************************************
4 * change header parameters of an rrd
5 *****************************************************************************
6 * $Id$
7 * $Log$
8 * Revision 1.6 2004/05/26 22:11:12 oetiker
9 * reduce compiler warnings. Many small fixes. -- Mike Slifcak <slif@bellsouth.net>
10 *
11 * Revision 1.5 2002/02/01 20:34:49 oetiker
12 * fixed version number and date/time
13 *
14 * Revision 1.4 2001/08/22 22:29:07 jake
15 * Contents of this patch:
16 * (1) Adds/revises documentation for rrd tune in rrd_tool.c and pod files.
17 * (2) Moves some initialization code from rrd_create.c to rrd_hw.c.
18 * (3) Adds another pass to smoothing for SEASONAL and DEVSEASONAL RRAs.
19 * This pass computes the coefficients as deviations from an average; the
20 * average is added the baseline coefficient of HWPREDICT. Statistical texts
21 * suggest this to preserve algorithm stability. It will not invalidate
22 * RRD files created and smoothed with the old code.
23 * (4) Adds the aberrant-reset flag to rrd tune. This operation, which is
24 * specified for a single data source, causes the holt-winters algorithm to
25 * forget everthing it has learned and start over.
26 * (5) Fixes a few out-of-date code comments.
27 *
28 * Revision 1.3 2001/03/07 21:21:54 oetiker
29 * complete rewrite of rrdgraph documentation. This also includs info
30 * on upcomming/planned changes to the rrdgraph interface and functionality
31 * -- Alex van den Bogaerdt <alex@slot.hollandcasino.nl>
32 *
33 * Revision 1.2 2001/03/04 13:01:55 oetiker
34 * Aberrant Behavior Detection support. A brief overview added to rrdtool.pod.
35 * Major updates to rrd_update.c, rrd_create.c. Minor update to other core files.
36 * This is backwards compatible! But new files using the Aberrant stuff are not readable
37 * by old rrdtool versions. See http://cricket.sourceforge.net/aberrant/rrd_hw.htm
38 * -- Jake Brutlag <jakeb@corp.webtv.net>
39 *
40 *****************************************************************************/
42 #include "rrd_tool.h"
43 #include "rrd_rpncalc.h"
44 #include "rrd_hw.h"
46 int set_hwarg(rrd_t *rrd,enum cf_en cf,enum rra_par_en rra_par,char *arg);
47 int set_deltaarg(rrd_t *rrd,enum rra_par_en rra_par,char *arg);
48 int set_windowarg(rrd_t *rrd,enum rra_par_en,char *arg);
50 int
51 rrd_tune(int argc, char **argv)
52 {
53 rrd_t rrd;
54 int matches;
55 int optcnt = 0;
56 long ds;
57 char ds_nam[DS_NAM_SIZE];
58 char ds_new[DS_NAM_SIZE];
59 long heartbeat;
60 double min;
61 double max;
62 char dst[DST_SIZE];
63 rrd_file_t *rrd_file;
64 optind = 0; opterr = 0; /* initialize getopt */
67 rrd_file = rrd_open(argv[1],&rrd,RRD_READWRITE);
68 if (rrd_file == NULL) {
69 return -1;
70 }
72 while (1){
73 static struct option long_options[] =
74 {
75 {"heartbeat", required_argument, 0, 'h'},
76 {"minimum", required_argument, 0, 'i'},
77 {"maximum", required_argument, 0, 'a'},
78 {"data-source-type", required_argument, 0, 'd'},
79 {"data-source-rename", required_argument, 0, 'r'},
80 /* added parameter tuning options for aberrant behavior detection */
81 {"deltapos",required_argument,0,'p'},
82 {"deltaneg",required_argument,0,'n'},
83 {"window-length",required_argument,0,'w'},
84 {"failure-threshold",required_argument,0,'f'},
85 {"alpha",required_argument,0,'x'},
86 {"beta",required_argument,0,'y'},
87 {"gamma",required_argument,0,'z'},
88 {"gamma-deviation",required_argument,0,'v'},
89 {"aberrant-reset",required_argument,0,'b'},
90 {0,0,0,0}
91 };
92 int option_index = 0;
93 int opt;
94 opt = getopt_long(argc, argv, "h:i:a:d:r:p:n:w:f:x:y:z:v:b:",
95 long_options, &option_index);
96 if (opt == EOF)
97 break;
99 optcnt++;
100 switch(opt) {
101 case 'h':
102 if ((matches = sscanf(optarg, DS_NAM_FMT ":%ld",ds_nam,&heartbeat)) != 2){
103 rrd_set_error("invalid arguments for heartbeat");
104 rrd_free(&rrd);
105 close(rrd_file->fd);
106 return -1;
107 }
108 if ((ds=ds_match(&rrd,ds_nam))==-1){
109 rrd_free(&rrd);
110 close(rrd_file->fd);
111 return -1;
112 }
113 rrd.ds_def[ds].par[DS_mrhb_cnt].u_cnt = heartbeat;
114 break;
116 case 'i':
117 if ((matches = sscanf(optarg,DS_NAM_FMT ":%lf",ds_nam,&min)) <1){
118 rrd_set_error("invalid arguments for minimum ds value");
119 rrd_free(&rrd);
120 close(rrd_file->fd);
121 return -1;
122 }
123 if ((ds=ds_match(&rrd,ds_nam))==-1){
124 rrd_free(&rrd);
125 close(rrd_file->fd);
126 return -1;
127 }
129 if(matches == 1)
130 min= DNAN;
131 rrd.ds_def[ds].par[DS_min_val].u_val = min;
132 break;
134 case 'a':
135 if ((matches = sscanf(optarg, DS_NAM_FMT ":%lf",ds_nam,&max)) <1){
136 rrd_set_error("invalid arguments for maximum ds value");
137 rrd_free(&rrd);
138 close(rrd_file->fd);
139 return -1;
140 }
141 if ((ds=ds_match(&rrd,ds_nam))==-1){
142 rrd_free(&rrd);
143 close(rrd_file->fd);
144 return -1;
145 }
146 if(matches == 1)
147 max= DNAN;
148 rrd.ds_def[ds].par[DS_max_val].u_val = max;
149 break;
151 case 'd':
152 if ((matches = sscanf(optarg, DS_NAM_FMT ":" DST_FMT ,ds_nam,dst)) != 2){
153 rrd_set_error("invalid arguments for data source type");
154 rrd_free(&rrd);
155 close(rrd_file->fd);
156 return -1;
157 }
158 if ((ds=ds_match(&rrd,ds_nam))==-1){
159 rrd_free(&rrd);
160 close(rrd_file->fd);
161 return -1;
162 }
163 if ((int)dst_conv(dst) == -1){
164 rrd_free(&rrd);
165 close(rrd_file->fd);
166 return -1;
167 }
168 strncpy(rrd.ds_def[ds].dst,dst,DST_SIZE-1);
169 rrd.ds_def[ds].dst[DST_SIZE-1]='\0';
171 rrd.pdp_prep[ds].last_ds[0] = 'U';
172 rrd.pdp_prep[ds].last_ds[1] = 'N';
173 rrd.pdp_prep[ds].last_ds[2] = 'K';
174 rrd.pdp_prep[ds].last_ds[3] = 'N';
175 rrd.pdp_prep[ds].last_ds[4] = '\0';
177 break;
178 case 'r':
179 if ((matches =
180 sscanf(optarg,DS_NAM_FMT ":" DS_NAM_FMT , ds_nam,ds_new)) != 2){
181 rrd_set_error("invalid arguments for data source type");
182 rrd_free(&rrd);
183 close(rrd_file->fd);
184 return -1;
185 }
186 if ((ds=ds_match(&rrd,ds_nam))==-1){
187 rrd_free(&rrd);
188 close(rrd_file->fd);
189 return -1;
190 }
191 strncpy(rrd.ds_def[ds].ds_nam,ds_new,DS_NAM_SIZE-1);
192 rrd.ds_def[ds].ds_nam[DS_NAM_SIZE-1]='\0';
193 break;
194 case 'p':
195 if (set_deltaarg(&rrd,RRA_delta_pos,optarg)) {
196 rrd_free(&rrd);
197 return -1;
198 }
199 break;
200 case 'n':
201 if (set_deltaarg(&rrd,RRA_delta_neg,optarg)) {
202 rrd_free(&rrd);
203 return -1;
204 }
205 break;
206 case 'f':
207 if (set_windowarg(&rrd,RRA_failure_threshold,optarg)) {
208 rrd_free(&rrd);
209 return -1;
210 }
211 break;
212 case 'w':
213 if (set_windowarg(&rrd,RRA_window_len,optarg)) {
214 rrd_free(&rrd);
215 return -1;
216 }
217 break;
218 case 'x':
219 if (set_hwarg(&rrd,CF_HWPREDICT,RRA_hw_alpha,optarg)) {
220 rrd_free(&rrd);
221 return -1;
222 }
223 break;
224 case 'y':
225 if (set_hwarg(&rrd,CF_HWPREDICT,RRA_hw_beta,optarg)) {
226 rrd_free(&rrd);
227 return -1;
228 }
229 break;
230 case 'z':
231 if (set_hwarg(&rrd,CF_SEASONAL,RRA_seasonal_gamma,optarg)) {
232 rrd_free(&rrd);
233 return -1;
234 }
235 break;
236 case 'v':
237 if (set_hwarg(&rrd,CF_DEVSEASONAL,RRA_seasonal_gamma,optarg)) {
238 rrd_free(&rrd);
239 return -1;
240 }
241 break;
242 case 'b':
243 if (sscanf(optarg,DS_NAM_FMT,ds_nam) != 1){
244 rrd_set_error("invalid argument for aberrant-reset");
245 rrd_free(&rrd);
246 close(rrd_file->fd);
247 return -1;
248 }
249 if ((ds=ds_match(&rrd,ds_nam))==-1){
250 /* ds_match handles it own errors */
251 rrd_free(&rrd);
252 close(rrd_file->fd);
253 return -1;
254 }
255 reset_aberrant_coefficients(&rrd,rrd_file,(unsigned long) ds);
256 if (rrd_test_error()) {
257 rrd_free(&rrd);
258 close(rrd_file->fd);
259 return -1;
260 }
261 break;
262 case '?':
263 if (optopt != 0)
264 rrd_set_error("unknown option '%c'", optopt);
265 else
266 rrd_set_error("unknown option '%s'",argv[optind-1]);
267 rrd_free(&rrd);
268 close(rrd_file->fd);
269 return -1;
270 }
271 }
272 if(optcnt>0){
274 rrd_seek(rrd_file,0,SEEK_SET);
275 rrd_write(rrd_file,rrd.stat_head, sizeof(stat_head_t)*1);
276 rrd_write(rrd_file,rrd.ds_def,
277 sizeof(ds_def_t)* rrd.stat_head->ds_cnt);
278 /* need to write rra_defs for RRA parameter changes */
279 rrd_write(rrd_file,rrd.rra_def,
280 sizeof(rra_def_t)*rrd.stat_head->rra_cnt);
281 } else {
282 int i;
283 for(i=0;i< (int)rrd.stat_head->ds_cnt;i++)
284 if (dst_conv(rrd.ds_def[i].dst) != DST_CDEF) {
285 printf("DS[%s] typ: %s\thbt: %ld\tmin: %1.4f\tmax: %1.4f\n",
286 rrd.ds_def[i].ds_nam,
287 rrd.ds_def[i].dst,
288 rrd.ds_def[i].par[DS_mrhb_cnt].u_cnt,
289 rrd.ds_def[i].par[DS_min_val].u_val,
290 rrd.ds_def[i].par[DS_max_val].u_val);
291 } else {
292 char *buffer = NULL;
293 rpn_compact2str((rpn_cdefds_t *) &(rrd.ds_def[i].par[DS_cdef]),rrd.ds_def,&buffer);
294 printf("DS[%s] typ: %s\tcdef: %s\n", rrd.ds_def[i].ds_nam,rrd.ds_def[i].dst,buffer);
295 free(buffer);
296 }
297 }
298 close(rrd_file->fd);
299 rrd_free(&rrd);
300 return 0;
301 }
303 int set_hwarg(rrd_t *rrd,enum cf_en cf,enum rra_par_en rra_par,char *arg)
304 {
305 double param;
306 unsigned long i;
307 signed short rra_idx = -1;
308 /* read the value */
309 param = atof(arg);
310 if (param <= 0.0 || param >= 1.0)
311 {
312 rrd_set_error("Holt-Winters parameter must be between 0 and 1");
313 return -1;
314 }
315 /* does the appropriate RRA exist? */
316 for (i = 0; i < rrd -> stat_head -> rra_cnt; ++i)
317 {
318 if (cf_conv(rrd -> rra_def[i].cf_nam) == cf)
319 {
320 rra_idx = i;
321 break;
322 }
323 }
324 if (rra_idx == -1)
325 {
326 rrd_set_error("Holt-Winters RRA does not exist in this RRD");
327 return -1;
328 }
330 /* set the value */
331 rrd -> rra_def[rra_idx].par[rra_par].u_val = param;
332 return 0;
333 }
335 int set_deltaarg(rrd_t *rrd,enum rra_par_en rra_par,char *arg)
336 {
337 rrd_value_t param;
338 unsigned long i;
339 signed short rra_idx = -1;
341 param = atof(arg);
342 if (param < 0.1)
343 {
344 rrd_set_error("Parameter specified is too small");
345 return -1;
346 }
347 /* does the appropriate RRA exist? */
348 for (i = 0; i < rrd -> stat_head -> rra_cnt; ++i)
349 {
350 if (cf_conv(rrd -> rra_def[i].cf_nam) == CF_FAILURES)
351 {
352 rra_idx = i;
353 break;
354 }
355 }
356 if (rra_idx == -1)
357 {
358 rrd_set_error("Failures RRA does not exist in this RRD");
359 return -1;
360 }
362 /* set the value */
363 rrd -> rra_def[rra_idx].par[rra_par].u_val = param;
364 return 0;
365 }
367 int set_windowarg(rrd_t *rrd,enum rra_par_en rra_par,char *arg)
368 {
369 unsigned long param;
370 unsigned long i, cdp_idx;
371 signed short rra_idx = -1;
372 /* read the value */
373 param = atoi(arg);
374 if (param < 1 || param > MAX_FAILURES_WINDOW_LEN)
375 {
376 rrd_set_error("Parameter must be between %d and %d",
377 1, MAX_FAILURES_WINDOW_LEN);
378 return -1;
379 }
380 /* does the appropriate RRA exist? */
381 for (i = 0; i < rrd -> stat_head -> rra_cnt; ++i)
382 {
383 if (cf_conv(rrd -> rra_def[i].cf_nam) == CF_FAILURES)
384 {
385 rra_idx = i;
386 break;
387 }
388 }
389 if (rra_idx == -1)
390 {
391 rrd_set_error("Failures RRA does not exist in this RRD");
392 return -1;
393 }
395 /* set the value */
396 rrd -> rra_def[rra_idx].par[rra_par].u_cnt = param;
398 /* erase existing violations */
399 for (i = 0; i < rrd -> stat_head -> ds_cnt; i++)
400 {
401 cdp_idx = rra_idx * (rrd -> stat_head -> ds_cnt) + i;
402 erase_violations(rrd,cdp_idx,rra_idx);
403 }
404 return 0;
405 }