minix3/commands/udpstat/udpstat.c

326 lines
7.0 KiB
C

/*
udpstat.c
Created: March 2001 by Philip Homburg <philip@f-mnx.phicoh.com>
*/
#define _POSIX_C_SOURCE 2
#define _NETBSD_SOURCE 1
#include <inet/inet.h>
#undef printf
#undef send
#include <assert.h>
#include <fcntl.h>
#include <stdio.h>
#include <time.h>
#include <unistd.h>
#include <sys/svrctl.h>
#ifndef __minix_vmd
#include <sys/times.h>
#endif
#include <net/netlib.h>
#include <net/gen/inet.h>
#include <netdb.h>
#include <net/gen/socket.h>
#include <minix/queryparam.h>
#include <inet/generic/buf.h>
#include <inet/generic/clock.h>
#include <inet/generic/event.h>
#include <inet/generic/type.h>
#include <inet/generic/udp_int.h>
char *prog_name;
udp_fd_t udp_fd_table[UDP_FD_NR];
udp_port_t *udp_port_table;
udp_port_t *udp_port_tablep;
size_t udp_port_table_s;
size_t udp_port_table_rs;
int udp_port_nr;
char values[6 * sizeof(void *) + 3];
char *valuesl= NULL;
size_t v_size;
int inclSel, numerical;
void print_fd(int i, clock_t now);
void usage(void);
int main(int argc, char*argv[])
{
char *ipstat_device;
int fd, i, r;
size_t psize;
char *pval, *param;
#ifdef __minix_vmd
struct timeval uptime;
#endif
clock_t now;
int fl;
int a_flag, n_flag;
struct tms tmsbuf;
(prog_name=strrchr(argv[0], '/')) ? prog_name++ : (prog_name=argv[0]);
a_flag= 0;
n_flag= 0;
while ((fl= getopt(argc, argv, "?an")) != -1)
{
switch(fl)
{
case '?':
usage();
case 'a':
a_flag= 1;
break;
case 'n':
n_flag= 1;
break;
default:
fprintf(stderr, "%s: getopt failed: '%c'\n",
prog_name, fl);
exit(1);
}
}
inclSel= !!a_flag;
numerical= !!n_flag;
ipstat_device= IPSTAT_DEVICE;
if ((fd= open(ipstat_device, O_RDWR)) == -1)
{
fprintf(stderr, "%s: unable to open '%s': %s\n", prog_name,
ipstat_device, strerror(errno));
exit(1);
}
v_size= 2*sizeof(udp_fd_table)+1;
valuesl= realloc(valuesl, v_size);
if (!valuesl)
{
fprintf(stderr, "%s: unable to malloc %u bytes for valuesl\n",
prog_name, v_size);
exit(1);
}
param= "udp_fd_table";
psize= strlen(param);
r= write(fd, param, psize);
if (r != psize)
{
fprintf(stderr, "%s: write to %s failed: %s\n", prog_name,
ipstat_device,
r < 0 ? strerror(errno) : "short write");
exit(1);
}
r= read(fd, valuesl, v_size);
if (r < 0)
{
fprintf(stderr, "%s: read from %s failed: %s\n", prog_name,
ipstat_device, strerror(errno));
exit(1);
}
pval= valuesl;
if (paramvalue(&pval, udp_fd_table, sizeof(udp_fd_table)) !=
sizeof(udp_fd_table))
{
fprintf(stderr,
"%s: unable to decode the results from queryparam (udp_fd_table)\n",
prog_name);
exit(1);
}
/* Get address, size, and element size of the UDP port table */
param = "&udp_port_table,$udp_port_table,$udp_port_table[0]";
psize = strlen(param);
r= write(fd, param, psize);
if (r != psize)
{
fprintf(stderr, "%s: write to %s failed: %s\n", prog_name,
ipstat_device,
r < 0 ? strerror(errno) : "short write");
exit(1);
}
r= read(fd, values, sizeof(values));
if (r < 0)
{
fprintf(stderr, "%s: read from %s failed: %s\n", prog_name,
ipstat_device, strerror(errno));
exit(1);
}
pval= values;
if (paramvalue(&pval, &udp_port_tablep, sizeof(udp_port_tablep)) !=
sizeof(udp_port_tablep) ||
paramvalue(&pval, &udp_port_table_s, sizeof(udp_port_table_s))
!= sizeof(udp_port_table_s) ||
paramvalue(&pval, &udp_port_table_rs, sizeof(udp_port_table_rs))
!= sizeof(udp_port_table_rs))
{
fprintf(stderr,
"%s: unable to decode the results from queryparam (&udp_port_table, ...)\n",
prog_name);
exit(1);
}
if (udp_port_table_rs != sizeof(udp_port_table[0]))
{
fprintf(stderr,
"%s: size mismatch in udp_port_table (different version of inet?)\n",
prog_name);
exit(1);
}
udp_port_nr= udp_port_table_s/udp_port_table_rs;
assert(udp_port_table_s == udp_port_nr*udp_port_table_rs);
udp_port_table= malloc(udp_port_table_s);
if (!udp_port_table)
{
fprintf(stderr,
"%s: unable to malloc %u bytes for udp_port_table\n",
prog_name, udp_port_table_s);
exit(1);
}
v_size= 2*udp_port_table_s+1;
valuesl= realloc(valuesl, v_size);
if (!valuesl)
{
fprintf(stderr, "%s: unable to malloc %u bytes for valuesl\n",
prog_name, v_size);
exit(1);
}
param = "udp_port_table";
psize = strlen(param);
r= write(fd, param, psize);
if (r != psize)
{
fprintf(stderr, "%s: write to %s failed: %s\n", prog_name,
ipstat_device,
r < 0 ? strerror(errno) : "short write");
exit(1);
}
r= read(fd, valuesl, v_size);
if (r < 0)
{
fprintf(stderr, "%s: read from %s failed: %s\n", prog_name,
ipstat_device, strerror(errno));
exit(1);
}
pval= valuesl;
if (paramvalue(&pval, udp_port_table, udp_port_table_s) !=
udp_port_table_s)
{
fprintf(stderr,
"%s: unable to decode the results from queryparam (udp_port_table)\n",
prog_name);
exit(1);
}
/* Get the uptime in clock ticks. */
#ifdef __minix_vmd
if (sysutime(UTIME_UPTIME, &uptime) == -1)
{
fprintf(stderr, "%s: sysutime failed: %s\n", prog_name,
strerror(errno));
exit(1);
}
now= uptime.tv_sec * HZ + (uptime.tv_usec*HZ/1000000);
#else /* Minix 3 */
now= times(&tmsbuf);
#endif
for (i= 0; i<UDP_FD_NR; i++)
print_fd(i, now);
exit(0);
}
void print_fd(int i, clock_t now)
{
unsigned long nwuo_flags;
int port_nr;
udp_fd_t *udp_fd;
udp_port_t *udp_port;
char *locaddr_str, *remaddr_str;
struct hostent *hostent;
struct servent *servent;
nwio_udpopt_t uf_udpopt;
udp_fd= &udp_fd_table[i];
if (!(udp_fd->uf_flags & UFF_INUSE))
return;
uf_udpopt= udp_fd->uf_udpopt;
nwuo_flags= uf_udpopt.nwuo_flags;
if (((nwuo_flags & NWUO_LOCPORT_MASK) != NWUO_LP_SET) && !inclSel)
return;
port_nr= udp_fd->uf_port-udp_port_tablep;
udp_port= &udp_port_table[port_nr];
printf("%3d", i);
if (nwuo_flags & NWUO_EN_LOC)
{
if (!numerical && (hostent=
gethostbyaddr((char *)&udp_port->up_ipaddr,
sizeof(ipaddr_t), AF_INET)) != NULL)
{
locaddr_str= hostent->h_name;
}
else
locaddr_str= inet_ntoa(udp_port->up_ipaddr);
}
else if (nwuo_flags & NWUO_EN_BROAD)
locaddr_str= "255.255.255.255";
else
locaddr_str= "0.0.0.0";
printf(" %s:", locaddr_str);
if ((nwuo_flags & NWUO_LOCPORT_MASK) != NWUO_LP_SEL &&
(nwuo_flags & NWUO_LOCPORT_MASK) != NWUO_LP_SET)
{
printf("*");
}
else if ((servent= getservbyport(uf_udpopt.nwuo_locport, "udp")) !=
NULL)
{
printf("%s", servent->s_name);
}
else
printf("%u", ntohs(uf_udpopt.nwuo_locport));
printf(" -> ");
if (!(nwuo_flags & NWUO_RA_SET))
remaddr_str= "*";
else if (!numerical &&
(hostent= gethostbyaddr((char *)&uf_udpopt.nwuo_remaddr,
sizeof(ipaddr_t), AF_INET)) != NULL)
{
remaddr_str= hostent->h_name;
}
else
remaddr_str= inet_ntoa(uf_udpopt.nwuo_remaddr);
printf("%s:", remaddr_str);
if (!(nwuo_flags & NWUO_RP_SET))
printf("*");
else if ((servent= getservbyport(uf_udpopt.nwuo_remport, "udp")) !=
NULL)
{
printf("%s", servent->s_name);
}
else
printf("%u", ntohs(uf_udpopt.nwuo_remport));
printf("\n");
}
void usage(void)
{
fprintf(stderr, "Usage: %s [-a] [-n]\n", prog_name);
exit(1);
}
/*
* $PchId: udpstat.c,v 1.4 2005/01/30 01:04:57 philip Exp $
*/