Did some more work on the daemon.
This commit is contained in:
259
src/daemon/main.c
Normal file
259
src/daemon/main.c
Normal file
@@ -0,0 +1,259 @@
|
||||
/* $Id$ */
|
||||
|
||||
/* Copyright (C) 1995, 1996, 1997 Free Software Foundation, Inc.
|
||||
This file is part of the Gnome Top Library.
|
||||
Contributed by Martin Baulig <martin@home-of-linux.org>, April 1998.
|
||||
|
||||
The Gnome Top Library is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU Library General Public License as
|
||||
published by the Free Software Foundation; either version 2 of the
|
||||
License, or (at your option) any later version.
|
||||
|
||||
The Gnome Top Library 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
|
||||
Library General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Library General Public
|
||||
License along with the GNU C Library; see the file COPYING.LIB. If not,
|
||||
write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
||||
Boston, MA 02111-1307, USA. */
|
||||
|
||||
#include <glibtop/gnuserv.h>
|
||||
|
||||
#include <glibtop/command.h>
|
||||
#include <glibtop/version.h>
|
||||
#include <glibtop/xmalloc.h>
|
||||
#include <glibtop/union.h>
|
||||
#include <glibtop/open.h>
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <locale.h>
|
||||
|
||||
#if defined(HAVE_GETDTABLESIZE)
|
||||
#define GET_MAX_FDS() getdtablesize()
|
||||
#else
|
||||
/* Fallthrough case - please add other #elif cases above
|
||||
for different OS's as necessary */
|
||||
#define GET_MAX_FDS() 256
|
||||
#endif
|
||||
|
||||
static void
|
||||
do_output (int s, size_t size, const void *buf)
|
||||
{
|
||||
#ifdef REAL_DEBUG
|
||||
fprintf (stderr, "Writing %d bytes = %d.\n", sizeof (size_t), size);
|
||||
#endif
|
||||
|
||||
if (send (s, &size, sizeof (size_t), 0) < 0)
|
||||
glibtop_warn_io ("send");
|
||||
|
||||
if (!size)
|
||||
return;
|
||||
|
||||
#ifdef REAL_DEBUG
|
||||
fprintf (stderr, "Really writing %d bytes.\n", size);
|
||||
#endif
|
||||
|
||||
if (send (s, buf, size, 0) < 0)
|
||||
glibtop_warn_io ("send");
|
||||
}
|
||||
|
||||
void
|
||||
handle_socket_connection (int s)
|
||||
{
|
||||
pid_t pid;
|
||||
int nread;
|
||||
size_t size, dsize;
|
||||
char parameter [BUFSIZ];
|
||||
struct timeval tv;
|
||||
glibtop_union data;
|
||||
glibtop_command cmnd;
|
||||
glibtop_sysdeps sysdeps;
|
||||
glibtop server;
|
||||
void *ptr;
|
||||
|
||||
tv.tv_sec = 5;
|
||||
tv.tv_usec = 0;
|
||||
|
||||
while(1) {
|
||||
nread = recv (s, &size, sizeof (size_t), 0);
|
||||
|
||||
if (nread < 0) {
|
||||
glibtop_warn_io ("recv");
|
||||
return;
|
||||
}
|
||||
|
||||
if (nread == 0)
|
||||
return;
|
||||
|
||||
if (nread != sizeof (size_t)) {
|
||||
glibtop_warn ("Expected %d bytes but got %d", sizeof (size_t), nread);
|
||||
return;
|
||||
}
|
||||
|
||||
if (size != sizeof (glibtop_command)) {
|
||||
glibtop_warn ("Expected %d bytes but got %d", sizeof (size_t), nread);
|
||||
return;
|
||||
}
|
||||
|
||||
nread = recv (s, &cmnd, size, 0);
|
||||
|
||||
if (nread < 0) {
|
||||
glibtop_warn_io ("recv");
|
||||
return;
|
||||
}
|
||||
|
||||
if (nread == 0)
|
||||
return;
|
||||
|
||||
#ifdef REAL_DEBUG
|
||||
fprintf (stderr, "Received command %d from client.\n", cmnd.command);
|
||||
#endif
|
||||
|
||||
nread = recv (s, &dsize, sizeof (size_t), 0);
|
||||
|
||||
/* will return 0 if parent exits. */
|
||||
|
||||
if (nread < 0) {
|
||||
glibtop_warn_io ("recv");
|
||||
return;
|
||||
}
|
||||
|
||||
if (nread == 0)
|
||||
return;
|
||||
|
||||
if (dsize >= BUFSIZ) {
|
||||
glibtop_warn ("Client sent %d bytes, but buffer is %d", dsize, BUFSIZ);
|
||||
return;
|
||||
}
|
||||
|
||||
memset (parameter, 0, sizeof (parameter));
|
||||
|
||||
if (dsize) {
|
||||
#ifdef REAL_DEBUG
|
||||
fprintf (stderr, "Client has %d bytes of data.\n", dsize);
|
||||
#endif
|
||||
|
||||
nread = recv (s, parameter, dsize, 0);
|
||||
|
||||
/* will return 0 if parent exits. */
|
||||
|
||||
if (nread < 0) {
|
||||
glibtop_warn_io ("recv");
|
||||
return;
|
||||
}
|
||||
|
||||
if (nread == 0)
|
||||
return;
|
||||
|
||||
if (nread != (int) dsize) {
|
||||
glibtop_warn ("Expected %d bytes but got %d", dsize, nread);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
switch (cmnd.command) {
|
||||
case GLIBTOP_CMND_SYSDEPS:
|
||||
sysdeps.features = GLIBTOP_SYSDEPS_ALL;
|
||||
do_output (s, sizeof (glibtop_sysdeps), &sysdeps);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_CPU:
|
||||
glibtop_get_cpu_l (&server, &data.cpu);
|
||||
do_output (s, sizeof (glibtop_cpu), &data.cpu);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_MEM:
|
||||
glibtop_get_mem_l (&server, &data.mem);
|
||||
do_output (s, sizeof (glibtop_mem), &data.mem);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_SWAP:
|
||||
glibtop_get_swap_l (&server, &data.swap);
|
||||
do_output (s, sizeof (glibtop_swap), &data.swap);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_UPTIME:
|
||||
glibtop_get_uptime_l (&server, &data.uptime);
|
||||
do_output (s, sizeof (glibtop_uptime), &data.uptime);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_LOADAVG:
|
||||
glibtop_get_loadavg_l (&server, &data.loadavg);
|
||||
do_output (s, sizeof (glibtop_loadavg), &data.loadavg);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_SHM_LIMITS:
|
||||
glibtop_get_shm_limits_l (&server, &data.shm_limits);
|
||||
do_output (s, sizeof (glibtop_shm_limits), &data.shm_limits);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_MSG_LIMITS:
|
||||
glibtop_get_msg_limits_l (&server, &data.msg_limits);
|
||||
do_output (s, sizeof (glibtop_msg_limits), &data.msg_limits);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_SEM_LIMITS:
|
||||
glibtop_get_sem_limits_l (&server, &data.sem_limits);
|
||||
do_output (s, sizeof (glibtop_sem_limits), &data.sem_limits);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROCLIST:
|
||||
ptr = glibtop_get_proclist_l (&server, &data.proclist);
|
||||
do_output (s, sizeof (glibtop_proclist), &data.proclist);
|
||||
do_output (s, data.proclist.total, ptr);
|
||||
glibtop_free_r (&server, ptr);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_STATE:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_state_l
|
||||
(&server, &data.proc_state, pid);
|
||||
do_output (s, sizeof (glibtop_proc_state), &data.proc_state);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_UID:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_uid_l
|
||||
(&server, &data.proc_uid, pid);
|
||||
do_output (s, sizeof (glibtop_proc_uid), &data.proc_uid);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_MEM:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_mem_l
|
||||
(&server, &data.proc_mem, pid);
|
||||
do_output (s, sizeof (glibtop_proc_mem), &data.proc_mem);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_TIME:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_time_l
|
||||
(&server, &data.proc_time, pid);
|
||||
do_output (s, sizeof (glibtop_proc_time), &data.proc_time);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_SIGNAL:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_signal_l
|
||||
(&server, &data.proc_signal, pid);
|
||||
do_output (s, sizeof (glibtop_proc_signal), &data.proc_signal);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_KERNEL:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_kernel_l
|
||||
(&server, &data.proc_kernel, pid);
|
||||
do_output (s, sizeof (glibtop_proc_kernel), &data.proc_kernel);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
case GLIBTOP_CMND_PROC_SEGMENT:
|
||||
memcpy (&pid, parameter, sizeof (pid_t));
|
||||
glibtop_get_proc_segment_l
|
||||
(&server, &data.proc_segment, pid);
|
||||
do_output (s, sizeof (glibtop_proc_segment), &data.proc_segment);
|
||||
do_output (s, 0, NULL);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user