s3-libads: use ldap_init_fd() to initialize a ldap session if possible
[bbaumbach/samba-autobuild/.git] / examples / perfcounter / perf_writer_util.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  Performance Counter Daemon
4  *
5  *  Copyright (C) Marcin Krzysztof Porwit    2005
6  *  
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 3 of the License, or
10  *  (at your option) any later version.
11  *  
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *  
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include "perf.h"
22
23 extern sig_atomic_t keep_running;
24
25 void fatal(char *msg)
26 {
27   perror(msg);
28   exit(1);
29 }
30
31 void add_key_raw(TDB_CONTEXT *db, char *keystring, void *databuf, size_t datasize, int flags)
32 {
33   TDB_DATA key, data;
34
35   key.dptr = keystring;
36   key.dsize = strlen(keystring);
37   data.dptr = databuf;
38   data.dsize = datasize;
39
40   tdb_store(db, key, data, flags);
41 }
42
43 void add_key(TDB_CONTEXT *db, char *keystring, char *datastring, int flags)
44 {
45   TDB_DATA key, data;
46
47   key.dptr = keystring;
48   key.dsize = strlen(keystring);
49   data.dptr = datastring;
50   data.dsize = strlen(datastring);
51
52   tdb_store(db, key, data, flags);
53 }
54
55 void make_key(char *buf, int buflen, int key_part1, char *key_part2)
56 {
57     memset(buf, 0, buflen);
58     if(key_part2 != NULL)
59         sprintf(buf, "%d%s", key_part1, key_part2);
60     else
61         sprintf(buf, "%d", key_part1);
62
63     return;
64 }
65
66 void usage(char *progname)
67 {
68     fprintf(stderr, "Usage: %s [-d] [-f <file_path>].\n", progname);
69     fprintf(stderr, "\t-d: run as a daemon.\n");
70     fprintf(stderr, "\t-f <file_path>: path where the TDB files reside.\n");
71     fprintf(stderr, "\t\tDEFAULT is /var/lib/samba/perfmon\n");
72     exit(1);
73 }
74
75 void parse_flags(RuntimeSettings *rt, int argc, char **argv)
76 {
77     int flag;
78     
79     while((flag = getopt(argc, argv, "df:")) != -1)
80     {
81         switch(flag)
82         {
83             case 'd':
84             {
85                 rt->dflag = TRUE;
86                 break;
87             }
88             case 'f':
89             {
90                 memcpy(rt->dbDir, optarg, strlen(optarg));
91                 break;
92             }
93             default:
94             {
95                 usage(argv[0]);
96             }
97         }
98     }
99
100     return;
101 }
102
103 void setup_file_paths(RuntimeSettings *rt)
104 {
105     int status;
106
107     if(strlen(rt->dbDir) == 0)
108     {
109         /* No file path was passed in, use default */
110         sprintf(rt->dbDir, "/var/lib/samba/perfmon");
111     }
112
113     sprintf(rt->nameFile, "%s/names.tdb", rt->dbDir);
114     sprintf(rt->counterFile, "%s/data.tdb", rt->dbDir);
115
116     mkdir(rt->dbDir, 0755);
117     rt->cnames = tdb_open(rt->nameFile, 0, TDB_CLEAR_IF_FIRST, O_RDWR | O_CREAT, 0644);
118     rt->cdata = tdb_open(rt->counterFile, 0, TDB_CLEAR_IF_FIRST, O_RDWR | O_CREAT, 0644);
119
120     if(rt->cnames == NULL || rt->cdata == NULL)
121     {
122         perror("setup_file_paths");
123         exit(1);
124     }
125
126     return;
127 }
128
129 void sigterm_handler()
130 {
131     keep_running = FALSE;
132     return;
133 }
134
135 void daemonize(RuntimeSettings *rt)
136 {
137     pid_t pid;
138     int i;
139     int fd;
140
141     /* Check if we're already a daemon */
142     if(getppid() == 1)
143         return;
144     pid = fork();
145     if(pid < 0)
146         /* can't fork */
147         exit(1);
148     else if(pid > 0)
149     {
150         /* we're the parent */
151         tdb_close(rt->cnames);
152         tdb_close(rt->cdata);
153         exit(0);
154     }
155
156     /* get a new session */
157     if(setsid() == -1)
158         exit(2);
159
160     /* Change CWD */
161     chdir("/");
162
163     /* close file descriptors */
164     close(STDIN_FILENO);
165     close(STDOUT_FILENO);
166     close(STDERR_FILENO);
167
168     /* And reopen them as safe defaults */
169     fd = open("/dev/null", O_RDONLY);
170     if(fd != 0)
171     {
172         dup2(fd, 0);
173         close(fd);
174     }
175     fd = open("/dev/null", O_WRONLY);
176     if(fd != 1)
177     {
178         dup2(fd, 1);
179         close(fd);
180     }
181     fd = open("/dev/null", O_WRONLY);
182     if(fd != 2)
183     {
184         dup2(fd, 2);
185         close(fd);
186     }
187
188     /* handle signals */
189     signal(SIGINT, SIG_IGN);
190     signal(SIGHUP, SIG_IGN);
191     signal(SIGTERM, sigterm_handler);
192
193     return;
194 }
195
196 int get_counter_id(PERF_DATA_BLOCK *data)
197 {
198     data->counter_id += 2;
199     data->num_counters++;
200
201     return data->counter_id;
202 }
203
204 void init_perf_counter(PerfCounter *counter,
205                        PerfCounter *parent,
206                        unsigned int index,
207                        char *name,
208                        char *help,
209                        int counter_type,
210                        int record_type)
211 {
212     counter->index = index;
213     memcpy(counter->name, name, strlen(name));
214     memcpy(counter->help, help, strlen(help));
215     counter->counter_type = counter_type;
216     counter->record_type = record_type;
217
218     switch(record_type)
219     {
220         case PERF_OBJECT:
221             sprintf(counter->relationships, "p");
222             break;
223         case PERF_COUNTER:
224             sprintf(counter->relationships, "c[%d]", parent->index);
225             break;
226         case PERF_INSTANCE:
227             sprintf(counter->relationships, "i[%d]", parent->index);
228             break;
229         default:
230             perror("init_perf_counter: unknown record type");
231             exit(1);
232     }
233
234     return;
235 }