3 Copyright (C) 2014 Proxmox Server Solutions GmbH
5 This software is written by Proxmox Server Solutions GmbH <support@proxmox.com>
7 This program is free software: you can redistribute it and/or modify
8 it under the terms of the GNU Affero General Public License as published by
9 the Free Software Foundation, either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU Affero General Public License for more details.
17 You should have received a copy of the GNU Affero General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
20 Author: Dietmar Maurer <dietmar@proxmox.com>
32 #include <sys/signalfd.h>
33 #include <sys/types.h>
37 #include <arpa/inet.h>
38 #include <sys/socket.h>
40 #include <linux/netlink.h>
41 #include <libnfnetlink/libnfnetlink.h>
42 #include <libnetfilter_log/libnetfilter_log.h>
43 #include <netinet/ip.h>
44 #include <netinet/ip_icmp.h>
45 #include <netinet/ip6.h>
46 #include <netinet/icmp6.h>
47 #include <netinet/udp.h>
48 #include <netinet/tcp.h>
49 #include <netinet/if_ether.h>
54 static struct nflog_handle
*logh
= NULL
;
55 static struct nlif_handle
*nlifh
= NULL
;
58 gboolean foreground
= FALSE
;
59 gboolean debug
= FALSE
;
65 Special care was taken to allow fast parsing (and filer messages for a singl VM).
67 <VMID> <LOGLEVEL> <CHAIN> <TIME> <TIMEZONE> <MSG>
71 117 6 tap117i0-IN 14/Mar/2014:12:47:07 +0100 policy REJECT: IN=vmbr1 ...
75 #define LOGFILE "/var/log/pve-firewall.log"
77 #define LOCKFILE "/var/lock/pvefw-logger.lck"
78 #define PIDFILE "/var/run/pvefw-logger.pid"
81 #define LE_MAX (512 - 4) // try to fit into 512 bytes
83 #define MAX_CHAIN_LEN 28
86 guint32 len
; // max LE_MAX chars
90 #define STATIC_ASSERT(cond) \
91 extern void pve_static_assert(int test[(cond) ? 1 : -1])
93 STATIC_ASSERT(sizeof(struct log_entry
) == 512);
97 gboolean terminate_threads
= FALSE
;
99 static gboolean
write_pidfile(pid_t pid
)
103 char *strpid
= g_strdup_printf("%d\n", pid
);
104 res
= g_file_set_contents(PIDFILE
, strpid
, strlen(strpid
), NULL
);
110 static GAsyncQueue
*queue
;
113 safe_write(int fd
, char *buf
, size_t count
)
118 n
= write(fd
, buf
, count
);
119 } while (n
< 0 && errno
== EINTR
);
125 log_writer_thread(gpointer data
)
128 struct log_entry
*le
= (struct log_entry
*)g_async_queue_timeout_pop(queue
, 250000);
130 if (terminate_threads
) {
136 if (debug
) fputs(le
->buf
, stdout
);
138 int res
= safe_write(outfd
, le
->buf
, le
->len
);
143 syslog(3, "writing log failed, stopping daemon - %s", strerror (errno
));
144 g_main_loop_quit(main_loop
);
152 static int skipped_logs
= 0;
154 static void log_status_message(guint loglevel
, const char *fmt
, ...);
157 queue_log_entry(struct log_entry
*le
)
159 gint len
= g_async_queue_length(queue
);
161 if (skipped_logs
> 0) {
162 if (len
>= (LQ_LEN
- 1)) {
165 int skip_tmp
= skipped_logs
;
166 skipped_logs
= 0; // clear before calling log_status_message()
167 log_status_message(3, "skipped %d log entries (queue full)", skip_tmp
);
168 g_async_queue_push(queue
, le
);
174 g_async_queue_push(queue
, le
);
180 #define LEPRINTF(format, ...) \
182 if (le->len < LE_MAX) \
183 le->len += snprintf(le->buf + le->len, LE_MAX - le->len, format, ##__VA_ARGS__); \
185 #define LEPRINTTIME(sec) \
187 time_t tmp_sec = sec; \
188 if (le->len < (LE_MAX - 30)) \
189 le->len += strftime(le->buf + le->len, LE_MAX - le->len, "%d/%b/%Y:%H:%M:%S %z ", localtime(&tmp_sec)); \
193 log_status_message(guint loglevel
, const char *fmt
, ...)
198 if (loglevel
> 7 ) loglevel
= 7; // syslog defines level 0-7
200 struct log_entry
*le
= g_new0(struct log_entry
, 1);
202 LEPRINTF("0 %d - ", loglevel
);
204 LEPRINTTIME(time(NULL
));
206 le
->len
+= vsnprintf(le
->buf
+ le
->len
, LE_MAX
- le
->len
, fmt
, ap
);
212 // also log to syslog
214 vsyslog(loglevel
, fmt
, ap
);
218 print_tcp(struct log_entry
*le
, struct tcphdr
*h
, int payload_len
)
220 LEPRINTF("PROTO=TCP ");
222 if (payload_len
< sizeof(struct tcphdr
)) {
223 LEPRINTF("LEN=%d ", payload_len
);
224 LEPRINTF("INVALID=LEN ");
228 LEPRINTF("SPT=%u DPT=%u ", ntohs(h
->source
), ntohs(h
->dest
));
229 LEPRINTF("SEQ=%u ACK=%u ", ntohl(h
->seq
), ntohl(h
->ack_seq
));
230 LEPRINTF("WINDOW=%u ", ntohs(h
->window
));
232 if (h
->urg
) LEPRINTF("URG ");
233 if (h
->ack
) LEPRINTF("ACK ");
234 if (h
->psh
) LEPRINTF("PSH ");
235 if (h
->rst
) LEPRINTF("RST ");
236 if (h
->syn
) LEPRINTF("SYN ");
237 if (h
->fin
) LEPRINTF("FIN ");
239 if (h
->urg
) LEPRINTF("URGP=%u ",ntohs(h
->urg_ptr
));
245 print_udp(struct log_entry
*le
, struct udphdr
*h
, int payload_len
)
247 LEPRINTF("PROTO=UDP ");
249 if (payload_len
< sizeof(struct udphdr
)) {
250 LEPRINTF("LEN=%d ", payload_len
);
251 LEPRINTF("INVALID=LEN ");
255 LEPRINTF("SPT=%u DPT=%u LEN=%u", ntohs(h
->source
), ntohs(h
->dest
), ntohs(h
->len
));
261 print_icmp(struct log_entry
*le
, struct icmphdr
*h
, int payload_len
)
263 char tmp
[INET_ADDRSTRLEN
];
266 LEPRINTF("PROTO=ICMP ");
268 if (payload_len
< sizeof(struct icmphdr
)) {
269 LEPRINTF("LEN=%d ", payload_len
);
270 LEPRINTF("INVALID=LEN ");
274 LEPRINTF("TYPE=%u CODE=%u ", h
->type
, h
->code
);
279 LEPRINTF("ID=%u SEQ=%u ", ntohs(h
->un
.echo
.id
), ntohs(h
->un
.echo
.sequence
));
281 case ICMP_PARAMETERPROB
:
282 LEPRINTF("PARAMETER=%u ", ntohl(h
->un
.gateway
) >> 24);
285 gateway
= ntohl(h
->un
.gateway
);
286 inet_ntop(AF_INET
, &gateway
, tmp
, sizeof(tmp
));
287 LEPRINTF("GATEWAY=%s ", tmp
);
289 case ICMP_DEST_UNREACH
:
290 if (h
->code
== ICMP_FRAG_NEEDED
) {
291 LEPRINTF("MTU=%u ", ntohs(h
->un
.frag
.mtu
));
299 /* Section 3.1. SCTP Common Header Format */
300 typedef struct sctphdr
{
305 } __attribute__((packed
)) sctp_sctphdr_t
;
308 print_sctp(struct log_entry
*le
, struct sctphdr
*h
, int payload_len
)
310 LEPRINTF("PROTO=SCTP ");
312 if (payload_len
< sizeof(struct sctphdr
)) {
313 LEPRINTF("LEN=%d ", payload_len
);
314 LEPRINTF("INVALID=LEN ");
318 LEPRINTF("SPT=%u DPT=%u ", ntohs(h
->source
), ntohs(h
->dest
));
324 print_ipproto(struct log_entry
*le
, char * nexthdr
, int payload_len
, u_int8_t proto
)
328 print_tcp(le
, (struct tcphdr
*)nexthdr
, payload_len
);
331 print_udp(le
, (struct udphdr
*)nexthdr
, payload_len
);
334 print_icmp(le
, (struct icmphdr
*)nexthdr
, payload_len
);
337 print_sctp(le
, (struct sctphdr
*)nexthdr
, payload_len
);
340 LEPRINTF("PROTO=AH ");
343 LEPRINTF("PROTO=ESP ");
346 LEPRINTF("PROTO=IGMP ");
355 print_iphdr(struct log_entry
*le
, char * payload
, int payload_len
)
357 if (payload_len
< sizeof(struct iphdr
)) {
358 LEPRINTF("LEN=%d ", payload_len
);
359 LEPRINTF("INVALID=LEN ");
363 struct iphdr
*h
= (struct iphdr
*)payload
;
365 if (payload_len
<= (u_int32_t
)(h
->ihl
* 4)) {
366 LEPRINTF("INVALID=IHL ");
370 char tmp
[INET_ADDRSTRLEN
];
372 inet_ntop(AF_INET
, &h
->saddr
, tmp
, sizeof(tmp
));
373 LEPRINTF("SRC=%s ", tmp
);
374 inet_ntop(AF_INET
, &h
->daddr
, tmp
, sizeof(tmp
));
375 LEPRINTF("DST=%s ", tmp
);
377 LEPRINTF("LEN=%u TOS=0x%02X PREC=0x%02X TTL=%u ID=%u ",
378 ntohs(h
->tot_len
), h
->tos
& IPTOS_TOS_MASK
,
379 h
->tos
& IPTOS_PREC_MASK
, h
->ttl
, ntohs(h
->id
));
381 short ip_off
= ntohs(h
->frag_off
);
382 if (ip_off
& IP_OFFMASK
)
383 LEPRINTF("FRAG=%u ", ip_off
& IP_OFFMASK
);
385 if (ip_off
& IP_DF
) LEPRINTF("DF ");
386 if (ip_off
& IP_MF
) LEPRINTF("MF ");
388 void *nexthdr
= (u_int32_t
*)h
+ h
->ihl
;
389 payload_len
-= h
->ihl
* 4;
391 if (print_ipproto(le
, nexthdr
, payload_len
, h
->protocol
) < 0) {
392 LEPRINTF("PROTO=%u ", h
->protocol
);
399 print_routing(struct log_entry
*le
, struct ip6_rthdr
*rthdr
, int payload_len
)
401 char tmp
[INET6_ADDRSTRLEN
];
402 LEPRINTF("TYPE=%u SEGMENTS=%u", rthdr
->ip6r_type
, rthdr
->ip6r_segleft
);
404 if (payload_len
< sizeof(*rthdr
) || payload_len
< rthdr
->ip6r_len
*8) {
405 LEPRINTF("LEN=%d ", payload_len
);
406 LEPRINTF("INVALID=LEN ");
410 if (rthdr
->ip6r_type
== 0) {
411 /* Route via waypoints (deprecated), this contains a list of waypoints
412 * to visit. (RFC2460 (4.4))
415 struct ip6_rthdr0
*h
= (struct ip6_rthdr0
*)rthdr
;
416 if (rthdr
->ip6r_len
*8 < sizeof(*h
) + rthdr
->ip6r_segleft
* sizeof(struct in6_addr
)) {
417 LEPRINTF("INVALID=SEGMENTS ");
421 } else if (rthdr
->ip6r_type
== 1) {
422 /* nimrod routing (RFC1992) */
424 } else if (rthdr
->ip6r_type
== 2) {
425 /* RFC3375 (6.4), the layout is like type-0 but with exactly 1 address */
426 struct ip6_rthdr0
*h
= (struct ip6_rthdr0
*)rthdr
;
427 if (rthdr
->ip6r_len
*8 < sizeof(*h
) + sizeof(struct in6_addr
)) {
428 LEPRINTF("LEN=%d ", payload_len
);
429 LEPRINTF("INVALID=LEN ");
432 inet_ntop(AF_INET6
, &h
->ip6r0_addr
[0], tmp
, sizeof(tmp
));
433 LEPRINTF("HOME=%s ", tmp
);
441 print_fragment(struct log_entry
*le
, struct ip6_frag
*frag
, int payload_len
)
445 if (payload_len
< sizeof(*frag
)) {
446 LEPRINTF("LEN=%d ", payload_len
);
447 LEPRINTF("INVALID=LEN ");
451 offlg
= ntohs(frag
->ip6f_offlg
);
452 LEPRINTF("FRAG=%d ID=%d ", (offlg
&0x2FFF), ntohl(frag
->ip6f_ident
));
460 print_icmp6(struct log_entry
*le
, struct icmp6_hdr
*h
, int payload_len
)
462 struct nd_router_advert
*ra
;
463 struct nd_neighbor_advert
*na
;
464 struct nd_redirect
*re
;
465 char tmp
[INET6_ADDRSTRLEN
];
467 if (payload_len
< sizeof(struct icmp6_hdr
)) {
468 LEPRINTF("LEN=%d ", payload_len
);
469 LEPRINTF("INVALID=LEN ");
473 LEPRINTF("TYPE=%u CODE=%u ", h
->icmp6_type
, h
->icmp6_code
);
475 switch (h
->icmp6_type
) {
476 case ICMP6_ECHO_REQUEST
:
477 case ICMP6_ECHO_REPLY
:
478 LEPRINTF("ID=%u SEQ=%u ", ntohs(h
->icmp6_id
), ntohs(h
->icmp6_seq
));
481 case ND_ROUTER_SOLICIT
:
482 /* can be followed by options, otherwise nothing to print */
485 case ND_ROUTER_ADVERT
:
486 ra
= (struct nd_router_advert
*)h
;
487 LEPRINTF("HOPLIMIT=%d ", ra
->nd_ra_curhoplimit
);
488 /* nd_ra_flags_reserved is only 8 bit, so no swapping here as
489 * opposed to the neighbor advertisement flags (see below).
491 LEPRINTF("RA=%02x LIFETIME=%d REACHABLE=%d RETRANSMIT=%d ",
492 ra
->nd_ra_flags_reserved
,
493 ntohs(ra
->nd_ra_router_lifetime
),
494 ntohl(ra
->nd_ra_reachable
),
495 ntohl(ra
->nd_ra_retransmit
));
496 /* can be followed by options */
499 case ND_NEIGHBOR_SOLICIT
:
500 /* can be followed by options */
503 case ND_NEIGHBOR_ADVERT
:
504 na
= (struct nd_neighbor_advert
*)h
;
505 LEPRINTF("NA=%08x ", ntohl(na
->nd_na_flags_reserved
));
506 /* can be followed by options */
510 re
= (struct nd_redirect
*)h
;
511 inet_ntop(AF_INET6
, &re
->nd_rd_target
, tmp
, sizeof(tmp
));
512 LEPRINTF("TARGET=%s ", tmp
);
513 inet_ntop(AF_INET6
, &re
->nd_rd_dst
, tmp
, sizeof(tmp
));
514 LEPRINTF("GATEWAY=%s ", tmp
);
515 /* can be followed by options */
518 case ICMP6_DST_UNREACH
:
519 /* CODE shows the type, no extra parameters available in ipv6 */
522 case ICMP6_PACKET_TOO_BIG
:
523 LEPRINTF("MTU=%u ", ntohl(h
->icmp6_mtu
));
526 case ICMP6_TIME_EXCEEDED
:
527 /* CODE shows the type (0 = hop limit, 1 = reassembly timed out) */
530 case ICMP6_PARAM_PROB
:
531 switch (ntohl(h
->icmp6_pptr
)) {
532 case ICMP6_PARAMPROB_HEADER
:
533 LEPRINTF("PARAMETER=HEADER "); /* erroneous header */
535 case ICMP6_PARAMPROB_NEXTHEADER
:
536 LEPRINTF("PARAMETER=NEXTHEADER "); /* bad next-header field */
538 case ICMP6_PARAMPROB_OPTION
:
539 LEPRINTF("PARAMETER=OPTION "); /* bad ipv6 option (hop/dst header?) */
542 LEPRINTF("PARAMETER=%u ", ntohl(h
->icmp6_pptr
)); /* unknown */
552 check_ip6ext(struct log_entry
*le
, struct ip6_ext
*exthdr
, int payload_len
)
554 if (payload_len
< sizeof(*exthdr
) ||
555 payload_len
< exthdr
->ip6e_len
)
557 LEPRINTF("LEN=%d ", payload_len
);
558 LEPRINTF("INVALID=LEN ");
565 print_nexthdr(struct log_entry
*le
, char *hdr
, int payload_len
, u_int8_t proto
)
568 if (print_ipproto(le
, hdr
, payload_len
, proto
) == 0)
571 struct ip6_ext
*exthdr
= (struct ip6_ext
*)hdr
;
574 /* protocols (these return) */
576 LEPRINTF("PROTO=ICMPV6 ");
577 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
579 if (print_icmp6(le
, (struct icmp6_hdr
*)(hdr
+ exthdr
->ip6e_len
),
580 payload_len
- exthdr
->ip6e_len
) < 0)
586 /* extension headers (these break to keep iterating) */
587 case IPPROTO_ROUTING
:
588 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
590 if (print_routing(le
, (struct ip6_rthdr
*)hdr
, payload_len
) < 0)
593 case IPPROTO_FRAGMENT
:
594 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
596 if (print_fragment(le
, (struct ip6_frag
*)hdr
, payload_len
) < 0)
599 case IPPROTO_HOPOPTS
:
600 LEPRINTF("NEXTHDR=HOPOPTS ");
601 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
603 /* do we want to print these? */
605 case IPPROTO_DSTOPTS
:
606 LEPRINTF("NEXTHDR=DSTOPTS ");
607 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
609 /* do we want to print these? */
612 LEPRINTF("NEXTHDR=MH ");
613 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
617 /* unknown protocol */
619 LEPRINTF("PROTO=%u ", proto
);
623 if (check_ip6ext(le
, exthdr
, payload_len
) < 0)
625 hdr
+= exthdr
->ip6e_len
;
626 payload_len
-= exthdr
->ip6e_len
;
631 print_ip6hdr(struct log_entry
*le
, char * payload
, int payload_len
)
633 if (payload_len
< sizeof(struct ip6_hdr
)) {
634 LEPRINTF("LEN=%d ", payload_len
);
635 LEPRINTF("INVALID=LEN ");
639 struct ip6_hdr
*h
= (struct ip6_hdr
*)payload
;
641 char tmp
[INET6_ADDRSTRLEN
];
642 inet_ntop(AF_INET6
, &h
->ip6_src
, tmp
, sizeof(tmp
));
643 LEPRINTF("SRC=%s ", tmp
);
644 inet_ntop(AF_INET6
, &h
->ip6_dst
, tmp
, sizeof(tmp
));
645 LEPRINTF("DST=%s ", tmp
);
647 LEPRINTF("LEN=%u ", ntohs(h
->ip6_plen
));
649 u_int32_t flow
= ntohl(h
->ip6_flow
);
650 LEPRINTF("TC=%d FLOWLBL=%d ", (flow
>>20)&0xFF, flow
&0xFFFFF);
652 LEPRINTF("HOPLIMIT=%d ", h
->ip6_hlim
);
654 return print_nexthdr(le
, (char *)(h
+1), payload_len
- sizeof(*h
), h
->ip6_nxt
);
657 // ebtables -I FORWARD --nflog --nflog-group 0
659 print_arp(struct log_entry
*le
, struct ether_arp
*h
, int payload_len
)
661 if (payload_len
< sizeof(struct ether_arp
)) {
662 LEPRINTF("LEN=%d ", payload_len
);
663 LEPRINTF("INVALID=LEN ");
667 LEPRINTF("SRC=%u.%u.%u.%u ", h
->arp_spa
[0], h
->arp_spa
[1],
668 h
->arp_spa
[2], h
->arp_spa
[3]);
670 LEPRINTF("DST=%u.%u.%u.%u ", h
->arp_tpa
[0], h
->arp_tpa
[1],
671 h
->arp_tpa
[2], h
->arp_tpa
[3]);
673 LEPRINTF("PROTO=ARP ");
675 unsigned short code
= ntohs(h
->arp_op
);
678 LEPRINTF("REQUEST ");
681 LEPRINTF("REPLY MAC=%02x:%02x:%02x:%02x:%02x:%02x ",
682 h
->arp_sha
[0], h
->arp_sha
[1], h
->arp_sha
[2],
683 h
->arp_sha
[3], h
->arp_sha
[4], h
->arp_sha
[5]);
689 LEPRINTF("CODE=%u ", code
);
693 // LEPRINTF("HTYPE=%u ", ntohs(h->arp_hrd));
695 // LEPRINTF("PTYPE=%u ", ntohs(h->arp_pro));
701 static int print_pkt(struct log_entry
*le
, struct nflog_data
*ldata
, u_int8_t family
)
703 u_int32_t mark
= nflog_get_nfmark(ldata
);
704 u_int32_t indev
= nflog_get_indev(ldata
);
705 u_int32_t outdev
= nflog_get_outdev(ldata
);
706 u_int32_t physindev
= nflog_get_physindev(ldata
);
707 u_int32_t physoutdev
= nflog_get_physoutdev(ldata
);
709 char *prefix
= nflog_get_prefix(ldata
);
715 guint8 log_level
= 6; // info
717 char *chain_name
= "-";
719 if (prefix
!= NULL
) {
720 // Note: parse ":$vmid:$loglevel:$chain: $msg"
721 if (prefix
[0] == ':') {
722 char *p
= prefix
+ 1;
724 while(*p
>= '0' && *p
<= '9') { tmpid
*= 10; tmpid
+= *p
- '0'; p
++; }
727 (p
[1] >= '0' && p
[1] <= '7') &&
730 guint8 tmp_level
= p
[1] - '0'; // store for later use
731 char *chain_start
= p
+ 3; // store for later use
733 while (*p
&& *p
!= ':' && *p
!= ' ') p
++;
734 int len
= p
- chain_start
;
736 if (*p
== ':' && p
[1] == ' ' && len
&& (len
<= MAX_CHAIN_LEN
)) {
737 // parsing successful
739 *p
= 0; // terminate string
742 log_level
= tmp_level
;
743 chain_name
= chain_start
;
744 prefix
= p
+ 2; // the rest
750 LEPRINTF("%d ", vmid
);
752 LEPRINTF("%d ", log_level
);
754 LEPRINTF("%s ", chain_name
);
757 nflog_get_timestamp(ldata
, &ts
);
759 LEPRINTTIME(ts
.tv_sec
);
761 if (prefix
!= NULL
) {
762 LEPRINTF("%s", prefix
);
766 if (nlif_index2name(nlifh
, indev
, devname
) != -1) {
767 LEPRINTF("IN=%s ", devname
);
769 LEPRINTF("IN=%u ", indev
);
774 if (nlif_index2name(nlifh
, outdev
, devname
) != -1) {
775 LEPRINTF("OUT=%s ", devname
);
777 LEPRINTF("OUT=%u ", outdev
);
782 if (nlif_index2name(nlifh
, physindev
, devname
) != -1) {
783 LEPRINTF("PHYSIN=%s ", devname
);
785 LEPRINTF("PHYSIN=%u ", physindev
);
789 if (physoutdev
> 0) {
790 if (nlif_index2name(nlifh
, physoutdev
, devname
) != -1) {
791 LEPRINTF("PHYSOUT=%s ", devname
);
793 LEPRINTF("PHYSOUT=%u ", physoutdev
);
797 int payload_len
= nflog_get_payload(ldata
, &payload
);
799 int hwhdrlen
= nflog_get_msg_packet_hwhdrlen(ldata
);
801 unsigned char *hwhdr
= (unsigned char *)nflog_get_msg_packet_hwhdr(ldata
);
805 for (i
= 0; i
< hwhdrlen
; i
++) {
806 LEPRINTF("%02x", hwhdr
[i
]);
807 if (i
< (hwhdrlen
-1 )) LEPRINTF(":");
813 u_int16_t hw_protocol
= 0;
814 struct nfulnl_msg_packet_hdr
*ph
= NULL
;
818 print_iphdr(le
, payload
, payload_len
);
821 print_ip6hdr(le
, payload
, payload_len
);
824 ph
= nflog_get_msg_packet_hdr(ldata
);
825 if (ph
) hw_protocol
= ntohs(ph
->hw_protocol
);
827 switch (hw_protocol
) {
829 print_iphdr(le
, payload
, payload_len
);
832 print_ip6hdr(le
, payload
, payload_len
);
835 print_arp(le
, (struct ether_arp
*)payload
, payload_len
);
841 if (mark
) LEPRINTF("mark=%u ", mark
);
849 nflog_cb(struct nflog_g_handle
*gh
, struct nfgenmsg
*nfmsg
,
850 struct nflog_data
*nfa
, void *data
)
852 struct log_entry
*le
= g_new0(struct log_entry
, 1);
854 print_pkt(le
, nfa
, nfmsg
->nfgen_family
);
856 LEPRINTF("\n"); // add newline
864 nflog_read_cb(GIOChannel
*source
,
865 GIOCondition condition
,
871 int fd
= g_io_channel_unix_get_fd(source
);
873 if ((rv
= recv(fd
, buf
, sizeof(buf
), 0)) && rv
>= 0) {
874 nflog_handle_packet(logh
, buf
, rv
);
881 nlif_read_cb(GIOChannel
*source
,
882 GIOCondition condition
,
885 static int last_res
= 0;
888 if ((res
= nlif_catch(nlifh
)) < 0) {
889 if (last_res
== 0) { // only report once
890 log_status_message(3, "nlif_catch failed (res = %d)", res
);
901 signal_read_cb(GIOChannel
*source
,
902 GIOCondition condition
,
906 struct signalfd_siginfo si
;
908 int fd
= g_io_channel_unix_get_fd(source
);
910 if ((rv
= read(fd
, &si
, sizeof(si
))) && rv
>= 0) {
911 terminate_threads
= TRUE
;
912 log_status_message(5, "received terminate request (signal)");
913 g_main_loop_quit(main_loop
);
920 main(int argc
, char *argv
[])
925 gboolean wrote_pidfile
= FALSE
;
927 openlog("pvepw-logger", LOG_CONS
|LOG_PID
, LOG_DAEMON
);
929 GOptionContext
*context
;
931 GOptionEntry entries
[] = {
932 { "debug", 'd', 0, G_OPTION_ARG_NONE
, &debug
, "Turn on debug messages", NULL
},
933 { "foreground", 'f', 0, G_OPTION_ARG_NONE
, &foreground
, "Do not daemonize server", NULL
},
937 context
= g_option_context_new("");
938 g_option_context_add_main_entries (context
, entries
, NULL
);
941 if (!g_option_context_parse (context
, &argc
, &argv
, &err
)) {
942 fprintf(stderr
, "error: %s\n", err
->message
);
943 fprintf(stderr
, "%s", g_option_context_get_help(context
, FALSE
, NULL
));
949 fprintf(stderr
, "error: too many arguments\n");
950 fprintf(stderr
, "%s", g_option_context_get_help(context
, FALSE
, NULL
));
954 g_option_context_free(context
);
956 if (debug
) foreground
= TRUE
;
958 if ((lockfd
= open(LOCKFILE
, O_RDWR
|O_CREAT
|O_APPEND
, 0644)) == -1) {
959 fprintf(stderr
, "unable to create lock '%s': %s\n", LOCKFILE
, strerror (errno
) );
963 for (int i
= 10; i
>= 0; i
--) {
964 if (flock(lockfd
, LOCK_EX
|LOCK_NB
) != 0) {
966 fprintf(stderr
, "unable to aquire lock '%s': %s\n", LOCKFILE
, strerror (errno
));
970 fprintf(stderr
, "unable to aquire lock '%s' - trying again.\n", LOCKFILE
);
976 if ((outfd
= open(LOGFILE
, O_WRONLY
|O_CREAT
|O_APPEND
, 0644)) == -1) {
977 fprintf(stderr
, "unable to open file '%s': %s\n", LOGFILE
, strerror (errno
));
981 if ((logh
= nflog_open()) == NULL
) {
982 fprintf(stderr
, "unable to open nflog\n");
986 if (!nflog_bind_pf(logh
, AF_INET
) <= 0) {
987 fprintf(stderr
, "nflog_bind_pf AF_INET failed\n");
992 if (!nflog_bind_pf(logh
, AF_INET6
) <= 0) {
993 fprintf(stderr
, "nflog_bind_pf AF_INET6 failed\n");
998 if (!nflog_bind_pf(logh
, AF_BRIDGE
) <= 0) {
999 fprintf(stderr
, "nflog_bind_pf AF_BRIDGE failed\n");
1003 struct nflog_g_handle
*qh
= nflog_bind_group(logh
, 0);
1005 fprintf(stderr
, "no nflog handle for group 0\n");
1009 if (nflog_set_mode(qh
, NFULNL_COPY_PACKET
, 0xffff) < 0) {
1010 fprintf(stderr
, "can't set packet copy mode\n");
1014 if ((nlifh
= nlif_open()) == NULL
) {
1015 fprintf(stderr
, "unable to open netlink interface handle\n");
1021 sigaddset(&mask
, SIGINT
);
1022 sigaddset(&mask
, SIGTERM
);
1024 sigprocmask(SIG_BLOCK
, &mask
, NULL
);
1026 if ((sigfd
= signalfd(-1, &mask
, SFD_NONBLOCK
)) < 0) {
1027 fprintf(stderr
, "unable to open signalfd: %s\n", strerror (errno
));
1032 pid_t cpid
= fork();
1035 fprintf(stderr
, "failed to daemonize program - %s\n", strerror (errno
));
1038 write_pidfile(cpid
);
1043 if (chroot("/") != 0) fprintf(stderr
, "chroot '/' failed - %s\n", strerror (errno
));
1045 if ((nullfd
= open("/dev/null", O_RDWR
, 0)) != -1) {
1056 write_pidfile(getpid());
1059 wrote_pidfile
= TRUE
;
1061 nflog_callback_register(qh
, &nflog_cb
, logh
);
1063 queue
= g_async_queue_new_full(g_free
);
1065 log_status_message(5, "starting pvefw logger");
1069 GIOChannel
*nlif_ch
= g_io_channel_unix_new(nlif_fd(nlifh
));
1071 g_io_add_watch(nlif_ch
, G_IO_IN
, nlif_read_cb
, NULL
);
1073 int logfd
= nflog_fd(logh
);
1074 GIOChannel
*nflog_ch
= g_io_channel_unix_new(logfd
);
1076 g_io_add_watch(nflog_ch
, G_IO_IN
, nflog_read_cb
, NULL
);
1078 GIOChannel
*sig_ch
= g_io_channel_unix_new(sigfd
);
1079 if (!g_io_add_watch(sig_ch
, G_IO_IN
, signal_read_cb
, NULL
)) {
1083 GThread
*wthread
= g_thread_new("log_writer_thread", log_writer_thread
, NULL
);
1085 main_loop
= g_main_loop_new(NULL
, TRUE
);
1087 g_main_loop_run(main_loop
);
1089 log_status_message(5, "stopping pvefw logger");
1091 g_thread_join(wthread
);