Annotation of src/usr.bin/systat/pftop.c, Revision 1.18
1.18 ! jasper 1: /* $OpenBSD: pftop.c,v 1.17 2011/02/06 05:20:25 lum Exp $ */
1.1 canacar 2: /*
3: * Copyright (c) 2001, 2007 Can Erkin Acar
4: * Copyright (c) 2001 Daniel Hartmeier
5: * All rights reserved.
6: *
7: * Redistribution and use in source and binary forms, with or without
8: * modification, are permitted provided that the following conditions
9: * are met:
10: *
11: * - Redistributions of source code must retain the above copyright
12: * notice, this list of conditions and the following disclaimer.
13: * - Redistributions in binary form must reproduce the above
14: * copyright notice, this list of conditions and the following
15: * disclaimer in the documentation and/or other materials provided
16: * with the distribution.
17: *
18: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
21: * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22: * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
23: * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
24: * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25: * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
26: * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28: * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29: * POSSIBILITY OF SUCH DAMAGE.
30: *
31: */
32:
33: #include <sys/types.h>
34: #include <sys/ioctl.h>
35: #include <sys/socket.h>
36:
37: #include <net/if.h>
38: #include <netinet/in.h>
1.8 mcbride 39: #include <netinet/tcp.h>
1.1 canacar 40: #include <netinet/tcp_fsm.h>
41: #include <net/pfvar.h>
42: #include <arpa/inet.h>
43:
44: #include <altq/altq.h>
45: #include <altq/altq_cbq.h>
46: #include <altq/altq_priq.h>
47: #include <altq/altq_hfsc.h>
48:
49: #include <ctype.h>
50: #include <curses.h>
51: #include <err.h>
52: #include <errno.h>
53: #include <fcntl.h>
54: #include <netdb.h>
55: #include <signal.h>
56: #include <stdio.h>
57: #include <stdlib.h>
58: #include <string.h>
59: #include <unistd.h>
60: #include <stdarg.h>
61:
1.6 canacar 62: #include "systat.h"
1.1 canacar 63: #include "engine.h"
64: #include "cache.h"
65:
66: extern const char *tcpstates[];
67:
68: #define MIN_NUM_STATES 1024
69: #define NUM_STATE_INC 1024
70:
71: #define DEFAULT_CACHE_SIZE 10000
72:
73: /* XXX must also check type before use */
74: #define PT_ADDR(x) (&(x)->addr.v.a.addr)
75:
76: /* XXX must also check type before use */
77: #define PT_MASK(x) (&(x)->addr.v.a.mask)
78:
79: #define PT_NOROUTE(x) ((x)->addr.type == PF_ADDR_NOROUTE)
80:
81: /* view management */
82: int select_states(void);
83: int read_states(void);
84: void sort_states(void);
85: void print_states(void);
86:
87: int select_rules(void);
88: int read_rules(void);
89: void print_rules(void);
90:
91: int select_queues(void);
92: int read_queues(void);
93: void print_queues(void);
94:
1.7 canacar 95: void update_cache(void);
96:
1.1 canacar 97: /* qsort callbacks */
98: int sort_size_callback(const void *s1, const void *s2);
99: int sort_exp_callback(const void *s1, const void *s2);
100: int sort_pkt_callback(const void *s1, const void *s2);
101: int sort_age_callback(const void *s1, const void *s2);
102: int sort_sa_callback(const void *s1, const void *s2);
103: int sort_sp_callback(const void *s1, const void *s2);
104: int sort_da_callback(const void *s1, const void *s2);
105: int sort_dp_callback(const void *s1, const void *s2);
106: int sort_rate_callback(const void *s1, const void *s2);
107: int sort_peak_callback(const void *s1, const void *s2);
108: int pf_dev = -1;
109:
110: struct sc_ent **state_cache = NULL;
1.4 canacar 111: struct pfsync_state *state_buf = NULL;
1.1 canacar 112: int state_buf_len = 0;
113: u_int32_t *state_ord = NULL;
114: u_int32_t num_states = 0;
115: u_int32_t num_states_all = 0;
116: u_int32_t num_rules = 0;
117: u_int32_t num_queues = 0;
118: int cachestates = 0;
119:
120: char *filter_string = NULL;
121: int dumpfilter = 0;
122:
123: #define MIN_LABEL_SIZE 5
124: #define ANCHOR_FLD_SIZE 12
125:
126: /* Define fields */
127: field_def fields[] = {
128: {"SRC", 20, 45, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
129: {"DEST", 20, 45, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
130: {"GW", 20, 45, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
131: {"STATE", 5, 23, 18, FLD_ALIGN_COLUMN, -1, 0, 0, 0},
132: {"AGE", 5, 9, 4, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
133: {"EXP", 5, 9, 4, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
134: {"PR ", 4, 9, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
135: {"DIR", 1, 3, 2, FLD_ALIGN_CENTER, -1, 0, 0, 0},
136: {"PKTS", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
137: {"BYTES", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
138: {"RULE", 2, 4, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
139: {"LABEL", MIN_LABEL_SIZE, MIN_LABEL_SIZE, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
140: {"STATES", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
141: {"EVAL", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
142: {"ACTION", 1, 8, 4, FLD_ALIGN_LEFT, -1, 0, 0, 0},
143: {"LOG", 1, 3, 2, FLD_ALIGN_LEFT, -1, 0, 0, 0},
144: {"QUICK", 1, 1, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
145: {"KS", 1, 1, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
146: {"IF", 4, 6, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
147: {"INFO", 40, 80, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
1.4 canacar 148: {"MAX", 3, 5, 2, FLD_ALIGN_RIGHT, -1, 0, 0},
1.1 canacar 149: {"RATE", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
150: {"AVG", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
151: {"PEAK", 5, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
1.4 canacar 152: {"ANCHOR", 6, 16, 1, FLD_ALIGN_LEFT, -1, 0, 0},
1.1 canacar 153: {"QUEUE", 15, 30, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
154: {"BW", 4, 5, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
155: {"SCH", 3, 4, 1, FLD_ALIGN_LEFT, -1, 0, 0, 0},
156: {"PRIO", 1, 4, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
157: {"DROP_P", 6, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
158: {"DROP_B", 6, 8, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
159: {"QLEN", 4, 4, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
160: {"BORROW", 4, 6, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
161: {"SUSPENDS", 4, 6, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
162: {"P/S", 3, 7, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0},
163: {"B/S", 4, 7, 1, FLD_ALIGN_RIGHT, -1, 0, 0, 0}
164: };
165:
166:
167: /* for states */
1.18 ! jasper 168: #define FLD_SRC FIELD_ADDR(fields,0)
! 169: #define FLD_DEST FIELD_ADDR(fields,1)
! 170: #define FLD_GW FIELD_ADDR(fields,2)
! 171: #define FLD_STATE FIELD_ADDR(fields,3)
! 172: #define FLD_AGE FIELD_ADDR(fields,4)
! 173: #define FLD_EXP FIELD_ADDR(fields,5)
1.1 canacar 174: /* common */
1.18 ! jasper 175: #define FLD_PROTO FIELD_ADDR(fields,6)
! 176: #define FLD_DIR FIELD_ADDR(fields,7)
! 177: #define FLD_PKTS FIELD_ADDR(fields,8)
! 178: #define FLD_BYTES FIELD_ADDR(fields,9)
! 179: #define FLD_RULE FIELD_ADDR(fields,10)
1.1 canacar 180: /* for rules */
1.18 ! jasper 181: #define FLD_LABEL FIELD_ADDR(fields,11)
! 182: #define FLD_STATS FIELD_ADDR(fields,12)
! 183: #define FLD_EVAL FIELD_ADDR(fields,13)
! 184: #define FLD_ACTION FIELD_ADDR(fields,14)
! 185: #define FLD_LOG FIELD_ADDR(fields,15)
! 186: #define FLD_QUICK FIELD_ADDR(fields,16)
! 187: #define FLD_KST FIELD_ADDR(fields,17)
! 188: #define FLD_IF FIELD_ADDR(fields,18)
! 189: #define FLD_RINFO FIELD_ADDR(fields,19)
! 190: #define FLD_STMAX FIELD_ADDR(fields,20)
1.1 canacar 191: /* other */
1.18 ! jasper 192: #define FLD_SI FIELD_ADDR(fields,21) /* instantaneous speed */
! 193: #define FLD_SA FIELD_ADDR(fields,22) /* average speed */
! 194: #define FLD_SP FIELD_ADDR(fields,23) /* peak speed */
! 195: #define FLD_ANCHOR FIELD_ADDR(fields,24)
1.1 canacar 196: /* for queues */
1.18 ! jasper 197: #define FLD_QUEUE FIELD_ADDR(fields,25)
! 198: #define FLD_BANDW FIELD_ADDR(fields,26)
! 199: #define FLD_SCHED FIELD_ADDR(fields,27)
! 200: #define FLD_PRIO FIELD_ADDR(fields,28)
! 201: #define FLD_DROPP FIELD_ADDR(fields,29)
! 202: #define FLD_DROPB FIELD_ADDR(fields,30)
! 203: #define FLD_QLEN FIELD_ADDR(fields,31)
! 204: #define FLD_BORR FIELD_ADDR(fields,32)
! 205: #define FLD_SUSP FIELD_ADDR(fields,33)
! 206: #define FLD_PKTSPS FIELD_ADDR(fields,34)
! 207: #define FLD_BYTESPS FIELD_ADDR(fields,35)
1.1 canacar 208:
209: /* Define views */
210: field_def *view0[] = {
211: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_STATE,
212: FLD_AGE, FLD_EXP, FLD_PKTS, FLD_BYTES, NULL
213: };
214:
215: field_def *view1[] = {
216: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_GW, FLD_STATE, FLD_AGE,
217: FLD_EXP, FLD_PKTS, FLD_BYTES, FLD_SI, FLD_SP, FLD_SA, FLD_RULE, NULL
218: };
219:
220: field_def *view2[] = {
221: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_STATE, FLD_AGE, FLD_EXP,
222: FLD_PKTS, FLD_BYTES, FLD_SI, FLD_SP, FLD_SA, FLD_RULE, FLD_GW, NULL
223: };
224:
225: field_def *view3[] = {
226: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_AGE, FLD_EXP, FLD_PKTS,
227: FLD_BYTES, FLD_STATE, FLD_SI, FLD_SP, FLD_SA, FLD_RULE, FLD_GW, NULL
228: };
229:
230: field_def *view4[] = {
231: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_PKTS, FLD_BYTES, FLD_STATE,
232: FLD_AGE, FLD_EXP, FLD_SI, FLD_SP, FLD_SA, FLD_RULE, FLD_GW, NULL
233: };
234:
235: field_def *view5[] = {
236: FLD_RULE, FLD_ANCHOR, FLD_ACTION, FLD_DIR, FLD_LOG, FLD_QUICK, FLD_IF,
237: FLD_PROTO, FLD_KST, FLD_PKTS, FLD_BYTES, FLD_STATS, FLD_STMAX,
238: FLD_RINFO, NULL
239: };
240:
241: field_def *view6[] = {
242: FLD_RULE, FLD_LABEL, FLD_PKTS, FLD_BYTES, FLD_STATS, FLD_STMAX,
243: FLD_ACTION, FLD_DIR, FLD_LOG, FLD_QUICK, FLD_IF, FLD_PROTO,
244: FLD_ANCHOR, FLD_KST, NULL
245: };
246:
247: field_def *view7[] = {
248: FLD_PROTO, FLD_DIR, FLD_SRC, FLD_DEST, FLD_SI, FLD_SP, FLD_SA,
249: FLD_BYTES, FLD_STATE, FLD_PKTS, FLD_AGE, FLD_EXP, FLD_RULE, FLD_GW, NULL
250: };
251:
252: field_def *view8[] = {
253: FLD_QUEUE, FLD_BANDW, FLD_SCHED, FLD_PRIO, FLD_PKTS, FLD_BYTES,
254: FLD_DROPP, FLD_DROPB, FLD_QLEN, FLD_BORR, FLD_SUSP, FLD_PKTSPS,
255: FLD_BYTESPS, NULL
256: };
257:
258: /* Define orderings */
259: order_type order_list[] = {
260: {"none", "none", 'N', NULL},
261: {"bytes", "bytes", 'B', sort_size_callback},
262: {"expiry", "exp", 'E', sort_exp_callback},
263: {"packets", "pkt", 'P', sort_pkt_callback},
264: {"age", "age", 'A', sort_age_callback},
265: {"source addr", "src", 'F', sort_sa_callback},
266: {"dest. addr", "dest", 'T', sort_da_callback},
267: {"source port", "sport", 'S', sort_sp_callback},
268: {"dest. port", "dport", 'D', sort_dp_callback},
269: {"rate", "rate", 'R', sort_rate_callback},
270: {"peak", "peak", 'K', sort_peak_callback},
271: {NULL, NULL, 0, NULL}
272: };
273:
274: /* Define view managers */
275: struct view_manager state_mgr = {
276: "States", select_states, read_states, sort_states, print_header,
277: print_states, keyboard_callback, order_list, NULL
278: };
279:
280: struct view_manager rule_mgr = {
281: "Rules", select_rules, read_rules, NULL, print_header,
282: print_rules, keyboard_callback, NULL, NULL
283: };
284:
285: struct view_manager queue_mgr = {
286: "Queues", select_queues, read_queues, NULL, print_header,
287: print_queues, keyboard_callback, NULL, NULL
288: };
289:
290: field_view views[] = {
291: {view2, "states", '8', &state_mgr},
292: {view5, "rules", '9', &rule_mgr},
293: {view8, "queues", 'Q', &queue_mgr},
294: {NULL, NULL, 0, NULL}
295: };
296:
297:
298: /* altq structures from pfctl */
299:
300: union class_stats {
301: class_stats_t cbq_stats;
302: struct priq_classstats priq_stats;
303: struct hfsc_classstats hfsc_stats;
304: };
305:
306: struct queue_stats {
307: union class_stats data;
308: struct timeval timestamp;
309: u_int8_t valid;
310: };
311:
312: struct pf_altq_node {
313: struct pf_altq altq;
314: struct pf_altq_node *next;
315: struct pf_altq_node *children;
316: struct pf_altq_node *next_flat;
317: struct queue_stats qstats;
318: struct queue_stats qstats_last;
319: u_int8_t depth;
320: u_int8_t visited;
321: };
322:
323:
324: /* ordering functions */
325:
326: int
327: sort_size_callback(const void *s1, const void *s2)
328: {
329: u_int64_t b1 = COUNTER(state_buf[* (u_int32_t *) s1].bytes[0]) +
330: COUNTER(state_buf[* (u_int32_t *) s1].bytes[1]);
331: u_int64_t b2 = COUNTER(state_buf[* (u_int32_t *) s2].bytes[0]) +
332: COUNTER(state_buf[* (u_int32_t *) s2].bytes[1]);
333: if (b2 > b1)
334: return sortdir;
335: if (b2 < b1)
336: return -sortdir;
337: return 0;
338: }
339:
340: int
341: sort_pkt_callback(const void *s1, const void *s2)
342: {
343: u_int64_t p1 = COUNTER(state_buf[* (u_int32_t *) s1].packets[0]) +
344: COUNTER(state_buf[* (u_int32_t *) s1].packets[1]);
345: u_int64_t p2 = COUNTER(state_buf[* (u_int32_t *) s2].packets[0]) +
346: COUNTER(state_buf[* (u_int32_t *) s2].packets[1]);
347: if (p2 > p1)
348: return sortdir;
349: if (p2 < p1)
350: return -sortdir;
351: return 0;
352: }
353:
354: int
355: sort_age_callback(const void *s1, const void *s2)
356: {
1.3 mcbride 357: if (ntohl(state_buf[* (u_int32_t *) s2].creation) >
358: ntohl(state_buf[* (u_int32_t *) s1].creation))
1.1 canacar 359: return sortdir;
1.3 mcbride 360: if (ntohl(state_buf[* (u_int32_t *) s2].creation) <
361: ntohl(state_buf[* (u_int32_t *) s1].creation))
1.1 canacar 362: return -sortdir;
363: return 0;
364: }
365:
366: int
367: sort_exp_callback(const void *s1, const void *s2)
368: {
1.3 mcbride 369: if (ntohl(state_buf[* (u_int32_t *) s2].expire) >
370: ntohl(state_buf[* (u_int32_t *) s1].expire))
1.1 canacar 371: return sortdir;
1.3 mcbride 372: if (ntohl(state_buf[* (u_int32_t *) s2].expire) <
373: ntohl(state_buf[* (u_int32_t *) s1].expire))
1.1 canacar 374: return -sortdir;
375: return 0;
376: }
377:
378: int
379: sort_rate_callback(const void *s1, const void *s2)
380: {
381: struct sc_ent *e1 = state_cache[* (u_int32_t *) s1];
382: struct sc_ent *e2 = state_cache[* (u_int32_t *) s2];
383:
384: if (e1 == NULL)
385: return sortdir;
386: if (e2 == NULL)
387: return -sortdir;
388:
389: if (e2->rate > e1 -> rate)
390: return sortdir;
391: if (e2->rate < e1 -> rate)
392: return -sortdir;
393: return 0;
394: }
395:
396: int
397: sort_peak_callback(const void *s1, const void *s2)
398: {
399: struct sc_ent *e1 = state_cache[* (u_int32_t *) s1];
400: struct sc_ent *e2 = state_cache[* (u_int32_t *) s2];
401:
402: if (e2 == NULL)
403: return -sortdir;
404: if (e1 == NULL || e2 == NULL)
405: return 0;
406:
407: if (e2->peak > e1 -> peak)
408: return sortdir;
409: if (e2->peak < e1 -> peak)
410: return -sortdir;
411: return 0;
412: }
413:
414: int
415: compare_addr(int af, const struct pf_addr *a, const struct pf_addr *b)
416: {
417: switch (af) {
418: case AF_INET:
419: if (ntohl(a->addr32[0]) > ntohl(b->addr32[0]))
420: return 1;
421: if (a->addr32[0] != b->addr32[0])
422: return -1;
423: break;
424: case AF_INET6:
425: if (ntohl(a->addr32[0]) > ntohl(b->addr32[0]))
426: return 1;
427: if (a->addr32[0] != b->addr32[0])
428: return -1;
429: if (ntohl(a->addr32[1]) > ntohl(b->addr32[1]))
430: return 1;
431: if (a->addr32[1] != b->addr32[1])
432: return -1;
433: if (ntohl(a->addr32[2]) > ntohl(b->addr32[2]))
434: return 1;
435: if (a->addr32[2] != b->addr32[2])
436: return -1;
437: if (ntohl(a->addr32[3]) > ntohl(b->addr32[3]))
438: return 1;
439: if (a->addr32[3] != b->addr32[3])
440: return -1;
441: break;
442: }
443:
444: return 0;
445: }
446:
1.13 jsg 447: static __inline int
1.4 canacar 448: sort_addr_callback(const struct pfsync_state *s1,
449: const struct pfsync_state *s2, int dir)
1.1 canacar 450: {
451: const struct pf_addr *aa, *ab;
452: u_int16_t pa, pb;
453: int af, ret, ii, io;
454:
455: af = s1->af;
456:
457: if (af > s2->af)
458: return sortdir;
459: if (af < s2->af)
460: return -sortdir;
461:
462: ii = io = 0;
463:
464: if (dir == PF_OUT) /* looking for source addr */
465: io = 1;
466: else /* looking for dest addr */
467: ii = 1;
468:
469: if (s1->direction == PF_IN) {
470: aa = &s1->key[PF_SK_STACK].addr[ii];
471: pa = s1->key[PF_SK_STACK].port[ii];
472: } else {
473: aa = &s1->key[PF_SK_WIRE].addr[io];
474: pa = s1->key[PF_SK_WIRE].port[io];
475: }
476:
477: if (s2->direction == PF_IN) {
1.16 deraadt 478: ab = &s2->key[PF_SK_STACK].addr[ii];
1.1 canacar 479: pb = s2->key[PF_SK_STACK].port[ii];
480: } else {
1.16 deraadt 481: ab = &s2->key[PF_SK_WIRE].addr[io];
1.1 canacar 482: pb = s2->key[PF_SK_WIRE].port[io];
483: }
484:
485: ret = compare_addr(af, aa, ab);
486: if (ret)
487: return ret * sortdir;
488:
489: if (ntohs(pa) > ntohs(pb))
490: return sortdir;
491: return -sortdir;
492: }
493:
1.13 jsg 494: static __inline int
1.4 canacar 495: sort_port_callback(const struct pfsync_state *s1,
496: const struct pfsync_state *s2, int dir)
1.1 canacar 497: {
498: const struct pf_addr *aa, *ab;
499: u_int16_t pa, pb;
500: int af, ret, ii, io;
501:
502: af = s1->af;
503:
504:
505: if (af > s2->af)
506: return sortdir;
507: if (af < s2->af)
508: return -sortdir;
509:
510: ii = io = 0;
511:
512: if (dir == PF_OUT) /* looking for source addr */
513: io = 1;
514: else /* looking for dest addr */
515: ii = 1;
516:
517: if (s1->direction == PF_IN) {
518: aa = &s1->key[PF_SK_STACK].addr[ii];
519: pa = s1->key[PF_SK_STACK].port[ii];
520: } else {
521: aa = &s1->key[PF_SK_WIRE].addr[io];
522: pa = s1->key[PF_SK_WIRE].port[io];
523: }
524:
525: if (s2->direction == PF_IN) {
1.16 deraadt 526: ab = &s2->key[PF_SK_STACK].addr[ii];
1.1 canacar 527: pb = s2->key[PF_SK_STACK].port[ii];
528: } else {
1.16 deraadt 529: ab = &s2->key[PF_SK_WIRE].addr[io];
1.1 canacar 530: pb = s2->key[PF_SK_WIRE].port[io];
531: }
532:
533:
534: if (ntohs(pa) > ntohs(pb))
535: return sortdir;
536: if (ntohs(pa) < ntohs(pb))
537: return - sortdir;
538:
539: ret = compare_addr(af, aa, ab);
540: if (ret)
541: return ret * sortdir;
542: return -sortdir;
543: }
544:
545: int
1.4 canacar 546: sort_sa_callback(const void *p1, const void *p2)
1.1 canacar 547: {
1.4 canacar 548: struct pfsync_state *s1 = state_buf + (* (u_int32_t *) p1);
549: struct pfsync_state *s2 = state_buf + (* (u_int32_t *) p2);
550: return sort_addr_callback(s1, s2, PF_OUT);
1.1 canacar 551: }
552:
553: int
1.4 canacar 554: sort_da_callback(const void *p1, const void *p2)
1.1 canacar 555: {
1.4 canacar 556: struct pfsync_state *s1 = state_buf + (* (u_int32_t *) p1);
557: struct pfsync_state *s2 = state_buf + (* (u_int32_t *) p2);
1.1 canacar 558: return sort_addr_callback(s1, s2, PF_IN);
559: }
560:
561: int
562: sort_sp_callback(const void *p1, const void *p2)
563: {
1.4 canacar 564: struct pfsync_state *s1 = state_buf + (* (u_int32_t *) p1);
565: struct pfsync_state *s2 = state_buf + (* (u_int32_t *) p2);
1.1 canacar 566: return sort_port_callback(s1, s2, PF_OUT);
567: }
568:
569: int
570: sort_dp_callback(const void *p1, const void *p2)
571: {
1.4 canacar 572: struct pfsync_state *s1 = state_buf + (* (u_int32_t *) p1);
573: struct pfsync_state *s2 = state_buf + (* (u_int32_t *) p2);
1.1 canacar 574: return sort_port_callback(s1, s2, PF_IN);
575: }
576:
577: void
578: sort_states(void)
579: {
580: order_type *ordering;
581:
582: if (curr_mgr == NULL)
583: return;
584:
585: ordering = curr_mgr->order_curr;
586:
587: if (ordering == NULL)
588: return;
589: if (ordering->func == NULL)
590: return;
591: if (state_buf == NULL)
592: return;
593: if (num_states <= 0)
594: return;
595:
596: mergesort(state_ord, num_states, sizeof(u_int32_t), ordering->func);
597: }
598:
599: /* state management functions */
600:
601: void
602: alloc_buf(int ns)
603: {
604: int len;
605:
606: if (ns < MIN_NUM_STATES)
607: ns = MIN_NUM_STATES;
608:
609: len = ns;
610:
611: if (len >= state_buf_len) {
612: len += NUM_STATE_INC;
1.4 canacar 613: state_buf = realloc(state_buf, len * sizeof(struct pfsync_state));
1.1 canacar 614: state_ord = realloc(state_ord, len * sizeof(u_int32_t));
615: state_cache = realloc(state_cache,
616: len * sizeof(struct sc_ent *));
617: if (state_buf == NULL || state_ord == NULL ||
618: state_cache == NULL)
619: err(1, "realloc");
620: state_buf_len = len;
621: }
622: }
623:
624: int
625: select_states(void)
626: {
627: num_disp = num_states;
628: return (0);
629: }
630:
631: int
632: read_states(void)
633: {
634: struct pfioc_states ps;
635: int n;
636:
637: if (pf_dev == -1)
638: return -1;
639:
640: for (;;) {
1.4 canacar 641: int sbytes = state_buf_len * sizeof(struct pfsync_state);
1.1 canacar 642:
643: ps.ps_len = sbytes;
644: ps.ps_buf = (char *) state_buf;
645:
646: if (ioctl(pf_dev, DIOCGETSTATES, &ps) < 0) {
647: error("DIOCGETSTATES");
648: }
1.4 canacar 649: num_states_all = ps.ps_len / sizeof(struct pfsync_state);
1.1 canacar 650:
651: if (ps.ps_len < sbytes)
652: break;
653:
654: alloc_buf(num_states_all);
655: }
656:
657: if (dumpfilter) {
658: int fd = open("state.dmp", O_WRONLY|O_CREAT|O_EXCL, 0);
659: if (fd > 0) {
660: write(fd, state_buf, ps.ps_len);
661: close(fd);
662: }
663: }
664:
665: num_states = num_states_all;
666: for (n = 0; n<num_states_all; n++)
667: state_ord[n] = n;
668:
669: if (cachestates) {
670: for (n = 0; n < num_states; n++)
671: state_cache[n] = cache_state(state_buf + n);
672: cache_endupdate();
673: }
674:
675: num_disp = num_states;
676: return 0;
677: }
678:
679: int
680: unmask(struct pf_addr * m, u_int8_t af)
681: {
682: int i = 31, j = 0, b = 0, msize;
683: u_int32_t tmp;
684:
685: if (af == AF_INET)
686: msize = 1;
687: else
688: msize = 4;
689: while (j < msize && m->addr32[j] == 0xffffffff) {
690: b += 32;
691: j++;
692: }
693: if (j < msize) {
694: tmp = ntohl(m->addr32[j]);
695: for (i = 31; tmp & (1 << i); --i)
696: b++;
697: }
698: return (b);
699: }
700:
701: /* display functions */
702:
703: void
704: tb_print_addr(struct pf_addr * addr, struct pf_addr * mask, int af)
705: {
1.15 giovanni 706: switch (af) {
707: case AF_INET: {
708: tbprintf("%s", inetname(addr->v4));
709: break;
710: }
711: case AF_INET6: {
712: tbprintf("%s", inet6name(&addr->v6));
713: break;
714: }
715: }
1.1 canacar 716:
717: if (mask != NULL) {
718: if (!PF_AZERO(mask, af))
719: tbprintf("/%u", unmask(mask, af));
720: }
721: }
1.4 canacar 722:
1.1 canacar 723: void
724: print_fld_host2(field_def *fld, struct pfsync_state_key *ks,
725: struct pfsync_state_key *kn, int idx, int af)
726: {
727: struct pf_addr *as = &ks->addr[idx];
728: struct pf_addr *an = &kn->addr[idx];
729:
730: u_int16_t ps = ntohs(ks->port[idx]);
731: u_int16_t pn = ntohs(kn->port[idx]);
732:
733: if (fld == NULL)
734: return;
735:
736: if (fld->width < 3) {
737: print_fld_str(fld, "*");
738: return;
739: }
740:
741: tb_start();
742: tb_print_addr(as, NULL, af);
743:
744: if (af == AF_INET)
745: tbprintf(":%u", ps);
746: else
747: tbprintf("[%u]", ps);
748:
749: print_fld_tb(fld);
750:
751: if (PF_ANEQ(as, an, af) || ps != pn) {
752: tb_start();
753: tb_print_addr(an, NULL, af);
754:
755: if (af == AF_INET)
756: tbprintf(":%u", pn);
757: else
758: tbprintf("[%u]", pn);
759: print_fld_tb(FLD_GW);
760: }
761:
762: }
763:
764: void
765: print_fld_state(field_def *fld, unsigned int proto,
766: unsigned int s1, unsigned int s2)
767: {
768: int len;
769:
770: if (fld == NULL)
771: return;
772:
773: len = fld->width;
774: if (len < 1)
775: return;
776:
777: tb_start();
778:
779: if (proto == IPPROTO_TCP) {
780: if (s1 <= TCPS_TIME_WAIT && s2 <= TCPS_TIME_WAIT)
781: tbprintf("%s:%s", tcpstates[s1], tcpstates[s2]);
782: #ifdef PF_TCPS_PROXY_SRC
783: else if (s1 == PF_TCPS_PROXY_SRC ||
784: s2 == PF_TCPS_PROXY_SRC)
785: tbprintf("PROXY:SRC\n");
786: else if (s1 == PF_TCPS_PROXY_DST ||
787: s2 == PF_TCPS_PROXY_DST)
788: tbprintf("PROXY:DST\n");
789: #endif
790: else
791: tbprintf("<BAD STATE LEVELS>");
792: } else if (proto == IPPROTO_UDP && s1 < PFUDPS_NSTATES &&
793: s2 < PFUDPS_NSTATES) {
794: const char *states[] = PFUDPS_NAMES;
795: tbprintf("%s:%s", states[s1], states[s2]);
796: } else if (proto != IPPROTO_ICMP && s1 < PFOTHERS_NSTATES &&
797: s2 < PFOTHERS_NSTATES) {
798: /* XXX ICMP doesn't really have state levels */
799: const char *states[] = PFOTHERS_NAMES;
800: tbprintf("%s:%s", states[s1], states[s2]);
801: } else {
802: tbprintf("%u:%u", s1, s2);
803: }
804:
805: if (strlen(tmp_buf) > len) {
806: tb_start();
807: tbprintf("%u:%u", s1, s2);
808: }
809:
810: print_fld_tb(fld);
811: }
812:
813: int
1.4 canacar 814: print_state(struct pfsync_state * s, struct sc_ent * ent)
1.1 canacar 815: {
1.4 canacar 816: struct pfsync_state_peer *src, *dst;
1.1 canacar 817: struct protoent *p;
1.4 canacar 818: u_int64_t sz;
1.1 canacar 819:
820: if (s->direction == PF_OUT) {
821: src = &s->src;
822: dst = &s->dst;
823: } else {
824: src = &s->dst;
825: dst = &s->src;
826: }
827:
828: p = getprotobynumber(s->proto);
829:
830: if (p != NULL)
831: print_fld_str(FLD_PROTO, p->p_name);
832: else
833: print_fld_uint(FLD_PROTO, s->proto);
834:
835: if (s->direction == PF_OUT) {
836: print_fld_host2(FLD_SRC, &s->key[PF_SK_WIRE],
837: &s->key[PF_SK_STACK], 1, s->af);
838: print_fld_host2(FLD_DEST, &s->key[PF_SK_WIRE],
839: &s->key[PF_SK_STACK], 0, s->af);
840: } else {
841: print_fld_host2(FLD_SRC, &s->key[PF_SK_STACK],
842: &s->key[PF_SK_WIRE], 0, s->af);
843: print_fld_host2(FLD_DEST, &s->key[PF_SK_STACK],
844: &s->key[PF_SK_WIRE], 1, s->af);
845: }
846:
847: if (s->direction == PF_OUT)
848: print_fld_str(FLD_DIR, "Out");
849: else
850: print_fld_str(FLD_DIR, "In");
851:
852: print_fld_state(FLD_STATE, s->proto, src->state, dst->state);
1.3 mcbride 853: print_fld_age(FLD_AGE, ntohl(s->creation));
854: print_fld_age(FLD_EXP, ntohl(s->expire));
1.4 canacar 855:
856: sz = COUNTER(s->bytes[0]) + COUNTER(s->bytes[1]);
857:
858: print_fld_size(FLD_PKTS, COUNTER(s->packets[0]) +
859: COUNTER(s->packets[1]));
860: print_fld_size(FLD_BYTES, sz);
1.3 mcbride 861: print_fld_rate(FLD_SA, (s->creation) ?
1.4 canacar 862: ((double)sz/ntohl((double)s->creation)) : -1);
1.1 canacar 863:
1.9 canacar 864: print_fld_uint(FLD_RULE, ntohl(s->rule));
1.1 canacar 865: if (cachestates && ent != NULL) {
866: print_fld_rate(FLD_SI, ent->rate);
867: print_fld_rate(FLD_SP, ent->peak);
868: }
869:
870: end_line();
871: return 1;
872: }
873:
874: void
875: print_states(void)
876: {
877: int n, count = 0;
878:
879: for (n = dispstart; n < num_disp; n++) {
880: count += print_state(state_buf + state_ord[n],
881: state_cache[state_ord[n]]);
882: if (maxprint > 0 && count >= maxprint)
883: break;
884: }
885: }
886:
887: /* rule display */
888:
889: struct pf_rule *rules = NULL;
890: u_int32_t alloc_rules = 0;
891:
892: int
893: select_rules(void)
894: {
895: num_disp = num_rules;
896: return (0);
897: }
898:
899:
900: void
901: add_rule_alloc(u_int32_t nr)
902: {
903: if (nr == 0)
904: return;
905:
906: num_rules += nr;
907:
908: if (rules == NULL) {
909: rules = malloc(num_rules * sizeof(struct pf_rule));
910: if (rules == NULL)
911: err(1, "malloc");
912: alloc_rules = num_rules;
913: } else if (num_rules > alloc_rules) {
914: rules = realloc(rules, num_rules * sizeof(struct pf_rule));
915: if (rules == NULL)
916: err(1, "realloc");
917: alloc_rules = num_rules;
918: }
919: }
920:
921: int label_length;
922:
923: int
924: read_anchor_rules(char *anchor)
925: {
926: struct pfioc_rule pr;
927: u_int32_t nr, num, off;
1.4 canacar 928: int len;
1.1 canacar 929:
930: if (pf_dev < 0)
931: return (-1);
932:
933: memset(&pr, 0, sizeof(pr));
934: strlcpy(pr.anchor, anchor, sizeof(pr.anchor));
1.4 canacar 935:
1.1 canacar 936: if (ioctl(pf_dev, DIOCGETRULES, &pr)) {
937: error("anchor %s: %s", anchor, strerror(errno));
938: return (-1);
939: }
940:
941: off = num_rules;
942: num = pr.nr;
943: add_rule_alloc(num);
944:
945: for (nr = 0; nr < num; ++nr) {
946: pr.nr = nr;
947: if (ioctl(pf_dev, DIOCGETRULE, &pr)) {
948: error("DIOCGETRULE: %s", strerror(errno));
949: return (-1);
950: }
951: /* XXX overload pr.anchor, to store a pointer to
952: * anchor name */
953: pr.rule.anchor = (struct pf_anchor *) anchor;
1.4 canacar 954: len = strlen(pr.rule.label);
955: if (len > label_length)
956: label_length = len;
1.1 canacar 957: rules[off + nr] = pr.rule;
958: }
959:
960: return (num);
961: }
962:
963: struct anchor_name {
964: char name[MAXPATHLEN];
965: struct anchor_name *next;
966: u_int32_t ref;
967: };
968:
969: struct anchor_name *anchor_root = NULL;
970: struct anchor_name *anchor_end = NULL;
971: struct anchor_name *anchor_free = NULL;
972:
973: struct anchor_name*
974: alloc_anchor_name(const char *path)
975: {
976: struct anchor_name *a;
977:
978: a = anchor_free;
979: if (a == NULL) {
980: a = (struct anchor_name *)malloc(sizeof(struct anchor_name));
981: if (a == NULL)
982: return (NULL);
983: } else
984: anchor_free = a->next;
985:
986: if (anchor_root == NULL)
987: anchor_end = a;
988:
989: a->next = anchor_root;
990: anchor_root = a;
991:
992: a->ref = 0;
993: strlcpy(a->name, path, sizeof(a->name));
994: return (a);
995: }
996:
997: void
998: reset_anchor_names(void)
999: {
1000: if (anchor_end == NULL)
1001: return;
1002:
1003: anchor_end->next = anchor_free;
1004: anchor_free = anchor_root;
1005: anchor_root = anchor_end = NULL;
1006: }
1007:
1008: struct pfioc_ruleset ruleset;
1009: char *rs_end = NULL;
1010:
1011: int
1012: read_rulesets(const char *path)
1013: {
1014: char *pre;
1015: struct anchor_name *a;
1016: u_int32_t nr, ns;
1017: int len;
1018:
1019: if (path == NULL)
1020: ruleset.path[0] = '\0';
1021: else if (strlcpy(ruleset.path, path, sizeof(ruleset.path)) >=
1022: sizeof(ruleset.path))
1023: return (-1);
1024:
1025: /* a persistent storage for anchor names */
1026: a = alloc_anchor_name(ruleset.path);
1027: if (a == NULL)
1028: return (-1);
1029:
1030: len = read_anchor_rules(a->name);
1031: if (len < 0)
1032: return (-1);
1033:
1034: a->ref += len;
1035:
1036: if (ioctl(pf_dev, DIOCGETRULESETS, &ruleset)) {
1037: error("DIOCGETRULESETS: %s", strerror(errno));
1038: return (-1);
1039: }
1040:
1041: ns = ruleset.nr;
1042:
1043: if (rs_end == NULL)
1044: rs_end = ruleset.path + sizeof(ruleset.path);
1045:
1046: /* 'pre' tracks the previous level on the anchor */
1047: pre = strchr(ruleset.path, 0);
1048: len = rs_end - pre;
1049: if (len < 1)
1050: return (-1);
1051: --len;
1052:
1053: for (nr = 0; nr < ns; ++nr) {
1054: ruleset.nr = nr;
1055: if (ioctl(pf_dev, DIOCGETRULESET, &ruleset)) {
1056: error("DIOCGETRULESET: %s", strerror(errno));
1057: return (-1);
1058: }
1059: *pre = '/';
1060: if (strlcpy(pre + 1, ruleset.name, len) < len)
1061: read_rulesets(ruleset.path);
1062: *pre = '\0';
1063: }
1064:
1065: return (0);
1066: }
1067:
1068: void
1069: compute_anchor_field(void)
1070: {
1071: struct anchor_name *a;
1072: int sum, cnt, mx, nx;
1073: sum = cnt = mx = 0;
1074:
1075: for (a = anchor_root; a != NULL; a = a->next, cnt++) {
1076: int len;
1077: if (a->ref == 0)
1078: continue;
1079: len = strlen(a->name);
1080: sum += len;
1081: if (len > mx)
1082: mx = len;
1083: }
1084:
1085: nx = sum/cnt;
1086: if (nx < ANCHOR_FLD_SIZE)
1087: nx = (mx < ANCHOR_FLD_SIZE) ? mx : ANCHOR_FLD_SIZE;
1088:
1089: if (FLD_ANCHOR->max_width != mx ||
1090: FLD_ANCHOR->norm_width != nx) {
1091: FLD_ANCHOR->max_width = mx;
1092: FLD_ANCHOR->norm_width = nx;
1093: field_setup();
1094: need_update = 1;
1095: }
1096: }
1097:
1098: int
1099: read_rules(void)
1100: {
1.4 canacar 1101: int ret, nw, mw;
1.1 canacar 1102: num_rules = 0;
1103:
1104: if (pf_dev == -1)
1105: return (-1);
1106:
1107: label_length = MIN_LABEL_SIZE;
1108:
1109: reset_anchor_names();
1110: ret = read_rulesets(NULL);
1111: compute_anchor_field();
1112:
1.4 canacar 1113: nw = mw = label_length;
1114: if (nw > 16)
1115: nw = 16;
1116:
1117: if (FLD_LABEL->norm_width != nw ||
1118: FLD_LABEL->max_width != mw) {
1119: FLD_LABEL->norm_width = nw;
1120: FLD_LABEL->max_width = mw;
1121: field_setup();
1122: need_update = 1;
1.1 canacar 1123: }
1124:
1125: num_disp = num_rules;
1126: return (ret);
1127: }
1128:
1129: void
1130: tb_print_addrw(struct pf_addr_wrap *addr, struct pf_addr *mask, u_int8_t af)
1131: {
1132: switch (addr->type) {
1133: case PF_ADDR_ADDRMASK:
1134: tb_print_addr(&addr->v.a.addr, mask, af);
1135: break;
1136: case PF_ADDR_NOROUTE:
1137: tbprintf("noroute");
1138: break;
1139: case PF_ADDR_DYNIFTL:
1140: tbprintf("(%s)", addr->v.ifname);
1141: break;
1142: case PF_ADDR_TABLE:
1143: tbprintf("<%s>", addr->v.tblname);
1144: break;
1145: default:
1146: tbprintf("UNKNOWN");
1147: break;
1148: }
1149: }
1150:
1151: void
1152: tb_print_op(u_int8_t op, const char *a1, const char *a2)
1153: {
1154: if (op == PF_OP_IRG)
1155: tbprintf("%s >< %s ", a1, a2);
1156: else if (op == PF_OP_XRG)
1157: tbprintf("%s <> %s ", a1, a2);
1158: else if (op == PF_OP_RRG)
1159: tbprintf("%s:%s ", a1, a2);
1160: else if (op == PF_OP_EQ)
1161: tbprintf("= %s ", a1);
1162: else if (op == PF_OP_NE)
1163: tbprintf("!= %s ", a1);
1164: else if (op == PF_OP_LT)
1165: tbprintf("< %s ", a1);
1166: else if (op == PF_OP_LE)
1167: tbprintf("<= %s ", a1);
1168: else if (op == PF_OP_GT)
1169: tbprintf("> %s ", a1);
1170: else if (op == PF_OP_GE)
1171: tbprintf(">= %s ", a1);
1172: }
1173:
1174: void
1175: tb_print_port(u_int8_t op, u_int16_t p1, u_int16_t p2, char *proto)
1176: {
1177: char a1[6], a2[6];
1178: struct servent *s = getservbyport(p1, proto);
1179:
1180: p1 = ntohs(p1);
1181: p2 = ntohs(p2);
1182: snprintf(a1, sizeof(a1), "%u", p1);
1183: snprintf(a2, sizeof(a2), "%u", p2);
1184: tbprintf("port ");
1185: if (s != NULL && (op == PF_OP_EQ || op == PF_OP_NE))
1186: tb_print_op(op, s->s_name, a2);
1187: else
1188: tb_print_op(op, a1, a2);
1189: }
1190:
1191: void
1192: tb_print_fromto(struct pf_rule_addr *src, struct pf_rule_addr *dst,
1193: u_int8_t af, u_int8_t proto)
1194: {
1195: if (
1196: PF_AZERO(PT_ADDR(src), AF_INET6) &&
1197: PF_AZERO(PT_ADDR(dst), AF_INET6) &&
1198: ! PT_NOROUTE(src) && ! PT_NOROUTE(dst) &&
1199: PF_AZERO(PT_MASK(src), AF_INET6) &&
1200: PF_AZERO(PT_MASK(dst), AF_INET6) &&
1201: !src->port_op && !dst->port_op)
1202: tbprintf("all ");
1203: else {
1204: tbprintf("from ");
1205: if (PT_NOROUTE(src))
1206: tbprintf("no-route ");
1207: else if (PF_AZERO(PT_ADDR(src), AF_INET6) &&
1208: PF_AZERO(PT_MASK(src), AF_INET6))
1209: tbprintf("any ");
1210: else {
1211: if (src->neg)
1212: tbprintf("! ");
1213: tb_print_addrw(&src->addr, PT_MASK(src), af);
1214: tbprintf(" ");
1215: }
1216: if (src->port_op)
1217: tb_print_port(src->port_op, src->port[0],
1218: src->port[1],
1219: proto == IPPROTO_TCP ? "tcp" : "udp");
1220:
1221: tbprintf("to ");
1222: if (PT_NOROUTE(dst))
1223: tbprintf("no-route ");
1224: else if (PF_AZERO(PT_ADDR(dst), AF_INET6) &&
1225: PF_AZERO(PT_MASK(dst), AF_INET6))
1226: tbprintf("any ");
1227: else {
1228: if (dst->neg)
1229: tbprintf("! ");
1230: tb_print_addrw(&dst->addr, PT_MASK(dst), af);
1231: tbprintf(" ");
1232: }
1233: if (dst->port_op)
1234: tb_print_port(dst->port_op, dst->port[0],
1235: dst->port[1],
1236: proto == IPPROTO_TCP ? "tcp" : "udp");
1237: }
1238: }
1239:
1240: void
1241: tb_print_ugid(u_int8_t op, unsigned u1, unsigned u2,
1242: const char *t, unsigned umax)
1243: {
1244: char a1[11], a2[11];
1245:
1246: snprintf(a1, sizeof(a1), "%u", u1);
1247: snprintf(a2, sizeof(a2), "%u", u2);
1248:
1249: tbprintf("%s ", t);
1250: if (u1 == umax && (op == PF_OP_EQ || op == PF_OP_NE))
1251: tb_print_op(op, "unknown", a2);
1252: else
1253: tb_print_op(op, a1, a2);
1254: }
1255:
1256: void
1257: tb_print_flags(u_int8_t f)
1258: {
1259: const char *tcpflags = "FSRPAUEW";
1260: int i;
1261:
1262: for (i = 0; tcpflags[i]; ++i)
1263: if (f & (1 << i))
1264: tbprintf("%c", tcpflags[i]);
1265: }
1266:
1267: void
1268: print_rule(struct pf_rule *pr)
1269: {
1.11 henning 1270: static const char *actiontypes[] = { "Pass", "Block", "Scrub",
1271: "no Scrub", "Nat", "no Nat", "Binat", "no Binat", "Rdr",
1272: "no Rdr", "SynProxy Block", "Defer", "Match" };
1.1 canacar 1273: int numact = sizeof(actiontypes) / sizeof(char *);
1274:
1275: static const char *routetypes[] = { "", "fastroute", "route-to",
1276: "dup-to", "reply-to" };
1277:
1278: int numroute = sizeof(routetypes) / sizeof(char *);
1279:
1280: if (pr == NULL) return;
1281:
1282: print_fld_str(FLD_LABEL, pr->label);
1283: print_fld_size(FLD_STATS, pr->states_tot);
1284:
1285: print_fld_size(FLD_PKTS, pr->packets[0] + pr->packets[1]);
1286: print_fld_size(FLD_BYTES, pr->bytes[0] + pr->bytes[1]);
1.4 canacar 1287:
1.1 canacar 1288: print_fld_uint(FLD_RULE, pr->nr);
1.5 sthen 1289: if (pr->direction == PF_OUT)
1290: print_fld_str(FLD_DIR, "Out");
1291: else if (pr->direction == PF_IN)
1292: print_fld_str(FLD_DIR, "In");
1293: else
1294: print_fld_str(FLD_DIR, "Any");
1295:
1.1 canacar 1296: if (pr->quick)
1297: print_fld_str(FLD_QUICK, "Quick");
1298:
1299: if (pr->keep_state == PF_STATE_NORMAL)
1300: print_fld_str(FLD_KST, "Keep");
1301: else if (pr->keep_state == PF_STATE_MODULATE)
1302: print_fld_str(FLD_KST, "Mod");
1303: #ifdef PF_STATE_SYNPROXY
1304: else if (pr->keep_state == PF_STATE_MODULATE)
1305: print_fld_str(FLD_KST, "Syn");
1306: #endif
1307: if (pr->log == 1)
1308: print_fld_str(FLD_LOG, "Log");
1309: else if (pr->log == 2)
1310: print_fld_str(FLD_LOG, "All");
1311:
1.12 canacar 1312: if (pr->action >= numact)
1.1 canacar 1313: print_fld_uint(FLD_ACTION, pr->action);
1314: else print_fld_str(FLD_ACTION, actiontypes[pr->action]);
1.12 canacar 1315:
1.1 canacar 1316: if (pr->proto) {
1317: struct protoent *p = getprotobynumber(pr->proto);
1318:
1319: if (p != NULL)
1320: print_fld_str(FLD_PROTO, p->p_name);
1321: else
1322: print_fld_uint(FLD_PROTO, pr->proto);
1323: }
1324:
1325: if (pr->ifname[0]) {
1326: tb_start();
1327: if (pr->ifnot)
1328: tbprintf("!");
1329: tbprintf("%s", pr->ifname);
1330: print_fld_tb(FLD_IF);
1331: }
1332: if (pr->max_states)
1333: print_fld_uint(FLD_STMAX, pr->max_states);
1.4 canacar 1334:
1.1 canacar 1335: /* print info field */
1336:
1337: tb_start();
1.4 canacar 1338:
1.1 canacar 1339: if (pr->action == PF_DROP) {
1340: if (pr->rule_flag & PFRULE_RETURNRST)
1341: tbprintf("return-rst ");
1342: #ifdef PFRULE_RETURN
1343: else if (pr->rule_flag & PFRULE_RETURN)
1344: tbprintf("return ");
1345: #endif
1346: #ifdef PFRULE_RETURNICMP
1347: else if (pr->rule_flag & PFRULE_RETURNICMP)
1348: tbprintf("return-icmp ");
1349: #endif
1350: else
1351: tbprintf("drop ");
1352: }
1353:
1354: if (pr->rt > 0 && pr->rt < numroute) {
1355: tbprintf("%s ", routetypes[pr->rt]);
1356: if (pr->rt != PF_FASTROUTE)
1357: tbprintf("... ");
1358: }
1.4 canacar 1359:
1.1 canacar 1360: if (pr->af) {
1361: if (pr->af == AF_INET)
1362: tbprintf("inet ");
1363: else
1364: tbprintf("inet6 ");
1365: }
1366:
1367: tb_print_fromto(&pr->src, &pr->dst, pr->af, pr->proto);
1.4 canacar 1368:
1.1 canacar 1369: if (pr->uid.op)
1370: tb_print_ugid(pr->uid.op, pr->uid.uid[0], pr->uid.uid[1],
1371: "user", UID_MAX);
1372: if (pr->gid.op)
1373: tb_print_ugid(pr->gid.op, pr->gid.gid[0], pr->gid.gid[1],
1374: "group", GID_MAX);
1375:
1.8 mcbride 1376: if (pr->action == PF_PASS &&
1377: (pr->proto == 0 || pr->proto == IPPROTO_TCP) &&
1378: (pr->flags != TH_SYN || pr->flagset != (TH_SYN | TH_ACK) )) {
1379: tbprintf("flags ");
1380: if (pr->flags || pr->flagset) {
1381: tb_print_flags(pr->flags);
1382: tbprintf("/");
1383: tb_print_flags(pr->flagset);
1384: } else
1385: tbprintf("any ");
1.1 canacar 1386: }
1387:
1388: tbprintf(" ");
1389:
1390: if (pr->tos)
1391: tbprintf("tos 0x%2.2x ", pr->tos);
1392: #ifdef PFRULE_FRAGMENT
1393: if (pr->rule_flag & PFRULE_FRAGMENT)
1394: tbprintf("fragment ");
1395: #endif
1396: #ifdef PFRULE_NODF
1397: if (pr->rule_flag & PFRULE_NODF)
1398: tbprintf("no-df ");
1399: #endif
1400: #ifdef PFRULE_RANDOMID
1401: if (pr->rule_flag & PFRULE_RANDOMID)
1402: tbprintf("random-id ");
1403: #endif
1404: if (pr->min_ttl)
1405: tbprintf("min-ttl %d ", pr->min_ttl);
1406: if (pr->max_mss)
1407: tbprintf("max-mss %d ", pr->max_mss);
1408: if (pr->allow_opts)
1409: tbprintf("allow-opts ");
1410:
1.10 henning 1411: /* XXX more missing */
1.1 canacar 1412:
1413: if (pr->qname[0] && pr->pqname[0])
1414: tbprintf("queue(%s, %s) ", pr->qname, pr->pqname);
1415: else if (pr->qname[0])
1416: tbprintf("queue %s ", pr->qname);
1.4 canacar 1417:
1.1 canacar 1418: if (pr->tagname[0])
1419: tbprintf("tag %s ", pr->tagname);
1420: if (pr->match_tagname[0]) {
1421: if (pr->match_tag_not)
1422: tbprintf("! ");
1423: tbprintf("tagged %s ", pr->match_tagname);
1424: }
1.4 canacar 1425:
1.1 canacar 1426: print_fld_tb(FLD_RINFO);
1427:
1428: /* XXX anchor field overloaded with anchor name */
1429: print_fld_str(FLD_ANCHOR, (char *)pr->anchor);
1430: tb_end();
1431:
1432: end_line();
1433: }
1434:
1435: void
1436: print_rules(void)
1437: {
1438: u_int32_t n, count = 0;
1439:
1440: for (n = dispstart; n < num_rules; n++) {
1441: print_rule(rules + n);
1442: count ++;
1443: if (maxprint > 0 && count >= maxprint)
1444: break;
1445: }
1446: }
1447:
1448: /* queue display */
1449:
1450: struct pf_altq_node *
1451: pfctl_find_altq_node(struct pf_altq_node *root, const char *qname,
1452: const char *ifname)
1453: {
1454: struct pf_altq_node *node, *child;
1455:
1456: for (node = root; node != NULL; node = node->next) {
1457: if (!strcmp(node->altq.qname, qname)
1458: && !(strcmp(node->altq.ifname, ifname)))
1459: return (node);
1460: if (node->children != NULL) {
1461: child = pfctl_find_altq_node(node->children, qname,
1462: ifname);
1463: if (child != NULL)
1464: return (child);
1465: }
1466: }
1467: return (NULL);
1468: }
1469:
1470: void
1471: pfctl_insert_altq_node(struct pf_altq_node **root,
1472: const struct pf_altq altq, const struct queue_stats qstats)
1473: {
1474: struct pf_altq_node *node;
1475:
1476: node = calloc(1, sizeof(struct pf_altq_node));
1477: if (node == NULL)
1478: err(1, "pfctl_insert_altq_node: calloc");
1479: memcpy(&node->altq, &altq, sizeof(struct pf_altq));
1480: memcpy(&node->qstats, &qstats, sizeof(qstats));
1481: node->next = node->children = node->next_flat = NULL;
1482: node->depth = 0;
1483: node->visited = 1;
1484:
1485: if (*root == NULL)
1486: *root = node;
1487: else if (!altq.parent[0]) {
1488: struct pf_altq_node *prev = *root;
1489:
1490: while (prev->next != NULL)
1491: prev = prev->next;
1492: prev->next = node;
1493: } else {
1494: struct pf_altq_node *parent;
1495:
1496: parent = pfctl_find_altq_node(*root, altq.parent, altq.ifname);
1497: if (parent == NULL)
1498: errx(1, "parent %s not found", altq.parent);
1499: node->depth = parent->depth+1;
1500: if (parent->children == NULL)
1501: parent->children = node;
1502: else {
1503: struct pf_altq_node *prev = parent->children;
1504:
1505: while (prev->next != NULL)
1506: prev = prev->next;
1507: prev->next = node;
1508: }
1509: }
1.6 canacar 1510: }
1511:
1512: void
1513: pfctl_set_next_flat(struct pf_altq_node *node, struct pf_altq_node *up)
1514: {
1515: while (node) {
1516: struct pf_altq_node *next = node->next ? node->next : up;
1517: if (node->children) {
1518: node->next_flat = node->children;
1519: pfctl_set_next_flat(node->children, next);
1520: } else
1521: node->next_flat = next;
1522: node = node->next;
1.1 canacar 1523: }
1524: }
1525:
1526: int
1527: pfctl_update_qstats(struct pf_altq_node **root, int *inserts)
1528: {
1529: struct pf_altq_node *node;
1530: struct pfioc_altq pa;
1531: struct pfioc_qstats pq;
1532: u_int32_t nr;
1533: struct queue_stats qstats;
1534: u_int32_t nr_queues;
1.6 canacar 1535: int ret = 0;
1.1 canacar 1536:
1537: *inserts = 0;
1538: memset(&pa, 0, sizeof(pa));
1539: memset(&pq, 0, sizeof(pq));
1540: memset(&qstats, 0, sizeof(qstats));
1541:
1542: if (pf_dev < 0)
1543: return (-1);
1544:
1545: if (ioctl(pf_dev, DIOCGETALTQS, &pa)) {
1546: error("DIOCGETALTQS: %s", strerror(errno));
1547: return (-1);
1548: }
1.6 canacar 1549:
1.1 canacar 1550: num_queues = nr_queues = pa.nr;
1551: for (nr = 0; nr < nr_queues; ++nr) {
1552: pa.nr = nr;
1553: if (ioctl(pf_dev, DIOCGETALTQ, &pa)) {
1554: error("DIOCGETALTQ: %s", strerror(errno));
1.6 canacar 1555: ret = -1;
1556: break;
1.1 canacar 1557: }
1558: if (pa.altq.qid > 0) {
1559: pq.nr = nr;
1560: pq.ticket = pa.ticket;
1561: pq.buf = &qstats;
1562: pq.nbytes = sizeof(qstats);
1563: if (ioctl(pf_dev, DIOCGETQSTATS, &pq)) {
1564: error("DIOCGETQSTATS: %s", strerror(errno));
1.6 canacar 1565: ret = -1;
1566: break;
1.1 canacar 1567: }
1568: qstats.valid = 1;
1569: gettimeofday(&qstats.timestamp, NULL);
1570: if ((node = pfctl_find_altq_node(*root, pa.altq.qname,
1571: pa.altq.ifname)) != NULL) {
1.7 canacar 1572: /* update altq data too as bandwidth may have changed */
1.1 canacar 1573: memcpy(&node->altq, &pa.altq, sizeof(struct pf_altq));
1574: memcpy(&node->qstats_last, &node->qstats,
1575: sizeof(struct queue_stats));
1576: memcpy(&node->qstats, &qstats,
1577: sizeof(qstats));
1578: node->visited = 1;
1579: } else {
1580: pfctl_insert_altq_node(root, pa.altq, qstats);
1581: *inserts = 1;
1582: }
1583: }
1584: else
1585: --num_queues;
1586: }
1.6 canacar 1587:
1588: pfctl_set_next_flat(*root, NULL);
1589:
1590: return (ret);
1.1 canacar 1591: }
1592:
1593: void
1594: pfctl_free_altq_node(struct pf_altq_node *node)
1595: {
1596: while (node != NULL) {
1597: struct pf_altq_node *prev;
1598:
1599: if (node->children != NULL)
1600: pfctl_free_altq_node(node->children);
1601: prev = node;
1602: node = node->next;
1603: free(prev);
1604: }
1605: }
1606:
1607: void
1608: pfctl_mark_all_unvisited(struct pf_altq_node *root)
1609: {
1610: if (root != NULL) {
1611: struct pf_altq_node *node = root;
1612: while (node != NULL) {
1613: node->visited = 0;
1614: node = node->next_flat;
1615: }
1616: }
1617: }
1618:
1619: int
1620: pfctl_have_unvisited(struct pf_altq_node *root)
1621: {
1622: if (root == NULL)
1623: return(0);
1624: else {
1625: struct pf_altq_node *node = root;
1626: while (node != NULL) {
1627: if (node->visited == 0)
1628: return(1);
1629: node = node->next_flat;
1630: }
1631: return(0);
1632: }
1633: }
1634:
1635: struct pf_altq_node *altq_root = NULL;
1636:
1637: int
1638: select_queues(void)
1639: {
1640: num_disp = num_queues;
1641: return (0);
1642: }
1643:
1644: int
1645: read_queues(void)
1646: {
1647: static int first_read = 1;
1648: int inserts;
1649: num_disp = num_queues = 0;
1650:
1651: pfctl_mark_all_unvisited(altq_root);
1652: if (pfctl_update_qstats(&altq_root, &inserts))
1653: return (-1);
1654:
1.7 canacar 1655: /* Allow inserts only on first read;
1656: * on subsequent reads clear and reload
1657: */
1.1 canacar 1658: if (first_read == 0 &&
1659: (inserts != 0 || pfctl_have_unvisited(altq_root) != 0)) {
1660: pfctl_free_altq_node(altq_root);
1661: altq_root = NULL;
1662: first_read = 1;
1663: if (pfctl_update_qstats(&altq_root, &inserts))
1664: return (-1);
1665: }
1666:
1667: first_read = 0;
1668: num_disp = num_queues;
1669:
1670: return(0);
1671: }
1672:
1673: double
1674: calc_interval(struct timeval *cur_time, struct timeval *last_time)
1675: {
1676: double sec;
1677:
1678: sec = (double)(cur_time->tv_sec - last_time->tv_sec) +
1679: (double)(cur_time->tv_usec - last_time->tv_usec) / 1000000;
1680:
1681: return (sec);
1682: }
1683:
1684: double
1685: calc_rate(u_int64_t new_bytes, u_int64_t last_bytes, double interval)
1686: {
1687: double rate;
1688:
1689: rate = (double)(new_bytes - last_bytes) / interval;
1690: return (rate);
1691: }
1692:
1693: double
1694: calc_pps(u_int64_t new_pkts, u_int64_t last_pkts, double interval)
1695: {
1696: double pps;
1697:
1698: pps = (double)(new_pkts - last_pkts) / interval;
1699: return (pps);
1700: }
1701:
1702: #define DEFAULT_PRIORITY 1
1703:
1704: void
1705: print_queue(struct pf_altq_node *node)
1706: {
1707: u_int8_t d;
1708: double interval, pps, bps;
1709: pps = bps = 0;
1710:
1711: tb_start();
1712: for (d = 0; d < node->depth; d++)
1713: tbprintf(" ");
1714: tbprintf(node->altq.qname);
1715: print_fld_tb(FLD_QUEUE);
1716:
1717: if (node->altq.scheduler == ALTQT_CBQ ||
1718: node->altq.scheduler == ALTQT_HFSC
1719: )
1720: print_fld_bw(FLD_BANDW, (double)node->altq.bandwidth);
1721:
1722: if (node->altq.priority != DEFAULT_PRIORITY)
1723: print_fld_uint(FLD_PRIO,
1724: node->altq.priority);
1725:
1726: if (node->qstats.valid && node->qstats_last.valid)
1727: interval = calc_interval(&node->qstats.timestamp,
1728: &node->qstats_last.timestamp);
1729: else
1730: interval = 0;
1731:
1732: switch (node->altq.scheduler) {
1733: case ALTQT_CBQ:
1734: print_fld_str(FLD_SCHED, "cbq");
1735: print_fld_size(FLD_PKTS,
1736: node->qstats.data.cbq_stats.xmit_cnt.packets);
1737: print_fld_size(FLD_BYTES,
1738: node->qstats.data.cbq_stats.xmit_cnt.bytes);
1739: print_fld_size(FLD_DROPP,
1740: node->qstats.data.cbq_stats.drop_cnt.packets);
1741: print_fld_size(FLD_DROPB,
1742: node->qstats.data.cbq_stats.drop_cnt.bytes);
1743: print_fld_size(FLD_QLEN, node->qstats.data.cbq_stats.qcnt);
1744: print_fld_size(FLD_BORR, node->qstats.data.cbq_stats.borrows);
1745: print_fld_size(FLD_SUSP, node->qstats.data.cbq_stats.delays);
1746: if (interval > 0) {
1747: pps = calc_pps(node->qstats.data.cbq_stats.xmit_cnt.packets,
1748: node->qstats_last.data.cbq_stats.xmit_cnt.packets, interval);
1749: bps = calc_rate(node->qstats.data.cbq_stats.xmit_cnt.bytes,
1750: node->qstats_last.data.cbq_stats.xmit_cnt.bytes, interval);
1751: }
1752: break;
1753: case ALTQT_PRIQ:
1754: print_fld_str(FLD_SCHED, "priq");
1755: print_fld_size(FLD_PKTS,
1756: node->qstats.data.priq_stats.xmitcnt.packets);
1757: print_fld_size(FLD_BYTES,
1758: node->qstats.data.priq_stats.xmitcnt.bytes);
1759: print_fld_size(FLD_DROPP,
1760: node->qstats.data.priq_stats.dropcnt.packets);
1761: print_fld_size(FLD_DROPB,
1762: node->qstats.data.priq_stats.dropcnt.bytes);
1763: print_fld_size(FLD_QLEN, node->qstats.data.priq_stats.qlength);
1764: if (interval > 0) {
1765: pps = calc_pps(node->qstats.data.priq_stats.xmitcnt.packets,
1766: node->qstats_last.data.priq_stats.xmitcnt.packets, interval);
1767: bps = calc_rate(node->qstats.data.priq_stats.xmitcnt.bytes,
1768: node->qstats_last.data.priq_stats.xmitcnt.bytes, interval);
1769: }
1770: break;
1771: case ALTQT_HFSC:
1772: print_fld_str(FLD_SCHED, "hfsc");
1773: print_fld_size(FLD_PKTS,
1774: node->qstats.data.hfsc_stats.xmit_cnt.packets);
1775: print_fld_size(FLD_BYTES,
1776: node->qstats.data.hfsc_stats.xmit_cnt.bytes);
1777: print_fld_size(FLD_DROPP,
1778: node->qstats.data.hfsc_stats.drop_cnt.packets);
1779: print_fld_size(FLD_DROPB,
1780: node->qstats.data.hfsc_stats.drop_cnt.bytes);
1781: print_fld_size(FLD_QLEN, node->qstats.data.hfsc_stats.qlength);
1782: if (interval > 0) {
1783: pps = calc_pps(node->qstats.data.hfsc_stats.xmit_cnt.packets,
1784: node->qstats_last.data.hfsc_stats.xmit_cnt.packets, interval);
1785: bps = calc_rate(node->qstats.data.hfsc_stats.xmit_cnt.bytes,
1786: node->qstats_last.data.hfsc_stats.xmit_cnt.bytes, interval);
1787: }
1788: break;
1789: }
1790:
1791: /* if (node->altq.scheduler != ALTQT_HFSC && interval > 0) { */
1792: if (node->altq.scheduler && interval > 0) {
1793: tb_start();
1794: if (pps > 0 && pps < 1)
1795: tbprintf("%-3.1lf", pps);
1796: else
1797: tbprintf("%u", (unsigned int) pps);
1798:
1799: print_fld_tb(FLD_PKTSPS);
1800: print_fld_bw(FLD_BYTESPS, bps);
1801: }
1802: }
1803:
1804: void
1805: print_queues(void)
1806: {
1807: u_int32_t n, count = 0;
1808: struct pf_altq_node *node = altq_root;
1809:
1810: for (n = 0; n < dispstart; n++)
1811: node = node->next_flat;
1812:
1813: for (; n < num_disp; n++) {
1814: print_queue(node);
1815: node = node->next_flat;
1816: end_line();
1817: count ++;
1818: if (maxprint > 0 && count >= maxprint)
1819: break;
1820: }
1821: }
1822:
1823: /* main program functions */
1824:
1825: void
1.7 canacar 1826: update_cache(void)
1.1 canacar 1827: {
1828: static int pstate = -1;
1829: if (pstate == cachestates)
1830: return;
1831:
1832: pstate = cachestates;
1833: if (cachestates) {
1834: show_field(FLD_SI);
1835: show_field(FLD_SP);
1836: gotsig_alarm = 1;
1837: } else {
1838: hide_field(FLD_SI);
1839: hide_field(FLD_SP);
1840: need_update = 1;
1841: }
1842: field_setup();
1843: }
1844:
1.7 canacar 1845: int
1.1 canacar 1846: initpftop(void)
1847: {
1848: struct pf_status status;
1849: field_view *v;
1850: int cachesize = DEFAULT_CACHE_SIZE;
1851:
1852: v = views;
1853: while(v->name != NULL)
1854: add_view(v++);
1855:
1856: pf_dev = open("/dev/pf", O_RDONLY);
1857: if (pf_dev == -1) {
1858: alloc_buf(0);
1859: } else if (ioctl(pf_dev, DIOCGETSTATUS, &status)) {
1860: warn("DIOCGETSTATUS");
1861: alloc_buf(0);
1862: } else
1863: alloc_buf(status.states);
1864:
1865: /* initialize cache with given size */
1866: if (cache_init(cachesize))
1867: warnx("Failed to initialize cache.");
1868: else if (interactive && cachesize > 0)
1869: cachestates = 1;
1870:
1871: update_cache();
1872:
1873: show_field(FLD_STMAX);
1874: show_field(FLD_ANCHOR);
1.7 canacar 1875:
1876: return (1);
1.1 canacar 1877: }