4 * Copyright (c) 2016 Tom Herbert <tom@herbertland.com>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2
8 * as published by the Free Software Foundation.
11 #ifndef __NET_STRPARSER_H_
12 #define __NET_STRPARSER_H_
14 #include <linux/skbuff.h>
17 #define STRP_STATS_ADD(stat, count) ((stat) += (count))
18 #define STRP_STATS_INCR(stat) ((stat)++)
21 unsigned long long msgs
;
22 unsigned long long bytes
;
23 unsigned int mem_fail
;
24 unsigned int need_more_hdr
;
25 unsigned int msg_too_big
;
26 unsigned int msg_timeouts
;
27 unsigned int bad_hdr_len
;
30 struct strp_aggr_stats
{
31 unsigned long long msgs
;
32 unsigned long long bytes
;
33 unsigned int mem_fail
;
34 unsigned int need_more_hdr
;
35 unsigned int msg_too_big
;
36 unsigned int msg_timeouts
;
37 unsigned int bad_hdr_len
;
39 unsigned int interrupted
;
40 unsigned int unrecov_intr
;
45 /* Callbacks are called with lock held for the attached socket */
46 struct strp_callbacks
{
47 int (*parse_msg
)(struct strparser
*strp
, struct sk_buff
*skb
);
48 void (*rcv_msg
)(struct strparser
*strp
, struct sk_buff
*skb
);
49 int (*read_sock_done
)(struct strparser
*strp
, int err
);
50 void (*abort_parser
)(struct strparser
*strp
, int err
);
51 void (*lock
)(struct strparser
*strp
);
52 void (*unlock
)(struct strparser
*strp
);
60 static inline struct strp_msg
*strp_msg(struct sk_buff
*skb
)
62 return (struct strp_msg
*)((void *)skb
->cb
+
63 offsetof(struct qdisc_skb_cb
, data
));
66 /* Structure for an attached lower socket */
76 struct sk_buff
**skb_nextp
;
77 struct timer_list msg_timer
;
78 struct sk_buff
*skb_head
;
79 unsigned int need_bytes
;
80 struct delayed_work delayed_work
;
81 struct work_struct work
;
82 struct strp_stats stats
;
83 struct strp_callbacks cb
;
86 /* Must be called with lock held for attached socket */
87 static inline void strp_pause(struct strparser
*strp
)
92 /* May be called without holding lock for attached socket */
93 void strp_unpause(struct strparser
*strp
);
95 static inline void save_strp_stats(struct strparser
*strp
,
96 struct strp_aggr_stats
*agg_stats
)
98 /* Save psock statistics in the mux when psock is being unattached. */
100 #define SAVE_PSOCK_STATS(_stat) (agg_stats->_stat += \
102 SAVE_PSOCK_STATS(msgs
);
103 SAVE_PSOCK_STATS(bytes
);
104 SAVE_PSOCK_STATS(mem_fail
);
105 SAVE_PSOCK_STATS(need_more_hdr
);
106 SAVE_PSOCK_STATS(msg_too_big
);
107 SAVE_PSOCK_STATS(msg_timeouts
);
108 SAVE_PSOCK_STATS(bad_hdr_len
);
109 #undef SAVE_PSOCK_STATS
113 if (strp
->interrupted
)
114 agg_stats
->interrupted
++;
115 if (strp
->unrecov_intr
)
116 agg_stats
->unrecov_intr
++;
119 static inline void aggregate_strp_stats(struct strp_aggr_stats
*stats
,
120 struct strp_aggr_stats
*agg_stats
)
122 #define SAVE_PSOCK_STATS(_stat) (agg_stats->_stat += stats->_stat)
123 SAVE_PSOCK_STATS(msgs
);
124 SAVE_PSOCK_STATS(bytes
);
125 SAVE_PSOCK_STATS(mem_fail
);
126 SAVE_PSOCK_STATS(need_more_hdr
);
127 SAVE_PSOCK_STATS(msg_too_big
);
128 SAVE_PSOCK_STATS(msg_timeouts
);
129 SAVE_PSOCK_STATS(bad_hdr_len
);
130 SAVE_PSOCK_STATS(aborts
);
131 SAVE_PSOCK_STATS(interrupted
);
132 SAVE_PSOCK_STATS(unrecov_intr
);
133 #undef SAVE_PSOCK_STATS
137 void strp_done(struct strparser
*strp
);
138 void strp_stop(struct strparser
*strp
);
139 void strp_check_rcv(struct strparser
*strp
);
140 int strp_init(struct strparser
*strp
, struct sock
*sk
,
141 const struct strp_callbacks
*cb
);
142 void strp_data_ready(struct strparser
*strp
);
143 int strp_process(struct strparser
*strp
, struct sk_buff
*orig_skb
,
144 unsigned int orig_offset
, size_t orig_len
,
145 size_t max_msg_size
, long timeo
);
147 #endif /* __NET_STRPARSER_H_ */