ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/user_stats.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/user_stats.c (file contents):
Revision 1.2 by tdb, Thu Feb 20 13:19:52 2003 UTC vs.
Revision 1.15 by tdb, Mon Jan 19 16:49:21 2004 UTC

# Line 1 | Line 1
1 < /*
1 > /*
2   * i-scream central monitoring system
3 < * http://www.i-scream.org.uk
4 < * Copyright (C) 2000-2002 i-scream
3 > * http://www.i-scream.org
4 > * Copyright (C) 2000-2004 i-scream
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.
6 > * This library is free software; you can redistribute it and/or
7 > * modify it under the terms of the GNU Lesser General Public
8 > * License as published by the Free Software Foundation; either
9 > * version 2.1 of the License, or (at your option) any later version.
10   *
11 < * This program is distributed in the hope that it will be useful,
11 > * This library 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.
13 > * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 > * Lesser 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.
16 > * You should have received a copy of the GNU Lesser General Public
17 > * License along with this library; if not, write to the Free Software
18 > * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19 > * 02111-1307 USA
20 > *
21 > * $Id$
22   */
23  
24   #ifdef HAVE_CONFIG_H
25   #include "config.h"
26   #endif
27 < #include "statgrab.h"
27 >
28   #include <stdlib.h>
29   #include <stdio.h>
30   #include <string.h>
31 < #ifdef FREEBSD
31 > #include "statgrab.h"
32 > #ifdef ALLBSD
33   #include <sys/types.h>
34   #endif
35   #include <utmp.h>
36  
37 < #define START_VAL 5
37 > #ifdef SOLARIS
38 > #define MAX_LOGIN_NAME_SIZE 8
39 > #endif
40 > #if defined(LINUX) || defined(FREEBSD)
41 > #define MAX_LOGIN_NAME_SIZE UT_NAMESIZE
42 > #endif
43 > #ifdef NETBSD
44 > #define MAX_LOGIN_NAME_SIZE _POSIX_LOGIN_NAME_MAX
45 > #endif
46 > #if defined(CYGWIN)
47 > #define MAX_LOGIN_NAME_SIZE _SC_LOGIN_NAME_MAX
48 > #endif
49  
50 + #define START_VAL (5*(1+MAX_LOGIN_NAME_SIZE))
51 +
52   user_stat_t *get_user_stats(){
53          int num_users=0;
54  
55 <        static user_stat_t user_stat;
56 <        static int watermark=-1;
55 >        static user_stat_t user_stats;
56 >        static int size_of_namelist=-1;
57 >        char *tmp;
58 > #if defined(SOLARIS) || defined(LINUX) || defined(CYGWIN)
59          struct utmp *entry;
60 + #endif
61 + #ifdef ALLBSD
62 +        struct utmp entry;
63 +        FILE *f;
64 + #endif
65  
42        name *name_ptr;
43
66          /* First case call */
67 <        if (watermark==-1){
68 <                user_stat.name_list=malloc(START_VAL * sizeof *user_stat.name_list);
69 <                if(user_stat.name_list==NULL){
67 >        if (size_of_namelist==-1){
68 >                user_stats.name_list=malloc(START_VAL);
69 >                if(user_stats.name_list==NULL){
70                          return NULL;
71                  }
72 <                watermark=START_VAL;
72 >                size_of_namelist=START_VAL;
73          }      
74  
75 +        /* Essentially blank the list, or give it a inital starting string */
76 +        strcpy(user_stats.name_list, "");
77 +
78 + #if defined(SOLARIS) || defined(LINUX) || defined(CYGWIN)
79          setutent();
80          while((entry=getutent()) != NULL) {
81                  if(entry->ut_type==USER_PROCESS) {
82 <                        if(num_users>watermark-1){
83 <                                name_ptr=user_stat.name_list;
84 <                                if((user_stat.name_list=realloc(user_stat.name_list, (watermark*2* sizeof *user_stat.name_list)))==NULL){
85 <                                        user_stat.name_list=name_ptr;
82 >                        if((strlen(user_stats.name_list)+MAX_LOGIN_NAME_SIZE+2) > size_of_namelist){
83 >                                tmp=user_stats.name_list;
84 >                                user_stats.name_list=realloc(user_stats.name_list, 1+(size_of_namelist*2));
85 >                                if(user_stats.name_list==NULL){
86 >                                        user_stats.name_list=tmp;
87                                          return NULL;
88                                  }
89 <                                watermark=watermark*2;
89 >                                size_of_namelist=1+(size_of_namelist*2);
90                          }
91  
92 <                        strncpy(user_stat.name_list[num_users], entry->ut_user, MAX_LOGIN_NAME_SIZE);
93 <                        /* NULL terminate just in case , the size of user_stat.name_list is MAX_LOGIN_NAME_SIZE+1 */
67 <
68 <                        user_stat.name_list[num_users][MAX_LOGIN_NAME_SIZE]='\0';
69 <                                
92 >                        strncat(user_stats.name_list, entry->ut_user, MAX_LOGIN_NAME_SIZE);
93 >                        strcat(user_stats.name_list, " ");
94                          num_users++;
95                  }
96          }
97          endutent();
98 <        user_stat.num_entries=num_users;
98 > #endif
99 > #ifdef ALLBSD
100 >        if ((f=fopen(_PATH_UTMP, "r")) == NULL){
101 >                return NULL;
102 >        }
103 >        while((fread(&entry, sizeof(entry),1,f)) != 0){
104 >                if (entry.ut_name[0] == '\0') continue;
105 >                if((strlen(user_stats.name_list)+MAX_LOGIN_NAME_SIZE+2) > size_of_namelist){
106 >                        tmp=user_stats.name_list;
107 >                        user_stats.name_list=realloc(user_stats.name_list, 1+(size_of_namelist*2));
108 >                        if(user_stats.name_list==NULL){
109 >                                user_stats.name_list=tmp;
110 >                                return NULL;
111 >                        }
112 >                        size_of_namelist=1+(size_of_namelist*2);
113 >                        
114 >                }
115 >                strncat(user_stats.name_list, entry.ut_name, MAX_LOGIN_NAME_SIZE);
116 >                strcat(user_stats.name_list, " ");
117 >                num_users++;
118 >        }
119 >        fclose(f);
120  
121 <        return &user_stat;
121 > #endif  
122 >        /* We want to remove the last " " */
123 >        if(num_users!=0){
124 >                tmp=strrchr(user_stats.name_list, ' ');
125 >                if(tmp!=NULL){
126 >                        *tmp='\0';
127 >                        user_stats.num_entries=num_users;
128 >                }
129 >        }
130 >
131 >        return &user_stats;
132  
133   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines