Add Linux support, code cleanups

[remotetop.git] / sendtop.c

diff --git a/sendtop.c b/sendtop.c
index 297eabc..4a48662 100644 (file)
--- a/sendtop.c
+++ b/sendtop.c
@@ -33,6 +33,7 @@
 #include <unistd.h>
 #include <signal.h>
 #include <errno.h>
+#include <ctype.h>
 #include <sys/sysctl.h>
 
 #include "tlog.h"
@@ -41,18 +42,18 @@
 
 #include "top.h"
 
-#define NITEM   0
+#ifdef FreeBSD
 
 static int
 getsysctl(char *name, void *ptr, size_t len) {
        size_t nlen = len;
        if (sysctlbyname(name, ptr, &nlen, NULL, 0) == -1) {
-               tlog(TL_ALARM, "sysctl(%s...) failed: %s\n", name,
+               tlog(TL_ALARM, "sysctl(\"%s\", ...) failed: %s\n", name,
                        strerror(errno));
                return 1;
        }   
        if (nlen != len) {
-               tlog(TL_ALARM, "sysctl(%s...) expected %lu, got %lu\n", name,
+               tlog(TL_ALARM, "sysctl(\"%s\", ...) expected %lu, got %lu\n", name,
                        (unsigned long)len, (unsigned long)nlen);
                return 1;
        }
@@ -63,25 +64,15 @@ getsysctl(char *name, void *ptr, size_t len) {
 #define GETSYSCTL(name, var) getsysctl(name, &(var), sizeof(var))
 
 static void
-fillMsgTop(TCMsgTop *m) {
-       double lll[3];
+getmeminfo(TCMsgTop *m) {
        int    memory_tmp;
+       long    memory_tmp1;
 
-       memset(m,0,sizeof(TCMsgTop));
-       if ( getloadavg(lll,3) >= NITEM+1 )
-               m->load=lll[NITEM];
-       else
-               m->load=-1.0;
-
-       m->freemem = 0;
        if ( GETSYSCTL("vm.stats.vm.v_inactive_count", memory_tmp) == 0 )
                m->freemem += memory_tmp;
        else
                m->freemem = -1;
-       if ( m->freemem>=0 && GETSYSCTL("vm.stats.vm.v_cache_count", memory_tmp) == 0 )
-               m->freemem += memory_tmp;
-       else
-               m->freemem = -1;
+
        if ( m->freemem>=0 && GETSYSCTL("vm.stats.vm.v_free_count", memory_tmp) == 0 )
                m->freemem += memory_tmp;
        else
@@ -90,11 +81,85 @@ fillMsgTop(TCMsgTop *m) {
        if (  m->freemem>=0 )
                m->freemem *= getpagesize();
         
-       m->usermem =-1;
-       if ( GETSYSCTL("hw.usermem", memory_tmp) == 0 )
-               m->usermem = memory_tmp;
+       if ( GETSYSCTL("hw.physmem", memory_tmp1) == 0 )
+               m->physmem = memory_tmp1;
+}
+
+
+#elif defined  Linux /* end FreeBSD */
+
+#define MEMBUFSIZE 256
+
+static char*
+cmpPattern( char *str, char *pattern ) {
+       while( *str != '\0' && *str == *pattern ) {
+               str++;
+               pattern++;
+       }
+
+       if ( *pattern == '\0' ) {
+               if ( *str == '\0' )
+                       return NULL;
+
+               /*
+                * str matches by pattern, so skip leading spaces 
+                * before actual value
+                */
+               while( *str != '\0' ) {
+                       if ( isspace( *str ) )
+                               str++;
+                       else 
+                               return str;
+               }
+       }
+       
+       return NULL;
+}
+
+static void
+getmeminfo(TCMsgTop *m) {
+       FILE *fh;
+       static char buf[MEMBUFSIZE];
+       char *ptr;
+
+       fh=fopen("/proc/meminfo", "r");
+       if (fh == NULL) {
+               tlog(TL_ALARM, "fopen(\"/proc/meminfo\", ...) failed: %s", strerror(errno));
+               return;
+       }
+
+       while( fgets(buf, MEMBUFSIZE, fh) &&  (m->physmem == 0 || m->freemem == 0) ) {
+               if ( (ptr = cmpPattern(buf, "MemTotal:")) != NULL ) {
+                       m->physmem = atoll(ptr) * 1024; /* mem in Kb */ 
+               } else if ( (ptr = cmpPattern(buf, "MemFree:")) != NULL ) {
+                       m->freemem = atoll(ptr) * 1024; /* mem in Kb */ 
+               }
+       }
+
+       fclose(fh);
+}
+
+#else /* end Linux*/
+#error No memory stat for current OS
+#endif 
+
+#define NITEM   0
+
+static void
+fillMsgTop(TCMsgTop *m) {
+       double lll[3];
+
+       memset(m,0,sizeof(TCMsgTop));
+       if ( getloadavg(lll,3) >= NITEM+1 )
+               m->load=lll[NITEM];
+       else
+               m->load=-1.0;
+
+       m->freemem = 0;
+       m->physmem = 0;
+       getmeminfo(m);
 
-       tlog( TL_DEBUG, "Sended topdata: %.2f load, %d free, %d total", m->load, m->freemem, m->usermem);
+       tlog( TL_DEBUG, "Sended topdata: %.2f load, %lld free, %lld total", m->load, m->freemem, m->physmem);
 }
 
 static int