aboutsummaryrefslogtreecommitdiff
path: root/main.c
blob: 6cfd203016d21757cd84fccb867002e8a63c6f8d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include <u.h>
#include <libc.h>

typedef struct IPNet IPNet;

struct IPNet
{
	u32int addr;
	u32int mask;
	u32int minaddr;
	u32int maxaddr;
	u32int bcast;
	uint nhosts;
	int cidr;
};

#pragma varargck type "I" u32int
#pragma varargck type "N" IPNet

int
Ifmt(Fmt *f)
{
	uchar ip4[4];
	u32int addr;

	addr = va_arg(f->args, u32int);

	ip4[0] = addr>>24 & 0xff;
	ip4[1] = addr>>16 & 0xff;
	ip4[2] = addr>>8  & 0xff;
	ip4[3] = addr     & 0xff;

	return fmtprint(f, "%d.%d.%d.%d", ip4[0], ip4[1], ip4[2], ip4[3]);
}

int
Nfmt(Fmt *f)
{
	IPNet net;
	int n;

	net = va_arg(f->args, IPNet);

	n = fmtprint(f, "network %I\n", net.addr);
	n += fmtprint(f, "netmask %I\n", net.mask);
	n += fmtprint(f, "minaddr %I\n", net.minaddr);
	n += fmtprint(f, "maxaddr %I\n", net.maxaddr);
	n += fmtprint(f, "bcast %I\n", net.bcast);
	n += fmtprint(f, "hosts %d\n", net.nhosts);
	n += fmtprint(f, "cidr %d\n", net.cidr);
	return n;
}

int
countones(u32int addr)
{
	int cnt, shift;

	for(cnt = 0, shift = 31; cnt < 32 && (addr & 1<<shift) != 0; cnt++, shift--)
		;
	return cnt;
}

void
usage(void)
{
	fprint(2, "usage: %s addr mask\n", argv0);
	exits("usage");
}

void
main(int argc, char *argv[])
{
	IPNet net;
	u32int addr, mask;
	char *a, *m;
	int cidr;

	fmtinstall('I', Ifmt);
	fmtinstall('N', Nfmt);
	ARGBEGIN{
	default: usage();
	}ARGEND;
	if(argc != 2)
		usage();

	a = argv[0];
	addr = strtoul(a, &a, 10) << 24;
	addr |= strtoul(++a, &a, 10) << 16;
	addr |= strtoul(++a, &a, 10) << 8;
	addr |= strtoul(++a, &a, 10);
	m = argv[1];
	if(strlen(m) <= 2){
		cidr = strtol(m, nil, 10);
		mask = ~0 << (32-cidr);
	}else{
		mask = strtoul(m, &m, 10) << 24;
		mask |= strtoul(++m, &m, 10) << 16;
		mask |= strtoul(++m, &m, 10) << 8;
		mask |= strtoul(++m, &m, 10);
	}

	net.addr = addr&mask;
	net.mask = mask;
	net.bcast = addr|~mask;
	net.minaddr = net.addr+1;
	net.maxaddr = net.bcast-1;
	net.nhosts = net.maxaddr-net.minaddr;
	net.cidr = countones(net.mask);
	print("%N", net);

	exits(nil);
}