src/libstatgrab/tools.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2003 i-scream
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <regex.h>
#ifdef ALLBSD
#include <fcntl.h>
#include <kvm.h>
#endif
#ifdef NETBSD
#include <uvm/uvm_extern.h>
#endif

#include "tools.h"

char *f_read_line(FILE *f, const char *string){
        /* Max line length. 8k should be more than enough */
        static char line[8192];

        while((fgets(line, sizeof(line), f))!=NULL){
                if(strncmp(string, line, strlen(string))==0){
                        return line;
                }
        }

        return NULL;
}

char *get_string_match(char *line, regmatch_t *match){
        int len=match->rm_eo - match->rm_so;
        char *match_string=malloc(len+1);

        match_string=strncpy(match_string, line+match->rm_so, len);
        match_string[len]='\0';

        return match_string;
}

#ifdef HAVE_ATOLL
long long get_ll_match(char *line, regmatch_t *match){
        char *ptr;
        long long num;

        ptr=line+match->rm_so;
        num=atoll(ptr);

        return num;
}
#endif

#ifdef ALLBSD
kvm_t *get_kvm() {
        static kvm_t *kvmd = NULL;

        if (kvmd != NULL) {
                return kvmd;
        }

        kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
        return kvmd;
}
#endif

#ifdef NETBSD
struct uvmexp *get_uvmexp() {
        static u_long addr = 0;
        static struct uvmexp uvm;
        kvm_t *kvmd = get_kvm();

        if (kvmd == NULL) {
                return NULL;
        }

        if (addr == 0) {
                struct nlist symbols[] = {
                        { "_uvmexp" },
                        { NULL }
                };

                if (kvm_nlist(kvmd, symbols) != 0) {
                        return NULL;
                }
                addr = symbols[0].n_value;
        }

        if (kvm_read(kvmd, addr, &uvm, sizeof uvm) != sizeof uvm) {
                return NULL;
        }
        return &uvm;
}
#endif

int statgrab_init(){

#ifdef ALLBSD
        kvm_t *kvmd;
#endif
#ifdef NETBSD
        struct uvmexp *uvm;
#endif

#ifdef ALLBSD
        kvmd = get_kvm();
        if (kvmd == NULL) return 1;
#endif
#ifdef NETBSD
        uvm = get_uvmexp();
        if (uvm == NULL) return 1;
#endif
        return 0;
}
Revision:	1.10
Committed:	Mon Oct 20 15:25:47 2003 UTC (20 years, 6 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.9:	+19 -0 lines
Log Message:	Added statgrab_init() This function just calls the get_kvm type functions so all special privilages can be dropped after this function has been called. It currently only does stuff on *bsd (and needs testing on netbsd). All other OS's will just have a function that returns 0 Solaris at somepoint may need to do something here so we can do the drive mappings.
#	User	Rev	Content
1	tdb	1.5	/*
2			* i-scream central monitoring system
3	tdb	1.6	* http://www.i-scream.org
4			* Copyright (C) 2000-2003 i-scream
5	tdb	1.5	*
6			* This program is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU General Public License
8			* as published by the Free Software Foundation; either version 2
9			* of the License, or (at your option) any later version.
10			*
11			* This program is distributed in the hope that it will be useful,
12			* but WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU General Public License for more details.
15			*
16			* You should have received a copy of the GNU General Public License
17			* along with this program; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19			*/
20
21			#ifdef HAVE_CONFIG_H
22			#include "config.h"
23			#endif
24
25	pajs	1.1	#include <stdio.h>
26			#include <string.h>
27	pajs	1.3	#include <stdlib.h>
28			#include <sys/types.h>
29			#include <regex.h>
30	ats	1.9	#ifdef ALLBSD
31	ats	1.8	#include <fcntl.h>
32			#include <kvm.h>
33			#endif
34	ats	1.9	#ifdef NETBSD
35			#include <uvm/uvm_extern.h>
36			#endif
37	pajs	1.1
38	ats	1.7	#include "tools.h"
39
40	pajs	1.1	char f_read_line(FILE f, const char *string){
41			/* Max line length. 8k should be more than enough */
42			static char line[8192];
43
44	pajs	1.2	while((fgets(line, sizeof(line), f))!=NULL){
45	pajs	1.1	if(strncmp(string, line, strlen(string))==0){
46			return line;
47			}
48			}
49
50			return NULL;
51	pajs	1.3	}
52
53			char get_string_match(char line, regmatch_t *match){
54			int len=match->rm_eo - match->rm_so;
55			char *match_string=malloc(len+1);
56
57			match_string=strncpy(match_string, line+match->rm_so, len);
58			match_string[len]='\0';
59
60			return match_string;
61			}
62	ats	1.7
63	pajs	1.4	#ifdef HAVE_ATOLL
64	pajs	1.3	long long get_ll_match(char line, regmatch_t match){
65			char *ptr;
66			long long num;
67
68			ptr=line+match->rm_so;
69			num=atoll(ptr);
70
71			return num;
72	ats	1.8	}
73			#endif
74
75	ats	1.9	#ifdef ALLBSD
76	ats	1.8	kvm_t *get_kvm() {
77			static kvm_t *kvmd = NULL;
78
79			if (kvmd != NULL) {
80			return kvmd;
81			}
82
83			kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
84			return kvmd;
85	pajs	1.1	}
86	pajs	1.4	#endif
87	ats	1.9
88			#ifdef NETBSD
89			struct uvmexp *get_uvmexp() {
90			static u_long addr = 0;
91			static struct uvmexp uvm;
92			kvm_t *kvmd = get_kvm();
93
94			if (kvmd == NULL) {
95			return NULL;
96			}
97
98			if (addr == 0) {
99			struct nlist symbols[] = {
100			{ "_uvmexp" },
101			{ NULL }
102			};
103
104			if (kvm_nlist(kvmd, symbols) != 0) {
105			return NULL;
106			}
107			addr = symbols[0].n_value;
108			}
109
110			if (kvm_read(kvmd, addr, &uvm, sizeof uvm) != sizeof uvm) {
111			return NULL;
112			}
113			return &uvm;
114			}
115			#endif
116
117	pajs	1.10	int statgrab_init(){
118
119			#ifdef ALLBSD
120			kvm_t *kvmd;
121			#endif
122			#ifdef NETBSD
123			struct uvmexp *uvm;
124			#endif
125
126			#ifdef ALLBSD
127			kvmd = get_kvm();
128			if (kvmd == NULL) return 1;
129			#endif
130			#ifdef NETBSD
131			uvm = get_uvmexp();
132			if (uvm == NULL) return 1;
133			#endif
134			return 0;
135			}