8623f94f80
PR: 29775 Submitted by: pat@databits.net Approved by: maintainer
474 lines
14 KiB
Text
474 lines
14 KiB
Text
--- wminet.c.orig Fri Dec 18 06:07:27 1998
|
|
+++ wminet.c Thu Aug 16 11:08:44 2001
|
|
@@ -10,6 +10,8 @@
|
|
|
|
ProFTPD support by Mike Kershaw aka Dragorn (dragorn@melchior.nerv-un.ml.org)
|
|
ProFTPD support was made 64bit clean by Martijn Pieterse (pieterse@xs4all.nl)
|
|
+
|
|
+ FreeBSD port by Stephen Kiernan (sk-ports@vegamuse.org)
|
|
|
|
see http://windowmaker.mezaway.org for more awesome wm dock apps :)
|
|
|
|
@@ -23,6 +25,11 @@
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <ctype.h>
|
|
+#ifdef __FreeBSD__
|
|
+#include <limits.h>
|
|
+#include <kvm.h>
|
|
+#include <netdb.h>
|
|
+#endif /* __FreeBSD__ */
|
|
|
|
#include <sys/wait.h>
|
|
#include <sys/stat.h>
|
|
@@ -30,6 +37,23 @@
|
|
#include <sys/types.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
+#ifdef __FreeBSD__
|
|
+#include <sys/socketvar.h>
|
|
+#include <sys/sysctl.h>
|
|
+
|
|
+#include <net/route.h>
|
|
+#include <netinet/in.h>
|
|
+#include <netinet/in_systm.h>
|
|
+#include <netinet/ip.h>
|
|
+#include <netinet/in_pcb.h>
|
|
+#include <netinet/ip_var.h>
|
|
+#include <netinet/tcp.h>
|
|
+#include <netinet/tcp_fsm.h>
|
|
+#include <netinet/tcp_timer.h>
|
|
+#include <netinet/tcp_var.h>
|
|
+
|
|
+#include <arpa/inet.h>
|
|
+#endif /* __FreeBSD__ */
|
|
|
|
#include <utmp.h>
|
|
#include <dirent.h>
|
|
@@ -57,6 +81,10 @@
|
|
// Lame work-around... Sigh... when will they standardize the headers!?!?
|
|
#define TCP_ESTABLISHED 1
|
|
|
|
+#ifndef DEFAULT_WMINETRC
|
|
+#define DEFAULT_WMINETRC "/etc/wminetrc"
|
|
+#endif /* DEFAULT_WMINETRC */
|
|
+
|
|
extern char **environ;
|
|
|
|
char *ProgName;
|
|
@@ -101,6 +129,13 @@
|
|
|
|
char uconfig_file[256];
|
|
|
|
+#ifdef __FreeBSD__
|
|
+struct utmp *_wminet_getutent(FILE *);
|
|
+#define setutent() { FILE *_wminet__ufp = fopen(_PATH_UTMP, "r")
|
|
+#define getutent() _wminet_getutent(_wminet__ufp)
|
|
+#define endutent() fclose(_wminet__ufp); }
|
|
+#endif /* __FreeBSD__ */
|
|
+
|
|
void usage(void);
|
|
void printversion(void);
|
|
void BlitString(char *name, int x, int y);
|
|
@@ -109,10 +144,22 @@
|
|
void wminet_routine(int, char **);
|
|
int PortWatch( short port );
|
|
int ReadConfigInt(FILE *fp, char *setting, int *value);
|
|
-int ReadConfigString(FILE *fp, char *setting, char *value);
|
|
+int ReadConfigString(FILE *fp, char *setting, char *value, int len);
|
|
int Read_Config_File( char *filename );
|
|
|
|
|
|
+#ifdef __FreeBSD__
|
|
+inline
|
|
+struct utmp *_wminet_getutent( FILE *fp )
|
|
+{
|
|
+ static struct utmp usr;
|
|
+ if( fread((char *)&usr, sizeof(usr), 1, fp) == 1 )
|
|
+ return &usr;
|
|
+ else
|
|
+ return NULL;
|
|
+}
|
|
+#endif /* __FreeBSD__ */
|
|
+
|
|
int main(int argc, char *argv[]) {
|
|
|
|
int i;
|
|
@@ -149,7 +196,7 @@
|
|
case 'c' :
|
|
if (argc > (i+1))
|
|
{
|
|
- strcpy(uconfig_file, argv[i+1]);
|
|
+ strncpy(uconfig_file, argv[i+1], 256);
|
|
i++;
|
|
}
|
|
break;
|
|
@@ -214,12 +261,12 @@
|
|
}
|
|
else
|
|
{
|
|
- sprintf(config_file, "%s/.wminetrc", getenv("HOME"));
|
|
+ snprintf(config_file, 256, "%s/.wminetrc", getenv("HOME"));
|
|
|
|
if (!Read_Config_File(config_file))
|
|
{
|
|
// Fall back to /etc/wminetrc
|
|
- sprintf(config_file, "/etc/wminetrc");
|
|
+ snprintf(config_file, 256, DEFAULT_WMINETRC);
|
|
|
|
Read_Config_File(config_file);
|
|
}
|
|
@@ -240,13 +287,13 @@
|
|
if (use_proftpd)
|
|
{
|
|
if (strstr(dent->d_name, "proftpd-") != NULL)
|
|
- strcpy(ftpclasses[0], dent->d_name);
|
|
+ strncpy(ftpclasses[0], dent->d_name, 64);
|
|
}
|
|
else
|
|
{
|
|
if (strstr(dent->d_name, "ftp.pids-") != NULL)
|
|
{
|
|
- strcpy(ftpclasses[numftpclasses++], dent->d_name);
|
|
+ strncpy(ftpclasses[numftpclasses++], dent->d_name, 64);
|
|
//printf("ftppidfile: %s\n", dent->d_name);
|
|
}
|
|
}
|
|
@@ -422,7 +469,6 @@
|
|
char seps[]={"/"};
|
|
char sep2[]={":"};
|
|
char sep3[]={" "};
|
|
-
|
|
#ifdef HTTP_MONITOR_PROC
|
|
DIR *dir;
|
|
struct dirent *dent;
|
|
@@ -439,8 +485,13 @@
|
|
setutent();
|
|
while ((ut = getutent()))
|
|
{
|
|
- if ((ut->ut_type == USER_PROCESS) &&
|
|
- (ut->ut_name[0] != '\0'))
|
|
+ if (
|
|
+#ifdef __FreeBSD__
|
|
+ (ut->ut_line[0] != '\0')
|
|
+#else
|
|
+ (ut->ut_type == USER_PROCESS)
|
|
+#endif /* __FreeBSD__ */
|
|
+ && (ut->ut_name[0] != '\0'))
|
|
{
|
|
nUsers++;
|
|
}
|
|
@@ -459,7 +510,7 @@
|
|
logrun_t runent;
|
|
logrun_header_t head;
|
|
|
|
- sprintf(buf, "%s/%s", ftp_pid_path, ftpclasses[0]);
|
|
+ snprintf(buf, 1024, "%s/%s", ftp_pid_path, ftpclasses[0]);
|
|
|
|
if (( fd = open(buf, O_RDONLY, 0644)) == -1)
|
|
{
|
|
@@ -486,7 +537,7 @@
|
|
} else {
|
|
for (i=0; i!= numftpclasses; i++)
|
|
{
|
|
- sprintf(buf, "%s/%s", ftp_pid_path, ftpclasses[i]);
|
|
+ snprintf(buf, 1024, "%s/%s", ftp_pid_path, ftpclasses[i]);
|
|
//printf("opening '%s'\n", buf);
|
|
fp = fopen(buf, "r");
|
|
if (fp)
|
|
@@ -508,11 +559,10 @@
|
|
// httpd processes
|
|
nHttp = 0;
|
|
|
|
-#ifdef HTTP_MONITOR_PROC
|
|
+#if defined(HTTP_MONITOR_PROC) && !defined(__FreeBSD__)
|
|
|
|
if ( monitor_http )
|
|
{
|
|
-
|
|
dir = opendir("/proc");
|
|
if (dir)
|
|
{
|
|
@@ -520,7 +570,7 @@
|
|
{
|
|
if (!isalpha(dent->d_name[0]))
|
|
{
|
|
- sprintf(buf, "/proc/%s/stat", dent->d_name);
|
|
+ snprintf(buf, 1024, "/proc/%s/stat", dent->d_name);
|
|
//printf("opening '%s'\n", buf);
|
|
fp=fopen(buf, "r");
|
|
if (fp)
|
|
@@ -544,41 +594,7 @@
|
|
#ifdef HTTP_MONITOR_NET
|
|
|
|
if ( monitor_http )
|
|
- {
|
|
-
|
|
- fp = fopen("/proc/net/tcp", "r");
|
|
- if (fp)
|
|
- {
|
|
- fgets(buf, 512, fp); // get rid of text header
|
|
-
|
|
- while ( (fgets(buf, 512, fp)) )
|
|
- {
|
|
- tok = strtok(buf, sep2);
|
|
- tok = strtok(NULL, sep2);
|
|
- tok = strtok(NULL, sep2);
|
|
-
|
|
- tok[4]=0;
|
|
- tok1 = strtok(NULL, sep2);
|
|
- tok1 += 5;
|
|
- tok1[2] = 0;
|
|
-
|
|
- // printf("port: %i\n", strtol(tok, NULL, 16));
|
|
- // printf("state: %i\n", strtol(tok1, NULL, 16));
|
|
-
|
|
- i = strtol(tok, NULL, 16);
|
|
- j = strtol(tok1, NULL, 16);
|
|
-
|
|
- // should make this configurable
|
|
- if (( i == 80 || i == 8080) && (j == TCP_ESTABLISHED))
|
|
- {
|
|
- nHttp++;
|
|
- }
|
|
-
|
|
- }
|
|
-
|
|
- fclose(fp);
|
|
- }
|
|
- }
|
|
+ nHttp = PortWatch( 80 ) + PortWatch( 8080 );
|
|
#endif
|
|
|
|
|
|
@@ -587,7 +603,11 @@
|
|
|
|
if ( monitor_nfs )
|
|
{
|
|
+#ifdef __FreeBSD__
|
|
+ fp = popen("/usr/bin/showmount -a", "r");
|
|
+#else
|
|
fp = popen("/usr/sbin/showmount -a", "r");
|
|
+#endif /* __FreeBSD__ */
|
|
if (fp)
|
|
{
|
|
while ( (fgets(buf, 128, fp)) )
|
|
@@ -604,9 +624,46 @@
|
|
// Total Processes
|
|
nProc = 0;
|
|
|
|
+#if defined(__FreeBSD__) && defined(HTTP_MONITOR_PROC)
|
|
+ if( monitor_proc || monitor_http )
|
|
+#else
|
|
if ( monitor_proc )
|
|
+#endif /* __FreeBSD__ && HTTP_MONITOR_PROC */
|
|
{
|
|
-
|
|
+#ifdef __FreeBSD__
|
|
+ char errbuf[_POSIX2_LINE_MAX];
|
|
+ kvm_t *kd = kvm_openfiles( NULL, NULL, NULL, O_RDONLY, errbuf );
|
|
+ if( kd == 0 )
|
|
+ fprintf( stderr, "%s", errbuf );
|
|
+ else
|
|
+ {
|
|
+ int procs;
|
|
+
|
|
+#ifdef HTTP_MONITOR_PROC
|
|
+ struct kinfo_proc *kinfo =
|
|
+#endif /* HTTP_MONITOR_PROC */
|
|
+ kvm_getprocs(kd,KERN_PROC_ALL,0,&procs);
|
|
+
|
|
+#ifdef HTTP_MONITOR_PROC
|
|
+ if( monitor_http )
|
|
+ {
|
|
+ int i;
|
|
+ for( i = 0; i < procs; i++ )
|
|
+ {
|
|
+ if( !strncmp( KI_PROC(&kinfo[i])->p_comm, "(httpd)", 7 ) )
|
|
+ {
|
|
+ nHttp++;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+#endif /* HTTP_MONITOR_PROC */
|
|
+
|
|
+ kvm_close( kd );
|
|
+
|
|
+ if( monitor_proc )
|
|
+ nProc = procs;
|
|
+ }
|
|
+#else
|
|
fp = fopen("/proc/loadavg", "r");
|
|
if (fp)
|
|
{
|
|
@@ -617,6 +674,7 @@
|
|
|
|
fclose(fp);
|
|
}
|
|
+#endif /* __FreeBSD__ */
|
|
}
|
|
|
|
// lpd
|
|
@@ -654,14 +712,61 @@
|
|
|
|
int PortWatch( short port )
|
|
{
|
|
+
|
|
+ int count=0;
|
|
+
|
|
+#ifdef __FreeBSD__
|
|
+ struct protoent *p;
|
|
+
|
|
+ setprotoent(1);
|
|
+ setservent(1);
|
|
+ while((p = getprotoent()))
|
|
+ {
|
|
+ if( !strcmp( p->p_name, "tcp" ) )
|
|
+ {
|
|
+ int len = 0;
|
|
+ if( sysctlbyname( "net.inet.tcp.pcblist", 0, &len, 0, 0 ) >= 0 )
|
|
+ {
|
|
+ char *buf = malloc(len);
|
|
+ if( buf )
|
|
+ {
|
|
+ if( sysctlbyname("net.inet.tcp.pcblist", buf, &len, 0, 0) >= 0 )
|
|
+ {
|
|
+ struct xinpgen *xig, *oxig;
|
|
+
|
|
+ oxig = xig = (struct xinpgen *)buf;
|
|
+ for( xig = (struct xinpgen *)((char *)xig + xig->xig_len);
|
|
+ xig->xig_len > sizeof(struct xinpgen);
|
|
+ xig = (struct xinpgen *)((char *)xig + xig->xig_len))
|
|
+ {
|
|
+ struct tcpcb *tp = &((struct xtcpcb *)xig)->xt_tp;
|
|
+ struct inpcb *inp = &((struct xtcpcb *)xig)->xt_inp;
|
|
+ struct xsocket *so = &((struct xtcpcb *)xig)->xt_socket;
|
|
+
|
|
+ if( ( so->xso_protocol != IPPROTO_TCP ) ||
|
|
+ ( inp->inp_gencnt > oxig->xig_gen ) ||
|
|
+ ( inet_lnaof(inp->inp_laddr) == INADDR_ANY ) )
|
|
+ continue;
|
|
+
|
|
+ if( ( ntohs((u_short)inp->inp_lport) == port ) &&
|
|
+ ( tp->t_state == TCPS_ESTABLISHED ) )
|
|
+ {
|
|
+ count++;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ free(buf);
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+#else
|
|
FILE *fp;
|
|
char buf[1024];
|
|
char *tok,*tok1;
|
|
int i,j;
|
|
char sep2[]={":"};
|
|
|
|
- int count=0;
|
|
-
|
|
fp = fopen("/proc/net/tcp", "r");
|
|
if (fp)
|
|
{
|
|
@@ -693,6 +798,7 @@
|
|
|
|
fclose(fp);
|
|
}
|
|
+#endif /* __FreeBSD__ */
|
|
|
|
return count;
|
|
}
|
|
@@ -745,14 +851,14 @@
|
|
newx -= CHAR_WIDTH;
|
|
}
|
|
|
|
- sprintf(buf, "%02i", num);
|
|
+ snprintf(buf, 1024, "%02i", num);
|
|
|
|
BlitString(buf, newx, y);
|
|
}
|
|
|
|
|
|
// ReadConfigSetting
|
|
-int ReadConfigString(FILE *fp, char *setting, char *value)
|
|
+int ReadConfigString(FILE *fp, char *setting, char *value, int vallen)
|
|
{
|
|
char str[1024];
|
|
char buf[1024];
|
|
@@ -767,7 +873,7 @@
|
|
return 0;
|
|
}
|
|
|
|
- sprintf(str, "%s=", setting);
|
|
+ snprintf(str, 1024, "%s=", setting);
|
|
slen = strlen(str);
|
|
|
|
fseek(fp, 0, SEEK_SET);
|
|
@@ -799,7 +905,7 @@
|
|
if ( buf[i] == '=' )
|
|
{
|
|
p=buf+i+1;
|
|
- strcpy(value, p);
|
|
+ strncpy(value, p, vallen);
|
|
return 1;
|
|
}
|
|
}
|
|
@@ -814,7 +920,7 @@
|
|
{
|
|
char buf[1024];
|
|
|
|
- if (ReadConfigString(fp, setting, (char *) &buf))
|
|
+ if (ReadConfigString(fp, setting, (char *) &buf, 1024))
|
|
{
|
|
*value = atoi(buf);
|
|
return 1;
|
|
@@ -831,11 +937,11 @@
|
|
if (fp)
|
|
{
|
|
ReadConfigInt(fp, "interval", &loopinterval);
|
|
- ReadConfigString(fp, "action1", action1);
|
|
- ReadConfigString(fp, "action2", action2);
|
|
- ReadConfigString(fp, "action3", action3);
|
|
- ReadConfigString(fp, "action4", action4);
|
|
- ReadConfigString(fp, "action5", action5);
|
|
+ ReadConfigString(fp, "action1", action1, 256);
|
|
+ ReadConfigString(fp, "action2", action2, 256);
|
|
+ ReadConfigString(fp, "action3", action3, 256);
|
|
+ ReadConfigString(fp, "action4", action4, 256);
|
|
+ ReadConfigString(fp, "action5", action5, 256);
|
|
ReadConfigInt(fp, "monitor_proc", &monitor_proc);
|
|
ReadConfigInt(fp, "monitor_users", &monitor_users);
|
|
ReadConfigInt(fp, "monitor_ftp", &monitor_ftp);
|
|
@@ -843,23 +949,23 @@
|
|
ReadConfigInt(fp, "monitor_nfs", &monitor_nfs);
|
|
ReadConfigInt(fp, "monitor_lpd", &monitor_lpd);
|
|
ReadConfigInt(fp, "use_proftpd", &use_proftpd);
|
|
- ReadConfigString(fp, "ftp_pid_path", ftp_pid_path);
|
|
+ ReadConfigString(fp, "ftp_pid_path", ftp_pid_path, 256);
|
|
|
|
ReadConfigInt(fp, "portwatch1.pos", &portwatch[1].pos);
|
|
ReadConfigInt(fp, "portwatch1.port", &portwatch[1].port);
|
|
- ReadConfigString(fp, "portwatch1.label", portwatch[1].label);
|
|
+ ReadConfigString(fp, "portwatch1.label", portwatch[1].label, 10);
|
|
ReadConfigInt(fp, "portwatch2.pos", &portwatch[2].pos);
|
|
ReadConfigInt(fp, "portwatch2.port", &portwatch[2].port);
|
|
- ReadConfigString(fp, "portwatch2.label", portwatch[2].label);
|
|
+ ReadConfigString(fp, "portwatch2.label", portwatch[2].label, 10);
|
|
ReadConfigInt(fp, "portwatch3.pos", &portwatch[3].pos);
|
|
ReadConfigInt(fp, "portwatch3.port", &portwatch[3].port);
|
|
- ReadConfigString(fp, "portwatch3.label", portwatch[3].label);
|
|
+ ReadConfigString(fp, "portwatch3.label", portwatch[3].label, 10);
|
|
ReadConfigInt(fp, "portwatch4.pos", &portwatch[4].pos);
|
|
ReadConfigInt(fp, "portwatch4.port", &portwatch[4].port);
|
|
- ReadConfigString(fp, "portwatch4.label", portwatch[4].label);
|
|
+ ReadConfigString(fp, "portwatch4.label", portwatch[4].label, 10);
|
|
ReadConfigInt(fp, "portwatch5.pos", &portwatch[5].pos);
|
|
ReadConfigInt(fp, "portwatch5.port", &portwatch[5].port);
|
|
- ReadConfigString(fp, "portwatch5.label", portwatch[5].label);
|
|
+ ReadConfigString(fp, "portwatch5.label", portwatch[5].label, 10);
|
|
|
|
fclose(fp);
|
|
return 1;
|